uwflooding-sec.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 UWFLOODINGSEC_H
#define UWFLOODINGSEC_H
 
#define TTL_EQUALS_TO_ZERO \
    "TEZ" 
#include "uwflooding-hdr.h"
 
#include <uwip-module.h>
#include <uwip-clmsg.h>
#include <uwcbr-module.h>
 
#include "mphy.h"
#include "packet.h"
#include <module.h>
#include <tclcl.h>
 
#include <cmath>
#include <limits>
#include <sstream>
#include <string>
#include <iostream>
#include <rng.h>
#include <ctime>
#include <vector>
#include <fstream>
#include <map>
#include <list>
 
#include "uwbase_reputation.h"
 
 
class UwFloodingSec;
 
class NeighborReputationHandler : public TimerHandler
{
public:
 
    NeighborReputationHandler(uint8_t neigh_addr, UwFloodingSec* m, 
            double alpha_val, int dbg);
 
    virtual ~NeighborReputationHandler();
 
    bool addUnconfirmedPkt(int uid, double expire_time);
 
    bool checkUnconfirmedPkt(int uid);
 
    void updateChannelMetrics(double val_snr, double last_noise_val);
 
    void updateInstantNoise(double inst_noise_val);
 
protected:
 
    bool getNextPacket(int& uid, double& exp_time) const;
 
    void removeOldPackets();
 
    virtual void expire(Event *e);
 
    UwFloodingSec* module; 
    uint8_t neighbor_addr; 
    std::map<int, double> unconfirmed_pkts; 
    bool is_running; 
    int waiting_uid; 
    double avg_snr; 
    double alpha; 
    bool is_first_pkt; 
    double last_noise; 
    double inst_noise; 
    int debug; 
};
 
class UwFloodingSec : public Module
{
    friend class NeighborReputationHandler;
 
public:
    UwFloodingSec();
 
    virtual ~UwFloodingSec();
 
protected:
    /*****************************
     |     Internal Functions    |
     *****************************/
    virtual int command(int, const char *const *);
 
    virtual void recv(Packet *);
 
    virtual int recvSyncClMsg(ClMessage *);
 
    virtual int recvAsyncClMsg(ClMessage *);
 
    static nsaddr_t str2addr(const char *);
 
    virtual void writePathInTrace(const Packet *, const string &);
 
    static string printIP(const nsaddr_t &);
 
    void addToNeighbor(uint8_t neighbor_addr);
 
    void printNeighbor();
 
    bool checkUnconfirmedPkt(uint8_t neighbor_addr, int uid);
 
    void sendDown(Packet* p,  double delay=0);
 
    virtual void sendStatsClMsg(int neighbor_addr);
 
    void retrieveInstantNoise(int neighbor_addr);
 
private:
    // Variables
 
    uint8_t ipAddr_;
    int ttl_; 
    double maximum_cache_time_; 
    int optimize_; 
    long packets_forwarded_; 
    bool trace_path_; 
    char
            *trace_file_path_name_; 
    ofstream trace_file_path_; 
    ostringstream osstream_; 
    typedef std::map<uint16_t, double> map_packets; 
    typedef std::map<uint8_t, map_packets>
            map_forwarded_packets; 
    map_forwarded_packets
            my_forwarded_packets_; 
    std::map<uint16_t,uint8_t> ttl_traffic_map; 
    bool use_reputation; 
    typedef std::map<uint8_t, uint> neighbor_map; 
    neighbor_map neighbor; 
    typedef std::map<uint8_t, NeighborReputationHandler> neighbor_timer_map;
    neighbor_timer_map neighbor_tmr; 
    double fwd_to; 
    uint8_t prev_hop_temp; 
    UwReputationInterface* reputation; 
    double alpha_snr; 
    bool valid_phy_id; 
    int stats_phy_id; 
    UwFloodingSec(const UwFloodingSec &);
 
    uint8_t getTTL(Packet* p) const;
};
 
#endif // UWFLOODING_H