1check_dependencies()
2{
3    # Check if qemu is installed
4    if [ -z "$(which qemu-system-x86_64)" ]; then
5        echo "Please install qemu first!"
6        exit 1
7    fi
8
9    # Check if brctl is installed
10    if [ -z "$(which brctl)" ]; then
11        echo "Please install bridge-utils first!"
12        exit 1
13    fi
14
15    # Check if dnsmasq is installed
16    if [ -z "$(which dnsmasq)" ]; then
17        echo "Please install dnsmasq first!"
18        exit 1
19    fi
20
21    # Check if iptable is installed
22    if [ -z "$(which iptables)" ]; then
23        echo "Please install iptables first!"
24        exit 1
25    fi
26
27}
28
29check_dependencies
30
31# 进行启动前检查
32flag_can_run=1
33ARGS=`getopt -o p -l bios:,display: -- "$@"`
34eval set -- "${ARGS}"
35echo "$@"
36allflags=
37# allflags=$(qemu-system-x86_64 -cpu help | awk '/flags/ {y=1; getline}; y {print}' | tr ' ' '\n' | grep -Ev "^$" | sed -r 's|^|+|' | tr '\n' ',' | sed -r "s|,$||")
38# 设置ARCH环境变量，如果没有设置，就默认为x86_64
39export ARCH=${ARCH:=x86_64}
40echo "ARCH=${ARCH}"
41#ARCH="i386"
42# 请根据自己的需要，在-d 后方加入所需的 trace 事件
43
44# 标准的trace events
45qemu_trace_std=cpu_reset,guest_errors,trace:virtio*,trace:e1000e_rx*,trace:e1000e_tx*,trace:e1000e_irq*
46# 调试usb的trace
47qemu_trace_usb=trace:usb_xhci_reset,trace:usb_xhci_run,trace:usb_xhci_stop,trace:usb_xhci_irq_msi,trace:usb_xhci_irq_msix,trace:usb_xhci_port_reset,trace:msix_write_config,trace:usb_xhci_irq_msix,trace:usb_xhci_irq_msix_use,trace:usb_xhci_irq_msix_unuse,trace:usb_xhci_irq_msi,trace:usb_xhci_*
48
49# 根据架构设置qemu的加速方式
50if [ ${ARCH} == "i386" ] || [ ${ARCH} == "x86_64" ]; then
51    qemu_accel="kvm"
52    if [ $(uname) == Darwin ]; then
53        qemu_accel=hvf
54    fi
55fi
56
57# uboot版本
58UBOOT_VERSION="v2023.10"
59RISCV64_UBOOT_PATH="arch/riscv64/u-boot-${UBOOT_VERSION}-riscv64"
60
61
62DISK_NAME="disk-${ARCH}.img"
63
64QEMU=qemu-system-${ARCH}
65QEMU_DISK_IMAGE="../bin/${DISK_NAME}"
66QEMU_MEMORY="512M"
67QEMU_MEMORY_BACKEND="dragonos-qemu-shm.ram"
68QEMU_MEMORY_BACKEND_PATH_PREFIX="/dev/shm"
69QEMU_SHM_OBJECT="-object memory-backend-file,size=${QEMU_MEMORY},id=${QEMU_MEMORY_BACKEND},mem-path=${QEMU_MEMORY_BACKEND_PATH_PREFIX}/${QEMU_MEMORY_BACKEND},share=on "
70QEMU_SMP="2,cores=2,threads=1,sockets=1"
71QEMU_MONITOR="-monitor stdio"
72QEMU_TRACE="${qemu_trace_std}"
73QEMU_CPU_FEATURES=""
74QEMU_RTC_CLOCK=""
75QEMU_SERIAL="-serial file:../serial_opt.txt"
76QEMU_DRIVE="id=disk,file=${QEMU_DISK_IMAGE},if=none"
77QEMU_ACCELARATE=""
78QEMU_ARGUMENT=""
79
80# 如果qemu_accel不为空
81if [ -n "${qemu_accel}" ]; then
82    QEMU_ACCELARATE="-machine accel=${qemu_accel} -enable-kvm "
83fi
84
85if [ ${ARCH} == "i386" ] || [ ${ARCH} == "x86_64" ]; then
86    QEMU_MACHINE=" -machine q35,memory-backend=${QEMU_MEMORY_BACKEND} "
87    QEMU_CPU_FEATURES+="-cpu IvyBridge,apic,x2apic,+fpu,check,+vmx,${allflags}"
88    QEMU_RTC_CLOCK+=" -rtc clock=host,base=localtime"
89else
90    QEMU_MACHINE=" -machine virt,memory-backend=${QEMU_MEMORY_BACKEND} -cpu sifive-u54 "
91fi
92
93if [ ${ARCH} == "riscv64" ]; then
94# 如果是riscv64架构，就不需要图形界面
95    QEMU_ARGUMENT+=" --nographic "
96    # 从控制台显示
97    QEMU_MONITOR=""
98    QEMU_SERIAL=""
99fi
100
101
102# ps: 下面这条使用tap的方式，无法dhcp获取到ip，暂时不知道为什么
103# QEMU_DEVICES="-device ahci,id=ahci -device ide-hd,drive=disk,bus=ahci.0 -net nic,netdev=nic0 -netdev tap,id=nic0,model=virtio-net-pci,script=qemu/ifup-nat,downscript=qemu/ifdown-nat -usb -device qemu-xhci,id=xhci,p2=8,p3=4 "
104QEMU_DEVICES="-device ahci,id=ahci -device ide-hd,drive=disk,bus=ahci.0 -netdev user,id=hostnet0,hostfwd=tcp::12580-:12580 -device virtio-net-pci,vectors=5,netdev=hostnet0,id=net0 -usb -device qemu-xhci,id=xhci,p2=8,p3=4 "
105# E1000E
106# QEMU_DEVICES="-device ahci,id=ahci -device ide-hd,drive=disk,bus=ahci.0 -netdev user,id=hostnet0,hostfwd=tcp::12580-:12580 -net nic,model=e1000e,netdev=hostnet0,id=net0 -netdev user,id=hostnet1,hostfwd=tcp::12581-:12581 -device virtio-net-pci,vectors=5,netdev=hostnet1,id=net1 -usb -device qemu-xhci,id=xhci,p2=8,p3=4 "
107QEMU_ARGUMENT+="-d ${QEMU_DISK_IMAGE} -m ${QEMU_MEMORY} -smp ${QEMU_SMP} -boot order=d ${QEMU_MONITOR} -d ${qemu_trace_std} "
108
109QEMU_ARGUMENT+="-s ${QEMU_MACHINE} ${QEMU_CPU_FEATURES} ${QEMU_RTC_CLOCK} ${QEMU_SERIAL} -drive ${QEMU_DRIVE} ${QEMU_DEVICES} "
110QEMU_ARGUMENT+=" ${QEMU_SHM_OBJECT} "
111QEMU_ARGUMENT+=" ${QEMU_ACCELARATE} "
112
113
114
115# 安装riscv64的uboot
116install_riscv_uboot()
117{
118
119    if [ ! -d ${RISCV64_UBOOT_PATH} ]; then
120        echo "正在下载u-boot..."
121        uboot_tar_name="u-boot-${UBOOT_VERSION}-riscv64.tar.xz"
122
123        uboot_parent_path=$(dirname ${RISCV64_UBOOT_PATH}) || (echo "获取riscv u-boot 版本 ${UBOOT_VERSION} 的父目录失败" && exit 1)
124
125        if [ ! -f ${uboot_tar_name} ]; then
126            wget https://mirrors.dragonos.org.cn/pub/third_party/u-boot/${uboot_tar_name} || (echo "下载riscv u-boot 版本 ${UBOOT_VERSION} 失败" && exit 1)
127        fi
128        echo "下载完成"
129        echo "正在解压u-boot到 '$uboot_parent_path'..."
130        mkdir -p $uboot_parent_path
131        tar xvf u-boot-${UBOOT_VERSION}-riscv64.tar.xz -C ${uboot_parent_path} || (echo "解压riscv u-boot 版本 ${UBOOT_VERSION} 失败" && exit 1)
132        echo "解压完成"
133        rm -rf u-boot-${UBOOT_VERSION}-riscv64.tar.xz
134    fi
135    echo "riscv u-boot 版本 ${UBOOT_VERSION} 已经安装"
136}
137
138
139if [ $flag_can_run -eq 1 ]; then
140  while true;do
141    case "$1" in
142        --bios)
143        case "$2" in
144              uefi) #uefi启动新增ovmf.fd固件
145              BIOS_TYPE=uefi
146            ;;
147              legacy)
148              BIOS_TYPE=legacy
149              ;;
150        esac;shift 2;;
151        --display)
152        case "$2" in
153              vnc)
154              QEMU_ARGUMENT+=" -display vnc=:00"
155              ;;
156              window)
157              ;;
158        esac;shift 2;;
159        *) break
160      esac
161  done
162
163# 删除共享内存
164sudo rm -rf ${QEMU_MEMORY_BACKEND_PATH_PREFIX}/${QEMU_MEMORY_BACKEND}
165
166if [ ${BIOS_TYPE} == uefi ] ;then
167  if [ ${ARCH} == x86_64 ] ;then
168    sudo ${QEMU} -bios arch/x86_64/efi/OVMF-pure-efi.fd ${QEMU_ARGUMENT}
169  elif [ ${ARCH} == i386 ] ;then
170    sudo ${QEMU} -bios arch/i386/efi/OVMF-pure-efi.fd ${QEMU_ARGUMENT}
171  elif [ ${ARCH} == riscv64 ] ;then
172    install_riscv_uboot
173    sudo ${QEMU} -kernel ${RISCV64_UBOOT_PATH}/u-boot.bin ${QEMU_ARGUMENT}
174  else
175    echo "不支持的架构: ${ARCH}"
176  fi
177else
178  # 如果是i386架构或者x86_64架构，就直接启动
179  if [ ${ARCH} == x86_64 ] || [ ${ARCH} == i386 ] ;then
180    sudo ${QEMU} ${QEMU_ARGUMENT}
181  elif [ ${ARCH} == riscv64 ] ;then
182    # 如果是riscv64架构，就与efi启动一样
183    install_riscv_uboot
184    sudo ${QEMU} -kernel ${RISCV64_UBOOT_PATH}/u-boot.bin ${QEMU_ARGUMENT}
185  else
186    echo "不支持的架构: ${ARCH}"
187  fi
188fi
189
190# 删除共享内存
191sudo rm -rf ${QEMU_MEMORY_BACKEND_PATH_PREFIX}/${QEMU_MEMORY_BACKEND}
192else
193  echo "不满足运行条件"
194fi
195