-/* @(#) $Id: ./src/logcollector/read_postgresql_log.c, 2011/09/08 dcid Exp $
- */
-
/* Copyright (C) 2009 Trend Micro Inc.
* All rights reserved.
*
* and/or modify it under the terms of the GNU General Public
* License (version 2) as published by the FSF - Free Software
* Foundation.
- *
- * License details at the LICENSE file included with OSSEC or
- * online at: http://www.ossec.net/en/licensing.html
*/
/* Read PostgreSQL logs */
-
#include "shared.h"
#include "logcollector.h"
-
-/* Send pgsql message and check the return code.
- */
-void __send_pgsql_msg(int pos, int drop_it, char *buffer)
+/* Send pgsql message and check the return code */
+static void __send_pgsql_msg(int pos, int drop_it, char *buffer)
{
debug2("%s: DEBUG: Reading PostgreSQL message: '%s'", ARGV0, buffer);
- if(drop_it == 0)
- {
- if(SendMSG(logr_queue, buffer, logff[pos].file, POSTGRESQL_MQ) < 0)
- {
+ if (drop_it == 0) {
+ if (SendMSG(logr_queue, buffer, logff[pos].file, POSTGRESQL_MQ) < 0) {
merror(QUEUE_SEND, ARGV0);
- if((logr_queue = StartMQ(DEFAULTQPATH,WRITE)) < 0)
- {
+ if ((logr_queue = StartMQ(DEFAULTQPATH, WRITE)) < 0) {
ErrorExit(QUEUE_FATAL, ARGV0, DEFAULTQPATH);
}
}
}
}
-
-
/* Read PostgreSQL log files */
void *read_postgresql_log(int pos, int *rc, int drop_it)
{
- int str_len = 0;
+ size_t str_len = 0;
int need_clear = 0;
char *p;
char str[OS_MAXSTR + 1];
char buffer[OS_MAXSTR + 1];
-
- /* Zeroing buffer and str */
+ /* Zero buffer and str */
buffer[0] = '\0';
buffer[OS_MAXSTR] = '\0';
- str[OS_MAXSTR]= '\0';
+ str[OS_MAXSTR] = '\0';
*rc = 0;
-
- /* Getting new entry */
- while(fgets(str, OS_MAXSTR - OS_LOG_HEADER, logff[pos].fp) != NULL)
- {
-
- /* Getting buffer size */
+ /* Get new entry */
+ while (fgets(str, OS_MAXSTR - OS_LOG_HEADER, logff[pos].fp) != NULL) {
+ /* Get buffer size */
str_len = strlen(str);
-
- /* Checking str_len size. Very useless, but just to make sure.. */
- if(str_len >= sizeof(buffer) -2)
- {
- str_len = sizeof(buffer) -10;
+ /* Check str_len size. Very useless, but just to make sure.. */
+ if (str_len >= sizeof(buffer) - 2) {
+ str_len = sizeof(buffer) - 10;
}
-
- /* Getting the last occurence of \n */
- if ((p = strrchr(str, '\n')) != NULL)
- {
+ /* Get the last occurrence of \n */
+ if ((p = strrchr(str, '\n')) != NULL) {
*p = '\0';
- /* If need clear is set, we just get the line and ignore it. */
- if(need_clear)
- {
+ /* If need_clear is set, we just get the line and ignore it. */
+ if (need_clear) {
need_clear = 0;
continue;
}
- }
- else
- {
+ } else {
need_clear = 1;
}
-
- #ifdef WIN32
- if ((p = strrchr(str, '\r')) != NULL)
- {
+#ifdef WIN32
+ if ((p = strrchr(str, '\r')) != NULL) {
*p = '\0';
}
-
- /* Looking for empty string (only on windows) */
- if(str_len <= 1)
- {
+ /* Look for empty string (only on Windows) */
+ if (str_len <= 1) {
continue;
}
-
/* Windows can have comment on their logs */
- if(str[0] == '#')
- {
+ if (str[0] == '#') {
continue;
}
- #endif
-
-
+#endif
/* PostgreSQL messages have the following format:
* [2007-08-31 19:17:32.186 ADT] 192.168.2.99:db_name
*/
- if((str_len > 32) &&
- (str[0] == '[') &&
- (str[5] == '-') &&
- (str[8] == '-') &&
- (str[11] == ' ') &&
- (str[14] == ':') &&
- (str[17] == ':') &&
- isdigit((int)str[1]) &&
- isdigit((int)str[12]))
- {
-
- /* If the saved message is empty, set it and continue. */
- if(buffer[0] == '\0')
- {
+ if ((str_len > 32) &&
+ (str[0] == '[') &&
+ (str[5] == '-') &&
+ (str[8] == '-') &&
+ (str[11] == ' ') &&
+ (str[14] == ':') &&
+ (str[17] == ':') &&
+ isdigit((int)str[1]) &&
+ isdigit((int)str[12])) {
+
+ /* If the saved message is empty, set it and continue */
+ if (buffer[0] == '\0') {
strncpy(buffer, str, str_len + 2);
continue;
}
/* If not, send the saved one and store the new one for later */
- else
- {
+ else {
__send_pgsql_msg(pos, drop_it, buffer);
-
-
- /* Storing current one at the buffer */
+ /* Store current one at the buffer */
strncpy(buffer, str, str_len + 2);
}
}
-
- /* Query logs can be in multiple lines.
- * They always start with a tab in the additional ones.
+ /* Query logs can be in multiple lines
+ * They always start with a tab in the additional ones
*/
- else if((str_len > 2) && (buffer[0] != '\0') &&
- (str[0] == '\t'))
- {
+ else if ((str_len > 2) && (buffer[0] != '\0') &&
+ (str[0] == '\t')) {
/* Size of the buffer */
- int buffer_len = strlen(buffer);
+ size_t buffer_len = strlen(buffer);
- p = str +1;
+ p = str + 1;
- /* Removing extra spaces and tabs */
- while(*p == ' ' || *p == '\t')
- {
+ /* Remove extra spaces and tabs */
+ while (*p == ' ' || *p == '\t') {
p++;
}
-
- /* Adding additional message to the saved buffer. */
- if(sizeof(buffer) - buffer_len > str_len +256)
- {
+ /* Add additional message to the saved buffer */
+ if (sizeof(buffer) - buffer_len > str_len + 256) {
/* Here we make sure that the size of the buffer
* minus what was used (strlen) is greater than
* the length of the received message.
*/
- buffer[buffer_len] = ' ';
- buffer[buffer_len +1] = '\0';
- strncat(buffer, str, str_len +3);
+ buffer[buffer_len] = ' ';
+ buffer[buffer_len + 1] = '\0';
+ strncat(buffer, str, str_len + 3);
}
}
continue;
}
-
- /* Send whatever is stored. */
- if(buffer[0] != '\0')
- {
+ /* Send whatever is stored */
+ if (buffer[0] != '\0') {
__send_pgsql_msg(pos, drop_it, buffer);
}
- return(NULL);
+ return (NULL);
}
-/* EOF */