uwudp-module.h

Go to the documentation of this file.

//
// Copyright (c) 2017 Regents of the SIGNET lab, University of Padova.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
// 1. Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
// 2. 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.
// 3. Neither the name of the University of Padova (SIGNET lab) nor the
//    names of its contributors may be used to endorse or promote products
//    derived from this software without specific prior written permission.
//
// 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 _UWUDP_H_
#define _UWUDP_H_
 
#include <sstream>
#include <uwip-module.h>
 
#include <map>
#include <module.h>
#include <set>
 
#define DROP_UNKNOWN_PORT_NUMBER \
    "UPN" 
#define DROP_RECEIVED_DUPLICATED_PACKET \
    "RDP" 
#define HDR_UWUDP(P) (hdr_uwudp::access(P))
 
extern packet_t PT_UWUDP;
 
typedef std::map<uint8_t, std::set<int>> map_packets_el;
 
typedef struct hdr_uwudp {
    uint8_t sport_; 
    uint8_t dport_; 
    static int offset_; 
    static int &
    offset()
    {
        return offset_;
    }
 
    static struct hdr_uwudp *
    access(const Packet *p)
    {
        return (struct hdr_uwudp *) p->access(offset_);
    }
 
    uint8_t &
    sport()
    {
        return sport_;
    }
 
    uint8_t &
    dport()
    {
        return dport_;
    }
} hdr_uwudp;
 
class UwUdp : public Module
{
public:
    UwUdp();
 
    virtual ~UwUdp() = default;
 
    virtual void recv(Packet *) override;
 
    virtual void recv(Packet *, int) override;
 
    virtual int command(int, const char *const *) override;
 
    virtual int assignPort(Module *);
 
    void
    printIdsPkts() const
    {
        std::cout << "UWUDP packets IDs:" << std::endl;
        std::cout << "PT_UWUDP: \t\t" << PT_UWUDP << std::endl;
    }
 
protected:
    uint16_t portcounter; 
    int drop_duplicated_packets_; 
    int debug_; 
    std::map<int, int> port_map; 
    std::map<int, int> id_map; 
    std::map<uint16_t, map_packets_el>
            map_packets; 
    static int
    getUdpHeaderSize()
    {
        return sizeof(hdr_uwudp);
    }
};
 
#endif // _UWUDP_H_