mirror of https://github.com/grpc/grpc.git
Fix google c2p resolver shutdown during metadata server queries (#28519)
* Fix c2p resolver shutdown during metadata server queries * handle lame channels in XDS clientpull/28803/head
apolcyn
3 years ago
committed by
GitHub
parent
9ffd1a7b0a
commit
64082940a5
9 changed files with 284 additions and 41 deletions
@ -0,0 +1,114 @@ |
||||
//
|
||||
// Copyright 2017 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 <atomic> |
||||
#include <memory> |
||||
#include <mutex> |
||||
#include <random> |
||||
#include <sstream> |
||||
#include <thread> |
||||
|
||||
#include <gmock/gmock.h> |
||||
|
||||
#include "absl/strings/str_format.h" |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <grpc/support/string_util.h> |
||||
#include <grpc/support/time.h> |
||||
#include <grpcpp/channel.h> |
||||
#include <grpcpp/client_context.h> |
||||
#include <grpcpp/create_channel.h> |
||||
#include <grpcpp/impl/codegen/sync.h> |
||||
#include <grpcpp/server.h> |
||||
#include <grpcpp/server_builder.h> |
||||
|
||||
#include "src/core/lib/gpr/env.h" |
||||
#include "src/core/lib/gprpp/thd.h" |
||||
#include "test/core/util/fake_udp_and_tcp_server.h" |
||||
#include "test/core/util/port.h" |
||||
#include "test/core/util/test_config.h" |
||||
|
||||
namespace { |
||||
|
||||
void TryConnectAndDestroy(const char* fake_metadata_server_address) { |
||||
grpc::ChannelArguments args; |
||||
std::string target = "google-c2p-experimental:///servername_not_used"; |
||||
args.SetInt("grpc.testing.google_c2p_resolver_pretend_running_on_gcp", 1); |
||||
args.SetString("grpc.testing.google_c2p_resolver_metadata_server_override", |
||||
fake_metadata_server_address); |
||||
auto channel = ::grpc::CreateCustomChannel( |
||||
target, grpc::InsecureChannelCredentials(), args); |
||||
// Start connecting, and give some time for the google-c2p resolver to begin
|
||||
// resolution and start trying to contact the metadata server.
|
||||
channel->GetState(true /* try_to_connect */); |
||||
ASSERT_FALSE( |
||||
channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(100))); |
||||
channel.reset(); |
||||
}; |
||||
|
||||
// Exercise the machinery involved with shutting down the C2P resolver while
|
||||
// it's waiting for its initial metadata server queries to finish.
|
||||
TEST(DestroyGoogleC2pChannelWithActiveConnectStressTest, |
||||
LoopTryConnectAndDestroyWithHangingMetadataServer) { |
||||
// Create a fake metadata server which hangs.
|
||||
grpc_core::testing::FakeUdpAndTcpServer fake_metadata_server( |
||||
grpc_core::testing::FakeUdpAndTcpServer::AcceptMode:: |
||||
kWaitForClientToSendFirstBytes, |
||||
grpc_core::testing::FakeUdpAndTcpServer::CloseSocketUponCloseFromPeer); |
||||
std::vector<std::unique_ptr<std::thread>> threads; |
||||
const int kNumThreads = 100; |
||||
threads.reserve(kNumThreads); |
||||
for (int i = 0; i < kNumThreads; i++) { |
||||
threads.emplace_back( |
||||
new std::thread(TryConnectAndDestroy, fake_metadata_server.address())); |
||||
} |
||||
for (size_t i = 0; i < threads.size(); i++) { |
||||
threads[i]->join(); |
||||
} |
||||
} |
||||
|
||||
// Exercise the machinery involved with shutting down the C2P resolver while
|
||||
// it's waiting for its initial metadata server queries to finish.
|
||||
TEST(DestroyGoogleC2pChannelWithActiveConnectStressTest, |
||||
LoopTryConnectAndDestroyWithFastFailingMetadataServer) { |
||||
// Create a fake metadata server address which rejects connections
|
||||
int port = grpc_pick_unused_port_or_die(); |
||||
std::string address = absl::StrFormat("[::1]:%d", port); |
||||
std::vector<std::unique_ptr<std::thread>> threads; |
||||
const int kNumThreads = 100; |
||||
threads.reserve(kNumThreads); |
||||
for (int i = 0; i < kNumThreads; i++) { |
||||
threads.emplace_back( |
||||
new std::thread(TryConnectAndDestroy, address.c_str())); |
||||
} |
||||
for (size_t i = 0; i < threads.size(); i++) { |
||||
threads[i]->join(); |
||||
} |
||||
} |
||||
|
||||
} // namespace
|
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc::testing::TestEnvironment env(argc, argv); |
||||
gpr_setenv("GRPC_EXPERIMENTAL_GOOGLE_C2P_RESOLVER", "true"); |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
grpc_init(); |
||||
auto result = RUN_ALL_TESTS(); |
||||
grpc_shutdown(); |
||||
return result; |
||||
} |
||||
Loading…
Reference in new issue