X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=AcessNative%2Fld-acess_src%2Fexports.c;h=756febaf10ac6d8389ce9cd6998527ca23daa31e;hb=96d72268a3a1cfddcbd71353881e8bfadc17db5e;hp=af9cc900037286386e61705343adf87a2c5b15d6;hpb=b8b25ae01e13c655608fb6705574e5218b1a519f;p=tpg%2Facess2.git

diff --git a/AcessNative/ld-acess_src/exports.c b/AcessNative/ld-acess_src/exports.c
index af9cc900..756febaf 100644
--- a/AcessNative/ld-acess_src/exports.c
+++ b/AcessNative/ld-acess_src/exports.c
@@ -5,16 +5,18 @@
  * - Exported functions
  */
 #define DONT_INCLUDE_SYSCALL_NAMES 1
-#include "../../Usermode/include/acess/sys.h"
+#include "../../Usermode/Libraries/ld-acess.so_src/include_exp/acess/sys.h"
 #include "../syscalls.h"
 #include "exports.h"
 #include <stdarg.h>
+#include <stddef.h>
 
 #define DEBUG(v...)	Debug(v)
+//#define DEBUG(v...)	do{}while(0)//Debug(v)
+#define PAGE_SIZE	4096
 
 typedef struct sFILE	FILE;
 
-extern FILE	*stderr;
 extern void	exit(int) __attribute__ ((noreturn));
 extern int	printf(const char *, ...);
 extern int	fprintf(FILE *,const char *, ...);
@@ -22,21 +24,30 @@ 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	Warning(const char *Format, ...);
 extern void	Debug(const char *Format, ...);
 extern int	AllocateMemory(uintptr_t VirtAddr, size_t ByteCount);
 
 // === CONSTANTS ===
 #define NATIVE_FILE_MASK	0x40000000
 
+// === GLOBALS ===
+int	acess__errno;
+char	*gsExecutablePath = "./ld-acess";
+
 // === CODE ===
 // --- VFS Calls
-int acess_chdir(const char *Path)
+int acess__SysChdir(const char *Path)
 {
 	return _Syscall(SYS_CHDIR, ">s", Path);
 }
 
