1 files changed, 41 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" );
+	}
+}