mirror of https://github.com/grpc/grpc.git
Yash Tibrewal
4 years ago
parent
9f4c2e4a75
commit
cdda406115
20 changed files with 302 additions and 1 deletions
@ -0,0 +1,103 @@ |
|||||||
|
//
|
||||||
|
//
|
||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
//
|
||||||
|
//
|
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "absl/container/inlined_vector.h" |
||||||
|
|
||||||
|
#include "src/core/ext/xds/certificate_provider_registry.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
namespace { |
||||||
|
|
||||||
|
class RegistryState { |
||||||
|
public: |
||||||
|
void RegisterCertificateProviderFactory( |
||||||
|
std::unique_ptr<CertificateProviderFactory> factory) { |
||||||
|
gpr_log(GPR_DEBUG, "registering certificate provider factory for \"%s\"", |
||||||
|
factory->name()); |
||||||
|
for (size_t i = 0; i < factories_.size(); ++i) { |
||||||
|
GPR_ASSERT(strcmp(factories_[i]->name(), factory->name()) != 0); |
||||||
|
} |
||||||
|
factories_.push_back(std::move(factory)); |
||||||
|
} |
||||||
|
|
||||||
|
CertificateProviderFactory* LookupCertificateProviderFactory( |
||||||
|
absl::string_view name) const { |
||||||
|
for (size_t i = 0; i < factories_.size(); ++i) { |
||||||
|
if (name == factories_[i]->name()) { |
||||||
|
return factories_[i].get(); |
||||||
|
} |
||||||
|
} |
||||||
|
return nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
// We currently support 3 factories without doing additional
|
||||||
|
// allocation. This number could be raised if there is a case where
|
||||||
|
// more factories are needed and the additional allocations are
|
||||||
|
// hurting performance (which is unlikely, since these allocations
|
||||||
|
// only occur at gRPC initialization time).
|
||||||
|
absl::InlinedVector<std::unique_ptr<CertificateProviderFactory>, 3> |
||||||
|
factories_; |
||||||
|
}; |
||||||
|
|
||||||
|
static RegistryState* g_state = nullptr; |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
//
|
||||||
|
// CertificateProviderRegistry
|
||||||
|
//
|
||||||
|
|
||||||
|
CertificateProviderFactory* |
||||||
|
CertificateProviderRegistry::LookupCertificateProviderFactory( |
||||||
|
absl::string_view name) { |
||||||
|
GPR_ASSERT(g_state != nullptr); |
||||||
|
return g_state->LookupCertificateProviderFactory(name); |
||||||
|
} |
||||||
|
|
||||||
|
void CertificateProviderRegistry::InitRegistry() { |
||||||
|
if (g_state == nullptr) g_state = new RegistryState(); |
||||||
|
} |
||||||
|
|
||||||
|
void CertificateProviderRegistry::ShutdownRegistry() { |
||||||
|
delete g_state; |
||||||
|
g_state = nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
void CertificateProviderRegistry::RegisterCertificateProviderFactory( |
||||||
|
std::unique_ptr<CertificateProviderFactory> factory) { |
||||||
|
InitRegistry(); |
||||||
|
g_state->RegisterCertificateProviderFactory(std::move(factory)); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
//
|
||||||
|
// Plugin registration
|
||||||
|
//
|
||||||
|
|
||||||
|
void grpc_certificate_provider_registry_init() { |
||||||
|
grpc_core::CertificateProviderRegistry::InitRegistry(); |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_certificate_provider_registry_shutdown() { |
||||||
|
grpc_core::CertificateProviderRegistry::ShutdownRegistry(); |
||||||
|
} |
||||||
@ -0,0 +1,90 @@ |
|||||||
|
//
|
||||||
|
//
|
||||||
|
// Copyright 2020 gRPC authors.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
//
|
||||||
|
//
|
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include <gmock/gmock.h> |
||||||
|
|
||||||
|
#include "src/core/ext/xds/certificate_provider_registry.h" |
||||||
|
|
||||||
|
#include "test/core/util/test_config.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
namespace testing { |
||||||
|
namespace { |
||||||
|
|
||||||
|
class FakeCertificateProviderFactory1 : public CertificateProviderFactory { |
||||||
|
public: |
||||||
|
const char* name() const override { return "fake1"; } |
||||||
|
|
||||||
|
std::unique_ptr<Config> CreateCertificateProviderConfig( |
||||||
|
const Json& config_json, grpc_error** error) override { |
||||||
|
return nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
RefCountedPtr<grpc_tls_certificate_provider> CreateCertificateProvider( |
||||||
|
std::unique_ptr<Config> config) override { |
||||||
|
return nullptr; |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
class FakeCertificateProviderFactory2 : public CertificateProviderFactory { |
||||||
|
public: |
||||||
|
const char* name() const override { return "fake2"; } |
||||||
|
|
||||||
|
std::unique_ptr<Config> CreateCertificateProviderConfig( |
||||||
|
const Json& config_json, grpc_error** error) override { |
||||||
|
return nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
RefCountedPtr<grpc_tls_certificate_provider> CreateCertificateProvider( |
||||||
|
std::unique_ptr<Config> config) override { |
||||||
|
return nullptr; |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
TEST(CertificateProviderRegistryTest, Basic) { |
||||||
|
CertificateProviderRegistry::InitRegistry(); |
||||||
|
auto* fake_factory_1 = new FakeCertificateProviderFactory1; |
||||||
|
auto* fake_factory_2 = new FakeCertificateProviderFactory2; |
||||||
|
CertificateProviderRegistry::RegisterCertificateProviderFactory( |
||||||
|
std::unique_ptr<CertificateProviderFactory>(fake_factory_1)); |
||||||
|
CertificateProviderRegistry::RegisterCertificateProviderFactory( |
||||||
|
std::unique_ptr<CertificateProviderFactory>(fake_factory_2)); |
||||||
|
EXPECT_EQ( |
||||||
|
CertificateProviderRegistry::LookupCertificateProviderFactory("fake1"), |
||||||
|
fake_factory_1); |
||||||
|
EXPECT_EQ( |
||||||
|
CertificateProviderRegistry::LookupCertificateProviderFactory("fake2"), |
||||||
|
fake_factory_2); |
||||||
|
EXPECT_EQ( |
||||||
|
CertificateProviderRegistry::LookupCertificateProviderFactory("fake3"), |
||||||
|
nullptr); |
||||||
|
CertificateProviderRegistry::ShutdownRegistry(); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
::testing::InitGoogleTest(&argc, argv); |
||||||
|
grpc::testing::TestEnvironment env(argc, argv); |
||||||
|
auto result = RUN_ALL_TESTS(); |
||||||
|
return result; |
||||||
|
} |
||||||
Loading…
Reference in new issue