|
|
|
@@ -13,14 +13,19 @@
|
|
|
|
|
*
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#ifndef EXAMPLES_READING_LOGS_VIA_RULE_MESSAGE_READING_LOGS_VIA_RULE_MESSAGE_H_
|
|
|
|
|
#define EXAMPLES_READING_LOGS_VIA_RULE_MESSAGE_READING_LOGS_VIA_RULE_MESSAGE_H_
|
|
|
|
|
|
|
|
|
|
#include <string>
|
|
|
|
|
#include <memory>
|
|
|
|
|
#include <thread>
|
|
|
|
|
#include <array>
|
|
|
|
|
#include <chrono>
|
|
|
|
|
#include <pthread.h>
|
|
|
|
|
|
|
|
|
|
#include "modsecurity/rule_message.h"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define NUM_THREADS 100
|
|
|
|
|
constexpr auto NUM_THREADS = 100;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
char request_header[] = "" \
|
|
|
|
@@ -62,40 +67,21 @@ char response_body[] = "" \
|
|
|
|
|
|
|
|
|
|
char ip[] = "200.249.12.31";
|
|
|
|
|
|
|
|
|
|
#include "modsecurity/rule_message.h"
|
|
|
|
|
|
|
|
|
|
#ifndef EXAMPLES_READING_LOGS_VIA_RULE_MESSAGE_READING_LOGS_VIA_RULE_MESSAGE_H_
|
|
|
|
|
#define EXAMPLES_READING_LOGS_VIA_RULE_MESSAGE_READING_LOGS_VIA_RULE_MESSAGE_H_
|
|
|
|
|
static void process_request(modsecurity::ModSecurity *modsec, modsecurity::RulesSet *rules) {
|
|
|
|
|
for (auto z = 0; z < 10000; z++) {
|
|
|
|
|
auto modsecTransaction = std::make_unique<modsecurity::Transaction>(modsec, rules, nullptr);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
struct data_ms {
|
|
|
|
|
modsecurity::ModSecurity *modsec;
|
|
|
|
|
modsecurity::RulesSet *rules;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
#if defined _MSC_VER
|
|
|
|
|
#pragma warning(push)
|
|
|
|
|
#pragma warning(disable:4716) // avoid error C4716: 'process_request': must return a value, as MSVC C++ compiler doesn't support [[noreturn]]
|
|
|
|
|
#pragma warning(disable:4715) // avoid warning c4715: 'process_request' : not all control paths return a value
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
[[noreturn]] static void *process_request(void *data) {
|
|
|
|
|
struct data_ms *a = (struct data_ms *)data;
|
|
|
|
|
modsecurity::ModSecurity *modsec = a->modsec;
|
|
|
|
|
modsecurity::RulesSet *rules = a->rules;
|
|
|
|
|
int z = 0;
|
|
|
|
|
|
|
|
|
|
for (z = 0; z < 10000; z++) {
|
|
|
|
|
modsecurity::Transaction *modsecTransaction = \
|
|
|
|
|
new modsecurity::Transaction(modsec, rules, NULL);
|
|
|
|
|
modsecTransaction->processConnection(ip, 12345, "127.0.0.1", 80);
|
|
|
|
|
modsecTransaction->processURI(request_uri, "GET", "1.1");
|
|
|
|
|
|
|
|
|
|
std::this_thread::sleep_for(std::chrono::microseconds(10));
|
|
|
|
|
|
|
|
|
|
modsecTransaction->addRequestHeader("Host",
|
|
|
|
|
"net.tutsplus.com");
|
|
|
|
|
modsecTransaction->processRequestHeaders();
|
|
|
|
|
modsecTransaction->processRequestBody();
|
|
|
|
|
|
|
|
|
|
modsecTransaction->addResponseHeader("HTTP/1.1",
|
|
|
|
|
"200 OK");
|
|
|
|
|
modsecTransaction->processResponseHeaders(200, "HTTP 1.2");
|
|
|
|
@@ -103,18 +89,11 @@ struct data_ms {
|
|
|
|
|
(const unsigned char*)response_body,
|
|
|
|
|
strlen((const char*)response_body));
|
|
|
|
|
modsecTransaction->processResponseBody();
|
|
|
|
|
|
|
|
|
|
modsecTransaction->processLogging();
|
|
|
|
|
|
|
|
|
|
delete modsecTransaction;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pthread_exit(nullptr);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#if defined _MSC_VER
|
|
|
|
|
#pragma warning(pop)
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
class ReadingLogsViaRuleMessage {
|
|
|
|
|
public:
|
|
|
|
|
ReadingLogsViaRuleMessage(char *request_header,
|
|
|
|
@@ -134,11 +113,6 @@ class ReadingLogsViaRuleMessage {
|
|
|
|
|
{ }
|
|
|
|
|
|
|
|
|
|
int process() const {
|
|
|
|
|
pthread_t threads[NUM_THREADS];
|
|
|
|
|
int i;
|
|
|
|
|
struct data_ms dms;
|
|
|
|
|
void *status;
|
|
|
|
|
|
|
|
|
|
auto modsec = std::make_unique<modsecurity::ModSecurity>();
|
|
|
|
|
modsec->setConnectorInformation("ModSecurity-test v0.0.1-alpha" \
|
|
|
|
|
" (ModSecurity test)");
|
|
|
|
@@ -152,18 +126,19 @@ class ReadingLogsViaRuleMessage {
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
dms.modsec = modsec.get();
|
|
|
|
|
dms.rules = rules.get();
|
|
|
|
|
std::array<std::thread, NUM_THREADS> threads;
|
|
|
|
|
|
|
|
|
|
for (i = 0; i < NUM_THREADS; i++) {
|
|
|
|
|
pthread_create(&threads[i], NULL, process_request,
|
|
|
|
|
reinterpret_cast<void *>(&dms));
|
|
|
|
|
for (auto i = 0; i != threads.size(); ++i) {
|
|
|
|
|
threads[i] = std::thread(
|
|
|
|
|
[&modsec, &rules]() {
|
|
|
|
|
process_request(modsec.get(), rules.get());
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::this_thread::sleep_for(std::chrono::microseconds(10000));
|
|
|
|
|
|
|
|
|
|
for (i=0; i < NUM_THREADS; i++) {
|
|
|
|
|
pthread_join(threads[i], &status);
|
|
|
|
|
for (auto i = 0; i != threads.size(); ++i) {
|
|
|
|
|
threads[i].join();
|
|
|
|
|
std::cout << "Main: completed thread id :" << i << std::endl;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|