AcessNative - Fix compile errors from API changes (and from compiler upgrade)
[tpg/acess2.git] / 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

UCC git Repository :: git.ucc.asn.au