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

Remove debug prints and add @bpfglobal support

Browse Source
This commit is contained in:
varun-r-mallya
2025-09-08 20:16:31 +05:30
parent 627ee9265a
commit 83937dc11a
2 changed files with 14 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
pythonbpf/globals_pass.py
View File

@ -16,9 +16,6 @@ def emit_globals(module: ir.Module, names: list[str]):
g = module.get_global(name)
else:
g = ir.GlobalValue(module, ptr_ty, name)
print("global emitted:", name)
print(isinstance(g, ir.GlobalVariable))
print(isinstance(g, ir.Function))
elems.append(g.bitcast(ptr_ty))
gv = ir.GlobalVariable(module, used_array_ty, "llvm.compiler.used")
@ -31,16 +28,19 @@ def globals_processing(tree, module: ir.Module):
collected = ["LICENSE"]
for node in tree.body:
if isinstance(node, ast.FunctionDef) and len(node.decorator_list) == 2:
dec = node.decorator_list[1]
if (
isinstance(dec, ast.Call)
and isinstance(dec.func, ast.Name)
and dec.func.id == "section"
and len(dec.args) == 1
and isinstance(dec.args[0], ast.Constant)
and isinstance(dec.args[0].value, str)
):
collected.append(node.name)
if isinstance(node, ast.FunctionDef):
for dec in node.decorator_list:
if (
isinstance(dec, ast.Call)
and isinstance(dec.func, ast.Name)
and dec.func.id == "section"
and len(dec.args) == 1
and isinstance(dec.args[0], ast.Constant)
and isinstance(dec.args[0].value, str)
):
collected.append(node.name)
elif isinstance(dec, ast.Name) and dec.id == "bpfglobal":
collected.append(node.name)
emit_globals(module, collected)