source: nscp/modules/NSClientServer/NSClientServer.h

/**************************************************************************
*   Copyright (C) 2004-2007 by Michael Medin <michael@medin.name>         *
*                                                                         *
*   This code is part of NSClient++ - http://trac.nakednuns.org/nscp      *
*                                                                         *
*   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; either version 2 of the License, or     *
*   (at your option) any later version.                                   *
*                                                                         *
*   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, write to the                         *
*   Free Software Foundation, Inc.,                                       *
*   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
***************************************************************************/
#include <socket_helpers.hpp>
#include <check_nt/server/protocol.hpp>

class NSClientServer : public nscapi::impl::simple_plugin, public check_nt::server::handler {
public:
        NSClientServer();
        virtual ~NSClientServer();
        // Module calls
        bool loadModuleEx(std::string alias, NSCAPI::moduleLoadMode mode);
        bool unloadModule();

        check_nt::packet handle(check_nt::packet packet);

        check_nt::packet create_error(std::wstring msg) {
                return check_nt::packet("ERROR: Failed to parse");
        }

        void log_debug(std::string module, std::string file, int line, std::string msg) const {
                if (GET_CORE()->should_log(NSCAPI::log_level::debug)) {
                        GET_CORE()->log(NSCAPI::log_level::debug, file, line, msg);
                }
        }
        void log_error(std::string module, std::string file, int line, std::string msg) const {
                if (GET_CORE()->should_log(NSCAPI::log_level::error)) {
                        GET_CORE()->log(NSCAPI::log_level::error, file, line, msg);
                }
        }

        std::string get_password() const {
                return password_;
        }

private:
        void set_password(std::string password) {
                password_ = password;
        }
        virtual void set_allow_arguments(bool v)  {
                allowArgs_ = v;
        }
        virtual void set_allow_nasty_arguments(bool v) {
                allowNasty_ = v;
        }
        virtual void set_perf_data(bool v) {
                noPerfData_ = !v;
        }
        bool isPasswordOk(std::string remotePassword);

private:
        bool noPerfData_;
        bool allowNasty_;
        bool allowArgs_;

        socket_helpers::connection_info info_;
        boost::shared_ptr<check_nt::server::server> server_;
        std::string password_;

};