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

2008-11-26 Esteban Sanchez <estebans@artica.es> 

* pandora_strutils.cc, pandora_strutils.h, ftp/pandora_ftp_client.cc,
	ftp/pandora_ftp_test.cc, ftp/pandora_ftp_client.h,
	ftp/pandora_ftp_test.h, windows/pandora_windows_info.h,
	windows/pandora_wmi.cc, windows/pandora_windows_info.cc,
	misc/pandora_file.cc, misc/pandora_file.h, pandora_agent_conf.cc,
	ssh/pandora_ssh_client.cc, ssh/pandora_ssh_test.cc,
	ssh/pandora_ssh_client.h, ssh/pandora_ssh_test.h,
	pandora_agent_conf.h, windows_service.cc, windows_service.h,
	modules/pandora_module.h, modules/pandora_module_logevent.cc,
	modules/pandora_module_exec.cc, modules/pandora_module_logevent.h,
	modules/pandora_module_exec.h, modules/pandora_module_freedisk.cc,
	modules/pandora_module_freedisk.h, modules/pandora_module_service.cc,
	modules/pandora_module_service.h, modules/pandora_module_proc.cc,
	modules/pandora_data.cc, modules/pandora_module_proc.h,
	modules/pandora_data.h, modules/pandora_module_factory.cc,
	modules/pandora_module_odbc.cc, modules/pandora_module_odbc.h,
	modules/pandora_module_factory.h,
	modules/pandora_module_freememory.cc, modules/pandora_module_list.cc,
	modules/pandora_module_freememory.h,
	modules/pandora_module_cpuusage.cc, modules/pandora_module_cpuusage.h,
	modules/pandora_module.cc, pandora.h: Tab style correction.
	Indentation blankspaces moved to tab characters.



git-svn-id: https://svn.code.sf.net/p/pandora/code/trunk@1262 c3f86ba8-e40f-0410-aaad-9ba5e7f4b01f
This commit is contained in:
esanchezm 2008-11-26 10:08:19 +00:00
parent c387a3ccc9
commit 9cd9ecb462
44 changed files with 1582 additions and 1558 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pandora_agents/win32/ChangeLog
View File

@ -1,3 +1,28 @@
2008-11-26 Esteban Sanchez <estebans@artica.es>
* pandora_strutils.cc, pandora_strutils.h, ftp/pandora_ftp_client.cc,
ftp/pandora_ftp_test.cc, ftp/pandora_ftp_client.h,
ftp/pandora_ftp_test.h, windows/pandora_windows_info.h,
windows/pandora_wmi.cc, windows/pandora_windows_info.cc,
misc/pandora_file.cc, misc/pandora_file.h, pandora_agent_conf.cc,
ssh/pandora_ssh_client.cc, ssh/pandora_ssh_test.cc,
ssh/pandora_ssh_client.h, ssh/pandora_ssh_test.h,
pandora_agent_conf.h, windows_service.cc, windows_service.h,
modules/pandora_module.h, modules/pandora_module_logevent.cc,
modules/pandora_module_exec.cc, modules/pandora_module_logevent.h,
modules/pandora_module_exec.h, modules/pandora_module_freedisk.cc,
modules/pandora_module_freedisk.h, modules/pandora_module_service.cc,
modules/pandora_module_service.h, modules/pandora_module_proc.cc,
modules/pandora_data.cc, modules/pandora_module_proc.h,
modules/pandora_data.h, modules/pandora_module_factory.cc,
modules/pandora_module_odbc.cc, modules/pandora_module_odbc.h,
modules/pandora_module_factory.h,
modules/pandora_module_freememory.cc, modules/pandora_module_list.cc,
modules/pandora_module_freememory.h,
modules/pandora_module_cpuusage.cc, modules/pandora_module_cpuusage.h,
modules/pandora_module.cc, pandora.h: Tab style correction.
Indentation blankspaces moved to tab characters.
2008-10-30 Raul Mateos <raulofpandora@gmail.com>
* win32/installer/pandora_2.0.mpi: Updated installer to let the user start

14
pandora_agents/win32/ftp/pandora_ftp_client.cc
View File

