[mirror_edk2.git] / InOsEmuPkg / Unix / Sec / SecMain.h

/*++ @file\r
\r
Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>\r
Portions copyright (c) 2008 - 2011, Apple Inc. All rights reserved.<BR>\r
\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
**/\r
\r
#ifndef _SEC_MAIN_H__\r
#define _SEC_MAIN_H__\r
\r
#include <PiPei.h>\r
#include <Uefi.h>\r
\r
#include <Library/PeCoffLib.h>\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/PrintLib.h>\r
#include <Library/PcdLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/ReportStatusCodeLib.h>\r
\r
#include <Library/ThunkPpiList.h>\r
#include <Library/ThunkProtocolList.h>\r
#include <Library/PpiListLib.h>\r
#include <Library/PeiServicesLib.h>\r
#include <Library/PeCoffGetEntryPointLib.h>\r
\r
#include <Ppi/EmuThunk.h>\r
#include <Ppi/StatusCode.h>\r
\r
#include <Protocol/SimplePointer.h>\r
#include <Protocol/SimpleTextIn.h>\r
#include <Protocol/SimpleTextInEx.h>\r
#include <Protocol/UgaDraw.h>\r
#include <Protocol/SimpleFileSystem.h>\r
\r
#include <Protocol/EmuThunk.h>\r
#include <Protocol/EmuIoThunk.h>\r
#include <Protocol/EmuGraphicsWindow.h>\r
#include <Protocol/EmuThread.h>\r
#include <Protocol/EmuBlockIo.h>\r
\r
#include <Guid/FileInfo.h>\r
#include <Guid/FileSystemInfo.h>\r
#include <Guid/FileSystemVolumeLabelInfo.h>\r
\r
#include <stdio.h>\r
#include <stdlib.h>\r
\r
#include <sys/types.h>\r
#include <sys/stat.h>\r
#include <sys/termios.h>\r
#include <sys/time.h>\r
\r
#if __CYGWIN__\r
#include <sys/dirent.h>\r
#else\r
#include <sys/dir.h>\r
#endif\r
\r
#include <sys/mman.h>\r
#include <dlfcn.h>\r
\r
#include <unistd.h>\r
#include <poll.h>\r
#include <fcntl.h>\r
#include <time.h>\r
#include <signal.h>\r
#include <errno.h>\r
#include <string.h>\r
#include <stdlib.h>\r
#include <sys/ioctl.h>\r
#include <sys/statvfs.h>\r
\r
#include <sys/socket.h>\r
#include <netdb.h>\r
#include <netinet/in.h>\r
#include <net/if.h>\r
#include <ifaddrs.h>\r
\r
#ifdef __APPLE__\r
#include <net/if_dl.h>\r
#include <net/bpf.h>\r
#include <sys/param.h>\r
#include <sys/mount.h>\r
#include <sys/disk.h>\r
#define _XOPEN_SOURCE\r
#ifndef _Bool\r
  #define _Bool char // for clang debug\r
#endif\r
#else\r
#include <termio.h>\r
#include <sys/vfs.h>\r
#include <linux/fs.h>\r
#endif \r
\r
#include <utime.h>\r
\r
#include "Gasket.h"\r
\r
\r
#define STACK_SIZE                0x20000      \r
\r
typedef struct {\r
  EFI_PHYSICAL_ADDRESS  Address;\r
  UINT64                Size;\r
} EMU_FD_INFO;\r
\r
typedef struct {\r
  EFI_PHYSICAL_ADDRESS  Memory;\r
  UINT64                Size;\r
} EMU_SYSTEM_MEMORY;\r
\r
\r
#define MAX_IMAGE_CONTEXT_TO_MOD_HANDLE_ARRAY_SIZE 0x100\r
\r
typedef struct {\r
  PE_COFF_LOADER_IMAGE_CONTEXT   *ImageContext;\r
  VOID                           *ModHandle;\r
} IMAGE_CONTEXT_TO_MOD_HANDLE;\r
\r
\r
EFI_STATUS\r
EFIAPI\r
SecUnixPeiLoadFile (\r
  VOID                  *Pe32Data,  \r
  EFI_PHYSICAL_ADDRESS  *ImageAddress,  \r
  UINT64                *ImageSize,  \r
  EFI_PHYSICAL_ADDRESS  *EntryPoint  \r
  );\r
