]> git.proxmox.com Git - mirror_edk2.git/blob - MdeModulePkg/Universal/PCD/Dxe/Service.c
Update string offset and default offset from UINT16 to STRING_HEAD to match the gener...
[mirror_edk2.git] / MdeModulePkg / Universal / PCD / Dxe / Service.c
1 /** @file
2 Help functions used by PCD DXE driver.
3
4 Copyright (c) 2006 - 2012, Intel Corporation. All rights reserved.<BR>
5 This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
9
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
12
13 **/
14
15 #include "Service.h"
16
17 PCD_DATABASE *mPcdDatabase;
18
19 LIST_ENTRY *mCallbackFnTable;
20
21 /**
22 Get the PCD entry pointer in PCD database.
23
24 This routine will visit PCD database to find the PCD entry according to given
25 token number. The given token number is autogened by build tools and it will be
26 translated to local token number. Local token number contains PCD's type and
27 offset of PCD entry in PCD database.
28
29 @param TokenNumber Token's number, it is autogened by build tools
30 @param GetSize The size of token's value
31
32 @return PCD entry pointer in PCD database
33
34 **/
35 VOID *
36 GetWorker (
37 IN UINTN TokenNumber,
38 IN UINTN GetSize
39 )
40 {
41 UINT32 *LocalTokenNumberTable;
42 EFI_GUID *GuidTable;
43 UINT8 *StringTable;
44 EFI_GUID *Guid;
45 UINT16 *Name;
46 VARIABLE_HEAD *VariableHead;
47 UINT8 *VaraiableDefaultBuffer;
48 UINT8 *Data;
49 VPD_HEAD *VpdHead;
50 UINT8 *PcdDb;
51 VOID *RetPtr;
52 UINTN MaxSize;
53 UINTN TmpTokenNumber;
54 UINTN DataSize;
55 EFI_STATUS Status;
56 UINT32 LocalTokenNumber;
57 UINT32 Offset;
58 STRING_HEAD StringTableIdx;
59 BOOLEAN IsPeiDb;
60
61 //
62 // Aquire lock to prevent reentrance from TPL_CALLBACK level
63 //
64 EfiAcquireLock (&mPcdDatabaseLock);
65
66 RetPtr = NULL;
67 //
68 // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.
69 // We have to decrement TokenNumber by 1 to make it usable
70 // as the array index.
71 //
72 TokenNumber--;
73
74 TmpTokenNumber = TokenNumber;
75
76 //
77 // PCD_TOTAL_TOKEN_NUMBER is a auto-generated constant.
78 // It could be zero. EBC compiler is very choosy. It may
79 // report warning. So we add 1 in each size of the
80 // comparison.
81 //
82 ASSERT (TokenNumber + 1 < PCD_TOTAL_TOKEN_NUMBER + 1);
83
84 ASSERT ((GetSize == DxePcdGetSize (TokenNumber + 1)) || (GetSize == 0));
85
86 // EBC compiler is very choosy. It may report warning about comparison
87 // between UINTN and 0 . So we add 1 in each size of the
88 // comparison.
89 IsPeiDb = (BOOLEAN) ((TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1) ? TRUE : FALSE);
90
91 LocalTokenNumberTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable :
92 mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
93
94 TokenNumber = IsPeiDb ? TokenNumber :
95 TokenNumber - PEI_LOCAL_TOKEN_NUMBER;
96
97 LocalTokenNumber = LocalTokenNumberTable[TokenNumber];
98
99 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {
100 if (GetSize == 0) {
101 GetPtrTypeSize (TmpTokenNumber, &MaxSize);
102 } else {
103 MaxSize = GetSize;
104 }
105 LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize, IsPeiDb);
106 }
107
108 PcdDb = IsPeiDb ? ((UINT8 *) &mPcdDatabase->PeiDb) : ((UINT8 *) &mPcdDatabase->DxeDb);
109
110 if (IsPeiDb) {
111 StringTable = (UINT8 *) (&mPcdDatabase->PeiDb.Init.StringTable[0]);
112 } else {
113 StringTable = (UINT8 *) (&mPcdDatabase->DxeDb.Init.StringTable[0]);
114 }
115
116
117 Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;
118
119 switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {
120 case PCD_TYPE_VPD:
121 VpdHead = (VPD_HEAD *) ((UINT8 *) PcdDb + Offset);
122 RetPtr = (VOID *) (UINTN) (PcdGet32 (PcdVpdBaseAddress) + VpdHead->Offset);
123 break;
124
125 case PCD_TYPE_HII|PCD_TYPE_STRING:
126 case PCD_TYPE_HII:
127 if (IsPeiDb) {
128 GuidTable = (EFI_GUID *) (&mPcdDatabase->PeiDb.Init.GuidTable[0]);
129 } else {
130 GuidTable = (EFI_GUID *) (&mPcdDatabase->DxeDb.Init.GuidTable[0]);
131 }
132
133 VariableHead = (VARIABLE_HEAD *) (PcdDb + Offset);
134 Guid = GuidTable + VariableHead->GuidTableIndex;
135 Name = (UINT16*)(StringTable + VariableHead->StringIndex);
136
137 if ((LocalTokenNumber & PCD_TYPE_ALL_SET) == (PCD_TYPE_HII|PCD_TYPE_STRING)) {
138 //
139 // If a HII type PCD's datum type is VOID*, the DefaultValueOffset is the index of
140 // string array in string table.
141 //
142 StringTableIdx = *(STRING_HEAD*)((UINT8 *) PcdDb + VariableHead->DefaultValueOffset);
143 VaraiableDefaultBuffer = (VOID *) (StringTable + StringTableIdx);
144 Status = GetHiiVariable (Guid, Name, &Data, &DataSize);
145 if (Status == EFI_SUCCESS) {
146 if (GetSize == 0) {
147 //
148 // It is a pointer type. So get the MaxSize reserved for
149 // this PCD entry.
150 //
151 GetPtrTypeSize (TmpTokenNumber, &GetSize);
152 }
153 CopyMem (VaraiableDefaultBuffer, Data + VariableHead->Offset, GetSize);
154 FreePool (Data);
155 }
156 RetPtr = (VOID *) VaraiableDefaultBuffer;
157 } else {
158 VaraiableDefaultBuffer = (UINT8 *) PcdDb + VariableHead->DefaultValueOffset;
159
160 Status = GetHiiVariable (Guid, Name, &Data, &DataSize);
161 if (Status == EFI_SUCCESS) {
162 if (GetSize == 0) {
163 //
164 // It is a pointer type. So get the MaxSize reserved for
165 // this PCD entry.
166 //
167 GetPtrTypeSize (TmpTokenNumber, &GetSize);
168 }
169 CopyMem (VaraiableDefaultBuffer, Data + VariableHead->Offset, GetSize);
170 FreePool (Data);
171 }
172 //
173 // If the operation is successful, we copy the data
174 // to the default value buffer in the PCD Database.
175 // So that we can free the Data allocated in GetHiiVariable.
176 //
177 //
178 // If the operation is not successful,
179 // Return 1) either the default value specified by Platform Integrator
180 // 2) Or the value Set by a PCD set operation.
181 //
182 RetPtr = (VOID *) VaraiableDefaultBuffer;
183 }
184 break;
185
186 case PCD_TYPE_STRING:
187 StringTableIdx = *(STRING_HEAD*)((UINT8 *) PcdDb + Offset);
188 RetPtr = (VOID *) (StringTable + StringTableIdx);
189 break;
190
191 case PCD_TYPE_DATA:
192 RetPtr = (VOID *) ((UINT8 *) PcdDb + Offset);
193 break;
194
195 default:
196 ASSERT (FALSE);
197 break;
198
199 }
200
201 EfiReleaseLock (&mPcdDatabaseLock);
202
203 return RetPtr;
204
205 }
206
207 /**
208 Register the callback function for a PCD entry.
209
210 This routine will register a callback function to a PCD entry by given token number
211 and token space guid.
212
213 @param TokenNumber PCD token's number, it is autogened by build tools.
214 @param Guid PCD token space's guid,
215 if not NULL, this PCD is dynamicEx type PCD.
216 @param CallBackFunction Callback function pointer
217
218 @return EFI_SUCCESS Always success for registering callback function.
219
220 **/
221 EFI_STATUS
222 DxeRegisterCallBackWorker (
223 IN UINTN TokenNumber,
224 IN CONST EFI_GUID *Guid, OPTIONAL
225 IN PCD_PROTOCOL_CALLBACK CallBackFunction
226 )
227 {
228 CALLBACK_FN_ENTRY *FnTableEntry;
229 LIST_ENTRY *ListHead;
230 LIST_ENTRY *ListNode;
231
232 if (Guid != NULL) {
233 TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) TokenNumber);
234 }
235
236 //
237 // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.
238 // We have to decrement TokenNumber by 1 to make it usable
239 // as the array index of mCallbackFnTable[].
240 //
241 ListHead = &mCallbackFnTable[TokenNumber - 1];
242 ListNode = GetFirstNode (ListHead);
243
244 while (ListNode != ListHead) {
245 FnTableEntry = CR_FNENTRY_FROM_LISTNODE(ListNode, CALLBACK_FN_ENTRY, Node);
246
247 if (FnTableEntry->CallbackFn == CallBackFunction) {
248 //
249 // We only allow a Callback function to be register once
250 // for a TokenNumber. So just return EFI_SUCCESS
251 //
252 return EFI_SUCCESS;
253 }
254 ListNode = GetNextNode (ListHead, ListNode);
255 }
256
257 FnTableEntry = AllocatePool (sizeof(CALLBACK_FN_ENTRY));
258 ASSERT (FnTableEntry != NULL);
259
260 FnTableEntry->CallbackFn = CallBackFunction;
261 InsertTailList (ListHead, &FnTableEntry->Node);
262
263 return EFI_SUCCESS;
264 }
265
266 /**
267 UnRegister the callback function for a PCD entry.
268
269 This routine will unregister a callback function to a PCD entry by given token number
270 and token space guid.
271
272 @param TokenNumber PCD token's number, it is autogened by build tools.
273 @param Guid PCD token space's guid.
274 if not NULL, this PCD is dynamicEx type PCD.
275 @param CallBackFunction Callback function pointer
276
277 @retval EFI_SUCCESS Callback function is success to be unregister.
278 @retval EFI_INVALID_PARAMETER Can not find the PCD entry by given token number.
279 **/
280 EFI_STATUS
281 DxeUnRegisterCallBackWorker (
282 IN UINTN TokenNumber,
283 IN CONST EFI_GUID *Guid, OPTIONAL
284 IN PCD_PROTOCOL_CALLBACK CallBackFunction
285 )
286 {
287 CALLBACK_FN_ENTRY *FnTableEntry;
288 LIST_ENTRY *ListHead;
289 LIST_ENTRY *ListNode;
290
291 if (Guid != NULL) {
292 TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) TokenNumber);
293 }
294
295 //
296 // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.
297 // We have to decrement TokenNumber by 1 to make it usable
298 // as the array index of mCallbackFnTable[].
299 //
300 ListHead = &mCallbackFnTable[TokenNumber - 1];
301 ListNode = GetFirstNode (ListHead);
302
303 while (ListNode != ListHead) {
304 FnTableEntry = CR_FNENTRY_FROM_LISTNODE(ListNode, CALLBACK_FN_ENTRY, Node);
305
306 if (FnTableEntry->CallbackFn == CallBackFunction) {
307 //
308 // We only allow a Callback function to be register once
309 // for a TokenNumber. So we can safely remove the Node from
310 // the Link List and return EFI_SUCCESS.
311 //
312 RemoveEntryList (ListNode);
313 FreePool (FnTableEntry);
314
315 return EFI_SUCCESS;
316 }
317 ListNode = GetNextNode (ListHead, ListNode);
318 }
319
320 return EFI_INVALID_PARAMETER;
321 }
322
323 /**
324 Get next token number in given token space.
325
326 This routine is used for dynamicEx type PCD. It will firstly scan token space
327 table to get token space according to given token space guid. Then scan given
328 token number in found token space, if found, then return next token number in
329 this token space.
330
331 @param Guid Token space guid. Next token number will be scaned in
332 this token space.
333 @param TokenNumber Token number.
334 If PCD_INVALID_TOKEN_NUMBER, return first token number in
335 token space table.
336 If not PCD_INVALID_TOKEN_NUMBER, return next token number
337 in token space table.
338 @param GuidTable Token space guid table. It will be used for scan token space
339 by given token space guid.
340 @param SizeOfGuidTable The size of guid table.
341 @param ExMapTable DynamicEx token number mapping table.
342 @param SizeOfExMapTable The size of dynamicEx token number mapping table.
343
344 @retval EFI_NOT_FOUND Can not given token space or token number.
345 @retval EFI_SUCCESS Success to get next token number.
346
347 **/
348 EFI_STATUS
349 ExGetNextTokeNumber (
350 IN CONST EFI_GUID *Guid,
351 IN OUT UINTN *TokenNumber,
352 IN EFI_GUID *GuidTable,
353 IN UINTN SizeOfGuidTable,
354 IN DYNAMICEX_MAPPING *ExMapTable,
355 IN UINTN SizeOfExMapTable
356 )
357 {
358 EFI_GUID *MatchGuid;
359 UINTN Index;
360 UINTN GuidTableIdx;
361 BOOLEAN Found;
362
363 //
364 // Scan token space guid
365 //
366 MatchGuid = ScanGuid (GuidTable, SizeOfGuidTable, Guid);
367 if (MatchGuid == NULL) {
368 return EFI_NOT_FOUND;
369 }
370
371 //
372 // Find the token space table in dynamicEx mapping table.
373 //
374 Found = FALSE;
375 GuidTableIdx = MatchGuid - GuidTable;
376 for (Index = 0; Index < SizeOfExMapTable; Index++) {
377 if (ExMapTable[Index].ExGuidIndex == GuidTableIdx) {
378 Found = TRUE;
379 break;
380 }
381 }
382
383 if (Found) {
384 //
385 // If given token number is PCD_INVALID_TOKEN_NUMBER, then return the first
386 // token number in found token space.
387 //
388 if (*TokenNumber == PCD_INVALID_TOKEN_NUMBER) {
389 *TokenNumber = ExMapTable[Index].ExTokenNumber;
390 return EFI_SUCCESS;
391 }
392
393 for ( ; Index < SizeOfExMapTable; Index++) {
394 if (ExMapTable[Index].ExTokenNumber == *TokenNumber) {
395 Index ++;
396 if (Index == SizeOfExMapTable) {
397 //
398 // Exceed the length of ExMap Table
399 //
400 *TokenNumber = PCD_INVALID_TOKEN_NUMBER;
401 return EFI_SUCCESS;
402 } else if (ExMapTable[Index].ExGuidIndex == GuidTableIdx) {
403 //
404 // Found the next match
405 //
406 *TokenNumber = ExMapTable[Index].ExTokenNumber;
407 return EFI_SUCCESS;
408 } else {
409 //
410 // Guid has been changed. It is the next Token Space Guid.
411 // We should flag no more TokenNumber.
412 //
413 *TokenNumber = PCD_INVALID_TOKEN_NUMBER;
414 return EFI_SUCCESS;
415 }
416 }
417 }
418 }
419
420 return EFI_NOT_FOUND;
421 }
422
423
424 /**
425 Initialize the PCD database in DXE phase.
426
427 PCD database in DXE phase also contains PCD database in PEI phase which is copied
428 from GUID Hob.
429
430 **/
431 VOID
432 BuildPcdDxeDataBase (
433 VOID
434 )
435 {
436 PEI_PCD_DATABASE *PeiDatabase;
437 EFI_HOB_GUID_TYPE *GuidHob;
438 UINTN Index;
439
440 mPcdDatabase = AllocateZeroPool (sizeof(PCD_DATABASE));
441 ASSERT (mPcdDatabase != NULL);
442
443 GuidHob = GetFirstGuidHob (&gPcdDataBaseHobGuid);
444 if (GuidHob != NULL) {
445
446 //
447 // We will copy over the PEI phase's PCD Database.
448 //
449 // If no PEIMs use dynamic Pcd Entry, the Pcd Service PEIM
450 // should not be included at all. So the GuidHob could
451 // be NULL. If it is NULL, we just copy over the DXE Default
452 // Value to PCD Database.
453 //
454
455 PeiDatabase = (PEI_PCD_DATABASE *) GET_GUID_HOB_DATA (GuidHob);
456 //
457 // Copy PCD Entries refereneced in PEI phase to PCD DATABASE
458 //
459 CopyMem (&mPcdDatabase->PeiDb, PeiDatabase, sizeof (PEI_PCD_DATABASE));
460 }
461
462 //
463 // Copy PCD Entries with default value to PCD DATABASE
464 //
465 CopyMem (&mPcdDatabase->DxeDb.Init, &gDXEPcdDbInit, sizeof(DXE_PCD_DATABASE_INIT));
466
467
468 //
469 // Initialized the Callback Function Table
470 //
471
472 mCallbackFnTable = AllocateZeroPool (PCD_TOTAL_TOKEN_NUMBER * sizeof (LIST_ENTRY));
473 ASSERT(mCallbackFnTable != NULL);
474
475 // EBC compiler is very choosy. It may report warning about comparison
476 // between UINTN and 0 . So we add 1 in each size of the
477 // comparison.
478 for (Index = 0; Index + 1 < PCD_TOTAL_TOKEN_NUMBER + 1; Index++) {
479 InitializeListHead (&mCallbackFnTable[Index]);
480 }
481 }
482
483 /**
484 Get Variable which contains HII type PCD entry.
485
486 @param VariableGuid Variable's guid
487 @param VariableName Variable's unicode name string
488 @param VariableData Variable's data pointer,
489 @param VariableSize Variable's size.
490
491 @return the status of gRT->GetVariable
492 **/
493 EFI_STATUS
494 GetHiiVariable (
495 IN EFI_GUID *VariableGuid,
496 IN UINT16 *VariableName,
497 OUT UINT8 **VariableData,
498 OUT UINTN *VariableSize
499 )
500 {
501 UINTN Size;
502 EFI_STATUS Status;
503 UINT8 *Buffer;
504
505 Size = 0;
506 Buffer = NULL;
507
508 //
509 // Firstly get the real size of HII variable
510 //
511 Status = gRT->GetVariable (
512 (UINT16 *)VariableName,
513 VariableGuid,
514 NULL,
515 &Size,
516 Buffer
517 );
518
519 //
520 // Allocate buffer to hold whole variable data according to variable size.
521 //
522 if (Status == EFI_BUFFER_TOO_SMALL) {
523 Buffer = (UINT8 *) AllocatePool (Size);
524
525 ASSERT (Buffer != NULL);
526
527 Status = gRT->GetVariable (
528 VariableName,
529 VariableGuid,
530 NULL,
531 &Size,
532 Buffer
533 );
534
535 ASSERT (Status == EFI_SUCCESS);
536 *VariableData = Buffer;
537 *VariableSize = Size;
538 }
539
540 return Status;
541 }
542
543 /**
544 Find the local token number according to system SKU ID.
545
546 @param LocalTokenNumber PCD token number
547 @param Size The size of PCD entry.
548 @param IsPeiDb If TRUE, the PCD entry is initialized in PEI phase.
549 If False, the PCD entry is initialized in DXE phase.
550
551 @return Token number according to system SKU ID.
552
553 **/
554 UINT32
555 GetSkuEnabledTokenNumber (
556 UINT32 LocalTokenNumber,
557 UINTN Size,
558 BOOLEAN IsPeiDb
559 )
560 {
561 SKU_HEAD *SkuHead;
562 SKU_ID *SkuIdTable;
563 INTN Index;
564 UINT8 *Value;
565 SKU_ID *PhaseSkuIdTable;
566 UINT8 *PcdDb;
567
568 ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0);
569
570 PcdDb = IsPeiDb ? (UINT8 *) &mPcdDatabase->PeiDb : (UINT8 *) &mPcdDatabase->DxeDb;
571
572 SkuHead = (SKU_HEAD *) (PcdDb + (LocalTokenNumber & PCD_DATABASE_OFFSET_MASK));
573 Value = (UINT8 *) (PcdDb + SkuHead->SkuDataStartOffset);
574
575 PhaseSkuIdTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.SkuIdTable :
576 mPcdDatabase->DxeDb.Init.SkuIdTable;
577
578 SkuIdTable = &PhaseSkuIdTable[SkuHead->SkuIdTableOffset];
579
580 //
581 // Find the current system's SKU ID entry in SKU ID table.
582 //
583 for (Index = 0; Index < SkuIdTable[0]; Index++) {
584 if (mPcdDatabase->PeiDb.Init.SystemSkuId == SkuIdTable[Index + 1]) {
585 break;
586 }
587 }
588 ASSERT (Index < SkuIdTable[0]);
589
590 switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {
591 case PCD_TYPE_VPD:
592 Value = (UINT8 *) &(((VPD_HEAD *) Value)[Index]);
593 return (UINT32) ((Value - PcdDb) | PCD_TYPE_VPD);
594
595 case PCD_TYPE_HII:
596 Value = (UINT8 *) &(((VARIABLE_HEAD *) Value)[Index]);
597 return (UINT32) ((Value - PcdDb) | PCD_TYPE_HII);
598
599 case PCD_TYPE_STRING:
600 Value = (UINT8 *) &(((STRING_HEAD *) Value)[Index]);
601 return (UINT32) ((Value - PcdDb) | PCD_TYPE_STRING);
602
603 case PCD_TYPE_DATA:
604 Value += Size * Index;
605 return (UINT32) (Value - PcdDb);
606
607 default:
608 ASSERT (FALSE);
609 }
610
611 ASSERT (FALSE);
612
613 return 0;
614
615 }
616
617 /**
618 Invoke the callback function when dynamic PCD entry was set, if this PCD entry
619 has registered callback function.
620
621 @param ExTokenNumber DynamicEx PCD's token number, if this PCD entry is dyanmicEx
622 type PCD.
623 @param Guid DynamicEx PCD's guid, if this PCD entry is dynamicEx type
624 PCD.
625 @param TokenNumber PCD token number generated by build tools.
626 @param Data Value want to be set for this PCD entry
627 @param Size The size of value
628
629 **/
630 VOID
631 InvokeCallbackOnSet (
632 UINT32 ExTokenNumber,
633 CONST EFI_GUID *Guid, OPTIONAL
634 UINTN TokenNumber,
635 VOID *Data,
636 UINTN Size
637 )
638 {
639 CALLBACK_FN_ENTRY *FnTableEntry;
640 LIST_ENTRY *ListHead;
641 LIST_ENTRY *ListNode;
642
643 //
644 // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.
645 // We have to decrement TokenNumber by 1 to make it usable
646 // as the array index of mCallbackFnTable[].
647 //
648 ListHead = &mCallbackFnTable[TokenNumber - 1];
649 ListNode = GetFirstNode (ListHead);
650
651 while (ListNode != ListHead) {
652 FnTableEntry = CR_FNENTRY_FROM_LISTNODE (ListNode, CALLBACK_FN_ENTRY, Node);
653
654 FnTableEntry->CallbackFn(Guid,
655 (Guid == NULL) ? TokenNumber : ExTokenNumber,
656 Data,
657 Size);
658
659 ListNode = GetNextNode (ListHead, ListNode);
660 }
661
662 return;
663 }
664
665
666 /**
667 Wrapper function for setting non-pointer type value for a PCD entry.
668
669 @param TokenNumber Pcd token number autogenerated by build tools.
670 @param Data Value want to be set for PCD entry
671 @param Size Size of value.
672
673 @return status of SetWorker.
674
675 **/
676 EFI_STATUS
677 SetValueWorker (
678 IN UINTN TokenNumber,
679 IN VOID *Data,
680 IN UINTN Size
681 )
682 {
683 return SetWorker (TokenNumber, Data, &Size, FALSE);
684 }
685
686
687 /**
688 Set value for an PCD entry
689
690 @param TokenNumber Pcd token number autogenerated by build tools.
691 @param Data Value want to be set for PCD entry
692 @param Size Size of value.
693 @param PtrType If TRUE, the type of PCD entry's value is Pointer.
694 If False, the type of PCD entry's value is not Pointer.
695
696 @retval EFI_INVALID_PARAMETER If this PCD type is VPD, VPD PCD can not be set.
697 @retval EFI_INVALID_PARAMETER If Size can not be set to size table.
698 @retval EFI_INVALID_PARAMETER If Size of non-Ptr type PCD does not match the size information in PCD database.
699 @retval EFI_NOT_FOUND If value type of PCD entry is intergrate, but not in
700 range of UINT8, UINT16, UINT32, UINT64
701 @retval EFI_NOT_FOUND Can not find the PCD type according to token number.
702 **/
703 EFI_STATUS
704 SetWorker (
705 IN UINTN TokenNumber,
706 IN VOID *Data,
707 IN OUT UINTN *Size,
708 IN BOOLEAN PtrType
709 )
710 {
711 UINT32 *LocalTokenNumberTable;
712 BOOLEAN IsPeiDb;
713 UINT32 LocalTokenNumber;
714 EFI_GUID *GuidTable;
715 UINT8 *StringTable;
716 EFI_GUID *Guid;
717 UINT16 *Name;
718 UINTN VariableOffset;
719 VOID *InternalData;
720 VARIABLE_HEAD *VariableHead;
721 UINTN Offset;
722 UINT8 *PcdDb;
723 EFI_STATUS Status;
724 UINTN MaxSize;
725 UINTN TmpTokenNumber;
726
727 //
728 // TokenNumber Zero is reserved as PCD_INVALID_TOKEN_NUMBER.
729 // We have to decrement TokenNumber by 1 to make it usable
730 // as the array index.
731 //
732 TokenNumber--;
733
734 TmpTokenNumber = TokenNumber;
735
736 //
737 // EBC compiler is very choosy. It may report warning about comparison
738 // between UINTN and 0 . So we add 1 in each size of the
739 // comparison.
740 //
741 ASSERT (TokenNumber + 1 < PCD_TOTAL_TOKEN_NUMBER + 1);
742
743 if (PtrType) {
744 //
745 // Get MaxSize first, then check new size with max buffer size.
746 //
747 GetPtrTypeSize (TokenNumber, &MaxSize);
748 if (*Size > MaxSize) {
749 *Size = MaxSize;
750 return EFI_INVALID_PARAMETER;
751 }
752 } else {
753 if (*Size != DxePcdGetSize (TokenNumber + 1)) {
754 return EFI_INVALID_PARAMETER;
755 }
756 }
757
758 //
759 // EBC compiler is very choosy. It may report warning about comparison
760 // between UINTN and 0 . So we add 1 in each size of the
761 // comparison.
762 //
763 if ((TokenNumber + 1 < PEI_NEX_TOKEN_NUMBER + 1) ||
764 (TokenNumber + 1 >= PEI_LOCAL_TOKEN_NUMBER + 1 || TokenNumber + 1 < (PEI_LOCAL_TOKEN_NUMBER + DXE_NEX_TOKEN_NUMBER + 1))) {
765 InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, *Size);
766 }
767
768 //
769 // Aquire lock to prevent reentrance from TPL_CALLBACK level
770 //
771 EfiAcquireLock (&mPcdDatabaseLock);
772
773 //
774 // EBC compiler is very choosy. It may report warning about comparison
775 // between UINTN and 0 . So we add 1 in each size of the
776 // comparison.
777 //
778 IsPeiDb = (BOOLEAN) ((TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1) ? TRUE : FALSE);
779
780 LocalTokenNumberTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable :
781 mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
782
783 TokenNumber = IsPeiDb ? TokenNumber
784 : TokenNumber - PEI_LOCAL_TOKEN_NUMBER;
785
786 LocalTokenNumber = LocalTokenNumberTable[TokenNumber];
787
788 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {
789 if (PtrType) {
790 GetPtrTypeSize (TmpTokenNumber, &MaxSize);
791 } else {
792 MaxSize = *Size;
793 }
794 LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize, IsPeiDb);
795 }
796
797 Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;
798
799 PcdDb = IsPeiDb ? ((UINT8 *) &mPcdDatabase->PeiDb) : ((UINT8 *) &mPcdDatabase->DxeDb);
800
801 if (IsPeiDb) {
802 StringTable = (UINT8 *) (&mPcdDatabase->PeiDb.Init.StringTable[0]);
803 } else {
804 StringTable = (UINT8 *) (&mPcdDatabase->DxeDb.Init.StringTable[0]);
805 }
806
807
808 InternalData = PcdDb + Offset;
809
810 switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {
811 case PCD_TYPE_VPD:
812 ASSERT (FALSE);
813 Status = EFI_INVALID_PARAMETER;
814 break;
815
816 case PCD_TYPE_STRING:
817 if (SetPtrTypeSize (TmpTokenNumber, Size)) {
818 CopyMem (StringTable + *((STRING_HEAD *)InternalData), Data, *Size);
819 Status = EFI_SUCCESS;
820 } else {
821 Status = EFI_INVALID_PARAMETER;
822 }
823 break;
824
825 case PCD_TYPE_HII|PCD_TYPE_STRING:
826 case PCD_TYPE_HII:
827 if (PtrType) {
828 if (!SetPtrTypeSize (TmpTokenNumber, Size)) {
829 Status = EFI_INVALID_PARAMETER;
830 break;
831 }
832 }
833
834 if (IsPeiDb) {
835 GuidTable = (EFI_GUID *) (&mPcdDatabase->PeiDb.Init.GuidTable[0]);
836 } else {
837 GuidTable = (EFI_GUID *) (&mPcdDatabase->DxeDb.Init.GuidTable[0]);
838 }
839
840 VariableHead = (VARIABLE_HEAD *) (PcdDb + Offset);
841
842 Guid = GuidTable + VariableHead->GuidTableIndex;
843 Name = (UINT16*) (StringTable + VariableHead->StringIndex);
844 VariableOffset = VariableHead->Offset;
845 Status = SetHiiVariable (Guid, Name, Data, *Size, VariableOffset);
846
847 if (EFI_NOT_FOUND == Status) {
848 if ((LocalTokenNumber & PCD_TYPE_ALL_SET) == (PCD_TYPE_HII|PCD_TYPE_STRING)) {
849 CopyMem (
850 StringTable + *(STRING_HEAD *)(PcdDb + VariableHead->DefaultValueOffset),
851 Data,
852 *Size
853 );
854 } else {
855 CopyMem (PcdDb + VariableHead->DefaultValueOffset, Data, *Size);
856 }
857 Status = EFI_SUCCESS;
858 }
859 break;
860
861 case PCD_TYPE_DATA:
862 if (PtrType) {
863 if (SetPtrTypeSize (TmpTokenNumber, Size)) {
864 CopyMem (InternalData, Data, *Size);
865 Status = EFI_SUCCESS;
866 } else {
867 Status = EFI_INVALID_PARAMETER;
868 }
869 break;
870 }
871
872 Status = EFI_SUCCESS;
873 switch (*Size) {
874 case sizeof(UINT8):
875 *((UINT8 *) InternalData) = *((UINT8 *) Data);
876 break;
877
878 case sizeof(UINT16):
879 *((UINT16 *) InternalData) = *((UINT16 *) Data);
880 break;
881
882 case sizeof(UINT32):
883 *((UINT32 *) InternalData) = *((UINT32 *) Data);
884 break;
885
886 case sizeof(UINT64):
887 *((UINT64 *) InternalData) = *((UINT64 *) Data);
888 break;
889
890 default:
891 ASSERT (FALSE);
892 Status = EFI_NOT_FOUND;
893 break;
894 }
895 break;
896
897 default:
898 ASSERT (FALSE);
899 Status = EFI_NOT_FOUND;
900 break;
901 }
902
903 EfiReleaseLock (&mPcdDatabaseLock);
904
905 return Status;
906 }
907
908 /**
909 Wrapper function for get PCD value for dynamic-ex PCD.
910
911 @param Guid Token space guid for dynamic-ex PCD.
912 @param ExTokenNumber Token number for dynamic-ex PCD.
913 @param GetSize The size of dynamic-ex PCD value.
914
915 @return PCD entry in PCD database.
916
917 **/
918 VOID *
919 ExGetWorker (
920 IN CONST EFI_GUID *Guid,
921 IN UINTN ExTokenNumber,
922 IN UINTN GetSize
923 )
924 {
925 return GetWorker(GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber), GetSize);
926 }
927
928 /**
929 Wrapper function for set PCD value for non-Pointer type dynamic-ex PCD.
930
931 @param ExTokenNumber Token number for dynamic-ex PCD.
932 @param Guid Token space guid for dynamic-ex PCD.
933 @param Data Value want to be set.
934 @param SetSize The size of value.
935
936 @return status of ExSetWorker().
937
938 **/
939 EFI_STATUS
940 ExSetValueWorker (
941 IN UINTN ExTokenNumber,
942 IN CONST EFI_GUID *Guid,
943 IN VOID *Data,
944 IN UINTN SetSize
945 )
946 {
947 return ExSetWorker (ExTokenNumber, Guid, Data, &SetSize, FALSE);
948 }
949
950 /**
951 Set value for a dynamic PCD entry.
952
953 This routine find the local token number according to dynamic-ex PCD's token
954 space guid and token number firstly, and invoke callback function if this PCD
955 entry registered callback function. Finally, invoken general SetWorker to set
956 PCD value.
957
958 @param ExTokenNumber Dynamic-ex PCD token number.
959 @param Guid Token space guid for dynamic-ex PCD.
960 @param Data PCD value want to be set
961 @param SetSize Size of value.
962 @param PtrType If TRUE, this PCD entry is pointer type.
963 If FALSE, this PCD entry is not pointer type.
964
965 @return status of SetWorker().
966
967 **/
968 EFI_STATUS
969 ExSetWorker (
970 IN UINTN ExTokenNumber,
971 IN CONST EFI_GUID *Guid,
972 IN VOID *Data,
973 IN OUT UINTN *SetSize,
974 IN BOOLEAN PtrType
975 )
976 {
977 UINTN TokenNumber;
978
979 TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber);
980
981 InvokeCallbackOnSet ((UINT32) ExTokenNumber, Guid, TokenNumber, Data, *SetSize);
982
983 return SetWorker (TokenNumber, Data, SetSize, PtrType);
984
985 }
986
987 /**
988 Set value for HII-type PCD.
989
990 A HII-type PCD's value is stored in a variable. Setting/Getting the value of
991 HII-type PCD is to visit this variable.
992
993 @param VariableGuid Guid of variable which stored value of a HII-type PCD.
994 @param VariableName Unicode name of variable which stored value of a HII-type PCD.
995 @param Data Value want to be set.
996 @param DataSize Size of value
997 @param Offset Value offset of HII-type PCD in variable.
998
999 @return status of GetVariable()/SetVariable().
1000
1001 **/
1002 EFI_STATUS
1003 SetHiiVariable (
1004 IN EFI_GUID *VariableGuid,
1005 IN UINT16 *VariableName,
1006 IN CONST VOID *Data,
1007 IN UINTN DataSize,
1008 IN UINTN Offset
1009 )
1010 {
1011 UINTN Size;
1012 VOID *Buffer;
1013 EFI_STATUS Status;
1014 UINT32 Attribute;
1015 UINTN SetSize;
1016
1017 Size = 0;
1018 SetSize = 0;
1019
1020 //
1021 // Try to get original variable size information.
1022 //
1023 Status = gRT->GetVariable (
1024 (UINT16 *)VariableName,
1025 VariableGuid,
1026 NULL,
1027 &Size,
1028 NULL
1029 );
1030
1031 if (Status == EFI_BUFFER_TOO_SMALL) {
1032 //
1033 // Patch new PCD's value to offset in given HII variable.
1034 //
1035 if (Size >= (DataSize + Offset)) {
1036 SetSize = Size;
1037 } else {
1038 SetSize = DataSize + Offset;
1039 }
1040 Buffer = AllocatePool (SetSize);
1041 ASSERT (Buffer != NULL);
1042
1043 Status = gRT->GetVariable (
1044 VariableName,
1045 VariableGuid,
1046 &Attribute,
1047 &Size,
1048 Buffer
1049 );
1050
1051 ASSERT_EFI_ERROR (Status);
1052
1053 CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);
1054
1055 Status = gRT->SetVariable (
1056 VariableName,
1057 VariableGuid,
1058 Attribute,
1059 SetSize,
1060 Buffer
1061 );
1062
1063 FreePool (Buffer);
1064 return Status;
1065 } else if (Status == EFI_NOT_FOUND) {
1066 //
1067 // If variable does not exist, a new variable need to be created.
1068 //
1069
1070 Size = Offset + DataSize;
1071
1072 Buffer = AllocateZeroPool (Size);
1073 ASSERT (Buffer != NULL);
1074
1075 CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);
1076
1077 Status = gRT->SetVariable (
1078 VariableName,
1079 VariableGuid,
1080 EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,
1081 Size,
1082 Buffer
1083 );
1084
1085 FreePool (Buffer);
1086 return Status;
1087 }
1088
1089 //
1090 // If we drop to here, the value is failed to be written in to variable area
1091 // So, we will save the data in the PCD Database's volatile area.
1092 //
1093 return Status;
1094 }
1095
1096 /**
1097 Get local token number according to dynamic-ex PCD's {token space guid:token number}
1098
1099 A dynamic-ex type PCD, developer must provide pair of token space guid: token number
1100 in DEC file. PCD database maintain a mapping table that translate pair of {token
1101 space guid: token number} to local token number.
1102
1103 @param Guid Token space guid for dynamic-ex PCD entry.
1104 @param ExTokenNumber Dynamic-ex PCD token number.
1105
1106 @return local token number for dynamic-ex PCD.
1107
1108 **/
1109 UINTN
1110 GetExPcdTokenNumber (
1111 IN CONST EFI_GUID *Guid,
1112 IN UINT32 ExTokenNumber
1113 )
1114 {
1115 UINT32 Index;
1116 DYNAMICEX_MAPPING *ExMap;
1117 EFI_GUID *GuidTable;
1118 EFI_GUID *MatchGuid;
1119 UINTN MatchGuidIdx;
1120
1121 if (!PEI_DATABASE_EMPTY) {
1122 ExMap = mPcdDatabase->PeiDb.Init.ExMapTable;
1123 GuidTable = mPcdDatabase->PeiDb.Init.GuidTable;
1124
1125 MatchGuid = ScanGuid (GuidTable, sizeof(mPcdDatabase->PeiDb.Init.GuidTable), Guid);
1126
1127 if (MatchGuid != NULL) {
1128
1129 MatchGuidIdx = MatchGuid - GuidTable;
1130
1131 for (Index = 0; Index < PEI_EXMAPPING_TABLE_SIZE; Index++) {
1132 if ((ExTokenNumber == ExMap[Index].ExTokenNumber) &&
1133 (MatchGuidIdx == ExMap[Index].ExGuidIndex)) {
1134 return ExMap[Index].LocalTokenNumber;
1135 }
1136 }
1137 }
1138 }
1139
1140 ExMap = mPcdDatabase->DxeDb.Init.ExMapTable;
1141 GuidTable = mPcdDatabase->DxeDb.Init.GuidTable;
1142
1143 MatchGuid = ScanGuid (GuidTable, sizeof(mPcdDatabase->DxeDb.Init.GuidTable), Guid);
1144 //
1145 // We need to ASSERT here. If GUID can't be found in GuidTable, this is a
1146 // error in the BUILD system.
1147 //
1148 ASSERT (MatchGuid != NULL);
1149
1150 MatchGuidIdx = MatchGuid - GuidTable;
1151
1152 for (Index = 0; Index < DXE_EXMAPPING_TABLE_SIZE; Index++) {
1153 if ((ExTokenNumber == ExMap[Index].ExTokenNumber) &&
1154 (MatchGuidIdx == ExMap[Index].ExGuidIndex)) {
1155 return ExMap[Index].LocalTokenNumber;
1156 }
1157 }
1158
1159 ASSERT (FALSE);
1160
1161 return 0;
1162 }
1163
1164 /**
1165 Get SKU ID table from PCD database.
1166
1167 @param LocalTokenNumberTableIdx Index of local token number in token number table.
1168 @param IsPeiPcd If TRUE,
1169
1170 @return Pointer to SKU ID array table
1171
1172 **/
1173 SKU_ID *
1174 GetSkuIdArray (
1175 IN UINTN LocalTokenNumberTableIdx,
1176 IN BOOLEAN IsPeiPcd
1177 )
1178 {
1179 SKU_HEAD *SkuHead;
1180 UINTN LocalTokenNumber;
1181 UINT8 *Database;
1182
1183 if (IsPeiPcd) {
1184 LocalTokenNumber = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];
1185 Database = (UINT8 *) &mPcdDatabase->PeiDb;
1186 } else {
1187 LocalTokenNumber = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx - PEI_LOCAL_TOKEN_NUMBER];
1188 Database = (UINT8 *) &mPcdDatabase->DxeDb;
1189 }
1190
1191 ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) != 0);
1192
1193 SkuHead = (SKU_HEAD *) ((UINT8 *)Database + (LocalTokenNumber & PCD_DATABASE_OFFSET_MASK));
1194
1195 return (SKU_ID *) (Database + SkuHead->SkuIdTableOffset);
1196
1197 }
1198
1199 /**
1200 Wrapper function of getting index of PCD entry in size table.
1201
1202 @param LocalTokenNumberTableIdx Index of this PCD in local token number table.
1203 @param IsPeiDb If TRUE, the pcd entry is initialized in PEI phase,
1204 If FALSE, the pcd entry is initialized in DXE phase.
1205
1206 @return index of PCD entry in size table.
1207 **/
1208 UINTN
1209 GetSizeTableIndex (
1210 IN UINTN LocalTokenNumberTableIdx,
1211 IN BOOLEAN IsPeiDb
1212 )
1213 {
1214 UINT32 *LocalTokenNumberTable;
1215 UINTN LocalTokenNumber;
1216 UINTN Index;
1217 UINTN SizeTableIdx;
1218 SKU_ID *SkuIdTable;
1219
1220 if (IsPeiDb) {
1221 LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;
1222 } else {
1223 LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
1224 }
1225
1226 SizeTableIdx = 0;
1227
1228 for (Index = 0; Index < LocalTokenNumberTableIdx; Index ++) {
1229 LocalTokenNumber = LocalTokenNumberTable[Index];
1230
1231 if ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER) {
1232 //
1233 // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type
1234 // PCD entry.
1235 //
1236 if ((LocalTokenNumber & PCD_TYPE_VPD) != 0) {
1237 //
1238 // We have only one entry for VPD enabled PCD entry:
1239 // 1) MAX Size.
1240 // We consider current size is equal to MAX size.
1241 //
1242 SizeTableIdx++;
1243 } else {
1244 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {
1245 //
1246 // We have only two entry for Non-Sku enabled PCD entry:
1247 // 1) MAX SIZE
1248 // 2) Current Size
1249 //
1250 SizeTableIdx += 2;
1251 } else {
1252 //
1253 // We have these entry for SKU enabled PCD entry
1254 // 1) MAX SIZE
1255 // 2) Current Size for each SKU_ID (It is equal to MaxSku).
1256 //
1257 SkuIdTable = GetSkuIdArray (Index, IsPeiDb);
1258 SizeTableIdx += (UINTN)*SkuIdTable + 1;
1259 }
1260 }
1261 }
1262
1263 }
1264
1265 return SizeTableIdx;
1266 }
1267
1268 /**
1269 Get size of POINTER type PCD value.
1270
1271 @param LocalTokenNumberTableIdx Index of local token number in local token number table.
1272 @param MaxSize Maxmium size of POINTER type PCD value.
1273
1274 @return size of POINTER type PCD value.
1275
1276 **/
1277 UINTN
1278 GetPtrTypeSize (
1279 IN UINTN LocalTokenNumberTableIdx,
1280 OUT UINTN *MaxSize
1281 )
1282 {
1283 INTN SizeTableIdx;
1284 UINTN LocalTokenNumber;
1285 SKU_ID *SkuIdTable;
1286 SIZE_INFO *SizeTable;
1287 UINTN Index;
1288 BOOLEAN IsPeiDb;
1289 UINT32 *LocalTokenNumberTable;
1290
1291 // EBC compiler is very choosy. It may report warning about comparison
1292 // between UINTN and 0 . So we add 1 in each size of the
1293 // comparison.
1294 IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);
1295
1296
1297 if (IsPeiDb) {
1298 LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;
1299 SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;
1300 } else {
1301 LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;
1302 LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
1303 SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;
1304 }
1305
1306 LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];
1307
1308 ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);
1309
1310 SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);
1311
1312 *MaxSize = SizeTable[SizeTableIdx];
1313 //
1314 // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type
1315 // PCD entry.
1316 //
1317 if ((LocalTokenNumber & PCD_TYPE_VPD) != 0) {
1318 //
1319 // We have only one entry for VPD enabled PCD entry:
1320 // 1) MAX Size.
1321 // We consider current size is equal to MAX size.
1322 //
1323 return *MaxSize;
1324 } else {
1325 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {
1326 //
1327 // We have only two entry for Non-Sku enabled PCD entry:
1328 // 1) MAX SIZE
1329 // 2) Current Size
1330 //
1331 return SizeTable[SizeTableIdx + 1];
1332 } else {
1333 //
1334 // We have these entry for SKU enabled PCD entry
1335 // 1) MAX SIZE
1336 // 2) Current Size for each SKU_ID (It is equal to MaxSku).
1337 //
1338 SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);
1339 for (Index = 0; Index < SkuIdTable[0]; Index++) {
1340 if (SkuIdTable[1 + Index] == mPcdDatabase->PeiDb.Init.SystemSkuId) {
1341 return SizeTable[SizeTableIdx + 1 + Index];
1342 }
1343 }
1344 return SizeTable[SizeTableIdx + 1];
1345 }
1346 }
1347 }
1348
1349 /**
1350 Set size of POINTER type PCD value. The size should not exceed the maximum size
1351 of this PCD value.
1352
1353 @param LocalTokenNumberTableIdx Index of local token number in local token number table.
1354 @param CurrentSize Size of POINTER type PCD value.
1355
1356 @retval TRUE Success to set size of PCD value.
1357 @retval FALSE Fail to set size of PCD value.
1358 **/
1359 BOOLEAN
1360 SetPtrTypeSize (
1361 IN UINTN LocalTokenNumberTableIdx,
1362 IN OUT UINTN *CurrentSize
1363 )
1364 {
1365 INTN SizeTableIdx;
1366 UINTN LocalTokenNumber;
1367 SKU_ID *SkuIdTable;
1368 SIZE_INFO *SizeTable;
1369 UINTN Index;
1370 UINTN MaxSize;
1371 BOOLEAN IsPeiDb;
1372 UINT32 *LocalTokenNumberTable;
1373
1374 //
1375 // EBC compiler is very choosy. It may report warning about comparison
1376 // between UINTN and 0 . So we add 1 in each size of the
1377 // comparison.
1378 //
1379 IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);
1380
1381 if (IsPeiDb) {
1382 LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;
1383 SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;
1384 } else {
1385 LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;
1386 LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
1387 SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;
1388 }
1389
1390 LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];
1391
1392 ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);
1393
1394 SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);
1395
1396 MaxSize = SizeTable[SizeTableIdx];
1397 //
1398 // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type
1399 // PCD entry.
1400 //
1401 if ((LocalTokenNumber & PCD_TYPE_VPD) != 0) {
1402 //
1403 // We shouldn't come here as we don't support SET for VPD
1404 //
1405 ASSERT (FALSE);
1406 return FALSE;
1407 } else {
1408 if ((*CurrentSize > MaxSize) ||
1409 (*CurrentSize == MAX_ADDRESS)) {
1410 *CurrentSize = MaxSize;
1411 return FALSE;
1412 }
1413
1414 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {
1415 //
1416 // We have only two entry for Non-Sku enabled PCD entry:
1417 // 1) MAX SIZE
1418 // 2) Current Size
1419 //
1420 SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;
1421 return TRUE;
1422 } else {
1423 //
1424 // We have these entry for SKU enabled PCD entry
1425 // 1) MAX SIZE
1426 // 2) Current Size for each SKU_ID (It is equal to MaxSku).
1427 //
1428 SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);
1429 for (Index = 0; Index < SkuIdTable[0]; Index++) {
1430 if (SkuIdTable[1 + Index] == mPcdDatabase->PeiDb.Init.SystemSkuId) {
1431 SizeTable[SizeTableIdx + 1 + Index] = (SIZE_INFO) *CurrentSize;
1432 return TRUE;
1433 }
1434 }
1435 SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;
1436 return TRUE;
1437 }
1438 }
1439 }