1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
| #include <cstdio>
#include <cstring>
#include <string>
#include <vector>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/errno.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string>
using namespace std;
const int BUF_SIZE = 1024;
string addr_to_string(const sockaddr_in *addr) {
char addr_str[INET_ADDRSTRLEN];
const char *addr_str_ptr = inet_ntop(AF_INET, &addr->sin_addr, addr_str, sizeof(addr_str));
if (addr_str_ptr == NULL) {
fprintf(stderr, "inet_ntop failed\n");
return "";
}
return string(addr_str_ptr) + ":" + to_string(ntohs(addr->sin_port));
}
string get_sockname_string(int sockfd) {
sockaddr_in addr;
socklen_t len = sizeof(addr);
if (getsockname(sockfd, (sockaddr *) &addr, &len) == -1) {
fprintf(stderr, "getsockname failed\n");
return "";
}
return addr_to_string(&addr);
}
string get_peername_string(int sockfd) {
sockaddr_in addr;
socklen_t len = sizeof(addr);
if (getpeername(sockfd, (sockaddr *) &addr, &len) == -1) {
fprintf(stderr, "getpeername failed, errno: %d, str: %s\n", errno, strerror(errno));
return "";
}
return addr_to_string(&addr);
}
int init_server(int type, const sockaddr *addr, socklen_t alen, int qlen) {
int fd;
if ((fd = socket(addr->sa_family, type, 0)) == -1) {
fprintf(stderr, "create socket failed, errno: %d, str: %s\n", errno, strerror(errno));
return -1;
}
printf("create socket fd: %d, sock_name: %s, peer_name: %s\n", fd, get_sockname_string(fd).c_str(), get_peername_string(fd).c_str());
if (bind(fd, addr, alen) == -1) {
fprintf(stderr, "bind failed, errno: %d, str: %s\n", errno, strerror(errno));
close(fd);
return -1;
}
printf("bind socket fd: %d, sock_name: %s, peer_name: %s\n", fd, get_sockname_string(fd).c_str(), get_peername_string(fd).c_str());
if (type == SOCK_STREAM || type == SOCK_SEQPACKET) {
if (listen(fd, qlen) == -1) {
fprintf(stderr, "listen failed, errno: %d, str: %s\n", errno, strerror(errno));
close(fd);
return -1;
}
printf("listen fd: %d, sock_name: %s, peer_name: %s\n", fd, get_sockname_string(fd).c_str(), get_peername_string(fd).c_str());
}
return fd;
}
int main(int argc, char **argv) {
int sleep_before_recv = 0;
int sleep_after_listen = 0;
int qlen = 0;
int port = 27015;
char *ip = "0.0.0.0";
for (int i = 1; i < argc;) {
if (argv[i] == string("--port")) {
port = atoi(argv[i + 1]);
i += 2;
continue;
}
if (argv[i] == string("--qlen")) {
qlen = atoi(argv[i + 1]);
i += 2;
continue;
}
if (argv[i] == string("--sleep-before-recv")) {
sleep_before_recv = atoi(argv[i + 1]);
i += 2;
continue;
}
if (argv[i] == string("--sleep-after-listen")) {
sleep_after_listen = atoi(argv[i + 1]);
i += 2;
continue;
}
if (argv[i] == string("--ip")) {
ip = argv[i + 1];
i += 2;
continue;
}
i++;
}
printf("port: %d, qlen: %d, sleep-before-recv: %d, sleep-after-listen: %d\n", port, qlen, sleep_before_recv, sleep_after_listen);
sockaddr_in addr;
bzero(&addr, sizeof(addr));
addr.sin_family = AF_INET;
inet_pton(addr.sin_family, ip, &addr.sin_addr);
addr.sin_port = htons(port);
int sockfd = init_server(SOCK_STREAM, (sockaddr *) &addr, sizeof(addr), qlen);
if (sockfd == -1) return -1;
sleep(sleep_after_listen);
while (true) {
int clfd, n;
sockaddr_in accepted_addr;
socklen_t len = sizeof(accepted_addr);
if ((clfd = accept(sockfd, (sockaddr *) &accepted_addr, &len)) == -1) {
fprintf(stderr, "accept failed, errno: %d, str: %s\n", errno, strerror(errno));
continue;
}
printf("after accept sockfd: %d, sock_name: %s, peer_name: %s\n",
sockfd, get_sockname_string(sockfd).c_str(), get_peername_string(sockfd).c_str());
printf("accepted fd: %d, sock_name: %s, peer_name: %s, addr: %s\n",
clfd, get_sockname_string(clfd).c_str(), get_peername_string(clfd).c_str(), addr_to_string(&accepted_addr).c_str());
sleep(sleep_before_recv);
char buf[BUF_SIZE];
if ((n = recv(clfd, buf, BUF_SIZE - 1, 0)) == -1) {
fprintf(stderr, "recv failed, errno: %d, str: %s\n", errno, strerror(errno));
close(clfd);
continue;
}
buf[n] = '\0';
printf("read %d bytes from fd: %d, buf: %s\n", n, clfd, buf);
close(clfd);
}
return 0;
}
|