/*
 *  Copyright (C) 2013-2019 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
 *  Copyright (C) 2007-2013 Sourcefire, Inc.
 *
 *  Authors: Tomasz Kojm
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2 as
 *  published by the Free Software Foundation.
 *
 *  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., 51 Franklin Street, Fifth Floor, Boston,
 *  MA 02110-1301, USA.
 */

#if HAVE_CONFIG_H
#include "clamav-config.h"
#endif

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include <zlib.h>

#include "clamav.h"
#include "dconf.h"
#include "readdb.h"
#include "str.h"
#include "others.h"

#include "mpool.h"

struct dconf_module {
    const char *mname; /* module name */
    const char *sname; /* submodule name */
    uint32_t bflag;    /* bit flag */
    uint8_t state;     /* default state (on/off) */
};

#ifdef CL_EXPERIMENTAL
#define DCONF_ENABLE_EXPERIMENTAL 1
#else
#define DCONF_ENABLE_EXPERIMENTAL 0
#endif

static struct dconf_module modules[] = {

    {"PE", "PARITE", PE_CONF_PARITE, 1},
    {"PE", "KRIZ", PE_CONF_KRIZ, 1},
    {"PE", "MAGISTR", PE_CONF_MAGISTR, 1},
    {"PE", "POLIPOS", PE_CONF_POLIPOS, 1},
    {"PE", "MD5SECT", PE_CONF_MD5SECT, 1},
    {"PE", "UPX", PE_CONF_UPX, 1},
    {"PE", "FSG", PE_CONF_FSG, 1},
    {"PE", "SWIZZOR", PE_CONF_SWIZZOR, 0},

    {"PE", "PETITE", PE_CONF_PETITE, 1},
    {"PE", "PESPIN", PE_CONF_PESPIN, 1},
    {"PE", "YC", PE_CONF_YC, 1},
    {"PE", "WWPACK", PE_CONF_WWPACK, 1},

    {"PE", "NSPACK", PE_CONF_NSPACK, 1},
    {"PE", "MEW", PE_CONF_MEW, 1},
    {"PE", "UPACK", PE_CONF_UPACK, 1},
    {"PE", "ASPACK", PE_CONF_ASPACK, 1},
    {"PE", "CATALOG", PE_CONF_CATALOG, 1},
    {"PE", "CERTS", PE_CONF_CERTS, 1},
    {"PE", "MATCHICON", PE_CONF_MATCHICON, 1},
    {"PE", "IMPTBL", PE_CONF_IMPTBL, 1},

    {"ELF", NULL, 0x1, 1},

    {"MACHO", NULL, 0x1, 1},

    {"ARCHIVE", "RAR", ARCH_CONF_RAR, 1},
    {"ARCHIVE", "ZIP", ARCH_CONF_ZIP, 1},
    {"ARCHIVE", "GZIP", ARCH_CONF_GZ, 1},
    {"ARCHIVE", "BZIP", ARCH_CONF_BZ, 1},
    {"ARCHIVE", "ARJ", ARCH_CONF_ARJ, 1},
    {"ARCHIVE", "SZDD", ARCH_CONF_SZDD, 1},
    {"ARCHIVE", "CAB", ARCH_CONF_CAB, 1},
    {"ARCHIVE", "CHM", ARCH_CONF_CHM, 1},
    {"ARCHIVE", "OLE2", ARCH_CONF_OLE2, 1},
    {"ARCHIVE", "TAR", ARCH_CONF_TAR, 1},
    {"ARCHIVE", "CPIO", ARCH_CONF_CPIO, 1},
    {"ARCHIVE", "BINHEX", ARCH_CONF_BINHEX, 1},
    {"ARCHIVE", "SIS", ARCH_CONF_SIS, 1},
    {"ARCHIVE", "NSIS", ARCH_CONF_NSIS, 1},
    {"ARCHIVE", "AUTOIT", ARCH_CONF_AUTOIT, 1},
    {"ARCHIVE", "ISHIELD", ARCH_CONF_ISHIELD, 1},
    {"ARCHIVE", "7zip", ARCH_CONF_7Z, 1},
    {"ARCHIVE", "ISO9660", ARCH_CONF_ISO9660, 1},
    {"ARCHIVE", "DMG", ARCH_CONF_DMG, 1},
    {"ARCHIVE", "XAR", ARCH_CONF_XAR, 1},
    {"ARCHIVE", "HFSPLUS", ARCH_CONF_HFSPLUS, 1},
    {"ARCHIVE", "XZ", ARCH_CONF_XZ, 1},
    {"ARCHIVE", "PASSWD", ARCH_CONF_PASSWD, 1},
    {"ARCHIVE", "MBR", ARCH_CONF_MBR, 1},
    {"ARCHIVE", "GPT", ARCH_CONF_GPT, 1},
    {"ARCHIVE", "APM", ARCH_CONF_APM, 1},
    {"ARCHIVE", "EGG", ARCH_CONF_EGG, 1},

