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

diff --git a/firmware/goodfet.c b/firmware/goodfet.c
index c310c0c..b9485aa 100644
--- a/firmware/goodfet.c
+++ b/firmware/goodfet.c
@@ -32,10 +32,11 @@ 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:
@@ -44,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;
@@ -57,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;
   }
 }
@@ -70,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();
   
@@ -81,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);
   }
 }