Use local Logger instead of the global one

src/binexport/__main__.py

@@ -12,6 +12,7 @@
 from multiprocessing import Pool, Queue, Manager
 import queue
 from binexport import ProgramBinExport
+from binexport.utils import logger
 
 BINARY_FORMAT = {
     "application/x-dosexec",
@@ -111,7 +112,7 @@ def main(ida_path: str, input_file: str, threads: int, verbose: bool) -> None:
         ingress.put(file)
         total += 1
 
-    logging.info(f"Start exporting {total} binaries")
+    logger.info(f"Start exporting {total} binaries")
 
     i = 0
     while True:
@@ -122,7 +123,7 @@ def main(ida_path: str, input_file: str, threads: int, verbose: bool) -> None:
             pp_res = Bcolors.OKGREEN + "OK" + Bcolors.ENDC
         else:
             pp_res = Bcolors.FAIL + "KO" + Bcolors.ENDC
-        logging.info(f"[{i}/{total}] {str(path) + '.BinExport'} [{pp_res}]")
+        logger.info(f"[{i}/{total}] {str(path) + '.BinExport'} [{pp_res}]")
         if i == total:
             break
 

src/binexport/expression.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
-import logging
 from functools import cached_property
 from typing import TYPE_CHECKING
 
 from binexport.binexport2_pb2 import BinExport2
 from binexport.types import ExpressionType
+from binexport.utils import logger
 
 if TYPE_CHECKING:
     from .program import ProgramBinExport
@@ -156,6 +156,4 @@ def _parse_protobuf(
             self._value = self.pb_expr.symbol
 
         else:
-            logging.error(
-                f"Malformed protobuf message. Invalid expression type {self.pb_expr.type}"
-            )
+            logger.error(f"Malformed protobuf message. Invalid expression type {self.pb_expr.type}")

src/binexport/function.py

@@ -1,11 +1,10 @@
 from __future__ import annotations
-import logging
 import weakref
 import networkx
 from functools import cached_property
 from typing import TYPE_CHECKING
 
-from binexport.utils import get_basic_block_addr
+from binexport.utils import get_basic_block_addr, logger
 from binexport.basic_block import BasicBlockBinExport
 from binexport.types import FunctionType
 
@@ -56,7 +55,7 @@ def __init__(
 
         if is_import:
             if self.addr is None:
-                logging.error("Missing function address for imported function")
+                logger.error("Missing function address for imported function")
             return
 
         assert pb_fun is not None, "pb_fun must be provided"
@@ -196,7 +195,7 @@ def blocks(self) -> dict[Addr, BasicBlockBinExport]:
             )
 
             if basic_block.addr in bblocks:
-                logging.error(
+                logger.error(
                     f"0x{self.addr:x} basic block address (0x{basic_block.addr:x}) already in(idx:{bb_idx})"
                 )
 

src/binexport/program.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 import pathlib
-import logging
 import networkx
 import weakref
 from collections import defaultdict
@@ -9,6 +8,7 @@
 from binexport.binexport2_pb2 import BinExport2
 from binexport.function import FunctionBinExport
 from binexport.types import FunctionType
+from binexport.utils import logger
 
 if TYPE_CHECKING:
     from binexport.types import Addr
@@ -66,7 +66,7 @@ def __init__(self, file: pathlib.Path | str):
         for i, pb_fun in enumerate(self.proto.flow_graph):
             f = FunctionBinExport(weakref.ref(self), pb_fun=pb_fun)
             if f.addr in self:
-                logging.error(f"Address collision for 0x{f.addr:x}")
+                logger.error(f"Address collision for 0x{f.addr:x}")
                 coll += 1
             self[f.addr] = f
             count_f += 1
@@ -81,7 +81,7 @@ def __init__(self, file: pathlib.Path | str):
                 )
                 count_imp += 1
             if node.address not in self:
-                logging.error(f"Missing function address: 0x{node.address:x} ({node.type})")
+                logger.error(f"Missing function address: 0x{node.address:x} ({node.type})")
                 continue
 
             self[node.address].type = FunctionType.from_proto(node.type)
@@ -103,7 +103,7 @@ def __init__(self, file: pathlib.Path | str):
         for f in self.values():
             self.fun_names[f.name] = f
 
-        logging.debug(
+        logger.debug(
             f"total all:{count_f}, imported:{count_imp} collision:{coll} (total:{count_f + count_imp + coll})"
         )
 
@@ -160,15 +160,15 @@ def from_binary_file(
 
         if retcode != 0 and not binexport_file.exists():
             # Still continue if retcode != 0, because idat64 something crashes but still manage to export file
-            logging.warning(
+            logger.warning(
                 f"{exec_file.name} failed to export [ret:{retcode}, binexport:{binexport_file.exists()}]"
             )
             return False
 
         if binexport_file.exists():
             return ProgramBinExport(binexport_file) if open_export else True
         else:
-            logging.error(f"{exec_file} can't find binexport generated")
+            logger.error(f"{exec_file} can't find binexport generated")
             return False
 
     @property

src/binexport/utils.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+import logging
 from collections.abc import Iterator
 from typing import TYPE_CHECKING
 
@@ -69,3 +70,7 @@ def instruction_index_range(rng: Binexport2.BasicBlock.IndexRange) -> Iterator[i
     :return: iterator over the indices
     """
     return range(rng.begin_index, (rng.end_index if rng.end_index else rng.begin_index + 1))
+
+
+# Main logger object
+logger = logging.getLogger("python-binexport")