Allow latency thresholds for TST009 testing

[samplevnf.git] / VNFs / DPPD-PROX / run.c

diff --git a/VNFs/DPPD-PROX/run.c b/VNFs/DPPD-PROX/run.c
index bed0c75..c05f0a9 100644 (file)
--- a/VNFs/DPPD-PROX/run.c
+++ b/VNFs/DPPD-PROX/run.c
@@ -237,6 +237,13 @@ void __attribute__((noreturn)) run(uint32_t flags)
                        if (stop_tsc && rte_rdtsc() >= stop_tsc) {
                                stop_prox = 1;
                        }
+                       if ((prox_cfg.heartbeat_tsc) && (prox_cfg.heartbeat_timeout) && (rte_rdtsc() >= prox_cfg.heartbeat_tsc)) {
+                               plog_info("Stopping to handle client as heartbeat timed out\n");
+                               stop_core_all(-1);
+                               stop_handling_client();
+                               req_refresh();
+                               prox_cfg.heartbeat_tsc = 0;
+                       }
                }
        } else {
                while (stop_prox == 0) {
@@ -254,6 +261,13 @@ void __attribute__((noreturn)) run(uint32_t flags)
                        if (stop_tsc && rte_rdtsc() >= stop_tsc) {
                                stop_prox = 1;
                        }
+                       if ((prox_cfg.heartbeat_tsc) && (prox_cfg.heartbeat_timeout) && (rte_rdtsc() >= prox_cfg.heartbeat_tsc)) {
+                               plog_info("Stopping to handle client as heartbeat timed out\n");
+                               stop_core_all(-1);
+                               stop_handling_client();
+                               req_refresh();
+                               prox_cfg.heartbeat_tsc = 0;
+                       }
                }
        }