From be4f53e1a8ffdb7f098451f712bb33295f93654b Mon Sep 17 00:00:00 2001
From: Unix-Ninja <chris@unix-ninja.com>
Date: Thu, 6 Mar 2014 14:06:24 -0500
Subject: [PATCH] added case-sensitive toggle switch & cleaned up garbage
 results (still super-fast)

---
 searchsploit | 99 ++++++++++++++++++++++++++++++++++------------------
 1 file changed, 65 insertions(+), 34 deletions(-)

diff --git a/searchsploit b/searchsploit
index 757606fdb..166ae4764 100755
--- a/searchsploit
+++ b/searchsploit
@@ -5,59 +5,89 @@
 
 csvpath=/usr/share/exploitdb/files.csv
 progname=`basename $0`
+TAGS=
+SCASE='-i'
 VERBOSE=0
 
 # if files.csv is in the searchsploit path, use that
 if [ -f "$( dirname $0 )/files.csv" ]; then
-	csvpath="$( dirname $0 )/files.csv"
+    csvpath="$( dirname $0 )/files.csv"
 fi
 
 # usage info
 function usage()
 {
-	echo "Usage: $progname [options] term1 [term2] ... [termN]"
-	echo "Example: $progname oracle windows local"
-	echo 
-	echo "======="
-	echo "Options"
-	echo "======="
-	echo
-	echo "  -h, --help   Show help screen" 
-	echo "  -v           By setting verbose output, description lines are allowed to"
-	echo "               overflow their columns"
-	echo
-	echo "*NOTES*"
-	echo "Use any number of search terms you would like (minimum of one)."
-	echo "Search terms are not case sensitive, and order is irrelevant."
-	exit 1
+    echo "Usage: $progname [options] term1 [term2] ... [termN]"
+    echo "Example: $progname oracle windows local"
+    echo 
+    echo "======="
+    echo "Options"
+    echo "======="
+    echo
+    echo "   -c                Perform case-sensitive searches; by default, searches will"
+    echo "                      try to be greedy"
+    echo "   -h, --help    Show help screen" 
+    echo "   -v                By setting verbose output, description lines are allowed to"
+    echo "                      overflow their columns"
+    echo
+    echo "*NOTES*"
+    echo "Use any number of search terms you would like (minimum of one)."
+    echo "Search terms are not case sensitive, and order is irrelevant."
+    exit 1
 }
-if [ $# -eq 0 -o "$1" == "-h" -o "$1" == "--help" ]; then
-	usage >&2
-fi
 
 # dynamically set column widths
 COL2=35
 COL1=$(( `tput cols` - $COL2 - 1 ))
 
-if [ "$1" == "-v" ]; then
-	VERBOSE=1
-	shift
+# check for empty args
+if [ $# -eq 0 ]; then
+    usage >&2
 fi
 
+# parse long arguments
+ARGS="-"
+for param in $@; do
+    if [ "$param" == "--help" ]; then
+        usage >&2
+    else
+        if [ "${param:0:1}" == "-" ]; then
+            ARGS=$ARGS${param:1}
+            shift
+            continue
+        fi
+        TAGS="$TAGS $param"
+    fi
+done
+
+# parse short arguments
+while getopts "chv" arg $ARGS; do
+    if [ "$arg" = "?" ]; then
+        usage >&2;
+    fi
+    case $arg in
+        c) SCASE='';;
+        h) usage >&2;;
+        v) VERBOSE=1;;
+    esac
+    shift $((OPTIND-1))
+done
+
 # print header
 printf "%-${COL1}s %s" " Description"
-echo "  Path"
+echo "   Path"
 printf "%0.s-" `eval echo {1..$(( $COL1 + 1 ))}`
 echo -n " "
 printf "%0.s-" `eval echo {1..$(( $COL2 - 1 ))}`
 echo
 
 # create search command
-SEARCH="fgrep -i \"$1\" $csvpath"
-shift
-while (( "$#" )); do
-  SEARCH="$SEARCH | fgrep -i \"$1\""
-  shift
+SEARCH=
+for tag in $TAGS; do
+    if [ "$SEARCH" ]; then
+        SEARCH="$SEARCH |"
+    fi
+    SEARCH="$SEARCH fgrep $SCASE \"$tag\""
 done
 
 # set LANG variable to avoid illegal byte sequence errors in sed
@@ -65,12 +95,13 @@ LANG=C
 
 # search, format, and print results
 if [ "$VERBOSE" -eq 0 ]; then
-	eval $SEARCH \
-	| awk -F "\"*,\"*" '{ printf "%-'$COL1'.'$COL1's | %s\n", $3, $2}' \
-	| sed "	s/| platforms/| /"
+    FORMAT=$COL1'.'$COL1
 else
-	eval $SEARCH \
-	| awk -F "\"*,\"*" '{ printf "%-'$COL1's | %s\n", $3, $2}' \
-	| sed "	s/| platforms/| /"
+    FORMAT=$COL1
 fi
+    cat $csvpath \
+    | eval $SEARCH \
+    | awk -F "\"*,\"*" '{ printf "%-'$FORMAT's | %s\n", $3, $2}' \
+    | sed "   s/| platforms/| /" \
+    | eval $SEARCH
 exit 0