fuzz.h

Go to the documentation of this file.

// Copyright (C) 2016-2019  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 FUZZ_H
#define FUZZ_H
 
#ifdef ENABLE_AFL
 
#include <exceptions/exceptions.h>
 
#include <arpa/inet.h>
#include <net/if.h>
#include <sys/socket.h>
#include <unistd.h>
 
#include <condition_variable>
#include <mutex>
#include <string>
#include <thread>
 
namespace isc {
    
 
 
class Fuzz {
public:
    static constexpr size_t BUFFER_SIZE = 256 * 1024;
 
    static constexpr size_t MAX_SEND_SIZE = 64000;
 
    static constexpr long MAX_LOOP_COUNT = 1000;
 
 
    Fuzz(int ipversion, uint16_t port);
 
    ~Fuzz();
 
    void transfer(void) const;
 
    long maxLoopCount() const {
        return loop_max_;
    }
 
private:
    void createAddressStructures(int ipversion, const char* interface,
                                 const char* address, uint16_t port);
 
    // 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 FuzzInitFail : public Exception {
public:
    FuzzInitFail(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) { };
};
 
}
 
#endif // ENABLE_AFL
 
#endif // FUZZ_H