@echo off & setlocal enabledelayedexpansion
echo [+] Getting Host IP Address...

:: 打印获取宿主机的 IP 地址
for /f "tokens=1,2 delims=:" %%A in ('ipconfig ^| findstr "IPv4"') do (
set "ip=%%B"
set "ip=!ip:~1!"
echo IPv4 Address: !ip!
)

echo [+] Starting Cobalt Strike server...

:: 获取 WSL 的 IP 地址
for /f "delims=" %%i in ('wsl -u root -- bash -c "hostname -I"') do (
set wslIP=%%i
goto :done
)

:done

:: 设置默认密码
set defaultPassword=password

:: 打印获取的WSL IP 和密码
echo WSL IP Address: !wslIP!
echo Default Password: !defaultPassword!

:: 启动 Cobalt Strike 服务器并在后台运行
start /B wsl -u root -- bash -c "cd ./Server && ./teamserver !wslIP! !defaultPassword!"
echo [+] Cobalt Strike server started.

:: 等待一段时间（例如 10 秒）
timeout /t 10 >nul

echo [+] Starting Cobalt Strike client...

:: 启动 Cobalt Strike 客户端并在后台运行
start /B java -XX:ParallelGCThreads=4 -XX:+AggressiveHeap -XX:+UseParallelGC -javaagent:./Client/uHook.jar -Xms512M -Xmx1024M -jar ./Client/cobaltstrike-client.jar

:: 等待用户输入以保持窗口打开
pause