EmulatorPkg/Unix/Host/SecMain.h

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