Add a new program (prog_num = 4) that will not parse packets and will
use the meta data hash to spread/redirect traffic into different cpus.

For the new program we set on bpf_set_link_xdp_fd:
        xdp_flags |= XDP_FLAGS_META_HASH | XDP_FLAGS_META_VLAN;

On mlx5 it will succeed since mlx5 already supports these flags.

The new program will read the value of the hash from the data_meta
pointer from the xdp_md and will use it to compute the destination cpu.

Note: I didn't test this patch to show redirect works with the hash!
I only used it to see that the hash and vlan values are set correctly
by the driver and can be seen by the xdp program.

* I faced some difficulties to read the hash value using the helper
functions defined in the previous patches, but once i used the same logic
with out these functions it worked ! Will have to figure this out later.

Signed-off-by: Saeed Mahameed <sae...@mellanox.com>
---
 samples/bpf/xdp_redirect_cpu_kern.c | 67 +++++++++++++++++++++++++++++
 samples/bpf/xdp_redirect_cpu_user.c |  7 +++
 2 files changed, 74 insertions(+)

diff --git a/samples/bpf/xdp_redirect_cpu_kern.c 
b/samples/bpf/xdp_redirect_cpu_kern.c
index 303e9e7161f3..d6b3f55f342a 100644
--- a/samples/bpf/xdp_redirect_cpu_kern.c
+++ b/samples/bpf/xdp_redirect_cpu_kern.c
@@ -376,6 +376,73 @@ int  xdp_prognum3_proto_separate(struct xdp_md *ctx)
        return bpf_redirect_map(&cpu_map, cpu_dest, 0);
 }
 
+#if 0
+xdp_md_info_arr mdi = {
+       [XDP_DATA_META_HASH] = {.offset = 0, .present = 1},
+       [XDP_DATA_META_VLAN] = {.offset = sizeof(struct xdp_md_hash), .present 
= 1},
+};
+#endif
+
+SEC("xdp_cpu_map4_hash_separate")
+int  xdp_prognum4_hash_separate(struct xdp_md *ctx)
+{
+       void *data_meta = (void *)(long)ctx->data_meta;
+       void *data_end  = (void *)(long)ctx->data_end;
+       void *data      = (void *)(long)ctx->data;
+       struct xdp_md_hash *hash;
+       struct xdp_md_vlan *vlan;
+       struct datarec *rec;
+       u32 cpu_dest = 0;
+       u32 cpu_idx = 0;
+       u32 *cpu_lookup;
+       u32 key = 0;
+
+       /* Count RX packet in map */
+       rec = bpf_map_lookup_elem(&rx_cnt, &key);
+       if (!rec)
+               return XDP_ABORTED;
+       rec->processed++;
+
+       /* for some reason this code fails to be verified */
+#if 0
+       hash = xdp_data_meta_get_hash(mdi, data_meta);
+       if (hash + 1 > data)
+               return XDP_ABORTED;
+
+       vlan = xdp_data_meta_get_vlan(mdi, data_meta);
+       if (vlan + 1 > data)
+               return XDP_ABORTED;
+#endif
+
+       /* Work around for the above code */
+       hash = data_meta; /* since we know hash will appear first */
+        if (hash + 1 > data)
+               return XDP_ABORTED;
+
+#if 0
+       // Just for testing
+       /* We know that vlan will appear after the hash */
+       vlan = (void *)((char *)data_meta + sizeof(*hash));
+       if (vlan + 1 > data) {
+               return XDP_ABORTED;
+       }
+#endif
+
+       cpu_idx = reciprocal_scale(hash->hash, MAX_CPUS);
+
+       cpu_lookup = bpf_map_lookup_elem(&cpus_available, &cpu_idx);
+       if (!cpu_lookup)
+               return XDP_ABORTED;
+       cpu_dest = *cpu_lookup;
+
+       if (cpu_dest >= MAX_CPUS) {
+               rec->issue++;
+               return XDP_ABORTED;
+       }
+
+       return bpf_redirect_map(&cpu_map, cpu_dest, 0);
+}
+
 SEC("xdp_cpu_map4_ddos_filter_pktgen")
 int  xdp_prognum4_ddos_filter_pktgen(struct xdp_md *ctx)
 {
diff --git a/samples/bpf/xdp_redirect_cpu_user.c 
b/samples/bpf/xdp_redirect_cpu_user.c
index f6efaefd485b..3429215d5a7b 100644
--- a/samples/bpf/xdp_redirect_cpu_user.c
+++ b/samples/bpf/xdp_redirect_cpu_user.c
@@ -679,6 +679,13 @@ int main(int argc, char **argv)
                return EXIT_FAIL_OPTION;
        }
 
+       /*
+        * prog_num 4 requires xdp meta data hash
+        * Vlan is not required but added just for testing..
+        */
+       if (prog_num == 4)
+               xdp_flags |= XDP_FLAGS_META_HASH | XDP_FLAGS_META_VLAN;
+
        /* Remove XDP program when program is interrupted */
        signal(SIGINT, int_exit);
 
-- 
2.17.0

Reply via email to