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         }
  84 }
  85
  86 static struct stats_cons stats_cons[8];
  87 static size_t n_stats_cons = 0;
  88 static uint16_t stats_cons_flags = 0;
  89
  90 static void stats_cons_add(struct stats_cons *sc)
  91 {
  92         if (n_stats_cons == sizeof(stats_cons)/sizeof(stats_cons[0]))
  93                 return;
  94
  95         stats_cons[n_stats_cons++] = *sc;
  96         sc->init();
  97         stats_cons_flags |= sc->flags;
  98 }
  99
 100 static void stats_cons_notify(void)
 101 {
 102         for (size_t i = 0; i < n_stats_cons; ++i) {
 103                 stats_cons[i].notify();
 104         }
 105 }
 106
 107 static void stats_cons_refresh(void)
 108 {
 109         for (size_t i = 0; i < n_stats_cons; ++i) {
 110                 if (stats_cons[i].refresh)
 111                         stats_cons[i].refresh();
 112         }
 113 }
 114
 115 static void stats_cons_finish(void)
 116 {
 117         for (size_t i = 0; i < n_stats_cons; ++i) {
 118                 if (stats_cons[i].finish)
 119                         stats_cons[i].finish();
 120         }
 121 }
 122
 123 static void busy_wait_until(uint64_t deadline)
 124 {
 125         while (rte_rdtsc() < deadline)
 126                 ;
 127 }
 128
 129 static void multiplexed_input_stats(uint64_t deadline)
 130 {
 131         if (deadline)
 132                 input_proc_until(deadline);
 133         else
 134                 input_proc();
 135
 136         if (needs_refresh) {
 137                 needs_refresh = 0;
 138                 stats_cons_refresh();
 139         }
 140
 141         if (rte_atomic32_read(&lsc)) {
 142                 rte_atomic32_dec(&lsc);
 143                 update_link_states();
 144                 stats_cons_refresh();
 145         }
 146 }
 147
 148 static void print_warnings(void)
 149 {
 150         if (get_n_warnings() == -1) {
 151                 plog_info("Warnings disabled\n");
 152         }
 153         else if (get_n_warnings() > 0) {
 154                 int n_print = get_n_warnings() < 5? get_n_warnings(): 5;
 155                 plog_info("Started with %d warnings, last %d warnings: \n", get_n_warnings(), n_print);
 156                 for (int i = -n_print + 1; i <= 0; ++i) {
 157                         plog_info("%s", get_warning(i));
 158                 }
 159         }
 160         else {
 161                 plog_info("Started without warnings\n");
 162         }
 163 }
 164
 165 /* start main loop */
 166 void __attribute__((noreturn)) run(uint32_t flags)
 167 {
 168         uint64_t cur_tsc;
 169         uint64_t next_update;
 170         uint64_t stop_tsc = 0;
 171         int ret = 0;
 172         const uint64_t update_interval_threshold = usec_to_tsc(1);
 173
 174         if (flags & DSF_LISTEN_TCP)
 175                 PROX_PANIC(reg_input_tcp(), "Failed to start listening on TCP port 8474: %s\n", strerror(errno));
 176         if (flags & DSF_LISTEN_UDS)
 177                 PROX_PANIC(reg_input_uds(), "Failed to start listening on UDS /tmp/prox.sock: %s\n", strerror(errno));
 178
 179         if (prox_cfg.use_stats_logger)
 180                 stats_cons_add(stats_cons_log_get());
 181
 182         stats_init(prox_cfg.start_time, prox_cfg.duration_time);
 183         stats_update(STATS_CONS_F_ALL);
 184
 185         switch (prox_cfg.ui) {
 186         case PROX_UI_CURSES:
 187                 reg_input_curses();
 188                 stats_cons_add(&display);
 189                 break;
 190         case PROX_UI_CLI:
 191                 stats_cons_add(stats_cons_cli_get());
 192                 break;
 193         case PROX_UI_NONE:
 194         default:
 195                 break;
 196         }
 197
 198         if (flags & DSF_AUTOSTART)
 199                 start_core_all(-1);
 200         else
 201                 stop_core_all(-1);
 202
 203         cur_tsc = rte_rdtsc();
 204         if (prox_cfg.duration_time != 0) {
 205                 stop_tsc = cur_tsc + sec_to_tsc(prox_cfg.start_time + prox_cfg.duration_time);
 206         }
 207
 208         stats_cons_notify();
 209         stats_cons_refresh();
 210
 211         update_interval = str_to_tsc(prox_cfg.update_interval_str);
 212         next_update = cur_tsc + update_interval;
 213
 214         cmd_rx_tx_info();
 215         print_warnings();
 216
 217         struct task_master *task = (struct task_master *)lcore_cfg[prox_cfg.master].tasks_all[0];
 218         if (handle_ctrl_plane) {
 219                 while (stop_prox == 0) {
 220                         ret = 1;
 221                         // Run ctrl plane for max 10 msec to let screen and keyboard updates
 222                         if (prox_cfg.flags & DSF_CTRL_PLANE_ENABLED) {
 223                                 uint64_t ctrl_plane_update = rte_rdtsc() + msec_to_tsc(10);
 224                                 while ((ret) && (rte_rdtsc() < ctrl_plane_update))
 225                                         ret = handle_ctrl_plane(lcore_cfg[prox_cfg.master].tasks_all[0], NULL, 0);
 226                         }
 227                         multiplexed_input_stats(0);
 228                         if (rte_rdtsc() < next_update)
 229                                 continue;
 230                         next_update += update_interval;
 231                         stats_update(stats_cons_flags);
 232                         stats_cons_notify();
 233
 234                         if (stop_tsc && rte_rdtsc() >= stop_tsc) {
 235                                 stop_prox = 1;
 236                         }
 237                 }
 238         } else {
 239                 while (stop_prox == 0) {
 240
 241                         if (update_interval < update_interval_threshold)
 242                                 busy_wait_until(next_update);
 243                         else
 244                                 multiplexed_input_stats(next_update);
 245
 246                         next_update += update_interval;
 247
 248                         stats_update(stats_cons_flags);
 249                         stats_cons_notify();
 250
 251                         if (stop_tsc && rte_rdtsc() >= stop_tsc) {
 252                                 stop_prox = 1;
 253                         }
 254                 }
 255         }
 256
 257         stats_cons_finish();
 258
 259         if (prox_cfg.flags & DSF_WAIT_ON_QUIT) {
 260                 stop_core_all(-1);
 261         }
 262
 263         if (prox_cfg.logbuf) {
 264                 file_print(prox_cfg.logbuf);
 265         }
 266
 267         display_end();
 268         exit(EXIT_SUCCESS);
 269 }