|
@@ -21,9 +21,7 @@
|
|
#include <gtest/gtest.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
|
|
#include <dhcp/dhcp4.h>
|
|
#include <dhcp/dhcp4.h>
|
|
-#include <dhcp4/dhcp4_srv.h>
|
|
|
|
-#include <dhcp/option.h>
|
|
|
|
-#include <asiolink/io_address.h>
|
|
|
|
|
|
+#include <dhcp4/ctrl_dhcp4_srv.h>
|
|
|
|
|
|
using namespace std;
|
|
using namespace std;
|
|
using namespace isc;
|
|
using namespace isc;
|
|
@@ -31,244 +29,30 @@ using namespace isc::dhcp;
|
|
using namespace isc::asiolink;
|
|
using namespace isc::asiolink;
|
|
|
|
|
|
namespace {
|
|
namespace {
|
|
-const char* const INTERFACE_FILE = "interfaces.txt";
|
|
|
|
|
|
|
|
-class NakedDhcpv4Srv: public Dhcpv4Srv {
|
|
|
|
|
|
+class NakedControlledDhcpv4Srv: public ControlledDhcpv4Srv {
|
|
// "naked" DHCPv4 server, exposes internal fields
|
|
// "naked" DHCPv4 server, exposes internal fields
|
|
public:
|
|
public:
|
|
- NakedDhcpv4Srv():Dhcpv4Srv(DHCP4_SERVER_PORT + 10000) { }
|
|
|
|
-
|
|
|
|
- boost::shared_ptr<Pkt4> processDiscover(boost::shared_ptr<Pkt4>& discover) {
|
|
|
|
- return Dhcpv4Srv::processDiscover(discover);
|
|
|
|
- }
|
|
|
|
- boost::shared_ptr<Pkt4> processRequest(boost::shared_ptr<Pkt4>& request) {
|
|
|
|
- return Dhcpv4Srv::processRequest(request);
|
|
|
|
- }
|
|
|
|
- void processRelease(boost::shared_ptr<Pkt4>& release) {
|
|
|
|
- return Dhcpv4Srv::processRelease(release);
|
|
|
|
- }
|
|
|
|
- void processDecline(boost::shared_ptr<Pkt4>& decline) {
|
|
|
|
- Dhcpv4Srv::processDecline(decline);
|
|
|
|
- }
|
|
|
|
- boost::shared_ptr<Pkt4> processInform(boost::shared_ptr<Pkt4>& inform) {
|
|
|
|
- return Dhcpv4Srv::processInform(inform);
|
|
|
|
- }
|
|
|
|
|
|
+ NakedControlledDhcpv4Srv():ControlledDhcpv4Srv(DHCP4_SERVER_PORT + 10000) { }
|
|
};
|
|
};
|
|
|
|
|
|
-class Dhcpv4SrvTest : public ::testing::Test {
|
|
|
|
|
|
+class CtrlDhcpv4SrvTest : public ::testing::Test {
|
|
public:
|
|
public:
|
|
- Dhcpv4SrvTest() {
|
|
|
|
- unlink(INTERFACE_FILE);
|
|
|
|
- fstream fakeifaces(INTERFACE_FILE, ios::out | ios::trunc);
|
|
|
|
- if (if_nametoindex("lo") > 0) {
|
|
|
|
- fakeifaces << "lo 127.0.0.1";
|
|
|
|
- } else if (if_nametoindex("lo0") > 0) {
|
|
|
|
- fakeifaces << "lo0 127.0.0.1";
|
|
|
|
- }
|
|
|
|
- fakeifaces.close();
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- void MessageCheck(const boost::shared_ptr<Pkt4>& q,
|
|
|
|
- const boost::shared_ptr<Pkt4>& a) {
|
|
|
|
- ASSERT_TRUE(q);
|
|
|
|
- ASSERT_TRUE(a);
|
|
|
|
-
|
|
|
|
- EXPECT_EQ(q->getHops(), a->getHops());
|
|
|
|
- EXPECT_EQ(q->getIface(), a->getIface());
|
|
|
|
- EXPECT_EQ(q->getIndex(), a->getIndex());
|
|
|
|
- EXPECT_EQ(q->getGiaddr(), a->getGiaddr());
|
|
|
|
-
|
|
|
|
- // check that bare minimum of required options are there
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_SUBNET_MASK));
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_ROUTERS));
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_DHCP_SERVER_IDENTIFIER));
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_DHCP_LEASE_TIME));
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_SUBNET_MASK));
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_ROUTERS));
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_DOMAIN_NAME));
|
|
|
|
- EXPECT_TRUE(a->getOption(DHO_DOMAIN_NAME_SERVERS));
|
|
|
|
-
|
|
|
|
- // check that something is offered
|
|
|
|
- EXPECT_TRUE(a->getYiaddr().toText() != "0.0.0.0");
|
|
|
|
|
|
+ CtrlDhcpv4SrvTest() {
|
|
}
|
|
}
|
|
|
|
|
|
- ~Dhcpv4SrvTest() {
|
|
|
|
- unlink(INTERFACE_FILE);
|
|
|
|
|
|
+ ~CtrlDhcpv4SrvTest() {
|
|
};
|
|
};
|
|
};
|
|
};
|
|
|
|
|
|
-TEST_F(Dhcpv4SrvTest, basic) {
|
|
|
|
- // nothing to test. DHCPv4_srv instance is created
|
|
|
|
- // in test fixture. It is destroyed in destructor
|
|
|
|
|
|
+TEST_F(CtrlDhcpv4SrvTest, basic) {
|
|
|
|
|
|
- Dhcpv4Srv* srv = NULL;
|
|
|
|
|
|
+ ControlledDhcpv4Srv* srv = NULL;
|
|
ASSERT_NO_THROW({
|
|
ASSERT_NO_THROW({
|
|
- srv = new Dhcpv4Srv(DHCP4_SERVER_PORT + 10000);
|
|
|
|
|
|
+ srv = new ControlledDhcpv4Srv(DHCP4_SERVER_PORT + 10000);
|
|
});
|
|
});
|
|
|
|
|
|
delete srv;
|
|
delete srv;
|
|
}
|
|
}
|
|
|
|
|
|
-TEST_F(Dhcpv4SrvTest, processDiscover) {
|
|
|
|
- NakedDhcpv4Srv* srv = new NakedDhcpv4Srv();
|
|
|
|
- vector<uint8_t> mac(6);
|
|
|
|
- for (int i = 0; i < 6; i++) {
|
|
|
|
- mac[i] = 255 - i;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- boost::shared_ptr<Pkt4> pkt(new Pkt4(DHCPDISCOVER, 1234));
|
|
|
|
- boost::shared_ptr<Pkt4> offer;
|
|
|
|
-
|
|
|
|
- pkt->setIface("eth0");
|
|
|
|
- pkt->setIndex(17);
|
|
|
|
- pkt->setHWAddr(1, 6, mac);
|
|
|
|
- pkt->setRemoteAddr(IOAddress("192.0.2.56"));
|
|
|
|
- pkt->setGiaddr(IOAddress("192.0.2.67"));
|
|
|
|
-
|
|
|
|
- // let's make it a relayed message
|
|
|
|
- pkt->setHops(3);
|
|
|
|
- pkt->setRemotePort(DHCP4_SERVER_PORT);
|
|
|
|
-
|
|
|
|
- // should not throw
|
|
|
|
- EXPECT_NO_THROW(
|
|
|
|
- offer = srv->processDiscover(pkt);
|
|
|
|
- );
|
|
|
|
-
|
|
|
|
- // should return something
|
|
|
|
- ASSERT_TRUE(offer);
|
|
|
|
-
|
|
|
|
- EXPECT_EQ(DHCPOFFER, offer->getType());
|
|
|
|
-
|
|
|
|
- // this is relayed message. It should be sent back to relay address.
|
|
|
|
- EXPECT_EQ(pkt->getGiaddr(), offer->getRemoteAddr());
|
|
|
|
-
|
|
|
|
- MessageCheck(pkt, offer);
|
|
|
|
-
|
|
|
|
- // now repeat the test for directly sent message
|
|
|
|
- pkt->setHops(0);
|
|
|
|
- pkt->setGiaddr(IOAddress("0.0.0.0"));
|
|
|
|
- pkt->setRemotePort(DHCP4_CLIENT_PORT);
|
|
|
|
-
|
|
|
|
- EXPECT_NO_THROW(
|
|
|
|
- offer = srv->processDiscover(pkt);
|
|
|
|
- );
|
|
|
|
-
|
|
|
|
- // should return something
|
|
|
|
- ASSERT_TRUE(offer);
|
|
|
|
-
|
|
|
|
- EXPECT_EQ(DHCPOFFER, offer->getType());
|
|
|
|
-
|
|
|
|
- // this is direct message. It should be sent back to origin, not
|
|
|
|
- // to relay.
|
|
|
|
- EXPECT_EQ(pkt->getRemoteAddr(), offer->getRemoteAddr());
|
|
|
|
-
|
|
|
|
- MessageCheck(pkt, offer);
|
|
|
|
-
|
|
|
|
- delete srv;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-TEST_F(Dhcpv4SrvTest, processRequest) {
|
|
|
|
- NakedDhcpv4Srv* srv = new NakedDhcpv4Srv();
|
|
|
|
- vector<uint8_t> mac(6);
|
|
|
|
- for (int i = 0; i < 6; i++) {
|
|
|
|
- mac[i] = i*10;
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- boost::shared_ptr<Pkt4> req(new Pkt4(DHCPREQUEST, 1234));
|
|
|
|
- boost::shared_ptr<Pkt4> ack;
|
|
|
|
-
|
|
|
|
- req->setIface("eth0");
|
|
|
|
- req->setIndex(17);
|
|
|
|
- req->setHWAddr(1, 6, mac);
|
|
|
|
- req->setRemoteAddr(IOAddress("192.0.2.56"));
|
|
|
|
- req->setGiaddr(IOAddress("192.0.2.67"));
|
|
|
|
-
|
|
|
|
- // should not throw
|
|
|
|
- ASSERT_NO_THROW(
|
|
|
|
- ack = srv->processRequest(req);
|
|
|
|
- );
|
|
|
|
-
|
|
|
|
- // should return something
|
|
|
|
- ASSERT_TRUE(ack);
|
|
|
|
-
|
|
|
|
- EXPECT_EQ(DHCPACK, ack->getType());
|
|
|
|
-
|
|
|
|
- // this is relayed message. It should be sent back to relay address.
|
|
|
|
- EXPECT_EQ(req->getGiaddr(), ack->getRemoteAddr());
|
|
|
|
-
|
|
|
|
- MessageCheck(req, ack);
|
|
|
|
-
|
|
|
|
- // now repeat the test for directly sent message
|
|
|
|
- req->setHops(0);
|
|
|
|
- req->setGiaddr(IOAddress("0.0.0.0"));
|
|
|
|
- req->setRemotePort(DHCP4_CLIENT_PORT);
|
|
|
|
-
|
|
|
|
- EXPECT_NO_THROW(
|
|
|
|
- ack = srv->processDiscover(req);
|
|
|
|
- );
|
|
|
|
-
|
|
|
|
- // should return something
|
|
|
|
- ASSERT_TRUE(ack);
|
|
|
|
-
|
|
|
|
- EXPECT_EQ(DHCPOFFER, ack->getType());
|
|
|
|
-
|
|
|
|
- // this is direct message. It should be sent back to origin, not
|
|
|
|
- // to relay.
|
|
|
|
- EXPECT_EQ(ack->getRemoteAddr(), req->getRemoteAddr());
|
|
|
|
-
|
|
|
|
- MessageCheck(req, ack);
|
|
|
|
-
|
|
|
|
- delete srv;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-TEST_F(Dhcpv4SrvTest, processRelease) {
|
|
|
|
- NakedDhcpv4Srv* srv = new NakedDhcpv4Srv();
|
|
|
|
-
|
|
|
|
- boost::shared_ptr<Pkt4> pkt(new Pkt4(DHCPRELEASE, 1234));
|
|
|
|
-
|
|
|
|
- // should not throw
|
|
|
|
- EXPECT_NO_THROW(
|
|
|
|
- srv->processRelease(pkt);
|
|
|
|
- );
|
|
|
|
-
|
|
|
|
- // TODO: Implement more reasonable tests before starting
|
|
|
|
- // work on processSomething() method.
|
|
|
|
-
|
|
|
|
- delete srv;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-TEST_F(Dhcpv4SrvTest, processDecline) {
|
|
|
|
- NakedDhcpv4Srv* srv = new NakedDhcpv4Srv();
|
|
|
|
-
|
|
|
|
- boost::shared_ptr<Pkt4> pkt(new Pkt4(DHCPDECLINE, 1234));
|
|
|
|
-
|
|
|
|
- // should not throw
|
|
|
|
- EXPECT_NO_THROW(
|
|
|
|
- srv->processDecline(pkt);
|
|
|
|
- );
|
|
|
|
-
|
|
|
|
- // TODO: Implement more reasonable tests before starting
|
|
|
|
- // work on processSomething() method.
|
|
|
|
- delete srv;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-TEST_F(Dhcpv4SrvTest, processInform) {
|
|
|
|
- NakedDhcpv4Srv* srv = new NakedDhcpv4Srv();
|
|
|
|
-
|
|
|
|
- boost::shared_ptr<Pkt4> pkt(new Pkt4(DHCPINFORM, 1234));
|
|
|
|
-
|
|
|
|
- // should not throw
|
|
|
|
- EXPECT_NO_THROW(
|
|
|
|
- srv->processInform(pkt);
|
|
|
|
- );
|
|
|
|
-
|
|
|
|
- // should return something
|
|
|
|
- EXPECT_TRUE(srv->processInform(pkt));
|
|
|
|
-
|
|
|
|
- // TODO: Implement more reasonable tests before starting
|
|
|
|
- // work on processSomething() method.
|
|
|
|
-
|
|
|
|
- delete srv;
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
} // end of anonymous namespace
|
|
} // end of anonymous namespace
|