2 files changed, 82 insertions, 0 deletions
diff --git a/src/tests/ftptest.c b/src/tests/ftptest.c
new file mode 100644
index 000000000..f8208de73
--- /dev/null
+++ b/src/tests/ftptest.c
@@ -0,0 +1,41 @@
+#include <stdint.h>
+#include <string.h>
+#include <byteswap.h>
+#include <console.h>
+#include <vsprintf.h>
+#include <gpxe/async.h>
+#include <gpxe/ftp.h>
+
+static void test_ftp_callback ( char *data, size_t len ) {
+	unsigned int i;
+	char c;
+
+	for ( i = 0 ; i < len ; i++ ) {
+		c = data[i];
+		if ( c == '\r' ) {
+			/* Print nothing */
+		} else if ( ( c == '\n' ) || ( c >= 32 ) || ( c <= 126 ) ) {
+			putchar ( c );
+		} else {
+			putchar ( '.' );
+		}
+	}
+}
+
+void test_ftp ( struct in_addr server, const char *filename ) {
+	struct ftp_request ftp;
+	int rc;
+
+	printf ( "FTP fetching %s:%s\n", inet_ntoa ( server ), filename );
+	
+	memset ( &ftp, 0, sizeof ( ftp ) );
+	ftp.tcp.sin.sin_addr.s_addr = server.s_addr;
+	ftp.tcp.sin.sin_port = htons ( FTP_PORT );
+	ftp.filename = filename;
+	ftp.callback = test_ftp_callback;
+
+	rc = async_wait ( ftp_get ( &ftp ) );
+	if ( rc ) {
+		printf ( "FTP fetch failed\n" );
+	}
+}
diff --git a/src/tests/hellotest.c b/src/tests/hellotest.c
new file mode 100644
index 000000000..e38856ee8
--- /dev/null
+++ b/src/tests/hellotest.c
@@ -0,0 +1,41 @@
+#include <stdint.h>
+#include <string.h>
+#include <byteswap.h>
+#include <console.h>
+#include <vsprintf.h>
+#include <gpxe/async.h>
+#include <gpxe/hello.h>
+
+static void test_hello_callback ( char *data, size_t len ) {
+	unsigned int i;
+	char c;
+
+	for ( i = 0 ; i < len ; i++ ) {
+		c = data[i];
+		if ( c == '\r' ) {
+			/* Print nothing */
+		} else if ( ( c == '\n' ) || ( c >= 32 ) || ( c <= 126 ) ) {
+			putchar ( c );
+		} else {
+			putchar ( '.' );
+		}
+	}
+}
+
+void test_hello ( struct sockaddr_in *server, const char *message ) {
+	struct hello_request hello;
+	int rc;
+
+	printf ( "Saying \"%s\" to %s:%d\n", message,
+		 inet_ntoa ( server->sin_addr ), ntohs ( server->sin_port ) );
+	
+	memset ( &hello, 0, sizeof ( hello ) );
+	hello.tcp.sin = *server;
+	hello.message = message;
+	hello.callback = test_hello_callback;
+
+	rc = async_wait ( say_hello ( &hello ) );
+	if ( rc ) {
+		printf ( "HELLO fetch failed\n" );
+	}
+}