-/* @(#) $Id$ */
+/* @(#) $Id: ./src/os_maild/maild.c, 2011/09/08 dcid Exp $
+ */
/* Copyright (C) 2009 Trend Micro Inc.
* All rights reserved.
* License (version 2) as published by the FSF - Free Software
* Foundation.
*
- * License details at the LICENSE file included with OSSEC or
+ * License details at the LICENSE file included with OSSEC or
* online at: http://www.ossec.net/en/licensing.html
*/
#include "maild.h"
#include "mail_list.h"
-
void OS_Run(MailConfig *mail);
int main(int argc, char **argv)
/* Setting the name */
OS_SetName(ARGV0);
-
+
while((c = getopt(argc, argv, "Vdhtfu:g:D:c:")) != -1){
switch(c){
if(!optarg)
ErrorExit("%s: -D needs an argument",ARGV0);
dir=optarg;
+ break;
case 'c':
if(!optarg)
ErrorExit("%s: -c needs an argument",ARGV0);
cfg = optarg;
break;
case 't':
- test_config = 1;
+ test_config = 1;
break;
default:
help(ARGV0);
if((uid < 0)||(gid < 0))
ErrorExit(USER_ERROR,ARGV0,user,group);
-
/* Reading configuration */
if(MailConf(test_config, cfg, &mail) < 0)
ErrorExit(CONFIG_ERROR, ARGV0, cfg);
mail.strict_checking = getDefine_Int("maild",
"strict_checking",
0, 1);
-
+
/* Get groupping */
mail.groupping = getDefine_Int("maild",
"groupping",
0, 1);
-
+
/* Getting subject type */
mail.subject_full = getDefine_Int("maild",
"full_subject",
0, 1);
-
-
+
+#ifdef GEOIP
+ /* Get GeoIP */
+ mail.geoip = getDefine_Int("maild",
+ "geoip",
+ 0, 1);
+#endif
+
+
/* Exit here if test config is set */
if(test_config)
exit(0);
-
- if(!run_foreground)
+
+ if(!run_foreground)
{
nowDaemon();
goDaemon();
}
-
+
/* Privilege separation */
if(Privsep_SetGroup(gid) < 0)
ErrorExit(SETGID_ERROR,ARGV0,group);
-
+
/* chrooting */
if(Privsep_Chroot(dir) < 0)
ErrorExit(CHROOT_ERROR,ARGV0,dir);
nowChroot();
-
- /* Changing user */
+
+ /* Changing user */
if(Privsep_SetUser(uid) < 0)
ErrorExit(SETUID_ERROR,ARGV0,user);
/* Signal manipulation */
StartSIG(ARGV0);
-
+
/* Creating PID files */
if(CreatePID(ARGV0, getpid()) < 0)
ErrorExit(PID_ERROR, ARGV0);
-
+
/* Start up message */
verbose(STARTUP_MSG, ARGV0, (int)getpid());
-
+
/* the real daemon now */
OS_Run(&mail);
MailMsg *s_msg = NULL;
MailMsg *msg_sms = NULL;
- time_t tm;
- struct tm *p;
+ time_t tm;
+ struct tm *p;
int i = 0;
int mailtosend = 0;
int childcount = 0;
- int today = 0;
+ int today = 0;
int thishour = 0;
int n_errs = 0;
/* Creating the list */
- OS_CreateMailList(MAIL_LIST_SIZE);
-
-
+ OS_CreateMailList(MAIL_LIST_SIZE);
+
+
/* Setting default timeout */
mail_timeout = DEFAULT_TIMEOUT;
-
+
/* Clearing global vars */
_g_subject_level = 0;
memset(_g_subject, '\0', SUBJECT_SIZE +2);
tm = time(NULL);
p = localtime(&tm);
-
+
/* SMS messages are sent without delay */
if(msg_sms)
{
pid_t pid;
-
+
pid = fork();
-
+
if(pid < 0)
{
merror("%s: Fork failed. cause: %d - %s", ARGV0, errno, strerror(errno));
/* Increasing child count */
childcount++;
}
-
+
/* If mail_timeout == NEXTMAIL_TIMEOUT, we will try to get
* more messages, before sending anything
{
/* getting more messages */
}
-
-
- /* Hour changed. Send all supressed mails */
+
+
+ /* Hour changed. Send all supressed mails */
else if(((mailtosend < mail->maxperhour) && (mailtosend != 0))||
((p->tm_hour != thishour) && (childcount < MAXCHILDPROCESS)))
{
{
if(OS_Sendmail(mail, p) < 0)
merror(SNDMAIL_ERROR,ARGV0,mail->smtpserver);
-
- exit(0);
+
+ exit(0);
}
-
+
/* Cleaning the memory */
- mailmsg = OS_PopLastMail();
+ mailmsg = OS_PopLastMail();
do
{
- FreeMail(mailmsg);
+ FreeMail(mailmsg);
mailmsg = OS_PopLastMail();
}while(mailmsg);
-
-
- /* Increasing child count */
- childcount++;
+
+
+ /* Increasing child count */
+ childcount++;
/* Clearing global vars */
_g_subject[0] = '\0';
_g_subject[SUBJECT_SIZE -1] = '\0';
_g_subject_level = 0;
-
-
+
+
/* Cleaning up set values */
if(mail->gran_to)
{
/* If we sent everything */
if(p->tm_hour != thishour)
{
- thishour = p->tm_hour;
+ thishour = p->tm_hour;
mailtosend = 0;
}
}
-
+
/* Saved message for the do_not_group option.
*/
if(s_msg)
i++;
}
}
-
+
OS_AddMailtoList(s_msg);
s_msg = NULL;
mailtosend++;
continue;
}
-
-
+
+
/* Receive message from queue */
if((msg = OS_RecvMailQ(fileq, p, mail, &msg_sms)) != NULL)
{
{
OS_AddMailtoList(msg);
}
-
+
/* Change timeout to see if any new message is coming shortly */
if(mail->groupping)
/* Waiting for the childs .. */
- while (childcount)
+ while (childcount)
{
int wp;
int p_status;
wp = waitpid((pid_t) -1, &p_status, WNOHANG);
if (wp < 0)
{
- merror(WAITPID_ERROR, ARGV0);
+ merror(WAITPID_ERROR, ARGV0);
n_errs++;
}
- /* if = 0, we still need to wait for the child process */
- else if (wp == 0)
+ /* if = 0, we still need to wait for the child process */
+ else if (wp == 0)
break;
else
{
if(p_status != 0)
{
+ merror(CHLDWAIT_ERROR,ARGV0,p_status);
merror(SNDMAIL_ERROR,ARGV0,mail->smtpserver);
n_errs++;
}
/* Too many errors */
if(n_errs > 6)
{
+ merror(TOOMANY_WAIT_ERROR,ARGV0);
merror(SNDMAIL_ERROR,ARGV0,mail->smtpserver);
exit(1);
}
}
-
+
}
}
projects / ossec-hids.git / blobdiff