260 lines
8.4 KiB
C++
260 lines
8.4 KiB
C++
/* Defines a parent class for a Pandora module.
|
|
|
|
Copyright (C) 2006 Artica ST.
|
|
Written by Esteban Sanchez.
|
|
|
|
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, 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.
|
|
*/
|
|
|
|
#ifndef __PANDORA_MODULE_H__
|
|
#define __PANDORA_MODULE_H__
|
|
|
|
#include "../pandora.h"
|
|
#include "pandora_data.h"
|
|
#include "boost/regex.h"
|
|
#include <list>
|
|
#include <string>
|
|
#include <ctime>
|
|
|
|
using namespace Pandora;
|
|
|
|
/**
|
|
* Definition of Pandora modules.
|
|
*/
|
|
namespace Pandora_Modules {
|
|
|
|
/**
|
|
* Defines the type of the module.
|
|
*
|
|
* The type of a module is the value type the module can have.
|
|
*/
|
|
typedef enum {
|
|
TYPE_0, /**< Invalid value */
|
|
TYPE_GENERIC_DATA, /**< The value is an integer */
|
|
TYPE_GENERIC_DATA_INC, /**< The value is an integer with
|
|
* incremental diferences */
|
|
TYPE_GENERIC_PROC, /**< The value is a 0 or a 1 */
|
|
TYPE_GENERIC_DATA_STRING, /**< The value is a string */
|
|
TYPE_ASYNC_DATA, /**< Asynchronous generic_data */
|
|
TYPE_ASYNC_PROC, /**< Asynchronous generic_proc */
|
|
TYPE_ASYNC_STRING /**< Asynchronous generic_data_string */
|
|
} Module_Type;
|
|
|
|
const string module_generic_data_str = "generic_data";
|
|
const string module_generic_data_inc_str = "generic_data_inc";
|
|
const string module_generic_proc_str = "generic_proc";
|
|
const string module_generic_data_string_str = "generic_data_string";
|
|
const string module_async_data_str = "async_data";
|
|
const string module_async_proc_str = "async_proc";
|
|
const string module_async_string_str = "async_string";
|
|
|
|
/**
|
|
* Defines the kind of the module.
|
|
*
|
|
* The kind of a module is the work the module does.
|
|
*/
|
|
typedef enum {
|
|
MODULE_0, /**< Invalid kind */
|
|
MODULE_EXEC, /**< The module run a custom command */
|
|
MODULE_PROC, /**< The module checks for a running
|
|
* process */
|
|
MODULE_SERVICE, /**< The module checks for a running
|
|
* service */
|
|
MODULE_FREEDISK, /**< The module checks the free */
|
|
MODULE_FREEDISK_PERCENT, /**< The module checks the free */
|
|
MODULE_CPUUSAGE, /**< The module checks the CPU usage */
|
|
MODULE_INVENTORY, /**< The module gets the inventory of the machine */
|
|
MODULE_FREEMEMORY, /**< The module checks the percentage of
|
|
* freememory in the system */
|
|
MODULE_FREEMEMORY_PERCENT, /**< The module checks the amount of
|
|
* freememory in the system */
|
|
MODULE_ODBC, /**< The module performs a SQL query via ODBC */
|
|
MODULE_LOGEVENT, /**< The module checks for log events */
|
|
MODULE_WMIQUERY, /**< The module runs WQL queries */
|
|
MODULE_PERFCOUNTER, /**< The module reads performance counters */
|
|
MODULE_TCPCHECK, /**< The module checks whether a tcp port is open */
|
|
MODULE_REGEXP, /**< The module searches a file for matches of a regular expression */
|
|
MODULE_PLUGIN /**< Plugin */
|
|
} Module_Kind;
|
|
|
|
/**
|
|
* Defines the structure that holds module conditions.
|
|
*/
|
|
typedef struct {
|
|
double value_1;
|
|
double value_2;
|
|
string string_value;
|
|
string operation;
|
|
string command;
|
|
regex_t regexp;
|
|
} Condition;
|
|
|
|
/**
|
|
* Defines the structure that holds the module cron.
|
|
*/
|
|
typedef struct {
|
|
time_t utimestamp;
|
|
int interval;
|
|
int params[5][2];
|
|
} Cron;
|
|
|
|
const string module_exec_str = "module_exec";
|
|
const string module_proc_str = "module_proc";
|
|
const string module_service_str = "module_service";
|
|
const string module_freedisk_str = "module_freedisk";
|
|
const string module_freedisk_percent_str = "module_freedisk_percent";
|
|
const string module_freememory_str = "module_freememory";
|
|
const string module_freememory_percent_str = "module_freememory_percent";
|
|
const string module_cpuusage_str = "module_cpuusage";
|
|
const string module_inventory_str = "module_inventory";
|
|
const string module_odbc_str = "module_odbc";
|
|
const string module_logevent_str = "module_logevent";
|
|
const string module_wmiquery_str = "module_wmiquery";
|
|
const string module_perfcounter_str = "module_perfcounter";
|
|
const string module_tcpcheck_str = "module_tcpcheck";
|
|
const string module_regexp_str = "module_regexp";
|
|
const string module_plugin_str = "module_plugin";
|
|
|
|
/**
|
|
* Pandora module super-class exception.
|
|
*/
|
|
class Module_Exception : public Pandora::Pandora_Exception { };
|
|
|
|
/**
|
|
* An error happened with the module output.
|
|
*/
|
|
class Output_Error : public Pandora_Modules::Module_Exception { };
|
|
|
|
/**
|
|
* The module value is not correct, usually beacause of the limits.
|
|
*/
|
|
class Value_Error : public Pandora_Modules::Module_Exception { };
|
|
|
|
/**
|
|
* The module does not satisfy its interval.
|
|
*/
|
|
class Interval_Not_Fulfilled : public Pandora_Modules::Module_Exception { };
|
|
|
|
/**
|
|
* Pandora module super-class.
|
|
*
|
|
* Every defined module must inherit of this class.
|
|
*/
|
|
class Pandora_Module {
|
|
private:
|
|
int module_interval;
|
|
int module_timeout;
|
|
int executions;
|
|
int max, min;
|
|
string post_process;
|
|
bool has_limits, has_min, has_max;
|
|
Module_Type module_type;
|
|
string module_kind_str;
|
|
Module_Kind module_kind;
|
|
string save;
|
|
list<Condition *> *condition_list;
|
|
Cron *cron;
|
|
|
|
protected:
|
|
|
|
list<Pandora_Data *> *data_list;
|
|
|
|
/**
|
|
* Indicates if the module generated output in
|
|
* his last execution.
|
|
*/
|
|
bool has_output;
|
|
/**
|
|
* The name of the module.
|
|
*/
|
|
string module_name;
|
|
/**
|
|
* The description of the module.
|
|
*/
|
|
string module_description;
|
|
|
|
/**
|
|
* Flag to set a module as asynchronous
|
|
*/
|
|
bool async;
|
|
/**
|
|
* List of items of the inventory
|
|
*/
|
|
list<Pandora_Data *> *inventory_list;
|
|
/**
|
|
* Data passed to the latest setOutput
|
|
*/
|
|
string latest_output;
|
|
/**
|
|
* String containing the module type
|
|
*/
|
|
string module_type_str;
|
|
|
|
string getDataOutput (Pandora_Data *data);
|
|
void cleanDataList ();
|
|
public:
|
|
Pandora_Module (string name);
|
|
virtual ~Pandora_Module ();
|
|
|
|
static Module_Type
|
|
parseModuleTypeFromString (string type);
|
|
|
|
static Module_Kind
|
|
parseModuleKindFromString (string kind);
|
|
|
|
void setInterval (int interval);
|
|
int getInterval ();
|
|
void setTimeout (int timeout);
|
|
int getTimeout ();
|
|
string getSave ();
|
|
|
|
virtual string getXml ();
|
|
|
|
|
|
virtual void run ();
|
|
|
|
virtual void setOutput (string output);
|
|
virtual void setOutput (string output,
|
|
SYSTEMTIME *system_time);
|
|
|
|
|
|
string getName () const;
|
|
string getDescription () const;
|
|
string getTypeString () const;
|
|
string getLatestOutput () const;
|
|
Module_Type getTypeInt () const;
|
|
Module_Type getModuleType () const;
|
|
Module_Kind getModuleKind () const;
|
|
|
|
void setType (string type);
|
|
void setKind (string kind);
|
|
void setDescription (string description);
|
|
void setMax (int value);
|
|
void setMin (int value);
|
|
void setPostProcess (string value);
|
|
void setAsync (bool async);
|
|
void setSave (string save);
|
|
|
|
void exportDataOutput ();
|
|
void addCondition (string condition);
|
|
void evaluateConditions ();
|
|
int checkCron ();
|
|
void setCron (string cron_string);
|
|
void setCronInterval (int interval);
|
|
};
|
|
}
|
|
|
|
#endif
|