rpython: prevent defmacro! from mutating its argument

asarhaddon · Oct 15, 2024 · 5a010e7 · 5a010e7
1 parent 8deb5b6
commit 5a010e7
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/impls/rpython/step8_macros.py b/impls/rpython/step8_macros.py
@@ -93,8 +93,10 @@ def EVAL(ast, env):
             ast = quasiquote(ast[1]) # Continue loop (TCO)
         elif u"defmacro!" == a0sym:
             func = EVAL(ast[2], env)
-            func.ismacro = True
-            return env.set(ast[1], func)
+            return env.set(ast[1],
+                           MalFunc(func.fn, ast=func.ast, env=fn.env,
+                           params=fn.params, EvalFunc=fn.EvalFunc,
+                           ismacro=True))
         elif u"do" == a0sym:
             if len(ast) == 0:
                 return nil

diff --git a/impls/rpython/step9_try.py b/impls/rpython/step9_try.py
@@ -93,8 +93,10 @@ def EVAL(ast, env):
             ast = quasiquote(ast[1]) # Continue loop (TCO)
         elif u"defmacro!" == a0sym:
             func = EVAL(ast[2], env)
-            func.ismacro = True
-            return env.set(ast[1], func)
+            return env.set(ast[1],
+                           MalFunc(func.fn, ast=func.ast, env=fn.env,
+                           params=fn.params, EvalFunc=fn.EvalFunc,
+                           ismacro=True))
         elif u"try*" == a0sym:
             if len(ast) < 3:
                 return EVAL(ast[1], env);

diff --git a/impls/rpython/stepA_mal.py b/impls/rpython/stepA_mal.py
@@ -102,8 +102,10 @@ def EVAL(ast, env):
             ast = quasiquote(ast[1]) # Continue loop (TCO)
         elif u"defmacro!" == a0sym:
             func = EVAL(ast[2], env)
-            func.ismacro = True
-            return env.set(ast[1], func)
+            return env.set(ast[1],
+                           MalFunc(func.fn, ast=func.ast, env=fn.env,
+                           params=fn.params, EvalFunc=fn.EvalFunc,
+                           ismacro=True))
         elif u"try*" == a0sym:
             if len(ast) < 3:
                 return EVAL(ast[1], env);