    {"DOCUMENT", "HTML", DOC_CONF_HTML, 1},
    {"DOCUMENT", "RTF", DOC_CONF_RTF, 1},
    {"DOCUMENT", "PDF", DOC_CONF_PDF, 1},
    {"DOCUMENT", "SCRIPT", DOC_CONF_SCRIPT, 1},
    {"DOCUMENT", "HTMLSKIPRAW", DOC_CONF_HTML_SKIPRAW, 1},
    {"DOCUMENT", "JSNORM", DOC_CONF_JSNORM, 1},
    {"DOCUMENT", "SWF", DOC_CONF_SWF, 1},
    {"DOCUMENT", "OOXML", DOC_CONF_OOXML, 1},
    {"DOCUMENT", "MSPML", DOC_CONF_MSXML, 1},
    {"DOCUMENT", "HWP", DOC_CONF_HWP, 1},

    {"MAIL", "MBOX", MAIL_CONF_MBOX, 1},
    {"MAIL", "TNEF", MAIL_CONF_TNEF, 1},

    {"OTHER", "UUENCODED", OTHER_CONF_UUENC, 1},
    {"OTHER", "SCRENC", OTHER_CONF_SCRENC, 1},
    {"OTHER", "RIFF", OTHER_CONF_RIFF, 1},
    {"OTHER", "JPEG", OTHER_CONF_JPEG, 1},
    {"OTHER", "CRYPTFF", OTHER_CONF_CRYPTFF, 1},
    {"OTHER", "DLP", OTHER_CONF_DLP, 1},
    {"OTHER", "MYDOOMLOG", OTHER_CONF_MYDOOMLOG, 1},
    {"OTHER", "PREFILTERING", OTHER_CONF_PREFILTERING, 1},
    {"OTHER", "PDFNAMEOBJ", OTHER_CONF_PDFNAMEOBJ, 1},
    {"OTHER", "PRTNINTXN", OTHER_CONF_PRTNINTXN, 1},
    {"OTHER", "LZW", OTHER_CONF_LZW, 1},

    {"PHISHING", "ENGINE", PHISHING_CONF_ENGINE, 1},
    {"PHISHING", "ENTCONV", PHISHING_CONF_ENTCONV, 1},

    {"BYTECODE", "INTERPRETER", BYTECODE_INTERPRETER, 1},
    {"BYTECODE", "JIT X86", BYTECODE_JIT_X86, 1},
    {"BYTECODE", "JIT PPC", BYTECODE_JIT_PPC, 1},
    {"BYTECODE", "JIT ARM", BYTECODE_JIT_ARM, 0},

    {"STATS", "DISABLED", DCONF_STATS_DISABLED, 0},
    {"STATS", "PESECTION DISABLED", DCONF_STATS_PE_SECTION_DISABLED, 0},

    {"PCRE", "SUPPORT", PCRE_CONF_SUPPORT, 1},
    {"PCRE", "OPTIONS", PCRE_CONF_OPTIONS, 1},
    {"PCRE", "GLOBAL", PCRE_CONF_GLOBAL, 1},

    {NULL, NULL, 0, 0}};

