Date: Sun, 28 Jun 2009 01:25:18 GMT From: Fang Wang <fangwang@FreeBSD.org> To: Perforce Change Reviews <perforce@FreeBSD.org> Subject: PERFORCE change 165334 for review Message-ID: <200906280125.n5S1PI4E026853@repoman.freebsd.org>
next in thread | raw e-mail | index | archive | help
http://perforce.freebsd.org/chv.cgi?CH=165334 Change 165334 by fangwang@fangwang_utobsd on 2009/06/28 01:25:08 Impelement tcputo regression test code, not finish yet. Affected files ... .. //depot/projects/soc2009/tcputo/src/tools/regression/netinet/tcputo/Makefile#1 add .. //depot/projects/soc2009/tcputo/src/tools/regression/netinet/tcputo/tcputo.c#2 edit Differences ... ==== //depot/projects/soc2009/tcputo/src/tools/regression/netinet/tcputo/tcputo.c#2 (text+ko) ==== @@ -1,0 +1,179 @@ +#include <sys/types.h> +#include <sys/socket.h> + +#include <netinet/in.h> +#include <netinet/tcp.h> + +#include <arpa/inet.h> + +#include <errno.h> +#include <fcntl.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> +#include <time.h> + +static void +usage(void) +{ + + fprintf(stderr, "tcpconnect server port\n"); + fprintf(stderr, "tcpconnect client ip port\n"); + exit(-1); +} + +static int +tcputo_timer(void) +{ + static time_t start_time = 0; + time_t end_time; + time_t interval; + + if (start_time == 0) { + time(&start_time); + interval = 0; + } + else { + time(&end_time); + interval = end_time - start_time; + start_time = 0; + } + + return (int)(interval); +} + +static void +tcputo_server(int argc, char *argv[]) +{ + int listen_sock, accept_sock; + struct sockaddr_in sin; + char *dummy; + char buf[16*1024]; + long port; + int user_timeout; + int optval = 4*1024; + + if (argc != 1) + usage(); + + bzero(&sin, sizeof(sin)); + sin.sin_len = sizeof(sin); + sin.sin_family = AF_INET; + sin.sin_addr.s_addr = htonl(INADDR_ANY); + + port = strtoul(argv[0], &dummy, 10); + if (port < 1 || port > 65535 || *dummy != '\0') + usage(); + sin.sin_port = htons(port); + + listen_sock = socket(PF_INET, SOCK_STREAM, 0); + if (listen_sock == -1) + err(-1, "socket"); + + if (bind(listen_sock, (struct sockaddr *)&sin, sizeof(sin)) == -1) + err(-1, "bind"); + + if (listen(listen_sock, -1) == -1) + err(-1, "listen"); + + accept_sock = accept(listen_sock, NULL, NULL); + close(listen_sock); + if (setsockopt(accept_sock, SOL_SOCKET, SO_SNDBUF, &optval, sizeof(optval)) == -1) + err(-1, "setsockopt"); + while(1) { + sleep(1); + printf("server again %d\n", optval++); + (void)recv(accept_sock, buf, 16*1024, MSG_DONTWAIT); + (void)tcputo_timer(); + if (send(accept_sock, buf, 16*1024, MSG_NOSIGNAL) >= 0) { + (void)tcputo_timer(); + continue; + } + puts(strerror(errno)); + user_timeout = tcputo_timer(); + printf("Connection timeout, %d seconds.\n", user_timeout); + break; + } + + close(accept_sock); +} + +static void +tcputo_client(int argc, char *argv[]) +{ + struct sockaddr_in sin; + long count, i, port; + char *dummy; + char buf[16*1024]; + int sock; + int nonblock = 0, md5enable = 0; + int user_timeout; + int optval = 4*1024; + + if (argc != 2) + usage(); + + bzero(&sin, sizeof(sin)); + sin.sin_len = sizeof(sin); + sin.sin_family = AF_INET; + if (inet_aton(argv[0], &sin.sin_addr) == 0) + err(-1, "convert address"); + + port = strtoul(argv[1], &dummy, 10); + if (port < 1 || port > 65535 || *dummy != '\0') + usage(); + sin.sin_port = htons(port); + + sock = socket(PF_INET, SOCK_STREAM, 0); + if (sock == -1) + err(-1, "socket"); + if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &optval, sizeof(optval)) == -1) + err(-1, "setsockopt"); + + /* No warning in default case on ENOPROTOOPT. */ + /* if (setsockopt(sock, IPPROTO_TCP, TCP_MD5SIG, + &md5enable, sizeof(md5enable)) != 0) { + if (errno == ENOPROTOOPT && md5enable > 0) + err(-1, "setsockopt(TCP_MD5SIG)"); + else if (errno != ENOPROTOOPT) + warn("setsockopt(TCP_MD5SIG)"); + } */ + + if (connect(sock, (struct sockaddr *)&sin, sizeof(sin)) == -1) + err(-1, "connect"); + + while (1) { + sleep(1); + printf("client again %d\n", optval++); + (void)recv(sock, buf, 16*1024, MSG_DONTWAIT); + (void)tcputo_timer(); + if (send(sock, buf, 16*1024, MSG_NOSIGNAL) > 0) { + (void)tcputo_timer(); + continue; + } + puts(strerror(errno)); + user_timeout = tcputo_timer(); + printf("Connection timeout, %d seconds.\n", user_timeout); + break; + } + + close(sock); +} + +int +main(int argc, char *argv[]) +{ + + if (argc < 2) + usage(); + + if (strcmp(argv[1], "server") == 0) + tcputo_server(argc - 2, argv + 2); + else if (strcmp(argv[1], "client") == 0) + tcputo_client(argc - 2, argv + 2); + else + usage(); + + exit(0); +}
Want to link to this message? Use this URL: <https://mail-archive.FreeBSD.org/cgi/mid.cgi?200906280125.n5S1PI4E026853>