TestApplLayer.cc

/***************************************************************************
 * file:        TestApplLayer.cc
 *
 * author:      Daniel Willkomm
 *
 * copyright:   (C) 2004 Telecommunication Networks Group (TKN) at
 *              Technische Universitaet Berlin, Germany.
 *
 *              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.
 *              For further information see file COPYING
 *              in the top level directory
 ***************************************************************************
 * part of:     framework implementation developed by tkn
 * description: application layer: test class for the application layer
 ***************************************************************************/


#include "TestApplLayer.h"
#include "NetwControlInfo.h"

#include <SimpleAddress.h>


void TestApplLayer::initialize(int stage)
{
    BaseApplLayer::initialize(stage);
    if(stage == 0) {
      hasPar("coreDebug") ? coreDebug = par("coreDebug").boolValue() : coreDebug = false;
        delayTimer = new cMessage( "delay-timer", SEND_BROADCAST_TIMER );
    }
    else if(stage==1) {
        scheduleAt(simTime() + dblrand() * 10, delayTimer);
    }
}


void TestApplLayer::handleLowerMsg( cMessage* msg )
{
    ApplPkt *m;
    switch( msg->getKind() ){
    case BROADCAST_MESSAGE:
        m = static_cast<ApplPkt *>(msg);
        coreEV << "Received a broadcast packet from host["<<m->getSrcAddr()<<"] -> sending reply\n";
        sendReply(m);
        break;
    case BROADCAST_REPLY_MESSAGE:
        m = static_cast<ApplPkt *>(msg);
        coreEV << "Received reply from host["<<m->getSrcAddr()<<"]; delete msg\n";
        delete msg;
  break;
    default:
  EV <<"Error! got packet with unknown kind: " << msg->getKind()<<endl;
        delete msg;
    }
}

void TestApplLayer::handleSelfMsg(cMessage *msg) {
    switch( msg->getKind() ){
    case SEND_BROADCAST_TIMER:
        sendBroadcast();
    delete msg;
    delayTimer = NULL;
  break;
    default:
      EV << "Unknown selfmessage! -> delete, kind: "<<msg->getKind() <<endl;
  delete msg;
    }
}

void TestApplLayer::sendBroadcast()
{
    ApplPkt *pkt = new ApplPkt("BROADCAST_MESSAGE", BROADCAST_MESSAGE);
    pkt->setDestAddr(-1);
    // we use the host modules getIndex() as a appl address
    pkt->setSrcAddr( myApplAddr() );
    pkt->setBitLength(headerLength);

    // set the control info to tell the network layer the layer 3
    // address;
    pkt->setControlInfo( new NetwControlInfo(L3BROADCAST) );

    coreEV << "Sending broadcast packet!\n";
    sendDown( pkt );
}

void TestApplLayer::sendReply(ApplPkt *msg)
{
    simtime_t delay;

    delay = uniform(0, 0.01);

    msg->setDestAddr(msg->getSrcAddr());
    msg->setSrcAddr(myApplAddr());
    msg->setKind(BROADCAST_REPLY_MESSAGE);
    msg->setName("BROADCAST_REPLY_MESSAGE");
    sendDelayedDown(msg, delay);

    coreEV << "sent message with delay " << delay << endl;

    //NOTE: the NetwControl info was already ste by the network layer
    //and stays the same
}

TestApplLayer::~TestApplLayer()
{
  cancelAndDelete(delayTimer);
}