1 files changed, 159 insertions, 0 deletions
diff --git a/tftpTest/test.c b/tftpTest/test.c
new file mode 100644
index 0000000..b900030
--- /dev/null
+++ b/tftpTest/test.c
@@ -0,0 +1,159 @@
+/*
+ * Test RTEMS/KA9Q TFTP device driver
+ *
+ * This program may be distributed and used for any purpose.
+ * I ask only that you:
+ *      1. Leave this author information intact.
+ *      2. Document any changes you make.
+ *
+ * W. Eric Norum
+ * Saskatchewan Accelerator Laboratory
+ * University of Saskatchewan
+ * Saskatoon, Saskatchewan, CANADA
+ * eric@skatter.usask.ca
+ */
+
+#include <stdio.h>
+#include <rtems.h>
+#include <rtems/error.h>
+#include <sys/fcntl.h>
+#include <unistd.h>
+#include <bootp.h>
+#include <netuser.h>
+
+static char cbuf[1024];
+static char *fullname;
+static rtems_interval then, now;
+
+static void
+showRate (unsigned long totalRead)
+{
+	int elapsed;
+
+	printf ("Read %lu bytes", totalRead);
+	elapsed = now - then;
+	if (elapsed) {
+		rtems_interval ticksPerSecond;
+		rtems_clock_get (RTEMS_CLOCK_GET_TICKS_PER_SECOND, &ticksPerSecond);
+		printf (" (%ld bytes/sec)",
+				(long)(((long long)totalRead * ticksPerSecond)
+								/ elapsed));
+	}
+	printf (".\n");
+rtems_ka9q_execute_command ("ifconfig rtems");
+}
+
+static void
+testRawRead (void)
+{
+	int fd;
+	int nread;
+	unsigned long totalRead = 0;
+
+	fd = open (fullname, O_RDONLY);
+	if (fd < 0) {
+		printf ("Open failed: %s\n", strerror (errno));
+		return;
+	}
+
+	rtems_clock_get (RTEMS_CLOCK_GET_TICKS_SINCE_BOOT, &then);
+	for (;;) {
+		nread = read (fd, cbuf, sizeof cbuf);
+		if (nread < 0) {
+			printf ("Read failed: %s\n", strerror (errno));
+			close (fd);
+			return;
+		}
+		if (nread == 0)
+			break;
+		totalRead += nread;
+	}
+	rtems_clock_get (RTEMS_CLOCK_GET_TICKS_SINCE_BOOT, &now);
+	close (fd);
+	showRate (totalRead);
+}
+
+static void
+testFread (void)
+{
+	FILE *fp;
+	int nread;
+	unsigned long totalRead = 0;
+
+	fp = fopen (fullname, "r");
+	if (fp == NULL) {
+		printf ("Open failed: %s\n", strerror (errno));
+		return;
+	}
+
+	rtems_clock_get (RTEMS_CLOCK_GET_TICKS_SINCE_BOOT, &then);
+	for (;;) {
+		nread = fread (cbuf, sizeof cbuf[0], sizeof cbuf, fp);
+		if (nread < 0) {
+			printf ("Read failed: %s\n", strerror (errno));
+			fclose (fp);
+			return;
+		}
+		if (nread == 0)
+			break;
+		totalRead += nread;
+	}
+	rtems_clock_get (RTEMS_CLOCK_GET_TICKS_SINCE_BOOT, &now);
+	fclose (fp);
+	showRate (totalRead);
+}
+
+static int
+makeFullname (rtems_unsigned32 host, const char *file)
+{
+	const char *hostname;
+
+	if (host) {
+		hostname = rtems_hostname_for_address (BootpHost.s_addr, 0);
+		if (hostname == NULL) {
+			printf ("No host to read from!\n");
+			return 0;
+		}
+	}
+	else {
+		hostname = "";
+	}
+	fullname = realloc (fullname, 8 + strlen (file) + strlen (hostname));
+	sprintf (fullname, "/TFTP/%s/%s", hostname, file);
+	printf ("Read `%s'.\n", fullname);
+	return 1;
+}
+
+void
+testTFTP (void)
+{
+	/*
+	 * Check that invalid file names are reported as such
+	 */
+	if (makeFullname (0, "")) {
+		testRawRead ();
+		testFread ();
+	}
+
+	/*
+	 * Check that non-existent files are reported as such
+	 */
+	if (makeFullname (0, "BAD-FILE-NAME")) {
+		testRawRead ();
+		testFread ();
+	}
+
+	/*
+	 * Check read speed
+	 */
+	if (BootpFileName == NULL) {
+		printf ("Nothing to read!\n");
+		return;
+	}
+	if (makeFullname (BootpHost.s_addr, BootpFileName)) {
+		testRawRead ();
+		testFread ();
+	}
+}
+	#include <limits.h>
+	int foo = INT_MAX;