From 2aa47107451e62c395bb4b48bffc27e8d00ef8e8 Mon Sep 17 00:00:00 2001 From: Yang Tse Date: Mon, 17 Jul 2006 03:38:13 +0000 Subject: [PATCH] Minor cleanup --- tests/server/sockfilt.c | 83 +++++++++++++++++++++++------------------ 1 file changed, 47 insertions(+), 36 deletions(-) diff --git a/tests/server/sockfilt.c b/tests/server/sockfilt.c index 4b9fec4ef..415bb3576 100644 --- a/tests/server/sockfilt.c +++ b/tests/server/sockfilt.c @@ -168,16 +168,15 @@ static int juggle(curl_socket_t *sockfdp, fd_set fds_read; fd_set fds_write; fd_set fds_err; + curl_socket_t sockfd; curl_socket_t maxfd; ssize_t rc; - ssize_t len; - ssize_t nread; + ssize_t nread_stdin; + ssize_t nread_socket; ssize_t bytes_written; - ssize_t bytes_read; - ssize_t port_strlen; + ssize_t buffer_len; unsigned char buffer[256]; /* FIX: bigger buffer */ char data[256]; - curl_socket_t sockfd; timeout.tv_sec = 120; timeout.tv_usec = 0; @@ -189,7 +188,9 @@ static int juggle(curl_socket_t *sockfdp, FD_SET(fileno(stdin), &fds_read); switch(*mode) { + case PASSIVE_LISTEN: + /* server mode */ sockfd = listenfd; /* there's always a socket to wait for */ @@ -198,8 +199,9 @@ static int juggle(curl_socket_t *sockfdp, break; case PASSIVE_CONNECT: + sockfd = *sockfdp; - if(-1 == sockfd) { + if(CURL_SOCKET_BAD == sockfd) { /* eeek, we are supposedly connected and then this cannot be -1 ! */ logmsg("socket is -1! on %s:%d", __FILE__, __LINE__); maxfd = 0; /* stdin */ @@ -212,10 +214,10 @@ static int juggle(curl_socket_t *sockfdp, break; case ACTIVE: - sockfd = *sockfdp; + sockfd = *sockfdp; /* sockfd turns CURL_SOCKET_BAD when our connection has been closed */ - if(sockfd != CURL_SOCKET_BAD) { + if(CURL_SOCKET_BAD != sockfd) { FD_SET(sockfd, &fds_read); maxfd = sockfd; } @@ -226,11 +228,13 @@ static int juggle(curl_socket_t *sockfdp, break; case ACTIVE_DISCONNECT: + logmsg("disconnected, no socket to read on"); maxfd = 0; sockfd = CURL_SOCKET_BAD; break; - } + + } /* switch(*mode) */ do { rc = select(maxfd + 1, &fds_read, &fds_write, &fds_err, &timeout); @@ -260,8 +264,8 @@ static int juggle(curl_socket_t *sockfdp, DATA - plain pass-thru data */ - nread = read(fileno(stdin), buffer, 5); - if(5 == nread) { + nread_stdin = read(fileno(stdin), buffer, 5); + if(5 == nread_stdin) { logmsg("Received %c%c%c%c (on stdin)", buffer[0], buffer[1], buffer[2], buffer[3] ); @@ -272,13 +276,13 @@ static int juggle(curl_socket_t *sockfdp, } else if(!memcmp("PORT", buffer, 4)) { - /* question asking us what PORT number we are listening to. - Replies with PORT with "IPv[num]/[port]" */ - sprintf((char *)buffer, "IPv%d/%d\n", use_ipv6?6:4, port); - port_strlen = (ssize_t)strlen((char *)buffer); - sprintf(data, "PORT\n%04x\n", port_strlen); + /* Question asking us what PORT number we are listening to. + Replies to PORT with "IPv[num]/[port]" */ + sprintf((char *)buffer, "IPv%d/%d\n", use_ipv6?6:4, (int)port); + buffer_len = (ssize_t)strlen((char *)buffer); + sprintf(data, "PORT\n%04x\n", buffer_len); write(fileno(stdout), data, 10); - write(fileno(stdout), buffer, port_strlen); + write(fileno(stdout), buffer, buffer_len); } else if(!memcmp("QUIT", buffer, 4)) { /* just die */ @@ -290,14 +294,20 @@ static int juggle(curl_socket_t *sockfdp, if(5 != read(fileno(stdin), buffer, 5)) return FALSE; - buffer[4] = '\0'; + buffer[5] = '\0'; - len = (ssize_t)strtol((char *)buffer, NULL, 16); - if(len != read(fileno(stdin), buffer, len)) + buffer_len = (ssize_t)strtol((char *)buffer, NULL, 16); + if (buffer_len > (ssize_t)sizeof(buffer)) { + logmsg("Buffer size %d too small for data size %d", + (int)sizeof(buffer), buffer_len); + return FALSE; + } + nread_stdin = read(fileno(stdin), buffer, buffer_len); + if(nread_stdin != buffer_len) return FALSE; - logmsg("> %d bytes data, server => client", len); - lograw(buffer, len); + logmsg("> %d bytes data, server => client", buffer_len); + lograw(buffer, buffer_len); if(*mode == PASSIVE_LISTEN) { logmsg("*** We are disconnected!"); @@ -305,10 +315,10 @@ static int juggle(curl_socket_t *sockfdp, } else { /* send away on the socket */ - bytes_written = swrite(sockfd, buffer, len); - if(bytes_written != len) { + bytes_written = swrite(sockfd, buffer, buffer_len); + if(bytes_written != buffer_len) { logmsg("Not all data was sent. Bytes to send: %d sent: %d", - len, bytes_written); + buffer_len, bytes_written); } } } @@ -329,19 +339,20 @@ static int juggle(curl_socket_t *sockfdp, return TRUE; } } - else if(nread == -1){ - logmsg("read %d from stdin, exiting", nread); + else if(-1 == nread_stdin) { + logmsg("read %d from stdin, exiting", nread_stdin); return FALSE; } } + if((sockfd != CURL_SOCKET_BAD) && (FD_ISSET(sockfd, &fds_read)) ) { if(*mode == PASSIVE_LISTEN) { /* there's no stream set up yet, this is an indication that there's a client connecting. */ sockfd = accept(sockfd, NULL, NULL); - if(-1 == sockfd) + if(CURL_SOCKET_BAD == sockfd) logmsg("accept() failed\n"); else { logmsg("====> Client connect"); @@ -353,9 +364,9 @@ static int juggle(curl_socket_t *sockfdp, } /* read from socket, pass on data to stdout */ - bytes_read = sread(sockfd, buffer, sizeof(buffer)); + nread_socket = sread(sockfd, buffer, sizeof(buffer)); - if(bytes_read <= 0) { + if(nread_socket <= 0) { logmsg("====> Client disconnect"); write(fileno(stdout), "DISC\n", 5); sclose(sockfd); @@ -367,12 +378,12 @@ static int juggle(curl_socket_t *sockfdp, return TRUE; } - sprintf(data, "DATA\n%04x\n", bytes_read); + sprintf(data, "DATA\n%04x\n", nread_socket); write(fileno(stdout), data, 10); - write(fileno(stdout), buffer, bytes_read); + write(fileno(stdout), buffer, nread_socket); - logmsg("< %d bytes data, client => server", bytes_read); - lograw(buffer, bytes_read); + logmsg("< %d bytes data, client => server", nread_socket); + lograw(buffer, nread_socket); } return TRUE; @@ -390,8 +401,8 @@ static curl_socket_t sockdaemon(curl_socket_t sock, int rc; if (setsockopt - (sock, SOL_SOCKET, SO_REUSEADDR, (const void *) &flag, - sizeof(int)) < 0) { + (sock, SOL_SOCKET, SO_REUSEADDR, (void *)&flag, + sizeof(flag)) < 0) { perror("setsockopt(SO_REUSEADDR)"); }