fix bug when setting default port for iscsi
[tridge/dbench.git] / tbench_srv.c
1 /* 
2    dbench version 2
3    Copyright (C) Andrew Tridgell 1999
4    
5    This program is free software; you can redistribute it and/or modify
6    it under the terms of the GNU General Public License as published by
7    the Free Software Foundation; either version 3 of the License, or
8    (at your option) any later version.
9    
10    This program is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13    GNU General Public License for more details.
14    
15    You should have received a copy of the GNU General Public License
16    along with this program; if not, see <http://www.gnu.org/licenses/>.
17 */
18
19 #include "dbench.h"
20
21 struct options options = {
22         .tcp_options = TCP_OPTIONS
23 };
24
25 static void server(int fd)
26 {
27         char buf[70000];
28         unsigned *ibuf = (unsigned *)buf;
29         uint32_t n;
30
31         signal(SIGPIPE, SIG_IGN);
32         
33         printf("^"); fflush(stdout);
34
35         while (1) {
36                 if (read_sock(fd, buf, 4) != 4) break;
37                 n = ntohl(ibuf[0]);
38                 if (n+4 >= sizeof(buf)) {
39                         printf("overflow in server!\n");
40                         exit(1);
41                 }
42                 if (read_sock(fd, buf+4, n) != (int)n) break;
43                 n = ntohl(ibuf[1]);
44                 ibuf[0] = htonl(n);
45                 if (write_sock(fd, buf, n+4) != (int)(n+4)) break;
46         }
47
48         exit(0);
49 }
50
51 static void listener(void)
52 {
53         int sock;
54
55         sock = open_socket_in(SOCK_STREAM, TCP_PORT);
56
57         if (listen(sock, 20) == -1) {
58                 fprintf(stderr,"listen failed\n");
59                 exit(1);
60         }
61
62         printf("waiting for connections\n");
63
64         signal(SIGCHLD, SIG_IGN);
65
66         while (1) {
67                 struct sockaddr addr;
68                 socklen_t in_addrlen = sizeof(addr);
69                 int fd;
70
71                 while (waitpid((pid_t)-1,(int *)NULL, WNOHANG) > 0) ;
72
73                 fd = accept(sock, &addr, &in_addrlen);
74
75                 if (fd != -1) {
76                         if (fork() == 0) server(fd);
77                         close(fd);
78                 }
79         }
80 }
81
82
83 static void usage(void)
84 {
85         printf("usage: tbench_srv [OPTIONS]\n"
86                "options:\n"
87                "  -t options       set socket options\n");
88         exit(1);
89 }
90
91
92 static void process_opts(int argc, char **argv)
93 {
94         int c;
95
96         while ((c = getopt(argc, argv, "t:")) != -1) {
97                 switch (c) {
98                 case 't':
99                         options.tcp_options = optarg;
100                         break;
101                 default:
102                         usage();
103                 }
104         }
105 }
106
107
108  int main(int argc, char *argv[])
109 {
110         process_opts(argc, argv);
111
112         listener();
113         return 0;
114 }