mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-08-13 13:26:01 +03:00
Adding multithreaded example from issue #3054 (by airween)
- Rewritten to use C++ libModSecurity API and std::thread (instead of pthreads)
This commit is contained in:
parent
293cd214c7
commit
4bf9616f9e
1
.gitignore
vendored
1
.gitignore
vendored
@ -50,6 +50,7 @@ ltmain.sh
|
|||||||
examples/simple_example_using_c/test
|
examples/simple_example_using_c/test
|
||||||
/tools/rules-check/modsec-rules-check
|
/tools/rules-check/modsec-rules-check
|
||||||
examples/multiprocess_c/multi
|
examples/multiprocess_c/multi
|
||||||
|
examples/multithread/multithread
|
||||||
examples/reading_logs_via_rule_message/simple_request
|
examples/reading_logs_via_rule_message/simple_request
|
||||||
examples/reading_logs_with_offset/read
|
examples/reading_logs_with_offset/read
|
||||||
examples/using_bodies_in_chunks/simple_request
|
examples/using_bodies_in_chunks/simple_request
|
||||||
|
@ -259,6 +259,10 @@ setExampleTargetProperties(reading_logs_via_rule_message)
|
|||||||
add_executable(reading_logs_with_offset ${BASE_DIR}/examples/reading_logs_with_offset/read.cc)
|
add_executable(reading_logs_with_offset ${BASE_DIR}/examples/reading_logs_with_offset/read.cc)
|
||||||
setExampleTargetProperties(reading_logs_with_offset)
|
setExampleTargetProperties(reading_logs_with_offset)
|
||||||
|
|
||||||
|
# multithread
|
||||||
|
add_executable(multithread ${BASE_DIR}/examples/multithread/multithread.cc)
|
||||||
|
setExampleTargetProperties(multithread)
|
||||||
|
|
||||||
# tools
|
# tools
|
||||||
#
|
#
|
||||||
|
|
||||||
|
@ -51,6 +51,7 @@ Built files will be located in the directory: `build\win32\build\[build_configur
|
|||||||
* `using_bodies_in_chunks.exe`
|
* `using_bodies_in_chunks.exe`
|
||||||
* `reading_logs_via_rule_message.exe`
|
* `reading_logs_via_rule_message.exe`
|
||||||
* `reading_logs_with_offset.exe`
|
* `reading_logs_with_offset.exe`
|
||||||
|
* `multithread.exe`
|
||||||
* Executable files for tools
|
* Executable files for tools
|
||||||
* `rules_check.exe`
|
* `rules_check.exe`
|
||||||
|
|
||||||
|
@ -423,6 +423,7 @@ AM_COND_IF([EXAMPLES],
|
|||||||
examples/Makefile \
|
examples/Makefile \
|
||||||
examples/simple_example_using_c/Makefile \
|
examples/simple_example_using_c/Makefile \
|
||||||
examples/multiprocess_c/Makefile \
|
examples/multiprocess_c/Makefile \
|
||||||
|
examples/multithread/Makefile \
|
||||||
examples/reading_logs_with_offset/Makefile \
|
examples/reading_logs_with_offset/Makefile \
|
||||||
examples/reading_logs_via_rule_message/Makefile \
|
examples/reading_logs_via_rule_message/Makefile \
|
||||||
examples/using_bodies_in_chunks/Makefile \
|
examples/using_bodies_in_chunks/Makefile \
|
||||||
|
@ -4,6 +4,7 @@ ACLOCAL_AMFLAGS = -I build
|
|||||||
|
|
||||||
SUBDIRS = \
|
SUBDIRS = \
|
||||||
multiprocess_c \
|
multiprocess_c \
|
||||||
|
multithread \
|
||||||
reading_logs_with_offset \
|
reading_logs_with_offset \
|
||||||
reading_logs_via_rule_message \
|
reading_logs_via_rule_message \
|
||||||
simple_example_using_c \
|
simple_example_using_c \
|
||||||
|
54
examples/multithread/Makefile.am
Normal file
54
examples/multithread/Makefile.am
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
|
||||||
|
|
||||||
|
noinst_PROGRAMS = multithread
|
||||||
|
|
||||||
|
multithread_SOURCES = \
|
||||||
|
multithread.cc
|
||||||
|
|
||||||
|
multithread_LDADD = \
|
||||||
|
$(CURL_LDADD) \
|
||||||
|
$(GEOIP_LDADD) \
|
||||||
|
$(GLOBAL_LDADD) \
|
||||||
|
$(LIBXML2_LDADD) \
|
||||||
|
$(LMDB_LDADD) \
|
||||||
|
$(MAXMIND_LDADD) \
|
||||||
|
$(LUA_LDADD) \
|
||||||
|
$(PCRE_LDADD) \
|
||||||
|
$(SSDEEP_LDADD) \
|
||||||
|
$(YAJL_LDADD)
|
||||||
|
|
||||||
|
multithread_LDFLAGS = \
|
||||||
|
-L$(top_builddir)/src/.libs/ \
|
||||||
|
$(GEOIP_LDFLAGS) \
|
||||||
|
-lmodsecurity \
|
||||||
|
-lm \
|
||||||
|
-lstdc++ \
|
||||||
|
$(LMDB_LDFLAGS) \
|
||||||
|
$(LUA_LDFLAGS) \
|
||||||
|
$(MAXMIND_LDFLAGS) \
|
||||||
|
$(SSDEEP_LDFLAGS) \
|
||||||
|
$(YAJL_LDFLAGS)
|
||||||
|
|
||||||
|
multithread_CPPFLAGS = \
|
||||||
|
$(GLOBAL_CFLAGS) \
|
||||||
|
-I$(top_builddir)/headers \
|
||||||
|
-I$(top_builddir) \
|
||||||
|
-g \
|
||||||
|
-I../others \
|
||||||
|
-fPIC \
|
||||||
|
-O3 \
|
||||||
|
$(CURL_CFLAGS) \
|
||||||
|
$(GEOIP_CFLAGS) \
|
||||||
|
$(GLOBAL_CPPFLAGS) \
|
||||||
|
$(MODSEC_NO_LOGS) \
|
||||||
|
$(YAJL_CFLAGS) \
|
||||||
|
$(LMDB_CFLAGS) \
|
||||||
|
$(LUA_CFLAGS) \
|
||||||
|
$(PCRE_CFLAGS) \
|
||||||
|
$(LIBXML2_CFLAGS)
|
||||||
|
|
||||||
|
|
||||||
|
MAINTAINERCLEANFILES = \
|
||||||
|
Makefile.in
|
||||||
|
|
||||||
|
|
14
examples/multithread/basic_rules.conf
Normal file
14
examples/multithread/basic_rules.conf
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
SecDebugLog debug.log
|
||||||
|
SecDebugLogLevel 9
|
||||||
|
|
||||||
|
|
||||||
|
SecRule REQUEST_HEADERS:User-Agent ".*" "id:1,phase:1,t:sha1,t:hexEncode,setvar:tx.ua_hash=%{MATCHED_VAR}"
|
||||||
|
|
||||||
|
SecAction "id:2,phase:2,initcol:ip=%{REMOTE_ADDR}_%{tx.ua_hash}"
|
||||||
|
|
||||||
|
SecRule REQUEST_HEADERS:User-Agent "@rx .*" "id:3,phase:2,setvar:ip.auth_attempt=+1"
|
||||||
|
|
||||||
|
SecRule ARGS:foo "@rx herewego" "id:4,phase:2,setvar:ip.foo=bar,expirevar:ip.foo=2"
|
||||||
|
#SecRule ARGS:foo "@rx herewego" "id:4,phase:2,setvar:ip.foo=bar"
|
||||||
|
SecRule IP "@rx bar" "id:5,phase:2,pass"
|
||||||
|
SecRule IP:auth_attempt "@rx bar" "id:6,phase:2,pass"
|
68
examples/multithread/multithread.cc
Normal file
68
examples/multithread/multithread.cc
Normal file
@ -0,0 +1,68 @@
|
|||||||
|
#include <iostream>
|
||||||
|
#include <thread>
|
||||||
|
#include <array>
|
||||||
|
|
||||||
|
#include <modsecurity/modsecurity.h>
|
||||||
|
#include <modsecurity/transaction.h>
|
||||||
|
#include <modsecurity/rules_set.h>
|
||||||
|
|
||||||
|
static void process_request(modsecurity::ModSecurity *modsec, modsecurity::RulesSet *rules, int tid) {
|
||||||
|
std::cout << "Hello World! It's me, thread #" << tid << std::endl;
|
||||||
|
|
||||||
|
for(int i = 0; i != 1'000; i++) {
|
||||||
|
auto modsecTransaction = std::make_unique<modsecurity::Transaction>(modsec, rules, nullptr);
|
||||||
|
|
||||||
|
modsecTransaction->processConnection("127.0.0.1", 12345, "127.0.0.1", 80);
|
||||||
|
modsecTransaction->processURI(
|
||||||
|
"https://www.modsecurity.org/test?foo=herewego",
|
||||||
|
"GET", "1.1");
|
||||||
|
|
||||||
|
modsecTransaction->addRequestHeader("User-Agent",
|
||||||
|
"Basic ModSecurity example");
|
||||||
|
modsecTransaction->processRequestHeaders();
|
||||||
|
modsecTransaction->processRequestBody();
|
||||||
|
|
||||||
|
modsecTransaction->addResponseHeader("HTTP/1.1",
|
||||||
|
"200 OK");
|
||||||
|
modsecTransaction->processResponseHeaders(200, "HTTP 1.2");
|
||||||
|
modsecTransaction->processResponseBody();
|
||||||
|
|
||||||
|
modsecTransaction->processLogging();
|
||||||
|
|
||||||
|
std::this_thread::sleep_for(std::chrono::microseconds(100));
|
||||||
|
}
|
||||||
|
|
||||||
|
std::cout << "Thread #" << tid << " exits" << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
int main (int argc, char *argv[]) {
|
||||||
|
auto modsec = std::make_unique<modsecurity::ModSecurity>();
|
||||||
|
modsec->setConnectorInformation("ModSecurity-test v0.0.1-alpha (Simple " \
|
||||||
|
"example on how to use ModSecurity API");
|
||||||
|
|
||||||
|
char main_rule_uri[] = "basic_rules.conf";
|
||||||
|
auto rules = std::make_unique<modsecurity::RulesSet>();
|
||||||
|
if (rules->loadFromUri(main_rule_uri) < 0) {
|
||||||
|
std::cerr << "Problems loading the rules..." << std::endl;
|
||||||
|
std::cerr << rules->m_parserError.str() << std::endl;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
constexpr auto NUM_THREADS = 100;
|
||||||
|
std::array<std::thread, NUM_THREADS> threads;
|
||||||
|
|
||||||
|
for (auto i = 0; i != threads.size(); ++i) {
|
||||||
|
threads[i] = std::thread(
|
||||||
|
[&modsec, &rules, i]() {
|
||||||
|
process_request(modsec.get(), rules.get(), i);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
std::this_thread::sleep_for(std::chrono::microseconds(10000));
|
||||||
|
|
||||||
|
for (auto i = 0; i != threads.size(); ++i) {
|
||||||
|
threads[i].join();
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
Loading…
x
Reference in New Issue
Block a user