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

34
pythonbpf/license_pass.py
View File

@ -18,20 +18,28 @@ def emit_license(module: ir.Module, license_str: str):
return gvar
def license_processing(tree, module):
"""Process the LICENSE assignment in the given AST tree and return the section name"""
"""Process the LICENSE function decorated with @bpf and @bpfglobal and return the section name"""
count = 0
for node in tree.body:
if isinstance(node, ast.Assign):
for target in node.targets:
if isinstance(target, ast.Name) and target.id == "LICENSE":
if count == 0:
count += 1
if isinstance(node.value, ast.Constant) and isinstance(node.value.value, str):
emit_license(module, node.value.value)
return "LICENSE"
else:
print("ERROR: LICENSE must be a string literal")
return None
if isinstance(node, ast.FunctionDef) and node.name == "LICENSE":
# check decorators
decorators = [dec.id for dec in node.decorator_list if isinstance(dec, ast.Name)]
if "bpf" in decorators and "bpfglobal" in decorators:
if count == 0:
count += 1
# check function body has a return string
if (
len(node.body) == 1
and isinstance(node.body[0], ast.Return)
and isinstance(node.body[0].value, ast.Constant)
and isinstance(node.body[0].value.value, str)
):
emit_license(module, node.body[0].value.value)
return "LICENSE"
else:
print("ERROR: LICENSE already assigned")
print("ERROR: LICENSE() must return a string literal")
return None
else:
print("ERROR: LICENSE already defined")
return None
return None