#ifdef USE_MPOOL
struct cli_dconf *cli_dconf_init(mpool_t *mempool)
#else
struct cli_dconf *cli_dconf_init(void)
#endif
{
    unsigned int i;
    struct cli_dconf *dconf;

    dconf = (struct cli_dconf *)MPOOL_CALLOC(mempool, sizeof(struct cli_dconf), 1);
    if (!dconf)
        return NULL;

    for (i = 0; modules[i].mname; i++) {
        if (!strcmp(modules[i].mname, "PE")) {
            if (modules[i].state)
                dconf->pe |= modules[i].bflag;

        } else if (!strcmp(modules[i].mname, "ELF")) {
            if (modules[i].state)
                dconf->elf |= modules[i].bflag;

        } else if (!strcmp(modules[i].mname, "MACHO")) {
            if (modules[i].state)
                dconf->macho |= modules[i].bflag;

        } else if (!strcmp(modules[i].mname, "ARCHIVE")) {
            if (modules[i].state)
                dconf->archive |= modules[i].bflag;

        } else if (!strcmp(modules[i].mname, "DOCUMENT")) {
            if (modules[i].state)
                dconf->doc |= modules[i].bflag;

        } else if (!strcmp(modules[i].mname, "MAIL")) {
            if (modules[i].state)
                dconf->mail |= modules[i].bflag;

        } else if (!strcmp(modules[i].mname, "OTHER")) {
            if (modules[i].state)
                dconf->other |= modules[i].bflag;
        } else if (!strcmp(modules[i].mname, "PHISHING")) {
            if (modules[i].state)
                dconf->phishing |= modules[i].bflag;
        } else if (!strcmp(modules[i].mname, "BYTECODE")) {
            if (modules[i].state)
                dconf->bytecode |= modules[i].bflag;
        } else if (!strcmp(modules[i].mname, "STATS")) {
            if (modules[i].state)
                dconf->stats |= modules[i].bflag;
        } else if (!strcmp(modules[i].mname, "PCRE")) {
            if (modules[i].state)
                dconf->pcre |= modules[i].bflag;
        }
    }

    return dconf;
}

void cli_dconf_print(struct cli_dconf *dconf)
{
    unsigned int pe = 0, elf = 0, macho = 0, arch = 0, doc = 0, mail = 0;
    unsigned int other = 0, phishing = 0, i, bytecode = 0, stats = 0, pcre = 0;

    cli_dbgmsg("Dynamic engine configuration settings:\n");
    cli_dbgmsg("--------------------------------------\n");

    for (i = 0; modules[i].mname; i++) {
        if (!strcmp(modules[i].mname, "PE")) {
            if (!pe) {
                cli_dbgmsg("Module PE: %s\n", dconf->pe ? "On" : "Off");
                pe = 1;
            }

            if (dconf->pe)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->pe & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "ELF")) {
            if (!elf) {
                cli_dbgmsg("Module ELF: %s\n", dconf->elf ? "On" : "Off");
                elf = 1;
            }
        } else if (!strcmp(modules[i].mname, "MACHO")) {
            if (!macho) {
                cli_dbgmsg("Module MACHO: %s\n", dconf->elf ? "On" : "Off");
                macho = 1;
            }
        } else if (!strcmp(modules[i].mname, "ARCHIVE")) {
            if (!arch) {
                cli_dbgmsg("Module ARCHIVE: %s\n", dconf->archive ? "On" : "Off");
                arch = 1;
            }

            if (dconf->archive)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->archive & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "DOCUMENT")) {
            if (!doc) {
                cli_dbgmsg("Module DOCUMENT: %s\n", dconf->doc ? "On" : "Off");
                doc = 1;
            }

            if (dconf->doc)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->doc & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "MAIL")) {
            if (!mail) {
                cli_dbgmsg("Module MAIL: %s\n", dconf->mail ? "On" : "Off");
                mail = 1;
            }

            if (dconf->mail)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->mail & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "OTHER")) {
            if (!other) {
                cli_dbgmsg("Module OTHER: %s\n", dconf->other ? "On" : "Off");
                other = 1;
            }

            if (dconf->other)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->other & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "PHISHING")) {
            if (!phishing) {
                cli_dbgmsg("Module PHISHING %s\n", dconf->phishing ? "On" : "Off");
                phishing = 1;
            }

            if (dconf->phishing)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->phishing & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "BYTECODE")) {
            if (!bytecode) {
                cli_dbgmsg("Module BYTECODE %s\n", dconf->bytecode ? "On" : "Off");
                bytecode = 1;
            }

            if (dconf->bytecode)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->bytecode & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "STATS")) {
            if (!stats) {
                cli_dbgmsg("Module STATS %s\n", dconf->stats ? "On" : "Off");
                stats = 1;
            }

            if (dconf->stats)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->stats & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
        } else if (!strcmp(modules[i].mname, "PCRE")) {
#if HAVE_PCRE
            if (!pcre) {
                cli_dbgmsg("Module PCRE %s\n", dconf->pcre ? "On" : "Off");
                pcre = 1;
            }

            if (dconf->pcre)
                cli_dbgmsg("   * Submodule %10s:\t%s\n", modules[i].sname, (dconf->pcre & modules[i].bflag) ? "On" : "** Off **");
            else
                continue;
#else
            if (!pcre) {
                cli_dbgmsg("Module PCRE Off\n");
                pcre = 1;
            }

            continue;
#endif
        }
    }
}

