mirror of
git://git.yoctoproject.org/linux-yocto.git
synced 2025-10-22 23:13:01 +02:00
vsock/test: Introduce vsock_bind()
Add a helper for socket()+bind(). Adapt callers. Reviewed-by: Stefano Garzarella <sgarzare@redhat.com> Reviewed-by: Luigi Leonardi <leonardi@redhat.com> Signed-off-by: Michal Luczaj <mhal@rbox.co> Link: https://patch.msgid.link/20250128-vsock-transport-vs-autobind-v3-3-1cf57065b770@rbox.co Signed-off-by: Jakub Kicinski <kuba@kernel.org>
This commit is contained in:
parent
aa388c7211
commit
852a00c428
|
@ -96,14 +96,33 @@ void vsock_wait_remote_close(int fd)
|
|||
close(epollfd);
|
||||
}
|
||||
|
||||
/* Create socket <type>, bind to <cid, port> and return the file descriptor. */
|
||||
int vsock_bind(unsigned int cid, unsigned int port, int type)
|
||||
{
|
||||
struct sockaddr_vm sa = {
|
||||
.svm_family = AF_VSOCK,
|
||||
.svm_cid = cid,
|
||||
.svm_port = port,
|
||||
};
|
||||
int fd;
|
||||
|
||||
fd = socket(AF_VSOCK, type, 0);
|
||||
if (fd < 0) {
|
||||
perror("socket");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
if (bind(fd, (struct sockaddr *)&sa, sizeof(sa))) {
|
||||
perror("bind");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
return fd;
|
||||
}
|
||||
|
||||
/* Bind to <bind_port>, connect to <cid, port> and return the file descriptor. */
|
||||
int vsock_bind_connect(unsigned int cid, unsigned int port, unsigned int bind_port, int type)
|
||||
{
|
||||
struct sockaddr_vm sa_client = {
|
||||
.svm_family = AF_VSOCK,
|
||||
.svm_cid = VMADDR_CID_ANY,
|
||||
.svm_port = bind_port,
|
||||
};
|
||||
struct sockaddr_vm sa_server = {
|
||||
.svm_family = AF_VSOCK,
|
||||
.svm_cid = cid,
|
||||
|
@ -112,16 +131,7 @@ int vsock_bind_connect(unsigned int cid, unsigned int port, unsigned int bind_po
|
|||
|
||||
int client_fd, ret;
|
||||
|
||||
client_fd = socket(AF_VSOCK, type, 0);
|
||||
if (client_fd < 0) {
|
||||
perror("socket");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
if (bind(client_fd, (struct sockaddr *)&sa_client, sizeof(sa_client))) {
|
||||
perror("bind");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
client_fd = vsock_bind(VMADDR_CID_ANY, bind_port, type);
|
||||
|
||||
timeout_begin(TIMEOUT);
|
||||
do {
|
||||
|
@ -192,28 +202,9 @@ int vsock_seqpacket_connect(unsigned int cid, unsigned int port)
|
|||
/* Listen on <cid, port> and return the file descriptor. */
|
||||
static int vsock_listen(unsigned int cid, unsigned int port, int type)
|
||||
{
|
||||
union {
|
||||
struct sockaddr sa;
|
||||
struct sockaddr_vm svm;
|
||||
} addr = {
|
||||
.svm = {
|
||||
.svm_family = AF_VSOCK,
|
||||
.svm_port = port,
|
||||
.svm_cid = cid,
|
||||
},
|
||||
};
|
||||
int fd;
|
||||
|
||||
fd = socket(AF_VSOCK, type, 0);
|
||||
if (fd < 0) {
|
||||
perror("socket");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) {
|
||||
perror("bind");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
fd = vsock_bind(cid, port, type);
|
||||
|
||||
if (listen(fd, 1) < 0) {
|
||||
perror("listen");
|
||||
|
|
|
@ -43,6 +43,7 @@ int vsock_connect(unsigned int cid, unsigned int port, int type);
|
|||
int vsock_accept(unsigned int cid, unsigned int port,
|
||||
struct sockaddr_vm *clientaddrp, int type);
|
||||
int vsock_stream_connect(unsigned int cid, unsigned int port);
|
||||
int vsock_bind(unsigned int cid, unsigned int port, int type);
|
||||
int vsock_bind_connect(unsigned int cid, unsigned int port,
|
||||
unsigned int bind_port, int type);
|
||||
int vsock_seqpacket_connect(unsigned int cid, unsigned int port);
|
||||
|
|
|
@ -113,24 +113,9 @@ static void test_stream_bind_only_client(const struct test_opts *opts)
|
|||
|
||||
static void test_stream_bind_only_server(const struct test_opts *opts)
|
||||
{
|
||||
union {
|
||||
struct sockaddr sa;
|
||||
struct sockaddr_vm svm;
|
||||
} addr = {
|
||||
.svm = {
|
||||
.svm_family = AF_VSOCK,
|
||||
.svm_port = opts->peer_port,
|
||||
.svm_cid = VMADDR_CID_ANY,
|
||||
},
|
||||
};
|
||||
int fd;
|
||||
|
||||
fd = socket(AF_VSOCK, SOCK_STREAM, 0);
|
||||
|
||||
if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) {
|
||||
perror("bind");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
fd = vsock_bind(VMADDR_CID_ANY, opts->peer_port, SOCK_STREAM);
|
||||
|
||||
/* Notify the client that the server is ready */
|
||||
control_writeln("BIND");
|
||||
|
|
Loading…
Reference in New Issue
Block a user