70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Focused Visual C++ 6.0 C89 regression checks for moduleobject.c."""
|
|
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
source = Path(sys.argv[1]).read_text()
|
|
|
|
|
|
def function_body(name):
|
|
match = re.search(
|
|
re.escape(name) + r"\s*\([^;]*\)\s*\{(?P<body>.*?)\n\}",
|
|
source,
|
|
re.S,
|
|
)
|
|
assert match, name + " definition not found"
|
|
return match.group("body")
|
|
|
|
|
|
def ordered(body, tokens, message):
|
|
positions = [body.find(token) for token in tokens]
|
|
assert all(position >= 0 for position in positions) and positions == sorted(positions), message
|
|
|
|
|
|
body = function_body("_PyModule_IsExtension")
|
|
ordered(body, (
|
|
"PyModuleObject *module;",
|
|
"PyModuleDef *def;",
|
|
"if (!PyModule_Check(obj))",
|
|
"module = (PyModuleObject*)obj;",
|
|
"def = module->md_def;",
|
|
), "_PyModule_IsExtension must declare its locals before validation and preserve lookup order")
|
|
assert "PyModuleObject *module =" not in body and "PyModuleDef *def =" not in body
|
|
print("moduleobject _PyModule_IsExtension C89 locals and lookup order: PASS")
|
|
|
|
body = function_body("_PyModuleSpec_IsUninitializedSubmodule")
|
|
ordered(body, (
|
|
"PyObject *value;",
|
|
"int is_uninitialized;",
|
|
"if (spec == NULL)",
|
|
"value = PyObject_GetAttr(spec, &_Py_ID(_uninitialized_submodules));",
|
|
"is_uninitialized = PySequence_Contains(value, name);",
|
|
"Py_DECREF(value);",
|
|
), "_PyModuleSpec_IsUninitializedSubmodule must hoist declarations while preserving reference order")
|
|
assert "PyObject *value =" not in body and "int is_uninitialized =" not in body
|
|
print("moduleobject uninitialized-submodule C89 locals and reference order: PASS")
|
|
|
|
body = function_body("module_getattro")
|
|
ordered(body, (
|
|
"PyObject *attr, *mod_name, *getattr, *spec;",
|
|
"attr = PyObject_GenericGetAttr((PyObject *)m, name);",
|
|
"Py_INCREF(mod_name);",
|
|
"spec = PyDict_GetItemWithError(m->md_dict, &_Py_ID(__spec__));",
|
|
), "module_getattro must hoist spec without changing lookup timing")
|
|
assert "PyObject *spec =" not in body
|
|
print("moduleobject module_getattro C89 local and lookup order: PASS")
|
|
|
|
body = function_body("module_get_annotations")
|
|
ordered(body, (
|
|
"PyObject *dict;",
|
|
"PyObject *annotations;",
|
|
"int result;",
|
|
"dict = PyObject_GetAttr((PyObject *)m, &_Py_ID(__dict__));",
|
|
"annotations = PyDict_GetItemWithError(dict, &_Py_ID(__annotations__));",
|
|
"result = PyDict_SetItem(",
|
|
), "module_get_annotations must hoist locals and preserve dictionary/error order")
|
|
assert "PyObject *annotations =" not in body and "int result =" not in body
|
|
print("moduleobject module_get_annotations C89 locals and dictionary order: PASS")
|