Commit 44c4aa2b authored by Hangbin Liu's avatar Hangbin Liu Committed by Alexei Starovoitov
Browse files

selftest/bpf: Test pinning map with reused map fd 



This add a test to make sure that we can still pin maps with
reused map fd.

Signed-off-by: default avatarHangbin Liu <liuhangbin@gmail.com>
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
Acked-by: default avatarAndrii Nakryiko <andrii@kernel.org>
Link: https://lore.kernel.org/bpf/20201006021345.3817033-4-liuhangbin@gmail.com
parent 2c193d32

tools/testing/selftests/bpf/prog_tests/pinning.c

+48 −1
Original line number Diff line number Diff line
@@ -37,7 +37,7 @@ void test_pinning(void) 
	struct stat statbuf = {};

	struct bpf_object *obj;

	struct bpf_map *map;

	int err;

	int err, map_fd;

	DECLARE_LIBBPF_OPTS(bpf_object_open_opts, opts,

		.pin_root_path = custpath,

	);
@@ -213,6 +213,53 @@ void test_pinning(void) 
	if (CHECK(err, "stat custpinpath", "err %d errno %d\n", err, errno))

		goto out;



	/* remove the custom pin path to re-test it with reuse fd below */

	err = unlink(custpinpath);

	if (CHECK(err, "unlink custpinpath", "err %d errno %d\n", err, errno))

		goto out;



	err = rmdir(custpath);

	if (CHECK(err, "rmdir custpindir", "err %d errno %d\n", err, errno))

		goto out;



	bpf_object__close(obj);



	/* test pinning at custom path with reuse fd */

	obj = bpf_object__open_file(file, NULL);

	err = libbpf_get_error(obj);

	if (CHECK(err, "default open", "err %d errno %d\n", err, errno)) {

		obj = NULL;

		goto out;

	}



	map_fd = bpf_create_map(BPF_MAP_TYPE_ARRAY, sizeof(__u32),

				sizeof(__u64), 1, 0);

	if (CHECK(map_fd < 0, "create pinmap manually", "fd %d\n", map_fd))

		goto out;



	map = bpf_object__find_map_by_name(obj, "pinmap");

	if (CHECK(!map, "find map", "NULL map"))

		goto close_map_fd;



	err = bpf_map__reuse_fd(map, map_fd);

	if (CHECK(err, "reuse pinmap fd", "err %d errno %d\n", err, errno))

		goto close_map_fd;



	err = bpf_map__set_pin_path(map, custpinpath);

	if (CHECK(err, "set pin path", "err %d errno %d\n", err, errno))

		goto close_map_fd;



	err = bpf_object__load(obj);

	if (CHECK(err, "custom load", "err %d errno %d\n", err, errno))

		goto close_map_fd;



	/* check that pinmap was pinned at the custom path */

	err = stat(custpinpath, &statbuf);

	if (CHECK(err, "stat custpinpath", "err %d errno %d\n", err, errno))

		goto close_map_fd;



close_map_fd:

	close(map_fd);

out:

	unlink(pinpath);

	unlink(nopinpath);