pinba plugin: pinba_socket_open: Rewrote the function ...
[collectd.git] / src / pinba.c
index 6110429..30a1426 100644 (file)
@@ -1,5 +1,6 @@
 /**
- * collectd - src/pinba.c
+ * collectd - src/pinba.c (based on code from pinba_engine 0.0.5)
+ * Copyright (c) 2007-2009  Antony Dovgal
  * Copyright (C) 2010       Phoenix Kayo
  *
  * This program is free software; you can redistribute it and/or modify it
  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  *
  * Authors:
+ *   Antony Dovgal <tony at daylessday.org>
  *   Phoenix Kayo <kayo.k11.4 at gmail.com>
  **/
 
+#define _XOPEN_SOURCE 500
+
 #include "collectd.h"
 #include "common.h"
 #include "plugin.h"
 #include "configfile.h"
 
-/*
- *  Service declaration section
- */
+#include <pthread.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <poll.h>
 
-#ifndef PINBA_UDP_BUFFER_SIZE
-#define PINBA_UDP_BUFFER_SIZE 65536
-#endif
+#include "pinba.pb-c.h"
 
-typedef struct _pinba_statres_ pinba_statres;
-struct _pinba_statres_ {
-  const char *name;
-  float req_per_sec;
-  float req_time;
-  float ru_utime;
-  float ru_stime;
-  float doc_size;
-  float mem_peak;
-};
+typedef uint8_t u_char;
 
-/*
- *  Fetch collection datas
- *
- *  Return value: id of next data, zero if no data.
- */
-static unsigned int
-service_statnode_collect(pinba_statres *res,
-                        unsigned int id);
-static void
-service_statnode_begin();
-static void
-service_statnode_end();
-static void
-service_statnode_free();
-static void
-service_statnode_init();
-static void
-service_statnode_add(const char *name,
-                    const char *host,
-                    const char *server,
-                    const char *script);
-
-static void
-service_config(const char* address,
-              unsigned int port);
-static int
-service_start();
-static int
-service_stop();
+#include <event.h>
 
 /*
- *  Service implementation section
+ *  Service declaration section
  */
+#ifndef PINBA_UDP_BUFFER_SIZE
+# define PINBA_UDP_BUFFER_SIZE 65536
+#endif
 
-#include"pinba.pb-c.h"
+#ifndef PINBA_DEFAULT_ADDRESS
+# define PINBA_DEFAULT_ADDRESS "127.0.0.1" /* FIXME */
+#endif
 
-#include<stdarg.h>
-#include<stdlib.h>
-#include<stdio.h>
-#include<string.h>
-#include<unistd.h>
-#include<errno.h>
-#include<fcntl.h>
+#ifndef PINBA_DEFAULT_PORT
+# define PINBA_DEFAULT_PORT 12345 /* FIXME */
+#endif
 
-#include<sys/types.h>
-#include<math.h>
-#include<pthread.h>
+#ifndef PINBA_MAX_SOCKETS
+# define PINBA_MAX_SOCKETS 16
+#endif
 
-#include<sys/time.h>
-#include<event.h>
+#ifndef NI_MAXSERV
+# define NI_MAXSERV 32
+#endif
 
-#include<netinet/in.h>
-#include<arpa/inet.h>
-#include<sys/socket.h>
+/*
+ * Private data structures
+ */
+typedef struct _pinba_statres_ pinba_statres;
+struct _pinba_statres_ {
+  const char *name;
+  double req_per_sec;
+  double req_time;
+  double ru_utime;
+  double ru_stime;
+  double doc_size;
+  double mem_peak;
+};
 
