RemoteControl.hpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac

    This file is part of Leosac.

    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Leosac 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 Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once

#include "RemoteControlSecurity.hpp"
#include "core/config/ConfigManager.hpp"
#include <boost/property_tree/ptree_fwd.hpp>
#include <zmqpp/zmqpp.hpp>

namespace Leosac
{

class Kernel;

class RemoteControl
{
  public:
    RemoteControl(zmqpp::context &ctx, Kernel &kernel,
                  const boost::property_tree::ptree &cfg);

    void update();

  private:
    bool handle_module_config(zmqpp::message *msg_in, zmqpp::message *msg_out);

    bool handle_module_list(zmqpp::message *msg_in, zmqpp::message *msg_out);

    bool handle_sync_from(zmqpp::message *msg_in, zmqpp::message *msg_out);

    bool handle_save(zmqpp::message *msg_in, zmqpp::message *msg_out);

    bool handle_general_config(zmqpp::message *msg_in, zmqpp::message *msg_out);

    bool handle_config_version(zmqpp::message *msg_in, zmqpp::message *msg_out);

    void module_list(zmqpp::message *message_out);

    void module_config(const std::string &module,
                       ConfigManager::ConfigFormat cfg_format,
                       zmqpp::message *message_out);

    void general_config(ConfigManager::ConfigFormat cfg_format,
                        zmqpp::message *msg_out);

    void handle_msg();

    void process_config(const boost::property_tree::ptree &cfg);

    bool gather_remote_module_list(zmqpp::socket &s,
                                   std::list<std::string> &remote_modules);

    bool receive_remote_config(zmqpp::socket &s, std::map<std::string, bool> &cfg);

    bool gather_remote_config(zmqpp::socket &s, std::list<std::string> &start_list,
                              std::list<std::string> &stop_list);

    Kernel &kernel_;

    zmqpp::socket socket_;

    std::string public_key_;

    std::string secret_key_;

    zmqpp::auth auth_;
    zmqpp::context &context_;

    // Function is not really void (), we use placeholder and bind.
    using CommandHandlerMap =
        std::map<std::string, std::function<bool(zmqpp::message *msg_in,
                                                 zmqpp::message *msg_out)>>;

    CommandHandlerMap command_handlers_;

    RemoteControlSecurity security_;

    std::string current_client_idt_;

    // Allow kernel full access to this class.
    friend class Kernel;
};
}