如何在CPython中将全局变量缓存为本地const以使用LOAD_CONST指令？

可以做到，核心思路是直接修改函数的字节码和常量池——这两个信息存在函数的__code__属性对应的CodeType对象中，无需修改或重新编译CPython解释器。以下是具体实现步骤和代码：

原理概述 #

函数执行时的指令序列（字节码）和常量池都是CodeType对象的不可变属性。我们需要：

• 找到字节码中加载目标全局变量的LOAD_GLOBAL指令
• 将其替换为加载常量的LOAD_CONST指令
• 把全局变量的值加入到函数的常量池中
• 用新的CodeType对象替换原函数的__code__属性

具体实现代码 #

原代码（待修改的函数） #

a = 7

def f():
    return a + 1

修改函数字节码的代码 #

import types

# 获取原函数的代码对象
orig_code = f.__code__

# 1. 定位加载全局变量'a'的LOAD_GLOBAL指令
# 先拿到'a'在co_names中的索引
name_index = orig_code.co_names.index('a')
# Python 3.11中LOAD_GLOBAL的操作码是0x6C，后跟双字节小端序的操作数
load_global_bytes = bytes([0x6C, name_index & 0xFF, (name_index >> 8) & 0xFF])

# 在字节码中找到该指令的位置
code_bytearray = bytearray(orig_code.co_code)
instruction_pos = code_bytearray.find(load_global_bytes)
if instruction_pos == -1:
    raise ValueError("未找到加载目标全局变量的LOAD_GLOBAL指令")

# 2. 更新常量池，加入全局变量a的值
new_consts = orig_code.co_consts + (a,)
new_const_index = len(new_consts) - 1  # 新常量的索引

# 3. 替换LOAD_GLOBAL为LOAD_CONST指令
# Python 3.11中LOAD_CONST的操作码是0x64，后跟双字节小端序的操作数
load_const_bytes = bytes([0x64, new_const_index & 0xFF, (new_const_index >> 8) & 0xFF])
code_bytearray[instruction_pos:instruction_pos+3] = load_const_bytes

# 4. 创建新的CodeType对象（3.11的参数顺序必须严格对应）
new_code = types.CodeType(
    orig_code.co_argcount,
    orig_code.co_posonlyargcount,
    orig_code.co_kwonlyargcount,
    orig_code.co_nlocals,
    orig_code.co_stacksize,
    orig_code.co_flags,
    bytes(code_bytearray),
    new_consts,
    orig_code.co_names,
    orig_code.co_varnames,
    orig_code.co_filename,
    orig_code.co_name,
    orig_code.co_qualname,
    orig_code.co_firstlineno,
    orig_code.co_lnotab,
    orig_code.co_exceptiontable,
    orig_code.co_freevars,
    orig_code.co_cellvars
)

# 5. 替换原函数的__code__属性
f.__code__ = new_code

验证修改结果 #

运行以下代码验证：

# 查看函数的常量池，现在包含7
print(f.__code__.co_consts)  # 输出示例：(None, 1, 7)

# 反编译字节码，确认使用LOAD_CONST加载7
import dis
dis.dis(f)

反编译后的输出会类似：

4           0 LOAD_CONST               2 (7)
              2 LOAD_CONST               1 (1)
              4 BINARY_ADD
              6 RETURN_VALUE

可以看到已经没有LOAD_GLOBAL指令，而是通过LOAD_CONST直接加载常量7。

注意事项 #

• 操作必须在函数定义后、调用前执行，因为我们捕获的是函数定义时全局变量的当前值。
• 仅适用于全局变量的值后续不会改变的场景，因为我们使用的是定义时的副本，而非引用。
• 该代码仅适配Python 3.11.x版本——不同Python版本的字节码操作码、参数格式（比如操作数字节数）可能不同，需要对应调整。
• 修改字节码时要确保操作数索引正确，否则会导致函数执行报错。

内容的提问来源于stack exchange，提问作者Liam