added sock_set functions for select() support

[palacios.git] / palacios / src / palacios / vmm_socket.c

diff --git a/palacios/src/palacios/vmm_socket.c b/palacios/src/palacios/vmm_socket.c
index 8546688..dbc19c0 100644 (file)
--- a/palacios/src/palacios/vmm_socket.c
+++ b/palacios/src/palacios/vmm_socket.c
@@ -19,6 +19,8 @@
 
 
 #include <palacios/vmm_socket.h>
+#include <palacios/vmm.h>
+
 
 struct v3_socket_hooks * sock_hooks = 0;
 
@@ -29,3 +31,79 @@ void V3_Init_Sockets(struct v3_socket_hooks * hooks) {
   PrintInfo("Initializing Socket Interface\n");
   sock_hooks = hooks;
 }
+
+
+
+void v3_init_sock_set(struct v3_sock_set * sock_set) {
+  sock_set->num_socks = 0;
+  sock_set->socks = NULL;
+
+}
+
+
+
+
+
+/* This should probably check if the socket is already added */
+// adds socket to the sockset
+void v3_set_sock(struct v3_sock_set * sock_set, V3_SOCK sock) {
+  struct v3_sock_entry * new_entry = V3_Malloc(sizeof(struct v3_sock_entry));
+
+  new_entry->sock = sock;
+  new_entry->is_set = 0;
+
+  if (sock_set->socks) {
+    new_entry->next = sock_set->socks;
+  }
+
+  sock_set->socks = new_entry;
+
+  sock_set->num_socks++;
+}
+
+
+// deletes socket from sockset
+void v3_clr_sock(struct v3_sock_set * sock_set, V3_SOCK sock) {
+  struct v3_sock_entry * iter, * back_ptr;
+
+  iter = sock_set->socks;
+  back_ptr = NULL;
+
+  v3_foreach_sock(sock_set, iter) {
+    if (iter->sock == sock) {
+      if (back_ptr == NULL) {
+       sock_set->socks = iter->next;
+      } else {
+       back_ptr->next = iter->next;
+      }
+
+      V3_Free(iter);
+
+      sock_set->num_socks--;
+      break;
+    }
+
+    back_ptr = iter;
+  }
+}
+
+// checks is_set vairable 
+int v3_isset_sock(struct v3_sock_set * sock_set, V3_SOCK sock) {
+  struct v3_sock_entry * iter;
+
+  v3_foreach_sock(sock_set, iter) {
+    if (iter->sock == sock) {
+      return iter->is_set;
+    }
+  }
+  return -1;
+}
+
+
+// clears all is_set variables.
+void v3_zero_sockset(struct v3_sock_set * sock_set) {
+  struct v3_sock_entry * iter;
+  v3_foreach_sock(sock_set, iter) {
+    iter->is_set = 0;
+  }
+}