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 <grpc/credentials.h>
22 #include <grpc/grpc_security.h>
23 #include <grpc/status.h>
24 #include <string.h>
25
26 #include <utility>
27
28 #include "absl/log/check.h"
29
process_auth_failure(void * state,grpc_auth_context *,const grpc_metadata *,size_t,grpc_process_auth_metadata_done_cb cb,void * user_data)30 static void process_auth_failure(void* state, grpc_auth_context* /*ctx*/,
31 const grpc_metadata* /*md*/,
32 size_t /*md_count*/,
33 grpc_process_auth_metadata_done_cb cb,
34 void* user_data) {
35 CHECK_EQ(state, nullptr);
36 cb(user_data, nullptr, 0, nullptr, 0, GRPC_STATUS_UNAUTHENTICATED, nullptr);
37 }
38
LocalTestFixture(std::string localaddr,grpc_local_connect_type type)39 LocalTestFixture::LocalTestFixture(std::string localaddr,
40 grpc_local_connect_type type)
41 : localaddr_(std::move(localaddr)), type_(type) {}
42
MakeServer(const grpc_core::ChannelArgs & args,grpc_completion_queue * cq,absl::AnyInvocable<void (grpc_server *)> & pre_server_start)43 grpc_server* LocalTestFixture::MakeServer(
44 const grpc_core::ChannelArgs& args, grpc_completion_queue* cq,
45 absl::AnyInvocable<void(grpc_server*)>& pre_server_start) {
46 grpc_server_credentials* server_creds =
47 grpc_local_server_credentials_create(type_);
48 auto* server = grpc_server_create(args.ToC().get(), nullptr);
49 grpc_server_register_completion_queue(server, cq, nullptr);
50 if (args.Contains(FAIL_AUTH_CHECK_SERVER_ARG_NAME)) {
51 grpc_auth_metadata_processor processor = {process_auth_failure, nullptr,
52 nullptr};
53 grpc_server_credentials_set_auth_metadata_processor(server_creds,
54 processor);
55 }
56 CHECK(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 CHECK_NE(client, nullptr);
69 grpc_channel_credentials_release(creds);
70 return client;
71 }
72