BootstrapServer.cpp

Go to the documentation of this file.

/*
 * Copyright (C) 2006-2020 Istituto Italiano di Tecnologia (IIT)
 * Copyright (C) 2006-2010 RobotCub Consortium
 * All rights reserved.
 *
 * This software may be modified and distributed under the terms of the
 * BSD-3-Clause license. See the accompanying LICENSE file for details.
 */
 
#include <yarp/conf/system.h>
 
#include <yarp/os/DummyConnector.h>
#include <yarp/os/impl/FallbackNameServer.h>
#include <yarp/os/impl/NameServer.h>
#include <yarp/os/impl/NameConfig.h>
 
#include <yarp/name/NameServerConnectionHandler.h>
 
#include <yarp/name/BootstrapServer.h>
 
using namespace yarp::os;
using namespace yarp::name;
using namespace yarp::os::impl;
 
class BootstrapServerAdapter : public NameServerStub {
private:
    FallbackNameServer *fallback;
    NameServerConnectionHandler handler;
public:
    BootstrapServerAdapter(NameService& owner) : handler(&owner) {
        fallback = new FallbackNameServer(*this);
        if (fallback==nullptr) {
            fprintf(stderr,"Cannot allocate ServerAdapter\n");
            ::exit(1);
        }
    }
 
    virtual ~BootstrapServerAdapter() {
        if (fallback!=nullptr) delete fallback;
        fallback = nullptr;
    }
 
    std::string apply(const std::string& txt, const Contact& remote) override {
        DummyConnector con, con2;
        con.setTextMode(true);
        ConnectionWriter& writer = con.getWriter();
        writer.appendText(txt.c_str());
        bool ok = handler.apply(con.getReader(),&(con2.getWriter()));
        std::string result;
        if (ok) {
            result = con2.getReader().expectText();
        }
        printf("ASKED %s, GAVE %s\n", txt.c_str(), result.c_str());
        return result;
    }
 
    bool start() {
        return fallback->start();
    }
 
    bool stop() {
        fallback->close();
        return true;
    }
};
 
BootstrapServer::BootstrapServer(NameService& owner) {
 
    implementation = new BootstrapServerAdapter(owner);
    if (implementation==nullptr) {
        fprintf(stderr,"Cannot allocate ServerAdapter\n");
        ::exit(1);
    }
}
 
BootstrapServer::~BootstrapServer() {
    if (implementation!=nullptr) {
        delete ((BootstrapServerAdapter*)implementation);
        implementation = nullptr;
    }
 
}
 
bool BootstrapServer::start() {
    if (implementation!=nullptr) {
        return ((BootstrapServerAdapter*)implementation)->start();
    }
    return false;
}
 
bool BootstrapServer::stop() {
    if (implementation!=nullptr) {
        return ((BootstrapServerAdapter*)implementation)->stop();
    }
    return false;
}
 
 
bool BootstrapServer::configFileBootstrap(yarp::os::Contact& contact,
                                          bool configFileRequired,
                                          bool mayEditConfigFile) {
    Contact suggest = contact;
 
    // see what address is lying around
    Contact prev;
    NameConfig conf;
    if (conf.fromFile()) {
        prev = conf.getAddress();
    } else if (configFileRequired) {
        fprintf(stderr,"Could not read configuration file %s\n",
                conf.getConfigFileName().c_str());
        return false;
    }
 
    // merge
    if (prev.isValid()) {
        if (suggest.getHost() == "...") {
            suggest.setHost(prev.getHost());
        }
        if (suggest.getCarrier() == "...") {
            suggest.setCarrier(prev.getCarrier());
        }
        if (suggest.getPort() == 0) {
            suggest.setPort(prev.getPort());
        }
    }
 
    if (suggest.getRegName() == "...") {
        suggest.setName(conf.getNamespace());
    }
 
    // still something not set?
    if (suggest.getPort() == 0) {
        suggest.setPort(10000);
    }
    if (suggest.getHost() == "...") {
        // should get my IP
        suggest.setHost(conf.getHostName());
    }
 
    if (!configFileRequired)  {
        // finally, should make sure IP is local, and if not, correct it
        if (!conf.isLocalName(suggest.getHost())) {
            fprintf(stderr,"Overriding non-local address for name server\n");
            suggest.setHost(conf.getHostName());
        } else {
            // Let's just check we're not a loopback
            std::string betterHost = conf.getHostName(false,suggest.getHost());
            if (betterHost!=suggest.getHost()) {
                fprintf(stderr,"Overriding loopback address for name server\n");
                suggest.setHost(betterHost);
            }
        }
    }
 
    bool changed = false;
    if (prev.isValid()) {
        changed = (prev.getHost() != suggest.getHost()) ||
            (prev.getPort() != suggest.getPort()) ||
            (conf.getMode() != "yarp" && conf.getMode() != "");
    }
    if (changed && !mayEditConfigFile) {
        fprintf(stderr,"PROBLEM: need to change settings in %s\n",
                conf.getConfigFileName().c_str());
        fprintf(stderr,"  Current settings: host %s port %d family %s\n",
                prev.getHost().c_str(), prev.getPort(),
                (conf.getMode()=="")?"yarp":conf.getMode().c_str());
        fprintf(stderr,"  Desired settings:  host %s port %d family %s\n",
                suggest.getHost().c_str(), suggest.getPort(), "yarp");
        fprintf(stderr,"Please specify '--write' if it is ok to overwrite current settings, or\n");
        fprintf(stderr,"Please specify '--read' to use the current settings, or\n");
        fprintf(stderr,"delete %s\n", conf.getConfigFileName().c_str());
        return false;
    }
    bool shouldSave = changed || !prev.isValid();
 
    if (shouldSave) {
        // and save
        conf.setAddress(suggest);
        if (!conf.toFile()) {
            fprintf(stderr,"Could not save configuration file %s\n",
                    conf.getConfigFileName().c_str());
        }
    }
 
    contact = suggest;
    return true;
}
 
 
Contact BootstrapServer::where() {
    Contact addr = FallbackNameServer::getAddress();
    addr.setName("fallback");
    return addr;
}
 
//#endif