vpn-manager: Add l2tp/ipsec vpn manager

ipsec_manager_test.cc

Index: ipsec_manager_test.cc
diff --git a/ipsec_manager_test.cc b/ipsec_manager_test.cc
new file mode 100644
index 0000000000000000000000000000000000000000..738341951d9c73b8d7f85e4854370167e9a345cc
--- /dev/null
+++ b/ipsec_manager_test.cc
@@ -0,0 +1,258 @@
+// 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 "base/command_line.h"
+#include "base/file_util.h"
+#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/ipsec_manager.h"
+
+using ::chromeos::FindLog;
+using ::chromeos::ProcessMock;
+using ::testing::_;
+using ::testing::InSequence;
+using ::testing::Return;
+
+const int kMockFd = 123;
+
+const int kMockStarterPid = 10001;
+
+DECLARE_int32(ipsec_timeout);
+DECLARE_string(local);
+
+class IpsecManagerTest : public ::testing::Test {
+ public:
+  void SetUp() {
+    file_util::GetCurrentDirectory(&test_path_);
+    test_path_ = test_path_.Append("test");
+    file_util::Delete(test_path_, true);
+    file_util::CreateDirectory(test_path_);
+    stateful_container_ = test_path_.Append("etc");
+    file_util::CreateDirectory(stateful_container_);
+    remote_ = "1.2.3.4";
+    ServiceManager::temp_path_ = new FilePath(test_path_);
+    psk_file_ = test_path_.Append("psk").value();
+    server_ca_file_ = test_path_.Append("server.ca").value();
+    client_key_file_ = test_path_.Append("client.key").value();
+    client_cert_file_ = test_path_.Append("client.cert").value();
+    ipsec_up_file_ = test_path_.Append("ipsec-up").value();
+    WriteFile(psk_file_, "secret");
+    WriteFile(server_ca_file_, "");
+    WriteFile(client_key_file_, "");
+    WriteFile(client_cert_file_, "");
+    chromeos::ClearLog();
+    starter_ = new ProcessMock;
+    ipsec_.starter_.reset(starter_);
+    ipsec_.stateful_container_ = stateful_container_.value();
+    ipsec_.ipsec_group_ = getgid();
+    ipsec_.ipsec_up_file_ = ipsec_up_file_;
+  }
+
+  void TearDown() {
+    // Removes and deletes temp path.
+    ServiceManager::DeleteTemp();
+  }
+
+  void SetStartStarterExpectations(bool already_running);
+
+ protected:
+  void WriteFile(const std::string& file_path, const char* contents) {
+    file_util::WriteFile(FilePath(file_path), contents, strlen(contents));
+  }
+
+  void DoInitialize(int ike_version, bool use_psk);
+
+  IpsecManager ipsec_;
+  FilePath stateful_container_;
+  FilePath test_path_;
+  std::string remote_;
+  std::string psk_file_;
+  std::string server_ca_file_;
+  std::string client_key_file_;
+  std::string client_cert_file_;
+  std::string ipsec_up_file_;
+  ProcessMock* starter_;
+};
+
+void IpsecManagerTest::DoInitialize(int ike_version, bool use_psk) {
+  if (use_psk) {
+    ASSERT_TRUE(ipsec_.Initialize(ike_version, remote_, psk_file_, "", "", ""));
+  } else {
+    ASSERT_TRUE(ipsec_.Initialize(ike_version, remote_, "", server_ca_file_,
+                                  client_key_file_, client_cert_file_));
+  }
+}
+
+void IpsecManagerTest::SetStartStarterExpectations(bool already_running) {
+  if (already_running) {
+    ipsec_.starter_pid_file_ = test_path_.Append("starter_pid").value();
+    // File must exist.
+    file_util::WriteFile(FilePath(ipsec_.starter_pid_file_), "", 0);
+    // Test that it attempts to kill the running starter.
+    EXPECT_CALL(*starter_, ResetPidByFile(ipsec_.starter_pid_file_)).
+        WillOnce(Return(true));
+    EXPECT_CALL(*starter_, pid()).WillOnce(Return(1));
+    EXPECT_CALL(*starter_, Reset(0));
+  }
+
+  EXPECT_CALL(*starter_, AddArg(IPSEC_STARTER));
+  EXPECT_CALL(*starter_, AddArg("--nofork"));
+  EXPECT_CALL(*starter_, RedirectUsingPipe(STDERR_FILENO, false));
+  EXPECT_CALL(*starter_, Start()).WillOnce(Return(true));
+  EXPECT_CALL(*starter_, GetPipe(STDERR_FILENO)).WillOnce(Return(kMockFd));
+  EXPECT_CALL(*starter_, pid()).WillOnce(Return(kMockStarterPid));
+}
+
+TEST_F(IpsecManagerTest, InitializeBadRemote) {
+  EXPECT_FALSE(ipsec_.Initialize(1, "", psk_file_, "", "", ""));
+  EXPECT_TRUE(FindLog("Missing remote"));
+}
+
+TEST_F(IpsecManagerTest, InitializeNoAuth) {
+  EXPECT_FALSE(ipsec_.Initialize(1, remote_, "", "", "", ""));
+  EXPECT_TRUE(FindLog("Must specify either PSK or certificates"));
+}
+
+TEST_F(IpsecManagerTest, InitializeNotBoth) {
+  EXPECT_FALSE(ipsec_.Initialize(1, remote_,
+                                 psk_file_,
+                                 server_ca_file_,
+                                 client_key_file_,
+                                 client_cert_file_));
+  EXPECT_TRUE(FindLog("Specified both PSK and certificates"));
+}
+
+TEST_F(IpsecManagerTest, InitializeUnsupportedVersion) {
+  EXPECT_FALSE(ipsec_.Initialize(3, remote_, psk_file_, "", "", ""));
+  EXPECT_TRUE(FindLog("Unsupported IKE version"));
+}
+
+TEST_F(IpsecManagerTest, PollWaitIfNotUpYet) {
+  ipsec_.start_ticks_ = base::TimeTicks::Now();
+  EXPECT_EQ(1000, ipsec_.Poll());
+}
+
+TEST_F(IpsecManagerTest, PollTimeoutWaiting) {
+  ipsec_.start_ticks_ = base::TimeTicks::Now() -
+      base::TimeDelta::FromSeconds(FLAGS_ipsec_timeout + 1);
+  EXPECT_EQ(1000, ipsec_.Poll());
+  EXPECT_TRUE(FindLog("IPsec connection timed out"));
+  EXPECT_TRUE(ipsec_.was_stopped_);
+}
+
+TEST_F(IpsecManagerTest, PollTransitionToUp) {
+  ipsec_.start_ticks_ = base::TimeTicks::Now();
+  WriteFile(ipsec_up_file_, "");
+  EXPECT_FALSE(ipsec_.is_running_);
+  EXPECT_EQ(-1, ipsec_.Poll());
+  EXPECT_TRUE(FindLog("IPsec connection now up"));
+  EXPECT_TRUE(ipsec_.is_running_);
+}
+
+TEST_F(IpsecManagerTest, PollNothingIfRunning) {
+  ipsec_.is_running_ = true;
+  EXPECT_EQ(-1, ipsec_.Poll());
+}
+
+class IpsecManagerTestIkeV1Psk : public IpsecManagerTest {
+ public:
+  void SetUp() {
+    IpsecManagerTest::SetUp();
+    DoInitialize(1, true);
+  }
+
+ protected:
+  void CheckStarter(const std::string& actual);
+};
+
+TEST_F(IpsecManagerTestIkeV1Psk, Initialize) {
+}
+
+TEST_F(IpsecManagerTestIkeV1Psk, FormatPsk) {
+  const char kLocal[] = "5.6.7.8";
+  FilePath input(test_path_.Append("psk"));
+  const char psk[] = "pAssword\n";
+  file_util::WriteFile(input, psk, strlen(psk));
+  FilePath output;
+  FLAGS_local = std::string(kLocal);
+  std::string formatted;
+  EXPECT_TRUE(ipsec_.FormatPsk(input, &formatted));
+  EXPECT_EQ("5.6.7.8 1.2.3.4 : PSK \"pAssword\"\n", formatted);
+}
+
+TEST_F(IpsecManagerTestIkeV1Psk, StartStarterNotYetRunning) {
+  InSequence unused;
+  SetStartStarterExpectations(false);
+  EXPECT_TRUE(ipsec_.StartStarter());
+  EXPECT_EQ(kMockFd, ipsec_.output_fd());
+  EXPECT_EQ("ipsec[10001]: ", ipsec_.ipsec_prefix_);
+}
+
+TEST_F(IpsecManagerTestIkeV1Psk, StartStarterAlreadyRunning) {
+  InSequence unused;
+  SetStartStarterExpectations(true);
+  EXPECT_TRUE(ipsec_.StartStarter());
+  EXPECT_EQ(kMockFd, ipsec_.output_fd());
+  EXPECT_EQ("ipsec[10001]: ", ipsec_.ipsec_prefix_);
+}
+
+void IpsecManagerTestIkeV1Psk::CheckStarter(const std::string& actual) {
+  const char kExpected[] =
+      "config setup\n"
+      "\tcharonstart=no\n"
+      "conn managed\n"
+      "\tkeyexchange=ikev1\n"
+      "\tauthby=psk\n"
+      "\tpfs=no\n"
+      "\trekey=no\n"
+      "\tleft=%defaultroute\n"
+      "\tleftprotoport=17/1701\n"
+      "\tleftupdown=/usr/libexec/l2tpipsec_vpn/pluto_updown\n"
+      "\tright=1.2.3.4\n"
+      "\trightprotoport=17/1701\n"
+      "\tauto=start\n";
+  EXPECT_EQ(kExpected, actual);
+}
+
+TEST_F(IpsecManagerTestIkeV1Psk, FormatStarterConfigFile) {
+  CheckStarter(ipsec_.FormatStarterConfigFile());
+}
+
+TEST_F(IpsecManagerTestIkeV1Psk, Start) {
+  InSequence unused;
+  SetStartStarterExpectations(false);
+  EXPECT_TRUE(ipsec_.Start());
+  EXPECT_FALSE(ipsec_.start_ticks_.is_null());
+}
+
+TEST_F(IpsecManagerTestIkeV1Psk, WriteConfigFiles) {
+  EXPECT_TRUE(ipsec_.WriteConfigFiles());
+  std::string conf_contents;
+  ASSERT_TRUE(file_util::ReadFileToString(
+      stateful_container_.Append("ipsec.conf"), &conf_contents));
+  CheckStarter(conf_contents);
+  ASSERT_TRUE(file_util::PathExists(stateful_container_.Append(
+      "ipsec.secrets")));
+}
+
+class IpsecManagerTestIkeV1Certs : public IpsecManagerTest {
+ public:
+  void SetUp() {
+    IpsecManagerTest::SetUp();
+    DoInitialize(1, false);
+  }
+};
+
+TEST_F(IpsecManagerTestIkeV1Certs, Initialize) {
+  DoInitialize(1, false);
+}
+
+
+int main(int argc, char** argv) {
+  SetUpTests(&argc, argv, true);
+  return RUN_ALL_TESTS();
+}