NetworkConfigurator.cpp

//
// This file is part of the LOTI distribution (https://github.com/levy/loti/).
// Copyright (c) 2018 Levente Mészáros.
//
// 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, version 3.
//
// 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, see <http://www.gnu.org/licenses/>.
//

#include "inet/networklayer/common/L3AddressResolver.h"
#include "inet/networklayer/ipv4/Ipv4InterfaceData.h"
#include "NetworkConfigurator.h"

namespace loti {

Define_Module(NetworkConfigurator);

void NetworkConfigurator::initialize(int stage)
{
    if (stage == INITSTAGE_STATIC_ROUTING) {
        cTopology topology;
        L3AddressResolver resolver;
        topology.extractByProperty("networkNode");
        for (int i = 0; i < topology.getNumNodes(); i++) {
            auto destinationNode = topology.getNode(i);
            auto destinationDaemon = findDaemon(destinationNode->getModule());
            networkNodes.push_back(destinationNode->getModule());
            topology.calculateUnweightedSingleShortestPathsTo(destinationNode);
            for (int j = 0; j < topology.getNumNodes(); j++) {
                auto sourceNode = topology.getNode(j);
                if (destinationNode == sourceNode)
                    continue;
                if (sourceNode->getNumPaths() == 0)
                    continue;
                auto sourceDaemon = findDaemon(sourceNode->getModule());
                auto linkOut = sourceNode->getPath(0);
                auto nextHopModule = linkOut->getRemoteNode()->getModule();
                auto nextHopDaemon = findDaemon(nextHopModule);
                auto nextHopInterfaceTable = resolver.findInterfaceTableOf(nextHopModule);
                auto nextHopInterfaceEntry = nextHopInterfaceTable->findInterfaceByNodeInputGateId(linkOut->getRemoteGateId());
                auto nextHopIpAddress = nextHopInterfaceEntry->getProtocolData<Ipv4InterfaceData>()->getIPAddress();
                sourceDaemon->learnRoute(destinationDaemon->nodeId(), nextHopDaemon->nodeId());
                sourceDaemon->addNeighbor(nextHopDaemon->nodeId(), nextHopIpAddress);
            }
        }
        WATCH_VECTOR(networkNodes);
    }
}

Daemon *NetworkConfigurator::findDaemon(const cModule *networkNode)
{
    return check_and_cast<Daemon *>(networkNode->getSubmodule("app", 0));
}

}