-typedef struct _pinba_socket_ pinba_socket;
-struct _pinba_socket_ {
+struct pinba_socket_s {
   int listen_sock;
   struct event *accept_event;
+
+  struct pollfd fd[PINBA_MAX_SOCKETS];
+  nfds_t fd_num;
 };
+typedef struct pinba_socket_s pinba_socket_t;
 
-typedef double pinba_time;
-typedef uint32_t pinba_size;
+typedef double pinba_time_t;
+typedef uint32_t pinba_size_t;
 
-static pinba_time
-now(){
+static pinba_time_t now (void)
+{
   static struct timeval tv;
-  static struct timezone tz;
   
-  gettimeofday(&tv, &tz);
+  gettimeofday (&tv, /* tz = */ NULL);
   
   return (double)tv.tv_sec+((double)tv.tv_usec/(double)1000000);
 }
 
-static pthread_rwlock_t
-temp_lock;
+static pthread_rwlock_t temp_lock;
 
-static struct event_base *
-temp_base=NULL;
+static struct event_base *temp_base = NULL;
 
-static pinba_socket *
-temp_sock=NULL;
+static pinba_socket_t *temp_sock = NULL;
 
-static pthread_t
-temp_thrd;
+static pthread_t temp_thrd;
 
 typedef struct _pinba_statnode_ pinba_statnode;
 struct _pinba_statnode_{
@@ -138,23 +114,25 @@ struct _pinba_statnode_{
   char *server;
   char *script;
   /* collected data */
-  pinba_time last_coll;
-  pinba_size req_count;
-  pinba_time req_time;
-  pinba_time ru_utime;
-  pinba_time ru_stime;
-  pinba_size doc_size;
-  pinba_size mem_peak;
+  pinba_time_t last_coll;
+  pinba_size_t req_count;
+  pinba_time_t req_time;
+  pinba_time_t ru_utime;
+  pinba_time_t ru_stime;
+  pinba_size_t doc_size;
+  pinba_size_t mem_peak;
 };
 
-static unsigned int
-stat_nodes_count=0;
+static unsigned int stat_nodes_count=0;
+
+static pinba_statnode *stat_nodes = NULL;
 
-static pinba_statnode *
-stat_nodes=NULL;
+char service_status=0;
+char *service_address = PINBA_DEFAULT_ADDRESS;
+unsigned int service_port=PINBA_DEFAULT_PORT;
 
-static void
-service_statnode_reset(pinba_statnode *node){
+static void service_statnode_reset (pinba_statnode *node) /* {{{ */
+{
   node->last_coll=now();
   node->req_count=0;
   node->req_time=0.0;
@@ -162,25 +140,28 @@ service_statnode_reset(pinba_statnode *node){
   node->ru_stime=0.0;
   node->doc_size=0;
   node->mem_peak=0;
-}
+} /* }}} void service_statnode_reset */
 
-static inline void
-strdel(char **str){
-  if(*str)free(*str);
-  *str=NULL;
-}
+static void strset (char **str, const char *new) /* {{{ */
+{
+  char *tmp;
 
-static inline void
-strset(char **str, const char *new){
-  strdel(str);
-  new && (*str=strdup(new));
-}
+  if (!str || !new)
+    return;
+
+  tmp = strdup (new);
+  if (tmp == NULL)
+    return;
+
+  sfree (*str);
+  *str = tmp;
+} /* }}} void strset */
 
-static void
-service_statnode_add(const char *name,
-                    const char *host,
-                    const char *server,
-                    const char *script){
+static void service_statnode_add(const char *name, /* {{{ */
+    const char *host,
+    const char *server,
+    const char *script)
+{
   pinba_statnode *node;
   DEBUG("adding node `%s' to collector { %s, %s, %s }", name, host?host:"", server?server:"", script?script:"");
   
@@ -209,31 +190,31 @@ service_statnode_add(const char *name,
   
   /* increment counter */
   stat_nodes_count++;
-}
+} /* }}} void service_statnode_add */
 
-static void
-service_statnode_free(){
-  if(stat_nodes_count){
-    DEBUG("shutdowning collector..");
-    unsigned int i;
-    
-    for(i=0;i<stat_nodes_count;i++){
-      /* free strings */
-      strdel(&stat_nodes[i].name);
-      strdel(&stat_nodes[i].host);
-      strdel(&stat_nodes[i].server);
-      strdel(&stat_nodes[i].script);
-    }
-    
-    free(stat_nodes);
-    stat_nodes_count=0;
-    
-    pthread_rwlock_destroy(&temp_lock);
+static void service_statnode_free (void)
+{
+  unsigned int i;
+
+  if(stat_nodes_count < 1)
+    return;
+
+  for (i = 0; i < stat_nodes_count; i++)
+  {
+    sfree (stat_nodes[i].name);
+    sfree (stat_nodes[i].host);
+    sfree (stat_nodes[i].server);
+    sfree (stat_nodes[i].script);
   }
+
+  sfree (stat_nodes);
+  stat_nodes_count = 0;
+
+  pthread_rwlock_destroy (&temp_lock);
 }
 
-static void
-service_statnode_init(){
+static void service_statnode_init (void)
+{
   /* only total info collect by default */
   service_statnode_free();
   
@@ -241,62 +222,62 @@ service_statnode_init(){
   pthread_rwlock_init(&temp_lock, 0);
 }
 
-static void
-service_statnode_begin(){
+static void service_statnode_begin (void)
+{
   service_statnode_init();
   pthread_rwlock_wrlock(&temp_lock);
   
   service_statnode_add("total", NULL, NULL, NULL);
 }
 
-static void
-service_statnode_end(){
+static void service_statnode_end (void)
+{
   pthread_rwlock_unlock(&temp_lock);
 }
 
-static unsigned int
-service_statnode_collect(pinba_statres *res, unsigned int i){
+static unsigned int service_statnode_collect (pinba_statres *res, /* {{{ */
+    unsigned int index)
+{
   pinba_statnode* node;
+  pinba_time_t delta;
   
-  if(stat_nodes_count==0) return 0;
+  if (stat_nodes_count == 0)
+    return 0;
   
   /* begin collecting */
-  if(i==0){
-    pthread_rwlock_wrlock(&temp_lock);
-  }
-  
-  /* find non-empty node */
-  //for(node=stat_nodes+i; node->req_count==0 && ++i<stat_nodes_count; node=stat_nodes+i);
+  if (index == 0)
+    pthread_rwlock_wrlock (&temp_lock);
   
   /* end collecting */
-  if(i>=stat_nodes_count){
-    pthread_rwlock_unlock(&temp_lock);
+  if (index >= stat_nodes_count)
+  {
+    pthread_rwlock_unlock (&temp_lock);
     return 0;
   }
   
-  node=stat_nodes+i;
-  
-  pinba_time delta=now()-node->last_coll;
-  
-  res->name=node->name;
+  node = stat_nodes + index;
+  delta = now() - node->last_coll;
   
-  res->req_per_sec=node->req_count/delta;
+  res->name = node->name;
+  res->req_per_sec = node->req_count / delta;
   
-  if(node->req_count==0)node->req_count=1;
-  res->req_time=node->req_time/node->req_count;
-  res->ru_utime=node->ru_utime/node->req_count;
-  res->ru_stime=node->ru_stime/node->req_count;
-  res->ru_stime=node->ru_stime/node->req_count;
-  res->doc_size=node->doc_size/node->req_count;
-  res->mem_peak=node->mem_peak/node->req_count;
+  if (node->req_count == 0)
+    node->req_count = 1;
+
+  res->req_time = node->req_time / node->req_count;
+  res->ru_utime = node->ru_utime / node->req_count;
+  res->ru_stime = node->ru_stime / node->req_count;
+  res->ru_stime = node->ru_stime / node->req_count;
+  res->doc_size = node->doc_size / node->req_count;
+  res->mem_peak = node->mem_peak / node->req_count;
   
-  service_statnode_reset(node);
-  return ++i;
-}
+  service_statnode_reset (node);
+  return (index + 1);
+} /* }}} unsigned int service_statnode_collect */
 
-static void
-service_statnode_process(pinba_statnode *node,
-                        Pinba__Request* request){
+static void service_statnode_process (pinba_statnode *node,
+    Pinba__Request* request)
+{
   node->req_count++;
   node->req_time+=request->request_time;
   node->ru_utime+=request->ru_utime;
@@ -305,72 +286,238 @@ service_statnode_process(pinba_statnode *node,
   node->mem_peak+=request->memory_peak;
 }
 
-static void
-service_process_request(Pinba__Request* request){
+static void service_process_request (Pinba__Request *request)
+{
   unsigned int i;
-  pthread_rwlock_wrlock(&temp_lock);
+
+  pthread_rwlock_wrlock (&temp_lock);
   
-  DEBUG("Process request..");
-  for(i=0;i<stat_nodes_count;i++){
-    if(stat_nodes[i].host && strcmp(request->hostname, stat_nodes[i].host))continue;
-    if(stat_nodes[i].server && strcmp(request->server_name, stat_nodes[i].server))continue;
-    if(stat_nodes[i].script && strcmp(request->script_name, stat_nodes[i].script))continue;
+  for (i = 0; i < stat_nodes_count; i++)
+  {
+    if(stat_nodes[i].host && strcmp(request->hostname, stat_nodes[i].host))
+      continue;
+    if(stat_nodes[i].server && strcmp(request->server_name, stat_nodes[i].server))
+      continue;
+    if(stat_nodes[i].script && strcmp(request->script_name, stat_nodes[i].script))
+      continue;
+
     service_statnode_process(&stat_nodes[i], request);
   }
   
   pthread_rwlock_unlock(&temp_lock);
 }
 
-char
-service_status=0;
-
-char *
-service_address=PINBA_DEFAULT_ADDRESS;
-
-unsigned int
-service_port=PINBA_DEFAULT_PORT;
+static void *pinba_main (void *arg)
+{
+  DEBUG("entering listen-loop..");
+  
+  service_status=1;
+  event_base_dispatch(temp_base);
+  
+  /* unreachable */
+  return NULL;
+}
 
-static void
-service_config(const char* address,
-              unsigned int port){
-  char need_restart=0;
+static void pinba_socket_free (pinba_socket_t *socket) /* {{{ */
+{
+  if (!socket)
+    return;
   
-  if(address && (service_address && strcmp(service_address, address))){
-    strset(&service_address, address);
-    need_restart++;
+  if (socket->listen_sock >= 0)
+  {
+    close(socket->listen_sock);
+    socket->listen_sock = -1;
   }
   
-  if(port>0 && port<65536 && service_port!=port){
-    service_port=port;
-    need_restart++;
+  if (socket->accept_event)
+  {
+    event_del(socket->accept_event);
+    free(socket->accept_event);
+    socket->accept_event = NULL;
   }
   
-  if(service_status && need_restart){
-    service_stop();
-    service_start();
+  free(socket);
+} /* }}} void pinba_socket_free */
+
+static int pinba_process_stats_packet (const uint8_t *buffer, /* {{{ */
+    size_t buffer_size)
+{
+  Pinba__Request *request;  
+  
+  request = pinba__request__unpack (NULL, buffer_size, buffer);
+  
+  if (!request)
+    return (-1);
+
+  service_process_request(request);
+  pinba__request__free_unpacked (request, NULL);
+    
+  return (0);
+} /* }}} int pinba_process_stats_packet */
+
+static void pinba_udp_read_callback_fn (int sock, short event, void *arg) /* {{{ */
+{
+  uint8_t buffer[PINBA_UDP_BUFFER_SIZE];
+  size_t buffer_size;
+  int status;
+
+  if ((event & EV_READ) == 0)
+    return;
+
+  while (42)
+  {
+    buffer_size = sizeof (buffer);
+    status = recvfrom (sock, buffer, buffer_size - 1, MSG_DONTWAIT, /* from = */ NULL, /* from len = */ 0);
+    if (status < 0)
+    {
+      char errbuf[1024];
+
+      if ((errno == EINTR)
+#ifdef EWOULDBLOCK
+          || (errno == EWOULDBLOCK)
+#endif
+          || (errno == EAGAIN))
+      {
+        continue;
+      }
+
+      WARNING("pinba plugin: recvfrom(2) failed: %s",
+          sstrerror (errno, errbuf, sizeof (errbuf)));
+      return;
+    }
+    else if (status == 0)
+    {
+      DEBUG ("pinba plugin: recvfrom(2) returned unexpected status zero.");
+      return;
+    }
+    else /* if (status > 0) */
+    {
+      assert (((size_t) status) < buffer_size);
+      buffer_size = (size_t) status;
+      buffer[buffer_size] = 0;
+
+      status = pinba_process_stats_packet (buffer, buffer_size);
+      if (status != 0)
+        DEBUG("pinba plugin: Parsing packet failed.");
+      return;
+    }
+
+    /* not reached */
+    assert (23 == 42);
+  } /* while (42) */
+} /* }}} void pinba_udp_read_callback_fn */
+
+static int pb_add_socket (pinba_socket_t *s, /* {{{ */
+    const struct addrinfo *ai)
+{
+  int fd;
+  int tmp;
+  int status;
+
+  if (s->fd_num == PINBA_MAX_SOCKETS)
+  {
+    WARNING ("pinba plugin: Sorry, you have hit the built-in limit of "
+        "%i sockets. Please complain to the collectd developers so we can "
+        "raise the limit.", PINBA_MAX_SOCKETS);
+    return (-1);
   }
-}
 
-static void
-pinba_socket_free (pinba_socket *socket);
+  fd = socket (ai->ai_family, ai->ai_socktype, ai->ai_protocol);
+  if (fd < 0)
+  {
+    char errbuf[1024];
+    ERROR ("pinba plugin: socket(2) failed: %s",
+        sstrerror (errno, errbuf, sizeof (errbuf)));
+    return (0);
+  }
 
-static pinba_socket *
-pinba_socket_open (const char *ip,
-                  int listen_port);
+  tmp = 1;
+  status = setsockopt (fd, SOL_SOCKET, SO_REUSEADDR, &tmp, sizeof (tmp));
+  if (status != 0)
+  {
+    char errbuf[1024];
+    WARNING ("pinba plugin: setsockopt(SO_REUSEADDR) failed: %s",
+        sstrerror (errno, errbuf, sizeof (errbuf)));
+  }
 
-static void *
-pinba_main(void *arg){
-  DEBUG("entering listen-loop..");
-  
-  service_status=1;
-  event_base_dispatch(temp_base);
-  
-  /* unreachable */
-  return NULL;
-}
+  status = bind (fd, ai->ai_addr, ai->ai_addrlen);
+  if (status != 0)
+  {
+    char errbuf[1024];
+    ERROR ("pinba plugin: bind(2) failed: %s",
+        sstrerror (errno, errbuf, sizeof (errbuf)));
+    return (0);
+  }
+
+  s->fd[s->fd_num].fd = fd;
+  s->fd[s->fd_num].events = POLLIN | POLLPRI;
+  s->fd[s->fd_num].revents = 0;
+  s->fd_num++;
+
+  return (0);
+} /* }}} int pb_add_socket */
+
+static pinba_socket_t *pinba_socket_open (const char *node, /* {{{ */
+    int listen_port)
+{
+  pinba_socket_t *s;
+  struct addrinfo *ai_list;
+  struct addrinfo *ai_ptr;
+  struct addrinfo  ai_hints;
+  int status;
+
+  char service[NI_MAXSERV]; /* FIXME */
+  snprintf (service, sizeof (service), "%i", listen_port);
+
+  memset (&ai_hints, 0, sizeof (ai_hints));
+  ai_hints.ai_flags = AI_PASSIVE;
+  ai_hints.ai_family = AF_UNSPEC;
+  ai_hints.ai_socktype = SOCK_DGRAM;
+  ai_hints.ai_addr = NULL;
+  ai_hints.ai_canonname = NULL;
+  ai_hints.ai_next = NULL;
+
+  ai_list = NULL;
+  status = getaddrinfo (node, service,
+      &ai_hints, &ai_list);
+  if (status != 0)
+  {
+    ERROR ("pinba plugin: getaddrinfo(3) failed: %s",
+        gai_strerror (status));
+    return (NULL);
+  }
+  assert (ai_list != NULL);
+
+  s = malloc (sizeof (*s));
+  if (s != NULL)
+  {
+    freeaddrinfo (ai_list);
+    ERROR ("pinba plugin: malloc failed.");
+    return (NULL);
+  }
+  memset (s, 0, sizeof (*s));
+
+  for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next)
+  {
+    status = pb_add_socket (s, ai_ptr);
+    if (status != 0)
+      break;
+  } /* for (ai_list) */
+  
+  freeaddrinfo (ai_list);
+
+  if (s->fd_num < 1)
+  {
+    WARNING ("pinba plugin: Unable to open socket for address %s.", node);
+    sfree (s);
+    s = NULL;
+  }
+
+  return (s);
+} /* }}} pinba_socket_open */
 
-static int
-service_cleanup(){
+static int service_cleanup (void)
+{
   DEBUG("closing socket..");
   if(temp_sock){
     pthread_rwlock_wrlock(&temp_lock);
@@ -382,10 +529,12 @@ service_cleanup(){
   event_base_free(temp_base);
   
   DEBUG("shutting down..");
+
+  return (0);
 }
 
-static int
-service_start(){
+static int service_start(void)
+{
   DEBUG("starting up..");
   
   DEBUG("initializing event..");
@@ -408,8 +557,8 @@ service_start(){
   return 0;
 }
 
-static int
-service_stop(){
+static int service_stop (void)
+{
   pthread_cancel(temp_thrd);
   pthread_join(temp_thrd, NULL);
   service_status=0;
@@ -420,140 +569,35 @@ service_stop(){
   return 0;
 }
 
-static int
-pinba_process_stats_packet (const unsigned char *buf,
-                           int buf_len) {
-  Pinba__Request *request;  
-  
-  request = pinba__request__unpack(NULL, buf_len, buf);
-  
-  if (!request) {
-    return P_FAILURE;
-  } else {
-    service_process_request(request);
-    
-    pinba__request__free_unpacked(request, NULL);
-    
-    return P_SUCCESS;
-  }
-}
+static void service_config (const char *address, unsigned int port) /* {{{ */
+{
+  int need_restart = 0;
 
-static void
-pinba_udp_read_callback_fn (int sock,
-                           short event,
-                           void *arg) {
-  if (event & EV_READ) {
-    int ret;
-    unsigned char buf[PINBA_UDP_BUFFER_SIZE];
-    struct sockaddr_in from;
-    socklen_t fromlen = sizeof(struct sockaddr_in);
-    
-    ret = recvfrom(sock, buf, PINBA_UDP_BUFFER_SIZE-1, MSG_DONTWAIT, (struct sockaddr *)&from, &fromlen);
-    if (ret > 0) {
-      if (pinba_process_stats_packet(buf, ret) != P_SUCCESS) {
-       DEBUG("failed to parse data received from %s", inet_ntoa(from.sin_addr));
-      }
-    } else if (ret < 0) {
-      if (errno == EINTR || errno == EAGAIN || errno == EWOULDBLOCK) {
-       return;
-      }
-      WARNING("recv() failed: %s (%d)", strerror(errno), errno);
-    } else {
-      WARNING("recv() returned 0");
-    }
+  if (address && service_address && (strcmp(service_address, address) != 0))
+  {
+    strset (&service_address, address);
+    need_restart++;
   }
-}
 
-static void
-pinba_socket_free (pinba_socket *socket) {
-  if (!socket) return;
-  
-  if (socket->listen_sock >= 0) {
-    close(socket->listen_sock);
-    socket->listen_sock = -1;
-  }
-  
-  if (socket->accept_event) {
-    event_del(socket->accept_event);
-    free(socket->accept_event);
-    socket->accept_event = NULL;
+  if ((port > 0) && (port < 65536) && (service_port != port))
+  {
+    service_port=port;
+    need_restart++;
   }
-  
-  free(socket);
-}
 
-static pinba_socket *
-pinba_socket_open (const char *ip,
-                  int listen_port) {
-  struct sockaddr_in addr;
-  pinba_socket *s;
-  int sfd, flags, yes = 1;
-  
-  if ((sfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {
-    ERROR("socket() failed: %s (%d)", strerror(errno), errno);
-    return NULL;
-  }
-  
-  if ((flags = fcntl(sfd, F_GETFL, 0)) < 0 || fcntl(sfd, F_SETFL, flags | O_NONBLOCK) < 0) {
-    close(sfd);
-    return NULL;
-  }
-  
-  if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1) {
-    close(sfd);
-    return NULL;
-  }
-  
-  s = (pinba_socket *)calloc(1, sizeof(pinba_socket));
-  if (!s) {
-    return NULL;
-  }
-  s->listen_sock = sfd;
-  
-  memset(&addr, 0, sizeof(addr));
-  
-  addr.sin_family = AF_INET;
-  addr.sin_port = htons(listen_port);
-  addr.sin_addr.s_addr = htonl(INADDR_ANY);
-  
-  if (ip && *ip) {
-    struct in_addr tmp;
-    
-    if (inet_aton(ip, &tmp)) {
-      addr.sin_addr.s_addr = tmp.s_addr;
-    } else {
-      WARNING("inet_aton(%s) failed, listening on ANY IP-address", ip);
-    }
-  }
-  
-  if (bind(s->listen_sock, (struct sockaddr *)&addr, sizeof(addr))) {
-    pinba_socket_free(s);
-    ERROR("bind() failed: %s (%d)", strerror(errno), errno);
-    return NULL;
-  }
-  
-  s->accept_event = (struct event *)calloc(1, sizeof(struct event));
-  if (!s->accept_event) {
-    ERROR("calloc() failed: %s (%d)", strerror(errno), errno);
-    pinba_socket_free(s);
-    return NULL;
+  if(service_status && need_restart)
+  {
+    service_stop();
+    service_start();
   }
-  
-  event_set(s->accept_event, s->listen_sock, EV_READ | EV_PERSIST, pinba_udp_read_callback_fn, s);
-  event_base_set(temp_base, s->accept_event);
-  event_add(s->accept_event, NULL);
-  
-  return s;
-}
+} /* }}} void service_config */
 
 /*
  * Plugin declaration section
  */
 
-#include<stdint.h>
-
-static int
-config_set (char **var, const char *value) {
+static int config_set (char **var, const char *value)
+{
   /* code from nginx plugin for collectd */
   if (*var != NULL) {
     free (*var);
@@ -564,8 +608,8 @@ config_set (char **var, const char *value) {
   else return (0);
 }
 
-static int
-plugin_config (oconfig_item_t *ci) {
+static int plugin_config (oconfig_item_t *ci)
+{
   unsigned int i, o;
   int pinba_port = 0;
   char *pinba_address = NULL;
@@ -637,23 +681,22 @@ plugin_config (oconfig_item_t *ci) {
   service_config(pinba_address, pinba_port);
 } /* int pinba_config */
 
-static int
-plugin_init (void) {
+static int plugin_init (void)
+{
   INFO("Pinba Starting..");
   service_start();
   return 0;
 }
 
-static int
-plugin_shutdown (void) {
+static int plugin_shutdown (void)
+{
   INFO("Pinba Stopping..");
   service_stop();
   service_statnode_free();
   return 0;
 }
 
-static int
-plugin_submit (const char *plugin_instance,
+static int plugin_submit (const char *plugin_instance,
               const char *type,
               const pinba_statres *res) {
   value_t values[6];
@@ -675,25 +718,29 @@ plugin_submit (const char *plugin_instance,
   sstrncpy (vl.type, type, sizeof (vl.type));
   INFO("Pinba Dispatch");
   plugin_dispatch_values (&vl);
+
+  return (0);
 }
 
-static int
-plugin_read (void) {
-  DEBUG("Pinba Read..");
+static int plugin_read (void)
+{
   unsigned int i=0;
   static pinba_statres res;
   
-  for(; i=service_statnode_collect(&res, i); ){
+  while ((i = service_statnode_collect (&res, i)) != 0)
+  {
     plugin_submit(res.name, "pinba_view", &res);
   }
   
   return 0;
 }
 
-void module_register (void) {
+void module_register (void)
+{
   plugin_register_complex_config ("pinba", plugin_config);
   plugin_register_init ("pinba", plugin_init);
   plugin_register_read ("pinba", plugin_read);
   plugin_register_shutdown ("pinba", plugin_shutdown);
 } /* void module_register */
 
+/* vim: set sw=2 sts=2 et fdm=marker : */