\r
int\r
main (\r
  IN  int   Argc,\r
  IN  char  **Argv,\r
  IN  char  **Envp\r
  );\r
\r
VOID\r
SecLoadFromCore (\r
  IN  UINTN   LargestRegion,\r
  IN  UINTN   LargestRegionSize,\r
  IN  UINTN   BootFirmwareVolumeBase,\r
  IN  VOID    *PeiCoreFile\r
  );\r
\r
EFI_STATUS\r
SecLoadFile (\r
  IN  VOID                    *Pe32Data,\r
  IN  EFI_PHYSICAL_ADDRESS    *ImageAddress,\r
  IN  UINT64                  *ImageSize,\r
  IN  EFI_PHYSICAL_ADDRESS    *EntryPoint\r
  );\r
\r
EFI_STATUS\r
SecFfsFindPeiCore (\r
  IN  EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader,\r
  OUT VOID                        **Pe32Data\r
  );\r
\r
EFI_STATUS\r
SecFfsFindNextFile (\r
  IN EFI_FV_FILETYPE             SearchType,\r
  IN EFI_FIRMWARE_VOLUME_HEADER  *FwVolHeader,\r
  IN OUT EFI_FFS_FILE_HEADER     **FileHeader\r
  );\r
\r
EFI_STATUS\r
SecFfsFindSectionData (\r
  IN EFI_SECTION_TYPE      SectionType,\r
  IN EFI_FFS_FILE_HEADER   *FfsFileHeader,\r
  IN OUT VOID              **SectionData\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
SecUnixPeCoffLoaderLoadAsDll (\r
  IN CHAR8    *PdbFileName,\r
  IN VOID     **ImageEntryPoint,\r
  OUT VOID    **ModHandle\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
SecUnixPeCoffLoaderFreeLibrary (\r
  OUT VOID    *ModHandle\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
SecUnixFdAddress (\r
  IN     UINTN                 Index,\r
  IN OUT EFI_PHYSICAL_ADDRESS  *FdBase,\r
  IN OUT UINT64                *FdSize,\r
  IN OUT EFI_PHYSICAL_ADDRESS  *FixUp\r
  )\r
;\r
\r
EFI_STATUS\r
EFIAPI\r
GasketSecUnixFdAddress (\r
  IN     UINTN                 Index,\r
  IN OUT EFI_PHYSICAL_ADDRESS  *FdBase,\r
  IN OUT UINT64                *FdSize,\r
  IN OUT EFI_PHYSICAL_ADDRESS  *FixUp\r
  )\r
;\r
\r
\r
EFI_STATUS\r
GetImageReadFunction (\r
  IN PE_COFF_LOADER_IMAGE_CONTEXT          *ImageContext,\r
  IN EFI_PHYSICAL_ADDRESS                  *TopOfMemory\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
SecImageRead (\r
  IN     VOID    *FileHandle,\r
  IN     UINTN   FileOffset,\r
  IN OUT UINTN   *ReadSize,\r
  OUT    VOID    *Buffer\r
  );\r
\r
CHAR16                            *\r
AsciiToUnicode (\r
  IN  CHAR8   *Ascii,\r
  IN  UINTN   *StrLen OPTIONAL\r
  );\r
\r
UINTN\r
CountSeperatorsInString (\r
  IN  const CHAR16   *String,\r
  IN  CHAR16   Seperator\r
  );\r
\r
EFI_STATUS\r
EFIAPI\r
SecTemporaryRamSupport (\r
  IN CONST EFI_PEI_SERVICES   **PeiServices,\r
  IN EFI_PHYSICAL_ADDRESS     TemporaryMemoryBase,\r
  IN EFI_PHYSICAL_ADDRESS     PermanentMemoryBase,\r
  IN UINTN                    CopySize\r
  );\r
  \r
EFI_STATUS\r
EFIAPI\r
GasketSecTemporaryRamSupport (\r
  IN CONST EFI_PEI_SERVICES   **PeiServices,\r
  IN EFI_PHYSICAL_ADDRESS     TemporaryMemoryBase,\r
  IN EFI_PHYSICAL_ADDRESS     PermanentMemoryBase,\r
  IN UINTN                    CopySize\r
  );\r
\r
\r
RETURN_STATUS\r
EFIAPI\r
SecPeCoffGetEntryPoint (\r
  IN     VOID  *Pe32Data,\r
  IN OUT VOID  **EntryPoint\r
  );\r
\r
VOID\r
EFIAPI\r
SecPeCoffRelocateImageExtraAction (\r
  IN OUT PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext\r
  );\r
\r
VOID\r
EFIAPI\r
SecPeCoffLoaderUnloadImageExtraAction (\r
  IN OUT PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext\r
  );\r
\r
\r
VOID\r
EFIAPI\r
PeiSwitchStacks (\r
  IN      SWITCH_STACK_ENTRY_POINT  EntryPoint,\r
  IN      VOID                      *Context1,  OPTIONAL\r
  IN      VOID                      *Context2,  OPTIONAL\r
  IN      VOID                      *NewStack\r
  );\r
\r
VOID\r
SecInitThunkProtocol (\r
  VOID\r
  );\r
 \r
  \r
EFI_PHYSICAL_ADDRESS *\r
MapMemory (\r
  INTN fd,\r
  UINT64 length,\r
  INTN   prot,\r
  INTN   flags);\r
\r
EFI_STATUS\r
MapFile (\r
  IN  CHAR8                     *FileName,\r
  IN OUT  EFI_PHYSICAL_ADDRESS  *BaseAddress,\r
  OUT UINT64                    *Length\r
  );\r
\r
 \r
\r
VOID SecSleep (UINT64 Milliseconds);\r
VOID SecEnableInterrupt (VOID);\r
VOID SecDisableInterrupt (VOID);\r
BOOLEAN SecInterruptEanbled (VOID);\r
\r
\r
extern EMU_THUNK_PROTOCOL    gEmuThunkProtocol;\r
extern EMU_IO_THUNK_PROTOCOL gX11ThunkIo;\r
extern EMU_IO_THUNK_PROTOCOL gPosixFileSystemThunkIo;\r
extern EMU_IO_THUNK_PROTOCOL gPthreadThunkIo;\r
extern EMU_IO_THUNK_PROTOCOL gBlockIoThunkIo;\r
\r
#endif\r
Commit	Line	Data
949f388f	1	/*++ @file\r
949f388f	2	\r
10d1be3e	3	Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>\r
949f388f	4	Portions copyright (c) 2008 - 2011, Apple Inc. All rights reserved.<BR>\r
10d1be3e	5	\r
	6	This program and the accompanying materials\r
	7	are licensed and made available under the terms and conditions of the BSD License\r
	8	which accompanies this distribution. The full text of the license may be found at\r
	9	http://opensource.org/licenses/bsd-license.php\r
	10	\r
	11	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	12	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
949f388f	13	\r
	14	**/\r
	15	\r
	16	#ifndef _SEC_MAIN_H__\r
	17	#define _SEC_MAIN_H__\r
	18	\r
	19	#include <PiPei.h>\r
	20	#include <Uefi.h>\r
	21	\r
	22	#include <Library/PeCoffLib.h>\r
	23	#include <Library/BaseLib.h>\r
	24	#include <Library/BaseMemoryLib.h>\r
	25	#include <Library/PrintLib.h>\r
	26	#include <Library/PcdLib.h>\r
	27	#include <Library/DebugLib.h>\r
	28	#include <Library/ReportStatusCodeLib.h>\r
	29	\r
	30	#include <Library/ThunkPpiList.h>\r
	31	#include <Library/ThunkProtocolList.h>\r
65e3f333	32	#include <Library/PpiListLib.h>\r
	33	#include <Library/PeiServicesLib.h>\r
	34	#include <Library/PeCoffGetEntryPointLib.h>\r
949f388f	35	\r
	36	#include <Ppi/EmuThunk.h>\r
	37	#include <Ppi/StatusCode.h>\r
949f388f	38	\r
	39	#include <Protocol/SimplePointer.h>\r
	40	#include <Protocol/SimpleTextIn.h>\r
	41	#include <Protocol/SimpleTextInEx.h>\r
	42	#include <Protocol/UgaDraw.h>\r
	43	#include <Protocol/SimpleFileSystem.h>\r
	44	\r
	45	#include <Protocol/EmuThunk.h>\r
	46	#include <Protocol/EmuIoThunk.h>\r
	47	#include <Protocol/EmuGraphicsWindow.h>\r
10d1be3e	48	#include <Protocol/EmuThread.h>\r
033d0e5f	49	#include <Protocol/EmuBlockIo.h>\r
949f388f	50	\r
	51	#include <Guid/FileInfo.h>\r
	52	#include <Guid/FileSystemInfo.h>\r
	53	#include <Guid/FileSystemVolumeLabelInfo.h>\r
	54	\r
	55	#include <stdio.h>\r
	56	#include <stdlib.h>\r
	57	\r
	58	#include <sys/types.h>\r
	59	#include <sys/stat.h>\r
	60	#include <sys/termios.h>\r
	61	#include <sys/time.h>\r
	62	\r
	63	#if __CYGWIN__\r
	64	#include <sys/dirent.h>\r
	65	#else\r
	66	#include <sys/dir.h>\r
	67	#endif\r
	68	\r
	69	#include <sys/mman.h>\r
	70	#include <dlfcn.h>\r
	71	\r
	72	#include <unistd.h>\r
	73	#include <poll.h>\r
	74	#include <fcntl.h>\r
	75	#include <time.h>\r
	76	#include <signal.h>\r
	77	#include <errno.h>\r
	78	#include <string.h>\r
	79	#include <stdlib.h>\r
	80	#include <sys/ioctl.h>\r
63947cc4	81	#include <sys/statvfs.h>\r
949f388f	82	\r
	83	#include <sys/socket.h>\r
	84	#include <netdb.h>\r
	85	#include <netinet/in.h>\r
	86	#include <net/if.h>\r
	87	#include <ifaddrs.h>\r
	88	\r
	89	#ifdef __APPLE__\r
	90	#include <net/if_dl.h>\r
	91	#include <net/bpf.h>\r
	92	#include <sys/param.h>\r
	93	#include <sys/mount.h>\r
63947cc4	94	#include <sys/disk.h>\r
949f388f	95	#define _XOPEN_SOURCE\r
	96	#ifndef _Bool\r
	97	#define _Bool char // for clang debug\r
	98	#endif\r
	99	#else\r
	100	#include <termio.h>\r
	101	#include <sys/vfs.h>\r
63947cc4	102	#include <linux/fs.h>\r
949f388f	103	#endif \r
	104	\r
	105	#include <utime.h>\r
	106	\r
	107	#include "Gasket.h"\r
	108	\r
	109	\r
	110	#define STACK_SIZE 0x20000 \r
	111	\r
	112	typedef struct {\r
	113	EFI_PHYSICAL_ADDRESS Address;\r
	114	UINT64 Size;\r
	115	} EMU_FD_INFO;\r
	116	\r
	117	typedef struct {\r
	118	EFI_PHYSICAL_ADDRESS Memory;\r
	119	UINT64 Size;\r
	120	} EMU_SYSTEM_MEMORY;\r
	121	\r
	122	\r
	123	#define MAX_IMAGE_CONTEXT_TO_MOD_HANDLE_ARRAY_SIZE 0x100\r
	124	\r
	125	typedef struct {\r
	126	PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext;\r
	127	VOID *ModHandle;\r
	128	} IMAGE_CONTEXT_TO_MOD_HANDLE;\r
	129	\r
	130	\r
	131	EFI_STATUS\r
	132	EFIAPI\r
	133	SecUnixPeiLoadFile (\r
	134	VOID *Pe32Data, \r
	135	EFI_PHYSICAL_ADDRESS *ImageAddress, \r
	136	UINT64 *ImageSize, \r
	137	EFI_PHYSICAL_ADDRESS *EntryPoint \r
	138	);\r
	139	\r
	140	int\r
	141	main (\r
	142	IN int Argc,\r
	143	IN char **Argv,\r
	144	IN char **Envp\r
	145	);\r
	146	\r
	147	VOID\r
	148	SecLoadFromCore (\r
	149	IN UINTN LargestRegion,\r
	150	IN UINTN LargestRegionSize,\r
	151	IN UINTN BootFirmwareVolumeBase,\r
	152	IN VOID *PeiCoreFile\r
	153	);\r
	154	\r
	155	EFI_STATUS\r
	156	SecLoadFile (\r
	157	IN VOID *Pe32Data,\r
	158	IN EFI_PHYSICAL_ADDRESS *ImageAddress,\r
	159	IN UINT64 *ImageSize,\r
	160	IN EFI_PHYSICAL_ADDRESS *EntryPoint\r
	161	);\r
	162	\r
	163	EFI_STATUS\r
	164	SecFfsFindPeiCore (\r
	165	IN EFI_FIRMWARE_VOLUME_HEADER *FwVolHeader,\r
	166	OUT VOID **Pe32Data\r
167	);\r
168	\r
169	EFI_STATUS\r
170	SecFfsFindNextFile (\r
171	IN EFI_FV_FILETYPE SearchType,\r
172	IN EFI_FIRMWARE_VOLUME_HEADER *FwVolHeader,\r
173	IN OUT EFI_FFS_FILE_HEADER **FileHeader\r
174	);\r
175	\r
176	EFI_STATUS\r
177	SecFfsFindSectionData (\r
178	IN EFI_SECTION_TYPE SectionType,\r
179	IN EFI_FFS_FILE_HEADER *FfsFileHeader,\r
180	IN OUT VOID **SectionData\r
181	);\r
182	\r
183	EFI_STATUS\r
184	EFIAPI\r
185	SecUnixPeCoffLoaderLoadAsDll (\r
186	IN CHAR8 *PdbFileName,\r
187	IN VOID **ImageEntryPoint,\r
188	OUT VOID **ModHandle\r
189	);\r
190	\r
191	EFI_STATUS\r
192	EFIAPI\r
193	SecUnixPeCoffLoaderFreeLibrary (\r
194	OUT VOID *ModHandle\r
195	);\r
196	\r
197	EFI_STATUS\r
198	EFIAPI\r
199	SecUnixFdAddress (\r
200	IN UINTN Index,\r
201	IN OUT EFI_PHYSICAL_ADDRESS *FdBase,\r
202	IN OUT UINT64 *FdSize,\r
203	IN OUT EFI_PHYSICAL_ADDRESS *FixUp\r
204	)\r
205	;\r
206	\r
207	EFI_STATUS\r
208	EFIAPI\r
209	GasketSecUnixFdAddress (\r
210	IN UINTN Index,\r
211	IN OUT EFI_PHYSICAL_ADDRESS *FdBase,\r
212	IN OUT UINT64 *FdSize,\r
213	IN OUT EFI_PHYSICAL_ADDRESS *FixUp\r
214	)\r
215	;\r
216	\r
217	\r
218	EFI_STATUS\r
219	GetImageReadFunction (\r
220	IN PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext,\r
221	IN EFI_PHYSICAL_ADDRESS *TopOfMemory\r
222	);\r
223	\r
224	EFI_STATUS\r
225	EFIAPI\r
226	SecImageRead (\r
227	IN VOID *FileHandle,\r
228	IN UINTN FileOffset,\r
229	IN OUT UINTN *ReadSize,\r
230	OUT VOID *Buffer\r
231	);\r
232	\r
233	CHAR16 *\r
234	AsciiToUnicode (\r
235	IN CHAR8 *Ascii,\r
236	IN UINTN *StrLen OPTIONAL\r
237	);\r
238	\r
239	UINTN\r
240	CountSeperatorsInString (\r
241	IN const CHAR16 *String,\r
242	IN CHAR16 Seperator\r
243	);\r
244	\r
245	EFI_STATUS\r
246	EFIAPI\r
247	SecTemporaryRamSupport (\r
248	IN CONST EFI_PEI_SERVICES **PeiServices,\r
249	IN EFI_PHYSICAL_ADDRESS TemporaryMemoryBase,\r
250	IN EFI_PHYSICAL_ADDRESS PermanentMemoryBase,\r
251	IN UINTN CopySize\r
252	);\r
253	\r
254	EFI_STATUS\r
255	EFIAPI\r
256	GasketSecTemporaryRamSupport (\r
257	IN CONST EFI_PEI_SERVICES **PeiServices,\r
258	IN EFI_PHYSICAL_ADDRESS TemporaryMemoryBase,\r
259	IN EFI_PHYSICAL_ADDRESS PermanentMemoryBase,\r
260	IN UINTN CopySize\r
261	);\r
262	\r
263	\r
264	RETURN_STATUS\r
265	EFIAPI\r
266	SecPeCoffGetEntryPoint (\r
267	IN VOID *Pe32Data,\r
268	IN OUT VOID **EntryPoint\r
269	);\r
270	\r
271	VOID\r
272	EFIAPI\r
273	SecPeCoffRelocateImageExtraAction (\r
274	IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r
275	);\r
276	\r
277	VOID\r
278	EFIAPI\r
279	SecPeCoffLoaderUnloadImageExtraAction (\r
280	IN OUT PE_COFF_LOADER_IMAGE_CONTEXT *ImageContext\r
281	);\r
282	\r
283	\r
284	VOID\r
285	EFIAPI\r
286	PeiSwitchStacks (\r
287	IN SWITCH_STACK_ENTRY_POINT EntryPoint,\r
288	IN VOID *Context1, OPTIONAL\r
289	IN VOID *Context2, OPTIONAL\r
949f388f	290	IN VOID *NewStack\r
	291	);\r
	292	\r
	293	VOID\r
	294	SecInitThunkProtocol (\r
	295	VOID\r
	296	);\r
65e3f333	297	\r
949f388f	298	\r
65e3f333	299	EFI_PHYSICAL_ADDRESS *\r
	300	MapMemory (\r
	301	INTN fd,\r
	302	UINT64 length,\r
	303	INTN prot,\r
	304	INTN flags);\r
	305	\r
	306	EFI_STATUS\r
	307	MapFile (\r
	308	IN CHAR8 *FileName,\r
	309	IN OUT EFI_PHYSICAL_ADDRESS *BaseAddress,\r
	310	OUT UINT64 *Length\r
	311	);\r
	312	\r
	313	\r
949f388f	314	\r
	315	VOID SecSleep (UINT64 Milliseconds);\r
	316	VOID SecEnableInterrupt (VOID);\r
	317	VOID SecDisableInterrupt (VOID);\r
	318	BOOLEAN SecInterruptEanbled (VOID);\r
	319	\r
	320	\r
	321	extern EMU_THUNK_PROTOCOL gEmuThunkProtocol;\r
	322	extern EMU_IO_THUNK_PROTOCOL gX11ThunkIo;\r
	323	extern EMU_IO_THUNK_PROTOCOL gPosixFileSystemThunkIo;\r
	324	extern EMU_IO_THUNK_PROTOCOL gPthreadThunkIo;\r
033d0e5f	325	extern EMU_IO_THUNK_PROTOCOL gBlockIoThunkIo;\r
949f388f	326	\r
949f388f	327	#endif\r