info: merge jsoninfo and txtinfo common code
[olsrd.git] / lib / jsoninfo / src / olsrd_jsoninfo.c
index 8f27a4a..f0acfff 100644 (file)
  * Dynamic linked library for the olsr.org olsr daemon
  */
 
-#include <arpa/inet.h>
-#include <string.h>
-#include <unistd.h>
-
-#include "ipcalc.h"
-#include "olsr.h"
-#include "scheduler.h"
-#include "../../info/http_headers.h"
-#include "../../info/info_types.h"
-#include "jsoninfo_printers.h"
 #include "olsrd_jsoninfo.h"
+#include "jsoninfo_printers.h"
+#include "../../info/olsrd_info.h"
 
-#ifdef _WIN32
-#define close(x) closesocket(x)
-#endif /* _WIN32 */
-
-/* defines to make txtinfo and jsoninfo look alike */
 #define PLUGIN_NAME "JSONINFO"
 
-static int ipc_socket;
-
-/* IPC initialization function */
-static int plugin_ipc_init(void);
-
-static void send_info(unsigned int /*send_what*/, int /*socket*/);
-
-static void ipc_action(int, void *, unsigned int);
-
-#define TXT_IPC_BUFSIZE 256
-
-static outbuffer_t outbuffer;
-
-static struct timer_entry *writetimer_entry;
-
 static printer_functions_t printer_functions = { //
     //
         .init = &plugin_init, //
@@ -99,59 +71,6 @@ static printer_functions_t printer_functions = { //
         .plugins = &ipc_print_plugins //
     };
 
-static void determine_action(unsigned int *send_what, char *requ) {
-  if (!printer_functions.is_command)
-    *send_what = 0;
-  else if ((*printer_functions.is_command)(requ, SIW_OLSRD_CONF))
-    *send_what |= SIW_OLSRD_CONF;
-  else if ((*printer_functions.is_command)(requ, SIW_ALL))
-    *send_what = SIW_ALL;
-  else {
-    // these are the two overarching categories
-    if ((*printer_functions.is_command)(requ, SIW_RUNTIME_ALL))
-      *send_what |= SIW_RUNTIME_ALL;
-    if ((*printer_functions.is_command)(requ, SIW_STARTUP_ALL))
-      *send_what |= SIW_STARTUP_ALL;
-
-    // these are the individual sections
-    if ((*printer_functions.is_command)(requ, SIW_NEIGHBORS))
-      *send_what |= SIW_NEIGHBORS;
-    if ((*printer_functions.is_command)(requ, SIW_LINKS))
-      *send_what |= SIW_LINKS;
-    if ((*printer_functions.is_command)(requ, SIW_ROUTES))
-      *send_what |= SIW_ROUTES;
-    if ((*printer_functions.is_command)(requ, SIW_HNA))
-      *send_what |= SIW_HNA;
-    if ((*printer_functions.is_command)(requ, SIW_MID))
-      *send_what |= SIW_MID;
-    if ((*printer_functions.is_command)(requ, SIW_TOPOLOGY))
-      *send_what |= SIW_TOPOLOGY;
-    if ((*printer_functions.is_command)(requ, SIW_GATEWAYS))
-      *send_what |= SIW_GATEWAYS;
-    if ((*printer_functions.is_command)(requ, SIW_INTERFACES))
-      *send_what |= SIW_INTERFACES;
-    if ((*printer_functions.is_command)(requ, SIW_2HOP))
-      *send_what |= SIW_2HOP;
-    if ((*printer_functions.is_command)(requ, SIW_SGW))
-      *send_what |= SIW_SGW;
-
-    // specials
-    if ((*printer_functions.is_command)(requ, SIW_VERSION))
-      *send_what |= SIW_VERSION;
-    if ((*printer_functions.is_command)(requ, SIW_CONFIG))
-      *send_what |= SIW_CONFIG;
-    if ((*printer_functions.is_command)(requ, SIW_PLUGINS))
-      *send_what |= SIW_PLUGINS;
-
-    /* To print out neighbours only on the Freifunk Status
-     * page the normal output is somewhat lengthy. The
-     * header parsing is sufficient for standard wget.
-     */
-    if ((*printer_functions.is_command)(requ, SIW_NEIGHBORS_FREIFUNK))
-      *send_what = SIW_NEIGHBORS_FREIFUNK;
-  }
-}
-
 /**
  *Do initialization here
  *
@@ -159,306 +78,14 @@ static void determine_action(unsigned int *send_what, char *requ) {
  *function in uolsrd_plugin.c
  */
 int olsrd_plugin_init(void) {
-  /* Initial IPC value */
-  ipc_socket = -1;
-  memset(&outbuffer, 0, sizeof(outbuffer));
-
-  if (printer_functions.init) {
-    (*printer_functions.init)(PLUGIN_NAME);
-  }
-
-  plugin_ipc_init();
-  return 1;
+  return info_plugin_init(PLUGIN_NAME, &printer_functions, &info_plugin_config);
 }
 
 /**
  * destructor - called at unload
  */
 void olsr_plugin_exit(void) {
-  if (ipc_socket != -1)
-    close(ipc_socket);
-}
-
-static int plugin_ipc_init(void) {
-  union olsr_sockaddr sst;
-  uint32_t yes = 1;
-  socklen_t addrlen;
-
-  /* Init ipc socket */
-  if ((ipc_socket = socket(olsr_cnf->ip_version, SOCK_STREAM, 0)) == -1) {
-#ifndef NODEBUG
-    olsr_printf(1, "("PLUGIN_NAME") socket()=%s\n", strerror(errno));
-#endif /* NODEBUG */
-    return 0;
-  } else {
-    if (setsockopt(ipc_socket, SOL_SOCKET, SO_REUSEADDR, (char *) &yes, sizeof(yes)) < 0) {
-#ifndef NODEBUG
-      olsr_printf(1, "("PLUGIN_NAME") setsockopt()=%s\n", strerror(errno));
-#endif /* NODEBUG */
-      return 0;
-    }
-#if (defined __FreeBSD__ || defined __FreeBSD_kernel__) && defined SO_NOSIGPIPE
-    if (setsockopt(ipc_socket, SOL_SOCKET, SO_NOSIGPIPE, (char *) &yes, sizeof(yes)) < 0) {
-      perror("SO_REUSEADDR failed");
-      return 0;
-    }
-#endif /* (defined __FreeBSD__ || defined __FreeBSD_kernel__) && defined SO_NOSIGPIPE */
-#if defined linux && defined IPV6_V6ONLY
-    if (info_plugin_config.ipv6_only && olsr_cnf->ip_version == AF_INET6) {
-      if (setsockopt(ipc_socket, IPPROTO_IPV6, IPV6_V6ONLY, (char *) &yes, sizeof(yes)) < 0) {
-        perror("IPV6_V6ONLY failed");
-        return 0;
-      }
-    }
-#endif /* defined linux && defined IPV6_V6ONLY */
-    /* Bind the socket */
-
-    /* complete the socket structure */
-    memset(&sst, 0, sizeof(sst));
-    if (olsr_cnf->ip_version == AF_INET) {
-      sst.in4.sin_family = AF_INET;
-      addrlen = sizeof(struct sockaddr_in);
-#ifdef SIN6_LEN
-      sst.in4.sin_len = addrlen;
-#endif /* SIN6_LEN */
-      sst.in4.sin_addr.s_addr = info_plugin_config.listen_ip.v4.s_addr;
-      sst.in4.sin_port = htons(info_plugin_config.ipc_port);
-    } else {
-      sst.in6.sin6_family = AF_INET6;
-      addrlen = sizeof(struct sockaddr_in6);
-#ifdef SIN6_LEN
-      sst.in6.sin6_len = addrlen;
-#endif /* SIN6_LEN */
-      sst.in6.sin6_addr = info_plugin_config.listen_ip.v6;
-      sst.in6.sin6_port = htons(info_plugin_config.ipc_port);
-    }
-
-    /* bind the socket to the port number */
-    if (bind(ipc_socket, &sst.in, addrlen) == -1) {
-#ifndef NODEBUG
-      olsr_printf(1, "("PLUGIN_NAME") bind()=%s\n", strerror(errno));
-#endif /* NODEBUG */
-      return 0;
-    }
-
-    /* show that we are willing to listen */
-    if (listen(ipc_socket, 1) == -1) {
-#ifndef NODEBUG
-      olsr_printf(1, "("PLUGIN_NAME") listen()=%s\n", strerror(errno));
-#endif /* NODEBUG */
-      return 0;
-    }
-
-    /* Register with olsrd */
-    add_olsr_socket(ipc_socket, &ipc_action, NULL, NULL, SP_PR_READ);
-
-#ifndef NODEBUG
-    olsr_printf(2, "("PLUGIN_NAME") listening on port %d\n", info_plugin_config.ipc_port);
-#endif /* NODEBUG */
-  }
-  return 1;
-}
-
-static void ipc_action(int fd, void *data __attribute__ ((unused)), unsigned int flags __attribute__ ((unused))) {
-  union olsr_sockaddr pin;
-
-  char addr[INET6_ADDRSTRLEN];
-  fd_set rfds;
-  struct timeval tv;
-  unsigned int send_what = 0;
-  int ipc_connection;
-
-  socklen_t addrlen = sizeof(pin);
-
-  if (outbuffer.count >= MAX_CLIENTS) {
-    return;
-  }
-
-  if ((ipc_connection = accept(fd, &pin.in, &addrlen)) == -1) {
-#ifndef NODEBUG
-    olsr_printf(1, "("PLUGIN_NAME") accept()=%s\n", strerror(errno));
-#endif /* NODEBUG */
-    return;
-  }
-
-  tv.tv_sec = tv.tv_usec = 0;
-  if (olsr_cnf->ip_version == AF_INET) {
-    if (inet_ntop(olsr_cnf->ip_version, &pin.in4.sin_addr, addr, INET6_ADDRSTRLEN) == NULL)
-      addr[0] = '\0';
-    if (!ip4equal(&pin.in4.sin_addr, &info_plugin_config.accept_ip.v4) && info_plugin_config.accept_ip.v4.s_addr != INADDR_ANY) {
-      if (!info_plugin_config.allow_localhost || ntohl(pin.in4.sin_addr.s_addr) != INADDR_LOOPBACK) {
-        olsr_printf(1, "("PLUGIN_NAME") From host(%s) not allowed!\n", addr);
-        close(ipc_connection);
-        return;
-      }
-    }
-  } else {
-    if (inet_ntop(olsr_cnf->ip_version, &pin.in6.sin6_addr, addr, INET6_ADDRSTRLEN) == NULL)
-      addr[0] = '\0';
-    /* Use in6addr_any (::) in olsr.conf to allow anybody. */
-    if (!ip6equal(&in6addr_any, &info_plugin_config.accept_ip.v6) && !ip6equal(&pin.in6.sin6_addr, &info_plugin_config.accept_ip.v6)) {
-      olsr_printf(1, "("PLUGIN_NAME") From host(%s) not allowed!\n", addr);
-      close(ipc_connection);
-      return;
-    }
-  }
-
-#ifndef NODEBUG
-  olsr_printf(2, "("PLUGIN_NAME") Connect from %s\n", addr);
-#endif /* NODEBUG */
-
-  /* purge read buffer to prevent blocking on linux */
-  FD_ZERO(&rfds);
-  FD_SET((unsigned int )ipc_connection, &rfds); /* Win32 needs the cast here */
-  if (0 <= select(ipc_connection + 1, &rfds, NULL, NULL, &tv)) {
-    char requ[1024];
-    ssize_t s = recv(ipc_connection, (void *) &requ, sizeof(requ) - 1, 0); /* Win32 needs the cast here */
-
-    if (s == sizeof(requ) - 1) {
-      /* input was much too long, just skip the rest */
-      char dummy[1024];
-
-      while (recv(ipc_connection, (void *) &dummy, sizeof(dummy), 0) == sizeof(dummy))
-        ;
-    }
-
-    if (0 < s) {
-      requ[s] = 0;
-      determine_action(&send_what, requ);
-    }
-
-    if (!send_what)
-      send_what = SIW_ALL;
-  }
-
-  send_info(send_what, ipc_connection);
-}
-
-static void info_write_data(void *foo __attribute__ ((unused))) {
-  fd_set set;
-  int result, i, j, max;
-  struct timeval tv;
-
-  FD_ZERO(&set);
-  max = 0;
-  for (i = 0; i < outbuffer.count; i++) {
-    /* And we cast here since we get a warning on Win32 */
-    FD_SET((unsigned int )(outbuffer.socket[i]), &set);
-
-    if (outbuffer.socket[i] > max) {
-      max = outbuffer.socket[i];
-    }
-  }
-
-  tv.tv_sec = 0;
-  tv.tv_usec = 0;
-
-  result = select(max + 1, NULL, &set, NULL, &tv);
-  if (result <= 0) {
-    return;
-  }
-
-  for (i = 0; i < outbuffer.count; i++) {
-    if (FD_ISSET(outbuffer.socket[i], &set)) {
-      result = send(outbuffer.socket[i], outbuffer.buffer[i] + outbuffer.written[i], outbuffer.size[i] - outbuffer.written[i], 0);
-      if (result > 0) {
-        outbuffer.written[i] += result;
-      }
-
-      if (result <= 0 || outbuffer.written[i] == outbuffer.size[i]) {
-        /* close this socket and cleanup*/
-        close(outbuffer.socket[i]);
-        free(outbuffer.buffer[i]);
-        outbuffer.buffer[i] = NULL;
-
-        for (j = i + 1; j < outbuffer.count; j++) {
-          outbuffer.buffer[j - 1] = outbuffer.buffer[j];
-          outbuffer.size[j - 1] = outbuffer.size[j];
-          outbuffer.socket[j - 1] = outbuffer.socket[j];
-          outbuffer.written[j - 1] = outbuffer.written[j];
-        }
-        outbuffer.count--;
-      }
-    }
-  }
-  if (!outbuffer.count) {
-    olsr_stop_timer(writetimer_entry);
-  }
-}
-
-static void send_info(unsigned int send_what, int the_socket) {
-  struct autobuf abuf;
-
-  const char *content_type = (printer_functions.determine_mime_type) ? (*printer_functions.determine_mime_type)(send_what) : "text/plain; charset=utf-8";
-  int contentLengthPlaceholderStart = 0;
-  int headerLength = 0;
-
-  abuf_init(&abuf, 2 * 4096);
-
-  if (info_plugin_config.http_headers) {
-    build_http_header(PLUGIN_NAME, HTTP_200, content_type, &abuf, &contentLengthPlaceholderStart);
-    headerLength = abuf.len;
-  }
-
-  // only add if normal format
-  if (send_what & SIW_ALL) {
-    if (printer_functions.output_start)
-      (*printer_functions.output_start)(&abuf);
-
-    if ((send_what & SIW_LINKS) && printer_functions.links)
-      (*printer_functions.links)(&abuf);
-    if ((send_what & SIW_NEIGHBORS) && printer_functions.neighbors)
-      (*printer_functions.neighbors)(&abuf, false);
-    if ((send_what & SIW_TOPOLOGY) && printer_functions.topology)
-      (*printer_functions.topology)(&abuf);
-    if ((send_what & SIW_HNA) && printer_functions.hna)
-      (*printer_functions.hna)(&abuf);
-    if ((send_what & SIW_SGW) && printer_functions.sgw)
-      (*printer_functions.sgw)(&abuf);
-    if ((send_what & SIW_MID) && printer_functions.mid)
-      (*printer_functions.mid)(&abuf);
-    if ((send_what & SIW_ROUTES) && printer_functions.routes)
-      (*printer_functions.routes)(&abuf);
-    if ((send_what & SIW_GATEWAYS) && printer_functions.gateways)
-      (*printer_functions.gateways)(&abuf);
-    if ((send_what & SIW_CONFIG) && printer_functions.config)
-      (*printer_functions.config)(&abuf);
-    if ((send_what & SIW_INTERFACES) && printer_functions.interfaces)
-      (*printer_functions.interfaces)(&abuf);
-    if ((send_what & SIW_2HOP) && printer_functions.neighbors)
-      (*printer_functions.neighbors)(&abuf, true);
-    if ((send_what & SIW_VERSION) && printer_functions.version)
-      (*printer_functions.version)(&abuf);
-    if ((send_what & SIW_PLUGINS) && printer_functions.plugins)
-      (*printer_functions.plugins)(&abuf);
-
-    if (printer_functions.output_end)
-      (*printer_functions.output_end)(&abuf);
-  } else if ((send_what & SIW_OLSRD_CONF) && printer_functions.olsrd_conf) {
-    /* this outputs the olsrd.conf text directly, not normal format */
-    (*printer_functions.olsrd_conf)(&abuf);
-  }
-
-  if (info_plugin_config.http_headers) {
-    http_header_adjust_content_length(&abuf, contentLengthPlaceholderStart, abuf.len - headerLength);
-  }
-
-  /* avoid a memcpy: just move the abuf.buf pointer and clear abuf */
-  outbuffer.buffer[outbuffer.count] = abuf.buf;
-  outbuffer.size[outbuffer.count] = abuf.len;
-  outbuffer.written[outbuffer.count] = 0;
-  outbuffer.socket[outbuffer.count] = the_socket;
-  abuf.buf = NULL;
-  abuf.len = 0;
-  abuf.size = 0;
-
-  outbuffer.count++;
-
-  if (outbuffer.count == 1) {
-    writetimer_entry = olsr_start_timer(100, 0, OLSR_TIMER_PERIODIC, &info_write_data, NULL, 0);
-  }
-
-  abuf_free(&abuf);
+  info_plugin_exit();
 }
 
 /*