|
|
@@ -0,0 +1,422 @@
|
|
|
+/*
|
|
|
+ *
|
|
|
+ * Copyright 2019 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/grpc.h>
|
|
|
+#include <grpc/support/log.h>
|
|
|
+#include <grpc/support/time.h>
|
|
|
+#include <grpcpp/channel.h>
|
|
|
+#include <grpcpp/client_context.h>
|
|
|
+#include <grpcpp/create_channel.h>
|
|
|
+#include <grpcpp/server.h>
|
|
|
+#include <grpcpp/server_builder.h>
|
|
|
+#include <grpcpp/server_context.h>
|
|
|
+
|
|
|
+#include "src/core/lib/iomgr/timer.h"
|
|
|
+#include "src/proto/grpc/testing/echo.grpc.pb.h"
|
|
|
+#include "test/core/util/port.h"
|
|
|
+#include "test/core/util/test_config.h"
|
|
|
+#include "test/cpp/end2end/test_service_impl.h"
|
|
|
+#include "test/cpp/util/subprocess.h"
|
|
|
+
|
|
|
+#include <gtest/gtest.h>
|
|
|
+#include <pthread.h>
|
|
|
+#include <sys/time.h>
|
|
|
+#include <thread>
|
|
|
+
|
|
|
+using grpc::testing::EchoRequest;
|
|
|
+using grpc::testing::EchoResponse;
|
|
|
+
|
|
|
+static std::string g_root;
|
|
|
+
|
|
|
+static gpr_mu g_mu;
|
|
|
+extern gpr_timespec (*gpr_now_impl)(gpr_clock_type clock_type);
|
|
|
+gpr_timespec (*gpr_now_impl_orig)(gpr_clock_type clock_type) = gpr_now_impl;
|
|
|
+static int g_time_shift_sec = 0;
|
|
|
+static int g_time_shift_nsec = 0;
|
|
|
+static gpr_timespec now_impl(gpr_clock_type clock) {
|
|
|
+ auto ts = gpr_now_impl_orig(clock);
|
|
|
+ // We only manipulate the realtime clock to simulate changes in wall-clock
|
|
|
+ // time
|
|
|
+ if (clock != GPR_CLOCK_REALTIME) {
|
|
|
+ return ts;
|
|
|
+ }
|
|
|
+ GPR_ASSERT(ts.tv_nsec >= 0);
|
|
|
+ GPR_ASSERT(ts.tv_nsec < GPR_NS_PER_SEC);
|
|
|
+ gpr_mu_lock(&g_mu);
|
|
|
+ ts.tv_sec += g_time_shift_sec;
|
|
|
+ ts.tv_nsec += g_time_shift_nsec;
|
|
|
+ gpr_mu_unlock(&g_mu);
|
|
|
+ if (ts.tv_nsec >= GPR_NS_PER_SEC) {
|
|
|
+ ts.tv_nsec -= GPR_NS_PER_SEC;
|
|
|
+ ++ts.tv_sec;
|
|
|
+ } else if (ts.tv_nsec < 0) {
|
|
|
+ --ts.tv_sec;
|
|
|
+ ts.tv_nsec = GPR_NS_PER_SEC + ts.tv_nsec;
|
|
|
+ }
|
|
|
+ return ts;
|
|
|
+}
|
|
|
+
|
|
|
+// offset the value returned by gpr_now(GPR_CLOCK_REALTIME) by msecs
|
|
|
+// milliseconds
|
|
|
+static void set_now_offset(int msecs) {
|
|
|
+ g_time_shift_sec = msecs / 1000;
|
|
|
+ g_time_shift_nsec = (msecs % 1000) * 1e6;
|
|
|
+}
|
|
|
+
|
|
|
+// restore the original implementation of gpr_now()
|
|
|
+static void reset_now_offset() {
|
|
|
+ g_time_shift_sec = 0;
|
|
|
+ g_time_shift_nsec = 0;
|
|
|
+}
|
|
|
+
|
|
|
+namespace grpc {
|
|
|
+namespace testing {
|
|
|
+
|
|
|
+namespace {
|
|
|
+
|
|
|
+// gpr_now() is called with invalid clock_type
|
|
|
+TEST(TimespecTest, GprNowInvalidClockType) {
|
|
|
+ // initialize to some junk value
|
|
|
+ gpr_clock_type invalid_clock_type = (gpr_clock_type)32641;
|
|
|
+ EXPECT_DEATH(gpr_now(invalid_clock_type), ".*");
|
|
|
+}
|
|
|
+
|
|
|
+// Add timespan with negative nanoseconds
|
|
|
+TEST(TimespecTest, GprTimeAddNegativeNs) {
|
|
|
+ gpr_timespec now = gpr_now(GPR_CLOCK_MONOTONIC);
|
|
|
+ gpr_timespec bad_ts = {1, -1000, GPR_TIMESPAN};
|
|
|
+ EXPECT_DEATH(gpr_time_add(now, bad_ts), ".*");
|
|
|
+}
|
|
|
+
|
|
|
+// Subtract timespan with negative nanoseconds
|
|
|
+TEST(TimespecTest, GprTimeSubNegativeNs) {
|
|
|
+ // Nanoseconds must always be positive. Negative timestamps are represented by
|
|
|
+ // (negative seconds, positive nanoseconds)
|
|
|
+ gpr_timespec now = gpr_now(GPR_CLOCK_MONOTONIC);
|
|
|
+ gpr_timespec bad_ts = {1, -1000, GPR_TIMESPAN};
|
|
|
+ EXPECT_DEATH(gpr_time_sub(now, bad_ts), ".*");
|
|
|
+}
|
|
|
+
|
|
|
+// Add negative milliseconds to gpr_timespec
|
|
|
+TEST(TimespecTest, GrpcNegativeMillisToTimespec) {
|
|
|
+ // -1500 milliseconds converts to timespec (-2 secs, 5 * 10^8 nsec)
|
|
|
+ gpr_timespec ts = grpc_millis_to_timespec(-1500, GPR_CLOCK_MONOTONIC);
|
|
|
+ GPR_ASSERT(ts.tv_sec = -2);
|
|
|
+ GPR_ASSERT(ts.tv_nsec = 5e8);
|
|
|
+ GPR_ASSERT(ts.clock_type == GPR_CLOCK_MONOTONIC);
|
|
|
+}
|
|
|
+
|
|
|
+class TimeChangeTest : public ::testing::Test {
|
|
|
+ protected:
|
|
|
+ TimeChangeTest() {}
|
|
|
+
|
|
|
+ void SetUp() {
|
|
|
+ auto port = grpc_pick_unused_port_or_die();
|
|
|
+ std::ostringstream addr_stream;
|
|
|
+ addr_stream << "localhost:" << port;
|
|
|
+ auto addr = addr_stream.str();
|
|
|
+ server_.reset(new SubProcess({
|
|
|
+ g_root + "/client_crash_test_server",
|
|
|
+ "--address=" + addr,
|
|
|
+ }));
|
|
|
+ GPR_ASSERT(server_);
|
|
|
+ channel_ = CreateChannel(addr, InsecureChannelCredentials());
|
|
|
+ GPR_ASSERT(channel_);
|
|
|
+ stub_ = grpc::testing::EchoTestService::NewStub(channel_);
|
|
|
+ }
|
|
|
+
|
|
|
+ void TearDown() {
|
|
|
+ server_.reset();
|
|
|
+ reset_now_offset();
|
|
|
+ }
|
|
|
+
|
|
|
+ std::unique_ptr<grpc::testing::EchoTestService::Stub> CreateStub() {
|
|
|
+ return grpc::testing::EchoTestService::NewStub(channel_);
|
|
|
+ }
|
|
|
+
|
|
|
+ std::shared_ptr<Channel> GetChannel() { return channel_; }
|
|
|
+ // time jump offsets in milliseconds
|
|
|
+ const int TIME_OFFSET1 = 20123;
|
|
|
+ const int TIME_OFFSET2 = 5678;
|
|
|
+
|
|
|
+ private:
|
|
|
+ std::unique_ptr<SubProcess> server_;
|
|
|
+ std::shared_ptr<Channel> channel_;
|
|
|
+ std::unique_ptr<grpc::testing::EchoTestService::Stub> stub_;
|
|
|
+};
|
|
|
+
|
|
|
+// Wall-clock time jumps forward on client before bidi stream is created
|
|
|
+TEST_F(TimeChangeTest, TimeJumpForwardBeforeStreamCreated) {
|
|
|
+ EchoRequest request;
|
|
|
+ EchoResponse response;
|
|
|
+ ClientContext context;
|
|
|
+ context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ context.AddMetadata(kServerResponseStreamsToSend, "1");
|
|
|
+
|
|
|
+ auto channel = GetChannel();
|
|
|
+ GPR_ASSERT(channel);
|
|
|
+ EXPECT_TRUE(
|
|
|
+ channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000)));
|
|
|
+ auto stub = CreateStub();
|
|
|
+
|
|
|
+ // time jumps forward by TIME_OFFSET1 milliseconds
|
|
|
+ set_now_offset(TIME_OFFSET1);
|
|
|
+ auto stream = stub->BidiStream(&context);
|
|
|
+ request.set_message("Hello");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+
|
|
|
+ EXPECT_TRUE(stream->WritesDone());
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ auto status = stream->Finish();
|
|
|
+ EXPECT_TRUE(status.ok());
|
|
|
+}
|
|
|
+
|
|
|
+// Wall-clock time jumps back on client before bidi stream is created
|
|
|
+TEST_F(TimeChangeTest, TimeJumpBackBeforeStreamCreated) {
|
|
|
+ EchoRequest request;
|
|
|
+ EchoResponse response;
|
|
|
+ ClientContext context;
|
|
|
+ context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ context.AddMetadata(kServerResponseStreamsToSend, "1");
|
|
|
+
|
|
|
+ auto channel = GetChannel();
|
|
|
+ GPR_ASSERT(channel);
|
|
|
+ EXPECT_TRUE(
|
|
|
+ channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000)));
|
|
|
+ auto stub = CreateStub();
|
|
|
+
|
|
|
+ // time jumps back by TIME_OFFSET1 milliseconds
|
|
|
+ set_now_offset(-TIME_OFFSET1);
|
|
|
+ auto stream = stub->BidiStream(&context);
|
|
|
+ request.set_message("Hello");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+
|
|
|
+ EXPECT_TRUE(stream->WritesDone());
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+ EXPECT_EQ(request.message(), response.message());
|
|
|
+
|
|
|
+ auto status = stream->Finish();
|
|
|
+ EXPECT_TRUE(status.ok());
|
|
|
+}
|
|
|
+
|
|
|
+// Wall-clock time jumps forward on client while call is in progress
|
|
|
+TEST_F(TimeChangeTest, TimeJumpForwardAfterStreamCreated) {
|
|
|
+ EchoRequest request;
|
|
|
+ EchoResponse response;
|
|
|
+ ClientContext context;
|
|
|
+ context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ context.AddMetadata(kServerResponseStreamsToSend, "2");
|
|
|
+
|
|
|
+ auto channel = GetChannel();
|
|
|
+ GPR_ASSERT(channel);
|
|
|
+ EXPECT_TRUE(
|
|
|
+ channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000)));
|
|
|
+ auto stub = CreateStub();
|
|
|
+
|
|
|
+ auto stream = stub->BidiStream(&context);
|
|
|
+
|
|
|
+ request.set_message("Hello");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ // time jumps forward by TIME_OFFSET1 milliseconds.
|
|
|
+ set_now_offset(TIME_OFFSET1);
|
|
|
+
|
|
|
+ request.set_message("World");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->WritesDone());
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ auto status = stream->Finish();
|
|
|
+ EXPECT_TRUE(status.ok());
|
|
|
+}
|
|
|
+
|
|
|
+// Wall-clock time jumps back on client while call is in progress
|
|
|
+TEST_F(TimeChangeTest, TimeJumpBackAfterStreamCreated) {
|
|
|
+ EchoRequest request;
|
|
|
+ EchoResponse response;
|
|
|
+ ClientContext context;
|
|
|
+ context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ context.AddMetadata(kServerResponseStreamsToSend, "2");
|
|
|
+
|
|
|
+ auto channel = GetChannel();
|
|
|
+ GPR_ASSERT(channel);
|
|
|
+ EXPECT_TRUE(
|
|
|
+ channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000)));
|
|
|
+ auto stub = CreateStub();
|
|
|
+
|
|
|
+ auto stream = stub->BidiStream(&context);
|
|
|
+
|
|
|
+ request.set_message("Hello");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ // time jumps back TIME_OFFSET1 milliseconds.
|
|
|
+ set_now_offset(-TIME_OFFSET1);
|
|
|
+
|
|
|
+ request.set_message("World");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->WritesDone());
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ auto status = stream->Finish();
|
|
|
+ EXPECT_TRUE(status.ok());
|
|
|
+}
|
|
|
+
|
|
|
+// Wall-clock time jumps forward on client before connection to server is up
|
|
|
+TEST_F(TimeChangeTest, TimeJumpForwardBeforeServerConnect) {
|
|
|
+ EchoRequest request;
|
|
|
+ EchoResponse response;
|
|
|
+ ClientContext context;
|
|
|
+ context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ context.AddMetadata(kServerResponseStreamsToSend, "2");
|
|
|
+
|
|
|
+ auto channel = GetChannel();
|
|
|
+ GPR_ASSERT(channel);
|
|
|
+
|
|
|
+ // time jumps forward by TIME_OFFSET2 milliseconds
|
|
|
+ set_now_offset(TIME_OFFSET2);
|
|
|
+
|
|
|
+ auto ret =
|
|
|
+ channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ // We use monotonic clock for pthread_cond_timedwait() deadline on linux, and
|
|
|
+ // realtime clock on other platforms - see gpr_cv_wait() in sync_posix.cc.
|
|
|
+ // So changes in system clock affect deadlines on non-linux platforms
|
|
|
+#ifdef GPR_LINUX
|
|
|
+ EXPECT_TRUE(ret);
|
|
|
+ auto stub = CreateStub();
|
|
|
+ auto stream = stub->BidiStream(&context);
|
|
|
+
|
|
|
+ request.set_message("Hello");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+ request.set_message("World");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->WritesDone());
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ auto status = stream->Finish();
|
|
|
+ EXPECT_TRUE(status.ok());
|
|
|
+#else
|
|
|
+ EXPECT_FALSE(ret);
|
|
|
+#endif
|
|
|
+}
|
|
|
+
|
|
|
+// Wall-clock time jumps back on client before connection to server is up
|
|
|
+TEST_F(TimeChangeTest, TimeJumpBackBeforeServerConnect) {
|
|
|
+ EchoRequest request;
|
|
|
+ EchoResponse response;
|
|
|
+ ClientContext context;
|
|
|
+ context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ context.AddMetadata(kServerResponseStreamsToSend, "2");
|
|
|
+
|
|
|
+ auto channel = GetChannel();
|
|
|
+ GPR_ASSERT(channel);
|
|
|
+
|
|
|
+ // time jumps back by TIME_OFFSET2 milliseconds
|
|
|
+ set_now_offset(-TIME_OFFSET2);
|
|
|
+
|
|
|
+ EXPECT_TRUE(
|
|
|
+ channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000)));
|
|
|
+ auto stub = CreateStub();
|
|
|
+ auto stream = stub->BidiStream(&context);
|
|
|
+
|
|
|
+ request.set_message("Hello");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+ request.set_message("World");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+ EXPECT_TRUE(stream->WritesDone());
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ auto status = stream->Finish();
|
|
|
+ EXPECT_TRUE(status.ok());
|
|
|
+}
|
|
|
+
|
|
|
+// Wall-clock time jumps forward and backwards during call
|
|
|
+TEST_F(TimeChangeTest, TimeJumpForwardAndBackDuringCall) {
|
|
|
+ EchoRequest request;
|
|
|
+ EchoResponse response;
|
|
|
+ ClientContext context;
|
|
|
+ context.set_deadline(grpc_timeout_milliseconds_to_deadline(5000));
|
|
|
+ context.AddMetadata(kServerResponseStreamsToSend, "2");
|
|
|
+
|
|
|
+ auto channel = GetChannel();
|
|
|
+ GPR_ASSERT(channel);
|
|
|
+
|
|
|
+ EXPECT_TRUE(
|
|
|
+ channel->WaitForConnected(grpc_timeout_milliseconds_to_deadline(5000)));
|
|
|
+ auto stub = CreateStub();
|
|
|
+ auto stream = stub->BidiStream(&context);
|
|
|
+
|
|
|
+ request.set_message("Hello");
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+
|
|
|
+ // time jumps back by TIME_OFFSET2 milliseconds
|
|
|
+ set_now_offset(-TIME_OFFSET2);
|
|
|
+
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+ request.set_message("World");
|
|
|
+
|
|
|
+ // time jumps forward by TIME_OFFSET milliseconds
|
|
|
+ set_now_offset(TIME_OFFSET1);
|
|
|
+
|
|
|
+ EXPECT_TRUE(stream->Write(request));
|
|
|
+
|
|
|
+ // time jumps back by TIME_OFFSET2 milliseconds
|
|
|
+ set_now_offset(-TIME_OFFSET2);
|
|
|
+
|
|
|
+ EXPECT_TRUE(stream->WritesDone());
|
|
|
+
|
|
|
+ // time jumps back by TIME_OFFSET2 milliseconds
|
|
|
+ set_now_offset(-TIME_OFFSET2);
|
|
|
+
|
|
|
+ EXPECT_TRUE(stream->Read(&response));
|
|
|
+
|
|
|
+ // time jumps back by TIME_OFFSET2 milliseconds
|
|
|
+ set_now_offset(-TIME_OFFSET2);
|
|
|
+
|
|
|
+ auto status = stream->Finish();
|
|
|
+ EXPECT_TRUE(status.ok());
|
|
|
+}
|
|
|
+
|
|
|
+} // namespace
|
|
|
+
|
|
|
+} // namespace testing
|
|
|
+} // namespace grpc
|
|
|
+
|
|
|
+int main(int argc, char** argv) {
|
|
|
+ std::string me = argv[0];
|
|
|
+ // get index of last slash in path to test binary
|
|
|
+ auto lslash = me.rfind('/');
|
|
|
+ // set g_root = path to directory containing test binary
|
|
|
+ if (lslash != std::string::npos) {
|
|
|
+ g_root = me.substr(0, lslash);
|
|
|
+ } else {
|
|
|
+ g_root = ".";
|
|
|
+ }
|
|
|
+
|
|
|
+ gpr_mu_init(&g_mu);
|
|
|
+ gpr_now_impl = now_impl;
|
|
|
+
|
|
|
+ grpc::testing::TestEnvironment env(argc, argv);
|
|
|
+ ::testing::InitGoogleTest(&argc, argv);
|
|
|
+ auto ret = RUN_ALL_TESTS();
|
|
|
+ return ret;
|
|
|
+}
|
rmstar