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
|
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include "fdleak.h"
void server ()
{
int s, x;
struct sockaddr_in baddr;
struct sockaddr_in addr;
socklen_t baddrsize = sizeof(baddr);
int one = 1;
s = DO( socket(PF_INET, SOCK_STREAM, 0) );
(void) DO( setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(int)) );
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = inet_addr("127.0.0.1");
addr.sin_port = htons(12321);
(void) DO( bind(s, (struct sockaddr *)&addr, sizeof(addr)) );
(void) DO( listen(s, 5) );
memset(&baddr, 0, sizeof(baddr));
x = DO( accept(s, (struct sockaddr *)&baddr, &baddrsize) );
(void) DO( write(x, "hello", 6) );
}
void client ()
{
int s, count = 0, ret;
struct sockaddr_in addr;
char buf[1024];
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = inet_addr("127.0.0.1");
addr.sin_port = htons(12321);
do {
count++;
s = DO( socket(PF_INET, SOCK_STREAM, 0) );
ret = connect(s, (struct sockaddr *)&addr, sizeof(addr));
if (ret == -1) {
// If the connect() failed, we close the socket and reopen it before
// trying again. This isn't necessary on Linux, but it is on Darwin.
(void) DO( close(s) );
sleep(1);
}
} while (count < 10 && ret == -1);
if (ret == -1) {
perror("connect");
exit(1);
}
(void) DO( read(s, buf, sizeof(buf)) );
printf("%s\n", buf);
}
int main (int argc, char **argv)
{
int pid, status;
CLOSE_INHERITED_FDS;
if ((pid = fork()) == 0) {
server();
return 0;
}
client();
wait(&status);
return 0;
}
|
