svn commit: r218193 - head/sbin/hastd
Pawel Jakub Dawidek
pjd at FreeBSD.org
Wed Feb 2 15:46:29 UTC 2011
Author: pjd
Date: Wed Feb 2 15:46:28 2011
New Revision: 218193
URL: http://svn.freebsd.org/changeset/base/218193
Log:
Add proto_connect_wait() to wait for connection to finish.
If timeout argument to proto_connect() is -1, then the caller needs to use
this new function to wait for connection.
This change is in preparation for capsicum, where sandboxed worker wants
to ask main process to connect in worker's behalf and pass descriptor
to the worker. Because we don't want the main process to wait for the
connection, it will start async connection and pass descriptor to the
worker who will be responsible for waiting for the connection to finish.
MFC after: 1 week
Modified:
head/sbin/hastd/proto.c
head/sbin/hastd/proto.h
head/sbin/hastd/proto_impl.h
head/sbin/hastd/proto_tcp4.c
head/sbin/hastd/proto_uds.c
Modified: head/sbin/hastd/proto.c
==============================================================================
--- head/sbin/hastd/proto.c Wed Feb 2 15:42:00 2011 (r218192)
+++ head/sbin/hastd/proto.c Wed Feb 2 15:46:28 2011 (r218193)
@@ -174,7 +174,7 @@ proto_connect(struct proto_conn *conn, i
PJDLOG_ASSERT(conn->pc_side == PROTO_SIDE_CLIENT);
PJDLOG_ASSERT(conn->pc_proto != NULL);
PJDLOG_ASSERT(conn->pc_proto->hp_connect != NULL);
- PJDLOG_ASSERT(timeout >= 0);
+ PJDLOG_ASSERT(timeout >= -1);
ret = conn->pc_proto->hp_connect(conn->pc_ctx, timeout);
if (ret != 0) {
@@ -186,6 +186,27 @@ proto_connect(struct proto_conn *conn, i
}
int
+proto_connect_wait(struct proto_conn *conn, int timeout)
+{
+ int ret;
+
+ PJDLOG_ASSERT(conn != NULL);
+ PJDLOG_ASSERT(conn->pc_magic == PROTO_CONN_MAGIC);
+ PJDLOG_ASSERT(conn->pc_side == PROTO_SIDE_CLIENT);
+ PJDLOG_ASSERT(conn->pc_proto != NULL);
+ PJDLOG_ASSERT(conn->pc_proto->hp_connect_wait != NULL);
+ PJDLOG_ASSERT(timeout >= 0);
+
+ ret = conn->pc_proto->hp_connect_wait(conn->pc_ctx, timeout);
+ if (ret != 0) {
+ errno = ret;
+ return (-1);
+ }
+
+ return (0);
+}
+
+int
proto_server(const char *addr, struct proto_conn **connp)
{
Modified: head/sbin/hastd/proto.h
==============================================================================
--- head/sbin/hastd/proto.h Wed Feb 2 15:42:00 2011 (r218192)
+++ head/sbin/hastd/proto.h Wed Feb 2 15:46:28 2011 (r218193)
@@ -39,6 +39,7 @@ struct proto_conn;
int proto_client(const char *addr, struct proto_conn **connp);
int proto_connect(struct proto_conn *conn, int timeout);
+int proto_connect_wait(struct proto_conn *conn, int timeout);
int proto_server(const char *addr, struct proto_conn **connp);
int proto_accept(struct proto_conn *conn, struct proto_conn **newconnp);
int proto_send(const struct proto_conn *conn, const void *data, size_t size);
Modified: head/sbin/hastd/proto_impl.h
==============================================================================
--- head/sbin/hastd/proto_impl.h Wed Feb 2 15:42:00 2011 (r218192)
+++ head/sbin/hastd/proto_impl.h Wed Feb 2 15:46:28 2011 (r218193)
@@ -41,6 +41,7 @@
typedef int hp_client_t(const char *, void **);
typedef int hp_connect_t(void *, int);
+typedef int hp_connect_wait_t(void *, int);
typedef int hp_server_t(const char *, void **);
typedef int hp_accept_t(void *, void **);
typedef int hp_send_t(void *, const unsigned char *, size_t);
@@ -57,6 +58,7 @@ struct hast_proto {
const char *hp_name;
hp_client_t *hp_client;
hp_connect_t *hp_connect;
+ hp_connect_wait_t *hp_connect_wait;
hp_server_t *hp_server;
hp_accept_t *hp_accept;
hp_send_t *hp_send;
Modified: head/sbin/hastd/proto_tcp4.c
==============================================================================
--- head/sbin/hastd/proto_tcp4.c Wed Feb 2 15:42:00 2011 (r218192)
+++ head/sbin/hastd/proto_tcp4.c Wed Feb 2 15:46:28 2011 (r218193)
@@ -60,6 +60,7 @@ struct tcp4_ctx {
#define TCP4_SIDE_SERVER_WORK 2
};
+static int tcp4_connect_wait(void *ctx, int timeout);
static void tcp4_close(void *ctx);
static in_addr_t
@@ -214,16 +215,14 @@ static int
tcp4_connect(void *ctx, int timeout)
{
struct tcp4_ctx *tctx = ctx;
- struct timeval tv;
- fd_set fdset;
- socklen_t esize;
- int error, flags, ret;
+ int error, flags;
PJDLOG_ASSERT(tctx != NULL);
PJDLOG_ASSERT(tctx->tc_magic == TCP4_CTX_MAGIC);
PJDLOG_ASSERT(tctx->tc_side == TCP4_SIDE_CLIENT);
PJDLOG_ASSERT(tctx->tc_fd >= 0);
- PJDLOG_ASSERT(timeout >= 0);
+ PJDLOG_ASSERT(tctx->tc_sin.sin_family != AF_UNSPEC);
+ PJDLOG_ASSERT(timeout >= -1);
flags = fcntl(tctx->tc_fd, F_GETFL);
if (flags == -1) {
@@ -244,6 +243,8 @@ tcp4_connect(void *ctx, int timeout)
if (connect(tctx->tc_fd, (struct sockaddr *)&tctx->tc_sin,
sizeof(tctx->tc_sin)) == 0) {
+ if (timeout == -1)
+ return (0);
error = 0;
goto done;
}
@@ -252,10 +253,35 @@ tcp4_connect(void *ctx, int timeout)
pjdlog_common(LOG_DEBUG, 1, errno, "connect() failed");
goto done;
}
- /*
- * Connection can't be established immediately, let's wait
- * for HAST_TIMEOUT seconds.
- */
+ if (timeout == -1)
+ return (0);
+ return (tcp4_connect_wait(ctx, timeout));
+done:
+ flags &= ~O_NONBLOCK;
+ if (fcntl(tctx->tc_fd, F_SETFL, flags) == -1) {
+ if (error == 0)
+ error = errno;
+ pjdlog_common(LOG_DEBUG, 1, errno,
+ "fcntl(F_SETFL, ~O_NONBLOCK) failed");
+ }
+ return (error);
+}
+
+static int
+tcp4_connect_wait(void *ctx, int timeout)
+{
+ struct tcp4_ctx *tctx = ctx;
+ struct timeval tv;
+ fd_set fdset;
+ socklen_t esize;
+ int error, flags, ret;
+
+ PJDLOG_ASSERT(tctx != NULL);
+ PJDLOG_ASSERT(tctx->tc_magic == TCP4_CTX_MAGIC);
+ PJDLOG_ASSERT(tctx->tc_side == TCP4_SIDE_CLIENT);
+ PJDLOG_ASSERT(tctx->tc_fd >= 0);
+ PJDLOG_ASSERT(timeout >= 0);
+
tv.tv_sec = timeout;
tv.tv_usec = 0;
again:
@@ -289,6 +315,13 @@ again:
}
error = 0;
done:
+ flags = fcntl(tctx->tc_fd, F_GETFL);
+ if (flags == -1) {
+ if (error == 0)
+ error = errno;
+ pjdlog_common(LOG_DEBUG, 1, errno, "fcntl(F_GETFL) failed");
+ return (error);
+ }
flags &= ~O_NONBLOCK;
if (fcntl(tctx->tc_fd, F_SETFL, flags) == -1) {
if (error == 0)
@@ -492,6 +525,7 @@ static struct hast_proto tcp4_proto = {
.hp_name = "tcp4",
.hp_client = tcp4_client,
.hp_connect = tcp4_connect,
+ .hp_connect_wait = tcp4_connect_wait,
.hp_server = tcp4_server,
.hp_accept = tcp4_accept,
.hp_send = tcp4_send,
Modified: head/sbin/hastd/proto_uds.c
==============================================================================
--- head/sbin/hastd/proto_uds.c Wed Feb 2 15:42:00 2011 (r218192)
+++ head/sbin/hastd/proto_uds.c Wed Feb 2 15:46:28 2011 (r218193)
@@ -131,7 +131,7 @@ uds_connect(void *ctx, int timeout)
PJDLOG_ASSERT(uctx->uc_magic == UDS_CTX_MAGIC);
PJDLOG_ASSERT(uctx->uc_side == UDS_SIDE_CLIENT);
PJDLOG_ASSERT(uctx->uc_fd >= 0);
- PJDLOG_ASSERT(timeout >= 0);
+ PJDLOG_ASSERT(timeout >= -1);
if (connect(uctx->uc_fd, (struct sockaddr *)&uctx->uc_sun,
sizeof(uctx->uc_sun)) < 0) {
@@ -142,6 +142,20 @@ uds_connect(void *ctx, int timeout)
}
static int
+uds_connect_wait(void *ctx, int timeout)
+{
+ struct uds_ctx *uctx = ctx;
+
+ PJDLOG_ASSERT(uctx != NULL);
+ PJDLOG_ASSERT(uctx->uc_magic == UDS_CTX_MAGIC);
+ PJDLOG_ASSERT(uctx->uc_side == UDS_SIDE_CLIENT);
+ PJDLOG_ASSERT(uctx->uc_fd >= 0);
+ PJDLOG_ASSERT(timeout >= 0);
+
+ return (0);
+}
+
+static int
uds_server(const char *addr, void **ctxp)
{
struct uds_ctx *uctx;
@@ -330,6 +344,7 @@ static struct hast_proto uds_proto = {
.hp_name = "uds",
.hp_client = uds_client,
.hp_connect = uds_connect,
+ .hp_connect_wait = uds_connect_wait,
.hp_server = uds_server,
.hp_accept = uds_accept,
.hp_send = uds_send,
More information about the svn-src-head
mailing list