usb communication (device side) refactoring

[proxmark3-svn] / bootrom / bootrom.c

diff --git a/bootrom/bootrom.c b/bootrom/bootrom.c
index c2c81a9da1594ce4bb1205ac0e89dae9f954b085..b2500821728daf00d5ddcad767f41867c1ce81db 100644 (file)
--- a/bootrom/bootrom.c
+++ b/bootrom/bootrom.c
@@ -6,17 +6,15 @@
 // Main code for the bootloader
 //-----------------------------------------------------------------------------
 
 // Main code for the bootloader
 //-----------------------------------------------------------------------------
 

-#include <proxmark3.h>
+#include "proxmark3.h"

 #include "usb_cdc.h"
 #include "usb_cdc.h"

-#include "cmd.h"
-//#include "usb_hid.h"

 
 void DbpString(char *str) {
 
 void DbpString(char *str) {

-  byte_t len = 0;
+  uint8_t len = 0;

   while (str[len] != 0x00) {
     len++;
   }
   while (str[len] != 0x00) {
     len++;
   }

-  cmd_send(CMD_DEBUG_PRINT_STRING,len,0,0,(byte_t*)str,len);
+  cmd_send(CMD_DEBUG_PRINT_STRING,len,0,0,(uint8_t*)str,len);

 }
 
 struct common_area common_area __attribute__((section(".commonarea")));
 }
 
 struct common_area common_area __attribute__((section(".commonarea")));


@@ -89,15 +87,10 @@ static void Fatal(void)
   for(;;);
 }
 
   for(;;);
 }
 

-void UsbPacketReceived(uint8_t *packet, int len) {
+void UsbPacketReceived(UsbCommand *c) {

   int i, dont_ack=0;
   int i, dont_ack=0;

-  UsbCommand* c = (UsbCommand *)packet;

   volatile uint32_t *p;
   
   volatile uint32_t *p;
   

-  if(len != sizeof(UsbCommand)) {
-    Fatal();
-  }
-  

   uint32_t arg0 = (uint32_t)c->arg[0];
   
   switch(c->cmd) {
   uint32_t arg0 = (uint32_t)c->arg[0];
   
   switch(c->cmd) {


@@ -199,21 +192,17 @@ static void flash_mode(int externally_entered)
        start_addr = 0;
        end_addr = 0;
        bootrom_unlocked = 0;
        start_addr = 0;
        end_addr = 0;
        bootrom_unlocked = 0;

-  byte_t rx[sizeof(UsbCommand)];
-       size_t rx_len;
+    UsbCommand rx;

 
 

-  usb_enable();
-  for (volatile size_t i=0; i<0x100000; i++);
+       usb_enable();
+       for (volatile size_t i=0; i<0x100000; i++) {};

 
        for(;;) {
                WDT_HIT();
 
 
        for(;;) {
                WDT_HIT();
 

-    if (usb_poll()) {
-      rx_len = usb_read(rx,sizeof(UsbCommand));
-      if (rx_len) {
-        UsbPacketReceived(rx,rx_len);
-      }
-    }
+               if (cmd_receive(&rx)) {
+                       UsbPacketReceived(&rx);
+               }

 
                if(!externally_entered && !BUTTON_PRESS()) {
                        /* Perform a reset to leave flash mode */
 
                if(!externally_entered && !BUTTON_PRESS()) {
                        /* Perform a reset to leave flash mode */