Astral/kernel-src/sys/syscalls/socket.c

#include <kernel/syscalls.h>
#include <kernel/abi.h>
#include <kernel/auth.h>
#include <kernel/sock.h>
#include <errno.h>

#define SOCK_CLOEXEC   0x80000
#define SOCK_NONBLOCK  0x800

syscallret_t syscall_socket(context_t *, int domain, int type, int protocol) {
	syscallret_t ret = {
		.ret = -1
	};

	int socktype = -1;
	bool non_blocking = false, close_exec = false;

	if (type & SOCK_NONBLOCK) {
		non_blocking = true;
		type &= ~SOCK_NONBLOCK;
	}

	if (type & SOCK_CLOEXEC) {
		close_exec = true;
		type &= ~SOCK_CLOEXEC;
	}

	switch (domain) {
		case AF_INET:
			if (type == SOCK_STREAM)
				socktype = SOCKET_TYPE_TCP;
			if (type == SOCK_DGRAM)
				socktype = SOCKET_TYPE_UDP;
			break;
		case AF_LOCAL:
			if (type == SOCK_STREAM)
				socktype = SOCKET_TYPE_LOCAL;
			if (type == SOCK_SEQPACKET)
				socktype = SOCKET_TYPE_LOCAL_SEQPACKET;
			break;
		case AF_PACKET:
			if (type == SOCK_DGRAM)
				socktype = SOCKET_TYPE_RAW_STRIPPED;
			break;
	}

	if (socktype == -1) {
		ret.errno = EAFNOSUPPORT;
		return ret;
	}

	if (socktype == SOCKET_TYPE_RAW_STRIPPED) {
		ret.errno = auth_network_check(&current_thread()->proc->cred, AUTH_ACTIONS_NETWORK_CREATERAW, NULL, NULL);
		if (ret.errno)
			return ret;
	}

	socket_t *socket = socket_create(socktype, protocol);
	if (socket == NULL) {
		ret.errno = ENOMEM;
		return ret;
	}

	vnode_t *vnode;
	ret.errno = sockfs_newsocket(&vnode, socket);
	if (ret.errno) {
		socket->ops->destroy(socket);
		return ret;
	}

	file_t *file;
	int fd;
	
	ret.errno = fd_new(close_exec ? O_CLOEXEC : 0, &file, &fd);
	if (ret.errno) {
		// socket gets deleted by node cleanup
		VOP_RELEASE(vnode);
		return ret;
	}

	file->vnode = vnode;
	file->flags = FILE_READ | FILE_WRITE | (non_blocking ? O_NONBLOCK : 0);
	file->offset = 0;
	file->mode = 0777;

	ret.ret = fd;

	return ret;
}