1 /* 2 * 3 * Copyright 2016 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 #ifndef GRPC_TEST_CPP_UTIL_TEST_CREDENTIALS_PROVIDER_H 20 #define GRPC_TEST_CPP_UTIL_TEST_CREDENTIALS_PROVIDER_H 21 22 #include <memory> 23 24 #include <grpcpp/security/credentials.h> 25 #include <grpcpp/security/server_credentials.h> 26 #include <grpcpp/support/channel_arguments.h> 27 28 namespace grpc { 29 namespace testing { 30 31 const char kInsecureCredentialsType[] = "INSECURE_CREDENTIALS"; 32 // For real credentials, like tls/ssl, this name should match the AuthContext 33 // property "transport_security_type". 34 const char kTlsCredentialsType[] = "ssl"; 35 const char kAltsCredentialsType[] = "alts"; 36 const char kGoogleDefaultCredentialsType[] = "google_default_credentials"; 37 38 // Provide test credentials of a particular type. 39 class CredentialTypeProvider { 40 public: ~CredentialTypeProvider()41 virtual ~CredentialTypeProvider() {} 42 43 virtual std::shared_ptr<ChannelCredentials> GetChannelCredentials( 44 ChannelArguments* args) = 0; 45 virtual std::shared_ptr<ServerCredentials> GetServerCredentials() = 0; 46 }; 47 48 // Provide test credentials. Thread-safe. 49 class CredentialsProvider { 50 public: ~CredentialsProvider()51 virtual ~CredentialsProvider() {} 52 53 // Add a secure type in addition to the defaults. The default provider has 54 // (kInsecureCredentialsType, kTlsCredentialsType). 55 virtual void AddSecureType( 56 const std::string& type, 57 std::unique_ptr<CredentialTypeProvider> type_provider) = 0; 58 59 // Provide channel credentials according to the given type. Alter the channel 60 // arguments if needed. Return nullptr if type is not registered. 61 virtual std::shared_ptr<ChannelCredentials> GetChannelCredentials( 62 const std::string& type, ChannelArguments* args) = 0; 63 64 // Provide server credentials according to the given type. 65 // Return nullptr if type is not registered. 66 virtual std::shared_ptr<ServerCredentials> GetServerCredentials( 67 const std::string& type) = 0; 68 69 // Provide a list of secure credentials type. 70 virtual std::vector<std::string> GetSecureCredentialsTypeList() = 0; 71 }; 72 73 // Get the current provider. Create a default one if not set. 74 // Not thread-safe. 75 CredentialsProvider* GetCredentialsProvider(); 76 77 // Set the global provider. Takes ownership. The previous set provider will be 78 // destroyed. 79 // Not thread-safe. 80 void SetCredentialsProvider(CredentialsProvider* provider); 81 82 } // namespace testing 83 } // namespace grpc 84 85 #endif // GRPC_TEST_CPP_UTIL_TEST_CREDENTIALS_PROVIDER_H 86