mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-13 17:50:29 +08:00
301 lines
9.5 KiB
Bash
Executable File
301 lines
9.5 KiB
Bash
Executable File
#!/bin/bash
|
||
# ========================================================================== #
|
||
# #
|
||
# KVMD - The main PiKVM daemon. #
|
||
# #
|
||
# Copyright (C) 2023-2025 SilentWind <mofeng654321@hotmail.com> #
|
||
# #
|
||
# This program is free software: you can redistribute it and/or modify #
|
||
# it under the terms of the GNU General Public License as published by #
|
||
# the Free Software Foundation, either version 3 of the License, or #
|
||
# (at your option) any later version. #
|
||
# #
|
||
# This program is distributed in the hope that it will be useful, #
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
||
# GNU General Public License for more details. #
|
||
# #
|
||
# You should have received a copy of the GNU General Public License #
|
||
# along with this program. If not, see <https://www.gnu.org/licenses/>. #
|
||
# #
|
||
# ========================================================================== #
|
||
|
||
# 定义颜色代码
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[0;33m'
|
||
BLUE='\033[0;34m'
|
||
NC='\033[0m'
|
||
|
||
# 输出日志的函数
|
||
log_info() {
|
||
echo -e "${GREEN}[INFO] $1${NC}"
|
||
}
|
||
|
||
log_warn() {
|
||
echo -e "${YELLOW}[WARN] $1${NC}"
|
||
}
|
||
|
||
log_error() {
|
||
echo -e "${RED}[ERROR] $1${NC}"
|
||
}
|
||
|
||
# 初始化检查
|
||
log_info "One-KVM 正在启动..."
|
||
|
||
# 首次初始化配置
|
||
if [ ! -f /etc/kvmd/.init_flag ]; then
|
||
log_info "首次初始化配置..."
|
||
|
||
# 创建必要目录并移动配置文件
|
||
if mkdir -p /etc/kvmd/ && \
|
||
mv /etc/kvmd_backup/* /etc/kvmd/ && \
|
||
touch /etc/kvmd/.docker_flag && \
|
||
sed -i 's/localhost.localdomain/docker/g' /etc/kvmd/meta.yaml && \
|
||
sed -i 's/localhost/localhost:4430/g' /etc/kvmd/kvm_input.sh; then
|
||
log_info "移动配置文件完成"
|
||
else
|
||
log_error "移动配置文件失败"
|
||
exit 1
|
||
fi
|
||
|
||
# SSL证书配置
|
||
if ! /usr/share/kvmd/kvmd-gencert --do-the-thing; then
|
||
log_error "Nginx SSL 证书生成失败"
|
||
exit 1
|
||
fi
|
||
if ! /usr/share/kvmd/kvmd-gencert --do-the-thing --vnc; then
|
||
log_error "VNC SSL 证书生成失败"
|
||
exit 1
|
||
fi
|
||
|
||
# 设置用户名和密码
|
||
if [ ! -z "$USERNAME" ] && [ ! -z "$PASSWORD" ]; then
|
||
# 设置自定义用户名和密码
|
||
if python -m kvmd.apps.htpasswd del admin \
|
||
&& echo "$PASSWORD" | python -m kvmd.apps.htpasswd add -i "$USERNAME" \
|
||
&& echo "$PASSWORD -> $USERNAME:$PASSWORD" > /etc/kvmd/vncpasswd \
|
||
&& echo "$USERNAME:$PASSWORD -> $USERNAME:$PASSWORD" > /etc/kvmd/ipmipasswd; then
|
||
log_info "用户凭据设置成功"
|
||
else
|
||
log_error "用户凭据设置失败"
|
||
exit 1
|
||
fi
|
||
elif [ ! -z "$PASSWORD" ] && [ -z "$USERNAME" ]; then
|
||
# 只设置密码，保持admin用户名
|
||
if echo "$PASSWORD" | python -m kvmd.apps.htpasswd set -i "admin" \
|
||
&& echo "$PASSWORD -> admin:$PASSWORD" > /etc/kvmd/vncpasswd \
|
||
&& echo "admin:$PASSWORD -> admin:$PASSWORD" > /etc/kvmd/ipmipasswd; then
|
||
log_info "admin 用户密码设置成功"
|
||
else
|
||
log_error "admin 用户密码设置失败"
|
||
exit 1
|
||
fi
|
||
else
|
||
log_warn "未设置 USERNAME 和 PASSWORD 环境变量，使用默认值(admin/admin)"
|
||
fi
|
||
|
||
# SSL开关配置
|
||
if [ "$NOSSL" == 1 ]; then
|
||
log_info "已禁用SSL"
|
||
if ! python -m kvmd.apps.ngxmkconf /etc/kvmd/nginx/nginx.conf.mako /etc/kvmd/nginx/nginx.conf -o nginx/https/enabled=false; then
|
||
log_error "Nginx 配置失败"
|
||
exit 1
|
||
fi
|
||
else
|
||
if ! python -m kvmd.apps.ngxmkconf /etc/kvmd/nginx/nginx.conf.mako /etc/kvmd/nginx/nginx.conf; then
|
||
log_error "Nginx 配置失败"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
# 认证配置
|
||
if [ "$NOAUTH" == "1" ]; then
|
||
sed -i "s/enabled: true/enabled: false/g" /etc/kvmd/override.yaml
|
||
log_info "已禁用认证"
|
||
fi
|
||
|
||
#add supervisord conf
|
||
if [ "$NOWEBTERM" == "1" ]; then
|
||
log_info "已禁用 WebTerm 功能"
|
||
rm -r /usr/share/kvmd/extras/webterm
|
||
else
|
||
cat >> /etc/kvmd/supervisord.conf << EOF
|
||
|
||
[program:kvmd-webterm]
|
||
command=/usr/local/bin/ttyd --interface=/run/kvmd/ttyd.sock --port=0 --writable /bin/bash -c '/etc/kvmd/armbain-motd; bash'
|
||
directory=/
|
||
autostart=true
|
||
autorestart=true
|
||
priority=14
|
||
stopasgroup=true
|
||
stdout_logfile=/dev/stdout
|
||
stdout_logfile_maxbytes = 0
|
||
redirect_stderr=true
|
||
EOF
|
||
fi
|
||
|
||
if [ "$NOWEBTERMWRITE" == "1" ]; then
|
||
sed -i "s/--writable//g" /etc/kvmd/supervisord.conf
|
||
fi
|
||
|
||
if [ "$NOVNC" == "1" ]; then
|
||
log_info "已禁用 VNC 功能"
|
||
rm -r /usr/share/kvmd/extras/vnc
|
||
else
|
||
cat >> /etc/kvmd/supervisord.conf << EOF
|
||
|
||
[program:kvmd-vnc]
|
||
command=python -m kvmd.apps.vnc --run
|
||
directory=/
|
||
autostart=true
|
||
autorestart=true
|
||
priority=11
|
||
stopasgroup=true
|
||
stdout_logfile=/dev/stdout
|
||
stdout_logfile_maxbytes = 0
|
||
redirect_stderr=true
|
||
EOF
|
||
fi
|
||
|
||
if [ "$NOIPMI" == "1" ]; then
|
||
log_info "已禁用 IPMI 功能"
|
||
rm -r /usr/share/kvmd/extras/ipmi
|
||
else
|
||
cat >> /etc/kvmd/supervisord.conf << EOF
|
||
|
||
[program:kvmd-ipmi]
|
||
command=python -m kvmd.apps.ipmi --run
|
||
directory=/
|
||
autostart=true
|
||
autorestart=true
|
||
priority=12
|
||
stopasgroup=true
|
||
stdout_logfile=/dev/stdout
|
||
stdout_logfile_maxbytes = 0
|
||
redirect_stderr=true
|
||
EOF
|
||
fi
|
||
|
||
if [ "$NOGOSTC" == "1" ]; then
|
||
log_info "已禁用 GOSTC 功能"
|
||
rm -rf /usr/share/kvmd/extras/gostc
|
||
else
|
||
cat >> /etc/kvmd/supervisord.conf << EOF
|
||
|
||
[program:kvmd-gostc]
|
||
command=/usr/bin/gostc -web-addr 0.0.0.0:18080
|
||
autostart=true
|
||
autorestart=true
|
||
startsecs=5
|
||
priority=300
|
||
stopasgroup=true
|
||
stdout_logfile=/dev/stdout
|
||
stdout_logfile_maxbytes = 0
|
||
redirect_stderr=true
|
||
EOF
|
||
fi
|
||
|
||
#switch OTG config
|
||
if [ "$OTG" == "1" ]; then
|
||
log_info "已启用 OTG 功能"
|
||
sed -i "s/ch9329/otg/g" /etc/kvmd/override.yaml
|
||
sed -i "s|device: /dev/ttyUSB0||g" /etc/kvmd/override.yaml
|
||
if [ "$NOMSD" == 1 ]; then
|
||
log_info "已禁用 MSD 功能"
|
||
else
|
||
sed -i "s/#type: otg/type: otg/g" /etc/kvmd/override.yaml
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$VIDEONUM" ]; then
|
||
if sed -i "s|/dev/video0|/dev/video$VIDEONUM|g" /etc/kvmd/override.yaml && \
|
||
sed -i "s|/dev/video0|/dev/video$VIDEONUM|g" /etc/kvmd/janus/janus.plugin.ustreamer.jcfg; then
|
||
log_info "视频设备已设置为 /dev/video$VIDEONUM"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$AUDIONUM" ]; then
|
||
if sed -i "s/hw:0/hw:$AUDIONUM/g" /etc/kvmd/janus/janus.plugin.ustreamer.jcfg; then
|
||
log_info "音频设备已设置为 hw:$AUDIONUM"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$CH9329SPEED" ]; then
|
||
if sed -i "s/speed: 9600/speed: $CH9329SPEED/g" /etc/kvmd/override.yaml; then
|
||
log_info "CH9329 串口速率已设置为 $CH9329SPEED"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$CH9329NUM" ]; then
|
||
if sed -i "s|/dev/ttyUSB0|/dev/ttyUSB$CH9329NUM|g" /etc/kvmd/override.yaml; then
|
||
log_info "CH9329 串口设备已设置为 $CH9329NUM"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$CH9329TIMEOUT" ]; then
|
||
if sed -i "s/read_timeout: 0.3/read_timeout: $CH9329TIMEOUT/g" /etc/kvmd/override.yaml; then
|
||
log_info "CH9329 超时已设置为 $CH9329TIMEOUT 秒"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$H264PRESET" ]; then
|
||
if sed -i "s/ultrafast/$H264PRESET/g" /etc/kvmd/override.yaml; then
|
||
log_info "H264 预设已设置为 $H264PRESET"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$VIDEOFORMAT" ]; then
|
||
if sed -i "s/--format=mjpeg/--format=$VIDEOFORMAT/g" /etc/kvmd/override.yaml; then
|
||
log_info "视频输入格式已设置为 $VIDEOFORMAT"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$HWENCODER" ]; then
|
||
if sed -i "s/--h264-hwenc=disabled/--h264-hwenc=$HWENCODER/g" /etc/kvmd/override.yaml; then
|
||
log_info "硬件编码器已设置为 $HWENCODER"
|
||
fi
|
||
fi
|
||
|
||
# 设置WEB端口
|
||
if [ ! -z "$HTTPPORT" ]; then
|
||
if sed -i "s/port: 8080/port: $HTTPPORT/g" /etc/kvmd/override.yaml; then
|
||
log_info "HTTP 端口已设置为 $HTTPPORT"
|
||
fi
|
||
fi
|
||
|
||
if [ ! -z "$HTTPSPORT" ]; then
|
||
if sed -i "s/port: 4430/port: $HTTPSPORT/g" /etc/kvmd/override.yaml; then
|
||
log_info "HTTPS 端口已设置为 $HTTPSPORT"
|
||
fi
|
||
fi
|
||
|
||
|
||
touch /etc/kvmd/.init_flag
|
||
log_info "初始化配置完成"
|
||
fi
|
||
|
||
# OTG设备配置
|
||
if [ "$OTG" == "1" ]; then
|
||
log_info "正在配置 OTG 设备..."
|
||
rm -r /run/kvmd/otg &> /dev/null
|
||
|
||
if ! modprobe libcomposite; then
|
||
log_error "加载 libcomposite 模块失败"
|
||
exit 1
|
||
fi
|
||
|
||
if python -m kvmd.apps.otg start; then
|
||
ln -s /dev/hidg1 /dev/kvmd-hid-mouse
|
||
ln -s /dev/hidg0 /dev/kvmd-hid-keyboard
|
||
ln -s /dev/hidg2 /dev/kvmd-hid-mouse-alt
|
||
log_info "OTG 设备配置完成"
|
||
else
|
||
log_warn "OTG 设备挂载失败"
|
||
#exit 1
|
||
fi
|
||
fi
|
||
|
||
log_info "One-KVM 配置文件准备完成，正在启动服务..."
|
||
exec supervisord -c /etc/kvmd/supervisord.conf
|