1 #include <ccan/net/net.h>
2 #include <ccan/net/net.c>
3 #include <ccan/tap/tap.h>
9 static unsigned int server(int protocol, int type)
14 struct sockaddr_in ipv4;
15 struct sockaddr_in6 ipv6;
17 socklen_t addlen = sizeof(addr);
19 sock = socket(protocol, type, 0);
21 /* Bind to free port. */
24 /* Figure out what port it gave us. */
25 getsockname(sock, &addr.addr, &addlen);
32 fd = accept(sock, NULL, 0);
34 err(1, "Accepting from socket %i", sock);
36 ret = write(fd, "Yay!", strlen("Yay!"));
37 if (ret != strlen("Yay!"))
38 err(1, "Write returned %i", ret);
42 return ntohs(protocol == AF_INET
43 ? addr.ipv4.sin_port : addr.ipv6.sin6_port);
46 static bool we_faked_double = false;
48 /* Get a localhost on ipv4 and IPv6. Fake it if we have to. */
49 static struct addrinfo* double_addr_lookup(char* buf)
51 struct addrinfo *addr, *addr2;
53 addr = net_client_lookup("localhost", buf, AF_UNSPEC, SOCK_STREAM);
57 /* If we only got one, we need to fake up the other one. */
59 addr2 = addr->ai_next;
61 we_faked_double = true;
64 if (addr->ai_family == AF_INET) {
65 /* These are the names I found on my Ubuntu system. */
66 addr2 = net_client_lookup("ip6-localhost", buf,
67 AF_UNSPEC, SOCK_STREAM);
69 addr2 = net_client_lookup("localhost6", buf,
73 addr2 = net_client_lookup("::1", buf,
76 } else if (addr->ai_family == AF_INET6)
77 /* IPv6 only? This is a guess... */
78 addr2 = net_client_lookup("ip4-localhost", buf,
79 AF_UNSPEC, SOCK_STREAM);
82 addr->ai_next = addr2;
88 /* One IPv4 and one IPv6? */
89 if (addr->ai_family == AF_INET && addr2->ai_family == AF_INET6)
91 if (addr->ai_family == AF_INET6 && addr2->ai_family == AF_INET)
96 static void double_addr_free(struct addrinfo* addr)
98 struct addrinfo *addr2;
99 if (we_faked_double) {
100 addr2 = addr->ai_next;
101 addr->ai_next = NULL;
110 struct addrinfo *addr;
112 struct sockaddr saddr;
118 port = server(AF_INET, SOCK_STREAM);
119 sprintf(buf, "%u", port);
121 addr = double_addr_lookup(buf);
123 fd = net_connect(addr);
126 slen = sizeof(saddr);
127 ok1(getsockname(fd, &saddr, &slen) == 0);
128 diag("family = %d", saddr.sa_family);
129 ok1(saddr.sa_family == AF_INET);
130 status = read(fd, buf, sizeof(buf));
131 ok(status == strlen("Yay!"),
132 "Read returned %i (%s)", status, strerror(errno));
133 ok1(strncmp(buf, "Yay!", strlen("Yay!")) == 0);
135 double_addr_free(addr);
137 port = server(AF_INET6, SOCK_STREAM);
138 sprintf(buf, "%u", port);
140 addr = double_addr_lookup(buf);
142 fd = net_connect(addr);
145 slen = sizeof(saddr);
146 ok1(getsockname(fd, &saddr, &slen) == 0);
147 ok1(saddr.sa_family == AF_INET6);
148 status = read(fd, buf, sizeof(buf));
149 ok(status == strlen("Yay!"),
150 "Read returned %i (%s)", status, strerror(errno));
151 ok1(strncmp(buf, "Yay!", strlen("Yay!")) == 0);
153 double_addr_free(addr);
156 ok1(WIFEXITED(status));
157 ok1(WEXITSTATUS(status) == 0);
159 /* This exits depending on whether all tests passed */
160 return exit_status();