1 #include "toys.h" 2 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 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 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 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
