mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-09-30 19:47:47 +03:00
Adds support to the multithread example
This commit is contained in:
141
examples/multithread_c/multi.c
Normal file
141
examples/multithread_c/multi.c
Normal file
@@ -0,0 +1,141 @@
|
||||
/*
|
||||
* 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/modsecurity.h>
|
||||
#include <modsecurity/transaction.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <pthread.h>
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/ipc.h>
|
||||
#include <sys/shm.h>
|
||||
#include <sys/wait.h>
|
||||
|
||||
#define SHM_SIZE 1024
|
||||
#define FORKS 10
|
||||
#define TRHREADS 10
|
||||
|
||||
|
||||
char main_rule_uri[] = "basic_rules.conf";
|
||||
Rules *rules = NULL;
|
||||
ModSecurity *modsec = NULL;
|
||||
|
||||
key_t key;
|
||||
|
||||
|
||||
void process_request (void *ptr)
|
||||
{
|
||||
int i;
|
||||
for (i = 0; i < TRHREADS; i++) {
|
||||
struct timeval tv;
|
||||
|
||||
int shmid = shmget(key, SHM_SIZE, 0644 | IPC_CREAT);
|
||||
char *data;
|
||||
int *j;
|
||||
data = shmat(shmid, (void *)0, 0);
|
||||
j = (int *)data;
|
||||
(*j)++;
|
||||
|
||||
|
||||
Transaction *transaction = NULL;
|
||||
transaction = msc_new_transaction(modsec, rules, NULL);
|
||||
|
||||
msc_process_connection(transaction, "127.0.0.1", 12345, "127.0.0.1", 80);
|
||||
msc_process_uri(transaction,
|
||||
"http://www.modsecurity.org/test?key1=value1&key2=value2&key3=value3",
|
||||
"GET", "1.1");
|
||||
msc_add_request_header(transaction, "User-Agent",
|
||||
"Basic ModSecurity example");
|
||||
msc_process_request_headers(transaction);
|
||||
msc_process_request_body(transaction);
|
||||
msc_add_response_header(transaction, "Content-type", "text/html");
|
||||
msc_process_response_headers(transaction);
|
||||
msc_process_response_body(transaction);
|
||||
msc_process_logging(transaction, 200);
|
||||
msc_transaction_cleanup(transaction);
|
||||
tv.tv_sec = 0;
|
||||
tv.tv_usec = 1000;
|
||||
select(0, NULL, NULL, NULL, &tv);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
int main (int argc, char **argv)
|
||||
{
|
||||
int ret = 1;
|
||||
const char *error = NULL;
|
||||
pthread_t thread[TRHREADS*FORKS];
|
||||
int i = 0;
|
||||
pid_t pid;
|
||||
int shmid;
|
||||
|
||||
modsec = msc_init();
|
||||
|
||||
key = ftok("shmdemo.c", 'R');
|
||||
|
||||
shmid = shmget(key, SHM_SIZE, 0644 | IPC_CREAT);
|
||||
char *data = shmat(shmid, (void *)0, 0);
|
||||
memset(data, '\0', SHM_SIZE);
|
||||
int *z = (int *) data;
|
||||
z = 0;
|
||||
|
||||
msc_set_connector_info(modsec, "ModSecurity-test v0.0.1-alpha (Simple " \
|
||||
"example on how to use ModSecurity API");
|
||||
|
||||
rules = msc_create_rules_set();
|
||||
|
||||
ret = msc_rules_add_file(rules, main_rule_uri, &error);
|
||||
if (ret < 0) {
|
||||
fprintf(stderr, "Problems loading the rules --\n");
|
||||
fprintf(stderr, "%s\n", error);
|
||||
goto end;
|
||||
}
|
||||
|
||||
msc_rules_dump(rules);
|
||||
|
||||
for (int h = 0; h < FORKS; h++) {
|
||||
pid = fork();
|
||||
if (pid == 0) {
|
||||
pthread_create (&thread[h*TRHREADS], NULL, (void *) &process_request, (void *) NULL);
|
||||
pthread_join(thread[h*TRHREADS], NULL);
|
||||
goto child;
|
||||
}
|
||||
}
|
||||
|
||||
wait(NULL);
|
||||
|
||||
int *j;
|
||||
data = shmat(shmid, (void *)0, 0);
|
||||
j = (int *) data;
|
||||
fprintf(stderr, "The final count is: %d\n", *j);
|
||||
|
||||
child:
|
||||
|
||||
if (pid == 0) {
|
||||
return 0;
|
||||
}
|
||||
end:
|
||||
msc_rules_cleanup(rules);
|
||||
msc_cleanup(modsec);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user