@ -35,9 +35,9 @@ using namespace Pandora;
*/
Pandora_Ftp_Client::Pandora_Ftp_Client ()
{
curl = NULL;
curl = NULL;
return;
return;
}
/**
@ -49,9 +49,9 @@ Pandora_Ftp_Client::Pandora_Ftp_Client ()
*/
Pandora_Ftp_Client::~Pandora_Ftp_Client ()
{
this->disconnect ();
this->disconnect ();
return;
return;
}
/**
@ -62,10 +62,10 @@ Pandora_Ftp_Client::~Pandora_Ftp_Client ()
void
Pandora_Ftp_Client::disconnect ()
{
if (curl != NULL) {
curl_easy_cleanup (curl);
if (curl != NULL) {
curl_easy_cleanup (curl);
curl = NULL;
}
}
}
/**

44
pandora_agents/win32/ftp/pandora_ftp_client.h
View File

@ -31,50 +31,50 @@ using namespace std;
* FTP connection classes.
*/
namespace FTP {
/**
/**
* A FTP super-class exception.
*/
class FTP_Exception : public Pandora::Pandora_Exception {
};
class FTP_Exception : public Pandora::Pandora_Exception {
};
/**
* The FTP authentication fails when connecting.
*/
class Authentication_Failed : public FTP::FTP_Exception {
};
class Authentication_Failed : public FTP::FTP_Exception {
};
/**
* The FTP host is unknown.
*/
class Unknown_Host : public FTP::FTP_Exception {
};
class Unknown_Host : public FTP::FTP_Exception {
};
/**
/**
* Client to perform a FTP connection to a host.
*/
class Pandora_Ftp_Client {
private:
class Pandora_Ftp_Client {
private:
string host;
string username;
string password;
CURL *curl;
CURLcode result;
public:
Pandora_Ftp_Client ();
~Pandora_Ftp_Client ();
void connect (const string host,
CURL *curl;
CURLcode result;
public:
Pandora_Ftp_Client ();
~Pandora_Ftp_Client ();
void connect (const string host,
const int port,
const string username,
const string password);
void disconnect ();
void ftpFileFilename (const string remote_filename,
void disconnect ();
void ftpFileFilename (const string remote_filename,
const string filepath);
string getError ();
};
};
}
#endif

6
pandora_agents/win32/ftp/pandora_ftp_test.cc
View File

@ -114,7 +114,7 @@ Pandora_FTP_Test::test () {
tmp_filepath += "\\";
}
tmp_filepath += tmp_filename;
decl = new TiXmlDeclaration( "1.0", "ISO-8859-1", "" );
doc = new TiXmlDocument (tmp_filepath);
doc->InsertEndChild (*decl);
@ -132,9 +132,9 @@ Pandora_FTP_Test::test () {
throw e;
}
delete doc;
cout << "Created a blank XML file in " << tmp_filepath<< endl;
remote_filepath = conf->getValue ("server_path");
if (remote_filepath[remote_filepath.length () - 1] != '/') {
remote_filepath += "/";

28
pandora_agents/win32/ftp/pandora_ftp_test.h
View File

@ -25,18 +25,18 @@ using namespace Pandora;
namespace FTP {
/**
* Class to perform a test of the FTP configuration.
*
* An object of this class will read the configuration file
* and copy a blank xml file into remote server path.
*/
class Pandora_FTP_Test {
private:
Pandora_Ftp_Client *ftp_client;
Pandora_Agent_Conf *conf;
public:
Pandora_FTP_Test ();
~Pandora_FTP_Test ();
void test ();
};
* Class to perform a test of the FTP configuration.
*
* An object of this class will read the configuration file
* and copy a blank xml file into remote server path.
*/
class Pandora_FTP_Test {
private:
Pandora_Ftp_Client *ftp_client;
Pandora_Agent_Conf *conf;
public:
Pandora_FTP_Test ();
~Pandora_FTP_Test ();
void test ();
};
}

86
pandora_agents/win32/misc/pandora_file.cc
View File

@ -35,14 +35,14 @@ using namespace std;
**/
bool
Pandora_File::fileExists (const string filepath) {
ifstream myfile (filepath.c_str ());
if (! myfile.is_open ()) {
return false;
}
ifstream myfile (filepath.c_str ());
if (! myfile.is_open ()) {
return false;
}
myfile.close();
return true;
return true;
}
/**
@ -57,21 +57,21 @@ Pandora_File::fileExists (const string filepath) {
**/
string
Pandora_File::readFile (const string filepath) {
string line, result;
ifstream myfile (filepath.c_str ());
if (! myfile.is_open ()) {
throw File_Not_Found ();
}
if (myfile.is_open()) {
while (! myfile.eof()) {
getline (myfile,line);
result += line + '\n';
}
myfile.close();
}
return result;
string line, result;
ifstream myfile (filepath.c_str ());
if (! myfile.is_open ()) {
throw File_Not_Found ();
}
if (myfile.is_open()) {
while (! myfile.eof()) {
getline (myfile,line);
result += line + '\n';
}
myfile.close();
}
return result;
}
/**
@ -128,9 +128,9 @@ Pandora_File::readBinFile (const string filepath, char **buffer) {
*/
void
Pandora_File::removeFile (const string filepath) {
if (remove (filepath.c_str ()) == -1) {
throw Delete_Error ();
}
if (remove (filepath.c_str ()) == -1) {
throw Delete_Error ();
}
}
/**
@ -144,13 +144,13 @@ Pandora_File::removeFile (const string filepath) {
*/
void
Pandora_File::writeFile (const string filepath, const string data) {
ofstream file (filepath.c_str ());
if (! file.is_open ()) {
throw File_Not_Found ();
}
file.write (data.c_str (), data.length ());
file.close ();
ofstream file (filepath.c_str ());
if (! file.is_open ()) {
throw File_Not_Found ();
}
file.write (data.c_str (), data.length ());
file.close ();
}
/**
@ -165,18 +165,18 @@ Pandora_File::writeFile (const string filepath, const string data) {
*/
void
Pandora_File::writeBinFile (const string filepath, const char *buffer, int size) {
ofstream file;
if (buffer == NULL) {
throw File_Exception ();
}
file.open(filepath.c_str (), ios_base::binary | ios_base::trunc);
if (! file.is_open ()) {
throw File_Not_Found ();
}
file.write (buffer, size);
file.close ();
ofstream file;
if (buffer == NULL) {
throw File_Exception ();
}
file.open(filepath.c_str (), ios_base::binary | ios_base::trunc);
if (! file.is_open ()) {
throw File_Not_Found ();
}
file.write (buffer, size);
file.close ();
}
/**

26
pandora_agents/win32/misc/pandora_file.h
View File

@ -34,30 +34,30 @@ namespace Pandora_File {
/**
* File super-class exception.
*/
class File_Exception : Pandora::Pandora_Exception {
};
class File_Exception : Pandora::Pandora_Exception {
};
/**
* Exception throwed when a file could not be found when doing
* a file operation.
*/
class File_Not_Found : Pandora_File::File_Exception {
};
class File_Not_Found : Pandora_File::File_Exception {
};
/**
/**
* Exception throwed when a file could not be deleted on a delete
* operation.
*/
class Delete_Error : Pandora_File::File_Exception {
};
class Delete_Error : Pandora_File::File_Exception {
};
bool fileExists (const string filename);
string readFile (const string filename);
int readBinFile (const string filepath, char **buffer);
void removeFile (const string filename);
void writeFile (const string filename, const string data);
void writeBinFile (const string filepath, const char *buffer, int size);
string readFile (const string filename);
int readBinFile (const string filepath, char **buffer);
void removeFile (const string filename);
void writeFile (const string filename, const string data);
void writeBinFile (const string filepath, const char *buffer, int size);
string fileName (const string filepath);
void md5 (const char *data, int size, char *buffer);
}

4
pandora_agents/win32/modules/pandora_data.cc
View File

@ -31,7 +31,7 @@ using namespace Pandora;
* @param timestamp Timeestamp value.
*/
Pandora_Data::Pandora_Data (string value) {
this->value = value;
this->value = value;
GetSystemTime (&(this->timestamp));
}
@ -90,7 +90,7 @@ Pandora_Data::getTimestamp () const {
string retval;
sprintf (strtime, "%d-%02d-%02d %02d:%02d:%02d", this->timestamp.wYear, this->timestamp.wMonth, this->timestamp.wDay,
this->timestamp.wHour, this->timestamp.wMinute, this->timestamp.wSecond);
this->timestamp.wHour, this->timestamp.wMinute, this->timestamp.wSecond);
retval = strtime;
return retval;
}

2
pandora_agents/win32/modules/pandora_data.h
View File

@ -38,7 +38,7 @@ namespace Pandora {
public:
Pandora_Data ();
Pandora_Data (string value);
Pandora_Data (string value, SYSTEMTIME *system_time);
Pandora_Data (string value, SYSTEMTIME *system_time);
~Pandora_Data ();
string getValue () const;

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

@ -34,13 +34,13 @@ using namespace Pandora_Strutils;
* @param name Module name.
*/
Pandora_Module::Pandora_Module (string name) {
this->module_name = name;
this->executions = 0;
this->module_interval = 1;
this->max = 0;
this->min = 0;
this->has_limits = false;
this->data_list = NULL;
this->module_name = name;
this->executions = 0;
this->module_interval = 1;
this->max = 0;
this->min = 0;
this->has_limits = false;
this->data_list = NULL;
}
/**
@ -82,17 +82,17 @@ Pandora_Module::cleanDataList () {
*/
Module_Type
Pandora_Module::parseModuleTypeFromString (string type) {
if (type == module_generic_data_str) {
return TYPE_GENERIC_DATA;
} else if (type == module_generic_data_inc_str) {
return TYPE_GENERIC_DATA_INC;
} else if (type == module_generic_data_string_str) {
return TYPE_GENERIC_DATA_STRING;
} else if (type == module_generic_proc_str) {
return TYPE_GENERIC_PROC;
} else {
return TYPE_0;
}
if (type == module_generic_data_str) {
return TYPE_GENERIC_DATA;
} else if (type == module_generic_data_inc_str) {
return TYPE_GENERIC_DATA_INC;
} else if (type == module_generic_data_string_str) {
return TYPE_GENERIC_DATA_STRING;
} else if (type == module_generic_proc_str) {
return TYPE_GENERIC_PROC;
} else {
return TYPE_0;
}
}
/**
@ -104,26 +104,26 @@ Pandora_Module::parseModuleTypeFromString (string type) {
*/
Module_Kind
Pandora_Module::parseModuleKindFromString (string kind) {
if (kind == module_exec_str) {
return MODULE_EXEC;
} else if (kind == module_proc_str) {
return MODULE_PROC;
} else if (kind == module_service_str) {
return MODULE_SERVICE;
} else if (kind == module_freedisk_str) {
return MODULE_FREEDISK;
} else if (kind == module_freememory_str) {
return MODULE_FREEMEMORY;
} else if (kind == module_cpuusage_str) {
return MODULE_CPUUSAGE;
if (kind == module_exec_str) {
return MODULE_EXEC;
} else if (kind == module_proc_str) {
return MODULE_PROC;
} else if (kind == module_service_str) {
return MODULE_SERVICE;
} else if (kind == module_freedisk_str) {
return MODULE_FREEDISK;
} else if (kind == module_freememory_str) {
return MODULE_FREEMEMORY;
} else if (kind == module_cpuusage_str) {
return MODULE_CPUUSAGE;
} else if (kind == module_odbc_str) {
return MODULE_ODBC;
} else if (kind == module_logevent_str) {
return MODULE_ODBC;
} else if (kind == module_logevent_str) {
return MODULE_LOGEVENT;
} else {
return MODULE_0;
}
return MODULE_LOGEVENT;
} else {
return MODULE_0;
}
}
/**
@ -133,7 +133,7 @@ Pandora_Module::parseModuleKindFromString (string kind) {
*/
string
Pandora_Module::getName () const {
return this->module_name;
return this->module_name;
}
/**
@ -153,7 +153,7 @@ Pandora_Module::getDescription () const {
*/
string
Pandora_Module::getTypeString () const {
return this->module_type_str;
return this->module_type_str;
}
/**
@ -163,7 +163,7 @@ Pandora_Module::getTypeString () const {
*/
Module_Type
Pandora_Module::getTypeInt () const {
return this->module_type;
return this->module_type;
}
/**
@ -173,7 +173,7 @@ Pandora_Module::getTypeInt () const {
*/
Module_Kind
Pandora_Module::getModuleKind () const {
return this->module_kind;
return this->module_kind;
}
/**
@ -183,7 +183,7 @@ Pandora_Module::getModuleKind () const {
*/
Module_Type
Pandora_Module::getModuleType () const {
return this->module_type;
return this->module_type;
}
/**
@ -202,7 +202,7 @@ string
Pandora_Module::getDataOutput (Pandora_Data *data) {
double value;
if (this->module_type == TYPE_GENERIC_DATA_STRING) {
if (this->module_type == TYPE_GENERIC_DATA_STRING) {
return data->getValue ();
}
@ -274,19 +274,19 @@ Pandora_Module::setOutput (string output, SYSTEMTIME *system_time) {
*/
void
Pandora_Module::run () {
/* Check the interval */
if (this->executions % this->module_interval != 0) {
pandoraDebug ("%s: Interval is not fulfilled",
this->module_name.c_str ());
this->executions++;
has_output = false;
throw Interval_Not_Fulfilled ();
}
/* Increment the executions after check. This is done to execute the
first time */
this->executions++;
has_output = true;
/* Check the interval */
if (this->executions % this->module_interval != 0) {
pandoraDebug ("%s: Interval is not fulfilled",
this->module_name.c_str ());
this->executions++;
has_output = false;
throw Interval_Not_Fulfilled ();
}
/* Increment the executions after check. This is done to execute the
first time */
this->executions++;
has_output = true;
}
/**
@ -319,24 +319,24 @@ Pandora_Module::getXml () {
pandoraDebug ("%s getXML begin", module_name.c_str ());
if (!this->has_output || this->data_list == NULL) {
return NULL;
}
root = new TiXmlElement ("module");
element = new TiXmlElement ("name");
text = new TiXmlText (this->module_name);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete element;
delete text;
element = new TiXmlElement ("type");
text = new TiXmlText (this->module_type_str);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete element;
delete text;
return NULL;
}
root = new TiXmlElement ("module");
element = new TiXmlElement ("name");
text = new TiXmlText (this->module_name);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete element;
delete text;
element = new TiXmlElement ("type");
text = new TiXmlText (this->module_type_str);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete element;
delete text;
if (this->data_list && this->data_list->size () > 1) {
list<Pandora_Data *>::iterator iter;
@ -351,15 +351,15 @@ Pandora_Module::getXml () {
data_element = new TiXmlElement ("data");
element = new TiXmlElement ("value");
try {
data_clean = strreplace (this->getDataOutput (data), "%", "%%" );
} catch (Output_Error e) {
data_clean = strreplace (this->getDataOutput (data), "%", "%%" );
} catch (Output_Error e) {
delete element;
continue;
}
continue;
}
text = new TiXmlText (data_clean);
element->InsertEndChild (*text);
data_element->InsertEndChild (*element);
text = new TiXmlText (data_clean);
element->InsertEndChild (*text);
data_element->InsertEndChild (*element);
delete text;
delete element;
@ -379,26 +379,26 @@ Pandora_Module::getXml () {
data = data_list->front ();
element = new TiXmlElement ("data");
try {
data_clean = strreplace (this->getDataOutput (data), "%", "%%" );
text = new TiXmlText (data_clean);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete text;
} catch (Output_Error e) {
}
data_clean = strreplace (this->getDataOutput (data), "%", "%%" );
text = new TiXmlText (data_clean);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete text;
} catch (Output_Error e) {
}
delete element;
}
element = new TiXmlElement ("description");
text = new TiXmlText (this->module_description);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete text;
delete element;
element = new TiXmlElement ("description");
text = new TiXmlText (this->module_description);
element->InsertEndChild (*text);
root->InsertEndChild (*element);
delete text;
delete element;
this->cleanDataList ();
pandoraDebug ("%s getXML end", module_name.c_str ());
return root;
pandoraDebug ("%s getXML end", module_name.c_str ());
return root;
}
/**
@ -410,8 +410,8 @@ Pandora_Module::getXml () {
*/
void
Pandora_Module::setMax (int value) {
this->has_limits = true;
this->max = value;
this->has_limits = true;
this->max = value;
}
/**
@ -423,8 +423,8 @@ Pandora_Module::setMax (int value) {
*/
void
Pandora_Module::setMin (int value) {
this->has_limits = true;
this->min = value;
this->has_limits = true;
this->min = value;
}
/**
@ -434,8 +434,8 @@ Pandora_Module::setMin (int value) {
*/
void
Pandora_Module::setType (string type) {
this->module_type_str = type;
this->module_type = parseModuleTypeFromString (type);
this->module_type_str = type;
this->module_type = parseModuleTypeFromString (type);
}
/**
@ -445,8 +445,8 @@ Pandora_Module::setType (string type) {
*/
void
Pandora_Module::setKind (string kind) {
this->module_kind_str = kind;
this->module_kind = parseModuleKindFromString (kind);
this->module_kind_str = kind;
this->module_kind = parseModuleKindFromString (kind);
}
/**
@ -456,7 +456,7 @@ Pandora_Module::setKind (string kind) {
*/
void
Pandora_Module::setInterval (int interval) {
this->module_interval = interval;
this->module_interval = interval;
}
/**
@ -466,7 +466,7 @@ Pandora_Module::setInterval (int interval) {
*/
int
Pandora_Module::getInterval () {
return this->module_interval;
return this->module_interval;
}
/**
@ -476,5 +476,5 @@ Pandora_Module::getInterval () {
*/
void
Pandora_Module::setDescription (string description) {
this->module_description = description;
this->module_description = description;
}

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

@ -39,29 +39,29 @@ namespace Pandora_Modules {
*
* 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
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 */
} Module_Type;
TYPE_GENERIC_PROC, /**< The value is a 0 or a 1 */
TYPE_GENERIC_DATA_STRING /**< The value is a 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_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";
/**
* 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
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 */
@ -71,10 +71,10 @@ namespace Pandora_Modules {
* freememory in the system */
MODULE_ODBC, /**< The module performs a SQL query via ODBC */
MODULE_LOGEVENT /**< The module checks for log events */
} Module_Kind;
const string module_exec_str = "module_exec";
const string module_proc_str = "module_proc";
} Module_Kind;
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_freememory_str = "module_freememory";
@ -85,43 +85,43 @@ namespace Pandora_Modules {
/**
* Pandora module super-class exception.
*/
class Module_Exception : public Pandora::Pandora_Exception { };
class Module_Exception : public Pandora::Pandora_Exception { };
/**
* An error happened with the module output.
*/
class Output_Error : public Pandora_Modules::Module_Exception { };
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 { };
class Value_Error : public Pandora_Modules::Module_Exception { };
/**
* The module does not satisfy its interval.
*/
class Interval_Not_Fulfilled : public Pandora_Modules::Module_Exception { };
class Interval_Not_Fulfilled : public Pandora_Modules::Module_Exception { };
/**
/**
* Pandora module super-class.
*
* Every defined module must inherit of this class.
*/
class Pandora_Module {
class Pandora_Module {
private:
int module_interval;
int executions;
int max, min;
bool has_limits;
int executions;
int max, min;
bool has_limits;
string module_type_str;
Module_Type module_type;
Module_Type module_type;
string module_kind_str;
Module_Kind module_kind;
Module_Kind module_kind;
list<Pandora_Data *> *data_list;
string getDataOutput (Pandora_Data *data);
void cleanDataList ();
protected:
protected:
/**
* Indicates if the module generated output in
* his last execution.
@ -130,45 +130,45 @@ namespace Pandora_Modules {
/**
* The name of the module.
*/
string module_name;
string module_name;
/**
* The description of the module.
*/
string module_description;
public:
Pandora_Module (string name);
virtual ~Pandora_Module ();
string module_description;
public:
Pandora_Module (string name);
virtual ~Pandora_Module ();
static Module_Type
static Module_Type
parseModuleTypeFromString (string type);
static Module_Kind
parseModuleKindFromString (string kind);
void setInterval (int interval);
int getInterval ();
TiXmlElement *getXml ();
virtual void run ();
void setInterval (int interval);
int getInterval ();
TiXmlElement *getXml ();
virtual void run ();
virtual void setOutput (string output);
virtual void setOutput (string output, SYSTEMTIME *system_time);
virtual void setOutput (string output, SYSTEMTIME *system_time);
string getName () const;
string getName () const;
string getDescription () const;
string getTypeString () const;
Module_Type getTypeInt () const;
Module_Type getModuleType () const;
string getTypeString () const;
Module_Type getTypeInt () const;
Module_Type getModuleType () const;
Module_Kind getModuleKind () const;
void setType (string type);
void setType (string type);
void setKind (string kind);
void setDescription (string description);
void setMax (int value);
void setMin (int value);
};
void setDescription (string description);
void setMax (int value);
void setMin (int value);
};
}
#endif

14
pandora_agents/win32/modules/pandora_module_cpuusage.cc
View File

@ -35,10 +35,10 @@ using namespace Pandora_Strutils;
*/
Pandora_Module_Cpuusage::Pandora_Module_Cpuusage (string name, int cpu_id)
: Pandora_Module (name) {
this->cpu_id = cpu_id;
this->setKind (module_cpuusage_str);
this->cpu_id = cpu_id;
this->setKind (module_cpuusage_str);
}
void
@ -46,10 +46,10 @@ Pandora_Module_Cpuusage::run () {
int res;
try {
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
try {
res = Pandora_Wmi::getCpuUsagePercentage (this->cpu_id);

16
pandora_agents/win32/modules/pandora_module_cpuusage.h
View File

@ -28,14 +28,14 @@ namespace Pandora_Modules {
/**
* Module to retrieve the CPU usage percentage.
*/
class Pandora_Module_Cpuusage : public Pandora_Module {
private:
int cpu_id;
public:
Pandora_Module_Cpuusage (string name, int cpu_id);
void run ();
};
class Pandora_Module_Cpuusage : public Pandora_Module {
private:
int cpu_id;
public:
Pandora_Module_Cpuusage (string name, int cpu_id);
void run ();
};
}
#endif

2
pandora_agents/win32/modules/pandora_module_exec.cc
View File

@ -35,7 +35,7 @@ using namespace Pandora_Modules;
* @param exec Command to be executed.
*/
Pandora_Module_Exec::Pandora_Module_Exec (string name, string exec)
: Pandora_Module (name) {
: Pandora_Module (name) {
this->module_exec = "cmd.exe /c \"" + exec + "\"";
this->setKind (module_exec_str);

16
pandora_agents/win32/modules/pandora_module_exec.h
View File

@ -31,14 +31,14 @@ namespace Pandora_Modules {
* Any custom order that want to be executed can be put in
* the <code>util</code> directory into the Pandora agent path.
*/
class Pandora_Module_Exec : public Pandora_Module {
private:
string module_exec;
public:
Pandora_Module_Exec (string name, string exec);
void run ();
};
class Pandora_Module_Exec : public Pandora_Module {
private:
string module_exec;
public:
Pandora_Module_Exec (string name, string exec);
void run ();
};
}
#endif

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

@ -56,18 +56,18 @@ using namespace Pandora_Strutils;
string
parseLine (string line, string token) {
unsigned int pos;
string retstr = "";
unsigned int pos;
string retstr = "";
pos = line.find (token);
if (pos == 0) {
retstr = line.substr (token.length ());
pos = line.find (token);
if (pos == 0) {
retstr = line.substr (token.length ());
if (retstr == "") {
retstr = " ";
}
}
}
return retstr;
return retstr;
}
/**
@ -80,16 +80,16 @@ parseLine (string line, string token) {
*/
Pandora_Module *
Pandora_Module_Factory::getModuleFromDefinition (string definition) {
list<string> tokens;
list<string>::iterator iter;
string module_name, module_type, module_exec;
string module_min, module_max, module_description;
string module_interval, module_proc, module_service;
list<string> tokens;
list<string>::iterator iter;
string module_name, module_type, module_exec;
string module_min, module_max, module_description;
string module_interval, module_proc, module_service;
string module_freedisk, module_cpuusage, module_odbc;
string module_odbc_query, module_dsn, module_freememory;
string module_logevent, module_source, module_eventtype, module_pattern;
Pandora_Module *module;
bool numeric;
Pandora_Module *module;
bool numeric;
Module_Type type;
module_name = "";
@ -109,88 +109,88 @@ Pandora_Module_Factory::getModuleFromDefinition (string definition) {
module_eventtype = "";
module_pattern = "";
stringtok (tokens, definition, "\n");
/* Pick the first and the last value of the token list */
iter = tokens.begin ();
while (iter != tokens.end()) {
string line;
line = trim (*iter);
if (module_name == "") {
module_name = parseLine (line, TOKEN_NAME);
}
if (module_type == "") {
module_type = parseLine (line, TOKEN_TYPE);
}
if (module_interval == "") {
module_interval = parseLine (line, TOKEN_INTERVAL);
}
if (module_exec == "") {
module_exec = parseLine (line, TOKEN_EXEC);
}
if (module_proc == "") {
module_proc = parseLine (line, TOKEN_PROC);
}
stringtok (tokens, definition, "\n");
/* Pick the first and the last value of the token list */
iter = tokens.begin ();
while (iter != tokens.end()) {
string line;
line = trim (*iter);
if (module_name == "") {
module_name = parseLine (line, TOKEN_NAME);
}
if (module_type == "") {
module_type = parseLine (line, TOKEN_TYPE);
}
if (module_interval == "") {
module_interval = parseLine (line, TOKEN_INTERVAL);
}
if (module_exec == "") {
module_exec = parseLine (line, TOKEN_EXEC);
}
if (module_proc == "") {
module_proc = parseLine (line, TOKEN_PROC);
}
if (module_service == "") {
module_service = parseLine (line, TOKEN_SERVICE);
module_service = parseLine (line, TOKEN_SERVICE);
}
if (module_freedisk == "") {
module_freedisk = parseLine (line, TOKEN_FREEDISK);
module_freedisk = parseLine (line, TOKEN_FREEDISK);
}
if (module_freememory == "") {
module_freememory = parseLine (line, TOKEN_FREEMEMORY);
module_freememory = parseLine (line, TOKEN_FREEMEMORY);
}
if (module_cpuusage == "") {
module_cpuusage = parseLine (line, TOKEN_CPUUSAGE);
module_cpuusage = parseLine (line, TOKEN_CPUUSAGE);
}
if (module_odbc == "") {
module_odbc = parseLine (line, TOKEN_ODBC);
}
if (module_max == "") {
module_max = parseLine (line, TOKEN_MAX);
}
if (module_min == "") {
module_min = parseLine (line, TOKEN_MIN);
}
if (module_description == "") {
module_description = parseLine (line, TOKEN_DESCRIPTION);
}
if (module_odbc_query == "") {
module_odbc_query = parseLine (line, TOKEN_ODBC_QUERY);
}
if (module_logevent == "") {
module_logevent = parseLine (line, TOKEN_LOGEVENT);
}
if (module_source == "") {
module_source = parseLine (line, TOKEN_SOURCE);
}
if (module_eventtype == "") {
module_eventtype = parseLine (line, TOKEN_EVENTTYPE);
}
if (module_pattern == "") {
module_pattern = parseLine (line, TOKEN_PATTERN);
}
module_odbc = parseLine (line, TOKEN_ODBC);
}
if (module_max == "") {
module_max = parseLine (line, TOKEN_MAX);
}
if (module_min == "") {
module_min = parseLine (line, TOKEN_MIN);
}
if (module_description == "") {
module_description = parseLine (line, TOKEN_DESCRIPTION);
}
if (module_odbc_query == "") {
module_odbc_query = parseLine (line, TOKEN_ODBC_QUERY);
}
if (module_logevent == "") {
module_logevent = parseLine (line, TOKEN_LOGEVENT);
}
if (module_source == "") {
module_source = parseLine (line, TOKEN_SOURCE);
}
if (module_eventtype == "") {
module_eventtype = parseLine (line, TOKEN_EVENTTYPE);
}
if (module_pattern == "") {
module_pattern = parseLine (line, TOKEN_PATTERN);
}
iter++;
}
iter++;
}
/* Create module objects */
if (module_exec != "") {
module = new Pandora_Module_Exec (module_name,
module_exec);
} else if (module_proc != "") {
module = new Pandora_Module_Proc (module_name,
module_proc);
if (module_exec != "") {
module = new Pandora_Module_Exec (module_name,
module_exec);
} else if (module_proc != "") {
module = new Pandora_Module_Proc (module_name,
module_proc);
} else if (module_service != "") {
module = new Pandora_Module_Service (module_name,
module = new Pandora_Module_Service (module_name,
module_service);
} else if (module_freedisk != "") {
module = new Pandora_Module_Freedisk (module_name,
module = new Pandora_Module_Freedisk (module_name,
module_freedisk);
} else if (module_freememory != "") {
module = new Pandora_Module_Freememory (module_name);
module = new Pandora_Module_Freememory (module_name);
} else if (module_cpuusage != "") {
int cpu_id;
@ -202,7 +202,7 @@ Pandora_Module_Factory::getModuleFromDefinition (string definition) {
return NULL;
}
module = new Pandora_Module_Cpuusage (module_name,
module = new Pandora_Module_Cpuusage (module_name,
cpu_id);
} else if (module_odbc != "") {
@ -211,76 +211,76 @@ Pandora_Module_Factory::getModuleFromDefinition (string definition) {
module_odbc_query);
} else if (module_logevent != "") {
module = new Pandora_Module_Logevent (module_name,
module_source,
module_eventtype,
module_pattern);
} else {
return NULL;
}
module_source,
module_eventtype,
module_pattern);
} else {
return NULL;
}
if (module_description != "") {
module->setDescription (module_description);
}
module->setDescription (module_description);
}
type = Pandora_Module::parseModuleTypeFromString (module_type);
switch (type) {
case TYPE_GENERIC_DATA:
case TYPE_GENERIC_DATA_INC:
case TYPE_GENERIC_PROC:
module->setType (module_type);
numeric = true;
break;
case TYPE_GENERIC_DATA_STRING:
module->setType (module_type);
numeric = false;
break;
default:
pandoraDebug ("Bad module type \"%s\" while parsing %s module",
module_type.c_str (), module_name.c_str ());
delete module;
return NULL;
}
if (numeric) {
if (module_max != "") {
try {
int value = strtoint (module_max);
module->setMax (value);
} catch (Invalid_Conversion e) {
pandoraLog ("Invalid max value %s for module %s",
module_max.c_str (),
module_name.c_str ());
}
}
if (module_min != "") {
try {
int value = strtoint (module_min);
module->setMin (value);
} catch (Invalid_Conversion e) {
pandoraLog ("Invalid min value %s for module %s",
module_min.c_str (),
module_name.c_str ());
}
}
}
if (module_interval != "") {
int interval;
try {
interval = strtoint (module_interval);
module->setInterval (interval);
} catch (Invalid_Conversion e) {
pandoraLog ("Invalid interval value \"%s\" for module %s",
module_interval.c_str (),
module_name.c_str ());
}
}
return module;
switch (type) {
case TYPE_GENERIC_DATA:
case TYPE_GENERIC_DATA_INC:
case TYPE_GENERIC_PROC:
module->setType (module_type);
numeric = true;
break;
case TYPE_GENERIC_DATA_STRING:
module->setType (module_type);
numeric = false;
break;
default:
pandoraDebug ("Bad module type \"%s\" while parsing %s module",
module_type.c_str (), module_name.c_str ());
delete module;
return NULL;
}
if (numeric) {
if (module_max != "") {
try {
int value = strtoint (module_max);
module->setMax (value);
} catch (Invalid_Conversion e) {
pandoraLog ("Invalid max value %s for module %s",
module_max.c_str (),
module_name.c_str ());
}
}
if (module_min != "") {
try {
int value = strtoint (module_min);
module->setMin (value);
} catch (Invalid_Conversion e) {
pandoraLog ("Invalid min value %s for module %s",
module_min.c_str (),
module_name.c_str ());
}
}
}
if (module_interval != "") {
int interval;
try {
interval = strtoint (module_interval);
module->setInterval (interval);
} catch (Invalid_Conversion e) {
pandoraLog ("Invalid interval value \"%s\" for module %s",
module_interval.c_str (),
module_name.c_str ());
}
}
return module;
}

2
pandora_agents/win32/modules/pandora_module_factory.h
View File

@ -32,7 +32,7 @@ using namespace Pandora_Modules;
* Factoy to create Pandora_Module objects by parsing a definition.
*/
namespace Pandora_Module_Factory {
Pandora_Module * getModuleFromDefinition (string definition);
Pandora_Module * getModuleFromDefinition (string definition);
}
#endif

20
pandora_agents/win32/modules/pandora_module_freedisk.cc
View File

@ -37,13 +37,13 @@ using namespace Pandora_Strutils;
*/
Pandora_Module_Freedisk::Pandora_Module_Freedisk (string name, string disk_id)
: Pandora_Module (name) {
this->disk_id = disk_id;
transform (disk_id.begin (), disk_id.end (),
this->disk_id.begin (), (int (*) (int)) toupper);
this->setKind (module_freedisk_str);
this->disk_id = disk_id;
transform (disk_id.begin (), disk_id.end (),
this->disk_id.begin (), (int (*) (int)) toupper);
this->setKind (module_freedisk_str);
}
void
@ -51,10 +51,10 @@ Pandora_Module_Freedisk::run () {
long res;
try {
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
try {
res = Pandora_Wmi::getDiskFreeSpace (this->disk_id);

16
pandora_agents/win32/modules/pandora_module_freedisk.h
View File

@ -29,14 +29,14 @@ namespace Pandora_Modules {
* Module to retrieve the free space available in a logical volume
* disk.
*/
class Pandora_Module_Freedisk : public Pandora_Module {
private:
string disk_id;
public:
Pandora_Module_Freedisk (string name, string disk_id);
void run ();
};
class Pandora_Module_Freedisk : public Pandora_Module {
private:
string disk_id;
public:
Pandora_Module_Freedisk (string name, string disk_id);
void run ();
};
}
#endif

10
pandora_agents/win32/modules/pandora_module_freememory.cc
View File

@ -35,7 +35,7 @@ using namespace Pandora_Strutils;
Pandora_Module_Freememory::Pandora_Module_Freememory (string name)
: Pandora_Module (name) {
this->setKind (module_freememory_str);
this->setKind (module_freememory_str);
}
void
@ -43,10 +43,10 @@ Pandora_Module_Freememory::run () {
long res;
try {
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
try {
res = Pandora_Wmi::getFreememory ();

12
pandora_agents/win32/modules/pandora_module_freememory.h
View File

@ -27,12 +27,12 @@ namespace Pandora_Modules {
/**
* Module to retrieve the free memory amount of the system.
*/
class Pandora_Module_Freememory : public Pandora_Module {
public:
Pandora_Module_Freememory (string name);
void run ();
};
class Pandora_Module_Freememory : public Pandora_Module {
public:
Pandora_Module_Freememory (string name);
void run ();
};
}
#endif

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

@ -41,46 +41,46 @@ using namespace std;
* module definitions.
*/
Pandora_Modules::Pandora_Module_List::Pandora_Module_List (string filename) {
ifstream file (filename.c_str ());
string buffer;
unsigned int pos;
ifstream file (filename.c_str ());
string buffer;
unsigned int pos;
this->modules = new list<Pandora_Module *> ();
if (!file.is_open ()) {
return;
}
/* Read and set the file */
while (!file.eof ()) {
/* Set the value from each line */
getline (file, buffer);
/* Ignore blank or commented lines */
if (buffer[0] != '#' && buffer[0] != '\n' && buffer[0] != '\0') {
pos = buffer.find ("module_begin");
if (pos != string::npos) {
string str_module = buffer + "\n";
bool module_end = false;
while (!module_end) {
if (file.eof ()) {
break;
}
getline (file, buffer);
pos = buffer.find ("module_end");
module_end = (pos != string::npos);
str_module += buffer + "\n";
}
this->parseModuleDefinition (str_module);
}
}
}
file.close ();
current = new std::list<Pandora_Module *>::iterator ();
*current = modules->begin ();
this->modules = new list<Pandora_Module *> ();
if (!file.is_open ()) {
return;
}
/* Read and set the file */
while (!file.eof ()) {
/* Set the value from each line */
getline (file, buffer);
/* Ignore blank or commented lines */
if (buffer[0] != '#' && buffer[0] != '\n' && buffer[0] != '\0') {
pos = buffer.find ("module_begin");
if (pos != string::npos) {
string str_module = buffer + "\n";
bool module_end = false;
while (!module_end) {
if (file.eof ()) {
break;
}
getline (file, buffer);
pos = buffer.find ("module_end");
module_end = (pos != string::npos);
str_module += buffer + "\n";
}
this->parseModuleDefinition (str_module);
}
}
}
file.close ();
current = new std::list<Pandora_Module *>::iterator ();
*current = modules->begin ();
}
/**
@ -92,14 +92,14 @@ Pandora_Modules::Pandora_Module_List::~Pandora_Module_List () {
Pandora_Module *module;
std::list<Pandora_Module *>::iterator iter;
if (modules->size () > 0) {
iter = modules->begin ();
do {
module = *iter;
if (modules->size () > 0) {
iter = modules->begin ();
do {
module = *iter;
delete module;
iter++;
} while (iter != modules->end ());
}
iter++;
} while (iter != modules->end ());
}
delete modules;
delete current;
modules = NULL;
@ -108,64 +108,64 @@ Pandora_Modules::Pandora_Module_List::~Pandora_Module_List () {
void
Pandora_Modules::Pandora_Module_List::parseModuleDefinition (string definition) {
Pandora_Module *module;
Pandora_Module_Exec *module_exec;
Pandora_Module_Proc *module_proc;
Pandora_Module *module;
Pandora_Module_Exec *module_exec;
Pandora_Module_Proc *module_proc;
Pandora_Module_Service *module_service;
Pandora_Module_Freedisk *module_freedisk;
Pandora_Module_Cpuusage *module_cpuusage;
Pandora_Module_Freememory *module_freememory;
Pandora_Module_Odbc *module_odbc;
Pandora_Module_Logevent *module_logevent;
module = Pandora_Module_Factory::getModuleFromDefinition (definition);
if (module != NULL) {
switch (module->getModuleKind ()) {
case MODULE_EXEC:
module_exec = (Pandora_Module_Exec *) module;
modules->push_back (module_exec);
break;
case MODULE_PROC:
module_proc = (Pandora_Module_Proc *) module;
modules->push_back (module_proc);
break;
module = Pandora_Module_Factory::getModuleFromDefinition (definition);
if (module != NULL) {
switch (module->getModuleKind ()) {
case MODULE_EXEC:
module_exec = (Pandora_Module_Exec *) module;
modules->push_back (module_exec);
break;
case MODULE_PROC:
module_proc = (Pandora_Module_Proc *) module;
modules->push_back (module_proc);
break;
case MODULE_SERVICE:
module_service = (Pandora_Module_Service *) module;
modules->push_back (module_service);
break;
module_service = (Pandora_Module_Service *) module;
modules->push_back (module_service);
break;
case MODULE_FREEDISK:
module_freedisk = (Pandora_Module_Freedisk *) module;
modules->push_back (module_freedisk);
break;
modules->push_back (module_freedisk);
break;
case MODULE_FREEMEMORY:
module_freememory = (Pandora_Module_Freememory *) module;
modules->push_back (module_freememory);
break;
modules->push_back (module_freememory);
break;
case MODULE_CPUUSAGE:
module_cpuusage = (Pandora_Module_Cpuusage *) module;
modules->push_back (module_cpuusage);
break;
modules->push_back (module_cpuusage);
break;
case MODULE_ODBC:
module_odbc = (Pandora_Module_Odbc *) module;
modules->push_back (module_odbc);
break;
modules->push_back (module_odbc);
break;
case MODULE_LOGEVENT:
module_logevent = (Pandora_Module_Logevent *) module;
modules->push_back (module_logevent);
break;
default:
break;
}
}
modules->push_back (module_logevent);
break;
default:
break;
}
}
}
@ -176,7 +176,7 @@ Pandora_Modules::Pandora_Module_List::parseModuleDefinition (string definition)
*/
Pandora_Module *
Pandora_Modules::Pandora_Module_List::getCurrentValue () {
return *(*current);
return *(*current);
}
/**
@ -184,9 +184,9 @@ Pandora_Modules::Pandora_Module_List::getCurrentValue () {
*/
void
Pandora_Modules::Pandora_Module_List::goFirst () {
if (modules != NULL) {
*current = modules->begin ();
}
if (modules != NULL) {
*current = modules->begin ();
}
}
/**
@ -194,9 +194,9 @@ Pandora_Modules::Pandora_Module_List::goFirst () {
*/
void
Pandora_Modules::Pandora_Module_List::goLast () {
if (modules != NULL) {
*current = modules->end ();
}
if (modules != NULL) {
*current = modules->end ();
}
}
/**
@ -204,9 +204,9 @@ Pandora_Modules::Pandora_Module_List::goLast () {
*/
void
Pandora_Modules::Pandora_Module_List::goNext () {
if (current != NULL && !isLast ()) {
(*current)++;
}
if (current != NULL && !isLast ()) {
(*current)++;
}
}
/**
@ -214,9 +214,9 @@ Pandora_Modules::Pandora_Module_List::goNext () {
*/
void
Pandora_Modules::Pandora_Module_List::goPrev () {
if (current != NULL && !isFirst ()) {
(*current)--;
}
if (current != NULL && !isFirst ()) {
(*current)--;
}
}
/**
@ -224,10 +224,10 @@ Pandora_Modules::Pandora_Module_List::goPrev () {
*/
bool
Pandora_Modules::Pandora_Module_List::isLast () {
if (current == NULL || modules == NULL) {
return true;
}
return *current == modules->end ();
if (current == NULL || modules == NULL) {
return true;
}
return *current == modules->end ();
}
/**
@ -235,8 +235,8 @@ Pandora_Modules::Pandora_Module_List::isLast () {
*/
bool
Pandora_Modules::Pandora_Module_List::isFirst () {
if (current == NULL || modules == NULL) {
return true;
}
return *current == modules->begin ();
if (current == NULL || modules == NULL) {
return true;
}
return *current == modules->begin ();
}

8
pandora_agents/win32/modules/pandora_module_logevent.cc
View File

@ -35,10 +35,10 @@ using namespace Pandora_Modules;
Pandora_Module_Logevent::Pandora_Module_Logevent (string name, string source, string type, string pattern)
: Pandora_Module (name) {
this->source = source;
this->type = type;
this->pattern = pattern;
this->setKind (module_logevent_str);
this->source = source;
this->type = type;
this->pattern = pattern;
this->setKind (module_logevent_str);
}
void

18
pandora_agents/win32/modules/pandora_module_logevent.h
View File

@ -31,15 +31,15 @@ namespace Pandora_Modules {
* pattern. Events can be filtered by source and type.
*/
class Pandora_Module_Logevent : public Pandora_Module {
private:
string source;
string type;
string pattern;
public:
Pandora_Module_Logevent (string name, string source, string type, string pattern);
void run ();
};
class Pandora_Module_Logevent : public Pandora_Module {
private:
string source;
string type;
string pattern;
public:
Pandora_Module_Logevent (string name, string source, string type, string pattern);
void run ();
};
}
#endif

14
pandora_agents/win32/modules/pandora_module_odbc.cc
View File

@ -49,8 +49,8 @@ Pandora_Module_Odbc::Pandora_Module_Odbc (string name,
conf = Pandora_Agent_Conf::getInstance ();
this->setKind (module_odbc_str);
this->dsn = dsn;
this->username = conf->getValue ("odbc_" + dsn + "_username");
this->dsn = dsn;
this->username = conf->getValue ("odbc_" + dsn + "_username");
this->password = conf->getValue ("odbc_" + dsn + "_password");
this->query = query;
}
@ -72,7 +72,7 @@ Pandora_Module_Odbc::setDsn (string dsn) {
*/
void
Pandora_Module_Odbc::setUsername (string username) {
this->username = username;
this->username = username;
}
/**
@ -171,10 +171,10 @@ Pandora_Module_Odbc::doQuery () {
void
Pandora_Module_Odbc::run () {
try {
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
if (this->query == "") {
pandoraLog ("Error on module ODBC '%s': No query to execute",

14
pandora_agents/win32/modules/pandora_module_odbc.h
View File

@ -32,17 +32,17 @@ namespace Pandora_Modules {
* Module to retrieve a value based on a SQL query to an ODBC
* connection.
*/
class Pandora_Module_Odbc : public Pandora_Module {
private:
string dsn;
class Pandora_Module_Odbc : public Pandora_Module {
private:
string dsn;
string username;
string password;
string query;
Connection *con;
void doQuery ();
public:
Pandora_Module_Odbc (string name,
public:
Pandora_Module_Odbc (string name,
string dsn,
string query);
@ -56,8 +56,8 @@ namespace Pandora_Modules {
string getUsername ();
string getQuery ();
void run ();
};
void run ();
};
}
#endif

22
pandora_agents/win32/modules/pandora_module_proc.cc
View File

@ -36,12 +36,12 @@ using namespace Pandora_Strutils;
*/
Pandora_Module_Proc::Pandora_Module_Proc (string name, string process_name)
: Pandora_Module (name) {
this->process_name = process_name;
transform (process_name.begin (), process_name.end (),
this->process_name.begin (), (int (*) (int)) tolower);
this->setKind (module_proc_str);
this->process_name = process_name;
transform (process_name.begin (), process_name.end (),
this->process_name.begin (), (int (*) (int)) tolower);
this->setKind (module_proc_str);
}
void
@ -49,12 +49,12 @@ Pandora_Module_Proc::run () {
int res;
try {
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
res = Pandora_Wmi::isProcessRunning (this->process_name);
this->setOutput (inttostr (res));
this->setOutput (inttostr (res));
}

16
pandora_agents/win32/modules/pandora_module_proc.h
View File

@ -27,14 +27,14 @@ namespace Pandora_Modules {
/**
* Module to check that a process is running on the system.
*/
class Pandora_Module_Proc : public Pandora_Module {
private:
string process_name;
public:
Pandora_Module_Proc (string name, string process_name);
void run ();
};
class Pandora_Module_Proc : public Pandora_Module {
private:
string process_name;
public:
Pandora_Module_Proc (string name, string process_name);
void run ();
};
}
#endif

26
pandora_agents/win32/modules/pandora_module_service.cc
View File

@ -37,13 +37,13 @@ using namespace Pandora_Strutils;
*/
Pandora_Module_Service::Pandora_Module_Service (string name, string service_name)
: Pandora_Module (name) {
this->service_name = service_name;
transform (service_name.begin (), service_name.end (),
this->service_name.begin (), (int (*) (int)) tolower);
this->setKind (module_service_str);
this->service_name = service_name;
transform (service_name.begin (), service_name.end (),
this->service_name.begin (), (int (*) (int)) tolower);
this->setKind (module_service_str);
}
void
@ -51,11 +51,11 @@ Pandora_Module_Service::run () {
int res;
try {
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
res = Pandora_Wmi::isServiceRunning (this->service_name);
this->setOutput (inttostr (res));
Pandora_Module::run ();
} catch (Interval_Not_Fulfilled e) {
return;
}
res = Pandora_Wmi::isServiceRunning (this->service_name);
this->setOutput (inttostr (res));
}

16
pandora_agents/win32/modules/pandora_module_service.h
View File

@ -29,14 +29,14 @@ namespace Pandora_Modules {
* Module to check that a windows service is running on the
* system.
*/
class Pandora_Module_Service : public Pandora_Module {
private:
string service_name;
public:
Pandora_Module_Service (string name, string service_name);
void run ();
};
class Pandora_Module_Service : public Pandora_Module {
private:
string service_name;
public:
Pandora_Module_Service (string name, string service_name);
void run ();
};
}
#endif

46
pandora_agents/win32/pandora.h
View File

@ -40,32 +40,32 @@ namespace Pandora {
* A Key_Value object holds a relation between a value and a
* unique key value.
*/
class Key_Value {
private:
class Key_Value {
private:
string key;
string value;
public:
void parseLine (string str);
string getKey ();
string getValue ();
};
static const HKEY hkey = HKEY_LOCAL_MACHINE;
const char * const name = "PandoraFMSAgent";
const char * const display_name = "Pandora FMS agent";
const char * const description = "The Pandora FMS Agent service";
public:
void parseLine (string str);
string getKey ();
string getValue ();
};
static const HKEY hkey = HKEY_LOCAL_MACHINE;
const char * const name = "PandoraFMSAgent";
const char * const display_name = "Pandora FMS agent";
const char * const description = "The Pandora FMS Agent service";
void setPandoraInstallDir (string dir);
string getPandoraInstallDir ();
void setPandoraInstallPath (string path);
string getPandoraInstallPath ();
void setPandoraDebug (bool dbg);
void setPandoraInstallDir (string dir);
string getPandoraInstallDir ();
void setPandoraInstallPath (string path);
string getPandoraInstallPath ();
void setPandoraDebug (bool dbg);
bool getPandoraDebug ();
string getPandoraAgentVersion ();
void pandoraDebug (char *format, ...);
void pandoraLog (char *format, ...);
void pandoraFree (void * e);
string getPandoraAgentVersion ();
void pandoraDebug (char *format, ...);
void pandoraLog (char *format, ...);
void pandoraFree (void * e);
/**
* Super-class exception.
@ -73,7 +73,7 @@ namespace Pandora {
* Other exceptions generated in the application should inherate from
* this class. This allow a easier handling on throw and catch blocks.
*/
class Pandora_Exception { };
class Pandora_Exception { };
}
#endif /* __PANDORA_H__ */

72
pandora_agents/win32/pandora_agent_conf.cc
View File

@ -28,14 +28,14 @@ using namespace Pandora;
#define MAX_KEYS 100
Pandora::Pandora_Agent_Conf::Pandora_Agent_Conf () {
this->key_values = NULL;
this->key_values = NULL;
}
/**
* Destroy a Pandora_Agent_Conf object.
*/
Pandora::Pandora_Agent_Conf::~Pandora_Agent_Conf () {
delete key_values;
delete key_values;
}
Pandora_Agent_Conf *
@ -62,34 +62,34 @@ Pandora::Pandora_Agent_Conf::getInstance () {
void
Pandora::Pandora_Agent_Conf::setFile (string filename) {
ifstream file (filename.c_str ());
string buffer;
unsigned int pos;
string buffer;
unsigned int pos;
if (this->key_values)
delete this->key_values;
this->key_values = new list<Key_Value> ();
if (!file.is_open ()) {
return;
}
/* Read and set the file */
while (!file.eof ()) {
/* Set the value from each line */
getline (file, buffer);
/* Ignore blank or commented lines */
if (buffer[0] != '#' && buffer[0] != '\n' && buffer[0] != '\0') {
pos = buffer.find ("module_");
if (pos == string::npos) {
Key_Value kv;
kv.parseLine (buffer);
key_values->push_back (kv);
}
}
}
file.close ();
this->key_values = new list<Key_Value> ();
if (!file.is_open ()) {
return;
}
/* Read and set the file */
while (!file.eof ()) {
/* Set the value from each line */
getline (file, buffer);
/* Ignore blank or commented lines */
if (buffer[0] != '#' && buffer[0] != '\n' && buffer[0] != '\0') {
pos = buffer.find ("module_");
if (pos == string::npos) {
Key_Value kv;
kv.parseLine (buffer);
key_values->push_back (kv);
}
}
}
file.close ();
}
/**
@ -106,14 +106,14 @@ Pandora::Pandora_Agent_Conf::setFile (string filename) {
string
Pandora::Pandora_Agent_Conf::getValue (const string key)
{
std::list<Key_Value>::iterator i;
for (i = key_values->begin (); i != key_values->end (); i++) {
if ((*i).getKey () == key) {
return (*i).getValue ();
}
}
return "";
std::list<Key_Value>::iterator i;
for (i = key_values->begin (); i != key_values->end (); i++) {
if ((*i).getKey () == key) {
return (*i).getValue ();
}
}
return "";
}

2
pandora_agents/win32/pandora_agent_conf.h
View File

@ -28,7 +28,7 @@
using namespace std;
namespace Pandora {
/**
/**
* Agent main configuration class.
*
* Stores a list of Key_Value objects with the agent configuration.

174
pandora_agents/win32/pandora_strutils.cc
View File

@ -38,22 +38,22 @@ using namespace Pandora;
*/
string
Pandora_Strutils::trim (const string str) {
char * delims = " \t\r\n";
string result = str;
string::size_type index = result.find_last_not_of (delims);
if (index != string::npos) {
result.erase (++index);
}
index = result.find_first_not_of (delims);
if (index != std::string::npos) {
result.erase (0, index);
} else {
result.erase ();
}
return result;
char * delims = " \t\r\n";
string result = str;
string::size_type index = result.find_last_not_of (delims);
if (index != string::npos) {
result.erase (++index);
}
index = result.find_first_not_of (delims);
if (index != std::string::npos) {
result.erase (0, index);
} else {
result.erase ();
}
return result;
}
/**
@ -100,7 +100,7 @@ Pandora_Strutils::strUnicodeToAnsi (LPCWSTR s) {
*/
string
Pandora_Strutils::inttostr (const int i) {
return longtostr (i);
return longtostr (i);
}
/**
@ -113,10 +113,10 @@ Pandora_Strutils::inttostr (const int i) {
string
Pandora_Strutils::longtostr (const long i) {
std::ostringstream o;
o << i;
return o.str();
o << i;
return o.str();
}
/**
@ -128,10 +128,10 @@ Pandora_Strutils::longtostr (const long i) {
*/
string
Pandora_Strutils::longtohex (const long i) {
std::ostringstream o;
o << std::hex << i;
return o.str();
std::ostringstream o;
o << std::hex << i;
return o.str();
}
/**
@ -146,12 +146,12 @@ Pandora_Strutils::longtohex (const long i) {
*/
int
Pandora_Strutils::strtoint (const string str) {
int result;
if (! std::sscanf (str.c_str (), "%d", &result)) {
throw Invalid_Conversion ();
}
return result;
int result;
if (! std::sscanf (str.c_str (), "%d", &result)) {
throw Invalid_Conversion ();
}
return result;
}
/**
@ -165,12 +165,12 @@ Pandora_Strutils::strtoint (const string str) {
*/
double
Pandora_Strutils::strtodouble (const string str) {
double result;
if (! std::sscanf (str.c_str (), "%le", &result)) {
throw Invalid_Conversion ();
}
return result;
double result;
if (! std::sscanf (str.c_str (), "%le", &result)) {
throw Invalid_Conversion ();
}
return result;
}
/**
@ -185,13 +185,13 @@ Pandora_Strutils::strtodouble (const string str) {
*/
unsigned long long
Pandora_Strutils::strtoulong (const string str) {
unsigned long long result;
unsigned long long result;
if (! std::sscanf (str.c_str (), "%I64d", &result)) {
throw Invalid_Conversion ();
}
if (! std::sscanf (str.c_str (), "%I64d", &result)) {
throw Invalid_Conversion ();
}
return result;
return result;
}
/**
@ -205,30 +205,30 @@ Pandora_Strutils::strtoulong (const string str) {
*/
string
Pandora_Strutils::strreplace (string in, string pattern, string rep) {
int i = in.find (pattern);
int j;
if (i < 0) {
return in;
}
int plen = pattern.length ();
int rlen = rep.length ();
do {
in.replace(i, plen, rep);
i += rlen;
string rest = in.substr (i, in.length () - i);
j = rest.find (pattern);
i += j;
} while (j >= 0);
return in;
int i = in.find (pattern);
int j;
if (i < 0) {
return in;
}
int plen = pattern.length ();
int rlen = rep.length ();
do {
in.replace(i, plen, rep);
i += rlen;
string rest = in.substr (i, in.length () - i);
j = rest.find (pattern);
i += j;
} while (j >= 0);
return in;
}
inline bool
isseparator (char c, char const * const wstr) {
return (strchr (wstr, c) != NULL);
return (strchr (wstr, c) != NULL);
}
/**
@ -243,30 +243,30 @@ isseparator (char c, char const * const wstr) {
*/
void
Pandora_Strutils::stringtok (list<string> &l, string const &s,
char const * const separators) {
char const * const separators) {
const string::size_type strsize = s.size();
string::size_type i = 0;
while (i < strsize) {
/* eat leading whitespace */
while ((i < strsize) && (isseparator (s[i], separators))) {
i++;
}
if (i == strsize) {
return; /* nothing left but WS */
}
/* find end of word */
string::size_type j = i + 1;
while ((j < strsize) && (!isseparator (s[j], separators))) {
j++;
}
/* add word */
l.push_back (s.substr (i, j - i));
/* set up for next loop */
i = j + 1;
}
const string::size_type strsize = s.size();
string::size_type i = 0;
while (i < strsize) {
/* eat leading whitespace */
while ((i < strsize) && (isseparator (s[i], separators))) {
i++;
}
if (i == strsize) {
return; /* nothing left but WS */
}
/* find end of word */
string::size_type j = i + 1;
while ((j < strsize) && (!isseparator (s[j], separators))) {
j++;
}
/* add word */
l.push_back (s.substr (i, j - i));
/* set up for next loop */
i = j + 1;
}
}

30
pandora_agents/win32/pandora_strutils.h
View File

@ -31,32 +31,32 @@ using namespace std;
* Operations with strings.
*/
namespace Pandora_Strutils {
/**
/**
* String super-class exception.
*/
class String_Exception : Pandora::Pandora_Exception {};
class String_Exception : Pandora::Pandora_Exception {};
/**
* Exception throwed when a conversion could not be success.
*/
class Invalid_Conversion : Pandora_Strutils::String_Exception {};
class Invalid_Conversion : Pandora_Strutils::String_Exception {};
string trim (const string str);
string trim (const string str);
LPSTR strUnicodeToAnsi (LPCWSTR s);
string inttostr (const int i);
string longtostr (const long i);
string longtohex (const long i);
int strtoint (const string str);
double strtodouble (const string str);
string inttostr (const int i);
string longtostr (const long i);
string longtohex (const long i);
int strtoint (const string str);
double strtodouble (const string str);
unsigned long long strtoulong (const string str);
string strreplace (string in, string pattern, string rep);
string strreplace (string in, string pattern, string rep);
void
stringtok (list<string> &l, string const &s,
char const * const separators = " \t\n");
void
stringtok (list<string> &l, string const &s,
char const * const separators = " \t\n");
}
#endif /* __STRUTILS_H__ */

296
pandora_agents/win32/ssh/pandora_ssh_client.cc
View File

@ -42,7 +42,7 @@ using namespace Pandora;
* @param e Numeric error code.
*/
Connection_Failed::Connection_Failed (int e) {
err_number = e;
err_number = e;
}
/**
@ -52,7 +52,7 @@ Connection_Failed::Connection_Failed (int e) {
*/
int
Connection_Failed::getError () {
return err_number;
return err_number;
}
/**
@ -61,18 +61,18 @@ Connection_Failed::getError () {
* @param e Error description.
*/
Scp_Failed::Scp_Failed (char *e) {
errmsg = strdup (e);
errmsg = strdup (e);
}
/**
* Creates a SSH client object and initialize its attributes.
*/
Pandora_Ssh_Client::Pandora_Ssh_Client () {
sock = 0;
fingerprint = "";
session = NULL;
channel = NULL;
return;
sock = 0;
fingerprint = "";
session = NULL;
channel = NULL;
return;
}
/**
@ -83,11 +83,11 @@ Pandora_Ssh_Client::Pandora_Ssh_Client () {
* @see disconnect
*/
Pandora_Ssh_Client::~Pandora_Ssh_Client () {
if (session != NULL) {
disconnect ();
}
return;
if (session != NULL) {
disconnect ();
}
return;
}
/**
@ -97,79 +97,79 @@ Pandora_Ssh_Client::~Pandora_Ssh_Client () {
*/
void
Pandora_Ssh_Client::disconnect () {
if (channel != NULL) {
libssh2_channel_send_eof (channel);
libssh2_channel_close (channel);
libssh2_channel_wait_closed (channel);
libssh2_channel_free (channel);
channel = NULL;
}
if (session != NULL) {
libssh2_session_disconnect (session, "");
libssh2_session_free (session);
session = NULL;
}
if (sock != 0) {
closesocket (sock);
sock = 0;
}
if (channel != NULL) {
libssh2_channel_send_eof (channel);
libssh2_channel_close (channel);
libssh2_channel_wait_closed (channel);
libssh2_channel_free (channel);
channel = NULL;
}
if (session != NULL) {
libssh2_session_disconnect (session, "");
libssh2_session_free (session);
session = NULL;
}
if (sock != 0) {
closesocket (sock);
sock = 0;
}
}
void
Pandora_Ssh_Client::newConnection (const string host, const int port) {
struct sockaddr_in sin;
struct hostent *resolv = NULL;
WSADATA wsadata;
string finger_aux;
char char_aux[3];
if (session != NULL) {
throw Session_Already_Opened ();
}
WSAStartup (2, &wsadata);
sock = socket (AF_INET, SOCK_STREAM, 0);
if (sock == -1) {
throw Socket_Error ();
}
resolv = (struct hostent *) gethostbyname (host.c_str ());
if (resolv == NULL) {
disconnect ();
throw Resolv_Failed ();
}
sin.sin_family = AF_INET;
sin.sin_port = htons (port);
sin.sin_addr = *((struct in_addr *)resolv->h_addr);
if (connect (sock, (struct sockaddr*) (&sin),
sizeof (struct sockaddr_in)) == -1) {
disconnect ();
throw Connection_Failed (WSAGetLastError ());
}
session = libssh2_session_init();
if (libssh2_session_startup (session, sock) != 0) {
disconnect ();
throw Session_Error ();
}
/* Get the fingerprint and transform it to a hexadecimal readable
string */
finger_aux = libssh2_hostkey_hash (session,
LIBSSH2_HOSTKEY_HASH_MD5);
fingerprint = "";
for (int i = 0; i < 16; i++) {
sprintf (char_aux, "%02X:", (unsigned char) finger_aux[i]);
fingerprint += (char *) char_aux;
}
fingerprint.erase (fingerprint.length () - 1, 2);
struct sockaddr_in sin;
struct hostent *resolv = NULL;
WSADATA wsadata;
string finger_aux;
char char_aux[3];
if (session != NULL) {
throw Session_Already_Opened ();
}
WSAStartup (2, &wsadata);
sock = socket (AF_INET, SOCK_STREAM, 0);
if (sock == -1) {
throw Socket_Error ();
}
resolv = (struct hostent *) gethostbyname (host.c_str ());
if (resolv == NULL) {
disconnect ();
throw Resolv_Failed ();
}
sin.sin_family = AF_INET;
sin.sin_port = htons (port);
sin.sin_addr = *((struct in_addr *)resolv->h_addr);
if (connect (sock, (struct sockaddr*) (&sin),
sizeof (struct sockaddr_in)) == -1) {
disconnect ();
throw Connection_Failed (WSAGetLastError ());
}
session = libssh2_session_init();
if (libssh2_session_startup (session, sock) != 0) {
disconnect ();
throw Session_Error ();
}
/* Get the fingerprint and transform it to a hexadecimal readable
string */
finger_aux = libssh2_hostkey_hash (session,
LIBSSH2_HOSTKEY_HASH_MD5);
fingerprint = "";
for (int i = 0; i < 16; i++) {
sprintf (char_aux, "%02X:", (unsigned char) finger_aux[i]);
fingerprint += (char *) char_aux;
}
fingerprint.erase (fingerprint.length () - 1, 2);
}
/**
@ -190,26 +190,26 @@ Pandora_Ssh_Client::newConnection (const string host, const int port) {
*/
void
Pandora_Ssh_Client::connectWithPublicKey (const string host, const int port,
const string username, const string filename_pubkey,
const string filename_privkey, const string passphrase) {
try {
newConnection (host, port);
} catch (Session_Already_Opened e) {
}
if (session != NULL) {
if (libssh2_userauth_publickey_fromfile (session,
username.c_str (),
filename_pubkey.c_str (),
filename_privkey.c_str (),
passphrase.c_str ())) {
disconnect ();
throw Authentication_Failed ();
}
}
const string username, const string filename_pubkey,
const string filename_privkey, const string passphrase) {
try {
newConnection (host, port);
} catch (Session_Already_Opened e) {
}
if (session != NULL) {
if (libssh2_userauth_publickey_fromfile (session,
username.c_str (),
filename_pubkey.c_str (),
filename_privkey.c_str (),
passphrase.c_str ())) {
disconnect ();
throw Authentication_Failed ();
}
}
return;
}
/**
* Copy a file using a SSH connection via scp method.
*
@ -232,54 +232,54 @@ Pandora_Ssh_Client::connectWithPublicKey (const string host, const int port,
void
Pandora_Ssh_Client::scpFileFilename (const string remote_filename,
const string filename) {
LIBSSH2_CHANNEL *scp_channel;
size_t to_send, sent;
char *errmsg;
int errmsg_len;
string buffer;
if (session == NULL) {
throw Session_Not_Opened ();
}
try {
buffer = Pandora_File::readFile (filename);
} catch (Pandora_File::File_Not_Found e) {
pandoraLog ("Pandora_Ssh_Client: File %s not found",
filename.c_str());
LIBSSH2_CHANNEL *scp_channel;
size_t to_send, sent;
char *errmsg;
int errmsg_len;
string buffer;
if (session == NULL) {
throw Session_Not_Opened ();
}
try {
buffer = Pandora_File::readFile (filename);
} catch (Pandora_File::File_Not_Found e) {
pandoraLog ("Pandora_Ssh_Client: File %s not found",
filename.c_str());
throw e;
}
to_send = buffer.length ();
scp_channel = libssh2_scp_send (session, remote_filename.c_str (), 0666,
to_send);
if (scp_channel == NULL) {
throw Channel_Error ();
}
libssh2_channel_set_blocking (scp_channel, 1);
/* FIXME: It may crash if the scp fails, maybe because of a libssh2 bug */
sent = libssh2_channel_write (scp_channel, buffer.c_str (), to_send);
if (sent < 0) {
Scp_Failed *e;
errmsg = (char *) malloc (sizeof (char) * 1000);
libssh2_session_last_error (session, &errmsg, &errmsg_len, 1);
pandoraLog ("Error %d on SCP %s", sent, errmsg);
e = new Scp_Failed (errmsg);
libssh2_channel_close (scp_channel);
libssh2_channel_wait_closed (scp_channel);
libssh2_channel_free (scp_channel);
Pandora::pandoraFree (errmsg);
throw *e;
}
libssh2_channel_send_eof (scp_channel);
libssh2_channel_close (scp_channel);
libssh2_channel_wait_closed (scp_channel);
libssh2_channel_free (scp_channel);
}
to_send = buffer.length ();
scp_channel = libssh2_scp_send (session, remote_filename.c_str (), 0666,
to_send);
if (scp_channel == NULL) {
throw Channel_Error ();
}
libssh2_channel_set_blocking (scp_channel, 1);
/* FIXME: It may crash if the scp fails, maybe because of a libssh2 bug */
sent = libssh2_channel_write (scp_channel, buffer.c_str (), to_send);
if (sent < 0) {
Scp_Failed *e;
errmsg = (char *) malloc (sizeof (char) * 1000);
libssh2_session_last_error (session, &errmsg, &errmsg_len, 1);
pandoraLog ("Error %d on SCP %s", sent, errmsg);
e = new Scp_Failed (errmsg);
libssh2_channel_close (scp_channel);
libssh2_channel_wait_closed (scp_channel);
libssh2_channel_free (scp_channel);
Pandora::pandoraFree (errmsg);
throw *e;
}
libssh2_channel_send_eof (scp_channel);
libssh2_channel_close (scp_channel);
libssh2_channel_wait_closed (scp_channel);
libssh2_channel_free (scp_channel);
}
/**
@ -292,5 +292,5 @@ Pandora_Ssh_Client::scpFileFilename (const string remote_filename,
*/
string
Pandora_Ssh_Client::getFingerprint () {
return this->fingerprint;
return this->fingerprint;
}

120
pandora_agents/win32/ssh/pandora_ssh_client.h
View File

@ -31,110 +31,110 @@ using namespace std;
* SSH connection classes.
*/
namespace SSH {
/**
/**
* A SSH super-class exception.
*/
class SSH_Exception : public Pandora::Pandora_Exception {
};
class SSH_Exception : public Pandora::Pandora_Exception {
};
/**
/**
* A SSH session has already be opened and a new
* connection is attemped without closing it.
*/
class Session_Already_Opened : public SSH::SSH_Exception {
};
class Session_Already_Opened : public SSH::SSH_Exception {
};
/**
* A SSH operations is tried and a session
* has still not be opened.
*/
class Session_Not_Opened : public SSH::SSH_Exception {
};
class Session_Not_Opened : public SSH::SSH_Exception {
};
/**
* There were unknown problems with the SSH session.
*/
class Session_Error : public SSH::SSH_Exception {
};
class Session_Error : public SSH::SSH_Exception {
};
/**
* The SSH authentication fails when connecting.
*/
class Authentication_Failed : public SSH::SSH_Exception {
};
class Authentication_Failed : public SSH::SSH_Exception {
};
/**
* The host could not be resolved.
*/
class Resolv_Failed : public SSH::SSH_Exception {
};
class Resolv_Failed : public SSH::SSH_Exception {
};
/**
* Unknown socket error.
*/
class Socket_Error : public SSH::SSH_Exception {
};
class Socket_Error : public SSH::SSH_Exception {
};
/**
* An error happened with a file.
*/
class File_Error : public SSH::SSH_Exception {
};
class File_Error : public SSH::SSH_Exception {
};
/**
* An error occured with the SSH channel.
*/
class Channel_Error : public SSH::SSH_Exception {
};
class Channel_Error : public SSH::SSH_Exception {
};
/**
* Connection failed with the host.
*/
class Connection_Failed : public SSH::SSH_Exception {
private:
int err_number;
public:
Connection_Failed (int e);
int getError ();
};
class Connection_Failed : public SSH::SSH_Exception {
private:
int err_number;
public:
Connection_Failed (int e);
int getError ();
};
/**
* The scp operation failed due to some unknow error.
*/
class Scp_Failed : public SSH::SSH_Exception {
private:
char *errmsg;
public:
Scp_Failed (char *e);
~Scp_Failed () { Pandora::pandoraFree (errmsg); };
};
/**
class Scp_Failed : public SSH::SSH_Exception {
private:
char *errmsg;
public:
Scp_Failed (char *e);
~Scp_Failed () { Pandora::pandoraFree (errmsg); };
};
/**
* Client to perform a SSH connection to a host.
*/
class Pandora_Ssh_Client {
private:
int sock;
string fingerprint;
LIBSSH2_SESSION *session;
LIBSSH2_CHANNEL *channel;
void newConnection (const string host, const int port);
public:
Pandora_Ssh_Client ();
~Pandora_Ssh_Client ();
void connectWithPublicKey (const string host, const int port,
const string username, const string filename_pubkey,
const string filename_privkey, const string passphrase);
void disconnect ();
void scpFileFilename (const string remote_filename,
const string filename);
string getFingerprint ();
};
class Pandora_Ssh_Client {
private:
int sock;
string fingerprint;
LIBSSH2_SESSION *session;
LIBSSH2_CHANNEL *channel;
void newConnection (const string host, const int port);
public:
Pandora_Ssh_Client ();
~Pandora_Ssh_Client ();
void connectWithPublicKey (const string host, const int port,
const string username, const string filename_pubkey,
const string filename_privkey, const string passphrase);
void disconnect ();
void scpFileFilename (const string remote_filename,
const string filename);
string getFingerprint ();
};
}
#endif

240
pandora_agents/win32/ssh/pandora_ssh_test.cc
View File

@ -33,22 +33,22 @@ using namespace SSH;
* all the information to perform a SSH test.
*/
Pandora_SSH_Test::Pandora_SSH_Test () {
string conf_file;
conf_file = Pandora::getPandoraInstallDir ();
conf_file += "pandora_agent.conf";
conf = Pandora::Pandora_Agent_Conf::getInstance ();
string conf_file;
conf_file = Pandora::getPandoraInstallDir ();
conf_file += "pandora_agent.conf";
conf = Pandora::Pandora_Agent_Conf::getInstance ();
conf->setFile (conf_file);
ssh_client = new SSH::Pandora_Ssh_Client ();
ssh_client = new SSH::Pandora_Ssh_Client ();
}
/**
* Deletes a Pandora_SSH_Test object.
*/
Pandora_SSH_Test::~Pandora_SSH_Test () {
delete conf;
delete ssh_client;
delete conf;
delete ssh_client;
}
/**
@ -70,128 +70,128 @@ Pandora_SSH_Test::~Pandora_SSH_Test () {
*/
void
Pandora_SSH_Test::test () {
string pubkey_file, privkey_file, tmp_filename;
string remote_host, remote_filepath, tmp_filepath;
TiXmlDocument *doc;
TiXmlDeclaration *decl;
bool saved;
pubkey_file = Pandora::getPandoraInstallDir ();
pubkey_file += "key\\id_dsa.pub";
string pubkey_file, privkey_file, tmp_filename;
string remote_host, remote_filepath, tmp_filepath;
TiXmlDocument *doc;
TiXmlDeclaration *decl;
bool saved;
pubkey_file = Pandora::getPandoraInstallDir ();
pubkey_file += "key\\id_dsa.pub";
if (! Pandora_File::fileExists (pubkey_file)) {
cout << "Public key file " << pubkey_file << " not found."
<< endl;
return;
}
cout << "Public key file " << pubkey_file << " exists." << endl;
cout << "Public key file " << pubkey_file << " exists." << endl;
privkey_file = Pandora::getPandoraInstallDir ();
privkey_file += "key\\id_dsa";
privkey_file = Pandora::getPandoraInstallDir ();
privkey_file += "key\\id_dsa";
if (! Pandora_File::fileExists (privkey_file)) {
cout << "Private key file " << privkey_file << " not found."
<< endl;
return;
}
cout << "Private key file: " << privkey_file << " exists." << endl;
cout << "Private key file: " << privkey_file << " exists." << endl;
remote_host = this->conf->getValue ("server_ip");
cout << "Connecting with " << remote_host << "." << endl;
try {
this->ssh_client->connectWithPublicKey (remote_host.c_str (), 22,
"pandora", pubkey_file,
privkey_file, "");
} catch (Authentication_Failed e) {
cout << "Authentication Failed when connecting to "
<< remote_host << endl;
cout << "Check the remote host configuration and the public/private key files."
<< endl;
throw e;
} catch (Socket_Error e) {
cout << "Socket error when connecting to "
<< remote_host << endl;
cout << "Check the network configuration." << endl;
throw e;
} catch (Resolv_Failed e) {
cout << "Could not resolv "
<< remote_host << endl;
cout << "Check the network configuration." << endl;
throw e;
} catch (Connection_Failed e) {
cout << "Connection error number " << e.getError () << endl;
cout << "Check the network configuration." << endl;
throw e;
} catch (Session_Error e) {
cout << "Error while opening SSH session." << endl;
cout << "Check the network configuration." << endl;
throw e;
}
cout << "Authentication successful." << endl;
cout << "Host fingerprint: " << this->ssh_client->getFingerprint ()
<< endl;
tmp_filename = "ssh.test";
tmp_filepath = conf->getValue ("temporal");
if (tmp_filepath[tmp_filepath.length () - 1] != '\\') {
tmp_filepath += "\\";
}
tmp_filepath += tmp_filename;
decl = new TiXmlDeclaration( "1.0", "ISO-8859-1", "" );
doc = new TiXmlDocument (tmp_filepath);
doc->InsertEndChild (*decl);
saved = doc->SaveFile();
delete doc;
if (!saved) {
Pandora::Pandora_Exception e;
cout << "Error when saving the XML in " << tmp_filepath << endl;
cout << "Check the configuration file" << endl;
throw e;
}
cout << "Created a blank XML file in " << tmp_filepath<< endl;
remote_filepath = conf->getValue ("server_path");
if (remote_filepath[remote_filepath.length () - 1] != '/') {
remote_filepath += "/";
}
cout << "Remote copying " << tmp_filepath << " on server " << remote_host
<< " at " << remote_filepath << tmp_filename << endl;
try {
ssh_client->scpFileFilename (remote_filepath + tmp_filename,
tmp_filepath);
} catch (Session_Not_Opened e) {
ssh_client->disconnect();
cout << "The SSH session could not be created." << endl;
cout << "Check the network configuration." << endl;
try {
Pandora_File::removeFile (tmp_filepath);
} catch (Pandora::Pandora_Exception e) {
}
throw e;
} catch (Scp_Failed e) {
ssh_client->disconnect();
cout << "The copying operation could not finished." << endl;
cout << "Check the network configuration." << endl;
try {
Pandora_File::removeFile (tmp_filepath);
} catch (Pandora::Pandora_Exception e) {
}
throw e;
} catch (Pandora::Pandora_Exception e) {
ssh_client->disconnect();
cout << "An unhandled exception happened." << endl;
throw e;
}
cout << "Successfuly file copied to remote host " << endl;
ssh_client->disconnect();
cout << "Successfuly disconnected from remote host " << endl;
try {
Pandora_File::removeFile (tmp_filepath);
} catch (Pandora_File::Delete_Error e) {
}
cout << "The SSH test was successful!" << endl;
cout << "Connecting with " << remote_host << "." << endl;
try {
this->ssh_client->connectWithPublicKey (remote_host.c_str (), 22,
"pandora", pubkey_file,
privkey_file, "");
} catch (Authentication_Failed e) {
cout << "Authentication Failed when connecting to "
<< remote_host << endl;
cout << "Check the remote host configuration and the public/private key files."
<< endl;
throw e;
} catch (Socket_Error e) {
cout << "Socket error when connecting to "
<< remote_host << endl;
cout << "Check the network configuration." << endl;
throw e;
} catch (Resolv_Failed e) {
cout << "Could not resolv "
<< remote_host << endl;
cout << "Check the network configuration." << endl;
throw e;
} catch (Connection_Failed e) {
cout << "Connection error number " << e.getError () << endl;
cout << "Check the network configuration." << endl;
throw e;
} catch (Session_Error e) {
cout << "Error while opening SSH session." << endl;
cout << "Check the network configuration." << endl;
throw e;
}
cout << "Authentication successful." << endl;
cout << "Host fingerprint: " << this->ssh_client->getFingerprint ()
<< endl;
tmp_filename = "ssh.test";
tmp_filepath = conf->getValue ("temporal");
if (tmp_filepath[tmp_filepath.length () - 1] != '\\') {
tmp_filepath += "\\";
}
tmp_filepath += tmp_filename;
decl = new TiXmlDeclaration( "1.0", "ISO-8859-1", "" );
doc = new TiXmlDocument (tmp_filepath);
doc->InsertEndChild (*decl);
saved = doc->SaveFile();
delete doc;
if (!saved) {
Pandora::Pandora_Exception e;
cout << "Error when saving the XML in " << tmp_filepath << endl;
cout << "Check the configuration file" << endl;
throw e;
}
cout << "Created a blank XML file in " << tmp_filepath<< endl;
remote_filepath = conf->getValue ("server_path");
if (remote_filepath[remote_filepath.length () - 1] != '/') {
remote_filepath += "/";
}
cout << "Remote copying " << tmp_filepath << " on server " << remote_host
<< " at " << remote_filepath << tmp_filename << endl;
try {
ssh_client->scpFileFilename (remote_filepath + tmp_filename,
tmp_filepath);
} catch (Session_Not_Opened e) {
ssh_client->disconnect();
cout << "The SSH session could not be created." << endl;
cout << "Check the network configuration." << endl;
try {
Pandora_File::removeFile (tmp_filepath);
} catch (Pandora::Pandora_Exception e) {
}
throw e;
} catch (Scp_Failed e) {
ssh_client->disconnect();
cout << "The copying operation could not finished." << endl;
cout << "Check the network configuration." << endl;
try {
Pandora_File::removeFile (tmp_filepath);
} catch (Pandora::Pandora_Exception e) {
}
throw e;
} catch (Pandora::Pandora_Exception e) {
ssh_client->disconnect();
cout << "An unhandled exception happened." << endl;
throw e;
}
cout << "Successfuly file copied to remote host " << endl;
ssh_client->disconnect();
cout << "Successfuly disconnected from remote host " << endl;
try {
Pandora_File::removeFile (tmp_filepath);
} catch (Pandora_File::Delete_Error e) {
}
cout << "The SSH test was successful!" << endl;
}

18
pandora_agents/win32/ssh/pandora_ssh_test.h
View File

@ -31,15 +31,15 @@ namespace SSH {
* An object of this class will read the configuration file
* and copy a blank xml file into remote server path.
*/
class Pandora_SSH_Test {
private:
Pandora_Ssh_Client *ssh_client;
Pandora::Pandora_Agent_Conf *conf;
public:
Pandora_SSH_Test ();
~Pandora_SSH_Test ();
void test ();
};
class Pandora_SSH_Test {
private:
Pandora_Ssh_Client *ssh_client;
Pandora::Pandora_Agent_Conf *conf;
public:
Pandora_SSH_Test ();
~Pandora_SSH_Test ();
void test ();
};
}
#endif

14
pandora_agents/win32/windows/pandora_windows_info.cc
View File

@ -33,7 +33,7 @@ using namespace Pandora_Windows_Info;
*/
string
Pandora_Windows_Info::getOSName () {
return Pandora_Wmi::getOSName ();
return Pandora_Wmi::getOSName ();
}
/**
@ -43,7 +43,7 @@ Pandora_Windows_Info::getOSName () {
*/
string
Pandora_Windows_Info::getOSVersion () {
return Pandora_Wmi::getOSVersion ();
return Pandora_Wmi::getOSVersion ();
}
/**
@ -53,7 +53,7 @@ Pandora_Windows_Info::getOSVersion () {
*/
string
Pandora_Windows_Info::getOSBuild () {
return Pandora_Wmi::getOSBuild();
return Pandora_Wmi::getOSBuild();
}
/**
@ -63,7 +63,7 @@ Pandora_Windows_Info::getOSBuild () {
*/
string
Pandora_Windows_Info::getSystemName () {
return Pandora_Wmi::getSystemName ();
return Pandora_Wmi::getSystemName ();
}
/**
@ -75,9 +75,9 @@ string
Pandora_Windows_Info::getSystemPath () {
char buffer[MAX_PATH];
::GetWindowsDirectory (buffer, MAX_PATH+1);
string str_path = buffer;
::GetWindowsDirectory (buffer, MAX_PATH+1);
string str_path = buffer;
str_path = trim (str_path);
return str_path;
}

11
pandora_agents/win32/windows/pandora_windows_info.h
View File

@ -34,11 +34,10 @@ using namespace std;
* Windows information functions.
*/
namespace Pandora_Windows_Info {
string getOSName ();
string getOSVersion ();
string getOSBuild ();
string getSystemName ();
string getSystemPath ();
string getOSName ();
string getOSVersion ();
string getOSBuild ();
string getSystemName ();
string getSystemPath ();
}
#endif

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

@ -337,7 +337,7 @@ Pandora_Wmi::getOSBuild () {
FOR_EACH (quickfix, quickfixes, NULL) {
dhGetValue (L"%s", &build, quickfix,
L".Version");
ret = build;
dhFreeString (build);
@ -345,7 +345,7 @@ Pandora_Wmi::getOSBuild () {
} catch (string errstr) {
pandoraLog ("getOSBuild error. %s", errstr.c_str ());
}
return ret;
}
@ -370,7 +370,7 @@ Pandora_Wmi::getSystemName () {
FOR_EACH (quickfix, quickfixes, NULL) {
dhGetValue (L"%s", &name, quickfix,
L".CSName");
ret = name;
dhFreeString (name);
@ -378,7 +378,7 @@ Pandora_Wmi::getSystemName () {
} catch (string errstr) {
pandoraLog ("getSystemName error. %s", errstr.c_str ());
}
return ret;
}

750
pandora_agents/win32/windows_service.cc
View File

@ -35,9 +35,9 @@ static void WINAPI windows_service_start (DWORD argc, LPTSTR *argv);
static VOID WINAPI windows_service_ctrl_handler (DWORD dwOpcode);
static VOID svc_format_message (LPSTR szString, int iCount);
static void SetWindowsServiceStatus (DWORD dwCurrentState,
DWORD dwWin32ExitCode,
DWORD dwCheckPoint,
DWORD dwWaitHint);
DWORD dwWin32ExitCode,
DWORD dwCheckPoint,
DWORD dwWaitHint);
static void ErrorStopService (LPTSTR lpszAPI);
/**
@ -46,9 +46,9 @@ static void ErrorStopService (LPTSTR lpszAPI);
* All the attributes are set to NULL.
*/
Windows_Service::Windows_Service () {
service_name = NULL;
service_display_name = NULL;
service_description = NULL;
service_name = NULL;
service_display_name = NULL;
service_description = NULL;
}
/**
* Set the values of the service to run.
@ -59,16 +59,16 @@ Windows_Service::Windows_Service () {
* @param svc_description Long description of the service.
*/
Windows_Service::Windows_Service (const char * svc_name,
const char * svc_display_name,
const char * svc_description) {
sleep_time = 0;
run_function = NULL;
init_function = NULL;
stop_event = CreateEvent (NULL, TRUE, FALSE, NULL);
service_name = (char *) svc_name;
service_display_name = (char *) svc_display_name;
service_description = (char *) svc_description;
current_service = this;
const char * svc_display_name,
const char * svc_description) {
sleep_time = 0;
run_function = NULL;
init_function = NULL;
stop_event = CreateEvent (NULL, TRUE, FALSE, NULL);
service_name = (char *) svc_name;
service_display_name = (char *) svc_display_name;
service_description = (char *) svc_description;
current_service = this;
}
/**
@ -84,8 +84,8 @@ Windows_Service::~Windows_Service () {
*/
void
Windows_Service::setRunFunction (void (Windows_Service::*f) ()) {
run_function = f;
current_service->run_function = f;
run_function = f;
current_service->run_function = f;
}
/**
@ -97,8 +97,8 @@ Windows_Service::setRunFunction (void (Windows_Service::*f) ()) {
*/
void
Windows_Service::setInitFunction (void (Windows_Service::*f) ()) {
init_function = f;
current_service->init_function = f;
init_function = f;
current_service->init_function = f;
}
/**
@ -111,14 +111,14 @@ Windows_Service::setInitFunction (void (Windows_Service::*f) ()) {
*/
void
Windows_Service::execRunFunction () {
if (run_function != NULL) {
(this->*run_function) ();
if (sleep_time > 0) {
while (WaitForSingleObject (stop_event, sleep_time) != WAIT_OBJECT_0) {
(this->*run_function) ();
}
}
}
if (run_function != NULL) {
(this->*run_function) ();
if (sleep_time > 0) {
while (WaitForSingleObject (stop_event, sleep_time) != WAIT_OBJECT_0) {
(this->*run_function) ();
}
}
}
}
/**
@ -126,9 +126,9 @@ Windows_Service::execRunFunction () {
*/
void
Windows_Service::execInitFunction () {
if (init_function != NULL) {
(this->*init_function) ();
}
if (init_function != NULL) {
(this->*init_function) ();
}
}
/**
@ -138,7 +138,7 @@ Windows_Service::execInitFunction () {
*/
LPSTR
Windows_Service::getServiceName () {
return service_name;
return service_name;
}
/**
@ -152,8 +152,8 @@ Windows_Service::getServiceName () {
*/
void
Windows_Service::setSleepTime (unsigned int s) {
sleep_time = s;
current_service->sleep_time = sleep_time;
sleep_time = s;
current_service->sleep_time = sleep_time;
}
/**
@ -163,123 +163,123 @@ Windows_Service::setSleepTime (unsigned int s) {
*/
void
Windows_Service::install (LPCTSTR application_binary_path) {
SC_HANDLE sc_manager;
SERVICE_DESCRIPTION sd_buf;
cout << " [SERVICE] Attempting to install the service.\n";
cout << " [SERVICE] The full path to the binary is: " << application_binary_path << endl;
/* Add program to the Services database */
sc_manager = OpenSCManager (NULL, /* local machine */
NULL, /* defaults to SERVICES_ACTIVE_DATABASE */
SC_MANAGER_ALL_ACCESS /* full access rights */);
if (sc_manager == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to open a connection to the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch (err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << ". Access is denied. " << msg << endl;
break;
SC_HANDLE sc_manager;
SERVICE_DESCRIPTION sd_buf;
cout << " [SERVICE] Attempting to install the service.\n";
cout << " [SERVICE] The full path to the binary is: " << application_binary_path << endl;
/* Add program to the Services database */
sc_manager = OpenSCManager (NULL, /* local machine */
NULL, /* defaults to SERVICES_ACTIVE_DATABASE */
SC_MANAGER_ALL_ACCESS /* full access rights */);
if (sc_manager == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to open a connection to the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch (err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << ". Access is denied. " << msg << endl;
break;
case ERROR_DATABASE_DOES_NOT_EXIST:
cout << " [SERVICE] " << basic_message << " Services database does not exist. " << msg << endl;
break;
case ERROR_INVALID_PARAMETER:
cout << " [SERVICE] Invalid parameter. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << ") " << msg << endl;
break;
}
}
/* Crerate the service */
sc_service = CreateService (sc_manager, /* SCManager database */
service_name, /* name of service */
service_display_name, /* service name to display */
SERVICE_ALL_ACCESS, /* desired access */
SERVICE_WIN32_OWN_PROCESS, /* service type */
SERVICE_AUTO_START, /* start type */
SERVICE_ERROR_NORMAL, /* error control type */
application_binary_path, /* service's binary */
NULL, /* no load ordering group */
NULL, /* no tag identifier */
NULL, /* no dependencies */
NULL, /* LocalSystem account */
NULL /* no password */ );
if (sc_service == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Error while adding the service to the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch (err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_CIRCULAR_DEPENDENCY:
cout << " [SERVICE] " << basic_message << " Circular dependency. " << msg << endl;
break;
case ERROR_DUP_NAME:
cout << " [SERVICE] " << basic_message << " The display name (\"" << service_display_name;
cout << "\") is already in use. " << msg << endl;
break;
case ERROR_INVALID_HANDLE:
cout << " [SERVICE] " << basic_message << " Invalid handle. " << msg << endl;
break;
case ERROR_INVALID_NAME:
cout << " [SERVICE] " << basic_message << " Invalid service name. " << msg << endl;
break;
case ERROR_INVALID_PARAMETER:
cout << " [SERVICE] " << basic_message << " Invalid parameter. " << msg << endl;
break;
case ERROR_INVALID_SERVICE_ACCOUNT:
cout << " [SERVICE] " << basic_message << " Invalid service account. " << msg << endl;
break;
case ERROR_SERVICE_EXISTS:
cout << " [SERVICE] " << basic_message << " Service already exists. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << ") " << msg << endl;
break;
}
}
/* Apparently, the call to ChangeServiceConfig2() only works on Windows >= 2000 */
sd_buf.lpDescription = service_description;
if (!ChangeServiceConfig2 (sc_service, /* handle to service */
SERVICE_CONFIG_DESCRIPTION, /* change: description */
&sd_buf)) /* value: new description */ {
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
cout << " [SERVICE] Unable to add a description to the service. " << msg << endl;
}
cout << " [SERVICE] Successfully added the service to the Services database." << endl;
CloseServiceHandle (sc_service);
CloseServiceHandle (sc_manager);
case ERROR_DATABASE_DOES_NOT_EXIST:
cout << " [SERVICE] " << basic_message << " Services database does not exist. " << msg << endl;
break;
case ERROR_INVALID_PARAMETER:
cout << " [SERVICE] Invalid parameter. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << ") " << msg << endl;
break;
}
}
/* Crerate the service */
sc_service = CreateService (sc_manager, /* SCManager database */
service_name, /* name of service */
service_display_name, /* service name to display */
SERVICE_ALL_ACCESS, /* desired access */
SERVICE_WIN32_OWN_PROCESS, /* service type */
SERVICE_AUTO_START, /* start type */
SERVICE_ERROR_NORMAL, /* error control type */
application_binary_path, /* service's binary */
NULL, /* no load ordering group */
NULL, /* no tag identifier */
NULL, /* no dependencies */
NULL, /* LocalSystem account */
NULL /* no password */ );
if (sc_service == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Error while adding the service to the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch (err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_CIRCULAR_DEPENDENCY:
cout << " [SERVICE] " << basic_message << " Circular dependency. " << msg << endl;
break;
case ERROR_DUP_NAME:
cout << " [SERVICE] " << basic_message << " The display name (\"" << service_display_name;
cout << "\") is already in use. " << msg << endl;
break;
case ERROR_INVALID_HANDLE:
cout << " [SERVICE] " << basic_message << " Invalid handle. " << msg << endl;
break;
case ERROR_INVALID_NAME:
cout << " [SERVICE] " << basic_message << " Invalid service name. " << msg << endl;
break;
case ERROR_INVALID_PARAMETER:
cout << " [SERVICE] " << basic_message << " Invalid parameter. " << msg << endl;
break;
case ERROR_INVALID_SERVICE_ACCOUNT:
cout << " [SERVICE] " << basic_message << " Invalid service account. " << msg << endl;
break;
case ERROR_SERVICE_EXISTS:
cout << " [SERVICE] " << basic_message << " Service already exists. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << ") " << msg << endl;
break;
}
}
/* Apparently, the call to ChangeServiceConfig2() only works on Windows >= 2000 */
sd_buf.lpDescription = service_description;
if (!ChangeServiceConfig2 (sc_service, /* handle to service */
SERVICE_CONFIG_DESCRIPTION, /* change: description */
&sd_buf)) /* value: new description */ {
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
cout << " [SERVICE] Unable to add a description to the service. " << msg << endl;
}
cout << " [SERVICE] Successfully added the service to the Services database." << endl;
CloseServiceHandle (sc_service);
CloseServiceHandle (sc_manager);
}
/**
@ -287,112 +287,112 @@ Windows_Service::install (LPCTSTR application_binary_path) {
*/
void
Windows_Service::uninstall () {
SC_HANDLE sc_manager, sc_service;
cout << " [SERVICE] Attempting to uninstall the service." << endl;
/* Remove from the Services database */
sc_manager = OpenSCManager (NULL, /* local machine */
NULL, /* ServicesActive database */
SC_MANAGER_ALL_ACCESS); /* full access rights */
if (sc_manager == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to open a connection to the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch(err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_DATABASE_DOES_NOT_EXIST:
cout << " [SERVICE] " << basic_message << " Services database does not exist. " << msg << endl;
break;
case ERROR_INVALID_PARAMETER:
cout << " [SERVICE] " << basic_message << " Invalid parameter. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << "). " << msg << endl;
break;
}
}
/* Open the service with DELETE access */
sc_service = OpenService (sc_manager, /* SCManager database */
service_name, /* name of service */
DELETE); /* only need DELETE access */
if (sc_service == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to locate in the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch (err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_INVALID_HANDLE:
cout << " [SERVICE] " << basic_message << " Invalid handle. " << msg << endl;
break;
case ERROR_INVALID_NAME:
cout << " [SERVICE] " << basic_message << " Invalid name. " << msg << endl;
break;
case ERROR_SERVICE_DOES_NOT_EXIST:
cout << " [SERVICE] " << basic_message << " Service does not exist. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << "Unrecognized error (" << err << "). " << msg << endl;
break;
}
CloseServiceHandle (sc_manager);
return;
}
if (!DeleteService (sc_service)) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to remove from the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch(err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_INVALID_HANDLE:
cout << " [SERVICE] " << basic_message << " Invalid handle. " << msg << endl;
break;
case ERROR_SERVICE_MARKED_FOR_DELETE:
cout << " [SERVICE] " << basic_message << " Service already marked for delete. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << "). " << msg << endl;
break;
}
}
cout << " [SERVICE] Successfully removed the service from the Services database.";
CloseServiceHandle (sc_service);
CloseServiceHandle (sc_manager);
SC_HANDLE sc_manager, sc_service;
cout << " [SERVICE] Attempting to uninstall the service." << endl;
/* Remove from the Services database */
sc_manager = OpenSCManager (NULL, /* local machine */
NULL, /* ServicesActive database */
SC_MANAGER_ALL_ACCESS); /* full access rights */
if (sc_manager == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to open a connection to the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch(err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_DATABASE_DOES_NOT_EXIST:
cout << " [SERVICE] " << basic_message << " Services database does not exist. " << msg << endl;
break;
case ERROR_INVALID_PARAMETER:
cout << " [SERVICE] " << basic_message << " Invalid parameter. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << "). " << msg << endl;
break;
}
}
/* Open the service with DELETE access */
sc_service = OpenService (sc_manager, /* SCManager database */
service_name, /* name of service */
DELETE); /* only need DELETE access */
if (sc_service == NULL) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to locate in the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch (err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_INVALID_HANDLE:
cout << " [SERVICE] " << basic_message << " Invalid handle. " << msg << endl;
break;
case ERROR_INVALID_NAME:
cout << " [SERVICE] " << basic_message << " Invalid name. " << msg << endl;
break;
case ERROR_SERVICE_DOES_NOT_EXIST:
cout << " [SERVICE] " << basic_message << " Service does not exist. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << "Unrecognized error (" << err << "). " << msg << endl;
break;
}
CloseServiceHandle (sc_manager);
return;
}
if (!DeleteService (sc_service)) {
DWORD err = GetLastError();
LPCTSTR basic_message = "Unable to remove from the Services database.";
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
switch(err) {
case ERROR_ACCESS_DENIED:
cout << " [SERVICE] " << basic_message << " Access is denied. " << msg << endl;
break;
case ERROR_INVALID_HANDLE:
cout << " [SERVICE] " << basic_message << " Invalid handle. " << msg << endl;
break;
case ERROR_SERVICE_MARKED_FOR_DELETE:
cout << " [SERVICE] " << basic_message << " Service already marked for delete. " << msg << endl;
break;
default:
cout << " [SERVICE] " << basic_message;
cout << " Unrecognized error (" << err << "). " << msg << endl;
break;
}
}
cout << " [SERVICE] Successfully removed the service from the Services database.";
CloseServiceHandle (sc_service);
CloseServiceHandle (sc_manager);
}
@ -404,138 +404,138 @@ Windows_Service::uninstall () {
*/
void
Windows_Service::run () {
SERVICE_TABLE_ENTRY ste_dispatch_table[] =
{
{ service_name, windows_service_start },
{ NULL, NULL }
};
int err = StartServiceCtrlDispatcher (ste_dispatch_table);
/* Start up the Win32 Service */
if (!err) {
char msg[1024];
memset (msg, sizeof (msg), '\0');
svc_format_message (msg, sizeof (msg));
}
SERVICE_TABLE_ENTRY ste_dispatch_table[] =
{
{ service_name, windows_service_start },
{ NULL, NULL }
};
int err = StartServiceCtrlDispatcher (ste_dispatch_table);
/* Start up the Win32 Service */
if (!err) {
char msg[1024];
memset (msg, sizeof (msg), '\0');
svc_format_message (msg, sizeof (msg));
}
}
static void WINAPI
windows_service_start (DWORD argc, LPTSTR *argv) {
service_status_handle = RegisterServiceCtrlHandler (current_service->getServiceName (),
windows_service_ctrl_handler);
service_status_handle = RegisterServiceCtrlHandler (current_service->getServiceName (),
windows_service_ctrl_handler);
if (service_status_handle == (SERVICE_STATUS_HANDLE) 0) {
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
return;
}
/* Initialization code should go here. */
current_service->execInitFunction ();
if (service_status_handle == (SERVICE_STATUS_HANDLE) 0) {
TCHAR msg[1000];
svc_format_message (msg, sizeof (msg));
return;
}
/* Initialization code should go here. */
current_service->execInitFunction ();
/* Initialization complete - report running status. */
SetWindowsServiceStatus (SERVICE_RUNNING, 0, 0, 0);
/* This is where the service should do its work. */
current_service->execRunFunction ();
/* Initialization complete - report running status. */
SetWindowsServiceStatus (SERVICE_RUNNING, 0, 0, 0);
/* This is where the service should do its work. */
current_service->execRunFunction ();
return;
return;
}
static VOID WINAPI
windows_service_ctrl_handler (DWORD opcode) {
switch (opcode) {
case SERVICE_CONTROL_PAUSE:
SetWindowsServiceStatus (SERVICE_CONTROL_PAUSE, 0, 0, 0);
break;
case SERVICE_CONTROL_CONTINUE:
SetWindowsServiceStatus (SERVICE_CONTROL_CONTINUE, 0, 0, 0);
break;
case SERVICE_CONTROL_STOP:
Sleep (READ_TIMEOUT * 2); /* wait for 2x the timeout, just to ensure that things
* the service has processed any last packets
*/
SetWindowsServiceStatus (SERVICE_STOPPED, 0, 0, 0);
return;
case SERVICE_CONTROL_INTERROGATE:
/* Fall through to send current status. */
break;
default:
break;
}
return;
switch (opcode) {
case SERVICE_CONTROL_PAUSE:
SetWindowsServiceStatus (SERVICE_CONTROL_PAUSE, 0, 0, 0);
break;
case SERVICE_CONTROL_CONTINUE:
SetWindowsServiceStatus (SERVICE_CONTROL_CONTINUE, 0, 0, 0);
break;
case SERVICE_CONTROL_STOP:
Sleep (READ_TIMEOUT * 2); /* wait for 2x the timeout, just to ensure that things
* the service has processed any last packets
*/
SetWindowsServiceStatus (SERVICE_STOPPED, 0, 0, 0);
return;
case SERVICE_CONTROL_INTERROGATE:
/* Fall through to send current status. */
break;
default:
break;
}
return;
}
static void
SetWindowsServiceStatus (DWORD dwCurrentState, DWORD dwWin32ExitCode,
DWORD dwCheckPoint, DWORD dwWaitHint) {
SERVICE_STATUS ss; /* Current status of the service. */
/* Disable control requests until the service is started.*/
if (dwCurrentState == SERVICE_START_PENDING)
ss.dwControlsAccepted = 0;
else
ss.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
/* Initialize ss structure. */
ss.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
ss.dwServiceSpecificExitCode = 0;
ss.dwCurrentState = dwCurrentState;
ss.dwWin32ExitCode = dwWin32ExitCode;
ss.dwCheckPoint = dwCheckPoint;
ss.dwWaitHint = dwWaitHint;
/* Send status of the service to the Service Controller. */
if (!SetServiceStatus (service_status_handle, &ss))
ErrorStopService (TEXT ("SetServiceStatus"));
DWORD dwCheckPoint, DWORD dwWaitHint) {
SERVICE_STATUS ss; /* Current status of the service. */
/* Disable control requests until the service is started.*/
if (dwCurrentState == SERVICE_START_PENDING)
ss.dwControlsAccepted = 0;
else
ss.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
/* Initialize ss structure. */
ss.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
ss.dwServiceSpecificExitCode = 0;
ss.dwCurrentState = dwCurrentState;
ss.dwWin32ExitCode = dwWin32ExitCode;
ss.dwCheckPoint = dwCheckPoint;
ss.dwWaitHint = dwWaitHint;
/* Send status of the service to the Service Controller. */
if (!SetServiceStatus (service_status_handle, &ss))
ErrorStopService (TEXT ("SetServiceStatus"));
}
static void
ErrorStopService (LPTSTR lpszAPI)
{
TCHAR buffer[256] = TEXT("");
TCHAR error[1024] = TEXT("");
LPVOID lpvMessageBuffer;
wsprintf (buffer, TEXT("API = %s, "), lpszAPI);
lstrcat (error, buffer);
ZeroMemory(buffer, sizeof(buffer));
wsprintf(buffer,TEXT("error code = %d, "), GetLastError());
lstrcat(error, buffer);
// Obtain the error string.
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM,
NULL, GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR)&lpvMessageBuffer, 0, NULL);
ZeroMemory((LPVOID)buffer, (DWORD)sizeof(buffer));
wsprintf(buffer,TEXT("message = %s"), (TCHAR *)lpvMessageBuffer);
lstrcat(error, buffer);
// Free the buffer allocated by the system.
LocalFree (lpvMessageBuffer);
// Write the error string to the debugger.
// If you have threads running, tell them to stop. Something went
// wrong, and you need to stop them so you can inform the SCM.
TCHAR buffer[256] = TEXT("");
TCHAR error[1024] = TEXT("");
LPVOID lpvMessageBuffer;
wsprintf (buffer, TEXT("API = %s, "), lpszAPI);
lstrcat (error, buffer);
ZeroMemory(buffer, sizeof(buffer));
wsprintf(buffer,TEXT("error code = %d, "), GetLastError());
lstrcat(error, buffer);
// Obtain the error string.
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_SYSTEM,
NULL, GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR)&lpvMessageBuffer, 0, NULL);
ZeroMemory((LPVOID)buffer, (DWORD)sizeof(buffer));
wsprintf(buffer,TEXT("message = %s"), (TCHAR *)lpvMessageBuffer);
lstrcat(error, buffer);
// Free the buffer allocated by the system.
LocalFree (lpvMessageBuffer);
// Write the error string to the debugger.
// If you have threads running, tell them to stop. Something went
// wrong, and you need to stop them so you can inform the SCM.
// SetEvent (g_stop_event);
// Stop the service.
SetWindowsServiceStatus (SERVICE_STOPPED, GetLastError(), 0, 0);
// Stop the service.
SetWindowsServiceStatus (SERVICE_STOPPED, GetLastError(), 0, 0);
}
static VOID
@ -544,15 +544,15 @@ svc_format_message (LPSTR msg, int count)
LPVOID msg_buf;
if (msg != NULL && count > 0) {
memset (msg, 0, count);
FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL, GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), /* Default language */
(LPTSTR) &msg_buf, 0, NULL);
memset (msg, 0, count);
FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
NULL, GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), /* Default language */
(LPTSTR) &msg_buf, 0, NULL);
strncpy (msg, (LPCTSTR) msg_buf, count);
/* Free the buffer. */
LocalFree (msg_buf);
msg_buf = NULL;
strncpy (msg, (LPCTSTR) msg_buf, count);
/* Free the buffer. */
LocalFree (msg_buf);
msg_buf = NULL;
}
}

