static void clear_stats(int fd) { unsigned int nr_cpus = sysconf(_SC_NPROCESSORS_CONF); __u64 values[nr_cpus]; __u32 key; memset(values, 0, sizeof(values)); for (key = 0; key < SLOTS; key++) bpf_update_elem(fd, &key, values, BPF_ANY); }
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; }
static void update_elem(int key, unsigned long value) { if (bpf_update_elem(mapfd, &key, &value, 0)) { fail("bpf_update_elem failed '%s'\n", strerror(errno)); } }
int main(int ac, char **argv) { int serverfd, serverconnfd, clientfd; socklen_t sockaddr_len; struct sockaddr serv_addr, mapped_addr, tmp_addr; struct sockaddr_in *serv_addr_in, *mapped_addr_in, *tmp_addr_in; char filename[256]; char *ip; serv_addr_in = (struct sockaddr_in *)&serv_addr; mapped_addr_in = (struct sockaddr_in *)&mapped_addr; tmp_addr_in = (struct sockaddr_in *)&tmp_addr; snprintf(filename, sizeof(filename), "%s_kern.o", argv[0]); if (load_bpf_file(filename)) { printf("%s", bpf_log_buf); return 1; } assert((serverfd = socket(AF_INET, SOCK_STREAM, 0)) > 0); assert((clientfd = socket(AF_INET, SOCK_STREAM, 0)) > 0); /* Bind server to ephemeral port on lo */ memset(&serv_addr, 0, sizeof(serv_addr)); serv_addr_in->sin_family = AF_INET; serv_addr_in->sin_port = 0; serv_addr_in->sin_addr.s_addr = htonl(INADDR_LOOPBACK); assert(bind(serverfd, &serv_addr, sizeof(serv_addr)) == 0); sockaddr_len = sizeof(serv_addr); assert(getsockname(serverfd, &serv_addr, &sockaddr_len) == 0); ip = inet_ntoa(serv_addr_in->sin_addr); printf("Server bound to: %s:%d\n", ip, ntohs(serv_addr_in->sin_port)); memset(&mapped_addr, 0, sizeof(mapped_addr)); mapped_addr_in->sin_family = AF_INET; mapped_addr_in->sin_port = htons(5555); mapped_addr_in->sin_addr.s_addr = inet_addr("255.255.255.255"); assert(!bpf_update_elem(map_fd[0], &mapped_addr, &serv_addr, BPF_ANY)); assert(listen(serverfd, 5) == 0); ip = inet_ntoa(mapped_addr_in->sin_addr); printf("Client connecting to: %s:%d\n", ip, ntohs(mapped_addr_in->sin_port)); assert(connect(clientfd, &mapped_addr, sizeof(mapped_addr)) == 0); sockaddr_len = sizeof(tmp_addr); ip = inet_ntoa(tmp_addr_in->sin_addr); assert((serverconnfd = accept(serverfd, &tmp_addr, &sockaddr_len)) > 0); printf("Server received connection from: %s:%d\n", ip, ntohs(tmp_addr_in->sin_port)); sockaddr_len = sizeof(tmp_addr); assert(getpeername(clientfd, &tmp_addr, &sockaddr_len) == 0); ip = inet_ntoa(tmp_addr_in->sin_addr); printf("Client's peer address: %s:%d\n", ip, ntohs(tmp_addr_in->sin_port)); /* Is the server's getsockname = the socket getpeername */ assert(memcmp(&serv_addr, &tmp_addr, sizeof(struct sockaddr_in)) == 0); return 0; }