fix: panic if plugin_fn applied to main fn (#12)

extism · Dec 14, 2022 · ec173ad · ec173ad
1 parent 1458d41
commit ec173ad
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/derive/src/lib.rs b/derive/src/lib.rs
@@ -30,6 +30,12 @@ pub fn plugin_fn(
         panic!("extism_pdk::plugin_fn expects a function with one argument, `()` may be used if no input is needed");
     }
 
+    if name == "main" {
+        panic!(
+            "extism_pdk::plugin_fn must not be applied to a `main` function. To fix, rename this to something other than `main`."
+        )
+    }
+
     match output {
         syn::ReturnType::Default => panic!(
             "extism_pdk::plugin_fn expects a return value, `()` may be used if no output is needed"