[mirror_edk2.git] / BaseTools / Source / C / EfiLdrImage / EfiLdrImage.c

/** @file\r
\r
Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials                          \r
are licensed and made available under the terms and conditions of the BSD License         \r
which accompanies this distribution.  The full text of the license may be found at        \r
http://opensource.org/licenses/bsd-license.php                                            \r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     \r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             \r
\r
Module Name:\r
\r
  efildrimage.c\r
\r
Abstract:\r
\r
  Creates and EFILDR image.\r
  This tool combines several PE Image files together using following format denoted as EBNF:\r
  FILE := EFILDR_HEADER\r
          EFILDR_IMAGE +\r
          <PeImageFileContent> +\r
  The order of EFILDR_IMAGE is same as the order of placing PeImageFileContent.\r
\r
Revision History\r
\r
**/\r
\r
\r
#include <stdio.h>\r
#include <stdlib.h>\r
#include <string.h>\r
#include "ParseInf.h"\r
#include "CommonLib.h"\r
#include "EfiUtilityMsgs.h"\r
\r
#define MAX_PE_IMAGES                  63\r
#define FILE_TYPE_FIXED_LOADER         0\r
#define FILE_TYPE_RELOCATABLE_PE_IMAGE 1\r
\r
typedef struct {\r
  UINT32 CheckSum;\r
  UINT32 Offset;\r
  UINT32 Length;\r
  UINT8  FileName[52];\r
} EFILDR_IMAGE;\r
\r
typedef struct {          \r
  UINT32       Signature;     \r
  UINT32       HeaderCheckSum;\r
  UINT32       FileLength;\r
  UINT32       NumberOfImages;\r
} EFILDR_HEADER;\r
\r
//\r
// Utility Name\r
//\r
#define UTILITY_NAME  "EfiLdrImage"\r
\r
//\r
// Utility version information\r
//\r
#define UTILITY_MAJOR_VERSION 0\r
#define UTILITY_MINOR_VERSION 1\r
\r
void\r
Version (\r
  void\r
  )\r
/*++\r
\r
Routine Description:\r
\r
  Displays the standard utility information to SDTOUT\r
\r
Arguments:\r
\r
  None\r
\r
Returns:\r
\r
  None\r
\r
--*/\r
{\r
  printf ("%s v%d.%d %s -Utility to break a file into two pieces at the request offset.\n", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION, __BUILD_VERSION);\r
  printf ("Copyright (c) 1999-2010 Intel Corporation. All rights reserved.\n");\r
}\r
\r
VOID\r
Usage (\r
  VOID\r
  )\r
{\r
  printf ("Usage: EfiLdrImage -o OutImage LoaderImage PeImage1 PeImage2 ... PeImageN\n");\r
  exit (1);\r
}\r
\r
EFI_STATUS\r
CountVerboseLevel (\r
  IN CONST CHAR8* VerboseLevelString,\r
  IN CONST UINT64 Length,\r
  OUT UINT64 *ReturnValue\r
)\r
{\r
  UINT64 i = 0;\r
  for (;i < Length; ++i) {\r
    if (VerboseLevelString[i] != 'v' && VerboseLevelString[i] != 'V') {\r
      return EFI_ABORTED;\r
    }\r
    ++(*ReturnValue);\r
  }\r
  \r
  return EFI_SUCCESS;\r
}\r
\r
UINT64\r
FCopyFile (\r
  FILE    *in,\r
  FILE    *out\r
  )\r
/*++\r
Routine Description:\r
  Write all the content of input file to output file.\r
\r
Arguments:\r
  in  - input file pointer\r
  out - output file pointer\r
\r
Return:\r
  UINT64 : file size of input file\r
--*/\r
{\r
  UINT32          filesize, offset, length;\r
  CHAR8           Buffer[8*1024];\r
\r
  fseek (in, 0, SEEK_END);\r
  filesize = ftell(in);\r
\r
  fseek (in, 0, SEEK_SET);\r
\r
  offset = 0;\r
  while (offset < filesize)  {\r
    length = sizeof(Buffer);\r
    if (filesize-offset < length) {\r
      length = filesize-offset;\r
    }\r
\r
    fread (Buffer, length, 1, in);\r
    fwrite (Buffer, length, 1, out);\r
    offset += length;\r
  }\r
\r
  return filesize;\r
}\r
\r
\r
int\r
main (\r
  int argc,\r
  char *argv[]\r
  )\r
