X-Git-Url: http://git.rot13.org/?p=goodfet;a=blobdiff_plain;f=firmware%2Fgoodfet.c;h=53591509c91d11d567ba031b7a09803f35a642a5;hp=00dd921ad7e21cca776f35312f9542f54c75d3d0;hb=6e4a1014e1ddfdf63dbf07fa3fdcaa1fdca99bab;hpb=0fb25630e9970a4e6d18e190c484f654490565b0

diff --git a/firmware/goodfet.c b/firmware/goodfet.c
index 00dd921..5359150 100644
--- a/firmware/goodfet.c
+++ b/firmware/goodfet.c
@@ -1,10 +1,16 @@
-//GOODFET Main File
-//Includes several applications.
+/*! \file goodfet.c
+  \author Travis Goodspeed
+  \brief Main module.
+  
+  This is the main module of the GoodFET, which calls the initialization
+  routines and delegates commands to the various applications.
+*/
+
 
 #include "platform.h"
 #include "command.h"
 #include "apps.h"
-
+#include "glitch.h"
 
 
 //LED on P1.0
@@ -16,27 +22,41 @@ void init(){
   
   //LED out and on.
   PLEDDIR |= PLEDPIN;
-  PLEDOUT |= PLEDPIN;
+  PLEDOUT &= ~PLEDPIN;
   
   //Setup clocks, unique to each '430.
   msp430_init_dco();
   msp430_init_uart();
   
+  //DAC should be at full voltage if it exists.
+  #ifdef DAC12IR
+  glitchvoltages(0xfff,0xfff);
+  #endif
+  
   //Enable Interrupts.
   //eint();
 }
 
+
 //! Handle a command.
 void handle(unsigned char app,
 	    unsigned char verb,
-	    unsigned char len){
+	    unsigned long len){
+  //debugstr("GoodFET");
+  P1OUT&=~1;
   switch(app){
+  case GLITCH:
+    glitchhandle(app,verb,len);
+    break;
   case MONITOR:
     monitorhandle(app,verb,len);
     break;
   case SPI:
     spihandle(app,verb,len);
     break;
+  case AVR:
+    avrhandle(app,verb,len);
+    break;
   case I2CAPP:
     i2chandle(app,verb,len);
     break;
@@ -50,11 +70,13 @@ void handle(unsigned char app,
     jtag430x2handle(app,verb,len);
     break;
   default:
-    #ifdef HANDLEOTHER
-    HANDLEOTHER(app,verb,len);
-    #else
-    txdata(app,NOK,0);
-    #endif
+    if(pluginhandle){
+      pluginhandle(app,verb,len);
+    }else{
+      debugstr("Plugin missing.");
+      debughex(app);
+      txdata(app,NOK,0);
+    }
     break;
   }
 }
@@ -63,24 +85,58 @@ void handle(unsigned char app,
 int main(void)
 {
   volatile unsigned int i;
-  unsigned char app, verb, len;
+  unsigned char app, verb;
+  unsigned long len;
+  // MSP reboot count for reset input & reboot function located at 0xFFFE
+  volatile unsigned int reset_count = 0;
+  void (*reboot_function)(void) = (void *) 0xFFFE;
   
   init();
-  
+
   txstring(MONITOR,OK,"http://goodfet.sf.net/");
   
+  
   //Command loop.  There's no end!
   while(1){
     //Magic 3
     app=serial_rx();
+
+	// If the app is the reset byte (0x80) increment and loop
+	if (app == 0x80) {
+		reset_count++;
+
+		if (reset_count > 4) {
+			// We could trigger the WDT with either:
+			// WDTCTL = 0;
+			// or
+			// WDTCTL = WDTPW + WDTCNTCL + WDTSSEL + 0x00;
+			// but instead we'll jump to our reboot function pointer
+			(*reboot_function)();
+		}
+
+		continue;
+	} else {
+		reset_count = 0;
+	}
+
     verb=serial_rx();
-    len=serial_rx();
+    //len=serial_rx();
+    len=rxword();
     
-    //Read data, if any
-    for(i=0;i<len;i++){
-      cmddata[i]=serial_rx();
+    //Read data, looking for buffer overflow.y
+    if(len<=CMDDATALEN){
+      for(i=0;i<len;i++){
+	cmddata[i]=serial_rx();
+      }
+      handle(app,verb,len);
+    }else{
+      //Listen to the blaberring.
+      for(i-0;i<len;i++)
+	serial_rx();
+      //Reply with an error.
+      debugstr("Buffer length exceeded.");
+      txdata(MONITOR,NOK,0);
     }
-    handle(app,verb,len);
   }
 }