From e5ac848901659fc846dce29f399e091a2d70a89d Mon Sep 17 00:00:00 2001
From: Stefan Lankes <lankes@lfbs.rwth-aachen.de>
Date: Sat, 17 Sep 2011 00:56:52 +0200
Subject: [PATCH] extend BSD socket examples (create more messages for a stress
 test)

---
 newlib/examples/client.c | 22 ++++++++++++++++++++--
 newlib/examples/server.c | 15 ++++++++++++++-
 2 files changed, 34 insertions(+), 3 deletions(-)

diff --git a/newlib/examples/client.c b/newlib/examples/client.c
index ae1f0a0a..1b6da07f 100644
--- a/newlib/examples/client.c
+++ b/newlib/examples/client.c
@@ -12,10 +12,14 @@
 #undef errno
 extern int errno;
 
+#define QUIT	0
+#define S2C	1
 #define MAX_BUF 100
+#define N	10
 
 int main(int argc, char* argv[])
 {
+	int i, cmd;
 	int sockd;
 	int count;
 	struct sockaddr_in serv_name;
@@ -48,10 +52,24 @@ int main(int argc, char* argv[])
 		exit(1);
 	}
 
-	count = read(sockd, buf, MAX_BUF);
-	write(1, buf, count);
+	for(i=0, cmd=S2C; i<N; i++) {
+		// request new message
+		write(sockd, &cmd, sizeof(int));
+
+		// dump current value of i
+		snprintf(buf, MAX_BUF, "%d. ", i);
+		write(1, buf, strlen(buf));
+
+		// read new message and dump the incoming message on the screen
+		count = read(sockd, buf, MAX_BUF);
+		write(1, buf, count);
+	}
+	cmd = QUIT;
+	write(sockd, &cmd, sizeof(int));
 
 	close(sockd);
 
+	printf("Leave client...\n");
+
 	return 0;
 }
diff --git a/newlib/examples/server.c b/newlib/examples/server.c
index 35e09e43..c7f0d03f 100644
--- a/newlib/examples/server.c
+++ b/newlib/examples/server.c
@@ -14,12 +14,16 @@ extern int errno;
 
 static char msg[] =" Hello from server!\n";
 
+#define QUIT	0
+#define S2C	1
+
 int main(int argc, char* argv[])
 {
 	int sockd, sockd2;
 	unsigned int addrlen;
 	struct sockaddr_in my_name, peer_name;
 	int status;
+	int cmd, i=0;
 
 	/* create a socket */
 	sockd = socket(AF_INET, SOCK_STREAM, 0);
@@ -64,7 +68,16 @@ int main(int argc, char* argv[])
 			perror("Wrong connection");
 			exit(1);
 		}
-		write(sockd2, msg, strlen(msg)+1);
+
+again:
+		read(sockd2, &cmd, sizeof(cmd));
+		printf("server got %d. command: %d\n", i, cmd);
+		i++;
+		if (cmd == S2C)	
+			write(sockd2, msg, strlen(msg)+1);
+		if (cmd != QUIT)
+			goto again;
+
 		close(sockd2);
 	}