1 files changed, 107 insertions, 0 deletions
diff --git a/libc-bottom-half/sources/socket.c b/libc-bottom-half/sources/socket.c
new file mode 100644
index 0000000..3e61638
--- /dev/null
+++ b/libc-bottom-half/sources/socket.c
@@ -0,0 +1,107 @@
+#include <errno.h>
+#include <netinet/in.h>
+
+#include <wasi/descriptor_table.h>
+#include <wasi/sockets_utils.h>
+
+static int tcp_socket(network_ip_address_family_t family, bool blocking)
+{
+	tcp_create_socket_error_code_t error;
+	tcp_own_tcp_socket_t socket;
+	if (!tcp_create_socket_create_tcp_socket(family, &socket, &error)) {
+		errno = __wasi_sockets_utils__map_error(error);
+		return -1;
+	}
+
+	tcp_borrow_tcp_socket_t socket_borrow = tcp_borrow_tcp_socket(socket);
+	poll_own_pollable_t socket_pollable =
+		tcp_method_tcp_socket_subscribe(socket_borrow);
+
+	descriptor_table_entry_t
+		entry = { .tag = DESCRIPTOR_TABLE_ENTRY_TCP_SOCKET,
+			  .tcp_socket = {
+				  .socket = socket,
+				  .socket_pollable = socket_pollable,
+				  .blocking = blocking,
+				  .fake_nodelay = false,
+				  .family = family,
+				  .state = { .tag = TCP_SOCKET_STATE_UNBOUND,
+					     .unbound = {
+						     /* No additional state. */ } },
+			  } };
+
+	int fd;
+	if (!descriptor_table_insert(entry, &fd)) {
+		errno = EMFILE;
+		return -1;
+	}
+	return fd;
+}
+
+static int udp_socket(network_ip_address_family_t family, bool blocking)
+{
+	udp_create_socket_error_code_t error;
+	udp_own_udp_socket_t socket;
+	if (!udp_create_socket_create_udp_socket(family, &socket, &error)) {
+		errno = __wasi_sockets_utils__map_error(error);
+		return -1;
+	}
+
+	udp_borrow_udp_socket_t socket_borrow = udp_borrow_udp_socket(socket);
+	poll_own_pollable_t socket_pollable =
+		udp_method_udp_socket_subscribe(socket_borrow);
+
+	descriptor_table_entry_t
+		entry = { .tag = DESCRIPTOR_TABLE_ENTRY_UDP_SOCKET,
+			  .udp_socket = {
+				  .socket = socket,
+				  .socket_pollable = socket_pollable,
+				  .blocking = blocking,
+				  .family = family,
+				  .state = { .tag = UDP_SOCKET_STATE_UNBOUND,
+					     .unbound = {
+						     /* No additional state. */ } },
+			  } };
+
+	int fd;
+	if (!descriptor_table_insert(entry, &fd)) {
+		errno = EMFILE;
+		return -1;
+	}
+	return fd;
+}
+
+int socket(int domain, int type, int protocol)
+{
+	network_ip_address_family_t family;
+	switch (domain) {
+	case PF_INET:
+		family = NETWORK_IP_ADDRESS_FAMILY_IPV4;
+		break;
+
+	case PF_INET6:
+		family = NETWORK_IP_ADDRESS_FAMILY_IPV6;
+		break;
+
+	default:
+		errno = EAFNOSUPPORT;
+		return -1;
+	}
+
+	int real_type = type & ~(SOCK_NONBLOCK | SOCK_CLOEXEC);
+	bool blocking = (type & SOCK_NONBLOCK) == 0;
+	// Ignore SOCK_CLOEXEC flag. That concept does not exist in WASI.
+
+	if (real_type == SOCK_STREAM &&
+	    (protocol == 0 || protocol == IPPROTO_TCP)) {
+		return tcp_socket(family, blocking);
+
+	} else if (real_type == SOCK_DGRAM &&
+		   (protocol == 0 || protocol == IPPROTO_UDP)) {
+		return udp_socket(family, blocking);
+
+	} else {
+		errno = EPROTONOSUPPORT;
+		return -1;
+	}
+}