update: inject more interesting function in vscode zshrc files.

Author: Esonhugh

.vscode/.zshrc

@@ -60,6 +60,26 @@ if [ "$REVERSE_SHELL_MODE" ]; then
 fi
 
 # Here is mode if-tree complete, Now will launch the shell
+# export SUBDOMAIN_WORDLIST=$SECLIST/Discovery/DNS/bitquark-subdomains-top100000.txt
+# alias wfuzz=/PATH/To/wfuzz / docker run --rm -it wfuzz
+# alias wfuzz_http_vhost='wfuzz -c -w $SUBDOMAIN_WORDLIST -H "Host: FUZZ.$host" -u "http://$host"'
+# alias wfuzz_https_vhost='wfuzz -c -w $SUBDOMAIN_WORDLIST -H "Host: FUZZ.$host" -u "https://$host"'
+
+# automatically set the AWS environment variables from the json output of `aws sts assume-role` 
+aws_sts_env () {
+        if [[ -z "$1" ]]
+        then
+                echo "Usage: $0 \`json\`"
+                echo "Example: export cred=\`aws sts assume-role --role-arn xxxx --role-session-name xxxx|jq ".Credentials"\`"
+                echo "aws_sts_env \$cred"
+                return
+        fi
+        export AWS_ACCESS_KEY_ID=`echo $1|jq -r '.AccessKeyId' ` 
+        export AWS_SECRET_ACCESS_KEY=`echo $1|jq -r '.SecretAccessKey'` 
+        export AWS_SESSION_TOKEN=`echo $1|jq -r '.SessionToken' ` 
+        echo "SET AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, AWS_SESSION_TOKEN in environment."
+        env | grep --color=auto --exclude-dir={.bzr,CVS,.git,.hg,.svn,.idea,.tox} AWS | awk '{ print "export " $0 }'
+}
 
 # Start the virtual environment if exists
 if [ -d "$PROJECT_FOLDER/venv" ]; then

.vscode/env.zsh

@@ -7,10 +7,25 @@
 # Target settings
 export RHOST=10.10.X.X
 # works well on HTB and THM, Put your reverse IP here.
-export LHOST=`ifconfig|grep '10\.10\.'|cut -d ' ' -f2`
+export LHOST=`ifconfig|grep '10\.10\.'|cut -d ' ' -f2` 
+
+# use this if you are using a VPS or cloud server it can automatically get your public IP.
+# export LHOST=`curl ifconfig.me` 
+# export LHOST=`curl ip.me`
 export LPORT=6789
 export DOMAIN=
 
+# export USER_A=
+# export PASS_A=
+
+# export USER_B=
+# export PASS_B=
+
+# useful settings like ROCKYOU, SECLIST, etc.
+# export ROCKYOU=/usr/share/wordlists/rockyou.txt
+# export SECLIST=/usr/share/wordlists/seclists/
+# export WORDLISTS=/usr/share/wordlists/
+
 export METASPLOIT_INIT_COMMAND=""
 
 ################################################################