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

Make sure all results from WMI queries are consumed. 

If there are unconsumed results left from a WMI query an error is
generated in the event log.
This commit is contained in:
Ramon Novoa 2023-04-25 19:03:41 +02:00
parent ca7e1777e5
commit 7e829dd03c
1 changed files with 21 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -99,7 +99,7 @@ Pandora_Wmi::isServiceRunning (string service_name) {
string query;
char *state;
string str_state;
int retval;
int retval = -1;
query = "SELECT * FROM Win32_Service WHERE Name = \"" + service_name + "\"";
@ -112,23 +112,27 @@ Pandora_Wmi::isServiceRunning (string service_name) {
FOR_EACH (quickfix, quickfixes, NULL) {
dhGetValue (L"%s", &state, quickfix,
L".State");
str_state = state;
if (str_state == "Running") {
retval = 1;
}
else {
retval = 0;
if (retval == -1) {
str_state = state;
if (str_state == "Running") {
retval = 1;
}
else {
retval = 0;
}
}
dhFreeString (state);
return retval;
} NEXT_THROW (quickfix);
} catch (string errstr) {
pandoraLog ("isServiceRunning error. %s", errstr.c_str ());
}
pandoraDebug ("Service %s not found.", service_name.c_str ());
return 0;
if (retval == -1) {
pandoraDebug ("Service %s not found.", service_name.c_str ());
return 0;
}
return retval;
}
/**
@ -1261,7 +1265,7 @@ Pandora_Wmi::getSystemAddress () {
CDispPtr wmi_svc = NULL, nic_info = NULL;
VARIANT ip_addresses;
char *caption = NULL, *mac_address = NULL;
string ret = "";
string str_addr, ret = "";
try {
@ -1274,12 +1278,12 @@ Pandora_Wmi::getSystemAddress () {
dhGetValue (L"%v", &ip_addresses, nic_info_item,
L".IPAddress");
if (&ip_addresses != NULL)
if (&ip_addresses != NULL && ret == "")
{
ret = getIPs(&ip_addresses);
if(ret != "0.0.0.0") {
break;
}
str_addr = getIPs(&ip_addresses);
if (str_addr != "0.0.0.0") {
ret = str_addr;
}
}
} NEXT_THROW (nic_info_item);
} catch (string errstr) {