1 // Copyright 2023 gRPC authors. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 #ifndef GRPC_TEST_CORE_END2END_FIXTURES_SECURE_FIXTURE_H 16 #define GRPC_TEST_CORE_END2END_FIXTURES_SECURE_FIXTURE_H 17 18 #include <string> 19 #include <utility> 20 21 #include "absl/functional/any_invocable.h" 22 23 #include <grpc/grpc.h> 24 #include <grpc/grpc_security.h> 25 #include <grpc/support/log.h> 26 27 #include "src/core/lib/channel/channel_args.h" 28 #include "src/core/lib/gprpp/host_port.h" 29 #include "test/core/end2end/end2end_tests.h" 30 #include "test/core/util/port.h" 31 32 // Base class for a fixture that just needs to select cred types (or mutate 33 // client/server channel args). 34 class SecureFixture : public grpc_core::CoreTestFixture { 35 public: 36 explicit SecureFixture(std::string localaddr = grpc_core::JoinHostPort( 37 "localhost", grpc_pick_unused_port_or_die())) localaddr_(std::move (localaddr))38 : localaddr_(std::move(localaddr)) {} 39 40 protected: localaddr()41 const std::string& localaddr() const { return localaddr_; } 42 43 private: 44 virtual grpc_channel_credentials* MakeClientCreds( 45 const grpc_core::ChannelArgs& args) = 0; 46 virtual grpc_server_credentials* MakeServerCreds( 47 const grpc_core::ChannelArgs& args) = 0; MutateClientArgs(grpc_core::ChannelArgs args)48 virtual grpc_core::ChannelArgs MutateClientArgs(grpc_core::ChannelArgs args) { 49 return args; 50 } MutateServerArgs(grpc_core::ChannelArgs args)51 virtual grpc_core::ChannelArgs MutateServerArgs(grpc_core::ChannelArgs args) { 52 return args; 53 } MakeServer(const grpc_core::ChannelArgs & in_args,grpc_completion_queue * cq,absl::AnyInvocable<void (grpc_server *)> & pre_server_start)54 grpc_server* MakeServer( 55 const grpc_core::ChannelArgs& in_args, grpc_completion_queue* cq, 56 absl::AnyInvocable<void(grpc_server*)>& pre_server_start) override { 57 auto args = MutateServerArgs(in_args); 58 auto* creds = MakeServerCreds(args); 59 auto* server = grpc_server_create(args.ToC().get(), nullptr); 60 grpc_server_register_completion_queue(server, cq, nullptr); 61 GPR_ASSERT(grpc_server_add_http2_port(server, localaddr_.c_str(), creds)); 62 grpc_server_credentials_release(creds); 63 pre_server_start(server); 64 grpc_server_start(server); 65 return server; 66 } MakeClient(const grpc_core::ChannelArgs & in_args,grpc_completion_queue *)67 grpc_channel* MakeClient(const grpc_core::ChannelArgs& in_args, 68 grpc_completion_queue*) override { 69 auto args = MutateClientArgs(in_args); 70 auto* creds = MakeClientCreds(args); 71 auto* client = 72 grpc_channel_create(localaddr_.c_str(), creds, args.ToC().get()); 73 GPR_ASSERT(client != nullptr); 74 grpc_channel_credentials_release(creds); 75 return client; 76 } 77 78 std::string localaddr_; 79 }; 80 81 // Fixture that uses insecure credentials. 82 class InsecureFixture : public SecureFixture { 83 public: 84 using SecureFixture::SecureFixture; 85 86 private: MakeClientCreds(const grpc_core::ChannelArgs &)87 grpc_channel_credentials* MakeClientCreds( 88 const grpc_core::ChannelArgs&) override { 89 return grpc_insecure_credentials_create(); 90 } MakeServerCreds(const grpc_core::ChannelArgs &)91 grpc_server_credentials* MakeServerCreds( 92 const grpc_core::ChannelArgs&) override { 93 return grpc_insecure_server_credentials_create(); 94 } 95 }; 96 97 #endif // GRPC_TEST_CORE_END2END_FIXTURES_SECURE_FIXTURE_H 98