source: nscp/NSCPlugin.h @ 75d5e70

#pragma once

#include <NSCAPI.h>
#include <NSCHelper.h>
#include <sstream>

/**
 * @ingroup NSClient++
 * Exception class for the NSCPlugin class.
 * When an unexpected error occurs in NSCPlugin this exception is thrown.
 *
 * @version 1.0
 * first version
 *
 * @date 02-12-2005
 *
 * @author mickem
 *
 * @par license
 * This code is absolutely free to use and modify. The code is provided "as is" with
 * no expressed or implied warranty. The author accepts no liability if it causes
 * any damage to your computer, causes your pet to fall ill, increases baldness
 * or makes your car start emitting strange noises when you start it up.
 * This code has no bugs, just undocumented features!
 * 
 * @todo 
 * Add status codes to make error type simpler to parse out.
 *
 * @bug 
 *
 */
class NSPluginException {
public:
        std::string file_;      // DLL filename (for which the exception was thrown)
        std::string error_;     // An error message (human readable format)
        /**
         * @param file DLL filename (for which the exception is thrown)
         * @param error An error message (human readable format)
         */
        NSPluginException(std::string file, std::string error) : file_(file), error_(error) {
        }
        /**
         *
         * @param file DLL filename (for which the exception is thrown)
         * @param sError An error message (human readable format)
         * @param nError Error code to be appended at the end of the string
         * @todo Change this to be some form of standard error code and merge with above.
         */
        NSPluginException(std::string file, std::string sError, int nError) : file_(file) {
                std::stringstream s;
                s << sError;
                s << nError;
                error_ = s.str();

        }
};

/**
 * @ingroup NSClient++
 * NSCPlugin is a wrapper class to wrap all DLL calls and make things simple and clean inside the actual application.<br>
 * Things tend to be one-to-one by which I mean that a call to a function here should call the corresponding function in the plug in (if loaded).
 * If things are "broken" NSPluginException is called to indicate this. Error states are returned for normal "conditions".
 *
 *
 * @version 1.0
 * first version
 *
 * @date 02-12-2005
 *
 * @author mickem
 *
 * @par license
 * This code is absolutely free to use and modify. The code is provided "as is" with
 * no expressed or implied warranty. The author accepts no liability if it causes
 * any damage to your computer, causes your pet to fall ill, increases baldness
 * or makes your car start emitting strange noises when you start it up.
 * This code has no bugs, just undocumented features!
 * 
 * @todo 
 * getVersion() is not implemented as of yet.
 *
 * @bug 
 *
 */
class NSCPlugin {
private:
        bool bLoaded_;                  // Status of plug in
        HMODULE hModule_;               // module handle to the DLL (once it is loaded)
        std::string file_;              // Name of the DLL file

        typedef INT (*lpModuleHelperInit)(NSCModuleHelper::lpNSAPILoader f);
        typedef INT (*lpLoadModule)();
        typedef INT (*lpGetName)(char*,unsigned int);
        typedef INT (*lpGetDescription)(char*,unsigned int);
        typedef INT (*lpGetVersion)(int*,int*,int*);
        typedef INT (*lpHasCommandHandler)();
        typedef INT (*lpHasMessageHandler)();
        typedef NSCAPI::nagiosReturn (*lpHandleCommand)(const char*,const unsigned int, char**,char*,unsigned int,char *,unsigned int);
        typedef INT (*lpHandleMessage)(int,const char*,const int,const char*);
        typedef INT (*lpUnLoadModule)();
        typedef INT (*lpGetConfigurationMeta)(int, char*);


        lpModuleHelperInit fModuleHelperInit;
        lpLoadModule fLoadModule;
        lpGetName fGetName;
        lpGetVersion fGetVersion;
        lpGetDescription fGetDescription;
        lpHasCommandHandler fHasCommandHandler;
        lpHasMessageHandler fHasMessageHandler;
        lpHandleCommand fHandleCommand;
        lpHandleMessage fHandleMessage;
        lpUnLoadModule fUnLoadModule;
        lpGetConfigurationMeta fGetConfigurationMeta;

public:
        NSCPlugin(const std::string file);
        NSCPlugin(NSCPlugin &other);
        virtual ~NSCPlugin(void);

        std::string getName(void);
        std::string getDescription();
        void load_dll(void);
        void load_plugin(void);
        bool getVersion(int *major, int *minor, int *revision);
        bool hasCommandHandler(void);
        bool hasMessageHandler(void);
        NSCAPI::nagiosReturn handleCommand(const char *command, const unsigned int argLen, char **arguments, char* returnMessageBuffer, unsigned int returnMessageBufferLen, char* returnPerfBuffer, unsigned int returnPerfBufferLen);
        void handleMessage(int msgType, const char* file, const int line, const char *message);
        void unload(void);
        std::string getCongifurationMeta();

private:
        bool isLoaded() const {
                return bLoaded_;
        }
        bool getName_(char* buf, unsigned int buflen);
        bool getDescription_(char* buf, unsigned int buflen);
        void loadRemoteProcs_(void);
        bool getConfigurationMeta_(char* buf, unsigned int buflen);
};