mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-08-15 23:55:03 +03:00
115 lines
2.8 KiB
C++
115 lines
2.8 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 "parser/driver.h"
|
|
|
|
#include "parser/seclang-parser.hh"
|
|
|
|
Driver::Driver()
|
|
: trace_scanning(false),
|
|
trace_parsing(false),
|
|
requestBodyLimit(0),
|
|
responseBodyLimit(0),
|
|
audit_log(new ModSecurity::AuditLog()) {
|
|
}
|
|
|
|
|
|
Driver::~Driver() {
|
|
}
|
|
|
|
|
|
int Driver::addSecRule(ModSecurity::Rule *rule) {
|
|
if (rule->phase >= ModSecurity::ModSecurity::Phases::NUMBER_OF_PHASES) {
|
|
/** TODO: return an error message */
|
|
return -1;
|
|
}
|
|
|
|
int size = this->rules[rule->phase].size();
|
|
|
|
if (size == 0) {
|
|
this->rules[rule->phase].push_back(rule);
|
|
return true;
|
|
}
|
|
|
|
ModSecurity::Rule *lastRule = this->rules[rule->phase][size-1];
|
|
if (lastRule->chained && lastRule->chainedRule == NULL) {
|
|
lastRule->chainedRule = rule;
|
|
return true;
|
|
}
|
|
if (lastRule->chained && lastRule->chainedRule != NULL) {
|
|
ModSecurity::Rule *a = lastRule->chainedRule;
|
|
while (a->chained && a->chainedRule != NULL) {
|
|
a = a->chainedRule;
|
|
}
|
|
if (a->chained && a->chainedRule == NULL) {
|
|
a->chainedRule = rule;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
this->rules[rule->phase].push_back(rule);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
int Driver::parse(const std::string &f) {
|
|
file = f;
|
|
scan_begin();
|
|
yy::seclang_parser parser(*this);
|
|
parser.set_debug_level(trace_parsing);
|
|
// yy_scan_buffer
|
|
|
|
int res = parser.parse();
|
|
|
|
if (this->audit_log->init() == false) {
|
|
return false;
|
|
}
|
|
|
|
scan_end();
|
|
return res;
|
|
}
|
|
|
|
|
|
int Driver::parseFile(const std::string &f) {
|
|
file = f;
|
|
scan_begin();
|
|
yy::seclang_parser parser(*this);
|
|
parser.set_debug_level(trace_parsing);
|
|
int res = parser.parse();
|
|
|
|
// std::cout << "Leaving the parser: " << res << std::endl;
|
|
|
|
scan_end();
|
|
return res;
|
|
}
|
|
|
|
|
|
void Driver::error(const yy::location& l, const std::string& m,
|
|
const std::string& c) {
|
|
if (parserError.tellp() == 0) {
|
|
parserError << "Parser error, ";
|
|
parserError << "Filename: " << file << ". ";
|
|
parserError << "Line: " << l.end.line << ". ";
|
|
parserError << "Column: " << l.end.column << ". ";
|
|
}
|
|
parserError << c;
|
|
}
|
|
|
|
|
|
void Driver::parser_error(const yy::location& l, const std::string& m) {
|
|
parserError << ". " << m << "." << std::endl;
|
|
}
|