mirror of https://github.com/grpc/grpc.git
Nicolas "Pixel" Noble
10 years ago
committed by
Nicolas "Pixel" Noble
parent
8623ad0847
commit
94964fd0b2
14 changed files with 266 additions and 22 deletions
@ -0,0 +1,222 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015, 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 <grpc/support/port_platform.h> |
||||||
|
#ifdef GPR_WINSOCK_SOCKET |
||||||
|
|
||||||
|
#include "src/core/iomgr/sockaddr.h" |
||||||
|
#include "src/core/iomgr/resolve_address.h" |
||||||
|
|
||||||
|
#include <sys/types.h> |
||||||
|
#include <string.h> |
||||||
|
|
||||||
|
#include "src/core/iomgr/iomgr_internal.h" |
||||||
|
#include "src/core/iomgr/sockaddr_utils.h" |
||||||
|
#include "src/core/support/string.h" |
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
#include <grpc/support/thd.h> |
||||||
|
#include <grpc/support/time.h> |
||||||
|
|
||||||
|
typedef struct { |
||||||
|
char *name; |
||||||
|
char *default_port; |
||||||
|
grpc_resolve_cb cb; |
||||||
|
void *arg; |
||||||
|
} request; |
||||||
|
|
||||||
|
static void split_host_port(const char *name, char **host, char **port) { |
||||||
|
const char *host_start; |
||||||
|
size_t host_len; |
||||||
|
const char *port_start; |
||||||
|
|
||||||
|
*host = NULL; |
||||||
|
*port = NULL; |
||||||
|
|
||||||
|
if (name[0] == '[') { |
||||||
|
/* Parse a bracketed host, typically an IPv6 literal. */ |
||||||
|
const char *rbracket = strchr(name, ']'); |
||||||
|
if (rbracket == NULL) { |
||||||
|
/* Unmatched [ */ |
||||||
|
return; |
||||||
|
} |
||||||
|
if (rbracket[1] == '\0') { |
||||||
|
/* ]<end> */ |
||||||
|
port_start = NULL; |
||||||
|
} else if (rbracket[1] == ':') { |
||||||
|
/* ]:<port?> */ |
||||||
|
port_start = rbracket + 2; |
||||||
|
} else { |
||||||
|
/* ]<invalid> */ |
||||||
|
return; |
||||||
|
} |
||||||
|
host_start = name + 1; |
||||||
|
host_len = rbracket - host_start; |
||||||
|
if (memchr(host_start, ':', host_len) == NULL) { |
||||||
|
/* Require all bracketed hosts to contain a colon, because a hostname or
|
||||||
|
IPv4 address should never use brackets. */ |
||||||
|
return; |
||||||
|
} |
||||||
|
} else { |
||||||
|
const char *colon = strchr(name, ':'); |
||||||
|
if (colon != NULL && strchr(colon + 1, ':') == NULL) { |
||||||
|
/* Exactly 1 colon. Split into host:port. */ |
||||||
|
host_start = name; |
||||||
|
host_len = colon - name; |
||||||
|
port_start = colon + 1; |
||||||
|
} else { |
||||||
|
/* 0 or 2+ colons. Bare hostname or IPv6 litearal. */ |
||||||
|
host_start = name; |
||||||
|
host_len = strlen(name); |
||||||
|
port_start = NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/* Allocate return values. */ |
||||||
|
*host = gpr_malloc(host_len + 1); |
||||||
|
memcpy(*host, host_start, host_len); |
||||||
|
(*host)[host_len] = '\0'; |
||||||
|
|
||||||
|
if (port_start != NULL) { |
||||||
|
*port = gpr_strdup(port_start); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
grpc_resolved_addresses *grpc_blocking_resolve_address( |
||||||
|
const char *name, const char *default_port) { |
||||||
|
struct addrinfo hints; |
||||||
|
struct addrinfo *result = NULL, *resp; |
||||||
|
char *host; |
||||||
|
char *port; |
||||||
|
int s; |
||||||
|
size_t i; |
||||||
|
grpc_resolved_addresses *addrs = NULL; |
||||||
|
const gpr_timespec start_time = gpr_now(); |
||||||
|
|
||||||
|
/* parse name, splitting it into host and port parts */ |
||||||
|
split_host_port(name, &host, &port); |
||||||
|
if (host == NULL) { |
||||||
|
gpr_log(GPR_ERROR, "unparseable host:port: '%s'", name); |
||||||
|
goto done; |
||||||
|
} |
||||||
|
if (port == NULL) { |
||||||
|
if (default_port == NULL) { |
||||||
|
gpr_log(GPR_ERROR, "no port in name '%s'", name); |
||||||
|
goto done; |
||||||
|
} |
||||||
|
port = gpr_strdup(default_port); |
||||||
|
} |
||||||
|
|
||||||
|
/* Call getaddrinfo */ |
||||||
|
memset(&hints, 0, sizeof(hints)); |
||||||
|
hints.ai_family = AF_UNSPEC; /* ipv4 or ipv6 */ |
||||||
|
hints.ai_socktype = SOCK_STREAM; /* stream socket */ |
||||||
|
hints.ai_flags = AI_PASSIVE; /* for wildcard IP address */ |
||||||
|
|
||||||
|
s = getaddrinfo(host, port, &hints, &result); |
||||||
|
if (s != 0) { |
||||||
|
gpr_log(GPR_ERROR, "getaddrinfo: %s", gai_strerror(s)); |
||||||
|
goto done; |
||||||
|
} |
||||||
|
|
||||||
|
/* Success path: set addrs non-NULL, fill it in */ |
||||||
|
addrs = gpr_malloc(sizeof(grpc_resolved_addresses)); |
||||||
|
addrs->naddrs = 0; |
||||||
|
for (resp = result; resp != NULL; resp = resp->ai_next) { |
||||||
|
addrs->naddrs++; |
||||||
|
} |
||||||
|
addrs->addrs = gpr_malloc(sizeof(grpc_resolved_address) * addrs->naddrs); |
||||||
|
i = 0; |
||||||
|
for (resp = result; resp != NULL; resp = resp->ai_next) { |
||||||
|
memcpy(&addrs->addrs[i].addr, resp->ai_addr, resp->ai_addrlen); |
||||||
|
addrs->addrs[i].len = resp->ai_addrlen; |
||||||
|
i++; |
||||||
|
} |
||||||
|
|
||||||
|
/* Temporary logging, to help identify flakiness in dualstack_socket_test. */ |
||||||
|
{ |
||||||
|
const gpr_timespec delay = gpr_time_sub(gpr_now(), start_time); |
||||||
|
const int delay_ms = |
||||||
|
delay.tv_sec * GPR_MS_PER_SEC + delay.tv_nsec / GPR_NS_PER_MS; |
||||||
|
gpr_log(GPR_INFO, "logspam: getaddrinfo(%s, %s) resolved %d addrs in %dms:", |
||||||
|
host, port, addrs->naddrs, delay_ms); |
||||||
|
for (i = 0; i < addrs->naddrs; i++) { |
||||||
|
char *buf; |
||||||
|
grpc_sockaddr_to_string(&buf, (struct sockaddr *)&addrs->addrs[i].addr, |
||||||
|
0); |
||||||
|
gpr_log(GPR_INFO, "logspam: [%d] %s", i, buf); |
||||||
|
gpr_free(buf); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
done: |
||||||
|
gpr_free(host); |
||||||
|
gpr_free(port); |
||||||
|
if (result) { |
||||||
|
freeaddrinfo(result); |
||||||
|
} |
||||||
|
return addrs; |
||||||
|
} |
||||||
|
|
||||||
|
/* Thread function to asynch-ify grpc_blocking_resolve_address */ |
||||||
|
static void do_request(void *rp) { |
||||||
|
request *r = rp; |
||||||
|
grpc_resolved_addresses *resolved = |
||||||
|
grpc_blocking_resolve_address(r->name, r->default_port); |
||||||
|
void *arg = r->arg; |
||||||
|
grpc_resolve_cb cb = r->cb; |
||||||
|
gpr_free(r->name); |
||||||
|
gpr_free(r->default_port); |
||||||
|
gpr_free(r); |
||||||
|
cb(arg, resolved); |
||||||
|
grpc_iomgr_unref(); |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_resolved_addresses_destroy(grpc_resolved_addresses *addrs) { |
||||||
|
gpr_free(addrs->addrs); |
||||||
|
gpr_free(addrs); |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_resolve_address(const char *name, const char *default_port, |
||||||
|
grpc_resolve_cb cb, void *arg) { |
||||||
|
request *r = gpr_malloc(sizeof(request)); |
||||||
|
gpr_thd_id id; |
||||||
|
grpc_iomgr_ref(); |
||||||
|
r->name = gpr_strdup(name); |
||||||
|
r->default_port = gpr_strdup(default_port); |
||||||
|
r->cb = cb; |
||||||
|
r->arg = arg; |
||||||
|
gpr_thd_new(&id, do_request, r, NULL); |
||||||
|
} |
||||||
|
|
||||||
|
#endif |
||||||
Loading…
Reference in new issue