1 files changed, 9 insertions, 1 deletions
diff --git a/updater/updater.cpp b/updater/updater.cpp
index 0da7a4e82..9fa01a53c 100644
--- a/updater/updater.cpp
+++ b/updater/updater.cpp
@@ -53,7 +53,7 @@ int main(int argc, char** argv) {
     setbuf(stdout, NULL);
     setbuf(stderr, NULL);
 
-    if (argc != 4) {
+    if (argc != 4 && argc != 5) {
         printf("unexpected number of arguments (%d)\n", argc);
         return 1;
     }
@@ -146,6 +146,14 @@ int main(int argc, char** argv) {
     state.script = &script[0];
     state.errmsg = NULL;
 
+    if (argc == 5) {
+        if (strcmp(argv[4], "retry") == 0) {
+            state.is_retry = true;
+        } else {
+            printf("unexpected argument: %s", argv[4]);
+        }
+    }
+
     char* result = Evaluate(&state, root);
 
     if (have_eio_error) {