git.rot13.org Git - goodfet/blob - client/goodfet.cc

   1 #!/usr/bin/env python
   2
   3 import sys;
   4 import binascii;
   5
   6 from GoodFET import GoodFET;
   7 from intelhex import IntelHex;
   8
   9
  10
  11 if(len(sys.argv)==1):
  12     print "Usage: %s verb [objects]\n" % sys.argv[0];
  13     print "%s test" % sys.argv[0];
  14     print "%s dumpcode $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  15     print "%s erase" % sys.argv[0];
  16     print "%s flash $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  17     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  18     sys.exit();
  19
  20 #Initailize FET and set baud rate
  21 client=GoodFET();
  22 client.serInit("/dev/ttyUSB0")
  23
  24
  25
  26 #Connect to target
  27 client.CCsetup();
  28 client.CCstart();
  29
  30 if(sys.argv[1]=="test"):
  31     client.CCtest();
  32 if(sys.argv[1]=="dumpcode"):
  33     f = sys.argv[2];
  34     start=0x0000;
  35     stop=0xFFFF;
  36     if(len(sys.argv)>3):
  37         start=int(sys.argv[3],16);
  38     if(len(sys.argv)>4):
  39         stop=int(sys.argv[4],16);
  40
  41     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
  42     h = IntelHex(None);
  43     i=start;
  44     while i<=stop:
  45         h[i>>1]=client.CCpeekcodebyte(i);
  46         if(i%0x100==0):
  47             print "Dumped %04x."%i;
  48         i+=1;
  49     h.write_hex_file(f);
  50 if(sys.argv[1]=="dumpdata"):
  51     f = sys.argv[2];
  52     start=0xE000;
  53     stop=0xFFFF;
  54     if(len(sys.argv)>3):
  55         start=int(sys.argv[3],16);
  56     if(len(sys.argv)>4):
  57         stop=int(sys.argv[4],16);
  58
  59     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
  60     h = IntelHex(None);
  61     i=start;
  62     while i<=stop:
  63         h[i]=client.CCpeekdatabyte(i);
  64         if(i%0x100==0):
  65             print "Dumped %04x."%i;
  66         i+=1;
  67     h.write_hex_file(f);
  68 if(sys.argv[1]=="erase"):
  69   print "Status: %s" % client.CCstatusstr();
  70   client.CCchiperase();
  71   print "Status: %s" %client.CCstatusstr();
  72
  73 if(sys.argv[1]=="flash"):
  74     f=sys.argv[2];
  75     start=0;
  76     stop=0xFFFF;
  77     if(len(sys.argv)>3):
  78         start=int(sys.argv[3],16);
  79     if(len(sys.argv)>4):
  80         stop=int(sys.argv[4],16);
  81
  82     h = IntelHex(f);
  83
  84     client.MSP430masserase();
  85     for i in h._buf.keys():
  86         #print "%04x: %04x"%(i,h[i>>1]);
  87         if(i>=start and i<=stop  and i&1==0):
  88             client.MSP430writeflash(i,h[i>>1]);
  89             if(i%0x100==0):
  90                 print "%04x" % i;
  91 if(sys.argv[1]=="writedata"):
  92     f=sys.argv[2];
  93     start=0;
  94     stop=0xFFFF;
  95     if(len(sys.argv)>3):
  96         start=int(sys.argv[3],16);
  97     if(len(sys.argv)>4):
  98         stop=int(sys.argv[4],16);
  99
 100     h = IntelHex(f);
 101
 102     for i in h._buf.keys():
 103         if(i>=start and i<=stop):
 104             client.CCpokedatabyte(i,h[i]);
 105             if(i%0x100==0):
 106                 print "%04x" % i;
 107 if(sys.argv[1]=="flashtest"):
 108     client.MSP430flashtest();
 109 if(sys.argv[1]=="verifycode"):
 110     f=sys.argv[2];
 111     start=0;
 112     stop=0xFFFF;
 113     if(len(sys.argv)>3):
 114         start=int(sys.argv[3],16);
 115     if(len(sys.argv)>4):
 116         stop=int(sys.argv[4],16);
 117
 118     h = IntelHex(f);
 119     for i in h._buf.keys():
 120         if(i>=start and i<=stop):
 121             peek=client.MSP430peek(i)
 122             if(h[i>>1]!=peek):
 123                 print "ERROR at %04x, found %04x not %04x"%(i,peek,h[i>>1]);
 124             if(i%0x100==0):
 125                 print "%04x" % i;
 126
 127 client.CCstop();