tyler.durden
/
pandorafms
mirror of https://github.com/pandorafms/pandorafms.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

2008-12-16 Ramon Novoa <rnovoa@artica.es> 

* win32/modules/pandora_module_wmiquery.cc,
          win32/modules/pandora_module_wmiquery.h: Added to repository. WMI
          query module.

        * win32/windows/pandora_wmi.cc,
          win32/windows/pandora_wmi.h,
          win32/modules/pandora_module.h,
          win32/modules/pandora_module_factory.cc,
          win32/modules/pandora_module_list.cc,
          win32/modules/pandora_module.cc,
          win32/PandoraAgent.dev: Added support for WMI query module.




git-svn-id: https://svn.code.sf.net/p/pandora/code/trunk@1294 c3f86ba8-e40f-0410-aaad-9ba5e7f4b01f
This commit is contained in:
ramonn 2008-12-16 11:30:12 +00:00
parent e22e10a3a2
commit 4cdc153c03
10 changed files with 292 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pandora_agents/ChangeLog
View File

@ -1,3 +1,17 @@
2008-12-16 Ramon Novoa <rnovoa@artica.es>
* win32/modules/pandora_module_wmiquery.cc,
win32/modules/pandora_module_wmiquery.h: Added to repository. WMI
query module.
* win32/windows/pandora_wmi.cc,
win32/windows/pandora_wmi.h,
win32/modules/pandora_module.h,
win32/modules/pandora_module_factory.cc,
win32/modules/pandora_module_list.cc,
win32/modules/pandora_module.cc,
win32/PandoraAgent.dev: Added support for WMI query module.
2008-12-11 Evi Vanoost <vanooste@rcbi.rochester.edu>
* linux/plugins/SGE: Added the SGE plugin

22
pandora_agents/win32/PandoraAgent.dev
View File

@ -1,7 +1,7 @@
[Project]
FileName=PandoraAgent.dev
Name=PandoraAgent
UnitCount=79
UnitCount=81
Type=1
Ver=1
ObjFiles=
@ -837,3 +837,23 @@ Priority=1000
OverrideBuildCmd=0
BuildCmd=
[Unit80]
FileName=modules\pandora_module_wmiquery.cc
CompileCpp=1
Folder=Modules
Compile=1
Link=1
Priority=1000
OverrideBuildCmd=0
BuildCmd=
[Unit81]
FileName=modules\pandora_module_wmiquery.h
CompileCpp=1
Folder=Modules
Compile=1
Link=1
Priority=1000
OverrideBuildCmd=0
BuildCmd=

3
pandora_agents/win32/modules/pandora_module.cc
View File

@ -120,8 +120,9 @@ Pandora_Module::parseModuleKindFromString (string kind) {
} else if (kind == module_odbc_str) {
return MODULE_ODBC;
} else if (kind == module_logevent_str) {
return MODULE_LOGEVENT;
} else if (kind == module_wmiquery_str) {
return MODULE_WMIQUERY;
} else {
return MODULE_0;
}

7
pandora_agents/win32/modules/pandora_module.h
View File

