From 6798ba771c84807ebba3fdf5f43d52adef4bd9e2 Mon Sep 17 00:00:00 2001
From: hxAri <hxari@proton.me>
Date: Sun, 30 Jun 2024 03:41:34 +0700
Subject: [PATCH] [FEATURE] Added kex utility into Kali Nethunter distro

---
 penguin | 75 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)

diff --git a/penguin b/penguin
index ca7c814..3391dc4 100755
--- a/penguin
+++ b/penguin
@@ -2596,6 +2596,81 @@ function distroConfig() {
 					exit 1
 				fi
 			fi
+			
+			echo "$(stdio stdout $distro creating /:usr/bin/kex)"
+			cat > $source/$folder/usr/bin/kex <<- EOF
+			#!/bin/env/bash
+			
+			function start-kex() {
+			    if [[ ! -f ~/.vnc/passwd ]]; then
+			        passwd-kex
+			    fi
+			    USR=\$(whoami)
+			    if [ \$USR == "root" ]; then
+			        SCREEN=":2"
+			    else
+			        SCREEN=":1"
+			    fi 
+			    export MOZ_FAKE_NO_SANDBOX=1; export HOME=\${HOME}; export USER=\${USR}; LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgcc_s.so.1 nohup vncserver \$SCREEN >/dev/null 2>&1 </dev/null
+			    starting_kex=1
+			    return 0
+			}
+			
+			function stop-kex() {
+			    vncserver -kill :1 | sed s/"Xtigervnc"/"NetHunter KeX"/
+			    vncserver -kill :2 | sed s/"Xtigervnc"/"NetHunter KeX"/
+			    return $?
+			}
+			
+			function passwd-kex() {
+			    vncpasswd
+			    return $?
+			}
+			
+			function status-kex() {
+			    sessions=\$(vncserver -list | sed s/"TigerVNC"/"NetHunter KeX"/)
+			    if [[ \$sessions == *"590"* ]]; then
+			        printf "\n\${sessions}\n"
+			        printf "\nYou can use the KeX client to connect to any of these displays.\n\n"
+			    else
+			        if [ ! -z \$starting_kex ]; then
+			            printf '\nError starting the KeX server.\nPlease try "nethunter kex kill" or restart your termux session and try again.\n\n'
+			        fi
+			    fi
+			    return 0
+			}
+			
+			function kill-kex() {
+			    pkill Xtigervnc
+			    return \$?
+			}
+			
+			case \$1 in
+			    start)
+			        start-kex
+			        ;;
+			    stop)
+			        stop-kex
+			        ;;
+			    status)
+			        status-kex
+			        ;;
+			    passwd)
+			        passwd-kex
+			        ;;
+			    kill)
+			        kill-kex
+			        ;;
+			    *)
+			        stop-kex
+			        start-kex
+			        status-kex
+			        ;;
+			esac
+			EOF
+			
+			echo "$(stdio stdout $distro chmod+s /:usr/bin/kex)"
+			chmod 700 $source/$folder/usr/bin/kex
 						
 			echo "$(stdio stdout $distro fixing sudo)"
 			echo "$(stdio stdout $distro chmod+s /:usr/bin/sudo)"