inet::TCPEchoApp Class Reference

Accepts any number of incoming connections, and sends back whatever arrives on them. More...

#include <TCPEchoApp.h>

Inheritance diagram for inet::TCPEchoApp:

Public Member Functions
	TCPEchoApp ()
Public Member Functions inherited from inet::ILifecycle
virtual	~ILifecycle ()

Protected Member Functions
virtual bool	isNodeUp ()
virtual void	sendDown (cMessage *msg)
virtual void	startListening ()
virtual void	stopListening ()
virtual void	initialize (int stage) override
virtual int	numInitStages () const override
virtual void	handleMessage (cMessage *msg) override
virtual void	finish () override
virtual void	refreshDisplay () const override
virtual bool	handleOperationStage (LifecycleOperation *operation, int stage, IDoneCallback *doneCallback) override
	Perform one stage of a lifecycle operation. More...

Protected Attributes
simtime_t	delay
double	echoFactor = NaN
TCPSocket	socket
NodeStatus *	nodeStatus = nullptr
long	bytesRcvd = 0
long	bytesSent = 0

Static Protected Attributes
static simsignal_t	rcvdPkSignal = registerSignal("rcvdPk")
static simsignal_t	sentPkSignal = registerSignal("sentPk")

Detailed Description

Accepts any number of incoming connections, and sends back whatever arrives on them.

Constructor & Destructor Documentation

inet::TCPEchoApp::TCPEchoApp ( )

inline

{}

Member Function Documentation

void inet::TCPEchoApp::finish ( )

overrideprotectedvirtual

{
    recordScalar("bytesRcvd", bytesRcvd);
    recordScalar("bytesSent", bytesSent);
}

void inet::TCPEchoApp::handleMessage ( cMessage *  msg )

overrideprotectedvirtual

{
    if (!isNodeUp())
        throw cRuntimeError("Application is not running");
    if (msg->isSelfMessage()) {
        sendDown(msg);
    }
    else if (msg->getKind() == TCP_I_PEER_CLOSED) {
        // we'll close too
        msg->setName("close");
        msg->setKind(TCP_C_CLOSE);

        if (delay == 0)
            sendDown(msg);
        else
            scheduleAt(simTime() + delay, msg); // send after a delay
    }
    else if (msg->getKind() == TCP_I_DATA || msg->getKind() == TCP_I_URGENT_DATA) {
        cPacket *pkt = check_and_cast<cPacket *>(msg);
        emit(rcvdPkSignal, pkt);
        bytesRcvd += pkt->getByteLength();

        if (echoFactor == 0) {
            delete pkt;
        }
        else {
            // reverse direction, modify length, and send it back
            pkt->setKind(TCP_C_SEND);
            TCPCommand *ind = check_and_cast<TCPCommand *>(pkt->removeControlInfo());
            TCPSendCommand *cmd = new TCPSendCommand();
            cmd->setConnId(ind->getConnId());
            pkt->setControlInfo(cmd);
            delete ind;

            long byteLen = pkt->getByteLength() * echoFactor;

            if (byteLen < 1)
                byteLen = 1;

            pkt->setByteLength(byteLen);

            RawPacket *baMsg = dynamic_cast<RawPacket *>(pkt);

            // if (dataTransferMode == TCP_TRANSFER_BYTESTREAM)
            if (baMsg) {
                ByteArray& outdata = baMsg->getByteArray();
                ByteArray indata = outdata;
                outdata.setDataArraySize(byteLen);

                for (long i = 0; i < byteLen; i++)
                    outdata.setData(i, indata.getData(i / echoFactor));
            }

            if (delay == 0)
                sendDown(pkt);
            else
                scheduleAt(simTime() + delay, pkt); // send after a delay
        }
    }
    else {
        // some indication -- ignore
        delete msg;
    }
}

bool inet::TCPEchoApp::handleOperationStage ( LifecycleOperation *  operation, int  stage, IDoneCallback *  doneCallback  )

overrideprotectedvirtual

Perform one stage of a lifecycle operation.

Processing may be done entirely within this method, or may be a longer process that involves nonzero simulation time or several events, and is triggered by this method call.

Return value: true = "done"; false = "not yet done, will invoke doneCallback when done"

