From a1955d09e3664ccfc4ab859ce1d963309abebe3b Mon Sep 17 00:00:00 2001
From: ivanr <ivanr@9017d574-64ec-4062-9424-5e00b32a252b>
Date: Mon, 24 Sep 2007 23:59:42 +0000
Subject: [PATCH] Add crude performance measurement.

---
 apache2/Makefile |  1 +
 apache2/re.c     | 74 ++++++++++++++++++++++++++++++++++++++++++++++++
 apache2/re.h     |  3 ++
 3 files changed, 78 insertions(+)

diff --git a/apache2/Makefile b/apache2/Makefile
index 07f5e7bc..7cd4c577 100644
--- a/apache2/Makefile
+++ b/apache2/Makefile
@@ -29,6 +29,7 @@ APACHECTL = apachectl
 
 INCLUDES = -I /usr/include/libxml2
 DEFS = -DWITH_LIBXML2
+#DEFS = -DWITH_LIBXML2 -DPERFORMANCE_MEASUREMENT
 #DEFS = -DWITH_LIBXML2 -DNO_MODSEC_API
 #DEFS = -DWITH_LIBXML2 -DDEBUG_CONF
 #DEFS = -DWITH_LIBXML2 -DCACHE_DEBUG
diff --git a/apache2/re.c b/apache2/re.c
index 8424c34b..421df5cc 100644
--- a/apache2/re.c
+++ b/apache2/re.c
@@ -614,12 +614,18 @@ void msre_engine_destroy(msre_engine *engine) {
 #define NEXT_RULE   2
 #define SKIP_RULES  3
 
+
+
 /**
  * Default implementation of the ruleset phase processing; it processes
  * the rules in the ruleset attached to the currently active
  * transaction phase.
  */
+#if defined(PERFORMANCE_MEASUREMENT)
+static apr_status_t msre_ruleset_process_phase_(msre_ruleset *ruleset, modsec_rec *msr) {
+#else
 apr_status_t msre_ruleset_process_phase(msre_ruleset *ruleset, modsec_rec *msr) {
+#endif
     apr_array_header_t *arr = NULL;
     msre_rule **rules;
     apr_status_t rc;
@@ -657,6 +663,7 @@ apr_status_t msre_ruleset_process_phase(msre_ruleset *ruleset, modsec_rec *msr)
     rules = (msre_rule **)arr->elts;
     for (i = 0; i < arr->nelts; i++) {
         msre_rule *rule = rules[i];
+        apr_time_t time1 = 0;
 
         /* NEXT_CHAIN is used when one of the rules in a chain
          * fails to match and then we need to skip the remaining
@@ -712,8 +719,16 @@ apr_status_t msre_ruleset_process_phase(msre_ruleset *ruleset, modsec_rec *msr)
                     rule, (fn ? fn : ""), (id ? id : ""), (rev ? rev : ""));
         }
 
+#if defined(PERFORMANCE_MEASUREMENT)
+        time1 = apr_time_now();
+#endif
+
         rc = msre_rule_process(rule, msr);
 
+#if defined(PERFORMANCE_MEASUREMENT)
+        rule->execution_time += (apr_time_now() - time1);
+#endif
+
         if (msr->txcfg->debuglog_level >= 4) {
             msr_log(msr, 4, "Rule returned %i.", rc);
         }
@@ -796,6 +811,63 @@ apr_status_t msre_ruleset_process_phase(msre_ruleset *ruleset, modsec_rec *msr)
     return 0;
 }
 
+#if defined(PERFORMANCE_MEASUREMENT)
+apr_status_t msre_ruleset_process_phase(msre_ruleset *ruleset, modsec_rec *msr) {
+    apr_array_header_t *arr = NULL;
+    msre_rule **rules = NULL;
+    apr_status_t rc;
+    apr_time_t time1;
+    int i;
+
+    switch (msr->phase) {
+        case PHASE_REQUEST_HEADERS :
+            arr = ruleset->phase_request_headers;
+            break;
+        case PHASE_REQUEST_BODY :
+            arr = ruleset->phase_request_body;
+            break;
+        case PHASE_RESPONSE_HEADERS :
+            arr = ruleset->phase_response_headers;
+            break;
+        case PHASE_RESPONSE_BODY :
+            arr = ruleset->phase_response_body;
+            break;
+        case PHASE_LOGGING :
+            arr = ruleset->phase_logging;
+            break;
+        default :
+            msr_log(msr, 1, "Internal Error: Invalid phase %d", msr->phase);
+            return -1;
+    }
+
+    rules = (msre_rule **)arr->elts;
+    for (i = 0; i < arr->nelts; i++) {
+        msre_rule *rule = rules[i];
+        rule->execution_time = 0;
+    }    
+
+    time1 = apr_time_now();
+
+    for (i = 0; i < 10000; i++) {
+        rc = msre_ruleset_process_phase_(ruleset, msr);
+    }
+
+    msr_log(msr, 1, "Phase %i: %" APR_TIME_T_FMT " usec", msr->phase, ((apr_time_now() - time1) / 10000));
+
+    rules = (msre_rule **)arr->elts;
+    for (i = 0; i < arr->nelts; i++) {
+        msre_rule *rule = rules[i];
+        msr_log(msr, 1, "Rule %x [id \"%s\"][file \"%s\"][line \"%d\"]: %lu usec", rule,
+            ((rule->actionset != NULL)&&(rule->actionset->id != NULL)) ? rule->actionset->id : "-",
+            rule->filename != NULL ? rule->filename : "-",
+            rule->line_num,
+            (rule->execution_time / 10000));
+    }
+    
+    return rc;
+}
+#endif
+
 /**
  * Creates a ruleset that will be handled by the default
  * implementation.
@@ -1241,7 +1313,9 @@ static int execute_operator(msre_var *var, msre_rule *rule, modsec_rec *msr,
     if (msr->txcfg->debuglog_level >= 4) {
         time_before_regex = apr_time_now();
     }
+
     rc = rule->op_metadata->execute(msr, rule, var, &my_error_msg);
+
     if (msr->txcfg->debuglog_level >= 4) {
         msr_log(msr, 4, "Operator completed in %" APR_TIME_T_FMT " usec.",
             (apr_time_now() - time_before_regex));
diff --git a/apache2/re.h b/apache2/re.h
index f214f117..ca05dbbe 100644
--- a/apache2/re.h
+++ b/apache2/re.h
@@ -123,6 +123,9 @@ struct msre_rule {
     
     msre_ruleset            *ruleset;
     msre_rule               *chain_starter;
+#if defined(PERFORMANCE_MEASUREMENT)
+    unsigned int             execution_time;
+#endif
 };
 
 msre_rule DSOLOCAL *msre_rule_create(msre_ruleset *ruleset,