raw: Update list-based socket management functions.
[swifty.git] / src / raw / swift_list.c
index 786c6fc..04832ac 100644 (file)
@@ -1,14 +1,8 @@
 /*
- * swift implementation of syscall API
+ * Linked list management for sockets
  *
- * Simulates the classic socket syscalls (socket, bind, send, recv).
- * Implementation uses raw sockets (AF_INET, SOCK_RAW).
- *
- * Subsequently, implementation is to be ported into kernel space and
- * the interface is going to be offered by the Linux syscall API.
- *
- * Heavily inspired by GLIBC's <sys/socket.h>
- * (/usr/include/sys/socket.h).
+ * Use sock_list structure to store information about sockets. Use functions
+ * to add, remove, find and update information in list.
  *
  * 2011, Razvan Deaconescu, razvan.deaconescu@cs.pub.ro
  */
 #include <unistd.h>
 #include <errno.h>
 
-#include "include/swift_types.h"
-#include "include/swift_list.h"
+#include "swift_types.h"
+#include "socket_manager.h"
 #include "debug.h"
+#include "util.h"
+
+enum sock_bind_state {
+       STATE_NOTBOUND,
+       STATE_BOUND
+};
+
+/* socket management structure */
+struct sock_list {
+       int s;
+       struct sockaddr addr;
+       enum sock_bind_state bind_state;
+       struct sock_list *next;
+       struct sock_list *prev;
+};
+
+/* socket management list head */
+static struct sock_list sock_list_head = {
+       .next = &sock_list_head,
+       .prev = &sock_list_head
+};
 
 /*
- * Add new socket to list. Called by sw_socket "syscall".
+ * Find socket in socket management list.
  */
 
-struct sock_list *list_add_socket(int s)
+static struct sock_list *list_get_link(int s)
 {
-        struct sock_list *ptr = malloc(sizeof(*ptr));
-        if (ptr == NULL)
-                return NULL;
+       struct sock_list *ptr;
 
-        ptr->next = &sock_list_head;
-        ptr->prev = sock_list_head.prev;
-        sock_list_head.prev->next = ptr;
-        sock_list_head.prev = ptr;
-        ptr->s = s;
+       for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
+               if (ptr->s == s)
+                       return ptr;
 
-        return ptr;
+       return NULL;
 }
 
 /*
- * Bind socket to given address. Called by sw_bind "syscall".
+ * Find socket in socket management list by address.
  */
 
-struct sock_list *list_update_socket_address(int s, __CONST_SOCKADDR_ARG addr)
+static struct sock_list *list_get_link_by_address(const struct sockaddr *addr)
 {
-        struct sock_list *ptr;
+       struct sock_list *ptr;
 
-        for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
-                if (ptr->s == s) {
-                        memcpy(&ptr->addr, addr, sizeof(ptr->addr));
-                       ptr->bind_state = STATE_BOUND;
-                        return ptr;
-                }
+       for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
+               if (memcmp(addr, &ptr->addr, sizeof(*addr)) == 0)
+                       return ptr;
 
-        return NULL;
+       return NULL;
 }
 
 /*
- * Get list element containing socket s. Called by sw_send* "syscalls".
+ * Unlink socket from list. Called by sm_del.
  */
 
-struct sock_list *list_elem_from_socket(int s)
+static void list_unlink(struct sock_list *ptr)
 {
-        struct sock_list *ptr;
+       ptr->next->prev = ptr->prev;
+       ptr->prev->next = ptr->next;
+       ptr->next = ptr;
+       ptr->prev = ptr;
+}
 
-        for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
-                if (ptr->s == s)
-                        return ptr;
+/*
+ * Link socket to list. Add socket to tail of list.
+ */
 
-        return NULL;
+static void list_link(struct sock_list *ptr)
+{
+       ptr->next = &sock_list_head;
+       ptr->prev = sock_list_head.prev;
+       sock_list_head.prev->next = ptr;
+       sock_list_head.prev = ptr;
 }
 
 /*
- * Get list element containing address addr. Called by sw_bind "syscall".
+ * Add new socket to list. Called by sw_socket "syscall".
  */
 
