1
0
mirror of https://github.com/mfontanini/libtins synced 2026-01-30 05:24:26 +01:00

Add active tests for IPv4

This commit is contained in:
Matias Fontanini
2016-03-20 16:27:43 -07:00
parent a70ce10bed
commit 5dc7b20a43
15 changed files with 865 additions and 2 deletions

View File

@@ -0,0 +1,73 @@
/*
* Copyright (c) 2016, Matias Fontanini
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#ifndef TINS_ACTIVE_TEST_H
#define TINS_ACTIVE_TEST_H
#include <memory>
#include <stdexcept>
#include "tins/packet_sender.h"
#include "configuration.h"
namespace Tins {
class PDU;
} // Tins
class TestFailed : public std::runtime_error {
public:
using std::runtime_error::runtime_error;
};
class ActiveTest {
public:
using PacketSenderPtr = std::shared_ptr<Tins::PacketSender>;
using ConfigurationPtr = std::shared_ptr<Configuration>;
ActiveTest(const PacketSenderPtr& packet_sender,
const ConfigurationPtr& configuration);
virtual ~ActiveTest() = default;
void execute();
virtual std::string name() const = 0;
bool matches_packet(const Tins::PDU& pdu) const;
virtual void validate_packet(const Tins::PDU& pdu) = 0;
protected:
const PacketSenderPtr& packet_sender() const;
const ConfigurationPtr& configuration() const;
virtual bool test_matches_packet(const Tins::PDU& pdu) const = 0;
virtual void execute_test() = 0;
private:
PacketSenderPtr packet_sender_;
ConfigurationPtr configuration_;
};
#endif // TINS_ACTIVE_TEST_H

View File

@@ -0,0 +1,68 @@
/*
* Copyright (c) 2016, Matias Fontanini
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#ifndef TINS_ACTIVE_TEST_RUNNER_H
#define TINS_ACTIVE_TEST_RUNNER_H
#include <vector>
#include <memory>
#include "active_test.h"
#include "configuration.h"
#include "packet_capturer.h"
class ActiveTestRunner {
public:
ActiveTestRunner(const Configuration& configuration);
template <typename T>
void add_test();
bool validate_tests();
void run();
private:
using ConfigurationPtr = ActiveTest::ConfigurationPtr;
using ActiveTestPtr = std::unique_ptr<ActiveTest>;
template <typename... EmptyTail>
void add_tests() { }
void do_run();
ConfigurationPtr configuration_;
ActiveTest::PacketSenderPtr packet_sender_;
PacketCapturer capturer_;
std::vector<ActiveTestPtr> tests_;
};
template <typename T>
void ActiveTestRunner::add_test() {
tests_.emplace_back(new T(packet_sender_, configuration_));
}
#endif // TINS_ACTIVE_TEST_RUNNER_H

View File

@@ -0,0 +1,52 @@
/*
* Copyright (c) 2016, Matias Fontanini
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#ifndef TINS_ACTIVE_TEST_CONFIGURATION_H
#define TINS_ACTIVE_TEST_CONFIGURATION_H
#include <cstdint>
#include <string>
#include "tins/network_interface.h"
class Configuration {
public:
void interface(const Tins::NetworkInterface& interface);
void source_port(uint16_t value);
void destination_port(uint16_t value);
const Tins::NetworkInterface& interface() const;
uint16_t source_port() const;
uint16_t destination_port() const;
private:
Tins::NetworkInterface interface_;
uint16_t source_port_ = 0;
uint16_t destination_port_ = 0;
};
#endif // TINS_ACTIVE_TEST_CONFIGURATION_H

View File

@@ -0,0 +1,59 @@
/*
* Copyright (c) 2016, Matias Fontanini
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#ifndef TINS_IPV4_TESTS_H
#define TINS_IPV4_TESTS_H
#include <cstdint>
#include <string>
#include "active_test.h"
class IPv4SourceAddressTest : public ActiveTest {
public:
using ActiveTest::ActiveTest;
std::string name() const;
private:
void execute_test();
void validate_packet(const Tins::PDU& pdu);
bool test_matches_packet(const Tins::PDU& pdu) const;
};
class IPv4FragmentationTest : public ActiveTest {
public:
using ActiveTest::ActiveTest;
std::string name() const;
private:
void execute_test();
void validate_packet(const Tins::PDU& pdu);
bool test_matches_packet(const Tins::PDU& pdu) const;
};
#endif // TINS_IPV4_TESTS_H

View File

@@ -0,0 +1,62 @@
/*
* Copyright (c) 2016, Matias Fontanini
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following disclaimer
* in the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
#ifndef TINS_PACKET_CAPTURER_H
#define TINS_PACKET_CAPTURER_H
#include <string>
#include <memory>
#include <atomic>
#include <thread>
#include <vector>
#include "tins/sniffer.h"
#include "configuration.h"
class PacketCapturer {
public:
using PacketPtr = std::unique_ptr<Tins::PDU>;
using PacketStorage = std::vector<PacketPtr>;
PacketCapturer(const Configuration& configuration);
void start_capture();
void stop_capture();
PacketStorage captured_packets();
private:
bool callback(const Tins::PDU& pdu);
std::string make_filter(const Configuration& configuration) const;
std::unique_ptr<Tins::Sniffer> sniffer_;
std::thread sniffer_thread_;
PacketStorage storage_;
std::atomic<bool> running_;
};
#endif // TINS_PACKET_CAPTURER_H