1 files changed, 125 insertions, 0 deletions
diff --git a/util.c b/util.c
new file mode 100644
index 0000000..4a2f1ad
--- /dev/null
+++ b/util.c
@@ -0,0 +1,125 @@
+/*
+ * Copyright (C) 2006 Stig Venaas <venaas@uninett.no>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ */
+
+#include <netdb.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <stdarg.h>
+
+void errx(char *format, ...) {
+    extern int errno;
+
+    va_list ap;
+    va_start(ap, format);
+    vfprintf(stderr, format, ap);
+    va_end(ap);
+    if (errno) {
+        fprintf(stderr, ": ");
+        perror(NULL);
+        fprintf(stderr, "errno=%d\n", errno);
+    } else
+        fprintf(stderr, "\n");
+    exit(1);
+}
+
+void err(char *format, ...) {
+    extern int errno;
+
+    va_list ap;
+    va_start(ap, format);
+    vfprintf(stderr, format, ap);
+    va_end(ap);
+    if (errno) {
+        fprintf(stderr, ": ");
+        perror(NULL);
+        fprintf(stderr, "errno=%d\n", errno);
+    } else
+        fprintf(stderr, "\n");
+}
+
+char *addr2string(struct sockaddr *addr, socklen_t len) {
+    struct sockaddr_in6 *sa6;
+    struct sockaddr_in sa4;
+    static char addr_buf[2][INET6_ADDRSTRLEN];
+    static int i = 0;
+    i = !i;
+    if (addr->sa_family == AF_INET6) {
+	sa6 = (struct sockaddr_in6 *)addr;
+	if (IN6_IS_ADDR_V4MAPPED(&sa6->sin6_addr)) {
+	    memset(&sa4, 0, sizeof(sa4));
+	    sa4.sin_family = AF_INET;
+	    sa4.sin_port = sa6->sin6_port;
+	    memcpy(&sa4.sin_addr, &sa6->sin6_addr.s6_addr[12], 4);
+	    addr = (struct sockaddr *)&sa4;
+	}
+    }
+    if (getnameinfo(addr, len, addr_buf[i], sizeof(addr_buf[i]),
+                    NULL, 0, NI_NUMERICHOST)) {
+        err("getnameinfo");
+        return NULL;
+    }
+    return addr_buf[i];
+}
+
+int bindport(int type, char *port) {
+    struct addrinfo hints, *res0, *res;
+    int s;
+    
+    memset(&hints, 0, sizeof(hints));
+    hints.ai_socktype = type;
+    hints.ai_family = AF_UNSPEC;
+    hints.ai_flags = AI_PASSIVE;
+
+    if (getaddrinfo(NULL, port, &hints, &res0) != 0) {
+        err("bindport: can't resolve port %s", port);
+	return -1;
+    }
+    for (res = res0; res; res = res->ai_next) {
+        s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
+        if (s >= 0) {
+            if (bind(s, res->ai_addr, res->ai_addrlen) == 0)
+                break;
+            close(s);
+            s = -1;
+        }
+    }
+    freeaddrinfo(res0);
+    return s;
+}
+
+int connectport(int type, char *host, char *port) {
+    struct addrinfo hints, *res0, *res;
+    int s;
+    
+    memset(&hints, 0, sizeof(hints));
+    hints.ai_socktype = type;
+    hints.ai_family = AF_UNSPEC;
+
+    if (getaddrinfo(host, port, &hints, &res0) != 0) {
+        err("connectport: can't resolve host %s port %s", host, port);
+	return -1;
+    }
+
+    for (res = res0; res; res = res->ai_next) {
+	s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
+	if (s < 0) {
+	    err("connectport: socket failed");
+	    continue;
+	}
+	if (connect(s, res->ai_addr, res->ai_addrlen) == 0)
+	    break;
+	err("connectport: connect failed");
+	close(s);
+	s = -1;
+    }
+    freeaddrinfo(res0);
+    return s;
+}