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 info" % sys.argv[0];
  15     print "%s dumpcode $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  16     print "%s dumpdata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  17     print "%s erase" % sys.argv[0];
  18     print "%s writedata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  19     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  20     print "%s peekdata 0x$start [0x$stop]" % sys.argv[0];
  21     print "%s pokedata 0x$adr 0x$val" % sys.argv[0];
  22     #print "%s peekcode 0x$start [0x$stop]" % sys.argv[0];
  23     sys.exit();
  24
  25 #Initailize FET and set baud rate
  26 client=GoodFET();
  27 client.serInit("/dev/ttyUSB0")
  28
  29 #Connect to target
  30 client.CCsetup();
  31 client.CCstart();
  32
  33 if(sys.argv[1]=="test"):
  34     client.CCtest();
  35 if(sys.argv[1]=="dumpcode"):
  36     f = sys.argv[2];
  37     start=0x0000;
  38     stop=0xFFFF;
  39     if(len(sys.argv)>3):
  40         start=int(sys.argv[3],16);
  41     if(len(sys.argv)>4):
  42         stop=int(sys.argv[4],16);
  43
  44     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
  45     h = IntelHex(None);
  46     i=start;
  47     while i<=stop:
  48         h[i>>1]=client.CCpeekcodebyte(i);
  49         if(i%0x100==0):
  50             print "Dumped %04x."%i;
  51         i+=1;
  52     h.write_hex_file(f);
  53 if(sys.argv[1]=="dumpdata"):
  54     f = sys.argv[2];
  55     start=0xE000;
  56     stop=0xFFFF;
  57     if(len(sys.argv)>3):
  58         start=int(sys.argv[3],16);
  59     if(len(sys.argv)>4):
  60         stop=int(sys.argv[4],16);
  61
  62     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
  63     h = IntelHex(None);
  64     i=start;
  65     while i<=stop:
  66         h[i]=client.CCpeekdatabyte(i);
  67         if(i%0x100==0):
  68             print "Dumped %04x."%i;
  69         i+=1;
  70     h.write_hex_file(f);
  71 if(sys.argv[1]=="erase"):
  72   print "Status: %s" % client.CCstatusstr();
  73   client.CCchiperase();
  74   print "Status: %s" %client.CCstatusstr();
  75
  76 if(sys.argv[1]=="flash"):
  77     f=sys.argv[2];
  78     start=0;
  79     stop=0xFFFF;
  80     if(len(sys.argv)>3):
  81         start=int(sys.argv[3],16);
  82     if(len(sys.argv)>4):
  83         stop=int(sys.argv[4],16);
  84
  85     h = IntelHex(f);
  86
  87     client.MSP430masserase();
  88     for i in h._buf.keys():
  89         #print "%04x: %04x"%(i,h[i>>1]);
  90         if(i>=start and i<=stop  and i&1==0):
  91             client.MSP430writeflash(i,h[i>>1]);
  92             if(i%0x100==0):
  93                 print "%04x" % i;
  94 if(sys.argv[1]=="writedata"):
  95     f=sys.argv[2];
  96     start=0;
  97     stop=0xFFFF;
  98     if(len(sys.argv)>3):
  99         start=int(sys.argv[3],16);
 100     if(len(sys.argv)>4):
 101         stop=int(sys.argv[4],16);
 102
 103     h = IntelHex(f);
 104
 105     for i in h._buf.keys():
 106         if(i>=start and i<=stop):
 107             client.CCpokedatabyte(i,h[i]);
 108             if(i%0x100==0):
 109                 print "%04x" % i;
 110 if(sys.argv[1]=="flashtest"):
 111     client.MSP430flashtest();
 112 if(sys.argv[1]=="verifycode"):
 113     f=sys.argv[2];
 114     start=0;
 115     stop=0xFFFF;
 116     if(len(sys.argv)>3):
 117         start=int(sys.argv[3],16);
 118     if(len(sys.argv)>4):
 119         stop=int(sys.argv[4],16);
 120
 121     h = IntelHex(f);
 122     for i in h._buf.keys():
 123         if(i>=start and i<=stop):
 124             peek=client.MSP430peek(i)
 125             if(h[i>>1]!=peek):
 126                 print "ERROR at %04x, found %04x not %04x"%(i,peek,h[i>>1]);
 127             if(i%0x100==0):
 128                 print "%04x" % i;
 129 if(sys.argv[1]=="peekdata"):
 130     start=0x0000;
 131     if(len(sys.argv)>2):
 132         start=int(sys.argv[2],16);
 133     stop=start;
 134     if(len(sys.argv)>3):
 135         stop=int(sys.argv[3],16);
 136     print "Peeking from %04x to %04x." % (start,stop);
 137     while start<=stop:
 138         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
 139         start=start+1;
 140 if(sys.argv[1]=="pokedata"):
 141     start=0x0000;
 142     val=0x00;
 143     if(len(sys.argv)>2):
 144         start=int(sys.argv[2],16);
 145     if(len(sys.argv)>3):
 146         val=int(sys.argv[3],16);
 147     print "Poking %04x to become %02x." % (start,val);
 148     client.CCpokedatabyte(start,val);
 149
 150 client.CCstop();