Adds initial support to initcol action

This commit is contained in:
Felipe Zimmerle
2015-12-15 17:53:27 -03:00
parent fb3696ac04
commit 913e22a77d
6 changed files with 147 additions and 2 deletions

65
src/actions/init_col.cc Normal file
View File

@@ -0,0 +1,65 @@
/*
* 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 "actions/init_col.h"
#include <iostream>
#include <string>
#include "actions/action.h"
#include "modsecurity/assay.h"
#include "src/utils.h"
#include "modsecurity/rule.h"
#include "src/macro_expansion.h"
namespace modsecurity {
namespace actions {
InitCol::InitCol(std::string action)
: Action(action, RunTimeOnlyIfMatchKind) {
}
bool InitCol::init(std::string *error) {
int posEquals = action.find("=");
int posInit = strlen("initcol:");
std::cout << "Init collection!!!!!!!!!!!!!" << std::endl;
if (action.size() < 8) {
// return false;
}
if (posEquals == std::string::npos) {
// return false;
}
m_collection_key = std::string(action, posInit, posEquals - posInit);
m_collection_value = std::string(action, posEquals + 1);
}
bool InitCol::evaluate(Rule *rule, Assay *assay) {
std::string collectionName;
collectionName = MacroExpansion::expand(m_collection_value, assay);
std::cout << std::endl;
std::cout << "Evaluate collection!!!!!!!!!!!!!" << std::endl;
std::cout << "Collection key: " << m_collection_key << std::endl;
std::cout << "Collection value: " << m_collection_value << std::endl;
std::cout << "Collection value (expanded): " << collectionName << std::endl;
return true;
}
} // namespace actions
} // namespace modsecurity

45
src/actions/init_col.h Normal file
View File

@@ -0,0 +1,45 @@
/*
* 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 <string>
#include "actions/action.h"
#ifndef SRC_ACTIONS_INIT_COL_H_
#define SRC_ACTIONS_INIT_COL_H_
class Assay;
namespace modsecurity {
class Assay;
namespace actions {
class InitCol : public Action {
public:
explicit InitCol(std::string action);
bool evaluate(Rule *rule, Assay *assay) override;
bool init(std::string *error) override;
private:
std::string m_collection_key;
std::string m_collection_value;
};
} // namespace actions
} // namespace modsecurity
#endif // SRC_ACTIONS_INIT_COL_H_