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

   1 #!/usr/bin/env python
   2
   3 import sys;
   4 import binascii;
   5
   6 from GoodFETMSP430 import GoodFETMSP430;
   7 from intelhex import IntelHex16bit, IntelHex;
   8
   9
  10 if(len(sys.argv)==1):
  11     print "Usage: %s verb [objects]\n" % sys.argv[0];
  12     print "%s test" % sys.argv[0];
  13     print "%s selftest" % sys.argv[0];
  14     print "%s dump $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  15     print "%s erase" % sys.argv[0];
  16     print "%s eraseinfo" % sys.argv[0];
  17     print "%s flash $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  18     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  19     print "%s poke 0x$adr 0x$val" % sys.argv[0];
  20     print "%s serial [$val]" % sys.argv[0];
  21     print "%s peek 0x$start [0x$stop]" % sys.argv[0];
  22     print "%s run" % sys.argv[0];
  23     sys.exit();
  24
  25 #Initialize FET and set baud rate
  26 client=GoodFETMSP430();
  27 client.serInit()
  28
  29 client.setup();
  30 client.start();
  31
  32 if(sys.argv[1]=="starttest"):
  33     for foo in range(1,10):
  34         client.start();
  35 if(sys.argv[1]=="info"):
  36     #print "Model    %08x " % client.MSP430deviceid();
  37     #print "Core     %04x " % client.MSP430coreid();
  38     #print "Identity %04x" % client.MSP430ident();
  39     print "Identifies as %s (%04x)" % (
  40         client.MSP430identstr(),
  41         client.MSP430ident());
  42 if(sys.argv[1]=="test"):
  43     client.MSP430test();
  44 if(sys.argv[1]=="glitch"):
  45     for foo in range(0,100):
  46         print "Identifies as %04x" % client.MSP430ident();
  47 if(sys.argv[1]=="dump"):
  48     f = sys.argv[2];
  49     start=0x0200;
  50     stop=0xFFFF;
  51     if(len(sys.argv)>3):
  52         start=int(sys.argv[3],16);
  53     if(len(sys.argv)>4):
  54         stop=int(sys.argv[4],16);
  55
  56     print "Dumping from %04x to %04x as %s." % (start,stop,f);
  57     #h = IntelHex16bit(None);
  58     h = IntelHex(None);
  59     i=start;
  60     while i<=stop:
  61         data=client.MSP430peekblock(i);
  62         print "Dumped %06x."%i;
  63         for j in data:
  64             if i<=stop: h[i]=ord(j);
  65             i+=1;
  66     h.write_hex_file(f);
  67 if(sys.argv[1]=="erase"):
  68     print "Erasing main flash memory."
  69     client.MSP430masserase();
  70 if(sys.argv[1]=="eraseinfo"):
  71     print "Erasing info memory."
  72     client.MSP430infoerase();
  73
  74 if(sys.argv[1]=="selftest"):
  75     import time;
  76
  77     entry=client.peek16(0xFFC0);
  78     print "Calling self-test at 0x%04x." %entry;
  79     print "Memory from 0x3020 to 0x3030 should be zeroed.";
  80
  81     for i in range(0x3020,0x3030):
  82         client.MSP430poke(i,0xFFFF);
  83     client.MSP430setPC(entry);
  84     time.sleep(5);
  85     errors=0;
  86
  87     if(client.peek8(0x302F)==0xFF):
  88         print "Test not run.  Try again.";
  89         exit(0);
  90
  91     for i in range(0x3020,0x3030,2):
  92         ret=client.peek16(i);
  93         if ret!=0:
  94             errors=errors+1;
  95             print "Failure at 0x%04x, returned %02xh." % (i,ret);
  96
  97     if errors==0:
  98         print "Test succeeded.  Board is ready to ship."
  99     else:
 100         print "Test failed.  Is it soldered correctly?"
 101     #client.MSP430dumpmem(0x3020,0x3030);
 102
 103 if(sys.argv[1]=="ivt"):
 104     client.MSP430dumpmem(0xFFC0,0xFFFF);
 105 if(sys.argv[1]=="regs"):
 106     for i in range(0,16):
 107         print "r%i=%04x" % (i,client.MSP430getreg(i));
 108
 109 if(sys.argv[1]=="flash"):
 110     f=sys.argv[2];
 111     start=0;
 112     stop=0x10000;
 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 = IntelHex16bit(f);
 119
 120     #Should this be default?
 121     #Makes flashing multiple images inconvenient.
 122     #client.MSP430masserase();
 123
 124     count=0; #Bytes in commit.
 125     first=0;
 126     vals=[];
 127     last=0;  #Last address committed.
 128     for i in h._buf.keys():
 129         if((count>0x40 or last+2!=i) and count>0 and i&1==0):
 130             #print "%i, %x, %x" % (len(vals), last, i);
 131             client.MSP430pokeflashblock(first,vals);
 132             count=0;
 133             first=0;
 134             last=0;
 135             vals=[];
 136         if(i>=start and i<stop  and i&1==0):
 137             val=h[i>>1];
 138             if(count==0):
 139                 first=i;
 140             last=i;
 141             count+=2;
 142             vals+=[val&0xff,(val&0xff00)>>8];
 143             if(i%0x100==0):
 144                 print "%04x" % i;
 145     if count>0: #last commit, ivt
 146         client.MSP430pokeflashblock(first,vals);
 147 if(sys.argv[1]=="verify"):
 148     f=sys.argv[2];
 149     start=0;
 150     stop=0xFFFF;
 151     if(len(sys.argv)>3):
 152         start=int(sys.argv[3],16);
 153     if(len(sys.argv)>4):
 154         stop=int(sys.argv[4],16);
 155
 156     h = IntelHex16bit(f);
 157     for i in h._buf.keys():
 158         if(i>=start and i<stop and i&1==0):
 159             peek=client.MSP430peek(i)
 160             if(h[i>>1]!=peek):
 161                 print "ERROR at %04x, found %04x not %04x"%(i,peek,h[i>>1]);
 162             if(i%0x100==0):
 163                 print "%04x" % i;
 164 if(sys.argv[1]=="peek"):
 165     start=0x0000;
 166     if(len(sys.argv)>2):
 167         start=int(sys.argv[2],16);
 168     stop=start;
 169     if(len(sys.argv)>3):
 170         stop=int(sys.argv[3],16);
 171     print "Peeking from %04x to %04x." % (start,stop);
 172     while start<=stop:
 173         print "%04x: %04x" % (start,client.MSP430peek(start));
 174         start=start+2;
 175
 176 if(sys.argv[1]=="poke"):
 177     start=0x0000;
 178     val=0x00;
 179     if(len(sys.argv)>2):
 180         start=int(sys.argv[2],16);
 181     if(len(sys.argv)>3):
 182         val=int(sys.argv[3],16);
 183     print "Poking %06x to become %04x." % (start,val);
 184
 185     while client.MSP430peek(start)&(~val)>0:
 186         client.MSP430pokeflash(start,val);
 187         print "Poked to %04x" % client.MSP430peek(start);
 188 if(sys.argv[1]=="serial"):
 189     oid=0xdeadbeef;
 190     if(len(sys.argv)>2):
 191         oid=eval(sys.argv[2]);
 192     else:
 193         #Crashes Windows.
 194         #simpchin=u=unichr(24207)+unichr(21015)+unichr(21495);  #Chinese for "Serial Number".
 195         #print simpchin;
 196
 197         oid=input(u"Serial Number? ");
 198     adr=0xFFB0;
 199     if(len(sys.argv)>2):
 200         start=int(sys.argv[2],16);
 201     if(len(sys.argv)>3):
 202         val=int(sys.argv[3],16);
 203     print "Poking 32-bit OID to 0x%04x as %08x." % (adr,oid);
 204     client.MSP430pokeflash(adr,oid&0xFFFF);
 205     client.MSP430pokeflash(adr+2,(oid>>16)&0xFFFF);
 206
 207 if(sys.argv[1]=="run"):
 208     #Set PC to RESET vector's value.
 209     #client.MSP430setPC(client.MSP430peek(0xfffe));
 210     #client.MSP430releasecpu();
 211     client.MSP430run();
 212
 213 if(sys.argv[1]=="whatever"):
 214     for i in [0x24FF, 0x2500, 0x2502, 0x2504]:
 215         print "%04x" % client.MSP430peek(i);
 216
 217 #client.MSP430releasecpu();
 218 #client.MSP430stop();