diff --git a/pythonbpf/codegen.py b/pythonbpf/codegen.py index 36f7a37..5de23a5 100644 --- a/pythonbpf/codegen.py +++ b/pythonbpf/codegen.py @@ -62,12 +62,14 @@ def compile_to_ir(filename: str, output: str, loglevel=logging.WARNING): if not hasattr(module, "_debug_compile_unit"): debug_generator = DebugInfoGenerator(module) debug_generator.generate_file_metadata(filename, os.path.dirname(filename)) - debug_generator.generate_debug_cu(DW_LANG_C11, - f"PythonBPF {VERSION}", - True # TODO: This is probably not true - # TODO: add a global field here that keeps track of all the globals. Works without it, but I think it might - # be required for kprobes. - , True) + debug_generator.generate_debug_cu( + DW_LANG_C11, + f"PythonBPF {VERSION}", + True, # TODO: This is probably not true + # TODO: add a global field here that keeps track of all the globals. Works without it, but I think it might + # be required for kprobes. + True, + ) processor(source, filename, module) @@ -129,7 +131,7 @@ def compile(loglevel=logging.WARNING) -> bool: success = True success = ( - compile_to_ir(str(caller_file), str(ll_file), loglevel=loglevel) and success + compile_to_ir(str(caller_file), str(ll_file), loglevel=loglevel) and success ) success = bool( @@ -156,7 +158,7 @@ def BPF(loglevel=logging.WARNING) -> BpfProgram: caller_frame = inspect.stack()[1] src = inspect.getsource(caller_frame.frame) with tempfile.NamedTemporaryFile( - mode="w+", delete=True, suffix=".py" + mode="w+", delete=True, suffix=".py" ) as f, tempfile.NamedTemporaryFile( mode="w+", delete=True, suffix=".ll" ) as inter, tempfile.NamedTemporaryFile( diff --git a/pythonbpf/debuginfo/debug_info_generator.py b/pythonbpf/debuginfo/debug_info_generator.py index 6de705b..ab9fed4 100644 --- a/pythonbpf/debuginfo/debug_info_generator.py +++ b/pythonbpf/debuginfo/debug_info_generator.py @@ -21,7 +21,9 @@ class DebugInfoGenerator: }, ) - def generate_debug_cu(self, language, producer: str, is_optimized: bool, is_distinct: bool): + def generate_debug_cu( + self, language, producer: str, is_optimized: bool, is_distinct: bool + ): self.module._debug_compile_unit = self.module.add_debug_info( "DICompileUnit", { # type: ignore @@ -100,7 +102,7 @@ class DebugInfoGenerator: ) def create_struct_type( - self, members: List[Any], size: int, is_distinct: bool + self, members: List[Any], size: int, is_distinct: bool ) -> Any: """Create a struct type with the given members and size""" return self.module.add_debug_info( @@ -115,7 +117,7 @@ class DebugInfoGenerator: ) def create_global_var_debug_info( - self, name: str, var_type: Any, is_local: bool = False + self, name: str, var_type: Any, is_local: bool = False ) -> Any: """Create debug info for a global variable""" global_var = self.module.add_debug_info(