AP_HAL_SITL: allow overriding args on reboot

magicrub · Sep 11, 2024 · 3ad0d11 · 3ad0d11
1 parent fcbf1a7
commit 3ad0d11
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/libraries/AP_HAL_SITL/HAL_SITL_Class.cpp b/libraries/AP_HAL_SITL/HAL_SITL_Class.cpp
@@ -305,6 +305,22 @@ void HAL_SITL::run(int argc, char * const argv[], Callbacks* callbacks) const
     actually_reboot();
 }
 
+void HAL_SITL::update_new_args(const char* opt, const char* val)
+{
+    for (uint8_t i=0; i<ARRAY_SIZE(new_argv)-1; i++) {
+        if (new_argv[i] == nullptr && new_argv[i+1] == nullptr) {
+            // add new option and value
+            new_argv[i] = (char*)opt;
+            new_argv[i+1] = (char*)val;
+            return;
+        } else if (!strcmp(new_argv[i], opt)) {
+            // update existing value
+            strncpy(new_argv[i+1], val, 100);
+            return;
+        }
+    }
+}
+
 void HAL_SITL::actually_reboot()
 {
     execv(new_argv[0], new_argv);

diff --git a/libraries/AP_HAL_SITL/HAL_SITL_Class.h b/libraries/AP_HAL_SITL/HAL_SITL_Class.h
@@ -13,6 +13,7 @@ class HAL_SITL : public AP_HAL::HAL {
     HAL_SITL();
     void run(int argc, char * const argv[], Callbacks* callbacks) const override;
     static void actually_reboot();
+    static void update_new_args(const char* opt, const char* val);
 
     void set_storage_posix_enabled(bool _enabled) {
         storage_posix_enabled = _enabled;