VNFs/DPPD-PROX/run.c

   1 /*
   2 // Copyright (c) 2010-2017 Intel Corporation
   3 //
   4 // Licensed under the Apache License, Version 2.0 (the "License");
   5 // you may not use this file except in compliance with the License.
   6 // You may obtain a copy of the License at
   7 //
   8 //     http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 // Unless required by applicable law or agreed to in writing, software
  11 // distributed under the License is distributed on an "AS IS" BASIS,
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 // See the License for the specific language governing permissions and
  14 // limitations under the License.
  15 */
  16
  17 #include <inttypes.h>
  18 #include <string.h>
  19
  20 #include <rte_launch.h>
  21 #include <rte_cycles.h>
  22 #include <rte_atomic.h>
  23
  24 #include "run.h"
  25 #include "prox_cfg.h"
  26 #include "prox_port_cfg.h"
  27 #include "quit.h"
  28 #include "commands.h"
  29 #include "main.h"
  30 #include "log.h"
  31 #include "display.h"
  32 #include "stats.h"
  33 #include "stats_cons.h"
  34 #include "stats_cons_log.h"
  35 #include "stats_cons_cli.h"
  36
  37 #include "input.h"
  38 #include "input_curses.h"
  39 #include "input_conn.h"
  40 #include "handle_master.h"
  41
  42 static int needs_refresh;
  43 static uint64_t update_interval;
  44 static int stop_prox = 0; /* set to 1 to stop prox */
  45
  46 void set_update_interval(uint32_t msec)
  47 {
  48         update_interval = msec_to_tsc(msec);
  49 }
  50
  51 void req_refresh(void)
  52 {
  53         needs_refresh = 1;
  54 }
  55
  56 void quit(void)
  57 {
  58         static rte_atomic32_t already_leaving = RTE_ATOMIC32_INIT(0);
  59         if (!rte_atomic32_test_and_set(&already_leaving))
  60                 return;
  61
  62         plog_info("Leaving...\n");
  63         if (lcore_cfg == NULL)
  64                 exit(EXIT_SUCCESS);
  65         stop_core_all(-1);
  66         stop_prox = 1;
  67 }
  68
  69 static void update_link_states(void)
  70 {
  71         struct prox_port_cfg *port_cfg;
  72         struct rte_eth_link link;
  73
  74         for (uint8_t portid = 0; portid < PROX_MAX_PORTS; ++portid) {
  75                 if (!prox_port_cfg[portid].active) {
  76                         continue;
  77                 }
  78
  79                 port_cfg  = &prox_port_cfg[portid];
  80                 rte_eth_link_get_nowait(portid, &link);
  81                 port_cfg->link_up = link.link_status;
  82                 port_cfg->link_speed = link.link_speed;
  83                 plog_info("Link speed now %d Mbps\n", port_cfg->link_speed);
  84         }
  85 }
  86
  87 static struct stats_cons stats_cons[8];
  88 static size_t n_stats_cons = 0;
  89 static uint16_t stats_cons_flags = 0;
  90
  91 static void stats_cons_add(struct stats_cons *sc)
  92 {
  93         if (n_stats_cons == sizeof(stats_cons)/sizeof(stats_cons[0]))
  94                 return;
  95
  96         stats_cons[n_stats_cons++] = *sc;
  97         sc->init();
  98         stats_cons_flags |= sc->flags;
  99 }
 100
 101 static void stats_cons_notify(void)
 102 {
 103         for (size_t i = 0; i < n_stats_cons; ++i) {
 104                 stats_cons[i].notify();
 105         }
 106 }
 107
 108 static void stats_cons_refresh(void)
 109 {
 110         for (size_t i = 0; i < n_stats_cons; ++i) {
 111                 if (stats_cons[i].refresh)
 112                         stats_cons[i].refresh();
 113         }
 114 }
 115
 116 static void stats_cons_finish(void)
 117 {
 118         for (size_t i = 0; i < n_stats_cons; ++i) {
 119                 if (stats_cons[i].finish)
 120                         stats_cons[i].finish();
 121         }
 122 }
 123
 124 static void busy_wait_until(uint64_t deadline)
 125 {
 126         while (rte_rdtsc() < deadline)
 127                 ;
 128 }
 129
 130 static void multiplexed_input_stats(uint64_t deadline)
 131 {
 132         if (deadline)
 133                 input_proc_until(deadline);
 134         else
 135                 input_proc();
 136
 137         if (needs_refresh) {
 138                 needs_refresh = 0;
 139                 stats_cons_refresh();
 140         }
 141
 142         if (rte_atomic32_read(&lsc)) {
 143                 rte_atomic32_dec(&lsc);
 144                 update_link_states();
 145                 stats_cons_refresh();
 146         }
 147 }
 148
 149 static void print_warnings(void)
 150 {
 151         if (get_n_warnings() == -1) {
 152                 plog_info("Warnings disabled\n");
 153         }
 154         else if (get_n_warnings() > 0) {
 155                 int n_print = get_n_warnings() < 5? get_n_warnings(): 5;
 156                 plog_info("Started with %d warnings, last %d warnings: \n", get_n_warnings(), n_print);
 157                 for (int i = -n_print + 1; i <= 0; ++i) {
 158                         plog_info("%s", get_warning(i));
 159                 }
 160         }
 161         else {
 162                 plog_info("Started without warnings\n");
 163         }
 164 }
 165
 166 /* start main loop */
 167 void __attribute__((noreturn)) run(uint32_t flags)
 168 {
 169         uint64_t cur_tsc;
 170         uint64_t next_update;
 171         uint64_t stop_tsc = 0;
 172         int ret = 0;
 173         const uint64_t update_interval_threshold = usec_to_tsc(1);
 174
 175         if (flags & DSF_LISTEN_TCP)
 176                 PROX_PANIC(reg_input_tcp(), "Failed to start listening on TCP port 8474: %s\n", strerror(errno));
 177         if (flags & DSF_LISTEN_UDS)
 178                 PROX_PANIC(reg_input_uds(), "Failed to start listening on UDS /tmp/prox.sock: %s\n", strerror(errno));
 179
 180         if (prox_cfg.use_stats_logger)
 181                 stats_cons_add(stats_cons_log_get());
 182
 183         stats_init(prox_cfg.start_time, prox_cfg.duration_time);
 184         stats_update(STATS_CONS_F_ALL);
 185
 186         switch (prox_cfg.ui) {
 187         case PROX_UI_CURSES:
 188                 reg_input_curses();
 189                 stats_cons_add(&display);
 190                 break;
 191         case PROX_UI_CLI:
 192                 stats_cons_add(stats_cons_cli_get());
 193                 break;
 194         case PROX_UI_NONE:
 195         default:
 196                 break;
 197         }
 198
 199         if (flags & DSF_AUTOSTART)
 200                 start_core_all(-1);
 201         else
 202                 stop_core_all(-1);
 203
 204         cur_tsc = rte_rdtsc();
 205         if (prox_cfg.duration_time != 0) {
 206                 stop_tsc = cur_tsc + sec_to_tsc(prox_cfg.start_time + prox_cfg.duration_time);
 207         }
 208
 209         stats_cons_notify();
 210         stats_cons_refresh();
 211
 212         update_interval = str_to_tsc(prox_cfg.update_interval_str);
 213         next_update = cur_tsc + update_interval;
 214
 215         cmd_rx_tx_info();
 216         print_warnings();
 217
 218         struct task_master *task = (struct task_master *)lcore_cfg[prox_cfg.master].tasks_all[0];
 219         if (handle_ctrl_plane) {
 220                 while (stop_prox == 0) {
 221                         ret = 1;
 222                         // Run ctrl plane for max 10 msec to let screen and keyboard updates
 223                         if (prox_cfg.flags & DSF_CTRL_PLANE_ENABLED) {
 224                                 uint64_t ctrl_plane_update = rte_rdtsc() + msec_to_tsc(10);
 225                                 while ((ret) && (rte_rdtsc() < ctrl_plane_update))
 226                                         ret = handle_ctrl_plane(lcore_cfg[prox_cfg.master].tasks_all[0], NULL, 0);
 227                         }
 228                         multiplexed_input_stats(0);
 229                         if (rte_rdtsc() < next_update)
 230                                 continue;
 231                         next_update += update_interval;
 232                         stats_update(stats_cons_flags);
 233                         stats_cons_notify();
 234
 235                         if (stop_tsc && rte_rdtsc() >= stop_tsc) {
 236                                 stop_prox = 1;
 237                         }
 238                 }
 239         } else {
 240                 while (stop_prox == 0) {
 241
 242                         if (update_interval < update_interval_threshold)
 243                                 busy_wait_until(next_update);
 244                         else
 245                                 multiplexed_input_stats(next_update);
 246
 247                         next_update += update_interval;
 248
 249                         stats_update(stats_cons_flags);
 250                         stats_cons_notify();
 251
 252                         if (stop_tsc && rte_rdtsc() >= stop_tsc) {
 253                                 stop_prox = 1;
 254                         }
 255                 }
 256         }
 257
 258         stats_cons_finish();
 259
 260         if (prox_cfg.flags & DSF_WAIT_ON_QUIT) {
 261                 stop_core_all(-1);
 262         }
 263
 264         if (prox_cfg.logbuf) {
 265                 file_print(prox_cfg.logbuf);
 266         }
 267
 268         display_end();
 269         exit(EXIT_SUCCESS);
 270 }