src/vpp-api/client/stat_client.c

   1 /*
   2  *------------------------------------------------------------------
   3  * stat_client.c - Library for access to VPP statistics segment
   4  *
   5  * Copyright (c) 2018 Cisco and/or its affiliates.
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at:
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *------------------------------------------------------------------
  18  */
  19
  20 #include <stdio.h>
  21 #include <errno.h>
  22 #include <sys/types.h>
  23 #include <sys/socket.h>
  24 #include <sys/un.h>
  25 #include <stdbool.h>
  26 #include <sys/stat.h>
  27 #include <regex.h>
  28 #include <assert.h>
  29 #include <vppinfra/vec.h>
  30 #include <vppinfra/lock.h>
  31 #include <stdatomic.h>
  32 #include <vpp/stats/stat_segment.h>
  33 #include <vpp-api/client/stat_client.h>
  34
  35 stat_client_main_t stat_client_main;
  36
  37 stat_client_main_t *
  38 stat_client_get (void)
  39 {
  40   stat_client_main_t *sm;
  41   sm = (stat_client_main_t *) malloc (sizeof (stat_client_main_t));
  42   clib_memset (sm, 0, sizeof (stat_client_main_t));
  43   return sm;
  44 }
  45
  46 void
  47 stat_client_free (stat_client_main_t * sm)
  48 {
  49   free (sm);
  50 }
  51
  52 static int
  53 recv_fd (int sock)
  54 {
  55   struct msghdr msg = { 0 };
  56   struct cmsghdr *cmsg;
  57   int fd = -1;
  58   char iobuf[1];
  59   struct iovec io = {.iov_base = iobuf,.iov_len = sizeof (iobuf) };
  60   union
  61   {
  62     char buf[CMSG_SPACE (sizeof (fd))];
  63     struct cmsghdr align;
  64   } u;
  65   msg.msg_iov = &io;
  66   msg.msg_iovlen = 1;
  67   msg.msg_control = u.buf;
  68   msg.msg_controllen = sizeof (u.buf);
  69
  70   ssize_t size;
  71   if ((size = recvmsg (sock, &msg, 0)) < 0)
  72     {
  73       perror ("recvmsg failed");
  74       return -1;
  75     }
  76   cmsg = CMSG_FIRSTHDR (&msg);
  77   if (cmsg && cmsg->cmsg_level == SOL_SOCKET && cmsg->cmsg_type == SCM_RIGHTS)
  78     {
  79       memmove (&fd, CMSG_DATA (cmsg), sizeof (fd));
  80     }
  81   return fd;
  82 }
  83
  84 static stat_segment_directory_entry_t *
  85 get_stat_vector_r (stat_client_main_t * sm)
  86 {
  87   ASSERT (sm->shared_header);
  88   return stat_segment_adjust (sm,
  89                               (void *) sm->shared_header->directory_vector);
  90 }
  91
  92 int
  93 stat_segment_connect_r (const char *socket_name, stat_client_main_t * sm)
  94 {
  95   int mfd = -1;
  96   int sock;
  97
  98   clib_memset (sm, 0, sizeof (*sm));
  99   if ((sock = socket (AF_UNIX, SOCK_SEQPACKET, 0)) < 0)
 100     {
 101       perror ("Stat client couldn't open socket");
 102       return -1;
 103     }
 104
 105   struct sockaddr_un un = { 0 };
 106   un.sun_family = AF_UNIX;
 107   strncpy ((char *) un.sun_path, socket_name, sizeof (un.sun_path) - 1);
 108   if (connect (sock, (struct sockaddr *) &un, sizeof (struct sockaddr_un)) <
 109       0)
 110     {
 111       close (sock);
 112       return -2;
 113     }
 114
 115   if ((mfd = recv_fd (sock)) < 0)
 116     {
 117       close (sock);
 118       fprintf (stderr, "Receiving file descriptor failed\n");
 119       return -3;
 120     }
 121   close (sock);
 122
 123   /* mmap shared memory segment. */
 124   void *memaddr;
 125   struct stat st = { 0 };
 126
 127   if (fstat (mfd, &st) == -1)
 128     {
 129       close (mfd);
 130       perror ("mmap fstat failed");
 131       return -4;
 132     }
 133   if ((memaddr =
 134        mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, mfd, 0)) == MAP_FAILED)
 135     {
 136       close (mfd);
 137       perror ("mmap map failed");
 138       return -5;
 139     }
 140
 141   close (mfd);
 142   sm->memory_size = st.st_size;
 143   sm->shared_header = memaddr;
 144   sm->directory_vector =
 145     stat_segment_adjust (sm, (void *) sm->shared_header->directory_vector);
 146
 147   return 0;
 148 }
 149
 150 int
 151 stat_segment_connect (const char *socket_name)
 152 {
 153   stat_client_main_t *sm = &stat_client_main;
 154   return stat_segment_connect_r (socket_name, sm);
 155 }
 156
 157 void
 158 stat_segment_disconnect_r (stat_client_main_t * sm)
 159 {
 160   munmap (sm->shared_header, sm->memory_size);
 161   return;
 162 }
 163
 164 void
 165 stat_segment_disconnect (void)
 166 {
 167   stat_client_main_t *sm = &stat_client_main;
 168   return stat_segment_disconnect_r (sm);
 169 }
 170
 171 double
 172 stat_segment_heartbeat_r (stat_client_main_t * sm)
 173 {
 174   stat_segment_access_t sa;
 175   stat_segment_directory_entry_t *ep;
 176
 177   /* Has directory been updated? */
 178   if (sm->shared_header->epoch != sm->current_epoch)
 179     return 0;
 180   if (stat_segment_access_start (&sa, sm))
 181     return 0;
 182   ep = vec_elt_at_index (sm->directory_vector, STAT_COUNTER_HEARTBEAT);
 183   if (!stat_segment_access_end (&sa, sm))
 184     return 0.0;
 185   return ep->value;
 186 }
 187
 188 double
 189 stat_segment_heartbeat (void)
 190 {
 191   stat_client_main_t *sm = &stat_client_main;
 192   return stat_segment_heartbeat_r (sm);
 193 }
 194
 195 #define stat_vec_dup(S,V)                             \
 196   ({                                                  \
 197   __typeof__ ((V)[0]) * _v(v) = 0;                    \
 198   if (V && ((void *)V > (void *)S->shared_header) &&  \
 199       (((void*)V + vec_bytes(V)) <                    \
 200        ((void *)S->shared_header + S->memory_size)))  \
 201     _v(v) = vec_dup(V);                               \
 202    _v(v);                                             \
 203 })
 204
 205 static counter_t *
 206 stat_vec_simple_init (counter_t c)
 207 {
 208   counter_t *v = 0;
 209   vec_add1 (v, c);
 210   return v;
 211 }
 212
 213 static vlib_counter_t *
 214 stat_vec_combined_init (vlib_counter_t c)
 215 {
 216   vlib_counter_t *v = 0;
 217   vec_add1 (v, c);
 218   return v;
 219 }
 220
 221 /*
 222  * If index2 is specified copy out the column (the indexed value across all
 223  * threads), otherwise copy out all values.
 224  */
 225 static stat_segment_data_t
 226 copy_data (stat_segment_directory_entry_t *ep, u32 index2, char *name,
 227            stat_client_main_t *sm)
 228 {
 229   stat_segment_data_t result = { 0 };
 230   int i;
 231   vlib_counter_t **combined_c;  /* Combined counter */
 232   counter_t **simple_c;         /* Simple counter */
 233   uint64_t *error_vector;
 234
 235   assert (sm->shared_header);
 236
 237   result.type = ep->type;
 238   result.name = strdup (name ? name : ep->name);
 239
 240   switch (ep->type)
 241     {
 242     case STAT_DIR_TYPE_SCALAR_INDEX:
 243       result.scalar_value = ep->value;
 244       break;
 245
 246     case STAT_DIR_TYPE_COUNTER_VECTOR_SIMPLE:
 247       simple_c = stat_segment_adjust (sm, ep->data);
 248       result.simple_counter_vec = stat_vec_dup (sm, simple_c);
 249       for (i = 0; i < vec_len (simple_c); i++)
 250         {
 251           counter_t *cb = stat_segment_adjust (sm, simple_c[i]);
 252           if (index2 != ~0)
 253             result.simple_counter_vec[i] = stat_vec_simple_init (cb[index2]);
 254           else
 255             result.simple_counter_vec[i] = stat_vec_dup (sm, cb);
 256         }
 257       break;
 258
 259     case STAT_DIR_TYPE_COUNTER_VECTOR_COMBINED:
 260       combined_c = stat_segment_adjust (sm, ep->data);
 261       result.combined_counter_vec = stat_vec_dup (sm, combined_c);
 262       for (i = 0; i < vec_len (combined_c); i++)
 263         {
 264           vlib_counter_t *cb = stat_segment_adjust (sm, combined_c[i]);
 265           if (index2 != ~0)
 266             result.combined_counter_vec[i] =
 267               stat_vec_combined_init (cb[index2]);
 268           else
 269             result.combined_counter_vec[i] = stat_vec_dup (sm, cb);
 270         }
 271       break;
 272
 273     case STAT_DIR_TYPE_ERROR_INDEX:
 274       /* Gather errors from all threads into a vector */
 275       error_vector =
 276         stat_segment_adjust (sm, (void *) sm->shared_header->error_vector);
 277       vec_validate (result.error_vector, vec_len (error_vector) - 1);
 278       for (i = 0; i < vec_len (error_vector); i++)
 279         {
 280           counter_t *cb = stat_segment_adjust (sm, (void *) error_vector[i]);
 281           result.error_vector[i] = cb[ep->index];
 282         }
 283       break;
 284
 285     case STAT_DIR_TYPE_NAME_VECTOR:
 286       {
 287         uint8_t **name_vector = stat_segment_adjust (sm, ep->data);
 288         result.name_vector = stat_vec_dup (sm, name_vector);
 289         for (i = 0; i < vec_len (name_vector); i++)
 290           {
 291             u8 *name = stat_segment_adjust (sm, name_vector[i]);
 292             result.name_vector[i] = stat_vec_dup (sm, name);
 293           }
 294       }
 295       break;
 296
 297     case STAT_DIR_TYPE_SYMLINK:
 298       /* Gather info from all threads into a vector */
 299       {
 300         stat_segment_directory_entry_t *ep2;
 301         ep2 = vec_elt_at_index (sm->directory_vector, ep->index1);
 302         return copy_data (ep2, ep->index2, ep->name, sm);
 303       }
 304
 305     case STAT_DIR_TYPE_EMPTY:
 306       break;
 307
 308     default:
 309       fprintf (stderr, "Unknown type: %d\n", ep->type);
 310     }
 311   return result;
 312 }
 313
 314 void
 315 stat_segment_data_free (stat_segment_data_t * res)
 316 {
 317   int i, j;
 318   for (i = 0; i < vec_len (res); i++)
 319     {
 320       switch (res[i].type)
 321         {
 322         case STAT_DIR_TYPE_COUNTER_VECTOR_SIMPLE:
 323           for (j = 0; j < vec_len (res[i].simple_counter_vec); j++)
 324             vec_free (res[i].simple_counter_vec[j]);
 325           vec_free (res[i].simple_counter_vec);
 326           break;
 327         case STAT_DIR_TYPE_COUNTER_VECTOR_COMBINED:
 328           for (j = 0; j < vec_len (res[i].combined_counter_vec); j++)
 329             vec_free (res[i].combined_counter_vec[j]);
 330           vec_free (res[i].combined_counter_vec);
 331           break;
 332         case STAT_DIR_TYPE_NAME_VECTOR:
 333           for (j = 0; j < vec_len (res[i].name_vector); j++)
 334             vec_free (res[i].name_vector[j]);
 335           vec_free (res[i].name_vector);
 336           break;
 337         case STAT_DIR_TYPE_ERROR_INDEX:
 338           vec_free (res[i].error_vector);
 339           break;
 340         case STAT_DIR_TYPE_SCALAR_INDEX:
 341         case STAT_DIR_TYPE_EMPTY:
 342           break;
 343         default:
 344           assert (0);
 345         }
 346       free (res[i].name);
 347     }
 348   vec_free (res);
 349 }
 350
 351 uint32_t *
 352 stat_segment_ls_r (uint8_t ** patterns, stat_client_main_t * sm)
 353 {
 354   stat_segment_access_t sa;
 355
 356   uint32_t *dir = 0;
 357   regex_t regex[vec_len (patterns)];
 358
 359   int i, j;
 360   for (i = 0; i < vec_len (patterns); i++)
 361     {
 362       int rv = regcomp (&regex[i], (const char *) patterns[i], 0);
 363       if (rv)
 364         {
 365           fprintf (stderr, "Could not compile regex %s\n", patterns[i]);
 366           return dir;
 367         }
 368     }
 369
 370   if (stat_segment_access_start (&sa, sm))
 371     return 0;
 372
 373   stat_segment_directory_entry_t *counter_vec = get_stat_vector_r (sm);
 374   for (j = 0; j < vec_len (counter_vec); j++)
 375     {
 376       for (i = 0; i < vec_len (patterns); i++)
 377         {
 378           int rv = regexec (&regex[i], counter_vec[j].name, 0, NULL, 0);
 379           if (rv == 0)
 380             {
 381               vec_add1 (dir, j);
 382               break;
 383             }
 384         }
 385       if (vec_len (patterns) == 0)
 386         vec_add1 (dir, j);
 387     }
 388
 389   for (i = 0; i < vec_len (patterns); i++)
 390     regfree (&regex[i]);
 391
 392   if (!stat_segment_access_end (&sa, sm))
 393     {
 394       /* Failed, clean up */
 395       vec_free (dir);
 396       return 0;
 397
 398     }
 399
 400   /* Update last version */
 401   sm->current_epoch = sa.epoch;
 402   return dir;
 403 }
 404
 405 uint32_t *
 406 stat_segment_ls (uint8_t ** patterns)
 407 {
 408   stat_client_main_t *sm = &stat_client_main;
 409   return stat_segment_ls_r ((uint8_t **) patterns, sm);
 410 }
 411
 412 stat_segment_data_t *
 413 stat_segment_dump_r (uint32_t * stats, stat_client_main_t * sm)
 414 {
 415   int i;
 416   stat_segment_directory_entry_t *ep;
 417   stat_segment_data_t *res = 0;
 418   stat_segment_access_t sa;
 419
 420   /* Has directory been update? */
 421   if (sm->shared_header->epoch != sm->current_epoch)
 422     return 0;
 423
 424   if (stat_segment_access_start (&sa, sm))
 425     return 0;
 426
 427   for (i = 0; i < vec_len (stats); i++)
 428     {
 429       /* Collect counter */
 430       ep = vec_elt_at_index (sm->directory_vector, stats[i]);
 431       vec_add1 (res, copy_data (ep, ~0, 0, sm));
 432     }
 433
 434   if (stat_segment_access_end (&sa, sm))
 435     return res;
 436
 437   fprintf (stderr, "Epoch changed while reading, invalid results\n");
 438   // TODO increase counter
 439   return 0;
 440 }
 441
 442 stat_segment_data_t *
 443 stat_segment_dump (uint32_t * stats)
 444 {
 445   stat_client_main_t *sm = &stat_client_main;
 446   return stat_segment_dump_r (stats, sm);
 447 }
 448
 449 /* Wrapper for accessing vectors from other languages */
 450 int
 451 stat_segment_vec_len (void *vec)
 452 {
 453   return vec_len (vec);
 454 }
 455
 456 void
 457 stat_segment_vec_free (void *vec)
 458 {
 459   vec_free (vec);
 460 }
 461
 462 /* Create a vector from a string (or add to existing) */
 463 uint8_t **
 464 stat_segment_string_vector (uint8_t ** string_vector, const char *string)
 465 {
 466   uint8_t *name = 0;
 467   size_t len = strlen (string);
 468
 469   vec_validate_init_c_string (name, string, len);
 470   vec_add1 (string_vector, name);
 471   return string_vector;
 472 }
 473
 474 stat_segment_data_t *
 475 stat_segment_dump_entry_r (uint32_t index, stat_client_main_t * sm)
 476 {
 477   stat_segment_directory_entry_t *ep;
 478   stat_segment_data_t *res = 0;
 479   stat_segment_access_t sa;
 480
 481   /* Has directory been update? */
 482   if (sm->shared_header->epoch != sm->current_epoch)
 483     return 0;
 484
 485   if (stat_segment_access_start (&sa, sm))
 486     return 0;
 487
 488   /* Collect counter */
 489   ep = vec_elt_at_index (sm->directory_vector, index);
 490   vec_add1 (res, copy_data (ep, ~0, 0, sm));
 491
 492   if (stat_segment_access_end (&sa, sm))
 493     return res;
 494   return 0;
 495 }
 496
 497 stat_segment_data_t *
 498 stat_segment_dump_entry (uint32_t index)
 499 {
 500   stat_client_main_t *sm = &stat_client_main;
 501   return stat_segment_dump_entry_r (index, sm);
 502 }
 503
 504 char *
 505 stat_segment_index_to_name_r (uint32_t index, stat_client_main_t * sm)
 506 {
 507   stat_segment_directory_entry_t *ep;
 508   stat_segment_access_t sa;
 509   stat_segment_directory_entry_t *vec;
 510
 511   /* Has directory been update? */
 512   if (sm->shared_header->epoch != sm->current_epoch)
 513     return 0;
 514   if (stat_segment_access_start (&sa, sm))
 515     return 0;
 516   vec = get_stat_vector_r (sm);
 517   ep = vec_elt_at_index (vec, index);
 518   if (!stat_segment_access_end (&sa, sm))
 519     return 0;
 520   return strdup (ep->name);
 521 }
 522
 523 char *
 524 stat_segment_index_to_name (uint32_t index)
 525 {
 526   stat_client_main_t *sm = &stat_client_main;
 527   return stat_segment_index_to_name_r (index, sm);
 528 }
 529
 530 uint64_t
 531 stat_segment_version_r (stat_client_main_t * sm)
 532 {
 533   ASSERT (sm->shared_header);
 534   return sm->shared_header->version;
 535 }
 536
 537 uint64_t
 538 stat_segment_version (void)
 539 {
 540   stat_client_main_t *sm = &stat_client_main;
 541   return stat_segment_version_r (sm);
 542 }
 543
 544 /*
 545  * fd.io coding-style-patch-verification: ON
 546  *
 547  * Local Variables:
 548  * eval: (c-set-style "gnu")
 549  * End:
 550  */