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;
};
}