KernelLand/Kernel/binary.c

   1 /*
   2  * Acess2
   3  * Common Binary Loader
   4  */
   5 #define DEBUG   0
   6 #include <acess.h>
   7 #include <binary.h>
   8 #include <mm_virt.h>
   9 #include <hal_proc.h>
  10 #include <vfs_threads.h>
  11
  12 // === CONSTANTS ===
  13 #define BIN_LOWEST      MM_USER_MIN             // 1MiB
  14 #define BIN_GRANUALITY  0x10000         // 64KiB
  15 #define BIN_HIGHEST     (USER_LIB_MAX-BIN_GRANUALITY)           // Just below the kernel
  16 #define KLIB_LOWEST     MM_MODULE_MIN
  17 #define KLIB_GRANUALITY 0x10000         // 32KiB
  18 #define KLIB_HIGHEST    (MM_MODULE_MAX-KLIB_GRANUALITY)
  19
  20 // === TYPES ===
  21 typedef struct sKernelBin {
  22         struct sKernelBin       *Next;
  23         void    *Base;
  24         tBinary *Info;
  25 } tKernelBin;
  26
  27 // === IMPORTS ===
  28 extern char     *Threads_GetName(int ID);
  29 extern tKernelSymbol    gKernelSymbols[];
  30 extern tKernelSymbol    gKernelSymbolsEnd[];
  31 extern tBinaryType      gELF_Info;
  32
  33 // === PROTOTYPES ===
  34 size_t  Binary_int_CacheArgs(const char **Path, const char ***ArgV, const char ***EnvP, void *DestBuffer);
  35  int    Proc_int_Execve(const char *File, const char **ArgV, const char **EnvP, int DataSize, bool bClearUser);
  36 tVAddr  Binary_Load(const char *Path, tVAddr *EntryPoint);
  37 tBinary *Binary_GetInfo(tMount MountID, tInode InodeID);
  38 tVAddr  Binary_MapIn(tBinary *Binary, const char *Path, tVAddr LoadMin, tVAddr LoadMax);
  39 tVAddr  Binary_IsMapped(tBinary *Binary);
  40 tBinary *Binary_DoLoad(tMount MountID, tInode Inode, const char *Path);
  41 void    Binary_Dereference(tBinary *Info);
  42 #if 0
  43 Uint    Binary_Relocate(void *Base);
  44 #endif
  45 Uint    Binary_GetSymbolEx(const char *Name, Uint *Value);
  46 #if 0
  47 Uint    Binary_FindSymbol(void *Base, const char *Name, Uint *Val);
  48 #endif
  49  int    Binary_int_CheckMemFree( tVAddr _start, size_t _len );
  50
  51 // === GLOBALS ===
  52 tShortSpinlock  glBinListLock;
  53 tBinary *glLoadedBinaries = NULL;
  54 char    **gsaRegInterps = NULL;
  55  int    giRegInterps = 0;
  56 tShortSpinlock  glKBinListLock;
  57 tKernelBin      *glLoadedKernelLibs;
  58 tBinaryType     *gRegBinTypes = &gELF_Info;
  59
  60 // === FUNCTIONS ===
  61 /**
  62  * \brief Registers a binary type
  63  */
  64 int Binary_RegisterType(tBinaryType *Type)
  65 {
  66         Type->Next = gRegBinTypes;
  67         gRegBinTypes = Type;
  68         return 1;
  69 }
  70
  71 /**
  72  * \fn int Proc_Spawn(const char *Path)
  73  */
  74 int Proc_Spawn(const char *Path)
  75 {
  76         char    stackPath[strlen(Path)+1];
  77         ENTER("sPath", Path);
  78
  79         strcpy(stackPath, Path);
  80
  81         LOG("stackPath = '%s'", stackPath);
  82
  83         if(Proc_Clone(CLONE_VM|CLONE_NOUSER) == 0)
  84         {
  85                 // CHILD
  86                 const char      *args[2] = {stackPath, NULL};
  87                 LOG("stackPath = '%s'", stackPath);
  88                 Proc_Execve(stackPath, args, &args[1], 0);
  89                 for(;;);
  90         }
  91         LEAVE('i', 0);
  92         return 0;
  93 }
  94
  95 /**
  96  * \todo Document
  97  */
  98 size_t Binary_int_CacheArgs(const char **Path, const char ***ArgV, const char ***EnvP, void *DestBuffer)
  99 {
 100         size_t  size;
 101          int    argc=0, envc=0;
 102          int    i;
 103         char    *strbuf;
 104         const char      **arrays;
 105
 106         // Calculate size
 107         size = 0;
 108         if( ArgV && *ArgV )
 109         {
 110                 const char      **argv = *ArgV;
 111                 for( argc = 0; argv[argc]; argc ++ )
 112                         size += strlen( argv[argc] ) + 1;
 113         }
 114         if( EnvP && *EnvP )
 115         {
 116                 const char      **envp = *EnvP;
 117                 for( envc = 0; envp[envc]; envc ++ )
 118                         size += strlen( envp[envc] ) + 1;
 119         }
 120         size = (size + sizeof(void*)-1) & ~(sizeof(void*)-1);   // Word align
 121         size += (argc+1+envc+1)*sizeof(void*);  // Arrays
 122         if( Path )
 123         {
 124                 size += strlen( *Path ) + 1;
 125         }
 126
 127         if( DestBuffer )
 128         {
 129                 arrays = DestBuffer;
 130                 strbuf = (void*)&arrays[argc+1+envc+1];
 131
 132                 // Fill ArgV
 133                 if( ArgV && *ArgV )
 134                 {
 135                         const char      **argv = *ArgV;
 136                         for( i = 0; argv[i]; i ++ )
 137                         {
 138                                 arrays[i] = strbuf;
 139                                 strcpy(strbuf, argv[i]);
 140                                 strbuf += strlen( argv[i] ) + 1;
 141                         }
 142                         *ArgV = arrays;
 143                         arrays += i;
 144                 }
 145                 *arrays++ = NULL;
 146                 // Fill EnvP
 147                 if( EnvP && *EnvP )
 148                 {
 149                         const char      **envp = *EnvP;
 150                         for( i = 0; envp[i]; i ++ )
 151                         {
 152                                 arrays[i] = strbuf;
 153                                 strcpy(strbuf, envp[i]);
 154                                 strbuf += strlen( envp[i] ) + 1;
 155                         }
 156                         *EnvP = arrays;
 157                         arrays += i;
 158                 }
 159                 *arrays++ = NULL;
 160                 // Fill path
 161                 if( Path )
 162                 {
 163                         strcpy(strbuf, *Path);
 164                         *Path = strbuf;
 165                 }
 166         }
 167
 168         return size;
 169 }
 170
 171 /**
 172  * \brief Create a new process with the specified set of file descriptors
 173  */
 174 int Proc_SysSpawn(const char *Binary, const char **ArgV, const char **EnvP, int nFD, int *FDs)
 175 {
 176
 177         // --- Save File, ArgV and EnvP
 178         size_t size = Binary_int_CacheArgs( &Binary, &ArgV, &EnvP, NULL );
 179         void *cachebuf = malloc( size );
 180         Binary_int_CacheArgs( &Binary, &ArgV, &EnvP, cachebuf );
 181
 182         // Cache the VFS handles
 183         void *handles = VFS_SaveHandles(nFD, FDs);
 184
 185         // Create new process
 186         tPID ret = Proc_Clone(CLONE_VM|CLONE_NOUSER);
 187         if( ret == 0 )
 188         {
 189                 VFS_RestoreHandles(nFD, handles);
 190                 VFS_FreeSavedHandles(nFD, handles);
 191                 // Frees cachebuf
 192                 Proc_int_Execve(Binary, ArgV, EnvP, size, 0);
 193                 for(;;);
 194         }
 195         if( ret == -1 )
 196         {
 197                 VFS_FreeSavedHandles(nFD, handles);
 198                 free(cachebuf);
 199         }
 200
 201         return ret;
 202 }
 203
 204 /**
 205  * \brief Replace the current user image with another
 206  * \param File  File to load as the next image
 207  * \param ArgV  Arguments to pass to user
 208  * \param EnvP  User's environment
 209  * \note Called Proc_ for historical reasons
 210  */
 211 int Proc_Execve(const char *File, const char **ArgV, const char **EnvP, int DataSize)
 212 {
 213         return Proc_int_Execve(File, ArgV, EnvP, DataSize, 1);
 214 }
 215
 216 int Proc_int_Execve(const char *File, const char **ArgV, const char **EnvP, int DataSize, bool bClearUser)
 217 {
 218         void    *cachebuf;
 219         tVAddr  entry;
 220         Uint    base;   // Uint because Proc_StartUser wants it
 221          int    argc;
 222
 223         ENTER("sFile pArgV pEnvP", File, ArgV, EnvP);
 224
 225         // --- Save File, ArgV and EnvP
 226         if( DataSize == 0 )
 227         {
 228                 DataSize = Binary_int_CacheArgs( &File, &ArgV, &EnvP, NULL );
 229                 cachebuf = malloc( DataSize );
 230                 Binary_int_CacheArgs( &File, &ArgV, &EnvP, cachebuf );
 231         }
 232
 233         // --- Get argc
 234         for( argc = 0; ArgV && ArgV[argc]; argc ++ )
 235                 ;
 236
 237         // --- Set Process Name
 238         Threads_SetName(File);
 239
 240         // --- Clear User Address space
 241         if( bClearUser )
 242         {
 243                 // MM_ClearUser should preserve handles
 244                 MM_ClearUser();
 245                 // - NOTE: Not a reliable test, but helps for now
 246                 ASSERTC( VFS_IOCtl(0, 0, NULL), !=, -1 );
 247         }
 248
 249         // --- Load new binary
 250         base = Binary_Load(File, &entry);
 251         if(base == 0)
 252         {
 253                 Log_Warning("Binary", "Proc_Execve - Unable to load '%s' [errno=%i]", File, errno);
 254                 LEAVE('-');
 255                 Threads_Exit(0, -10);
 256                 for(;;);
 257         }
 258
 259         LOG("entry = 0x%x, base = 0x%x", entry, base);
 260         LEAVE('-');
 261         // --- And... Jump to it
 262         Proc_StartUser(entry, base, argc, ArgV, DataSize);
 263         for(;;);        // Tell GCC that we never return
 264 }
 265
 266 /**
 267  * \brief Load a binary into the current address space
 268  * \param Path  Path to binary to load
 269  * \param EntryPoint    Pointer for exectuable entry point
 270  * \return Virtual address where the binary has been loaded
 271  */
 272 tVAddr Binary_Load(const char *Path, tVAddr *EntryPoint)
 273 {
 274         tMount  mount_id;
 275         tInode  inode;
 276         tBinary *pBinary;
 277         tVAddr  base = -1;
 278
 279         ENTER("sPath pEntryPoint", Path, EntryPoint);
 280
 281         // Sanity Check Argument
 282         if(Path == NULL) {
 283                 LEAVE('x', 0);
 284                 return 0;
 285         }
 286
 287         // Check if this path has been loaded before.
 288         #if 0
 289         // TODO: Implement a list of string/tBinary pairs for loaded bins
 290         #endif
 291
 292         // Get Inode
 293         {
 294                 int fd;
 295                 tFInfo  info;
 296                 fd = VFS_Open(Path, VFS_OPENFLAG_READ|VFS_OPENFLAG_EXEC);
 297                 if( fd == -1 ) {
 298                         LOG("%s does not exist", Path);
 299                         LEAVE_RET('x', 0);
 300                 }
 301                 VFS_FInfo(fd, &info, 0);
 302                 VFS_Close(fd);
 303                 mount_id = info.mount;
 304                 inode = info.inode;
 305                 LOG("mount_id = %i, inode = %i", mount_id, inode);
 306         }
 307
 308         // TODO: Also get modifcation time?
 309
 310         // Check if the binary has already been loaded
 311         if( !(pBinary = Binary_GetInfo(mount_id, inode)) )
 312                 pBinary = Binary_DoLoad(mount_id, inode, Path); // Else load it
 313
 314         // Error Check
 315         if(pBinary == NULL) {
 316                 LEAVE('x', 0);
 317                 return 0;
 318         }
 319
 320         // Map into process space
 321         base = Binary_MapIn(pBinary, Path, BIN_LOWEST, BIN_HIGHEST);
 322
 323         // Check for errors
 324         if(base == 0) {
 325                 LEAVE('x', 0);
 326                 return 0;
 327         }
 328
 329         // Interpret
 330         if(pBinary->Interpreter) {
 331                 tVAddr  start;
 332                 if( Binary_Load(pBinary->Interpreter, &start) == 0 ) {
 333                         Log_Error("Binary", "Can't load interpeter '%s' for '%s'",
 334                                 pBinary->Interpreter, Path);
 335                         LEAVE('x', 0);
 336                         return 0;
 337                 }
 338                 *EntryPoint = start;
 339         }
 340         else
 341                 *EntryPoint = pBinary->Entry - pBinary->Base + base;
 342
 343         // Return
 344         LOG("*EntryPoint = 0x%x", *EntryPoint);
 345         LEAVE('x', base);
 346         return base;    // Pass the base as an argument to the user if there is an interpreter
 347 }
 348
 349 /**
 350  * \brief Finds a matching binary entry
 351  * \param MountID       Mountpoint ID of binary file
 352  * \param InodeID       Inode ID of the file
 353  * \return Pointer to the binary definition (if already loaded)
 354  */
 355 tBinary *Binary_GetInfo(tMount MountID, tInode InodeID)
 356 {
 357         tBinary *pBinary;
 358         for(pBinary = glLoadedBinaries; pBinary; pBinary = pBinary->Next)
 359         {
 360                 if(pBinary->MountID == MountID && pBinary->Inode == InodeID)
 361                         return pBinary;
 362         }
 363         return NULL;
 364 }
 365
 366 /**
 367  * \brief Maps an already-loaded binary into an address space.
 368  * \param Binary        Pointer to globally stored binary definition
 369  * \param Path  Path to the binary's file (for debug)
 370  * \param LoadMin       Lowest location to map to
 371  * \param LoadMax       Highest location to map to
 372  * \return Base load address
 373  */
 374 tVAddr Binary_MapIn(tBinary *Binary, const char *Path, tVAddr LoadMin, tVAddr LoadMax)
 375 {
 376         tVAddr  base;
 377          int    i, fd;
 378
 379         ENTER("pBinary sPath pLoadMin pLoadMax", Binary, Path, LoadMin, LoadMax);
 380
 381         // Reference Executable (Makes sure that it isn't unloaded)
 382         Binary->ReferenceCount ++;
 383
 384         // Get Binary Base
 385         base = Binary->Base;
 386
 387         // Check if base is free
 388         if(base != 0)
 389         {
 390                 LOG("Checking base %p", base);
 391                 for( i = 0; i < Binary->NumSections; i ++ )
 392                 {
 393                         if( Binary_int_CheckMemFree( Binary->LoadSections[i].Virtual, Binary->LoadSections[i].MemSize ) )
 394                         {
 395                                 base = 0;
 396                                 LOG("Address 0x%x is taken\n", Binary->LoadSections[i].Virtual);
 397                                 break;
 398                         }
 399                 }
 400         }
 401
 402         // Check if the executable has no base or it is not free
 403         if(base == 0)
 404         {
 405                 // If so, give it a base
 406                 base = LoadMax;
 407                 while(base >= LoadMin)
 408                 {
 409                         for( i = 0; i < Binary->NumSections; i ++ )
 410                         {
 411                                 tVAddr  addr = Binary->LoadSections[i].Virtual - Binary->Base + base;
 412                                 size_t  size = Binary->LoadSections[i].MemSize;
 413                                 if( addr + size > LoadMax )
 414                                         break;
 415                                 if( Binary_int_CheckMemFree( addr, size ) )
 416                                         break;
 417                         }
 418                         // If space was found, break
 419                         if(i == Binary->NumSections)            break;
 420                         // Else decrement pointer and try again
 421                         base -= BIN_GRANUALITY;
 422                 }
 423                 LOG("Allocated base %p", base);
 424         }
 425
 426         // Error Check
 427         if(base < LoadMin) {
 428                 Log_Warning("Binary", "Executable '%s' cannot be loaded, no space", Path);
 429                 LEAVE('i', 0);
 430                 return 0;
 431         }
 432
 433         // Map Executable In
 434         if( Binary->MountID )
 435                 fd = VFS_OpenInode(Binary->MountID, Binary->Inode, VFS_OPENFLAG_READ);
 436         else
 437                 fd = VFS_Open(Path, VFS_OPENFLAG_READ);
 438         for( i = 0; i < Binary->NumSections; i ++ )
 439         {
 440                 tBinarySection  *sect = &Binary->LoadSections[i];
 441                 Uint    protflags, mapflags;
 442                 tVAddr  addr = sect->Virtual - Binary->Base + base;
 443                 LOG("%i - %p, 0x%x bytes from offset 0x%llx (%x)", i, addr, sect->FileSize, sect->Offset, sect->Flags);
 444
 445                 protflags = MMAP_PROT_READ;
 446                 mapflags = MMAP_MAP_FIXED;
 447
 448                 if( sect->Flags & BIN_SECTFLAG_EXEC )
 449                         protflags |= MMAP_PROT_EXEC;
 450                 // Read only pages are COW
 451                 if( sect->Flags & BIN_SECTFLAG_RO  ) {
 452                         VFS_MMap( (void*)addr, sect->FileSize, protflags, MMAP_MAP_SHARED|mapflags, fd, sect->Offset );
 453                 }
 454                 else {
 455                         protflags |= MMAP_PROT_WRITE;
 456                         VFS_MMap( (void*)addr, sect->FileSize, protflags, MMAP_MAP_PRIVATE|mapflags, fd, sect->Offset );
 457                 }
 458
 459                 // Apply anonymous memory for BSS
 460                 if( sect->FileSize < sect->MemSize ) {
 461                         mapflags |= MMAP_MAP_ANONYMOUS;
 462                         VFS_MMap(
 463                                 (void*)(addr + sect->FileSize), sect->MemSize - sect->FileSize,
 464                                 protflags, MMAP_MAP_PRIVATE|mapflags,
 465                                 0, 0
 466                                 );
 467                 }
 468         }
 469
 470         Log_Debug("Binary", "PID %i - Mapped '%s' to %p", Threads_GetPID(), Path, base);
 471         VFS_Close(fd);
 472
 473         LEAVE('p', base);
 474         return base;
 475 }
 476
 477 #if 0
 478 /**
 479  * \fn Uint Binary_IsMapped(tBinary *binary)
 480  * \brief Check if a binary is already mapped into the address space
 481  * \param binary        Binary information to check
 482  * \return Current Base or 0
 483  */
 484 Uint Binary_IsMapped(tBinary *binary)
 485 {
 486         Uint    iBase;
 487
 488         // Check prefered base
 489         iBase = binary->Base;
 490         if(MM_GetPage( iBase ) == (binary->Pages[0].Physical & ~0xFFF))
 491                 return iBase;
 492
 493         for(iBase = BIN_HIGHEST;
 494                 iBase >= BIN_LOWEST;
 495                 iBase -= BIN_GRANUALITY)
 496         {
 497                 if(MM_GetPage( iBase ) == (binary->Pages[0].Physical & ~0xFFF))
 498                         return iBase;
 499         }
 500
 501         return 0;
 502 }
 503 #endif
 504
 505 /**
 506  * \fn tBinary *Binary_DoLoad(char *truePath)
 507  * \brief Loads a binary file into memory
 508  * \param truePath      Absolute filename of binary
 509  */
 510 tBinary *Binary_DoLoad(tMount MountID, tInode Inode, const char *Path)
 511 {
 512         tBinary *pBinary;
 513          int    fp;
 514         Uint32  ident;
 515         tBinaryType     *bt = gRegBinTypes;
 516
 517         ENTER("iMountID XInode sPath", MountID, Inode, Path);
 518
 519         // Open File
 520         if( MountID )
 521         {
 522                 fp = VFS_OpenInode(MountID, Inode, VFS_OPENFLAG_READ);
 523         }
 524         else
 525         {
 526                 fp = VFS_Open(Path, VFS_OPENFLAG_READ);
 527         }
 528         if(fp == -1) {
 529                 LOG("Unable to load file, access denied");
 530                 LEAVE('n');
 531                 return NULL;
 532         }
 533
 534         LOG("fp = 0x%x", fp);
 535
 536         // Read File Type
 537         VFS_Read(fp, 4, &ident);
 538         VFS_Seek(fp, 0, SEEK_SET);
 539
 540         LOG("ident = 0x%x", ident);
 541
 542         // Determine the type
 543         for(; bt; bt = bt->Next)
 544         {
 545                 if( (ident & bt->Mask) != (Uint32)bt->Ident )
 546                         continue;
 547                 LOG("bt = %p (%s)", bt, bt->Name);
 548                 pBinary = bt->Load(fp);
 549                 break;
 550         }
 551
 552         // Close File
 553         VFS_Close(fp);
 554
 555         // Catch errors
 556         if(!bt) {
 557                 Log_Warning("Binary", "'%s' is an unknown file type. (%02x %02x %02x %02x)",
 558                         Path, ident&0xFF, (ident>>8)&0xFF, (ident>>16)&0xFF, (ident>>24)&0xFF);
 559                 LEAVE('n');
 560                 return NULL;
 561         }
 562
 563         LOG("pBinary = %p", pBinary);
 564
 565         // Error Check
 566         if(pBinary == NULL) {
 567                 LEAVE('n');
 568                 return NULL;
 569         }
 570
 571         // Initialise Structure
 572         pBinary->ReferenceCount = 0;
 573         pBinary->MountID = MountID;
 574         pBinary->Inode = Inode;
 575
 576         // Debug Information
 577         LOG("Interpreter: '%s'", pBinary->Interpreter);
 578         LOG("Base: 0x%x, Entry: 0x%x", pBinary->Base, pBinary->Entry);
 579         LOG("NumSections: %i", pBinary->NumSections);
 580
 581         // Add to the list
 582         SHORTLOCK(&glBinListLock);
 583         pBinary->Next = glLoadedBinaries;
 584         glLoadedBinaries = pBinary;
 585         SHORTREL(&glBinListLock);
 586
 587         // TODO: Register the path with the binary
 588
 589         // Return
 590         LEAVE('p', pBinary);
 591         return pBinary;
 592 }
 593
 594 /**
 595  * \fn void Binary_Unload(void *Base)
 596  * \brief Unload / Unmap a binary
 597  * \param Base  Loaded Base
 598  * \note Currently used only for kernel libaries
 599  */
 600 void Binary_Unload(void *Base)
 601 {
 602         tKernelBin      *pKBin;
 603         tKernelBin      *prev = NULL;
 604          int    i;
 605
 606         if((Uint)Base < 0xC0000000)
 607         {
 608                 // TODO: User Binaries
 609                 Log_Warning("BIN", "Unloading user binaries is currently unimplemented");
 610                 return;
 611         }
 612
 613         // Kernel Libraries
 614         for(pKBin = glLoadedKernelLibs;
 615                 pKBin;
 616                 prev = pKBin, pKBin = pKBin->Next)
 617         {
 618                 // Check the base
 619                 if(pKBin->Base != Base) continue;
 620                 // Deallocate Memory
 621                 for(i = 0; i < pKBin->Info->NumSections; i++)
 622                 {
 623                         // TODO: VFS_MUnmap();
 624                 }
 625                 // Dereference Binary
 626                 Binary_Dereference( pKBin->Info );
 627                 // Remove from list
 628                 if(prev)        prev->Next = pKBin->Next;
 629                 else            glLoadedKernelLibs = pKBin->Next;
 630                 // Free Kernel Lib
 631                 free(pKBin);
 632                 return;
 633         }
 634 }
 635
 636 /**
 637  * \fn void Binary_Dereference(tBinary *Info)
 638  * \brief Dereferences and if nessasary, deletes a binary
 639  * \param Info  Binary information structure
 640  */
 641 void Binary_Dereference(tBinary *Info)
 642 {
 643         // Decrement reference count
 644         Info->ReferenceCount --;
 645
 646         // Check if it is still in use
 647         if(Info->ReferenceCount)        return;
 648
 649         /// \todo Implement binary freeing
 650 }
 651
 652 /**
 653  * \fn char *Binary_RegInterp(char *Path)
 654  * \brief Registers an Interpreter
 655  * \param Path  Path to interpreter provided by executable
 656  */
 657 char *Binary_RegInterp(char *Path)
 658 {
 659          int    i;
 660         // NULL Check Argument
 661         if(Path == NULL)        return NULL;
 662         // NULL Check the array
 663         if(gsaRegInterps == NULL)
 664         {
 665                 giRegInterps = 1;
 666                 gsaRegInterps = malloc( sizeof(char*) );
 667                 gsaRegInterps[0] = malloc( strlen(Path) );
 668                 strcpy(gsaRegInterps[0], Path);
 669                 return gsaRegInterps[0];
 670         }
 671
 672         // Scan Array
 673         for( i = 0; i < giRegInterps; i++ )
 674         {
 675                 if(strcmp(gsaRegInterps[i], Path) == 0)
 676                         return gsaRegInterps[i];
 677         }
 678
 679         // Interpreter is not in list
 680         giRegInterps ++;
 681         gsaRegInterps = malloc( sizeof(char*)*giRegInterps );
 682         gsaRegInterps[i] = malloc( strlen(Path) );
 683         strcpy(gsaRegInterps[i], Path);
 684         return gsaRegInterps[i];
 685 }
 686
 687 // ============
 688 // Kernel Binary Handling
 689 // ============
 690 /**
 691  * \fn void *Binary_LoadKernel(const char *File)
 692  * \brief Load a binary into kernel space
 693  * \note This function shares much with #Binary_Load, but does it's own mapping
 694  * \param File  File to load into the kernel
 695  */
 696 void *Binary_LoadKernel(const char *File)
 697 {
 698         tBinary *pBinary;
 699         tKernelBin      *pKBinary;
 700         tVAddr  base = -1;
 701         tMount  mount_id;
 702         tInode  inode;
 703
 704         ENTER("sFile", File);
 705
 706         // Sanity Check Argument
 707         if(File == NULL) {
 708                 LEAVE('n');
 709                 return 0;
 710         }
 711
 712         {
 713                 int fd = VFS_Open(File, VFS_OPENFLAG_READ);
 714                 tFInfo  info;
 715                 if(fd == -1) {
 716                         LOG("Opening failed");
 717                         LEAVE('n');
 718                         return NULL;
 719                 }
 720                 VFS_FInfo(fd, &info, 0);
 721                 mount_id = info.mount;
 722                 inode = info.inode;
 723                 VFS_Close(fd);
 724                 LOG("Mount %i, Inode %lli", mount_id, inode);
 725         }
 726
 727         // Check if the binary has already been loaded
 728         if( (pBinary = Binary_GetInfo(mount_id, inode)) )
 729         {
 730                 for(pKBinary = glLoadedKernelLibs;
 731                         pKBinary;
 732                         pKBinary = pKBinary->Next )
 733                 {
 734                         if(pKBinary->Info == pBinary) {
 735                                 LOG("Already loaded");
 736                                 LEAVE('p', pKBinary->Base);
 737                                 return pKBinary->Base;
 738                         }
 739                 }
 740         }
 741         else
 742                 pBinary = Binary_DoLoad(mount_id, inode, File); // Else load it
 743
 744         // Error Check
 745         if(pBinary == NULL) {
 746                 LEAVE('n');
 747                 return NULL;
 748         }
 749
 750         LOG("Loaded as %p", pBinary);
 751         // --------------
 752         // Now pBinary is valid (either freshly loaded or only user mapped)
 753         // So, map it into kernel space
 754         // --------------
 755
 756         // Reference Executable (Makes sure that it isn't unloaded)
 757         pBinary->ReferenceCount ++;
 758
 759         base = Binary_MapIn(pBinary, File, KLIB_LOWEST, KLIB_HIGHEST);
 760         if( base == 0 ) {
 761                 LEAVE('n');
 762                 return 0;
 763         }
 764
 765         // Add to list
 766         // TODO: Could this cause race conditions if a binary isn't fully loaded when used
 767         pKBinary = malloc(sizeof(*pKBinary));
 768         pKBinary->Base = (void*)base;
 769         pKBinary->Info = pBinary;
 770         SHORTLOCK( &glKBinListLock );
 771         pKBinary->Next = glLoadedKernelLibs;
 772         glLoadedKernelLibs = pKBinary;
 773         SHORTREL( &glKBinListLock );
 774
 775         LEAVE('p', base);
 776         return (void*)base;
 777 }
 778
 779 /**
 780  * \fn Uint Binary_Relocate(void *Base)
 781  * \brief Relocates a loaded binary (used by kernel libraries)
 782  * \param Base  Loaded base address of binary
 783  * \return Boolean Success
 784  */
 785 Uint Binary_Relocate(void *Base)
 786 {
 787         Uint32  ident = *(Uint32*) Base;
 788         tBinaryType     *bt = gRegBinTypes;
 789
 790         for( ; bt; bt = bt->Next)
 791         {
 792                 if( (ident & bt->Mask) == (Uint)bt->Ident )
 793                         return bt->Relocate( (void*)Base);
 794         }
 795
 796         Log_Warning("BIN", "%p is an unknown file type. (%02x %02x %02x %02x)",
 797                 Base, ident&0xFF, (ident>>8)&0xFF, (ident>>16)&0xFF, (ident>>24)&0xFF);
 798         return 0;
 799 }
 800
 801 /**
 802  * \fn int Binary_GetSymbol(char *Name, Uint *Val)
 803  * \brief Get a symbol value
 804  * \return Value of symbol or -1 on error
 805  *
 806  * Gets the value of a symbol from either the currently loaded
 807  * libraries or the kernel's exports.
 808  */
 809 int Binary_GetSymbol(const char *Name, Uint *Val)
 810 {
 811         if( Binary_GetSymbolEx(Name, Val) )     return 1;
 812         return 0;
 813 }
 814
 815 /**
 816  * \fn Uint Binary_GetSymbolEx(char *Name, Uint *Value)
 817  * \brief Get a symbol value
 818  *
 819  * Gets the value of a symbol from either the currently loaded
 820  * libraries or the kernel's exports.
 821  */
 822 Uint Binary_GetSymbolEx(const char *Name, Uint *Value)
 823 {
 824         tKernelBin      *pKBin;
 825          int    numKSyms = ((Uint)&gKernelSymbolsEnd-(Uint)&gKernelSymbols)/sizeof(tKernelSymbol);
 826
 827         // Scan Kernel
 828         for( int i = 0; i < numKSyms; i++ )
 829         {
 830                 if(strcmp(Name, gKernelSymbols[i].Name) == 0) {
 831                         LOG("KSym %s = %p", gKernelSymbols[i].Name, gKernelSymbols[i].Value);
 832                         *Value = gKernelSymbols[i].Value;
 833                         return 1;
 834                 }
 835         }
 836
 837
 838         // Scan Loaded Libraries
 839         for(pKBin = glLoadedKernelLibs;
 840                 pKBin;
 841                 pKBin = pKBin->Next )
 842         {
 843                 if( Binary_FindSymbol(pKBin->Base, Name, Value) ) {
 844                         return 1;
 845                 }
 846         }
 847
 848         Log_Warning("BIN", "Unable to find symbol '%s'", Name);
 849         return 0;
 850 }
 851
 852 /**
 853  * \fn Uint Binary_FindSymbol(void *Base, char *Name, Uint *Val)
 854  * \brief Get a symbol from the specified library
 855  * \param Base  Base address
 856  * \param Name  Name of symbol to find
 857  * \param Val   Pointer to place final value
 858  */
 859 Uint Binary_FindSymbol(void *Base, const char *Name, Uint *Val)
 860 {
 861         Uint32  ident = *(Uint32*) Base;
 862         tBinaryType     *bt = gRegBinTypes;
 863
 864         for(; bt; bt = bt->Next)
 865         {
 866                 if( (ident & bt->Mask) == (Uint)bt->Ident )
 867                         return bt->GetSymbol(Base, Name, Val);
 868         }
 869
 870         Log_Warning("BIN", "Binary_FindSymbol - %p is an unknown file type. (%02x %02x %02x %02x)",
 871                 Base, ident&0xFF, ident>>8, ident>>16, ident>>24);
 872         return 0;
 873 }
 874
 875 /**
 876  * \brief Check if a range of memory is fully free
 877  * \return Inverse boolean free (0 if all pages are unmapped)
 878  */
 879 int Binary_int_CheckMemFree( tVAddr _start, size_t _len )
 880 {
 881         ENTER("p_start x_len", _start, _len);
 882
 883         _len += _start & (PAGE_SIZE-1);
 884         _len = (_len + PAGE_SIZE - 1) & ~(PAGE_SIZE-1);
 885         _start &= ~(PAGE_SIZE-1);
 886         LOG("_start = %p, _len = 0x%x", _start, _len);
 887         for( ; _len > PAGE_SIZE; _len -= PAGE_SIZE, _start += PAGE_SIZE ) {
 888                 if( MM_GetPhysAddr( (void*)_start ) != 0 ) {
 889                         LEAVE('i', 1);
 890                         return 1;
 891                 }
 892         }
 893         if( _len == PAGE_SIZE && MM_GetPhysAddr( (void*)_start ) != 0 ) {
 894                 LEAVE('i', 1);
 895                 return 1;
 896         }
 897         LEAVE('i', 0);
 898         return 0;
 899 }
 900
 901
 902 // === EXPORTS ===
 903 EXPORT(Binary_FindSymbol);
 904 EXPORT(Binary_Unload);