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