86 lines
2.2 KiB
Python
Executable File
86 lines
2.2 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
import subprocess
|
|
import os
|
|
|
|
color = {
|
|
'black': '\033[30m',
|
|
'red': '\033[31m',
|
|
'green': '\033[32m',
|
|
'yellow': '\033[33m',
|
|
'blue': '\033[34m',
|
|
'pink': '\033[35m',
|
|
'cyan': '\033[36m',
|
|
'white': '\033[37m',
|
|
'lblack': '\033[1;30m',
|
|
'lred': '\033[1;31m',
|
|
'lgreen': '\033[1;32m',
|
|
'lyellow': '\033[1;33m',
|
|
'lblue': '\033[1;34m',
|
|
'lpink': '\033[1;35m',
|
|
'lcyan': '\033[1;36m',
|
|
'lwhite': '\033[1;37m',
|
|
'reset': '\033[0m',
|
|
}
|
|
|
|
arch = subprocess.check_output("uname -m", shell=True).decode().strip()
|
|
|
|
|
|
kernel_modules = [
|
|
"nbd",
|
|
]
|
|
|
|
softwares = {
|
|
"doas": ["doas"],
|
|
"gcc": ["gcc"],
|
|
"nasm": ["nasm"],
|
|
"rust": ["rustup", "rustc", "rustdoc", "rust-lldb"],
|
|
"qemu": ["qemu-img", "qemu-nbd", "qemu-system-"+arch],
|
|
}
|
|
|
|
|
|
# 检查内核模块
|
|
lsmodoutput = subprocess.check_output("lsmod", shell=True).decode()
|
|
print("Checking kernel modules ...")
|
|
for mod in kernel_modules:
|
|
mmod = "\n" + mod
|
|
mmod += " "
|
|
if mmod in lsmodoutput:
|
|
print(" " + color['lpink'] + f"{mod}" + color['reset'] + " exists.")
|
|
else:
|
|
print(" " + color['lred'] + f"{mod}" + color['reset'] + " not found.")
|
|
print(
|
|
"Kernel module " +
|
|
color['lred'] + f"{mod}" + color['reset'] + " is not installed.")
|
|
exit(-1)
|
|
|
|
# 检查软件依赖
|
|
pathsenv = os.environ.get("PATH").split(":")
|
|
for software, progs in softwares.items():
|
|
print(
|
|
"Checking " + color['lcyan'] +
|
|
f"{software}" + color['reset'] + " ...")
|
|
for program in progs:
|
|
exists = False
|
|
for path in pathsenv:
|
|
if os.path.isfile(os.path.join(path, program)):
|
|
exists = True
|
|
break
|
|
if exists:
|
|
print(
|
|
" " + color['lyellow'] +
|
|
f"{program}" + color['reset'] + " existed.")
|
|
else:
|
|
print(
|
|
" " + color['lred'] +
|
|
f"{program}" + color['reset'] + " not found.")
|
|
|
|
print(
|
|
"Software " + color['lred'] + f"{software}" +
|
|
color['reset'] + " is not installed or completed.")
|
|
exit(-2)
|
|
|
|
print(
|
|
"All dependencies are " +
|
|
color['lgreen'] + "satisfied." + color['reset'])
|