/* * * Copyright 2015, Google Inc. * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are * met: * * * Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above * copyright notice, this list of conditions and the following disclaimer * in the documentation and/or other materials provided with the * distribution. * * Neither the name of Google Inc. nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * */ #include #include #include #include #include #include #include #include #include #include #include #include #include "test/core/util/reconnect_server.h" #include "test/cpp/util/test_config.h" #include "test/proto/test.grpc.pb.h" #include "test/proto/empty.grpc.pb.h" #include "test/proto/messages.grpc.pb.h" DEFINE_int32(control_port, 0, "Server port for controlling the server."); DEFINE_int32(retry_port, 0, "Server port for raw tcp connections. All incoming " "connections will be closed immediately."); using grpc::Server; using grpc::ServerBuilder; using grpc::ServerContext; using grpc::ServerCredentials; using grpc::ServerReader; using grpc::ServerReaderWriter; using grpc::ServerWriter; using grpc::SslServerCredentialsOptions; using grpc::Status; using grpc::testing::Empty; using grpc::testing::ReconnectService; using grpc::testing::ReconnectInfo; static bool got_sigint = false; class ReconnectServiceImpl : public ReconnectService::Service { public: explicit ReconnectServiceImpl(int retry_port) : retry_port_(retry_port), serving_(false), shutdown_(false) { reconnect_server_init(&tcp_server_); } ~ReconnectServiceImpl() { if (tcp_server_.tcp_server) { reconnect_server_destroy(&tcp_server_); } } void Poll(int seconds) { reconnect_server_poll(&tcp_server_, seconds); } Status Start(ServerContext* context, const Empty* request, Empty* response) { std::unique_lock lock(mu_); while (serving_ && !shutdown_) { cv_.wait(lock); } if (shutdown_) { return Status(grpc::StatusCode::UNAVAILABLE, "shutting down"); } serving_ = true; lock.unlock(); if (!tcp_server_.tcp_server) { reconnect_server_start(&tcp_server_, retry_port_); } else { reconnect_server_clear_timestamps(&tcp_server_); } return Status::OK; } Status Stop(ServerContext* context, const Empty* request, ReconnectInfo* response) { // extract timestamps and set response Verify(response); reconnect_server_clear_timestamps(&tcp_server_); std::lock_guard lock(mu_); serving_ = false; cv_.notify_one(); return Status::OK; } void Verify(ReconnectInfo* response) { double expected_backoff = 1000.0; const double kTransmissionDelay = 100.0; const double kBackoffMultiplier = 1.6; const double kJitterFactor = 0.2; const int kMaxBackoffMs = 120 * 1000; bool passed = true; for (timestamp_list* cur = tcp_server_.head; cur && cur->next; cur = cur->next) { double backoff = gpr_time_to_millis( gpr_time_sub(cur->next->timestamp, cur->timestamp)); double min_backoff = expected_backoff * (1 - kJitterFactor); double max_backoff = expected_backoff * (1 + kJitterFactor); if (backoff < min_backoff - kTransmissionDelay || backoff > max_backoff + kTransmissionDelay) { passed = false; } response->add_backoff_ms(static_cast(backoff)); expected_backoff *= kBackoffMultiplier; expected_backoff = expected_backoff > kMaxBackoffMs ? kMaxBackoffMs : expected_backoff; } response->set_passed(passed); } void Shutdown() { std::lock_guard lock(mu_); shutdown_ = true; cv_.notify_all(); } private: int retry_port_; reconnect_server tcp_server_; bool serving_; bool shutdown_; std::mutex mu_; std::condition_variable cv_; }; void RunServer() { std::ostringstream server_address; server_address << "0.0.0.0:" << FLAGS_control_port; ReconnectServiceImpl service(FLAGS_retry_port); ServerBuilder builder; builder.RegisterService(&service); builder.AddListeningPort(server_address.str(), grpc::InsecureServerCredentials()); std::unique_ptr server(builder.BuildAndStart()); gpr_log(GPR_INFO, "Server listening on %s", server_address.str().c_str()); while (!got_sigint) { service.Poll(5); } service.Shutdown(); } static void sigint_handler(int x) { got_sigint = true; } int main(int argc, char** argv) { grpc::testing::InitTest(&argc, &argv, true); signal(SIGINT, sigint_handler); GPR_ASSERT(FLAGS_control_port != 0); GPR_ASSERT(FLAGS_retry_port != 0); RunServer(); return 0; }