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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
//*************************************************************************
//*************************************************************************
//** **
//** (C)Copyright 1985-2011, American Megatrends, Inc. **
//** **
//** All Rights Reserved. **
//** **
//** 5555 Oakbrook Parkway, Suite 200, Norcross, GA 30093 **
//** **
//** Phone: (770)-246-8600 **
//** **
//*************************************************************************
//*************************************************************************
//*****************************************************************************
// $Header: /Alaska/SOURCE/Modules/OemActivation/OemActivationHook.c 3 6/10/11 10:57a Vyacheslava $
//
// $Revision: 3 $
//
// $Date: 6/10/11 10:57a $
//*****************************************************************************
// Revision History
// ----------------
// $Log: /Alaska/SOURCE/Modules/OemActivation/OemActivationHook.c $
//
// 3 6/10/11 10:57a Vyacheslava
//
// 2 5/26/11 11:34a Vyacheslava
//
// 1 4/29/11 12:39p Vyacheslava
// [TAG] EIP58139
// [Category] New Feature
// [Description] Added SW SMI hook to place Oem Activation Key in NCB.
//
//
//*****************************************************************************
//<AMI_FHDR_START>
//
// Name: OemActivationHook.c
//
// Description: SW SMI hook.
//
//<AMI_FHDR_END>
//*****************************************************************************
#include <AmiDxeLib.h>
#include <Protocol/SmiFlash.h>
#include "OemActivation.h"
//----------------------------------------------------------------------------
#if OEM_ACTIVATION_TABLE_LOCATION == 1
//<AMI_PHDR_START>
//----------------------------------------------------------------------------
//
// Procedure: OemActivationUpdateBlockTypeId
//
// Description: This function is SW SMI hook that sets Flash Block Description
// type for AMI AFU utility. (EIP#58139)
//
// Input:
// SwSmiNum - SW SMI value number
// Buffer - Flash descriptor address
//
// Output: VOID
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
VOID OemActivationUpdateBlockTypeId (
IN UINT8 SwSmiNum,
IN UINT64 Buffer )
{
BLOCK_DESC *BlockDesc;
UINTN i;
// return if SW SMI value is not "Get Flash Info"
if (SwSmiNum != SMIFLASH_GET_FLASH_INFO)
return;
BlockDesc = (BLOCK_DESC*)&((INFO_BLOCK*)Buffer)->Blocks;
for (i = 0; i < ((INFO_BLOCK*)Buffer)->TotalBlocks; i++) {
TRACE((TRACE_ALWAYS,"OemActivationUpdateBlockTypeId: %08X(%08X), Block %08X\n",OEM_ACTIVATION_BLOCK_ADDRESS,OEM_ACTIVATION_BLOCK_END,BlockDesc[i].StartAddress));
if (BlockDesc[i].StartAddress < OEM_ACTIVATION_BLOCK_ADDRESS)
continue;
if (BlockDesc[i].StartAddress >= OEM_ACTIVATION_BLOCK_END)
continue;
TRACE((TRACE_ALWAYS,"OemActivationUpdateBlockTypeId: Found Blocks %08X\n",BlockDesc[i].StartAddress));
BlockDesc[i].Type = OA3_FLASH_BLOCK_DESC_TYPE;
}
}
#endif
//*************************************************************************
//*************************************************************************
//** **
//** (C)Copyright 1985-2011, American Megatrends, Inc. **
//** **
//** All Rights Reserved. **
//** **
//** 5555 Oakbrook Parkway, Suite 200, Norcross, GA 30093 **
//** **
//** Phone: (770)-246-8600 **
//** **
//*************************************************************************
//*************************************************************************
|
