|
 |
7444ce |
diff --git a/Tests/test_imagemath.py b/Tests/test_imagemath.py
|
|
|
7444ce |
index 2329b74..d3b7ba3 100644
|
|
|
7444ce |
--- a/Tests/test_imagemath.py
|
|
|
7444ce |
+++ b/Tests/test_imagemath.py
|
|
|
7444ce |
@@ -58,6 +58,12 @@ class TestImageMath(PillowTestCase):
|
|
|
7444ce |
self.assertEqual(pixel(
|
|
|
7444ce |
ImageMath.eval("float(B)**33", images)), "F 8589934592.0")
|
|
|
7444ce |
|
|
|
7444ce |
+ def test_prevent_exec(self):
|
|
|
7444ce |
+ self.assertRaises(ValueError, ImageMath.eval("exec('pass')"))
|
|
|
7444ce |
+ self.assertRaises(ValueError, ImageMath.eval("(lambda: exec('pass'))()"))
|
|
|
7444ce |
+ self.assertRaises(ValueError, ImageMath.eval("(lambda: (lambda: exec('pass'))())()"))
|
|
|
7444ce |
+
|
|
|
7444ce |
+
|
|
|
7444ce |
def test_logical(self):
|
|
|
7444ce |
self.assertEqual(pixel(ImageMath.eval("not A", images)), 0)
|
|
|
7444ce |
self.assertEqual(pixel(ImageMath.eval("A and B", images)), "L 2")
|
|
|
7444ce |
diff --git a/src/PIL/ImageMath.py b/src/PIL/ImageMath.py
|
|
|
7444ce |
index c5bea70..13839e4 100644
|
|
|
7444ce |
--- a/src/PIL/ImageMath.py
|
|
|
7444ce |
+++ b/src/PIL/ImageMath.py
|
|
|
7444ce |
@@ -263,7 +263,18 @@ def eval(expression, _dict={}, **kw):
|
|
|
7444ce |
if hasattr(v, "im"):
|
|
|
7444ce |
args[k] = _Operand(v)
|
|
|
7444ce |
|
|
|
7444ce |
- out = builtins.eval(expression, args)
|
|
|
7444ce |
+ compiled_code = compile(expression, "<string>", "eval")
|
|
|
7444ce |
+ def scan(code):
|
|
|
7444ce |
+ for const in code.co_consts:
|
|
|
7444ce |
+ if type(const) == type(compiled_code):
|
|
|
7444ce |
+ scan(const)
|
|
|
7444ce |
+
|
|
|
7444ce |
+ for name in code.co_names:
|
|
|
7444ce |
+ if name not in args and name != "abs":
|
|
|
7444ce |
+ raise ValueError(f"'{name}' not allowed")
|
|
|
7444ce |
+
|
|
|
7444ce |
+ scan(compiled_code)
|
|
|
7444ce |
+ out = builtins.eval(expression, {"__builtins": {"abs": abs}}, args)
|
|
|
7444ce |
try:
|
|
|
7444ce |
return out.im
|
|
|
7444ce |
except AttributeError:
|