#!/bin/bash
# vim: expandtab tabstop=4 shiftwidth=4

# Fun fact is this project made by
# A tinkerer, not developer xD
# You might see this line but still ignore it then continues calling me a developer, but after all, i'm just a loser :D

# Initial startup
mkdir -p ~/mbbpatch/MBCPApp/mbapk
mkdir -p ~/mbbpatch/MBCPApp/mbcpapp_apk
mkdir -p ~/mbbpatch/MBCPApp/tools
clear

# Check if user runs on actual Linux environment
if uname -a | grep -i Linux ; then
    echo "INFO : You have Linux environment :)"
fi

# Check if user is trying to run under WSL (Windows Subsystem for Linux)
# Removing this detection part might cause project development to be STOPPED, You've been warned !
if [ -f /proc/sys/fs/binfmt_misc/WSLInterop ]; then
    echo "Detected WSL (Windows Subsystem for Linux) in current Linux environment !!"
    echo "Warn : You won't be supported for any issues under WSL environment !!"
fi

# Check if figlet exists or not
if command -v figlet ; then
    FIGLET=figlet
else
    echo "WARN : figlet not found, ignoring banner !"
    FIGLET=:
fi

COMMIT="$(git rev-parse --short HEAD)"

# Check if java exists on /usr/bin/java
echo 'Checking if Java exists...'
if ! java -version ; then
    echo "ERROR : Java not found !!!"
    echo "INFO : Please install Java for your Linux distribution ! "
    exit 127
fi

clear

download_tools() {
    apktool_link="https://bitbucket.org/iBotPeaches/apktool/downloads/apktool_2.12.0.jar"
    apkeditor_link="https://github.com/REAndroid/APKEditor/releases/download/V1.4.4/APKEditor-1.4.4.jar"

    cd tools && rm -rf *.jar
    wget -q --show-progress -O apktool.jar "$apktool_link" || echo "ERROR : downloading apktool failed"
    wget -q --show-progress -O apkeditor.jar "$apkeditor_link" || echo "ERROR : downloading apkeditor failed"
    cd .. && echo "You can now continue with other operations !"
}

copy_assets() {
    adb shell su -c cp -rf "/data/user/0/com.mbmobile/files/$1" /sdcard/assets
}

