mirror of https://github.com/grpc/grpc.git
Donna Dionne
10 years ago
commit
2c77f75b74
35 changed files with 934 additions and 170 deletions
@ -0,0 +1,166 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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/host_port.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; |
||||||
|
|
||||||
|
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 */ |
||||||
|
gpr_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 |
||||||
@ -0,0 +1,110 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 <stdio.h> |
||||||
|
#include <string.h> |
||||||
|
|
||||||
|
#include "src/core/security/credentials.h" |
||||||
|
#include "src/core/security/json_token.h" |
||||||
|
#include "src/core/support/file.h" |
||||||
|
|
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/cmdline.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
#include <grpc/support/slice.h> |
||||||
|
|
||||||
|
void create_jwt(const char *json_key_file_path, const char *service_url, |
||||||
|
const char *scope) { |
||||||
|
grpc_auth_json_key key; |
||||||
|
int ok = 0; |
||||||
|
char *jwt; |
||||||
|
gpr_slice json_key_data = gpr_load_file(json_key_file_path, &ok); |
||||||
|
if (!ok) { |
||||||
|
fprintf(stderr, "Could not read %s.\n", json_key_file_path); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
key = grpc_auth_json_key_create_from_string( |
||||||
|
(const char *)GPR_SLICE_START_PTR(json_key_data)); |
||||||
|
gpr_slice_unref(json_key_data); |
||||||
|
if (!grpc_auth_json_key_is_valid(&key)) { |
||||||
|
fprintf(stderr, "Could not parse json key.\n"); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
jwt = grpc_jwt_encode_and_sign( |
||||||
|
&key, service_url == NULL ? GRPC_JWT_OAUTH2_AUDIENCE : service_url, |
||||||
|
grpc_max_auth_token_lifetime, scope); |
||||||
|
grpc_auth_json_key_destruct(&key); |
||||||
|
if (jwt == NULL) { |
||||||
|
fprintf(stderr, "Could not create JWT.\n"); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
fprintf(stdout, "%s\n", jwt); |
||||||
|
gpr_free(jwt); |
||||||
|
} |
||||||
|
|
||||||
|
int main(int argc, char **argv) { |
||||||
|
char *scope = NULL; |
||||||
|
char *json_key_file_path = NULL; |
||||||
|
char *service_url = NULL; |
||||||
|
gpr_cmdline *cl = gpr_cmdline_create("create_jwt"); |
||||||
|
gpr_cmdline_add_string(cl, "json_key", "File path of the json key.", |
||||||
|
&json_key_file_path); |
||||||
|
gpr_cmdline_add_string(cl, "scope", |
||||||
|
"OPTIONAL Space delimited permissions. Mutually " |
||||||
|
"exclusive with service_url", |
||||||
|
&scope); |
||||||
|
gpr_cmdline_add_string(cl, "service_url", |
||||||
|
"OPTIONAL service URL. Mutually exclusive with scope.", |
||||||
|
&service_url); |
||||||
|
gpr_cmdline_parse(cl, argc, argv); |
||||||
|
|
||||||
|
if (json_key_file_path == NULL) { |
||||||
|
fprintf(stderr, "Missing --json_key option.\n"); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
if (scope != NULL) { |
||||||
|
if (service_url != NULL) { |
||||||
|
fprintf(stderr, |
||||||
|
"Options --scope and --service_url are mutually exclusive.\n"); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
} else if (service_url == NULL) { |
||||||
|
fprintf(stderr, "Need one of --service_url or --scope options.\n"); |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
|
||||||
|
create_jwt(json_key_file_path, service_url, scope); |
||||||
|
|
||||||
|
gpr_cmdline_destroy(cl); |
||||||
|
return 0; |
||||||
|
} |
||||||
Loading…
Reference in new issue