fluentnmap.sh

#!/bin/bash

#The Path of where running this script
CURRENT_PATH=`pwd`

#The Path to store dataFile
DATAFILE_PATH="/var/tmp"

#Path /To modify the path by yourself
SCRIPT_PATH="/usr/share/nmap/scripts"
cd $SCRIPT_PATH

#ver
VERSION=`nmap --version |grep version`

#Counter
counter(){
	grep -r 'categories =' *.nse | grep -P ''".*?"'' |grep $1 |wc |awk '{print $1}'
}

SCRIPT_TOTAL=$(expr `ls -la |wc |awk '{ print $1 }'` - 2)
AUTH_TOTAL=`counter auth`
BROADCAST_TOTAL=`counter broadcast`
BRUTE_TOTAL=`counter brute`
DEFAULT_TOTAL=`counter default`
DISCOVER_TOTAL=`counter discovery`
DOS_TOTAL=`counter dos`
EXPLOIT_TOTAL=`counter exploit`
EXTERNAL_TOTAL=`counter external`
FUZZER_TOTAL=`counter fuzzer`
INTRUSIVE_TOTAL=`counter intrusive`
MALWARE_TOTAL=`counter malware`
SAFE_TOTAL=`counter safe`
VERSION_TOTAL=`counter version`
VULN_TOTAL=`counter vuln`

