diff options
author | 2015-09-22 10:42:19 -0700 | |
---|---|---|
committer | 2015-09-22 10:42:19 -0700 | |
commit | 45724b35e411fef7c5da66a74c78428c11d56843 (patch) | |
tree | 9264034aca675c89444e02f72ef58e67d7043604 /src/core/iomgr/tcp_client_posix.c | |
parent | 298751c1195523ef6228595043b583c3a6270e08 (diff) |
indent pass to get logical source lines on one physical line
Diffstat (limited to 'src/core/iomgr/tcp_client_posix.c')
-rw-r--r-- | src/core/iomgr/tcp_client_posix.c | 363 |
1 files changed, 195 insertions, 168 deletions
diff --git a/src/core/iomgr/tcp_client_posix.c b/src/core/iomgr/tcp_client_posix.c index e23551c4c0..14255bda07 100644 --- a/src/core/iomgr/tcp_client_posix.c +++ b/src/core/iomgr/tcp_client_posix.c @@ -56,7 +56,8 @@ extern int grpc_tcp_trace; -typedef struct { +typedef struct +{ gpr_mu mu; grpc_fd *fd; gpr_timespec deadline; @@ -69,51 +70,57 @@ typedef struct { grpc_closure *closure; } async_connect; -static int prepare_socket(const struct sockaddr *addr, int fd) { - if (fd < 0) { - goto error; - } +static int +prepare_socket (const struct sockaddr *addr, int fd) +{ + if (fd < 0) + { + goto error; + } - if (!grpc_set_socket_nonblocking(fd, 1) || !grpc_set_socket_cloexec(fd, 1) || - (addr->sa_family != AF_UNIX && !grpc_set_socket_low_latency(fd, 1)) || - !grpc_set_socket_no_sigpipe_if_possible(fd)) { - gpr_log(GPR_ERROR, "Unable to configure socket %d: %s", fd, - strerror(errno)); - goto error; - } + if (!grpc_set_socket_nonblocking (fd, 1) || !grpc_set_socket_cloexec (fd, 1) || (addr->sa_family != AF_UNIX && !grpc_set_socket_low_latency (fd, 1)) || !grpc_set_socket_no_sigpipe_if_possible (fd)) + { + gpr_log (GPR_ERROR, "Unable to configure socket %d: %s", fd, strerror (errno)); + goto error; + } return 1; error: - if (fd >= 0) { - close(fd); - } + if (fd >= 0) + { + close (fd); + } return 0; } -static void tc_on_alarm(void *acp, int success, - grpc_closure_list *closure_list) { +static void +tc_on_alarm (void *acp, int success, grpc_closure_list * closure_list) +{ int done; async_connect *ac = acp; - if (grpc_tcp_trace) { - gpr_log(GPR_DEBUG, "CLIENT_CONNECT: %s: on_alarm: success=%d", ac->addr_str, - success); - } - gpr_mu_lock(&ac->mu); - if (ac->fd != NULL) { - grpc_fd_shutdown(ac->fd, closure_list); - } + if (grpc_tcp_trace) + { + gpr_log (GPR_DEBUG, "CLIENT_CONNECT: %s: on_alarm: success=%d", ac->addr_str, success); + } + gpr_mu_lock (&ac->mu); + if (ac->fd != NULL) + { + grpc_fd_shutdown (ac->fd, closure_list); + } done = (--ac->refs == 0); - gpr_mu_unlock(&ac->mu); - if (done) { - gpr_mu_destroy(&ac->mu); - gpr_free(ac->addr_str); - gpr_free(ac); - } + gpr_mu_unlock (&ac->mu); + if (done) + { + gpr_mu_destroy (&ac->mu); + gpr_free (ac->addr_str); + gpr_free (ac); + } } -static void on_writable(void *acp, int success, - grpc_closure_list *closure_list) { +static void +on_writable (void *acp, int success, grpc_closure_list * closure_list) +{ async_connect *ac = acp; int so_error = 0; socklen_t so_error_size; @@ -123,93 +130,107 @@ static void on_writable(void *acp, int success, grpc_closure *closure = ac->closure; grpc_fd *fd; - if (grpc_tcp_trace) { - gpr_log(GPR_DEBUG, "CLIENT_CONNECT: %s: on_writable: success=%d", - ac->addr_str, success); - } + if (grpc_tcp_trace) + { + gpr_log (GPR_DEBUG, "CLIENT_CONNECT: %s: on_writable: success=%d", ac->addr_str, success); + } - gpr_mu_lock(&ac->mu); - GPR_ASSERT(ac->fd); + gpr_mu_lock (&ac->mu); + GPR_ASSERT (ac->fd); fd = ac->fd; ac->fd = NULL; - gpr_mu_unlock(&ac->mu); - - grpc_alarm_cancel(&ac->alarm, closure_list); - - gpr_mu_lock(&ac->mu); - if (success) { - do { - so_error_size = sizeof(so_error); - err = getsockopt(fd->fd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_size); - } while (err < 0 && errno == EINTR); - if (err < 0) { - gpr_log(GPR_ERROR, "getsockopt(ERROR): %s", strerror(errno)); - goto finish; - } else if (so_error != 0) { - if (so_error == ENOBUFS) { - /* We will get one of these errors if we have run out of - memory in the kernel for the data structures allocated - when you connect a socket. If this happens it is very - likely that if we wait a little bit then try again the - connection will work (since other programs or this - program will close their network connections and free up - memory). This does _not_ indicate that there is anything - wrong with the server we are connecting to, this is a - local problem. - - If you are looking at this code, then chances are that - your program or another program on the same computer - opened too many network connections. The "easy" fix: - don't do that! */ - gpr_log(GPR_ERROR, "kernel out of buffers"); - gpr_mu_unlock(&ac->mu); - grpc_fd_notify_on_write(fd, &ac->write_closure, closure_list); - return; - } else { - switch (so_error) { - case ECONNREFUSED: - gpr_log(GPR_ERROR, "socket error: connection refused"); - break; - default: - gpr_log(GPR_ERROR, "socket error: %d", so_error); - break; - } - goto finish; - } - } else { - grpc_pollset_set_del_fd(ac->interested_parties, fd, closure_list); - *ep = grpc_tcp_create(fd, GRPC_TCP_DEFAULT_READ_SLICE_SIZE, ac->addr_str); - fd = NULL; + gpr_mu_unlock (&ac->mu); + + grpc_alarm_cancel (&ac->alarm, closure_list); + + gpr_mu_lock (&ac->mu); + if (success) + { + do + { + so_error_size = sizeof (so_error); + err = getsockopt (fd->fd, SOL_SOCKET, SO_ERROR, &so_error, &so_error_size); + } + while (err < 0 && errno == EINTR); + if (err < 0) + { + gpr_log (GPR_ERROR, "getsockopt(ERROR): %s", strerror (errno)); + goto finish; + } + else if (so_error != 0) + { + if (so_error == ENOBUFS) + { + /* We will get one of these errors if we have run out of + memory in the kernel for the data structures allocated + when you connect a socket. If this happens it is very + likely that if we wait a little bit then try again the + connection will work (since other programs or this + program will close their network connections and free up + memory). This does _not_ indicate that there is anything + wrong with the server we are connecting to, this is a + local problem. + + If you are looking at this code, then chances are that + your program or another program on the same computer + opened too many network connections. The "easy" fix: + don't do that! */ + gpr_log (GPR_ERROR, "kernel out of buffers"); + gpr_mu_unlock (&ac->mu); + grpc_fd_notify_on_write (fd, &ac->write_closure, closure_list); + return; + } + else + { + switch (so_error) + { + case ECONNREFUSED: + gpr_log (GPR_ERROR, "socket error: connection refused"); + break; + default: + gpr_log (GPR_ERROR, "socket error: %d", so_error); + break; + } + goto finish; + } + } + else + { + grpc_pollset_set_del_fd (ac->interested_parties, fd, closure_list); + *ep = grpc_tcp_create (fd, GRPC_TCP_DEFAULT_READ_SLICE_SIZE, ac->addr_str); + fd = NULL; + goto finish; + } + } + else + { + gpr_log (GPR_ERROR, "on_writable failed during connect"); goto finish; } - } else { - gpr_log(GPR_ERROR, "on_writable failed during connect"); - goto finish; - } - abort(); + abort (); finish: - if (fd != NULL) { - grpc_pollset_set_del_fd(ac->interested_parties, fd, closure_list); - grpc_fd_orphan(fd, NULL, "tcp_client_orphan", closure_list); - fd = NULL; - } + if (fd != NULL) + { + grpc_pollset_set_del_fd (ac->interested_parties, fd, closure_list); + grpc_fd_orphan (fd, NULL, "tcp_client_orphan", closure_list); + fd = NULL; + } done = (--ac->refs == 0); - gpr_mu_unlock(&ac->mu); - if (done) { - gpr_mu_destroy(&ac->mu); - gpr_free(ac->addr_str); - gpr_free(ac); - } - grpc_closure_list_add(closure_list, closure, *ep != NULL); + gpr_mu_unlock (&ac->mu); + if (done) + { + gpr_mu_destroy (&ac->mu); + gpr_free (ac->addr_str); + gpr_free (ac); + } + grpc_closure_list_add (closure_list, closure, *ep != NULL); } -void grpc_tcp_client_connect(grpc_closure *closure, grpc_endpoint **ep, - grpc_pollset_set *interested_parties, - const struct sockaddr *addr, size_t addr_len, - gpr_timespec deadline, - grpc_closure_list *closure_list) { +void +grpc_tcp_client_connect (grpc_closure * closure, grpc_endpoint ** ep, grpc_pollset_set * interested_parties, const struct sockaddr *addr, size_t addr_len, gpr_timespec deadline, grpc_closure_list * closure_list) +{ int fd; grpc_dualstack_mode dsmode; int err; @@ -223,78 +244,84 @@ void grpc_tcp_client_connect(grpc_closure *closure, grpc_endpoint **ep, *ep = NULL; /* Use dualstack sockets where available. */ - if (grpc_sockaddr_to_v4mapped(addr, &addr6_v4mapped)) { - addr = (const struct sockaddr *)&addr6_v4mapped; - addr_len = sizeof(addr6_v4mapped); - } - - fd = grpc_create_dualstack_socket(addr, SOCK_STREAM, 0, &dsmode); - if (fd < 0) { - gpr_log(GPR_ERROR, "Unable to create socket: %s", strerror(errno)); - } - if (dsmode == GRPC_DSMODE_IPV4) { - /* If we got an AF_INET socket, map the address back to IPv4. */ - GPR_ASSERT(grpc_sockaddr_is_v4mapped(addr, &addr4_copy)); - addr = (struct sockaddr *)&addr4_copy; - addr_len = sizeof(addr4_copy); - } - if (!prepare_socket(addr, fd)) { - grpc_closure_list_add(closure_list, closure, 0); - return; - } - - do { - GPR_ASSERT(addr_len < ~(socklen_t)0); - err = connect(fd, addr, (socklen_t)addr_len); - } while (err < 0 && errno == EINTR); - - addr_str = grpc_sockaddr_to_uri(addr); - gpr_asprintf(&name, "tcp-client:%s", addr_str); - - fdobj = grpc_fd_create(fd, name); - - if (err >= 0) { - *ep = grpc_tcp_create(fdobj, GRPC_TCP_DEFAULT_READ_SLICE_SIZE, addr_str); - grpc_closure_list_add(closure_list, closure, 1); - goto done; - } - - if (errno != EWOULDBLOCK && errno != EINPROGRESS) { - gpr_log(GPR_ERROR, "connect error to '%s': %s", addr_str, strerror(errno)); - grpc_fd_orphan(fdobj, NULL, "tcp_client_connect_error", closure_list); - grpc_closure_list_add(closure_list, closure, 0); - goto done; - } - - grpc_pollset_set_add_fd(interested_parties, fdobj, closure_list); - - ac = gpr_malloc(sizeof(async_connect)); + if (grpc_sockaddr_to_v4mapped (addr, &addr6_v4mapped)) + { + addr = (const struct sockaddr *) &addr6_v4mapped; + addr_len = sizeof (addr6_v4mapped); + } + + fd = grpc_create_dualstack_socket (addr, SOCK_STREAM, 0, &dsmode); + if (fd < 0) + { + gpr_log (GPR_ERROR, "Unable to create socket: %s", strerror (errno)); + } + if (dsmode == GRPC_DSMODE_IPV4) + { + /* If we got an AF_INET socket, map the address back to IPv4. */ + GPR_ASSERT (grpc_sockaddr_is_v4mapped (addr, &addr4_copy)); + addr = (struct sockaddr *) &addr4_copy; + addr_len = sizeof (addr4_copy); + } + if (!prepare_socket (addr, fd)) + { + grpc_closure_list_add (closure_list, closure, 0); + return; + } + + do + { + GPR_ASSERT (addr_len < ~(socklen_t) 0); + err = connect (fd, addr, (socklen_t) addr_len); + } + while (err < 0 && errno == EINTR); + + addr_str = grpc_sockaddr_to_uri (addr); + gpr_asprintf (&name, "tcp-client:%s", addr_str); + + fdobj = grpc_fd_create (fd, name); + + if (err >= 0) + { + *ep = grpc_tcp_create (fdobj, GRPC_TCP_DEFAULT_READ_SLICE_SIZE, addr_str); + grpc_closure_list_add (closure_list, closure, 1); + goto done; + } + + if (errno != EWOULDBLOCK && errno != EINPROGRESS) + { + gpr_log (GPR_ERROR, "connect error to '%s': %s", addr_str, strerror (errno)); + grpc_fd_orphan (fdobj, NULL, "tcp_client_connect_error", closure_list); + grpc_closure_list_add (closure_list, closure, 0); + goto done; + } + + grpc_pollset_set_add_fd (interested_parties, fdobj, closure_list); + + ac = gpr_malloc (sizeof (async_connect)); ac->closure = closure; ac->ep = ep; ac->fd = fdobj; ac->interested_parties = interested_parties; ac->addr_str = addr_str; addr_str = NULL; - gpr_mu_init(&ac->mu); + gpr_mu_init (&ac->mu); ac->refs = 2; ac->write_closure.cb = on_writable; ac->write_closure.cb_arg = ac; - if (grpc_tcp_trace) { - gpr_log(GPR_DEBUG, "CLIENT_CONNECT: %s: asynchronously connecting", - ac->addr_str); - } + if (grpc_tcp_trace) + { + gpr_log (GPR_DEBUG, "CLIENT_CONNECT: %s: asynchronously connecting", ac->addr_str); + } - gpr_mu_lock(&ac->mu); - grpc_alarm_init(&ac->alarm, - gpr_convert_clock_type(deadline, GPR_CLOCK_MONOTONIC), - tc_on_alarm, ac, gpr_now(GPR_CLOCK_MONOTONIC), closure_list); - grpc_fd_notify_on_write(ac->fd, &ac->write_closure, closure_list); - gpr_mu_unlock(&ac->mu); + gpr_mu_lock (&ac->mu); + grpc_alarm_init (&ac->alarm, gpr_convert_clock_type (deadline, GPR_CLOCK_MONOTONIC), tc_on_alarm, ac, gpr_now (GPR_CLOCK_MONOTONIC), closure_list); + grpc_fd_notify_on_write (ac->fd, &ac->write_closure, closure_list); + gpr_mu_unlock (&ac->mu); done: - gpr_free(name); - gpr_free(addr_str); + gpr_free (name); + gpr_free (addr_str); } #endif |