ARMv7 - Fix compiler and partial compilation

[tpg/acess2.git] / Tools / NetTest / tcpserver.c

diff --git a/Tools/NetTest/tcpserver.c b/Tools/NetTest/tcpserver.c
index e69de29..5e4aa72 100644 (file)
--- a/Tools/NetTest/tcpserver.c
+++ b/Tools/NetTest/tcpserver.c
@@ -0,0 +1,106 @@
+/*
+ * Acess2 Networking Test Suite (NetTest)
+ * - By John Hodge (thePowersGang)
+ *
+ * tcpserver.c
+ * - TCP Client tester
+ */
+#define DEBUG  1
+#include <vfs.h>
+#include <vfs_ext.h>
+#include <nettest.h>
+#include "tcpserver.h"
+
+#define MAX_CLIENTS    4
+
+struct sNetTest_TCPServer
+{
+        int    ServerFD;
+        int    nClients;
+       struct sClient {
+                int    FD;
+       } Clients[MAX_CLIENTS];
+};
+
+// === CODE ===
+tNetTest_TCPServer *NetTest_TCPServer_Create(int Port)
+{
+       tNetTest_TCPServer *ret;
+       ret = calloc(sizeof(*ret), 1);
+       ret->ServerFD = VFS_Open("/Devices/ip/1/tcps", VFS_OPENFLAG_READ);
+       ASSERT(ret->ServerFD >= 0);
+       VFS_IOCtl(ret->ServerFD, 4, &Port);
+       return ret;
+}
+
+void NetTest_TCPServer_Close(tNetTest_TCPServer *Srv)
+{
+       VFS_Close(Srv->ServerFD);
+       for( int i = 0; i < Srv->nClients; i ++ )
+       {
+               struct sClient  *client = &Srv->Clients[i];
+               VFS_Close(client->FD);
+       }
+       free(Srv);
+}
+
+int NetTest_TCPServer_FillSelect(tNetTest_TCPServer *Srv, fd_set *fds)
+{
+        int    max = -1;
+       
+       if( Srv->nClients < MAX_CLIENTS ) {
+               max = Srv->ServerFD;
+               FD_SET(Srv->ServerFD, fds);
+       }
+       for( int i = 0; i < Srv->nClients; i ++ )
+       {
+                int    fd = Srv->Clients[i].FD;
+               if( fd > max )  max = fd;
+               FD_SET(fd, fds);
+       }
+       return max;
+}
+
+void NetTest_TCPServer_HandleSelect(tNetTest_TCPServer *Srv, const fd_set *rfds, const fd_set *wfds, const fd_set *efds)
+{
+       LOG("Srv=%p", Srv);
+       if( FD_ISSET(Srv->ServerFD, rfds) )
+       {
+               // New connection!
+               ASSERT(Srv->nClients != MAX_CLIENTS);
+               struct sClient *client = &Srv->Clients[Srv->nClients++];
+               LOG("Child?");
+               client->FD = VFS_OpenChild(Srv->ServerFD, "", VFS_OPENFLAG_READ|VFS_OPENFLAG_WRITE);
+               LOG("client->FD = %i", client->FD);
+       }
+       if( FD_ISSET(Srv->ServerFD, efds) )
+       {
+               LOG("Oops, error on server");
+               VFS_Close(Srv->ServerFD);
+               Srv->ServerFD = -1;
+       }
+
+       for( int i = 0; i < Srv->nClients; i ++ )
+       {
+               struct sClient  *client = &Srv->Clients[i];
+               if( FD_ISSET(client->FD, rfds) )
+               {
+                       // RX'd data on client
+                       // TODO: Cache until '.' is seen, and send all data before+incl that character
+                       char    buf[1024];
+                       size_t len = VFS_Read(client->FD, sizeof(buf), buf);
+                       Debug_HexDump("TCP Srv Rx", buf, len);
+                       VFS_Write(client->FD, len, buf);
+               }
+               
+               if( FD_ISSET(client->FD, efds) )
+               {
+                       // Drop client
+                       VFS_Close(client->FD);
+                       memmove(client, client+1, (Srv->nClients-i-1) * sizeof(*client));
+                       Srv->nClients --;
+                       i --;   // counteract i++
+               }
+       }
+}
+