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

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