X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_usr%2Fv3_ctrl.c;h=984045bf8b826f29b4b82ac7aa54ac13167daeaf;hb=caf85567e859e58d73a0214ff2a89671692814a5;hp=c8bf216722a0e43fb32684c7e5a4fa6565c770be;hpb=4801e692b7344051eb94fff6faf1be53f621d422;p=palacios.git

diff --git a/linux_usr/v3_ctrl.c b/linux_usr/v3_ctrl.c
index c8bf216..984045b 100644
--- a/linux_usr/v3_ctrl.c
+++ b/linux_usr/v3_ctrl.c
@@ -28,7 +28,7 @@ int main(int argc, char* argv[]) {
     memset(&guest_img, 0, sizeof(struct v3_guest_img));
 
     if (argc <= 2) {
-	printf("Usage: ./v3_ctrl <guest_img> <vm name>\n");
+	printf("usage: v3_ctrl <guest_img> <vm name>\n");
 	return -1;
     }
 
@@ -51,6 +51,10 @@ int main(int argc, char* argv[]) {
     
     // load guest image into user memory
     guest_img.guest_data = malloc(guest_img.size);
+    if (!guest_img.guest_data) {
+        printf("ERROR: Could not allocate memory for guest image\n");
+        return -1;
+    }
 
     read_file(guest_fd, guest_img.size, guest_img.guest_data);