BaseTcpSupport.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 BASETCPSUPPORT_H_
#define BASETCPSUPPORT_H_

#include <omnetpp.h>
#include <map>
#include <TransportAddress.h>
#include <TCPSocket.h>
#include <ExtTCPSocketMap.h>

class BaseTcpSupport : public TCPSocket::CallbackInterface
{
public:

    // Event codes
    enum EvCode {NO_EST_CONNECTION, PEER_CLOSED, PEER_TIMEDOUT, PEER_REFUSED, CONNECTION_RESET, CONNECTION_SUCC_ClOSED};

    // Utility methods of the CallbackInterface
    virtual void socketDataArrived(int connId, void *yourPtr, cPacket *msg, bool urgent);
    virtual void socketEstablished(int connId, void *yourPtr);
    virtual void socketPeerClosed(int connId, void *yourPtr);
//    virtual void socketClosed(int connId, void *yourPtr);
    virtual void socketFailure(int connId, void *yourPtr, int code);
    virtual void socketStatusArrived(int connId, void *yourPtr, TCPStatusInfo *status) {delete status;}

protected:

    void handleTCPMessage(cMessage* msg);

    void bindAndListenTcp(int port);

    bool isAlreadyConnected(TransportAddress address);

    void establishTcpConnection(TransportAddress address);

    void sendTcpData(cPacket* msg, TransportAddress address);

    virtual void handleConnectionEvent(EvCode code, TransportAddress address);

    virtual void handleDataReceived(TransportAddress address, cPacket* msg, bool urgent);

    virtual void handleIncomingConnection(TransportAddress address);

    void closeTcpConnection(TransportAddress address);

    void setTcpOut(cGate* gate) {tcpOut = gate;}

    cGate* getTcpOut() {return tcpOut;}

private:

    ExtTCPSocketMap sockets;  
    typedef std::vector<cMessage*> msgQueue;
    typedef std::map<TransportAddress, msgQueue*> transQueue;
    transQueue queuedTx; 

    cGate* tcpOut; 
};

#endif /* BASETCPSUPPORT_H_ */