|
@@ -0,0 +1,273 @@
|
|
|
+// Copyright (C) 2011 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 <iostream>
|
|
|
+#include <sstream>
|
|
|
+#include <arpa/inet.h>
|
|
|
+#include <gtest/gtest.h>
|
|
|
+#include <asiolink/io_address.h>
|
|
|
+#include <dhcp/dhcp4.h>
|
|
|
+#include <dhcp/option.h>
|
|
|
+#include <dhcp/option4_addrlst.h>
|
|
|
+#include <util/buffer.h>
|
|
|
+
|
|
|
+using namespace std;
|
|
|
+using namespace isc;
|
|
|
+using namespace isc::dhcp;
|
|
|
+using namespace isc::asiolink;
|
|
|
+using namespace isc::util;
|
|
|
+
|
|
|
+namespace {
|
|
|
+
|
|
|
+// a sample data (list of 4 addresses)
|
|
|
+const uint8_t sampledata[] = {
|
|
|
+ 192, 0, 2, 3, // 192.0.2.3
|
|
|
+ 255, 255, 255, 0, // 255.255.255.0 - popular netmask
|
|
|
+ 0, 0, 0 , 0, // used for default routes or (any address)
|
|
|
+ 127, 0, 0, 1 // loopback
|
|
|
+};
|
|
|
+
|
|
|
+// expected on-wire format for an option with 1 address
|
|
|
+const uint8_t expected1[] = { // 1 address
|
|
|
+ DHO_DOMAIN_NAME_SERVERS, 4, // type, length
|
|
|
+ 192, 0, 2, 3, // 192.0.2.3
|
|
|
+};
|
|
|
+
|
|
|
+// expected on-wire format for an option with 4 addresses
|
|
|
+const uint8_t expected4[] = { // 4 addresses
|
|
|
+ 254, 16, // type = 254, len = 16
|
|
|
+ 192, 0, 2, 3, // 192.0.2.3
|
|
|
+ 255, 255, 255, 0, // 255.255.255.0 - popular netmask
|
|
|
+ 0, 0, 0 ,0, // used for default routes or (any address)
|
|
|
+ 127, 0, 0, 1 // loopback
|
|
|
+};
|
|
|
+
|
|
|
+class Option4AddrLstTest : public ::testing::Test {
|
|
|
+protected:
|
|
|
+
|
|
|
+ Option4AddrLstTest():
|
|
|
+ vec_(vector<uint8_t>(300,0)) // 300 bytes long filled with 0s
|
|
|
+ {
|
|
|
+ sampleAddrs_.push_back(IOAddress("192.0.2.3"));
|
|
|
+ sampleAddrs_.push_back(IOAddress("255.255.255.0"));
|
|
|
+ sampleAddrs_.push_back(IOAddress("0.0.0.0"));
|
|
|
+ sampleAddrs_.push_back(IOAddress("127.0.0.1"));
|
|
|
+ }
|
|
|
+
|
|
|
+ vector<uint8_t> vec_;
|
|
|
+ Option4AddrLst::AddressContainer sampleAddrs_;
|
|
|
+
|
|
|
+};
|
|
|
+
|
|
|
+TEST_F(Option4AddrLstTest, parse1) {
|
|
|
+
|
|
|
+ memcpy(&vec_[0], sampledata, sizeof(sampledata));
|
|
|
+
|
|
|
+ // just one address
|
|
|
+ Option4AddrLst* opt1 = 0;
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt1 = new Option4AddrLst(DHO_DOMAIN_NAME_SERVERS,
|
|
|
+ vec_.begin(),
|
|
|
+ vec_.begin()+4);
|
|
|
+ // use just first address (4 bytes), not the whole
|
|
|
+ // sampledata
|
|
|
+ );
|
|
|
+
|
|
|
+ EXPECT_EQ(Option::V4, opt1->getUniverse());
|
|
|
+
|
|
|
+ EXPECT_EQ(DHO_DOMAIN_NAME_SERVERS, opt1->getType());
|
|
|
+ EXPECT_EQ(6, opt1->len()); // 2 (header) + 4 (1x IPv4 addr)
|
|
|
+
|
|
|
+ Option4AddrLst::AddressContainer addrs = opt1->getAddresses();
|
|
|
+ ASSERT_EQ(1, addrs.size());
|
|
|
+
|
|
|
+ EXPECT_EQ("192.0.2.3", addrs[0].toText());
|
|
|
+
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ delete opt1;
|
|
|
+ opt1 = 0;
|
|
|
+ );
|
|
|
+
|
|
|
+ // 1 address
|
|
|
+}
|
|
|
+
|
|
|
+TEST_F(Option4AddrLstTest, parse4) {
|
|
|
+
|
|
|
+ vector<uint8_t> buffer(300,0); // 300 bytes long filled with 0s
|
|
|
+
|
|
|
+ memcpy(&buffer[0], sampledata, sizeof(sampledata));
|
|
|
+
|
|
|
+ // 4 addresses
|
|
|
+ Option4AddrLst* opt4 = 0;
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt4 = new Option4AddrLst(254,
|
|
|
+ buffer.begin(),
|
|
|
+ buffer.begin()+sizeof(sampledata));
|
|
|
+ );
|
|
|
+
|
|
|
+ EXPECT_EQ(Option::V4, opt4->getUniverse());
|
|
|
+
|
|
|
+ EXPECT_EQ(254, opt4->getType());
|
|
|
+ EXPECT_EQ(18, opt4->len()); // 2 (header) + 16 (4x IPv4 addrs)
|
|
|
+
|
|
|
+ Option4AddrLst::AddressContainer addrs = opt4->getAddresses();
|
|
|
+ ASSERT_EQ(4, addrs.size());
|
|
|
+
|
|
|
+ EXPECT_EQ("192.0.2.3", addrs[0].toText());
|
|
|
+ EXPECT_EQ("255.255.255.0", addrs[1].toText());
|
|
|
+ EXPECT_EQ("0.0.0.0", addrs[2].toText());
|
|
|
+ EXPECT_EQ("127.0.0.1", addrs[3].toText());
|
|
|
+
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ delete opt4;
|
|
|
+ opt4 = 0;
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+TEST_F(Option4AddrLstTest, assembly1) {
|
|
|
+
|
|
|
+ Option4AddrLst* opt = 0;
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt = new Option4AddrLst(DHO_DOMAIN_NAME_SERVERS, IOAddress("192.0.2.3"));
|
|
|
+ );
|
|
|
+ EXPECT_EQ(Option::V4, opt->getUniverse());
|
|
|
+ EXPECT_EQ(DHO_DOMAIN_NAME_SERVERS, opt->getType());
|
|
|
+
|
|
|
+ Option4AddrLst::AddressContainer addrs = opt->getAddresses();
|
|
|
+ ASSERT_EQ(1, addrs.size() );
|
|
|
+ EXPECT_EQ("192.0.2.3", addrs[0].toText());
|
|
|
+
|
|
|
+ OutputBuffer buf(100);
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt->pack4(buf);
|
|
|
+ );
|
|
|
+
|
|
|
+ ASSERT_EQ(6, opt->len());
|
|
|
+ ASSERT_EQ(6, buf.getLength());
|
|
|
+
|
|
|
+ EXPECT_EQ(0, memcmp(expected1, buf.getData(), 6));
|
|
|
+
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ delete opt;
|
|
|
+ opt = 0;
|
|
|
+ );
|
|
|
+
|
|
|
+ // This is old-fashioned option. We don't serve IPv6 types here!
|
|
|
+ EXPECT_THROW(
|
|
|
+ opt = new Option4AddrLst(DHO_DOMAIN_NAME_SERVERS, IOAddress("2001:db8::1")),
|
|
|
+ BadValue
|
|
|
+ );
|
|
|
+ if (opt) {
|
|
|
+ // test failed. Execption was not thrown, but option was created instead.
|
|
|
+ delete opt;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+TEST_F(Option4AddrLstTest, assembly4) {
|
|
|
+
|
|
|
+
|
|
|
+ Option4AddrLst* opt = 0;
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt = new Option4AddrLst(254, sampleAddrs_);
|
|
|
+ );
|
|
|
+ EXPECT_EQ(Option::V4, opt->getUniverse());
|
|
|
+ EXPECT_EQ(254, opt->getType());
|
|
|
+
|
|
|
+ Option4AddrLst::AddressContainer addrs = opt->getAddresses();
|
|
|
+ ASSERT_EQ(4, addrs.size() );
|
|
|
+ EXPECT_EQ("192.0.2.3", addrs[0].toText());
|
|
|
+ EXPECT_EQ("255.255.255.0", addrs[1].toText());
|
|
|
+ EXPECT_EQ("0.0.0.0", addrs[2].toText());
|
|
|
+ EXPECT_EQ("127.0.0.1", addrs[3].toText());
|
|
|
+
|
|
|
+ OutputBuffer buf(100);
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt->pack4(buf);
|
|
|
+ );
|
|
|
+
|
|
|
+ ASSERT_EQ(18, opt->len()); // 2(header) + 4xsizeof(IPv4addr)
|
|
|
+ ASSERT_EQ(18, buf.getLength());
|
|
|
+
|
|
|
+ ASSERT_EQ(0, memcmp(expected4, buf.getData(), 18));
|
|
|
+
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ delete opt;
|
|
|
+ opt = 0;
|
|
|
+ );
|
|
|
+
|
|
|
+ // This is old-fashioned option. We don't serve IPv6 types here!
|
|
|
+ sampleAddrs_.push_back(IOAddress("2001:db8::1"));
|
|
|
+ EXPECT_THROW(
|
|
|
+ opt = new Option4AddrLst(DHO_DOMAIN_NAME_SERVERS, sampleAddrs_),
|
|
|
+ BadValue
|
|
|
+ );
|
|
|
+ if (opt) {
|
|
|
+ // test failed. Execption was not thrown, but option was created instead.
|
|
|
+ delete opt;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+TEST_F(Option4AddrLstTest, setAddress) {
|
|
|
+ Option4AddrLst* opt = 0;
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt = new Option4AddrLst(123, IOAddress("1.2.3.4"));
|
|
|
+ );
|
|
|
+ opt->setAddress(IOAddress("192.0.255.255"));
|
|
|
+
|
|
|
+ Option4AddrLst::AddressContainer addrs = opt->getAddresses();
|
|
|
+ ASSERT_EQ(1, addrs.size() );
|
|
|
+ EXPECT_EQ("192.0.255.255", addrs[0].toText());
|
|
|
+
|
|
|
+ // We should accept IPv4-only addresses.
|
|
|
+ EXPECT_THROW(
|
|
|
+ opt->setAddress(IOAddress("2001:db8::1")),
|
|
|
+ BadValue
|
|
|
+ );
|
|
|
+
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ delete opt;
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+TEST_F(Option4AddrLstTest, setAddresses) {
|
|
|
+
|
|
|
+ Option4AddrLst* opt = 0;
|
|
|
+
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ opt = new Option4AddrLst(123); // empty list
|
|
|
+ );
|
|
|
+
|
|
|
+ opt->setAddresses(sampleAddrs_);
|
|
|
+
|
|
|
+ Option4AddrLst::AddressContainer addrs = opt->getAddresses();
|
|
|
+ ASSERT_EQ(4, addrs.size() );
|
|
|
+ EXPECT_EQ("192.0.2.3", addrs[0].toText());
|
|
|
+ EXPECT_EQ("255.255.255.0", addrs[1].toText());
|
|
|
+ EXPECT_EQ("0.0.0.0", addrs[2].toText());
|
|
|
+ EXPECT_EQ("127.0.0.1", addrs[3].toText());
|
|
|
+
|
|
|
+ // We should accept IPv4-only addresses.
|
|
|
+ sampleAddrs_.push_back(IOAddress("2001:db8::1"));
|
|
|
+ EXPECT_THROW(
|
|
|
+ opt->setAddresses(sampleAddrs_),
|
|
|
+ BadValue
|
|
|
+ );
|
|
|
+
|
|
|
+ EXPECT_NO_THROW(
|
|
|
+ delete opt;
|
|
|
+ );
|
|
|
+}
|
|
|
+
|
|
|
+} // namespace
|