X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_usr%2Fv3_env_inject.c;h=f9eb854c9209960b53da8c079deebed17a3db8d8;hb=caf85567e859e58d73a0214ff2a89671692814a5;hp=466e8a5a5add8b419dc482b9e07b96d202693ec0;hpb=fd1de839d257d3ed2f0a8f27d231432b166b2cc1;p=palacios.git

diff --git a/linux_usr/v3_env_inject.c b/linux_usr/v3_env_inject.c
index 466e8a5..f9eb854 100644
--- 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++;
     }