static int bpf_do_prog(const char *file, uint32_t flags, const char *object)
{
int fd, sock, ret;
if (flags & BPF_F_PIN) {
fd = bpf_prog_create(object);
printf("bpf: prog fd:%d (%s)\n", fd, strerror(errno));
assert(fd > 0);
ret = bpf_obj_pin(fd, file);
printf("bpf: pin ret:(%d,%s)\n", ret, strerror(errno));
assert(ret == 0);
} else {
fd = bpf_obj_get(file);
printf("bpf: get fd:%d (%s)\n", fd, strerror(errno));
assert(fd > 0);
}
sock = open_raw_sock("lo");
assert(sock > 0);
ret = setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, &fd, sizeof(fd));
printf("bpf: sock:%d <- fd:%d attached ret:(%d,%s)\n", sock, fd,
ret, strerror(errno));
assert(ret == 0);
return 0;
}
int main (int ac, char **argv)
{
if (ac != 3) {
printf("Usage: master <module.bpf> <map_path>\n");
return -1;
}
char filename[256];
int sock;
snprintf(filename, sizeof(filename), "%s", argv[1]);
if (load_bpf_file(filename)){
printf("%s", bpf_log_buf);
return 1;
}
/* Initializing our map with key val pair ( IPPROTO_ICMP,0) */
int key = IPPROTO_ICMP;
long value = 0;
int ret = bpf_update_elem(map_fd[0], &key, &value, BPF_ANY);
if (ret != 0) {
printf("Failed to add key-value pair: %s\n",strerror(errno));
return -1;
}
sock = open_raw_sock("lo");
assert(setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, prog_fd,
sizeof(prog_fd[0])) == 0);
/* Persisting map */
assert(bpf_obj_pin(map_fd[0],argv[2]) == 0);
printf("MAP persisted to %s\n", argv[2]);
// Pinging to localhost
FILE * f = popen("ping -c5 localhost", "r");
(void) f; //Suppress unused variable warning from compiler
long icmp_cnt = 0;
char q;
while (q != 'q') {
assert(bpf_lookup_elem(map_fd[0], &key, &icmp_cnt) == 0);
printf("ICMP Count %ld | Enter 'q' to quit: ", icmp_cnt);
q = getchar();
}
return 0;
}
int main (int ac, char **argv)
{
if (ac != 3) {
printf("Usage: master_host <module.bpf> <prog_path>\n");
return -1;
}
char filename[256];
snprintf(filename, sizeof(filename), "%s", argv[1]);
if (load_bpf_file(filename)){
printf("%s", bpf_log_buf);
return 1;
}
int key = 0;
int fd = bpf_obj_get(argv[2]);
if (fd < 0) {
printf("Failed to load container module %s\n", argv[2]);
return -1;
}
int ret = bpf_update_elem(map_fd[0], &key, &fd, BPF_ANY);
if (ret != 0) {
printf("Failed to add key-value pair %s\n", strerror(errno));
return -1;
}
int sock = open_raw_sock("lo");
assert(setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, prog_fd,
sizeof(prog_fd[0])) == 0);
FILE *f = popen("ping -c5 localhost", "r");
(void) f;
char q;
while (q != 'q') {
printf("Enter 'q' to exit\n");
q = getchar();
}
return 0;
}
int main(int ac, char **argv)
{
char filename[256];
FILE *f;
int i, sock;
snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]);
if (load_bpf_file(filename)) {
printf("%s", bpf_log_buf);
return 1;
}
sock = open_raw_sock("lo");
assert(setsockopt(sock, SOL_SOCKET, SO_ATTACH_BPF, prog_fd,
sizeof(prog_fd[0])) == 0);
f = popen("ping -c5 localhost", "r");
(void) f;
for (i = 0; i < 5; i++) {
long long tcp_cnt, udp_cnt, icmp_cnt;
int key;
key = IPPROTO_TCP;
assert(bpf_map_lookup_elem(map_fd[0], &key, &tcp_cnt) == 0);
key = IPPROTO_UDP;
assert(bpf_map_lookup_elem(map_fd[0], &key, &udp_cnt) == 0);
key = IPPROTO_ICMP;
assert(bpf_map_lookup_elem(map_fd[0], &key, &icmp_cnt) == 0);
printf("TCP %lld UDP %lld ICMP %lld bytes\n",
tcp_cnt, udp_cnt, icmp_cnt);
sleep(1);
}
return 0;
}
