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

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

syscallret_t syscall_accept(context_t *, int oldfd, abisockaddr_t *abisockaddr, int *uaddrlen, int acceptflags) {
	syscallret_t ret = {
		.ret = -1
	};

	int addrlen = 0;
	if (abisockaddr) {
		ret.errno = usercopy_fromuser(&addrlen, uaddrlen, sizeof(int));
		if (ret.errno)
			return ret;
	}

	vnode_t *servernode;
	int fileflags;
	ret.errno = sockfd_get(oldfd, &servernode, &fileflags);
	if (ret.errno)
		return ret;

	socket_t *server = SOCKFS_SOCKET_FROM_NODE(servernode);

	if (server->ops->accept == NULL) {
		ret.errno = ENOTSUP;
		goto cleanup;
	}

	socket_t *client = socket_create(server->type, server->protocol);
	if (client == NULL) {
		ret.errno = ENOMEM;
		goto cleanup;
	}

	vnode_t *clientnode;
	ret.errno = sockfs_newsocket(&clientnode, client);
	if (ret.errno) {
		client->ops->destroy(client);
		goto cleanup;
	}

	sockaddr_t addr = {0};
	ret.errno = server->ops->accept(server, client, &addr, fileflagstovnodeflags(fileflags));
	if (ret.errno) {
		// socket gets deleted by node cleanup
		VOP_RELEASE(clientnode);
		goto cleanup;
	}

	file_t *newfile;
	int newfd;
	
	ret.errno = fd_new(acceptflags & O_CLOEXEC, &newfile, &newfd);
	if (ret.errno) {
		// socket gets deleted by node cleanup
		VOP_RELEASE(clientnode);
		goto cleanup;
	}

	newfile->vnode = clientnode;
	newfile->flags = FILE_READ | FILE_WRITE | (acceptflags & (O_NONBLOCK));
	newfile->offset = 0;
	newfile->mode = 0777;

	if (abisockaddr != NULL) {
		socklen_t actual_len;
		ret.errno = sock_copy_addr_to_user(client->type, &addr, abisockaddr, addrlen, &actual_len);
		if (ret.errno) {
			fd_close(newfd);
			goto cleanup;
		}

		addrlen = actual_len;
		ret.errno = usercopy_touser(uaddrlen, &addrlen, sizeof(addrlen));
		if (ret.errno) {
			fd_close(newfd);
			goto cleanup;
		}
	}

	ret.ret = newfd;

	cleanup:
	VOP_RELEASE(servernode);

	return ret;
}