added virtio console, finished the stream interface implementation, added v3_stream...

[palacios.git] / linux_usr / v3_stream.c

diff --git a/linux_usr/v3_stream.c b/linux_usr/v3_stream.c
new file mode 100644 (file)
index 0000000..0a1f46a
--- /dev/null
+++ b/linux_usr/v3_stream.c
@@ -0,0 +1,113 @@
+/* 
+ * V3 Console utility
+ * (c) Jack lange & Lei Xia, 2010
+ */
+
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <fcntl.h> 
+#include <sys/ioctl.h> 
+#include <sys/stat.h> 
+#include <sys/types.h> 
+#include <unistd.h>
+#include <string.h>
+#include <fcntl.h>
+#include <pthread.h>
+#include <errno.h>
+#include<linux/unistd.h>
+#include <curses.h>
+
+
+#include "v3_ctrl.h"
+
+#define BUF_LEN 1025
+#define STREAM_NAME_LEN 128
+
+int main(int argc, char* argv[]) {
+    int vm_fd;
+    fd_set rset;
+    char * vm_dev = NULL;
+    char stream[STREAM_NAME_LEN];
+    char cons_buf[BUF_LEN];
+    int stream_fd = 0;
+
+    if (argc < 2) {
+       printf("Usage: ./v3_cons vm_device serial_number\n");
+       return -1;
+    }
+
+    vm_dev = argv[1];
+
+    if (strlen(argv[2]) >= STREAM_NAME_LEN) {
+       printf("ERROR: Stream name longer than maximum size (%d)\n", STREAM_NAME_LEN);
+       return -1;
+    }
+
+    memcpy(stream, argv[2], strlen(argv[2]));
+
+    vm_fd = open(vm_dev, O_RDONLY);
+    if (vm_fd == -1) {
+       printf("Error opening VM device: %s\n", vm_dev);
+       return -1;
+    }
+
+    stream_fd = ioctl(vm_fd, V3_VM_SERIAL_CONNECT, stream); 
+
+    /* Close the file descriptor.  */ 
+    close(vm_fd);
+ 
+    if (stream_fd < 0) {
+       printf("Error opening stream Console\n");
+       return -1;
+    }
+
+    while (1) {
+       int ret; 
+       int bytes_read = 0;
+       char in_buf[512];
+
+       memset(cons_buf, 0, BUF_LEN);
+
+
+       FD_ZERO(&rset);
+       FD_SET(stream_fd, &rset);
+       FD_SET(STDIN_FILENO, &rset);
+
+       ret = select(stream_fd + 1, &rset, NULL, NULL, NULL);
+       
+       if (ret == 0) {
+           continue;
+       } else if (ret == -1) {
+           perror("Select returned error\n");
+           return -1;
+       }
+
+       if (FD_ISSET(stream_fd, &rset)) {
+
+           bytes_read = read(stream_fd, cons_buf, BUF_LEN - 1);
+
+           cons_buf[bytes_read]='\0';
+           printf("%s", cons_buf);
+           fflush(stdout);
+
+       } else if (FD_ISSET(STDIN_FILENO, &rset)) {
+           fgets(in_buf, 512, stdin);
+
+           if (write(stream_fd, in_buf, strlen(in_buf)) != strlen(in_buf)) {
+               fprintf(stderr, "Error sending input bufer\n");
+               return -1;
+           }
+       } else {
+           printf("v3_cons ERROR: select returned %d\n", ret);
+           return -1;
+       }
+       
+
+    } 
+
+
+    return 0; 
+}
+
+