1*215249f6SDave Marchevsky // SPDX-License-Identifier: GPL-2.0
2*215249f6SDave Marchevsky /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
3*215249f6SDave Marchevsky
4*215249f6SDave Marchevsky #include <vmlinux.h>
5*215249f6SDave Marchevsky #include <bpf/bpf_tracing.h>
6*215249f6SDave Marchevsky #include <bpf/bpf_helpers.h>
7*215249f6SDave Marchevsky #include <bpf/bpf_core_read.h>
8*215249f6SDave Marchevsky #include "bpf_experimental.h"
9*215249f6SDave Marchevsky
10*215249f6SDave Marchevsky struct node_data {
11*215249f6SDave Marchevsky int key;
12*215249f6SDave Marchevsky int data;
13*215249f6SDave Marchevsky struct bpf_rb_node node;
14*215249f6SDave Marchevsky };
15*215249f6SDave Marchevsky
16*215249f6SDave Marchevsky struct node_data2 {
17*215249f6SDave Marchevsky int key;
18*215249f6SDave Marchevsky struct bpf_rb_node node;
19*215249f6SDave Marchevsky int data;
20*215249f6SDave Marchevsky };
21*215249f6SDave Marchevsky
less2(struct bpf_rb_node * a,const struct bpf_rb_node * b)22*215249f6SDave Marchevsky static bool less2(struct bpf_rb_node *a, const struct bpf_rb_node *b)
23*215249f6SDave Marchevsky {
24*215249f6SDave Marchevsky struct node_data2 *node_a;
25*215249f6SDave Marchevsky struct node_data2 *node_b;
26*215249f6SDave Marchevsky
27*215249f6SDave Marchevsky node_a = container_of(a, struct node_data2, node);
28*215249f6SDave Marchevsky node_b = container_of(b, struct node_data2, node);
29*215249f6SDave Marchevsky
30*215249f6SDave Marchevsky return node_a->key < node_b->key;
31*215249f6SDave Marchevsky }
32*215249f6SDave Marchevsky
33*215249f6SDave Marchevsky #define private(name) SEC(".data." #name) __hidden __attribute__((aligned(8)))
34*215249f6SDave Marchevsky private(A) struct bpf_spin_lock glock;
35*215249f6SDave Marchevsky private(A) struct bpf_rb_root groot __contains(node_data, node);
36*215249f6SDave Marchevsky
37*215249f6SDave Marchevsky SEC("tc")
rbtree_api_add__add_wrong_type(void * ctx)38*215249f6SDave Marchevsky long rbtree_api_add__add_wrong_type(void *ctx)
39*215249f6SDave Marchevsky {
40*215249f6SDave Marchevsky struct node_data2 *n;
41*215249f6SDave Marchevsky
42*215249f6SDave Marchevsky n = bpf_obj_new(typeof(*n));
43*215249f6SDave Marchevsky if (!n)
44*215249f6SDave Marchevsky return 1;
45*215249f6SDave Marchevsky
46*215249f6SDave Marchevsky bpf_spin_lock(&glock);
47*215249f6SDave Marchevsky bpf_rbtree_add(&groot, &n->node, less2);
48*215249f6SDave Marchevsky bpf_spin_unlock(&glock);
49*215249f6SDave Marchevsky return 0;
50*215249f6SDave Marchevsky }
51*215249f6SDave Marchevsky
52*215249f6SDave Marchevsky char _license[] SEC("license") = "GPL";
53