Adds using bodies in chunks example

examples/using_bodies_in_chunks/simple_request.cc

@@ -0,0 +1,243 @@
+/*
+ * 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 <stdio.h>
+#include <string.h>
+#include <string>
+#include <memory>
+
+#include <unistd.h>
+
+#include <modsecurity/modsecurity.h>
+#include <modsecurity/rules.h>
+#include <modsecurity/rule_message.h>
+
+
+
+char request_uri[] = "/test.pl?param1=test&para2=test2";
+
+char request_body_first[] = "" \
+    "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n\r" \
+    "<soap:Envelope xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" " \
+    "xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\" ";
+char request_body_second[] = "" \
+    "xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\">\n\r" \
+    "  <soap:Body>\n\r" \
+    "  <EnlightenResponse xmlns=\"http://clearforest.com/\">\n\r" \
+    "  <EnlightenResult>string</EnlightenResult>\n\r";
+char request_body_third[] = "" \
+    "  </EnlightenResponse>\n\r" \
+    "  </soap:Body>\n\r" \
+    "</soap:Envelope>\n\r";
+
+
+char response_body_first[] = "" \
+    "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n\r" \
+    "<soap:Envelope xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" " \
+    "xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\" ";
+char response_body_second[] = "" \
+    "xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\">\n\r" \
+    "  <soap:Body>\n\r" \
+    "  <EnlightenResponse xmlns=\"http://clearforest.com/\">\n\r" \
+    "  <EnlightenResult>string</EnlightenResult>\n\r";
+char response_body_third[] = "" \
+    "  </EnlightenResponse>\n\r" \
+    "  </soap:Body>\n\r" \
+    "</soap:Envelope>\n\r";
+
+char ip[] = "200.249.12.31";
+
+static void logCb(void *data, const void *ruleMessagev) {
+    if (ruleMessagev == NULL) {
+        std::cout << "I've got a call but the message was null ;(";
+        std::cout << std::endl;
+        return;
+    }
+
+    const modsecurity::RuleMessage *ruleMessage = \
+        reinterpret_cast<const modsecurity::RuleMessage *>(ruleMessagev);
+
+    std::cout << "Rule Id: " << std::to_string(ruleMessage->m_ruleId);
+    std::cout << " phase: " << std::to_string(ruleMessage->m_phase);
+    std::cout << std::endl;
+    if (ruleMessage->m_isDisruptive) {
+        std::cout << " * Disruptive action: ";
+        std::cout << modsecurity::RuleMessage::log(ruleMessage);
+        std::cout << std::endl;
+        std::cout << " ** %d is meant to be informed by the webserver.";
+        std::cout << std::endl;
+    } else {
+        std::cout << " * Match, but no disruptive action: ";
+        std::cout << modsecurity::RuleMessage::log(ruleMessage);
+        std::cout << std::endl;
+    }
+}
+
+
+int main(int argc, char **argv) {
+    modsecurity::ModSecurity *modsec;
+    modsecurity::Rules *rules;
+    modsecurity::ModSecurityIntervention it;
+
+    *argv++;
+    if (*argv == NULL) {
+        *argv--;
+        std::cout << "Use " << *argv << " test-case-file.conf";
+        std::cout << std::endl << std::endl;
+        return -1;
+    }
+
+    std::string rules_arg(*argv);
+
+    /**
+     * ModSecurity initial setup
+     *
+     */
+    modsec = new modsecurity::ModSecurity();
+    modsec->setConnectorInformation("ModSecurity-test v0.0.1-alpha" \
+        " (ModSecurity test)");
+    modsec->setServerLogCb(logCb, modsecurity::RuleMessageLogProperty
+        | modsecurity::IncludeFullHighlightLogProperty);
+
+    /**
+     * loading the rules....
+     *
+     */
+    rules = new modsecurity::Rules();
+    if (rules->loadFromUri(rules_arg.c_str()) < 0) {
+        std::cout << "Problems loading the rules..." << std::endl;
+        std::cout << rules->m_parserError.str() << std::endl;
+        return -1;
+    }
+
+
+    /**
+     * We are going to have a transaction
+     *
+     */
+    modsecurity::Transaction *modsecTransaction = \
+        new modsecurity::Transaction(modsec, rules, NULL);
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * Initial connection setup
+     *
+     */
+    modsecTransaction->processConnection(ip, 12345, "127.0.0.1", 80);
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * Finally we've got the URI
+     *
+     */
+    modsecTransaction->processURI(request_uri, "GET", "1.1");
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * Lets add our request headers.
+     *
+     */
+    modsecTransaction->addRequestHeader("Host",
+        "net.tutsplus.com");
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * No other reuqest header to add, let process it.
+     *
+     */
+    modsecTransaction->processRequestHeaders();
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * There is a request body to be informed...
+     *
+     */
+    modsecTransaction->appendRequestBody(
+        (const unsigned char*)request_body_first,
+        strlen((const char*)request_body_first));
+    // TODO: verify if there is any disruptive action.
+
+    modsecTransaction->appendRequestBody(
+        (const unsigned char*)request_body_second,
+        strlen((const char*)request_body_second));
+    // TODO: verify if there is any disruptive action.
+
+    modsecTransaction->appendRequestBody(
+        (const unsigned char*)request_body_third,
+        strlen((const char*)request_body_third));
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * Request body is there ;) lets process it.
+     *
+     */
+    modsecTransaction->processRequestBody();
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * The webserver is giving back the response headers.
+     */
+    modsecTransaction->addResponseHeader("HTTP/1.1",
+        "200 OK");
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * The response headers are filled in, lets process.
+     *
+     */
+    modsecTransaction->processResponseHeaders(200, "HTTP 1.2");
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * It is time to let modsec aware of the response body
+     *
+     */
+    modsecTransaction->appendResponseBody(
+        (const unsigned char*)response_body_first,
+        strlen((const char*)response_body_first));
+    // TODO: verify if there is any disruptive action.
+
+    modsecTransaction->appendResponseBody(
+        (const unsigned char*)response_body_second,
+        strlen((const char*)response_body_second));
+    // TODO: verify if there is any disruptive action.
+
+    modsecTransaction->appendResponseBody(
+        (const unsigned char*)response_body_third,
+        strlen((const char*)response_body_third));
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * Finally, lets have the response body processed.
+     *
+     */
+    modsecTransaction->processResponseBody();
+    // TODO: verify if there is any disruptive action.
+
+    /**
+     * Keeping track of everything: saving the logs.
+     *
+     */
+    modsecTransaction->processLogging();
+    // TODO: verify if there is any disruptive action.
+
+
+    /**
+     * cleanup.
+     */
+    delete modsecTransaction;
+    delete rules;
+    delete modsec;
+}