From 218c26ff2b61e6c02ca246e0076337eb7b6cd1eb Mon Sep 17 00:00:00 2001
From: Michael Brown <mcb30@etherboot.org>
Date: Thu, 23 Mar 2006 13:06:08 +0000
Subject: [PATCH] Can now also print data sent by the remote side.

---
 src/util/prototester.c | 47 ++++++++++++++++++++++++++++++------------
 1 file changed, 34 insertions(+), 13 deletions(-)

diff --git a/src/util/prototester.c b/src/util/prototester.c
index 88a21638..b4a5bc9a 100644
--- a/src/util/prototester.c
+++ b/src/util/prototester.c
@@ -262,7 +262,8 @@ struct tcp_operations {
 	void ( * closed ) ( struct tcp_connection *conn );
 	void ( * connected ) ( struct tcp_connection *conn );
 	void ( * acked ) ( struct tcp_connection *conn, size_t len );
-	void ( * newdata ) ( struct tcp_connection *conn );
+	void ( * newdata ) ( struct tcp_connection *conn,
+			     void *data, size_t len );
 	void ( * senddata ) ( struct tcp_connection *conn );
 };
 
@@ -271,7 +272,7 @@ struct tcp_connection {
 	struct tcp_operations *tcp_op;
 };
 
-static int tcp_connect ( struct tcp_connection *conn ) {
+int tcp_connect ( struct tcp_connection *conn ) {
 	struct uip_conn *uip_conn;
 	u16_t ipaddr[2];
 
@@ -289,13 +290,13 @@ static int tcp_connect ( struct tcp_connection *conn ) {
 	return 0;
 }
 
-static void tcp_send ( struct tcp_connection *conn, const void *data,
+void tcp_send ( struct tcp_connection *conn, const void *data,
 		       size_t len ) {
 	assert ( conn = *( ( void ** ) uip_conn->appstate ) );
 	uip_send ( ( void * ) data, len );
 }
 
-static void tcp_close ( struct tcp_connection *conn ) {
+void tcp_close ( struct tcp_connection *conn ) {
 	assert ( conn = *( ( void ** ) uip_conn->appstate ) );
 	uip_close();
 }
@@ -321,7 +322,7 @@ void uip_tcp_appcall ( void ) {
 	if ( uip_acked() )
 		op->acked ( conn, uip_conn->len );
 	if ( uip_newdata() )
-		op->newdata ( conn );
+		op->newdata ( conn, ( void * ) uip_appdata, uip_len );
 	if ( uip_rexmit() || uip_newdata() || uip_acked() ||
 	     uip_connected() || uip_poll() )
 		op->senddata ( conn );
@@ -387,7 +388,7 @@ static void run_tcpip ( void ) {
 	(type *)( (char *)__mptr - offsetof(type,member) );})
 
 enum hello_state {
-	HELLO_SENDING_MESSAGE = 0,
+	HELLO_SENDING_MESSAGE = 1,
 	HELLO_SENDING_ENDL,
 };
 
@@ -395,8 +396,8 @@ struct hello_request {
 	struct tcp_connection tcp;
 	const char *message;
 	enum hello_state state;
-	int remaining;
-	void ( *callback ) ( struct hello_request *hello );
+	size_t remaining;
+	void ( *callback ) ( char *data, size_t len );
 	int complete;
 };
 
@@ -427,6 +428,9 @@ static void hello_closed ( struct tcp_connection *conn ) {
 
 static void hello_connected ( struct tcp_connection *conn ) {
 	struct hello_request *hello = tcp_to_hello ( conn );
+
+	printf ( "Connection established\n" );
+	hello->state = HELLO_SENDING_MESSAGE;
 }
 
 static void hello_acked ( struct tcp_connection *conn, size_t len ) {
@@ -434,7 +438,7 @@ static void hello_acked ( struct tcp_connection *conn, size_t len ) {
 
 	hello->message += len;
 	hello->remaining -= len;
-	if ( hello->remaining <= 0 ) {
+	if ( hello->remaining == 0 ) {
 		switch ( hello->state ) {
 		case HELLO_SENDING_MESSAGE:
 			hello->state = HELLO_SENDING_ENDL;
@@ -442,7 +446,9 @@ static void hello_acked ( struct tcp_connection *conn, size_t len ) {
 			hello->remaining = 2;
 			break;
 		case HELLO_SENDING_ENDL:
-			tcp_close ( conn );
+			/* Nothing to do once we've finished sending
+			 * the end-of-line indicator.
+			 */
 			break;
 		default:
 			assert ( 0 );
@@ -450,8 +456,11 @@ static void hello_acked ( struct tcp_connection *conn, size_t len ) {
 	}
 }
 
-static void hello_newdata ( struct tcp_connection *conn ) {
+static void hello_newdata ( struct tcp_connection *conn, void *data,
+			    size_t len ) {
 	struct hello_request *hello = tcp_to_hello ( conn );
+
+	hello->callback ( data, len );
 }
 
 static void hello_senddata ( struct tcp_connection *conn ) {
@@ -557,8 +566,20 @@ static int hello_parse_options ( int argc, char **argv,
 	return optind;
 }
 
-static void test_hello_callback ( struct hello_request *hello ) {
-	
+static void test_hello_callback ( char *data, size_t len ) {
+	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 ( '.' );
+		}
+	}	
 }
 
 static int test_hello ( int argc, char **argv ) {