vpn-manager: Add l2tp/ipsec vpn manager

l2tp_manager_test.cc

+// Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chromeos/process_mock.h"
+#include "chromeos/syslog_logging.h"
+#include "chromeos/test_helpers.h"
+#include "gflags/gflags.h"
+#include "gtest/gtest.h"
+
+#include "vpn-manager/l2tp_manager.h"
+
+DECLARE_string(password);
+DECLARE_int32(ppp_setup_timeout);
+DECLARE_string(pppd_plugin);
+DECLARE_string(user);
+
+using ::chromeos::FindLog;
+using ::chromeos::ProcessMock;
+using ::testing::_;
+using ::testing::InSequence;
+using ::testing::Return;
+
+class L2tpManagerTest : public ::testing::Test {
+ public:
+  void SetUp() {
+    test_path_ = FilePath("test");
+    ServiceManager::temp_path_ = new FilePath(test_path_);
+    file_util::Delete(test_path_, true);
+    file_util::CreateDirectory(test_path_);
+    remote_ = "1.2.3.4";
+    control_path_ = test_path_.Append("control");
+    pppd_config_path_ = test_path_.Append("pppd.config");
+    ppp_interface_path_ = test_path_.Append("ppp0");
+    l2tpd_ = new ProcessMock;
+    l2tp_.l2tpd_.reset(l2tpd_);
+    l2tp_.l2tpd_control_path_ = control_path_;
+    l2tp_.ppp_interface_path_ = ppp_interface_path_;
+    FLAGS_pppd_plugin = "";
+    FLAGS_user = "me";
+    EXPECT_TRUE(l2tp_.Initialize(remote_));
+  }
+  void TearDown() {
+    ServiceManager::DeleteTemp();
+  }
+
+ protected:
+  std::string remote_;
+  FilePath test_path_;
+  FilePath control_path_;
+  FilePath pppd_config_path_;
+  FilePath ppp_interface_path_;
+  L2tpManager l2tp_;
+  ProcessMock* l2tpd_;
+};
+
+TEST_F(L2tpManagerTest, FormatL2tpdConfiguration) {
+  static const char kBaseExpected[] =
+      "[lac managed]\n"
+      "lns = 1.2.3.4\n"
+      "require chap = yes\n"
+      "refuse pap = yes\n"
+      "require authentication = yes\n"
+      "name = me\n"
+      "ppp debug = yes\n"
+      "pppoptfile = test/pppd.config\n"
+      "length bit = yes\n";
+  EXPECT_EQ(kBaseExpected,
+            l2tp_.FormatL2tpdConfiguration(pppd_config_path_.value()));
+}
+
+TEST_F(L2tpManagerTest, FormatPppdConfiguration) {
+  const char kBaseExpected[] =
+      "ipcp-accept-local\n"
+      "ipcp-accept-remote\n"
+      "refuse-eap\n"
+      "noccp\n"
+      "noauth\n"
+      "crtscts\n"
+      "idle 1800\n"
+      "mtu 1410\n"
+      "mru 1410\n"
+      "nodefaultroute\n"
+      "debug\n"
+      "lock\n"
+      "connect-delay 5000\n";
+  std::string expected(kBaseExpected);
+  EXPECT_EQ(expected, l2tp_.FormatPppdConfiguration());
+  FLAGS_pppd_plugin = "myplugin";
+  expected.append("plugin myplugin\n");
+  EXPECT_EQ(expected, l2tp_.FormatPppdConfiguration());
+}
+
+TEST_F(L2tpManagerTest, Initiate) {
+  FLAGS_user = "me";
+  FLAGS_password = "password";
+  EXPECT_TRUE(l2tp_.Initiate());
+  ExpectFileEquals("c managed me password\n", control_path_.value().c_str());
+  FLAGS_pppd_plugin = "set";
+  EXPECT_TRUE(l2tp_.Initiate());
+  ExpectFileEquals("c managed\n", control_path_.value().c_str());
+}
+
+TEST_F(L2tpManagerTest, Terminate) {
+  EXPECT_TRUE(l2tp_.Terminate());
+  ExpectFileEquals("d managed\n", control_path_.value().c_str());
+}
+
+TEST_F(L2tpManagerTest, Start) {
+  InSequence unused;
+  const int kMockFd = 567;
+
+  EXPECT_CALL(*l2tpd_, Reset(0));
+  EXPECT_CALL(*l2tpd_, AddArg(L2TPD));
+  EXPECT_CALL(*l2tpd_, AddArg("-c"));
+  EXPECT_CALL(*l2tpd_, AddArg("test/l2tpd.conf"));
+  EXPECT_CALL(*l2tpd_, AddArg("-C"));
+  EXPECT_CALL(*l2tpd_, AddArg("test/l2tpd.control"));
+  EXPECT_CALL(*l2tpd_, AddArg("-D"));
+  EXPECT_CALL(*l2tpd_, RedirectUsingPipe(STDERR_FILENO, false));
+  EXPECT_CALL(*l2tpd_, Start());
+  EXPECT_CALL(*l2tpd_, GetPipe(STDERR_FILENO)).WillOnce(Return(kMockFd));
+
+  EXPECT_TRUE(l2tp_.Start());
+  EXPECT_EQ(kMockFd, l2tp_.output_fd());
+  EXPECT_FALSE(l2tp_.start_ticks_.is_null());
+  EXPECT_FALSE(l2tp_.was_initiated_);
+}
+
+TEST_F(L2tpManagerTest, PollWaitIfNotUpYet) {
+  l2tp_.start_ticks_ = base::TimeTicks::Now();
+  EXPECT_EQ(1000, l2tp_.Poll());
+}
+
+TEST_F(L2tpManagerTest, PollTimeoutWaitingForControl) {
+  l2tp_.start_ticks_ = base::TimeTicks::Now() -
+      base::TimeDelta::FromSeconds(FLAGS_ppp_setup_timeout + 1);
+  EXPECT_EQ(1000, l2tp_.Poll());
+  EXPECT_TRUE(FindLog("PPP setup timed out"));
+  EXPECT_TRUE(l2tp_.was_stopped_);
+  EXPECT_FALSE(file_util::PathExists(control_path_));
+}
+
+TEST_F(L2tpManagerTest, PollTimeoutWaitingForUp) {
+  l2tp_.start_ticks_ = base::TimeTicks::Now() -
+      base::TimeDelta::FromSeconds(FLAGS_ppp_setup_timeout + 1);
+  l2tp_.was_initiated_ = true;
+  EXPECT_EQ(1000, l2tp_.Poll());
+  EXPECT_TRUE(FindLog("PPP setup timed out"));
+  EXPECT_TRUE(l2tp_.was_stopped_);
+  ExpectFileEquals("d managed\n", control_path_.value().c_str());
+}
+
+TEST_F(L2tpManagerTest, PollInitiateConnection) {
+  l2tp_.start_ticks_ = base::TimeTicks::Now();
+  file_util::WriteFile(control_path_, "", 0);
+  EXPECT_FALSE(l2tp_.was_initiated_);
+  FLAGS_pppd_plugin = "set";
+  EXPECT_EQ(1000, l2tp_.Poll());
+  EXPECT_TRUE(l2tp_.was_initiated_);
+  ExpectFileEquals("c managed\n", control_path_.value().c_str());
+}
+
+TEST_F(L2tpManagerTest, PollTransitionToUp) {
+  l2tp_.start_ticks_ = base::TimeTicks::Now();
+  l2tp_.was_initiated_ = true;
+  EXPECT_FALSE(l2tp_.is_running_);
+  file_util::WriteFile(ppp_interface_path_, "", 0);
+  EXPECT_EQ(-1, l2tp_.Poll());
+  EXPECT_TRUE(FindLog("L2TP connection now up"));
+  EXPECT_TRUE(l2tp_.is_running_);
+}
+
+TEST_F(L2tpManagerTest, PollNothingIfRunning) {
+  l2tp_.is_running_ = true;
+  EXPECT_EQ(-1, l2tp_.Poll());
+}
+
+int main(int argc, char** argv) {
+  SetUpTests(&argc, argv, true);
+  return RUN_ALL_TESTS();
+}