Try to use artifacts instead

dhinakg · Jun 23, 2024 · 6e6b1ae · 6e6b1ae
1 parent e8b75f9
commit 6e6b1ae
Showing 1 changed file with 103 additions and 34 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -26,11 +26,33 @@ jobs:
           name: macOS artifacts for ${{ matrix.os }}
           path: aastuff*
 
+  download-test-files:
+    runs-on: ubuntu-latest
+
+    steps:
+      - run: |
+          cat <<EOF >> links.txt
+          "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea"
+          aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024SummerSeed/fullrestores/052-34764/D5D3D10C-E557-4A46-8EBD-290411A228AA/iPhone16,2_18.0_22A5282m_Restore.ipsw"
+          aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024/Iris/mobileassets/003-49672/A1233F60-3D17-491B-803A-DB26E20695AE/com_apple_MobileAsset_UAF_Siri_Understanding/6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar
+          EOF
+          aria2c -x 16 -s 16 -j 16 --file-allocation=none -i links.txt
+      - uses: actions/upload-artifact@v4
+        with:
+          path: e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea
+      - uses: actions/upload-artifact@v4
+        with:
+          path: iPhone16,2_18.0_22A5282m_Restore.ipsw
+      - uses: actions/upload-artifact@v4
+        with:
+          path: 6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar
+
   test-get-key:
     strategy:
       matrix:
         python-version: ["3.10", "3.11", "3.12"]
     runs-on: macos-latest
+    needs: download-test-files
 
     steps:
       - uses: actions/checkout@v4
@@ -40,61 +62,108 @@ jobs:
           cache: "pip"
       - name: Install Python dependencies
         run: pip install -r requirements.txt
-      - name: Cache test files
-        id: cache
-        uses: actions/cache@v4
-        with:
-          key: test-get-key-files
-          path: |
-            tests
-      - name: Download test files
-        if: ${{ steps.cache.outputs.cache-hit != 'true' }}
-        run: |
-          brew install aria2
+      # - name: Cache test files
+      #   id: cache
+      #   uses: actions/cache@v4
+      #   with:
+      #     key: test-get-key-files
+      #     path: |
+      #       tests
+      # - name: Download test files
+      #   if: ${{ steps.cache.outputs.cache-hit != 'true' }}
+      #   run: |
+      #     brew install aria2
+      #     mkdir tests tests/macOS_15_beta_1_OTA tests/iOS_18_beta_1_IPSW/
+      #     # curl -L "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/macOS_15_beta_1_OTA/encrypted.aea
+      #     aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/macOS_15_beta_1_OTA/encrypted.aea
+      #     printf "$MACOS_OTA_TEST_KEY" > tests/macOS_15_beta_1_OTA/expected.txt
+      #     # curl -L "https://updates.cdn-apple.com/2024SummerSeed/fullrestores/052-34764/D5D3D10C-E557-4A46-8EBD-290411A228AA/iPhone16,2_18.0_22A5282m_Restore.ipsw" -o tests/iPhone_15PM_18.0_22A5282m.ipsw
+      #     aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024SummerSeed/fullrestores/052-34764/D5D3D10C-E557-4A46-8EBD-290411A228AA/iPhone16,2_18.0_22A5282m_Restore.ipsw" -o tests/iPhone_15PM_18.0_22A5282m.ipsw
+      #     unzip -p tests/iPhone_15PM_18.0_22A5282m.ipsw 090-29713-049.dmg.aea > tests/iOS_18_beta_1_IPSW/encrypted.aea
+      #     rm tests/iPhone_15PM_18.0_22A5282m.ipsw
+      #     printf "$IOS_IPSW_TEST_KEY" > tests/iOS_18_beta_1_IPSW/expected.txt
+      #   env:
+      #     MACOS_OTA_TEST_KEY: ${{ vars.MACOS_OTA_TEST_KEY }}
+      #     IOS_IPSW_TEST_KEY: ${{ vars.IOS_IPSW_TEST_KEY }}
+      - run: |
           mkdir tests tests/macOS_15_beta_1_OTA tests/iOS_18_beta_1_IPSW/
-          # curl -L "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/macOS_15_beta_1_OTA/encrypted.aea
-          aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/macOS_15_beta_1_OTA/encrypted.aea
           printf "$MACOS_OTA_TEST_KEY" > tests/macOS_15_beta_1_OTA/expected.txt
-          # curl -L "https://updates.cdn-apple.com/2024SummerSeed/fullrestores/052-34764/D5D3D10C-E557-4A46-8EBD-290411A228AA/iPhone16,2_18.0_22A5282m_Restore.ipsw" -o tests/iPhone_15PM_18.0_22A5282m.ipsw
-          aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024SummerSeed/fullrestores/052-34764/D5D3D10C-E557-4A46-8EBD-290411A228AA/iPhone16,2_18.0_22A5282m_Restore.ipsw" -o tests/iPhone_15PM_18.0_22A5282m.ipsw
-          unzip -p tests/iPhone_15PM_18.0_22A5282m.ipsw 090-29713-049.dmg.aea > tests/iOS_18_beta_1_IPSW/encrypted.aea
-          rm tests/iPhone_15PM_18.0_22A5282m.ipsw
           printf "$IOS_IPSW_TEST_KEY" > tests/iOS_18_beta_1_IPSW/expected.txt
         env:
           MACOS_OTA_TEST_KEY: ${{ vars.MACOS_OTA_TEST_KEY }}
           IOS_IPSW_TEST_KEY: ${{ vars.IOS_IPSW_TEST_KEY }}
