]> git.proxmox.com Git - mirror_edk2.git/blob - MdeModulePkg/Universal/PCD/Dxe/Service.c
Fix PcdSetPtr() service to use the max buffer size to check whether new buffer data...
[mirror_edk2.git] / MdeModulePkg / Universal / PCD / Dxe / Service.c
1 /** @file
2 Help functions used by PCD DXE driver.
3
4 Copyright (c) 2006 - 2011, 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 UINT16 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 = *(UINT16*)((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 = *(UINT16*)((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 return EFI_INVALID_PARAMETER;
750 }
751 } else {
752 if (*Size != DxePcdGetSize (TokenNumber + 1)) {
753 return EFI_INVALID_PARAMETER;
754 }
755 }
756
757 //
758 // EBC compiler is very choosy. It may report warning about comparison
759 // between UINTN and 0 . So we add 1 in each size of the
760 // comparison.
761 //
762 if ((TokenNumber + 1 < PEI_NEX_TOKEN_NUMBER + 1) ||
763 (TokenNumber + 1 >= PEI_LOCAL_TOKEN_NUMBER + 1 || TokenNumber + 1 < (PEI_LOCAL_TOKEN_NUMBER + DXE_NEX_TOKEN_NUMBER + 1))) {
764 InvokeCallbackOnSet (0, NULL, TokenNumber + 1, Data, *Size);
765 }
766
767 //
768 // Aquire lock to prevent reentrance from TPL_CALLBACK level
769 //
770 EfiAcquireLock (&mPcdDatabaseLock);
771
772 //
773 // EBC compiler is very choosy. It may report warning about comparison
774 // between UINTN and 0 . So we add 1 in each size of the
775 // comparison.
776 //
777 IsPeiDb = (BOOLEAN) ((TokenNumber + 1 < PEI_LOCAL_TOKEN_NUMBER + 1) ? TRUE : FALSE);
778
779 LocalTokenNumberTable = IsPeiDb ? mPcdDatabase->PeiDb.Init.LocalTokenNumberTable :
780 mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
781
782 TokenNumber = IsPeiDb ? TokenNumber
783 : TokenNumber - PEI_LOCAL_TOKEN_NUMBER;
784
785 LocalTokenNumber = LocalTokenNumberTable[TokenNumber];
786
787 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == PCD_TYPE_SKU_ENABLED) {
788 if (PtrType) {
789 GetPtrTypeSize (TmpTokenNumber, &MaxSize);
790 } else {
791 MaxSize = *Size;
792 }
793 LocalTokenNumber = GetSkuEnabledTokenNumber (LocalTokenNumber & ~PCD_TYPE_SKU_ENABLED, MaxSize, IsPeiDb);
794 }
795
796 Offset = LocalTokenNumber & PCD_DATABASE_OFFSET_MASK;
797
798 PcdDb = IsPeiDb ? ((UINT8 *) &mPcdDatabase->PeiDb) : ((UINT8 *) &mPcdDatabase->DxeDb);
799
800 if (IsPeiDb) {
801 StringTable = (UINT8 *) (&mPcdDatabase->PeiDb.Init.StringTable[0]);
802 } else {
803 StringTable = (UINT8 *) (&mPcdDatabase->DxeDb.Init.StringTable[0]);
804 }
805
806
807 InternalData = PcdDb + Offset;
808
809 switch (LocalTokenNumber & PCD_TYPE_ALL_SET) {
810 case PCD_TYPE_VPD:
811 ASSERT (FALSE);
812 Status = EFI_INVALID_PARAMETER;
813 break;
814
815 case PCD_TYPE_STRING:
816 if (SetPtrTypeSize (TmpTokenNumber, Size)) {
817 CopyMem (StringTable + *((UINT16 *)InternalData), Data, *Size);
818 Status = EFI_SUCCESS;
819 } else {
820 Status = EFI_INVALID_PARAMETER;
821 }
822 break;
823
824 case PCD_TYPE_HII|PCD_TYPE_STRING:
825 case PCD_TYPE_HII:
826 if (PtrType) {
827 if (!SetPtrTypeSize (TmpTokenNumber, Size)) {
828 Status = EFI_INVALID_PARAMETER;
829 break;
830 }
831 }
832
833 if (IsPeiDb) {
834 GuidTable = (EFI_GUID *) (&mPcdDatabase->PeiDb.Init.GuidTable[0]);
835 } else {
836 GuidTable = (EFI_GUID *) (&mPcdDatabase->DxeDb.Init.GuidTable[0]);
837 }
838
839 VariableHead = (VARIABLE_HEAD *) (PcdDb + Offset);
840
841 Guid = GuidTable + VariableHead->GuidTableIndex;
842 Name = (UINT16*) (StringTable + VariableHead->StringIndex);
843 VariableOffset = VariableHead->Offset;
844 Status = SetHiiVariable (Guid, Name, Data, *Size, VariableOffset);
845
846 if (EFI_NOT_FOUND == Status) {
847 if ((LocalTokenNumber & PCD_TYPE_ALL_SET) == (PCD_TYPE_HII|PCD_TYPE_STRING)) {
848 CopyMem (
849 StringTable + *(UINT16 *)(PcdDb + VariableHead->DefaultValueOffset),
850 Data,
851 *Size
852 );
853 } else {
854 CopyMem (PcdDb + VariableHead->DefaultValueOffset, Data, *Size);
855 }
856 Status = EFI_SUCCESS;
857 }
858 break;
859
860 case PCD_TYPE_DATA:
861 if (PtrType) {
862 if (SetPtrTypeSize (TmpTokenNumber, Size)) {
863 CopyMem (InternalData, Data, *Size);
864 Status = EFI_SUCCESS;
865 } else {
866 Status = EFI_INVALID_PARAMETER;
867 }
868 break;
869 }
870
871 Status = EFI_SUCCESS;
872 switch (*Size) {
873 case sizeof(UINT8):
874 *((UINT8 *) InternalData) = *((UINT8 *) Data);
875 break;
876
877 case sizeof(UINT16):
878 *((UINT16 *) InternalData) = *((UINT16 *) Data);
879 break;
880
881 case sizeof(UINT32):
882 *((UINT32 *) InternalData) = *((UINT32 *) Data);
883 break;
884
885 case sizeof(UINT64):
886 *((UINT64 *) InternalData) = *((UINT64 *) Data);
887 break;
888
889 default:
890 ASSERT (FALSE);
891 Status = EFI_NOT_FOUND;
892 break;
893 }
894 break;
895
896 default:
897 ASSERT (FALSE);
898 Status = EFI_NOT_FOUND;
899 break;
900 }
901
902 EfiReleaseLock (&mPcdDatabaseLock);
903
904 return Status;
905 }
906
907 /**
908 Wrapper function for get PCD value for dynamic-ex PCD.
909
910 @param Guid Token space guid for dynamic-ex PCD.
911 @param ExTokenNumber Token number for dynamic-ex PCD.
912 @param GetSize The size of dynamic-ex PCD value.
913
914 @return PCD entry in PCD database.
915
916 **/
917 VOID *
918 ExGetWorker (
919 IN CONST EFI_GUID *Guid,
920 IN UINTN ExTokenNumber,
921 IN UINTN GetSize
922 )
923 {
924 return GetWorker(GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber), GetSize);
925 }
926
927 /**
928 Wrapper function for set PCD value for non-Pointer type dynamic-ex PCD.
929
930 @param ExTokenNumber Token number for dynamic-ex PCD.
931 @param Guid Token space guid for dynamic-ex PCD.
932 @param Data Value want to be set.
933 @param SetSize The size of value.
934
935 @return status of ExSetWorker().
936
937 **/
938 EFI_STATUS
939 ExSetValueWorker (
940 IN UINTN ExTokenNumber,
941 IN CONST EFI_GUID *Guid,
942 IN VOID *Data,
943 IN UINTN SetSize
944 )
945 {
946 return ExSetWorker (ExTokenNumber, Guid, Data, &SetSize, FALSE);
947 }
948
949 /**
950 Set value for a dynamic PCD entry.
951
952 This routine find the local token number according to dynamic-ex PCD's token
953 space guid and token number firstly, and invoke callback function if this PCD
954 entry registered callback function. Finally, invoken general SetWorker to set
955 PCD value.
956
957 @param ExTokenNumber Dynamic-ex PCD token number.
958 @param Guid Token space guid for dynamic-ex PCD.
959 @param Data PCD value want to be set
960 @param SetSize Size of value.
961 @param PtrType If TRUE, this PCD entry is pointer type.
962 If FALSE, this PCD entry is not pointer type.
963
964 @return status of SetWorker().
965
966 **/
967 EFI_STATUS
968 ExSetWorker (
969 IN UINTN ExTokenNumber,
970 IN CONST EFI_GUID *Guid,
971 IN VOID *Data,
972 IN OUT UINTN *SetSize,
973 IN BOOLEAN PtrType
974 )
975 {
976 UINTN TokenNumber;
977
978 TokenNumber = GetExPcdTokenNumber (Guid, (UINT32) ExTokenNumber);
979
980 InvokeCallbackOnSet ((UINT32) ExTokenNumber, Guid, TokenNumber, Data, *SetSize);
981
982 return SetWorker (TokenNumber, Data, SetSize, PtrType);
983
984 }
985
986 /**
987 Set value for HII-type PCD.
988
989 A HII-type PCD's value is stored in a variable. Setting/Getting the value of
990 HII-type PCD is to visit this variable.
991
992 @param VariableGuid Guid of variable which stored value of a HII-type PCD.
993 @param VariableName Unicode name of variable which stored value of a HII-type PCD.
994 @param Data Value want to be set.
995 @param DataSize Size of value
996 @param Offset Value offset of HII-type PCD in variable.
997
998 @return status of GetVariable()/SetVariable().
999
1000 **/
1001 EFI_STATUS
1002 SetHiiVariable (
1003 IN EFI_GUID *VariableGuid,
1004 IN UINT16 *VariableName,
1005 IN CONST VOID *Data,
1006 IN UINTN DataSize,
1007 IN UINTN Offset
1008 )
1009 {
1010 UINTN Size;
1011 VOID *Buffer;
1012 EFI_STATUS Status;
1013 UINT32 Attribute;
1014 UINTN SetSize;
1015
1016 Size = 0;
1017 SetSize = 0;
1018
1019 //
1020 // Try to get original variable size information.
1021 //
1022 Status = gRT->GetVariable (
1023 (UINT16 *)VariableName,
1024 VariableGuid,
1025 NULL,
1026 &Size,
1027 NULL
1028 );
1029
1030 if (Status == EFI_BUFFER_TOO_SMALL) {
1031 //
1032 // Patch new PCD's value to offset in given HII variable.
1033 //
1034 if (Size >= (DataSize + Offset)) {
1035 SetSize = Size;
1036 } else {
1037 SetSize = DataSize + Offset;
1038 }
1039 Buffer = AllocatePool (SetSize);
1040 ASSERT (Buffer != NULL);
1041
1042 Status = gRT->GetVariable (
1043 VariableName,
1044 VariableGuid,
1045 &Attribute,
1046 &Size,
1047 Buffer
1048 );
1049
1050 ASSERT_EFI_ERROR (Status);
1051
1052 CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);
1053
1054 Status = gRT->SetVariable (
1055 VariableName,
1056 VariableGuid,
1057 Attribute,
1058 SetSize,
1059 Buffer
1060 );
1061
1062 FreePool (Buffer);
1063 return Status;
1064 } else if (Status == EFI_NOT_FOUND) {
1065 //
1066 // If variable does not exist, a new variable need to be created.
1067 //
1068
1069 Size = Offset + DataSize;
1070
1071 Buffer = AllocateZeroPool (Size);
1072 ASSERT (Buffer != NULL);
1073
1074 CopyMem ((UINT8 *)Buffer + Offset, Data, DataSize);
1075
1076 Status = gRT->SetVariable (
1077 VariableName,
1078 VariableGuid,
1079 EFI_VARIABLE_BOOTSERVICE_ACCESS | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_NON_VOLATILE,
1080 Size,
1081 Buffer
1082 );
1083
1084 FreePool (Buffer);
1085 return Status;
1086 }
1087
1088 //
1089 // If we drop to here, the value is failed to be written in to variable area
1090 // So, we will save the data in the PCD Database's volatile area.
1091 //
1092 return Status;
1093 }
1094
1095 /**
1096 Get local token number according to dynamic-ex PCD's {token space guid:token number}
1097
1098 A dynamic-ex type PCD, developer must provide pair of token space guid: token number
1099 in DEC file. PCD database maintain a mapping table that translate pair of {token
1100 space guid: token number} to local token number.
1101
1102 @param Guid Token space guid for dynamic-ex PCD entry.
1103 @param ExTokenNumber Dynamic-ex PCD token number.
1104
1105 @return local token number for dynamic-ex PCD.
1106
1107 **/
1108 UINTN
1109 GetExPcdTokenNumber (
1110 IN CONST EFI_GUID *Guid,
1111 IN UINT32 ExTokenNumber
1112 )
1113 {
1114 UINT32 Index;
1115 DYNAMICEX_MAPPING *ExMap;
1116 EFI_GUID *GuidTable;
1117 EFI_GUID *MatchGuid;
1118 UINTN MatchGuidIdx;
1119
1120 if (!PEI_DATABASE_EMPTY) {
1121 ExMap = mPcdDatabase->PeiDb.Init.ExMapTable;
1122 GuidTable = mPcdDatabase->PeiDb.Init.GuidTable;
1123
1124 MatchGuid = ScanGuid (GuidTable, sizeof(mPcdDatabase->PeiDb.Init.GuidTable), Guid);
1125
1126 if (MatchGuid != NULL) {
1127
1128 MatchGuidIdx = MatchGuid - GuidTable;
1129
1130 for (Index = 0; Index < PEI_EXMAPPING_TABLE_SIZE; Index++) {
1131 if ((ExTokenNumber == ExMap[Index].ExTokenNumber) &&
1132 (MatchGuidIdx == ExMap[Index].ExGuidIndex)) {
1133 return ExMap[Index].LocalTokenNumber;
1134 }
1135 }
1136 }
1137 }
1138
1139 ExMap = mPcdDatabase->DxeDb.Init.ExMapTable;
1140 GuidTable = mPcdDatabase->DxeDb.Init.GuidTable;
1141
1142 MatchGuid = ScanGuid (GuidTable, sizeof(mPcdDatabase->DxeDb.Init.GuidTable), Guid);
1143 //
1144 // We need to ASSERT here. If GUID can't be found in GuidTable, this is a
1145 // error in the BUILD system.
1146 //
1147 ASSERT (MatchGuid != NULL);
1148
1149 MatchGuidIdx = MatchGuid - GuidTable;
1150
1151 for (Index = 0; Index < DXE_EXMAPPING_TABLE_SIZE; Index++) {
1152 if ((ExTokenNumber == ExMap[Index].ExTokenNumber) &&
1153 (MatchGuidIdx == ExMap[Index].ExGuidIndex)) {
1154 return ExMap[Index].LocalTokenNumber;
1155 }
1156 }
1157
1158 ASSERT (FALSE);
1159
1160 return 0;
1161 }
1162
1163 /**
1164 Get SKU ID table from PCD database.
1165
1166 @param LocalTokenNumberTableIdx Index of local token number in token number table.
1167 @param IsPeiPcd If TRUE,
1168
1169 @return Pointer to SKU ID array table
1170
1171 **/
1172 SKU_ID *
1173 GetSkuIdArray (
1174 IN UINTN LocalTokenNumberTableIdx,
1175 IN BOOLEAN IsPeiPcd
1176 )
1177 {
1178 SKU_HEAD *SkuHead;
1179 UINTN LocalTokenNumber;
1180 UINT8 *Database;
1181
1182 if (IsPeiPcd) {
1183 LocalTokenNumber = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx];
1184 Database = (UINT8 *) &mPcdDatabase->PeiDb;
1185 } else {
1186 LocalTokenNumber = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable[LocalTokenNumberTableIdx - PEI_LOCAL_TOKEN_NUMBER];
1187 Database = (UINT8 *) &mPcdDatabase->DxeDb;
1188 }
1189
1190 ASSERT ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) != 0);
1191
1192 SkuHead = (SKU_HEAD *) ((UINT8 *)Database + (LocalTokenNumber & PCD_DATABASE_OFFSET_MASK));
1193
1194 return (SKU_ID *) (Database + SkuHead->SkuIdTableOffset);
1195
1196 }
1197
1198 /**
1199 Wrapper function of getting index of PCD entry in size table.
1200
1201 @param LocalTokenNumberTableIdx Index of this PCD in local token number table.
1202 @param IsPeiDb If TRUE, the pcd entry is initialized in PEI phase,
1203 If FALSE, the pcd entry is initialized in DXE phase.
1204
1205 @return index of PCD entry in size table.
1206 **/
1207 UINTN
1208 GetSizeTableIndex (
1209 IN UINTN LocalTokenNumberTableIdx,
1210 IN BOOLEAN IsPeiDb
1211 )
1212 {
1213 UINT32 *LocalTokenNumberTable;
1214 UINTN LocalTokenNumber;
1215 UINTN Index;
1216 UINTN SizeTableIdx;
1217 SKU_ID *SkuIdTable;
1218
1219 if (IsPeiDb) {
1220 LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;
1221 } else {
1222 LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
1223 }
1224
1225 SizeTableIdx = 0;
1226
1227 for (Index = 0; Index < LocalTokenNumberTableIdx; Index ++) {
1228 LocalTokenNumber = LocalTokenNumberTable[Index];
1229
1230 if ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER) {
1231 //
1232 // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type
1233 // PCD entry.
1234 //
1235 if ((LocalTokenNumber & PCD_TYPE_VPD) != 0) {
1236 //
1237 // We have only one entry for VPD enabled PCD entry:
1238 // 1) MAX Size.
1239 // We consider current size is equal to MAX size.
1240 //
1241 SizeTableIdx++;
1242 } else {
1243 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {
1244 //
1245 // We have only two entry for Non-Sku enabled PCD entry:
1246 // 1) MAX SIZE
1247 // 2) Current Size
1248 //
1249 SizeTableIdx += 2;
1250 } else {
1251 //
1252 // We have these entry for SKU enabled PCD entry
1253 // 1) MAX SIZE
1254 // 2) Current Size for each SKU_ID (It is equal to MaxSku).
1255 //
1256 SkuIdTable = GetSkuIdArray (Index, IsPeiDb);
1257 SizeTableIdx += (UINTN)*SkuIdTable + 1;
1258 }
1259 }
1260 }
1261
1262 }
1263
1264 return SizeTableIdx;
1265 }
1266
1267 /**
1268 Get size of POINTER type PCD value.
1269
1270 @param LocalTokenNumberTableIdx Index of local token number in local token number table.
1271 @param MaxSize Maxmium size of POINTER type PCD value.
1272
1273 @return size of POINTER type PCD value.
1274
1275 **/
1276 UINTN
1277 GetPtrTypeSize (
1278 IN UINTN LocalTokenNumberTableIdx,
1279 OUT UINTN *MaxSize
1280 )
1281 {
1282 INTN SizeTableIdx;
1283 UINTN LocalTokenNumber;
1284 SKU_ID *SkuIdTable;
1285 SIZE_INFO *SizeTable;
1286 UINTN Index;
1287 BOOLEAN IsPeiDb;
1288 UINT32 *LocalTokenNumberTable;
1289
1290 // EBC compiler is very choosy. It may report warning about comparison
1291 // between UINTN and 0 . So we add 1 in each size of the
1292 // comparison.
1293 IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);
1294
1295
1296 if (IsPeiDb) {
1297 LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;
1298 SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;
1299 } else {
1300 LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;
1301 LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
1302 SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;
1303 }
1304
1305 LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];
1306
1307 ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);
1308
1309 SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);
1310
1311 *MaxSize = SizeTable[SizeTableIdx];
1312 //
1313 // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type
1314 // PCD entry.
1315 //
1316 if ((LocalTokenNumber & PCD_TYPE_VPD) != 0) {
1317 //
1318 // We have only one entry for VPD enabled PCD entry:
1319 // 1) MAX Size.
1320 // We consider current size is equal to MAX size.
1321 //
1322 return *MaxSize;
1323 } else {
1324 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {
1325 //
1326 // We have only two entry for Non-Sku enabled PCD entry:
1327 // 1) MAX SIZE
1328 // 2) Current Size
1329 //
1330 return SizeTable[SizeTableIdx + 1];
1331 } else {
1332 //
1333 // We have these entry for SKU enabled PCD entry
1334 // 1) MAX SIZE
1335 // 2) Current Size for each SKU_ID (It is equal to MaxSku).
1336 //
1337 SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);
1338 for (Index = 0; Index < SkuIdTable[0]; Index++) {
1339 if (SkuIdTable[1 + Index] == mPcdDatabase->PeiDb.Init.SystemSkuId) {
1340 return SizeTable[SizeTableIdx + 1 + Index];
1341 }
1342 }
1343 return SizeTable[SizeTableIdx + 1];
1344 }
1345 }
1346 }
1347
1348 /**
1349 Set size of POINTER type PCD value. The size should not exceed the maximum size
1350 of this PCD value.
1351
1352 @param LocalTokenNumberTableIdx Index of local token number in local token number table.
1353 @param CurrentSize Size of POINTER type PCD value.
1354
1355 @retval TRUE Success to set size of PCD value.
1356 @retval FALSE Fail to set size of PCD value.
1357 **/
1358 BOOLEAN
1359 SetPtrTypeSize (
1360 IN UINTN LocalTokenNumberTableIdx,
1361 IN OUT UINTN *CurrentSize
1362 )
1363 {
1364 INTN SizeTableIdx;
1365 UINTN LocalTokenNumber;
1366 SKU_ID *SkuIdTable;
1367 SIZE_INFO *SizeTable;
1368 UINTN Index;
1369 UINTN MaxSize;
1370 BOOLEAN IsPeiDb;
1371 UINT32 *LocalTokenNumberTable;
1372
1373 //
1374 // EBC compiler is very choosy. It may report warning about comparison
1375 // between UINTN and 0 . So we add 1 in each size of the
1376 // comparison.
1377 //
1378 IsPeiDb = (BOOLEAN) (LocalTokenNumberTableIdx + 1 < PEI_LOCAL_TOKEN_NUMBER + 1);
1379
1380 if (IsPeiDb) {
1381 LocalTokenNumberTable = mPcdDatabase->PeiDb.Init.LocalTokenNumberTable;
1382 SizeTable = mPcdDatabase->PeiDb.Init.SizeTable;
1383 } else {
1384 LocalTokenNumberTableIdx -= PEI_LOCAL_TOKEN_NUMBER;
1385 LocalTokenNumberTable = mPcdDatabase->DxeDb.Init.LocalTokenNumberTable;
1386 SizeTable = mPcdDatabase->DxeDb.Init.SizeTable;
1387 }
1388
1389 LocalTokenNumber = LocalTokenNumberTable[LocalTokenNumberTableIdx];
1390
1391 ASSERT ((LocalTokenNumber & PCD_DATUM_TYPE_ALL_SET) == PCD_DATUM_TYPE_POINTER);
1392
1393 SizeTableIdx = GetSizeTableIndex (LocalTokenNumberTableIdx, IsPeiDb);
1394
1395 MaxSize = SizeTable[SizeTableIdx];
1396 //
1397 // SizeTable only contain record for PCD_DATUM_TYPE_POINTER type
1398 // PCD entry.
1399 //
1400 if ((LocalTokenNumber & PCD_TYPE_VPD) != 0) {
1401 //
1402 // We shouldn't come here as we don't support SET for VPD
1403 //
1404 ASSERT (FALSE);
1405 return FALSE;
1406 } else {
1407 if ((*CurrentSize > MaxSize) ||
1408 (*CurrentSize == MAX_ADDRESS)) {
1409 *CurrentSize = MaxSize;
1410 return FALSE;
1411 }
1412
1413 if ((LocalTokenNumber & PCD_TYPE_SKU_ENABLED) == 0) {
1414 //
1415 // We have only two entry for Non-Sku enabled PCD entry:
1416 // 1) MAX SIZE
1417 // 2) Current Size
1418 //
1419 SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;
1420 return TRUE;
1421 } else {
1422 //
1423 // We have these entry for SKU enabled PCD entry
1424 // 1) MAX SIZE
1425 // 2) Current Size for each SKU_ID (It is equal to MaxSku).
1426 //
1427 SkuIdTable = GetSkuIdArray (LocalTokenNumberTableIdx, IsPeiDb);
1428 for (Index = 0; Index < SkuIdTable[0]; Index++) {
1429 if (SkuIdTable[1 + Index] == mPcdDatabase->PeiDb.Init.SystemSkuId) {
1430 SizeTable[SizeTableIdx + 1 + Index] = (SIZE_INFO) *CurrentSize;
1431 return TRUE;
1432 }
1433 }
1434 SizeTable[SizeTableIdx + 1] = (SIZE_INFO) *CurrentSize;
1435 return TRUE;
1436 }
1437 }
1438 }