eeda2e83e7126a123906c78f14039e2cf45844cf
[ossec-hids.git] / main.c
1 /* @(#) $Id: ./src/os_dbd/main.c, 2011/09/08 dcid Exp $
2  */
3
4 /* Copyright (C) 2009 Trend Micro Inc.
5  * All rights reserved.
6  *
7  * This program is a free software; you can redistribute it
8  * and/or modify it under the terms of the GNU General Public
9  * License (version 2) as published by the FSF - Free Software
10  * Foundation.
11  *
12  * License details at the LICENSE file included with OSSEC or
13  * online at: http://www.ossec.net/en/licensing.html
14  */
15
16
17 #ifndef DBD
18    #define DBD
19 #endif
20
21 #ifndef ARGV0
22    #define ARGV0 "ossec-dbd"
23 #endif
24
25 #include "shared.h"
26 #include "dbd.h"
27
28
29 /* Prints information regarding enabled databases */
30 void db_info()
31 {
32     print_out(" ");
33     print_out("%s %s - %s", __ossec_name, __version, __author);
34
35     #ifdef UMYSQL
36     print_out("Compiled with MySQL support.");
37     #endif
38
39     #ifdef UPOSTGRES
40     print_out("Compiled with PostgreSQL support.");
41     #endif
42
43     #if !defined(UMYSQL) && !defined(UPOSTGRES)
44     print_out("Compiled without any Database support.");
45     #endif
46
47     print_out(" ");
48     print_out("%s",__license);
49
50     exit(1);
51 }
52
53
54
55 int main(int argc, char **argv)
56 {
57     int c, test_config = 0, run_foreground = 0;
58     int uid = 0,gid = 0;
59
60     /* Using MAILUSER (read only) */
61     char *dir  = DEFAULTDIR;
62     char *user = MAILUSER;
63     char *group = GROUPGLOBAL;
64     char *cfg = DEFAULTCPATH;
65
66
67     /* Database Structure */
68     DBConfig db_config;
69     db_config.error_count = 0;
70
71
72     /* Setting the name */
73     OS_SetName(ARGV0);
74
75
76     while((c = getopt(argc, argv, "vVdhtfu:g:D:c:")) != -1){
77         switch(c){
78             case 'V':
79                 db_info();
80                 break;
81             case 'v':
82                 db_info();
83                 break;
84             case 'h':
85                 help(ARGV0);
86                 break;
87             case 'd':
88                 nowDebug();
89                 break;
90             case 'f':
91                 run_foreground = 1;
92                 break;
93             case 'u':
94                 if(!optarg)
95                     ErrorExit("%s: -u needs an argument",ARGV0);
96                 user=optarg;
97                 break;
98             case 'g':
99                 if(!optarg)
100                     ErrorExit("%s: -g needs an argument",ARGV0);
101                 group=optarg;
102                 break;
103             case 'D':
104                 if(!optarg)
105                     ErrorExit("%s: -D needs an argument",ARGV0);
106                 dir=optarg;
107                 break;
108             case 'c':
109                 if(!optarg)
110                     ErrorExit("%s: -c needs an argument",ARGV0);
111                 cfg = optarg;
112                 break;
113             case 't':
114                 test_config = 1;
115                 break;
116             default:
117                 help(ARGV0);
118                 break;
119         }
120
121     }
122
123
124     /* Starting daemon */
125     debug1(STARTED_MSG, ARGV0);
126
127
128     /* Check if the user/group given are valid */
129     uid = Privsep_GetUser(user);
130     gid = Privsep_GetGroup(group);
131     if((uid < 0)||(gid < 0))
132     {
133         ErrorExit(USER_ERROR, ARGV0, user, group);
134     }
135
136
137     /* Reading configuration */
138     if((c = OS_ReadDBConf(test_config, cfg, &db_config)) < 0)
139     {
140         ErrorExit(CONFIG_ERROR, ARGV0, cfg);
141     }
142
143
144     /* Exit here if test config is set */
145     if(test_config)
146         exit(0);
147
148
149     if(!run_foreground)
150     {
151         /* Going on daemon mode */
152         nowDaemon();
153         goDaemon();
154     }
155
156
157
158     /* Not configured */
159     if(c == 0)
160     {
161         verbose("%s: Database not configured. Clean exit.", ARGV0);
162         exit(0);
163     }
164
165
166     /* Maybe disable this debug? */
167     debug1("%s: DEBUG: Connecting to '%s', using '%s', '%s', '%s', %d,'%s'.",
168            ARGV0, db_config.host, db_config.user,
169            db_config.pass, db_config.db,db_config.port,db_config.sock);
170
171
172     /* Setting config pointer */
173     osdb_setconfig(&db_config);
174
175
176     /* Getting maximum reconned attempts */
177     db_config.maxreconnect = getDefine_Int("dbd",
178                                            "reconnect_attempts", 1, 9999);
179
180
181     /* Connecting to the database */
182     c = 0;
183     while(c <= (db_config.maxreconnect * 10))
184     {
185         db_config.conn = osdb_connect(db_config.host, db_config.user,
186                                       db_config.pass, db_config.db,
187                                       db_config.port,db_config.sock);
188
189         /* If we are able to reconnect, keep going */
190         if(db_config.conn)
191         {
192             break;
193         }
194
195         c++;
196         sleep(c * 60);
197
198     }
199
200
201     /* If after the maxreconnect attempts, it still didn't work, exit here. */
202     if(!db_config.conn)
203     {
204         merror(DB_CONFIGERR, ARGV0);
205         ErrorExit(CONFIG_ERROR, ARGV0, cfg);
206     }
207
208
209     /* We must notify that we connected -- easy debugging */
210     verbose("%s: Connected to database '%s' at '%s'.",
211             ARGV0, db_config.db, db_config.host);
212
213
214     /* Privilege separation */  
215     if(Privsep_SetGroup(gid) < 0)
216         ErrorExit(SETGID_ERROR,ARGV0,group);
217
218
219     /* chrooting */
220     if(Privsep_Chroot(dir) < 0)
221         ErrorExit(CHROOT_ERROR,ARGV0,dir);
222
223
224     /* Now on chroot */
225     nowChroot();
226
227
228     /* Inserting server info into the db */
229     db_config.server_id = OS_Server_ReadInsertDB(&db_config);
230     if(db_config.server_id <= 0)
231     {
232         ErrorExit(CONFIG_ERROR, ARGV0, cfg);
233     }
234
235
236     /* Read rules and insert into the db */
237     if(OS_InsertRulesDB(&db_config) < 0)
238     {
239         ErrorExit(CONFIG_ERROR, ARGV0, cfg);
240     }
241
242
243     /* Changing user */
244     if(Privsep_SetUser(uid) < 0)
245         ErrorExit(SETUID_ERROR,ARGV0,user);
246
247
248     /* Basic start up completed. */
249     debug1(PRIVSEP_MSG,ARGV0,dir,user);
250
251
252     /* Signal manipulation */
253     StartSIG(ARGV0);
254
255
256     /* Creating PID files */
257     if(CreatePID(ARGV0, getpid()) < 0)
258         ErrorExit(PID_ERROR,ARGV0);
259
260
261     /* Start up message */
262     verbose(STARTUP_MSG, ARGV0, (int)getpid());
263
264
265     /* the real daemon now */   
266     OS_DBD(&db_config);
267     exit(0);
268 }
269
270
271 /* EOF */