main-func: implicitly save argc/argv in DEFINE_MAIN_FUNCTION() functions

Let's remove the risk of forgetting to save argc/argv if
DEFINE_MAIN_FUNCTION() is used.

diff --git a/src/nspawn/nspawn.c b/src/nspawn/nspawn.c
index 19eb5b6..dfc4f68 100644 (file)
--- a/src/nspawn/nspawn.c
+++ b/src/nspawn/nspawn.c
@@ -4667,9 +4667,6 @@ static int run(int argc, char *argv[]) {
         log_parse_environment();
         log_open();
 
-        /* Make sure rename_process() in the stub init process can work */
-        save_argc_argv(argc, argv);
-
         r = parse_argv(argc, argv);
         if (r <= 0)
                 goto finish;

diff --git a/src/shared/main-func.h b/src/shared/main-func.h
index 3c182e8..486976f 100644 (file)
--- a/src/shared/main-func.h
+++ b/src/shared/main-func.h
@@ -8,10 +8,12 @@
 #include "spawn-ask-password-agent.h"
 #include "spawn-polkit-agent.h"
 #include "static-destruct.h"
+#include "util.h"
 
 #define _DEFINE_MAIN_FUNCTION(intro, impl, ret)                         \
         int main(int argc, char *argv[]) {                              \
                 int r;                                                  \
+                save_argc_argv(argc, argv);                             \
                 intro;                                                  \
                 r = impl;                                               \
                 static_destruct();                                      \