NA_IPv4.h

Go to the documentation of this file.

//
// Copyright (C) 2013, NESG (Network Engineering and Security Group), http://nesg.ugr.es,
// - Gabriel Maciá Fernández (gmacia@ugr.es)
// - Leovigildo Sánchez Casado (sancale@ugr.es)
// - Rafael A. Rodríguez Gómez (rodgom@ugr.es)
// - Roberto Magán Carrión (rmagan@ugr.es)
// - Pedro García Teodoro (pgteodor@ugr.es)
// - José Camacho Páez (josecamacho@ugr.es)
// - Jesús E. Díaz Verdejo (jedv@ugr.es)
//
// This file is part of NETA.
//
//    NETA is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    NETA is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with NETA.  If not, see <http://www.gnu.org/licenses/>.
//

#ifndef NA_IPV4_H_
#define NA_IPV4_H_

#include <IPv4.h>
#include <omnetpp.h>
#include "common/log/NA_NesgLog.h"
#include "NA_HackedModule.h"
#include "NA_DroppingMessage_m.h"
#include "NA_DelayMessage_m.h"

#define TCP_DATA "tcp"

#define UDP_DATA "UDP"

#define PING_DATA "ping"

class INET_API NA_IPv4 : public IPv4, public NA_HackedModule {

private:

    NA_NesgLog log;

    /*----------------- DROPPING ATTACK  -------------------------*/
    bool droppingAttackIsActive;

    double droppingAttackProbability;

    long numDrops;

    /*----------------- DELAY ATTACK  -------------------------*/
    bool delayAttackIsActive;

    double delayAttackProbability;

    long numDelays;

    long totalDelayTime;

    ParPtr delayAttackValue;

    /*---------------------OTHERs-------------------------------*/
    long numRecvPacket;

public:

    void handleMessageFromAttackController(cMessage *msg);

protected:

    virtual void initialize();

    /*----------------- DROPPING ATTACK  -------------------------*/
    static simsignal_t dropSignal;

    virtual void handlePacketFromNetwork(IPv4Datagram *datagram, InterfaceEntry *fromIE);

    /*----------------- DELAY ATTACK  -------------------------*/
    virtual simtime_t startService(cPacket *msg);

    static simsignal_t delaySignal;

    /*------------------OTHERs ---------------------------------*/
    static simsignal_t rcvdPktSignal;

};

#endif /* NA_IPV4_H_ */