packet_fuzzer.h

Go to the documentation of this file.

// Copyright (C) 2016-2024 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#ifndef DHCPSRV_PACKET_FUZZER_H
#define DHCPSRV_PACKET_FUZZER_H
 
#ifdef FUZZING
 
#include <asiolink/io_address.h>
#include <exceptions/exceptions.h>
 
#include <arpa/inet.h>
#include <net/if.h>
#include <sys/socket.h>
#include <unistd.h>
 
#include <string>
 
namespace isc {
 
 
class PacketFuzzer {
public:
    static constexpr size_t BUFFER_SIZE = 256 * 1024;
 
    static constexpr size_t MAX_SEND_SIZE = 64000;
 
    static constexpr long MAX_LOOP_COUNT = 1000;
 
    PacketFuzzer(uint16_t const port,
                 std::string const interface,
                 std::string const address);
 
    ~PacketFuzzer();
 
    void transfer() const;
    void transfer(uint8_t const* data, size_t size) const;
 
    long maxLoopCount() const {
        return loop_max_;
    }
 
private:
    void createAddressStructures(uint16_t const port,
                                 std::string const& interface,
                                 isc::asiolink::IOAddress const& io_address);
 
    // Other member variables.
    long                loop_max_;      //< Maximum number of loop iterations
    size_t              sockaddr_len_;  //< Length of the structure
    struct sockaddr*    sockaddr_ptr_;  //< Pointer to structure used
    struct sockaddr_in  servaddr4_;     //< IPv6 address information
    struct sockaddr_in6 servaddr6_;     //< IPv6 address information
    int                 sockfd_;        //< Socket used to transfer data
};  // class PacketFuzzer
 
class FuzzInitFail : public Exception {
public:
    FuzzInitFail(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) { }
};  // class FuzzInitFail
 
}  // namespace isc
 
#endif  // FUZZING
 
#endif  // DHCPSRV_PACKET_FUZZER_H