[linux-2.6-block.git] / samples / bpf / test_cgrp2_sock2.c

// SPDX-License-Identifier: GPL-2.0
/* eBPF example program:
 *
 * - Loads eBPF program
 *
 *   The eBPF program loads a filter from file and attaches the
 *   program to a cgroup using BPF_PROG_ATTACH
 */

#define _GNU_SOURCE

#include <stdio.h>
#include <stdlib.h>
#include <stddef.h>
#include <string.h>
#include <unistd.h>
#include <assert.h>
#include <errno.h>
#include <fcntl.h>
#include <net/if.h>
#include <linux/bpf.h>
#include <bpf/bpf.h>

#include "bpf_insn.h"
#include "bpf_load.h"

static int usage(const char *argv0)
{
	printf("Usage: %s cg-path filter-path [filter-id]\n", argv0);
	return EXIT_FAILURE;
}

int main(int argc, char **argv)
{
	int cg_fd, ret, filter_id = 0;

	if (argc < 3)
		return usage(argv[0]);

	cg_fd = open(argv[1], O_DIRECTORY | O_RDONLY);
	if (cg_fd < 0) {
		printf("Failed to open cgroup path: '%s'\n", strerror(errno));
		return EXIT_FAILURE;
	}

	if (load_bpf_file(argv[2]))
		return EXIT_FAILURE;

	printf("Output from kernel verifier:\n%s\n-------\n", bpf_log_buf);

	if (argc > 3)
		filter_id = atoi(argv[3]);

	if (filter_id >= prog_cnt) {
		printf("Invalid program id; program not found in file\n");
		return EXIT_FAILURE;
	}

	ret = bpf_prog_attach(prog_fd[filter_id], cg_fd,
			      BPF_CGROUP_INET_SOCK_CREATE, 0);
	if (ret < 0) {
		printf("Failed to attach prog to cgroup: '%s'\n",
		       strerror(errno));
		return EXIT_FAILURE;
	}

	return EXIT_SUCCESS;
}
Commit	Line	Data
b2441318	1	// SPDX-License-Identifier: GPL-2.0
554ae6e7 DA	2	/* eBPF example program:
	3	*
	4	* - Loads eBPF program
	5	*
	6	* The eBPF program loads a filter from file and attaches the
	7	* program to a cgroup using BPF_PROG_ATTACH
	8	*/
	9
	10	#define _GNU_SOURCE
	11
	12	#include <stdio.h>
	13	#include <stdlib.h>
	14	#include <stddef.h>
	15	#include <string.h>
	16	#include <unistd.h>
	17	#include <assert.h>
	18	#include <errno.h>
	19	#include <fcntl.h>
	20	#include <net/if.h>
	21	#include <linux/bpf.h>
8d930450	22	#include <bpf/bpf.h>
554ae6e7	23
8d930450	24	#include "bpf_insn.h"
554ae6e7 DA	25	#include "bpf_load.h"
	26
	27	static int usage(const char *argv0)
	28	{
	29	printf("Usage: %s cg-path filter-path [filter-id]\n", argv0);
	30	return EXIT_FAILURE;
	31	}
	32
	33	int main(int argc, char **argv)
	34	{
	35	int cg_fd, ret, filter_id = 0;
	36
	37	if (argc < 3)
	38	return usage(argv[0]);
	39
	40	cg_fd = open(argv[1], O_DIRECTORY \| O_RDONLY);
	41	if (cg_fd < 0) {
	42	printf("Failed to open cgroup path: '%s'\n", strerror(errno));
	43	return EXIT_FAILURE;
	44	}
	45
	46	if (load_bpf_file(argv[2]))
	47	return EXIT_FAILURE;
	48
	49	printf("Output from kernel verifier:\n%s\n-------\n", bpf_log_buf);
	50
	51	if (argc > 3)
	52	filter_id = atoi(argv[3]);
	53
ee583014	54	if (filter_id >= prog_cnt) {
554ae6e7 DA	55	printf("Invalid program id; program not found in file\n");
	56	return EXIT_FAILURE;
	57	}
	58
	59	ret = bpf_prog_attach(prog_fd[filter_id], cg_fd,
7f677633	60	BPF_CGROUP_INET_SOCK_CREATE, 0);
554ae6e7 DA	61	if (ret < 0) {
	62	printf("Failed to attach prog to cgroup: '%s'\n",
	63	strerror(errno));
	64	return EXIT_FAILURE;
	65	}
	66
	67	return EXIT_SUCCESS;
	68	}