AcessNative/ld-acess_src/exports.c

   1 /*
   2  * AcessNative
   3  *
   4  * exports.c
   5  * - Exported functions
   6  */
   7 #define DONT_INCLUDE_SYSCALL_NAMES 1
   8 #include "../../Usermode/Libraries/ld-acess.so_src/include_exp/acess/sys.h"
   9 #include "../syscalls.h"
  10 #include "exports.h"
  11 #include <stdarg.h>
  12 #include <stddef.h>
  13
  14 #define DEBUG(v...)     do{if(gbSyscallDebugEnabled)Debug(v);}while(0)
  15 #define PAGE_SIZE       4096
  16
  17 #define TODO()  Warning("TODO: %s", __func__)
  18
  19 typedef struct sFILE    FILE;
  20
  21 extern void     exit(int) __attribute__ ((noreturn));
  22 extern int      printf(const char *, ...);
  23 extern int      fprintf(FILE *,const char *, ...);
  24 extern int      sprintf(char *,const char *, ...);
  25 extern int      vprintf(const char *, va_list);
  26 extern int      strncmp(const char *, const char *, size_t);
  27
  28 extern int      gSocket;
  29 extern int      giSyscall_ClientID;     // Needed for execve
  30 extern void     _InitSyscalls(void);
  31 extern void     _CloseSyscalls(void);
  32
  33 extern void     Warning(const char *Format, ...);
  34 extern void     Debug(const char *Format, ...);
  35 extern int      AllocateMemory(uintptr_t VirtAddr, size_t ByteCount);
  36
  37 // === CONSTANTS ===
  38 #define NATIVE_FILE_MASK        0x40000000
  39
  40 // === GLOBALS ===
  41 int     acess__errno;
  42  int    gbSyscallDebugEnabled = 0;
  43 char    *gsExecutablePath = "./ld-acess";
  44
  45 // === CODE ===
  46 // --- VFS Calls
  47 int acess__SysChdir(const char *Path)
  48 {
  49         return _Syscall(SYS_CHDIR, ">s", Path);
  50 }
  51
  52 int acess__SysOpen(const char *Path, unsigned int Flags)
  53 {
  54         if( strncmp(Path, "$$$$", 4) == 0 )
  55         {
  56                 return native_open(Path, Flags) | NATIVE_FILE_MASK;
  57         }
  58         SYSTRACE("open(\"%s\", 0x%x)", Path, Flags);
  59         return _Syscall(SYS_OPEN, ">s >i", Path, Flags);
  60 }
  61
  62 void acess__SysClose(int FD)
  63 {
  64         if(FD & NATIVE_FILE_MASK) {
  65                 return native_close(FD & (NATIVE_FILE_MASK-1));
  66         }
  67         SYSTRACE("close(%i)", FD);
  68         _Syscall(SYS_CLOSE, ">i", FD);
  69 }
  70
  71 int acess__SysReopen(int FD, const char *Path, int Flags) {
  72         SYSTRACE("reopen(0x%x, \"%s\", 0x%x)", FD, Path, Flags);
  73         return _Syscall(SYS_REOPEN, ">i >s >i", FD, Path, Flags);
  74 }
  75
  76 int acess__SysCopyFD(int srcfd, int dstfd) {
  77         SYSTRACE("_SysCopyFD(%i, %i)", srcfd, dstfd);
  78         return _Syscall(SYS_COPYFD, ">i >i", srcfd, dstfd);
  79 }
  80
  81 int acess__SysFDFlags(int fd, int mask, int newflags) {
  82         return _Syscall(SYS_FDFLAGS, ">i >i >i", fd, mask, newflags);
  83 }
  84
  85 size_t acess__SysRead(int FD, void *Dest, size_t Bytes) {
  86         if(FD & NATIVE_FILE_MASK)
  87                 return native_read(FD & (NATIVE_FILE_MASK-1), Dest, Bytes);
  88         SYSTRACE("_SysRead(0x%x, 0x%x, *%p)", FD, Bytes, Dest);
  89         return _Syscall(SYS_READ, ">i >i <d", FD, Bytes, Bytes, Dest);
  90 }
  91
  92 size_t acess__SysWrite(int FD, const void *Src, size_t Bytes) {
  93         if(FD & NATIVE_FILE_MASK)
  94                 return native_write(FD & (NATIVE_FILE_MASK-1), Src, Bytes);
  95         SYSTRACE("_SysWrite(0x%x, 0x%x, %p\"%.*s\")", FD, Bytes, Src, Bytes, (char*)Src);
  96         return _Syscall(SYS_WRITE, ">i >i >d", FD, Bytes, Bytes, Src);
  97 }
  98 uint64_t acess__SysTruncate(int fd, uint64_t size) {
  99         TODO();
 100         return 0;
 101 }
 102
 103 int acess__SysSeek(int FD, int64_t Ofs, int Dir)
 104 {
 105         if(FD & NATIVE_FILE_MASK) {
 106                 return native_seek(FD & (NATIVE_FILE_MASK-1), Ofs, Dir);
 107         }
 108         SYSTRACE("_SysSeek(0x%x, 0x%llx, %i)", FD, Ofs, Dir);
 109         return _Syscall(SYS_SEEK, ">i >I >i", FD, Ofs, Dir);
 110 }
 111
 112 uint64_t acess__SysTell(int FD)
 113 {
 114         if(FD & NATIVE_FILE_MASK)
 115                 return native_tell( FD & (NATIVE_FILE_MASK-1) );
 116         SYSTRACE("_SysTell(0x%x)", FD);
 117         return _Syscall(SYS_TELL, ">i", FD);
 118 }
 119
 120 int acess__SysIOCtl(int fd, int id, void *data) {
 121          int    len;
 122         SYSTRACE("_SysIOCtl(%i, %i, %p)", fd, id, data);
 123         // NOTE: The length here is hacky and could break
 124         len = (data == NULL ? 0 : PAGE_SIZE - ((uintptr_t)data % PAGE_SIZE));
 125         return _Syscall(SYS_IOCTL, ">i >i ?d", fd, id, len, data);
 126 }
 127 int acess__SysFInfo(int fd, t_sysFInfo *info, int maxacls) {
 128         SYSTRACE("_SysFInfo(%i, %p, %i)", fd, info, maxacls);
 129         return _Syscall(SYS_FINFO, ">i <d >i",
 130                 fd,
 131                 sizeof(t_sysFInfo)+maxacls*sizeof(t_sysACL), info,
 132                 maxacls
 133                 );
 134 }
 135
 136 int acess__SysReadDir(int fd, char *dest) {
 137         SYSTRACE("_SysReadDir(%i, %p)", fd, dest);
 138         return _Syscall(SYS_READDIR, ">i <d", fd, 256, dest);
 139 }
 140
 141 int acess__SysSelect(int nfds, fd_set *read, fd_set *write, fd_set *error, int64_t *timeout, uint32_t events)
 142 {
 143         SYSTRACE("_SysSelect(%i, %p, %p, %p, %p, 0x%x)", nfds, read, write, error, timeout, events);
 144         return _Syscall(SYS_SELECT, ">i ?d ?d ?d >d >i", nfds,
 145                 read ? (nfds+7)/8 : 0, read,
 146                 write ? (nfds+7)/8 : 0, write,
 147                 error ? (nfds+7)/8 : 0, error,
 148                 sizeof(*timeout), timeout,
 149                 events
 150                 );
 151 }
 152 int acess__SysMkDir(const char *pathname)
 153 {
 154         TODO();
 155         return 0;
 156 }
 157 int acess__SysUnlink(const char *pathname)
 158 {
 159         // TODO:
 160         TODO();
 161         return 0;
 162 }
 163 void* acess__SysMMap(void *addr, size_t length, unsigned int _flags, int fd, uint64_t offset)
 164 {
 165         TODO();
 166         return NULL;
 167 }
 168 int acess__SysMUnMap(void *addr, size_t length)
 169 {
 170         TODO();
 171         return 0;
 172 }
 173 uint64_t acess__SysMarshalFD(int FD)
 174 {
 175         TODO();
 176         return 0;
 177 }
 178 int acess__SysUnMarshalFD(uint64_t Handle)
 179 {
 180         TODO();
 181         return -1;
 182 }
 183
 184 int acess__SysOpenChild(int fd, char *name, int flags) {
 185         SYSTRACE("_SysOpenChild(0x%x, '%s', 0x%x)", fd, name, flags);
 186         return _Syscall(SYS_OPENCHILD, ">i >s >i", fd, name, flags);
 187 }
 188
 189 int acess__SysGetACL(int fd, t_sysACL *dest) {
 190         SYSTRACE("%s(0x%x, %p)", __func__, fd, dest);
 191         return _Syscall(SYS_GETACL, ">i <d", fd, sizeof(t_sysACL), dest);
 192 }
 193
 194 int acess__SysMount(const char *Device, const char *Directory, const char *Type, const char *Options) {
 195         SYSTRACE("_SysMount('%s', '%s', '%s', '%s')", Device, Directory, Type, Options);
 196         return _Syscall(SYS_MOUNT, ">s >s >s >s", Device, Directory, Type, Options);
 197 }
 198
 199
 200 // --- Error Handler
 201 int acess__SysSetFaultHandler(int (*Handler)(int)) {
 202         printf("TODO: Set fault handler (asked to set to %p)\n", Handler);
 203         return 0;
 204 }
 205
 206 void acess__SysSetName(const char *Name)
 207 {
 208         // TODO:
 209         TODO();
 210 }
 211
 212 int acess__SysGetName(char *NameDest)
 213 {
 214         // TODO:
 215         TODO();
 216         return 0;
 217 }
 218
 219 int acess__SysSetPri(int Priority)
 220 {
 221         // TODO:
 222         TODO();
 223         return 0;
 224 }
 225
 226 // --- Binaries? ---
 227 void *acess_SysLoadBin(const char *path, void **entry)
 228 {
 229         // ERROR!
 230         TODO();
 231         return NULL;
 232 }
 233
 234 int acess__SysUnloadBin(void *base)
 235 {
 236         // ERROR!
 237         TODO();
 238         return -1;
 239 }
 240
 241 // --- System ---
 242 int acess__SysLoadModule(const char *Path)
 243 {
 244         TODO();
 245         return -1;
 246 }
 247
 248 // --- Timekeeping ---
 249 int64_t acess__SysTimestamp(void)
 250 {
 251         // TODO: Better impl
 252         TODO();
 253 //      return now()*1000;
 254         return 0;
 255 }
 256
 257 // --- Memory Management ---
 258 uint64_t acess__SysGetPhys(uintptr_t vaddr)
 259 {
 260         // TODO:
 261         TODO();
 262         return 0;
 263 }
 264
 265 uint64_t acess__SysAllocate(uintptr_t vaddr)
 266 {
 267         if( AllocateMemory(vaddr, 0x1000) == -1 )       // Allocate a page
 268                 return 0;
 269
 270         return vaddr;   // Just ignore the need for paddrs :)
 271 }
 272
 273 // --- Process Management ---
 274 int acess__SysClone(int flags, void *stack)
 275 {
 276         #ifdef __WIN32__
 277         Warning("Win32 does not support anything like fork(2), cannot emulate");
 278         exit(-1);
 279         #else
 280         extern int fork(void);
 281         if(flags & CLONE_VM) {
 282                  int    ret, newID, kernel_tid=0;
 283                 Debug("USERSIDE fork()");
 284
 285                 newID = _Syscall(SYS_AN_FORK, "<d", sizeof(int), &kernel_tid);
 286                 ret = fork();
 287                 if(ret < 0) {
 288                         return ret;
 289                 }
 290
 291                 if(ret == 0)
 292                 {
 293                         _CloseSyscalls();
 294                         giSyscall_ClientID = newID;
 295                         _InitSyscalls();
 296                         return 0;
 297                 }
 298
 299                 // Return the acess TID instead
 300                 return kernel_tid;
 301         }
 302         else
 303         {
 304                 Warning("ERROR: Threads currently unsupported\n");
 305                 exit(-1);
 306         }
 307         #endif
 308 }
 309
 310 int acess__SysKill(int pid, int sig)
 311 {
 312         // TODO: Impliment SysKill
 313         return -1;
 314 }
 315
 316 int acess__SysExecVE(char *path, char **argv, const char **envp)
 317 {
 318          int    i, argc;
 319
 320         DEBUG("acess_execve: (path='%s', argv=%p, envp=%p)", path, argv, envp);
 321
 322         // Get argument count
 323         for( argc = 0; argv[argc]; argc ++ ) ;
 324         DEBUG(" acess_execve: argc = %i", argc);
 325
 326         const char      *new_argv[7+argc+1];
 327         char    client_id_str[11];
 328         char    socket_fd_str[11];
 329         sprintf(client_id_str, "%i", giSyscall_ClientID);
 330         sprintf(socket_fd_str, "%i", gSocket);
 331         new_argv[0] = "ld-acess";       // TODO: Get path to ld-acess executable
 332         new_argv[1] = "--key";          // Set client ID for Request.c
 333         new_argv[2] = client_id_str;
 334         new_argv[3] = "--socket";       // Socket
 335         new_argv[4] = socket_fd_str;
 336         new_argv[5] = "--binary";       // Set the binary path (instead of using argv[0])
 337         new_argv[6] = path;
 338         for( i = 0; i < argc; i ++ )    new_argv[7+i] = argv[i];
 339         new_argv[7+i] = NULL;
 340
 341         #if 1
 342         argc += 7;
 343         for( i = 0; i < argc; i ++ )
 344                 printf("\"%s\" ", new_argv[i]);
 345         printf("\n");
 346         if(envp)
 347         {
 348                 printf("envp = %p\n", envp);
 349                 for( i = 0; envp[i]; i ++ )
 350                         printf("%i: \"%s\"\n", i, envp[i]);
 351                 printf("envc = %i\n", i);
 352         }
 353         #endif
 354
 355         // Call actual execve
 356         return native_execve("./ld-acess", new_argv, envp);
 357 }
 358
 359 int acess__SysSpawn(const char *binary, const char **argv, const char **envp, int nfd, int fds[], struct s_sys_spawninfo *info)
 360 {
 361          int    argc = 0;
 362         while( argv[argc++] );
 363
 364         Debug("_SysSpawn('%s', %p (%i), %p, %i, %p, %p)",
 365                 binary, argv, argc, envp, nfd, fds, info);
 366
 367          int    kernel_tid;
 368          int    newID;
 369         newID = _Syscall(SYS_AN_SPAWN, "<d >d >d",
 370                 sizeof(int), &kernel_tid,
 371                 nfd*sizeof(int), fds,
 372                 info ? sizeof(*info) : 0, info);
 373
 374         const char      *new_argv[5+argc+1];
 375          int    new_argc = 0, i;
 376         char    client_id_str[11];
 377         sprintf(client_id_str, "%i", newID);
 378         new_argv[new_argc++] = gsExecutablePath;       // TODO: Get path to ld-acess executable
 379         new_argv[new_argc++] = "--key";
 380         new_argv[new_argc++] = client_id_str;
 381         new_argv[new_argc++] = "--binary";
 382         new_argv[new_argc++] = binary;
 383         for( i = 0; argv[i]; i ++)
 384                 new_argv[new_argc++] = argv[i];
 385         new_argv[new_argc++] = NULL;
 386
 387         // TODO: Debug output?
 388
 389         native_spawn(gsExecutablePath, new_argv, envp);
 390
 391         return kernel_tid;
 392 }
 393
 394 //void acess_sleep(void)
 395 //{
 396 //      DEBUG("%s()", __func__);
 397 //      _Syscall(SYS_SLEEP, "");
 398 //}
 399
 400 void acess__SysTimedSleep(int64_t Delay)
 401 {
 402         DEBUG("%s(%lli)", __func__, Delay);
 403         // Not accurate, but fuck it
 404         //if( Delay > 1000 )    sleep(Delay / 1000);
 405         //if( Delay % 1000 )    usleep( (Delay % 1000) * 1000 );
 406         //_Syscall(SYS_TIMEDSLEEP, ">I", Delay);
 407 }
 408
 409 int acess__SysWaitTID(int TID, int *ExitStatus)
 410 {
 411         DEBUG("%s(%i, %p)", __func__, TID, ExitStatus);
 412         return _Syscall(SYS_WAITTID, ">i <d", TID, sizeof(int), &ExitStatus);
 413 }
 414
 415 int acess_setuid(int ID) { return _Syscall(SYS_SETUID, ">i", ID); }
 416 int acess_setgid(int ID) { return _Syscall(SYS_SETGID, ">i", ID); }
 417 int acess_gettid(void) { return _Syscall(SYS_GETTID, ""); }
 418 int acess__SysGetPID(void) { return _Syscall(SYS_GETPID, ""); }
 419 int acess__SysGetUID(void) { return _Syscall(SYS_GETUID, ""); }
 420 int acess__SysGetGID(void) { return _Syscall(SYS_GETGID, ""); }
 421 int acess_getgid(void) { return _Syscall(SYS_GETGID, ""); }
 422
 423 int acess__SysSendMessage(int DestTID, int Length, void *Data)
 424 {
 425         DEBUG("%s(%i, 0x%x, %p)", __func__, DestTID, Length, Data);
 426         return _Syscall(SYS_SENDMSG, ">i >d", DestTID, Length, Data);
 427 }
 428
 429 int acess__SysGetMessage(int *SourceTID, int BufLen, void *Data)
 430 {
 431         DEBUG("%s(%p, %p)", __func__, SourceTID, Data);
 432         return _Syscall(SYS_GETMSG, "<d <d",
 433                 SourceTID ? sizeof(uint32_t) : 0, SourceTID,
 434                 BufLen, Data
 435                 );
 436 }
 437
 438 int acess__SysWaitEvent(int Mask)
 439 {
 440         DEBUG("%s(%x)", __func__, Mask);
 441         return _Syscall(SYS_WAITEVENT, ">i", Mask);
 442 }
 443
 444 // --- Logging
 445 void acess__SysDebug(const char *Format, ...)
 446 {
 447         va_list args;
 448
 449         va_start(args, Format);
 450
 451         printf("[_SysDebug %i] ", giSyscall_ClientID);
 452         vprintf(Format, args);
 453         printf("\n");
 454
 455         va_end(args);
 456 }
 457
 458 void acess__exit(int Status)
 459 {
 460         DEBUG("_exit(%i)", Status);
 461         _Syscall(SYS_EXIT, ">i", Status);
 462         exit(Status);
 463 }
 464
 465 uint32_t acess__SysSetMemFlags(uintptr_t vaddr, uint32_t flags, uint32_t mask)
 466 {
 467         // TODO: Impliment acess__SysSetMemFlags?
 468         return 0;
 469 }
 470
 471
 472 // === Symbol List ===
 473 #ifndef DEFSYM
 474 # define DEFSYM(name)   {#name, &acess_##name}
 475 #endif
 476 const tSym      caBuiltinSymbols[] = {
 477         DEFSYM(_exit),
 478
 479         DEFSYM(_SysChdir),
 480         DEFSYM(_SysOpen),
 481         DEFSYM(_SysOpenChild),
 482         DEFSYM(_SysReopen),
 483         DEFSYM(_SysClose),
 484         DEFSYM(_SysRead),
 485         DEFSYM(_SysWrite),
 486         DEFSYM(_SysSeek),
 487         DEFSYM(_SysTell),
 488         DEFSYM(_SysIOCtl),
 489         DEFSYM(_SysFInfo),
 490         DEFSYM(_SysReadDir),
 491         DEFSYM(_SysGetACL),
 492         DEFSYM(_SysMount),
 493         DEFSYM(_SysSelect),
 494         DEFSYM(_SysMkDir),
 495         DEFSYM(_SysUnlink),
 496
 497         DEFSYM(_SysClone),
 498         DEFSYM(_SysExecVE),
 499         DEFSYM(_SysSpawn),
 500 //      DEFSYM(sleep),
 501
 502         DEFSYM(_SysWaitTID),
 503         DEFSYM(gettid),
 504         DEFSYM(_SysGetPID),
 505         DEFSYM(setuid),
 506         DEFSYM(setgid),
 507         DEFSYM(_SysGetUID),
 508         DEFSYM(getgid),
 509
 510         DEFSYM(_SysSendMessage),
 511         DEFSYM(_SysGetMessage),
 512
 513         DEFSYM(_SysAllocate),
 514         DEFSYM(_SysSetMemFlags),
 515         DEFSYM(_SysDebug),
 516         {"_ZN4_sys5debugEPKcz", &acess__SysDebug},
 517         DEFSYM(_SysSetFaultHandler),
 518         DEFSYM(_SysWaitEvent),
 519
 520         DEFSYM(_errno)
 521 };
 522
 523 const int       ciNumBuiltinSymbols = sizeof(caBuiltinSymbols)/sizeof(caBuiltinSymbols[0]);
 524