Implements inet::ILifecycle.

{
    Enter_Method_Silent();
    if (dynamic_cast<NodeStartOperation *>(operation)) {
        if ((NodeStartOperation::Stage)stage == NodeStartOperation::STAGE_APPLICATION_LAYER)
            startListening();
    }
    else if (dynamic_cast<NodeShutdownOperation *>(operation)) {
        if ((NodeShutdownOperation::Stage)stage == NodeShutdownOperation::STAGE_APPLICATION_LAYER)
            // TODO: wait until socket is closed
            stopListening();
    }
    else if (dynamic_cast<NodeCrashOperation *>(operation))
        ;
    else
        throw cRuntimeError("Unsupported lifecycle operation '%s'", operation->getClassName());
    return true;
}

void inet::TCPEchoApp::initialize ( int  stage )

overrideprotectedvirtual

{
    cSimpleModule::initialize(stage);

    if (stage == INITSTAGE_LOCAL) {
        delay = par("echoDelay");
        echoFactor = par("echoFactor");

        bytesRcvd = bytesSent = 0;
        WATCH(bytesRcvd);
        WATCH(bytesSent);

        socket.setOutputGate(gate("tcpOut"));
        socket.readDataTransferModePar(*this);

        nodeStatus = dynamic_cast<NodeStatus *>(findContainingNode(this)->getSubmodule("status"));
    }
    else if (stage == INITSTAGE_APPLICATION_LAYER) {
        if (isNodeUp())
            startListening();
    }
}

bool inet::TCPEchoApp::isNodeUp ( )

protectedvirtual

Referenced by handleMessage(), and initialize().

{
    return !nodeStatus || nodeStatus->getState() == NodeStatus::UP;
}

virtual int inet::TCPEchoApp::numInitStages ( ) const

inlineoverrideprotectedvirtual

{ return NUM_INIT_STAGES; }

void inet::TCPEchoApp::refreshDisplay ( ) const

overrideprotectedvirtual

{
    char buf[80];
    sprintf(buf, "rcvd: %ld bytes\nsent: %ld bytes", bytesRcvd, bytesSent);
    getDisplayString().setTagArg("t", 0, buf);
}

void inet::TCPEchoApp::sendDown ( cMessage *  msg )

protectedvirtual

Referenced by handleMessage().

{
    if (msg->isPacket()) {
        bytesSent += ((cPacket *)msg)->getByteLength();
        emit(sentPkSignal, (cPacket *)msg);
    }

    send(msg, "tcpOut");
}

void inet::TCPEchoApp::startListening ( )

protectedvirtual

Referenced by handleOperationStage(), and initialize().

{
    const char *localAddress = par("localAddress");
    int localPort = par("localPort");
    socket.renewSocket();
    socket.bind(localAddress[0] ? L3Address(localAddress) : L3Address(), localPort);
    socket.listen();
}

void inet::TCPEchoApp::stopListening ( )

protectedvirtual

Referenced by handleOperationStage().

{
    socket.close();
}

Member Data Documentation

long inet::TCPEchoApp::bytesRcvd = 0

protected

Referenced by finish(), handleMessage(), initialize(), and refreshDisplay().

long inet::TCPEchoApp::bytesSent = 0

protected

Referenced by finish(), initialize(), refreshDisplay(), and sendDown().

simtime_t inet::TCPEchoApp::delay

protected

Referenced by handleMessage(), and initialize().

double inet::TCPEchoApp::echoFactor = NaN

protected

Referenced by handleMessage(), and initialize().

NodeStatus* inet::TCPEchoApp::nodeStatus = nullptr

protected

Referenced by initialize(), and isNodeUp().

simsignal_t inet::TCPEchoApp::rcvdPkSignal = registerSignal("rcvdPk")

staticprotected

Referenced by handleMessage().

simsignal_t inet::TCPEchoApp::sentPkSignal = registerSignal("sentPk")

staticprotected

Referenced by sendDown().

TCPSocket inet::TCPEchoApp::socket

protected

Referenced by initialize(), startListening(), and stopListening().

---

The documentation for this class was generated from the following files:

• TCPEchoApp.h
• TCPEchoApp.cc