mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-08-14 05:45:59 +03:00
84 lines
2.0 KiB
C++
84 lines
2.0 KiB
C++
/*
|
|
* 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 <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <list>
|
|
#include <utility>
|
|
|
|
#ifndef SRC_VARIABLES_GEO_H_
|
|
#define SRC_VARIABLES_GEO_H_
|
|
|
|
#include "src/variables/variable.h"
|
|
|
|
namespace modsecurity {
|
|
|
|
class Transaction;
|
|
namespace Variables {
|
|
|
|
class Geo_DictElement : public Variable {
|
|
public:
|
|
explicit Geo_DictElement(std::string dictElement)
|
|
: Variable("GEO" + std::string(":") +
|
|
std::string(dictElement)),
|
|
m_dictElement(dictElement) { }
|
|
|
|
void evaluate(Transaction *transaction,
|
|
Rule *rule,
|
|
std::vector<const VariableValue *> *l) override {
|
|
transaction->m_variableGeo.resolve(m_dictElement, l);
|
|
}
|
|
|
|
std::string m_dictElement;
|
|
};
|
|
|
|
|
|
class Geo_NoDictElement : public Variable {
|
|
public:
|
|
Geo_NoDictElement()
|
|
: Variable("GEO") { }
|
|
|
|
void evaluate(Transaction *transaction,
|
|
Rule *rule,
|
|
std::vector<const VariableValue *> *l) override {
|
|
transaction->m_variableGeo.resolve(l, m_keyExclusion);
|
|
}
|
|
};
|
|
|
|
|
|
class Geo_DictElementRegexp : public Variable {
|
|
public:
|
|
explicit Geo_DictElementRegexp(std::string dictElement)
|
|
: Variable("GEO:regex(" + dictElement + ")"),
|
|
m_r(dictElement) { }
|
|
|
|
void evaluate(Transaction *transaction,
|
|
Rule *rule,
|
|
std::vector<const VariableValue *> *l) override {
|
|
transaction->m_variableGeo.resolveRegularExpression(
|
|
&m_r, l, m_keyExclusion);
|
|
}
|
|
|
|
Utils::Regex m_r;
|
|
};
|
|
|
|
|
|
} // namespace Variables
|
|
} // namespace modsecurity
|
|
|
|
#endif // SRC_VARIABLES_GEO_H_
|
|
|