enhanced_build.sh

#!/bin/bash
# 增强版CPU漏洞检测工具打包脚本
# 支持完整功能、CodeQL集成、多语言检测、PoC工具
APP_NAME="SpecProbe"
APP_VERSION="3.0.0"
BUILD_TYPE="enhanced"

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
PURPLE='\033[0;35m'
CYAN='\033[0;36m'
NC='\033[0m' # No Color

# 打印带颜色的消息
print_info() { echo -e "${BLUE}ℹ️  $1${NC}"; }
print_success() { echo -e "${GREEN}✅ $1${NC}"; }
print_warning() { echo -e "${YELLOW}⚠️  $1${NC}"; }
print_error() { echo -e "${RED}❌ $1${NC}"; }
print_step() { echo -e "${PURPLE}🔧 $1${NC}"; }

# 错误处理
set -e
trap 'print_error "脚本执行失败，退出码: $?"' ERR

print_step "开始构建 $APP_NAME v$APP_VERSION"

# 检查必要的系统依赖
check_dependencies() {
    print_step "检查系统依赖..."
    
    # 检查Python
    if ! command -v python3 &> /dev/null; then
        print_error "Python3 未安装"
        exit 1
    fi
    print_success "Python3: $(python3 --version)"
    
    # 检查必要的Python包管理工具
    if ! python3 -c "import pip" &> /dev/null; then
        print_error "pip 未安装"
        exit 1
    fi
    
    # 检查wget
    if ! command -v wget &> /dev/null; then
        print_warning "wget 未安装，尝试使用curl"
        if ! command -v curl &> /dev/null; then
            print_error "wget 和 curl 都未安装，请安装其中一个"
            exit 1
        fi
        USE_CURL=true
    else
        USE_CURL=false
    fi
}

