From 50505aefb336a8e768559876c28badefdd9cafe8 Mon Sep 17 00:00:00 2001
From: dqy <1016751306@qq.com>
Date: Wed, 24 Apr 2024 20:04:12 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=94=B9=E9=85=8D=E7=BD=AE?=
 =?UTF-8?q?=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .github/workflows/detection.yml | 28 +++++++++++++++++++++-------
 requirements.txt                |  6 ++++++
 2 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/detection.yml b/.github/workflows/detection.yml
index db5bb67..7ed62b4 100644
--- a/.github/workflows/detection.yml
+++ b/.github/workflows/detection.yml
@@ -21,11 +21,25 @@ jobs:
     # needs: build # 确保安全检查在构建后执行
     steps:
       - uses: actions/checkout@v2
-      # - name: Install dependencies
-      #   run: pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
-      - name: Run Backdoor and Vulnerability Detection
-        uses: sangge/BackDoorBuster@feature/match # 使用BackDoorBuster仓库的根目录下的 action.yml
         with:
-          code_path: "./tests"
-          requirements_file: "./requirements.txt"
-          output_format: "txt"
+          repository: "sangge/BackDoorBuster"
+          ref: "feature/match"
+          path: "BackDoorBuster"
+      - name: Install dependencies
+        run: pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
+      # - name: Run Backdoor and Vulnerability Detection
+      #   uses: sangge/BackDoorBuster@feature/match # 使用BackDoorBuster仓库的根目录下的 action.yml
+      #   with:
+      #     code_path: "./tests"
+      #     requirements_file: "./requirements.txt"
+      #     output_format: "txt"
+      - name: Clone custom Git repository
+        run: |
+          git clone -b feature/match https://git.mamahaha.work/sangge/BackDoorBuster
+          cd BackDoorBuster/detection
+
+      - name: Run Backdoor Detection
+        run: python ${{ github.workspace }}/BackDoorBuster/detection/backdoor_detection.py ${{ github.workspace }}/BackDoorBuster/tests txt
+
+      - name: Run Requirements Detection
+        run: python ${{ github.workspace }}/BackDoorBuster/detection/requirements_detection.py ${{ github.workspace }}/BackDoorBuster/crawler/trans_extracted_data.txt ${{ github.workspace }}/BackDoorBuster/requirements.txt txt
diff --git a/requirements.txt b/requirements.txt
index e69de29..e3e14f8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+apache-airflow==2.8.0
+mlflow==2.5.1
+torch==2.0.0
+aiohttp==3.6.2
+flask==1.1.2
+numpy==1.18.5