Fix arp handling when master core is on socket 1
[samplevnf.git] / VNFs / DPPD-PROX / handle_police.c
index 125e8c0..c897bc1 100644 (file)
@@ -44,7 +44,12 @@ struct task_police {
                struct rte_meter_srtcm *sr_flows;
                struct rte_meter_trtcm *tr_flows;
        };
-
+       union {
+#if RTE_VERSION >= RTE_VERSION_NUM(18,5,0,0)
+               struct rte_meter_srtcm_profile sr_profile;
+               struct rte_meter_trtcm_profile tr_profile;
+#endif
+       };
        uint16_t           *user_table;
        enum police_action police_act[3][3];
        uint16_t overhead;
@@ -58,8 +63,12 @@ static uint8_t handle_police(struct task_police *task, struct rte_mbuf *mbuf, ui
        enum rte_meter_color in_color = e_RTE_METER_GREEN;
        enum rte_meter_color out_color;
        uint32_t pkt_len = rte_pktmbuf_pkt_len(mbuf) + task->overhead;
-       out_color = rte_meter_srtcm_color_aware_check(&task->sr_flows[user], tsc, pkt_len, in_color);
 
+#if RTE_VERSION < RTE_VERSION_NUM(18,5,0,0)
+       out_color = rte_meter_srtcm_color_aware_check(&task->sr_flows[user], tsc, pkt_len, in_color);
+#else
+       out_color = rte_meter_srtcm_color_aware_check(&task->sr_flows[user], &task->sr_profile, tsc, pkt_len, in_color);
+#endif
        return task->police_act[in_color][out_color] == ACT_DROP? OUT_DISCARD : 0;
 }
 
@@ -68,7 +77,11 @@ static uint8_t handle_police_tr(struct task_police *task, struct rte_mbuf *mbuf,
        enum rte_meter_color in_color = e_RTE_METER_GREEN;
        enum rte_meter_color out_color;
        uint32_t pkt_len = rte_pktmbuf_pkt_len(mbuf) + task->overhead;
+#if RTE_VERSION < RTE_VERSION_NUM(18,5,0,0)
        out_color = rte_meter_trtcm_color_aware_check(&task->tr_flows[user], tsc, pkt_len, in_color);
+#else
+       out_color = rte_meter_trtcm_color_aware_check(&task->tr_flows[user], &task->tr_profile, tsc, pkt_len, in_color);
+#endif
 
        if (task->runtime_flags  & TASK_MARK) {
 #if RTE_VERSION >= RTE_VERSION_NUM(1,8,0,0)
@@ -214,10 +227,16 @@ static void init_task_police(struct task_base *tbase, struct task_args *targ)
                        .cbs = targ->cbs,
                        .ebs = targ->ebs,
                };
-
+#if RTE_VERSION >= RTE_VERSION_NUM(18,5,0,0)
+               PROX_PANIC(rte_meter_srtcm_profile_config(&task->sr_profile, &params) != 0, "Failed to rte_meter_srtcm_profile_config\n");
+               for (uint32_t i = 0; i < targ->n_flows; ++i) {
+                       PROX_PANIC(rte_meter_srtcm_config(&task->sr_flows[i], &task->sr_profile) != 0, "Failed to rte_meter_srtcm_config");
+               }
+#else
                for (uint32_t i = 0; i < targ->n_flows; ++i) {
                        rte_meter_srtcm_config(&task->sr_flows[i], &params);
                }
+#endif
        }
        else {
                task->tr_flows = prox_zmalloc(targ->n_flows * sizeof(*task->tr_flows), socket_id);
@@ -233,10 +252,17 @@ static void init_task_police(struct task_base *tbase, struct task_args *targ)
                        .cir = targ->cir,
                        .cbs = targ->cbs,
                };
+#if RTE_VERSION >= RTE_VERSION_NUM(18,5,0,0)
+               PROX_PANIC(rte_meter_trtcm_profile_config(&task->tr_profile, &params) != 0, "Failed to rte_meter_srtcm_profile_config\n");
+               for (uint32_t i = 0; i < targ->n_flows; ++i) {
+                       PROX_PANIC(rte_meter_trtcm_config(&task->tr_flows[i], &task->tr_profile) != 0, "Failed to rte_meter_trtcm_config\n");
+               }
+#else
 
                for (uint32_t i = 0; i < targ->n_flows; ++i) {
                        rte_meter_trtcm_config(&task->tr_flows[i], &params);
                }
+#endif
        }
 
        for (uint32_t i = 0; i < 3; ++i) {