mirror of https://github.com/pwndbg/pwndbg.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
from unittest.mock import MagicMock
|
|
|
|
# Replace `pwndbg.commands` module with a mock to prevent import errors, as well
|
|
# as the `load_commands` function
|
|
module_name = "pwndbg.commands"
|
|
module = MagicMock(__name__=module_name, load_commands=lambda: None)
|
|
sys.modules[module_name] = module
|
|
|
|
# Load the mocks for the `gdb` and `gdblib` modules
|
|
import mocks.gdb
|
|
import mocks.gdblib # noqa: F401
|
|
|
|
# We must import the function under test after all the mocks are imported
|
|
from pwndbg.lib.memory import round_down
|
|
from pwndbg.lib.memory import round_up
|
|
|
|
|
|
def test_basic_rounding():
|
|
assert round_down(0xDEADBEEF, 0x1000) == 0xDEADB000
|
|
assert round_up(0xDEADBEEF, 0x1000) == 0xDEADC000
|
|
|
|
|
|
def test_many_rounding():
|
|
for n in range(0x100):
|
|
for i in range(8):
|
|
alignment = 1 << i
|
|
down = round_down(n, alignment)
|
|
up = round_up(n, alignment)
|
|
assert down <= n and down + alignment > n and down % alignment == 0
|
|
assert up >= n and up - alignment < n and up % alignment == 0
|