mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-08-13 21:36:00 +03:00
120 lines
3.1 KiB
C++
120 lines
3.1 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.
|
|
*
|
|
*/
|
|
|
|
|
|
#ifdef __cplusplus
|
|
#include <algorithm>
|
|
#include <iostream>
|
|
#include <list>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
#endif
|
|
|
|
#include "modsecurity/collection/collection.h"
|
|
#include "modsecurity/collection/variable.h"
|
|
|
|
|
|
#ifndef HEADERS_MODSECURITY_TRANSACTION_GLOBAL_COLLECTION_H_
|
|
#define HEADERS_MODSECURITY_TRANSACTION_GLOBAL_COLLECTION_H_
|
|
|
|
#ifndef __cplusplus
|
|
typedef struct GlobalCollection_t GlobalCollection;
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
namespace modsecurity {
|
|
namespace collection {
|
|
|
|
class CollectionKey {
|
|
public:
|
|
CollectionKey()
|
|
: m_compartiment(""),
|
|
m_name("") { }
|
|
explicit CollectionKey(std::string name)
|
|
: m_compartiment(""),
|
|
m_name(name) { }
|
|
CollectionKey(std::string name, std::string compartiment)
|
|
: m_compartiment(compartiment),
|
|
m_name(name) { }
|
|
|
|
std::string m_name;
|
|
std::string m_compartiment;
|
|
};
|
|
|
|
|
|
class collection_hash {
|
|
public:
|
|
size_t operator()(const CollectionKey *v) const {
|
|
size_t h = 0;
|
|
std::for_each(v->m_name.begin(), v->m_name.end(), [&](char c) {
|
|
h += tolower(c);
|
|
});
|
|
std::for_each(v->m_compartiment.begin(),
|
|
v->m_compartiment.end(), [&](char c) {
|
|
h += tolower(c);
|
|
});
|
|
|
|
return h;
|
|
}
|
|
};
|
|
|
|
|
|
class collection_equal {
|
|
public:
|
|
bool operator()(const CollectionKey *u, const CollectionKey *v) const {
|
|
return u->m_name == v->m_name
|
|
&& u->m_compartiment == v->m_compartiment;
|
|
}
|
|
};
|
|
|
|
|
|
class GlobalCollection :
|
|
public std::unordered_multimap<CollectionKey *, std::string,
|
|
collection_hash, collection_equal> {
|
|
public:
|
|
GlobalCollection();
|
|
~GlobalCollection();
|
|
void store(std::string key, std::string compartment, std::string value);
|
|
|
|
bool storeOrUpdateFirst(const std::string &key, std::string compartment,
|
|
const std::string &value);
|
|
|
|
bool updateFirst(const std::string &key, std::string compartment,
|
|
const std::string &value);
|
|
|
|
void del(const std::string& key, std::string compartment);
|
|
|
|
std::string* resolveFirst(const std::string& var, std::string compartment);
|
|
void resolveSingleMatch(const std::string& var, std::string compartment,
|
|
std::vector<const Variable *> *l);
|
|
void resolveMultiMatches(const std::string& var, std::string compartment,
|
|
std::vector<const Variable *> *l);
|
|
|
|
void resolveRegularExpression(const std::string& var,
|
|
std::string compartment,
|
|
std::vector<const Variable *> *l);
|
|
};
|
|
|
|
} // namespace collection
|
|
} // namespace modsecurity
|
|
|
|
#endif
|
|
|
|
|
|
#endif // HEADERS_MODSECURITY_TRANSACTION_GLOBAL_COLLECTION_H_
|
|
|
|
|