Compare commits

..

15 Commits

Author SHA1 Message Date
shenmo7192 523c9f6d9e build: 更新版本号至1.2.4 2026-04-05 12:01:38 +08:00
shenmo7192 dc7916598e refactor(desktop-fix): 重构桌面修复工具并支持多发行版
- 将工具重命名为更通用的 amber-pm-desktop-fix
- 添加对不同发行版的支持,包括银河麒麟系统
- 实现旧链接自动迁移功能
- 改进目录创建和符号链接处理逻辑
2026-04-05 12:01:16 +08:00
shenmo7192 bf996af2c0 fix(desktop): 修复非麒麟系统下桌面链接迁移问题
添加对旧版桌面链接的迁移处理,确保非麒麟系统中应用程序链接正确指向新位置
更新桌面数据库以反映变更
2026-04-05 11:58:58 +08:00
shenmo7192 2ddfa59dee chore: drop掉fish的支持 2026-04-01 14:36:27 +08:00
shenmo7192 9d23672441 优先使用/usr/share/fonts
Signed-off-by: shenmo <jifengshenmo@outlook.com>
2026-03-30 02:49:33 +00:00
shenmo7192 8d91cd2c8f 修复icon themes 直通
Signed-off-by: shenmo <jifengshenmo@outlook.com>
2026-03-29 16:11:20 +00:00
shenmo7192 fec4cb7aeb bump ver to 123 2026-03-29 23:26:11 +08:00
shenmo7192 5eb10c1854 恢复XDG_DATA_DIRS 2026-03-29 23:21:43 +08:00
shenmo7192 c7c0e823b8 新增 spk 直通
Signed-off-by: shenmo <jifengshenmo@outlook.com>
2026-03-29 12:24:29 +00:00
shenmo7192 e37742dd6a feat(shell-completion): 为 apm 添加 fish 和 zsh 的自动补全支持
添加 fish 和 zsh 的自动补全脚本,支持所有 apm 命令及其选项
2026-03-29 13:34:10 +08:00
shenmo7192 586f7d7ad3 docs: 更新README命令说明并添加AGENTS文档
更新README.md中的命令列表,简化并重新组织命令显示方式,添加--help-all选项说明
添加AGENTS.md文档,详细描述APM项目中使用的代理和助手工具的功能和使用方法
2026-03-29 13:24:55 +08:00
shenmo7192 7347294443 feat(apm): 添加ssinstall命令支持并更新帮助信息
添加ssinstall命令用于本地软件安装,同时更新帮助信息显示完整选项
更新版本号至1.2.2并优化彩蛋函数位置
2026-03-29 13:22:28 +08:00
shenmo7192 1414b9d919 修复 Steam 上无法使用本地cursor,降低XDG_DATA_DIRS优先级以修复一些错误 2026-03-29 02:49:33 +08:00
shenmo7192 437bc35f21 bump version to 1.2.1 2026-03-29 01:39:25 +08:00
shenmo7192 69509d7a03 refactor(amber-ce-configure-nvidia): 简化NVIDIA驱动配置脚本并优化链接逻辑
移除冗余代码,简化驱动版本检测和库文件链接流程
使用ldconfig直接获取库文件路径,提高效率
保留核心功能并优化日志输出
2026-03-29 01:34:22 +08:00
15 changed files with 699 additions and 835 deletions
+199
View File
@@ -0,0 +1,199 @@
# APM 代理与助手(AGENTS
本文档描述了 APM 项目中使用的代理和助手工具,它们用于增强 APM 的功能和用户体验。
## 1. 核心代理工具
### 1.1 amber-pm-debug
**功能**:用于在 APM 环境中执行命令,提供调试和诊断功能。
**使用场景**
- 执行 apt 相关命令
- 执行 dpkg 相关命令
- 进入调试环境
**示例**
```bash
amber-pm-debug apt update
amber-pm-debug dpkg --configure -a
```
### 1.2 amber-pm-app-launcher
**功能**:通过应用启动器启动 APM 软件包。
**使用场景**
- 启动已安装的 APM 应用
- 传递参数给应用
**示例**
```bash
amber-pm-app-launcher firefox
amber-pm-app-launcher gedit --new-document
```
### 1.3 amber-pm-configure-nvidia
**功能**:配置 NVIDIA 驱动支持。
**使用场景**
- 自动从主机获取 NVIDIA 驱动文件
- 为 APM 应用提供 GPU 加速支持
**示例**
```bash
amber-pm-configure-nvidia /path/to/ace-env
```
## 2. 构建与转换工具
### 2.1 amber-pm-convert
**功能**:将普通 Deb 包转换为 APM 软件包。
**使用场景**
- 转换第三方 Deb 包为 APM 格式
- 自定义包名和版本号
**示例**
```bash
amber-pm-convert --base amber-pm-trixie /path/to/package.deb
amber-pm-convert --base amber-pm-bookworm-spark-wine /path/to/package.deb --pkgname new-pkg --version 1.0.0
```
### 2.2 amber-pm-dstore-patch
**功能**:修补应用商店相关配置。
**使用场景**
- 安装或更新软件包后自动执行
- 确保应用商店配置正确
### 2.3 amber-pm-desktop-fix
**功能**:修复桌面环境相关问题,维护应用程序和图标链接。
**使用场景**
- 安装或移除软件包后自动执行
- 确保桌面环境正常运行
- 自动迁移旧版链接到新目录结构
**说明**
- 银河麒麟系统:链接到 `/usr/share/applications/`
- 其他发行版:链接到 `/usr/local/share/applications/apm/`
## 3. 沙箱与安全工具
### 3.1 APM_USE_SANDBOX
**功能**:启用主目录沙箱化。
**使用场景**
- 运行不受信任的应用
- 保护用户主目录
**示例**
```bash
apm sandbox-run firefox
```
### 3.2 APM_USE_BWRAP
**功能**:使用 bwrap 进行额外的隔离。
**使用场景**
- 需要更强隔离性的应用
- 增强安全性
**示例**
```bash
apm bwrap-run firefox
```
## 4. 本地安装工具
### 4.1 ssinstall
**功能**:使用 ssinstall 进行本地软件安装。
**使用场景**
- 安装本地软件包
- 与 spark-store 集成
**示例**
```bash
apm ssinstall /path/to/package
```
### 4.2 ssaudit
**功能**:使用 ssaudit 进行本地软件安装。
**使用场景**
- 安装本地软件包并进行审计
- 与 spark-store 集成
**示例**
```bash
apm ssaudit /path/to/package
```
## 5. 环境变量
### 5.1 APM_PKG_NAME
**功能**:指定当前运行的包名。
**使用场景**
- 在脚本中识别当前包
- 为应用提供包信息
### 5.2 PATH_PREFIX
**功能**:指定 APM 基础路径。
**使用场景**
- 自定义 APM 安装位置
- 多环境管理
## 6. 工作原理
APM 代理和助手工具通过以下方式工作:
1. **环境隔离**:使用 fuse-overlayfs 创建隔离的文件系统环境
2. **命令转发**:将用户命令转发到适当的环境中执行
3. **资源共享**:从主机系统获取必要的资源(如 NVIDIA 驱动)
4. **安全增强**:提供沙箱和隔离机制
5. **用户体验**:简化应用的安装和运行过程
## 7. 故障排除
### 7.1 常见问题
- **NVIDIA 驱动问题**:确保主机已安装 NVIDIA 驱动,APM 会自动检测并使用
- **沙箱权限**:确保用户有足够的权限创建和管理沙箱
- **包依赖**:使用 `apm show <package>` 查看包依赖,确保所有依赖已安装
### 7.2 调试命令
```bash
# 查看调试信息
apm debug
# 检查包状态
amber-pm-debug dpkg -l | grep <package>
# 检查 NVIDIA 配置
apm-nvidia-toggle
```
## 8. 扩展与定制
APM 代理系统设计为可扩展的,您可以:
1. **添加自定义代理**:在 `src/var/lib/apm/apm/files/ace-env/usr/bin/` 目录添加新的代理脚本
2. **修改现有代理**:根据需要调整现有代理的行为
3. **创建自定义基础环境**:使用 `amber-pm-convert` 工具创建基于特定需求的基础环境
通过这些工具和技术,APM 提供了一个灵活、安全、高效的软件包管理系统,适用于各种 Linux 发行版。
+6 -10
View File
@@ -37,32 +37,28 @@ Usage:
Commands:
install 安装软件包
remove 卸载软件包
run <package> 运行指定软件包的可执行文件
sandbox-run <package> 运行指定软件包的可执行文件(主目录沙箱化
launch <package> [args...] 启动软件包(通过应用启动器)
run <package> [EXEC_PATH] [args...] 运行指定软件包的可执行文件(可指定容器内路径
update 更新软件包信息
hold 锁定软件包版本
unhold 解锁软件包版本
full-upgrade 升级全部软件包
list 查看可用软件包信息
search 搜索软件包
download 下载包
show 展示包信息
clean 清除缓存软件包
autoremove 自动移除不需要的包
ssaudit <path> 使用 ssaudit 进行本地软件安装,详情见 spark-store
debug 显示调试系统信息并进入调试环境
amber 彩蛋功能
xmp360 彩蛋功能
bronya 彩蛋功能
-h, --help 显示此帮助信息
--help-all 显示完整帮助信息
-v, --version 展示APM版本号
```
### 完整命令列表
使用 `apm --help-all` 查看完整的命令列表,包括高级命令如 `sandbox-run``bwrap-run``hold``unhold``full-upgrade``download``ssinstall``ssaudit``debug` 等。
+1 -1
View File
@@ -1 +1 @@
@VERSION@=1.2.0
@VERSION@=1.2.4
+1 -1
View File
@@ -3,7 +3,7 @@ Source: amber-ce
Version: @VERSION@
Architecture: amd64
Maintainer: shenmo <shenmo@spark-app.store>
Installed-Size: 49000
Installed-Size: 49388
Depends: zenity | garma,bubblewrap, flatpak, policykit-1 | pkexec | polkit-1 | polkit, systemd, procps,coreutils,fuse-overlayfs,xz-utils | xz,libnotify-bin,curl,xdg-user-dirs,bash
Recommends: dpkg, fakeroot, busybox
Section: misc
View File
View File
+98
View File
@@ -0,0 +1,98 @@
#!/bin/bash
# 确定目标目录
if grep -q "Kylin" /etc/os-release; then
TARGET_BASE="/usr/share"
APP_TARGET_DIR="$TARGET_BASE/applications"
echo "检测到麒麟系统,使用目标目录: $TARGET_BASE"
else
TARGET_BASE="/usr/local/share"
APP_TARGET_DIR="$TARGET_BASE/applications/apm"
echo "检测到非麒麟系统,使用目标目录: $APP_TARGET_DIR"
fi
function ensure_dir() {
local dir="$1"
# 检查目录是否为空
if [ -z "$dir" ]; then
echo "错误: 目录路径不能为空"
return 1
fi
# 检查目录是否存在
if [ ! -d "$dir" ]; then
echo "目录 '$dir' 不存在,正在创建..."
if mkdir -p "$dir"; then
echo "成功创建目录 '$dir'"
return 0
else
echo "错误: 无法创建目录 '$dir'"
return 1
fi
else
return 0
fi
}
# 函数:检查目录并创建符号链接
process_directory() {
local source_dir="$1"
local target_dir="$2"
local name="$3"
if [ -d "$source_dir" ] && [ -n "$(ls -A "$source_dir")" ]; then
ln -sv $source_dir/* "$target_dir" 2>/dev/null
find "$target_dir" -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink {} \; 2>/dev/null &
else
echo "$name directory is empty or does not exist, skipping..."
fi
}
# 使用动态确定的目标目录
ensure_dir "$APP_TARGET_DIR/"
ensure_dir "$TARGET_BASE/icons/"
# 处理 applications 目录
process_directory "/var/lib/apm/apm/files/ace-env/amber-ce-tools/data-dir/applications/" \
"$APP_TARGET_DIR/" "Applications"
# 处理 icons 目录
process_directory "/var/lib/apm/apm/files/ace-env/amber-ce-tools/data-dir/icons/" \
"$TARGET_BASE/icons/" "Icons"
# 等待所有后台任务完成
wait
# 迁移老链接(仅非麒麟系统需要)
if ! grep -q "Kylin" /etc/os-release; then
OLD_LINK_DIR="/usr/local/share/applications"
# 检查旧目录是否存在且不是apm子目录
if [ -d "$OLD_LINK_DIR" ] && [ "$OLD_LINK_DIR" != "$APP_TARGET_DIR" ]; then
echo "检查并迁移老链接..."
# 查找旧目录中指向APM数据目录的符号链接
find "$OLD_LINK_DIR" -maxdepth 1 -type l | while read -r link; do
target=$(readlink "$link")
# 如果链接指向APM的数据目录
if [[ "$target" == /var/lib/apm/apm/files/ace-env/amber-ce-tools/data-dir/applications/* ]]; then
filename=$(basename "$link")
# 如果新位置没有同名文件,则移动
if [ ! -e "$APP_TARGET_DIR/$filename" ]; then
echo "迁移老链接: $filename"
mv -v "$link" "$APP_TARGET_DIR/"
else
echo "新位置已存在 $filename,删除老链接"
rm -v "$link"
fi
fi
done
fi
fi
# 更新桌面数据库
if command -v update-desktop-database >/dev/null 2>&1; then
echo "正在更新桌面数据库..."
update-desktop-database "$TARGET_BASE/applications/"
else
echo "警告: update-desktop-database 命令未找到"
fi
-67
View File
@@ -1,67 +0,0 @@
#!/bin/bash
if [[ ! -e "/usr/share/gxde-api" ]] && ! grep -q "Kylin" /etc/os-release; then
exit 0 # No needed
fi
# 确定目标目录
if grep -q "Kylin" /etc/os-release; then
TARGET_BASE="/usr/share"
echo "检测到麒麟系统,使用目标目录: $TARGET_BASE"
else
TARGET_BASE="/usr/local/share"
echo "检测到GXDE环境,使用目标目录: $TARGET_BASE"
fi
function ensure_dir() {
local dir="$1"
# 检查目录是否为空
if [ -z "$dir" ]; then
echo "错误: 目录路径不能为空"
return 1
fi
# 检查目录是否存在
if [ ! -d "$dir" ]; then
echo "目录 '$dir' 不存在,正在创建..."
if mkdir -p "$dir"; then
echo "成功创建目录 '$dir'"
return 0
else
echo "错误: 无法创建目录 '$dir'"
return 1
fi
else
return 0
fi
}
# 函数:检查目录并创建符号链接
process_directory() {
local source_dir="$1"
local target_dir="$2"
local name="$3"
if [ -d "$source_dir" ] && [ -n "$(ls -A "$source_dir")" ]; then
ln -sv $source_dir/* "$target_dir" 2>/dev/null
find "$target_dir" -xtype l -exec echo '{} is invalid now and going to be cleaned' \; -exec unlink {} \; 2>/dev/null &
else
echo "$name directory is empty or does not exist, skipping..."
fi
}
# 使用动态确定的目标目录
ensure_dir "$TARGET_BASE/applications/"
ensure_dir "$TARGET_BASE/icons/"
# 处理 applications 目录
process_directory "/var/lib/apm/apm/files/ace-env/amber-ce-tools/data-dir/applications/" \
"$TARGET_BASE/applications/" "Applications"
# 处理 icons 目录
process_directory "/var/lib/apm/apm/files/ace-env/amber-ce-tools/data-dir/icons/" \
"$TARGET_BASE/icons/" "Icons"
# 等待所有后台任务完成
wait
+116 -64
View File
@@ -18,13 +18,105 @@ Usage:
$SCRIPT_NAME [COMMAND] [OPTIONS] [PACKAGES...]
Commands:
install 安装软件包
remove 卸载软件包
launch <package> [args...] 启动软件包(通过应用启动器)
run <package> [EXEC_PATH] [args...] 运行指定软件包的可执行文件(可指定容器内路径)
update 更新软件包信息
list 查看可用软件包信息
search 搜索软件包
show 展示包信息
clean 清除缓存软件包
autoremove 自动移除不需要的包
amber 彩蛋功能
xmp360 彩蛋功能
bronya 彩蛋功能
-h, --help 显示此帮助信息
--help-all 显示完整帮助信息
-v, --version 展示APM版本号
本 APM 具有兔兔伯爵,女武神装甲和超级大运之力。
EOF
}
# 彩蛋函数
amber_egg() {
cat <<'EOF'
____ ____
/ __ )____ __________ ____ / __ )__ ______ ____ __ __
/ __ / __ `/ ___/ __ \/ __ \ / __ / / / / __ \/ __ \/ / / /
/ /_/ / /_/ / / / /_/ / / / / / /_/ / /_/ / / / / / / / /_/ /
/_____/\__,_/_/ \____/_/ /_/ /_____/\__,_/_/ /_/_/ /_/\__, /
/____/
Amber Package Manager - Sparkling with magic! 安柏包管理器 - blingbling
💎 Another target tracked down by Outrider Amber! 侦察骑士,发现目标!
EOF
}
bronya_egg() {
cat <<'EOF'
_ __ ____ _ ____ __
| | / /__ _/ / /____ ______(_)__ / __/_ _____ / /____ __ _
| |/ / _ `/ / '_/ // / __/ / _-) _\ \/ // (_-</ __/ -_) ' \
|___/\_,_/_/_/\_\\_, /_/ /_/\__/ /___/\_, /___/\__/\__/_/_/_/
/ / ___ ___ __/___/____/ / /___/
/ /__/ _ `/ // / _ \/ __/ _ \
/____/\_,_/_,_/_//_/\__/_//_/
Valkyrie 系统启动 - 重装小兔,Fire!
💎 感谢 Anysets 为 AmberCE 和 AmberPM 的 Arch 架构支持提供帮助~
EOF
}
xmp360_egg() {
cat <<'EOF'
################################################################################
################################################################################
################################################################################
##################### ####################
####################. %###################
##########= ####################
############################ #=========*#########
############# .% .*****= :######
############################### =+ =#######* -#####
##### %: *######## :#####
######################% # ########* +#####
########: # %#####
########################## .# ######
##################* -* ######
##################= =###%: +- *###* ######
##################. # %: :* # -######
##################. % *###= % -- #### % %######
########################- ####+ ########################### .####. #########
######################### ############################% ##########
###########################=::+################################%-:-*############
################################################################################
哇——————袄 - 撞大运咯!
💎 感谢 潇湘·秀 为 APM 的 RPM 架构支持提供帮助~
EOF
}
# 完整帮助信息函数
show_help_all() {
cat <<EOF
APM - Amber Package Manager ${VERSION}
Usage:
$SCRIPT_NAME [COMMAND] [OPTIONS] [PACKAGES...]
Commands:
install 安装软件包
remove 卸载软件包
launch <package> [args...] 启动软件包(通过应用启动器)
run <package> [EXEC_PATH] [args...] 运行指定软件包的可执行文件(可指定容器内路径)
sandbox-run <package> [EXEC_PATH] [args...] 运行指定软件包的可执行文件(主目录沙箱化)
bwrap-run <package> [EXEC_PATH] [args...] 运行指定软件包的可执行文件(使用特殊的挂载参数以支持bwrap
bwrap-run <package> [EXEC_PATH] [args...] 运行指定软件包的可执行文件(使用 bwrap
update 更新软件包信息
hold 锁定软件包版本
@@ -37,6 +129,7 @@ Commands:
show 展示包信息
clean 清除缓存软件包
autoremove 自动移除不需要的包
ssinstall <path> 使用 ssinstall 进行本地软件安装,详情见 spark-store
ssaudit <path> 使用 ssaudit 进行本地软件安装,详情见 spark-store
debug 显示调试系统信息并进入调试环境
@@ -44,7 +137,8 @@ Commands:
xmp360 彩蛋功能
bronya 彩蛋功能
-h, --help 显示帮助信息
-h, --help 显示简要帮助信息
--help-all 显示此完整帮助信息
-v, --version 展示APM版本号
本 APM 具有兔兔伯爵,女武神装甲和超级大运之力。
@@ -241,65 +335,6 @@ amber-pm-debug "$@"
}
# 彩蛋函数
amber_egg() {
cat <<'EOF'
____ ____
/ __ )____ __________ ____ / __ )__ ______ ____ __ __
/ __ / __ `/ ___/ __ \/ __ \ / __ / / / / __ \/ __ \/ / / /
/ /_/ / /_/ / / / /_/ / / / / / /_/ / /_/ / / / / / / / /_/ /
/_____/\__,_/_/ \____/_/ /_/ /_____/\__,_/_/ /_/_/ /_/\__, /
/____/
Amber Package Manager - Sparkling with magic! 安柏包管理器 - blingbling
💎 Another target tracked down by Outrider Amber! 侦察骑士,发现目标!
EOF
}
bronya_egg() {
cat <<'EOF'
_ __ ____ _ ____ __
| | / /__ _/ / /____ ______(_)__ / __/_ _____ / /____ __ _
| |/ / _ `/ / '_/ // / __/ / _-) _\ \/ // (_-</ __/ -_) ' \
|___/\_,_/_/_/\_\\_, /_/ /_/\__/ /___/\_, /___/\__/\__/_/_/_/
/ / ___ ___ __/___/____/ / /___/
/ /__/ _ `/ // / _ \/ __/ _ \
/____/\_,_/_,_/_//_/\__/_//_/
Valkyrie 系统启动 - 重装小兔,Fire!
💎 感谢 Anysets 为 AmberCE 和 AmberPM 的 Arch 架构支持提供帮助~
EOF
}
xmp360_egg() {
cat <<'EOF'
################################################################################
################################################################################
################################################################################
##################### ####################
####################. %###################
##########= ####################
############################ #=========*#########
############# .% .*****= :######
############################### =+ =#######* -#####
##### %: *######## :#####
######################% # ########* +#####
########: # %#####
########################## .# ######
##################* -* ######
##################= =###%: +- *###* ######
##################. # %: :* # -######
##################. % *###= % -- #### % %######
########################- ####+ ########################### .####. #########
######################### ############################% ##########
###########################=::+################################%-:-*############
################################################################################
哇——————袄 - 撞大运咯!
💎 感谢 潇湘·秀 为 APM 的 RPM 架构支持提供帮助~
EOF
}
apm-nvidia-toggle(){
@@ -356,7 +391,7 @@ case "$1" in
fi
amber-pm-debug amber-pm-dstore-patch
apm-nvidia-toggle
amber-pm-gxde-desktop-fix
amber-pm-desktop-fix
update-mime-database /var/lib/apm/apm/files/ace-env/amber-ce-tools/data-dir/mime > /dev/null 2>&1 &
;;
download|search|policy|list|update|clean|show|depends|rdepends|changelog|moo)
@@ -397,7 +432,7 @@ case "$1" in
exit $exit_code
fi
amber-pm-debug amber-pm-dstore-patch
amber-pm-gxde-desktop-fix
amber-pm-desktop-fix
;;
launch)
shift
@@ -477,12 +512,29 @@ case "$1" in
exit $exit_code
fi
amber-pm-debug amber-pm-dstore-patch
amber-pm-gxde-desktop-fix
amber-pm-desktop-fix
;;
ssinstall)
amber-pm-debug dpkg --configure -a
amber-pm-debug ssinstall $@ --native
exit_code=$?
if [ $exit_code -eq 0 ]; then
log.info "Operation successful"
else
log.error "Error: Operation failed"
exit $exit_code
fi
amber-pm-debug amber-pm-dstore-patch
amber-pm-desktop-fix
;;
-h|--help)
show_help
;;
--help-all)
show_help_all
;;
-v|--version)
echo "$VERSION"
;;
@@ -37,6 +37,7 @@ _apm()
# see if the user selected a command already
local COMMANDS=(
"ssaudit"
"ssinstall"
"launch"
"list"
"search"
@@ -300,6 +301,11 @@ get_arg_position() {
_filedir
return 0
;;
ssinstall)
# ssinstall 命令总是匹配文件
_filedir
return 0
;;
esac
fi
View File
+191
View File
@@ -0,0 +1,191 @@
#compdef apm
local context state line
local -a commands options
# 定义命令列表
commands=(
'ssaudit:使用 ssaudit 进行本地软件安装'
'ssinstall:使用 ssinstall 进行本地软件安装'
'launch:启动软件包(通过应用启动器)'
'list:查看可用软件包信息'
'search:搜索软件包'
'show:展示包信息'
'showsrc:展示源码包信息'
'install:安装软件包'
'remove:卸载软件包'
'purge:彻底卸载软件包'
'autoremove:自动移除不需要的包'
'autopurge:自动彻底移除不需要的包'
'update:更新软件包信息'
'upgrade:升级软件包'
'full-upgrade:升级全部软件包'
'dist-upgrade:分发升级'
'run:运行指定软件包的可执行文件'
'sandbox-run:运行指定软件包的可执行文件(主目录沙箱化)'
'bwrap-run:运行指定软件包的可执行文件(使用 bwrap)'
'help:显示帮助信息'
'source:获取源码包'
'build-dep:安装构建依赖'
'clean:清除缓存软件包'
'autoclean:自动清除缓存软件包'
'download:下载包'
'changelog:查看包的变更日志'
'amber:彩蛋功能'
'xmp360:彩蛋功能'
'bronya:彩蛋功能'
'debug:显示调试系统信息并进入调试环境'
'depends:查看包依赖'
'rdepends:查看反向依赖'
'policy:查看包策略'
)
# 定义通用选项
options=(
'-h[显示帮助信息]'
'--help[显示帮助信息]'
'--help-all[显示完整帮助信息]'
'-v[展示APM版本号]'
'--version[展示APM版本号]'
)
# 定义命令特定选项
local -A command_options
command_options=(
'install' '-d[仅下载不安装] --download-only[仅下载不安装] -y[自动回答是] --assume-yes[自动回答是] --assume-no[自动回答否] -u[显示已升级的包] --show-upgraded[显示已升级的包] -m[忽略缺失的包] --ignore-missing[忽略缺失的包] -t[指定目标发行版] --target-release[指定目标发行版] --download[下载] --fix-missing[修复缺失的包] --ignore-hold[忽略被锁定的包] --upgrade[升级] --only-upgrade[仅升级] --allow-change-held-packages[允许更改被锁定的包] --allow-remove-essential[允许移除 essential 包] --allow-downgrades[允许降级] --print-uris[打印 URI] --trivial-only[仅处理简单的升级] --remove[移除] --arch-only[仅处理架构相关的包] --allow-unauthenticated[允许未认证的包] --allow-insecure-repositories[允许不安全的仓库] --install-recommends[安装推荐的包] --install-suggests[安装建议的包] --no-install-recommends[不安装推荐的包] --no-install-suggests[不安装建议的包] --fix-policy[修复策略] --show-progress[显示进度] --fix-broken[修复损坏的包] --purge[彻底移除] --verbose-versions[显示详细版本信息] --auto-remove[自动移除不需要的包] -s[模拟运行] --simulate[模拟运行] --dry-run[模拟运行] --force-yes[强制回答是] --reinstall[重新安装] --solver[指定解算器]'
'remove' '-d[仅下载不安装] --download-only[仅下载不安装] -y[自动回答是] --assume-yes[自动回答是] --assume-no[自动回答否] -u[显示已升级的包] --show-upgraded[显示已升级的包] -m[忽略缺失的包] --ignore-missing[忽略缺失的包] -t[指定目标发行版] --target-release[指定目标发行版] --download[下载] --fix-missing[修复缺失的包] --ignore-hold[忽略被锁定的包] --upgrade[升级] --only-upgrade[仅升级] --allow-change-held-packages[允许更改被锁定的包] --allow-remove-essential[允许移除 essential 包] --allow-downgrades[允许降级] --print-uris[打印 URI] --trivial-only[仅处理简单的升级] --remove[移除] --arch-only[仅处理架构相关的包] --allow-unauthenticated[允许未认证的包] --allow-insecure-repositories[允许不安全的仓库] --install-recommends[安装推荐的包] --install-suggests[安装建议的包] --no-install-recommends[不安装推荐的包] --no-install-suggests[不安装建议的包] --fix-policy[修复策略] --show-progress[显示进度] --fix-broken[修复损坏的包] --purge[彻底移除] --verbose-versions[显示详细版本信息] --auto-remove[自动移除不需要的包] -s[模拟运行] --simulate[模拟运行] --dry-run[模拟运行] --force-yes[强制回答是] --reinstall[重新安装] --solver[指定解算器]'
'update' '--list-cleanup[清理列表] --print-uris[打印 URI] --allow-insecure-repositories[允许不安全的仓库]'
'list' '--installed[显示已安装的包] --upgradable[显示可升级的包] --manual-installed[显示手动安装的包] -v[详细输出] --verbose[详细输出] -a[显示所有版本] --all-versions[显示所有版本] -t[指定目标发行版] --target-release[指定目标发行版]'
'show' '-a[显示所有版本] --all-versions[显示所有版本]'
'depends' '-i[重要依赖] --important[重要依赖] --installed[已安装的依赖] --pre-depends[预依赖] --depends[依赖] --recommends[推荐依赖] --suggests[建议依赖] --replaces[替换] --breaks[破坏] --conflicts[冲突] --enhances[增强] --recurse[递归] --implicit[隐式依赖]'
'rdepends' '-i[重要依赖] --important[重要依赖] --installed[已安装的依赖] --pre-depends[预依赖] --depends[依赖] --recommends[推荐依赖] --suggests[建议依赖] --replaces[替换] --breaks[破坏] --conflicts[冲突] --enhances[增强] --recurse[递归] --implicit[隐式依赖]'
'search' '-n[仅按名称搜索] --names-only[仅按名称搜索] -f[完整搜索] --full[完整搜索]'
'showsrc' '--only-source[仅源码]'
'source' '-d[仅下载不安装] --download-only[仅下载不安装] -y[自动回答是] --assume-yes[自动回答是] --assume-no[自动回答否] -u[显示已升级的包] --show-upgraded[显示已升级的包] -m[忽略缺失的包] --ignore-missing[忽略缺失的包] -t[指定目标发行版] --target-release[指定目标发行版] --download[下载] --fix-missing[修复缺失的包] --ignore-hold[忽略被锁定的包] --upgrade[升级] --only-upgrade[仅升级] --allow-change-held-packages[允许更改被锁定的包] --allow-remove-essential[允许移除 essential 包] --allow-downgrades[允许降级] --print-uris[打印 URI] --trivial-only[仅处理简单的升级] --remove[移除] --arch-only[仅处理架构相关的包] --allow-unauthenticated[允许未认证的包] --allow-insecure-repositories[允许不安全的仓库] --install-recommends[安装推荐的包] --install-suggests[安装建议的包] --no-install-recommends[不安装推荐的包] --no-install-suggests[不安装建议的包] --fix-policy[修复策略] -s[模拟运行] --simulate[模拟运行] --dry-run[模拟运行] -b[编译] --compile[编译] --build[编译] -P[构建配置文件] --build-profiles[构建配置文件] --diff-only[仅差异] --debian-only[仅 Debian] --tar-only[仅 tar] --dsc-only[仅 dsc]'
'build-dep' '-d[仅下载不安装] --download-only[仅下载不安装] -y[自动回答是] --assume-yes[自动回答是] --assume-no[自动回答否] -u[显示已升级的包] --show-upgraded[显示已升级的包] -m[忽略缺失的包] --ignore-missing[忽略缺失的包] -t[指定目标发行版] --target-release[指定目标发行版] --download[下载] --fix-missing[修复缺失的包] --ignore-hold[忽略被锁定的包] --upgrade[升级] --only-upgrade[仅升级] --allow-change-held-packages[允许更改被锁定的包] --allow-remove-essential[允许移除 essential 包] --allow-downgrades[允许降级] --print-uris[打印 URI] --trivial-only[仅处理简单的升级] --remove[移除] --arch-only[仅处理架构相关的包] --allow-unauthenticated[允许未认证的包] --allow-insecure-repositories[允许不安全的仓库] --install-recommends[安装推荐的包] --install-suggests[安装建议的包] --no-install-recommends[不安装推荐的包] --no-install-suggests[不安装建议的包] --fix-policy[修复策略] -a[主机架构] --host-architecture[主机架构] -s[模拟运行] --simulate[模拟运行] --dry-run[模拟运行] -P[构建配置文件] --build-profiles[构建配置文件] --purge[彻底移除] --solver[指定解算器]'
'clean' '-s[模拟运行] --simulate[模拟运行] --dry-run[模拟运行]'
'autoclean' '-s[模拟运行] --simulate[模拟运行] --dry-run[模拟运行]'
)
# 定义目录路径
local primary_dir="/var/lib/apm/apm/files/ace-env/var/lib/apm/"
local fallback_dir="/var/lib/apm/"
# 查找不包含特定子目录的目录
function find_directories_without_ace_env() {
local base_dir="$1"
local result=()
# 检查基础目录是否存在
if [[ ! -d "$base_dir" ]]; then
return 1
fi
# 查找所有直接子目录,排除包含ace-env子目录的目录
for dir in "$base_dir"/*; do
if [[ -d "$dir" ]] && [[ ! -d "$dir/files/ace-env" ]]; then
result+=("$(basename "$dir")")
fi
done
# 输出结果
if [[ ${#result[@]} -gt 0 ]]; then
print -l "${result[@]}"
return 0
fi
return 1
}
function apm_run_compgen() {
local result
result=$(find_directories_without_ace_env "$primary_dir")
if [[ -n "$result" ]]; then
echo "$result"
else
result=$(find_directories_without_ace_env "$fallback_dir")
if [[ -n "$result" ]]; then
echo "$result"
else
echo ""
fi
fi
}
# 主完成函数
_arguments -C \
"$options[@]" \
'*:: :->command'
case $state in
command)
if (( CURRENT == 1 )); then
_describe 'command' commands
else
local cmd="${words[1]}"
local arg_pos=0
local found_cmd=0
for (( i=1; i < CURRENT; i++ )); do
if [[ $found_cmd -eq 0 ]]; then
if [[ "${words[i]}" == "$cmd" ]]; then
found_cmd=1
fi
else
# 跳过选项参数(以-开头)
if [[ "${words[i]}" != -* ]]; then
((arg_pos++))
fi
fi
done
# 处理命令特定的选项
if [[ -n "${command_options[$cmd]}" ]]; then
_arguments "${(s: :)command_options[$cmd]}"
fi
# 处理命令特定的参数
case $cmd in
remove|purge|autoremove|autopurge)
if [[ $arg_pos -eq 1 ]]; then
_files -W "$primary_dir" -/n
fi
;;
show|list|download|changelog|depends|rdepends|policy)
if [[ $arg_pos -eq 1 ]]; then
local packages
packages=($(amber-pm-debug apt-cache --no-generate pkgnames "$PREFIX" -o Dir::Cache="/var/lib/aptss/" 2> /dev/null))
_describe 'package' packages
fi
;;
install)
if [[ $arg_pos -eq 1 ]]; then
local packages
packages=($(amber-pm-debug apt-cache --no-generate pkgnames "$PREFIX" -o Dir::Cache="/var/lib/aptss/" 2> /dev/null))
_describe 'package' packages
_files -g "*.deb"
fi
;;
source|build-dep|showsrc)
if [[ $arg_pos -eq 1 ]]; then
local packages
packages=($(amber-pm-debug apt-cache --no-generate pkgnames "$PREFIX" -o Dir::Cache="/var/lib/aptss/" 2> /dev/null))
_describe 'package' packages
fi
;;
run|sandbox-run|bwrap-run|launch)
if [[ $arg_pos -eq 1 ]]; then
local packages
packages=($(apm_run_compgen))
_describe 'package' packages
elif [[ $arg_pos -ge 2 ]]; then
_files
fi
;;
ssaudit|ssinstall)
_files
;;
esac
fi
;;
esac
+4 -1
View File
@@ -109,7 +109,7 @@ ENV_VARS=(
"PATH /amber-ce-tools/bin-override:\$PATH"
"IS_ACE_ENV 1"
"GTK_USE_PORTAL 1"
"XDG_DATA_DIRS /amber-ce-tools/additional-data-dir-in-container:/host/usr/local/share:/host/usr/share:\$XDG_DATA_DIRS"
"XDG_DATA_DIRS /amber-ce-tools/additional-data-dir-in-container:\$XDG_DATA_DIRS"
"XCURSOR_PATH /host/usr/share/icons:/host/usr/share/cursors:/usr/share/icons:/usr/share/cursors:\$XCURSOR_PATH"
)
@@ -127,6 +127,9 @@ BASE_BIND_DIRS=(
"--dev-bind-try /run/user/\$uid/pulse /run/user/\$uid/pulse"
"--dev-bind / /host"
"--dev-bind-try /etc/resolv.conf /etc/resolv.conf"
"--dev-bind-try /usr/share/icons /usr/local/share/icons"
"--dev-bind-try /usr/share/fonts /usr/local/share/fonts"
"--dev-bind-try /usr/share/themes /usr/share/themes"
"--dev-bind-try /home /home"
"--dev-bind-try $HOME/.apm/${APM_PKG_NAME}/.deepinwine $HOME/.deepinwine"
)
@@ -1,6 +1,6 @@
[Desktop Entry]
Exec=xdg-open %U
MimeType=inode/directory;application/x-mimearchive;x-scheme-handler/http;x-scheme-handler/https;application/msword;application/vnd.openxmlformats-officedocument.wordprocessingml.document;application/vnd.ms-excel;application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;application/vnd.ms-powerpoint;application/vnd.openxmlformats-officedocument.presentationml.presentation;application/x-gzip;application/zip;application/rar;application/x-tar;application/pdf;application/rtf;image/gif;image/jpeg;image/jp2;image/png;image/tiff;image/bmp;image/svg+xml;image/webp;image/x-icon;application/kswps;application/kset;application/ksdps;application/x-photoshop;application/x-coreldraw;application/x-shockwave-flash;text/plain;application/x-javascript;text/javascript;text/css;text/html;application/xhtml+xml;text/xml;text/x-vcard;application/x-httpd-php;application/java-archive;application/vnd.android.package-archive;application/octet-stream;application/x-x509-user-cert;audio/mpeg;audio/midi;audio/x-wav;audio/x-mpegurl;audio/x-m4a;audio/ogg;audio/x-realaudio;video/mp4;video/mpeg;video/quicktime;video/x-m4v;video/x-ms-wmv;video/x-msvideo;video/webm;video/x-flv;application/xhtml_xml;
MimeType=x-scheme-handler/spk;inode/directory;application/x-mimearchive;x-scheme-handler/http;x-scheme-handler/https;application/msword;application/vnd.openxmlformats-officedocument.wordprocessingml.document;application/vnd.ms-excel;application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;application/vnd.ms-powerpoint;application/vnd.openxmlformats-officedocument.presentationml.presentation;application/x-gzip;application/zip;application/rar;application/x-tar;application/pdf;application/rtf;image/gif;image/jpeg;image/jp2;image/png;image/tiff;image/bmp;image/svg+xml;image/webp;image/x-icon;application/kswps;application/kset;application/ksdps;application/x-photoshop;application/x-coreldraw;application/x-shockwave-flash;text/plain;application/x-javascript;text/javascript;text/css;text/html;application/xhtml+xml;text/xml;text/x-vcard;application/x-httpd-php;application/java-archive;application/vnd.android.package-archive;application/octet-stream;application/x-x509-user-cert;audio/mpeg;audio/midi;audio/x-wav;audio/x-mpegurl;audio/x-m4a;audio/ogg;audio/x-realaudio;video/mp4;video/mpeg;video/quicktime;video/x-m4v;video/x-ms-wmv;video/x-msvideo;video/webm;video/x-flv;application/xhtml_xml;
Name=ace-run-in-host-os
NoDisplay=true
Terminal=false
@@ -6,713 +6,99 @@ log.error() { echo -e "[\e[31mERROR\e[0m]: \e[1m$*\e[0m"; }
log.info() { echo -e "[\e[96mINFO\e[0m]: \e[1m$*\e[0m"; }
log.debug() { echo -e "[\e[32mDEBUG\e[0m]: \e[1m$*\e[0m"; }
# 检查权限
if [ "$UID" != "0" ]; then
if [ "$UID" != "0" ];then
log.error "需要以root权限运行 Need to be run as root."
exit 1
fi
# 检查参数
if [ -z "$1" ]; then
log.error "需要把ace-env所在的路径设置为第一个参数"
# 1. 获取宿主机 NVIDIA 驱动版本
nvidia_version=$(cat /sys/module/nvidia/version 2>/dev/null)
if [ -z "$nvidia_version" ]; then
#log.error "无法获取 NVIDIA 驱动版本 Can not determine NVIDIA Driver version"
exit
fi
# 2. 目标目录准备
ACE_DIR="$1"
if [[ ! -e "${ACE_DIR}" ]];then
log.error "未检测到 ace-env 安装,请安装后再试 ace-env is not detected. Please try again after installation"
exit 1
fi
# 辅助函数
trim() {
local str="$1"
str="${str#"${str%%[![:space:]]*}"}" # 移除前导空格
str="${str%"${str##*[![:space:]]}"}" # 移除尾部空格
echo "$str"
}
# 3. 检查容器内驱动版本是否与宿主机一致
version_file="$ACE_DIR/amber-ce-tools/nvidia_current_version"
if [ -f "$version_file" ] && [ "$(cat "$version_file")" = "$nvidia_version" ]; then
# log.info "容器内 NVIDIA 驱动版本 ($nvidia_version) 与宿主机一致,无需重新链接"
# log.info "Driver version in container matches host version, no need to relink"
exit 0
fi
is_regular_or_symlink() {
local path="$1"
if [ -f "$path" ] || [ -L "$path" ]; then
return 0
fi
return 1
}
mkdir -p "$ACE_DIR/usr/lib" "$ACE_DIR/usr/lib32"
is_directory() {
local path="$1"
if [ -d "$path" ]; then
return 0
fi
return 1
}
log.info "正在链接 NVIDIA 驱动库 Linking NVIDIA Driver Libs"
is_char_or_block_device() {
local path="$1"
if [ -c "$path" ] || [ -b "$path" ]; then
return 0
fi
return 1
}
# 4. 收集库文件路径
lib_list=$(ldconfig -p | grep -Ei "nvidia|libcuda|libvdpau_nvidia|libnvcuvid|libnvencode|libnvidia-encode" | cut -d'>' -f2)
read_driver_version() {
# 尝试从/sys目录读取
if [ -f "/sys/module/nvidia/version" ]; then
local version=$(cat "/sys/module/nvidia/version" 2>/dev/null)
version=$(trim "$version")
if [ -n "$version" ]; then
echo "$version"
return 0
fi
fi
# 尝试从/proc目录读取
if [ -f "/proc/driver/nvidia/version" ]; then
local proc_version=$(cat "/proc/driver/nvidia/version" 2>/dev/null)
# 提取版本号 (格式: 535.86.05)
if [[ "$proc_version" =~ [0-9]+\.[0-9]+\.[0-9]+ ]]; then
echo "${BASH_REMATCH[0]}"
return 0
fi
fi
return 1
}
# 5. 复制库文件
copied=0
has_32bit=false
has_64bit=false
has_glx=false
version_from_filename() {
local filename="$1"
local prefix="$2"
if [[ "$filename" != "$prefix"* ]]; then
return 1
for lib in $lib_list; do
resolved=$(readlink -f "$lib") # 解析符号链接
filename=$(basename "$lib")
if file "$resolved" | grep -q "32-bit"; then
ln -sf "/host/$resolved" "$ACE_DIR/usr/lib32/$filename"
has_32bit=true
log.debug "创建32位链接: $ACE_DIR/usr/lib32/$filename -> /host/$resolved"
else
ln -sf "/host/$resolved" "$ACE_DIR/usr/lib/$filename"
has_64bit=true
copied=1
log.debug "创建64位链接: $ACE_DIR/usr/lib/$filename -> /host/$resolved"
fi
local version="${filename#$prefix}"
if [[ -z "$version" ]]; then
return 1
fi
echo "$version"
return 0
}
compare_versions() {
local ver1="$1"
local ver2="$2"
# 分割版本号
IFS='.' read -r -a v1_parts <<< "$ver1"
IFS='.' read -r -a v2_parts <<< "$ver2"
local max_len=$(( ${#v1_parts[@]} > ${#v2_parts[@]} ? ${#v1_parts[@]} : ${#v2_parts[@]} ))
for ((i=0; i<max_len; i++)); do
local v1=${v1_parts[i]:-0}
local v2=${v2_parts[i]:-0}
if (( v1 < v2 )); then
echo "-1"
return
elif (( v1 > v2 )); then
echo "1"
return
fi
done
echo "0"
}
collect_files() {
local search_dirs=("${!1}")
local pattern="$2"
local results=()
for dir in "${search_dirs[@]}"; do
if [ ! -d "$dir" ]; then
continue
fi
# 使用find搜索文件
while IFS= read -r -d '' file; do
if is_regular_or_symlink "$file"; then
results+=("$file")
fi
done < <(find "$dir" -name "$pattern" -type f 2>/dev/null | head -100)
done
# 去重
local unique_results=()
declare -A seen
for file in "${results[@]}"; do
local realpath=$(readlink -f "$file" 2>/dev/null || echo "$file")
if [ -z "${seen[$realpath]}" ]; then
seen["$realpath"]=1
unique_results+=("$realpath")
fi
done
echo "${unique_results[@]}"
}
select_best_versioned_lib() {
local files=("${!1}")
local prefix="$2"
local prefer_version="$3"
local best=""
local best_version=""
for file in "${files[@]}"; do
local filename=$(basename "$file")
# 提取版本号
local version_result=$(version_from_filename "$filename" "$prefix")
if [ -z "$version_result" ]; then
continue
fi
local version="$version_result"
# 如果指定了首选版本,优先匹配
if [ -n "$prefer_version" ] && [ "$version" != "$prefer_version" ]; then
continue
fi
if [ -z "$best" ] || [ "$(compare_versions "$version" "$best_version")" -gt 0 ]; then
best="$file"
best_version="$version"
fi
done
# 如果没有匹配到首选版本,选择最高版本
if [ -z "$best" ] && [ -n "$prefer_version" ]; then
for file in "${files[@]}"; do
local filename=$(basename "$file")
local version_result=$(version_from_filename "$filename" "$prefix")
if [ -z "$version_result" ]; then
continue
fi
local version="$version_result"
if [ -z "$best" ] || [ "$(compare_versions "$version" "$best_version")" -gt 0 ]; then
best="$file"
best_version="$version"
fi
done
fi
echo "$best"
}
detect_driver_info() {
local info=""
# 读取驱动版本
local driver_version=$(read_driver_version)
if [ -z "$driver_version" ]; then
driver_version=""
fi
# 默认库搜索路径
local default_search_paths=(
"/usr/lib64"
"/usr/lib/x86_64-linux-gnu"
"/usr/lib/i386-linux-gnu"
"/usr/lib/aarch64-linux-gnu"
"/usr/lib/x86_64-linux-gnu/nvidia/current"
"/usr/lib/i386-linux-gnu/nvidia/current"
"/usr/lib/aarch64-linux-gnu/nvidia/current"
"/lib64"
"/lib/x86_64-linux-gnu"
"/lib/i386-linux-gnu"
"/lib/aarch64-linux-gnu"
"/lib/x86_64-linux-gnu/nvidia/current"
"/lib/i386-linux-gnu/nvidia/current"
"/lib/aarch64-linux-gnu/nvidia/current"
"/usr/lib"
"/lib"
)
# 搜索libcuda.so
local cuda_files=($(collect_files default_search_paths[@] "libcuda.so.*"))
local nvidia_ml_files=($(collect_files default_search_paths[@] "libnvidia-ml.so.*"))
local selected_lib=""
if [ ${#cuda_files[@]} -gt 0 ]; then
selected_lib=$(select_best_versioned_lib cuda_files[@] "libcuda.so." "$driver_version")
fi
if [ -z "$selected_lib" ] && [ ${#nvidia_ml_files[@]} -gt 0 ]; then
selected_lib=$(select_best_versioned_lib nvidia_ml_files[@] "libnvidia-ml.so." "$driver_version")
fi
local lib_dir=""
if [ -n "$selected_lib" ]; then
lib_dir=$(dirname "$selected_lib")
# 如果还没有驱动版本,从文件名提取
if [ -z "$driver_version" ]; then
local filename=$(basename "$selected_lib")
local cuda_version=$(version_from_filename "$filename" "libcuda.so.")
local ml_version=$(version_from_filename "$filename" "libnvidia-ml.so.")
if [ -n "$cuda_version" ]; then
driver_version="$cuda_version"
elif [ -n "$ml_version" ]; then
driver_version="$ml_version"
fi
fi
fi
# 如果还没找到库目录,使用默认的
if [ -z "$lib_dir" ]; then
for path in "${default_search_paths[@]}"; do
if is_directory "$path"; then
lib_dir="$path"
break
fi
done
fi
echo "$driver_version:$lib_dir"
}
read_elf_soname() {
local file="$1"
# 使用readelf读取SONAME
if command -v readelf >/dev/null 2>&1; then
local soname=$(readelf -d "$file" 2>/dev/null | grep -E "SONAME.*\[.*\]" | sed -E 's/.*\[(.*)\].*/\1/')
if [ -n "$soname" ]; then
echo "$soname"
return 0
fi
fi
# 使用objdump作为备选
if command -v objdump >/dev/null 2>&1; then
local soname=$(objdump -p "$file" 2>/dev/null | grep -E "SONAME" | awk '{print $2}')
if [ -n "$soname" ]; then
echo "$soname"
return 0
fi
fi
return 1
}
is_elf32() {
local file="$1"
if ! [ -f "$file" ]; then
return 1
fi
# 检查文件头部
local header=$(head -c 5 "$file" 2>/dev/null | od -An -t x1 | tr -d ' \n')
# ELF魔法字节: 7f 45 4c 46
if [[ "$header" == 7f454c46* ]]; then
# 检查第5个字节:01表示32位,02表示64位
local class_byte=${header:8:2}
if [ "$class_byte" = "01" ]; then
return 0
fi
fi
return 1
}
ensure_symlink() {
local target="$1"
local link_path="$2"
# 创建父目录
mkdir -p "$(dirname "$link_path")" 2>/dev/null
# 如果链接已存在且正确,跳过
if [ -L "$link_path" ]; then
local current_target=$(readlink -f "$link_path" 2>/dev/null || readlink "$link_path")
if [ "$current_target" = "$target" ]; then
return 0
fi
fi
# 删除现有文件/链接
rm -f "$link_path" 2>/dev/null
# 创建符号链接
ln -sf "$target" "$link_path" 2>/dev/null
return $?
}
# 主函数
main() {
ACE_DIR="$1"
# 检查目标目录
if [[ ! -e "${ACE_DIR}" ]]; then
log.error "ACE_DIR为空,退出。"
exit 1
fi
# 获取驱动信息
local driver_info=$(detect_driver_info)
if [ -z "$driver_info" ]; then
# log.error "无法检测到NVIDIA驱动信息 Cannot detect NVIDIA driver information"
exit
fi
local nvidia_version=$(echo "$driver_info" | cut -d':' -f1)
local lib_dir=$(echo "$driver_info" | cut -d':' -f2)
if [ -z "$nvidia_version" ]; then
# log.error "无法获取NVIDIA驱动版本 Cannot determine NVIDIA driver version"
exit
fi
# 检查版本是否已存在且匹配
if [ -f "$ACE_DIR/amber-ce-tools/nvidia_current_version" ]; then
local existing_version=$(cat "$ACE_DIR/amber-ce-tools/nvidia_current_version")
if [ "$existing_version" = "$nvidia_version" ]; then
# log.info "NVIDIA驱动版本未变化,跳过链接操作 NVIDIA Driver version unchanged, skipping linking."
exit 0
else
log.info "检测到NVIDIA驱动版本变化: $existing_version -> $nvidia_version"
fi
fi
log.info "正在链接NVIDIA驱动库和GLX组件 Linking NVIDIA Driver Libs and GLX components"
# 准备目录结构
mkdir -p "$ACE_DIR/usr/lib" "$ACE_DIR/usr/lib32"
mkdir -p "$ACE_DIR/orig" "$ACE_DIR/orig/32"
mkdir -p "$ACE_DIR/etc"
# 清理旧链接
# find "$ACE_DIR/usr/lib" -type l -name "*.so*" -delete 2>/dev/null
# find "$ACE_DIR/usr/lib32" -type l -name "*.so*" -delete 2>/dev/null
# find "$ACE_DIR/orig" -type l -name "*.so*" -delete 2>/dev/null
# find "$ACE_DIR/orig/32" -type l -name "*.so*" -delete 2>/dev/null
# 默认库搜索路径
local default_search_paths=(
"$lib_dir"
"/usr/lib64"
"/usr/lib/x86_64-linux-gnu"
"/usr/lib/i386-linux-gnu"
"/usr/lib/aarch64-linux-gnu"
"/usr/lib/x86_64-linux-gnu/nvidia/current"
"/usr/lib/i386-linux-gnu/nvidia/current"
"/usr/lib/aarch64-linux-gnu/nvidia/current"
"/lib64"
"/lib/x86_64-linux-gnu"
"/lib/i386-linux-gnu"
"/lib/aarch64-linux-gnu"
"/lib/x86_64-linux-gnu/nvidia/current"
"/lib/i386-linux-gnu/nvidia/current"
"/lib/aarch64-linux-gnu/nvidia/current"
"/usr/lib"
"/lib"
)
# 1. 首先收集核心NVIDIA库
log.debug "收集核心NVIDIA库..."
# 核心库列表
local core_libs=(
"libnvidia-ml.so.*"
"libcuda.so.*"
"libnvidia-ptxjitcompiler.so.*"
"libnvidia-fatbinaryloader.so.*"
"libnvidia-opencl.so.*"
"libnvidia-compiler.so.*"
"libnvidia-encode.so.*"
"libnvidia-opticalflow.so.*"
"libnvcuvid.so.*"
"libnvidia-cfg.so.*"
"libnvidia-allocator.so.*"
"libnvidia-nvvm.so.*"
)
# 2. 收集图形库(包含GLX)
log.debug "收集图形库..."
local graphics_libs=(
"libGLX_nvidia.so.*"
"libEGL_nvidia.so.*"
"libGLESv1_CM_nvidia.so.*"
"libGLESv2_nvidia.so.*"
"libnvidia-glcore.so.*"
"libnvidia-glsi.so.*"
"libnvidia-tls.so.*"
"libnvidia-egl-gbm.so.*"
"libnvidia-egl-wayland.so.*"
"libnvidia-vulkan-producer.so.*"
"libEGL.so*"
"libGL.so*"
"libGLESv1_CM.so*"
"libGLESv2.so*"
"libGLX.so*"
"libGLdispatch.so*"
"libOpenCL.so*"
"libOpenGL.so*"
"libnvidia-api.so*"
"libnvidia-egl-xcb.so*"
"libnvidia-egl-xlib.so*"
)
# 收集所有库文件
local all_libs=()
for pattern in "${core_libs[@]}" "${graphics_libs[@]}"; do
local files=($(collect_files default_search_paths[@] "$pattern"))
all_libs+=("${files[@]}")
done
# 去重
declare -A seen_libs
local unique_libs=()
for lib in "${all_libs[@]}"; do
local realpath=$(readlink -f "$lib" 2>/dev/null || echo "$lib")
if [ -z "${seen_libs[$realpath]}" ]; then
seen_libs["$realpath"]=1
unique_libs+=("$realpath")
fi
done
# 创建链接
local has_32bit=false
local has_64bit=false
local has_glx=false
for lib_path in "${unique_libs[@]}"; do
if [ ! -f "$lib_path" ] && [ ! -L "$lib_path" ]; then
continue
fi
local filename=$(basename "$lib_path")
local is_32bit=false
# 检查是否是32位库
if is_elf32 "$lib_path"; then
is_32bit=true
has_32bit=true
else
has_64bit=true
fi
# 创建原始链接(容器内路径)
local orig_dest_dir="$ACE_DIR/orig"
if [ "$is_32bit" = true ]; then
orig_dest_dir="$ACE_DIR/orig/32"
fi
# 创建主链接
local container_target="/host$lib_path"
local orig_link_path="$orig_dest_dir/$filename"
if ensure_symlink "$container_target" "$orig_link_path"; then
# 检查是否是GLX库
if [[ "$filename" == libGLX_nvidia.so.* ]]; then
has_glx=true
fi
# 创建SONAME链接
local soname=$(read_elf_soname "$lib_path")
if [ -n "$soname" ] && [ "$soname" != "$filename" ]; then
local soname_link_path="$orig_dest_dir/$soname"
ensure_symlink "$container_target" "$soname_link_path"
fi
fi
done
# 3. 收集X.Org模块
log.debug "收集X.Org模块..."
local xorg_paths=(
"$lib_dir/nvidia/xorg"
"$lib_dir/xorg/modules/drivers"
"$lib_dir/xorg/modules/extensions"
"$lib_dir/xorg/modules/updates/drivers"
"$lib_dir/xorg/modules/updates/extensions"
"/usr/lib/xorg/modules/drivers"
"/usr/lib/xorg/modules/extensions"
"/usr/lib/xorg/modules/updates/drivers"
"/usr/lib/xorg/modules/updates/extensions"
"/usr/lib64/xorg/modules/drivers"
"/usr/lib64/xorg/modules/extensions"
"/usr/lib64/xorg/modules/updates/drivers"
"/usr/lib64/xorg/modules/updates/extensions"
)
# 查找X.Org驱动程序
local xorg_driver=""
for xorg_dir in "${xorg_paths[@]}"; do
if [ -f "$xorg_dir/nvidia_drv.so" ]; then
xorg_driver="$xorg_dir/nvidia_drv.so"
break
fi
done
# 查找GLX服务器模块
local glx_server=""
for xorg_dir in "${xorg_paths[@]}"; do
if [ -f "$xorg_dir/libglxserver_nvidia.so.$nvidia_version" ]; then
glx_server="$xorg_dir/libglxserver_nvidia.so.$nvidia_version"
break
fi
done
# 如果没有找到特定版本,尝试通配符
if [ -z "$glx_server" ]; then
for xorg_dir in "${xorg_paths[@]}"; do
local found=$(find "$xorg_dir" -name "libglxserver_nvidia.so.*" -type f 2>/dev/null | head -1)
if [ -n "$found" ]; then
glx_server="$found"
break
fi
done
fi
# 创建X.Org文件链接
if [ -n "$xorg_driver" ]; then
local dest_dir="$ACE_DIR$(dirname "$xorg_driver")"
mkdir -p "$dest_dir"
local container_target="/host$xorg_driver"
ensure_symlink "$container_target" "$dest_dir/$(basename "$xorg_driver")"
fi
if [ -n "$glx_server" ]; then
# 检查是否是GLX库
if [[ "$filename" == libGLX_nvidia.so.* ]]; then
has_glx=true
local dest_dir="$ACE_DIR$(dirname "$glx_server")"
mkdir -p "$dest_dir"
local container_target="/host$glx_server"
ensure_symlink "$container_target" "$dest_dir/$(basename "$glx_server")"
fi
# 4. 复制配置文件和辅助文件
log.debug "处理配置和辅助文件..."
# Vulkan配置文件
local vulkan_files=(
"/usr/share/vulkan/icd.d/nvidia_icd.json"
"/usr/share/vulkan/icd.d/nvidia_icd.x86_64.json"
"/usr/share/vulkan/icd.d/nvidia_icd.aarch64.json"
"/usr/share/vulkan/implicit_layer.d/nvidia_layers.json"
)
# EGL配置文件
local egl_files=(
"/usr/share/egl/egl_external_platform.d/10_nvidia_wayland.json"
"/usr/share/egl/egl_external_platform.d/15_nvidia_gbm.json"
"/usr/share/egl/egl_external_platform.d/20_nvidia_xcb.json"
)
# GLVND配置文件
local glvnd_files=(
"/usr/share/glvnd/egl_vendor.d/10_nvidia.json"
)
# X11配置文件
local x11_files=(
"/usr/share/X11/xorg.conf.d/10-nvidia.conf"
"/usr/share/X11/xorg.conf.d/nvidia-drm-outputclass.conf"
)
# 处理所有配置文件
for file in "${vulkan_files[@]}" "${egl_files[@]}" "${glvnd_files[@]}" "${x11_files[@]}"; do
if [ -f "$file" ]; then
local dest_dir="$ACE_DIR$(dirname "$file")"
mkdir -p "$dest_dir"
local container_target="/host$file"
ensure_symlink "$container_target" "$dest_dir/$(basename "$file")"
fi
done
# 5. 生成ld.so.conf文件
if [ "$has_64bit" = true ] || [ "$has_32bit" = true ]; then
echo "/opt/extensions/nvidia/orig" > "$ACE_DIR/etc/ld.so.conf"
if [ "$has_32bit" = true ]; then
echo "/opt/extensions/nvidia/orig/32" >> "$ACE_DIR/etc/ld.so.conf"
fi
done
# 6. 复制辅助文件
additional_files=(
/usr/share/vulkan/icd.d/nvidia_icd.json
/usr/share/vulkan/icd.d/nvidia_icd.x86_64.json
/usr/share/vulkan/icd.d/nvidia_icd.aarch64.json
/usr/share/vulkan/implicit_layer.d/nvidia_layers.json
/usr/share/egl/egl_external_platform.d/10_nvidia_wayland.json
/usr/share/egl/egl_external_platform.d/15_nvidia_gbm.json
/usr/share/egl/egl_external_platform.d/20_nvidia_xcb.json
/usr/share/glvnd/egl_vendor.d/10_nvidia.json
)
for file in "${additional_files[@]}"; do
if [ -f "$file" ]; then
resolved=$(readlink -f "$file")
dir=$(dirname "$file")
mkdir -p "$ACE_DIR/$dir"
ln -sf "/host/$resolved" "$ACE_DIR/$dir/$(basename "$file")"
log.debug "复制配置文件: $ACE_DIR/$dir/$(basename "$file") -> /host/$resolved"
fi
# 6. 标记版本
done
# 7. 生成 ld.so.conf
echo "/usr/lib" > "$ACE_DIR/etc/ld.so.conf"
if [ "$has_32bit" = true ]; then
echo "/usr/lib32" >> "$ACE_DIR/etc/ld.so.conf"
fi
# 8. 标记版本
if [ $copied -eq 1 ]; then
echo "$nvidia_version" > "$ACE_DIR/amber-ce-tools/nvidia_current_version"
# 7. 生成环境变量脚本
cat > "$ACE_DIR/nvidia_env.sh" << EOF
#!/bin/bash
# NVIDIA驱动环境变量
export NVIDIA_DRIVER_VERSION="$nvidia_version"
# 库路径
if [ -d "/opt/extensions/nvidia/orig" ]; then
export LD_LIBRARY_PATH="/opt/extensions/nvidia/orig:\${LD_LIBRARY_PATH}"
fi
if [ -d "/opt/extensions/nvidia/orig/32" ]; then
export LD_LIBRARY_PATH="/opt/extensions/nvidia/orig/32:\${LD_LIBRARY_PATH}"
fi
# GLX和EGL配置
if [ "$has_glx" = true ]; then
export __GLX_VENDOR_LIBRARY_NAME="nvidia"
export __NV_PRIME_RENDER_OFFLOAD="1"
fi
# Vulkan ICD文件
if [ -f "/opt/extensions/nvidia/usr/share/vulkan/icd.d/nvidia_icd.json" ]; then
export VK_ICD_FILENAMES="/opt/extensions/nvidia/usr/share/vulkan/icd.d/nvidia_icd.json"
export VK_ADD_DRIVER_FILES="\${VK_ICD_FILENAMES}"
fi
# EGL外部平台配置
EGL_CONF_DIRS=""
for dir in /opt/extensions/nvidia/usr/share/egl/egl_external_platform.d \
/usr/share/egl/egl_external_platform.d; do
if [ -d "\$dir" ]; then
EGL_CONF_DIRS="\$dir:\${EGL_CONF_DIRS}"
fi
done
if [ -n "\${EGL_CONF_DIRS}" ]; then
export EGL_EXTERNAL_PLATFORM_CONFIG_DIRS="\${EGL_CONF_DIRS%:}"
export __EGL_EXTERNAL_PLATFORM_CONFIG_DIRS="\${EGL_CONF_DIRS%:}"
fi
# EGL供应商库目录
EGL_VENDOR_DIRS=""
for dir in /opt/extensions/nvidia/usr/share/glvnd/egl_vendor.d \
/usr/share/glvnd/egl_vendor.d; do
if [ -d "\$dir" ]; then
EGL_VENDOR_DIRS="\$dir:\${EGL_VENDOR_DIRS}"
fi
done
if [ -n "\${EGL_VENDOR_DIRS}" ]; then
export __EGL_VENDOR_LIBRARY_DIRS="\${EGL_VENDOR_DIRS%:}"
fi
export NVIDIA_CTK_LIBCUDA_DIR="/opt/extensions/nvidia/orig"
EOF
chmod +x "$ACE_DIR/nvidia_env.sh"
log.info "NVIDIA驱动库和GLX组件已成功链接 Nvidia Driver Libs and GLX components are successfully linked."
log.info "NVIDIA 驱动库已成功链接 Nvidia Driver Libs are successfully linked."
log.info "驱动版本: $nvidia_version"
log.info "64位库: $has_64bit, 32位库: $has_32bit, GLX支持: $has_glx"
log.info "环境变量脚本已生成: $ACE_DIR/nvidia_env.sh"
# 创建设备节点信息文件(供容器运行时使用)
cat > "$ACE_DIR/devices.info" << EOF
# NVIDIA设备节点
/dev/nvidiactl
/dev/nvidia-uvm
/dev/nvidia-uvm-tools
/dev/nvidia-modeset
/dev/nvidia[0-9]*
/dev/dri/card*
/dev/dri/renderD*
EOF
exit 0
}
# 执行主函数
main "$1"
else
log.error "未找到有效 NVIDIA 库文件 No valid NVIDIA Driver Libs found."
fi