colorful ipv6 status message

[i3/i3status] / src / print_ipv6_addr.c

diff --git a/src/print_ipv6_addr.c b/src/print_ipv6_addr.c
index b370011f0ac0e53fe24fa02a3aea37c418bbb0f2..abf45081b3ea6b1daaa67c9fde714f57e1cb4032 100644 (file)
--- a/src/print_ipv6_addr.c
+++ b/src/print_ipv6_addr.c
@@ -1,13 +1,19 @@
 // vim:ts=8:expandtab
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
 #include <stdio.h>
 #include <stdbool.h>
 #include <unistd.h>
 #include <stdlib.h>
-#include <sys/types.h>
-#include <sys/socket.h>
 #include <netdb.h>
 #include <string.h>
-#include <arpa/inet.h>
+#include <yajl/yajl_gen.h>
+#include <yajl/yajl_version.h>
+
+#include "i3status.h"
 
 static char *get_sockname(struct addrinfo *addr) {
         static char buf[INET6_ADDRSTRLEN+1];
@@ -45,7 +51,7 @@ static char *get_sockname(struct addrinfo *addr) {
         if ((ret = getnameinfo((struct sockaddr*)&local, local_len,
                                buf, sizeof(buf), NULL, 0,
                                NI_NUMERICHOST)) != 0) {
-                fprintf(stderr, "getnameinfo(): %s\n", gai_strerror(ret));
+                fprintf(stderr, "i3status: getnameinfo(): %s\n", gai_strerror(ret));
                 (void)close(fd);
                 return NULL;
         }
@@ -58,7 +64,7 @@ static char *get_sockname(struct addrinfo *addr) {
  * Returns the IPv6 address with which you have connectivity at the moment.
  * The char * is statically allocated and mustn't be freed
  */
-static char *get_ipv6_addr() {
+static char *get_ipv6_addr(void) {
         struct addrinfo hints;
         struct addrinfo *result, *resp;
         static struct addrinfo *cached = NULL;
@@ -70,6 +76,7 @@ static char *get_ipv6_addr() {
 
         memset(&hints, 0, sizeof(struct addrinfo));
         hints.ai_family = AF_INET6;
+        hints.ai_socktype = SOCK_DGRAM;
 
         /* We resolve the K root server to get a public IPv6 address. You can
          * replace this with any other host which has an AAAA record, but the
@@ -83,7 +90,11 @@ static char *get_ipv6_addr() {
 
         for (resp = result; resp != NULL; resp = resp->ai_next) {
                 char *addr_string = get_sockname(resp);
-                if (!addr_string)
+                /* If we could not get our own address and there is more than
+                 * one result for resolving k.root-servers.net, we cannot
+                 * cache. Otherwise, no matter if we got IPv6 connectivity or
+                 * not, we will cache the (single) result and are done. */
+                if (!addr_string && result->ai_next != NULL)
                         continue;
 
                 if ((cached = malloc(sizeof(struct addrinfo))) == NULL)
@@ -102,24 +113,30 @@ static char *get_ipv6_addr() {
         return NULL;
 }
 
-void print_ipv6_info(const char *format_up, const char *format_down) {
+void print_ipv6_info(yajl_gen json_gen, char *buffer, const char *format_up, const char *format_down) {
         const char *walk;
         char *addr_string = get_ipv6_addr();
+        char *outwalk = buffer;
 
         if (addr_string == NULL) {
-                printf("%s", format_down);
+                START_COLOR("color_bad");
+                OUTPUT_FULL_TEXT(format_down);
+                END_COLOR;
                 return;
         }
 
         for (walk = format_up; *walk != '\0'; walk++) {
                 if (*walk != '%') {
-                        putchar(*walk);
+                        *(outwalk++) = *walk;
                         continue;
                 }
 
                 if (strncmp(walk+1, "ip", strlen("ip")) == 0) {
-                        printf("%s", addr_string);
+                        outwalk += sprintf(outwalk, "%s", addr_string);
                         walk += strlen("ip");
                 }
         }
+        START_COLOR("color_good");
+        OUTPUT_FULL_TEXT(buffer);
+        END_COLOR;
 }