static int chkflevel(const char *entry, int field)
{
    char *pt;

    if ((pt = cli_strtok(entry, field, ":"))) { /* min version */
        if (!isdigit(*pt)) {
            free(pt);
            return 0;
        }

        if ((unsigned int)atoi(pt) > CL_FLEVEL_DCONF) {
            free(pt);
            return 0;
        }

        free(pt);

        if ((pt = cli_strtok(entry, field + 1, ":"))) { /* max version */
            if (!isdigit(*pt)) {
                free(pt);
                return 0;
            }

            if ((unsigned int)atoi(pt) < CL_FLEVEL_DCONF) {
                free(pt);
                return 0;
            }

            free(pt);
        }
    }

    return 1;
}

int cli_dconf_load(FILE *fs, struct cl_engine *engine, unsigned int options, struct cli_dbio *dbio)
{
    char buffer[FILEBUFF];
    unsigned int line = 0;
    int ret           = 0;
    uint32_t val;

    UNUSEDPARAM(options);

    while (cli_dbgets(buffer, FILEBUFF, fs, dbio)) {
        line++;
        cli_chomp(buffer);

        if (!strncmp(buffer, "PE:", 3) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 3, "0x%x", &val) == 1) {
                engine->dconf->pe = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "ELF:", 4) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 4, "0x%x", &val) == 1) {
                engine->dconf->elf = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "MACHO:", 6) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 6, "0x%x", &val) == 1) {
                engine->dconf->macho = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "ARCHIVE:", 8) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 8, "0x%x", &val) == 1) {
                engine->dconf->archive = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "DOCUMENT:", 9) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 9, "0x%x", &val) == 1) {
                engine->dconf->doc = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "MAIL:", 5) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 5, "0x%x", &val) == 1) {
                engine->dconf->mail = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "OTHER:", 6) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 6, "0x%x", &val) == 1) {
                engine->dconf->other = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "PHISHING:", 9) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 9, "0x%x", &val) == 1) {
                engine->dconf->phishing = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "BYTECODE:", 9) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 9, "0x%x", &val) == 1) {
                engine->dconf->bytecode = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "STATS:", 6) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 6, "0x%x", &val) == 1) {
                engine->dconf->stats = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }

        if (!strncmp(buffer, "PCRE:", 5) && chkflevel(buffer, 2)) {
            if (sscanf(buffer + 5, "0x%x", &val) == 1) {
                engine->dconf->pcre = val;
            } else {
                ret = CL_EMALFDB;
                break;
            }
        }
    }

    if (ret) {
        cli_errmsg("Problem parsing configuration file at line %u\n", line);
        return ret;
    }

    return CL_SUCCESS;
}