Fixed persistant

Signed-off-by: Frey Alfredsson <[email protected]>

Author: freysteinn

xdq-tester/sched_sprio.bpf.c

@@ -108,8 +108,12 @@ void *dequeue_prog(struct dequeue_ctx *ctx)
 		goto err;
 
 	flow->pkts--;
-	if (flow->pkts <= 0 && !flow->persistent)
-		bpf_map_delete_elem(&flow_states, &nt);
+	if (flow->pkts <= 0) {
+		if (!flow->persistent)
+			bpf_map_delete_elem(&flow_states, &nt);
+		else
+			flow->finish_bytes = 0;
+	}
 
 	bpf_printk("DEQUEUE SPRIO with priority %d", prio);
 	return pkt;

xdq-tester/sched_wfq.bpf.c

@@ -113,8 +113,12 @@ void *dequeue_prog(struct dequeue_ctx *ctx)
 		goto err;
 
 	flow->pkts--;
-	if (flow->pkts <= 0 && !flow->persistent)
-		bpf_map_delete_elem(&flow_states, &nt);
+	if (flow->pkts <= 0) {
+		if (!flow->persistent)
+			bpf_map_delete_elem(&flow_states, &nt);
+		else
+			flow->finish_bytes = 0;
+	}
 
 	virtual_time_bytes = prio;