|
@@ -0,0 +1,236 @@
|
|
|
+// Copyright (C) 2014 Internet Systems Consortium, Inc. ("ISC")
|
|
|
+//
|
|
|
+// Permission to use, copy, modify, and/or distribute this software for any
|
|
|
+// purpose with or without fee is hereby granted, provided that the above
|
|
|
+// copyright notice and this permission notice appear in all copies.
|
|
|
+//
|
|
|
+// THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
|
|
|
+// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
|
+// AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
|
+// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
|
+// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
|
|
|
+// OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
|
+// PERFORMANCE OF THIS SOFTWARE.
|
|
|
+
|
|
|
+#include <config.h>
|
|
|
+
|
|
|
+#include <config/ccsession.h>
|
|
|
+#include <dhcp/dhcp4.h>
|
|
|
+#include <dhcp6/ctrl_dhcp4_srv.h>
|
|
|
+#include <dhcpsrv/cfgmgr.h>
|
|
|
+
|
|
|
+#include <boost/scoped_ptr.hpp>
|
|
|
+#include <gtest/gtest.h>
|
|
|
+
|
|
|
+#include <fstream>
|
|
|
+#include <iostream>
|
|
|
+#include <sstream>
|
|
|
+
|
|
|
+#include <arpa/inet.h>
|
|
|
+#include <unistd.h>
|
|
|
+
|
|
|
+using namespace std;
|
|
|
+using namespace isc;
|
|
|
+using namespace isc::asiolink;
|
|
|
+using namespace isc::config;
|
|
|
+using namespace isc::data;
|
|
|
+using namespace isc::dhcp;
|
|
|
+using namespace isc::hooks;
|
|
|
+
|
|
|
+namespace {
|
|
|
+
|
|
|
+class NakedControlledDhcpv4Srv: public ControlledDhcpv4Srv {
|
|
|
+ // "Naked" DHCPv4 server, exposes internal fields
|
|
|
+public:
|
|
|
+ NakedControlledDhcpv4Srv():ControlledDhcpv4Srv(0) { }
|
|
|
+};
|
|
|
+
|
|
|
+
|
|
|
+class JSONFileBackendTest : public ::testing::Test {
|
|
|
+public:
|
|
|
+ JSONFileBackendTest() {
|
|
|
+ }
|
|
|
+
|
|
|
+ ~JSONFileBackendTest() {
|
|
|
+ static_cast<void>(unlink(TEST_FILE));
|
|
|
+ };
|
|
|
+
|
|
|
+ void writeFile(const std::string& file_name, const std::string& content) {
|
|
|
+ static_cast<void>(unlink(file_name.c_str()));
|
|
|
+
|
|
|
+ ofstream out(file_name.c_str(), ios::trunc);
|
|
|
+ EXPECT_TRUE(out.is_open());
|
|
|
+ out << content;
|
|
|
+ out.close();
|
|
|
+ }
|
|
|
+
|
|
|
+ static const char* TEST_FILE;
|
|
|
+};
|
|
|
+
|
|
|
+const char* JSONFileBackendTest::TEST_FILE = "test-config.json";
|
|
|
+
|
|
|
+// This test checks if configuration can be read from a JSON file.
|
|
|
+TEST_F(JSONFileBackendTest, jsonFile) {
|
|
|
+
|
|
|
+ // Prepare configuration file.
|
|
|
+ string config = "{ \"interfaces\": [ \"*\" ],"
|
|
|
+ "\"rebind-timer\": 2000, "
|
|
|
+ "\"renew-timer\": 1000, "
|
|
|
+ "\"subnet4\": [ { "
|
|
|
+ " \"pool\": [ \"192.0.2.1 - 192.0.2.100\" ],"
|
|
|
+ " \"subnet\": \"192.0.2.0/24\" "
|
|
|
+ " },"
|
|
|
+ " {"
|
|
|
+ " \"pool\": [ \"192.0.3.101 - 192.0.3.150\" ],"
|
|
|
+ " \"subnet\": \"192.0.3.0/24\", "
|
|
|
+ " \"id\": 0 "
|
|
|
+ " },"
|
|
|
+ " {"
|
|
|
+ " \"pool\": [ \"192.0.4.101 - 192.0.4.150\" ],"
|
|
|
+ " \"subnet\": \"192.0.4.0/24\" "
|
|
|
+ " } ],"
|
|
|
+ "\"valid-lifetime\": 4000 }";
|
|
|
+
|
|
|
+ writeFile(TEST_FILE, config);
|
|
|
+
|
|
|
+ // Now initialize the server
|
|
|
+ boost::scoped_ptr<ControlledDhcpv4Srv> srv;
|
|
|
+ ASSERT_NO_THROW(
|
|
|
+ srv.reset(new ControlledDhcpv4Srv(0))
|
|
|
+ );
|
|
|
+
|
|
|
+ // And configure it using the config file.
|
|
|
+ EXPECT_NO_THROW(srv->init(TEST_FILE));
|
|
|
+
|
|
|
+ // Now check if the configuration has been applied correctly.
|
|
|
+ const Subnet4Collection* subnets = CfgMgr::instance().getSubnets4();
|
|
|
+ ASSERT_TRUE(subnets);
|
|
|
+ ASSERT_EQ(3, subnets->size()); // We expect 3 subnets.
|
|
|
+
|
|
|
+
|
|
|
+ // Check subnet 1.
|
|
|
+ EXPECT_EQ("192.0.2.0", subnets->at(0)->get().first.toText());
|
|
|
+ EXPECT_EQ(24, subnets->at(0)->get().second);
|
|
|
+
|
|
|
+ // Check pools in the first subnet.
|
|
|
+ const PoolCollection& pools1 = subnets->at(0)->getPools(Lease::TYPE_V4);
|
|
|
+ ASSERT_EQ(1, pools1.size());
|
|
|
+ EXPECT_EQ("192.0.2.1", pools1.at(0)->getFirstAddress().toText());
|
|
|
+ EXPECT_EQ("192.0.2.100", pools1.at(0)->getLastAddress().toText());
|
|
|
+ EXPECT_EQ(Lease::TYPE_V4, pools1.at(0)->getType());
|
|
|
+
|
|
|
+ // Check subnet 2.
|
|
|
+ EXPECT_EQ("192.0.3.0", subnets->at(1)->get().first.toText());
|
|
|
+ EXPECT_EQ(24, subnets->at(1)->get().second);
|
|
|
+
|
|
|
+ // Check pools in the second subnet.
|
|
|
+ const PoolCollection& pools2 = subnets->at(1)->getPools(Lease::TYPE_V4);
|
|
|
+ ASSERT_EQ(1, pools2.size());
|
|
|
+ EXPECT_EQ("192.0.3.101", pools2.at(0)->getFirstAddress().toText());
|
|
|
+ EXPECT_EQ("192.0.3.150", pools2.at(0)->getLastAddress().toText());
|
|
|
+ EXPECT_EQ(Lease::TYPE_V4, pools2.at(0)->getType());
|
|
|
+
|
|
|
+ // And finally check subnet 3.
|
|
|
+ EXPECT_EQ("192.0.4.0", subnets->at(2)->get().first.toText());
|
|
|
+ EXPECT_EQ(24, subnets->at(2)->get().second);
|
|
|
+
|
|
|
+ // ... and it's only pool.
|
|
|
+ const PoolCollection& pools3 = subnets->at(2)->getPools(Lease::TYPE_V4);
|
|
|
+ EXPECT_EQ("192.0.4.101", pools3.at(0)->getFirstAddress().toText());
|
|
|
+ EXPECT_EQ("192.0.4.150", pools3.at(0)->getLastAddress().toText());
|
|
|
+ EXPECT_EQ(Lease::TYPE_NA, pools3.at(0)->getType());
|
|
|
+}
|
|
|
+
|
|
|
+// This test checks if configuration can be read from a JSON file.
|
|
|
+TEST_F(JSONFileBackendTest, comments) {
|
|
|
+
|
|
|
+ string config_hash_comments = "# This is a comment. It should be \n"
|
|
|
+ "#ignored. Real config starts in line below\n"
|
|
|
+ "{ \"Dhcp4\": { \"interfaces\": [ \"*\" ],"
|
|
|
+ "\"rebind-timer\": 2000, "
|
|
|
+ "\"renew-timer\": 1000, \n"
|
|
|
+ "# comments in the middle should be ignored, too\n"
|
|
|
+ "\"subnet4\": [ { "
|
|
|
+ " \"pool\": [ \"192.0.2.0/24\" ],"
|
|
|
+ " \"subnet\": \"192.0.2.0/22\" "
|
|
|
+ " } ],"
|
|
|
+ "\"valid-lifetime\": 4000 }"
|
|
|
+ "}";
|
|
|
+
|
|
|
+ /// @todo: Implement C++-style (// ...) comments
|
|
|
+ /// @todo: Implement C-style (/* ... */) comments
|
|
|
+
|
|
|
+ writeFile(TEST_FILE, config_hash_comments);
|
|
|
+
|
|
|
+ // Now initialize the server
|
|
|
+ boost::scoped_ptr<ControlledDhcpv4Srv> srv;
|
|
|
+ ASSERT_NO_THROW(
|
|
|
+ srv.reset(new ControlledDhcpv4Srv(0))
|
|
|
+ );
|
|
|
+
|
|
|
+ // And configure it using config without
|
|
|
+ EXPECT_NO_THROW(srv->init(TEST_FILE));
|
|
|
+
|
|
|
+ // Now check if the configuration has been applied correctly.
|
|
|
+ const Subnet4Collection* subnets = CfgMgr::instance().getSubnets4();
|
|
|
+ ASSERT_TRUE(subnets);
|
|
|
+ ASSERT_EQ(1, subnets->size());
|
|
|
+
|
|
|
+ // Check subnet 1.
|
|
|
+ EXPECT_EQ("192.0.2.0", subnets->at(0)->get().first.toText());
|
|
|
+ EXPECT_EQ(22, subnets->at(0)->get().second);
|
|
|
+
|
|
|
+ // Check pools in the first subnet.
|
|
|
+ const PoolCollection& pools1 = subnets->at(0)->getPools(Lease::TYPE_NA);
|
|
|
+ ASSERT_EQ(1, pools1.size());
|
|
|
+ EXPECT_EQ("192.0.2.0", pools1.at(0)->getFirstAddress().toText());
|
|
|
+ EXPECT_EQ("192.0.2.255", pools1.at(0)->getLastAddress().toText());
|
|
|
+ EXPECT_EQ(Lease::TYPE_NA, pools1.at(0)->getType());
|
|
|
+}
|
|
|
+
|
|
|
+// This test checks if configuration detects failure when trying:
|
|
|
+// - empty file
|
|
|
+// - empty filename
|
|
|
+// - no Dhcp4 element
|
|
|
+// - Config file that contains Dhcp4 but has a content error
|
|
|
+TEST_F(JSONFileBackendTest, configBroken) {
|
|
|
+
|
|
|
+ // Empty config is not allowed, because Dhcp4 element is missing
|
|
|
+ string config_empty = "";
|
|
|
+
|
|
|
+ // This config does not have mandatory Dhcp4 element
|
|
|
+ string config_v4 = "{ \"Dhcp6\": { \"interfaces\": [ \"*\" ],"
|
|
|
+ "\"preferred-lifetime\": 3000,"
|
|
|
+ "\"rebind-timer\": 2000, "
|
|
|
+ "\"renew-timer\": 1000, "
|
|
|
+ "\"subnet4\": [ { "
|
|
|
+ " \"pool\": [ \"2001:db8::/80\" ],"
|
|
|
+ " \"subnet\": \"2001:db8::/64\" "
|
|
|
+ " } ]}";
|
|
|
+
|
|
|
+ // This has Dhcp4 element, but it's utter nonsense
|
|
|
+ string config_nonsense = "{ \"Dhcp4\": { \"reviews\": \"are so much fun\" } }";
|
|
|
+
|
|
|
+ // Now initialize the server
|
|
|
+ boost::scoped_ptr<ControlledDhcpv4Srv> srv;
|
|
|
+ ASSERT_NO_THROW(
|
|
|
+ srv.reset(new ControlledDhcpv4Srv(0))
|
|
|
+ );
|
|
|
+
|
|
|
+ // Try to configure without filename. Should fail.
|
|
|
+ EXPECT_THROW(srv->init(""), BadValue);
|
|
|
+
|
|
|
+ // Try to configure it using empty file. Should fail.
|
|
|
+ writeFile(TEST_FILE, config_empty);
|
|
|
+ EXPECT_THROW(srv->init(TEST_FILE), BadValue);
|
|
|
+
|
|
|
+ // Now try to load a config that does not have Dhcp4 component.
|
|
|
+ writeFile(TEST_FILE, config_v4);
|
|
|
+ EXPECT_THROW(srv->init(TEST_FILE), BadValue);
|
|
|
+
|
|
|
+ // Now try to load a config with Dhcp4 full of nonsense.
|
|
|
+ writeFile(TEST_FILE, config_nonsense);
|
|
|
+ EXPECT_THROW(srv->init(TEST_FILE), BadValue);
|
|
|
+}
|
|
|
+
|
|
|
+} // End of anonymous namespace
|