src/config.cxx

   1 /*
   2  * inoclam - Inotify+ClamAV virus scanner
   3  * Copyright (C) 2007, 2008, 2009 Vermont Department of Taxes
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2 as
   7  * published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #include <list>
  20 #include <confuse.h>
  21 #include <libdaemon/dlog.h>
  22 #include <unistd.h>
  23 #include <sys/types.h>
  24 #include <sys/stat.h>
  25 #include <string.h>
  26 #include <strings.h>
  27 #include <errno.h>
  28 #include <unistd.h>
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31
  32 #include "config.hxx"
  33 #include "clam.hxx"
  34 #include "inotify.hxx"
  35 #include "smtp.hxx"
  36
  37 config::config()
  38 {
  39         directory = NULL;
  40         file_email_subject = NULL;
  41         virus_email_subject = NULL;
  42         host = NULL;
  43         from = NULL;
  44         to = NULL;
  45 }
  46
  47 void config::setDirectory(char *new_dir)
  48 {
  49         directory = new_dir;
  50 }
  51
  52 char *config::getDirectory()
  53 {
  54         return directory;
  55 }
  56
  57 void config::setFileEMailSubject(char *subject)
  58 {
  59         file_email_subject = subject;
  60 }
  61
  62 char *config::getFileEMailSubject()
  63 {
  64         return file_email_subject;
  65 }
  66
  67 void config::setVirusEMailSubject(char *subject)
  68 {
  69         virus_email_subject = subject;
  70 }
  71
  72 char *config::getVirusEMailSubject()
  73 {
  74         return virus_email_subject;
  75 }
  76
  77 void config::setFileEMailEnabled(cfg_bool_t enabled)
  78 {
  79         file_email_enabled = enabled;
  80 }
  81
  82 cfg_bool_t config::getFileEMailEnabled()
  83 {
  84         return file_email_enabled;
  85 }
  86
  87 void config::setVirusEMailEnabled(cfg_bool_t enabled)
  88 {
  89         virus_email_enabled = enabled;
  90 }
  91
  92
  93 cfg_bool_t config::getVirusEMailEnabled()
  94 {
  95         return virus_email_enabled;
  96 }
  97
  98 void config::setVirusRemovalEnabled(cfg_bool_t enabled)
  99 {
 100         virus_removal_enabled = enabled;
 101 }
 102
 103 cfg_bool_t config::getVirusRemovalEnabled()
 104 {
 105         return virus_removal_enabled;
 106 }
 107
 108 char *config::getHost()
 109 {
 110         return host;
 111 }
 112
 113 void config::setHost(char *h)
 114 {
 115         host = h;
 116 }
 117
 118 int config::getPort()
 119 {
 120         return port;
 121 }
 122
 123 void config::setPort(int p)
 124 {
 125         port = p;
 126 }
 127
 128 char *config::getFrom()
 129 {
 130         return from;
 131 }
 132
 133 void config::setFrom(char *f)
 134 {
 135         from = f;
 136 }
 137
 138 std::list < char *>*config::getTo()
 139 {
 140         return to;
 141 }
 142
 143 void config::setTo(std::list < char *>*t)
 144 {
 145         to = t;
 146 }
 147
 148 config::~config()
 149 {
 150         if (to) {
 151                 for (std::list < char *>::iterator itr = to->begin(); itr != to->end(); ++itr) {
 152                         free(*itr);
 153                 }
 154                 delete to;
 155                 to = NULL;
 156         }
 157         if (directory) {
 158                 free(directory);
 159                 directory = NULL;
 160         }
 161
 162         if (virus_email_subject) {
 163                 free(virus_email_subject);
 164                 virus_email_subject = NULL;
 165         }
 166
 167         if (file_email_subject) {
 168                 free(file_email_subject);
 169                 file_email_subject = NULL;
 170         }
 171
 172         if (host) {
 173                 free(host);
 174                 host = NULL;
 175         }
 176
 177         if (from) {
 178                 free(from);
 179                 from = NULL;
 180         }
 181 }
 182
 183 /**
 184  * The configuration file path.
 185  */
 186 char *configfile;
 187
 188 /**
 189  * Parses an inoclam.conf configuration file.
 190  */
 191 std::list < config::config * >*config_parse()
 192 {
 193         std::list < config::config * >*conf_list;
 194         conf_list = new std::list < config::config * >;
 195
 196         int n, i, j;
 197         int rc;
 198
 199         cfg_t *cfg;
 200
 201         cfg_opt_t file_opts[] = {
 202                 CFG_BOOL((char*)"email_alert", cfg_false, CFGF_NONE),
 203                 CFG_STR((char*)"email_subject", (char*)"[inoclam] File Created/Changed", CFGF_NONE),
 204                 CFG_END()
 205         };
 206
 207         cfg_opt_t virus_opts[] = {
 208                 CFG_BOOL((char*)"auto_remove", cfg_true, CFGF_NONE),
 209                 CFG_BOOL((char*)"email_alert", cfg_true, CFGF_NONE),
 210                 CFG_STR((char*)"email_subject", (char*)"[inoclam] Virus Detected", CFGF_NONE),
 211                 CFG_END()
 212         };
 213
 214         cfg_opt_t smtp_opts[] = {
 215                 CFG_STR((char*)"host", (char*)"localhost", CFGF_NONE),
 216                 CFG_INT((char*)"port", 25, CFGF_NONE),
 217                 CFG_STR((char*)"from", (char*)"root@localhost", CFGF_NONE),
 218                 CFG_STR_LIST((char*)"to", (char*)"{root@localhost}", CFGF_NONE),
 219                 CFG_END()
 220         };
 221
 222         cfg_opt_t watch_opts[] = {
 223                 CFG_STR((char*)"directory", (char*)"/tmp", CFGF_NONE),
 224                 CFG_SEC((char*)"file", file_opts, CFGF_NONE),
 225                 CFG_SEC((char*)"virus", virus_opts, CFGF_NONE),
 226                 CFG_SEC((char*)"smtp", smtp_opts, CFGF_NONE),
 227                 CFG_END()
 228         };
 229
 230         cfg_opt_t opts[] = {
 231                 CFG_SEC((char*)"watch", watch_opts, CFGF_MULTI),
 232                 CFG_END()
 233         };
 234
 235         cfg = NULL;
 236
 237         cfg = cfg_init(opts, 0);
 238         if (!cfg) {
 239                 daemon_log(LOG_ERR, "cfg_init failed!");
 240                 return conf_list;
 241         }
 242
 243         rc = cfg_parse(cfg, configfile);
 244         if (rc == CFG_PARSE_ERROR) {
 245                 daemon_log(LOG_ERR, "parser error '%s'", configfile);
 246         }
 247
 248         n = cfg_size(cfg, "watch");
 249         for (i = 0; i < n; i++) {
 250                 config::config * conf;
 251                 conf = new config::config();
 252
 253                 cfg_t *smtp_cfg = NULL;
 254                 cfg_t *file_cfg = NULL;
 255                 cfg_t *virus_cfg = NULL;
 256                 cfg_t *watch_cfg = cfg_getnsec(cfg, "watch", i);
 257
 258                 conf->setDirectory(strdup(cfg_getstr(watch_cfg, "directory")));
 259
 260                 file_cfg = cfg_getsec(watch_cfg, "file");
 261                 conf->setFileEMailSubject(strdup(cfg_getstr(file_cfg, "email_subject")));
 262                 conf->setFileEMailEnabled(cfg_getbool(file_cfg, "email_alert"));
 263
 264                 virus_cfg = cfg_getsec(watch_cfg, "virus");
 265                 conf->setVirusEMailSubject(strdup(cfg_getstr(virus_cfg, "email_subject")));
 266                 conf->setVirusEMailEnabled(cfg_getbool(virus_cfg, "email_alert"));
 267                 conf->setVirusRemovalEnabled(cfg_getbool(virus_cfg, "auto_remove"));
 268
 269                 smtp_cfg = cfg_getsec(watch_cfg, "smtp");
 270                 conf->setHost(strdup(cfg_getstr(smtp_cfg, "host")));
 271                 conf->setPort(cfg_getint(smtp_cfg, "port"));
 272                 conf->setFrom(strdup(cfg_getstr(smtp_cfg, "from")));
 273
 274
 275                 std::list < char *>*to;
 276                 to = new std::list < char *>;
 277                 for (j = 0; j < cfg_size(smtp_cfg, "to"); j++) {
 278                         to->push_front(strdup(cfg_getnstr(smtp_cfg, "to", j)));
 279                 }
 280                 conf->setTo(to);
 281
 282                 conf_list->push_front(conf);
 283         }
 284
 285         if (cfg) {
 286                 cfg_free(cfg);
 287                 cfg = NULL;
 288         }
 289
 290         return conf_list;
 291 }