40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Guard only insert_to_usedpool's entry locals and original operation order."""
|
|
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
source = Path(sys.argv[1]).read_text()
|
|
match = re.search(
|
|
r"\ninsert_to_usedpool\(poolp pool\)\s*\{(.*?)\n\}",
|
|
source, re.S,
|
|
)
|
|
assert match, "insert_to_usedpool definition not found"
|
|
body = re.sub(r"/\*.*?\*/", "", match.group(1), flags=re.S)
|
|
declarations = """
|
|
uint size;
|
|
poolp next;
|
|
poolp prev;
|
|
"""
|
|
assert body.split()[:len(declarations.split())] == declarations.split(), (
|
|
"insert_to_usedpool must declare size, next, and prev without initializers at function entry"
|
|
)
|
|
|
|
# Keep the assertion before size evaluation and both lookups, then preserve
|
|
# the four pointer-link writes in their original order.
|
|
expected = declarations + """
|
|
assert(pool->ref.count > 0);
|
|
size = pool->szidx;
|
|
next = usedpools[size + size];
|
|
prev = next->prevpool;
|
|
pool->nextpool = next;
|
|
pool->prevpool = prev;
|
|
next->prevpool = pool;
|
|
prev->nextpool = pool;
|
|
"""
|
|
assert body.split() == expected.split(), (
|
|
"preserve insert_to_usedpool assertion, size/lookup evaluation, and pointer-link mutation order"
|
|
)
|
|
print("obmalloc insert_to_usedpool entry declarations and original operation order: PASS")
|