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:
73
tests/active_tests/include/active_test.h
Normal file
73
tests/active_tests/include/active_test.h
Normal 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
|
||||
68
tests/active_tests/include/active_test_runner.h
Normal file
68
tests/active_tests/include/active_test_runner.h
Normal 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
|
||||
52
tests/active_tests/include/configuration.h
Normal file
52
tests/active_tests/include/configuration.h
Normal 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
|
||||
59
tests/active_tests/include/ipv4_tests.h
Normal file
59
tests/active_tests/include/ipv4_tests.h
Normal 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
|
||||
62
tests/active_tests/include/packet_capturer.h
Normal file
62
tests/active_tests/include/packet_capturer.h
Normal 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
|
||||
Reference in New Issue
Block a user