Implement task timeout

* Add argument: --task-timeout=1[s,m,h]
* Timed out tasks are SIGKILL'd
* If killing a task fails, the entire program ends

1 files changed, 12 insertions, 0 deletions

diff --git a/src/cli/stasis/stasis_main.c b/src/cli/stasis/stasis_main.c
index 967ecaf..44ee6d7 100644
--- a/src/cli/stasis/stasis_main.c
+++ b/src/cli/stasis/stasis_main.c
@@ -540,6 +540,18 @@ int main(int argc, char *argv[]) {
             case OPT_FAIL_FAST:
                 globals.parallel_fail_fast = true;
                 break;
+            case OPT_TASK_TIMEOUT:
+                globals.task_timeout = str_to_timeout(optarg);
+                if (globals.task_timeout < 0) {
+                    fprintf(stderr, "Invalid timeout: %s\n", optarg);
+                    if (globals.task_timeout == STR_TO_TIMEOUT_INVALID_TIME_SCALE) {
+                        fprintf(stderr, "Use format '#s' (seconds), '#m' (minutes), '#h' (hours)\n");
+                    } else if (globals.task_timeout == STR_TO_TIMEOUT_NEGATIVE) {
+                        fprintf(stderr, "Timeout cannot be negative\n");
+                    }
+                    exit(1);
+                }
+                break;
             case OPT_POOL_STATUS_INTERVAL:
                 globals.pool_status_interval = (int) strtol(optarg, NULL, 10);
                 if (globals.pool_status_interval < 1) {