Cleaned up places where MM_Allocate was used without checks
[tpg/acess2.git] / Kernel / syscalls.c
1 /*
2  * AcessOS Microkernel Version
3  * syscalls.c
4  */
5 #define DEBUG   0
6
7 #include <acess.h>
8 #include <syscalls.h>
9 #include <proc.h>
10 #include <errno.h>
11
12 #define CHECK_NUM_NULLOK(v,size)        \
13         if((v)&&!Syscall_Valid((size),(Uint)(v))){ret=-1;err=-EINVAL;break;}
14 #define CHECK_STR_NULLOK(v)     \
15         if((v)&&!Syscall_ValidString((Uint)(v))){ret=-1;err=-EINVAL;break;}
16 #define CHECK_NUM_NONULL(v,size)        \
17         if(!(v)||!Syscall_Valid((size),(Uint)(v))){ret=-1;err=-EINVAL;break;}
18 #define CHECK_STR_NONULL(v)     \
19         if(!(v)||!Syscall_ValidString((Uint)(v))){ret=-1;err=-EINVAL;break;}
20
21 // === IMPORTS ===
22 extern int      Proc_Clone(Uint *Err, Uint Flags);
23 extern int      Threads_WaitTID(int TID, int *status);
24 extern Uint     Proc_SendMessage(Uint *Err, Uint Dest, Uint Length, void *Data);
25 extern int      Proc_GetMessage(Uint *Err, Uint *Source, void *Buffer);
26 extern int      Proc_Execve(char *File, char **ArgV, char **EnvP);
27 extern Uint     Binary_Load(char *file, Uint *entryPoint);
28 extern int      Threads_SetUID(Uint *errno, tUID ID);
29 extern int      Threads_SetGID(Uint *errno, tGID ID);
30 extern int      Threads_SetFaultHandler(Uint Handler);
31
32 // === PROTOTYPES ===
33  int    Syscall_ValidString(Uint Addr);
34  int    Syscall_Valid(int Size, Uint Addr);
35
36 // === CODE ===
37 // TODO: Do sanity checking on arguments, ATM the user can really fuck with the kernel
38 void SyscallHandler(tSyscallRegs *Regs)
39 {
40         Uint64  ret = 0;
41         Uint    err = -EOK;
42          int    callNum = Regs->Num;
43         
44         #if DEBUG < 2
45         if(callNum != SYS_READ && callNum != SYS_WRITE) {
46         #endif
47         ENTER("iThread iNum", Threads_GetTID(), callNum);
48         if(callNum < NUM_SYSCALLS)
49                 LOG("Syscall %s", cSYSCALL_NAMES[callNum]);
50         LOG("Arg1: 0x%x, Arg2: 0x%x, Arg3: 0x%x, Arg4: 0x%x", Regs->Arg1, Regs->Arg2, Regs->Arg3, Regs->Arg4);
51         #if DEBUG < 2
52         }
53         #endif
54         
55         switch(Regs->Num)
56         {
57         // -- Exit the current thread
58         case SYS_EXIT:  Threads_Exit(0, Regs->Arg1);    break;
59         
60         // -- Put the current thread to sleep
61         case SYS_SLEEP: Threads_Sleep();        break;
62         
63         // -- Yield current timeslice
64         case SYS_YIELD: Threads_Yield();        break;
65         
66         // -- Set Error Handler
67         case SYS_SETFAULTHANDLER:
68                 Threads_SetFaultHandler(Regs->Arg1);
69                 break;
70         
71         // -- Clone the current thread
72         case SYS_CLONE:
73                 // Call clone system call
74                 ret = Proc_Clone(&err, Regs->Arg1);
75                 // Change user stack if a new stack address is passed
76                 if(ret == 0 && Regs->Arg2)
77                         Regs->StackPointer = Regs->Arg2;
78                 break;
79         
80         // -- Send a signal
81         case SYS_KILL:
82                 err = -ENOSYS;
83                 ret = -1;
84                 break;
85         
86         // -- Wait for a thread
87         case SYS_WAITTID:
88                 // Sanity Check (Status can be NULL)
89                 CHECK_NUM_NULLOK( Regs->Arg2, sizeof(int) );
90                 // TID, *Status
91                 ret = Threads_WaitTID(Regs->Arg1, (void*)Regs->Arg2);
92                 break;
93         
94         // -- Get the physical address of a page
95         case SYS_GETPHYS:
96                 ret = MM_GetPhysAddr(Regs->Arg1);
97                 break;
98         
99         // -- Map an address
100         case SYS_MAP:   MM_Map(Regs->Arg1, Regs->Arg2); break;
101         
102         // -- Allocate an address
103         case SYS_ALLOCATE:      ret = MM_Allocate(Regs->Arg1);  break;
104         
105         // -- Unmap an address
106         case SYS_UNMAP:         MM_Deallocate(Regs->Arg1);      break;
107         
108         // -- Get Thread/Process IDs
109         case SYS_GETTID:        ret = Threads_GetTID(); break;
110         case SYS_GETPID:        ret = Threads_GetPID(); break;
111         
112         // -- Get User/Group IDs
113         case SYS_GETUID:        ret = Threads_GetUID(); break;
114         case SYS_GETGID:        ret = Threads_GetGID(); break;
115         
116         // -- Set User/Group IDs
117         case SYS_SETUID:        ret = Threads_SetUID(&err, Regs->Arg1); break;
118         case SYS_SETGID:        ret = Threads_SetGID(&err, Regs->Arg1); break;
119         
120         // -- Send Message
121         case SYS_SENDMSG:
122                 CHECK_NUM_NONULL(Regs->Arg3, Regs->Arg2);
123                 // Destination, Size, *Data
124                 ret = Proc_SendMessage(&err, Regs->Arg1, Regs->Arg2, (void*)Regs->Arg3);
125                 break;
126         // -- Check for messages
127         case SYS_GETMSG:
128                 CHECK_NUM_NULLOK(Regs->Arg1, sizeof(Uint));
129                 if( Regs->Arg2 && Regs->Arg2 != -1 && !MM_IsUser(Regs->Arg2) ) {
130                         err = -EINVAL;  ret = -1;       break;
131                 }
132                 // *Source, *Data
133                 ret = Proc_GetMessage(&err, (Uint*)Regs->Arg1, (void*)Regs->Arg2);
134                 break;
135         
136         // -- Get the current timestamp
137         case SYS_GETTIME:
138                 ret = now();
139                 break;
140         
141         // -- Set the thread's name
142         case SYS_SETNAME:
143                 CHECK_STR_NONULL(Regs->Arg1);
144                 Threads_SetName( (char*)Regs->Arg1 );
145                 break;
146         
147         // ---
148         // Binary Control
149         // ---
150         // -- Replace the current process with another
151         case SYS_EXECVE:
152                 CHECK_STR_NONULL(Regs->Arg1);
153                 // Check the argument arrays
154                 {
155                          int    i;
156                         char    **tmp = (char**)Regs->Arg2;
157                         // Check ArgV (traverse array checking all string pointers)
158                         CHECK_NUM_NONULL( tmp, sizeof(char**) );
159                         //Log("tmp = %p", tmp);
160                         for(i=0;tmp[i];i++) {
161                                 CHECK_NUM_NONULL( &tmp[i], sizeof(char*) );
162                                 CHECK_STR_NONULL( tmp[i] );
163                         }
164                         // Check EnvP also
165                         // - EnvP can be NULL
166                         if( Regs->Arg3 )
167                         {
168                                 tmp = (char**)Regs->Arg3;
169                                 //Log("tmp = %p", tmp);
170                                 for(i=0;tmp[i];i++) {
171                                         CHECK_NUM_NULLOK( &tmp[i], sizeof(char*) );
172                                         CHECK_STR_NONULL( tmp[i] );
173                                 }
174                         }
175                 }
176                 LEAVE('s', "Assuming 0");
177                 // Path, **Argv, **Envp
178                 ret = Proc_Execve( (char*)Regs->Arg1, (char**)Regs->Arg2, (char**)Regs->Arg3 );
179                 break;
180         // -- Load a binary into the current process
181         case SYS_LOADBIN:
182                 if( !Syscall_ValidString(Regs->Arg1)
183                 ||  !Syscall_Valid(sizeof(Uint), Regs->Arg2) ) {
184                         err = -EINVAL;
185                         ret = 0;
186                         break;
187                 }
188                 // Path, *Entrypoint
189                 ret = Binary_Load((char*)Regs->Arg1, (Uint*)Regs->Arg2);
190                 break;
191         
192         // ---
193         // Virtual Filesystem
194         // ---
195         case SYS_OPEN:
196                 if( !Syscall_ValidString(Regs->Arg1) ) {
197                         err = -EINVAL;
198                         ret = -1;
199                         break;
200                 }
201                 LOG("VFS_Open(\"%s\", 0x%x)", (char*)Regs->Arg1, Regs->Arg2 | VFS_OPENFLAG_USER);
202                 ret = VFS_Open((char*)Regs->Arg1, Regs->Arg2 | VFS_OPENFLAG_USER);
203                 break;
204         
205         case SYS_CLOSE:
206                 LOG("VFS_Close(%i)", Regs->Arg1);
207                 VFS_Close( Regs->Arg1 );
208                 break;
209         
210         case SYS_SEEK:
211                 #if BITS == 64
212                 ret = VFS_Seek( Regs->Arg1, Regs->Arg2, Regs->Arg3 );
213                 #else
214                 ret = VFS_Seek( Regs->Arg1, Regs->Arg2|(((Uint64)Regs->Arg3)<<32), Regs->Arg4 );
215                 #endif
216                 break;
217                 
218         case SYS_TELL:
219                 ret = VFS_Tell( Regs->Arg1 );
220                 break;
221         
222         case SYS_WRITE:
223                 CHECK_NUM_NONULL( Regs->Arg3, Regs->Arg2 );
224                 ret = VFS_Write( Regs->Arg1, Regs->Arg2, (void*)Regs->Arg3 );
225                 break;
226         
227         case SYS_READ:
228                 CHECK_NUM_NONULL( Regs->Arg3, Regs->Arg2 );
229                 ret = VFS_Read( Regs->Arg1, Regs->Arg2, (void*)Regs->Arg3 );
230                 break;
231         
232         case SYS_FINFO:
233                 CHECK_NUM_NONULL( Regs->Arg2, sizeof(tFInfo) + Regs->Arg3*sizeof(tVFS_ACL) );
234                 // FP, Dest, MaxACLs
235                 ret = VFS_FInfo( Regs->Arg1, (void*)Regs->Arg2, Regs->Arg3 );
236                 break;
237         
238         // Get ACL Value
239         case SYS_GETACL:
240                 if( !Syscall_Valid(sizeof(tVFS_ACL), Regs->Arg1) ) {
241                         err = -EINVAL;
242                         ret = -1;
243                         break;
244                 }
245                 ret = VFS_GetACL( Regs->Arg1, (void*)Regs->Arg2 );
246                 break;
247         
248         // Read Directory
249         case SYS_READDIR:
250                 // TODO: What if the filename is longer?
251                 // Maybe force it to be a 256 byte buffer
252                 if( !Syscall_Valid(8, Regs->Arg2) ) {
253                         err = -EINVAL;
254                         ret = -1;
255                         break;
256                 }
257                 ret = VFS_ReadDir( Regs->Arg1, (void*)Regs->Arg2 );
258                 break;
259         
260         // Open a file that is a entry in an open directory
261         case SYS_OPENCHILD:
262                 if( !Syscall_ValidString(Regs->Arg2) ) {
263                         err = -EINVAL;
264                         ret = -1;
265                         break;
266                 }
267                 ret = VFS_OpenChild( &err, Regs->Arg1, (char*)Regs->Arg2, Regs->Arg3);
268                 break;
269         
270         // Change Directory
271         case SYS_CHDIR:
272                 if( !Syscall_ValidString(Regs->Arg1) ) {
273                         err = -EINVAL;
274                         ret = -1;
275                         break;
276                 }
277                 ret = VFS_ChDir( (void*)Regs->Arg1 );
278                 break;
279         
280         // IO Control
281         case SYS_IOCTL:
282                 // All sanity checking should be done by the driver
283                 ret = VFS_IOCtl( Regs->Arg1, Regs->Arg2, (void*)Regs->Arg3 );
284                 break;
285         
286         // Mount a filesystem
287         case SYS_MOUNT:
288                 // Only root can mount filesystems
289                 if(Threads_GetUID() != 0) {
290                         err = -EACCES;
291                         ret = -1;
292                         break;
293                 }
294                 // Sanity check the paths
295                 if(!Syscall_ValidString(Regs->Arg1)
296                 || !Syscall_ValidString(Regs->Arg2)
297                 || !Syscall_ValidString(Regs->Arg3)
298                 || !Syscall_ValidString(Regs->Arg4) ) {
299                         err = -EINVAL;
300                         ret = -1;
301                         break;
302                 }
303                 ret = VFS_Mount(
304                         (char*)Regs->Arg1,      // Device
305                         (char*)Regs->Arg2,      // Mount point
306                         (char*)Regs->Arg3,      // Filesystem
307                         (char*)Regs->Arg4       // Options
308                         );
309                 break;
310         
311         // -- Debug
312         //#if DEBUG_BUILD
313         case SYS_DEBUG:
314                 LogF("Log: [%i] ", Threads_GetTID());
315                 LogF((char*)Regs->Arg1,
316                         Regs->Arg2, Regs->Arg3, Regs->Arg4, Regs->Arg5, Regs->Arg6);
317                 LogF("\n");
318                 break;
319         //#endif
320         
321         // -- Default (Return Error)
322         default:
323                 Warning("SyscallHandler: Unknown System Call %i", Regs->Num);
324                 if(Regs->Num < NUM_SYSCALLS)
325                         Warning(" Syscall '%s'", cSYSCALL_NAMES[Regs->Num]);
326                 err = -ENOSYS;
327                 ret = -1;
328                 break;
329         }
330         
331         if(err != 0) {
332                 LOG("ID: %i, Return errno = %i", Regs->Num, err);
333         }
334         
335         #if BITS < 64
336         Regs->Return = ret&0xFFFFFFFF;
337         Regs->RetHi = ret >> 32;
338         #else
339         Regs->Return = ret;
340         #endif
341         Regs->Error = err;
342         #if DEBUG
343         # if DEBUG < 2
344         if( callNum != SYS_READ && callNum != SYS_WRITE ) {
345         # endif
346         LOG("err = %i", err);
347         if(Regs->Num != SYS_EXECVE)
348                 LEAVE('x', ret);
349         else
350                 LOG("Actual %i", ret);
351         # if DEBUG < 2
352         }
353         # endif
354         #endif
355 }
356
357 /**
358  * \fn int Syscall_ValidString(Uint Addr)
359  * \brief Checks if a memory address contains a valid string
360  */
361 int Syscall_ValidString(Uint Addr)
362 {
363         // Check if the memory is user memory
364         if(!MM_IsUser(Addr))    return 0;
365         
366         return CheckString( (char*)Addr );
367 }
368
369 /**
370  * \fn int Syscall_Valid(int Size, Uint Addr)
371  * \brief Checks if a memory address is valid
372  */
373 int Syscall_Valid(int Size, Uint Addr)
374 {
375         if(!MM_IsUser(Addr))    return 0;
376         
377         return CheckMem( (void*)Addr, Size );
378 }

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