else/uvms
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ch-proxy: support normal vsock

Browse source
This commit is contained in:
Else, Someone 2025-11-07 19:04:58 +02:00
parent 927db90791
commit 20451db88c
1 changed files with 81 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

89
pkgs/ch-proxy/proxy.c
View file

@ -7,6 +7,8 @@
#include <sys/socket.h>
#include <sys/un.h>
#include <linux/vm_sockets.h>
struct msghdr mk_msghdr();
int ch_connect(const char*, const char*);
ssize_t send_fd(int, int);
@ -18,7 +20,8 @@ void print_usage() {
"Usage:\n"
"\tch-proxy uvm/$USER_VM_NAME [PORT]\n"
"\tch-proxy uuvm/$VM_NAME [PORT]\n"
"\tch-proxy vsock-mux%$PATH [PORT]\n");
"\tch-proxy vsock-mux%$PATH [PORT]\n"
"\tch-proxy vsock/cid[:port]\n");
}
char *extract_vsock_mux(const char *host_string) {
@ -67,6 +70,50 @@ char *extract_muvm(const char *host_string) {
return result;
}
int extract_cid(const char *host, int *cid, int *port) {
if (errno != 0) {
perror("extract_cid(...): errno unclean");
exit(EXIT_FAILURE);
}
const char PREFIX[] = "vsock/";
const ssize_t PREFIX_LEN = sizeof(PREFIX) - 1;
if (strncmp(host, PREFIX, PREFIX_LEN) != 0) {
return 1;
}
char *in = NULL;
const char *end = host + strlen(host);
const char *sCid = host + PREFIX_LEN;
long x;
x = strtol(sCid, &in, 10);
if (errno != 0) {
perror("strtol(cid, ...)");
exit(EXIT_FAILURE);
}
*cid = (int) x;
if (in == end) {
*port = 22;
return 0;
} else if (in[0] != ':') {
perror("extract_cid(...): expected a string of the form cid[:port]");
perror(host);
exit(EXIT_FAILURE);
}
const char *sPort = in + 1;
x = strtol(sPort, &in, 10);
if (errno != 0) {
perror("strtol(port, ...)");
exit(EXIT_FAILURE);
}
*port = (int) x;
return 0;
}
int main(int argc, char** argv) {
if (!(2 <= argc && argc <= 3)) {
fprintf(stderr, "%s: Wrong # of arguments: %d\n", argv[0], argc);
@ -79,21 +126,47 @@ int main(int argc, char** argv) {
const char *ssh_host = argv[1];
const char *port_string = argc == 3 ? argv[2] : PORT_DEFAULT;
char *path_un;
int cid = -1, port = -1;
char *path_un = NULL;
if ((path_un = extract_uvm(ssh_host)) != NULL) {
} else if ((path_un = extract_muvm(ssh_host)) != NULL) {
} else if ((path_un = extract_vsock_mux(ssh_host)) != NULL) {
} else if (extract_cid(ssh_host, &cid, &port) == 0) {
} else {
fprintf(stderr, "ch-proxy/main: unexpected host stirng format: %s\n", ssh_host);
fprintf(stderr, "ch-proxy/main: unexpected host string format: %s\n", ssh_host);
print_usage();
return EXIT_FAILURE;
}
const int s = ch_connect(path_un, port_string);
if (s == -1) {
perror("ssh-vsock-proxy/main/ch_connect");
return EXIT_FAILURE;
};
int s = -1;
if (path_un != NULL) {
s = ch_connect(path_un, port_string);
if (s == -1) {
perror("ssh-vsock-proxy/main/ch_connect");
return EXIT_FAILURE;
};
} else if (cid != -1 && port != -1) {
struct sockaddr_vm sa = {
.svm_family = AF_VSOCK,
.svm_reserved1 = 0,
.svm_port = port,
.svm_cid = cid,
};
memset(sa.svm_zero, 0, sizeof(sa.svm_zero));
s = socket(AF_VSOCK, SOCK_STREAM, 0);
if (s == -1) {
perror("socket(AF_VSOCK, ...)");
exit(EXIT_FAILURE);
}
if (connect(s, (struct sockaddr*)&sa, sizeof(sa)) != 0) {
perror("connect(socket(AF_VSOCK, ...), ...)");
exit(EXIT_FAILURE);
}
} else {
perror("Couldn't parse neither uuvm/ strings nor vsock/cid[:port]");
exit(EXIT_FAILURE);
}
if (send_fd(1, s) == -1) {
perror("ssh-vsock-proxy/main/send_fd");