提交 b3a6dfe8 编写于 作者: A Asias He 提交者: David S. Miller

VSOCK: Introduce vsock_auto_bind helper

This peace of code is called three times, let's have a helper for it.
Signed-off-by: NAsias He <asias@redhat.com>
Acked-by: NAndy King <acking@vmware.com>
Signed-off-by: NDavid S. Miller <davem@davemloft.net>
上级 b33698e2
...@@ -165,6 +165,18 @@ static struct list_head vsock_bind_table[VSOCK_HASH_SIZE + 1]; ...@@ -165,6 +165,18 @@ static struct list_head vsock_bind_table[VSOCK_HASH_SIZE + 1];
static struct list_head vsock_connected_table[VSOCK_HASH_SIZE]; static struct list_head vsock_connected_table[VSOCK_HASH_SIZE];
static DEFINE_SPINLOCK(vsock_table_lock); static DEFINE_SPINLOCK(vsock_table_lock);
/* Autobind this socket to the local address if necessary. */
static int vsock_auto_bind(struct vsock_sock *vsk)
{
struct sock *sk = sk_vsock(vsk);
struct sockaddr_vm local_addr;
if (vsock_addr_bound(&vsk->local_addr))
return 0;
vsock_addr_init(&local_addr, VMADDR_CID_ANY, VMADDR_PORT_ANY);
return __vsock_bind(sk, &local_addr);
}
static void vsock_init_tables(void) static void vsock_init_tables(void)
{ {
int i; int i;
...@@ -956,15 +968,10 @@ static int vsock_dgram_sendmsg(struct kiocb *kiocb, struct socket *sock, ...@@ -956,15 +968,10 @@ static int vsock_dgram_sendmsg(struct kiocb *kiocb, struct socket *sock,
lock_sock(sk); lock_sock(sk);
if (!vsock_addr_bound(&vsk->local_addr)) { err = vsock_auto_bind(vsk);
struct sockaddr_vm local_addr; if (err)
goto out;
vsock_addr_init(&local_addr, VMADDR_CID_ANY, VMADDR_PORT_ANY);
err = __vsock_bind(sk, &local_addr);
if (err != 0)
goto out;
}
/* If the provided message contains an address, use that. Otherwise /* If the provided message contains an address, use that. Otherwise
* fall back on the socket's remote handle (if it has been connected). * fall back on the socket's remote handle (if it has been connected).
...@@ -1038,15 +1045,9 @@ static int vsock_dgram_connect(struct socket *sock, ...@@ -1038,15 +1045,9 @@ static int vsock_dgram_connect(struct socket *sock,
lock_sock(sk); lock_sock(sk);
if (!vsock_addr_bound(&vsk->local_addr)) { err = vsock_auto_bind(vsk);
struct sockaddr_vm local_addr; if (err)
goto out;
vsock_addr_init(&local_addr, VMADDR_CID_ANY, VMADDR_PORT_ANY);
err = __vsock_bind(sk, &local_addr);
if (err != 0)
goto out;
}
if (!transport->dgram_allow(remote_addr->svm_cid, if (!transport->dgram_allow(remote_addr->svm_cid,
remote_addr->svm_port)) { remote_addr->svm_port)) {
...@@ -1163,17 +1164,9 @@ static int vsock_stream_connect(struct socket *sock, struct sockaddr *addr, ...@@ -1163,17 +1164,9 @@ static int vsock_stream_connect(struct socket *sock, struct sockaddr *addr,
memcpy(&vsk->remote_addr, remote_addr, memcpy(&vsk->remote_addr, remote_addr,
sizeof(vsk->remote_addr)); sizeof(vsk->remote_addr));
/* Autobind this socket to the local address if necessary. */ err = vsock_auto_bind(vsk);
if (!vsock_addr_bound(&vsk->local_addr)) { if (err)
struct sockaddr_vm local_addr; goto out;
vsock_addr_init(&local_addr, VMADDR_CID_ANY,
VMADDR_PORT_ANY);
err = __vsock_bind(sk, &local_addr);
if (err != 0)
goto out;
}
sk->sk_state = SS_CONNECTING; sk->sk_state = SS_CONNECTING;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册