1 #include "toys.h"
2
xsocket(int domain,int type,int protocol)3 int xsocket(int domain, int type, int protocol)
4 {
5 int fd = socket(domain, type, protocol);
6
7 if (fd < 0) perror_exit("socket %x %x", type, protocol);
8 return fd;
9 }
10
xsetsockopt(int fd,int level,int opt,void * val,socklen_t len)11 void xsetsockopt(int fd, int level, int opt, void *val, socklen_t len)
12 {
13 if (-1 == setsockopt(fd, level, opt, val, len)) perror_exit("setsockopt");
14 }
15
xconnect(char * host,char * port,int family,int socktype,int protocol,int flags)16 int xconnect(char *host, char *port, int family, int socktype, int protocol,
17 int flags)
18 {
19 struct addrinfo info, *ai, *ai2;
20 int fd;
21
22 memset(&info, 0, sizeof(struct addrinfo));
23 info.ai_family = family;
24 info.ai_socktype = socktype;
25 info.ai_protocol = protocol;
26 info.ai_flags = flags;
27
28 fd = getaddrinfo(host, port, &info, &ai);
29 if (fd || !ai)
30 error_exit("Connect '%s%s%s': %s", host, port ? ":" : "", port ? port : "",
31 fd ? gai_strerror(fd) : "not found");
32
33 // Try all the returned addresses. Report errors if last entry can't connect.
34 for (ai2 = ai; ai; ai = ai->ai_next) {
35 fd = (ai->ai_next ? socket : xsocket)(ai->ai_family, ai->ai_socktype,
36 ai->ai_protocol);
37 if (!connect(fd, ai->ai_addr, ai->ai_addrlen)) break;
38 else if (!ai2->ai_next) perror_exit("connect");
39 close(fd);
40 }
41 freeaddrinfo(ai2);
42
43 return fd;
44 }
45
xpoll(struct pollfd * fds,int nfds,int timeout)46 int xpoll(struct pollfd *fds, int nfds, int timeout)
47 {
48 int i;
49
50 for (;;) {
51 if (0>(i = poll(fds, nfds, timeout))) {
52 if (toys.signal) return i;
53 if (errno != EINTR && errno != ENOMEM) perror_exit("xpoll");
54 else if (timeout>0) timeout--;
55 } else return i;
56 }
57 }
58