@ -70,7 +70,8 @@ namespace Pandora_Modules {
MODULE_FREEMEMORY, /**< 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_LOGEVENT, /**< The module checks for log events */
MODULE_WMIQUERY /**< The module runs WQL queries */
} Module_Kind;
const string module_exec_str = "module_exec";
@ -81,6 +82,7 @@ namespace Pandora_Modules {
const string module_cpuusage_str = "module_cpuusage";
const string module_odbc_str = "module_odbc";
const string module_logevent_str = "module_logevent";
const string module_wmiquery_str = "module_wmiquery";
/**
* Pandora module super-class exception.
@ -102,7 +104,8 @@ namespace Pandora_Modules {
*/
class Interval_Not_Fulfilled : public Pandora_Modules::Module_Exception { };
/** * Pandora module super-class.
/**
* Pandora module super-class.
*
* Every defined module must inherit of this class.
*/

15
pandora_agents/win32/modules/pandora_module_factory.cc
View File

@ -28,6 +28,7 @@
#include "pandora_module_cpuusage.h"
#include "pandora_module_odbc.h"
#include "pandora_module_logevent.h"
#include "pandora_module_wmiquery.h"
#include "../pandora_strutils.h"
#include <list>
@ -57,6 +58,8 @@ using namespace Pandora_Strutils;
#define TOKEN_ASYNC ("module_async")
#define TOKEN_WATCHDOG ("module_watchdog ")
#define TOKEN_START_COMMAND ("module_start_command ")
#define TOKEN_WMIQUERY ("module_wmiquery ")
#define TOKEN_WMICOLUMN ("module_wmicolumn ")
string
parseLine (string line, string token) {
@ -94,6 +97,7 @@ Pandora_Module_Factory::getModuleFromDefinition (string definition) {
string module_logevent, module_source, module_eventtype, module_eventcode;
string module_pattern, module_async;
string module_watchdog, module_start_command;
string module_wmiquery, module_wmicolumn;
Pandora_Module *module;
bool numeric;
Module_Type type;
@ -118,6 +122,8 @@ Pandora_Module_Factory::getModuleFromDefinition (string definition) {
module_async = "";
module_watchdog = "";
module_start_command = "";
module_wmiquery = "";
module_wmicolumn = "";
stringtok (tokens, definition, "\n");
@ -194,6 +200,12 @@ Pandora_Module_Factory::getModuleFromDefinition (string definition) {
if (module_watchdog == "") {
module_watchdog = parseLine (line, TOKEN_WATCHDOG);
}
if (module_wmiquery == "") {
module_wmiquery = parseLine (line, TOKEN_WMIQUERY);
}
if (module_wmicolumn == "") {
module_wmicolumn = parseLine (line, TOKEN_WMICOLUMN);
}
iter++;
}
@ -263,6 +275,9 @@ Pandora_Module_Factory::getModuleFromDefinition (string definition) {
module_eventtype,
module_eventcode,
module_pattern);
} else if (module_wmiquery != "") {
module = new Pandora_Module_WMIQuery (module_name,
module_wmiquery, module_wmicolumn);
} else {
return NULL;
}

6
pandora_agents/win32/modules/pandora_module_list.cc
View File

@ -28,6 +28,7 @@
#include "pandora_module_cpuusage.h"
#include "pandora_module_odbc.h"
#include "pandora_module_logevent.h"
#include "pandora_module_wmiquery.h"
#include <fstream>
using namespace std;
@ -135,6 +136,7 @@ Pandora_Modules::Pandora_Module_List::parseModuleDefinition (string definition)
Pandora_Module_Freememory *module_freememory;
Pandora_Module_Odbc *module_odbc;
Pandora_Module_Logevent *module_logevent;
Pandora_Module_WMIQuery *module_wmiquery;
module = Pandora_Module_Factory::getModuleFromDefinition (definition);
@ -180,6 +182,10 @@ Pandora_Modules::Pandora_Module_List::parseModuleDefinition (string definition)
module_logevent = (Pandora_Module_Logevent *) module;
modules->push_back (module_logevent);
break;
case MODULE_WMIQUERY:
module_wmiquery = (Pandora_Module_WMIQuery *) module;
modules->push_back (module_wmiquery);
break;
default:
break;
}

69
pandora_agents/win32/modules/pandora_module_wmiquery.cc Normal file
View File

@ -0,0 +1,69 @@
/* Pandora wmiquery module. This module runs WQL queries.
Copyright (C) 2008 Artica ST.
Written by Ramon Novoa.
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.
*/
#include "pandora_module_wmiquery.h"
#include "../windows/pandora_wmi.h"
#include "../pandora_windows_service.h"
using namespace Pandora;
using namespace Pandora_Modules;
/**
* Creates a Pandora_Module_WMIQuery object.
*
* @param name Module name.
*/
Pandora_Module_WMIQuery::Pandora_Module_WMIQuery (string name, string query, string column)
: Pandora_Module (name) {
this->query = query;
this->column = column;
this->setKind (module_wmiquery_str);
}
void
Pandora_Module_WMIQuery::run () {
string value;
list<string> rows;
list<string>::iterator row;
// Run
try {
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
if (this->query.empty () || this->column.empty ()) {
return;
}
Pandora_Wmi::runWMIQuery (this->query, this->column, rows);
// No data
if (rows.size () < 1) {
this->setOutput ("");
return;
}
for (row = rows.begin (); row != rows.end(); ++row) {
this->setOutput (*row);
}
}

42
pandora_agents/win32/modules/pandora_module_wmiquery.h Normal file
View File

@ -0,0 +1,42 @@
/* Pandora wmiquery module header file.
Copyright (C) 2008 Artica ST.
Written by Ramon Novoa.
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_WMIQUERY_H__
#define __PANDORA_MODULE_WMIQUERY_H__
#include "pandora_module.h"
namespace Pandora_Modules {
/**
* This module runs WQL queries.
*/
class Pandora_Module_WMIQuery : public Pandora_Module {
private:
string query;
string column;
public:
Pandora_Module_WMIQuery (string name, string query, string column);
void run ();
};
}
#endif

37
pandora_agents/win32/windows/pandora_wmi.cc
View File

@ -632,3 +632,40 @@ Pandora_Wmi::stopService (string service_name) {
return success;
}
/**
* Runs a generic WQL query.
*
* @param wmi_query WQL query.
* @param column Column to retrieve from the query result.
* @param rows List where the query result will be placed.
*/
void
Pandora_Wmi::runWMIQuery (string wmi_query, string column, list<string> &rows) {
CDhInitialize init;
CDispPtr wmi_svc, quickfixes;
char *value = NULL;
wstring column_w(column.length(), L' ');
wstring wmi_query_w(wmi_query.length(), L' ');
// Copy string to wstring.
std::copy(column.begin(), column.end(), column_w.begin());
std::copy(wmi_query.begin(), wmi_query.end(), wmi_query_w.begin());
try {
dhCheck (dhGetObject (getWmiStr (L"."), NULL, &wmi_svc));
dhCheck (dhGetValue (L"%o", &quickfixes, wmi_svc,
L".ExecQuery(%S)",
wmi_query_w.c_str ()));
FOR_EACH (quickfix, quickfixes, NULL) {
dhGetValue (L"%s", &value, quickfix,
column_w.c_str ());
rows.push_back (value);
dhFreeString (value);
} NEXT_THROW (quickfix);
} catch (string errstr) {
pandoraLog ("runWMIQuery error. %s", errstr.c_str ());
}
}

3
pandora_agents/win32/windows/pandora_wmi.h
View File

@ -61,6 +61,9 @@ namespace Pandora_Wmi {
bool runProgram (string command);
bool startService (string service_name);
bool stopService (string service_name);
void runWMIQuery (string wmi_query,
string var,
list<string> &rows);
};
#endif