Linux userspace tools for live migration

[palacios.git] / linux_usr / v3_env_inject.c

diff --git a/linux_usr/v3_env_inject.c b/linux_usr/v3_env_inject.c
index 466e8a5..f9eb854 100644 (file)
--- a/linux_usr/v3_env_inject.c
+++ b/linux_usr/v3_env_inject.c
@@ -18,7 +18,7 @@
 
 
 static void usage (char* bin) {
-       fprintf(stderr, "Usage: %s /dev/v3-vm<N> env-file inject-point-exe\n", bin);
+       fprintf(stderr, "usage: %s /dev/v3-vm<N> env-file inject-point-exe\n", bin);
 }
 
 int main (int argc, char **argv) {
@@ -52,6 +52,10 @@ int main (int argc, char **argv) {
         if (tmp_str[len] == '\n')
             tmp_str[len] = 0;
         strings[i] = (char*)malloc(MAX_STRING_LEN);
+        if (!strings[i]) {
+                fprintf(stderr, "Error allocating space for variable\n");
+                return -1;
+        }
         strcpy(strings[i], tmp_str);
         i++;
     }