3 * - Acess kernel emulation on another OS using SDL and UDP
16 # include <sys/socket.h>
17 # include <netinet/in.h>
19 #include "../syscalls.h"
23 #define MAX_CLIENTS 16
28 SDL_Thread *WorkerThread;
31 tRequestHeader *CurrentRequest;
32 struct sockaddr_in ClientAddr;
39 extern tRequestHeader *SyscallRecieve(tRequestHeader *Request, int *ReturnLength);
40 extern int Threads_CreateRootProcess(void);
41 extern void Threads_SetThread(int TID);
42 // HACK: Should have these in a header
43 extern void Log_Debug(const char *Subsys, const char *Message, ...);
44 extern void Log_Notice(const char *Subsys, const char *Message, ...);
47 tClient *Server_GetClient(int ClientID);
48 int Server_WorkerThread(void *ClientPtr);
49 int SyscallServer(void);
50 int Server_ListenThread(void *Unused);
55 SOCKET gSocket = INVALID_SOCKET;
57 # define INVALID_SOCKET -1
58 int gSocket = INVALID_SOCKET;
60 tClient gaServer_Clients[MAX_CLIENTS];
61 SDL_Thread *gpServer_ListenThread;
64 int Server_GetClientID(void)
67 Uint32 thisId = SDL_ThreadID();
69 for( i = 0; i < MAX_CLIENTS; i ++ )
71 if( SDL_GetThreadID(gaServer_Clients[i].WorkerThread) == thisId )
72 return gaServer_Clients[i].ClientID;
75 fprintf(stderr, "ERROR: Server_GetClientID - Thread is not allocated\n");
80 tClient *Server_GetClient(int ClientID)
85 // Allocate an ID if needed
87 ClientID = Threads_CreateRootProcess();
89 for( i = 0; i < MAX_CLIENTS; i ++ )
91 if( gaServer_Clients[i].ClientID == ClientID ) {
92 return &gaServer_Clients[i];
94 if(!ret && gaServer_Clients[i].ClientID == 0)
95 ret = &gaServer_Clients[i];
98 // Uh oh, no free slots
99 // TODO: Dynamic allocation
103 // Allocate a thread for the process
104 ret->ClientID = ClientID;
105 ret->CurrentRequest = NULL;
107 if( !ret->WorkerThread ) {
108 ret->WaitFlag = SDL_CreateCond();
109 ret->Mutex = SDL_CreateMutex();
110 SDL_mutexP( ret->Mutex );
111 ret->WorkerThread = SDL_CreateThread( Server_WorkerThread, ret );
117 int Server_WorkerThread(void *ClientPtr)
119 tClient *Client = ClientPtr;
120 tRequestHeader *retHeader;
121 tRequestHeader errorHeader;
124 int cur_client_id = 0;
130 // Wait for something to do
131 while( Client->CurrentRequest == NULL )
132 SDL_CondWait(Client->WaitFlag, Client->Mutex);
134 Log_Debug("AcessSrv", "Worker got message %p", Client->CurrentRequest);
136 if(Client->ClientID != cur_client_id) {
137 Threads_SetThread( Client->ClientID );
138 cur_client_id = Client->ClientID;
143 int callid = Client->CurrentRequest->CallID;
144 Log_Debug("AcessSrv", "Client %i request %i %s",
145 Client->ClientID, callid,
146 callid < N_SYSCALLS ? casSYSCALL_NAMES[callid] : "UNK"
151 retHeader = SyscallRecieve(Client->CurrentRequest, &retSize);
154 // Return an error to the client
155 printf("ERROR: SyscallRecieve failed\n");
156 errorHeader.CallID = Client->CurrentRequest->CallID;
157 errorHeader.NParams = 0;
158 retHeader = &errorHeader;
159 retSize = sizeof(errorHeader);
163 retHeader->ClientID = Client->ClientID;
165 // Mark the thread as ready for another job
166 free(Client->CurrentRequest);
167 Client->CurrentRequest = 0;
169 // Log_Debug("AcessSrv", "Sending %i to %x:%i (Client %i)",
170 // retSize, ntohl(Client->ClientAddr.sin_addr.s_addr),
171 // ntohs(Client->ClientAddr.sin_port),
176 sentSize = sendto(gSocket, retHeader, retSize, 0,
177 (struct sockaddr*)&Client->ClientAddr, sizeof(Client->ClientAddr)
179 if( sentSize != retSize ) {
180 perror("Server_WorkerThread - send");
183 // Free allocated header
184 if( retHeader != &errorHeader )
190 int SyscallServer(void)
192 struct sockaddr_in server;
195 /* Open windows connection */
196 if (WSAStartup(0x0101, &gWinsock) != 0)
198 fprintf(stderr, "Could not open Windows connection.\n");
204 // Open TCP Connection
205 gSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
207 // Open UDP Connection
208 gSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
210 if (gSocket == INVALID_SOCKET)
212 fprintf(stderr, "Could not create socket.\n");
219 // Set server address
220 memset(&server, 0, sizeof(struct sockaddr_in));
221 server.sin_family = AF_INET;
222 server.sin_port = htons(SERVER_PORT);
223 server.sin_addr.s_addr = htonl(INADDR_ANY);
226 if( bind(gSocket, (struct sockaddr *)&server, sizeof(struct sockaddr_in)) == -1 )
228 fprintf(stderr, "Cannot bind address to socket.\n");
229 perror("SyscallServer - bind");
231 closesocket(gSocket);
243 Log_Notice("AcessSrv", "Listening on 0.0.0.0:%i", SERVER_PORT);
244 gpServer_ListenThread = SDL_CreateThread( Server_ListenThread, NULL );
248 int Server_ListenThread(void *Unused)
250 // Wait for something to do :)
254 struct sockaddr_in client;
255 uint clientSize = sizeof(client);
256 int clientSock = accept(gSocket, (struct sockaddr*)&client, &clientSize);
257 if( clientSock < 0 ) {
258 perror("SyscallServer - accept");
262 Log("Client connection %x:%i\n",
263 ntohl(client.sin_addr), ntohs(client.sin_port)
268 tRequestHeader *req = (void*)data;
269 struct sockaddr_in addr;
270 uint clientSize = sizeof(addr);
274 length = recvfrom(gSocket, data, BUFSIZ, 0, (struct sockaddr*)&addr, &clientSize);
277 perror("SyscallServer - recv");
281 // Hand off to a worker thread
282 // - TODO: Actually have worker threads
283 // Log_Debug("Server", "%i bytes from %x:%i", length,
284 // ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port));
286 client = Server_GetClient(req->ClientID);
287 // NOTE: Hack - Should check if all zero
288 if( req->ClientID == 0 || client->ClientAddr.sin_port == 0 )
290 memcpy(&client->ClientAddr, &addr, sizeof(addr));
292 else if( memcmp(&client->ClientAddr, &addr, sizeof(addr)) != 0 )
294 printf("ClientID %i used by %x:%i\n",
295 client->ClientID, ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port));
296 printf(" actually owned by %x:%i\n",
297 ntohl(client->ClientAddr.sin_addr.s_addr), ntohs(client->ClientAddr.sin_port));
301 if( client->CurrentRequest ) {
302 printf("Worker thread for %x:%i is busy\n",
303 ntohl(client->ClientAddr.sin_addr.s_addr), ntohs(client->ClientAddr.sin_port));
307 Log_Debug("AcessSrv", "Message from Client %i (%p)",
308 client->ClientID, client);
310 // Make a copy of the request data
311 req = malloc(length);
312 memcpy(req, data, length);
313 client->CurrentRequest = req;
314 SDL_CondSignal(client->WaitFlag);