7 # define DEBUG_S printf
10 # define DONT_INCLUDE_SYSCALL_NAMES
22 # include <sys/socket.h>
23 # include <netinet/in.h>
26 #include "../syscalls.h"
31 void SendData(void *Data, int Length);
32 int ReadData(void *Dest, int MaxLen, int Timeout);
37 SOCKET gSocket = INVALID_SOCKET;
39 # define INVALID_SOCKET -1
40 int gSocket = INVALID_SOCKET;
42 // Client ID to pass to server
43 // TODO: Implement such that each thread gets a different one
44 int giSyscall_ClientID = 0;
45 struct sockaddr_in gSyscall_ServerAddr;
48 void Request_Preinit(void)
51 memset((void *)&gSyscall_ServerAddr, '\0', sizeof(struct sockaddr_in));
52 gSyscall_ServerAddr.sin_family = AF_INET;
53 gSyscall_ServerAddr.sin_port = htons(SERVER_PORT);
56 int _InitSyscalls(void)
59 /* Open windows connection */
60 if (WSAStartup(0x0101, &gWinsock) != 0)
62 fprintf(stderr, "Could not open Windows connection.\n");
68 // Open TCP Connection
69 gSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
71 // Open UDP Connection
72 gSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
74 if (gSocket == INVALID_SOCKET)
76 fprintf(stderr, "Could not create socket.\n");
85 memset((void *)&client, '\0', sizeof(struct sockaddr_in));
86 client.sin_family = AF_INET;
87 client.sin_port = htons(0);
88 client.sin_addr.s_addr = htonl(0x7F000001);
92 if( connect(gSocket, (struct sockaddr *)&gSyscall_ServerAddr, sizeof(struct sockaddr_in)) < 0 )
94 fprintf(stderr, "[ERROR -] Cannot connect to server (localhost:%i)\n", SERVER_PORT);
95 perror("_InitSyscalls");
108 if( bind(gSocket, (struct sockaddr *)&client, sizeof(struct sockaddr_in)) == -1 )
110 fprintf(stderr, "Cannot bind address to socket.\n");
112 closesocket(gSocket);
123 tRequestAuthHdr auth;
124 auth.pid = giSyscall_ClientID;
125 SendData(&auth, sizeof(auth));
126 int len = ReadData(&auth, sizeof(auth), 5);
128 fprintf(stderr, "Timeout waiting for auth response\n");
131 giSyscall_ClientID = auth.pid;
134 // Ask server for a client ID
135 if( !giSyscall_ClientID )
143 SendData(&req, sizeof(req));
145 len = ReadData(&req, sizeof(req), 5);
147 fprintf(stderr, "Unable to connect to server (localhost:%i)\n", SERVER_PORT);
151 giSyscall_ClientID = req.ClientID;
159 * \brief Close the syscall socket
160 * \note Used in acess_fork to get a different port number
162 void _CloseSyscalls(void)
165 closesocket(gSocket);
172 int SendRequest(tRequestHeader *Request, int RequestSize, int ResponseSize)
174 if( gSocket == INVALID_SOCKET )
180 Request->ClientID = giSyscall_ClientID;
184 for(i=0;i<RequestSize;i++)
186 printf("%02x ", ((uint8_t*)Request)[i]);
187 if( i % 16 == 15 ) printf("\n");
194 char *data = (char*)&Request->Params[Request->NParams];
195 DEBUG_S("Request #%i (%s) -", Request->CallID, casSYSCALL_NAMES[Request->CallID]);
196 for( i = 0; i < Request->NParams; i ++ )
198 switch(Request->Params[i].Type)
201 DEBUG_S(" 0x%08x", *(uint32_t*)data);
202 data += sizeof(uint32_t);
205 DEBUG_S(" 0x%016"PRIx64"", *(uint64_t*)data);
206 data += sizeof(uint64_t);
208 case ARG_TYPE_STRING:
209 DEBUG_S(" '%s'", (char*)data);
210 data += Request->Params[i].Length;
213 DEBUG_S(" %p:0x%x", (char*)data, Request->Params[i].Length);
214 if( !(Request->Params[i].Flags & ARG_FLAG_ZEROED) )
215 data += Request->Params[i].Length;
223 SendData(Request, RequestSize);
225 if( Request->CallID == SYS_EXIT ) return 0;
227 // Wait for a response (no timeout)
228 return ReadData(Request, ResponseSize, 0);
231 void SendData(void *Data, int Length)
236 len = send(gSocket, Data, Length, 0);
238 len = sendto(gSocket, Data, Length, 0,
239 (struct sockaddr*)&gSyscall_ServerAddr, sizeof(gSyscall_ServerAddr));
242 if( len != Length ) {
243 fprintf(stderr, "[ERROR %i] ", giSyscall_ClientID);
249 int ReadData(void *Dest, int MaxLength, int Timeout)
254 struct timeval *timeoutPtr;
257 FD_SET(gSocket, &fds);
268 ret = select(gSocket+1, &fds, NULL, NULL, timeoutPtr);
270 fprintf(stderr, "[ERROR %i] ", giSyscall_ClientID);
271 perror("ReadData - select");
276 printf("[ERROR %i] Timeout reading from socket\n", giSyscall_ClientID);
281 ret = recv(gSocket, Dest, MaxLength, 0);
283 ret = recvfrom(gSocket, Dest, MaxLength, 0, NULL, 0);
287 fprintf(stderr, "[ERROR %i] ", giSyscall_ClientID);
292 DEBUG_S("%i bytes read from socket\n", ret);