forked from iovisor/bcc
-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
332 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,332 @@ | ||
require("test_helper") | ||
|
||
TestClang = {} | ||
|
||
function TestClang:test_probe_read1() | ||
local text = [[ | ||
#include <linux/sched.h> | ||
#include <uapi/linux/ptrace.h> | ||
int count_sched(struct pt_regs *ctx, struct task_struct *prev) { | ||
pid_t p = prev->pid; | ||
return (p != -1); | ||
} | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
local fn = b:load_func("count_sched", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_probe_read2() | ||
local text = [[ | ||
#include <linux/sched.h> | ||
#include <uapi/linux/ptrace.h> | ||
int count_foo(struct pt_regs *ctx, unsigned long a, unsigned long b) { | ||
return (a != b); | ||
} | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
local fn = b:load_func("count_foo", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_probe_read_keys() | ||
local text = [[ | ||
#include <uapi/linux/ptrace.h> | ||
#include <linux/blkdev.h> | ||
BPF_HASH(start, struct request *); | ||
int do_request(struct pt_regs *ctx, struct request *req) { | ||
u64 ts = bpf_ktime_get_ns(); | ||
start.update(&req, &ts); | ||
return 0; | ||
} | ||
int do_completion(struct pt_regs *ctx, struct request *req) { | ||
u64 *tsp = start.lookup(&req); | ||
if (tsp != 0) { | ||
start.delete(&req); | ||
} | ||
return 0; | ||
} | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
local fns = b:load_funcs('BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_sscanf() | ||
local text = [[ | ||
BPF_TABLE("hash", int, struct { u64 a; u64 b; u32 c:18; u32 d:14; struct { u32 a; u32 b; } s; }, stats, 10); | ||
int foo(void *ctx) { | ||
return 0; | ||
} | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
local fn = b:load_func("foo", 'BPF_PROG_TYPE_KPROBE') | ||
local t = b:get_table("stats") | ||
local s1 = t:key_sprintf(2) | ||
|
||
assert_equals(s1, "0x2") | ||
|
||
local s2 = t:leaf_sprintf({{2, 3, 4, 1, {5, 6}}}) | ||
local l = t:leaf_scanf(s2) | ||
|
||
assert_equals(tonumber(l.a), 2) | ||
assert_equals(tonumber(l.b), 3) | ||
assert_equals(tonumber(l.c), 4) | ||
assert_equals(tonumber(l.d), 1) | ||
assert_equals(tonumber(l.s.a), 5) | ||
assert_equals(tonumber(l.s.b), 6) | ||
end | ||
|
||
function TestClang:test_sscanf_array() | ||
local text = [[ BPF_TABLE("hash", int, struct { u32 a[3]; u32 b; }, stats, 10); ]] | ||
|
||
local b = BPF:new{text=text, debug=0} | ||
local t = b:get_table("stats") | ||
|
||
local s1 = t:key_sprintf(2) | ||
assert_equals(s1, "0x2") | ||
|
||
local s2 = t:leaf_sprintf({{{1, 2, 3}, 4}}) | ||
assert_equals(s2, "{ [ 0x1 0x2 0x3 ] 0x4 }") | ||
|
||
local l = t:leaf_scanf(s2) | ||
assert_equals(l.a[0], 1) | ||
assert_equals(l.a[1], 2) | ||
assert_equals(l.a[2], 3) | ||
assert_equals(l.b, 4) | ||
end | ||
|
||
function TestClang:test_iosnoop() | ||
local text = [[ | ||
#include <linux/blkdev.h> | ||
#include <uapi/linux/ptrace.h> | ||
struct key_t { | ||
struct request *req; | ||
}; | ||
BPF_TABLE("hash", struct key_t, u64, start, 1024); | ||
int do_request(struct pt_regs *ctx, struct request *req) { | ||
struct key_t key = {}; | ||
bpf_trace_printk("traced start %d\\n", req->__data_len); | ||
return 0; | ||
} | ||
]] | ||
|
||
local b = BPF:new{text=text, debug=0} | ||
local fn = b:load_func("do_request", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_blk_start_request() | ||
local text = [[ | ||
#include <linux/blkdev.h> | ||
#include <uapi/linux/ptrace.h> | ||
int do_request(struct pt_regs *ctx, int req) { | ||
bpf_trace_printk("req ptr: 0x%x\n", req); | ||
return 0; | ||
} | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
local fn = b:load_func("do_request", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_bpf_hash() | ||
local text = [[ | ||
BPF_HASH(table1); | ||
BPF_HASH(table2, u32); | ||
BPF_HASH(table3, u32, int); | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
end | ||
|
||
function TestClang:test_consecutive_probe_read() | ||
local text = [[ | ||
#include <linux/fs.h> | ||
#include <linux/mount.h> | ||
BPF_HASH(table1, struct super_block *); | ||
int trace_entry(struct pt_regs *ctx, struct file *file) { | ||
if (!file) return 0; | ||
struct vfsmount *mnt = file->f_path.mnt; | ||
if (mnt) { | ||
struct super_block *k = mnt->mnt_sb; | ||
u64 zero = 0; | ||
table1.update(&k, &zero); | ||
k = mnt->mnt_sb; | ||
table1.update(&k, &zero); | ||
} | ||
return 0; | ||
} | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
local fn = b:load_func("trace_entry", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_nested_probe_read() | ||
local text = [[ | ||
#include <linux/fs.h> | ||
int trace_entry(struct pt_regs *ctx, struct file *file) { | ||
if (!file) return 0; | ||
const char *name = file->f_path.dentry->d_name.name; | ||
bpf_trace_printk("%s\\n", name); | ||
return 0; | ||
} | ||
]] | ||
local b = BPF:new{text=text, debug=0} | ||
local fn = b:load_func("trace_entry", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_char_array_probe() | ||
local b = BPF:new{text=[[#include <linux/blkdev.h> | ||
int kprobe__blk_update_request(struct pt_regs *ctx, struct request *req) { | ||
bpf_trace_printk("%s\\n", req->rq_disk->disk_name); | ||
return 0; | ||
}]]} | ||
end | ||
|
||
function TestClang:test_probe_read_helper() | ||
local b = BPF:new{text=[[ | ||
#include <linux/fs.h> | ||
static void print_file_name(struct file *file) { | ||
if (!file) return; | ||
const char *name = file->f_path.dentry->d_name.name; | ||
bpf_trace_printk("%s\\n", name); | ||
} | ||
static void print_file_name2(int unused, struct file *file) { | ||
print_file_name(file); | ||
} | ||
int trace_entry1(struct pt_regs *ctx, struct file *file) { | ||
print_file_name(file); | ||
return 0; | ||
} | ||
int trace_entry2(struct pt_regs *ctx, int unused, struct file *file) { | ||
print_file_name2(unused, file); | ||
return 0; | ||
} | ||
]]} | ||
local fn1 = b:load_func("trace_entry1", 'BPF_PROG_TYPE_KPROBE') | ||
local fn2 = b:load_func("trace_entry2", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_probe_struct_assign() | ||
local b = BPF:new{text = [[ | ||
#include <uapi/linux/ptrace.h> | ||
struct args_t { | ||
const char *filename; | ||
int flags; | ||
int mode; | ||
}; | ||
int kprobe__sys_open(struct pt_regs *ctx, const char *filename, | ||
int flags, int mode) { | ||
struct args_t args = {}; | ||
args.filename = filename; | ||
args.flags = flags; | ||
args.mode = mode; | ||
bpf_trace_printk("%s\\n", args.filename); | ||
return 0; | ||
}; | ||
]]} | ||
end | ||
|
||
function TestClang:test_task_switch() | ||
local b = BPF:new{text=[[ | ||
#include <uapi/linux/ptrace.h> | ||
#include <linux/sched.h> | ||
struct key_t { | ||
u32 prev_pid; | ||
u32 curr_pid; | ||
}; | ||
BPF_TABLE("hash", struct key_t, u64, stats, 1024); | ||
int kprobe__finish_task_switch(struct pt_regs *ctx, struct task_struct *prev) { | ||
struct key_t key = {}; | ||
u64 zero = 0, *val; | ||
key.curr_pid = bpf_get_current_pid_tgid(); | ||
key.prev_pid = prev->pid; | ||
val = stats.lookup_or_init(&key, &zero); | ||
(*val)++; | ||
return 0; | ||
} | ||
]]} | ||
end | ||
|
||
function TestClang:test_probe_simple_assign() | ||
local b = BPF:new{text=[[ | ||
#include <uapi/linux/ptrace.h> | ||
#include <linux/gfp.h> | ||
struct leaf { size_t size; }; | ||
BPF_HASH(simple_map, u32, struct leaf); | ||
int kprobe____kmalloc(struct pt_regs *ctx, size_t size) { | ||
u32 pid = bpf_get_current_pid_tgid(); | ||
struct leaf* leaf = simple_map.lookup(&pid); | ||
if (leaf) | ||
leaf->size += size; | ||
return 0; | ||
}]]} | ||
end | ||
|
||
function TestClang:test_unop_probe_read() | ||
local text = [[ | ||
#include <linux/blkdev.h> | ||
int trace_entry(struct pt_regs *ctx, struct request *req) { | ||
if (!(req->bio->bi_rw & 1)) | ||
return 1; | ||
if (((req->bio->bi_rw))) | ||
return 1; | ||
return 0; | ||
} | ||
]] | ||
local b = BPF:new{text=text} | ||
local fn = b:load_func("trace_entry", 'BPF_PROG_TYPE_KPROBE') | ||
end | ||
|
||
function TestClang:test_complex_leaf_types() | ||
local text = [[ | ||
struct list; | ||
struct list { | ||
struct list *selfp; | ||
struct list *another_selfp; | ||
struct list *selfp_array[2]; | ||
}; | ||
struct empty { | ||
}; | ||
union emptyu { | ||
struct empty *em1; | ||
struct empty em2; | ||
struct empty em3; | ||
struct empty em4; | ||
}; | ||
BPF_TABLE("array", int, struct list, t1, 1); | ||
BPF_TABLE("array", int, struct list *, t2, 1); | ||
BPF_TABLE("array", int, union emptyu, t3, 1); | ||
]] | ||
local b = BPF:new{text=text} | ||
local ffi = require("ffi") | ||
|
||
-- TODO: ptrs? | ||
assert_equals(ffi.sizeof(b:get_table("t3").c_leaf), 8) | ||
end | ||
|
||
function TestClang:test_cflags() | ||
local text = [[ | ||
#ifndef MYFLAG | ||
#error "MYFLAG not set as expected" | ||
#endif | ||
]] | ||
local b = BPF:new{text=text, cflags={"-DMYFLAG"}} | ||
end | ||
|
||
function TestClang:test_exported_maps() | ||
local b1 = BPF{text=[[BPF_TABLE_PUBLIC("hash", int, int, table1, 10);]]} | ||
local b2 = BPF{text=[[BPF_TABLE("extern", int, int, table1, 10);]]} | ||
end | ||
|
||
function TestClang:test_syntax_error() | ||
assert_error_msg_contains( | ||
"failed to compile BPF module", | ||
BPF.new, | ||
BPF, {text=[[int failure(void *ctx) { if (); return 0; }]]}) | ||
end | ||
|
||
os.exit(LuaUnit.run()) |