diff options
Diffstat (limited to 'contrib/testclient.c')
-rw-r--r-- | contrib/testclient.c | 159 |
1 files changed, 108 insertions, 51 deletions
diff --git a/contrib/testclient.c b/contrib/testclient.c index a5300ac..2c5324b 100644 --- a/contrib/testclient.c +++ b/contrib/testclient.c @@ -31,6 +31,7 @@ #include <arpa/inet.h> #include <stdio.h> #include <string.h> +#include <ctype.h> #include <unistd.h> #include <errno.h> #include <string.h> @@ -38,8 +39,8 @@ int main(int argc, char* argv[]) { - if(argc < 3) { - fprintf(stderr, "Usage: %s <addr> <port>\n", argv[0]); + if(argc < 4) { + fprintf(stderr, "Usage: %s <addr> <port> (A|B)\n", argv[0]); return -1; } @@ -67,67 +68,123 @@ int main(int argc, char* argv[]) return -1; } - char buf[1234567]; - unsigned int i; - for(i = 0; i<sizeof(buf); ++i) { - buf[i] = 'A' + i%62; - } - buf[sizeof(buf)-1] = '\n'; + switch(toupper(argv[3][0])) { +/*************** MODE A ***************/ + case 'A': { + printf("MODE: A\n"); - int nbwritten = send(c, buf, sizeof(buf), 0); - if(nbwritten <= 0) { - if(nbwritten < 0) - perror("send()"); - else - fprintf(stderr, "nothing sent... aborting\n"); - return -1; - } - printf("%d bytes sent\n", nbwritten); - if(nbwritten != sizeof(buf)) { - fprintf(stderr, "to few bytes sent ... aborting!"); - return -1; - } - - shutdown(c, SHUT_WR); - - int len = 0; - for(;;) { - int nbread = recv(c, &(buf[len]), sizeof(buf) - len, 0); - if(nbread <= 0) { - if(!nbread) { - fprintf(stderr, "connection closed\n"); - } else if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) { - perror("recv()"); - } else - continue; + char buf[1234567]; + unsigned int i; + for(i = 0; i<sizeof(buf); ++i) { + buf[i] = 'A' + i%62; + } + buf[sizeof(buf)-1] = '\n'; - return -1; + int wtot = 0; + for(;;) { + int nbwritten = send(c, &(buf[wtot]), sizeof(buf) - wtot, 0); + if(nbwritten <= 0) { + if(nbwritten < 0) + perror("send()"); + else + fprintf(stderr, "nothing sent... aborting\n"); + return -1; + } + wtot += nbwritten; + printf("%d bytes sent, total=%d\n", nbwritten, wtot); + if(wtot >= sizeof(buf)) { + break; + } } - len += nbread; - printf("%d bytes received, total = %d, left = %d", nbread, len, len - nbwritten); - if(len == nbwritten) { - printf(" .. finished\n"); - for(i = 0; i<nbwritten-1; ++i) { - if(buf[i] != ('A' + i%62)) { - fprintf(stderr, "error at index %d (expected 0x%02X, got 0x%02X)\n", i, 'A' + i%62, (unsigned int)(buf[i])); + shutdown(c, SHUT_WR); + + int rtot = 0; + for(;;) { + int nbread = recv(c, &(buf[rtot]), sizeof(buf) - rtot, 0); + if(nbread <= 0) { + if(!nbread) { + fprintf(stderr, "connection closed\n"); + } else if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) { + perror("recv()"); + } else + continue; + + return -1; + } + + rtot += nbread; + printf("%d bytes received, total = %d, left = %d", nbread, rtot, rtot - wtot); + if(rtot == wtot) { + printf(" .. finished\n"); + for(i = 0; i<wtot-1; ++i) { + if(buf[i] != ('A' + i%62)) { + fprintf(stderr, "error at index %d (expected 0x%02X, got 0x%02X)\n", i, 'A' + i%62, (unsigned int)(buf[i])); + return -1; + } + } + if(buf[wtot-1] != '\n') { + fprintf(stderr, "string doesn't end with new-line\n"); return -1; } - } - if(buf[nbwritten-1] != '\n') { - fprintf(stderr, "string doesn't end with new-line\n"); + break; + } else if(rtot > wtot) { + fprintf(stderr, "got too many bytes back???\n"); return -1; } - break; - } else if(len > nbwritten) { - fprintf(stderr, "got too many bytes back???\n"); - return -1; + + printf("\n"); } - printf("\n"); + while(recv(c, buf, 1, 0) > 0); + break; } +/*************** MODE A ***************/ + case 'B': { + printf("MODE: B\n"); + + char request[] = "hello world\n"; + int nbwritten = send(c, request, sizeof(request), 0); + if(nbwritten <= 0) { + if(nbwritten < 0) + perror("send()"); + else + fprintf(stderr, "nothing sent... aborting\n"); + return -1; + } + printf("%d bytes sent\n", nbwritten); + if(nbwritten != sizeof(request)) { + fprintf(stderr, "to few bytes sent ... aborting!"); + return -1; + } - while(recv(c, buf, 1, 0) > 0); + char buf[1000]; + int rtot = 0; + for(;;) { + int nbread = recv(c, buf, sizeof(buf), 0); + if(nbread <= 0) { + if(!nbread) { + fprintf(stderr, "connection closed\n"); + return 0; + } else if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) { + perror("recv()"); + return -1; + } + continue; + } + + rtot += nbread; + printf("%d bytes received, total = %d", nbread, rtot); + if(rtot >= 1234567) { + printf(" .. finished\n"); + return 0; + } + printf("\n"); + } + break; + } +/**************************************/ + } return 0; } |