-int acess_open(const char *Path, int Flags)
+int acess__SysOpen(const char *Path, int Flags)
 {
 	if( strncmp(Path, "$$$$", 4) == 0 )
 	{
@@ -46,7 +57,8 @@ int acess_open(const char *Path, int Flags)
 	return _Syscall(SYS_OPEN, ">s >i", Path, Flags);
 }
 
-void acess_close(int FD) {
+void acess__SysClose(int FD)
+{
 	if(FD & NATIVE_FILE_MASK) {
 		return native_close(FD & (NATIVE_FILE_MASK-1));
 	}
@@ -54,26 +66,29 @@ void acess_close(int FD) {
 	_Syscall(SYS_CLOSE, ">i", FD);
 }
 
-int acess_reopen(int FD, const char *Path, int Flags) {
+int acess__SysReopen(int FD, const char *Path, int Flags) {
 	DEBUG("reopen(0x%x, \"%s\", 0x%x)", FD, Path, Flags);
 	return _Syscall(SYS_REOPEN, ">i >s >i", FD, Path, Flags);
 }
 
-size_t acess_read(int FD, size_t Bytes, void *Dest) {
+size_t acess__SysRead(int FD, void *Dest, size_t Bytes) {
 	if(FD & NATIVE_FILE_MASK)
-		return native_read(FD & (NATIVE_FILE_MASK-1), Bytes, Dest);
-	DEBUG("read(0x%x, 0x%x, *%p)", FD, Bytes, Dest);
+		return native_read(FD & (NATIVE_FILE_MASK-1), Dest, Bytes);
+//	if( FD > 2 )
+		DEBUG("read(0x%x, 0x%x, *%p)", FD, Bytes, Dest);
 	return _Syscall(SYS_READ, ">i >i <d", FD, Bytes, Bytes, Dest);
 }
 
-size_t acess_write(int FD, size_t Bytes, const void *Src) {
+size_t acess__SysWrite(int FD, const void *Src, size_t Bytes) {
 	if(FD & NATIVE_FILE_MASK)
-		return native_write(FD & (NATIVE_FILE_MASK-1), Bytes, Src);
-	DEBUG("write(0x%x, 0x%x, %p\"%.*s\")", FD, Bytes, Src, Bytes, (char*)Src);
+		return native_write(FD & (NATIVE_FILE_MASK-1), Src, Bytes);
+//	if( FD > 2 )
+		DEBUG("write(0x%x, 0x%x, %p\"%.*s\")", FD, Bytes, Src, Bytes, (char*)Src);
 	return _Syscall(SYS_WRITE, ">i >i >d", FD, Bytes, Bytes, Src);
 }
 
-int acess_seek(int FD, int64_t Ofs, int Dir) {
+int acess__SysSeek(int FD, int64_t Ofs, int Dir)
+{
 	if(FD & NATIVE_FILE_MASK) {
 		return native_seek(FD & (NATIVE_FILE_MASK-1), Ofs, Dir);
 	}
@@ -81,61 +96,124 @@ int acess_seek(int FD, int64_t Ofs, int Dir) {
 	return _Syscall(SYS_SEEK, ">i >I >i", FD, Ofs, Dir);
 }
 
-uint64_t acess_tell(int FD) {
+uint64_t acess__SysTell(int FD)
+{
 	if(FD & NATIVE_FILE_MASK)
 		return native_tell( FD & (NATIVE_FILE_MASK-1) );
+	DEBUG("tell(0x%x)", FD);
 	return _Syscall(SYS_TELL, ">i", FD);
 }
 
-int acess_ioctl(int fd, int id, void *data) {
-	// NOTE: 1024 byte size is a hack
+int acess__SysIOCtl(int fd, int id, void *data) {
+	 int	len;
 	DEBUG("ioctl(%i, %i, %p)", fd, id, data);
-	return _Syscall(SYS_IOCTL, ">i >i ?d", fd, id, 1024, data);
+	// NOTE: The length here is hacky and could break
+	if( data == NULL )
+		len = 0;
+	else
+		len = PAGE_SIZE - ((uintptr_t)data % PAGE_SIZE);
+	return _Syscall(SYS_IOCTL, ">i >i ?d", fd, id, len, data);
 }
-int acess_finfo(int fd, t_sysFInfo *info, int maxacls) {
+int acess__SysFInfo(int fd, t_sysFInfo *info, int maxacls) {
+//	DEBUG("offsetof(size, t_sysFInfo) = %i", offsetof(t_sysFInfo, size));
+	DEBUG("finfo(%i, %p, %i)", fd, info, maxacls);
 	return _Syscall(SYS_FINFO, ">i <d >i",
 		fd,
 		sizeof(t_sysFInfo)+maxacls*sizeof(t_sysACL), info,
-		maxacls);
+		maxacls
+		);
 }
 
-int acess_readdir(int fd, char *dest) {
-	DEBUG("readdir(%i, %p)", fd, dest);
+int acess__SysReadDir(int fd, char *dest) {
+	DEBUG("SysReadDir(%i, %p)", fd, 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, int64_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__SysUnlink(const char *pathname)
+{
+	// TODO:
+	return 0;
+}
 
 int acess__SysOpenChild(int fd, char *name, int flags) {
+	DEBUG("_SysOpenChild(0x%x, '%s', 0x%x)", fd, name, flags);
 	return _Syscall(SYS_OPENCHILD, ">i >s >i", fd, name, flags);
 }
 
 int acess__SysGetACL(int fd, t_sysACL *dest) {
-	return _Syscall(SYS_GETACL, "<i >i <d", fd, sizeof(t_sysACL), dest);
+	DEBUG("%s(0x%x, %p)", __func__, fd, dest);
+	return _Syscall(SYS_GETACL, ">i <d", fd, sizeof(t_sysACL), dest);
 }
 
 int acess__SysMount(const char *Device, const char *Directory, const char *Type, const char *Options) {
+	DEBUG("%s('%s', '%s', '%s', '%s')", __func__, Device, Directory, Type, Options);
 	return _Syscall(SYS_MOUNT, ">s >s >s >s", Device, Directory, Type, Options);
 }
 
 
 // --- Error Handler
-int	acess__SysSetFaultHandler(int (*Handler)(int)) {
+int acess__SysSetFaultHandler(int (*Handler)(int)) {
 	printf("TODO: Set fault handler (asked to set to %p)\n", Handler);
 	return 0;
 }
 
+void acess__SysSetName(const char *Name)
+{
+	// TODO:
+}
+
+int acess__SysGetName(char *NameDest)
+{
+	// TODO: 
+	return 0;
+}
+
+int acess__SysSetPri(int Priority)
+{
+	// TODO:
+	return 0;
+}
+
+// --- Binaries? ---
+void *acess_SysLoadBin(const char *path, void **entry)
+{
+	// ERROR!
+	return NULL;
+}
+
+int acess__SysUnloadBin(void *base)
+{
+	// ERROR!
+	return -1;
+}
+
+// --- Timekeeping ---
+int64_t acess__SysTimestamp(void)
+{
+	// TODO: Better impl
+//	return now()*1000;
+	return 0;
+}
+
 // --- Memory Management ---
-uint64_t acess__SysAllocate(uint vaddr)
+uint64_t acess__SysGetPhys(uintptr_t vaddr)
+{
+	// TODO:
+	return 0;
+}
+
+uint64_t acess__SysAllocate(uintptr_t vaddr)
 {
 	if( AllocateMemory(vaddr, 0x1000) == -1 )	// Allocate a page
 		return 0;
@@ -144,20 +222,28 @@ uint64_t acess__SysAllocate(uint vaddr)
 }
 
 // --- Process Management ---
-int acess_clone(int flags, void *stack)
+int acess__SysClone(int flags, void *stack)
 {
+	#ifdef __WIN32__
+	Warning("Win32 does not support anything like fork(2), cannot emulate");
+	exit(-1);
+	#else
 	extern int fork(void);
 	if(flags & CLONE_VM) {
 		 int	ret, newID, kernel_tid=0;
-		printf("fork()");
+		Debug("USERSIDE fork()");
 		
-		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();
 			giSyscall_ClientID = newID;
+			_InitSyscalls();
 			return 0;
 		}
 		
@@ -166,12 +252,19 @@ int acess_clone(int flags, void *stack)
 	}
 	else
 	{
-		fprintf(stderr, "ERROR: Threads currently unsupported\n");
+		Warning("ERROR: Threads currently unsupported\n");
 		exit(-1);
 	}
+	#endif
 }
 
-int acess_execve(char *path, char **argv, char **envp)
+int acess__SysKill(int pid, int sig)
+{
+	// TODO: Impliment SysKill
+	return -1;
+}
+
+int acess__SysExecVE(char *path, char **argv, const char **envp)
 {
 	 int	i, argc;
 	
@@ -180,62 +273,114 @@ 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);
+
+	const 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");
+	if(envp)
+	{
+		printf("envp = %p\n", envp);
+		for( i = 0; envp[i]; i ++ )
+			printf("%i: \"%s\"\n", i, envp[i]);
+		printf("envc = %i\n", i);
+	}
 	#endif
 	
 	// Call actual execve
-	return execve("./ld-acess", new_argv, envp);
+	return native_execve("./ld-acess", new_argv, envp);
 }
 
-void acess_sleep(void)
+int acess__SysSpawn(const char *binary, const char **argv, const char **envp, int nfd, int fds[], struct s_sys_spawninfo *info)
 {
-	_Syscall(SYS_SLEEP, "");
-}
+	 int	argc = 0;
+	while( argv[argc++] );
+
+	Debug("_SysSpawn('%s', %p (%i), %p, %i, %p, %p)",
+		binary, argv, argc, envp, nfd, fds, info);
+
+	 int	kernel_tid;
+	 int	newID;
+	newID = _Syscall(SYS_AN_SPAWN, "<d >d >d", sizeof(int), &kernel_tid,
+		nfd*sizeof(int), fds,
+		info ? sizeof(*info) : 0, info);
+
+	const char	*new_argv[5+argc+1];
+	 int	new_argc = 0, i;
+	char	client_id_str[11];
+	sprintf(client_id_str, "%i", newID);
+	new_argv[new_argc++] = gsExecutablePath;       // TODO: Get path to ld-acess executable
+	new_argv[new_argc++] = "--key";
+	new_argv[new_argc++] = client_id_str;
+	new_argv[new_argc++] = "--binary";
+	new_argv[new_argc++] = binary;
+	for( i = 0; argv[i]; i ++)
+		new_argv[new_argc++] = argv[i];
+	new_argv[new_argc++] = NULL;
+	
+	// TODO: Debug output?
+	
+	native_spawn(gsExecutablePath, new_argv, envp);
 
-int acess_waittid(int TID, int *ExitStatus)
-{
-	return _Syscall(SYS_WAITTID, ">i <d", TID, sizeof(int), &ExitStatus);
+	return kernel_tid;
 }
 
-int acess_setuid(int ID)
-{
-	return _Syscall(SYS_SETUID, ">i", ID);
-}
+//void acess_sleep(void)
+//{
+//	DEBUG("%s()", __func__);
+//	_Syscall(SYS_SLEEP, "");
+//}
 
-int acess_setgid(int ID)
+int acess__SysWaitTID(int TID, int *ExitStatus)
 {
-	return _Syscall(SYS_SETGID, ">i", ID);
+	DEBUG("%s(%i, %p)", __func__, TID, ExitStatus);
+	return _Syscall(SYS_WAITTID, ">i <d", TID, sizeof(int), &ExitStatus);
 }
 
-int acess_SysSendMessage(int DestTID, int Length, void *Data)
+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)
 {
+	DEBUG("%s(%i, 0x%x, %p)", __func__, DestTID, Length, Data);
 	return _Syscall(SYS_SENDMSG, ">i >d", DestTID, Length, Data);
 }
 
-int acess_SysGetMessage(int *SourceTID, void *Data)
+int acess__SysGetMessage(int *SourceTID, int BufLen, void *Data)
 {
+	DEBUG("%s(%p, %p)", __func__, SourceTID, Data);
 	return _Syscall(SYS_GETMSG, "<d <d",
-		SourceTID ? sizeof(int) : 0, SourceTID,
-		Data ? 4096 : 0, Data
+		SourceTID ? sizeof(uint32_t) : 0, SourceTID,
+		BufLen, Data
 		);
 }
 
+int acess__SysWaitEvent(int Mask)
+{
+	DEBUG("%s(%x)", __func__, Mask);
+	return _Syscall(SYS_WAITEVENT, ">i", Mask);
+}
+
 // --- Logging
 void acess__SysDebug(const char *Format, ...)
 {
@@ -243,7 +388,7 @@ void acess__SysDebug(const char *Format, ...)
 	
 	va_start(args, Format);
 	
-	printf("[_SysDebug %i]", giSyscall_ClientID);
+	printf("[_SysDebug %i] ", giSyscall_ClientID);
 	vprintf(Format, args);
 	printf("\n");
 	
@@ -257,42 +402,58 @@ void acess__exit(int Status)
 	exit(Status);
 }
 
+uint32_t acess__SysSetMemFlags(uintptr_t vaddr, uint32_t flags, uint32_t mask)
+{
+	// TODO: Impliment acess__SysSetMemFlags?
+	return 0;
+}
+
 
 // === Symbol List ===
-#define DEFSYM(name)	{#name, acess_##name}
+#ifndef DEFSYM
+# define DEFSYM(name)	{#name, &acess_##name}
+#endif
 const tSym	caBuiltinSymbols[] = {
 	DEFSYM(_exit),
 	
-	DEFSYM(chdir),
-	DEFSYM(open),
-	DEFSYM(close),
-	DEFSYM(reopen),
-	DEFSYM(read),
-	DEFSYM(write),
-	DEFSYM(seek),
-	DEFSYM(tell),
-	DEFSYM(ioctl),
-	DEFSYM(finfo),
-	DEFSYM(readdir),
-	DEFSYM(select),
+	DEFSYM(_SysChdir),
+	DEFSYM(_SysOpen),
 	DEFSYM(_SysOpenChild),
+	DEFSYM(_SysReopen),
+	DEFSYM(_SysClose),
+	DEFSYM(_SysRead),
+	DEFSYM(_SysWrite),
+	DEFSYM(_SysSeek),
+	DEFSYM(_SysTell),
+	DEFSYM(_SysIOCtl),
+	DEFSYM(_SysFInfo),
+	DEFSYM(_SysReadDir),
 	DEFSYM(_SysGetACL),
 	DEFSYM(_SysMount),
+	DEFSYM(_SysSelect),
 	
-	DEFSYM(clone),
-	DEFSYM(execve),
-	DEFSYM(sleep),
+	DEFSYM(_SysClone),
+	DEFSYM(_SysExecVE),
+	DEFSYM(_SysSpawn),
+//	DEFSYM(sleep),
 	
-	DEFSYM(waittid),
+	DEFSYM(_SysWaitTID),
+	DEFSYM(gettid),
 	DEFSYM(setuid),
 	DEFSYM(setgid),
+	DEFSYM(getuid),
+	DEFSYM(getgid),
 
-	DEFSYM(SysSendMessage),
-	DEFSYM(SysGetMessage),
+	DEFSYM(_SysSendMessage),
+	DEFSYM(_SysGetMessage),
 	
 	DEFSYM(_SysAllocate),
+	DEFSYM(_SysSetMemFlags),
 	DEFSYM(_SysDebug),
-	DEFSYM(_SysSetFaultHandler)
+	DEFSYM(_SysSetFaultHandler),
+	DEFSYM(_SysWaitEvent),
+	
+	DEFSYM(_errno)
 };
 
 const int	ciNumBuiltinSymbols = sizeof(caBuiltinSymbols)/sizeof(caBuiltinSymbols[0]);