mirror of https://github.com/grpc/grpc.git
Craig Tiller
10 years ago
commit
b1c55ce13d
24 changed files with 1938 additions and 977 deletions
@ -0,0 +1,61 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_INTERNAL_CPP_CLIENT_SECURE_CREDENTIALS_H |
||||||
|
#define GRPC_INTERNAL_CPP_CLIENT_SECURE_CREDENTIALS_H |
||||||
|
|
||||||
|
#include <grpc/grpc_security.h> |
||||||
|
|
||||||
|
#include <grpc++/config.h> |
||||||
|
#include <grpc++/credentials.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
class SecureCredentials GRPC_FINAL : public Credentials { |
||||||
|
public: |
||||||
|
explicit SecureCredentials(grpc_credentials* c_creds) : c_creds_(c_creds) {} |
||||||
|
~SecureCredentials() GRPC_OVERRIDE { grpc_credentials_release(c_creds_); } |
||||||
|
grpc_credentials* GetRawCreds() { return c_creds_; } |
||||||
|
|
||||||
|
std::shared_ptr<grpc::ChannelInterface> CreateChannel( |
||||||
|
const string& target, const grpc::ChannelArguments& args) GRPC_OVERRIDE; |
||||||
|
SecureCredentials* AsSecureCredentials() GRPC_OVERRIDE { return this; } |
||||||
|
|
||||||
|
private: |
||||||
|
grpc_credentials* const c_creds_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPC_INTERNAL_CPP_CLIENT_SECURE_CREDENTIALS_H
|
||||||
|
|
||||||
@ -0,0 +1,60 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_INTERNAL_CPP_SERVER_SECURE_SERVER_CREDENTIALS_H |
||||||
|
#define GRPC_INTERNAL_CPP_SERVER_SECURE_SERVER_CREDENTIALS_H |
||||||
|
|
||||||
|
#include <grpc/grpc_security.h> |
||||||
|
|
||||||
|
#include <grpc++/server_credentials.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
class SecureServerCredentials GRPC_FINAL : public ServerCredentials { |
||||||
|
public: |
||||||
|
explicit SecureServerCredentials(grpc_server_credentials* creds) |
||||||
|
: creds_(creds) {} |
||||||
|
~SecureServerCredentials() GRPC_OVERRIDE { |
||||||
|
grpc_server_credentials_release(creds_); |
||||||
|
} |
||||||
|
|
||||||
|
int AddPortToServer(const grpc::string& addr, |
||||||
|
grpc_server* server) GRPC_OVERRIDE; |
||||||
|
|
||||||
|
private: |
||||||
|
grpc_server_credentials* const creds_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPC_INTERNAL_CPP_SERVER_SECURE_SERVER_CREDENTIALS_H
|
||||||
@ -0,0 +1,119 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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/cpp/interop/client_helper.h" |
||||||
|
|
||||||
|
#include <fstream> |
||||||
|
#include <memory> |
||||||
|
#include <sstream> |
||||||
|
|
||||||
|
#include <unistd.h> |
||||||
|
|
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
#include <gflags/gflags.h> |
||||||
|
#include <grpc++/channel_arguments.h> |
||||||
|
#include <grpc++/channel_interface.h> |
||||||
|
#include <grpc++/create_channel.h> |
||||||
|
#include <grpc++/credentials.h> |
||||||
|
#include <grpc++/stream.h> |
||||||
|
#include "test/cpp/util/create_test_channel.h" |
||||||
|
|
||||||
|
DECLARE_bool(enable_ssl); |
||||||
|
DECLARE_bool(use_prod_roots); |
||||||
|
DECLARE_int32(server_port); |
||||||
|
DECLARE_string(server_host); |
||||||
|
DECLARE_string(server_host_override); |
||||||
|
DECLARE_string(test_case); |
||||||
|
DECLARE_string(default_service_account); |
||||||
|
DECLARE_string(service_account_key_file); |
||||||
|
DECLARE_string(oauth_scope); |
||||||
|
|
||||||
|
// In some distros, gflags is in the namespace google, and in some others,
|
||||||
|
// in gflags. This hack is enabling us to find both.
|
||||||
|
namespace google {} |
||||||
|
namespace gflags {} |
||||||
|
using namespace google; |
||||||
|
using namespace gflags; |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
grpc::string GetServiceAccountJsonKey() { |
||||||
|
static grpc::string json_key; |
||||||
|
if (json_key.empty()) { |
||||||
|
std::ifstream json_key_file(FLAGS_service_account_key_file); |
||||||
|
std::stringstream key_stream; |
||||||
|
key_stream << json_key_file.rdbuf(); |
||||||
|
json_key = key_stream.str(); |
||||||
|
} |
||||||
|
return json_key; |
||||||
|
} |
||||||
|
|
||||||
|
std::shared_ptr<ChannelInterface> CreateChannelForTestCase( |
||||||
|
const grpc::string& test_case) { |
||||||
|
GPR_ASSERT(FLAGS_server_port); |
||||||
|
const int host_port_buf_size = 1024; |
||||||
|
char host_port[host_port_buf_size]; |
||||||
|
snprintf(host_port, host_port_buf_size, "%s:%d", FLAGS_server_host.c_str(), |
||||||
|
FLAGS_server_port); |
||||||
|
|
||||||
|
if (test_case == "service_account_creds") { |
||||||
|
std::unique_ptr<Credentials> creds; |
||||||
|
GPR_ASSERT(FLAGS_enable_ssl); |
||||||
|
grpc::string json_key = GetServiceAccountJsonKey(); |
||||||
|
creds = ServiceAccountCredentials(json_key, FLAGS_oauth_scope, |
||||||
|
std::chrono::hours(1)); |
||||||
|
return CreateTestChannel(host_port, FLAGS_server_host_override, |
||||||
|
FLAGS_enable_ssl, FLAGS_use_prod_roots, creds); |
||||||
|
} else if (test_case == "compute_engine_creds") { |
||||||
|
std::unique_ptr<Credentials> creds; |
||||||
|
GPR_ASSERT(FLAGS_enable_ssl); |
||||||
|
creds = ComputeEngineCredentials(); |
||||||
|
return CreateTestChannel(host_port, FLAGS_server_host_override, |
||||||
|
FLAGS_enable_ssl, FLAGS_use_prod_roots, creds); |
||||||
|
} else if (test_case == "jwt_token_creds") { |
||||||
|
std::unique_ptr<Credentials> creds; |
||||||
|
GPR_ASSERT(FLAGS_enable_ssl); |
||||||
|
grpc::string json_key = GetServiceAccountJsonKey(); |
||||||
|
creds = JWTCredentials(json_key, std::chrono::hours(1)); |
||||||
|
return CreateTestChannel(host_port, FLAGS_server_host_override, |
||||||
|
FLAGS_enable_ssl, FLAGS_use_prod_roots, creds); |
||||||
|
} else { |
||||||
|
return CreateTestChannel(host_port, FLAGS_server_host_override, |
||||||
|
FLAGS_enable_ssl, FLAGS_use_prod_roots); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc
|
||||||
@ -0,0 +1,53 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_CPP_INTEROP_CLIENT_HELPER_H |
||||||
|
#define GRPC_TEST_CPP_INTEROP_CLIENT_HELPER_H |
||||||
|
|
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include <grpc++/config.h> |
||||||
|
#include <grpc++/channel_interface.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
grpc::string GetServiceAccountJsonKey(); |
||||||
|
|
||||||
|
std::shared_ptr<ChannelInterface> CreateChannelForTestCase( |
||||||
|
const grpc::string& test_case); |
||||||
|
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPC_TEST_CPP_INTEROP_CLIENT_HELPER_H
|
||||||
@ -0,0 +1,311 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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/cpp/interop/interop_client.h" |
||||||
|
|
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include <unistd.h> |
||||||
|
|
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
#include <grpc++/channel_interface.h> |
||||||
|
#include <grpc++/client_context.h> |
||||||
|
#include <grpc++/status.h> |
||||||
|
#include <grpc++/stream.h> |
||||||
|
#include "test/cpp/interop/test.grpc.pb.h" |
||||||
|
#include "test/cpp/interop/empty.grpc.pb.h" |
||||||
|
#include "test/cpp/interop/messages.grpc.pb.h" |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
namespace { |
||||||
|
// The same value is defined by the Java client.
|
||||||
|
const std::vector<int> request_stream_sizes = {27182, 8, 1828, 45904}; |
||||||
|
const std::vector<int> response_stream_sizes = {31415, 9, 2653, 58979}; |
||||||
|
const int kNumResponseMessages = 2000; |
||||||
|
const int kResponseMessageSize = 1030; |
||||||
|
const int kReceiveDelayMilliSeconds = 20; |
||||||
|
const int kLargeRequestSize = 314159; |
||||||
|
const int kLargeResponseSize = 271812; |
||||||
|
} // namespace
|
||||||
|
|
||||||
|
InteropClient::InteropClient(std::shared_ptr<ChannelInterface> channel) |
||||||
|
: channel_(channel) {} |
||||||
|
|
||||||
|
void InteropClient::AssertOkOrPrintErrorStatus(const Status& s) { |
||||||
|
if (s.IsOk()) { |
||||||
|
return; |
||||||
|
} |
||||||
|
gpr_log(GPR_INFO, "Error status code: %d, message: %s", s.code(), |
||||||
|
s.details().c_str()); |
||||||
|
GPR_ASSERT(0); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoEmpty() { |
||||||
|
gpr_log(GPR_INFO, "Sending an empty rpc..."); |
||||||
|
std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); |
||||||
|
|
||||||
|
Empty request = Empty::default_instance(); |
||||||
|
Empty response = Empty::default_instance(); |
||||||
|
ClientContext context; |
||||||
|
|
||||||
|
Status s = stub->EmptyCall(&context, request, &response); |
||||||
|
AssertOkOrPrintErrorStatus(s); |
||||||
|
|
||||||
|
gpr_log(GPR_INFO, "Empty rpc done."); |
||||||
|
} |
||||||
|
|
||||||
|
// Shared code to set large payload, make rpc and check response payload.
|
||||||
|
void InteropClient::PerformLargeUnary(SimpleRequest* request, |
||||||
|
SimpleResponse* response) { |
||||||
|
std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); |
||||||
|
|
||||||
|
ClientContext context; |
||||||
|
request->set_response_type(PayloadType::COMPRESSABLE); |
||||||
|
request->set_response_size(kLargeResponseSize); |
||||||
|
grpc::string payload(kLargeRequestSize, '\0'); |
||||||
|
request->mutable_payload()->set_body(payload.c_str(), kLargeRequestSize); |
||||||
|
|
||||||
|
Status s = stub->UnaryCall(&context, *request, response); |
||||||
|
|
||||||
|
AssertOkOrPrintErrorStatus(s); |
||||||
|
GPR_ASSERT(response->payload().type() == PayloadType::COMPRESSABLE); |
||||||
|
GPR_ASSERT(response->payload().body() == |
||||||
|
grpc::string(kLargeResponseSize, '\0')); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoComputeEngineCreds( |
||||||
|
const grpc::string& default_service_account, |
||||||
|
const grpc::string& oauth_scope) { |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"Sending a large unary rpc with compute engine credentials ..."); |
||||||
|
SimpleRequest request; |
||||||
|
SimpleResponse response; |
||||||
|
request.set_fill_username(true); |
||||||
|
request.set_fill_oauth_scope(true); |
||||||
|
PerformLargeUnary(&request, &response); |
||||||
|
gpr_log(GPR_INFO, "Got username %s", response.username().c_str()); |
||||||
|
gpr_log(GPR_INFO, "Got oauth_scope %s", response.oauth_scope().c_str()); |
||||||
|
GPR_ASSERT(!response.username().empty()); |
||||||
|
GPR_ASSERT(response.username().c_str() == default_service_account); |
||||||
|
GPR_ASSERT(!response.oauth_scope().empty()); |
||||||
|
const char* oauth_scope_str = response.oauth_scope().c_str(); |
||||||
|
GPR_ASSERT(oauth_scope.find(oauth_scope_str) != grpc::string::npos); |
||||||
|
gpr_log(GPR_INFO, "Large unary with compute engine creds done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoServiceAccountCreds(const grpc::string& username, |
||||||
|
const grpc::string& oauth_scope) { |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"Sending a large unary rpc with service account credentials ..."); |
||||||
|
SimpleRequest request; |
||||||
|
SimpleResponse response; |
||||||
|
request.set_fill_username(true); |
||||||
|
request.set_fill_oauth_scope(true); |
||||||
|
PerformLargeUnary(&request, &response); |
||||||
|
GPR_ASSERT(!response.username().empty()); |
||||||
|
GPR_ASSERT(!response.oauth_scope().empty()); |
||||||
|
GPR_ASSERT(username.find(response.username()) != grpc::string::npos); |
||||||
|
const char* oauth_scope_str = response.oauth_scope().c_str(); |
||||||
|
GPR_ASSERT(oauth_scope.find(oauth_scope_str) != grpc::string::npos); |
||||||
|
gpr_log(GPR_INFO, "Large unary with service account creds done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoJwtTokenCreds(const grpc::string& username) { |
||||||
|
gpr_log(GPR_INFO, "Sending a large unary rpc with JWT token credentials ..."); |
||||||
|
SimpleRequest request; |
||||||
|
SimpleResponse response; |
||||||
|
request.set_fill_username(true); |
||||||
|
PerformLargeUnary(&request, &response); |
||||||
|
GPR_ASSERT(!response.username().empty()); |
||||||
|
GPR_ASSERT(username.find(response.username()) != grpc::string::npos); |
||||||
|
gpr_log(GPR_INFO, "Large unary with JWT token creds done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoLargeUnary() { |
||||||
|
gpr_log(GPR_INFO, "Sending a large unary rpc..."); |
||||||
|
SimpleRequest request; |
||||||
|
SimpleResponse response; |
||||||
|
PerformLargeUnary(&request, &response); |
||||||
|
gpr_log(GPR_INFO, "Large unary done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoRequestStreaming() { |
||||||
|
gpr_log(GPR_INFO, "Sending request steaming rpc ..."); |
||||||
|
std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); |
||||||
|
|
||||||
|
ClientContext context; |
||||||
|
StreamingInputCallRequest request; |
||||||
|
StreamingInputCallResponse response; |
||||||
|
|
||||||
|
std::unique_ptr<ClientWriter<StreamingInputCallRequest>> stream( |
||||||
|
stub->StreamingInputCall(&context, &response)); |
||||||
|
|
||||||
|
int aggregated_payload_size = 0; |
||||||
|
for (unsigned int i = 0; i < request_stream_sizes.size(); ++i) { |
||||||
|
Payload* payload = request.mutable_payload(); |
||||||
|
payload->set_body(grpc::string(request_stream_sizes[i], '\0')); |
||||||
|
GPR_ASSERT(stream->Write(request)); |
||||||
|
aggregated_payload_size += request_stream_sizes[i]; |
||||||
|
} |
||||||
|
stream->WritesDone(); |
||||||
|
Status s = stream->Finish(); |
||||||
|
|
||||||
|
GPR_ASSERT(response.aggregated_payload_size() == aggregated_payload_size); |
||||||
|
AssertOkOrPrintErrorStatus(s); |
||||||
|
gpr_log(GPR_INFO, "Request streaming done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoResponseStreaming() { |
||||||
|
gpr_log(GPR_INFO, "Receiving response steaming rpc ..."); |
||||||
|
std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); |
||||||
|
|
||||||
|
ClientContext context; |
||||||
|
StreamingOutputCallRequest request; |
||||||
|
for (unsigned int i = 0; i < response_stream_sizes.size(); ++i) { |
||||||
|
ResponseParameters* response_parameter = request.add_response_parameters(); |
||||||
|
response_parameter->set_size(response_stream_sizes[i]); |
||||||
|
} |
||||||
|
StreamingOutputCallResponse response; |
||||||
|
std::unique_ptr<ClientReader<StreamingOutputCallResponse>> stream( |
||||||
|
stub->StreamingOutputCall(&context, request)); |
||||||
|
|
||||||
|
unsigned int i = 0; |
||||||
|
while (stream->Read(&response)) { |
||||||
|
GPR_ASSERT(response.payload().body() == |
||||||
|
grpc::string(response_stream_sizes[i], '\0')); |
||||||
|
++i; |
||||||
|
} |
||||||
|
GPR_ASSERT(response_stream_sizes.size() == i); |
||||||
|
Status s = stream->Finish(); |
||||||
|
|
||||||
|
AssertOkOrPrintErrorStatus(s); |
||||||
|
gpr_log(GPR_INFO, "Response streaming done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoResponseStreamingWithSlowConsumer() { |
||||||
|
gpr_log(GPR_INFO, "Receiving response steaming rpc with slow consumer ..."); |
||||||
|
std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); |
||||||
|
|
||||||
|
ClientContext context; |
||||||
|
StreamingOutputCallRequest request; |
||||||
|
|
||||||
|
for (int i = 0; i < kNumResponseMessages; ++i) { |
||||||
|
ResponseParameters* response_parameter = request.add_response_parameters(); |
||||||
|
response_parameter->set_size(kResponseMessageSize); |
||||||
|
} |
||||||
|
StreamingOutputCallResponse response; |
||||||
|
std::unique_ptr<ClientReader<StreamingOutputCallResponse>> stream( |
||||||
|
stub->StreamingOutputCall(&context, request)); |
||||||
|
|
||||||
|
int i = 0; |
||||||
|
while (stream->Read(&response)) { |
||||||
|
GPR_ASSERT(response.payload().body() == |
||||||
|
grpc::string(kResponseMessageSize, '\0')); |
||||||
|
gpr_log(GPR_INFO, "received message %d", i); |
||||||
|
usleep(kReceiveDelayMilliSeconds * 1000); |
||||||
|
++i; |
||||||
|
} |
||||||
|
GPR_ASSERT(kNumResponseMessages == i); |
||||||
|
Status s = stream->Finish(); |
||||||
|
|
||||||
|
AssertOkOrPrintErrorStatus(s); |
||||||
|
gpr_log(GPR_INFO, "Response streaming done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoHalfDuplex() { |
||||||
|
gpr_log(GPR_INFO, "Sending half-duplex streaming rpc ..."); |
||||||
|
std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); |
||||||
|
|
||||||
|
ClientContext context; |
||||||
|
std::unique_ptr<ClientReaderWriter<StreamingOutputCallRequest, |
||||||
|
StreamingOutputCallResponse>> |
||||||
|
stream(stub->HalfDuplexCall(&context)); |
||||||
|
|
||||||
|
StreamingOutputCallRequest request; |
||||||
|
ResponseParameters* response_parameter = request.add_response_parameters(); |
||||||
|
for (unsigned int i = 0; i < response_stream_sizes.size(); ++i) { |
||||||
|
response_parameter->set_size(response_stream_sizes[i]); |
||||||
|
GPR_ASSERT(stream->Write(request)); |
||||||
|
} |
||||||
|
stream->WritesDone(); |
||||||
|
|
||||||
|
unsigned int i = 0; |
||||||
|
StreamingOutputCallResponse response; |
||||||
|
while (stream->Read(&response)) { |
||||||
|
GPR_ASSERT(response.payload().has_body()); |
||||||
|
GPR_ASSERT(response.payload().body() == |
||||||
|
grpc::string(response_stream_sizes[i], '\0')); |
||||||
|
++i; |
||||||
|
} |
||||||
|
GPR_ASSERT(response_stream_sizes.size() == i); |
||||||
|
Status s = stream->Finish(); |
||||||
|
AssertOkOrPrintErrorStatus(s); |
||||||
|
gpr_log(GPR_INFO, "Half-duplex streaming rpc done."); |
||||||
|
} |
||||||
|
|
||||||
|
void InteropClient::DoPingPong() { |
||||||
|
gpr_log(GPR_INFO, "Sending Ping Pong streaming rpc ..."); |
||||||
|
std::unique_ptr<TestService::Stub> stub(TestService::NewStub(channel_)); |
||||||
|
|
||||||
|
ClientContext context; |
||||||
|
std::unique_ptr<ClientReaderWriter<StreamingOutputCallRequest, |
||||||
|
StreamingOutputCallResponse>> |
||||||
|
stream(stub->FullDuplexCall(&context)); |
||||||
|
|
||||||
|
StreamingOutputCallRequest request; |
||||||
|
request.set_response_type(PayloadType::COMPRESSABLE); |
||||||
|
ResponseParameters* response_parameter = request.add_response_parameters(); |
||||||
|
Payload* payload = request.mutable_payload(); |
||||||
|
StreamingOutputCallResponse response; |
||||||
|
for (unsigned int i = 0; i < request_stream_sizes.size(); ++i) { |
||||||
|
response_parameter->set_size(response_stream_sizes[i]); |
||||||
|
payload->set_body(grpc::string(request_stream_sizes[i], '\0')); |
||||||
|
GPR_ASSERT(stream->Write(request)); |
||||||
|
GPR_ASSERT(stream->Read(&response)); |
||||||
|
GPR_ASSERT(response.payload().has_body()); |
||||||
|
GPR_ASSERT(response.payload().body() == |
||||||
|
grpc::string(response_stream_sizes[i], '\0')); |
||||||
|
} |
||||||
|
|
||||||
|
stream->WritesDone(); |
||||||
|
GPR_ASSERT(!stream->Read(&response)); |
||||||
|
Status s = stream->Finish(); |
||||||
|
AssertOkOrPrintErrorStatus(s); |
||||||
|
gpr_log(GPR_INFO, "Ping pong streaming done."); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc
|
||||||
@ -0,0 +1,79 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_CPP_INTEROP_INTEROP_CLIENT_H |
||||||
|
#define GRPC_TEST_CPP_INTEROP_INTEROP_CLIENT_H |
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <grpc++/channel_interface.h> |
||||||
|
#include <grpc++/status.h> |
||||||
|
#include "test/cpp/interop/messages.grpc.pb.h" |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
class InteropClient { |
||||||
|
public: |
||||||
|
explicit InteropClient(std::shared_ptr<ChannelInterface> channel); |
||||||
|
~InteropClient() {} |
||||||
|
|
||||||
|
void Reset(std::shared_ptr<ChannelInterface> channel) { channel_ = channel; } |
||||||
|
|
||||||
|
void DoEmpty(); |
||||||
|
void DoLargeUnary(); |
||||||
|
void DoPingPong(); |
||||||
|
void DoHalfDuplex(); |
||||||
|
void DoRequestStreaming(); |
||||||
|
void DoResponseStreaming(); |
||||||
|
void DoResponseStreamingWithSlowConsumer(); |
||||||
|
// Auth tests.
|
||||||
|
// username is a string containing the user email
|
||||||
|
void DoJwtTokenCreds(const grpc::string& username); |
||||||
|
void DoComputeEngineCreds(const grpc::string& default_service_account, |
||||||
|
const grpc::string& oauth_scope); |
||||||
|
// username is a string containing the user email
|
||||||
|
void DoServiceAccountCreds(const grpc::string& username, |
||||||
|
const grpc::string& oauth_scope); |
||||||
|
|
||||||
|
private: |
||||||
|
void PerformLargeUnary(SimpleRequest* request, SimpleResponse* response); |
||||||
|
void AssertOkOrPrintErrorStatus(const Status& s); |
||||||
|
|
||||||
|
std::shared_ptr<ChannelInterface> channel_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPC_TEST_CPP_INTEROP_INTEROP_CLIENT_H
|
||||||
@ -0,0 +1,69 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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/cpp/interop/server_helper.h" |
||||||
|
|
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include <gflags/gflags.h> |
||||||
|
#include "test/core/end2end/data/ssl_test_data.h" |
||||||
|
#include <grpc++/config.h> |
||||||
|
#include <grpc++/server_credentials.h> |
||||||
|
|
||||||
|
DECLARE_bool(enable_ssl); |
||||||
|
|
||||||
|
// In some distros, gflags is in the namespace google, and in some others,
|
||||||
|
// in gflags. This hack is enabling us to find both.
|
||||||
|
namespace google {} |
||||||
|
namespace gflags {} |
||||||
|
using namespace google; |
||||||
|
using namespace gflags; |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
std::shared_ptr<ServerCredentials> CreateInteropServerCredentials() { |
||||||
|
if (FLAGS_enable_ssl) { |
||||||
|
SslServerCredentialsOptions::PemKeyCertPair pkcp = {test_server1_key, |
||||||
|
test_server1_cert}; |
||||||
|
SslServerCredentialsOptions ssl_opts; |
||||||
|
ssl_opts.pem_root_certs = ""; |
||||||
|
ssl_opts.pem_key_cert_pairs.push_back(pkcp); |
||||||
|
return SslServerCredentials(ssl_opts); |
||||||
|
} else { |
||||||
|
return InsecureServerCredentials(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc
|
||||||
@ -0,0 +1,49 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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_CPP_INTEROP_SERVER_HELPER_H |
||||||
|
#define GRPC_TEST_CPP_INTEROP_SERVER_HELPER_H |
||||||
|
|
||||||
|
#include <memory> |
||||||
|
|
||||||
|
#include <grpc++/server_credentials.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
std::shared_ptr<ServerCredentials> CreateInteropServerCredentials(); |
||||||
|
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPC_TEST_CPP_INTEROP_SERVER_HELPER_H
|
||||||
Loading…
Reference in new issue