feat: add backend logic for unlimited teams

Author: thelicato

src/server/api/controller.ts

@@ -20,6 +20,7 @@ import {
   GetFilesResponse,
   IGenericErrRes,
   IGenericResultRes,
+  NewTeamResponse,
   StartActivityRequest,
   StartBroadcastRequest,
   StartExploitAppRequest,
@@ -70,6 +71,19 @@ class APIController {
     }
   };
 
+  newTeam: RequestHandler = async (req: Request, res: Response<NewTeamResponse | IGenericErrRes>) => {
+    Logger.info(`Received ${req.method} request on ${req.path}`);
+    try {
+      const singleton = ManagerSingleton.getInstance();
+      const teamToken = singleton.addTeam();
+
+      res.json({ teamToken }).end();
+    } catch (error: any) {
+      Logger.error(`Error generating new Team Token: ${error}`);
+      res.status(500).json({ error: "An error occurred while generating a new Team Token" }).end();
+    }
+  };
+
   restartApp: RequestHandler = async (req: Request, res: Response<IGenericResultRes | IGenericErrRes>) => {
     Logger.info(`Received ${req.method} request on ${req.path}`);
     try {

src/server/api/routes.ts

@@ -73,6 +73,7 @@ export default (app: Router) => {
     APIController.enqueueStartExploitApp,
   );
   endpoint.post(E.RESET, checkFeatureEnabled(features.resetEnabled), APIController.reset);
+  endpoint.post(E.NEW_TEAM, checkFeatureEnabled(features.unlimitedTeams), APIController.newTeam);
   endpoint.get(E.FEATURES, APIController.features);
   endpoint.get(E.INFO, APIController.info);
   endpoint.post(E.RESTART, APIController.restartApp);

src/server/manager.ts

@@ -86,7 +86,8 @@ export class ManagerSingleton {
         startServiceEnabled: !(process.env.DROIDGROUND_START_SERVICE_DISABLED === "true"),
         terminalEnabled: !(process.env.DROIDGROUND_TERMINAL_DISABLED === "true"),
         resetEnabled: !(process.env.DROIDGROUND_RESET_DISABLED === "true"),
-        teamModeEnabled: teamNum > 0,
+        teamModeEnabled: teamNum > 0 || teamNum === -1,
+        unlimitedTeams: teamNum === -1,
         fridaType: process.env.DROIDGROUND_FRIDA_TYPE === "full" ? "full" : "jail",
         exploitAppDuration:
           isNaN(exploitAppDuration) || exploitAppDuration.trim().length === 0 ? 10 : parseInt(exploitAppDuration),
@@ -217,6 +218,19 @@ export class ManagerSingleton {
     return tokens;
   }
 
+  public addTeam(): string {
+    let teamToken = "";
+    while (true) {
+      teamToken = randomString(32);
+      const el = this.config.teams.find(t => t.token === teamToken);
+      if (el === undefined) {
+        break;
+      }
+    }
+    this.config.teams.push({ token: teamToken, exploitApps: [] });
+    return teamToken;
+  }
+
   private async checkPackage() {
     const adb = this.adb as Adb;
     const res = await adb.subprocess.noneProtocol.spawnWaitText(`pm list packages | grep ${this.config.packageName}`);

src/shared/api.ts

@@ -100,3 +100,7 @@ export interface GetAttackSurfaceRequest {
 export interface TeamTokenGenericRequest {
   teamToken: string;
 }
+
+export interface NewTeamResponse {
+  teamToken: string;
+}

src/shared/endpoints.ts

@@ -4,6 +4,7 @@ export const REST_API_ENDPOINTS = {
   INFO: "/info",
   RESTART: "/restart", // restart the target app
   ACTIVITY: "/activity",
+  NEW_TEAM: "/newTeam",
   BROADCAST: "/broadcast",
   SERVICE: "/service",
   DIALOGS: "/closeDialogs",

src/shared/types.ts

@@ -16,6 +16,7 @@ export interface DroidGroundFeatures {
   logcatEnabled: boolean;
   resetEnabled: boolean;
   teamModeEnabled: boolean;
+  unlimitedTeams: boolean;
   fridaType: "full" | "jail";
   exploitAppDuration: number;
   ipAddress: string;