Added Dry Run Mode for testing.

Author: DeniedAccessLife

ArduinoStrike/ArduinoStrike/Arduino.cpp

@@ -1,6 +1,5 @@
 #include "Logger.h"
 #include "Arduino.h"
-
 extern volatile bool g_shouldExit;
 
 Arduino::~Arduino()
@@ -12,8 +11,13 @@ Arduino::~Arduino()
     }
 }
 
-Arduino::Arduino(LPCSTR name) : io_context(), serial_port(io_context)
+Arduino::Arduino(LPCSTR name, bool dry_run) : io_context(), serial_port(io_context), dry_run(dry_run)
 {
+    if (dry_run)
+    {
+        return;
+    }
+
     char port[100] = "\\.\\";
     Logger::LogMessage("Searching for device...");
 
@@ -275,6 +279,12 @@ bool Arduino::SelectDevice(const vector<DeviceInfo>& devices, LPSTR port)
 
 bool Arduino::WriteMessage(const string& message)
 {
+    if (dry_run)
+    {
+        Logger::LogMessage("Message sent: " + message);
+        return true;
+    }
+
     if (!serial_port.is_open())
     {
         Logger::LogMessage("Attempt to write to a closed serial port");

ArduinoStrike/ArduinoStrike/Arduino.h

@@ -4,7 +4,6 @@
 #include "Utils.h"
 #include <devguid.h>
 #include <SetupAPI.h>
-
 using namespace std;
 using namespace std::chrono;
 using namespace this_thread;
@@ -21,10 +20,11 @@ class Arduino
 {
 public:
     ~Arduino();
-    Arduino(LPCSTR name);
+    Arduino(LPCSTR name, bool dry_run = false);
     bool WriteMessage(const string& message);
 
 private:
+    bool dry_run;
     io_context io_context;
     serial_port serial_port;
 

ArduinoStrike/ArduinoStrike/ArduinoStrike.cpp

@@ -26,13 +26,32 @@ static BOOL WINAPI ConsoleHandler(DWORD dwCtrlType)
     return FALSE;
 }
 
-int main()
+int main(int argc, char* argv[])
 {
     Logger::Init();
 
     Utils utils;
     Config config;
-
+    
+    Logger::LogMessage("Main: argc = " + to_string(argc));
+    for (int i = 0; i < argc; i++)
+    {
+        Logger::LogMessage("Main: argv[" + to_string(i) + "] = " + string(argv[i]));
+    }
+    
+    bool dry_run = false;
+    for (int i = 1; i < argc; i++)
+    {
+        string arg = argv[i];
+        Logger::LogMessage("Main: Checking argument: " + arg);
+        
+        if (arg == "--dry-run" || arg == "-d")
+        {
+            dry_run = true;
+            Logger::LogMessage("Main: Dry run mode enabled via command line argument!");
+        }
+    }
+    
     if (!SetConsoleCtrlHandler(ConsoleHandler, TRUE))
     {
         Logger::LogMessage("Failed to set console control handler!", boost::log::trivial::error);
@@ -44,7 +63,7 @@ int main()
 
     try
     {
-        arduino = new Arduino("Arduino Leonardo");
+        arduino = new Arduino("Arduino Leonardo", dry_run);
     }
     catch (const runtime_error& e)
     {