Very first commit: libmodsecurity

Check the README.md file for further information about the libmodsecurity.

test/common/modsecurity_test.cc

@@ -0,0 +1,153 @@
+/**
+ * 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 "common/modsecurity_test.h"
+
+#include <yajl/yajl_tree.h>
+#include <dirent.h>
+#include <string.h>
+
+#include <fstream>
+#include <cstdlib>
+#include <sstream>
+#include <string>
+
+#include "modsecurity/modsecurity.h"
+
+namespace modsecurity_test {
+
+template <class T>
+std::string ModSecurityTest<T>::header() {
+    std::stringstream i;
+
+    i << "ModSecurity " << MODSECURITY_VERSION << " - tests" << std::endl;
+#if not HAS_GETOPT
+    i << "(options are not available -- missing GetOpt)" << std::endl;
+#endif
+    i << std::endl;
+
+    return i.str();
+}
+
+template <class T>
+bool ModSecurityTest<T>::load_test_json(std::string file) {
+    std::vector<yajl_val> tests;
+    char errbuf[1024];
+    yajl_val node;
+
+    std::ifstream myfile;
+    myfile.open(file.c_str());
+    if (myfile.is_open() == false) {
+        std::cout << "Problems opening file: " << file << std::endl;
+        return false;
+    }
+
+    std::string str((std::istreambuf_iterator<char>(myfile)),
+        std::istreambuf_iterator<char>());
+    node = yajl_tree_parse((const char *) str.c_str(), errbuf, sizeof(errbuf));
+    if (node == NULL) {
+        std::cout << "Problems parsing file: " << file << std::endl;
+        if (strlen(errbuf) > 0) {
+            std::cout << errbuf << std::endl;
+        }
+        return false;
+    }
+
+    size_t num_tests = node->u.array.len;
+    for ( int i = 0; i < num_tests; i++ ) {
+        yajl_val obj = node->u.array.values[i];
+
+        T *u = T::from_yajl_node(obj);
+        u->filename = file;
+        u->name = u->filename + ":" + u->name;
+
+        if (this->count(u->name) == 0) {
+            std::vector<T *> *vector = new std::vector<T *>;
+            vector->push_back(u);
+            std::pair<std::string, std::vector<T *> *> a(u->name, vector);
+            this->insert(a);
+        } else {
+            std::vector<T *> *vec = this->at(u->name);
+            vec->push_back(u);
+        }
+    }
+    return true;
+}
+
+
+
+template <class T>
+std::pair<std::string, std::vector<T *>>* ModSecurityTest<T>::load_tests() {
+    DIR *dir;
+    struct dirent *ent;
+
+    if ((dir = opendir(this->target_folder.c_str())) == NULL) {
+        std::cout << "Problems opening directory: " << this->target_folder;
+        std::cout << std::endl;
+        return NULL;
+    }
+
+    while ((ent = readdir(dir)) != NULL) {
+        std::string filename = ent->d_name;
+        std::string json = ".json";
+        if (filename.size() < json.size()
+            || !std::equal(json.rbegin(), json.rend(), filename.rbegin())) {
+            continue;
+        }
+        if (load_test_json(this->target_folder + "/" + filename) == false) {
+            std::cout << "Problems loading tests from: " << filename;
+            std::cout << std::endl;
+        }
+    }
+
+    closedir(dir);
+
+    return NULL;
+}
+
+template <class T>
+void ModSecurityTest<T>::cmd_options(int argc, char **argv) {
+    int option_char;
+
+#if HAS_GETOPT
+    GetOpt getopt(argc, argv, "hvct:");
+
+    while ((option_char = getopt()) != EOF) {
+        switch (option_char) {
+            case 'h':
+                print_help();
+                return;
+                break;
+            case 'v':
+                this->verbose = true;
+                break;
+            case 'c':
+                this->color = false;
+                break;
+            case 't':
+                this->target_folder = getopt.optarg;
+            break;
+          }
+    }
+#else
+    if (argv[1]) {
+        this->target_folder = argv[1];
+    } else {
+        this->target_folder = default_test_path;
+    }
+#endif
+}
+
+}  // namespace modsecurity_test