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
|
//*************************************************************************
//*************************************************************************
//** **
//** (C)Copyright 1985-2013, American Megatrends, Inc. **
//** **
//** All Rights Reserved. **
//** **
//** 5555 Oakbrook Parkway, Suite 200, Norcross, GA 30093 **
//** **
//** Phone: (770)-246-8600 **
//** **
//*************************************************************************
//*************************************************************************
//*****************************************************************************
// $Header: /Alaska/SOURCE/Modules/SharkBayRefCodes/FIT/Smm/FitHook.c 1 7/03/13 10:06p Bensonlai $
//
// $Revision: 1 $
//
// $Date: 7/03/13 10:06p $
//*****************************************************************************
// Revision History
// ----------------
// $Log: /Alaska/SOURCE/Modules/SharkBayRefCodes/FIT/Smm/FitHook.c $
//
// 1 7/03/13 10:06p Bensonlai
// [TAG] EIP128151
// [Category] Improvement
// [Description] Implement an option(/b) to flash FV_DATA region uisng
// AFU flash utility.
// [Files] FitHook.cif
// FitHook.c
// FitHook.sdl
// FitHook.mak
//
//*****************************************************************************
//<AMI_FHDR_START>
//
// Name: FitHook.c
//
// Description: SW SMI hook.
//
//<AMI_FHDR_END>
//*****************************************************************************
#include <AmiDxeLib.h>
#include <Protocol/SmiFlash.h>
#include "token.h"
#define FLASH_DEVICE_BASE_ADDRESS (0xFFFFFFFF-FLASH_SIZE+1)
#define FV_DATA_BLOCK_ADDRESS (FV_DATA_BASE-FLASH_DEVICE_BASE_ADDRESS)
#define FV_DATA_BLOCK_END (FV_DATA_BLOCK_ADDRESS+FV_DATA_SIZE)
#define FV_DATA_TABLE_OFFSET (0xFFFFFFFF-FV_DATA_BASE+1)
//<AMI_PHDR_START>
//----------------------------------------------------------------------------
//
// Procedure: AppendFvDataToAFU_UpdateBlockTypeId
//
// 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 AppendFvDataToAFU_UpdateBlockTypeId (
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,"AppendFvDataToAFU_UpdateBlockTypeId: %08X(%08X), Block %08X\n", FV_DATA_BLOCK_ADDRESS, FV_DATA_BLOCK_END, BlockDesc[i].StartAddress));
if (BlockDesc[i].StartAddress < FV_DATA_BLOCK_ADDRESS)
continue;
if (BlockDesc[i].StartAddress >= FV_DATA_BLOCK_END)
continue;
TRACE((TRACE_ALWAYS,"AppendFvDataToAFU_UpdateBlockTypeId: Found Blocks %08X\n",BlockDesc[i].StartAddress));
BlockDesc[i].Type = BOOT_BLOCK;
}
}
//*************************************************************************
//*************************************************************************
//** **
//** (C)Copyright 1985-2013, American Megatrends, Inc. **
//** **
//** All Rights Reserved. **
//** **
//** 5555 Oakbrook Parkway, Suite 200, Norcross, GA 30093 **
//** **
//** Phone: (770)-246-8600 **
//** **
//*************************************************************************
//*************************************************************************
|