Commit c366fcf2 authored by 4ast's avatar 4ast

Merge pull request #89 from iovisor/quick_fix

Cache function handles in BPF class
parents 5e414945 53d45da1
......@@ -182,6 +182,9 @@ class BPF(object):
raise Exception("Failed to compile BPF module %s" % src_file)
def load_func(self, func_name, prog_type):
if func_name in self.funcs:
return self.funcs[func_name]
if lib.bpf_function_start(self.module, func_name.encode("ascii")) == None:
raise Exception("Unknown program %s" % func_name)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment