X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Tools%2FNetTest%2Ftcpserver.c;h=5e4aa7277c61ced5404670dac968d1272a0f664e;hb=7e9bbefbdcbfdba27eb6cdacae0811f428483892;hp=e69de29bb2d1d6434b8b29ae775ad8c2e48c5391;hpb=7ba570fe3cc5418f42decf5b72ac2295cce9e60f;p=tpg%2Facess2.git

diff --git a/Tools/NetTest/tcpserver.c b/Tools/NetTest/tcpserver.c
index e69de29b..5e4aa727 100644
--- 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++
+		}
+	}
+}
+