Add scripts to generate a dictionary and seed corpus for the config f…

…uzzing
osquery · Oct 21, 2019 · 3807675 · 3807675
1 parent df86635
commit 3807675
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 0 deletions.
diff --git a/osquery/main/harnesses/gen_fuzz_config_dict.sh b/osquery/main/harnesses/gen_fuzz_config_dict.sh
@@ -0,0 +1,27 @@
+#!/bin/bash
+
+# If this is not commented out, we get 'sed' commands in our output below
+#set -e
+
+SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+
+function usage() {
+    echo "${BASH_SOURCE[0]} destination-file"
+}
+
+function main() {
+    if [[ $# < 1 ]]; then
+	usage
+	exit 1
+    fi
+
+    # dict format is keyword="value" or just "value" - so we need to make sure our output is quoted.
+
+    egrep -h -R -o "HasMember\\(\"([^\"]+)\"\\)" *  | sed 's/HasMember(//' | sed 's/)//' > tmp
+    egrep -h -o -e "\"([^\"]+)\"" $SCRIPT_DIR/../../../tools/tests/*.conf >> tmp
+
+    sort tmp | uniq > $1
+    rm tmp
+}
+
+main $@
diff --git a/osquery/main/harnesses/gen_fuzz_config_seed_corpus.sh b/osquery/main/harnesses/gen_fuzz_config_seed_corpus.sh
@@ -0,0 +1,20 @@
+#!/bin/bash
+
+set -e
+
+SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+
+function usage() {
+    echo "${BASH_SOURCE[0]} destination-file"
+}
+
+function main() {
+    if [[ $# < 1 ]]; then
+	usage
+	exit 1
+    fi
+
+    zip -j $1 $SCRIPT_DIR/../../../tools/tests/*.conf
+}
+
+main $@