[mirror_edk2.git] / Tools / CCode / Source / GenBsfFixup / GenBsfFixup.c

/*++\r
\r
Copyright (c)  1999 - 2006 Intel Corporation. All rights reserved\r
This software and associated documentation (if any) is furnished\r
under a license and may only be used or copied in accordance\r
with the terms of the license. Except as permitted by such\r
license, no part of this software or documentation may be\r
reproduced, stored in a retrieval system, or transmitted in any\r
form or by any means without the express written consent of\r
Intel Corporation.\r
\r
\r
Module Name:\r
\r
  GenBsfFixup.c\r
\r
Abstract:\r
\r
  Utility to Fixup the SEC component for IA32.  This is an \r
  interim tool in place of the full GenBsfImage support for \r
  IA32.\r
  This tool supports the Synch3 update\r
\r
--*/\r
\r
#include "BaseTypes.h"\r
#include "UefiBaseTypes.h"\r
#include "EfiImage.h"\r
#include "FirmwareVolumeHeader.h"\r
#include "FirmwareVolumeImageFormat.h"\r
#include "ParseInf.h"\r
#include "CommonLib.h"\r
#include "FirmwareFileSystem.h"\r
#include "FvLib.h"\r
#include <stdio.h>\r
#include <stdlib.h>\r
#include <string.h>\r
#include <assert.h>\r
\r
//\r
// BugBug -- this port to the new FFS is really weird.\r
//           A lot of the file-header stuff has been ported, but\r
//           not the section information.\r
//\r
\r
#define UTILITY_NAME "GenBsfFixup"\r
#define UTILITY_MAJOR_VERSION 0\r
#define UTILITY_MINOR_VERSION 1\r
\r
UINT32  gFixup;\r
\r
UINT32\r
GetOccupiedSize (\r
  IN UINT32  ActualSize,\r
  IN UINT32  Alignment\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  ActualSize  - GC_TODO: add argument description\r
  Alignment   - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  UINT32  OccupiedSize;\r
\r
  OccupiedSize = ActualSize;\r
  while ((OccupiedSize & (Alignment - 1)) != 0) {\r
    OccupiedSize++;\r
  }\r
\r
  return OccupiedSize;\r
}\r
\r
static\r
void\r
Version (\r
  VOID\r
  )\r
{\r
  printf ("%s v%d.%d -EDK Utility to Fixup the SEC component for IA32.\n", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION);\r
  printf ("Copyright (c) 1999-2006 Intel Corporation. All rights reserved.\n");\r
}\r
\r
\r
VOID\r
Usage (\r
  VOID\r
  )\r
{\r
  Version();\r
  printf ("\nUsage: " UTILITY_NAME " FvVolumeImageFile AddressOfFvInMemory OffsetOfFixup OutputFileName \n");\r
}\r
\r
int\r
ReadHeader (\r
  FILE    *In,\r
  UINT32  *FvSize\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Reads in Firmware Volume information from the volume header file.\r
\r
Arguments:\r
\r
  In: Firmware Volume header file to read from\r
  FvSize: Size of Firmware Volume\r
\r
Returns:\r
\r
  int: Number of bytes read\r
\r
--*/\r
{\r
  EFI_FIRMWARE_VOLUME_HEADER  VolumeHeader;\r
  EFI_FV_BLOCK_MAP_ENTRY      BlockMap;\r
  INT32                       SigTemp[2];\r
  INT32                       Invert;\r
  INT32                       bytesread;\r
  UINT32                      size;\r
\r
  size      = 0;\r
  Invert    = 0;\r
  bytesread = 0;\r
\r
  if (In == NULL) {\r
    printf ("Error: Input file is NULL.\n");\r
    return -1;\r
  }\r
\r
  fread (&VolumeHeader, sizeof (EFI_FIRMWARE_VOLUME_HEADER) - sizeof (EFI_FV_BLOCK_MAP_ENTRY), 1, In);\r
  bytesread   = sizeof (EFI_FIRMWARE_VOLUME_HEADER) - sizeof (EFI_FV_BLOCK_MAP_ENTRY);\r
  SigTemp[0]  = VolumeHeader.Signature;\r
  SigTemp[1]  = 0;\r
\r
  if (VolumeHeader.Attributes & EFI_FVB_ERASE_POLARITY) {\r
    Invert = 1;\r
  }\r
\r
  do {\r
    fread (&BlockMap, sizeof (EFI_FV_BLOCK_MAP_ENTRY), 1, In);\r
    bytesread += sizeof (EFI_FV_BLOCK_MAP_ENTRY);\r
\r
    if (BlockMap.NumBlocks != 0) {\r
      size += BlockMap.NumBlocks * BlockMap.BlockLength;\r
    }\r
\r
  } while (BlockMap.NumBlocks != 0);\r
\r
  *FvSize = size;\r
  rewind (In);\r
\r
  if (Invert == 1) {\r
    bytesread *= -1;\r
  }\r
\r
  return bytesread;\r
}\r
\r
UINT32\r
GetSectionLength (\r
  IN  UINT32  *Length\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Converts a UINT8[3] array to a UINT32\r
\r
Arguments:\r
\r
  Length        A pointer to a 3 byte array\r
\r
Returns:\r
\r
  UINT32:       The length.\r
\r
--*/\r
{\r
  return *Length & 0x00FFFFFF;\r
}\r
\r
int\r
Readfile (\r
  UINT8   *FvImage,\r
  int     bytes,\r
  int     Invert\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  GC_TODO: Add function description\r
\r
Arguments:\r
\r
  FvImage - GC_TODO: add argument description\r
  bytes   - GC_TODO: add argument description\r
  Invert  - GC_TODO: add argument description\r
\r
Returns:\r
\r
  GC_TODO: add return values\r
\r
--*/\r
{\r
  UINT32                    FileLength;\r
  UINT32                    OccupiedFileLength;\r
  EFI_FFS_FILE_HEADER       *FileHeader;\r
  UINT8                     FileState;\r
  UINT8                     Checksum;\r
  UINT8                     *Ptr;\r
  UINT32                    SectionLength;\r
  EFI_COMMON_SECTION_HEADER *SectionHeader;\r
  EFI_IMAGE_NT_HEADERS      *PeHeader;\r
  UINT32                    PeiCoreOffset;\r
\r
  Ptr         = FvImage + bytes;\r
\r
  FileHeader  = (EFI_FFS_FILE_HEADER *) Ptr;\r
\r
  FileState   = GetFileState ((UINT8) Invert, FileHeader);\r
\r
  switch (FileState) {\r
  case EFI_FILE_HEADER_CONSTRUCTION:\r
  case EFI_FILE_HEADER_INVALID:\r
    return sizeof (EFI_FFS_FILE_HEADER);\r
\r
  case EFI_FILE_HEADER_VALID:\r
    Checksum  = CalculateSum8 ((UINT8 *) FileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
    Checksum  = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
    Checksum  = (UINT8) (Checksum - FileHeader->State);\r
    if (Checksum != 0) {\r
      return -1;\r
    }\r
    //\r
    // Now do the fixup stuff - begin\r
    //\r
    if (FileHeader->Type == EFI_FV_FILETYPE_PEI_CORE) {\r
      SectionHeader = (EFI_COMMON_SECTION_HEADER *) FileHeader + sizeof (EFI_FFS_FILE_HEADER);\r
      SectionLength = GetSectionLength ((UINT32 *) &SectionHeader->Size[0]);\r
\r
      printf ("Section length is 0x%X\n", SectionLength);\r
\r
      if (SectionHeader->Type == EFI_SECTION_PE32) {\r
\r
        gFixup    = bytes + sizeof (EFI_FFS_FILE_HEADER) + sizeof (EFI_COMMON_SECTION_HEADER);\r
\r
        PeHeader  = (EFI_IMAGE_NT_HEADERS *) Ptr + sizeof (EFI_FFS_FILE_HEADER) + sizeof (EFI_COMMON_SECTION_HEADER);\r
\r
        if (((EFI_IMAGE_DOS_HEADER *) PeHeader)->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r
          //\r
          // DOS image header is present, so read the PE header after the DOS image header\r
          //\r
          PeHeader = (EFI_IMAGE_NT_HEADERS *) ((UINTN) PeHeader + (UINTN) ((((EFI_IMAGE_DOS_HEADER *) PeHeader)->e_lfanew) & 0x0ffff));\r
\r
        }\r
\r
        PeiCoreOffset = (UINTN) ((UINTN) (PeHeader->OptionalHeader.AddressOfEntryPoint & 0x0ffffffff));\r
\r
        gFixup += PeiCoreOffset;\r
      }\r
    }\r
\r
    FileLength          = GetLength (FileHeader->Size);\r
    OccupiedFileLength  = GetOccupiedSize (FileLength, 8);\r
    return OccupiedFileLength;\r
\r
  case EFI_FILE_DATA_VALID:\r
    //\r
    // Calculate header checksum\r
    //\r
    Checksum  = CalculateSum8 ((UINT8 *) FileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
    Checksum  = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
    Checksum  = (UINT8) (Checksum - FileHeader->State);\r
    if (Checksum != 0) {\r
      return -1;\r
    }\r
    //\r
    // Determine file length\r
    //\r
    FileLength          = GetLength (FileHeader->Size);\r
    OccupiedFileLength  = GetOccupiedSize (FileLength, 8);\r
\r
    //\r
    // Determine if file checksum is valid or fixed\r
    //\r
    if (FileHeader->Attributes & FFS_ATTRIB_CHECKSUM) {\r
      Checksum  = CalculateSum8 (Ptr, FileLength);\r
      Checksum  = (UINT8) (Checksum - FileHeader->State);\r
      if (Checksum != 0) {\r
        return -1;\r
      }\r
    } else {\r
      if (FileHeader->IntegrityCheck.Checksum.File != FFS_FIXED_CHECKSUM) {\r
        return -1;\r
      }\r
    }\r
    break;\r
\r
  case EFI_FILE_MARKED_FOR_UPDATE:\r
  case EFI_FILE_DELETED:\r
    //\r
    // Calculate header checksum\r
    //\r
    Checksum  = CalculateSum8 ((UINT8 *) FileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
    Checksum  = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
    Checksum  = (UINT8) (Checksum - FileHeader->State);\r
    if (Checksum != 0) {\r
      return -1;\r
    }\r
    //\r
    // Determine file length\r
    //\r
    FileLength          = GetLength (FileHeader->Size);\r
    OccupiedFileLength  = GetOccupiedSize (FileLength, 8);\r
\r
    //\r
    // Determine if file checksum is valid or fixed\r
    //\r
    if (FileHeader->Attributes & FFS_ATTRIB_CHECKSUM) {\r
      Checksum  = CalculateSum8 (Ptr, FileLength);\r
      Checksum  = (UINT8) (Checksum - FileHeader->State);\r
      if (Checksum != 0) {\r
        return -1;\r
      }\r
    } else {\r
      if (FileHeader->IntegrityCheck.Checksum.File != FFS_FIXED_CHECKSUM) {\r
        return -1;\r
      }\r
    }\r
\r
    return OccupiedFileLength;\r
\r
  default:\r
    return sizeof (EFI_FFS_FILE_HEADER);\r
  }\r
\r
  return OccupiedFileLength;\r
}\r
\r
int\r
main (\r
  int argc,\r
  char*argv[]\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Runs GenBsfFixup\r
\r
Arguments:\r
\r
  argc: number of command line arguments\r
\r
  arg[0] = This file name\r
  arg[1] = Firmware Volume Name\r
  arg[2] = Base Address to relocate\r
  arg[3] = Relative offset of the fixup to perform\r
  arg[4] = Output File Name\r
\r
Returns:\r
    \r
  int: 0 code success, -1 code failure\r
\r
--*/\r
{\r
  FILE        *In;\r
  FILE        *Out;\r
  int         ByteStart;\r
  int         Invert;\r
  int         Index;\r
  int         cnt;\r
  UINT8       *FvImage;\r
  int         ByteRead;\r
  UINT32      FvSize;\r
  UINT64      delta;\r
  UINT32      Idx;\r
  UINT64      FvOffset;\r
  EFI_STATUS  Status;\r
\r
  Index   = 0;\r
  Invert  = 0;\r
 \r
  if (argc == 1) {\r
    Usage();\r
    return -1;\r
  }\r
  \r
  if ((strcmp(argv[1], "-h") == 0) || (strcmp(argv[1], "--help") == 0) ||\r
      (strcmp(argv[1], "-?") == 0) || (strcmp(argv[1], "/?") == 0)) {\r
    Usage();\r
    return -1;\r
  }\r
  \r
  if ((strcmp(argv[1], "-V") == 0) || (strcmp(argv[1], "--version") == 0)) {\r
    Version();\r
    return -1;\r
  }\r
 \r
  if (argc != 5) {\r
    Usage();\r
    return -1;\r
  }\r
\r
  In = fopen (argv[1], "rb");\r
\r
  if (In == NULL) {\r
    printf ("Unable to open FV image file \"%s\"\n", argv[1]);\r
    return -1;\r
  }\r
\r
  ByteStart = ReadHeader (In, &FvSize);\r
\r
  if (ByteStart < 0) {\r
    Invert = 1;\r
    ByteStart *= -1;\r
  }\r
\r
  FvImage = malloc (FvSize);\r
  if (FvImage == NULL) {\r
    printf ("Cannot allocate memory\n");\r
    fclose (In);\r
    return -1;\r
  }\r
\r
  ByteRead = fread (FvImage, 1, FvSize, In);\r
\r
  if ((unsigned int) ByteRead != FvSize) {\r
    printf ("Read File error\n");\r
    fclose (In);\r
    return -1;\r
  }\r
\r
  cnt = 0;\r
  while ((unsigned int) ByteStart < FvSize && cnt != -1) {\r
    cnt = Readfile (FvImage, ByteStart, Invert);\r
\r
    if (cnt != -1) {\r
      ByteStart += cnt;\r
    }\r
\r
    if (cnt != sizeof (EFI_FFS_FILE_HEADER)) {\r
      Index++;\r
    }\r
  }\r
\r
  if (cnt == -1) {\r
    printf ("Firmware Volume image corrupted\n");\r
    return -1;\r
  }\r
\r
  fclose (In);\r
\r
  Out = fopen (argv[4], "wb");\r
\r
  if (Out == NULL) {\r
    printf ("Unable to open FV image file \"%s\"\n", argv[4]);\r
    return -1;\r
  }\r
\r
  In = fopen (argv[1], "rb");\r
\r
  if (In == NULL) {\r
    printf ("Unable to open FV image file \"%s\"\n", argv[1]);\r
    return -1;\r
  }\r
\r
  if (gFixup != 0) {\r
\r
    printf ("Fixup of 0x%X\n", gFixup);\r
\r
    Status = AsciiStringToUint64 (argv[2], TRUE, &FvOffset);\r
\r
    gFixup += (UINT32) FvOffset;\r
\r
    ByteStart = ReadHeader (In, &FvSize);\r
\r
    Readfile (FvImage, ByteStart, Invert);\r
\r
    cnt     = 0;\r
    Status  = AsciiStringToUint64 (argv[3], TRUE, &delta);\r
\r
    fclose (In);\r
    In = fopen (argv[1], "rb");\r
\r
    if (In == NULL) {\r
      printf ("Unable to open FV image file \"%s\"\n", argv[1]);\r
      return -1;\r
    }\r
\r
    for (Idx = 0; Idx < delta - FvOffset; Idx++) {\r
      fputc (fgetc (In), Out);\r
    }\r
\r
    fwrite (&gFixup, sizeof (UINT32), 1, Out);\r
    fseek (In, sizeof (UINT32), SEEK_CUR);\r
\r
    for (Idx = 0; Idx < FvSize - (delta - FvOffset) - sizeof (UINT32); Idx++) {\r
      fputc (fgetc (In), Out);\r
    }\r
\r
    fclose (In);\r
    fclose (Out);\r
  } else {\r
    printf ("There was no fixup to perform\n");\r
  }\r
\r
  free (FvImage);\r
\r
  return 0;\r
}\r
Commit	Line	Data
71ea530c	1	/*++\r
71ea530c	2	\r
f091efb3	3	Copyright (c) 1999 - 2006 Intel Corporation. All rights reserved\r
71ea530c	4	This software and associated documentation (if any) is furnished\r
	5	under a license and may only be used or copied in accordance\r
	6	with the terms of the license. Except as permitted by such\r
	7	license, no part of this software or documentation may be\r
	8	reproduced, stored in a retrieval system, or transmitted in any\r
	9	form or by any means without the express written consent of\r
	10	Intel Corporation.\r
	11	\r
	12	\r
	13	Module Name:\r
	14	\r
	15	GenBsfFixup.c\r
	16	\r
	17	Abstract:\r
	18	\r
	19	Utility to Fixup the SEC component for IA32. This is an \r
	20	interim tool in place of the full GenBsfImage support for \r
	21	IA32.\r
	22	This tool supports the Synch3 update\r
	23	\r
	24	--*/\r
	25	\r
	26	#include "BaseTypes.h"\r
	27	#include "UefiBaseTypes.h"\r
	28	#include "EfiImage.h"\r
	29	#include "FirmwareVolumeHeader.h"\r
	30	#include "FirmwareVolumeImageFormat.h"\r
	31	#include "ParseInf.h"\r
	32	#include "CommonLib.h"\r
	33	#include "FirmwareFileSystem.h"\r
	34	#include "FvLib.h"\r
	35	#include <stdio.h>\r
	36	#include <stdlib.h>\r
	37	#include <string.h>\r
	38	#include <assert.h>\r
	39	\r
	40	//\r
	41	// BugBug -- this port to the new FFS is really weird.\r
	42	// A lot of the file-header stuff has been ported, but\r
	43	// not the section information.\r
	44	//\r
f091efb3	45	\r
	46	#define UTILITY_NAME "GenBsfFixup"\r
	47	#define UTILITY_MAJOR_VERSION 0\r
	48	#define UTILITY_MINOR_VERSION 1\r
71ea530c	49	\r
	50	UINT32 gFixup;\r
	51	\r
	52	UINT32\r
	53	GetOccupiedSize (\r
	54	IN UINT32 ActualSize,\r
	55	IN UINT32 Alignment\r
	56	)\r
	57	/*++\r
	58	\r
	59	Routine Description:\r
	60	\r
	61	GC_TODO: Add function description\r
	62	\r
	63	Arguments:\r
	64	\r
	65	ActualSize - GC_TODO: add argument description\r
	66	Alignment - GC_TODO: add argument description\r
	67	\r
	68	Returns:\r
	69	\r
	70	GC_TODO: add return values\r
	71	\r
	72	--*/\r
	73	{\r
	74	UINT32 OccupiedSize;\r
	75	\r
	76	OccupiedSize = ActualSize;\r
	77	while ((OccupiedSize & (Alignment - 1)) != 0) {\r
	78	OccupiedSize++;\r
	79	}\r
	80	\r
	81	return OccupiedSize;\r
	82	}\r
	83	\r
f091efb3	84	static\r
	85	void\r
	86	Version (\r
	87	VOID\r
	88	)\r
	89	{\r
	90	printf ("%s v%d.%d -EDK Utility to Fixup the SEC component for IA32.\n", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION);\r
	91	printf ("Copyright (c) 1999-2006 Intel Corporation. All rights reserved.\n");\r
	92	}\r
	93	\r
	94	\r
	95	VOID\r
	96	Usage (\r
	97	VOID\r
	98	)\r
	99	{\r
	100	Version();\r
	101	printf ("\nUsage: " UTILITY_NAME " FvVolumeImageFile AddressOfFvInMemory OffsetOfFixup OutputFileName \n");\r
	102	}\r
	103	\r
71ea530c	104	int\r
	105	ReadHeader (\r
	106	FILE *In,\r
	107	UINT32 *FvSize\r
	108	)\r
	109	/*++\r
	110	\r
	111	Routine Description:\r
	112	\r
	113	Reads in Firmware Volume information from the volume header file.\r
	114	\r
	115	Arguments:\r
	116	\r
	117	In: Firmware Volume header file to read from\r
	118	FvSize: Size of Firmware Volume\r
	119	\r
	120	Returns:\r
	121	\r
	122	int: Number of bytes read\r
	123	\r
	124	--*/\r
	125	{\r
	126	EFI_FIRMWARE_VOLUME_HEADER VolumeHeader;\r
	127	EFI_FV_BLOCK_MAP_ENTRY BlockMap;\r
	128	INT32 SigTemp[2];\r
	129	INT32 Invert;\r
	130	INT32 bytesread;\r
	131	UINT32 size;\r
	132	\r
	133	size = 0;\r
	134	Invert = 0;\r
	135	bytesread = 0;\r
	136	\r
	137	if (In == NULL) {\r
	138	printf ("Error: Input file is NULL.\n");\r
	139	return -1;\r
	140	}\r
	141	\r
	142	fread (&VolumeHeader, sizeof (EFI_FIRMWARE_VOLUME_HEADER) - sizeof (EFI_FV_BLOCK_MAP_ENTRY), 1, In);\r
	143	bytesread = sizeof (EFI_FIRMWARE_VOLUME_HEADER) - sizeof (EFI_FV_BLOCK_MAP_ENTRY);\r
	144	SigTemp[0] = VolumeHeader.Signature;\r
	145	SigTemp[1] = 0;\r
	146	\r
	147	if (VolumeHeader.Attributes & EFI_FVB_ERASE_POLARITY) {\r
	148	Invert = 1;\r
	149	}\r
	150	\r
	151	do {\r
	152	fread (&BlockMap, sizeof (EFI_FV_BLOCK_MAP_ENTRY), 1, In);\r
	153	bytesread += sizeof (EFI_FV_BLOCK_MAP_ENTRY);\r
	154	\r
	155	if (BlockMap.NumBlocks != 0) {\r
	156	size += BlockMap.NumBlocks * BlockMap.BlockLength;\r
	157	}\r
	158	\r
	159	} while (BlockMap.NumBlocks != 0);\r
	160	\r
	161	*FvSize = size;\r
	162	rewind (In);\r
	163	\r
	164	if (Invert == 1) {\r
	165	bytesread *= -1;\r
	166	}\r
	167	\r
168	return bytesread;\r
169	}\r
170	\r
171	UINT32\r
172	GetSectionLength (\r
173	IN UINT32 *Length\r
174	)\r
175	/*++\r
176	\r
177	Routine Description:\r
178	\r
179	Converts a UINT8[3] array to a UINT32\r
180	\r
181	Arguments:\r
182	\r
183	Length A pointer to a 3 byte array\r
184	\r
185	Returns:\r
186	\r
187	UINT32: The length.\r
188	\r
189	--*/\r
190	{\r
191	return *Length & 0x00FFFFFF;\r
192	}\r
193	\r
194	int\r
195	Readfile (\r
196	UINT8 *FvImage,\r
197	int bytes,\r
198	int Invert\r
199	)\r
200	/*++\r
201	\r
202	Routine Description:\r
203	\r
204	GC_TODO: Add function description\r
205	\r
206	Arguments:\r
207	\r
208	FvImage - GC_TODO: add argument description\r
209	bytes - GC_TODO: add argument description\r
210	Invert - GC_TODO: add argument description\r
211	\r
212	Returns:\r
213	\r
214	GC_TODO: add return values\r
215	\r
216	--*/\r
217	{\r
218	UINT32 FileLength;\r
219	UINT32 OccupiedFileLength;\r
220	EFI_FFS_FILE_HEADER *FileHeader;\r
221	UINT8 FileState;\r
222	UINT8 Checksum;\r
223	UINT8 *Ptr;\r
224	UINT32 SectionLength;\r
225	EFI_COMMON_SECTION_HEADER *SectionHeader;\r
226	EFI_IMAGE_NT_HEADERS *PeHeader;\r
227	UINT32 PeiCoreOffset;\r
228	\r
229	Ptr = FvImage + bytes;\r
230	\r
231	FileHeader = (EFI_FFS_FILE_HEADER *) Ptr;\r
232	\r
233	FileState = GetFileState ((UINT8) Invert, FileHeader);\r
234	\r
235	switch (FileState) {\r
236	case EFI_FILE_HEADER_CONSTRUCTION:\r
237	case EFI_FILE_HEADER_INVALID:\r
238	return sizeof (EFI_FFS_FILE_HEADER);\r
239	\r
240	case EFI_FILE_HEADER_VALID:\r
241	Checksum = CalculateSum8 ((UINT8 *) FileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
242	Checksum = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
243	Checksum = (UINT8) (Checksum - FileHeader->State);\r
244	if (Checksum != 0) {\r
245	return -1;\r
246	}\r
247	//\r
248	// Now do the fixup stuff - begin\r
249	//\r
250	if (FileHeader->Type == EFI_FV_FILETYPE_PEI_CORE) {\r
251	SectionHeader = (EFI_COMMON_SECTION_HEADER *) FileHeader + sizeof (EFI_FFS_FILE_HEADER);\r
252	SectionLength = GetSectionLength ((UINT32 *) &SectionHeader->Size[0]);\r
253	\r
254	printf ("Section length is 0x%X\n", SectionLength);\r
255	\r
256	if (SectionHeader->Type == EFI_SECTION_PE32) {\r
257	\r
258	gFixup = bytes + sizeof (EFI_FFS_FILE_HEADER) + sizeof (EFI_COMMON_SECTION_HEADER);\r
259	\r
260	PeHeader = (EFI_IMAGE_NT_HEADERS *) Ptr + sizeof (EFI_FFS_FILE_HEADER) + sizeof (EFI_COMMON_SECTION_HEADER);\r
261	\r
262	if (((EFI_IMAGE_DOS_HEADER *) PeHeader)->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r
263	//\r
264	// DOS image header is present, so read the PE header after the DOS image header\r
265	//\r
266	PeHeader = (EFI_IMAGE_NT_HEADERS ) ((UINTN) PeHeader + (UINTN) ((((EFI_IMAGE_DOS_HEADER ) PeHeader)->e_lfanew) & 0x0ffff));\r
267	\r
268	}\r
269	\r
270	PeiCoreOffset = (UINTN) ((UINTN) (PeHeader->OptionalHeader.AddressOfEntryPoint & 0x0ffffffff));\r
271	\r
272	gFixup += PeiCoreOffset;\r
273	}\r
274	}\r
275	\r
276	FileLength = GetLength (FileHeader->Size);\r
277	OccupiedFileLength = GetOccupiedSize (FileLength, 8);\r
278	return OccupiedFileLength;\r
279	\r
280	case EFI_FILE_DATA_VALID:\r
281	//\r
282	// Calculate header checksum\r
283	//\r
284	Checksum = CalculateSum8 ((UINT8 *) FileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
285	Checksum = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
286	Checksum = (UINT8) (Checksum - FileHeader->State);\r
287	if (Checksum != 0) {\r
288	return -1;\r
289	}\r
290	//\r
291	// Determine file length\r
292	//\r
293	FileLength = GetLength (FileHeader->Size);\r
294	OccupiedFileLength = GetOccupiedSize (FileLength, 8);\r
295	\r
296	//\r
297	// Determine if file checksum is valid or fixed\r
298	//\r
299	if (FileHeader->Attributes & FFS_ATTRIB_CHECKSUM) {\r
300	Checksum = CalculateSum8 (Ptr, FileLength);\r
301	Checksum = (UINT8) (Checksum - FileHeader->State);\r
302	if (Checksum != 0) {\r
303	return -1;\r
304	}\r
305	} else {\r
306	if (FileHeader->IntegrityCheck.Checksum.File != FFS_FIXED_CHECKSUM) {\r
307	return -1;\r
308	}\r
309	}\r
310	break;\r
311	\r
312	case EFI_FILE_MARKED_FOR_UPDATE:\r
313	case EFI_FILE_DELETED:\r
314	//\r
315	// Calculate header checksum\r
316	//\r
317	Checksum = CalculateSum8 ((UINT8 *) FileHeader, sizeof (EFI_FFS_FILE_HEADER));\r
318	Checksum = (UINT8) (Checksum - FileHeader->IntegrityCheck.Checksum.File);\r
319	Checksum = (UINT8) (Checksum - FileHeader->State);\r
320	if (Checksum != 0) {\r
321	return -1;\r
322	}\r
323	//\r
324	// Determine file length\r
325	//\r
326	FileLength = GetLength (FileHeader->Size);\r
327	OccupiedFileLength = GetOccupiedSize (FileLength, 8);\r
328	\r
329	//\r
330	// Determine if file checksum is valid or fixed\r
331	//\r
332	if (FileHeader->Attributes & FFS_ATTRIB_CHECKSUM) {\r
333	Checksum = CalculateSum8 (Ptr, FileLength);\r
334	Checksum = (UINT8) (Checksum - FileHeader->State);\r
335	if (Checksum != 0) {\r
336	return -1;\r
337	}\r
338	} else {\r
339	if (FileHeader->IntegrityCheck.Checksum.File != FFS_FIXED_CHECKSUM) {\r
340	return -1;\r
341	}\r
342	}\r
343	\r
344	return OccupiedFileLength;\r
345	\r
346	default:\r
347	return sizeof (EFI_FFS_FILE_HEADER);\r
348	}\r
349	\r
350	return OccupiedFileLength;\r
351	}\r
352	\r
353	int\r
354	main (\r
355	int argc,\r
356	char*argv[]\r
357	)\r
358	/*++\r
359	\r
360	Routine Description:\r
361	\r
362	Runs GenBsfFixup\r
363	\r
364	Arguments:\r
365	\r
366	argc: number of command line arguments\r
367	\r
368	arg[0] = This file name\r
369	arg[1] = Firmware Volume Name\r
370	arg[2] = Base Address to relocate\r
371	arg[3] = Relative offset of the fixup to perform\r
372	arg[4] = Output File Name\r
373	\r
374	Returns:\r
375	\r
376	int: 0 code success, -1 code failure\r
377	\r
378	--*/\r
379	{\r
380	FILE *In;\r
381	FILE *Out;\r
382	int ByteStart;\r
383	int Invert;\r
384	int Index;\r
385	int cnt;\r
386	UINT8 *FvImage;\r
387	int ByteRead;\r
388	UINT32 FvSize;\r
389	UINT64 delta;\r
390	UINT32 Idx;\r
391	UINT64 FvOffset;\r
392	EFI_STATUS Status;\r
393	\r
394	Index = 0;\r
395	Invert = 0;\r
f091efb3	396	\r
db608e6b	397	if (argc == 1) {\r
f091efb3	398	Usage();\r
	399	return -1;\r
	400	}\r
	401	\r
	402	if ((strcmp(argv[1], "-h") == 0) \|\| (strcmp(argv[1], "--help") == 0) \|\|\r
	403	(strcmp(argv[1], "-?") == 0) \|\| (strcmp(argv[1], "/?") == 0)) {\r
	404	Usage();\r
	405	return -1;\r
	406	}\r
	407	\r
	408	if ((strcmp(argv[1], "-V") == 0) \|\| (strcmp(argv[1], "--version") == 0)) {\r
	409	Version();\r
	410	return -1;\r
	411	}\r
	412	\r
71ea530c	413	if (argc != 5) {\r
f091efb3	414	Usage();\r
71ea530c	415	return -1;\r
	416	}\r
	417	\r
	418	In = fopen (argv[1], "rb");\r
	419	\r
	420	if (In == NULL) {\r
	421	printf ("Unable to open FV image file \"%s\"\n", argv[1]);\r
	422	return -1;\r
	423	}\r
	424	\r
	425	ByteStart = ReadHeader (In, &FvSize);\r
	426	\r
	427	if (ByteStart < 0) {\r
	428	Invert = 1;\r
	429	ByteStart *= -1;\r
	430	}\r
	431	\r
	432	FvImage = malloc (FvSize);\r
	433	if (FvImage == NULL) {\r
	434	printf ("Cannot allocate memory\n");\r
	435	fclose (In);\r
	436	return -1;\r
	437	}\r
	438	\r
	439	ByteRead = fread (FvImage, 1, FvSize, In);\r
	440	\r
	441	if ((unsigned int) ByteRead != FvSize) {\r
	442	printf ("Read File error\n");\r
	443	fclose (In);\r
	444	return -1;\r
	445	}\r
	446	\r
	447	cnt = 0;\r
	448	while ((unsigned int) ByteStart < FvSize && cnt != -1) {\r
	449	cnt = Readfile (FvImage, ByteStart, Invert);\r
	450	\r
	451	if (cnt != -1) {\r
	452	ByteStart += cnt;\r
	453	}\r
	454	\r
	455	if (cnt != sizeof (EFI_FFS_FILE_HEADER)) {\r
	456	Index++;\r
	457	}\r
	458	}\r
	459	\r
	460	if (cnt == -1) {\r
	461	printf ("Firmware Volume image corrupted\n");\r
	462	return -1;\r
	463	}\r
	464	\r
	465	fclose (In);\r
	466	\r
	467	Out = fopen (argv[4], "wb");\r
	468	\r
	469	if (Out == NULL) {\r
	470	printf ("Unable to open FV image file \"%s\"\n", argv[4]);\r
	471	return -1;\r
	472	}\r
	473	\r
	474	In = fopen (argv[1], "rb");\r
	475	\r
	476	if (In == NULL) {\r
	477	printf ("Unable to open FV image file \"%s\"\n", argv[1]);\r
	478	return -1;\r
479	}\r
480	\r
481	if (gFixup != 0) {\r
482	\r
483	printf ("Fixup of 0x%X\n", gFixup);\r
484	\r
485	Status = AsciiStringToUint64 (argv[2], TRUE, &FvOffset);\r
486	\r
487	gFixup += (UINT32) FvOffset;\r
488	\r
489	ByteStart = ReadHeader (In, &FvSize);\r
490	\r
491	Readfile (FvImage, ByteStart, Invert);\r
492	\r
493	cnt = 0;\r
494	Status = AsciiStringToUint64 (argv[3], TRUE, &delta);\r
495	\r
496	fclose (In);\r
497	In = fopen (argv[1], "rb");\r
498	\r
499	if (In == NULL) {\r
500	printf ("Unable to open FV image file \"%s\"\n", argv[1]);\r
501	return -1;\r
502	}\r
503	\r
504	for (Idx = 0; Idx < delta - FvOffset; Idx++) {\r
505	fputc (fgetc (In), Out);\r
506	}\r
507	\r
508	fwrite (&gFixup, sizeof (UINT32), 1, Out);\r
509	fseek (In, sizeof (UINT32), SEEK_CUR);\r
510	\r
511	for (Idx = 0; Idx < FvSize - (delta - FvOffset) - sizeof (UINT32); Idx++) {\r
512	fputc (fgetc (In), Out);\r
513	}\r
514	\r
515	fclose (In);\r
516	fclose (Out);\r
517	} else {\r
518	printf ("There was no fixup to perform\n");\r
519	}\r
520	\r
521	free (FvImage);\r
522	\r
523	return 0;\r
524	}\r