X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=palacios%2Fsrc%2Finterfaces%2Fvmm_stream.c;h=3a3ccf761dd0a4fa919d1fb4325a6a2fa860b2b0;hb=4d1d8fadad33de7d3ebce2083d9782048f78b44e;hp=4b391ace2fc6fa49237b7f579cdc16821bb29213;hpb=504b1b73f85263b47bba96207b0a615933f87b15;p=palacios.git

diff --git a/palacios/src/interfaces/vmm_stream.c b/palacios/src/interfaces/vmm_stream.c
index 4b391ac..3a3ccf7 100644
--- a/palacios/src/interfaces/vmm_stream.c
+++ b/palacios/src/interfaces/vmm_stream.c
@@ -29,15 +29,20 @@ static struct v3_stream_hooks * stream_hooks = NULL;
 
 // VM can be NULL
 struct v3_stream * v3_stream_open(struct v3_vm_info * vm, const char * name, 
-				  uint64_t (*input)(struct v3_stream * stream, uint8_t * buf, uint64_t len),
+				  sint64_t (*input)(struct v3_stream * stream, uint8_t * buf, sint64_t len),
 				  void * guest_stream_data) {
     struct v3_stream * stream = NULL;
 
-    V3_ASSERT(stream_hooks != NULL);
-    V3_ASSERT(stream_hooks->open != NULL);
+    V3_ASSERT(vm, VCORE_NONE, stream_hooks != NULL);
+    V3_ASSERT(vm, VCORE_NONE, stream_hooks->open != NULL);
 
     stream = V3_Malloc(sizeof(struct v3_stream));
 
+    if (!stream) {
+	PrintError(vm, VCORE_NONE, "Cannot allocate in opening a stream\n");
+	return NULL;
+    }
+
     stream->input = input;
     stream->guest_stream_data = guest_stream_data;
     stream->host_stream_data = stream_hooks->open(stream, name, vm->host_priv_data);
@@ -45,16 +50,16 @@ struct v3_stream * v3_stream_open(struct v3_vm_info * vm, const char * name,
     return stream;
 }
 
-uint64_t v3_stream_output(struct v3_stream * stream, uint8_t * buf, uint32_t len) {
-    V3_ASSERT(stream_hooks != NULL);
-    V3_ASSERT(stream_hooks->output != NULL);
+sint64_t v3_stream_output(struct v3_stream * stream, uint8_t * buf, sint64_t len) {
+    V3_ASSERT(VM_NONE, VCORE_NONE, stream_hooks != NULL);
+    V3_ASSERT(VM_NONE, VCORE_NONE, stream_hooks->output != NULL);
 
     return stream_hooks->output(stream, buf, len);
 }
 
 void v3_stream_close(struct v3_stream * stream) {
-    V3_ASSERT(stream_hooks != NULL);
-    V3_ASSERT(stream_hooks->close != NULL);
+    V3_ASSERT(VM_NONE, VCORE_NONE, stream_hooks != NULL);
+    V3_ASSERT(VM_NONE, VCORE_NONE, stream_hooks->close != NULL);
 
     stream_hooks->close(stream);
 
@@ -65,7 +70,7 @@ void v3_stream_close(struct v3_stream * stream) {
 
 void V3_Init_Stream(struct v3_stream_hooks * hooks) {
     stream_hooks = hooks;
-    PrintDebug("V3 stream inited\n");
+    PrintDebug(VM_NONE, VCORE_NONE, "V3 stream inited\n");
 
     return;
 }