Ran black formatter

Author: acezxn

src/agent/dfbscan.py

@@ -544,7 +544,7 @@ def start_scan_sequential(self) -> None:
                         sink_values,
                         call_statements,
                         ret_values,
-                        non_locals=[]
+                        non_locals=[],
                     )
 
                     # Invoke the intra-procedural data-flow analysis

src/llmtool/dfbscan/intra_dataflow_analyzer.py

@@ -19,7 +19,7 @@ def __init__(
         sink_values: List[Tuple[str, int]],
         call_statements: List[Tuple[str, int]],
         ret_values: List[Tuple[str, int]],
-        non_locals: List[Tuple[str, int]]
+        non_locals: List[Tuple[str, int]],
     ) -> None:
         self.function = function
         self.summary_start = summary_start
@@ -110,15 +110,15 @@ def _get_prompt(self, input: LLMToolInput) -> str:
         for ret_val in input.ret_values:
             rets_str += f"- {ret_val[0]} at line {ret_val[1]}\n"
         prompt = prompt.replace("<RETURN_VALUES>", rets_str)
-        
+
         if input.non_locals:
             non_local_str = "Non local variables relevant to this function:\n"
             for non_local in input.non_locals:
                 non_local_str += f"- {non_local[0]} at line {non_local[1]}\n"
             prompt = prompt.replace("<NONLOCAL_VALUES>", non_local_str)
         else:
             prompt = prompt.replace("<NONLOCAL_VALUES>", "")
