[CRIU] [crtools-bot] sockets: Helper for binding unix socket

Cyrill Gorcunov gorcunov at openvz.org
Tue Feb 7 08:27:51 EST 2012


The commit is pushed to "master" and will appear on git://github.com/cyrillos/crtools.git
--------------->
commit 57eea3d3fd3d19987d62ac47b8e938d9559b112d
Author: Pavel Emelyanov <xemul at parallels.com>
Date:   Tue Feb 7 15:58:32 2012 +0400

    sockets: Helper for binding unix socket
    
    This one reads a name from image, does preparations for bind and calls bind.
    
    Signed-off-by: Pavel Emelyanov <xemul at parallels.com>
    Signed-off-by: Cyrill Gorcunov <gorcunov at openvz.org>
---
 sockets.c |  163 +++++++++++++++++++++++--------------------------------------
 1 files changed, 61 insertions(+), 102 deletions(-)

diff --git a/sockets.c b/sockets.c
index 7b283a4..b270862 100644
--- a/sockets.c
+++ b/sockets.c
@@ -811,118 +811,79 @@ static int run_accept_jobs(void)
 	return 0;
 }
 
-static int open_unix_sk_dgram(int sk, struct unix_sk_entry *ue, int img_fd)
+static int bind_unix_sk(int sk, struct unix_sk_entry *ue, int img_fd)
 {
-	if (ue->namelen) {
-
-		/*
-		 * This is trivial socket bind() case,
-		 * we don't have to wait for connect().
-		 */
-
-		struct unix_sk_listen *d;
-		struct sockaddr_un addr;
-		int ret;
-
-		memset(&addr, 0, sizeof(addr));
-		addr.sun_family = AF_UNIX;
-
-		ret = read(img_fd, &addr.sun_path, ue->namelen);
-		if (ret != ue->namelen) {
-			pr_err("Error reading socket name from image (%d)", ret);
-			goto err;
-		}
+	struct sockaddr_un addr;
+	struct unix_sk_listen *e;
+	int ret;
 
-		if (addr.sun_path[0] != '\0')
-			unlink(addr.sun_path);
-		if (bind(sk, (struct sockaddr *)&addr,
-			 sizeof(addr.sun_family) + ue->namelen) < 0) {
-			pr_perror("Can't bind socket");
-			goto err;
-		}
+	if (!ue->namelen || ue->namelen >= UNIX_PATH_MAX) {
+		pr_err("Bad unix name len %d\n", ue->namelen);
+		goto err;
+	}
 
-		/*
-		 * Just remember it and connect() if needed.
-		 */
-		d = xmalloc(sizeof(*d));
-		if (!d)
-			goto err;
+	memset(&addr, 0, sizeof(addr));
+	addr.sun_family = AF_UNIX;
 
-		memcpy(&d->addr, &addr, sizeof(d->addr));
-		d->addrlen = sizeof(addr.sun_family) + ue->namelen;
-		d->ino	= ue->id;
-		d->type = SOCK_DGRAM;
+	ret = read(img_fd, &addr.sun_path, ue->namelen);
+	if (ret != ue->namelen) {
+		pr_err("Error reading socket name from image (%d)", ret);
+		goto err;
+	}
 
-		SK_HASH_LINK(unix_listen, d->ino, d);
+	if (addr.sun_path[0] != '\0')
+		unlink(addr.sun_path);
+	if (bind(sk, (struct sockaddr *)&addr,
+				sizeof(addr.sun_family) + ue->namelen) < 0) {
+		pr_perror("Can't bind socket");
+		goto err;
 	}
 
-	if (ue->peer)
-		if (schedule_conn_job(CJ_DGRAM, ue))
-			goto err;
+	/*
+	 * Just remember it and connect() if needed.
+	 */
+	e = xmalloc(sizeof(*e));
+	if (!e)
+		goto err;
 
+	memcpy(&e->addr, &addr, sizeof(e->addr));
+	e->addrlen = sizeof(addr.sun_family) + ue->namelen;
+	e->ino	= ue->id;
+	e->type = ue->type;
+
+	SK_HASH_LINK(unix_listen, e->ino, e);
 	return 0;
 err:
 	return -1;
 }
 
-static int open_unix_sk_stream(int sk, struct unix_sk_entry *ue, int img_fd)
+static int open_unix_sk_dgram(int sk, struct unix_sk_entry *ue, int img_fd)
 {
-	int ret = -1;
+	int ret = 0;
 
-	if (ue->state == TCP_LISTEN) {
-		struct sockaddr_un addr;
-		struct unix_sk_listen *e;
-		int ret;
+	if (ue->namelen)
+		ret = bind_unix_sk(sk, ue, img_fd);
 
-		/*
-		 * Listen sockets are easiest ones -- simply
-		 * bind() and listen(), and that's all.
-		 */
-		if (!ue->namelen || ue->namelen >= UNIX_PATH_MAX) {
-			pr_err("Bad unix name len %d\n", ue->namelen);
-			goto err;
-		}
+	if (!ret && ue->peer)
+		ret = schedule_conn_job(CJ_DGRAM, ue);
 
-		memset(&addr, 0, sizeof(addr));
-		addr.sun_family = AF_UNIX;
+	return ret;
+}
 
-		ret = read(img_fd, &addr.sun_path, ue->namelen);
-		if (ret != ue->namelen) {
-			pr_err("Error reading socket name from image (%d)", ret);
-			goto err;
-		}
+static int open_unix_sk_stream(int sk, struct unix_sk_entry *ue, int img_fd)
+{
+	int ret;
 
-		if (addr.sun_path[0] != '\0')
-			unlink(addr.sun_path);
-		if (bind(sk, (struct sockaddr *)&addr,
-			 sizeof(addr.sun_family) + ue->namelen) < 0) {
-			pr_perror("Can't bind socket");
-			goto err;
-		}
+	if (ue->state == TCP_LISTEN) {
+		ret = bind_unix_sk(sk, ue, img_fd);
+		if (ret < 0)
+			goto out;
 
-		if (listen(sk, ue->backlog) < 0) {
+		ret = listen(sk, ue->backlog);
+		if (ret < 0) {
 			pr_perror("Can't listen socket");
-			goto err;
+			goto out;
 		}
-
-		/*
-		 * Collect listening sockets here, we will
-		 * need them to resolve in-flight connections
-		 * names.
-		 */
-		e = xzalloc(sizeof(*e));
-		if (!e)
-			goto err;
-
-		SK_HASH_LINK(unix_listen, ue->id, e);
-
-		memcpy(&e->addr, &addr, sizeof(e->addr));
-		e->addrlen	= sizeof(e->addr.sun_family) + ue->namelen;
-		e->ino		= ue->id;
-		e->type		= SOCK_STREAM;
-
-		dprintk("\tCollected listening socket %d\n", ue->id);
-
 	} else if (ue->state == TCP_ESTABLISHED) {
 
 		/*
@@ -943,37 +904,35 @@ static int open_unix_sk_stream(int sk, struct unix_sk_entry *ue, int img_fd)
 			 * Will become a server
 			 */
 
+			ret = -1;
 			prep_conn_addr(ue->id, &addr, &len);
 			if (bind(sk, (struct sockaddr *)&addr, len) < 0) {
 				pr_perror("Can't bind socket");
-				goto err;
+				goto out;
 			}
 
 			if (listen(sk, 1) < 0) {
 				pr_perror("Can't listen socket");
-				goto err;
+				goto out;
 			}
 
 			aj = xmalloc(sizeof(*aj));
 			if (aj == NULL)
-				goto err;
+				goto out;
 
 			aj->fd = ue->fd;
 			aj->next = accept_jobs;
 			accept_jobs = aj;
 			unix_show_job("Sched acc", ue->fd, ue->id);
-		} else {
-			if (schedule_conn_job((ue->flags & USK_INFLIGHT) ?
-						CJ_STREAM_INFLIGHT : CJ_STREAM, ue))
-				goto err;
-		}
+			ret = 0;
+		} else
+			ret = schedule_conn_job((ue->flags & USK_INFLIGHT) ?
+					CJ_STREAM_INFLIGHT : CJ_STREAM, ue);
 	} else {
 		pr_err("Unknown state %d\n", ue->state);
-		goto err;
+		ret = -1;
 	}
-
-	ret = 0;
-err:
+out:
 	return ret;
 }
 


More information about the CRIU mailing list