mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-08-15 23:55:03 +03:00
123 lines
2.8 KiB
C++
123 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 "modsecurity/transaction/variables.h"
|
|
|
|
#ifdef __cplusplus
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <unordered_map>
|
|
#include <list>
|
|
#endif
|
|
|
|
#include "modsecurity/transaction/variable.h"
|
|
#include "src/utils.h"
|
|
|
|
namespace ModSecurity {
|
|
namespace transaction {
|
|
|
|
|
|
Variables::Variables() {
|
|
this->reserve(1000);
|
|
}
|
|
|
|
|
|
void Variables::store(std::string key, std::string value) {
|
|
this->emplace(key, value);
|
|
}
|
|
|
|
|
|
bool Variables::storeOrUpdateFirst(const std::string &key,
|
|
const std::string &value) {
|
|
if (updateFirst(key, value) == false) {
|
|
store(key, value);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
bool Variables::updateFirst(const std::string &key, const std::string &value) {
|
|
auto range = this->equal_range(key);
|
|
|
|
for (auto it = range.first; it != range.second; ++it) {
|
|
it->second = value;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
void Variables::del(const std::string& key) {
|
|
this->erase(key);
|
|
}
|
|
|
|
|
|
void Variables::resolveSingleMatch(const std::string& var,
|
|
std::vector<const transaction::Variable *> *l) {
|
|
auto range = this->equal_range(var);
|
|
|
|
for (auto it = range.first; it != range.second; ++it) {
|
|
l->push_back(new transaction::Variable(var, it->second));
|
|
}
|
|
}
|
|
|
|
|
|
void Variables::resolveMultiMatches(const std::string& var,
|
|
std::vector<const transaction::Variable *> *l) {
|
|
size_t keySize = var.size();
|
|
l->reserve(15);
|
|
|
|
auto range = this->equal_range(var);
|
|
|
|
for (auto it = range.first; it != range.second; ++it) {
|
|
l->insert(l->begin(), new transaction::Variable(var, it->second));
|
|
}
|
|
|
|
for (const auto& x : *this) {
|
|
if (x.first.size() <= keySize + 1) {
|
|
continue;
|
|
}
|
|
if (x.first.at(keySize) != ':') {
|
|
continue;
|
|
}
|
|
if (x.first.compare(0, keySize, var) != 0) {
|
|
continue;
|
|
}
|
|
l->insert(l->begin(), new transaction::Variable(x.first, x.second));
|
|
}
|
|
}
|
|
|
|
|
|
void Variables::resolveRegularExpression(const std::string& var,
|
|
std::vector<const transaction::Variable *> *l) {
|
|
/* Not ready */
|
|
}
|
|
|
|
|
|
std::string* Variables::resolveFirst(const std::string& var) {
|
|
auto range = equal_range(var);
|
|
|
|
for (auto it = range.first; it != range.second; ++it) {
|
|
return &it->second;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
|
|
} // namespace transaction
|
|
} // namespace ModSecurity
|