/**
 * ModSecurity, http://www.modsecurity.org/
 * Copyright (c) 2015 Trustwave Holdings, Inc. (http://www.trustwave.com/)
 *
 * You may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * If any of the files related to licensing are missing or if you have any
 * other questions related to licensing please contact Trustwave Holdings, Inc.
 * directly using the email address security@modsecurity.org.
 *
 */

#include "src/rule.h"

#include <stdio.h>

#include <iostream>
#include <string>
#include <cstring>

#include "operators/operator.h"
#include "actions/action.h"
#include "../headers/modsecurity/modsecurity.h"


namespace ModSecurity {

using operators::Operator;
using actions::Action;

Rule::Rule(Operator *_op,
        std::vector<Variable> *_variables,
        std::vector<Action *> *_actions)
    : variables(_variables),
    op(_op),
    rule_id(0),
    phase(-1) {
    for (Action *a : *_actions) {
        if (a->action_kind == Action::ConfigurationKind) {
            actions_conf.push_back(a);
            a->evaluate(this);
        }

        if (a->action_kind == Action::RunTimeBeforeMatchAttemptKind) {
            actions_runtime_pre.push_back(a);
        }

        if (a->action_kind == Action::RunTimeOnlyIfMatchKind) {
            actions_runtime_pos.push_back(a);
        }
    }
    /**
     * If phase is not entered, we assume phase 2. For historical reasons.
     *
     */
    if (phase == -1) {
        phase = ModSecurity::Phases::RequestHeadersPhase;
    }
}

bool Rule::evaluate(Assay *assay) {
    std::vector<Variable> *variables = this->variables;

    assay->debug(4, "Executing operator \"" + this->op->op \
        + "\" with param \"" + this->op->param +  "\" against " \
        + Variable::to_s(variables) + ".");

    for (int i = 0; i < variables->size(); i++) {
        Variable variable = variables->at(i);

        if (std::strchr(variable.name.c_str(), ':') == NULL) {
            for (auto& x : assay->m_variables_strings) {
                if ((x.first.substr(0, variable.name.size() + 1).compare( \
                    variable.name + ":") != 0) && (x.first != variable.name)) {
                    continue;
                }

                int transformations = 0;
                std::string value = x.second;

                for (Action *a :
                    this->actions_runtime_pre) {
                    value = a->evaluate(value, assay);
                    assay->debug(9, " T (" + \
                            std::to_string(transformations) + ") " + \
                            a->name + ": \"" + value +"\"");

                    transformations++;
                }

                assay->debug(9, "Target value: \"" + value + "\"");
                clock_t begin = clock();

                bool ret = this->op->evaluate(assay, value);

                clock_t end = clock();
                double elapsed_secs = static_cast<double>(end - begin) \
                                      / CLOCKS_PER_SEC;

                assay->debug(4, "Operator completed in " + \
                        std::to_string(elapsed_secs) + " seconds");

                if (ret) {
                    assay->debug(4, "Rule returned 1.");

                    for (Action *a :
                        this->actions_runtime_pos) {
                        assay->debug(4, "Running action: " + a->action);
                        a->evaluate(assay);
                    }

                    return ret;
                } else {
                    assay->debug(4, "Rule returned 0.");
                }
            }
        } else {
            bool ret = false;
            try {
                ret = this->op->evaluate(assay,
                  assay->m_variables_strings.at(variable.name));
            } catch (...) {
            }

            return ret;
        }

        return false;
    }
}

}  // namespace ModSecurity