#! /bin/bash

if [[ $EUID -ne 0 ]]; then
	echo "This script must be run as root!"
	exit 1
fi

# make sure we have a wordlists directory
[[ ! -d /usr/share/wordlists ]] && mkdir -p /usr/share/wordlists || true

# add a link to the metasploit wordlists
ln -s /opt/metasploit-framework/embedded/framework/data/wordlists /usr/share/wordlists/metasploit

# add a link to the seclists
ln -s /usr/share/seclists /usr/share/wordlists/seclists

# add seclists webshells to generic webshells
[[ -d /usr/share/webshells && -d /usr/share/seclists/Web-Shells ]] && ln -s /usr/share/seclists/Web-Shells /usr/share/webshells/seclists || true