blob: 8683947be394eaad9f89b040a533a07e1ceaff10 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#include <Library/UefiBootServicesTableLib.h>
#include <Library/UefiLib.h>
#include <Pi/PiMultiPhase.h>
#include <Protocol/PiPcd.h>
#include <Protocol/PiPcdInfo.h>
//#include <Library/PcdLib.h>
VOID Usage()
{
Print(L"Program to set Sku number\n\n");
Print(L" Usage: SetSku <SkuNumber in hex>\n");
}
INTN
EFIAPI
ShellAppMain (
IN UINTN Argc,
IN CHAR16 **Argv
)
{
EFI_STATUS Status;
if (Argc != 2) {
Usage();
return EFI_INVALID_PARAMETER;
}
UINTN SkuNumber;
RETURN_STATUS ReturnStatus;
ReturnStatus = StrHexToUintnS(Argv[1], NULL, &SkuNumber);
if (RETURN_ERROR(ReturnStatus)) {
Print(L"Error! Can't convert SkuId string to number\n");
return EFI_INVALID_PARAMETER;
}
EFI_PCD_PROTOCOL* pcdProtocol;
Status = gBS->LocateProtocol(&gEfiPcdProtocolGuid,
NULL,
(VOID **)&pcdProtocol);
if (EFI_ERROR(Status)) {
Print(L"Error! Can't find EFI_PCD_PROTOCOL\n");
return EFI_UNSUPPORTED;
}
EFI_GET_PCD_INFO_PROTOCOL* getPcdInfoProtocol;
Status = gBS->LocateProtocol(&gEfiGetPcdInfoProtocolGuid,
NULL,
(VOID **)&getPcdInfoProtocol);
if (EFI_ERROR(Status)) {
Print(L"Error! Can't find EFI_GET_PCD_INFO_PROTOCOL\n");
return EFI_UNSUPPORTED;
}
pcdProtocol->SetSku(SkuNumber);
UINTN SkuId = getPcdInfoProtocol->GetSku();
/*
LibPcdSetSku(SkuNumber);
UINTN SkuId = LibPcdGetSku();
*/
if (SkuId != SkuNumber) {
Print(L"Failed to change SkuId to 0x%lx, SkuId is still 0x%lx\n", SkuNumber, SkuId);
return EFI_UNSUPPORTED;
}
Print(L"Sku is changed successfully to 0x%lx\n", SkuNumber);
return EFI_SUCCESS;
}
|