/*++\r
\r
Routine Description:\r
\r
\r
Arguments:\r
\r
\r
Returns:\r
\r
\r
--*/\r
{\r
  UINT64         i;\r
  UINT64         filesize;\r
  FILE          *fpIn, *fpOut;\r
  EFILDR_HEADER EfiLdrHeader;\r
  EFILDR_IMAGE  EfiLdrImage[MAX_PE_IMAGES];\r
  CHAR8* OutputFileName = NULL;\r
  CHAR8* InputFileNames[MAX_PE_IMAGES + 1];\r
  UINT8 InputFileCount = 0;\r
  BOOLEAN QuietFlag = FALSE;\r
  UINT64        DebugLevel = 0;\r
  UINT64        VerboseLevel = 0;\r
  EFI_STATUS Status = EFI_SUCCESS;\r
  \r
  SetUtilityName (UTILITY_NAME);\r
\r
  if (argc == 1) {\r
    Usage();\r
    return STATUS_ERROR;\r
  }\r
  \r
  argc --;\r
  argv ++;\r
\r
  if ((stricmp (argv[0], "-h") == 0) || (stricmp (argv[0], "--help") == 0)) {\r
    Usage();\r
    return STATUS_SUCCESS;    \r
  }\r
\r
  if (stricmp (argv[0], "--version") == 0) {\r
    Version();\r
    return STATUS_SUCCESS;    \r
  }\r
\r
  while (argc > 0) {\r
   \r
    if ((stricmp (argv[0], "-o") == 0) || (stricmp (argv[0], "--output") == 0)) {\r
      OutputFileName = argv[1];\r
      if (OutputFileName == NULL) {\r
        Error (NULL, 0, 1003, "Invalid option value", "Output file can't be null");\r
        return STATUS_ERROR;\r
      }\r
      argc -= 2;\r
      argv += 2;\r
      continue; \r
    }\r
    \r
    if ((stricmp (argv[0], "-q") == 0) || (stricmp (argv[0], "--quiet") == 0)) {\r
      QuietFlag = TRUE;\r
      argc --;\r
      argv ++;\r
      continue; \r
    }\r
    \r
    if ((strlen(argv[0]) >= 2 && argv[0][0] == '-' && (argv[0][1] == 'v' || argv[0][1] == 'V')) || (stricmp (argv[0], "--verbose") == 0)) {\r
      VerboseLevel = 1;\r
      if (strlen(argv[0]) > 2) {\r
        Status = CountVerboseLevel (&argv[0][2], strlen(argv[0]) - 2, &VerboseLevel);\r
        if (EFI_ERROR (Status)) {\r
          Error (NULL, 0, 1003, "Invalid option value", argv[0]);\r
          return STATUS_ERROR;        \r
        }\r
      }\r
      \r
      argc --;\r
      argv ++;\r
      continue; \r
    }\r
    \r
    if ((stricmp (argv[0], "-d") == 0) || (stricmp (argv[0], "--debug") == 0)) {\r
      Status = AsciiStringToUint64 (argv[1], FALSE, &DebugLevel);\r
      if (EFI_ERROR (Status)) {\r
        Error (NULL, 0, 1003, "Invalid option value", "%s = %s", argv[0], argv[1]);\r
        return STATUS_ERROR;        \r
      }\r
      argc -= 2;\r
      argv += 2;\r
      continue; \r
    }\r
    //\r
    // Don't recognize the parameter, should be regarded as the input file name.\r
    //\r
    InputFileNames[InputFileCount] = argv[0];\r
    InputFileCount++;\r
    argc--;\r
    argv++;\r
  }\r
\r
  if (InputFileCount == 0) {\r
    Error (NULL, 0, 1001, "Missing option", "No input file");\r
    return STATUS_ERROR;\r
  }\r
  //\r
  // Open output file for write\r
  //\r
  if (OutputFileName == NULL) {\r
    Error (NULL, 0, 1001, "Missing option", "No output file");\r
    return STATUS_ERROR;\r
  }\r
\r
  fpOut = fopen(OutputFileName, "w+b");\r
  if (!fpOut) {\r
    Error (NULL, 0, 0001, "Could not open output file", OutputFileName);\r
    return STATUS_ERROR;\r
  }\r
\r
  memset (&EfiLdrHeader, 0, sizeof (EfiLdrHeader));\r
  memset (&EfiLdrImage, 0, sizeof (EFILDR_IMAGE) * (InputFileCount));\r
\r
  memcpy (&EfiLdrHeader.Signature, "EFIL", 4);\r
  EfiLdrHeader.FileLength = sizeof(EFILDR_HEADER) + sizeof(EFILDR_IMAGE)*(InputFileCount);\r
\r
  //\r
  // Skip the file header first\r
  //\r
  fseek (fpOut, EfiLdrHeader.FileLength, SEEK_SET);\r
\r
  //\r
  // copy all the input files to the output file\r
  //\r
  for(i=0;i<InputFileCount;i++) {\r
    //\r
    // Copy the content of PeImage file to output file\r
    //\r
    fpIn = fopen (InputFileNames[i], "rb");\r
    if (!fpIn) {\r
      Error (NULL, 0, 0001, "Could not open input file", InputFileNames[i]);\r
      fclose (fpOut);\r
      return STATUS_ERROR;\r
    }\r
    filesize = FCopyFile (fpIn, fpOut);\r
    fclose(fpIn);\r
\r
    //\r
    //  And in the same time update the EfiLdrHeader and EfiLdrImage array\r
    //\r
    EfiLdrImage[i].Offset = EfiLdrHeader.FileLength;\r
    EfiLdrImage[i].Length = (UINT32) filesize;\r
    strncpy ((CHAR8*) EfiLdrImage[i].FileName, InputFileNames[i], sizeof (EfiLdrImage[i].FileName) - 1);\r
    EfiLdrHeader.FileLength += (UINT32) filesize;\r
    EfiLdrHeader.NumberOfImages++;\r
  }\r
\r
  //\r
  // Write the image header to the output file finally\r
  //\r
  fseek (fpOut, 0, SEEK_SET);\r
  fwrite (&EfiLdrHeader, sizeof(EFILDR_HEADER)        , 1, fpOut);\r
  fwrite (&EfiLdrImage , sizeof(EFILDR_IMAGE)*(InputFileCount), 1, fpOut);\r
\r
  fclose (fpOut);\r
  printf ("Created %s\n", OutputFileName);\r
  return 0;\r
}\r
\r
Commit	Line	Data
30fdf114 LG	1	/** @file\r
30fdf114 LG	2	\r
40d841f6 LG	3	Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>\r
40d841f6 LG	4	This program and the accompanying materials \r
30fdf114 LG	5	are licensed and made available under the terms and conditions of the BSD License \r
	6	which accompanies this distribution. The full text of the license may be found at \r
	7	http://opensource.org/licenses/bsd-license.php \r
	8	\r
	9	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
	10	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. \r
	11	\r
	12	Module Name:\r
	13	\r
	14	efildrimage.c\r
	15	\r
	16	Abstract:\r
	17	\r
	18	Creates and EFILDR image.\r
	19	This tool combines several PE Image files together using following format denoted as EBNF:\r
	20	FILE := EFILDR_HEADER\r
	21	EFILDR_IMAGE +\r
	22	<PeImageFileContent> +\r
	23	The order of EFILDR_IMAGE is same as the order of placing PeImageFileContent.\r
	24	\r
	25	Revision History\r
	26	\r
	27	**/\r
	28	\r
	29	\r
	30	#include <stdio.h>\r
	31	#include <stdlib.h>\r
	32	#include <string.h>\r
	33	#include "ParseInf.h"\r
	34	#include "CommonLib.h"\r
	35	#include "EfiUtilityMsgs.h"\r
	36	\r
	37	#define MAX_PE_IMAGES 63\r
	38	#define FILE_TYPE_FIXED_LOADER 0\r
	39	#define FILE_TYPE_RELOCATABLE_PE_IMAGE 1\r
	40	\r
	41	typedef struct {\r
	42	UINT32 CheckSum;\r
	43	UINT32 Offset;\r
	44	UINT32 Length;\r
	45	UINT8 FileName[52];\r
	46	} EFILDR_IMAGE;\r
	47	\r
	48	typedef struct { \r
	49	UINT32 Signature; \r
	50	UINT32 HeaderCheckSum;\r
	51	UINT32 FileLength;\r
	52	UINT32 NumberOfImages;\r
	53	} EFILDR_HEADER;\r
	54	\r
	55	//\r
	56	// Utility Name\r
	57	//\r
	58	#define UTILITY_NAME "EfiLdrImage"\r
	59	\r
	60	//\r
	61	// Utility version information\r
	62	//\r
	63	#define UTILITY_MAJOR_VERSION 0\r
	64	#define UTILITY_MINOR_VERSION 1\r
	65	\r
	66	void\r
	67	Version (\r
	68	void\r
69	)\r
70	/*++\r
71	\r
72	Routine Description:\r
73	\r
74	Displays the standard utility information to SDTOUT\r
75	\r
76	Arguments:\r
77	\r
78	None\r
79	\r
80	Returns:\r
81	\r
82	None\r
83	\r
84	--*/\r
85	{\r
b36d134f	86	printf ("%s v%d.%d %s -Utility to break a file into two pieces at the request offset.\n", UTILITY_NAME, UTILITY_MAJOR_VERSION, UTILITY_MINOR_VERSION, __BUILD_VERSION);\r
52302d4d	87	printf ("Copyright (c) 1999-2010 Intel Corporation. All rights reserved.\n");\r
30fdf114 LG	88	}\r
	89	\r
	90	VOID\r
	91	Usage (\r
	92	VOID\r
	93	)\r
	94	{\r
	95	printf ("Usage: EfiLdrImage -o OutImage LoaderImage PeImage1 PeImage2 ... PeImageN\n");\r
	96	exit (1);\r
	97	}\r
	98	\r
	99	EFI_STATUS\r
	100	CountVerboseLevel (\r
	101	IN CONST CHAR8* VerboseLevelString,\r
	102	IN CONST UINT64 Length,\r
	103	OUT UINT64 *ReturnValue\r
	104	)\r
	105	{\r
	106	UINT64 i = 0;\r
	107	for (;i < Length; ++i) {\r
	108	if (VerboseLevelString[i] != 'v' && VerboseLevelString[i] != 'V') {\r
	109	return EFI_ABORTED;\r
	110	}\r
	111	++(*ReturnValue);\r
	112	}\r
	113	\r
	114	return EFI_SUCCESS;\r
	115	}\r
	116	\r
	117	UINT64\r
	118	FCopyFile (\r
	119	FILE *in,\r
	120	FILE *out\r
	121	)\r
	122	/*++\r
	123	Routine Description:\r
	124	Write all the content of input file to output file.\r
	125	\r
	126	Arguments:\r
	127	in - input file pointer\r
	128	out - output file pointer\r
	129	\r
	130	Return:\r
	131	UINT64 : file size of input file\r
	132	--*/\r
	133	{\r
fd171542	134	UINT32 filesize, offset, length;\r
30fdf114 LG	135	CHAR8 Buffer[8*1024];\r
	136	\r
	137	fseek (in, 0, SEEK_END);\r
	138	filesize = ftell(in);\r
	139	\r
	140	fseek (in, 0, SEEK_SET);\r
	141	\r
	142	offset = 0;\r
	143	while (offset < filesize) {\r
	144	length = sizeof(Buffer);\r
	145	if (filesize-offset < length) {\r
	146	length = filesize-offset;\r
	147	}\r
	148	\r
	149	fread (Buffer, length, 1, in);\r
	150	fwrite (Buffer, length, 1, out);\r
	151	offset += length;\r
	152	}\r
	153	\r
	154	return filesize;\r
	155	}\r
	156	\r
	157	\r
	158	int\r
	159	main (\r
	160	int argc,\r
	161	char *argv[]\r
	162	)\r
	163	/*++\r
	164	\r
	165	Routine Description:\r
	166	\r
	167	\r
	168	Arguments:\r
	169	\r
	170	\r
	171	Returns:\r
	172	\r
	173	\r
	174	--*/\r
	175	{\r
	176	UINT64 i;\r
	177	UINT64 filesize;\r
	178	FILE fpIn, fpOut;\r
	179	EFILDR_HEADER EfiLdrHeader;\r
	180	EFILDR_IMAGE EfiLdrImage[MAX_PE_IMAGES];\r
	181	CHAR8* OutputFileName = NULL;\r
	182	CHAR8* InputFileNames[MAX_PE_IMAGES + 1];\r
	183	UINT8 InputFileCount = 0;\r
	184	BOOLEAN QuietFlag = FALSE;\r
	185	UINT64 DebugLevel = 0;\r
	186	UINT64 VerboseLevel = 0;\r
	187	EFI_STATUS Status = EFI_SUCCESS;\r
	188	\r
	189	SetUtilityName (UTILITY_NAME);\r
	190	\r
	191	if (argc == 1) {\r
	192	Usage();\r
	193	return STATUS_ERROR;\r
	194	}\r
	195	\r
	196	argc --;\r
	197	argv ++;\r
	198	\r
199	if ((stricmp (argv[0], "-h") == 0) \|\| (stricmp (argv[0], "--help") == 0)) {\r
200	Usage();\r
201	return STATUS_SUCCESS; \r
202	}\r
203	\r
204	if (stricmp (argv[0], "--version") == 0) {\r
205	Version();\r
206	return STATUS_SUCCESS; \r
207	}\r
208	\r
209	while (argc > 0) {\r
210	\r
211	if ((stricmp (argv[0], "-o") == 0) \|\| (stricmp (argv[0], "--output") == 0)) {\r
212	OutputFileName = argv[1];\r
213	if (OutputFileName == NULL) {\r
214	Error (NULL, 0, 1003, "Invalid option value", "Output file can't be null");\r
215	return STATUS_ERROR;\r
216	}\r
217	argc -= 2;\r
218	argv += 2;\r
219	continue; \r
220	}\r
221	\r
222	if ((stricmp (argv[0], "-q") == 0) \|\| (stricmp (argv[0], "--quiet") == 0)) {\r
223	QuietFlag = TRUE;\r
224	argc --;\r
225	argv ++;\r
226	continue; \r
227	}\r
228	\r
229	if ((strlen(argv[0]) >= 2 && argv[0][0] == '-' && (argv[0][1] == 'v' \|\| argv[0][1] == 'V')) \|\| (stricmp (argv[0], "--verbose") == 0)) {\r
230	VerboseLevel = 1;\r
231	if (strlen(argv[0]) > 2) {\r
232	Status = CountVerboseLevel (&argv[0][2], strlen(argv[0]) - 2, &VerboseLevel);\r
233	if (EFI_ERROR (Status)) {\r
234	Error (NULL, 0, 1003, "Invalid option value", argv[0]);\r
235	return STATUS_ERROR; \r
236	}\r
237	}\r
238	\r
239	argc --;\r
240	argv ++;\r
241	continue; \r
242	}\r
243	\r
244	if ((stricmp (argv[0], "-d") == 0) \|\| (stricmp (argv[0], "--debug") == 0)) {\r
245	Status = AsciiStringToUint64 (argv[1], FALSE, &DebugLevel);\r
246	if (EFI_ERROR (Status)) {\r
247	Error (NULL, 0, 1003, "Invalid option value", "%s = %s", argv[0], argv[1]);\r
248	return STATUS_ERROR; \r
249	}\r
250	argc -= 2;\r
251	argv += 2;\r
252	continue; \r
253	}\r
254	//\r
fd171542	255	// Don't recognize the parameter, should be regarded as the input file name.\r
30fdf114 LG	256	//\r
	257	InputFileNames[InputFileCount] = argv[0];\r
	258	InputFileCount++;\r
	259	argc--;\r
	260	argv++;\r
	261	}\r
	262	\r
	263	if (InputFileCount == 0) {\r
	264	Error (NULL, 0, 1001, "Missing option", "No input file");\r
	265	return STATUS_ERROR;\r
	266	}\r
	267	//\r
	268	// Open output file for write\r
	269	//\r
	270	if (OutputFileName == NULL) {\r
	271	Error (NULL, 0, 1001, "Missing option", "No output file");\r
	272	return STATUS_ERROR;\r
	273	}\r
	274	\r
	275	fpOut = fopen(OutputFileName, "w+b");\r
	276	if (!fpOut) {\r
	277	Error (NULL, 0, 0001, "Could not open output file", OutputFileName);\r
	278	return STATUS_ERROR;\r
	279	}\r
	280	\r
	281	memset (&EfiLdrHeader, 0, sizeof (EfiLdrHeader));\r
	282	memset (&EfiLdrImage, 0, sizeof (EFILDR_IMAGE) * (InputFileCount));\r
	283	\r
	284	memcpy (&EfiLdrHeader.Signature, "EFIL", 4);\r
	285	EfiLdrHeader.FileLength = sizeof(EFILDR_HEADER) + sizeof(EFILDR_IMAGE)*(InputFileCount);\r
	286	\r
	287	//\r
	288	// Skip the file header first\r
	289	//\r
	290	fseek (fpOut, EfiLdrHeader.FileLength, SEEK_SET);\r
	291	\r
	292	//\r
	293	// copy all the input files to the output file\r
	294	//\r
	295	for(i=0;i<InputFileCount;i++) {\r
	296	//\r
	297	// Copy the content of PeImage file to output file\r
	298	//\r
	299	fpIn = fopen (InputFileNames[i], "rb");\r
	300	if (!fpIn) {\r
	301	Error (NULL, 0, 0001, "Could not open input file", InputFileNames[i]);\r
	302	fclose (fpOut);\r
	303	return STATUS_ERROR;\r
	304	}\r
	305	filesize = FCopyFile (fpIn, fpOut);\r
	306	fclose(fpIn);\r
	307	\r
	308	//\r
	309	// And in the same time update the EfiLdrHeader and EfiLdrImage array\r
	310	//\r
	311	EfiLdrImage[i].Offset = EfiLdrHeader.FileLength;\r
fd171542	312	EfiLdrImage[i].Length = (UINT32) filesize;\r
30fdf114	313	strncpy ((CHAR8*) EfiLdrImage[i].FileName, InputFileNames[i], sizeof (EfiLdrImage[i].FileName) - 1);\r
fd171542	314	EfiLdrHeader.FileLength += (UINT32) filesize;\r
30fdf114 LG	315	EfiLdrHeader.NumberOfImages++;\r
	316	}\r
	317	\r
	318	//\r
	319	// Write the image header to the output file finally\r
	320	//\r
	321	fseek (fpOut, 0, SEEK_SET);\r
	322	fwrite (&EfiLdrHeader, sizeof(EFILDR_HEADER) , 1, fpOut);\r
	323	fwrite (&EfiLdrImage , sizeof(EFILDR_IMAGE)*(InputFileCount), 1, fpOut);\r
	324	\r
	325	fclose (fpOut);\r
	326	printf ("Created %s\n", OutputFileName);\r
	327	return 0;\r
	328	}\r
	329	\r