Cleanup old Frida server downloads from computer cache

Author: pimterry

src/dynamic-dep-store.ts

@@ -51,4 +51,25 @@ export async function getDependencyStream<K extends readonly string[]>(options:
     });
 
     return resultStream;
+}
+
+export async function cleanupDependencies(options: {
+    config: HtkConfig,
+    keyPrefix: string,
+    versionToKeep: string,
+    ext: `.${string}`
+}) {
+    const depFiles = await fs.readDir(options.config.configPath);
+
+    await Promise.all(depFiles.map(async (depFile) => {
+        if (
+            depFile.startsWith(options.keyPrefix + '-') &&
+            (depFile.endsWith(options.ext) || depFile.includes(`${options.ext}.tmp-`)) &&
+            !depFile.includes(options.versionToKeep)
+        ) {
+            await fs.deleteFile(path.join(options.config.configPath, depFile)).catch((e) => {
+                console.warn(`Failed to delete old dependency file ${depFile}:`, e);
+            });
+        }
+    }));
 }

src/interceptors/frida/frida-android-interceptor.ts

@@ -6,7 +6,7 @@ import { Interceptor } from "..";
 import { HtkConfig } from '../../config';
 
 import { createAdbClient } from '../android/adb-commands';
-import { FridaHost, FridaTarget, killProcess } from './frida-integration';
+import { cleanupOldFridaServers, FridaHost, FridaTarget, killProcess } from './frida-integration';
 import {
     getAndroidFridaHosts,
     getAndroidFridaTargets,
@@ -101,6 +101,8 @@ export class FridaAndroidInterceptor implements Interceptor {
     }
 
     async deactivateAll(): Promise<void | {}> {
+        cleanupOldFridaServers(this.config).catch(console.warn);
+
         const allPorts = new Set([
             ...Object.keys(this.fridaServers),
             ...Object.keys(this.interceptedApps)

src/interceptors/frida/frida-integration.ts

@@ -68,6 +68,15 @@ export const getFridaServer = (
     });
 }
 
+export const cleanupOldFridaServers = async (config: HtkConfig) => {
+    await dynamicDeps.cleanupDependencies({
+        config,
+        keyPrefix: 'frida-server',
+        versionToKeep: FRIDA_VERSION,
+        ext: '.bin'
+    } as const);
+};
+
 class FridaScriptError extends CustomError {
     constructor(
         message: FridaJs.ScriptAgentErrorMessage