From 77e73d70a9b81a7bbd8e49be52612fc62a9f9502 Mon Sep 17 00:00:00 2001
From: Qrius <qrius@quirio.net>
Date: Tue, 6 May 2025 14:22:33 +0200
Subject: Disable unsafe code execution by default

---
 src/smp/__init__.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/smp/__init__.py')

diff --git a/src/smp/__init__.py b/src/smp/__init__.py
index f7cb937..2cdf2fe 100644
--- a/src/smp/__init__.py
+++ b/src/smp/__init__.py
@@ -41,6 +41,13 @@ def main():
         action="store_true",
         help="Prefix builtins with smp_",
     )
+    parser.add_argument(
+        "-U",
+        "--unsafe",
+        default=False,
+        action="store_true",
+        help="Allow unsafe code execution",
+    )
     parser.add_argument(
         "file", nargs="?", default=None, help='Input file or "-" for stdin'
     )
@@ -50,6 +57,8 @@ def main():
     macro_processor_state = smp.macro_processor.MacroProcessorState()
     prefix = "smp_" if args.prefix_builtins else ""
     macro_processor = macro_processor_state.macro_processor(prefix=prefix)
+    if args.unsafe:
+        macro_processor.unsafe_code_execution = True
 
     for key, value in args.D.items():
         macro_processor.define_macro(key, value)
-- 
cgit v1.2.3