update to GoodFETMCPCANCommunication.py to allow for id sweeps

[goodfet] / client / GoodFETMCPCANCommunication.py

diff --git a/client/GoodFETMCPCANCommunication.py b/client/GoodFETMCPCANCommunication.py
index ab02306..e269a7b 100644 (file)
--- a/client/GoodFETMCPCANCommunication.py
+++ b/client/GoodFETMCPCANCommunication.py
@@ -17,7 +17,7 @@ import array;
 import csv, time, argparse;
 import datetime
 import os
-
+from random import randrange
 from GoodFETMCPCAN import GoodFETMCPCAN;
 from intelhex import IntelHex;
 
@@ -73,11 +73,15 @@ class GoodFETMCPCANCommunication:
     ##########################
          
     def sniff(self,freq,duration,description, verbose=True, comment=None, filename=None, standardid=None, debug=False, faster=False):
-                
+        
+        #reset eveything on the chip
+        self.client.serInit() 
+        self.reset()
+          
         #### ON-CHIP FILTERING
         if(standardid != None):
-            comment = comment+"Filtering for SID ";
-            
+            if( comment == None):
+                comment = ""
             self.client.MCPreqstatConfiguration();  
             self.client.poke8(0x60,0x26); # set RXB0 CTRL register to ONLY accept STANDARD messages with filter match (RXM1=0, RMX0=1, BUKT=1)
             self.client.poke8(0x20,0xFF); #set buffer 0 mask 1 (SID 10:3) to FF
@@ -117,8 +121,8 @@ class GoodFETMCPCANCommunication:
                self.client.poke8(RXFSIDL, SIDlow);
         
                if (verbose == True):
-                print "Filtering for SID %d (0x%02xh) with filter #%d"%(ID, ID, filter);
-            comment = comment + ("%d " %(ID))
+                   print "Filtering for SID %d (0x%02xh) with filter #%d"%(ID, ID, filter);
+               comment = comment + ("f%d" %(ID))
         
         
         self.client.MCPsetrate(freq);
@@ -197,6 +201,13 @@ class GoodFETMCPCANCommunication:
                     row.append(0);  #since we don't check for errors if we're not in debug mode...
                             
                 row.append(comment)
+                #how long the sniff was for
+                row.append(duration)
+                #boolean that tells us if there was filtering. 0 == no filters, 1 == filters
+                if(standardid != None):
+                    row.append(1)
+                else:
+                    row.append(0)
                 #write packet to file
                 for byte in packet:
                     row.append("%02x"%ord(byte));
@@ -223,6 +234,29 @@ class GoodFETMCPCANCommunication:
                         msgIDs.append(j)
         return msgIDs
     
+    def sweepRandom(self, freq, time = 5, number):
+        msgIDs = []
+        ids = []
+        for i in range(0,number,6):
+            idsTemp = []
+            comment = "sweepFilter"
+            for j in range(0,6,1):
+                id = randrange(2047)
+                comment += "_%d" % id
+                idsTemp.append(id)
+                ids.append(id)
+            print comment
+            description = "Running a sweep filer for all the possible standard IDs. This runs the following : " + comment
+            count = self.sniff(freq=freq, duration=time, description=description, comment = comment, filename, standardid = idsTemp)
+            if( count != 0):
+                for element in idsTemp:
+                    comment = "sweepFilter: %d" % (element)
+                    description = "Running a sweep filer for all the possible standard IDs. This run filters for: %d " % element
+                    count = self.sniff(freq=freq, duration = time, description = description,comment = comment, standardid = [element, element, element])
+                    if( count != 0):
+                        msgIDs.append(j)
+        return msgIDs, ids
+    
     def sniffTest(self, freq):
         
         rate = freq;