// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */

#include <vmlinux.h>
#include <bpf/bpf_tracing.h>
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_core_read.h>
#include "bpf_experimental.h"

struct node_data {
	int key;
	int data;
	struct bpf_rb_node node;
};

struct node_data2 {
	int key;
	struct bpf_rb_node node;
	int data;
};

static bool less2(struct bpf_rb_node *a, const struct bpf_rb_node *b)
{
	struct node_data2 *node_a;
	struct node_data2 *node_b;

	node_a = container_of(a, struct node_data2, node);
	node_b = container_of(b, struct node_data2, node);

	return node_a->key < node_b->key;
}

#define private(name) SEC(".data." #name) __hidden __attribute__((aligned(8)))
private(A) struct bpf_spin_lock glock;
private(A) struct bpf_rb_root groot __contains(node_data, node);

SEC("tc")
long rbtree_api_add__add_wrong_type(void *ctx)
{
	struct node_data2 *n;

	n = bpf_obj_new(typeof(*n));
	if (!n)
		return 1;

	bpf_spin_lock(&glock);
	bpf_rbtree_add(&groot, &n->node, less2);
	bpf_spin_unlock(&glock);
	return 0;
}

char _license[] SEC("license") = "GPL";