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

#include <vmlinux.h>
#include <string.h>
#include <stdbool.h>
#include <bpf/bpf_tracing.h>
#include "bpf_misc.h"

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

int err;
void *user_ptr;

SEC("lsm/file_open")
__failure
__msg("Unreleased reference id=")
int on_nanosleep_unreleased_ref(void *ctx)
{
	struct task_struct *task = bpf_get_current_task_btf();
	struct file *file = bpf_get_task_exe_file(task);
	struct bpf_dynptr dynptr;

	if (!file)
		return 0;

	err = bpf_dynptr_from_file(file, 0, &dynptr);
	return err ? 1 : 0;
}

SEC("xdp")
__failure
__msg("Expected a dynptr of type file as R1")
int xdp_wrong_dynptr_type(struct xdp_md *xdp)
{
	struct bpf_dynptr dynptr;

	bpf_dynptr_from_xdp(xdp, 0, &dynptr);
	bpf_dynptr_file_discard(&dynptr);
	return 0;
}

SEC("xdp")
__failure
__msg("Expected an initialized dynptr as R1")
int xdp_no_dynptr_type(struct xdp_md *xdp)
{
	struct bpf_dynptr dynptr;

	bpf_dynptr_file_discard(&dynptr);
	return 0;
}

SEC("lsm/file_open")
__failure
__msg("Leaking reference id={{[0-9]+}} alloc_insn={{[0-9]+}}. Release it first.")
int use_file_dynptr_after_put_file(void *ctx)
{
	struct task_struct *task = bpf_get_current_task_btf();
	struct file *file = bpf_get_task_exe_file(task);
	struct bpf_dynptr dynptr;
	char buf[64];

	if (!file)
		return 0;

	if (bpf_dynptr_from_file(file, 0, &dynptr))
		goto out;

	/* this should fail - file dynptr should be discarded first to prevent resource leak */
	bpf_put_file(file);

	bpf_dynptr_read(buf, sizeof(buf), &dynptr, 0, 0);
	return 0;

out:
	bpf_dynptr_file_discard(&dynptr);
	bpf_put_file(file);
	return 0;
}

SEC("lsm/file_open")
__failure
__msg("Leaking reference id={{[0-9]+}} alloc_insn={{[0-9]+}}. Release it first.")
int use_file_dynptr_slice_after_put_file(void *ctx)
{
	struct task_struct *task = bpf_get_current_task_btf();
	struct file *file = bpf_get_task_exe_file(task);
	struct bpf_dynptr dynptr;
	char buf[1];
	const char *data;

	if (!file)
		return 0;

	if (bpf_dynptr_from_file(file, 0, &dynptr))
		goto out;

	data = bpf_dynptr_slice(&dynptr, 0, buf, sizeof(buf));
	if (!data)
		goto out;

	/* this should fail - file dynptr should be discarded first to prevent resource leak */
	bpf_put_file(file);

	return data[0];

out:
	bpf_dynptr_file_discard(&dynptr);
	bpf_put_file(file);
	return 0;
}