From 00d38c4e8026425cb8f755b0ba70c65bcbbc0c6f Mon Sep 17 00:00:00 2001
From: Razvan Deaconescu <razvan.deaconescu@cs.pub.ro>
Date: Sat, 4 Jun 2011 20:46:30 +0300
Subject: [PATCH] raw: Add array-based socket management source code.

---
 src/raw/socket_manager_array.c | 207 +++++++++++++++++++++++++++++++++
 1 file changed, 207 insertions(+)
 create mode 100644 src/raw/socket_manager_array.c

diff --git a/src/raw/socket_manager_array.c b/src/raw/socket_manager_array.c
new file mode 100644
index 0000000..d86bc2d
--- /dev/null
+++ b/src/raw/socket_manager_array.c
@@ -0,0 +1,207 @@
+/*
+ * Array management for sockets
+ *
+ * Use an array similar to a descriptor table to store information about
+ * sockets. Use functions to add, remove, find and update information.
+ *
+ * 2011, Razvan Deaconescu, razvan.deaconescu@cs.pub.ro
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <unistd.h>
+#include <errno.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 {
+	int s;
+	struct sockaddr addr;
+	enum sock_bind_state bind_state;
+};
+
+/*
+ * Socket management array
+ * Size is assumed size of descriptor table. Socket descriptor is an array
+ * index.
+ */
+#define ARRAY_SIZE	1024
+static struct sock *sock_array[ARRAY_SIZE];
+
+/*
+ * Find socket in socket management array.
+ */
+
+static struct sock *array_get_sock(int s)
+{
+	if (s < 0 || s >= ARRAY_SIZE)
+		return NULL;
+
+	return sock_array[s];
+}
+
+/*
+ * Find socket in socket management array by address.
+ */
+
+static struct sock *array_get_sock_by_address(const struct sockaddr *addr)
+{
+	struct sock *ptr;
+	int i;
+
+	for (i = 0; i < ARRAY_SIZE; i++) {
+		ptr = sock_array[i];
+		if (ptr == NULL)
+			continue;
+		if (memcmp(addr, &ptr->addr, sizeof(*addr)) == 0)
+			return ptr;
+	}
+
+	return NULL;
+}
+
+/*
+ * Add new socket to list. Called by sw_socket "syscall".
+ */
+
+int sm_add(int s)
+{
+	struct sock *ptr = malloc(sizeof(*ptr));
+	if (ptr == NULL)
+		return -1;
+
+	ptr->s = s;
+	sock_array[s] = ptr;
+
+	return 0;
+}
+
+/*
+ * Bind socket to given address. Called by sw_bind "syscall".
+ */
+
+int sm_update_address(int s, const struct sockaddr *addr)
+{
+	struct sock *ptr;
+
+	ptr = array_get_sock(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 sm_del(int s)
+{
+	struct sock *ptr;
+
+	ptr = array_get_sock(s);
+	if (ptr == NULL)
+		return -1;
+
+	sock_array[s] = NULL;
+	free(ptr);
+
+	return 0;
+}
+
+/*
+ * Check if a socket is bound.
+ */
+
+int sm_is_bound(int s)
+{
+	struct sock *ptr;
+
+	ptr = array_get_sock(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 *ptr;
+
+	ptr = array_get_sock(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 *ptr;
+
+	ptr = array_get_sock(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 *ptr;
+
+	ptr = array_get_sock_by_address(addr);
+	if (ptr == NULL)
+		return 0;
+
+	return 1;
+}
+
+/*
+ * Find socket address.
+ */
+
+struct sockaddr *sm_get_address(int s)
+{
+	struct sock *ptr;
+
+	ptr = array_get_sock(s);
+	if (ptr == NULL)
+		return NULL;
+
+	return &ptr->addr;
+}
-- 
2.20.1