X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=AcessNative%2Fld-acess_src%2Fexports.c;h=0c87bda674ef882eae453f55e7928a90d797fd8a;hb=a41f3e5efdf853726d078dc03550de40e9d63bdd;hp=5f97bb27d2f741aaf80f83817c9991155dca9ecc;hpb=d497ef38c1c1e0aa5467722e115be0dc4baa47ab;p=tpg%2Facess2.git

diff --git a/AcessNative/ld-acess_src/exports.c b/AcessNative/ld-acess_src/exports.c
index 5f97bb27..0c87bda6 100644
--- a/AcessNative/ld-acess_src/exports.c
+++ b/AcessNative/ld-acess_src/exports.c
@@ -23,7 +23,11 @@ extern int	sprintf(char *,const char *, ...);
 extern int	vprintf(const char *, va_list);
 extern int	strncmp(const char *, const char *, size_t);
 
+extern int	gSocket;
 extern int	giSyscall_ClientID;	// Needed for execve
+extern void	_InitSyscalls(void);
+extern void	_CloseSyscalls(void);
+
 extern void	Debug(const char *Format, ...);
 extern int	AllocateMemory(uintptr_t VirtAddr, size_t ByteCount);
 
@@ -108,17 +112,24 @@ int acess_readdir(int fd, char *dest) {
 	return _Syscall(SYS_READDIR, ">i <d", fd, 256, dest);
 }
 
-int acess_select(int nfds, fd_set *read, fd_set *write, fd_set *error, time_t *timeout)
+int acess__SysSelect(int nfds, fd_set *read, fd_set *write, fd_set *error, time_t *timeout, uint32_t events)
 {
-	DEBUG("select(%i, %p, %p, %p, %p)", nfds, read, write, error, timeout);
-	return _Syscall(SYS_SELECT, ">i ?d ?d ?d >d", nfds,
+	DEBUG("_SysSelect(%i, %p, %p, %p, %p, 0x%x)", nfds, read, write, error, timeout, events);
+	return _Syscall(SYS_SELECT, ">i ?d ?d ?d >d >i", nfds,
 		read ? (nfds+7)/8 : 0, read,
 		write ? (nfds+7)/8 : 0, write,
 		error ? (nfds+7)/8 : 0, error,
-		sizeof(*timeout), timeout
+		sizeof(*timeout), timeout,
+		events
 		);
 }
 
+int acess_select(int nfds, fd_set *read, fd_set *write, fd_set *error, time_t *timeout)
+{
+	return acess__SysSelect(nfds, read, write, error, timeout, 0);
+}
+
+
 int acess__SysOpenChild(int fd, char *name, int flags) {
 	return _Syscall(SYS_OPENCHILD, ">i >s >i", fd, name, flags);
 }
@@ -153,14 +164,18 @@ int acess_clone(int flags, void *stack)
 	extern int fork(void);
 	if(flags & CLONE_VM) {
 		 int	ret, newID, kernel_tid=0;
-		printf("fork()");
+		printf("USERSIDE fork()\n");
 		
-		newID = _Syscall(SYS_FORK, "<d", sizeof(int), &kernel_tid);
+		newID = _Syscall(SYS_AN_FORK, "<d", sizeof(int), &kernel_tid);
 		ret = fork();
-		if(ret < 0)	return ret;
+		if(ret < 0) {
+			return ret;
+		}
 		
 		if(ret == 0)
 		{
+			_CloseSyscalls();
+			_InitSyscalls();
 			giSyscall_ClientID = newID;
 			return 0;
 		}
@@ -184,20 +199,24 @@ int acess_execve(char *path, char **argv, char **envp)
 	// Get argument count
 	for( argc = 0; argv[argc]; argc ++ ) ;
 	DEBUG(" acess_execve: argc = %i", argc);
-	
-	char	*new_argv[5+argc+1];
-	char	key[11];
-	sprintf(key, "%i", giSyscall_ClientID);
+
+	char	*new_argv[7+argc+1];
+	char	client_id_str[11];
+	char	socket_fd_str[11];
+	sprintf(client_id_str, "%i", giSyscall_ClientID);
+	sprintf(socket_fd_str, "%i", gSocket);
 	new_argv[0] = "ld-acess";	// TODO: Get path to ld-acess executable
-	new_argv[1] = "--key";	// Set socket/client ID for Request.c
-	new_argv[2] = key;
-	new_argv[3] = "--binary";	// Set the binary path (instead of using argv[0])
-	new_argv[4] = path;
-	for( i = 0; i < argc; i ++ )	new_argv[5+i] = argv[i];
-	new_argv[5+i] = NULL;
+	new_argv[1] = "--key";  	// Set client ID for Request.c
+	new_argv[2] = client_id_str;
+	new_argv[3] = "--socket";	// Socket
+	new_argv[4] = socket_fd_str;
+	new_argv[5] = "--binary";	// Set the binary path (instead of using argv[0])
+	new_argv[6] = path;
+	for( i = 0; i < argc; i ++ )	new_argv[7+i] = argv[i];
+	new_argv[7+i] = NULL;
 	
 	#if 1
-	argc += 5;
+	argc += 7;
 	for( i = 0; i < argc; i ++ )
 		printf("\"%s\" ", new_argv[i]);
 	printf("\n");
@@ -224,15 +243,12 @@ int acess_waittid(int TID, int *ExitStatus)
 	return _Syscall(SYS_WAITTID, ">i <d", TID, sizeof(int), &ExitStatus);
 }
 
-int acess_setuid(int ID)
-{
-	return _Syscall(SYS_SETUID, ">i", ID);
-}
-
-int acess_setgid(int ID)
-{
-	return _Syscall(SYS_SETGID, ">i", ID);
-}
+int acess_setuid(int ID) { return _Syscall(SYS_SETUID, ">i", ID); }
+int acess_setgid(int ID) { return _Syscall(SYS_SETGID, ">i", ID); }
+int acess_gettid(void) { return _Syscall(SYS_GETTID, ""); }
+int acess_getpid(void) { return _Syscall(SYS_GETPID, ""); }
+int acess_getuid(void) { return _Syscall(SYS_GETUID, ""); }
+int acess_getgid(void) { return _Syscall(SYS_GETGID, ""); }
 
 int acess_SysSendMessage(int DestTID, int Length, void *Data)
 {
@@ -289,6 +305,7 @@ const tSym	caBuiltinSymbols[] = {
 	DEFSYM(_SysOpenChild),
 	DEFSYM(_SysGetACL),
 	DEFSYM(_SysMount),
+	DEFSYM(_SysSelect),
 	
 	DEFSYM(clone),
 	DEFSYM(execve),
@@ -297,6 +314,7 @@ const tSym	caBuiltinSymbols[] = {
 	DEFSYM(waittid),
 	DEFSYM(setuid),
 	DEFSYM(setgid),
+	DEFSYM(gettid),
 
 	DEFSYM(SysSendMessage),
 	DEFSYM(SysGetMessage),