vnfs/VESreporting_vLB5.0/vpp_measurement_reporter.c

   1
   2 /*************************************************************************//**
   3  *
   4  * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   5  *
   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  *        http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  ****************************************************************************/
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <string.h>
  23 #include <sys/time.h>
  24 #include <math.h>
  25
  26 #include "evel.h"
  27
  28 #define BUFSIZE 128
  29 #define READ_INTERVAL 10
  30
  31 typedef struct dummy_vpp_metrics_struct {
  32   int bytes_in;
  33   int bytes_out;
  34   int packets_in;
  35   int packets_out;
  36 } vpp_metrics_struct;
  37
  38 void read_vpp_metrics(vpp_metrics_struct *, char *);
  39
  40 unsigned long long epoch_start = 0;
  41
  42
  43 #ifdef DOCKER
  44 int measure_traffic()
  45 {
  46
  47   EVEL_ERR_CODES evel_rc = EVEL_SUCCESS;
  48   FILE *fp;
  49   int status;
  50   char count[10];
  51   time_t rawtime;
  52   struct tm * timeinfo;
  53   char period [21];
  54   char cmd [100];
  55   int concurrent_sessions = 0;
  56   int configured_entities = 0;
  57   double mean_request_latency = 0;
  58   double measurement_interval = 1;
  59   double memory_configured = 0;
  60   double memory_used = 0;
  61   int request_rate=0;
  62   char secs [3];
  63   int sec;
  64   double loadavg;
  65
  66   printf("Checking app traffic\n");
  67   time (&rawtime);
  68   timeinfo = localtime (&rawtime);
  69   strftime(period,21,"%d/%b/%Y:%H:%M:",timeinfo);
  70   strftime(secs,3,"%S",timeinfo);
  71   sec = atoi(secs);
  72   if (sec == 0) sec = 59;
  73   sprintf(secs, "%02d", sec);
  74   strncat(period, secs, 21);
  75   // ....x....1....x....2.
  76   // 15/Oct/2016:17:51:19
  77   strcpy(cmd, "sudo docker logs vHello | grep -c ");
  78   strncat(cmd, period, 100);
  79
  80   fp = popen(cmd, "r");
  81   if (fp == NULL) {
  82     EVEL_ERROR("popen failed to execute command");
  83   }
  84
  85   if (fgets(count, 10, fp) != NULL) {
  86     request_rate = atoi(count);
  87     printf("Reporting request rate for second: %s as %d\n", period, request_rate);
  88
  89     }
  90     else {
  91       EVEL_ERROR("New Measurement failed");
  92     }
  93     printf("Processed measurement\n");
  94
  95   status = pclose(fp);
  96   if (status == -1) {
  97     EVEL_ERROR("pclose returned an error");
  98   }
  99   return request_rate;
 100 }
 101
 102 #endif
 103
 104
 105 int main(int argc, char** argv)
 106 {
 107   EVEL_ERR_CODES evel_rc = EVEL_SUCCESS;
 108   EVENT_MEASUREMENT* vpp_m = NULL;
 109   EVENT_HEADER* vpp_m_header = NULL;
 110   int bytes_in_this_round;
 111   int bytes_out_this_round;
 112   int packets_in_this_round;
 113   int packets_out_this_round;
 114   vpp_metrics_struct* last_vpp_metrics = malloc(sizeof(vpp_metrics_struct));
 115   vpp_metrics_struct* curr_vpp_metrics = malloc(sizeof(vpp_metrics_struct));
 116   struct timeval time_val;
 117   //time_t start_epoch;
 118   //time_t last_epoch;
 119   char hostname[BUFSIZE];
 120   char* fqdn = argv[1];
 121   int port = atoi(argv[2]);
 122   char* vnic = argv[3];
 123   char* fqdn2 = NULL;
 124   int port2 = 0;
 125
 126   if(argc == 6)
 127   {
 128      fqdn2 = argv[3];
 129      port2 = atoi(argv[4]);
 130      vnic = argv[5];
 131   }
 132   else
 133      vnic = argv[3];
 134
 135   MEASUREMENT_VNIC_PERFORMANCE * vnic_performance = NULL;
 136
 137   printf("\nVector Packet Processing (VPP) measurement collection\n");
 138   fflush(stdout);
 139
 140   if (!((argc == 6) || (argc == 4)))
 141   {
 142     fprintf(stderr, "Usage: %s <FQDN>|<IP address> <port> <FQDN>|<IP address> <port> <interface> \n", argv[0]);
 143     fprintf(stderr, "OR\n");
 144     fprintf(stderr, "Usage: %s <FQDN>|<IP address> <port> <interface> \n", argv[0]);
 145     exit(-1);
 146   }
 147
 148   srand(time(NULL));
 149
 150   /**************************************************************************/
 151   /* Initialize                                                             */
 152   /**************************************************************************/
 153   if(evel_initialize(fqdn,                         /* FQDN                  */
 154                        port,                         /* Port                  */
 155                        fqdn2,                        /* Backup FQDN           */
 156                        port2,                        /* Backup port           */
 157                        NULL,                         /* optional path         */
 158                        NULL,                         /* optional topic        */
 159                        100,                          /* Ring Buffer size      */
 160                        0,                            /* HTTPS?                */
 161                        NULL,                         /* cert file             */
 162                        NULL,                         /* key  file             */
 163                        NULL,                         /* ca   info             */
 164                        NULL,                         /* ca   file             */
 165                        0,                            /* verify peer           */
 166                        0,                            /* verify host           */
 167                        "sample1",                    /* Username              */
 168                        "sample1",                    /* Password              */
 169                        "sample1",                    /* Username2             */
 170                        "sample1",                    /* Password2             */
 171                        NULL,                         /* Source ip             */
 172                        NULL,                         /* Backup Source IP      */
 173                        EVEL_SOURCE_VIRTUAL_MACHINE,  /* Source type           */
 174                        "vLoadBalancer",              /* Role                  */
 175                        1))                           /* Verbosity             */
 176   {
 177     fprintf(stderr, "\nFailed to initialize the EVEL library!!!\n");
 178     exit(-1);
 179   }
 180   else
 181   {
 182     printf("\nInitialization completed\n");
 183   }
 184
 185   gethostname(hostname, BUFSIZE);
 186   memset(last_vpp_metrics, 0, sizeof(vpp_metrics_struct));
 187   read_vpp_metrics(last_vpp_metrics, vnic);
 188   gettimeofday(&time_val, NULL);
 189   epoch_start = time_val.tv_sec * 1000000 + time_val.tv_usec;
 190   sleep(READ_INTERVAL);
 191
 192   /***************************************************************************/
 193   /* Collect metrics from the VNIC                                           */
 194   /***************************************************************************/
 195   while(1) {
 196     // Read the number of vDNSs currently active
 197     int active_dns = 0;
 198     FILE* fd = fopen("active_dns.txt", "r");
 199     while(!feof(fd)) {
 200       if(fscanf(fd, "%d", &active_dns) != 1) /* Avoid infinite loops if the file doesn't contain exactly one number */
 201         break;
 202     }
 203
 204     if(fclose(fd))  {
 205       printf("Error when closing file\n");
 206       return 1;
 207     }
 208
 209     memset(curr_vpp_metrics, 0, sizeof(vpp_metrics_struct));
 210     read_vpp_metrics(curr_vpp_metrics, vnic);
 211
 212     if(active_dns > 0 && (curr_vpp_metrics->bytes_in - last_vpp_metrics->bytes_in > 0)) {
 213       bytes_in_this_round = (int) round((curr_vpp_metrics->bytes_in - last_vpp_metrics->bytes_in) / active_dns);
 214     }
 215     else {
 216       bytes_in_this_round = 0;
 217     }
 218     if(active_dns > 0 && (curr_vpp_metrics->bytes_out - last_vpp_metrics->bytes_out > 0)) {
 219       bytes_out_this_round = (int) round((curr_vpp_metrics->bytes_out - last_vpp_metrics->bytes_out) / active_dns);
 220     }
 221     else {
 222       bytes_out_this_round = 0;
 223     }
 224     if(active_dns > 0 && (curr_vpp_metrics->packets_in - last_vpp_metrics->packets_in > 0)) {
 225       packets_in_this_round = (int) round((curr_vpp_metrics->packets_in - last_vpp_metrics->packets_in) / active_dns);
 226     }
 227     else {
 228       packets_in_this_round = 0;
 229     }
 230     if(active_dns > 0 && (curr_vpp_metrics->packets_out - last_vpp_metrics->packets_out > 0)) {
 231       packets_out_this_round = (int) round((curr_vpp_metrics->packets_out - last_vpp_metrics->packets_out) / active_dns);
 232     }
 233     else {
 234       packets_out_this_round = 0;
 235     }
 236
 237     vpp_m = evel_new_measurement(READ_INTERVAL,"vLoadBalancer","TrafficStats_1.2.3.4");
 238     vnic_performance = (MEASUREMENT_VNIC_PERFORMANCE *)evel_measurement_new_vnic_performance(vnic, "true");
 239     evel_meas_vnic_performance_add(vpp_m, vnic_performance);
 240
 241     if(vpp_m != NULL) {
 242       printf("New measurement report created...\n");
 243
 244       evel_measurement_type_set(vpp_m, "HTTP request rate");
 245       evel_measurement_request_rate_set(vpp_m, rand()%10000);
 246
 247       evel_vnic_performance_rx_total_pkt_delta_set(vnic_performance, packets_in_this_round);
 248       evel_vnic_performance_tx_total_pkt_delta_set(vnic_performance, packets_out_this_round);
 249
 250       evel_vnic_performance_rx_octets_delta_set(vnic_performance, bytes_in_this_round);
 251       evel_vnic_performance_tx_octets_delta_set(vnic_performance, bytes_out_this_round);
 252
 253       /***************************************************************************/
 254       /* Set parameters in the MEASUREMENT header packet                         */
 255       /***************************************************************************/
 256       struct timeval tv_now;
 257       gettimeofday(&tv_now, NULL);
 258       unsigned long long epoch_now = tv_now.tv_usec + 1000000 * tv_now.tv_sec;
 259
 260       //last_epoch = start_epoch + READ_INTERVAL * 1000000;
 261       vpp_m_header = (EVENT_HEADER *)vpp_m;
 262       //vpp_m_header->start_epoch_microsec = start_epoch;
 263       //vpp_m_header->last_epoch_microsec = last_epoch;
 264       evel_start_epoch_set(&vpp_m->header, epoch_start);
 265       evel_last_epoch_set(&vpp_m->header, epoch_now);
 266       epoch_start = epoch_now;
 267
 268       evel_nfcnamingcode_set(&vpp_m->header, "vVNF");
 269       evel_nfnamingcode_set(&vpp_m->header, "vVNF");
 270       //strcpy(vpp_m_header->reporting_entity_id.value, "No UUID available");
 271       //strcpy(vpp_m_header->reporting_entity_name, hostname);
 272       evel_reporting_entity_name_set(&vpp_m->header, "lbll");
 273       evel_reporting_entity_id_set(&vpp_m->header, "No UUID available");
 274       evel_rc = evel_post_event(vpp_m_header);
 275
 276       if(evel_rc == EVEL_SUCCESS) {
 277         printf("Measurement report correctly sent to the collector!\n");
 278       }
 279       else {
 280         printf("Post failed %d (%s)\n", evel_rc, evel_error_string());
 281       }
 282     }
 283     else {
 284       printf("New measurement report failed (%s)\n", evel_error_string());
 285     }
 286
 287     last_vpp_metrics->bytes_in = curr_vpp_metrics->bytes_in;
 288     last_vpp_metrics->bytes_out = curr_vpp_metrics->bytes_out;
 289     last_vpp_metrics->packets_in = curr_vpp_metrics->packets_in;
 290     last_vpp_metrics->packets_out = curr_vpp_metrics->packets_out;
 291     //gettimeofday(&time_val, NULL);
 292     //start_epoch = time_val.tv_sec * 1000000 + time_val.tv_usec;
 293
 294     sleep(READ_INTERVAL);
 295   }
 296
 297   /***************************************************************************/
 298   /* Terminate                                                               */
 299   /***************************************************************************/
 300   sleep(1);
 301   evel_free_measurement(vpp_m);
 302   free(last_vpp_metrics);
 303   free(curr_vpp_metrics);
 304   evel_terminate();
 305   printf("Terminated\n");
 306
 307   return 0;
 308 }
 309
 310 void read_vpp_metrics(vpp_metrics_struct *vpp_metrics, char *vnic) {
 311   // Define an array of char that contains the parameters of the unix 'cut' command
 312   char* params[] = {"-f3", "-f11", "-f4", "-f12"};
 313   // Define the unix command to execute in order to read metrics from the vNIC
 314   char* cmd_prefix = "sudo cat /proc/net/dev | grep \"";
 315   char* cmd_mid = "\" | tr -s \' \' | cut -d\' \' ";
 316   char cmd[BUFSIZE];
 317   // Define other variables
 318   char buf[BUFSIZE];            /* buffer used to store VPP metrics     */
 319   int temp[] = {0, 0, 0, 0};    /* temp array that contains VPP values  */
 320   FILE *fp;                     /* file descriptor to pipe cmd to shell */
 321   int i;
 322
 323   for(i = 0; i < 4; i++) {
 324     // Clear buffers
 325     memset(buf, 0, BUFSIZE);
 326     memset(cmd, 0, BUFSIZE);
 327     // Build shell command to read metrics from the vNIC
 328     strcat(cmd, cmd_prefix);
 329     strcat(cmd, vnic);
 330     strcat(cmd, cmd_mid);
 331     strcat(cmd, params[i]);
 332
 333     // Open a pipe and read VPP values
 334     if ((fp = popen(cmd, "r")) == NULL) {
 335       printf("Error opening pipe!\n");
 336       return;
 337     }
 338
 339     while (fgets(buf, BUFSIZE, fp) != NULL);
 340     temp[i] = atoi(buf);
 341
 342     if(pclose(fp))  {
 343       printf("Command not found or exited with error status\n");
 344       return;
 345     }
 346   }
 347
 348   // Store metrics read from the vNIC in the struct passed from the main function
 349   vpp_metrics->bytes_in = temp[0];
 350   vpp_metrics->bytes_out = temp[1];
 351   vpp_metrics->packets_in = temp[2];
 352   vpp_metrics->packets_out = temp[3];
 353 }