1 //
2 //
3 // Copyright 2018 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 "test/core/end2end/fixtures/local_util.h"
20
21 #include <string.h>
22
23 #include <utility>
24
25 #include <grpc/grpc_security.h>
26 #include <grpc/status.h>
27 #include <grpc/support/log.h>
28
process_auth_failure(void * state,grpc_auth_context *,const grpc_metadata *,size_t,grpc_process_auth_metadata_done_cb cb,void * user_data)29 static void process_auth_failure(void* state, grpc_auth_context* /*ctx*/,
30 const grpc_metadata* /*md*/,
31 size_t /*md_count*/,
32 grpc_process_auth_metadata_done_cb cb,
33 void* user_data) {
34 GPR_ASSERT(state == nullptr);
35 cb(user_data, nullptr, 0, nullptr, 0, GRPC_STATUS_UNAUTHENTICATED, nullptr);
36 }
37
LocalTestFixture(std::string localaddr,grpc_local_connect_type type)38 LocalTestFixture::LocalTestFixture(std::string localaddr,
39 grpc_local_connect_type type)
40 : localaddr_(std::move(localaddr)), type_(type) {}
41
MakeServer(const grpc_core::ChannelArgs & args,grpc_completion_queue * cq,absl::AnyInvocable<void (grpc_server *)> & pre_server_start)42 grpc_server* LocalTestFixture::MakeServer(
43 const grpc_core::ChannelArgs& args, grpc_completion_queue* cq,
44 absl::AnyInvocable<void(grpc_server*)>& pre_server_start) {
45 grpc_server_credentials* server_creds =
46 grpc_local_server_credentials_create(type_);
47 auto* server = grpc_server_create(args.ToC().get(), nullptr);
48 grpc_server_register_completion_queue(server, cq, nullptr);
49 if (args.Contains(FAIL_AUTH_CHECK_SERVER_ARG_NAME)) {
50 grpc_auth_metadata_processor processor = {process_auth_failure, nullptr,
51 nullptr};
52 grpc_server_credentials_set_auth_metadata_processor(server_creds,
53 processor);
54 }
55 GPR_ASSERT(
56 grpc_server_add_http2_port(server, localaddr_.c_str(), server_creds));
57 grpc_server_credentials_release(server_creds);
58 pre_server_start(server);
59 grpc_server_start(server);
60 return server;
61 }
62
MakeClient(const grpc_core::ChannelArgs & args,grpc_completion_queue *)63 grpc_channel* LocalTestFixture::MakeClient(const grpc_core::ChannelArgs& args,
64 grpc_completion_queue*) {
65 grpc_channel_credentials* creds = grpc_local_credentials_create(type_);
66 auto* client =
67 grpc_channel_create(localaddr_.c_str(), creds, args.ToC().get());
68 GPR_ASSERT(client != nullptr);
69 grpc_channel_credentials_release(creds);
70 return client;
71 }
72