X-Git-Url: http://git.rot13.org/?p=goodfet;a=blobdiff_plain;f=firmware%2Fgoodfet.c;h=4cddc1976e05877388845dd529016f2b5626c530;hp=c310c0c8e2dfbd54c9eaaf3bb6d3c31c3c4f8de3;hb=f4a6b415e762bcdc560f3ea655851d16f483ea5a;hpb=84de7aa6b2ef9e83c7d36c3b2fed78933113c38a

diff --git a/firmware/goodfet.c b/firmware/goodfet.c
index c310c0c..4cddc19 100644
--- a/firmware/goodfet.c
+++ b/firmware/goodfet.c
@@ -35,7 +35,7 @@ void init(){
 //! Handle a command.
 void handle(unsigned char app,
 	    unsigned char verb,
-	    unsigned char len){
+	    unsigned long len){
   //debugstr("GoodFET");
   switch(app){
   case MONITOR:
@@ -70,7 +70,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 +82,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);
   }
 }