# 检查项目文件完整性
check_project_files() {
    print_step "检查项目文件完整性..."
    
    # 必需的Python文件
    required_files=(
        "main.py"
        "gui.py" 
        "scanner.py"
        "utils.py"
        "source_analysis_gui.py"
        "fixed_codeql_analyzer.py"
        "translations.py"
        "exploit.py"
    )
    
    # 必需的脚本文件
    required_scripts=(
        "main.sh"
        "tool.sh"
        "setup_pocs.sh"
    )
    
    missing_files=()
    
    for file in "${required_files[@]}"; do
        if [ ! -f "$file" ]; then
            missing_files+=("$file")
        fi
    done
    
    for script in "${required_scripts[@]}"; do
        if [ ! -f "$script" ]; then
            missing_files+=("$script")
        fi
    done
    
    if [ ${#missing_files[@]} -ne 0 ]; then
        print_error "缺少以下必需文件:"
        for file in "${missing_files[@]}"; do
            echo "  - $file"
        done
        exit 1
    fi
    
    print_success "项目文件完整性检查通过"
}

# 检查和准备CodeQL
check_codeql() {
    print_step "检查CodeQL环境..."
    
    CODEQL_INTEGRATED=false
    CODEQL_VERSION=""
    
    # 检查本地codeql文件夹
    if [ -d "codeql" ]; then
        if [ -f "codeql/codeql" ]; then
            CODEQL_VERSION=$(./codeql/codeql version --format=json 2>/dev/null | grep -o '"productVersion":"[^"]*"' | cut -d'"' -f4 || echo "unknown")
            print_success "发现本地CodeQL: v$CODEQL_VERSION ($(du -sh codeql | cut -f1))"
            CODEQL_INTEGRATED=true
        else
            print_warning "codeql文件夹存在但缺少可执行文件"
        fi
    # 检查系统CodeQL
    elif command -v codeql &> /dev/null; then
        CODEQL_VERSION=$(codeql version --format=json 2>/dev/null | grep -o '"productVersion":"[^"]*"' | cut -d'"' -f4 || echo "unknown")
        print_info "系统CodeQL: v$CODEQL_VERSION"
        CODEQL_INTEGRATED=false
    else
        print_warning "未找到CodeQL，源代码分析功能将受限"
        print_info "建议下载CodeQL: https://github.com/github/codeql-cli-binaries/releases"
        CODEQL_INTEGRATED=false
    fi
}

# 检查和准备PoC工具
check_poc_tools() {
    print_step "检查PoC工具环境..."
    
    if [ -d "tools" ]; then
        print_success "发现tools目录"
        
        # 检查编译的PoC
        if [ -d "tools/compiled" ]; then
            compiled_pocs=($(ls tools/compiled/ 2>/dev/null || true))
            if [ ${#compiled_pocs[@]} -gt 0 ]; then
                print_success "发现 ${#compiled_pocs[@]} 个编译的PoC工具: ${compiled_pocs[*]}"
            else
                print_warning "tools/compiled目录为空"
            fi
        else
            print_warning "未找到tools/compiled目录"
            print_info "运行 ./setup_pocs.sh 来下载和编译PoC工具"
        fi
        
        # 检查源码
        poc_sources=($(find tools -name "*.c" -o -name "*.cpp" | wc -l))
        if [ "$poc_sources" -gt 0 ]; then
            print_info "发现 $poc_sources 个PoC源文件"
        fi
    else
        print_warning "未找到tools目录"
        print_info "某些高级功能可能不可用"
    fi
}

# 创建虚拟环境并安装依赖
setup_python_env() {
    print_step "设置Python环境..."
    
    # 清理旧的构建
    rm -rf build_env dist *.spec __pycache__ .eggs *.egg-info
    rm -rf appimage_build
    
    # 创建虚拟环境
    python3 -m venv build_env
    source build_env/bin/activate
    
    # 升级pip
    pip install --upgrade pip setuptools wheel
    
    # 安装必要依赖
    print_info "安装Python依赖..."
    pip install PyQt5 psutil requests pyinstaller
    
    # 安装可选依赖（如果需要）
    pip install pillow || print_warning "Pillow安装失败，图标功能可能受限"
    
    print_success "Python环境设置完成"
}

# 修复PyQt5路径问题
fix_pyqt5_path() {
    print_step "修复PyQt5路径问题..."
    
    # 检查PyQt5安装
    python3 -c "import PyQt5.QtCore; print('PyQt5导入成功')" || {
        print_error "PyQt5导入失败，重新安装..."
        pip install --force-reinstall PyQt5
    }
    
    # 获取PyQt5路径
    PYQT5_PATH=$(python3 -c "import PyQt5; print(PyQt5.__path__[0])" 2>/dev/null)
    if [ -n "$PYQT5_PATH" ]; then
        print_info "PyQt5路径: $PYQT5_PATH"
        
        # 检查插件目录
        if [ ! -d "$PYQT5_PATH/Qt5/plugins" ]; then
            print_warning "PyQt5插件目录不存在，尝试修复..."
            # 尝试创建符号链接或复制
            if [ -d "$PYQT5_PATH/Qt/plugins" ]; then
                ln -sf "$PYQT5_PATH/Qt/plugins" "$PYQT5_PATH/Qt5/plugins" 2>/dev/null || {
                    mkdir -p "$PYQT5_PATH/Qt5"
                    cp -r "$PYQT5_PATH/Qt/plugins" "$PYQT5_PATH/Qt5/" 2>/dev/null || true
                }
            fi
        fi
    fi
}

# 构建Python应用
build_python_app() {
    print_step "使用PyInstaller构建Python应用..."
    
    # 修复PyQt5路径问题
    fix_pyqt5_path
    
    # 使用简化的构建方式避免复杂的spec文件
    print_info "使用简化构建模式..."
    
    # 创建临时构建目录
    mkdir -p temp_build
    
    # 复制所有Python文件到临时目录（避免路径编码问题）
    cp *.py temp_build/ 2>/dev/null || true
    cp *.sh temp_build/ 2>/dev/null || true
    [ -d "tools" ] && cp -r tools temp_build/ || print_warning "未找到tools目录"
    
    cd temp_build
    
    # 使用简化的PyInstaller命令
    pyinstaller --onefile --windowed \
        --name "$APP_NAME" \
        --add-data "*.sh:." \
        --add-data "*.py:." \
        $([ -d "tools" ] && echo "--add-data tools/:tools/") \
        --hidden-import PyQt5.sip \
        --hidden-import PyQt5.QtCore \
        --hidden-import PyQt5.QtGui \
        --hidden-import PyQt5.QtWidgets \
        --hidden-import PyQt5.QtPrintSupport \
        --hidden-import psutil \
        --hidden-import requests \
        --hidden-import json \
        --hidden-import subprocess \
        --hidden-import threading \
        --hidden-import platform \
        --hidden-import datetime \
        --hidden-import tempfile \
        --hidden-import shutil \
        --exclude-module tkinter \
        --exclude-module matplotlib \
        --exclude-module numpy \
        --clean \
        --noconfirm \
        main.py
    
    cd ..
    
    # 检查构建结果
    if [ -f "temp_build/dist/${APP_NAME}" ]; then
        # 移动构建结果
        mkdir -p dist
        cp "temp_build/dist/${APP_NAME}" "dist/"
        print_success "Python应用构建完成"
    else
        print_error "Python应用构建失败"
        
        # 尝试备用构建方法
        print_warning "尝试备用构建方法..."
        cd temp_build
        
        # 使用更简单的命令
        pyinstaller --onefile \
            --name "$APP_NAME" \
            --console \
            --clean \
            --noconfirm \
            main.py
        
        cd ..
        
        if [ -f "temp_build/dist/${APP_NAME}" ]; then
            mkdir -p dist
            cp "temp_build/dist/${APP_NAME}" "dist/"
            print_success "备用构建方法成功"
        else
            print_error "所有构建方法都失败"
            exit 1
        fi
    fi
    
    # 清理临时目录
    rm -rf temp_build
}

# 创建AppImage结构
create_appimage_structure() {
    print_step "创建AppImage目录结构..."
    
    mkdir -p appimage_build/AppDir/usr/bin
    mkdir -p appimage_build/AppDir/usr/lib
    mkdir -p appimage_build/AppDir/usr/share/applications
    mkdir -p appimage_build/AppDir/usr/share/icons/hicolor/256x256/apps
    
    # 复制主程序
    cp "dist/${APP_NAME}" "appimage_build/AppDir/usr/bin/"
    chmod +x "appimage_build/AppDir/usr/bin/${APP_NAME}"
    
    # 复制脚本文件
    cp *.sh "appimage_build/AppDir/usr/bin/" 2>/dev/null || true
    chmod +x appimage_build/AppDir/usr/bin/*.sh 2>/dev/null || true
    
    # 复制tools目录
    if [ -d "tools" ]; then
        cp -r tools "appimage_build/AppDir/usr/lib/"
        # 确保PoC工具可执行
        find "appimage_build/AppDir/usr/lib/tools" -type f -exec chmod +x {} \; 2>/dev/null || true
        print_success "PoC工具已复制"
    fi
    
    # 集成CodeQL（如果可用）
    if [ "$CODEQL_INTEGRATED" = true ]; then
        print_info "集成CodeQL到AppImage..."
        cp -r codeql "appimage_build/AppDir/usr/lib/"
        chmod +x "appimage_build/AppDir/usr/lib/codeql/codeql"
        
        # 创建CodeQL包装脚本
        cat > "appimage_build/AppDir/usr/bin/codeql" << 'CODEQL_WRAPPER'
#!/bin/bash
HERE="$(dirname "$(readlink -f "${0}")")"
export CODEQL_HOME="${HERE}/../lib/codeql"
exec "${CODEQL_HOME}/codeql" "$@"
CODEQL_WRAPPER
        chmod +x "appimage_build/AppDir/usr/bin/codeql"
        print_success "CodeQL已集成"
    fi
}

# 创建桌面文件
create_desktop_file() {
    print_step "创建桌面集成文件..."
    
    # 创建.desktop文件
    cat > "appimage_build/AppDir/${APP_NAME}.desktop" << EOF
[Desktop Entry]
Type=Application
Name=SpecProbe CPU漏洞检测工具
Name[en]=SpecProbe CPU Vulnerability Scanner
Comment=检测CPU硬件漏洞的专业安全工具
Comment[en]=Professional security tool for detecting CPU hardware vulnerabilities
GenericName=Security Scanner
GenericName[en]=Security Scanner
Exec=${APP_NAME}
Icon=${APP_NAME}
Terminal=false
StartupNotify=true
Categories=System;Security;Development;
Keywords=CPU;Security;Vulnerability;Spectre;Meltdown;EntryBleed;CodeQL;
StartupWMClass=${APP_NAME}
MimeType=text/x-csrc;text/x-c++src;text/x-java;application/javascript;text/x-python;
EOF

    # 创建应用图标（简单的文本图标）
    cat > "appimage_build/AppDir/usr/share/icons/hicolor/256x256/apps/${APP_NAME}.png" << 'EOF'
🛡️
EOF
    
    # 符合AppImage规范的根目录文件 - 修复重复复制问题
    if [ ! -f "appimage_build/AppDir/${APP_NAME}.desktop" ]; then
        cp "appimage_build/AppDir/usr/share/applications/${APP_NAME}.desktop" "appimage_build/AppDir/" 2>/dev/null || true
    fi
    if [ ! -f "appimage_build/AppDir/${APP_NAME}.png" ]; then
        cp "appimage_build/AppDir/usr/share/icons/hicolor/256x256/apps/${APP_NAME}.png" "appimage_build/AppDir/" 2>/dev/null || true
    fi
}

# 创建AppRun脚本
create_apprun() {
    print_step "创建AppRun启动脚本..."
    
    cat > "appimage_build/AppDir/AppRun" << 'APPRUN_SCRIPT'
#!/bin/bash
# SpecProbe AppRun启动脚本

HERE="$(dirname "$(readlink -f "${0}")")"
export APPDIR="${HERE}"

# 设置环境变量
export PATH="${HERE}/usr/bin:${PATH}"

# 设置CodeQL环境（如果存在）
if [ -f "${HERE}/usr/lib/codeql/codeql" ]; then
    export CODEQL_HOME="${HERE}/usr/lib/codeql"
    export CODEQL_PATH="${HERE}/usr/lib/codeql/codeql"
    export PATH="${HERE}/usr/lib/codeql:${PATH}"
fi

# 设置工具环境
if [ -d "${HERE}/usr/lib/tools" ]; then
    export SCANNER_TOOLS_DIR="${HERE}/usr/lib/tools"
    export SCANNER_APPIMAGE_DIR="${HERE}"
fi

# 设置工作目录为临时目录（解决只读问题）
export SCANNER_WORK_DIR=$(mktemp -d -t specprobe-XXXXXX)
if [ -d "${HERE}/usr/lib/tools" ]; then
    cp -r "${HERE}/usr/lib/tools"/* "${SCANNER_WORK_DIR}/" 2>/dev/null || true
fi

# 清理函数
cleanup() {
    if [ -n "$SCANNER_WORK_DIR" ] && [ -d "$SCANNER_WORK_DIR" ]; then
        rm -rf "$SCANNER_WORK_DIR"
    fi
}
trap cleanup EXIT

# 处理特殊调用
case "$1" in
    "main.sh"|"./main.sh")
        # 模拟原始main.sh调用
        shift
        cd "$SCANNER_WORK_DIR"
        exec "${HERE}/usr/bin/main.sh" "$@"
        ;;
    "codeql")
        # 直接调用CodeQL
        if [ -f "${HERE}/usr/lib/codeql/codeql" ]; then
            shift
            exec "${HERE}/usr/lib/codeql/codeql" "$@"
        else
            echo "CodeQL not integrated in this AppImage"
            exit 1
        fi
        ;;
    "tools")
        # 工具目录访问
        if [ -d "${HERE}/usr/lib/tools" ]; then
            echo "Tools directory: ${HERE}/usr/lib/tools"
            ls -la "${HERE}/usr/lib/tools"
        else
            echo "Tools not found"
            exit 1
        fi
        ;;
    "--help"|"-h")
        cat << 'HELP_EOF'
SpecProbe CPU漏洞检测工具 v3.0.0

用法:
  SpecProbe                    # 启动GUI界面
  SpecProbe main.sh [options]  # 命令行模式
  SpecProbe codeql [commands]  # 使用集成的CodeQL
  SpecProbe tools              # 查看集成的工具
  SpecProbe --version          # 显示版本信息

功能特性:
  🛡️  CPU硬件漏洞检测 (Spectre/Meltdown/EntryBleed)
  🔍 多语言源代码静态分析
  🧪 动态PoC验证
  📊 可视化报告导出
  🌍 多语言支持

更多信息: https://github.com/yourproject/specprobe
HELP_EOF
        ;;
    "--version"|"-v")
        echo "SpecProbe v3.0.0"
        if [ -f "${HERE}/usr/lib/codeql/codeql" ]; then
            echo "CodeQL: $("${HERE}/usr/lib/codeql/codeql" version 2>/dev/null | head -1)"
        fi
        if [ -d "${HERE}/usr/lib/tools/compiled" ]; then
            poc_count=$(ls "${HERE}/usr/lib/tools/compiled" 2>/dev/null | wc -l)
            echo "PoC工具: ${poc_count} 个"
        fi
        ;;
    *)
        # 默认启动GUI
        cd "$SCANNER_WORK_DIR"
        exec "${HERE}/usr/bin/SpecProbe" "$@"
        ;;
esac
APPRUN_SCRIPT

    chmod +x "appimage_build/AppDir/AppRun"
}

# 下载AppImageTool
download_appimagetool() {
    if [ ! -f "appimagetool-x86_64.AppImage" ]; then
        print_step "下载AppImageTool..."
        
        APPIMAGETOOL_URL="https://github.com/AppImage/AppImageKit/releases/download/continuous/appimagetool-x86_64.AppImage"
        
        if [ "$USE_CURL" = true ]; then
            curl -L -o "appimagetool-x86_64.AppImage" "$APPIMAGETOOL_URL"
        else
            wget -O "appimagetool-x86_64.AppImage" "$APPIMAGETOOL_URL"
        fi
        
        chmod +x "appimagetool-x86_64.AppImage"
        print_success "AppImageTool下载完成"
    else
        print_info "使用现有的AppImageTool"
    fi
}

# 构建最终AppImage
build_appimage() {
    print_step "构建最终AppImage..."
    
    # 确定输出文件名
    if [ "$CODEQL_INTEGRATED" = true ]; then
        OUTPUT_NAME="${APP_NAME}-${APP_VERSION}-complete-x86_64.AppImage"
        print_info "构建完整版本（包含CodeQL）"
    else
        OUTPUT_NAME="${APP_NAME}-${APP_VERSION}-standard-x86_64.AppImage"
        print_info "构建标准版本"
    fi
    
    # 设置AppImage环境变量
    export ARCH=x86_64
    export VERSION="$APP_VERSION"
    
    # 构建AppImage
    ./appimagetool-x86_64.AppImage "appimage_build/AppDir" "$OUTPUT_NAME"
    
    if [ -f "$OUTPUT_NAME" ]; then
        chmod +x "$OUTPUT_NAME"
        print_success "AppImage构建完成: $OUTPUT_NAME"
        
        # 显示文件信息
        FILE_SIZE=$(du -h "$OUTPUT_NAME" | cut -f1)
        print_info "文件大小: $FILE_SIZE"
        
        return 0
    else
        print_error "AppImage构建失败"
        return 1
    fi
}

# 验证构建结果
verify_build() {
    print_step "验证构建结果..."
    
    if [ -f "$OUTPUT_NAME" ]; then
        # 基本验证
        if ./"$OUTPUT_NAME" --version >/dev/null 2>&1; then
            print_success "基本功能验证通过"
        else
            print_warning "基本功能验证失败"
        fi
        
        # CodeQL验证
        if [ "$CODEQL_INTEGRATED" = true ]; then
            if ./"$OUTPUT_NAME" codeql version >/dev/null 2>&1; then
                print_success "CodeQL集成验证通过"
            else
                print_warning "CodeQL集成验证失败"
            fi
        fi
        
        # 工具验证
        if ./"$OUTPUT_NAME" tools >/dev/null 2>&1; then
            print_success "工具集成验证通过"
        else
            print_warning "工具集成验证失败"
        fi
        
        return 0
    else
        print_error "构建文件不存在"
        return 1
    fi
}

# 生成使用说明
generate_usage_info() {
    print_step "生成使用说明..."
    
    cat > "${APP_NAME}-README.txt" << EOF
SpecProbe CPU漏洞检测工具 v${APP_VERSION}
=====================================

文件信息:
- 主程序: ${OUTPUT_NAME}
- 版本: ${APP_VERSION}
- 构建时间: $(date)
- 文件大小: $(du -h "$OUTPUT_NAME" | cut -f1)

功能特性:
✅ CPU硬件漏洞检测 (Spectre/Meltdown/EntryBleed等)
✅ 多语言源代码静态分析 (C/C++/Java/Python/JS等)  
✅ 动态PoC漏洞验证
✅ 智能翻译与风险评估
✅ 可视化HTML报告导出
✅ 现代化用户界面
$([ "$CODEQL_INTEGRATED" = true ] && echo "✅ 集成CodeQL工具链" || echo "⚠️  需要单独安装CodeQL")

使用方法:
1. 基本使用:
   ./${OUTPUT_NAME}

2. 命令行模式:
   sudo ./${OUTPUT_NAME} main.sh [选项]

3. 查看帮助:
   ./${OUTPUT_NAME} --help

4. 查看版本:
   ./${OUTPUT_NAME} --version

$([ "$CODEQL_INTEGRATED" = true ] && cat << 'CODEQL_INFO'
5. 使用集成的CodeQL:
   ./${OUTPUT_NAME} codeql version
   ./${OUTPUT_NAME} codeql database create --language=cpp mydb
CODEQL_INFO
)

6. 查看集成工具:
   ./${OUTPUT_NAME} tools

系统要求:
- Linux x86_64 系统
- glibc 2.17 或更高版本
- sudo权限（用于硬件检测）
- 至少512MB可用内存

注意事项:
- 首次运行可能需要几秒钟初始化
- 硬件漏洞扫描需要管理员权限
- 源代码分析功能需要CodeQL$([ "$CODEQL_INTEGRATED" = false ] && echo "（请单独安装）")
- 大型项目分析可能需要较长时间

故障排除:
- 如果程序无法启动，请检查系统兼容性
- 如果缺少权限，请使用sudo运行相关功能
- 如果遇到问题，请查看控制台输出

更多信息:
- 项目主页: https://github.com/yourproject/specprobe  
- 问题反馈: https://github.com/yourproject/specprobe/issues
- 文档: https://github.com/yourproject/specprobe/wiki

构建信息:
- Python版本: $(python3 --version)
- PyQt5版本: $(pip show PyQt5 | grep Version | cut -d' ' -f2)
- 构建环境: $(uname -a)
EOF

    print_success "使用说明已生成: ${APP_NAME}-README.txt"
}

# 清理构建文件
cleanup_build() {
    print_step "清理构建文件..."
    
    read -p "是否清理构建文件？(y/N): " cleanup_confirm
    if [[ $cleanup_confirm =~ ^[Yy]$ ]]; then
        rm -rf build_env dist *.spec __pycache__ .eggs *.egg-info
        rm -rf appimage_build
        print_success "构建文件已清理"
    else
        print_info "保留构建文件"
    fi
}

# 主函数
main() {
    echo -e "${CYAN}"
    cat << 'EOF'
╔══════════════════════════════════════════════════════════╗
║               SpecProbe 构建工具 v3.0.0                  ║
║                                                          ║
║        🛡️  专业CPU漏洞检测工具一键打包系统              ║
║                                                          ║
╚══════════════════════════════════════════════════════════╝
EOF
    echo -e "${NC}"
    
    # 执行构建步骤
    check_dependencies
    check_project_files
    check_codeql
    check_poc_tools
    setup_python_env
    build_python_app
    create_appimage_structure
    create_desktop_file
    create_apprun
    download_appimagetool
    
    if build_appimage && verify_build; then
        generate_usage_info
        
        echo -e "${GREEN}"
        cat << EOF

🎉 构建成功完成！

📦 输出文件: ${OUTPUT_NAME}
📄 说明文档: ${APP_NAME}-README.txt
📊 文件大小: $(du -h "$OUTPUT_NAME" | cut -f1)

🚀 立即测试:
   ./${OUTPUT_NAME} --version
   ./${OUTPUT_NAME} --help
   ./${OUTPUT_NAME}

$([ "$CODEQL_INTEGRATED" = true ] && echo "✅ 包含完整CodeQL支持" || echo "ℹ️  标准版本，需要单独安装CodeQL")

EOF
        echo -e "${NC}"
        
        cleanup_build
        
    else
        print_error "构建失败"
        exit 1
    fi
}

# 执行主函数
main "$@"