is_unpacked() { [ -d ~/mbbpatch/MBCPApp/mbapk/mbapk_unpacked ] ; }
is_unpacked_lib() {
    [ -d ~/mbbpatch/MBCPApp/mbapk/mbapk_unpacked/lib ] || {
        echo "ERROR : [mbapk_unpacked/lib] folder not found ! Please unpack APK first !" ; exit 127
    }
}
apktool_exist() {
    [ -f ~/mbbpatch/MBCPApp/tools/apktool.jar ] || { echo "ERROR : apktool not found" ; exit 127 ; }
}
mb_apk_exist() {
    ls ~/mbbpatch/MBCPApp/mbapk/*.apk >/dev/null 2>&1 || {
        echo "No *.apk found in [mbapk] folder, please copy apk to [mbapk] folder !"
        echo "If you got apks from eMBee APKs, use [Convert apks to apk] option !"
    }
}

unpack_mbcp() {
    local - ; set -e
    apktool_exist && mb_apk_exist
    is_unpacked && rm -rf 'mbapk/mbapk_unpacked' || :
    java -jar tools/apktool.jar d mbapk/*.apk -o mbapk/mbapk_unpacked -j$(nproc) || { echo "ERROR : Unpacking failed !" ; exit 1  ; }
    echo "Cleaning useless files..."
    rm -rf 'mbapk/mbapk_unpacked/assets/_4A9w8flncUrhDOG8dyqLi_azBTYT3PlSXz0hiCzRQA_'
    rm -rf 'mbapk/mbapk_unpacked/assets/0QDl12M5S2hKxoKF4cNI4kEX1qDQRMiOd34TXjSjy4M_'
    rm -rf 'mbapk/mbapk_unpacked/assets/2GcdAWdkXGgxPfHSIhzLkrkf2LU6Z_cuZfUWnczoEHw_'
    rm -rf 'mbapk/mbapk_unpacked/assets/BiLlLhV4LAy_y_rvwmD82q_4yN_liwgN1X0mfwfqTeE_'
    rm -rf 'mbapk/mbapk_unpacked/assets/c7uFHgsHzlQ_T524_4N3h9neXvtcUg0m1ERhEjFpPI8_'
    rm -rf 'mbapk/mbapk_unpacked/assets/ccBu4aSM_mSaMBpFvr14pizu6EGMvwRiZltU_zVJMz0_'
    rm -rf 'mbapk/mbapk_unpacked/assets/DcM3z3tolULDhbn8YrUE7hxOlxZx2oFC5yNQFG1SX9k_'
    rm -rf 'mbapk/mbapk_unpacked/assets/fPGzhQFvISia1NiVGU8vQx9IpRm63E4_1Xv0Z2ypWUE_'
    rm -rf 'mbapk/mbapk_unpacked/assets/g9x205p_On9_RR4kI_GdsoQ2pfSrV_OQXSopshyhYVk_'
    rm -rf 'mbapk/mbapk_unpacked/assets/GXNv8xc5IcVF6TyUgjJpyYFeyHGqUqN7ZS0X_2WRbEE_'
    rm -rf 'mbapk/mbapk_unpacked/assets/H8leCCp85eX0YRvo4WCzqF5MEuj2_9ix6zL2bjnj6hk_'
    rm -rf 'mbapk/mbapk_unpacked/assets/Hput_BfAb8ftOt8hvazec24Eqk0gC5DDY6qALJMLtcI_'
    rm -rf 'mbapk/mbapk_unpacked/assets/MZuOOtdhO6pcGnZ5_kqMPh56hL3LqawBZd05vfIivww_'
    rm -rf 'mbapk/mbapk_unpacked/assets/noNmNoHJ0EUpV0TiKtyRPIVWI0im6K1RPIq3N_q0ho0_'
    rm -rf 'mbapk/mbapk_unpacked/assets/oOot9yCPD3GFqBix7JXjoz4Uui3k3NrwI8wHRlOpn4s_'
    rm -rf 'mbapk/mbapk_unpacked/assets/rqfr7ucXEZM4vIIYYPHXlo6IG1iEad_H61Z4bbEMR3s_'
    rm -rf 'mbapk/mbapk_unpacked/assets/Sje9R7eg_A4wlgbLNHwXS5xNFmS5qaoPnJ6Vg0RnIQ8_'
    rm -rf 'mbapk/mbapk_unpacked/assets/t8_bY_8ibDPQ9rngMwnd7WFI_uqoPlht9nBi26_llbw_'
    rm -rf 'mbapk/mbapk_unpacked/assets/VipqZ50mSUd28eIlPS_2t31IZ6tY24fl8sSl5YsQrbI_'
    rm -rf 'mbapk/mbapk_unpacked/assets/vLpH08bnGzkpsq_MmUdHqkh3bjrVN5tutp6s4jscm3w_'
    rm -rf 'mbapk/mbapk_unpacked/assets/WZ7CC4mxgoVWzgm8AO7rbOneE6TkqXwT2YV3htdswEQ_'
    rm -rf 'mbapk/mbapk_unpacked/assets/x_uYtCXdh_oYw_XLfZQkX10Bcqq2HFK3q7v1dJVw3zM_'
    rm -rf 'mbapk/mbapk_unpacked/assets/Y83jllvBi79vZIh2_UNtyo3_Lvd94lqie_q1dc5O3j0_'
    rm -rf 'mbapk/mbapk_unpacked/assets/ZN6CRAqtR7CKp7LFoivhHDbCS6iAtrEW6_sLvrfWgEM_'
    rm -rf 'mbapk/mbapk_unpacked/assets/ZOKOK1a6XCiqYb_a8bYPE9c0rjUf7_n7k77YxX5Ypdw_'
    rm -rf 'mbapk/mbapk_unpacked/assets/zxHdhL4_ZEQLQeo8R2DzSYl9PGss_cKSCQTcZGBtVmU_'
    rm -rf 'mbapk/mbapk_unpacked/assets/jNMI79raTvV7VHN5Vrzw4Xs66YP_a1ftP9SmUB24DYg_'
    echo "Creating [mbcp_info] folder..."
    mkdir 'mbapk/mbapk_unpacked/assets/mbcp_info/'
}

repack_mbcp() {
    apktool_exist
    is_unpacked || { echo "ERROR : [mbapk_unpacked] folder not found ! Please unpack APK first !" ; exit 127 ; }
    echo "Repacking APK..."
    echo "Compiled by MBCPApp Patcher on $(uname -s -r) with commit $COMMIT at $(date). That's all xD" > 'mbapk/mbapk_unpacked/assets/mbcp_info/mbcpinfo.txt'
    (
        set -e
        java -jar tools/apktool.jar b mbapk/mbapk_unpacked -o mbcpapp_apk/MBCP_Flutter_TMP.apk -j$(nproc)
        echo "Processing APK signature scheme v2/v3..."
        rm -rf 'mbcpapp_apk/MBCP_Flutter_TMP_decompile_xml'
        java -jar tools/apkeditor.jar d -i 'mbcpapp_apk/MBCP_Flutter_TMP.apk' 
        cp -r 'mbsig/signatures' 'mbcpapp_apk/MBCP_Flutter_TMP_decompile_xml/'
        java -jar tools/apkeditor.jar b -i 'mbcpapp_apk/MBCP_Flutter_TMP_decompile_xml' 
        mv 'mbcpapp_apk/MBCP_Flutter_TMP_decompile_xml_out.apk' 'mbcpapp_apk/MBCP_Flutter_SelfPatched.apk'
        rm -rf 'mbcpapp_apk/MBCP_Flutter_TMP_decompile_xml'
        rm 'mbcpapp_apk/MBCP_Flutter_TMP.apk'
    ) && {
        echo 'Completed! Repacked APK are saved as [mbcpapp_apk/MBCP_Flutter_SelfPatched.apk] !!!'
        echo 'Install and trying to open it when ಠ‿ಠ'
        echo 'If you are facing issues, report it on Telegram [@mbbpatch] or GitLab : mbbpatch !!'
    } || echo 'ERROR : Repacking failed !'
}

run_patcher() {
    is_unpacked_lib && bash ./patch.sh
}

run_legacy_patcher() {
    is_unpacked_lib && bash ./legacy_patch.sh
}

# Banner 
$FIGLET "MBCPApp Patcher"
echo -------------------------------------------------------------
echo "Patching-tool for MB Bank app with Flutter engine (v6.4.0+)"
echo "Original APK path must be inside [mbapk] folder !"
echo "IMPORTANT : Script must be run on [~/mbbpatch/MBCPApp] !"
echo -------------------------------------------------------------
echo "Current commit : $COMMIT (dev)"
echo "Source code : https://gitlab.com/mbcp/mbbpatch"
echo "Made possible by Cuynu with love <3"
echo -------------------------------------------------------------   
# Main functions      
PS3='Please select options to continue : '
select opt in 'Pull latest commit' 'Unpack APK' 'Convert apks to apk' 'Repack APK' 'Install patched app' 'MBShield Check' 'Patch App' 'Legacy patches' 'Extract assets [ROOT]' 'Launch MBCPApp/MBBank' 'Force close MBCPApp/MBBank' 'Clear MBCPApp/MBBank app data' 'Clean patched app' 'Download tools' 'Exit'
do
    if [ "$opt" == 'Download tools' ]; then
    download_tools

    elif [ "$opt" == 'Unpack APK' ]; then
    unpack_mbcp

    elif [ "$opt" == 'Repack APK' ]; then
    repack_mbcp

    elif [ "$opt" == 'MBShield Check' ]; then
    if [ -d ~/mbbpatch/MBCPApp/mbapk/mbapk_unpacked/ ]  
then
    if [ -f ~/mbbpatch/MBCPApp/mbapk/mbapk_unpacked/assets/mbshield.szip ]
then
    echo "MBShield found on [mbapk_unpacked/assets/mbshield.szip]" !!!
    echo "Patching operation is limited due to MBShield :("
    echo "You need to extract assets from Android device for some patching operation to works properly !!!"
else
    echo "MBShield NOT found on [mbapk_unpacked/assets/]"
    echo "Patching operation are all not limited at all !"
    echo "Asset extraction is not needed :)"
    fi
else 
    echo "[mbapk_unpacked] not found ! Please unpack APK first !"
fi


    elif [ "$opt" == 'Convert apks to apk' ]; then
    if ls ~/mbbpatch/MBCPApp/mbapk/*.apks >/dev/null 2>&1
 then
    echo "Converting apks to apk..."
    java -jar tools/apkeditor.jar m -i mbapk/*.apks 
    mv mbapk/*.apk mbapk/MBOriginal.apk
    echo 'Cleaning left over [apks] files...'
    rm -f mbapk/*.apks
    echo "You probably can continue to unpack APK!"
 else
    echo "APKs missing, cannot continue !"
 fi

    elif [ "$opt" == 'Patch App' ]; then
    run_patcher

    elif [ "$opt" == 'Legacy patches' ]; then
    run_legacy_patcher


    elif [ "$opt" == 'Extract assets [ROOT]' ]; then
    echo 'To extract encrypted assets [if current app has MBShield protection] you need rooted device'
    echo 'And trigger a bulit-in app assets extraction !'
    echo -------------------------------------------
    echo -------------------------------------------
    PS3='Select options continue, or [3] to quit : '
    select opt in 'Launch MBCPApp/MBBank' 'Extract assets' 'Exit'
do
    	if [ "$opt" == 'Launch MBCPApp/MBBank' ]; then
        adb shell am start -n com.mbmobile/io.flutter.plugins.MainActivity
        echo 'After app started, you MUST need to trigger assets extraction'
        echo 'By trigger eKYC authetication with reset account password option or register DigitalOTP'
        echo 'After you got into eKYC screen, close the app, then use Extract assets option to extract assets.'
        elif [ "$opt" == 'Extract assets' ]; then
        if [ -d ~/mbbpatch/MBCPApp/mbapk/mbapk_unpacked/ ]  
then
        if [ -f ~/mbbpatch/MBCPApp/mbapk/mbapk_unpacked/assets/mbshield.szip ]  
then
        echo "INFO : MBShield found ! Continuing !!!"
        echo 'You MUST grant root access to [com.android.shell] in order to extract assets !'
        echo 'Trying to extract assets...'
        adb shell am force-stop com.mbmobile
        adb shell rm -rf /sdcard/assets
        adb shell mkdir /sdcard/assets
        copy_assets flutter_assets
        copy_assets dexopt
        copy_assets font
        copy_assets mbshield
        copy_assets media_init_data
        copy_assets mlkit_barcode_models
        copy_assets models_bundled
        copy_assets mwebview
        copy_assets zfiles
        copy_assets crashlytics-build.properties
        copy_assets firmware
        copy_assets insider.ttf
        copy_assets dlangV5.dat
        copy_assets dlangV5.en.dat
        copy_assets policy0
        copy_assets policy1
        copy_assets policy2
        copy_assets policy3
        copy_assets policy4
        copy_assets policy5
        copy_assets policy6
        copy_assets policy8
        copy_assets policy9
        copy_assets policym
        copy_assets profile
        copy_assets rulesV5.dat
        copy_assets sgprofile
        copy_assets signature
        copy_assets smart_ekyc_finall.zip
        copy_assets version.json
        copy_assets vkeylicensepack
        copy_assets voscodesign.vky
        echo 'Copying assets...'
        adb pull /sdcard/assets mbapk/mbapk_unpacked/
        adb shell rm -rf /sdcard/assets
        else
        echo "INFO : MBShield not found ! No need to extract assets !"
        fi
else
        echo "ERROR : Can't find [mbapk/mbapk_unpacked] folder, cannot continue ! "
        echo "Please unpack APK first !"
        fi

        elif [ "$opt" == 'Exit' ]; then
        clear
        bash mbcpapp.sh
       break
        	fi
done

    elif [ "$opt" == 'Clear MBCPApp/MBBank app data' ]; then
    echo "Clearing [com.mbmobile] data..."
    adb shell pm clear com.mbmobile
    adb shell am start -n com.mbmobile/io.flutter.plugins.MainActivity
    echo "INFO : Current logged in account will remain present, even if app data is cleared !"

    elif [ "$opt" == 'Pull latest commit' ]; then
    git pull origin dev
    echo "Please run patcher again !"
    exit

    elif [ "$opt" == 'Launch MBCPApp/MBBank' ]; then
    adb shell am start -n com.mbmobile/io.flutter.plugins.MainActivity
    elif [ "$opt" == 'Force close MBCPApp/MBBank' ]; then
    adb shell am force-stop com.mbmobile

    elif [ "$opt" == 'Clean patched app' ]; then
    echo "Cleaning, please wait..." 
    rm -f mbcpapp_apk/*.apk
    rm -f mbcpapp_apk/*.zip
    rm -f mbapk/*.apk
    rm -f mbapk/*.apks
    rm -rf mbapk/mbapk_unpacked
    echo "Cleared !"

 
    elif [ "$opt" == 'Install patched app' ]; then
    echo 'Select your patched version to continue'
    PS3='Select patched version to continue, or [3] to quit : '
    select opt in 'App patched with newer version [v6.4.56 or higher]' 'App patched with older version [v6.4.55 or lower]' 'Exit'
do
    	if [ "$opt" == 'App patched with newer version [v6.4.56 or higher]' ]; then
        if [ -f ~/mbbpatch/MBCPApp/mbcpapp_apk/MBCP_Flutter_SelfPatched.apk ]
    then 
        echo "MBCP Helper / CorePatch must be installed with disable digest verify on !!!"
        echo "You must have connected Android device with USB debugging turned on in order to install !!!"
        adb kill-server
        adb start-server
        adb install 'mbcpapp_apk/MBCP_Flutter_SelfPatched.apk'
        echo "ATTENTION : Network traffic will be redirected to [medium.com] for 20 seconds !!!"
        adb shell su -c 'iptables -t nat -A OUTPUT -p tcp -d 0/0 -j DNAT --to-destination 162.159.153.4:443'
        adb shell am start -n com.mbmobile/io.flutter.plugins.MainActivity
        sleep 20
        echo "Restoring network traffic"
        adb shell su -c 'iptables -t nat -F OUTPUT'
        echo "Press [Try again] after got 1005/1007 error on MB, so it's can skip device not secure dialog !"
    else
        echo "[MBCP_Flutter_SelfPatched.apk] not found, cannot continue !"
    fi
        
        elif [ "$opt" == 'App patched with older version [v6.4.55 or lower]' ]; then
        if [ -f ~/mbbpatch/MBCPApp/mbcpapp_apk/MBCP_Flutter_SelfPatched.apk ]
    then 
        echo "MBCP Helper / CorePatch must be installed with disable digest verify on !!!"
        echo "You must have connected Android device with USB debugging turned on in order to install !!!"
        adb kill-server
        adb start-server
        adb install 'mbcpapp_apk/MBCP_Flutter_SelfPatched.apk'
        adb shell am start -n com.mbmobile/io.flutter.plugins.MainActivity
    else
        echo "[MBCP_Flutter_SelfPatched.apk] not found, cannot continue !"
    fi


        elif [ "$opt" == 'Exit' ]; then
        clear
        bash mbcpapp.sh
       break
        	fi
done


    elif [ "$opt" == 'Exit' ]; then
		exit
		break
	fi
done