varun-r-mallya/python-bpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/Python-BPF.git synced 2025-12-31 21:06:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

make license handling easier

Browse Source
This commit is contained in:
varun-r-mallya
2025-09-09 00:19:53 +05:30
parent d6ba30805e
commit 03404bb164
4 changed files with 35 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pythonbpf/functions_pass.py
View File

@ -12,6 +12,8 @@ def get_probe_string(func_node):
# For helper functions, we return "helper"
for decorator in func_node.decorator_list:
if isinstance(decorator, ast.Name) and decorator.id == "bpfglobal":
return None
if isinstance(decorator, ast.Call) and isinstance(decorator.func, ast.Name):
if decorator.func.id == "section" and len(decorator.args) == 1:
arg = decorator.args[0]
@ -134,7 +136,9 @@ def process_bpf_chunk(func_node, module, return_type, map_sym_tab):
param = func.args[0]
param.add_attribute("nocapture")
func.section = get_probe_string(func_node)
probe_string = get_probe_string(func_node)
if probe_string is not None:
func.section = probe_string
block = func.append_basic_block(name="entry")
builder = ir.IRBuilder(block)
@ -151,6 +155,9 @@ def func_proc(tree, module, chunks, map_sym_tab):
if isinstance(decorator, ast.Name) and decorator.id == "map":
is_global = True
break
elif isinstance(decorator, ast.Name) and decorator.id == "bpfglobal":
is_global = True
break
if is_global:
continue
func_type = get_probe_string(func_node)