+      - uses: actions/download-artifact@v4
+        with:
+          name: e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea
+          path: tests/macOS_15_beta_1_OTA/encrypted.aea
+      - uses: actions/download-artifact@v4
+        with:
+          name: iPhone16,2_18.0_22A5282m_Restore.ipsw
+          path: tests/iPhone_15PM_18.0_22A5282m.ipsw
+      - run: |
+          unzip -p tests/iPhone_15PM_18.0_22A5282m.ipsw 090-29713-049.dmg.aea > tests/iOS_18_beta_1_IPSW/encrypted.aea
+          rm tests/iPhone_15PM_18.0_22A5282m.ipsw
       - name: Run tests
         run: ./test_get_key.sh
 
   test-aastuff:
     runs-on: macos-latest
+    needs: download-test-files
 
     steps:
       - uses: actions/checkout@v4
-      - name: Cache test files
-        id: cache
-        uses: actions/cache@v4
-        with:
-          key: test-aastuff-files
-          path: |
-            tests
-      - name: Download test files
-        if: ${{ steps.cache.outputs.cache-hit != 'true' }}
-        run: |
-          brew install aria2
+      # - name: Cache test files
+      #   id: cache
+      #   uses: actions/cache@v4
+      #   with:
+      #     key: test-aastuff-files
+      #     path: |
+      #       tests
+      # - name: Download test files
+      #   if: ${{ steps.cache.outputs.cache-hit != 'true' }}
+      #   run: |
+      #     brew install aria2
+      #     mkdir tests tests/small tests/large
+      #     # This file uses a compressed inner layer
+      #     # curl -L "https://updates.cdn-apple.com/2024/Iris/mobileassets/003-49672/A1233F60-3D17-491B-803A-DB26E20695AE/com_apple_MobileAsset_UAF_Siri_Understanding/6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar" -o tests/small/encrypted.aea
+      #     aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024/Iris/mobileassets/003-49672/A1233F60-3D17-491B-803A-DB26E20695AE/com_apple_MobileAsset_UAF_Siri_Understanding/6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar" -o tests/small/encrypted.aea
+      #     printf "$SMALL_TEST_KEY" > tests/small/key.txt
+      #     # This file uses a raw inner layer
+      #     # curl -L "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/large/encrypted.aea
+      #     aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/large/encrypted.aea
+      #     printf "$LARGE_TEST_KEY" > tests/large/key.txt
+      #   env:
+      #     SMALL_TEST_KEY: ${{ vars.SMALL_TEST_KEY }}
+      #     LARGE_TEST_KEY: ${{ vars.LARGE_TEST_KEY }}
+      - run: |
           mkdir tests tests/small tests/large
-          # This file uses a compressed inner layer 
-          # curl -L "https://updates.cdn-apple.com/2024/Iris/mobileassets/003-49672/A1233F60-3D17-491B-803A-DB26E20695AE/com_apple_MobileAsset_UAF_Siri_Understanding/6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar" -o tests/small/encrypted.aea
-          aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024/Iris/mobileassets/003-49672/A1233F60-3D17-491B-803A-DB26E20695AE/com_apple_MobileAsset_UAF_Siri_Understanding/6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar" -o tests/small/encrypted.aea
           printf "$SMALL_TEST_KEY" > tests/small/key.txt
-          # This file uses a raw inner layer
-          # curl -L "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/large/encrypted.aea
-          aria2c -x 16 -s 16 -j 16 --file-allocation=none "https://updates.cdn-apple.com/2024SummerSeed/mobileassets/052-49061/CA7135A8-BAF6-4890-887C-35FB30C154D5/com_apple_MobileAsset_MacSoftwareUpdate/e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea" -o tests/large/encrypted.aea
           printf "$LARGE_TEST_KEY" > tests/large/key.txt
         env:
           SMALL_TEST_KEY: ${{ vars.SMALL_TEST_KEY }}
           LARGE_TEST_KEY: ${{ vars.LARGE_TEST_KEY }}
+      - uses: actions/download-artifact@v4
+        with:
+          name: 6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar
+          path: tests/small/encrypted.aea
+      - uses: actions/download-artifact@v4
+        with:
+          name: e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea
+          path: tests/large/encrypted.aea
       - name: Build
         run: make
       - name: Run tests
         run: ./test_extract.sh
+
+  delete-artifacts:
+    runs-on: ubuntu-latest
+    needs: [test-get-key, test-aastuff]
+    if: always()
+
+    steps:
+      - uses: geekyeggo/delete-artifact@v5
+        with:
+          name: |
+            e2de87f20576b2bdc021d36f74a2f836cf42afe576178388dfd0cde875f4f979.aea
+            iPhone16,2_18.0_22A5282m_Restore.ipsw
+            6FF3BAF0-FBEF-4C01-BB0E-30CD61DAFCC4.aar