| /* |
| * |
| * Copyright 2015 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 <gmock/gmock.h> |
| #include <grpc/grpc.h> |
| #include <grpc/grpc_security.h> |
| #include <grpcpp/security/credentials.h> |
| #include <grpcpp/security/server_credentials.h> |
| #include <grpcpp/security/tls_credentials_options.h> |
| #include <grpcpp/server_builder.h> |
| #include <gtest/gtest.h> |
| |
| #include <memory> |
| |
| #include "src/core/lib/gpr/env.h" |
| #include "src/core/lib/gpr/tmpfile.h" |
| #include "src/cpp/client/secure_credentials.h" |
| #include "src/cpp/common/tls_credentials_options_util.h" |
| |
| #define CA_CERT_PATH "src/core/tsi/test_creds/ca.pem" |
| #define SERVER_CERT_PATH "src/core/tsi/test_creds/server1.pem" |
| #define SERVER_KEY_PATH "src/core/tsi/test_creds/server1.key" |
| |
| namespace { |
| |
| constexpr const char* kRootCertName = "root_cert_name"; |
| constexpr const char* kRootCertContents = "root_cert_contents"; |
| constexpr const char* kIdentityCertName = "identity_cert_name"; |
| constexpr const char* kIdentityCertPrivateKey = "identity_private_key"; |
| constexpr const char* kIdentityCertContents = "identity_cert_contents"; |
| |
| using ::grpc::experimental::FileWatcherCertificateProvider; |
| using ::grpc::experimental::StaticDataCertificateProvider; |
| using ::grpc::experimental::TlsServerAuthorizationCheckArg; |
| using ::grpc::experimental::TlsServerAuthorizationCheckConfig; |
| using ::grpc::experimental::TlsServerAuthorizationCheckInterface; |
| |
| static void tls_server_authorization_check_callback( |
| grpc_tls_server_authorization_check_arg* arg) { |
| GPR_ASSERT(arg != nullptr); |
| std::string cb_user_data = "cb_user_data"; |
| arg->cb_user_data = static_cast<void*>(gpr_strdup(cb_user_data.c_str())); |
| arg->success = 1; |
| arg->target_name = gpr_strdup("callback_target_name"); |
| arg->peer_cert = gpr_strdup("callback_peer_cert"); |
| arg->status = GRPC_STATUS_OK; |
| arg->error_details->set_error_details("callback_error_details"); |
| } |
| |
| class TestTlsServerAuthorizationCheck |
| : public TlsServerAuthorizationCheckInterface { |
| int Schedule(TlsServerAuthorizationCheckArg* arg) override { |
| GPR_ASSERT(arg != nullptr); |
| std::string cb_user_data = "cb_user_data"; |
| arg->set_cb_user_data(static_cast<void*>(gpr_strdup(cb_user_data.c_str()))); |
| arg->set_success(1); |
| arg->set_target_name("sync_target_name"); |
| arg->set_peer_cert("sync_peer_cert"); |
| arg->set_status(GRPC_STATUS_OK); |
| arg->set_error_details("sync_error_details"); |
| return 1; |
| } |
| |
| void Cancel(TlsServerAuthorizationCheckArg* arg) override { |
| GPR_ASSERT(arg != nullptr); |
| arg->set_status(GRPC_STATUS_PERMISSION_DENIED); |
| arg->set_error_details("cancelled"); |
| } |
| }; |
| } // namespace |
| |
| namespace grpc { |
| namespace testing { |
| namespace { |
| |
| TEST(CredentialsTest, InvalidGoogleRefreshToken) { |
| std::shared_ptr<CallCredentials> bad1 = GoogleRefreshTokenCredentials(""); |
| EXPECT_EQ(static_cast<CallCredentials*>(nullptr), bad1.get()); |
| } |
| |
| TEST(CredentialsTest, DefaultCredentials) { |
| auto creds = GoogleDefaultCredentials(); |
| } |
| |
| TEST(CredentialsTest, StsCredentialsOptionsCppToCore) { |
| grpc::experimental::StsCredentialsOptions options; |
| options.token_exchange_service_uri = "https://foo.com/exchange"; |
| options.resource = "resource"; |
| options.audience = "audience"; |
| options.scope = "scope"; |
| // options.requested_token_type explicitly not set. |
| options.subject_token_path = "/foo/bar"; |
| options.subject_token_type = "nice_token_type"; |
| options.actor_token_path = "/foo/baz"; |
| options.actor_token_type = "even_nicer_token_type"; |
| grpc_sts_credentials_options core_opts = |
| grpc::experimental::StsCredentialsCppToCoreOptions(options); |
| EXPECT_EQ(options.token_exchange_service_uri, |
| core_opts.token_exchange_service_uri); |
| EXPECT_EQ(options.resource, core_opts.resource); |
| EXPECT_EQ(options.audience, core_opts.audience); |
| EXPECT_EQ(options.scope, core_opts.scope); |
| EXPECT_EQ(options.requested_token_type, core_opts.requested_token_type); |
| EXPECT_EQ(options.subject_token_path, core_opts.subject_token_path); |
| EXPECT_EQ(options.subject_token_type, core_opts.subject_token_type); |
| EXPECT_EQ(options.actor_token_path, core_opts.actor_token_path); |
| EXPECT_EQ(options.actor_token_type, core_opts.actor_token_type); |
| } |
| |
| TEST(CredentialsTest, StsCredentialsOptionsJson) { |
| const char valid_json[] = R"( |
| { |
| "token_exchange_service_uri": "https://foo/exchange", |
| "resource": "resource", |
| "audience": "audience", |
| "scope": "scope", |
| "requested_token_type": "requested_token_type", |
| "subject_token_path": "subject_token_path", |
| "subject_token_type": "subject_token_type", |
| "actor_token_path": "actor_token_path", |
| "actor_token_type": "actor_token_type" |
| })"; |
| grpc::experimental::StsCredentialsOptions options; |
| EXPECT_TRUE( |
| grpc::experimental::StsCredentialsOptionsFromJson(valid_json, &options) |
| .ok()); |
| EXPECT_EQ(options.token_exchange_service_uri, "https://foo/exchange"); |
| EXPECT_EQ(options.resource, "resource"); |
| EXPECT_EQ(options.audience, "audience"); |
| EXPECT_EQ(options.scope, "scope"); |
| EXPECT_EQ(options.requested_token_type, "requested_token_type"); |
| EXPECT_EQ(options.subject_token_path, "subject_token_path"); |
| EXPECT_EQ(options.subject_token_type, "subject_token_type"); |
| EXPECT_EQ(options.actor_token_path, "actor_token_path"); |
| EXPECT_EQ(options.actor_token_type, "actor_token_type"); |
| |
| const char minimum_valid_json[] = R"( |
| { |
| "token_exchange_service_uri": "https://foo/exchange", |
| "subject_token_path": "subject_token_path", |
| "subject_token_type": "subject_token_type" |
| })"; |
| EXPECT_TRUE(grpc::experimental::StsCredentialsOptionsFromJson( |
| minimum_valid_json, &options) |
| .ok()); |
| EXPECT_EQ(options.token_exchange_service_uri, "https://foo/exchange"); |
| EXPECT_EQ(options.resource, ""); |
| EXPECT_EQ(options.audience, ""); |
| EXPECT_EQ(options.scope, ""); |
| EXPECT_EQ(options.requested_token_type, ""); |
| EXPECT_EQ(options.subject_token_path, "subject_token_path"); |
| EXPECT_EQ(options.subject_token_type, "subject_token_type"); |
| EXPECT_EQ(options.actor_token_path, ""); |
| EXPECT_EQ(options.actor_token_type, ""); |
| |
| const char invalid_json[] = R"( |
| I'm not a valid JSON. |
| )"; |
| EXPECT_EQ( |
| grpc::StatusCode::INVALID_ARGUMENT, |
| grpc::experimental::StsCredentialsOptionsFromJson(invalid_json, &options) |
| .error_code()); |
| |
| const char invalid_json_missing_subject_token_type[] = R"( |
| { |
| "token_exchange_service_uri": "https://foo/exchange", |
| "subject_token_path": "subject_token_path" |
| })"; |
| auto status = grpc::experimental::StsCredentialsOptionsFromJson( |
| invalid_json_missing_subject_token_type, &options); |
| EXPECT_EQ(grpc::StatusCode::INVALID_ARGUMENT, status.error_code()); |
| EXPECT_THAT(status.error_message(), |
| ::testing::HasSubstr("subject_token_type")); |
| |
| const char invalid_json_missing_subject_token_path[] = R"( |
| { |
| "token_exchange_service_uri": "https://foo/exchange", |
| "subject_token_type": "subject_token_type" |
| })"; |
| status = grpc::experimental::StsCredentialsOptionsFromJson( |
| invalid_json_missing_subject_token_path, &options); |
| EXPECT_EQ(grpc::StatusCode::INVALID_ARGUMENT, status.error_code()); |
| EXPECT_THAT(status.error_message(), |
| ::testing::HasSubstr("subject_token_path")); |
| |
| const char invalid_json_missing_token_exchange_uri[] = R"( |
| { |
| "subject_token_path": "subject_token_path", |
| "subject_token_type": "subject_token_type" |
| })"; |
| status = grpc::experimental::StsCredentialsOptionsFromJson( |
| invalid_json_missing_token_exchange_uri, &options); |
| EXPECT_EQ(grpc::StatusCode::INVALID_ARGUMENT, status.error_code()); |
| EXPECT_THAT(status.error_message(), |
| ::testing::HasSubstr("token_exchange_service_uri")); |
| } |
| |
| TEST(CredentialsTest, StsCredentialsOptionsFromEnv) { |
| // Unset env and check expected failure. |
| gpr_unsetenv("STS_CREDENTIALS"); |
| grpc::experimental::StsCredentialsOptions options; |
| auto status = grpc::experimental::StsCredentialsOptionsFromEnv(&options); |
| EXPECT_EQ(grpc::StatusCode::NOT_FOUND, status.error_code()); |
| |
| // Set env and check for success. |
| const char valid_json[] = R"( |
| { |
| "token_exchange_service_uri": "https://foo/exchange", |
| "subject_token_path": "subject_token_path", |
| "subject_token_type": "subject_token_type" |
| })"; |
| char* creds_file_name; |
| FILE* creds_file = gpr_tmpfile("sts_creds_options", &creds_file_name); |
| ASSERT_NE(creds_file_name, nullptr); |
| ASSERT_NE(creds_file, nullptr); |
| ASSERT_EQ(sizeof(valid_json), |
| fwrite(valid_json, 1, sizeof(valid_json), creds_file)); |
| fclose(creds_file); |
| gpr_setenv("STS_CREDENTIALS", creds_file_name); |
| gpr_free(creds_file_name); |
| status = grpc::experimental::StsCredentialsOptionsFromEnv(&options); |
| EXPECT_TRUE(status.ok()); |
| EXPECT_EQ(options.token_exchange_service_uri, "https://foo/exchange"); |
| EXPECT_EQ(options.resource, ""); |
| EXPECT_EQ(options.audience, ""); |
| EXPECT_EQ(options.scope, ""); |
| EXPECT_EQ(options.requested_token_type, ""); |
| EXPECT_EQ(options.subject_token_path, "subject_token_path"); |
| EXPECT_EQ(options.subject_token_type, "subject_token_type"); |
| EXPECT_EQ(options.actor_token_path, ""); |
| EXPECT_EQ(options.actor_token_type, ""); |
| |
| // Cleanup. |
| gpr_unsetenv("STS_CREDENTIALS"); |
| } |
| |
| TEST(CredentialsTest, TlsServerAuthorizationCheckArgCallback) { |
| grpc_tls_server_authorization_check_arg* c_arg = |
| new grpc_tls_server_authorization_check_arg; |
| c_arg->cb = tls_server_authorization_check_callback; |
| c_arg->context = nullptr; |
| c_arg->error_details = new grpc_tls_error_details(); |
| TlsServerAuthorizationCheckArg* arg = |
| new TlsServerAuthorizationCheckArg(c_arg); |
| arg->set_cb_user_data(nullptr); |
| arg->set_success(0); |
| arg->set_target_name("target_name"); |
| arg->set_peer_cert("peer_cert"); |
| arg->set_status(GRPC_STATUS_UNAUTHENTICATED); |
| arg->set_error_details("error_details"); |
| const char* target_name_before_callback = c_arg->target_name; |
| const char* peer_cert_before_callback = c_arg->peer_cert; |
| |
| arg->OnServerAuthorizationCheckDoneCallback(); |
| EXPECT_STREQ(static_cast<char*>(arg->cb_user_data()), "cb_user_data"); |
| gpr_free(arg->cb_user_data()); |
| EXPECT_EQ(arg->success(), 1); |
| EXPECT_STREQ(arg->target_name().c_str(), "callback_target_name"); |
| EXPECT_STREQ(arg->peer_cert().c_str(), "callback_peer_cert"); |
| EXPECT_EQ(arg->status(), GRPC_STATUS_OK); |
| EXPECT_STREQ(arg->error_details().c_str(), "callback_error_details"); |
| |
| // Cleanup. |
| gpr_free(const_cast<char*>(target_name_before_callback)); |
| gpr_free(const_cast<char*>(peer_cert_before_callback)); |
| gpr_free(const_cast<char*>(c_arg->target_name)); |
| gpr_free(const_cast<char*>(c_arg->peer_cert)); |
| delete c_arg->error_details; |
| delete arg; |
| delete c_arg; |
| } |
| |
| TEST(CredentialsTest, TlsServerAuthorizationCheckConfigSchedule) { |
| std::shared_ptr<TestTlsServerAuthorizationCheck> |
| test_server_authorization_check(new TestTlsServerAuthorizationCheck()); |
| TlsServerAuthorizationCheckConfig config(test_server_authorization_check); |
| grpc_tls_server_authorization_check_arg* c_arg = |
| new grpc_tls_server_authorization_check_arg(); |
| c_arg->error_details = new grpc_tls_error_details(); |
| c_arg->context = nullptr; |
| TlsServerAuthorizationCheckArg* arg = |
| new TlsServerAuthorizationCheckArg(c_arg); |
| arg->set_cb_user_data(nullptr); |
| arg->set_success(0); |
| arg->set_target_name("target_name"); |
| arg->set_peer_cert("peer_cert"); |
| arg->set_status(GRPC_STATUS_PERMISSION_DENIED); |
| arg->set_error_details("error_details"); |
| const char* target_name_before_schedule = c_arg->target_name; |
| const char* peer_cert_before_schedule = c_arg->peer_cert; |
| |
| int schedule_output = config.Schedule(arg); |
| EXPECT_EQ(schedule_output, 1); |
| EXPECT_STREQ(static_cast<char*>(arg->cb_user_data()), "cb_user_data"); |
| EXPECT_EQ(arg->success(), 1); |
| EXPECT_STREQ(arg->target_name().c_str(), "sync_target_name"); |
| EXPECT_STREQ(arg->peer_cert().c_str(), "sync_peer_cert"); |
| EXPECT_EQ(arg->status(), GRPC_STATUS_OK); |
| EXPECT_STREQ(arg->error_details().c_str(), "sync_error_details"); |
| |
| // Cleanup. |
| gpr_free(arg->cb_user_data()); |
| gpr_free(const_cast<char*>(target_name_before_schedule)); |
| gpr_free(const_cast<char*>(peer_cert_before_schedule)); |
| gpr_free(const_cast<char*>(c_arg->target_name)); |
| gpr_free(const_cast<char*>(c_arg->peer_cert)); |
| delete c_arg->error_details; |
| if (c_arg->destroy_context != nullptr) { |
| c_arg->destroy_context(c_arg->context); |
| } |
| delete c_arg; |
| } |
| |
| TEST(CredentialsTest, TlsServerAuthorizationCheckConfigCppToC) { |
| std::shared_ptr<TestTlsServerAuthorizationCheck> |
| test_server_authorization_check(new TestTlsServerAuthorizationCheck()); |
| TlsServerAuthorizationCheckConfig config(test_server_authorization_check); |
| grpc_tls_server_authorization_check_arg c_arg; |
| c_arg.cb = tls_server_authorization_check_callback; |
| c_arg.cb_user_data = nullptr; |
| c_arg.success = 0; |
| c_arg.target_name = "target_name"; |
| c_arg.peer_cert = "peer_cert"; |
| c_arg.status = GRPC_STATUS_UNAUTHENTICATED; |
| c_arg.error_details = new grpc_tls_error_details(); |
| c_arg.error_details->set_error_details("error_details"); |
| c_arg.config = config.c_config(); |
| c_arg.context = nullptr; |
| int c_schedule_output = (c_arg.config)->Schedule(&c_arg); |
| EXPECT_EQ(c_schedule_output, 1); |
| EXPECT_STREQ(static_cast<char*>(c_arg.cb_user_data), "cb_user_data"); |
| EXPECT_EQ(c_arg.success, 1); |
| EXPECT_STREQ(c_arg.target_name, "sync_target_name"); |
| EXPECT_STREQ(c_arg.peer_cert, "sync_peer_cert"); |
| EXPECT_EQ(c_arg.status, GRPC_STATUS_OK); |
| EXPECT_STREQ(c_arg.error_details->error_details().c_str(), |
| "sync_error_details"); |
| |
| // Cleanup. |
| gpr_free(c_arg.cb_user_data); |
| c_arg.destroy_context(c_arg.context); |
| delete c_arg.error_details; |
| gpr_free(const_cast<char*>(c_arg.target_name)); |
| gpr_free(const_cast<char*>(c_arg.peer_cert)); |
| } |
| |
| TEST( |
| CredentialsTest, |
| TlsChannelCredentialsWithStaticDataCertificateProviderLoadingRootAndIdentity) { |
| experimental::IdentityKeyCertPair key_cert_pair; |
| key_cert_pair.private_key = kIdentityCertPrivateKey; |
| key_cert_pair.certificate_chain = kIdentityCertContents; |
| std::vector<experimental::IdentityKeyCertPair> identity_key_cert_pairs; |
| identity_key_cert_pairs.emplace_back(key_cert_pair); |
| auto certificate_provider = std::make_shared<StaticDataCertificateProvider>( |
| kRootCertContents, identity_key_cert_pairs); |
| auto test_server_authorization_check = |
| std::make_shared<TestTlsServerAuthorizationCheck>(); |
| auto server_authorization_check_config = |
| std::make_shared<TlsServerAuthorizationCheckConfig>( |
| test_server_authorization_check); |
| grpc::experimental::TlsChannelCredentialsOptions options( |
| certificate_provider); |
| options.watch_root_certs(); |
| options.set_root_cert_name(kRootCertName); |
| options.watch_identity_key_cert_pairs(); |
| options.set_identity_cert_name(kIdentityCertName); |
| options.set_server_verification_option(GRPC_TLS_SERVER_VERIFICATION); |
| options.set_server_authorization_check_config( |
| server_authorization_check_config); |
| auto channel_credentials = grpc::experimental::TlsCredentials(options); |
| GPR_ASSERT(channel_credentials.get() != nullptr); |
| } |
| |
| // ChannelCredentials should always have root credential presented. |
| // Otherwise the system root certificates will be loaded, which will cause |
| // failure in some tests under MacOS/Windows. |
| TEST(CredentialsTest, |
| TlsChannelCredentialsWithStaticDataCertificateProviderLoadingRootOnly) { |
| auto certificate_provider = |
| std::make_shared<StaticDataCertificateProvider>(kRootCertContents); |
| auto test_server_authorization_check = |
| std::make_shared<TestTlsServerAuthorizationCheck>(); |
| auto server_authorization_check_config = |
| std::make_shared<TlsServerAuthorizationCheckConfig>( |
| test_server_authorization_check); |
| GPR_ASSERT(certificate_provider != nullptr); |
| GPR_ASSERT(certificate_provider->c_provider() != nullptr); |
| grpc::experimental::TlsChannelCredentialsOptions options( |
| certificate_provider); |
| options.watch_root_certs(); |
| options.set_root_cert_name(kRootCertName); |
| options.set_server_verification_option(GRPC_TLS_SERVER_VERIFICATION); |
| options.set_server_authorization_check_config( |
| server_authorization_check_config); |
| auto channel_credentials = grpc::experimental::TlsCredentials(options); |
| GPR_ASSERT(channel_credentials.get() != nullptr); |
| } |
| |
| TEST( |
| CredentialsTest, |
| TlsChannelCredentialsWithFileWatcherCertificateProviderLoadingRootAndIdentity) { |
| auto certificate_provider = std::make_shared<FileWatcherCertificateProvider>( |
| SERVER_KEY_PATH, SERVER_CERT_PATH, CA_CERT_PATH, 1); |
| grpc::experimental::TlsChannelCredentialsOptions options( |
| certificate_provider); |
| options.watch_root_certs(); |
| options.set_root_cert_name(kRootCertName); |
| options.watch_identity_key_cert_pairs(); |
| options.set_identity_cert_name(kIdentityCertName); |
| options.set_server_verification_option(GRPC_TLS_SERVER_VERIFICATION); |
| auto test_server_authorization_check = |
| std::make_shared<TestTlsServerAuthorizationCheck>(); |
| auto server_authorization_check_config = |
| std::make_shared<TlsServerAuthorizationCheckConfig>( |
| test_server_authorization_check); |
| options.set_server_authorization_check_config( |
| server_authorization_check_config); |
| auto channel_credentials = grpc::experimental::TlsCredentials(options); |
| GPR_ASSERT(channel_credentials.get() != nullptr); |
| } |
| |
| // ChannelCredentials should always have root credential presented. |
| // Otherwise the system root certificates will be loaded, which will cause |
| // failure in some tests under MacOS/Windows. |
| TEST(CredentialsTest, |
| TlsChannelCredentialsWithFileWatcherCertificateProviderLoadingRootOnly) { |
| auto certificate_provider = |
| std::make_shared<FileWatcherCertificateProvider>(CA_CERT_PATH, 1); |
| grpc::experimental::TlsChannelCredentialsOptions options( |
| certificate_provider); |
| options.watch_root_certs(); |
| options.set_root_cert_name(kRootCertName); |
| options.set_server_verification_option(GRPC_TLS_SERVER_VERIFICATION); |
| auto test_server_authorization_check = |
| std::make_shared<TestTlsServerAuthorizationCheck>(); |
| auto server_authorization_check_config = |
| std::make_shared<TlsServerAuthorizationCheckConfig>( |
| test_server_authorization_check); |
| options.set_server_authorization_check_config( |
| server_authorization_check_config); |
| auto channel_credentials = grpc::experimental::TlsCredentials(options); |
| GPR_ASSERT(channel_credentials.get() != nullptr); |
| } |
| |
| TEST(CredentialsTest, TlsServerAuthorizationCheckConfigErrorMessages) { |
| std::shared_ptr<TlsServerAuthorizationCheckConfig> config( |
| new TlsServerAuthorizationCheckConfig(nullptr)); |
| grpc_tls_server_authorization_check_arg* c_arg = |
| new grpc_tls_server_authorization_check_arg; |
| c_arg->error_details = new grpc_tls_error_details(); |
| c_arg->context = nullptr; |
| TlsServerAuthorizationCheckArg* arg = |
| new TlsServerAuthorizationCheckArg(c_arg); |
| int schedule_output = config->Schedule(arg); |
| |
| EXPECT_EQ(schedule_output, 1); |
| EXPECT_EQ(arg->status(), GRPC_STATUS_NOT_FOUND); |
| EXPECT_STREQ( |
| arg->error_details().c_str(), |
| "the interface of the server authorization check config is nullptr"); |
| |
| arg->set_status(GRPC_STATUS_OK); |
| config->Cancel(arg); |
| EXPECT_EQ(arg->status(), GRPC_STATUS_NOT_FOUND); |
| EXPECT_STREQ( |
| arg->error_details().c_str(), |
| "the interface of the server authorization check config is nullptr"); |
| |
| // Cleanup. |
| delete c_arg->error_details; |
| if (c_arg->destroy_context != nullptr) { |
| c_arg->destroy_context(c_arg->context); |
| } |
| delete c_arg; |
| } |
| |
| } // namespace |
| } // namespace testing |
| } // namespace grpc |
| |
| int main(int argc, char** argv) { |
| ::testing::InitGoogleTest(&argc, argv); |
| int ret = RUN_ALL_TESTS(); |
| return ret; |
| } |