-struct sock_list *list_elem_from_address(__CONST_SOCKADDR_ARG addr)
+int sm_add(int s)
 {
-        struct sock_list *ptr;
-
-        for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next) {
-               dprintf("socket address to be checked\n");
-                if (ptr->bind_state == STATE_NOTBOUND)
-                        continue;
-               dprintf("bound socket address to be checked\n");
-                if (memcmp(&ptr->addr, addr, sizeof(addr)) == 0)
-                        return ptr;
-        }
-
-        return NULL;
+       struct sock_list *ptr = malloc(sizeof(*ptr));
+       if (ptr == NULL)
+               return -1;
+
+       ptr->s = s;
+       list_link(ptr);
+
+       return 0;
 }
 
 /*
- * Unlink socket from list. Called by list_remove_socket.
+ * Bind socket to given address. Called by sw_bind "syscall".
  */
-static struct sock_list *list_unlink_socket(int s)
+
+int sm_update_address(int s, const struct sockaddr *addr)
 {
-        struct sock_list *ptr;
-
-        for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
-                if (ptr->s == s) {
-                        ptr->next->prev = ptr->prev;
-                        ptr->prev->next = ptr->next;
-                        ptr->next = ptr;
-                        ptr->prev = ptr;
-                        return ptr;
-                }
-
-        return NULL;
+       struct sock_list *ptr;
+
+       ptr = list_get_link(s);
+       if (ptr == NULL)
+               return -1;
+
+       memcpy(&ptr->addr, addr, sizeof(ptr->addr));
+
+       return 0;
 }
 
 /*
  * Remove socket from list. Called by sw_close "syscall".
  */
 
-int list_remove_socket(int s)
+int sm_del(int s)
 {
-        struct sock_list *ptr;
+       struct sock_list *ptr;
+
+       ptr = list_get_link(s);
+       if (ptr == NULL)
+               return -1;
 
-        ptr = list_unlink_socket(s);
-        if (ptr == NULL)
-                return -1;
+       list_unlink(ptr);
+       free(ptr);
 
-        free(ptr);
-        return 0;
+       return 0;
 }
 
 /*
  * Check if a socket is bound.
  */
-int list_socket_is_bound(int s)
+
+int sm_is_bound(int s)
+{
+       struct sock_list *ptr;
+
+       ptr = list_get_link(s);
+       if (ptr == NULL)
+               return 0;
+
+       if (ptr->bind_state == STATE_BOUND)
+               return 1;
+
+       return 0;
+}
+
+/*
+ * Mark socket as bound.
+ */
+
+int sm_mark_bound(int s)
+{
+       struct sock_list *ptr;
+       
+       ptr = list_get_link(s);
+       if (ptr == NULL)
+               return -1;
+
+       ptr->bind_state = STATE_BOUND;
+
+       return 0;
+}
+
+/*
+ * Mark socket as unbound.
+ */
+
+int sm_mark_unbound(int s)
+{
+       struct sock_list *ptr;
+
+       ptr = list_get_link(s);
+       if (ptr == NULL)
+               return -1;
+
+       ptr->bind_state = STATE_NOTBOUND;
+
+       return 0;
+}
+
+/*
+ * Check if adress is asociated with a given socket.
+ */
+
+int sm_address_exists(const struct sockaddr *addr)
+{
+       struct sock_list *ptr;
+
+       ptr = list_get_link_by_address(addr);
+       if (ptr == NULL)
+               return 0;
+
+       return 1;
+}
+
+/*
+ * Find socket address.
+ */
+
+struct sockaddr *sm_get_address(int s)
 {
-        struct sock_list *ptr;
+       struct sock_list *ptr;
 
-        for (ptr = sock_list_head.next; ptr != &sock_list_head; ptr = ptr->next)
-                if (ptr->s == s) {
-                        if (ptr->bind_state == STATE_BOUND)
-                                return 1;
-                        break;
-                }
+       ptr = list_get_link(s);
+       if (ptr == NULL)
+               return NULL;
 
-        return 0;
+       return &ptr->addr;
 }