-        
+
         return prompt
 
     def _parse_response(
@@ -236,7 +236,9 @@ def _parse_response(
                     )
                 elif detail["type"] == "Nonlocal":
                     reachable_values_per_path.add(
-                        Value(detail["name"], line_number, ValueLabel.NONLOCAL, file_path)
+                        Value(
+                            detail["name"], line_number, ValueLabel.NONLOCAL, file_path
+                        )
                     )
             reachable_values.append(reachable_values_per_path)
 

src/tstool/analyzer/Cpp_TS_analyzer.py

@@ -15,15 +15,15 @@ class Cpp_TSAnalyzer(TSAnalyzer):
     TSAnalyzer for C/C++ source files using tree-sitter.
     Implements language-specific parsing and analysis.
     """
-    
+
     def extract_scope_info(self, tree: tree_sitter.Tree) -> None:
         """
         Parse source code to extract scope topography.
         Currently Not implemented.
         :param tree: Parsed syntax tree
         """
         pass
-    
+
     def extract_nonlocal_info(self) -> None:
         """
         Traverse the scopes to identify declarations of non locals.
@@ -431,7 +431,7 @@ def get_loop_statements(
                 loop_body_end_line,
             )
         return loop_statements
-    
+
     def get_global_expressions_by_identifier(
         self, identifier: str, program_root: Node
     ) -> List[Node]:

src/tstool/analyzer/Go_TS_analyzer.py

@@ -15,15 +15,15 @@ class Go_TSAnalyzer(TSAnalyzer):
     TSAnalyzer for Go source files using tree-sitter.
     Implements Go-specific parsing and analysis.
     """
-    
+
     def extract_scope_info(self, tree: tree_sitter.Tree) -> None:
         """
         Parse source code to extract scope topography.
         Currently Not implemented.
         :param tree: Parsed syntax tree
         """
         pass
-    
+
     def extract_nonlocal_info(self) -> None:
         """
         Traverse the scopes to identify declarations of non locals.
@@ -375,4 +375,4 @@ def get_global_expressions_by_identifier(
         :param program_root: Program root node
         :return: A list of extracted nodes
         """
-        return []
+        return []

src/tstool/analyzer/Java_TS_analyzer.py

@@ -15,15 +15,15 @@ class Java_TSAnalyzer(TSAnalyzer):
     TSAnalyzer for Java source files using tree-sitter.
     Implements Java-specific parsing and analysis.
     """
-    
+
     def extract_scope_info(self, tree: tree_sitter.Tree) -> None:
         """
         Parse source code to extract scope topography.
         Currently Not implemented.
         :param tree: Parsed syntax tree
         """
         pass
-    
+
     def extract_nonlocal_info(self) -> None:
         """
         Traverse the scopes to identify declarations of non locals.
@@ -387,4 +387,4 @@ def get_global_expressions_by_identifier(
         :param program_root: Program root node
         :return: A list of extracted nodes
         """
-        return []
+        return []

src/tstool/analyzer/Javascript_TS_analyzer.py

@@ -161,7 +161,8 @@ def extract_nonlocal_info(self) -> None:
                         if (
                             candidate_parent is not None
                             and candidate_parent.type == "variable_declarator"
-                            and candidate_parent.child_by_field_name("name") is candidate_node
+                            and candidate_parent.child_by_field_name("name")
+                            is candidate_node
                         ):
                             continue
 

src/tstool/analyzer/Python_TS_analyzer.py

@@ -15,15 +15,15 @@ class Python_TSAnalyzer(TSAnalyzer):
     TSAnalyzer for Python source files using tree-sitter.
     Implements Python-specific parsing and analysis.
     """
-    
+
     def extract_scope_info(self, tree: tree_sitter.Tree) -> None:
         """
         Parse source code to extract scope topography
         :param tree: Parsed syntax tree
         """
         # TODO: Add scope extraction if needed
         pass
-    
+
     def extract_nonlocal_info(self) -> None:
         """
         Traverse the scopes to identify declarations of non locals
@@ -294,7 +294,7 @@ def get_loop_statements(
                 end_line,
             )
         return loops
-    
+
     def get_global_expressions_by_identifier(
         self, identifier: str, program_root: Node
     ) -> List[Node]:
@@ -305,4 +305,4 @@ def get_global_expressions_by_identifier(
         :return: A list of extracted nodes
         """
         # TODO: implement if needed
-        return []
+        return []

src/tstool/analyzer/TS_analyzer.py

@@ -177,16 +177,16 @@ def __init__(
         self.globals_env: Dict[int, Value] = {}
         self.scope_env: Dict[int, Tuple[Node, Set[int]]] = {}
         self.api_env: Dict[int, API] = {}
-        
+
         # Dictionary storing mapping from the root node of the scope to its scope id
         self.scope_root_to_scope_id: Dict[Node, int] = {}
-        
+
         # Dictionary storing mapping from function root node to its scope id
         self.function_root_to_scope_id: Dict[Node, int] = {}
-                
+
         # Dictionary storing mapping from a scope id to all the non locals it is depended on
         self.child_scope_id_to_non_locals: Dict[int, Set[Value]] = {}
-        
+
         # Dictionary storing mapping from a non local value to its child scopes
         self.non_local_to_child_scopes: Dict[Value, Set[int]] = {}
 
@@ -266,9 +266,9 @@ def parse_project(self) -> None:
                 self.fileContentDic[file_path] = source
                 pbar.update(1)
             pbar.close()
-            
+
         self.extract_nonlocal_info()
-        
+
         # Analyzes extracted functions
         with concurrent.futures.ThreadPoolExecutor(
             max_workers=self.max_symbolic_workers_num
@@ -288,7 +288,7 @@ def parse_project(self) -> None:
                 self.function_env[func_id] = current_function
                 pbar.update(1)
             pbar.close()
-            
+
         # Analyzes extracted global variables
         pbar = tqdm(
             total=len(self.globalsRawDataDic), desc="Analyzing Global Variables"
@@ -343,14 +343,14 @@ def extract_scope_info(self, tree: tree_sitter.Tree) -> None:
         :param tree: Parsed syntax tree
         """
         pass
-    
+
     @abstractmethod
     def extract_nonlocal_info(self) -> None:
         """
         Traverse the scopes to identify declarations of non locals
         """
         pass
-    
+
     @abstractmethod
     def extract_function_info(
         self, file_path: str, source_code: str, tree: Tree
@@ -730,7 +730,7 @@ def get_loop_statements(
         :return: A dictionary mapping (start_line, end_line) to loop statement info.
         """
         pass
-    
+
     @abstractmethod
     def get_global_expressions_by_identifier(
         self, identifier: str, program_root: Node
@@ -871,7 +871,7 @@ def get_function_global_value_reference(
                     references.setdefault(function, []).append(ref_value)
 
         return references
-        
+
     def get_function_from_localvalue(self, value: Value) -> Optional[Function]:
         """
         Retrieve the function corresponding to a local value.

src/tstool/dfbscan_extractor/Cpp/Cpp_MLK_extractor.py

@@ -10,7 +10,7 @@ def is_global_source(self, global_declarator_node: Node) -> bool:
         Currently not implemented.
         """
         return False
-    
+
     def extract_sources(self, function: Function) -> List[Value]:
         """
         Extract the sources that can cause the memory leak bugs from C/C++ programs.

src/tstool/dfbscan_extractor/Cpp/Cpp_UAF_extractor.py

@@ -12,7 +12,7 @@ def is_global_source(self, global_declarator_node: Node) -> bool:
         Currently not implemented.
         """
         return False
-    
+
     def extract_sources(self, function: Function) -> List[Value]:
         """
         Extract the sources that can cause the use-after-free bugs from C/C++ programs.