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