From cf6eb857ea9f25074002bf660e844e6659626b2b Mon Sep 17 00:00:00 2001
From: Felix von Leitner <felix-libowfat@fefe.de>
Date: Thu, 26 Feb 2004 18:01:15 +0000
Subject: [PATCH] add two useful little test programs

---
 CHANGES       |   1 +
 test/client.c | 105 ++++++++++++++++++++++++++++++++++++++++++++++
 test/server.c | 113 ++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 219 insertions(+)
 create mode 100644 test/client.c
 create mode 100644 test/server.c

diff --git a/CHANGES b/CHANGES
index c85b428..7b27483 100644
--- a/CHANGES
+++ b/CHANGES
@@ -2,6 +2,7 @@
   make libowfat compile on BSD again (sorry, and thanks to everyone who
     sent me a patch :-D)
   add iob_addfile_close, needed for gatling 0.5+
+  add test/client and test/server, two simple TCP shell interfaces
 
 0.17:
   add Linux SIGIO support to IO
diff --git a/test/client.c b/test/client.c
new file mode 100644
index 0000000..921f4f9
--- /dev/null
+++ b/test/client.c
@@ -0,0 +1,105 @@
+#include "scan.h"
+#include "ip6.h"
+#include "str.h"
+#include <errno.h>
+#include <string.h>
+#include <sys/poll.h>
+#include <unistd.h>
+#include "socket.h"
+#include <sys/socket.h>
+#ifdef __dietlibc__
+#include <write12.h>
+#else
+static inline __write1(const char* s) { write(1,s,str_len(s)); }
+static inline __write2(const char* s) { write(2,s,str_len(s)); }
+#endif
+
+void panic(const char* s) {
+  int i=str_len(s);
+  __write2(s);
+  if (s[i-1]!='\n') {
+    __write2(": ");
+    __write2(strerror(errno));
+    __write2("\n");
+  }
+  exit(1);
+}
+
+void blockingcopy(int src,int dest) {
+  char buf[8192];
+  int i,j,k;
+  for (;;) {
+    i=read(src,buf,sizeof buf);
+    if (i==-1) panic("client: error: read() failed");
+    if (i==0) exit(0);
+    for (j=0; j<i; ) {
+      k=write(dest,buf+j,i-j);
+      if (k==-1) panic("client: error: write() failed");
+      if (k==0) panic("client: error: short write\n");
+      j+=k;
+    }
+  }
+}
+
+int main(int argc,char* argv[]) {
+  int s;
+  uint16 port=2342;
+  uint32 scope_id=0;
+  char ip[16];
+  char buf[8192];
+  int i;
+  struct pollfd p[2];
+
+  if (argc<3) {
+usage:
+    __write2("usage: client ip port\n\nexample: dump -0 -f - | client 10.0.0.149 8000\n");
+    return 0;
+  }
+  if (argv[1][i=scan_ip6if(argv[1],ip,&scope_id)]) {
+    __write2("client: error: invalid ip address!\n");
+    goto usage;
+  }
+  if (argv[2][scan_ushort(argv[2],&port)]) {
+    __write2("client: error: invalid port number!\n");
+    goto usage;
+  }
+  if (byte_equal(ip,12,V4mappedprefix)) {
+    s=socket_tcp4();
+    if (s==-1) panic("client: error: socket() failed");
+    if (socket_connect4(s,ip+12,port)==-1) panic("client: error: connect() failed");
+  } else {
+    s=socket_tcp6();
+    if (s==-1) panic("client: error: socket() failed");
+    if (socket_connect6(s,ip,port,scope_id)==-1) panic("client: error: connect() failed");
+  }
+  p[0].fd=0; p[0].events=POLLIN;
+  p[1].fd=s; p[1].events=POLLIN;
+  while (poll(p,2,5000)) {
+    if (p[0].revents) {
+      int j;
+      if (p[0].revents&POLLERR)
+	panic("client: error: poll() signals POLLERR\n");
+      i=read(0,buf,sizeof(buf));
+      if (i==0) { shutdown(s,SHUT_WR); blockingcopy(s,1); }
+      for (j=0; j<i; ) {
+	int k=write(s,buf+j,i-j);
+	if (k==-1) panic("client: error: write() failed");
+	if (k==0) panic("client: error: short write!\n");
+	j+=k;
+      }
+    }
+    if (p[1].revents) {
+      int j;
+      if (p[1].revents&POLLERR)
+	panic("client: error: poll() signals POLLERR\n");
+      i=read(s,buf,sizeof(buf));
+      if (i==0) { shutdown(1,SHUT_WR); blockingcopy(0,s); }
+      for (j=0; j<i; ) {
+	int k=write(1,buf+j,i-j);
+	if (k==-1) panic("client: error: write() failed");
+	if (k==0) panic("client: error: short write!\n");
+	j+=k;
+      }
+    }
+  }
+}
diff --git a/test/server.c b/test/server.c
new file mode 100644
index 0000000..af9d81b
--- /dev/null
+++ b/test/server.c
@@ -0,0 +1,113 @@
+#include "scan.h"
+#include "ip6.h"
+#include "str.h"
+#include <errno.h>
+#include <string.h>
+#include <sys/poll.h>
+#include <unistd.h>
+#include "socket.h"
+#include <sys/socket.h>
+#ifdef __dietlibc__
+#include <write12.h>
+#else
+static inline __write1(const char* s) { write(1,s,str_len(s)); }
+static inline __write2(const char* s) { write(2,s,str_len(s)); }
+#endif
+
+void panic(const char* s) {
+  int i=str_len(s);
+  __write2(s);
+  if (s[i-1]!='\n') {
+    __write2(": ");
+    __write2(strerror(errno));
+    __write2("\n");
+  }
+  exit(1);
+}
+
+void blockingcopy(int src,int dest) {
+  char buf[8192];
+  int i,j,k;
+  for (;;) {
+    i=read(src,buf,sizeof buf);
+    if (i==-1) panic("server: error: read() failed");
+    if (i==0) exit(0);
+    for (j=0; j<i; ) {
+      k=write(dest,buf+j,i-j);
+      if (k==-1) panic("server: error: write() failed");
+      if (k==0) panic("server: error: short write\n");
+      j+=k;
+    }
+  }
+}
+
+int main(int argc,char* argv[]) {
+  int s,t;
+  uint16 port=2342;
+  uint32 scope_id=0;
+  char ip[16];
+  char buf[8192];
+  int i;
+  struct pollfd p[2];
+
+  if (argc<3) {
+usage:
+    __write2("usage: server ip port\n\nexample: server 0 8000 | restore x -f -\n");
+    return 0;
+  }
+  if (!strcmp(argv[1],"0"))
+    byte_zero(ip,16);
+  else
+    if (argv[1][i=scan_ip6if(argv[1],ip,&scope_id)]) {
+      __write2("server: error: invalid ip address!\n");
+      goto usage;
+    }
+  if (argv[2][scan_ushort(argv[2],&port)]) {
+    __write2("server: error: invalid port number!\n");
+    goto usage;
+  }
+  if (byte_equal(ip,12,V4mappedprefix)) {
+    s=socket_tcp4();
+    if (s==-1) panic("server: error: socket() failed");
+    if (socket_bind4_reuse(s,ip+12,port)==-1) panic("server: error: bind() failed");
+    if (socket_listen(s,1)==-1) panic("server: error: listen() failed");
+    if ((t=socket_accept4(s,0,0))==-1) panic("server: error: accept() failed");
+  } else {
+    s=socket_tcp6();
+    if (s==-1) panic("server: error: socket() failed");
+    if (socket_bind6_reuse(s,ip,port,scope_id)==-1) panic("server: error: bind() failed");
+    if (socket_listen(s,1)==-1) panic("server: error: listen() failed");
+    if ((t=socket_accept6(s,0,0,0))==-1) panic("server: error: accept() failed");
+  }
+  close(s);
+  p[0].fd=0; p[0].events=POLLIN;
+  p[1].fd=t; p[1].events=POLLIN;
+  while (poll(p,2,5000)) {
+    if (p[0].revents) {
+      int j;
+      if (p[0].revents&POLLERR)
+	panic("server: error: poll() signals POLLERR\n");
+      i=read(0,buf,sizeof(buf));
+      if (i==0) { shutdown(t,SHUT_WR); blockingcopy(t,1); }
+      for (j=0; j<i; ) {
+	int k=write(t,buf+j,i-j);
+	if (k==-1) panic("server: error: write() failed");
+	if (k==0) panic("server: error: short write!\n");
+	j+=k;
+      }
+    }
+    if (p[1].revents) {
+      int j;
+      if (p[1].revents&POLLERR)
+	panic("server: error: poll() signals POLLERR\n");
+      i=read(t,buf,sizeof(buf));
+      if (i==0) { shutdown(1,SHUT_WR); blockingcopy(0,t); }
+      for (j=0; j<i; ) {
+	int k=write(1,buf+j,i-j);
+	if (k==-1) panic("server: error: write() failed");
+	if (k==0) panic("server: error: short write!\n");
+	j+=k;
+      }
+    }
+  }
+}