mirror of
https://github.com/varun-r-mallya/Python-BPF.git
synced 2025-12-31 21:06:25 +00:00
remove constants pass as it's functionality has been replicated
elsewhere
This commit is contained in:
@ -3,7 +3,6 @@ from llvmlite import ir
|
||||
from .license_pass import license_processing
|
||||
from .functions_pass import func_proc
|
||||
from .maps_pass import maps_proc
|
||||
# from .constants_pass import constants_processing
|
||||
from .globals_pass import globals_processing
|
||||
|
||||
|
||||
@ -29,12 +28,9 @@ def processor(source_code, filename, module):
|
||||
|
||||
map_sym_tab = maps_proc(tree, module, bpf_chunks)
|
||||
func_proc(tree, module, bpf_chunks, map_sym_tab)
|
||||
# For now, we will parse the BPF specific parts of AST
|
||||
|
||||
# constants_processing(tree, module)
|
||||
license_processing(tree, module)
|
||||
globals_processing(tree, module)
|
||||
# functions_processing(tree, module)
|
||||
|
||||
|
||||
def compile_to_ir(filename: str, output: str):
|
||||
|
||||
@ -1,52 +0,0 @@
|
||||
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
|
||||
|
||||
gvar.align = 1 # type: ignore
|
||||
gvar.linkage = "internal" # type: ignore
|
||||
gvar.global_constant = True
|
||||
|
||||
return gvar
|
||||
|
||||
|
||||
def constants_processing(tree, module):
|
||||
"""Process string constants in the given AST tree and emit them to rodata"""
|
||||
constant_count = 0
|
||||
current_function = None
|
||||
|
||||
class ConstantVisitor(ast.NodeVisitor):
|
||||
def visit_FunctionDef(self, node):
|
||||
nonlocal current_function
|
||||
old_function = current_function
|
||||
current_function = node.name
|
||||
for child in ast.iter_child_nodes(node):
|
||||
if not (hasattr(node, 'decorator_list') and child in node.decorator_list):
|
||||
self.visit(child)
|
||||
current_function = old_function
|
||||
|
||||
def visit_Constant(self, node):
|
||||
nonlocal constant_count
|
||||
if isinstance(node.value, str) and current_function is not None:
|
||||
if constant_count == 0:
|
||||
constant_name = f"{current_function}.____fmt"
|
||||
else:
|
||||
constant_name = f"""{current_function}.____fmt.{
|
||||
constant_count}"""
|
||||
emit_constants(module, node.value, constant_name)
|
||||
constant_count += 1
|
||||
self.generic_visit(node)
|
||||
|
||||
visitor = ConstantVisitor()
|
||||
visitor.visit(tree)
|
||||
|
||||
return constant_count
|
||||
Reference in New Issue
Block a user