1 | #define _GNU_SOURCE |
---|
2 | |
---|
3 | #include "socket.hfa" |
---|
4 | |
---|
5 | #include <errno.h> |
---|
6 | #include <string.h> |
---|
7 | extern "C" { |
---|
8 | #include <sys/socket.h> |
---|
9 | #include <netinet/in.h> |
---|
10 | } |
---|
11 | |
---|
12 | #include <fstream.hfa> |
---|
13 | #include <time.hfa> |
---|
14 | #include <thread.hfa> |
---|
15 | |
---|
16 | #include "options.hfa" |
---|
17 | |
---|
18 | int prepaddr(struct sockaddr_in & address) { |
---|
19 | int addrlen = sizeof(address); |
---|
20 | memset( (char *)&address, '\0', addrlen ); |
---|
21 | address.sin_family = AF_INET; |
---|
22 | address.sin_addr.s_addr = htonl(INADDR_ANY); |
---|
23 | address.sin_port = htons( options.socket.port ); |
---|
24 | return addrlen; |
---|
25 | } |
---|
26 | |
---|
27 | int listener(struct sockaddr_in & address, int addrlen) { |
---|
28 | int type = SOCK_STREAM; |
---|
29 | // if(options.socket.reuseport) type |= SOCK_NONBLOCK; |
---|
30 | int sockfd = socket(AF_INET, type, 0); |
---|
31 | if(sockfd < 0) { |
---|
32 | abort( "socket error: (%d) %s\n", (int)errno, strerror(errno) ); |
---|
33 | } |
---|
34 | |
---|
35 | if(options.socket.reuseport) { |
---|
36 | int value = 1; |
---|
37 | // if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (const void*)&on, sizeof(on))) |
---|
38 | // abort( "setsockopt SO_REUSEADDR error: (%d) %s\n", (int)errno, strerror(errno) ); |
---|
39 | if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEPORT, &value, sizeof(int)) < 0) |
---|
40 | abort( "setsockopt SO_REUSEPORT error: (%d) %s\n", (int)errno, strerror(errno) ); |
---|
41 | } |
---|
42 | |
---|
43 | int ret = 0; |
---|
44 | int waited = 0; |
---|
45 | for() { |
---|
46 | __CONST_SOCKADDR_ARG addr; |
---|
47 | addr.__sockaddr__ = (struct sockaddr *)&address; |
---|
48 | socklen_t addrlen = sizeof(address); |
---|
49 | ret = bind( sockfd, addr, addrlen ); |
---|
50 | if(ret < 0) { |
---|
51 | if(errno == EADDRINUSE) { |
---|
52 | if(waited == 0) { |
---|
53 | if(!options.interactive) abort | "Port already in use in non-interactive mode. Aborting"; |
---|
54 | sout | "Waiting for port"; |
---|
55 | } else { |
---|
56 | sout | "\r" | waited | nonl; |
---|
57 | flush( sout ); |
---|
58 | } |
---|
59 | waited ++; |
---|
60 | sleep( 1`s ); |
---|
61 | continue; |
---|
62 | } |
---|
63 | abort( "bind error: (%d) %s\n", (int)errno, strerror(errno) ); |
---|
64 | } |
---|
65 | break; |
---|
66 | } |
---|
67 | |
---|
68 | ret = listen( sockfd, options.socket.backlog ); |
---|
69 | if(ret < 0) { |
---|
70 | abort( "listen error: (%d) %s\n", (int)errno, strerror(errno) ); |
---|
71 | } |
---|
72 | |
---|
73 | return sockfd; |
---|
74 | } |
---|