-/* @(#) $Id: ./src/os_execd/win_execd.c, 2011/09/08 dcid Exp $
- */
-
/* Copyright (C) 2009 Trend Micro Inc.
* All right reserved.
*
* Foundation
*/
-
#ifdef WIN32
#include "shared.h"
#include "list_op.h"
#include "os_regex/os_regex.h"
#include "os_net/os_net.h"
-
#include "execd.h"
-
#ifdef ARGV0
- #undef ARGV0
+#undef ARGV0
#endif
#define ARGV0 "ossec-execd"
-
-
-
-/* Timeout data structure */
-typedef struct _timeout_data
-{
- time_t time_of_addition;
- int time_to_block;
- char **command;
-}timeout_data;
-
-
/* Timeout list */
OSList *timeout_list;
OSListNode *timeout_node;
-
-
-/** int main(int argc, char **argv) v0.1
- */
int WinExecd_Start()
{
int c;
int test_config = 0;
+ char *cfg = DEFAULTCPATH;
- char *xmlcfg = DEFAULTCPATH;
-
-
-
- /* Reading config */
- if((c = ExecdConfig(xmlcfg)) < 0)
- {
- ErrorExit(CONFIG_ERROR, ARGV0, xmlcfg);
+ /* Read config */
+ if ((c = ExecdConfig(cfg)) < 0) {
+ ErrorExit(CONFIG_ERROR, ARGV0, cfg);
}
-
/* Exit if test_config */
- if(test_config)
- return(0);
-
+ if (test_config) {
+ return (0);
+ }
/* Active response disabled */
- if(c == 1)
- {
+ if (c == 1) {
verbose(EXEC_DISABLED, ARGV0);
- return(0);
+ return (0);
}
-
- /* Creating list for timeout */
+ /* Create list for timeout */
timeout_list = OSList_Create();
- if(!timeout_list)
- {
+ if (!timeout_list) {
ErrorExit(LIST_ERROR, ARGV0);
}
-
-
/* Start up message */
verbose(STARTUP_MSG, ARGV0, getpid());
-
- return(1);
+ return (1);
}
-
-
void WinTimeoutRun(int curr_time)
{
- /* Checking if there is any timeouted command to execute. */
+ /* Check if there is any timed out command to execute */
timeout_node = OSList_GetFirstNode(timeout_list);
- while(timeout_node)
- {
+ while (timeout_node) {
timeout_data *list_entry;
list_entry = (timeout_data *)timeout_node->data;
- /* Timeouted */
- if((curr_time - list_entry->time_of_addition) >
- list_entry->time_to_block)
- {
+ /* Timed out */
+ if ((curr_time - list_entry->time_of_addition) >
+ list_entry->time_to_block) {
ExecCmd_Win32(list_entry->command[0]);
- /* Deletecurrently node already sets the pointer to next */
+ /* Delete currently node - already sets the pointer to next */
OSList_DeleteCurrentlyNode(timeout_list);
timeout_node = OSList_GetCurrentlyNode(timeout_list);
- /* Clearing the memory */
+ /* Clear the memory */
FreeTimeoutEntry(list_entry);
}
- else
- {
+ else {
timeout_node = OSList_GetNextNode(timeout_list);
}
}
}
-
-
-/** void WinExecdRun(char *exec_msg)
- */
void WinExecdRun(char *exec_msg)
{
time_t curr_time;
- int i,j;
+ int i, j;
int timeout_value;
int added_before = 0;
char **timeout_args;
-
char *tmp_msg = NULL;
char *name;
char *command;
char *cmd_ip;
char buffer[OS_MAXSTR + 1];
-
timeout_data *timeout_entry;
-
-
-
- /* Currently time */
+ /* Current time */
curr_time = time(0);
-
- /* Getting application name */
+ /* Get application name */
name = exec_msg;
-
- /* Zeroing the name */
+ /* Zero the name */
tmp_msg = strchr(exec_msg, ' ');
- if(!tmp_msg)
- {
+ if (!tmp_msg) {
merror(EXECD_INV_MSG, ARGV0, exec_msg);
return;
}
*tmp_msg = '\0';
tmp_msg++;
-
- /* Getting user. */
+ /* Get user */
cmd_user = tmp_msg;
tmp_msg = strchr(tmp_msg, ' ');
- if(!tmp_msg)
- {
+ if (!tmp_msg) {
merror(EXECD_INV_MSG, ARGV0, cmd_user);
return;
}
*tmp_msg = '\0';
tmp_msg++;
-
- /* Getting ip. */
+ /* Get IP */
cmd_ip = tmp_msg;
tmp_msg = strchr(tmp_msg, ' ');
- if(!tmp_msg)
- {
+ if (!tmp_msg) {
merror(EXECD_INV_MSG, ARGV0, cmd_ip);
return;
}
*tmp_msg = '\0';
tmp_msg++;
-
- /* Getting the command to execute (valid name) */
+ /* Get the command to execute (valid name) */
command = GetCommandbyName(name, &timeout_value);
- if(!command)
- {
+ if (!command) {
ReadExecConfig();
command = GetCommandbyName(name, &timeout_value);
- if(!command)
- {
+ if (!command) {
merror(EXEC_INV_NAME, ARGV0, name);
return;
}
}
-
- /* Command not present. */
- if(command[0] == '\0')
+ /* Command not present */
+ if (command[0] == '\0') {
return;
+ }
+ /* Allocate memory for the timeout argument */
+ os_calloc(MAX_ARGS + 2, sizeof(char *), timeout_args);
- /* Allocating memory for the timeout argument */
- os_calloc(MAX_ARGS+2, sizeof(char *), timeout_args);
-
-
- /* Adding initial variables to the timeout cmd */
+ /* Add initial variables to the timeout cmd */
snprintf(buffer, OS_MAXSTR, "\"%s\" %s \"%s\" \"%s\" \"%s\"",
command, DELETE_ENTRY, cmd_user, cmd_ip, tmp_msg);
os_strdup(buffer, timeout_args[0]);
timeout_args[1] = NULL;
-
-
- /* Getting size for the strncmp */
+ /* Get size for the strncmp */
i = 0, j = 0;
- while(buffer[i] != '\0')
- {
- if(buffer[i] == ' ')
+ while (buffer[i] != '\0') {
+ if (buffer[i] == ' ') {
j++;
+ }
i++;
- if(j == 4)
+ if (j == 4) {
break;
+ }
}
-
- /* Check this command was already executed. */
+ /* Check if this command was already executed */
timeout_node = OSList_GetFirstNode(timeout_list);
added_before = 0;
-
- while(timeout_node)
- {
+ while (timeout_node) {
timeout_data *list_entry;
list_entry = (timeout_data *)timeout_node->data;
- if(strncmp(list_entry->command[0], timeout_args[0], i) == 0)
- {
+ if (strncmp(list_entry->command[0], timeout_args[0], i) == 0) {
/* Means we executed this command before
- * and we don't need to add it again.
+ * and we don't need to add it again
*/
added_before = 1;
-
- /* updating the timeout */
+ /* Update the timeout */
list_entry->time_of_addition = curr_time;
break;
}
- /* Continue with the next entry in timeout list*/
+ /* Continue with the next entry in timeout list */
timeout_node = OSList_GetNextNode(timeout_list);
}
-
/* If it wasn't added before, do it now */
- if(!added_before)
- {
+ if (!added_before) {
snprintf(buffer, OS_MAXSTR, "\"%s\" %s \"%s\" \"%s\" \"%s\"", command,
- ADD_ENTRY, cmd_user, cmd_ip, tmp_msg);
- /* executing command */
-
+ ADD_ENTRY, cmd_user, cmd_ip, tmp_msg);
+ /* Execute command */
ExecCmd_Win32(buffer);
/* We don't need to add to the list if the timeout_value == 0 */
- if(timeout_value)
- {
- /* Creating the timeout entry */
+ if (timeout_value) {
+ /* Create the timeout entry */
os_calloc(1, sizeof(timeout_data), timeout_entry);
timeout_entry->command = timeout_args;
timeout_entry->time_of_addition = curr_time;
timeout_entry->time_to_block = timeout_value;
-
- /* Adding command to the timeout list */
- if(!OSList_AddData(timeout_list, timeout_entry))
- {
+ /* Add command to the timeout list */
+ if (!OSList_AddData(timeout_list, timeout_entry)) {
merror(LIST_ADD_ERROR, ARGV0);
FreeTimeoutEntry(timeout_entry);
}
}
/* If no timeout, we still need to free it in here */
- else
- {
+ else {
char **ss_ta = timeout_args;
- while(*timeout_args)
- {
+ while (*timeout_args) {
os_free(*timeout_args);
*timeout_args = NULL;
timeout_args++;
}
/* We didn't add it to the timeout list */
- else
- {
+ else {
char **ss_ta = timeout_args;
/* Clear the timeout arguments */
- while(*timeout_args)
- {
+ while (*timeout_args) {
os_free(*timeout_args);
*timeout_args = NULL;
timeout_args++;
}
}
-#endif
+#endif /* WIN32 */
-/* EOF */
projects / ossec-hids.git / blobdiff