SimpleTCP.h

Go to the documentation of this file.

//
// Copyright (C) 2010 Institut fuer Telematik, Karlsruher Institut fuer Technologie (KIT)
//
// This program 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 2
// of the License, or (at your option) any later version.
//
// This program 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 this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
//

#ifndef __SIMPLETCP_H
#define __SIMPLETCP_H

//#include <map>
//#include <set>
#include <omnetpp.h>
#include "IPvXAddress.h"
#include <TCPConnection.h>
#include <InitStages.h>

class SimpleNodeEntry;


struct StatisticsAndDelay {

public:

    // delay fault type string and corresponding map for switch..case
    std::string delayFaultTypeString;
    enum delayFaultTypeNum {
        delayFaultUndefined,
        delayFaultLiveAll,
        delayFaultLivePlanetlab,
        delayFaultSimulation
    };
    std::map<std::string, delayFaultTypeNum> delayFaultTypeMap;

    // statistics
    int numSent; 
    int numQueueLost; 
    int numPartitionLost; 
    int numDestUnavailableLost; 
    simtime_t delay; 
    simtime_t constantDelay; 
    bool useCoordinateBasedDelay; 
    double jitter; 
    bool faultyDelay; 
    GlobalNodeList* globalNodeList; 
    GlobalStatistics* globalStatistics; 
    SimpleNodeEntry* nodeEntry; 
};

class SimpleTCPConnection : public TCPConnection {

public:
    SimpleTCPConnection():TCPConnection(){};
    SimpleTCPConnection(TCP* mod, int appGateIndex, int connId):TCPConnection(mod, appGateIndex, connId){};

    virtual void sendToIP(TCPSegment *tcpseg);

    static StatisticsAndDelay sad;

    void sendRst(uint32 seq, IPvXAddress src, IPvXAddress dest, int srcPort, int destPort);
    void sendRstAck(uint32 seq, uint32 ack, IPvXAddress src, IPvXAddress dest, int srcPort, int destPort);

protected:
    virtual void sendToIP(TCPSegment *tcpseg, IPvXAddress src, IPvXAddress dest);

    SimpleTCPConnection *cloneListeningConnection();

};

class SimpleTCP : public TCP {

public:
  SimpleTCP() { sad.globalStatistics = NULL; };
  virtual ~SimpleTCP() {};

  void setNodeEntry (SimpleNodeEntry* entry);

  StatisticsAndDelay sad;

protected:
  // utility methods
  void segmentArrivalWhileClosed(TCPSegment *tcpseg, IPvXAddress srcAddr, IPvXAddress destAddr);
  SimpleTCPConnection *createConnection(int appGateIndex, int connId);
  virtual void initialize(int stage);
  virtual void handleMessage(cMessage *msg);

  virtual int numInitStages() const
  {
      return MAX_STAGE_UNDERLAY + 1;
  }

  void finish();
};

#endif