From 3705dc02cc18b79c77d8f5aba934e02c7a91467f Mon Sep 17 00:00:00 2001 From: Benjamin Franzke Date: Sat, 9 Mar 2013 06:30:21 +0100 Subject: Add util mostly for socket-based tasks --- util.c | 212 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ util.h | 28 +++++++++ 2 files changed, 240 insertions(+) create mode 100644 util.c create mode 100644 util.h diff --git a/util.c b/util.c new file mode 100644 index 0000000..be62ac1 --- /dev/null +++ b/util.c @@ -0,0 +1,212 @@ +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include + +#include "util.h" + +uint32_t +sock_get_ipv4_addr(int fd) +{ + struct sockaddr_storage storage; + struct sockaddr_in *own = (struct sockaddr_in *) &storage; + socklen_t addrlen = sizeof(struct sockaddr_storage); + + assert(getsockname(fd, (struct sockaddr *) &storage, &addrlen) == 0); + + return own->sin_addr.s_addr; +} + +uint32_t +sock_get_netmask(int fd) +{ + struct ifreq ifreq; + + memset(&ifreq, 0, sizeof ifreq); + + /* FIXME: do not hardcode interface name */ + strcpy(ifreq.ifr_name, "wlp3s0"); + + if (ioctl(fd, SIOCGIFNETMASK, &ifreq) < 0) { + fprintf(stderr, "get netmask failed: %s\n", strerror(errno)); + return 0; + } + + return ((struct sockaddr_in *) &ifreq.ifr_netmask)->sin_addr.s_addr; +} + +uint32_t +sock_get_peer_ipv4_addr(int fd) +{ + struct sockaddr_storage storage; + struct sockaddr_in *own = (struct sockaddr_in *) &storage; + socklen_t addrlen = sizeof(struct sockaddr_storage); + + assert(getpeername(fd, (struct sockaddr *) &storage, &addrlen) == 0); + + return own->sin_addr.s_addr; +} + +#if 0 +uint32_t +sock_get_peer_mac_addr(int fd) +{ + +} +#endif + +void +set_ip(uint8_t *ipbuf, uint32_t ip) +{ + ipbuf[0] = (ip & 0xFF); + ipbuf[1] = (ip & 0xFF00) >> 8; + ipbuf[2] = (ip & 0xFF0000) >> 16; + ipbuf[3] = (ip & 0xFF000000) >> 24; +} + + +int +bind_socket(int socktype, const char *host, const char *port) +{ + struct addrinfo hints, *result, *rp; + int reuseaddr = 1, s; + int fd; + + memset(&hints, 0, sizeof hints); + hints.ai_family = AF_INET; + hints.ai_socktype = socktype; + + printf("bind to host: %s, port: %s\n", host, port); + + s = getaddrinfo(host, port, &hints, &result); + if (s != 0) { + fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(s)); + return -1; + } + for (rp = result; rp != NULL; rp = rp->ai_next) { + fd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol); + if (fd == -1) + continue; + if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, + &reuseaddr, sizeof(reuseaddr)) == -1) + continue; + + /*ip = ((struct sockaddr_in *)rp->ai_addr)->sin_addr.s_addr;*/ + if (bind(fd, rp->ai_addr, rp->ai_addrlen) == 0) + break; + + close(fd); + } + freeaddrinfo(result); + if (rp == NULL) { + fprintf(stderr, "Failed to bind: %s\n", strerror(errno)); + return -1; + } + + if (socktype == SOCK_STREAM && + listen(fd, 1) < 0) { + fprintf(stderr, "Failed to listen: %s\n", strerror(errno)); + close(fd); + return -1; + } + + return fd; +} + +int +connect_to_host(int socktype, const char *host, const char *port) +{ + struct addrinfo hints, *result, *rp; + int fd, s; + + memset(&hints, 0, sizeof hints); + hints.ai_family = AF_INET; + hints.ai_socktype = socktype; + + printf("connect to host: %s, port: %s\n", host, port); + + s = getaddrinfo(host, port, &hints, &result); + if (s != 0) { + fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(s)); + return -1; + } + for (rp = result; rp != NULL; rp = rp->ai_next) { + fd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol); + if (fd == -1) + continue; + + if (connect(fd, rp->ai_addr, rp->ai_addrlen) != -1) + break; + + close(fd); + } + freeaddrinfo(result); + if (rp == NULL) { + fprintf(stderr, "Failed to bind: %s\n", strerror(errno)); + return -1; + } + + return fd; +} + +/* From systemd:src/shared/util.c */ +ssize_t +loop_read(int fd, void *buf, size_t nbytes, uint8_t do_poll) +{ + uint8_t *p; + ssize_t n = 0; + + assert(fd >= 0); + assert(buf); + + p = buf; + + while (nbytes > 0) { + ssize_t k; + + if ((k = read(fd, p, nbytes)) <= 0) { + + if (k < 0 && errno == EINTR) + continue; + + if (k < 0 && errno == EAGAIN && do_poll) { + struct pollfd pollfd; + + memset(&pollfd, 0, sizeof pollfd); + pollfd.fd = fd; + pollfd.events = POLLIN; + + if (poll(&pollfd, 1, -1) < 0) { + if (errno == EINTR) + continue; + + return n > 0 ? n : -errno; + } + + if (pollfd.revents != POLLIN) + return n > 0 ? n : -EIO; + + continue; + } + + return n > 0 ? n : (k < 0 ? -errno : 0); + } + + p += k; + nbytes -= k; + n += k; + } + + return n; +} diff --git a/util.h b/util.h new file mode 100644 index 0000000..4d0e27d --- /dev/null +++ b/util.h @@ -0,0 +1,28 @@ +#ifndef _UTIL_H_ +#define _UTIL_H_ + +#include + +#define MAX(a, b) (((a) > (b)) ? (a) : (b)) +#define ARRAY_SIZE(arr) (sizeof (arr) / sizeof ((arr)[0])) + +#define STR_EXPAND(tok) #tok +#define STR(str) STR_EXPAND(str) + +uint32_t +sock_get_ipv4_addr(int fd); +uint32_t +sock_get_peer_ipv4_addr(int fd); +uint32_t +sock_get_netmask(int fd); + +void +set_ip(uint8_t *ipbuf, uint32_t ip); +int +bind_socket(int socktype, const char *host, const char *port); +int +connect_to_host(int socktype, const char *host, const char *port); +ssize_t +loop_read(int fd, void *buf, size_t nbytes, uint8_t do_poll); + +#endif /* _UTIL_H_ */ -- cgit