tyler.durden/pandorafms
1
0
Fork 0
mirror of https://github.com/pandorafms/pandorafms.git synced 2025-07-28 16:24:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

Improved to be able to define agent name using external command.

Browse Source 
(cherry picked from commit 22d72ed80c9006f5fee76b361d4195f1b20eab07)
This commit is contained in:
Junichi Satoh 2014-11-16 12:04:00 +09:00
parent 41eabe19d3
commit d7ab377d29
5 changed files with 93 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pandora_agents/win32/pandora.cc
View File

@ -75,6 +75,19 @@ Key_Value::parseLine (string str) {
} }
} }
/**
* Parses a string by position and initialize the key and the value.
*/
void
Key_Value::parseLineByPosition (string str, int pos) {
string trimmedstr;
trimmedstr = trim (str);
key = trim (trimmedstr.substr (0, pos));
value = trim (trimmedstr.substr (pos));
}
/** /**
* Get the key of the object. * Get the key of the object.
* *

1
pandora_agents/win32/pandora.h
View File

@ -60,6 +60,7 @@ namespace Pandora {
string value; string value;
public: public:
void parseLine (string str); void parseLine (string str);
void parseLineByPosition (string str, int pos);
string getKey (); string getKey ();
string getValue (); string getValue ();
}; };

9
pandora_agents/win32/pandora_agent_conf.cc
View File

@ -220,6 +220,15 @@ Pandora::Pandora_Agent_Conf::setFile (string *all_conf){
} }
} }
/*Check if is a agent_name_cmd"*/
pos = buffer.find("agent_name_cmd");
if (pos != string::npos){
Key_Value kv;
kv.parseLineByPosition(buffer, 14);
key_values->push_back (kv);
continue;
}
/*Check if is a collection*/ /*Check if is a collection*/
pos = buffer.find("file_collection"); pos = buffer.find("file_collection");
if(pos != string::npos) { if(pos != string::npos) {

79
pandora_agents/win32/pandora_windows_service.cc
View File

@ -370,7 +370,7 @@ Pandora_Windows_Service::launchTentacleProxy() {
string string
Pandora_Windows_Service::getXmlHeader () { Pandora_Windows_Service::getXmlHeader () {
char timestamp[20]; char timestamp[20];
string agent_name, os_name, os_version, encoding, value, xml, address, parent_agent_name; string agent_name, os_name, os_version, encoding, value, xml, address, parent_agent_name, agent_name_cmd, temp_agent_name;
string custom_id, url_address, latitude, longitude, altitude, position_description, gis_exec, gis_result; string custom_id, url_address, latitude, longitude, altitude, position_description, gis_exec, gis_result;
time_t ctime; time_t ctime;
struct tm *ctime_tm = NULL; struct tm *ctime_tm = NULL;
@ -382,6 +382,27 @@ Pandora_Windows_Service::getXmlHeader () {
agent_name = Pandora_Windows_Info::getSystemName (); agent_name = Pandora_Windows_Info::getSystemName ();
} }
agent_name_cmd = conf->getValue ("agent_name_cmd");
if (agent_name_cmd != "") {
agent_name_cmd = "cmd.exe /c \"" + agent_name_cmd + "\"";
temp_agent_name = getCoordinatesFromCmdExec(agent_name_cmd);
// Delete carriage return if is provided
pos = temp_agent_name.find("\n");
if(pos != string::npos) {
temp_agent_name.erase(pos, temp_agent_name.size () - pos);
}
pos = temp_agent_name.find("\r");
if(pos != string::npos) {
temp_agent_name.erase(pos, temp_agent_name.size () - pos);
}
// Remove white spaces of the first and last.
temp_agent_name = trim(temp_agent_name);
if (temp_agent_name != "") {
agent_name = temp_agent_name;
}
}
// Get parent agent name // Get parent agent name
parent_agent_name = conf->getValue ("parent_agent_name"); parent_agent_name = conf->getValue ("parent_agent_name");
@ -448,7 +469,7 @@ Pandora_Windows_Service::getXmlHeader () {
gis_exec = conf->getValue ("gis_exec"); gis_exec = conf->getValue ("gis_exec");
if(gis_exec != "") { if(gis_exec != "") {
gis_result = getCoordinatesFromGisExec(gis_exec); gis_result = getCoordinatesFromCmdExec(gis_exec);
if(gis_result != "") { if(gis_result != "") {
// Delete carriage return if is provided // Delete carriage return if is provided
pos = gis_result.find("\n"); pos = gis_result.find("\n");
@ -461,7 +482,7 @@ Pandora_Windows_Service::getXmlHeader () {
} }
// Process the result as "latitude,longitude,altitude" // Process the result as "latitude,longitude,altitude"
pandoraDebug ("getCoordinatesFromGisExec: Parsing coordinates %s", gis_result.c_str ()); pandoraDebug ("getCoordinatesFromCmdExec: Parsing coordinates %s", gis_result.c_str ());
pos = gis_result.find(","); pos = gis_result.find(",");
if (pos != string::npos && pos != 0) { if (pos != string::npos && pos != 0) {
latitude = gis_result; latitude = gis_result;
@ -515,7 +536,7 @@ Pandora_Windows_Service::getXmlHeader () {
} }
string string
Pandora_Windows_Service::getCoordinatesFromGisExec (string gis_exec) Pandora_Windows_Service::getCoordinatesFromCmdExec (string cmd_exec)
{ {
PROCESS_INFORMATION pi; PROCESS_INFORMATION pi;
STARTUPINFO si; STARTUPINFO si;
@ -541,7 +562,7 @@ Pandora_Windows_Service::getCoordinatesFromGisExec (string gis_exec)
job = CreateJobObject (&attributes, NULL); job = CreateJobObject (&attributes, NULL);
if (job == NULL) { if (job == NULL) {
pandoraLog ("getCoordinatesFromGisExec: CreateJobObject failed. Err: %d", GetLastError ()); pandoraLog ("getCoordinatesFromCmdExec: CreateJobObject failed. Err: %d", GetLastError ());
return ""; return "";
} }
@ -549,7 +570,7 @@ Pandora_Windows_Service::getCoordinatesFromGisExec (string gis_exec)
out = GetStdHandle (STD_OUTPUT_HANDLE); out = GetStdHandle (STD_OUTPUT_HANDLE);
if (! CreatePipe (&out_read, &new_stdout, &attributes, 0)) { if (! CreatePipe (&out_read, &new_stdout, &attributes, 0)) {
pandoraLog ("getCoordinatesFromGisExec: CreatePipe failed. Err: %d", GetLastError ()); pandoraLog ("getCoordinatesFromCmdExec: CreatePipe failed. Err: %d", GetLastError ());
return ""; return "";
} }
@ -569,20 +590,20 @@ Pandora_Windows_Service::getCoordinatesFromGisExec (string gis_exec)
/* Set up members of the PROCESS_INFORMATION structure. */ /* Set up members of the PROCESS_INFORMATION structure. */
ZeroMemory (&pi, sizeof (pi)); ZeroMemory (&pi, sizeof (pi));
pandoraDebug ("Executing gis_exec: %s", gis_exec.c_str ()); pandoraDebug ("Executing cmd_exec: %s", cmd_exec.c_str ());
/* Create the child process. */ /* Create the child process. */
if (CreateProcess (NULL , (CHAR *)gis_exec.c_str (), NULL, NULL, TRUE, if (CreateProcess (NULL , (CHAR *)cmd_exec.c_str (), NULL, NULL, TRUE,
CREATE_SUSPENDED | CREATE_NO_WINDOW, NULL, NULL, &si, &pi) == 0) { CREATE_SUSPENDED | CREATE_NO_WINDOW, NULL, NULL, &si, &pi) == 0) {
pandoraLog ("getCoordinatesFromGisExec: %s CreateProcess failed. Err: %d", pandoraLog ("getCoordinatesFromCmdExec: %s CreateProcess failed. Err: %d",
gis_exec.c_str (), GetLastError ()); cmd_exec.c_str (), GetLastError ());
return ""; return "";
} else { } else {
char buffer[BUFSIZE + 1]; char buffer[BUFSIZE + 1];
unsigned long read, avail; unsigned long read, avail;
if (! AssignProcessToJobObject (job, pi.hProcess)) { if (! AssignProcessToJobObject (job, pi.hProcess)) {
pandoraLog ("getCoordinatesFromGisExec: could not assign proccess to job (error %d)", pandoraLog ("getCoordinatesFromCmdExec: could not assign proccess to job (error %d)",
GetLastError ()); GetLastError ());
} }
ResumeThread (pi.hThread); ResumeThread (pi.hThread);
@ -602,7 +623,7 @@ Pandora_Windows_Service::getCoordinatesFromGisExec (string gis_exec)
} else if(timeout < GetTickCount() - tickbase) { } else if(timeout < GetTickCount() - tickbase) {
/* STILL_ACTIVE */ /* STILL_ACTIVE */
TerminateProcess(pi.hThread, STILL_ACTIVE); TerminateProcess(pi.hThread, STILL_ACTIVE);
pandoraLog ("getCoordinatesFromGisExec: %s timed out (retcode: %d)", gis_exec.c_str (), STILL_ACTIVE); pandoraLog ("getCoordinatesFromCmdExec: %s timed out (retcode: %d)", cmd_exec.c_str (), STILL_ACTIVE);
break; break;
} }
} }
@ -611,12 +632,12 @@ Pandora_Windows_Service::getCoordinatesFromGisExec (string gis_exec)
if (retval != 0) { if (retval != 0) {
if (! TerminateJobObject (job, 0)) { if (! TerminateJobObject (job, 0)) {
pandoraLog ("getCoordinatesFromGisExec: TerminateJobObject failed. (error %d)", pandoraLog ("getCoordinatesFromCmdExec: TerminateJobObject failed. (error %d)",
GetLastError ()); GetLastError ());
} }
if (retval != STILL_ACTIVE) { if (retval != STILL_ACTIVE) {
pandoraLog ("getCoordinatesFromGisExec: %s did not executed well (retcode: %d)", pandoraLog ("getCoordinatesFromCmdExec: %s did not executed well (retcode: %d)",
gis_exec.c_str (), retval); cmd_exec.c_str (), retval);
} }
/* Close job, process and thread handles. */ /* Close job, process and thread handles. */
CloseHandle (job); CloseHandle (job);
@ -1366,7 +1387,7 @@ Pandora_Windows_Service::checkAgentName(string filename){
} }
int int
Pandora_Windows_Service::checkConfig (string file) { Pandora_Windows_Service::checkConfig (string file) {
int i, conf_size; int i, conf_size, pos;
char *conf_str = NULL, *remote_conf_str = NULL, *remote_conf_md5 = NULL; char *conf_str = NULL, *remote_conf_str = NULL, *remote_conf_md5 = NULL;
char agent_md5[33], conf_md5[33], flag; char agent_md5[33], conf_md5[33], flag;
string agent_name, conf_tmp_file, md5_tmp_file, temp_dir, tmp; string agent_name, conf_tmp_file, md5_tmp_file, temp_dir, tmp;
@ -1390,13 +1411,37 @@ Pandora_Windows_Service::checkConfig (string file) {
} }
agent_name = tmp; agent_name = tmp;
/* Get agent name cmd */
tmp = conf->getValue ("agent_name_cmd");
if (!tmp.empty ()) {
tmp = "cmd.exe /c \"" + tmp + "\"";
tmp = getCoordinatesFromCmdExec(tmp);
// Delete carriage return if is provided
pos = tmp.find("\n");
if(pos != string::npos) {
tmp.erase(pos, tmp.size () - pos);
}
pos = tmp.find("\r");
if(pos != string::npos) {
tmp.erase(pos, tmp.size () - pos);
}
// Remove white spaces of the first and last.
tmp = trim (tmp);
if (tmp != "") {
agent_name = tmp;
}
}
/* Error getting agent name */ /* Error getting agent name */
if (tmp.empty ()) { if (tmp.empty ()) {
pandoraDebug ("Pandora_Windows_Service::checkConfig: Error getting agent name"); pandoraDebug ("Pandora_Windows_Service::checkConfig: Error getting agent name");
return 0; return 0;
} }
Pandora_File::md5 (tmp.c_str(), tmp.size(), agent_md5); Pandora_File::md5 (agent_name.c_str(), agent_name.size(), agent_md5);
/* Calculate md5 hashes */ /* Calculate md5 hashes */
try { try {

2
pandora_agents/win32/pandora_windows_service.h
View File

@ -53,7 +53,7 @@ namespace Pandora {
string getXmlHeader (); string getXmlHeader ();
int copyDataFile (string filename); int copyDataFile (string filename);
string getCoordinatesFromGisExec (string gis_exec); string getCoordinatesFromCmdExec (string cmd_exec);
int copyTentacleDataFile (string host, int copyTentacleDataFile (string host,
string filename, string filename,
string port, string port,