angr 9.2.130__py3-none-manylinux2014_x86_64.whl → 9.2.132__py3-none-manylinux2014_x86_64.whl

angr/utils/bits.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import claripy
+
 
 def truncate_bits(value: int, nbits: int) -> int:
     """
@@ -14,3 +16,18 @@ def truncate_bits(value: int, nbits: int) -> int:
 
 def ffs(x: int) -> int:
     return (x & -x).bit_length() - 1
+
+
+def sign_extend(value: int, bits: int) -> int:
+    sign_bit = 1 << (bits - 1)
+    return (value & (sign_bit - 1)) - (value & sign_bit)
+
+
+def zeroextend_on_demand(op0: claripy.ast.BV, op1: claripy.ast.BV) -> claripy.ast.BV:
+    """
+    ZeroExtend op1 if the size of op1 is smaller than the size of op0. Otherwise, return op1.
+    """
+
+    if op0.size() > op1.size():
+        return claripy.ZeroExt(op0.size() - op1.size(), op1)
+    return op1

angr/utils/formatting.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+import os
 import sys
 from collections.abc import Sequence, Callable
 
@@ -23,7 +24,9 @@ def setup_terminal():
             colorama.init()
 
     global ansi_color_enabled  # pylint:disable=global-statement
-    ansi_color_enabled = isatty
+    # https://no-color.org/
+    no_color = os.environ.get("NO_COLOR", "")
+    ansi_color_enabled = isatty and not no_color
 
 
 def ansi_color(s: str, color: str | None) -> str:

angr/utils/orderedset.py

@@ -1,8 +1,11 @@
 from __future__ import annotations
+from typing import Generic, TypeVar
 import collections.abc
 
+T = TypeVar("T")
 
-class OrderedSet(collections.abc.MutableSet):
+
+class OrderedSet(Generic[T], collections.abc.MutableSet[T]):
     """
     Adapted from http://code.activestate.com/recipes/576694/
     Originally created by Raymond Hettinger and licensed under MIT.

angr/utils/ssa/__init__.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 from collections import defaultdict
-from typing import Any
+from typing import Any, Literal, overload
 
 import archinfo
 from ailment import Expression, Block
@@ -14,6 +14,16 @@ from .vvar_uses_collector import VVarUsesCollector
 from .tmp_uses_collector import TmpUsesCollector
 
 
+@overload
+def get_reg_offset_base_and_size(
+    reg_offset: int, arch: archinfo.Arch, size: int | None = None, resilient: Literal[True] = True
+) -> tuple[int, int]: ...
+@overload
+def get_reg_offset_base_and_size(
+    reg_offset: int, arch: archinfo.Arch, size: int | None = None, resilient: Literal[False] = False
+) -> tuple[int, int] | None: ...
+
+
 def get_reg_offset_base_and_size(
     reg_offset: int, arch: archinfo.Arch, size: int | None = None, resilient: bool = True
 ) -> tuple[int, int] | None:
@@ -34,9 +44,17 @@ def get_reg_offset_base_and_size(
     return base_reg_and_size
 
 
+@overload
 def get_reg_offset_base(
-    reg_offset: int, arch: archinfo.Arch, size: int | None = None, resilient: bool = True
-) -> int | None:
+    reg_offset: int, arch: archinfo.Arch, size: int | None = None, resilient: Literal[True] = True
+) -> int: ...
+@overload
+def get_reg_offset_base(
+    reg_offset: int, arch: archinfo.Arch, size: int | None = None, resilient: Literal[False] = False
+) -> int | None: ...
+
+
+def get_reg_offset_base(reg_offset, arch, size=None, resilient=True):
     """
     Translate a given register offset into the offset of its full register.
 

{angr-9.2.130.dist-info → angr-9.2.132.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: angr
-Version: 9.2.130
+Version: 9.2.132
 Summary: A multi-architecture binary analysis toolkit, with the ability to perform dynamic symbolic execution and various static analyses on binaries
 Home-page: https://github.com/angr/angr
 License: BSD-2-Clause
@@ -16,13 +16,13 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: CppHeaderParser
 Requires-Dist: GitPython
-Requires-Dist: ailment==9.2.130
-Requires-Dist: archinfo==9.2.130
+Requires-Dist: ailment==9.2.132
+Requires-Dist: archinfo==9.2.132
 Requires-Dist: cachetools
 Requires-Dist: capstone==5.0.3
 Requires-Dist: cffi>=1.14.0
-Requires-Dist: claripy==9.2.130
-Requires-Dist: cle==9.2.130
+Requires-Dist: claripy==9.2.132
+Requires-Dist: cle==9.2.132
 Requires-Dist: itanium-demangler
 Requires-Dist: mulpyplexer
 Requires-Dist: nampa
@@ -31,7 +31,7 @@ Requires-Dist: protobuf>=5.28.2
 Requires-Dist: psutil
 Requires-Dist: pycparser>=2.18
 Requires-Dist: pyformlang
-Requires-Dist: pyvex==9.2.130
+Requires-Dist: pyvex==9.2.132
 Requires-Dist: rich>=13.1.0
 Requires-Dist: sortedcontainers
 Requires-Dist: sympy