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