Implemented enter (Uf2 usb) DFU mode admin message on NRF52 (#3045)

* Added enter DFU mode admin message * Trunk
meshtastic · Dec 27, 2023 · dbac2b1 · dbac2b1
1 parent 2b7eb1e
commit dbac2b1
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 3 deletions.
diff --git a/protobufs b/protobufs
diff --git a/src/main.h b/src/main.h
@@ -77,7 +77,7 @@ extern int heltec_version;
 // This will suppress the current delay and instead try to run ASAP.
 extern bool runASAP;
 
-void nrf52Setup(), esp32Setup(), nrf52Loop(), esp32Loop(), rp2040Setup(), clearBonds();
+void nrf52Setup(), esp32Setup(), nrf52Loop(), esp32Loop(), rp2040Setup(), clearBonds(), enterDfuMode();
 
 meshtastic_DeviceMetadata getDeviceMetadata();
 

diff --git a/src/mesh/generated/meshtastic/admin.pb.h b/src/mesh/generated/meshtastic/admin.pb.h
@@ -131,7 +131,7 @@ typedef struct _meshtastic_AdminMessage {
         bool get_node_remote_hardware_pins_request;
         /* Respond with the mesh's nodes with their available gpio pins for RemoteHardware module use */
         meshtastic_NodeRemoteHardwarePinsResponse get_node_remote_hardware_pins_response;
-        /* Enter (serial) DFU mode
+        /* Enter (UF2) DFU mode
      Only implemented on NRF52 currently */
         bool enter_dfu_mode_request;
         /* Set the owner for this node */

diff --git a/src/modules/AdminModule.cpp b/src/modules/AdminModule.cpp
@@ -188,6 +188,13 @@ bool AdminModule::handleReceivedProtobuf(const meshtastic_MeshPacket &mp, meshta
         reboot(DEFAULT_REBOOT_SECONDS);
         break;
     }
+    case meshtastic_AdminMessage_enter_dfu_mode_request_tag: {
+        LOG_INFO("Client is requesting to enter DFU mode.\n");
+#ifdef ARCH_NRF52
+        enterDfuMode();
+#endif
+        break;
+    }
 #ifdef ARCH_PORTDUINO
     case meshtastic_AdminMessage_exit_simulator_tag:
         LOG_INFO("Exiting simulator\n");

diff --git a/src/platform/nrf52/main-nrf52.cpp b/src/platform/nrf52/main-nrf52.cpp
@@ -1,4 +1,5 @@
 #include "configuration.h"
+#include <Adafruit_TinyUSB.h>
 #include <Adafruit_nRFCrypto.h>
 #include <SPI.h>
 #include <Wire.h>
@@ -214,4 +215,9 @@ void clearBonds()
         nrf52Bluetooth->setup();
     }
     nrf52Bluetooth->clearBonds();
+}
+
+void enterDfuMode()
+{
+    enterUf2Dfu();
 }