#!/bin/bash

if [ "$HashUtilsVersion" ]; then return 0; fi
readonly HashUtilsVersion="1.0"

HashOutputDevice="/dev/stdout"

function hash_check_handshake() {
	local handshakeVerifier=$1
	local handshakePath=$2
	local handshakeAPSSID=$3
	local handshakeAPMAC=$4

	local analysis
	local hashData

	echo "Verifier Parameters: $handshakeVerifier, path $handshakePath, SSID \"$handshakeAPSSID\", MAC $handshakeAPMAC" > $HashOutputDevice

	case "$handshakeVerifier" in
		"pyrit")
			readarray analysis < <(pyrit -r "$handshakePath" analyze 2> $HashOutputDevice)
			if [ "${#analysis[@]}" -eq 0 -o $? != 0 ]; then
				echo "Error: pyrit seems to be broken!" > $HashOutputDevice
				return 1
			fi

			local hashMeta=$(echo "${analysis[@]}" | grep -F "AccessPoint ${handshakeAPMAC,,} ('$handshakeAPSSID')")

			if [ "$hashMeta" ]; then
				local hashID=$(echo "$hashMeta" |  awk -F'[ #:]' '{print $3}')
				hashData=$(echo "${analysis[@]}" | awk "\$0~/#$hashID: HMAC_SHA[0-9]+_AES/{ print \$0 }")
			else
			    echo "No valid hash meta was found for \"$handshakeAPSSID\"" > $HashOutputDevice
			fi;;
		"aircrack-ng")
			readarray analysis < <(aircrack-ng "$handshakePath" 2> $HashOutputDevice)
			if [ "${#analysis[@]}" -eq 0 -o $? != 0 ]; then
				echo "Error: aircrack-ng seems to be broken!" > $HashOutputDevice
				return 1
			fi

			hashData=$(echo "${analysis[@]}" | grep -E "${handshakeAPMAC^^}\s+" | grep -F "$handshakeAPSSID");;
		*) echo "Invalid verifier, quitting!"; return 1;;
	esac

	if [ -z "$hashData" ]; then
		echo "Handshake for $handshakeAPSSID ($handshakeAPMAC) is missing!"
		return 1
	fi

	local hashResult
	case "$handshakeVerifier" in
		"pyrit") hashResult=$(echo "$hashData" | grep "good");;
		"aircrack-ng") hashResult=$(echo "$hashData" | grep "(1 handshake)");;
	esac

	if [ -z "$hashResult" ]; then
		echo "Invalid hash for $handshakeAPSSID ($handshakeAPMAC)!"
		HASHCheckHandshake="invalid"
		return 1
	else
		echo "Valid hash for $handshakeAPSSID ($handshakeAPMAC)!"
		HASHCheckHandshake="valid"
	fi
}