Re-add codename to mail banner.

[inoclam.git] / config.cxx

/*
 * inoclam - Inotify+ClamAV virus scanner
 * Copyright (C) 2007 Vermont Department of Taxes
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Contributor(s):
 *   Tom Cort <tom.cort@state.vt.us>
 *   Matt Gagne <matt.gagne@state.vt.us>
 */

#include <confuse.h>
#include <libdaemon/dlog.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <string.h>
#include <strings.h>
#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>

#include "config.hxx"
#include "clam.hxx"
#include "inotify.hxx"
#include "smtp.hxx"

/**
 * Default Constructor
 */
config::config()
{
        virus_removal_enabled = cfg_false;
        smtp_enabled = cfg_true;
        watch_dir = NULL;
        smtp_host = NULL;
        smtp_port = 0;
        smtp_to = NULL;
        smtp_from = NULL;
        smtp_subject_virus = NULL;
        smtp_subject_file = NULL;
}

/**
 * Checks if the configuration file exists as a regular file.
 * @return Returns 1 if the file exists and is regular, else 0.
 */
int config::config_file_exists()
{
        struct stat s;

        /* No config file defined - NULL Pointer Check */
        if (!DEFAULT_CONFIGFILE) {
                return 0;
        }

        /* File exists? */
        if (stat(DEFAULT_CONFIGFILE, &s) < 0) {
                return 0;
        }

        /* File is a regular file? */
        if (!S_ISREG(s.st_mode)) {
                return 0;
        }

        return 1;
}

/**
 * Release any resources that hold configuration information.
 * This function effectively resets all configurable values.
 * It should be called at the end of the program.
 */
void config::config_free()
{
        if (watch_dir != NULL) {
                free(watch_dir);
                watch_dir = NULL;
        }

        virus_removal_enabled = cfg_true;

        smtp_enabled = cfg_true;

        if (smtp_host != NULL) {
                free(smtp_host);
                smtp_host = NULL;
        }

        smtp_port = 0;

        if (smtp_to != NULL) {
                free(smtp_to);
                smtp_to = NULL;
        }

        if (smtp_from != NULL) {
                free(smtp_from);
                smtp_from = NULL;
        }

        if (smtp_subject_virus != NULL) {
                free(smtp_subject_virus);
                smtp_subject_virus = NULL;
        }

        if (smtp_subject_file != NULL) {
                free(smtp_subject_file);
                smtp_subject_file = NULL;
        }

}

/**
 * Examines each value of the current configuration.
 * If a value was not set, it is set to the default.
 */
void config::config_with_defaults()
{
        if (watch_dir == NULL) {
                watch_dir = strdup(WATCH_DIR);
        }

        if (smtp_host == NULL) {
                smtp_host = strdup(SMTP_HOST);
        }

        if (smtp_port == 0) {
                smtp_port = SMTP_PORT;
        }

        if (smtp_to == NULL) {
                smtp_to = strdup(SMTP_TO);
        }

        if (smtp_from == NULL) {
                smtp_from = strdup(SMTP_FROM);
        }

        if (smtp_subject_virus == NULL) {
                smtp_subject_virus = strdup(SMTP_SUBJECT_VIRUS);
        }

        if (smtp_subject_file == NULL) {
                smtp_subject_file = strdup(SMTP_SUBJECT_FILE);
        }

}

/**
 * Parses an inoclam.conf configuration file.
 */
void config::parse()
{
        int rc;

        config_free();

        cfg_t *cfg;
        cfg_opt_t opts[] = {
                CFG_SIMPLE_STR("watch_dir", &watch_dir),
                CFG_SIMPLE_BOOL("virus_removal_enabled", &virus_removal_enabled),
                CFG_SIMPLE_BOOL("smtp_enabled", &smtp_enabled),
                CFG_SIMPLE_STR("smtp_host", &smtp_host),
                CFG_SIMPLE_INT("smtp_port", &smtp_port),
                CFG_SIMPLE_STR("smtp_to", &smtp_to),
                CFG_SIMPLE_STR("smtp_from", &smtp_from),
                CFG_SIMPLE_STR("smtp_subject_virus", &smtp_subject_virus),
                CFG_SIMPLE_STR("smtp_subject_file", &smtp_subject_file),
                CFG_END()
        };

        cfg = NULL;

        if (!config_file_exists()) {
                daemon_log(LOG_INFO, "Configuration file '%s' not found.", DEFAULT_CONFIGFILE);
                config_with_defaults();
                return;
        }

        cfg = cfg_init(opts, 0);
        if (!cfg) {
                daemon_log(LOG_ERR, "cfg_init failed!");
                config_with_defaults();
                return;
        }

        rc = cfg_parse(cfg, DEFAULT_CONFIGFILE);
        if (rc == CFG_PARSE_ERROR) {
                daemon_log(LOG_ERR, "parser error '%s'", DEFAULT_CONFIGFILE);
        }

        if (cfg) {
                cfg_free(cfg);
                cfg = NULL;
        }

        config_with_defaults();
}

cfg_bool_t config::getSMTPEnabled()
{
        return smtp_enabled;
}

cfg_bool_t config::getVirusRemovalEnabled()
{
        return virus_removal_enabled;
}

char *config::getHost()
{
        return smtp_host;
}

int config::getPort()
{
        return smtp_port;
}

char *config::getTo()
{
        return smtp_to;
}

char *config::getFrom()
{
        return smtp_from;
}

char *config::getVirusSubject()
{
        return smtp_subject_virus;
}

char *config::getFileSubject()
{
        return smtp_subject_file;
}

char *config::getWatchDir()
{
        return watch_dir;
}

/**
 * Deconstructor
 */
config::~config()
{
        config_free();  
}