X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Tools%2FNetTest%2Fmain.c;h=2aba06078af934ab99b7fd70aa0fa364f069d65b;hb=d7dcea0e5a8df0f479e99f168a10b9a9535c7ad6;hp=4f3e9f58fe24aa7b4b1620912b36707268aea5ca;hpb=2d39fe63812216118aec6520ccfb6ce2f11d6fd2;p=tpg%2Facess2.git

diff --git a/Tools/NetTest/main.c b/Tools/NetTest/main.c
index 4f3e9f58..2aba0607 100644
--- a/Tools/NetTest/main.c
+++ b/Tools/NetTest/main.c
@@ -9,7 +9,10 @@
 #include <acess_logging.h>
 #include <nettest.h>
 #include <string.h>
+#include <stdlib.h>
+#include <time.h>
 
+extern int	VFS_Init(void);
 extern int	IPStack_Install(char **Args);
 
 // === CODE ===
@@ -18,45 +21,106 @@ void PrintUsage(const char *ProgramName)
 	fprintf(stderr, "Usage: %s <commands...>\n", ProgramName);
 	fprintf(stderr, "\n");
 	fprintf(stderr,
-		"-dev <tapdev>:<mac>\n"
-		"-ip <dev>,<addr>,<mask>\n"
-		"-route <net>,<mask>,<nexthop>\n"
+		"Options:\n"
+		"-dev <mac>:<type>:<arg>\n"
+		"-ip <dev>,<addr>/<mask>\n"
+		"-route <net>/<mask>,<nexthop>\n"
+		"\n"
+		"Commands:\n"
+		"netcat <addr> <port>\n"
+		"\n"
+		"Device Types:\n"
+		"tun - Linux TUN/TAP device (takes an optional name)\n"
+		"unix - Unix named pipe (datagram)\n"
 		);
 }
 
 int main(int argc, char *argv[])
 {
+	 int	rv;
+	
 	if( argc <= 1 ) {
 		PrintUsage(argv[0]);
 		return 1;
 	}
-	
+
+	srand(time(NULL));	
+
 	// Startup
+	VFS_Init();
 	{
 		char	*ipstack_args[] = {NULL};
 		IPStack_Install( ipstack_args );
 	}
 	
-	for( int i = 0; i < argc; i ++ )
+	for( int i = 1; i < argc; i ++ )
 	{
-		if( argv[i][0] != '-' ) {
-		}
-		else if( strcmp(argv[i], "-dev") == 0 )
+		char *arg = argv[i];;
+		if( arg[0] != '-' )
 		{
-			if( ++i == argc ) { PrintUsage(argv[0]); return 1; }
-			NativeNic_AddDev(argv[i]);
+			if( strcmp(arg, "netcat") == 0 )
+			{
+				if( argc-i != 3 ) {
+					Log_Error("NetTest", "'netcat' <addr> <port>");
+					PrintUsage(argv[0]);
+					return -1;
+				}
+
+				NetTest_Suite_Netcat(argv[i+1], strtol(argv[i+2], NULL, 0));
+				i += 2;
+			}
+			else if( strcmp(arg, "cmdline") == 0 )
+			{
+				NetTest_Suite_Cmdline();
+			}
+			else
+			{
+				Log_Error("NetTest", "Unknown suite name '%s'", arg);
+				PrintUsage(argv[0]);
+			}
 		}
-		else if( strcmp(argv[i], "-ip") == 0 )
+		else
 		{
-			if( ++i == argc ) { PrintUsage(argv[0]); return 1; }
-			// TODO: parse argument and poke ipstack
+			if( strcmp(arg, "--help") == 0 )
+			{
+				PrintUsage(0);
+				return 0;
+			}
+			else if( strcmp(arg, "-dev") == 0 )
+			{
+				if( i+1 == argc ) { PrintUsage(argv[0]); return 1; }
+				rv = NativeNic_AddDev(argv[i+1]);
+				if( rv ) {
+					Log_Error("NetTest", "Failed to add device %s", argv[i+1]);
+					return -1;
+				}
+				i ++;
+			}
+			else if( strcmp(arg, "-ip") == 0 )
+			{
+				if( i+1 == argc ) { PrintUsage(argv[0]); return 1; }
+				// Parse argument and poke ipstack
+				if( NetTest_AddAddress(argv[i+1]) ) {
+					return -1;
+				}
+				i ++;
+			}
+			else
+			{
+				Log_Error("NetTest", "Unknown argument '%s'", arg);
+				PrintUsage(argv[0]);
+				return -1;
+			}
 		}
 	}
 
-	// Run suite
-	
-
 	// Teardown
-
+	Log_Log("NetTest", "Shutting down");
+	fflush(stdout);
 	return 0;
 }
+
+size_t NetTest_WriteStdout(const void *Data, size_t Size)
+{
+	return fwrite(Data, 1, Size, stdout);
+}