Fixed failing test_simple_library() in test_uprobes.py
@@ -24,18 +24,18 @@ int count(struct pt_regs *ctx) {
incr(0);
return 0;
}"""
- text = text.replace("PID", "%d" % os.getpid())
- print text
+ test_pid = os.getpid()
+ text = text.replace("PID", "%d" % test_pid)
b = bcc.BPF(text=text)
- b.attach_uprobe(name="c", sym="malloc_stats", fn_name="count")
- b.attach_uretprobe(name="c", sym="malloc_stats", fn_name="count")
+ b.attach_uprobe(name="c", sym="malloc_stats", fn_name="count", pid=test_pid)
+ b.attach_uretprobe(name="c", sym="malloc_stats", fn_name="count", pid=test_pid)
libc = ctypes.CDLL("libc.so.6")
libc.malloc_stats.restype = None
libc.malloc_stats.argtypes = []
libc.malloc_stats()
self.assertEqual(b["stats"][ctypes.c_int(0)].value, 2)
- b.detach_uretprobe(name="c", sym="malloc_stats")
- b.detach_uprobe(name="c", sym="malloc_stats")
+ b.detach_uretprobe(name="c", sym="malloc_stats", pid=test_pid)
+ b.detach_uprobe(name="c", sym="malloc_stats", pid=test_pid)
def test_simple_binary(self):
text = """
