mirror of https://github.com/grpc/grpc.git
parent
69563b9124
commit
be5f059219
12 changed files with 225 additions and 48 deletions
@ -0,0 +1,92 @@ |
||||
/*
|
||||
* |
||||
* 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 "test/core/security/oauth2_utils.h" |
||||
|
||||
#include <string.h> |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc/grpc_security.h> |
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <grpc/support/slice.h> |
||||
#include <grpc/support/sync.h> |
||||
|
||||
#include "src/core/security/credentials.h" |
||||
|
||||
typedef struct { |
||||
grpc_pollset pollset; |
||||
int is_done; |
||||
char *token; |
||||
} synchronizer; |
||||
|
||||
static void on_oauth2_response(void *user_data, grpc_credentials_md *md_elems, |
||||
size_t num_md, grpc_credentials_status status) { |
||||
synchronizer *sync = user_data; |
||||
char *token = NULL; |
||||
gpr_slice token_slice; |
||||
if (status == GRPC_CREDENTIALS_ERROR) { |
||||
gpr_log(GPR_ERROR, "Fetching token failed."); |
||||
} else { |
||||
GPR_ASSERT(num_md == 1); |
||||
token_slice = md_elems[0].value; |
||||
token = gpr_malloc(GPR_SLICE_LENGTH(token_slice) + 1); |
||||
memcpy(token, GPR_SLICE_START_PTR(token_slice), |
||||
GPR_SLICE_LENGTH(token_slice)); |
||||
token[GPR_SLICE_LENGTH(token_slice)] = '\0'; |
||||
} |
||||
gpr_mu_lock(GRPC_POLLSET_MU(&sync->pollset)); |
||||
sync->is_done = 1; |
||||
sync->token = token; |
||||
grpc_pollset_kick(&sync->pollset); |
||||
gpr_mu_unlock(GRPC_POLLSET_MU(&sync->pollset)); |
||||
} |
||||
|
||||
static void do_nothing(void *unused) {} |
||||
|
||||
char *grpc_test_fetch_oauth2_token_with_credentials(grpc_credentials *creds) { |
||||
synchronizer sync; |
||||
grpc_pollset_init(&sync.pollset); |
||||
sync.is_done = 0; |
||||
|
||||
grpc_credentials_get_request_metadata(creds, &sync.pollset, "", |
||||
on_oauth2_response, &sync); |
||||
|
||||
gpr_mu_lock(GRPC_POLLSET_MU(&sync.pollset)); |
||||
while (!sync.is_done) grpc_pollset_work(&sync.pollset, gpr_inf_future); |
||||
gpr_mu_unlock(GRPC_POLLSET_MU(&sync.pollset)); |
||||
|
||||
grpc_pollset_shutdown(&sync.pollset, do_nothing, NULL); |
||||
grpc_pollset_destroy(&sync.pollset); |
||||
return sync.token; |
||||
} |
@ -0,0 +1,51 @@ |
||||
/*
|
||||
* |
||||
* 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. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_TEST_CORE_SECURITY_OAUTH2_UTILS_H |
||||
#define GRPC_TEST_CORE_SECURITY_OAUTH2_UTILS_H |
||||
|
||||
#include "src/core/security/credentials.h" |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
/* Fetch oauth2 access token with a credentials object. Does not take ownership.
|
||||
Returns NULL on a failure. The caller should call gpr_free on the token. */ |
||||
char *grpc_test_fetch_oauth2_token_with_credentials(grpc_credentials *creds); |
||||
|
||||
#ifdef __cplusplus |
||||
} |
||||
#endif |
||||
|
||||
#endif /* GRPC_TEST_CORE_SECURITY_OAUTH2_UTILS_H */ |
Loading…
Reference in new issue