Initial commit

1 files changed, 72 insertions, 0 deletions

diff --git a/vsock.c b/vsock.c
new file mode 100644
index 0000000..3bcd8b3
--- /dev/null
+++ b/vsock.c
@@ -0,0 +1,72 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+// SPDX-FileCopyrightText: 2020 Alyssa Ross <hi@alyssa.is>
+
+#define _GNU_SOURCE
+
+#include "vsock.h"
+
+#include <sys/ioctl.h>
+#include <sys/socket.h>
+
+#include <linux/vm_sockets.h>
+
+static void fill_sockaddr(struct sockaddr_vm *addr, uint32_t cid, uint32_t port)
+{
+	addr->svm_family = AF_VSOCK;
+	addr->svm_cid = cid;
+	addr->svm_port = port;
+}
+
+int vsock_bind(int fd, uint32_t cid, uint32_t port)
+{
+	struct sockaddr_vm addr = { 0 };
+	fill_sockaddr(&addr, cid, port);
+
+	if (bind(fd, (struct sockaddr *)&addr, sizeof addr) == -1)
+		return -1;
+
+	return 0;
+}
+
+int vsock_accept(int sockfd, uint32_t *cid, uint32_t *port)
+{
+	struct sockaddr_vm addr = { 0 };
+	socklen_t addr_size = sizeof addr;
+
+	if (accept(sockfd, (struct sockaddr *)&addr, &addr_size) == -1)
+		return -1;
+
+	*cid = addr.svm_cid;
+	*port = addr.svm_port;
+
+	return 0;
+}
+
+int vsock_connect(int fd, uint32_t cid, uint32_t port)
+{
+	struct sockaddr_vm addr = { 0 };
+	fill_sockaddr(&addr, cid, port);
+	return connect(fd, (struct sockaddr *)&addr, sizeof addr);
+}
+
+int vsock_open(uint32_t cid, uint32_t port)
+{
+	int fd = socket(AF_VSOCK, SOCK_STREAM, 0);
+	if (fd == -1)
+		return -1;
+
+	return vsock_connect(fd, cid, port);
+}
+
+int vsock_get_port(int fd, uint32_t *port)
+{
+	struct sockaddr_vm addr;
+	socklen_t addrlen = sizeof addr;
+
+	if (getsockname(fd, (struct sockaddr *)&addr, &addrlen) == -1)
+		return -1;
+
+	*port = addr.svm_port;
+
+	return 0;
+}