mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-08-14 05:45:59 +03:00
96 lines
2.5 KiB
C++
96 lines
2.5 KiB
C++
/*
|
|
* ModSecurity, http://www.modsecurity.org/
|
|
* Copyright (c) 2015 - 2021 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 "src/operators/pm_from_file.h"
|
|
|
|
#include <string>
|
|
|
|
#include "src/operators/operator.h"
|
|
#include "src/utils/https_client.h"
|
|
#include "src/utils/system.h"
|
|
#include "src/utils/string.h"
|
|
|
|
using namespace modsecurity::utils::string;
|
|
|
|
namespace modsecurity {
|
|
namespace operators {
|
|
|
|
bool PmFromFile::isComment(const std::string &s) {
|
|
if (s.size() == 0) {
|
|
return true;
|
|
}
|
|
size_t pos = s.find("#");
|
|
if (pos != std::string::npos) {
|
|
for (int i = 0; i < pos; i++) {
|
|
if (!std::isspace(s[i])) {
|
|
return false;
|
|
}
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool PmFromFile::init(const std::string &config, std::string *error) {
|
|
std::vector<std::string> tokens = split(m_param, ' ');
|
|
|
|
for (const auto& token : tokens) {
|
|
if (! token.empty()) {
|
|
|
|
std::istream *iss;
|
|
|
|
if (token.compare(0, 8, "https://") == 0) {
|
|
Utils::HttpsClient client;
|
|
bool ret = client.download(token);
|
|
if (ret == false) {
|
|
error->assign(client.error);
|
|
return false;
|
|
}
|
|
iss = new std::stringstream(client.content);
|
|
} else {
|
|
std::string err;
|
|
std::string resource = utils::find_resource(token, config, &err);
|
|
iss = new std::ifstream(resource, std::ios::in);
|
|
|
|
if (((std::ifstream *)iss)->is_open() == false) {
|
|
error->assign("Failed to open file: '" + token + "'. " + err);
|
|
delete iss;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
for (std::string line; std::getline(*iss, line); ) {
|
|
if (isComment(line) == false) {
|
|
acmp_add_pattern(m_p, line.c_str(), NULL, NULL, line.length());
|
|
}
|
|
}
|
|
|
|
delete iss;
|
|
}
|
|
}
|
|
|
|
while (m_p->is_failtree_done == 0) {
|
|
acmp_prepare(m_p);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
} // namespace operators
|
|
} // namespace modsecurity
|