github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-09-05 04:23:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make hash map names more meaningful

Browse Source
This commit is contained in:
M. Mert Yildiran
2022-06-01 14:36:03 +03:00
parent 3b332d2a9a
commit 76bee7a9df
6 changed files with 18 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tap/tlstapper/bpf/golang_uprobes.c
View File

@@ -38,7 +38,7 @@ static __always_inline int golang_crypto_tls_write_uprobe(struct pt_regs *ctx) {
} }
__u64 key_dial_full = (pid << 32) + key_dial; __u64 key_dial_full = (pid << 32) + key_dial;
struct socket *s = bpf_map_lookup_elem(&golang_dial_writes, &key_dial_full); struct socket *s = bpf_map_lookup_elem(&golang_socket_to_write, &key_dial_full);
if (s == NULL) { if (s == NULL) {
bpf_printk("[golang_crypto_tls_write_uprobe] error getting socket"); bpf_printk("[golang_crypto_tls_write_uprobe] error getting socket");
return 0; return 0;
@@ -155,7 +155,7 @@ static __always_inline int golang_net_socket_uprobe(struct pt_regs *ctx) {
__u64 pid = pid_tgid >> 32; __u64 pid = pid_tgid >> 32;
// ctx->r14 is common between golang_net_socket_uprobe and golang_net_http_dialconn_uprobe // ctx->r14 is common between golang_net_socket_uprobe and golang_net_http_dialconn_uprobe
__u64 key_socket = (pid << 32) + ctx->r14; __u64 key_socket = (pid << 32) + ctx->r14;
struct socket *s = bpf_map_lookup_elem(&golang_socket_dials, &key_socket); struct socket *s = bpf_map_lookup_elem(&golang_dial_to_socket, &key_socket);
if (s == NULL) { if (s == NULL) {
return 0; return 0;
} }
@@ -163,7 +163,7 @@ static __always_inline int golang_net_socket_uprobe(struct pt_regs *ctx) {
struct socket b = { .pid = s->pid, .fd = ctx->rax, .key_dial = s->key_dial }; struct socket b = { .pid = s->pid, .fd = ctx->rax, .key_dial = s->key_dial };
__u64 key_dial_full = (pid << 32) + s->key_dial; __u64 key_dial_full = (pid << 32) + s->key_dial;
__u32 status = bpf_map_update_elem(&golang_dial_writes, &key_dial_full, &b, BPF_ANY); __u32 status = bpf_map_update_elem(&golang_socket_to_write, &key_dial_full, &b, BPF_ANY);
if (status != 0) { if (status != 0) {
bpf_printk("[golang_net_socket_uprobe] error updating socket file descriptor: %d", status); bpf_printk("[golang_net_socket_uprobe] error updating socket file descriptor: %d", status);
} }
@@ -188,7 +188,7 @@ static __always_inline int golang_net_http_dialconn_uprobe(struct pt_regs *ctx)
__u64 pid = b.pid; __u64 pid = b.pid;
// ctx->r14 is common between golang_net_socket_uprobe and golang_net_http_dialconn_uprobe // ctx->r14 is common between golang_net_socket_uprobe and golang_net_http_dialconn_uprobe
__u64 key_socket = (pid << 32) + ctx->r14; __u64 key_socket = (pid << 32) + ctx->r14;
status = bpf_map_update_elem(&golang_socket_dials, &key_socket, &b, BPF_ANY); status = bpf_map_update_elem(&golang_dial_to_socket, &key_socket, &b, BPF_ANY);
if (status != 0) { if (status != 0) {
bpf_printk("[golang_net_http_dialconn_uprobe] error setting socket: %d", status); bpf_printk("[golang_net_http_dialconn_uprobe] error setting socket: %d", status);
} }

4
tap/tlstapper/bpf/include/maps.h
View File

@@ -89,8 +89,8 @@ BPF_LRU_HASH(file_descriptor_to_ipv4, __u64, struct fd_info);
BPF_PERF_OUTPUT(chunks_buffer); BPF_PERF_OUTPUT(chunks_buffer);
BPF_PERF_OUTPUT(log_buffer); BPF_PERF_OUTPUT(log_buffer);
BPF_LRU_HASH(golang_socket_dials, __u64, struct socket); BPF_LRU_HASH(golang_dial_to_socket, __u64, struct socket);
BPF_LRU_HASH(golang_dial_writes, __u64, struct socket); BPF_LRU_HASH(golang_socket_to_write, __u64, struct socket);
BPF_RINGBUF(golang_read_writes); BPF_RINGBUF(golang_read_writes);
#endif /* __MAPS__ */ #endif /* __MAPS__ */

12
tap/tlstapper/tlstapper_bpfeb.go
View File

@@ -105,9 +105,9 @@ type tlsTapperMapSpecs struct {
ChunksBuffer *ebpf.MapSpec `ebpf:"chunks_buffer"` ChunksBuffer *ebpf.MapSpec `ebpf:"chunks_buffer"`
ConnectSyscallInfo *ebpf.MapSpec `ebpf:"connect_syscall_info"` ConnectSyscallInfo *ebpf.MapSpec `ebpf:"connect_syscall_info"`
FileDescriptorToIpv4 *ebpf.MapSpec `ebpf:"file_descriptor_to_ipv4"` FileDescriptorToIpv4 *ebpf.MapSpec `ebpf:"file_descriptor_to_ipv4"`
GolangDialWrites *ebpf.MapSpec `ebpf:"golang_dial_writes"` GolangDialToSocket *ebpf.MapSpec `ebpf:"golang_dial_to_socket"`
GolangReadWrites *ebpf.MapSpec `ebpf:"golang_read_writes"` GolangReadWrites *ebpf.MapSpec `ebpf:"golang_read_writes"`
GolangSocketDials *ebpf.MapSpec `ebpf:"golang_socket_dials"` GolangSocketToWrite *ebpf.MapSpec `ebpf:"golang_socket_to_write"`
Heap *ebpf.MapSpec `ebpf:"heap"` Heap *ebpf.MapSpec `ebpf:"heap"`
LogBuffer *ebpf.MapSpec `ebpf:"log_buffer"` LogBuffer *ebpf.MapSpec `ebpf:"log_buffer"`
PidsMap *ebpf.MapSpec `ebpf:"pids_map"` PidsMap *ebpf.MapSpec `ebpf:"pids_map"`
@@ -138,9 +138,9 @@ type tlsTapperMaps struct {
ChunksBuffer *ebpf.Map `ebpf:"chunks_buffer"` ChunksBuffer *ebpf.Map `ebpf:"chunks_buffer"`
ConnectSyscallInfo *ebpf.Map `ebpf:"connect_syscall_info"` ConnectSyscallInfo *ebpf.Map `ebpf:"connect_syscall_info"`
FileDescriptorToIpv4 *ebpf.Map `ebpf:"file_descriptor_to_ipv4"` FileDescriptorToIpv4 *ebpf.Map `ebpf:"file_descriptor_to_ipv4"`
GolangDialWrites *ebpf.Map `ebpf:"golang_dial_writes"` GolangDialToSocket *ebpf.Map `ebpf:"golang_dial_to_socket"`
GolangReadWrites *ebpf.Map `ebpf:"golang_read_writes"` GolangReadWrites *ebpf.Map `ebpf:"golang_read_writes"`
GolangSocketDials *ebpf.Map `ebpf:"golang_socket_dials"` GolangSocketToWrite *ebpf.Map `ebpf:"golang_socket_to_write"`
Heap *ebpf.Map `ebpf:"heap"` Heap *ebpf.Map `ebpf:"heap"`
LogBuffer *ebpf.Map `ebpf:"log_buffer"` LogBuffer *ebpf.Map `ebpf:"log_buffer"`
PidsMap *ebpf.Map `ebpf:"pids_map"` PidsMap *ebpf.Map `ebpf:"pids_map"`
@@ -154,9 +154,9 @@ func (m *tlsTapperMaps) Close() error {
m.ChunksBuffer, m.ChunksBuffer,
m.ConnectSyscallInfo, m.ConnectSyscallInfo,
m.FileDescriptorToIpv4, m.FileDescriptorToIpv4,
m.GolangDialWrites, m.GolangDialToSocket,
m.GolangReadWrites, m.GolangReadWrites,
m.GolangSocketDials, m.GolangSocketToWrite,
m.Heap, m.Heap,
m.LogBuffer, m.LogBuffer,
m.PidsMap, m.PidsMap,

BIN
tap/tlstapper/tlstapper_bpfeb.o
View File

Binary file not shown.

12
tap/tlstapper/tlstapper_bpfel.go
View File

@@ -105,9 +105,9 @@ type tlsTapperMapSpecs struct {
ChunksBuffer *ebpf.MapSpec `ebpf:"chunks_buffer"` ChunksBuffer *ebpf.MapSpec `ebpf:"chunks_buffer"`
ConnectSyscallInfo *ebpf.MapSpec `ebpf:"connect_syscall_info"` ConnectSyscallInfo *ebpf.MapSpec `ebpf:"connect_syscall_info"`
FileDescriptorToIpv4 *ebpf.MapSpec `ebpf:"file_descriptor_to_ipv4"` FileDescriptorToIpv4 *ebpf.MapSpec `ebpf:"file_descriptor_to_ipv4"`
GolangDialWrites *ebpf.MapSpec `ebpf:"golang_dial_writes"` GolangDialToSocket *ebpf.MapSpec `ebpf:"golang_dial_to_socket"`
GolangReadWrites *ebpf.MapSpec `ebpf:"golang_read_writes"` GolangReadWrites *ebpf.MapSpec `ebpf:"golang_read_writes"`
GolangSocketDials *ebpf.MapSpec `ebpf:"golang_socket_dials"` GolangSocketToWrite *ebpf.MapSpec `ebpf:"golang_socket_to_write"`
Heap *ebpf.MapSpec `ebpf:"heap"` Heap *ebpf.MapSpec `ebpf:"heap"`
LogBuffer *ebpf.MapSpec `ebpf:"log_buffer"` LogBuffer *ebpf.MapSpec `ebpf:"log_buffer"`
PidsMap *ebpf.MapSpec `ebpf:"pids_map"` PidsMap *ebpf.MapSpec `ebpf:"pids_map"`
@@ -138,9 +138,9 @@ type tlsTapperMaps struct {
ChunksBuffer *ebpf.Map `ebpf:"chunks_buffer"` ChunksBuffer *ebpf.Map `ebpf:"chunks_buffer"`
ConnectSyscallInfo *ebpf.Map `ebpf:"connect_syscall_info"` ConnectSyscallInfo *ebpf.Map `ebpf:"connect_syscall_info"`
FileDescriptorToIpv4 *ebpf.Map `ebpf:"file_descriptor_to_ipv4"` FileDescriptorToIpv4 *ebpf.Map `ebpf:"file_descriptor_to_ipv4"`
GolangDialWrites *ebpf.Map `ebpf:"golang_dial_writes"` GolangDialToSocket *ebpf.Map `ebpf:"golang_dial_to_socket"`
GolangReadWrites *ebpf.Map `ebpf:"golang_read_writes"` GolangReadWrites *ebpf.Map `ebpf:"golang_read_writes"`
GolangSocketDials *ebpf.Map `ebpf:"golang_socket_dials"` GolangSocketToWrite *ebpf.Map `ebpf:"golang_socket_to_write"`
Heap *ebpf.Map `ebpf:"heap"` Heap *ebpf.Map `ebpf:"heap"`
LogBuffer *ebpf.Map `ebpf:"log_buffer"` LogBuffer *ebpf.Map `ebpf:"log_buffer"`
PidsMap *ebpf.Map `ebpf:"pids_map"` PidsMap *ebpf.Map `ebpf:"pids_map"`
@@ -154,9 +154,9 @@ func (m *tlsTapperMaps) Close() error {
m.ChunksBuffer, m.ChunksBuffer,
m.ConnectSyscallInfo, m.ConnectSyscallInfo,
m.FileDescriptorToIpv4, m.FileDescriptorToIpv4,
m.GolangDialWrites, m.GolangDialToSocket,
m.GolangReadWrites, m.GolangReadWrites,
m.GolangSocketDials, m.GolangSocketToWrite,
m.Heap, m.Heap,
m.LogBuffer, m.LogBuffer,
m.PidsMap, m.PidsMap,

BIN
tap/tlstapper/tlstapper_bpfel.o
View File

Binary file not shown.