pentesting 0.52.0 → 0.52.2

package/dist/main.js

@@ -349,7 +349,7 @@ var ORPHAN_PROCESS_NAMES = [
 
 // src/shared/constants/agent.ts
 var APP_NAME = "Pentest AI";
-var APP_VERSION = "0.52.0";
+var APP_VERSION = "0.52.2";
 var APP_DESCRIPTION = "Autonomous Penetration Testing AI Agent";
 var LLM_ROLES = {
   SYSTEM: "system",
@@ -3150,13 +3150,21 @@ import { existsSync as existsSync4, readFileSync as readFileSync3, writeFileSync
 import { join as join3 } from "path";
 function extractFingerprint(tool, command) {
   const cmd = command || "";
+  let effectiveTool = tool.toLowerCase();
+  if (effectiveTool === "run_cmd" || effectiveTool === "run_background") {
+    const firstWord = cmd.trim().split(/\s+/)[0];
+    if (firstWord && !firstWord.startsWith("-")) {
+      effectiveTool = firstWord.toLowerCase();
+    }
+  }
   const targetMatch = cmd.match(
     /(?::\/\/|@)([\w.\-]+(?::\d+)?)|\b(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})(?::\d+)?\b/
   );
   const target = targetMatch?.[1] || targetMatch?.[2] || "";
-  const wordlistMatch = cmd.match(
-    /(?:-[wPUL]\s+|--(?:wordlist|passwords?|usernames?)(?:=|\s+))([^\s]+)/i
-  );
+  let wordlistMatch = cmd.match(/(?:-[wPUL]\s+)([^\s]+)/);
+  if (!wordlistMatch) {
+    wordlistMatch = cmd.match(/(?:--(?:wordlist|passwords?|usernames?)(?:=|\s+))([^\s]+)/i);
+  }
   const wordlist = wordlistMatch?.[1] || "";
   const portMatch = cmd.match(/(?:-p\s+|--port(?:=|\s+))(\S+)/);
   const port = portMatch?.[1] || "";
@@ -3178,7 +3186,7 @@ function extractFingerprint(tool, command) {
     const m = cmd.match(p);
     return m?.[0]?.trim() || "";
   }).filter(Boolean).join(" ");
-  return { tool: tool.toLowerCase(), target, wordlist, flags, port };
+  return { tool: effectiveTool, target, wordlist, flags, port };
 }
 function fingerprintsMatch(a, b) {
   return a.tool === b.tool && a.target === b.target && a.wordlist === b.wordlist && a.flags === b.flags && a.port === b.port;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pentesting",
-  "version": "0.52.0",
+  "version": "0.52.2",
   "description": "Autonomous Penetration Testing AI Agent",
   "type": "module",
   "main": "dist/main.js",
@@ -19,7 +19,7 @@
     "dev:tsx": "tsx src/platform/tui/main.tsx",
     "build": "tsup",
     "start": "node dist/main.js",
-    "test": "mkdir -p /tmp/pentesting-vitest && TMPDIR=/tmp/pentesting-vitest vitest run",
+    "test": "mkdir -p .vitest && TMPDIR=.vitest vitest run && rm -rf .vitest .pentesting",
     "test:watch": "vitest",
     "lint": "tsc --noEmit",
     "prepublishOnly": "npm run build",
@@ -29,7 +29,7 @@
     "release:minor": "npm version minor && npm run build && npm run publish:token",
     "release:major": "npm version major && npm run build && npm run publish:token",
     "release:docker": "docker buildx build --platform linux/amd64,linux/arm64 -t agnusdei1207/pentesting:latest --push . && docker system prune -af",
-    "check": "TMPDIR=/tmp npm run test && npm run build && npm run release:docker && bash test.sh"
+    "check": "npm run test && npm run build && npm run release:docker && bash test.sh"
   },
   "repository": {
     "type": "git",