# searching in specify category
searchInCate(){
	if [ $# -eq 1 ];
        then
                grep -H -r 'categories =' *.nse | grep -P ''".*?"'' |grep $1 |nl | cut -d ':' -f 1 > $DATAFILE_PATH'/dataFile'
		cat $DATAFILE_PATH'/dataFile'
        elif [ $# -eq 2 ];
        then
                grep -H -r 'categories =' *.nse | grep -P ''".*?"'' |grep $1 |grep $2 |nl | cut -d ':' -f 1 > $DATAFILE_PATH'/dataFile'
		cat $DATAFILE_PATH'/dataFile'
        fi
}


#Script
VULNER_STATU=`if [ $(ls nmap-vulners 2>/dev/null |wc |awk '{print $1}') -eq 0 ] ; then echo "[-] nmap-vulners - is not available now, use -h to see how to install it." ; else echo "[+] nmap-vulners is available, good luck !" ; fi`
VULSCAN_STATU=`if [ $(ls vulscan 2>/dev/null |wc |awk '{print $1}') -eq 0 ] ; then echo "[-] vulscan      - is not available now, use -h to see how to install it." ; else echo "[+] vulscan is available, good luck !" ; fi`

#Services
FTP=""
SSH=""
TELNET=""
SMTP=""
DNS=""
HTTP=""
POP=""
SMB=""

#print this help message
printHelp(){
	echo "[1] Usage: fluentnmap    <Keyword>                  - List related scripts"
	echo "[2] Usage: fluentnmap    <Full Script Name>         - Show the script's description"
	echo "[3] Usage: fluentnmap -c <Keyword/Category>         - List related scripts or scripts specified by script category"
	echo "[4] Usage: fluentnmap -c <Category> <Keyword>       - List related scripts are included in the specified category"
	echo "[5] Usage: fluentnmap -d <Full Script Name>         - Display the script detail according to the script name that you specified"
	echo "[6] Usage: fluentnmap    <Index>                    - Display the script detail according to the index you had specified"
	echo "[7] Usage: sudo fluentnmap install vulnscanner      - To install the vulnscanner packages (nmap-vulners & vulscan) from Github"
	echo "[8] Usage: sudo fluentnmap remove                   - To remove the vulnscanner packages"
	echo
	echo "Example: "
	echo "   fluentnmap smb"
	echo "   fluentnmap smb-brute.nse"
	echo "   fluentnmap 3"
	echo "   fluentnmap -d smb-brute.nse"
	echo "   fluentnmap -c auth"
	echo "   fluentnmap -c auth smb"
	echo "   nmap -sV --script <nmap-vulners|vulnscan> <Target>"
	echo "   To discovery the known vulnerabilities. Have fun :)"
	echo
	exit 1
}

printInfoBoard(){
         echo ' _____ _                  _   _   _                         '
         echo '|  ___| |_   _  ___ _ __ | |_| \ | |_ __ ___   __ _ _ __    '
         echo '| |_  | | | | |/ _ \ __ \| __|  \| | __ ` _ \ / _` | __ \   '
	 echo '|  _| | | |_| |  __/ | | | |_| |\  | | | | | | (_| | |_) |  '
	 echo '|_|   |_|\__,_|\___|_| |_|\__|_| \_|_| |_| |_|\__,_| .__/   '
         echo '                                                   |_|      '

 	 echo "------------------------------------------------------------"
	 echo " A mate of nmap."
	 echo " Clean, Concise and Straightforward."
	 echo " Help you to be familiar with nmap scripts."
         echo "------------------------------------------------------------"

	 echo "$VERSION"
	 echo "	<<<<<<<<<<<<<<<<<<< Total:$SCRIPT_TOTAL >>>>>>>>>>>>>>>>>>>>"
	 echo "          auth:$AUTH_TOTAL - brute:$BRUTE_TOTAL - broadcast:$BROADCAST_TOTAL - default:$DEFAULT_TOTAL"
	 echo "          discover:$DISCOVER_TOTAL - dos:$DOS_TOTAL - exploit:$EXPLOIT_TOTAL - external:$EXTERNAL_TOTAL"
	 echo "          fuzzer:$FUZZER_TOTAL - intrusive:$INTRUSIVE_TOTAL - malware:$MALWARE_TOTAL - safe:$SAFE_TOTAL"
	 echo "                      version:$VERSION_TOTAL - vuln:$VULN_TOTAL"
	 echo "           (use option -h/--help to see the help page)"
	 echo "$VULNER_STATU"
	 echo "$VULSCAN_STATU"
	 echo
}

# Check if dataFile was exist. If existed delete it
#if [ `test -e dataFile; echo $?` -eq 0 ] ; then  rm dataFile ; fi 

if [ $# -eq 1 ] ; then

	#filter unvalid character ( only alapa number and - are valid )
	flitered="$(echo $1 |sed -e 's/[^-.[:alnum:]]//g')"

	#check if the string format is nse script name 
	match=`echo $flitered |cut -d "." -f 2`

	#filter the input is purely number
	index="$(echo $1 |sed -e 's/[^[:digit:]]//g')"

	if [ $flitered = $1 ] ; then
		#handle the input: --help or -h
		if [ "$flitered" != "" ] && [ $flitered = "--help" ] || [ $flitered = "-h" ] ; then
			printInfoBoard
			printHelp
		#handle the input: remove
		elif [ "$flitered" != "" ] && [ $flitered = 'remove' ] ; then
			echo "[-] Sure to remove the vuln scanner packages ?"
			echo "[-] Please input 'y' to remove [y/n]" 
			read choice
			if [ "$choice" != "" ] && [ $choice = 'y' ] ; then
				message=`rm -r $SCRIPT_PATH/nmap-vulners; rm -r $SCRIPT_PATH/vulscan`
				echo $message
			fi
		#handle the input: script full name display the describe about nse script
		elif [ "$match" != "" ] && [ $match = "nse" ] ; then
		       	nmap --script-help $flitered
		#handle the input: index
		elif [ "$index" != "" ] && [ $index = $1 ] ; then
			scriptName="$(head -n $index $DATAFILE_PATH'/dataFile' |tail -n 1 |awk '{print $2}')"
			echo $scriptName
			#less $scriptName	
			vi $scriptName
			exit 0
		#handle the basic keyword matchs searching
		else
			cd $SCRIPT_PATH ; ls |grep $flitered |nl
			ls |grep $flitered |nl > $DATAFILE_PATH'/dataFile'
			exit 0
		fi
	else
		echo "[-] Parameter 3rror"
	fi
elif [ $# -eq 2 ] ; then
	#List scripts in specified category
	if [ $1 = '-c' ];then
		searchInCate $2
	#Dispaly all the detail about the choosen NSE script	
	elif [ $1 = '-d' ] ; then
		flitered="$(echo $2 |sed -e 's/[^-.[:alnum:]]//g')"
		match=`echo $flitered |cut -d "." -f 2`
		if [ $match = 'nse' ] ; then
			less $2
		else
			echo "[-] Invalid 4rgument: $2"
		fi
	elif [ $2 = '-d' ] ; then
		flitered="$(echo $1 |sed -e 's/[^-.[:alnum:]]//g')"
		match=`echo $flitered |cut -d "." -f 2`
		if [ $match = 'nse' ] ; then
			less $1
		else
			echo "[-] Invalid 4rgument: $2"
		fi
	#check the parameters
	elif [ $1 = 'install' ] && [ $2 = 'vulnscanner' ] ; then
		ls|grep nmap-vulners
		check1=`echo $?`
		ls|grep vulscan
		check2=`echo $?`
		echo check1: $check1
		echo check2: $check2 
		#if [ $check1 != 'nmap-vulners' ] && [ $check2 != 'vulnscan' ] ; then
		if [ $check1 -eq 1 ] || [ $check2 -eq 1 ] ; then
			if [ $check1 != 'nmap-vulners' ] && [ $check2 != 'vulnscan' ] ; then
				echo "[-] The 'nmap-vulners' or 'vulnscan' does not be installed in your machine "
				echo "[-] Please input 'y' to install [y/n]"
				read choice
			
				if [ $choice = 'y' ] ; then
					git clone https://github.com/vulnersCom/nmap-vulners.git
					git clone https://github.com/scipag/vulscan.git
					chmod -R 755 nmap-vulners
					chmod -R 755 vulscan
					echo "[+] Install success !"
					echo
					echo "Example: "
					echo "        nmap -sV --script <nmap-vulners|vulnscan> <Target>"
					echo "        To discovery the known vulnerabilities. Have fun :)"
					echo
				fi
			fi	
		
		elif [ $check1 -eq 0 ] && [ $check2 -eq 0 ] ; then
			echo "[!] Error: nmap-vulner is already exist"
			echo "[-] use: 'nmap -sV --script <nmap-vulners|vulnscan> <Target>'"
			echo "[-] If you want to remove it try to use: $0 remove nmap-vulners"
			echo
		fi
	else
		printInfoBoard
		printHelp
	fi
# search in specify category
elif [ $# -eq 3 ];
then
	if [ "$1" == '-c' ];
	then
		searchInCate $2 $3
	fi
else
	printInfoBoard
fi