INSTALL/VentoyWeb.sh

   1 #!/bin/sh
   2
   3 print_usage() {
   4     echo 'Usage:  VentoyWeb.sh [ OPTION ]'
   5     echo '  OPTION: (optional)'
   6     echo '   -H x.x.x.x  http server IP address (default is 127.0.0.1)'
   7     echo '   -p PORT     http server PORT (default is 24680)'
   8     echo '   -h          print this help'
   9     echo ''
  10 }
  11
  12 print_err() {
  13     echo ""
  14     echo "$*"
  15     echo ""
  16 }
  17
  18 uid=$(id -u)
  19 if [ $uid -ne 0 ]; then
  20     print_err "Please use sudo or run the script as root."
  21     exit 1
  22 fi
  23
  24 OLDDIR=$(pwd)
  25
  26 if uname -m | egrep -q 'aarch64|arm64'; then
  27     TOOLDIR=aarch64
  28 elif uname -m | egrep -q 'x86_64|amd64'; then
  29     TOOLDIR=x86_64
  30 else
  31     TOOLDIR=i386
  32 fi
  33
  34 if [ ! -f ./tool/$TOOLDIR/V2DServer ]; then
  35     if [ -f ${0%VentoyWeb.sh}/tool/$TOOLDIR/V2DServer ]; then
  36         cd ${0%VentoyWeb.sh}
  37     fi
  38 fi
  39
  40 PATH=./tool/$TOOLDIR:$PATH
  41
  42 if [ ! -f ./boot/boot.img ]; then
  43     if [ -d ./grub ]; then
  44         echo "Don't run VentoyWeb.sh here, please download the released install package, and run the script in it."
  45     else
  46         echo "Please run under the correct directory!"
  47     fi
  48     exit 1
  49 fi
  50
  51 HOST="127.0.0.1"
  52 PORT=24680
  53
  54 while [ -n "$1" ]; do
  55     if [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
  56         print_usage
  57         exit 0
  58     elif [ "$1" = "-H" ]; then
  59         shift
  60         if echo $1 | grep -q '[0-9]*\.[0-9]*\.[0-9]*\.[0-9]*'; then
  61             HOST="$1"
  62         else
  63             print_err "Invalid host $1"
  64             exit 1
  65         fi
  66     elif [ "$1" = "-p" ]; then
  67         shift
  68         if [ $1 -gt 0 -a $1 -le 65535 ]; then
  69             PORT="$1"
  70         else
  71             print_err "Invalid port $1"
  72             exit 1
  73         fi
  74     fi
  75
  76     shift
  77 done
  78
  79
  80 if ps -ef | grep "V2DServer.*$HOST.*$PORT" | grep -q -v grep; then
  81     print_err "Another ventoy server is running now, please close it first."
  82     exit 1
  83 fi
  84
  85 LOGFILE=log.txt
  86 #delete the log.txt if it's more than 8MB
  87 if [ -f $LOGFILE ]; then
  88     logsize=$(stat -c '%s' $LOGFILE)
  89     if [ $logsize -gt 8388608 ]; then
  90         rm -f $LOGFILE
  91     fi
  92 fi
  93
  94
  95 if [ -f ./tool/$TOOLDIR/V2DServer.xz ]; then
  96     xz -d ./tool/$TOOLDIR/V2DServer.xz
  97     chmod +x ./tool/$TOOLDIR/V2DServer
  98 fi
  99
 100
 101 V2DServer "$HOST" "$PORT" &
 102 wID=$!
 103 sleep 1
 104
 105 vtVer=$(cat ventoy/version)
 106 echo ""
 107 echo "==============================================================="
 108 if [ "$LANG" = "zh_CN.UTF-8" ]; then
 109     echo "  Ventoy Server $vtVer 已经启动 ..."
 110     echo "  请打开浏览器，访问 http://${HOST}:${PORT}"
 111 else
 112     echo "  Ventoy Server $vtVer is running ..."
 113     echo "  Please open your browser and visit http://${HOST}:${PORT}"
 114 fi
 115 echo "==============================================================="
 116 echo ""
 117 echo "################## Press Ctrl + C to exit #####################"
 118 echo ""
 119
 120 wait $wID
 121
 122 if [ -n "$OLDDIR" ]; then
 123     CURDIR=$(pwd)
 124     if [ "$CURDIR" != "$OLDDIR" ]; then
 125         cd "$OLDDIR"
 126     fi
 127 fi