From 7da73d98fe0b4bfdf28b4239bf4aa2a440bd9444 Mon Sep 17 00:00:00 2001 From: g0t mi1k Date: Thu, 5 Feb 2026 09:14:45 +0000 Subject: [PATCH] Add --debug flag to be more verbose --- MetasploitMCP.py | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) diff --git a/MetasploitMCP.py b/MetasploitMCP.py index e7606f7..94a0868 100644 --- a/MetasploitMCP.py +++ b/MetasploitMCP.py @@ -24,19 +24,21 @@ from starlette.routing import Mount, Route, Router # --- Configuration & Constants --- -logging.basicConfig( - level=os.environ.get("LOG_LEVEL", "INFO").upper(), - format='%(asctime)s - %(name)s - %(levelname)s - %(message)s' -) -logger = logging.getLogger("metasploit_mcp_server") -session_shell_type: Dict[str, str] = {} - # Metasploit Connection Config (from environment variables) MSF_PASSWORD = os.getenv('MSF_PASSWORD', 'yourpassword') MSF_SERVER = os.getenv('MSF_SERVER', '127.0.0.1') MSF_PORT_STR = os.getenv('MSF_PORT', '55553') MSF_SSL_STR = os.getenv('MSF_SSL', 'false') -PAYLOAD_SAVE_DIR = os.environ.get('PAYLOAD_SAVE_DIR', str(pathlib.Path.home() / "payloads")) +PAYLOAD_SAVE_DIR = os.getenv('PAYLOAD_SAVE_DIR', str(pathlib.Path.home() / "payloads")) +LOG_LEVEL = os.getenv('LOG_LEVEL', 'info') + +logging.basicConfig( + format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', +) +logger = logging.getLogger("metasploit_mcp_server") +logger.setLevel(LOG_LEVEL.upper()) + +session_shell_type: Dict[str, str] = {} # Timeouts and Polling Intervals (in seconds) DEFAULT_CONSOLE_READ_TIMEOUT = 15 # Default for quick console commands @@ -1718,8 +1720,13 @@ if __name__ == "__main__": parser.add_argument('--port', type=int, default=None, help='Port to listen on (default: find available from 8085)') parser.add_argument('--reload', action='store_true', help='Enable auto-reload (for development)') parser.add_argument('--find-port', action='store_true', help='Force finding an available port starting from --port or 8085') + parser.add_argument('--debug', action='store_true', help='Make output more verbose') args = parser.parse_args() + if args.debug: + LOG_LEVEL = 'debug' + logger.setLevel(LOG_LEVEL.upper()) + # Initialize MSF Client - Critical for server function try: initialize_msf_client() @@ -1756,5 +1763,5 @@ if __name__ == "__main__": host=args.host, port=selected_port, reload=args.reload, - log_level="info" + log_level=LOG_LEVEL.lower() )