X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=AcessNative%2Facesskernel_src%2Fserver.c;h=264f41220af27c1d08aa2197d171976de1472565;hb=a6f11cb266052c58ae7e2d6d6e8abd34a9e93214;hp=7491d74ac5622d0b1e54819a3e959c59e287e720;hpb=d599a063b3b453b705925c58180c93e1f9cb3d79;p=tpg%2Facess2.git

diff --git a/AcessNative/acesskernel_src/server.c b/AcessNative/acesskernel_src/server.c
index 7491d74a..264f4122 100644
--- a/AcessNative/acesskernel_src/server.c
+++ b/AcessNative/acesskernel_src/server.c
@@ -17,6 +17,7 @@
 # include <netinet/in.h>
 #endif
 #include "../syscalls.h"
+//#include <debug.h>
 
 #define	USE_TCP	0
 #define MAX_CLIENTS	16
@@ -36,6 +37,10 @@ typedef struct {
 
 // === IMPORTS ===
 extern tRequestHeader *SyscallRecieve(tRequestHeader *Request, int *ReturnLength);
+extern int	Threads_CreateRootProcess(void);
+// HACK: Should have these in a header
+extern void	Log_Debug(const char *Subsys, const char *Message, ...);
+extern void	Log_Notice(const char *Subsys, const char *Message, ...);
 
 // === PROTOTYPES ===
 tClient	*Server_GetClient(int ClientID);
@@ -50,21 +55,41 @@ SOCKET	gSocket = INVALID_SOCKET;
 # define INVALID_SOCKET -1
  int	gSocket = INVALID_SOCKET;
 #endif
- int	giServer_NextClientID = 1;
 tClient	gaServer_Clients[MAX_CLIENTS];
 
 // === CODE ===
+int Server_GetClientID(void)
+{
+	 int	i;
+	Uint32	thisId = SDL_ThreadID();
+	
+	for( i = 0; i < MAX_CLIENTS; i ++ )
+	{
+		if( SDL_GetThreadID(gaServer_Clients[i].WorkerThread) == thisId )
+			return gaServer_Clients[i].ClientID;
+	}
+	
+	fprintf(stderr, "ERROR: Server_GetClientID - Thread is not allocated\n");
+	
+	return 0;
+}
+
 tClient *Server_GetClient(int ClientID)
 {
 	tClient	*ret = NULL;
 	 int	i;
 	
+	// Allocate an ID if needed
+	if(ClientID == 0)
+		ClientID = Threads_CreateRootProcess();
+	
 	for( i = 0; i < MAX_CLIENTS; i ++ )
 	{
 		if( gaServer_Clients[i].ClientID == ClientID ) {
-			ret = &gaServer_Clients[i];
-			break;
+			return &gaServer_Clients[i];
 		}
+		if(!ret && gaServer_Clients[i].ClientID == 0)
+			ret = &gaServer_Clients[i];
 	}
 	
 	// Uh oh, no free slots
@@ -72,44 +97,69 @@ tClient *Server_GetClient(int ClientID)
 	if( !ret )
 		return NULL;
 	
-	if( ClientID == 0 )
-	{
-		ret->ClientID = giServer_NextClientID ++;
-		ret->CurrentRequest = NULL;
-		
-		if( !ret->WorkerThread ) {
-			ret->WorkerThread = SDL_CreateThread( Server_WorkerThread, ret );
-			ret->WaitFlag = SDL_CreateCond();
-			ret->Mutex = SDL_CreateMutex();
-			SDL_mutexP( ret->Mutex );
-		}
+	// Allocate a thread for the process
+	ret->ClientID = ClientID;
+	ret->CurrentRequest = NULL;
+		
+	if( !ret->WorkerThread ) {
+		ret->WaitFlag = SDL_CreateCond();
+		ret->Mutex = SDL_CreateMutex();
+		SDL_mutexP( ret->Mutex );
+		ret->WorkerThread = SDL_CreateThread( Server_WorkerThread, ret );
 	}
 	
-	return &gaServer_Clients[i];
+	return ret;
 }
 
 int Server_WorkerThread(void *ClientPtr)
 {
 	tClient	*Client = ClientPtr;
 	tRequestHeader	*retHeader;
+	tRequestHeader	errorHeader;
 	 int	retSize = 0;
 	 int	sentSize;
+	 int	cur_client_id = 0;
 	
 	#if USE_TCP
 	#else
 	for( ;; )
 	{
 		// Wait for something to do
-		while( !Client->CurrentRequest )	;
+		while( Client->CurrentRequest == NULL )
+			SDL_CondWait(Client->WaitFlag, Client->Mutex);
+		
+		if(Client->ClientID != cur_client_id) {
+			Threads_SetThread( Client->ClientID );
+			cur_client_id = Client->ClientID;
+		}
+		
+		Log_Debug("AcessSrv", "Worker %i takes %p",
+			Client->ClientID, Client->CurrentRequest);
 		
 		// Get the response
 		retHeader = SyscallRecieve(Client->CurrentRequest, &retSize);
 		
 		if( !retHeader ) {
 			// Return an error to the client
-			printf("Error returned by SyscallRecieve\n");
+			printf("ERROR: SyscallRecieve failed\n");
+			errorHeader.CallID = Client->CurrentRequest->CallID;
+			errorHeader.NParams = 0;
+			retHeader = &errorHeader;
+			retSize = sizeof(errorHeader);
 		}
 		
+		// Set ID
+		retHeader->ClientID = Client->ClientID;
+		
+		// Mark the thread as ready for another job
+		Client->CurrentRequest = 0;
+		
+		Log_Debug("AcessSrv", "Sending %i to %x:%i (Client %i)",
+			retSize, ntohl(Client->ClientAddr.sin_addr.s_addr),
+			ntohs(Client->ClientAddr.sin_port),
+			Client->ClientID
+			);
+		
 		// Return the data
 		sentSize = sendto(gSocket, retHeader, retSize, 0,
 			(struct sockaddr*)&Client->ClientAddr, sizeof(Client->ClientAddr)
@@ -119,12 +169,8 @@ int Server_WorkerThread(void *ClientPtr)
 		}
 		
 		// Free allocated header
-		free( retHeader );
-		
-		Client->CurrentRequest = 0;
-		
-		// Wait for something else
-		SDL_CondWait(Client->WaitFlag, Client->Mutex);
+		if( retHeader != &errorHeader )
+			free( retHeader );
 	}
 	#endif
 }
@@ -182,7 +228,7 @@ int SyscallServer(void)
 	listen(gSocket, 5);
 	#endif
 	
-	Log_Notice("Syscall", "Listening on 0.0.0.0:%i\n", SERVER_PORT);
+	Log_Notice("AcessSrv", "Listening on 0.0.0.0:%i", SERVER_PORT);
 	
 	// Wait for something to do :)
 	for( ;; )
@@ -196,7 +242,7 @@ int SyscallServer(void)
 			break ;
 		}
 		
-		printf("Client connection %x:%i",
+		Log("Client connection %x:%i\n",
 			ntohl(client.sin_addr), ntohs(client.sin_port)
 			);
 		
@@ -205,9 +251,11 @@ int SyscallServer(void)
 		tRequestHeader	*req = (void*)data;
 		struct sockaddr_in	addr;
 		uint	clientSize = sizeof(addr);
-		 int	length = recvfrom(gSocket, data, BUFSIZ, 0, (struct sockaddr*)&addr, &clientSize);
+		 int	length;
 		tClient	*client;
 		
+		length = recvfrom(gSocket, data, BUFSIZ, 0, (struct sockaddr*)&addr, &clientSize);
+		
 		if( length == -1 ) {
 			perror("SyscallServer - recv");
 			break;
@@ -219,7 +267,8 @@ int SyscallServer(void)
 			ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port));
 		
 		client = Server_GetClient(req->ClientID);
-		if( req->ClientID == 0 )
+		// NOTE: Hack - Should check if all zero
+		if( req->ClientID == 0 || client->ClientAddr.sin_port == 0 )
 		{
 			memcpy(&client->ClientAddr, &addr, sizeof(addr));
 		}
@@ -238,6 +287,9 @@ int SyscallServer(void)
 			continue;
 		}
 		
+		Log_Debug("AcessSrv", "Message from Client %i (%p)",
+			client->ClientID, client);
+		
 		client->CurrentRequest = req;
 		SDL_CondSignal(client->WaitFlag);
 		#endif