1 /* SPDX-License-Identifier: MIT */
2 #include <fcntl.h>
3 #include <stdint.h>
4 #include <netinet/in.h>
5 #include <arpa/inet.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <sys/socket.h>
10 #include <sys/time.h>
11 #include <unistd.h>
12
13 #include "helpers.h"
14
setup_listening_socket(int port,int ipv6)15 int setup_listening_socket(int port, int ipv6)
16 {
17 struct sockaddr_in srv_addr = { };
18 struct sockaddr_in6 srv_addr6 = { };
19 int fd, enable, ret, domain;
20
21 if (ipv6)
22 domain = AF_INET6;
23 else
24 domain = AF_INET;
25
26 fd = socket(domain, SOCK_STREAM, 0);
27 if (fd == -1) {
28 perror("socket()");
29 return -1;
30 }
31
32 enable = 1;
33 ret = setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &enable, sizeof(int));
34 if (ret < 0) {
35 perror("setsockopt(SO_REUSEADDR)");
36 return -1;
37 }
38
39 if (ipv6) {
40 srv_addr6.sin6_family = AF_INET6;
41 srv_addr6.sin6_port = htons(port);
42 srv_addr6.sin6_addr = in6addr_any;
43 ret = bind(fd, (const struct sockaddr *)&srv_addr6, sizeof(srv_addr6));
44 } else {
45 srv_addr.sin_family = AF_INET;
46 srv_addr.sin_port = htons(port);
47 srv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
48 ret = bind(fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
49 }
50
51 if (ret < 0) {
52 perror("bind()");
53 return -1;
54 }
55
56 if (listen(fd, 1024) < 0) {
57 perror("listen()");
58 return -1;
59 }
60
61 return fd;
62 }
63