mirror of
https://github.com/thelittlerocket/pve.git
synced 2024-11-14 19:16:19 +08:00
216 lines
7.8 KiB
Bash
216 lines
7.8 KiB
Bash
#!/bin/bash
|
||
# from
|
||
# https://github.com/spiritLHLS/pve
|
||
# 2023.07.31
|
||
|
||
|
||
# ./buildvm.sh VMID 用户名 密码 CPU核数 内存 硬盘 SSH端口 80端口 443端口 外网端口起 外网端口止 系统 存储盘
|
||
# ./buildvm.sh 102 test1 1234567 1 512 5 40001 40002 40003 50000 50025 debian11 local
|
||
|
||
cd /root >/dev/null 2>&1
|
||
# 创建NAT的虚拟机
|
||
vm_num="${1:-102}"
|
||
user="${2:-test}"
|
||
password="${3:-123456}"
|
||
core="${4:-1}"
|
||
memory="${5:-512}"
|
||
disk="${6:-5}"
|
||
sshn="${7:-40001}"
|
||
web1_port="${8:-40002}"
|
||
web2_port="${9:-40003}"
|
||
port_first="${10:-49975}"
|
||
port_last="${11:-50000}"
|
||
system="${12:-ubuntu22}"
|
||
storage="${13:-local}"
|
||
# in="${12:-300}"
|
||
# out="${13:-300}"
|
||
rm -rf "vm$name"
|
||
|
||
_red() { echo -e "\033[31m\033[01m$@\033[0m"; }
|
||
_green() { echo -e "\033[32m\033[01m$@\033[0m"; }
|
||
_yellow() { echo -e "\033[33m\033[01m$@\033[0m"; }
|
||
_blue() { echo -e "\033[36m\033[01m$@\033[0m"; }
|
||
reading(){ read -rp "$(_green "$1")" "$2"; }
|
||
utf8_locale=$(locale -a 2>/dev/null | grep -i -m 1 -E "utf8|UTF-8")
|
||
if [[ -z "$utf8_locale" ]]; then
|
||
_yellow "No UTF-8 locale found"
|
||
else
|
||
export LC_ALL="$utf8_locale"
|
||
export LANG="$utf8_locale"
|
||
export LANGUAGE="$utf8_locale"
|
||
_green "Locale set to $utf8_locale"
|
||
fi
|
||
|
||
get_system_arch() {
|
||
local sysarch="$(uname -m)"
|
||
if [ "${sysarch}" = "unknown" ] || [ "${sysarch}" = "" ]; then
|
||
local sysarch="$(arch)"
|
||
fi
|
||
# 根据架构信息设置系统位数并下载文件,其余 * 包括了 x86_64
|
||
case "${sysarch}" in
|
||
"i386" | "i686" | "x86_64")
|
||
system_arch="x86"
|
||
;;
|
||
"armv7l" | "armv8" | "armv8l" | "aarch64")
|
||
system_arch="arch"
|
||
;;
|
||
*)
|
||
system_arch=""
|
||
;;
|
||
esac
|
||
}
|
||
|
||
check_cdn() {
|
||
local o_url=$1
|
||
for cdn_url in "${cdn_urls[@]}"; do
|
||
if curl -sL -k "$cdn_url$o_url" --max-time 6 | grep -q "success" > /dev/null 2>&1; then
|
||
export cdn_success_url="$cdn_url"
|
||
return
|
||
fi
|
||
sleep 0.5
|
||
done
|
||
export cdn_success_url=""
|
||
}
|
||
|
||
check_cdn_file() {
|
||
check_cdn "https://raw.githubusercontent.com/spiritLHLS/ecs/main/back/test"
|
||
if [ -n "$cdn_success_url" ]; then
|
||
_yellow "CDN available, using CDN"
|
||
else
|
||
_yellow "No CDN available, no use CDN"
|
||
fi
|
||
}
|
||
|
||
cdn_urls=("https://cdn.spiritlhl.workers.dev/" "https://cdn3.spiritlhl.net/" "https://cdn1.spiritlhl.net/" "https://ghproxy.com/" "https://cdn2.spiritlhl.net/")
|
||
if [ ! -d "qcow" ]; then
|
||
mkdir qcow
|
||
fi
|
||
get_system_arch
|
||
if [ -z "${system_arch}" ] || [ ! -v system_arch ]; then
|
||
_red "This script can only run on machines under x86_64 or arm architecture."
|
||
exit 1
|
||
fi
|
||
if [ "$system_arch" = "x86" ]; then
|
||
systems=("debian10" "debian11" "debian9" "ubuntu18" "ubuntu20" "ubuntu22" "archlinux" "centos9-stream" "centos8-stream" "almalinux8" "almalinux9" "fedora33" "fedora34" "opensuse-leap-15")
|
||
for sys in ${systems[@]}; do
|
||
if [[ "$system" == "$sys" ]]; then
|
||
file_path="/root/qcow/${system}.qcow2"
|
||
break
|
||
fi
|
||
done
|
||
if [[ -z "$file_path" ]]; then
|
||
_red "Unable to install corresponding system, please check https://github.com/spiritLHLS/Images/ for supported system images "
|
||
_red "无法安装对应系统，请查看 https://github.com/spiritLHLS/Images/ 支持的系统镜像 "
|
||
exit 1
|
||
fi
|
||
if [ ! -f "$file_path" ]; then
|
||
# v1.0 基础安装包预安装
|
||
# v1.1 增加agent安装包预安装，方便在宿主机上看到虚拟机的进程
|
||
check_cdn_file
|
||
url="${cdn_success_url}https://github.com/spiritLHLS/Images/releases/download/v1.1/${system}.qcow2"
|
||
curl -L -o "$file_path" "$url"
|
||
fi
|
||
elif [ "$system_arch" = "arch" ]; then
|
||
systems=("ubuntu14" "ubuntu16" "ubuntu18" "ubuntu20" "ubuntu22")
|
||
for sys in ${systems[@]}; do
|
||
if [[ "$system" == "$sys" ]]; then
|
||
file_path="/root/qcow/${system}.img"
|
||
break
|
||
fi
|
||
done
|
||
if [[ -z "$file_path" ]]; then
|
||
# https://www.debian.org/mirror/list
|
||
_red "Unable to install corresponding system, please check http://cloud-images.ubuntu.com for supported system images "
|
||
_red "无法安装对应系统，请查看 http://cloud-images.ubuntu.com 支持的系统镜像 "
|
||
exit 1
|
||
fi
|
||
if [ -n "$file_path" ] && [ -f "$file_path" ]; then
|
||
case "$system" in
|
||
ubuntu14)
|
||
version="trusty"
|
||
;;
|
||
ubuntu16)
|
||
version="xenial"
|
||
;;
|
||
ubuntu18)
|
||
version="bionic"
|
||
;;
|
||
ubuntu20)
|
||
version="focal"
|
||
;;
|
||
ubuntu22)
|
||
version="jammy"
|
||
;;
|
||
*)
|
||
echo "Unsupported Ubuntu version."
|
||
exit 1
|
||
;;
|
||
esac
|
||
url="http://cloud-images.ubuntu.com/${version}/current/${version}-server-cloudimg-arm64.img"
|
||
curl -L -o "$file_path" "$url"
|
||
fi
|
||
fi
|
||
|
||
first_digit=${vm_num:0:1}
|
||
second_digit=${vm_num:1:1}
|
||
third_digit=${vm_num:2:1}
|
||
if [ $first_digit -le 2 ]; then
|
||
if [ $second_digit -eq 0 ]; then
|
||
num=$third_digit
|
||
else
|
||
num=$second_digit$third_digit
|
||
fi
|
||
else
|
||
num=$((first_digit - 2))$second_digit$third_digit
|
||
fi
|
||
|
||
qm create $vm_num --agent 1 --scsihw virtio-scsi-single --serial0 socket --cores $core --sockets 1 --cpu host --net0 virtio,bridge=vmbr1,firewall=0
|
||
if [ "$system_arch" = "x86" ]; then
|
||
qm importdisk $vm_num /root/qcow/${system}.qcow2 ${storage}
|
||
qm set $vm_num --scsihw virtio-scsi-pci --scsi0 ${storage}:${vm_num}/vm-${vm_num}-disk-0.raw
|
||
else
|
||
qm set $vm_num --bios ovmf
|
||
qm importdisk $vm_num /root/qcow/${system}.img ${storage}
|
||
qm set $vm_num --scsihw virtio-scsi-pci --scsi0 ${storage}:${vm_num}/vm-${vm_num}-disk-0.raw
|
||
fi
|
||
qm set $vm_num --bootdisk scsi0
|
||
qm set $vm_num --boot order=scsi0
|
||
qm set $vm_num --memory $memory
|
||
# --swap 256
|
||
qm set $vm_num --ide2 ${storage}:cloudinit
|
||
qm set $vm_num --nameserver 8.8.8.8
|
||
qm set $vm_num --searchdomain 8.8.4.4
|
||
user_ip="172.16.1.${num}"
|
||
qm set $vm_num --ipconfig0 ip=${user_ip}/24,gw=172.16.1.1
|
||
qm set $vm_num --cipassword $password --ciuser $user
|
||
# qm set $vm_num --agent 1
|
||
qm resize $vm_num scsi0 ${disk}G
|
||
qm start $vm_num
|
||
|
||
iptables -t nat -A PREROUTING -p tcp --dport ${sshn} -j DNAT --to-destination ${user_ip}:22
|
||
iptables -t nat -A PREROUTING -p tcp -m tcp --dport ${web1_port} -j DNAT --to-destination ${user_ip}:80
|
||
iptables -t nat -A PREROUTING -p tcp -m tcp --dport ${web2_port} -j DNAT --to-destination ${user_ip}:443
|
||
iptables -t nat -A PREROUTING -p tcp -m tcp --dport ${port_first}:${port_last} -j DNAT --to-destination ${user_ip}:${port_first}-${port_last}
|
||
iptables -t nat -A PREROUTING -p udp -m udp --dport ${port_first}:${port_last} -j DNAT --to-destination ${user_ip}:${port_first}-${port_last}
|
||
if [ ! -f "/etc/iptables/rules.v4" ]; then
|
||
touch /etc/iptables/rules.v4
|
||
fi
|
||
iptables-save > /etc/iptables/rules.v4
|
||
service netfilter-persistent restart
|
||
echo "$vm_num $user $password $core $memory $disk $sshn $web1_port $web2_port $port_first $port_last $system $storage" >> "vm${vm_num}"
|
||
# 虚拟机的相关信息将会存储到对应的虚拟机的NOTE中，可在WEB端查看
|
||
data=$(echo " VMID 用户名-username 密码-password CPU核数-CPU 内存-memory 硬盘-disk SSH端口 80端口 443端口 外网端口起-port-start 外网端口止-port-end 系统-system 存储盘-storage")
|
||
values=$(cat "vm${vm_num}")
|
||
IFS=' ' read -ra data_array <<< "$data"
|
||
IFS=' ' read -ra values_array <<< "$values"
|
||
length=${#data_array[@]}
|
||
for ((i=0; i<$length; i++))
|
||
do
|
||
echo "${data_array[$i]} ${values_array[$i]}"
|
||
echo ""
|
||
done > "/tmp/temp${vm_num}.txt"
|
||
sed -i 's/^/# /' "/tmp/temp${vm_num}.txt"
|
||
cat "/etc/pve/qemu-server/${vm_num}.conf" >> "/tmp/temp${vm_num}.txt"
|
||
cp "/tmp/temp${vm_num}.txt" "/etc/pve/qemu-server/${vm_num}.conf"
|
||
rm -rf "/tmp/temp${vm_num}.txt"
|
||
cat "vm${vm_num}"
|