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{}while(0)//Debug(v)
  15 #define PAGE_SIZE       4096
  16
  17 typedef struct sFILE    FILE;
  18
  19 extern void     exit(int) __attribute__ ((noreturn));
  20 extern int      printf(const char *, ...);
  21 extern int      fprintf(FILE *,const char *, ...);
  22 extern int      sprintf(char *,const char *, ...);
  23 extern int      vprintf(const char *, va_list);
  24 extern int      strncmp(const char *, const char *, size_t);
  25
  26 extern int      gSocket;
  27 extern int      giSyscall_ClientID;     // Needed for execve
  28 extern void     _InitSyscalls(void);
  29 extern void     _CloseSyscalls(void);
  30
  31 extern void     Warning(const char *Format, ...);
  32 extern void     Debug(const char *Format, ...);
  33 extern int      AllocateMemory(uintptr_t VirtAddr, size_t ByteCount);
  34
  35 // === CONSTANTS ===
  36 #define NATIVE_FILE_MASK        0x40000000
  37
  38 // === GLOBALS ===
  39 int     acess__errno;
  40
  41 // === CODE ===
  42 // --- VFS Calls
  43 int acess_chdir(const char *Path)
  44 {
  45         return _Syscall(SYS_CHDIR, ">s", Path);
  46 }
  47
  48 int acess_open(const char *Path, int Flags)
  49 {
  50         if( strncmp(Path, "$$$$", 4) == 0 )
  51         {
  52                 return native_open(Path, Flags) | NATIVE_FILE_MASK;
  53         }
  54         DEBUG("open(\"%s\", 0x%x)", Path, Flags);
  55         return _Syscall(SYS_OPEN, ">s >i", Path, Flags);
  56 }
  57
  58 void acess_close(int FD) {
  59         if(FD & NATIVE_FILE_MASK) {
  60                 return native_close(FD & (NATIVE_FILE_MASK-1));
  61         }
  62         DEBUG("close(%i)", FD);
  63         _Syscall(SYS_CLOSE, ">i", FD);
  64 }
  65
  66 int acess_reopen(int FD, const char *Path, int Flags) {
  67         DEBUG("reopen(0x%x, \"%s\", 0x%x)", FD, Path, Flags);
  68         return _Syscall(SYS_REOPEN, ">i >s >i", FD, Path, Flags);
  69 }
  70
  71 size_t acess_read(int FD, void *Dest, size_t Bytes) {
  72         if(FD & NATIVE_FILE_MASK)
  73                 return native_read(FD & (NATIVE_FILE_MASK-1), Dest, Bytes);
  74 //      if( FD > 2 )
  75                 DEBUG("read(0x%x, 0x%x, *%p)", FD, Bytes, Dest);
  76         return _Syscall(SYS_READ, ">i >i <d", FD, Bytes, Bytes, Dest);
  77 }
  78
  79 size_t acess_write(int FD, const void *Src, size_t Bytes) {
  80         if(FD & NATIVE_FILE_MASK)
  81                 return native_write(FD & (NATIVE_FILE_MASK-1), Src, Bytes);
  82 //      if( FD > 2 )
  83                 DEBUG("write(0x%x, 0x%x, %p\"%.*s\")", FD, Bytes, Src, Bytes, (char*)Src);
  84         return _Syscall(SYS_WRITE, ">i >i >d", FD, Bytes, Bytes, Src);
  85 }
  86
  87 int acess_seek(int FD, int64_t Ofs, int Dir) {
  88         if(FD & NATIVE_FILE_MASK) {
  89                 return native_seek(FD & (NATIVE_FILE_MASK-1), Ofs, Dir);
  90         }
  91         DEBUG("seek(0x%x, 0x%llx, %i)", FD, Ofs, Dir);
  92         return _Syscall(SYS_SEEK, ">i >I >i", FD, Ofs, Dir);
  93 }
  94
  95 uint64_t acess_tell(int FD) {
  96         if(FD & NATIVE_FILE_MASK)
  97                 return native_tell( FD & (NATIVE_FILE_MASK-1) );
  98         DEBUG("tell(0x%x)", FD);
  99         return _Syscall(SYS_TELL, ">i", FD);
 100 }
 101
 102 int acess_ioctl(int fd, int id, void *data) {
 103          int    len;
 104         DEBUG("ioctl(%i, %i, %p)", fd, id, data);
 105         // NOTE: The length here is hacky and could break
 106         if( data == NULL )
 107                 len = 0;
 108         else
 109                 len = PAGE_SIZE - ((uintptr_t)data % PAGE_SIZE);
 110         return _Syscall(SYS_IOCTL, ">i >i ?d", fd, id, len, data);
 111 }
 112 int acess_finfo(int fd, t_sysFInfo *info, int maxacls) {
 113 //      DEBUG("offsetof(size, t_sysFInfo) = %i", offsetof(t_sysFInfo, size));
 114         DEBUG("finfo(%i, %p, %i)", fd, info, maxacls);
 115         return _Syscall(SYS_FINFO, ">i <d >i",
 116                 fd,
 117                 sizeof(t_sysFInfo)+maxacls*sizeof(t_sysACL), info,
 118                 maxacls
 119                 );
 120 }
 121
 122 int acess_SysReadDir(int fd, char *dest) {
 123         DEBUG("SysReadDir(%i, %p)", fd, dest);
 124         return _Syscall(SYS_READDIR, ">i <d", fd, 256, dest);
 125 }
 126
 127 int acess__SysSelect(int nfds, fd_set *read, fd_set *write, fd_set *error, time_t *timeout, uint32_t events)
 128 {
 129         DEBUG("_SysSelect(%i, %p, %p, %p, %p, 0x%x)", nfds, read, write, error, timeout, events);
 130         return _Syscall(SYS_SELECT, ">i ?d ?d ?d >d >i", nfds,
 131                 read ? (nfds+7)/8 : 0, read,
 132                 write ? (nfds+7)/8 : 0, write,
 133                 error ? (nfds+7)/8 : 0, error,
 134                 sizeof(*timeout), timeout,
 135                 events
 136                 );
 137 }
 138
 139 int acess_select(int nfds, fd_set *read, fd_set *write, fd_set *error, time_t *timeout)
 140 {
 141         return acess__SysSelect(nfds, read, write, error, timeout, 0);
 142 }
 143
 144
 145 int acess__SysOpenChild(int fd, char *name, int flags) {
 146         DEBUG("_SysOpenChild(0x%x, '%s', 0x%x)", fd, name, flags);
 147         return _Syscall(SYS_OPENCHILD, ">i >s >i", fd, name, flags);
 148 }
 149
 150 int acess__SysGetACL(int fd, t_sysACL *dest) {
 151         DEBUG("%s(0x%x, %p)", __func__, fd, dest);
 152         return _Syscall(SYS_GETACL, ">i <d", fd, sizeof(t_sysACL), dest);
 153 }
 154
 155 int acess__SysMount(const char *Device, const char *Directory, const char *Type, const char *Options) {
 156         DEBUG("%s('%s', '%s', '%s', '%s')", __func__, Device, Directory, Type, Options);
 157         return _Syscall(SYS_MOUNT, ">s >s >s >s", Device, Directory, Type, Options);
 158 }
 159
 160
 161 // --- Error Handler
 162 int acess__SysSetFaultHandler(int (*Handler)(int)) {
 163         printf("TODO: Set fault handler (asked to set to %p)\n", Handler);
 164         return 0;
 165 }
 166
 167 // --- Memory Management ---
 168 uint64_t acess__SysAllocate(uint vaddr)
 169 {
 170         if( AllocateMemory(vaddr, 0x1000) == -1 )       // Allocate a page
 171                 return 0;
 172
 173         return vaddr;   // Just ignore the need for paddrs :)
 174 }
 175
 176 // --- Process Management ---
 177 int acess_clone(int flags, void *stack)
 178 {
 179         #ifdef __WIN32__
 180         Warning("Win32 does not support anything like fork(2), cannot emulate");
 181         exit(-1);
 182         #else
 183         extern int fork(void);
 184         if(flags & CLONE_VM) {
 185                  int    ret, newID, kernel_tid=0;
 186                 Debug("USERSIDE fork()");
 187
 188                 newID = _Syscall(SYS_AN_FORK, "<d", sizeof(int), &kernel_tid);
 189                 ret = fork();
 190                 if(ret < 0) {
 191                         return ret;
 192                 }
 193
 194                 if(ret == 0)
 195                 {
 196                         _CloseSyscalls();
 197                         giSyscall_ClientID = newID;
 198                         _InitSyscalls();
 199                         return 0;
 200                 }
 201
 202                 // Return the acess TID instead
 203                 return kernel_tid;
 204         }
 205         else
 206         {
 207                 Warning("ERROR: Threads currently unsupported\n");
 208                 exit(-1);
 209         }
 210         #endif
 211 }
 212
 213 int acess_execve(char *path, char **argv, const char **envp)
 214 {
 215          int    i, argc;
 216
 217         DEBUG("acess_execve: (path='%s', argv=%p, envp=%p)", path, argv, envp);
 218
 219         // Get argument count
 220         for( argc = 0; argv[argc]; argc ++ ) ;
 221         DEBUG(" acess_execve: argc = %i", argc);
 222
 223         const char      *new_argv[7+argc+1];
 224         char    client_id_str[11];
 225         char    socket_fd_str[11];
 226         sprintf(client_id_str, "%i", giSyscall_ClientID);
 227         sprintf(socket_fd_str, "%i", gSocket);
 228         new_argv[0] = "ld-acess";       // TODO: Get path to ld-acess executable
 229         new_argv[1] = "--key";          // Set client ID for Request.c
 230         new_argv[2] = client_id_str;
 231         new_argv[3] = "--socket";       // Socket
 232         new_argv[4] = socket_fd_str;
 233         new_argv[5] = "--binary";       // Set the binary path (instead of using argv[0])
 234         new_argv[6] = path;
 235         for( i = 0; i < argc; i ++ )    new_argv[7+i] = argv[i];
 236         new_argv[7+i] = NULL;
 237
 238         #if 1
 239         argc += 7;
 240         for( i = 0; i < argc; i ++ )
 241                 printf("\"%s\" ", new_argv[i]);
 242         printf("\n");
 243         if(envp)
 244         {
 245                 printf("envp = %p\n", envp);
 246                 for( i = 0; envp[i]; i ++ )
 247                         printf("%i: \"%s\"\n", i, envp[i]);
 248                 printf("envc = %i\n", i);
 249         }
 250         #endif
 251
 252         // Call actual execve
 253         return native_execve("./ld-acess", new_argv, envp);
 254 }
 255
 256 void acess_sleep(void)
 257 {
 258         DEBUG("%s()", __func__);
 259         _Syscall(SYS_SLEEP, "");
 260 }
 261
 262 int acess_waittid(int TID, int *ExitStatus)
 263 {
 264         DEBUG("%s(%i, %p)", __func__, TID, ExitStatus);
 265         return _Syscall(SYS_WAITTID, ">i <d", TID, sizeof(int), &ExitStatus);
 266 }
 267
 268 int acess_setuid(int ID) { return _Syscall(SYS_SETUID, ">i", ID); }
 269 int acess_setgid(int ID) { return _Syscall(SYS_SETGID, ">i", ID); }
 270 int acess_gettid(void) { return _Syscall(SYS_GETTID, ""); }
 271 int acess_getpid(void) { return _Syscall(SYS_GETPID, ""); }
 272 int acess_getuid(void) { return _Syscall(SYS_GETUID, ""); }
 273 int acess_getgid(void) { return _Syscall(SYS_GETGID, ""); }
 274
 275 int acess_SysSendMessage(int DestTID, int Length, void *Data)
 276 {
 277         DEBUG("%s(%i, 0x%x, %p)", __func__, DestTID, Length, Data);
 278         return _Syscall(SYS_SENDMSG, ">i >d", DestTID, Length, Data);
 279 }
 280
 281 int acess_SysGetMessage(int *SourceTID, int BufLen, void *Data)
 282 {
 283         DEBUG("%s(%p, %p)", __func__, SourceTID, Data);
 284         return _Syscall(SYS_GETMSG, "<d <d",
 285                 SourceTID ? sizeof(uint32_t) : 0, SourceTID,
 286                 BufLen, Data
 287                 );
 288 }
 289
 290 int acess__SysWaitEvent(int Mask)
 291 {
 292         DEBUG("%s(%x)", __func__, Mask);
 293         return _Syscall(SYS_WAITEVENT, ">i", Mask);
 294 }
 295
 296 // --- Logging
 297 void acess__SysDebug(const char *Format, ...)
 298 {
 299         va_list args;
 300
 301         va_start(args, Format);
 302
 303         printf("[_SysDebug %i] ", giSyscall_ClientID);
 304         vprintf(Format, args);
 305         printf("\n");
 306
 307         va_end(args);
 308 }
 309
 310 void acess__exit(int Status)
 311 {
 312         DEBUG("_exit(%i)", Status);
 313         _Syscall(SYS_EXIT, ">i", Status);
 314         exit(Status);
 315 }
 316
 317
 318 // === Symbol List ===
 319 #define DEFSYM(name)    {#name, &acess_##name}
 320 const tSym      caBuiltinSymbols[] = {
 321         DEFSYM(_exit),
 322
 323         DEFSYM(chdir),
 324         DEFSYM(open),
 325         DEFSYM(close),
 326         DEFSYM(reopen),
 327         DEFSYM(read),
 328         DEFSYM(write),
 329         DEFSYM(seek),
 330         DEFSYM(tell),
 331         DEFSYM(ioctl),
 332         DEFSYM(finfo),
 333         DEFSYM(SysReadDir),
 334         DEFSYM(select),
 335         DEFSYM(_SysOpenChild),
 336         DEFSYM(_SysGetACL),
 337         DEFSYM(_SysMount),
 338         DEFSYM(_SysSelect),
 339
 340         DEFSYM(clone),
 341         DEFSYM(execve),
 342         DEFSYM(sleep),
 343
 344         DEFSYM(waittid),
 345         DEFSYM(setuid),
 346         DEFSYM(setgid),
 347         DEFSYM(gettid),
 348
 349         DEFSYM(SysSendMessage),
 350         DEFSYM(SysGetMessage),
 351
 352         DEFSYM(_SysAllocate),
 353         DEFSYM(_SysDebug),
 354         DEFSYM(_SysSetFaultHandler),
 355         DEFSYM(_SysWaitEvent),
 356
 357         DEFSYM(_errno)
 358 };
 359
 360 const int       ciNumBuiltinSymbols = sizeof(caBuiltinSymbols)/sizeof(caBuiltinSymbols[0]);
 361