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

hardcoded func gen

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-06 02:38:59 +05:30
parent 8d7c869c47
commit c9e4910a89
4 changed files with 87 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pythonbpf/constants_pass.py
View File

@ -1,12 +1,14 @@
from llvmlite import ir
import ast
def emit_constants(module: ir.Module, constant_str: str, name: str):
constant_bytes = constant_str.encode("utf8") + b"\x00"
elems = [ir.Constant(ir.IntType(8), b) for b in constant_bytes]
ty = ir.ArrayType(ir.IntType(8), len(elems))
gvar = ir.GlobalVariable(module, ty, name=name)
print("constant emitted:", name)
gvar.initializer = ir.Constant(ty, elems) # type: ignore
@ -16,6 +18,7 @@ def emit_constants(module: ir.Module, constant_str: str, name: str):
return gvar
def constants_processing(tree, module):
"""Process string constants in the given AST tree and emit them to rodata"""
constant_count = 0
@ -37,7 +40,8 @@ def constants_processing(tree, module):
if constant_count == 0:
constant_name = f"{current_function}.____fmt"
else:
constant_name = f"{current_function}.____fmt.{constant_count}"
constant_name = f"""{current_function}.____fmt.{
constant_count}"""
emit_constants(module, node.value, constant_name)
constant_count += 1
self.generic_visit(node)
@ -45,4 +49,4 @@ def constants_processing(tree, module):
visitor = ConstantVisitor()
visitor.visit(tree)
return constant_count
return constant_count