50
pandora_agents/win32/windows_service.h
View File

@ -36,33 +36,33 @@
*/
class Windows_Service {
private:
char *service_name;
char *service_display_name;
char *service_description;
HANDLE stop_event;
int sleep_time;
SC_HANDLE sc_service;
void (Windows_Service::*run_function) ();
void (Windows_Service::*init_function) ();
char *service_name;
char *service_display_name;
char *service_description;
HANDLE stop_event;
int sleep_time;
SC_HANDLE sc_service;
void (Windows_Service::*run_function) ();
void (Windows_Service::*init_function) ();
public:
Windows_Service ();
Windows_Service (const char * svc_name,
const char * svc_display_name,
const char * svc_description);
~Windows_Service ();
void install (LPCTSTR application_binary_path);
void uninstall ();
void run ();
void setRunFunction (void (Windows_Service::*f) ());
void setInitFunction (void (Windows_Service::*f) ());
LPSTR getServiceName ();
void setSleepTime (unsigned int s);
Windows_Service ();
Windows_Service (const char * svc_name,
const char * svc_display_name,
const char * svc_description);
~Windows_Service ();
void install (LPCTSTR application_binary_path);
void uninstall ();
void run ();
void setRunFunction (void (Windows_Service::*f) ());
void setInitFunction (void (Windows_Service::*f) ());
LPSTR getServiceName ();
void setSleepTime (unsigned int s);
void execRunFunction ();
void execInitFunction ();
void execInitFunction ();
};
#endif /* __WINDOWS_SERVICE_H__ */