TermiNexus
b18d57fa38
背景（prof 基线 PROF/20260930_224857，主进程 433.69s）：正则这一族合计 ≈45s —— `re.match`（未预编译入口）被调 1970 万次 / 自身 10.70s、它内部 `Pattern.match` 占 8.06s， `re._compile`（缓存查找）1975 万次 / 7.19s，`re._make`（真正编译）274 万次 / 7.62s （说明 re 的 512 条缓存已被打穿，每 7 次调用就有 1 次重新编译）。 每次 `re.match(字面量, s)` 都要做"模式串哈希 → 缓存查找"；预编译后这层全省掉。 改动：把 15 个文件里 97 处"字面量当模式"的 `re.match/search/sub/subn/fullmatch/ findall/finditer/split(...)` 提升为模块级 `_RE_*: re.Pattern[str] = re.compile(...)`， 共 115 个调用点（含同字面量跨处共用）。模式串与 flags 一字未改（flags 从调用处 搬进 re.compile，含原先以**位置参数**传入的 flags）。4 处拼接型动态模式按设计不动 （FuncGen 2 处、HandlesImports 2 处）。 核验（不靠肉眼）：写了 AST 指纹对拍脚本，对每个文件取 HEAD 版与新工作区版，把 "旧 re.<m>(字面量, 其余实参, 关键字) + flags" 与 "新 _RE_X.<m>(实参, 关键字) + 常量 (模式, flags)" 做多重集比对 —— 15 个文件、115 个调用点全部逐一等价（0 失败）， 另查无重名常量、无遗留字面量调用、GetDiagnostics 全空。 对拍过程中查出一处子代理抄写错误：`_RE_GLOBAL_DSOLOCAL_MULTILINE` 丢了行首锚 `^` （原文注释明确"使用 ^ + MULTILINE 避免匹配字符串常量内容中的 define/declare"），已修正。 同轮经查证后**未做改动**的两项（避免把假热点当真事修）： - VLogger 的 `re.sub` 6.43s：profile 当时的 VLogger 版本里根本没有 `import re` （`git show 3658ab06:lib/core/VLogger.py` 已确认），属 C 帧错挂，不做。 - tokenize 274 万次：那是 `_generate_tokens_from_c_tokenizer` 的 **token 产出次数** （≈350 个整文件的词法量），SymbolUtils 的 4 个 tokenize 调用点 nc 仅数百，且已有 `_EXPAND_CACHE` 去重；再加一层 token 缓存无收益，不做。
623 lines
30 KiB
Python
623 lines
30 KiB
Python
from __future__ import annotations
|
||
from typing import TYPE_CHECKING, Any
|
||
import vast as ast
|
||
import re
|
||
import llvmlite.ir as ir
|
||
if TYPE_CHECKING:
|
||
from lib.core.translator import Translator
|
||
from lib.core.LlvmCodeGenerator import LlvmCodeGenerator
|
||
from lib.core.Handles.HandlesBase import BaseHandle
|
||
from lib.core.SymbolUtils import IsTModule
|
||
from lib.includes import t
|
||
|
||
# ---------------------------------------------------------------
|
||
# asm 模板重写用的正则**预编译**（性能：避免每次 `re.sub(字面量, ...)` 都要做
|
||
# "模式串哈希 → re 模块级缓存查找"。语义与逐字写法完全一致，模式串一字未改）。
|
||
# ---------------------------------------------------------------
|
||
# Intel 语法模板里的 `%0`/`%1` 占位操作数 → `$0`/`$1`
|
||
_RE_INTEL_FMT_OPERAND: re.Pattern[str] = re.compile(r'%(\d+)')
|
||
# Intel 语法模板里的 `%name` 具名操作数 → 去掉前缀 `%`
|
||
_RE_INTEL_FMT_NAMED: re.Pattern[str] = re.compile(r'%([a-zA-Z][a-zA-Z0-9]*)')
|
||
# asm 模板里 `call <符号>` 的函数名重写（保留 `call ` 前缀，只改写符号名）
|
||
_RE_ASM_CALL: re.Pattern[str] = re.compile(r'(call\s+)(\w+)')
|
||
|
||
# 猴子补丁 llvmlite.ir.InlineAsm 以支持 asm_dialect 参数
|
||
# 添加保护避免重复补丁
|
||
_InlineAsmPatched = getattr(ir.InlineAsm, '_transpyc_patched', False)
|
||
|
||
if not _InlineAsmPatched:
|
||
_OrigInlineAsmInit = ir.InlineAsm.__init__
|
||
_OrigInlineAsmDescr = ir.InlineAsm.descr
|
||
|
||
def _patched_inline_asm_init(self, ftype, asm, constraint, side_effect=False, asm_dialect=None):
|
||
_OrigInlineAsmInit(self, ftype, asm, constraint, side_effect)
|
||
self.asm_dialect = asm_dialect
|
||
|
||
def _patched_inline_asm_descr(self, buf):
|
||
sideeffect = 'sideeffect' if self.side_effect else ''
|
||
dialect_str = getattr(self, 'asm_dialect', None)
|
||
if dialect_str == 'intel':
|
||
dialect = 'inteldialect'
|
||
else:
|
||
dialect = ''
|
||
fmt = 'asm {sideeffect} {dialect} "{asm}", "{constraint}"'
|
||
buf.append(fmt.format(sideeffect=sideeffect, dialect=dialect, asm=self.asm,
|
||
constraint=self.constraint))
|
||
|
||
ir.InlineAsm.__init__ = _patched_inline_asm_init
|
||
ir.InlineAsm.descr = _patched_inline_asm_descr
|
||
ir.InlineAsm._transpyc_patched = True
|
||
|
||
|
||
class ExprAsmHandle(BaseHandle):
|
||
@staticmethod
|
||
def _format_to_dialect(asm_format: str | None) -> str | None:
|
||
"""将用户指定的 format 参数转换为 LLVM asm_dialect 值"""
|
||
fmt: str = asm_format.lower() if asm_format else 'intel'
|
||
if fmt == 'att':
|
||
return 'att'
|
||
elif fmt == 'arm':
|
||
return None # ARM 没有专门的 dialect 标记
|
||
else:
|
||
return 'intel' # 默认 Intel
|
||
|
||
def _HandleAsmLlvm(self, Node: ast.Call) -> ir.Value | None:
|
||
Gen: LlvmCodeGenerator = self.Trans.LlvmGen
|
||
# DEBUG: 验证 _HandleAsmLlvm 是否被调用
|
||
import sys as _dbg_sys
|
||
print(f"[DEBUG_ASM] _HandleAsmLlvm called, Node.args={len(Node.args)}, keywords={[kw.arg for kw in Node.keywords]}", file=_dbg_sys.stderr)
|
||
|
||
op_arg: Any = None
|
||
input_arg: Any = None
|
||
output_arg: Any = None
|
||
asm_format: str = 'Intel' # 汇编格式: Intel / AT&T / ARM
|
||
for kw in Node.keywords:
|
||
if kw.arg == 'op':
|
||
op_arg = kw.value
|
||
elif kw.arg in ('input', 'inp', 'inputs'):
|
||
input_arg = kw.value
|
||
elif kw.arg in ('output', 'out', 'outputs'):
|
||
output_arg = kw.value
|
||
elif kw.arg == 'format':
|
||
if isinstance(kw.value, ast.Constant) and isinstance(kw.value.value, str):
|
||
asm_format = kw.value.value
|
||
|
||
if len(Node.args) >= 1:
|
||
pos_op: Any = None
|
||
if len(Node.args) >= 2 and op_arg is None:
|
||
pos_op = Node.args[1]
|
||
return self._HandleAsmWithOpLlvm(Node.args[0], op_arg or pos_op, input_arg, output_arg, asm_format)
|
||
|
||
if len(Node.args) < 2:
|
||
return ir.Constant(ir.IntType(32), 1)
|
||
|
||
args: list[ast.AST] = Node.args
|
||
|
||
if len(args) == 4:
|
||
output_type_arg: ast.AST = args[0]
|
||
asm_template_arg: ast.AST = args[1]
|
||
constraint_arg: ast.AST = args[2]
|
||
clobber_arg: ast.AST = args[3]
|
||
|
||
output_type: ir.Type | None = self._infer_expr_llvm_type(output_type_arg)
|
||
if not output_type:
|
||
output_type = ir.IntType(32)
|
||
|
||
asm_template: str = ''
|
||
if isinstance(asm_template_arg, ast.Constant) and isinstance(asm_template_arg.value, str):
|
||
asm_template = self._prepare_intel_asm(self._mangle_asm_calls(asm_template_arg.value))
|
||
|
||
constraint: str = ''
|
||
if isinstance(constraint_arg, ast.Constant) and isinstance(constraint_arg.value, str):
|
||
constraint = constraint_arg.value
|
||
|
||
clobber: str = ''
|
||
if isinstance(clobber_arg, ast.Constant) and isinstance(clobber_arg.value, str):
|
||
clobber = clobber_arg.value
|
||
|
||
full_constraint: str = constraint
|
||
func_type: ir.FunctionType = ir.FunctionType(output_type, [])
|
||
asm_dialect: str | None = self._format_to_dialect(asm_format)
|
||
inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, constraint, side_effect=True, asm_dialect=asm_dialect)
|
||
result: Any = Gen.builder.call(inline_asm, [], name="asm_result")
|
||
return result
|
||
|
||
elif len(args) >= 2:
|
||
output_type_arg: ast.AST = args[0]
|
||
asm_template_arg: ast.AST = args[1]
|
||
|
||
output_type: ir.Type | None = self._infer_expr_llvm_type(output_type_arg)
|
||
if not output_type:
|
||
output_type = ir.IntType(32)
|
||
|
||
asm_template: str = ''
|
||
if isinstance(asm_template_arg, ast.Constant) and isinstance(asm_template_arg.value, str):
|
||
asm_template = self._prepare_intel_asm(self._mangle_asm_calls(asm_template_arg.value))
|
||
|
||
output_constraints: list[str] = []
|
||
output_values: list[Any] = []
|
||
if len(args) > 2 and isinstance(args[2], (ast.List, ast.Tuple)):
|
||
for item in args[2].elts:
|
||
if isinstance(item, (ast.List, ast.Tuple)) and len(item.elts) >= 2:
|
||
constraint_node: ast.AST = item.elts[0]
|
||
value_node: ast.AST = item.elts[1]
|
||
if isinstance(constraint_node, ast.Constant) and isinstance(constraint_node.value, str):
|
||
output_constraints.append(constraint_node.value)
|
||
value: Any = self.HandleExprLlvm(value_node)
|
||
if value:
|
||
output_values.append(value)
|
||
|
||
input_constraints: list[str] = []
|
||
input_values: list[Any] = []
|
||
if len(args) > 3 and isinstance(args[3], (ast.List, ast.Tuple)):
|
||
for item in args[3].elts:
|
||
if isinstance(item, (ast.List, ast.Tuple)) and len(item.elts) >= 2:
|
||
constraint_node: ast.AST = item.elts[0]
|
||
value_node: ast.AST = item.elts[1]
|
||
if isinstance(constraint_node, ast.Constant) and isinstance(constraint_node.value, str):
|
||
input_constraints.append(constraint_node.value)
|
||
value: Any = self.HandleExprLlvm(value_node)
|
||
if value:
|
||
input_values.append(value)
|
||
|
||
clobbers: list[str] = []
|
||
if len(args) > 4 and isinstance(args[4], (ast.List, ast.Tuple)):
|
||
for item in args[4].elts:
|
||
if isinstance(item, ast.Constant) and isinstance(item.value, str):
|
||
clobbers.append(item.value)
|
||
|
||
constraint_parts: list[str] = []
|
||
for oc in output_constraints:
|
||
constraint_parts.append(oc)
|
||
|
||
input_start_idx: int = len(output_constraints)
|
||
for ic in input_constraints:
|
||
constraint_parts.append(ic)
|
||
|
||
full_constraint: str = ",".join(constraint_parts)
|
||
|
||
if clobbers:
|
||
if full_constraint:
|
||
full_constraint += ","
|
||
for c in clobbers:
|
||
full_constraint += f"~{{{c}}}"
|
||
|
||
operands: list[Any] = output_values + input_values
|
||
param_types: list[ir.Type] = [v.type for v in operands]
|
||
func_type: ir.FunctionType = ir.FunctionType(output_type, param_types)
|
||
asm_dialect: str | None = self._format_to_dialect(asm_format)
|
||
inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, full_constraint, side_effect=False, asm_dialect=asm_dialect)
|
||
result: Any = Gen.builder.call(inline_asm, operands, name="asm_result")
|
||
return result
|
||
|
||
return None
|
||
|
||
def _prepare_intel_asm(self, template: str) -> str:
|
||
template = _RE_INTEL_FMT_OPERAND.sub(r'$\1', template)
|
||
template = _RE_INTEL_FMT_NAMED.sub(r'\1', template)
|
||
return template
|
||
|
||
def _normalize_asm_template(self, template: str) -> str:
|
||
lines = template.split('\n')
|
||
if len(lines) <= 1:
|
||
return template
|
||
first_line = lines[0]
|
||
rest_lines = lines[1:]
|
||
stripped = []
|
||
for line in rest_lines:
|
||
s = line.lstrip()
|
||
if s:
|
||
stripped.append(s)
|
||
else:
|
||
stripped.append('')
|
||
return '\n'.join([first_line] + stripped)
|
||
|
||
@staticmethod
|
||
def _intrinsic_marker(expr: ast.AST) -> str | None:
|
||
"""识别内建操作数标记调用，返回规范名或 None
|
||
|
||
支持两种等价写法（裸名写法省去 c. 前缀）：
|
||
- 内联汇编：c.AsmInp / c.AsmOut ↔ asminp / asmout
|
||
- LLVM IR：c.LInp / c.LOut ↔ irinp / irout
|
||
"""
|
||
if not isinstance(expr, ast.Call):
|
||
return None
|
||
func: ast.AST = expr.func
|
||
if isinstance(func, ast.Name):
|
||
if func.id == 'asminp':
|
||
return 'AsmInp'
|
||
if func.id == 'asmout':
|
||
return 'AsmOut'
|
||
if func.id == 'irinp':
|
||
return 'LInp'
|
||
if func.id == 'irout':
|
||
return 'LOut'
|
||
return None
|
||
if isinstance(func, ast.Attribute):
|
||
if isinstance(func.value, ast.Name) and func.value.id == 'c':
|
||
if func.attr in ('AsmInp', 'AsmOut', 'LInp', 'LOut'):
|
||
return func.attr
|
||
return None
|
||
return None
|
||
|
||
def _process_joined_str_template(self, node: ast.AST, out_input_ops: list[Any], out_input_constraints: list[str],
|
||
out_output_ops: list[Any], out_output_constraints: list[str]) -> str:
|
||
if not isinstance(node, ast.JoinedStr):
|
||
if isinstance(node, ast.Constant) and isinstance(node.value, str):
|
||
return node.value
|
||
return ''
|
||
|
||
raw_parts: list[tuple[str, str | int]] = []
|
||
for value in node.values:
|
||
if isinstance(value, ast.Constant) and isinstance(value.value, str):
|
||
raw_parts.append(('text', value.value))
|
||
elif isinstance(value, ast.FormattedValue):
|
||
expr: ast.AST = value.value
|
||
_marker: str | None = self._intrinsic_marker(expr)
|
||
if _marker == 'AsmInp':
|
||
if expr.args:
|
||
v: Any = self.HandleExprLlvm(expr.args[0])
|
||
if v:
|
||
c: str = 'r'
|
||
if len(expr.args) >= 2:
|
||
c = self._parse_asm_descr(expr.args[1])
|
||
if not c:
|
||
c = 'r'
|
||
out_input_ops.append(v)
|
||
out_input_constraints.append(c)
|
||
raw_parts.append(('input', len(out_input_ops) - 1))
|
||
elif _marker == 'AsmOut':
|
||
if expr.args:
|
||
target_ptr: Any = self._get_var_ptr_for_asm(expr.args[0])
|
||
if target_ptr:
|
||
v = target_ptr
|
||
else:
|
||
v = self.HandleExprLlvm(expr.args[0])
|
||
if v:
|
||
c: str = '=r'
|
||
if len(expr.args) >= 2:
|
||
c = self._parse_asm_descr(expr.args[1])
|
||
if not c:
|
||
c = '=r'
|
||
if not c.startswith('='):
|
||
c = '=' + c
|
||
out_output_ops.append(v)
|
||
out_output_constraints.append(c)
|
||
raw_parts.append(('output', len(out_output_ops) - 1))
|
||
elif not (isinstance(expr, ast.Call) and isinstance(expr.func, ast.Attribute)):
|
||
# 非 c.* 属性调用的表达式按普通输入操作数处理（保持原行为：
|
||
# c.* 属性调用中非 AsmInp/AsmOut 的形态不收集）
|
||
fallback: Any = self.HandleExprLlvm(expr)
|
||
if fallback and isinstance(fallback.type, ir.IntType):
|
||
out_input_ops.append(fallback)
|
||
out_input_constraints.append('r')
|
||
raw_parts.append(('input', len(out_input_ops) - 1))
|
||
else:
|
||
raw_parts.append(('text', ''))
|
||
|
||
num_outputs: int = len(out_output_ops)
|
||
parts: list[str] = []
|
||
for kind, data in raw_parts:
|
||
if kind == 'text':
|
||
parts.append(data)
|
||
elif kind == 'output':
|
||
parts.append(f'%{data}')
|
||
elif kind == 'input':
|
||
parts.append(f'%{num_outputs + data}')
|
||
return ''.join(parts)
|
||
|
||
_X86_REGISTERS = frozenset({
|
||
'rax', 'rbx', 'rcx', 'rdx', 'rsi', 'rdi', 'rbp', 'rsp',
|
||
'r8', 'r9', 'r10', 'r11', 'r12', 'r13', 'r14', 'r15',
|
||
'eax', 'ebx', 'ecx', 'edx', 'esi', 'edi', 'ebp', 'esp',
|
||
'ax', 'bx', 'cx', 'dx', 'si', 'di', 'bp', 'sp',
|
||
'al', 'bl', 'cl', 'dl', 'ah', 'bh', 'ch', 'dh',
|
||
'sil', 'dil', 'bpl', 'spl',
|
||
'r8d', 'r9d', 'r10d', 'r11d', 'r12d', 'r13d', 'r14d', 'r15d',
|
||
'r8w', 'r9w', 'r10w', 'r11w', 'r12w', 'r13w', 'r14w', 'r15w',
|
||
'r8b', 'r9b', 'r10b', 'r11b', 'r12b', 'r13b', 'r14b', 'r15b',
|
||
})
|
||
|
||
def _mangle_asm_calls(self, template: str) -> str:
|
||
Gen: LlvmCodeGenerator = self.Trans.LlvmGen
|
||
sha1: str = getattr(Gen, 'module_sha1', '')
|
||
if not sha1:
|
||
return template
|
||
|
||
def replace_call(m: re.Match[str]) -> str:
|
||
prefix: str = m.group(1)
|
||
func_name: str = m.group(2)
|
||
if func_name.lower() in self._X86_REGISTERS:
|
||
return m.group(0)
|
||
mangled: str = Gen._mangle_func_name(func_name)
|
||
if '.' in mangled:
|
||
return f'{prefix}\\22{mangled}\\22'
|
||
return f"{prefix}{mangled}"
|
||
|
||
return _RE_ASM_CALL.sub(replace_call, template)
|
||
|
||
def _HandleAsmWithOpLlvm(self, asm_template_arg: ast.AST, op_arg: ast.AST | None = None, input_arg: ast.AST | None = None, output_arg: ast.AST | None = None, asm_format: str = 'Intel') -> ir.Value:
|
||
Gen: LlvmCodeGenerator = self.Trans.LlvmGen
|
||
|
||
input_ops: list[Any] = []
|
||
input_constraints: list[str] = []
|
||
output_ops: list[Any] = []
|
||
output_constraints: list[str] = []
|
||
|
||
if output_arg and isinstance(output_arg, (ast.List, ast.Tuple)):
|
||
for item in output_arg.elts:
|
||
if self._intrinsic_marker(item) == 'AsmOut':
|
||
if len(item.args) >= 1:
|
||
target_ptr: Any = self._get_var_ptr_for_asm(item.args[0])
|
||
if target_ptr:
|
||
output_ops.append(target_ptr)
|
||
else:
|
||
value: Any = self.HandleExprLlvm(item.args[0])
|
||
if value:
|
||
output_ops.append(value)
|
||
constraint: str = '=r'
|
||
if len(item.args) >= 2:
|
||
base_constraint: str = self._parse_asm_descr(item.args[1])
|
||
if base_constraint:
|
||
if base_constraint.startswith('='):
|
||
constraint = base_constraint
|
||
else:
|
||
constraint = '=' + base_constraint
|
||
output_constraints.append(constraint)
|
||
|
||
raw_template: str
|
||
if isinstance(asm_template_arg, ast.JoinedStr):
|
||
raw_template = self._normalize_asm_template(
|
||
self._process_joined_str_template(
|
||
asm_template_arg, input_ops, input_constraints, output_ops, output_constraints
|
||
)
|
||
)
|
||
elif isinstance(asm_template_arg, ast.Constant) and isinstance(asm_template_arg.value, str):
|
||
raw_template = self._normalize_asm_template(asm_template_arg.value)
|
||
else:
|
||
raw_template = ''
|
||
|
||
raw_template = self._mangle_asm_calls(raw_template)
|
||
asm_dialect: str | None = self._format_to_dialect(asm_format)
|
||
asm_template: str
|
||
if asm_format == 'AT&T':
|
||
asm_template = raw_template # AT&T 不需要转换
|
||
elif asm_format == 'ARM':
|
||
asm_template = raw_template # ARM 不需要转换
|
||
else:
|
||
asm_template = self._prepare_intel_asm(raw_template) # Intel → LLVM IR 格式
|
||
|
||
if input_arg and isinstance(input_arg, (ast.List, ast.Tuple)):
|
||
for item in input_arg.elts:
|
||
if self._intrinsic_marker(item) == 'AsmInp':
|
||
if len(item.args) >= 1:
|
||
value: Any = self.HandleExprLlvm(item.args[0])
|
||
if value:
|
||
input_ops.append(value)
|
||
constraint: str = 'r'
|
||
if len(item.args) >= 2:
|
||
constraint = self._parse_asm_descr(item.args[1])
|
||
if not constraint:
|
||
constraint = 'r'
|
||
input_constraints.append(constraint)
|
||
elif isinstance(item, (ast.List, ast.Tuple)) and len(item.elts) >= 2:
|
||
value_node: ast.AST = item.elts[0]
|
||
descr_node: ast.AST = item.elts[1]
|
||
value: Any = self.HandleExprLlvm(value_node)
|
||
if value:
|
||
input_ops.append(value)
|
||
constraint: str = self._parse_asm_descr(descr_node)
|
||
if not constraint:
|
||
constraint = 'r'
|
||
input_constraints.append(constraint)
|
||
|
||
clobbers: list[str] = []
|
||
if op_arg and isinstance(op_arg, (ast.List, ast.Tuple)):
|
||
for item in op_arg.elts:
|
||
clobber: str = self._parse_asm_descr(item)
|
||
if clobber:
|
||
clobbers.append(clobber)
|
||
|
||
constraint_parts: list[str] = []
|
||
for oc in output_constraints:
|
||
c: str = oc if oc.startswith('=') else ('=' + oc)
|
||
constraint_parts.append(c)
|
||
for ic in input_constraints:
|
||
if ic in ('d', 'edx', 'rdx'):
|
||
constraint_parts.append('{dx}')
|
||
else:
|
||
constraint_parts.append(ic)
|
||
|
||
full_constraint: str = ",".join(constraint_parts)
|
||
|
||
constraint_to_reg: dict[str, str] = {
|
||
'a': 'rax', 'b': 'rbx', 'c': 'rcx', 'd': 'rdx',
|
||
'S': 'rsi', 'D': 'rdi',
|
||
'A': 'rax', 'U': 'r8',
|
||
}
|
||
used_regs: set[str] = set()
|
||
for ic in input_constraints:
|
||
if ic in constraint_to_reg:
|
||
used_regs.add(constraint_to_reg[ic])
|
||
for oc in output_constraints:
|
||
base_oc: str = oc.lstrip('=').lstrip('+')
|
||
if base_oc in constraint_to_reg:
|
||
used_regs.add(constraint_to_reg[base_oc])
|
||
|
||
if clobbers:
|
||
has_memory: bool = 'memory' in clobbers
|
||
if has_memory and output_ops:
|
||
clobbers = [c for c in clobbers if c != 'memory']
|
||
clobber_final: list[str]
|
||
if asm_format == 'ARM':
|
||
# ARM 不需要 x86 的 e→r 寄存器名转换
|
||
clobber_final = [c for c in clobbers if c not in used_regs]
|
||
else:
|
||
clobber_64bit: list[str] = []
|
||
for c in clobbers:
|
||
if c in ['eax', 'ebx', 'ecx', 'edx', 'esi', 'edi', 'ebp', 'esp']:
|
||
clobber_64bit.append(c.replace('e', 'r'))
|
||
else:
|
||
clobber_64bit.append(c)
|
||
clobber_final = [c for c in clobber_64bit if c not in used_regs]
|
||
if clobber_final:
|
||
if full_constraint:
|
||
full_constraint += ","
|
||
full_constraint += ",".join([f"~{{{c}}}" for c in clobber_final])
|
||
|
||
if output_ops:
|
||
single_out: bool = len(output_ops) == 1
|
||
if single_out:
|
||
op: Any = output_ops[0]
|
||
ret_type: ir.Type
|
||
if isinstance(op, ir.AllocaInstr):
|
||
ret_type = op.type.pointee
|
||
elif isinstance(op, ir.GlobalVariable):
|
||
ret_type = op.type.pointee
|
||
elif isinstance(op, (ir.GEPInstr, ir.CastInstr)):
|
||
ret_type = op.type.pointee if isinstance(op.type, ir.PointerType) else op.type
|
||
else:
|
||
ret_type = op.type
|
||
else:
|
||
def _get_ret_type(op: Any) -> ir.Type:
|
||
if isinstance(op, (ir.AllocaInstr, ir.GlobalVariable)):
|
||
return op.type.pointee
|
||
return op.type
|
||
ret_type = ir.LiteralStructType([_get_ret_type(op) for op in output_ops])
|
||
if isinstance(ret_type, ir.PointerType) and isinstance(ret_type.pointee, ir.IdentifiedStructType):
|
||
ret_type = ir.IntType(64)
|
||
operands: list[Any] = input_ops
|
||
param_types: list[ir.Type] = [op.type for op in operands]
|
||
func_type: ir.FunctionType = ir.FunctionType(ret_type, param_types)
|
||
inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, full_constraint, side_effect=len(output_ops) == 0, asm_dialect=asm_dialect)
|
||
result: Any = Gen.builder.call(inline_asm, operands, name="asm_result")
|
||
if single_out:
|
||
target: Any = output_ops[0]
|
||
if isinstance(target, ir.AllocaInstr):
|
||
if target.type.pointee != result.type:
|
||
i64t: ir.IntType = ir.IntType(64)
|
||
iv: Any = Gen.builder.ptrtoint(result, i64t)
|
||
ptr: Any = Gen.builder.bitcast(target, ir.PointerType(i64t))
|
||
Gen.builder.store(iv, ptr)
|
||
else:
|
||
Gen.builder.store(result, target)
|
||
elif isinstance(target, (ir.GlobalVariable, ir.GEPInstr, ir.CastInstr)):
|
||
target_pointee: ir.Type = target.type.pointee
|
||
if target_pointee != result.type:
|
||
if isinstance(target_pointee, ir.PointerType) and isinstance(result, ir.Instruction) and result.type == ir.IntType(64):
|
||
ptr_val: Any = Gen.builder.inttoptr(result, target_pointee)
|
||
Gen.builder.store(ptr_val, target)
|
||
else:
|
||
i64t: ir.IntType = ir.IntType(64)
|
||
iv: Any = Gen.builder.ptrtoint(result, i64t)
|
||
ptr: Any = Gen.builder.bitcast(target, ir.PointerType(i64t))
|
||
Gen.builder.store(iv, ptr)
|
||
else:
|
||
Gen.builder.store(result, target)
|
||
else:
|
||
dst: Any = Gen._allocaEntry(result.type)
|
||
Gen.builder.store(result, dst)
|
||
else:
|
||
for i, out_op in enumerate(output_ops):
|
||
val: Any = Gen.builder.extract_value(result, i, name=f"asm_out_{i}")
|
||
target: Any = out_op if isinstance(out_op, ir.AllocaInstr) else Gen._allocaEntry(val.type)
|
||
if isinstance(target, ir.AllocaInstr) and target.type.pointee != val.type:
|
||
i64t: ir.IntType = ir.IntType(64)
|
||
iv: Any = Gen.builder.ptrtoint(val, i64t)
|
||
ptr: Any = Gen.builder.bitcast(target, ir.PointerType(i64t))
|
||
Gen.builder.store(iv, ptr)
|
||
else:
|
||
Gen.builder.store(val, target)
|
||
else:
|
||
operands: list[Any] = input_ops
|
||
param_types: list[ir.Type] = [op.type for op in operands]
|
||
func_type: ir.FunctionType = ir.FunctionType(ir.VoidType(), param_types)
|
||
inline_asm: ir.InlineAsm = ir.InlineAsm(func_type, asm_template, full_constraint, side_effect=True, asm_dialect=asm_dialect)
|
||
Gen.builder.call(inline_asm, operands, name="asm_call")
|
||
return ir.Constant(ir.IntType(32), 1)
|
||
|
||
def _get_var_ptr_for_asm(self, node: ast.AST) -> ir.Value | None:
|
||
Gen: LlvmCodeGenerator = self.Trans.LlvmGen
|
||
if isinstance(node, ast.Name):
|
||
var_name: str = node.id
|
||
if var_name in Gen.variables and Gen.variables[var_name] is not None:
|
||
ptr: Any = Gen.variables[var_name]
|
||
if isinstance(ptr, ir.AllocaInstr):
|
||
return ptr
|
||
if isinstance(ptr, ir.GlobalVariable):
|
||
return ptr
|
||
elif isinstance(node, ast.Attribute):
|
||
obj_ptr: Any = self._get_var_ptr_for_asm(node.value)
|
||
if obj_ptr is not None:
|
||
obj_type: ir.Type = obj_ptr.type.pointee
|
||
if isinstance(obj_type, ir.PointerType) and isinstance(obj_type.pointee, ir.IdentifiedStructType):
|
||
Loaded: Any = Gen.builder.load(obj_ptr, name="asm_attr_Load")
|
||
struct_type: ir.IdentifiedStructType = obj_type.pointee
|
||
class_name: str | None = Gen._resolve_class_for_var(node.value.id) if isinstance(node.value, ast.Name) else None
|
||
if class_name:
|
||
offset: int = Gen._get_member_offset(node.attr, class_name)
|
||
gep: Any = Gen.builder.gep(Loaded, [ir.Constant(ir.IntType(32), 0),
|
||
ir.Constant(ir.IntType(32), offset)],
|
||
inbounds=True)
|
||
return gep
|
||
for idx, elem_type in enumerate(struct_type.elements):
|
||
gep: Any = Gen.builder.gep(Loaded, [ir.Constant(ir.IntType(32), 0),
|
||
ir.Constant(ir.IntType(32), idx)],
|
||
inbounds=True)
|
||
return gep
|
||
elif isinstance(obj_type, ir.IdentifiedStructType):
|
||
class_name: str | None = Gen._resolve_class_for_var(node.value.id) if isinstance(node.value, ast.Name) else None
|
||
if class_name:
|
||
offset: int = Gen._get_member_offset(node.attr, class_name)
|
||
gep: Any = Gen.builder.gep(obj_ptr, [ir.Constant(ir.IntType(32), 0),
|
||
ir.Constant(ir.IntType(32), offset)],
|
||
inbounds=True)
|
||
return gep
|
||
for idx, elem_type in enumerate(obj_type.elements):
|
||
gep: Any = Gen.builder.gep(obj_ptr, [ir.Constant(ir.IntType(32), 0),
|
||
ir.Constant(ir.IntType(32), idx)],
|
||
inbounds=True)
|
||
return gep
|
||
return None
|
||
|
||
def _parse_asm_descr(self, expr: ast.AST) -> str:
|
||
if isinstance(expr, ast.BinOp):
|
||
left_val: str = self._parse_asm_descr(expr.left)
|
||
right_val: str = self._parse_asm_descr(expr.right)
|
||
return left_val + right_val
|
||
|
||
elif isinstance(expr, ast.Attribute):
|
||
if isinstance(expr.value, ast.Attribute):
|
||
if (isinstance(expr.value.value, ast.Name) and
|
||
IsTModule(expr.value.value.id, self.Trans.SymbolTable) and
|
||
expr.value.attr == 'ASM_DESCR'):
|
||
AttrName: str = expr.attr
|
||
return getattr(t.ASM_DESCR, AttrName, "")
|
||
elif isinstance(expr.value, ast.Name) and (expr.value.id == 'ASM_DESCR' or IsTModule(expr.value.id, self.Trans.SymbolTable)):
|
||
# 裸名 ASM_DESCR.XXX：省去 t. 前缀的等价写法
|
||
AttrName: str = expr.attr
|
||
return getattr(t.ASM_DESCR, AttrName, "")
|
||
|
||
elif isinstance(expr, ast.Constant):
|
||
return expr.value
|
||
|
||
return ""
|
||
|
||
def _infer_expr_llvm_type(self, expr: ast.AST) -> ir.Type | None:
|
||
if isinstance(expr, ast.Name):
|
||
type_name: str = expr.id
|
||
if hasattr(t, type_name):
|
||
ctype: Any = getattr(t, type_name)
|
||
if isinstance(ctype, type) and issubclass(ctype, t.CType):
|
||
size: Any = getattr(ctype, 'Size', None)
|
||
if size is not None:
|
||
return ir.IntType(size)
|
||
if type_name == 'CFloat':
|
||
return ir.FloatType()
|
||
elif type_name == 'CDouble':
|
||
return ir.DoubleType()
|
||
elif isinstance(expr, ast.Attribute):
|
||
if isinstance(expr.value, ast.Name) and IsTModule(expr.value.id, self.Trans.SymbolTable):
|
||
return self._infer_expr_llvm_type(ast.Name(id=expr.attr, ctx=ast.Load()))
|
||
return None
|