Add and change error reporting for use in scripts.

2 files changed, 36 insertions, 11 deletions

diff --git a/src/rt_launch.c b/src/rt_launch.c
index 31e6447..c82248b 100644
--- a/src/rt_launch.c
+++ b/src/rt_launch.c
@@ -77,6 +77,7 @@ int main(int argc, char** argv)
         ret = __create_rt_task(launch, &info, cpu, wcet, period, class);
         if (ret < 0) {
                 perror("Could not create rt child process");
+                return 2;
         }
 
         return 0;       
diff --git a/src/set_rt_mode.c b/src/set_rt_mode.c
index 5936d7d..510da70 100644
--- a/src/set_rt_mode.c
+++ b/src/set_rt_mode.c
@@ -1,22 +1,46 @@
 #include <stdio.h>
 #include <strings.h>
 #include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+
 #include "litmus.h"
 
 
+void die(const char* str)
+{
+        fprintf(stderr, "%s\n%s\n", str, 
+                "Usage: set_rt_mode  [-v] {on|off}");
+        exit(127);
+}
+
 int main(int argc, char** argv) 
 {
-        if (argc == 2) {
-                if (!strcmp(argv[1], "on")) {
-                        printf("Enabling real-time mode.\n");
-                        set_rt_mode(MODE_RT_RUN);                       
-                        exit(0);
-                } else if (!strcmp(argv[1], "off")) {
-                        printf("Disabling real-time mode.\n");
-                        set_rt_mode(MODE_NON_RT);                       
-                        exit(0);
+        int ret;
+        int verbose = 0;
+        char* cmd = argv[1];
+
+        if (argc == 3 && !strcmp(argv[1], "-v")) {
+                verbose = 1;
+                cmd = argv[2];
+        } 
+        if (argc == 2 || verbose) {     
+                if (!strcmp(cmd, "on")) {
+                        if (verbose)
+                                printf("Enabling real-time mode.\n");
+                        ret = set_rt_mode(MODE_RT_RUN);                 
+                        if (ret && verbose) 
+                                perror("set_rt_mode failed");
+                        exit(ret ? errno : 0);
+                } else if (!strcmp(cmd, "off")) {
+                        if (verbose)
+                                printf("Disabling real-time mode.\n");
+                        ret = set_rt_mode(MODE_NON_RT); 
+                        if (ret && verbose) 
+                                perror("set_rt_mode failed");
+                        exit(ret ? errno : 0);          
                 } 
         }
-        printf("Usage: %s {on|off}\n", argv[0]);
-        return 1;       
+        die("Bad arguments.");
+        return 0;       
 }