1 /*
2 *
3 * Copyright 2017 gRPC authors.
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *
17 */
18
19 #include <grpc/support/log.h>
20 #include <grpcpp/channel.h>
21 #include <grpcpp/create_channel.h>
22 #include <grpcpp/impl/grpc_library.h>
23 #include <grpcpp/security/credentials.h>
24 #include <grpcpp/security/server_credentials.h>
25 #include <grpcpp/server.h>
26 #include <grpcpp/server_builder.h>
27 #include <gtest/gtest.h>
28
29 #include "src/core/ext/transport/chttp2/transport/chttp2_transport.h"
30 #include "src/core/lib/channel/channel_args.h"
31 #include "src/core/lib/iomgr/endpoint.h"
32 #include "src/core/lib/iomgr/endpoint_pair.h"
33 #include "src/core/lib/iomgr/exec_ctx.h"
34 #include "src/core/lib/iomgr/tcp_posix.h"
35 #include "src/core/lib/surface/channel.h"
36 #include "src/core/lib/surface/completion_queue.h"
37 #include "src/core/lib/surface/server.h"
38 #include "test/core/util/passthru_endpoint.h"
39 #include "test/core/util/port.h"
40
41 #include "src/cpp/client/create_channel_internal.h"
42 #include "src/proto/grpc/testing/echo.grpc.pb.h"
43 #include "test/core/util/test_config.h"
44
45 namespace grpc {
46 namespace testing {
47
tag(intptr_t x)48 static void* tag(intptr_t x) { return reinterpret_cast<void*>(x); }
49
ApplyCommonServerBuilderConfig(ServerBuilder * b)50 static void ApplyCommonServerBuilderConfig(ServerBuilder* b) {
51 b->SetMaxReceiveMessageSize(INT_MAX);
52 b->SetMaxSendMessageSize(INT_MAX);
53 }
54
ApplyCommonChannelArguments(ChannelArguments * c)55 static void ApplyCommonChannelArguments(ChannelArguments* c) {
56 c->SetInt(GRPC_ARG_MAX_RECEIVE_MESSAGE_LENGTH, INT_MAX);
57 c->SetInt(GRPC_ARG_MAX_SEND_MESSAGE_LENGTH, INT_MAX);
58 }
59
60 class EndpointPairFixture {
61 public:
EndpointPairFixture(Service * service,grpc_endpoint_pair endpoints)62 EndpointPairFixture(Service* service, grpc_endpoint_pair endpoints) {
63 ServerBuilder b;
64 cq_ = b.AddCompletionQueue(true);
65 b.RegisterService(service);
66 ApplyCommonServerBuilderConfig(&b);
67 server_ = b.BuildAndStart();
68
69 grpc_core::ExecCtx exec_ctx;
70
71 /* add server endpoint to server_ */
72 {
73 const grpc_channel_args* server_args =
74 grpc_server_get_channel_args(server_->c_server());
75 grpc_transport* transport = grpc_create_chttp2_transport(
76 server_args, endpoints.server, false /* is_client */);
77
78 for (grpc_pollset* pollset :
79 grpc_server_get_pollsets(server_->c_server())) {
80 grpc_endpoint_add_to_pollset(endpoints.server, pollset);
81 }
82
83 grpc_server_setup_transport(server_->c_server(), transport, nullptr,
84 server_args, nullptr);
85 grpc_chttp2_transport_start_reading(transport, nullptr, nullptr);
86 }
87
88 /* create channel */
89 {
90 ChannelArguments args;
91 args.SetString(GRPC_ARG_DEFAULT_AUTHORITY, "test.authority");
92 ApplyCommonChannelArguments(&args);
93
94 grpc_channel_args c_args = args.c_channel_args();
95 grpc_transport* transport =
96 grpc_create_chttp2_transport(&c_args, endpoints.client, true);
97 GPR_ASSERT(transport);
98 grpc_channel* channel = grpc_channel_create(
99 "target", &c_args, GRPC_CLIENT_DIRECT_CHANNEL, transport);
100 grpc_chttp2_transport_start_reading(transport, nullptr, nullptr);
101
102 channel_ = ::grpc::CreateChannelInternal(
103 "", channel,
104 std::vector<std::unique_ptr<
105 experimental::ClientInterceptorFactoryInterface>>());
106 }
107 }
108
~EndpointPairFixture()109 virtual ~EndpointPairFixture() {
110 server_->Shutdown();
111 cq_->Shutdown();
112 void* tag;
113 bool ok;
114 while (cq_->Next(&tag, &ok)) {
115 }
116 }
117
cq()118 ServerCompletionQueue* cq() { return cq_.get(); }
channel()119 std::shared_ptr<Channel> channel() { return channel_; }
120
121 private:
122 std::unique_ptr<Server> server_;
123 std::unique_ptr<ServerCompletionQueue> cq_;
124 std::shared_ptr<Channel> channel_;
125 };
126
127 class InProcessCHTTP2 : public EndpointPairFixture {
128 public:
InProcessCHTTP2(Service * service,grpc_passthru_endpoint_stats * stats)129 InProcessCHTTP2(Service* service, grpc_passthru_endpoint_stats* stats)
130 : EndpointPairFixture(service, MakeEndpoints(stats)), stats_(stats) {}
131
~InProcessCHTTP2()132 virtual ~InProcessCHTTP2() {
133 if (stats_ != nullptr) {
134 grpc_passthru_endpoint_stats_destroy(stats_);
135 }
136 }
137
writes_performed() const138 int writes_performed() const { return stats_->num_writes; }
139
140 private:
141 grpc_passthru_endpoint_stats* stats_;
142
MakeEndpoints(grpc_passthru_endpoint_stats * stats)143 static grpc_endpoint_pair MakeEndpoints(grpc_passthru_endpoint_stats* stats) {
144 static grpc_resource_quota* rq = grpc_resource_quota_create("bm");
145 grpc_endpoint_pair p;
146 grpc_passthru_endpoint_create(&p.client, &p.server, rq, stats);
147 return p;
148 }
149 };
150
UnaryPingPong(int request_size,int response_size)151 static double UnaryPingPong(int request_size, int response_size) {
152 const int kIterations = 10000;
153
154 EchoTestService::AsyncService service;
155 std::unique_ptr<InProcessCHTTP2> fixture(
156 new InProcessCHTTP2(&service, grpc_passthru_endpoint_stats_create()));
157 EchoRequest send_request;
158 EchoResponse send_response;
159 EchoResponse recv_response;
160 if (request_size > 0) {
161 send_request.set_message(std::string(request_size, 'a'));
162 }
163 if (response_size > 0) {
164 send_response.set_message(std::string(response_size, 'a'));
165 }
166 Status recv_status;
167 struct ServerEnv {
168 ServerContext ctx;
169 EchoRequest recv_request;
170 grpc::ServerAsyncResponseWriter<EchoResponse> response_writer;
171 ServerEnv() : response_writer(&ctx) {}
172 };
173 uint8_t server_env_buffer[2 * sizeof(ServerEnv)];
174 ServerEnv* server_env[2] = {
175 reinterpret_cast<ServerEnv*>(server_env_buffer),
176 reinterpret_cast<ServerEnv*>(server_env_buffer + sizeof(ServerEnv))};
177 new (server_env[0]) ServerEnv;
178 new (server_env[1]) ServerEnv;
179 service.RequestEcho(&server_env[0]->ctx, &server_env[0]->recv_request,
180 &server_env[0]->response_writer, fixture->cq(),
181 fixture->cq(), tag(0));
182 service.RequestEcho(&server_env[1]->ctx, &server_env[1]->recv_request,
183 &server_env[1]->response_writer, fixture->cq(),
184 fixture->cq(), tag(1));
185 std::unique_ptr<EchoTestService::Stub> stub(
186 EchoTestService::NewStub(fixture->channel()));
187 for (int iteration = 0; iteration < kIterations; iteration++) {
188 recv_response.Clear();
189 ClientContext cli_ctx;
190 std::unique_ptr<ClientAsyncResponseReader<EchoResponse>> response_reader(
191 stub->AsyncEcho(&cli_ctx, send_request, fixture->cq()));
192 void* t;
193 bool ok;
194 response_reader->Finish(&recv_response, &recv_status, tag(4));
195 GPR_ASSERT(fixture->cq()->Next(&t, &ok));
196 GPR_ASSERT(ok);
197 GPR_ASSERT(t == tag(0) || t == tag(1));
198 intptr_t slot = reinterpret_cast<intptr_t>(t);
199 ServerEnv* senv = server_env[slot];
200 senv->response_writer.Finish(send_response, Status::OK, tag(3));
201 for (int i = (1 << 3) | (1 << 4); i != 0;) {
202 GPR_ASSERT(fixture->cq()->Next(&t, &ok));
203 GPR_ASSERT(ok);
204 int tagnum = static_cast<int>(reinterpret_cast<intptr_t>(t));
205 GPR_ASSERT(i & (1 << tagnum));
206 i -= 1 << tagnum;
207 }
208 GPR_ASSERT(recv_status.ok());
209
210 senv->~ServerEnv();
211 senv = new (senv) ServerEnv();
212 service.RequestEcho(&senv->ctx, &senv->recv_request, &senv->response_writer,
213 fixture->cq(), fixture->cq(), tag(slot));
214 }
215
216 double writes_per_iteration =
217 static_cast<double>(fixture->writes_performed()) /
218 static_cast<double>(kIterations);
219
220 fixture.reset();
221 server_env[0]->~ServerEnv();
222 server_env[1]->~ServerEnv();
223
224 return writes_per_iteration;
225 }
226
TEST(WritesPerRpcTest,UnaryPingPong)227 TEST(WritesPerRpcTest, UnaryPingPong) {
228 EXPECT_LT(UnaryPingPong(0, 0), 2.05);
229 EXPECT_LT(UnaryPingPong(1, 0), 2.05);
230 EXPECT_LT(UnaryPingPong(0, 1), 2.05);
231 EXPECT_LT(UnaryPingPong(4096, 0), 2.5);
232 EXPECT_LT(UnaryPingPong(0, 4096), 2.5);
233 }
234
235 } // namespace testing
236 } // namespace grpc
237
main(int argc,char ** argv)238 int main(int argc, char** argv) {
239 ::testing::InitGoogleTest(&argc, argv);
240 grpc::testing::TestEnvironment env(argc, argv);
241 grpc_init();
242 int ret = RUN_ALL_TESTS();
243 grpc_shutdown();
244 return ret;
245 }
246