mirror of https://github.com/grpc/grpc.git
Yuchen Zeng
8 years ago
parent
ca294650ba
commit
47155ed254
22 changed files with 423 additions and 176 deletions
@ -0,0 +1,89 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_CORE_LIB_IOMGR_TCP_SERVER_UTILS_POSIX_H |
||||
#define GRPC_CORE_LIB_IOMGR_TCP_SERVER_UTILS_POSIX_H |
||||
|
||||
#include "src/core/lib/iomgr/ev_posix.h" |
||||
#include "src/core/lib/iomgr/resolve_address.h" |
||||
#include "src/core/lib/iomgr/socket_utils_posix.h" |
||||
#include "src/core/lib/iomgr/tcp_server.h" |
||||
|
||||
/* one listening port */ |
||||
typedef struct grpc_tcp_listener { |
||||
int fd; |
||||
grpc_fd *emfd; |
||||
grpc_tcp_server *server; |
||||
grpc_resolved_address addr; |
||||
int port; |
||||
unsigned port_index; |
||||
unsigned fd_index; |
||||
grpc_closure read_closure; |
||||
grpc_closure destroyed_closure; |
||||
struct grpc_tcp_listener *next; |
||||
/* sibling is a linked list of all listeners for a given port. add_port and
|
||||
clone_port place all new listeners in the same sibling list. A member of |
||||
the 'sibling' list is also a member of the 'next' list. The head of each |
||||
sibling list has is_sibling==0, and subsequent members of sibling lists |
||||
have is_sibling==1. is_sibling allows separate sibling lists to be |
||||
identified while iterating through 'next'. */ |
||||
struct grpc_tcp_listener *sibling; |
||||
int is_sibling; |
||||
} grpc_tcp_listener; |
||||
|
||||
/* If successful, add a listener to \a s for \a addr, set \a dsmode for the
|
||||
socket, and return the \a listener. */ |
||||
grpc_error *grpc_tcp_server_add_addr(grpc_tcp_server *s, |
||||
const grpc_resolved_address *addr, |
||||
unsigned port_index, unsigned fd_index, |
||||
grpc_dualstack_mode *dsmode, |
||||
grpc_tcp_listener **listener); |
||||
|
||||
/* Return the listener in \a s with address \a addr or NULL. */ |
||||
grpc_tcp_listener *grpc_tcp_server_find_listener_with_addr( |
||||
grpc_tcp_server *s, grpc_resolved_address *addr); |
||||
|
||||
/* Get all addresses assigned to network interfaces on the machine and create a
|
||||
listener for each. requested_port is the port to use for every listener, or 0 |
||||
to select one random port that will be used for every listener. Set *out_port |
||||
to the port selected. Return GRPC_ERROR_NONE only if all listeners were |
||||
added. */ |
||||
grpc_error *grpc_tcp_server_add_all_local_addrs(grpc_tcp_server *s, |
||||
unsigned port_index, |
||||
int requested_port, |
||||
int *out_port); |
||||
|
||||
/* Ruturn true if the platform supports ifaddrs */ |
||||
bool grpc_tcp_server_have_ifaddrs(void); |
||||
|
||||
#endif /* GRPC_CORE_LIB_IOMGR_TCP_SERVER_UTILS_POSIX_H */ |
||||
@ -0,0 +1,178 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#include "src/core/lib/iomgr/port.h" |
||||
|
||||
#ifdef GRPC_HAVE_IFADDRS |
||||
|
||||
#include "src/core/lib/iomgr/tcp_server_utils_posix.h" |
||||
|
||||
#include <errno.h> |
||||
#include <ifaddrs.h> |
||||
#include <stddef.h> |
||||
#include <string.h> |
||||
|
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <grpc/support/string_util.h> |
||||
|
||||
#include "src/core/lib/iomgr/error.h" |
||||
#include "src/core/lib/iomgr/sockaddr.h" |
||||
#include "src/core/lib/iomgr/sockaddr_utils.h" |
||||
|
||||
/* Bind to "::" to get a port number not used by any address. */ |
||||
static grpc_error *get_unused_port(int *port) { |
||||
grpc_resolved_address wild; |
||||
grpc_sockaddr_make_wildcard6(0, &wild); |
||||
grpc_dualstack_mode dsmode; |
||||
int fd; |
||||
grpc_error *err = |
||||
grpc_create_dualstack_socket(&wild, SOCK_STREAM, 0, &dsmode, &fd); |
||||
if (err != GRPC_ERROR_NONE) { |
||||
return err; |
||||
} |
||||
if (dsmode == GRPC_DSMODE_IPV4) { |
||||
grpc_sockaddr_make_wildcard4(0, &wild); |
||||
} |
||||
if (bind(fd, (const struct sockaddr *)wild.addr, (socklen_t)wild.len) != 0) { |
||||
err = GRPC_OS_ERROR(errno, "bind"); |
||||
close(fd); |
||||
return err; |
||||
} |
||||
if (getsockname(fd, (struct sockaddr *)wild.addr, (socklen_t *)&wild.len) != |
||||
0) { |
||||
err = GRPC_OS_ERROR(errno, "getsockname"); |
||||
close(fd); |
||||
return err; |
||||
} |
||||
close(fd); |
||||
*port = grpc_sockaddr_get_port(&wild); |
||||
return *port <= 0 ? GRPC_ERROR_CREATE("Bad port") : GRPC_ERROR_NONE; |
||||
} |
||||
|
||||
grpc_error *grpc_tcp_server_add_all_local_addrs(grpc_tcp_server *s, |
||||
unsigned port_index, |
||||
int requested_port, |
||||
int *out_port) { |
||||
struct ifaddrs *ifa = NULL; |
||||
struct ifaddrs *ifa_it; |
||||
unsigned fd_index = 0; |
||||
grpc_tcp_listener *sp = NULL; |
||||
grpc_error *err = GRPC_ERROR_NONE; |
||||
if (requested_port == 0) { |
||||
/* Note: There could be a race where some local addrs can listen on the
|
||||
selected port and some can't. The sane way to handle this would be to |
||||
retry by recreating the whole grpc_tcp_server. Backing out individual |
||||
listeners and orphaning the FDs looks like too much trouble. */ |
||||
if ((err = get_unused_port(&requested_port)) != GRPC_ERROR_NONE) { |
||||
return err; |
||||
} else if (requested_port <= 0) { |
||||
return GRPC_ERROR_CREATE("Bad get_unused_port()"); |
||||
} |
||||
gpr_log(GPR_DEBUG, "Picked unused port %d", requested_port); |
||||
} |
||||
if (getifaddrs(&ifa) != 0 || ifa == NULL) { |
||||
return GRPC_OS_ERROR(errno, "getifaddrs"); |
||||
} |
||||
for (ifa_it = ifa; ifa_it != NULL; ifa_it = ifa_it->ifa_next) { |
||||
grpc_resolved_address addr; |
||||
char *addr_str = NULL; |
||||
grpc_dualstack_mode dsmode; |
||||
grpc_tcp_listener *new_sp = NULL; |
||||
const char *ifa_name = (ifa_it->ifa_name ? ifa_it->ifa_name : "<unknown>"); |
||||
if (ifa_it->ifa_addr == NULL) { |
||||
continue; |
||||
} else if (ifa_it->ifa_addr->sa_family == AF_INET) { |
||||
addr.len = sizeof(struct sockaddr_in); |
||||
} else if (ifa_it->ifa_addr->sa_family == AF_INET6) { |
||||
addr.len = sizeof(struct sockaddr_in6); |
||||
} else { |
||||
continue; |
||||
} |
||||
memcpy(addr.addr, ifa_it->ifa_addr, addr.len); |
||||
if (!grpc_sockaddr_set_port(&addr, requested_port)) { |
||||
/* Should never happen, because we check sa_family above. */ |
||||
err = GRPC_ERROR_CREATE("Failed to set port"); |
||||
break; |
||||
} |
||||
if (grpc_sockaddr_to_string(&addr_str, &addr, 0) < 0) { |
||||
addr_str = gpr_strdup("<error>"); |
||||
} |
||||
gpr_log(GPR_DEBUG, |
||||
"Adding local addr from interface %s flags 0x%x to server: %s", |
||||
ifa_name, ifa_it->ifa_flags, addr_str); |
||||
/* We could have multiple interfaces with the same address (e.g., bonding),
|
||||
so look for duplicates. */ |
||||
if (grpc_tcp_server_find_listener_with_addr(s, &addr) != NULL) { |
||||
gpr_log(GPR_DEBUG, "Skipping duplicate addr %s on interface %s", addr_str, |
||||
ifa_name); |
||||
gpr_free(addr_str); |
||||
continue; |
||||
} |
||||
if ((err = grpc_tcp_server_add_addr(s, &addr, port_index, fd_index, &dsmode, |
||||
&new_sp)) != GRPC_ERROR_NONE) { |
||||
char *err_str = NULL; |
||||
grpc_error *root_err; |
||||
if (gpr_asprintf(&err_str, "Failed to add listener: %s", addr_str) < 0) { |
||||
err_str = gpr_strdup("Failed to add listener"); |
||||
} |
||||
root_err = GRPC_ERROR_CREATE(err_str); |
||||
gpr_free(err_str); |
||||
gpr_free(addr_str); |
||||
err = grpc_error_add_child(root_err, err); |
||||
break; |
||||
} else { |
||||
GPR_ASSERT(requested_port == new_sp->port); |
||||
++fd_index; |
||||
if (sp != NULL) { |
||||
new_sp->is_sibling = 1; |
||||
sp->sibling = new_sp; |
||||
} |
||||
sp = new_sp; |
||||
} |
||||
gpr_free(addr_str); |
||||
} |
||||
freeifaddrs(ifa); |
||||
if (err != GRPC_ERROR_NONE) { |
||||
return err; |
||||
} else if (sp == NULL) { |
||||
return GRPC_ERROR_CREATE("No local addresses"); |
||||
} else { |
||||
*out_port = sp->port; |
||||
return GRPC_ERROR_NONE; |
||||
} |
||||
} |
||||
|
||||
bool grpc_tcp_server_have_ifaddrs(void) { return true; } |
||||
|
||||
#endif /* GRPC_HAVE_IFADDRS */ |
||||
@ -0,0 +1,49 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#include "src/core/lib/iomgr/port.h" |
||||
|
||||
#ifndef GRPC_HAVE_IFADDRS |
||||
|
||||
#include "src/core/lib/iomgr/tcp_server_utils_posix.h" |
||||
|
||||
grpc_error *grpc_tcp_server_add_all_local_addrs(grpc_tcp_server *s, |
||||
unsigned port_index, |
||||
int requested_port, |
||||
int *out_port) { |
||||
return GRPC_ERROR_CREATE("no ifaddrs available"); |
||||
} |
||||
|
||||
bool grpc_tcp_server_have_ifaddrs(void) { return false; } |
||||
|
||||
#endif /* ifndef GRPC_HAVE_IFADDRS */ |
||||
Loading…
Reference in new issue