如何在 Python 中安全地使用 exec()？

Question

我的任务是构建一个应用程序，最终用户可以在其中使用自定义规则来评估返回的查询是否导致警告或警报（基于自己的阈值）。

我已经为用户构建了一种方法来模板化他们的逻辑。一个示例如下所示：

if (abs(<<21>>) >= abs(<<22>>)):
    retVal = <<21>>
else:
    retVal = <<22>>

的<<21>>和<<22>>参数将与前面在该程序中找到的值来代替。一旦所有这些替换发生，我就有一个非常简单的 if/else 块（在本例中），它看起来像这样存储在变量 ( execCd) 中：

if (abs(22.0) >= abs(-162.0)):
    retVal = 22.0
else:
    retVal = -162.0

这将exec()正确。现在，我怎样才能确保这一点？我看过这篇文章：http : //lybniz2.sourceforge.net/safeeval.html

我的代码最终看起来像这样：

safe_list = ['math','acos', 'asin', 'atan', 'atan2', 'ceil', 'cos', 'cosh', 'de grees', 'e', 'exp', 'fabs', 'floor', 'fmod', 'frexp', 'hypot', 'ldexp', 'log', 'log10', 'modf', 'pi', 'pow', 'radians', 'sin', 'sinh', 'sqrt', 'tan', 'tanh'] 
safe_dict = dict([ (k, locals().get(k, None)) for k in safe_list ]) 
safe_dict['abs'] = abs
exec(execCd,{"__builtins__":None},safe_dict)

但是，当我有第二个和第三个参数时，exec 失败，但有这个异常 - NameError: name 'retVal' is not defined

最终用户拥有的一些自定义逻辑非常广泛，而且其中大部分都会定期更改。我不想维护他们的自定义逻辑，最终用户希望能够快速测试各种警告/警报阈值逻辑。

如何从不安全（有意或无意）代码中保护此 exec 语句？

Answer 1

使用eval或exec不使用它们的唯一安全方法。

您不需要使用 exec。不是构建要执行的字符串，而是将其解析为对象，并使用它来驱动代码执行。

最简单的是，您可以将函数存储在 dict 中，并使用字符串来选择要调用的函数。如果您使用的是 python 语法，python 提供了所有用于解析自身的实用程序，您应该使用它们。

Answer 2

您的exec声明不是将 retVal 添加到您的本地环境，而是添加到safe_dict字典中。所以你可以从那里取回它：

execCd = """
if (abs(22.0) >= abs(-162.0)):
    retVal = 22.0
else:
    retVal = -162.0
"""

safe_list = ['math','acos', 'asin', 'atan', 'atan2', 'ceil', 'cos', 'cosh', 'de grees', 'e', 'exp', 'fabs', 'floor', 'fmod', 'frexp', 'hypot', 'ldexp', 'log', 'log10', 'modf', 'pi', 'pow', 'radians', 'sin', 'sinh', 'sqrt', 'tan', 'tanh'] 
safe_dict = dict([ (k, locals().get(k, None)) for k in safe_list ]) 
safe_dict['abs'] = abs
exec(execCd,{"__builtins__":None},safe_dict)
retVal = safe_dict["retVal"]