/*
 * Copyright (c) 2015-2017 Dmitry V. Levin <ldv@altlinux.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "tests.h"
#include <asm/unistd.h>

#if defined HAVE_UNION_BPF_ATTR_LOG_BUF && defined __NR_bpf
# include <stdio.h>
# include <stdint.h>
# include <unistd.h>
# include <linux/bpf.h>

static const struct bpf_insn insns[] = {
	{ .code = BPF_JMP | BPF_EXIT }
};

static const char *errstr;
static char log_buf[4096];

static long
sys_bpf(kernel_ulong_t cmd, kernel_ulong_t attr, kernel_ulong_t size)
{
	long rc = syscall(__NR_bpf, cmd, attr, size);
	errstr = sprintrc(rc);
	return rc;
}

static int
map_create(void)
{
	union bpf_attr attr = {
		.key_size = 4,
		.value_size = 8,
		.max_entries = 256
	};
	void *const t_attr = tail_memdup(&attr, sizeof(attr));
	return sys_bpf(BPF_MAP_CREATE, (unsigned long) t_attr, sizeof(attr));
}

static int
map_any(int cmd)
{
	union bpf_attr attr = {
		.map_fd = -1,
		.key = 0xdeadbeef,
		.value = 0xbadc0ded
	};
	void *const t_attr = tail_memdup(&attr, sizeof(attr));
	return sys_bpf(cmd, (unsigned long) t_attr, sizeof(attr));
}

static int
prog_load(void)
{
	union bpf_attr attr = {
		.insn_cnt = sizeof(insns) / sizeof(insns[0]),
		.insns = (unsigned long) insns,
		.license = (unsigned long) "GPL",
		.log_level = 42,
		.log_size = sizeof(log_buf),
		.log_buf = (unsigned long) log_buf
	};
	void *const t_attr = tail_memdup(&attr, sizeof(attr));
	return sys_bpf(BPF_PROG_LOAD, (unsigned long) t_attr, sizeof(attr));
}

/*
 * bpf() syscall and its first six commands were introduced in Linux kernel
 * 3.18. Some additional commands were added afterwards, so we need to take
 * precautions to make sure the tests compile.
 *
 * BPF_OBJ_PIN and BPF_OBJ_GET commands appear in kernel 4.4.
 */
# ifdef HAVE_UNION_BPF_ATTR_BPF_FD
static int
obj_manage(int cmd)
{
	union bpf_attr attr = {
		.pathname = (unsigned long) "/sys/fs/bpf/foo/bar",
		.bpf_fd = -1
	};
	void *const t_attr = tail_memdup(&attr, sizeof(attr));
	return sys_bpf(cmd, (unsigned long) t_attr, sizeof(attr));
}
# endif

/* BPF_PROG_ATTACH and BPF_PROG_DETACH commands appear in kernel 4.10. */
# ifdef HAVE_UNION_BPF_ATTR_ATTACH_TYPE
static int
prog_cgroup(int cmd)
{
	union bpf_attr attr = {
		.target_fd = -1,
		.attach_bpf_fd = -1,
		.attach_type = 0
	};
	void *const t_attr = tail_memdup(&attr, sizeof(attr));
	return sys_bpf(cmd, (unsigned long) t_attr, sizeof(attr));
}
# endif

int
main(void)
{
	map_create();
	printf("bpf(BPF_MAP_CREATE"
	       ", {map_type=BPF_MAP_TYPE_UNSPEC, key_size=4"
	       ", value_size=8, max_entries=256}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);

	map_any(BPF_MAP_LOOKUP_ELEM);
	printf("bpf(BPF_MAP_LOOKUP_ELEM"
	       ", {map_fd=-1, key=0xdeadbeef}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);

	map_any(BPF_MAP_UPDATE_ELEM);
	printf("bpf(BPF_MAP_UPDATE_ELEM"
	       ", {map_fd=-1, key=0xdeadbeef"
	       ", value=0xbadc0ded, flags=BPF_ANY}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);

	map_any(BPF_MAP_DELETE_ELEM);
	printf("bpf(BPF_MAP_DELETE_ELEM"
	       ", {map_fd=-1, key=0xdeadbeef}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);

	map_any(BPF_MAP_GET_NEXT_KEY);
	printf("bpf(BPF_MAP_GET_NEXT_KEY"
	       ", {map_fd=-1, key=0xdeadbeef}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);

	prog_load();
	printf("bpf(BPF_PROG_LOAD"
	       ", {prog_type=BPF_PROG_TYPE_UNSPEC, insn_cnt=1, insns=%p"
	       ", license=\"GPL\", log_level=42, log_size=4096, log_buf=%p"
	       ", kern_version=0}, %u) = %s\n",
	       insns, log_buf, (unsigned) sizeof(union bpf_attr), errstr);

# ifdef HAVE_UNION_BPF_ATTR_BPF_FD
	obj_manage(BPF_OBJ_PIN);
	printf("bpf(BPF_OBJ_PIN"
	       ", {pathname=\"/sys/fs/bpf/foo/bar\", bpf_fd=-1}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);

	obj_manage(BPF_OBJ_GET);
	printf("bpf(BPF_OBJ_GET"
	       ", {pathname=\"/sys/fs/bpf/foo/bar\", bpf_fd=-1}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);
# endif

# ifdef HAVE_UNION_BPF_ATTR_ATTACH_TYPE
	prog_cgroup(BPF_PROG_ATTACH);
	printf("bpf(BPF_PROG_ATTACH"
	       ", {target_fd=-1, attach_bpf_fd=-1"
	       ", attach_type=BPF_CGROUP_INET_INGRESS}, %u) = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);

	prog_cgroup(BPF_PROG_DETACH)
	printf("bpf(BPF_PROG_DETACH"
	       ", {target_fd=-1, attach_type=BPF_CGROUP_INET_INGRESS}, %u)"
	       " = %s\n",
	       (unsigned) sizeof(union bpf_attr), errstr);
# endif

	puts("+++ exited with 0 +++");
	return 0;
}

#else

SKIP_MAIN_UNDEFINED("__NR_bpf")

#endif