X-Git-Url: http://git.rot13.org/?p=goodfet;a=blobdiff_plain;f=firmware%2Fgoodfet.c;h=b9485aa5b87d233347adb8b8bf56b5428314ef66;hp=d09683cbda345a1d7267385e6b497b7db0ae0d51;hb=d64502ed4204f313ae89aa4e2b6c554d5a9c6563;hpb=ae09939eb8c62c83f244527e7916cee5f9145e6c

diff --git a/firmware/goodfet.c b/firmware/goodfet.c
index d09683c..b9485aa 100644
--- a/firmware/goodfet.c
+++ b/firmware/goodfet.c
@@ -32,10 +32,12 @@ void init(){
   //eint();
 }
 
+
 //! Handle a command.
 void handle(unsigned char app,
 	    unsigned char verb,
-	    unsigned char len){
+	    unsigned long len){
+  //debugstr("GoodFET");
   switch(app){
   case MONITOR:
     monitorhandle(app,verb,len);
@@ -43,6 +45,9 @@ void handle(unsigned char app,
   case SPI:
     spihandle(app,verb,len);
     break;
+  case AVR:
+    avrhandle(app,verb,len);
+    break;
   case I2CAPP:
     i2chandle(app,verb,len);
     break;
@@ -56,11 +61,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.");
+      txdata(app,NOK,0);
+    }
+      
     break;
   }
 }
@@ -69,7 +76,8 @@ void handle(unsigned char app,
 int main(void)
 {
   volatile unsigned int i;
-  unsigned char app, verb, len;
+  unsigned char app, verb;
+  unsigned long len;
   
   init();
   
@@ -80,13 +88,23 @@ int main(void)
     //Magic 3
     app=serial_rx();
     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);
   }
 }