1 files changed, 80 insertions, 0 deletions
diff --git a/libc-bottom-half/sources/shutdown.c b/libc-bottom-half/sources/shutdown.c
new file mode 100644
index 0000000..3ffd479
--- /dev/null
+++ b/libc-bottom-half/sources/shutdown.c
@@ -0,0 +1,80 @@
+#include <sys/socket.h>
+
+#include <errno.h>
+
+#include <wasi/api.h>
+#include <wasi/descriptor_table.h>
+#include <wasi/sockets_utils.h>
+
+int tcp_shutdown(tcp_socket_t *socket, int posix_how)
+{
+	tcp_shutdown_type_t wasi_how;
+	switch (posix_how) {
+	case SHUT_RD:
+		wasi_how = TCP_SHUTDOWN_TYPE_RECEIVE;
+		break;
+	case SHUT_WR:
+		wasi_how = TCP_SHUTDOWN_TYPE_SEND;
+		break;
+	case SHUT_RDWR:
+		wasi_how = TCP_SHUTDOWN_TYPE_BOTH;
+		break;
+	default:
+		errno = EINVAL;
+		return -1;
+	}
+
+	tcp_socket_state_connected_t connection;
+	if (socket->state.tag == TCP_SOCKET_STATE_CONNECTED) {
+		connection = socket->state.connected;
+	} else {
+		errno = ENOTCONN;
+		return -1;
+	}
+
+	network_error_code_t error;
+	tcp_borrow_tcp_socket_t socket_borrow =
+		tcp_borrow_tcp_socket(socket->socket);
+	if (!tcp_method_tcp_socket_shutdown(socket_borrow, wasi_how, &error)) {
+		errno = __wasi_sockets_utils__map_error(error);
+		return -1;
+	}
+
+	if (posix_how == SHUT_RD || posix_how == SHUT_RDWR) {
+		// TODO wasi-sockets: drop input stream (if not already). And
+		// update `recv` to take dropped input streams into account.
+	}
+
+	if (posix_how == SHUT_WR || posix_how == SHUT_RDWR) {
+		// TODO wasi-sockets: drop output stream (if not already). And
+		// update `send` to take dropped output streams into account.
+	}
+
+	return 0;
+}
+
+int udp_shutdown(udp_socket_t *socket, int posix_how)
+{
+	// UDP has nothing to shut down.
+	errno = EOPNOTSUPP;
+	return -1;
+}
+
+int shutdown(int socket, int how)
+{
+	descriptor_table_entry_t *entry;
+	if (!descriptor_table_get_ref(socket, &entry)) {
+		errno = EBADF;
+		return -1;
+	}
+
+	switch (entry->tag) {
+	case DESCRIPTOR_TABLE_ENTRY_TCP_SOCKET:
+		return tcp_shutdown(&entry->tcp_socket, how);
+	case DESCRIPTOR_TABLE_ENTRY_UDP_SOCKET:
+		return udp_shutdown(&entry->udp_socket, how);
+	default:
+		errno = EOPNOTSUPP;
+		return -1;
+	}
+}