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

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