@burtthecoder/mcp-dnstwist 1.0.3 → 1.0.5

package/README.md

@@ -38,7 +38,7 @@ npx -y @smithery/cli install @burtthecoder/mcp-dnstwist --client claude
 
 2. Install the server globally via npm:
 ```bash
-npm install -g mcp-dnstwist
+npm install -g @burtthecoder/mcp-dnstwist
 ```
 
 3. Add to your Claude Desktop configuration file:

package/build/index.js

@@ -2,32 +2,16 @@
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { CallToolRequestSchema, ErrorCode, ListToolsRequestSchema, McpError, } from '@modelcontextprotocol/sdk/types.js';
-import { exec } from 'child_process';
+import { execFile } from 'child_process';
 import { promisify } from 'util';
-const execAsync = promisify(exec);
+import { validateFuzzArgs } from './validate.js';
+// execFile runs docker directly with an argument array. No shell is involved,
+// so tool input can never be interpreted as shell syntax (CWE-78, issue #13).
+const execFileAsync = promisify(execFile);
 const DOCKER_IMAGE = 'elceef/dnstwist';
-function isFuzzDomainArgs(args) {
-    if (!args || typeof args !== 'object')
-        return false;
-    const a = args;
-    return typeof a.domain === 'string' &&
-        (a.nameservers === undefined || typeof a.nameservers === 'string') &&
-        (a.threads === undefined || typeof a.threads === 'number') &&
-        (a.format === undefined || ['json', 'csv', 'list'].includes(a.format)) &&
-        (a.registered_only === undefined || typeof a.registered_only === 'boolean') &&
-        (a.mxcheck === undefined || typeof a.mxcheck === 'boolean') &&
-        (a.ssdeep === undefined || typeof a.ssdeep === 'boolean') &&
-        (a.banners === undefined || typeof a.banners === 'boolean');
-}
 class DnstwistServer {
     constructor() {
-        this.server = new Server({
-            name: 'dnstwist-server',
-            version: '0.1.0',
-            capabilities: {
-                tools: {}
-            }
-        });
+        this.server = new Server({ name: 'dnstwist-server', version: '0.1.0' }, { capabilities: { tools: {} } });
         this.setupToolHandlers();
         this.server.onerror = (error) => console.error('[MCP Error]', error);
         process.on('SIGINT', async () => {
@@ -40,10 +24,10 @@ class DnstwistServer {
             process.exit(1);
         });
     }
-    async execCommand(command) {
-        console.error('Executing command:', command);
+    async execCommand(file, args) {
+        console.error('Executing:', file, JSON.stringify(args));
         try {
-            const result = await execAsync(command, {
+            const result = await execFileAsync(file, args, {
                 maxBuffer: 10 * 1024 * 1024
             });
             console.error('Command output:', result.stdout);
@@ -60,19 +44,19 @@ class DnstwistServer {
         try {
             console.error('Checking Docker...');
             try {
-                await this.execCommand('docker --version');
+                await this.execCommand('docker', ['--version']);
             }
             catch (error) {
                 throw new Error('Docker is not installed or not running. Please install Docker and try again.');
             }
             console.error('Checking if dnstwist image exists...');
             try {
-                await this.execCommand(`docker image inspect ${DOCKER_IMAGE}`);
+                await this.execCommand('docker', ['image', 'inspect', DOCKER_IMAGE]);
                 console.error('Dnstwist image found');
             }
             catch (error) {
                 console.error('Dnstwist image not found, pulling...');
-                await this.execCommand(`docker pull ${DOCKER_IMAGE}`);
+                await this.execCommand('docker', ['pull', DOCKER_IMAGE]);
                 console.error('Dnstwist image pulled successfully');
             }
         }
@@ -144,10 +128,7 @@ class DnstwistServer {
                 if (request.params.name !== 'fuzz_domain') {
                     throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${request.params.name}`);
                 }
-                if (!isFuzzDomainArgs(request.params.arguments)) {
-                    throw new McpError(ErrorCode.InvalidParams, 'Invalid arguments for fuzz_domain');
-                }
-                const { domain, nameservers = '1.1.1.1', threads = 50, format = 'json', registered_only = true, mxcheck = true, ssdeep = false, banners = true } = request.params.arguments;
+                const { domain, nameservers, threads, format, registered_only, mxcheck, ssdeep, banners } = validateFuzzArgs(request.params.arguments);
                 // Build command arguments
                 const args = [
                     domain,
@@ -166,7 +147,11 @@ class DnstwistServer {
                 if (banners)
                     args.push('-b');
                 // Run dnstwist in Docker
-                const { stdout, stderr } = await this.execCommand(`docker run --rm ${DOCKER_IMAGE} ${args.join(' ')}`);
+                const { stdout } = await this.execCommand('docker', ['run', '--rm', DOCKER_IMAGE, ...args]);
+                // csv and list output are returned as dnstwist printed them.
+                if (format !== 'json') {
+                    return { content: [{ type: 'text', text: stdout }] };
+                }
                 let results;
                 try {
                     results = JSON.parse(stdout);

package/build/validate.js

@@ -0,0 +1,70 @@
+import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
+import { isIP } from 'net';
+import { domainToASCII } from 'url';
+// A hostname: dot-separated labels of letters, digits and inner hyphens, ending
+// in an alphabetic or punycode TLD. It can't start with "-", so it can't be
+// mistaken for a dnstwist option either.
+const HOSTNAME = /^(?=.{1,253}$)(?:[a-z0-9](?:[a-z0-9-]{0,61}[a-z0-9])?\.)+(?:[a-z]{2,63}|xn--[a-z0-9-]{1,59})$/i;
+const FORMATS = ['json', 'csv', 'list'];
+const MAX_NAMESERVERS = 10;
+function invalid(message) {
+    throw new McpError(ErrorCode.InvalidParams, `Invalid arguments for fuzz_domain: ${message}`);
+}
+/** Lowercase ASCII (punycode) form of a hostname, or null if it isn't one. */
+function asciiHostname(value) {
+    const trimmed = value.trim();
+    // domainToASCII silently drops inner tabs and newlines ("a.com\nb" becomes
+    // "a.comb"), so reject whitespace and control characters first.
+    if (/[\s\x00-\x1f\x7f]/.test(trimmed))
+        return null;
+    // Handles Unicode (IDN) names; returns '' for invalid input.
+    const ascii = domainToASCII(trimmed.replace(/\.$/, ''));
+    return HOSTNAME.test(ascii) ? ascii : null;
+}
+function optionalBoolean(a, key, fallback) {
+    if (a[key] === undefined)
+        return fallback;
+    if (typeof a[key] !== 'boolean')
+        invalid(`${key} must be a boolean`);
+    return a[key];
+}
+/**
+ * Validate and normalise fuzz_domain arguments. Everything returned is safe to
+ * pass to dnstwist as a separate argv entry: no value can start with "-".
+ */
+export function validateFuzzArgs(args) {
+    if (!args || typeof args !== 'object')
+        invalid('arguments must be an object');
+    const a = args;
+    const domain = typeof a.domain === 'string' ? asciiHostname(a.domain) : null;
+    if (!domain)
+        invalid('domain must be a domain name such as example.com');
+    const rawNameservers = a.nameservers === undefined ? '1.1.1.1' : a.nameservers;
+    if (typeof rawNameservers !== 'string')
+        invalid('nameservers must be a string');
+    const servers = rawNameservers.split(',').map((ns) => {
+        const trimmed = ns.trim();
+        return isIP(trimmed) !== 0 ? trimmed : asciiHostname(trimmed);
+    });
+    if (servers.length > MAX_NAMESERVERS || servers.some((ns) => !ns)) {
+        invalid(`nameservers must be up to ${MAX_NAMESERVERS} comma-separated IP addresses or hostnames`);
+    }
+    const threads = a.threads === undefined ? 50 : a.threads;
+    if (typeof threads !== 'number' || !Number.isInteger(threads) || threads < 1 || threads > 100) {
+        invalid('threads must be an integer from 1 to 100');
+    }
+    const format = a.format === undefined ? 'json' : a.format;
+    if (!FORMATS.includes(format)) {
+        invalid(`format must be one of ${FORMATS.join(', ')}`);
+    }
+    return {
+        domain: domain,
+        nameservers: servers.join(','),
+        threads: threads,
+        format: format,
+        registered_only: optionalBoolean(a, 'registered_only', true),
+        mxcheck: optionalBoolean(a, 'mxcheck', true),
+        ssdeep: optionalBoolean(a, 'ssdeep', false),
+        banners: optionalBoolean(a, 'banners', true),
+    };
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@burtthecoder/mcp-dnstwist",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "MCP server for dnstwist - DNS fuzzing to detect typosquatting, phishing and corporate espionage",
   "main": "build/index.js",
   "type": "module",
@@ -14,18 +14,19 @@
   ],
   "scripts": {
     "build": "tsc && chmod +x build/index.js",
-    "prepublishOnly": "npm run build"
+    "prepublishOnly": "npm run build",
+    "test": "npm run build && node --test tests/*.test.mjs"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^0.4.0"
+    "@modelcontextprotocol/sdk": "^1.32.1"
   },
   "devDependencies": {
-    "@types/node": "^20.0.0",
-    "typescript": "^5.0.0"
+    "@types/node": "^20.19.43",
+    "typescript": "^5.9.3"
   },
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/burtthecoder/mcp-dnstwist.git"
+    "url": "git+https://github.com/w0h1v/mcp-dnstwist.git"
   },
   "keywords": [
     "mcp",
@@ -41,6 +42,6 @@
   "author": "burtmacklin",
   "license": "MIT",
   "engines": {
-    "node": ">=18"
+    "node": ">=20"
   }
 }