Code Review / vpp.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | review | tree
raw | inline | side by side
Stats: prometheus export coverity errors.
[vpp.git] / src / vpp / app / vpp_prometheus_export.c
diff --git a/src/vpp/app/vpp_prometheus_export.c b/src/vpp/app/vpp_prometheus_export.c
index cc85b81..b12ce6f 100644 (file)
--- a/src/vpp/app/vpp_prometheus_export.c
+++ b/src/vpp/app/vpp_prometheus_export.c
@@ -30,11 +30,10 @@
 #include <sys/socket.h>
 #include <vpp-api/client/stat_client.h>
 #include <vlib/vlib.h>
-#include <vpp/stats/stats.h>
 #include <ctype.h>
 
 /* https://github.com/prometheus/prometheus/wiki/Default-port-allocations */
-#define SERVER_PORT 9477
+#define SERVER_PORT 9482
 
 static char *
 prom_string (char *s)
@@ -50,12 +49,22 @@ prom_string (char *s)
 }
 
 static void
-dump_metrics (FILE * stream, stat_segment_cached_pointer_t * cp)
+dump_metrics (FILE * stream, u8 ** patterns)
 {
   stat_segment_data_t *res;
   int i, j, k;
+  static u32 *stats = 0;
+
+retry:
+  res = stat_segment_dump (stats);
+  if (res == 0)
+    {                          /* Memory layout has changed */
+      if (stats)
+       vec_free (stats);
+      stats = stat_segment_ls (patterns);
+      goto retry;
+    }
 
-  res = stat_segment_collect (cp);
   for (i = 0; i < vec_len (res); i++)
     {
       switch (res[i].type)
@@ -93,7 +102,7 @@ dump_metrics (FILE * stream, stat_segment_cached_pointer_t * cp)
                   prom_string (res[i].name), res[i].error_value);
          break;
 
-       case STAT_DIR_TYPE_SCALAR_POINTER:
+       case STAT_DIR_TYPE_SCALAR_INDEX:
          fformat (stream, "# TYPE %s counter\n", prom_string (res[i].name));
          fformat (stream, "%s %.2f\n", prom_string (res[i].name),
                   res[i].scalar_value);
@@ -113,7 +122,7 @@ dump_metrics (FILE * stream, stat_segment_cached_pointer_t * cp)
 #define NOT_FOUND_ERROR "<html><head><title>Document not found</title></head><body><h1>404 - Document not found</h1></body></html>"
 
 static void
-http_handler (FILE * stream, stat_segment_cached_pointer_t * cp)
+http_handler (FILE * stream, u8 ** patterns)
 {
   char status[80] = { 0 };
   if (fgets (status, sizeof (status) - 1, stream) == 0)
@@ -165,7 +174,7 @@ http_handler (FILE * stream, stat_segment_cached_pointer_t * cp)
       return;
     }
   fputs ("HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\n", stream);
-  dump_metrics (stream, cp);
+  dump_metrics (stream, patterns);
 }
 
 static int
@@ -179,6 +188,7 @@ start_listen (u16 port)
   if (listenfd == -1)
     {
       perror ("Failed opening socket");
+      return -1;
     }
 
   int rv =
@@ -186,6 +196,8 @@ start_listen (u16 port)
   if (rv < 0)
     {
       perror ("Failed setsockopt");
+      close (listenfd);
+      return -1;
     }
 
   memset (&serveraddr, 0, sizeof (serveraddr));
@@ -196,11 +208,13 @@ start_listen (u16 port)
   if (bind (listenfd, (struct sockaddr *) &serveraddr, addrlen) < 0)
     {
       fprintf (stderr, "bind() error %s\n", strerror (errno));
+      close (listenfd);
       return -1;
     }
   if (listen (listenfd, 1000000) != 0)
     {
       fprintf (stderr, "listen() error for %s\n", strerror (errno));
+      close (listenfd);
       return -1;
     }
   return listenfd;
@@ -223,10 +237,9 @@ main (int argc, char **argv)
   unformat_input_t _argv, *a = &_argv;
   u8 *stat_segment_name, *pattern = 0, **patterns = 0;
   int rv;
-  void *heap_base;
 
-  heap_base = clib_mem_vm_map ((void *) 0x10000000ULL, 128 << 20);
-  clib_mem_init (heap_base, 128 << 20);
+  /* Allocating 32MB heap */
+  clib_mem_init (0, 32 << 20);
 
   unformat_init_command_line (a, argv);
 
@@ -249,23 +262,18 @@ main (int argc, char **argv)
        }
     }
 
-  rv = stat_segment_connect ((char *) stat_segment_name);
-  if (rv)
+  if (vec_len (patterns) == 0)
     {
-      fformat (stderr, "Couldn't connect to vpp, does %s exist?\n",
-              stat_segment_name);
+      fformat (stderr,
+              "%s: usage [socket-name <name>] <patterns> ...\n", argv[0]);
       exit (1);
     }
 
-  u8 **dir;
-  stat_segment_cached_pointer_t *cp;
-
-  dir = stat_segment_ls (patterns);
-  cp = stat_segment_register (dir);
-  if (!cp)
+  rv = stat_segment_connect ((char *) stat_segment_name);
+  if (rv)
     {
-      fformat (stderr,
-              "Couldn't register required counters with stat segment\n");
+      fformat (stderr, "Couldn't connect to vpp, does %s exist?\n",
+              stat_segment_name);
       exit (1);
     }
 
@@ -284,15 +292,15 @@ main (int argc, char **argv)
        }
       else
        {
-         struct sockaddr_in6 clientaddr;
+         struct sockaddr_in6 clientaddr = { 0 };
          char address[INET6_ADDRSTRLEN];
          socklen_t addrlen;
          getpeername (conn_sock, (struct sockaddr *) &clientaddr, &addrlen);
          if (inet_ntop
              (AF_INET6, &clientaddr.sin6_addr, address, sizeof (address)))
            {
-             printf ("Client address is [%s]:%d\n", address,
-                     ntohs (clientaddr.sin6_port));
+             fprintf (stderr, "Client address is [%s]:%d\n", address,
+                      ntohs (clientaddr.sin6_port));
            }
        }
 
@@ -304,11 +312,12 @@ main (int argc, char **argv)
          continue;
        }
       /* Single reader at the moment */
-      http_handler (stream, cp);
+      http_handler (stream, patterns);
       fclose (stream);
     }
 
   stat_segment_disconnect ();
+  close (fd);
 
   exit (0);
 }
Vector Packet Processing