Open Source Projects Europe

#include <SimpleDV/SimpleDV.h>

namespace SimpleDV {

Register_Class(SimpleDV);

using namespace std;


RoutingUpdate::RoutingUpdate(const Address &_addr, const std::string &_src, const unsigned short &_qos):IntRoutingUpdate(_addr){
    src = _src;
    qos = _qos;
}

std::string RoutingUpdate::getSrc(){
    return src;
}
unsigned short RoutingUpdate::getQoS(){
    return qos;
}

void RoutingUpdate::addEntry(rtEntry entry){
    entries.push_back(entry);
}

entriesIt RoutingUpdate::entriesBegin(){
    return entries.begin();
}
entriesIt RoutingUpdate::entriesEnd(){
    return entries.end();
}


//Flow inserted/removed
void SimpleDV::insertFlow(const Address &addr, const std::string &dst, const unsigned short &qos, const unsigned short &metric){

    neig[qos][addr] = metric;

    rtEntry * oldEntry = &table[qos][dst];

    bool entryChangedDst = false;

    if(oldEntry->addr == dst){
        if(oldEntry->metric != metric){
            oldEntry->metric = metric;
        }
    } else if(oldEntry->metric >= metric){
        oldEntry->addr = dst;
        oldEntry->metric = metric;
        entryChangedDst = true;
    }

    if(oldEntry->metric >= infMetric){
        changes.insert(entries2NextItem(qosPaddr(qos, dst),""));
        table[qos].erase(dst);
    } else if(entryChangedDst){
        changes.insert(entries2NextItem(qosPaddr(qos, dst),dst));
    }
}
void SimpleDV::removeFlow(const Address &addr, const std::string &dst, const unsigned short &qos){
    neig[qos].erase(addr);
    if(neig[qos].size() <= 0){
        neig.erase(qos);
    }

    tTable * qosTable = &table[qos];

    for(tTableIt it = qosTable->begin(); it != qosTable->end();){
        rtEntry * oldEntry = &it->second;
        tTableIt actIt = it;
        it++;

        if(oldEntry->addr == dst){
            changes.insert(entries2NextItem(qosPaddr(qos, oldEntry->addr),""));
            qosTable->erase(actIt);
        }
    }
}

//Get Changes
entries2Next SimpleDV::getChanges(){
    entries2Next ret = changes;
    changes.clear();
    return ret;
}

entries2Next SimpleDV::getAll(){
    entries2Next ret;

    for(rtTableIt it = table.begin(); it != table.end(); it++){
        for(tTableIt it2 = it->second.begin(); it2 != it->second.end(); it2++){
            if(it2->second.metric < infMetric) {
                ret[qosPaddr(it->first, it2->first)] = it2->second.addr;
            } else {
                ret[qosPaddr(it->first, it2->first)] = "";
            }
        }
    }
    map<unsigned short, map<string, rtEntry> > table;

    return ret;
}

//Process a Routing Update, return true => inform FWDG of the update
bool SimpleDV::processUpdate(IntRoutingUpdate * update){
    RoutingUpdate * up = check_and_cast<RoutingUpdate*>(update);

    std::string src = up->getSrc();
    if(src == myAddr){
        return false;
    }

    unsigned short qos = up->getQoS();

    for(entriesIt it = up->entriesBegin(); it!= up->entriesEnd(); it++){
        if(it->addr == myAddr) {
            continue;
        }

        rtEntry * newEntry = &(*it);
        rtEntry * oldEntry = &table[qos][newEntry->addr];

        bool entryChangedDst = false;

        if(oldEntry->addr == newEntry->addr){
            if(oldEntry->metric == newEntry->metric){
                oldEntry->metric = newEntry->metric;
            }
        } else if(oldEntry->metric >= newEntry->metric){
            oldEntry->addr = src;
            oldEntry->metric = newEntry->metric;
            entryChangedDst = true;
        }

        if(oldEntry->metric >= infMetric){
            changes.insert(entries2NextItem(qosPaddr(qos, newEntry->addr),""));
            table[qos].erase(newEntry->addr);
        } else if(entryChangedDst){
            changes.insert(entries2NextItem(qosPaddr(qos, newEntry->addr),src));
        }
    }
    return ! changes.empty();
}

// Called after initialize
void SimpleDV::onPolicyInit(){
    myAddr = par("myAddr").stdstringValue();
    if(myAddr == "") {
        myAddr = myAddress.getIpcAddress().getName();
    }

    infMetric = 32;

    scheduleAt(simTime()+30, new cMessage("Time2Update"));
}


void SimpleDV::handleMessage(cMessage *msg){
    if(msg->isSelfMessage()){
        //Iterate all Qos
        for(qosNeighMetricIt it = neig.begin(); it!= neig.end(); it++){
            //getTable per QoS
            tTable * qosTable = &table[it->first];

            //iterate all Qos->Neighbour
            for(neighMetricIt it2 = it->second.begin(); it2 != it->second.end(); it2++){
                //New Update to QoS + Neighbour
                RoutingUpdate * update = new RoutingUpdate(it2->first, myAddr, it->first);

                //Populate the update
                for(tTableIt it3 = qosTable->begin(); it3 != qosTable->end();it3++){
                    update->addEntry(rtEntry(it3->first, it3->second.metric+ it2->second));
                }

                //Add our entry
                update->addEntry(rtEntry(myAddr, it2->second));

                //And send
                sendUpdate(update);
            }
        }

        scheduleAt(simTime()+30, msg);
    }
}


void SimpleDV::finish(){
    IntRouting::finish();

    EV << "I'm " << myAddr<<endl;

    for (rtTableIt it = table.begin(); it != table.end(); it++)
    {
        EV << "  QoS " << it->first << endl;
        for (tTableIt it2 = it->second.begin(); it2 != it->second.end(); it2++)
        {
            EV << "    " << it2->first << " via " << it2->second.addr << " ("
                    << it2->second.metric << " hops)" << endl;
        }
    }

}

}