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
  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 dump $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     print "%s poke 0x$adr 0x$val" % sys.argv[0];
  19     print "%s peek 0x$start [0x$stop]" % sys.argv[0];
  20     print "%s run" % sys.argv[0];
  21     sys.exit();
  22
  23 #Initialize FET and set baud rate
  24 client=GoodFETMSP430();
  25 client.serInit()
  26
  27 #Connect to target
  28 client.MSP430setup();
  29 #print "setup"
  30
  31 #Identify model number.
  32 client.MSP430start();
  33 client.MSP430haltcpu();
  34 #print "started"
  35
  36 if(sys.argv[1]=="info"):
  37     #print "Model    %08x " % client.MSP430deviceid();
  38     #print "Core     %04x " % client.MSP430coreid();
  39     #print "Identity %04x" % client.MSP430ident();
  40     print "Identifies as %s (%04x)" % (
  41         client.MSP430identstr(),
  42         client.MSP430ident());
  43 if(sys.argv[1]=="test"):
  44     client.MSP430test();
  45 if(sys.argv[1]=="glitch"):
  46     for foo in range(0,100):
  47         print "Identifies as %04x" % client.MSP430ident();
  48 if(sys.argv[1]=="dump"):
  49     f = sys.argv[2];
  50     start=0x0200;
  51     stop=0xFFFF;
  52     if(len(sys.argv)>3):
  53         start=int(sys.argv[3],16);
  54     if(len(sys.argv)>4):
  55         stop=int(sys.argv[4],16);
  56
  57     print "Dumping from %04x to %04x as %s." % (start,stop,f);
  58     #h = IntelHex16bit(None);
  59     h = IntelHex(None);
  60     i=start;
  61     while i<=stop:
  62         data=client.MSP430peekblock(i);
  63         print "Dumped %06x."%i;
  64         for j in data:
  65             if i<=stop: h[i]=ord(j);
  66             i+=1;
  67     h.write_hex_file(f);
  68 if(sys.argv[1]=="erase"):
  69     client.MSP430masserase();
  70 if(sys.argv[1]=="ivt"):
  71     client.MSP430dumpmem(0xFFE0,0xFFFF);
  72 if(sys.argv[1]=="flash"):
  73     f=sys.argv[2];
  74     start=0;
  75     stop=0xFFFF;
  76     if(len(sys.argv)>3):
  77         start=int(sys.argv[3],16);
  78     if(len(sys.argv)>4):
  79         stop=int(sys.argv[4],16);
  80
  81     h = IntelHex16bit(f);
  82
  83     #Should this be default?
  84     #Makes flashing multiple images inconvenient.
  85     client.MSP430masserase();
  86
  87     count=0; #Bytes in commit.
  88     first=0;
  89     vals=[];
  90     last=0;  #Last address committed.
  91     for i in h._buf.keys():
  92         if((count>0x40 or last+2!=i) and count>0 and i&1==0):
  93             #print "%i, %x, %x" % (len(vals), last, i);
  94             client.MSP430pokeflashblock(first,vals);
  95             count=0;
  96             first=0;
  97             last=0;
  98             vals=[];
  99         if(i>=start and i<stop  and i&1==0):
 100             val=h[i>>1];
 101             if(count==0):
 102                 first=i;
 103             last=i;
 104             count+=2;
 105             #val2=client.MSP430pokeflash(i,val);
 106             #if(val!=val2):
 107             #    print "Write error at %04x, %04x!=%04x." %(
 108             #        i, val, val2);
 109             vals+=[val&0xff,(val&0xff00)>>8];
 110             if(i%0x100==0):
 111                 print "%04x" % i;
 112     if count>0: #last commit, ivt
 113         client.MSP430pokeflashblock(first,vals);
 114 if(sys.argv[1]=="verify"):
 115     f=sys.argv[2];
 116     start=0;
 117     stop=0xFFFF;
 118     if(len(sys.argv)>3):
 119         start=int(sys.argv[3],16);
 120     if(len(sys.argv)>4):
 121         stop=int(sys.argv[4],16);
 122
 123     h = IntelHex16bit(f);
 124     for i in h._buf.keys():
 125         if(i>=start and i<stop and i&1==0):
 126             peek=client.MSP430peek(i)
 127             if(h[i>>1]!=peek):
 128                 print "ERROR at %04x, found %04x not %04x"%(i,peek,h[i>>1]);
 129             if(i%0x100==0):
 130                 print "%04x" % i;
 131 if(sys.argv[1]=="peek"):
 132     start=0x0000;
 133     if(len(sys.argv)>2):
 134         start=int(sys.argv[2],16);
 135     stop=start;
 136     if(len(sys.argv)>3):
 137         stop=int(sys.argv[3],16);
 138     print "Peeking from %04x to %04x." % (start,stop);
 139     while start<=stop:
 140         print "%04x: %04x" % (start,client.MSP430peek(start));
 141         start=start+2;
 142
 143 if(sys.argv[1]=="poke"):
 144     start=0x0000;
 145     val=0x00;
 146     if(len(sys.argv)>2):
 147         start=int(sys.argv[2],16);
 148     if(len(sys.argv)>3):
 149         val=int(sys.argv[3],16);
 150     print "Poking %06x to become %04x." % (start,val);
 151
 152     while client.MSP430peek(start)!=val:
 153         client.MSP430pokeflash(start,val);
 154         print "Poked to %04x" % client.MSP430peek(start);
 155
 156
 157 if(sys.argv[1]=="run"):
 158     #Set PC to RESET vector's value.
 159     #client.MSP430setPC(client.MSP430peek(0xfffe));
 160     #client.MSP430releasecpu();
 161     client.MSP430run();
 162
 163 if(sys.argv[1]=="whatever"):
 164     for i in [0x24FF, 0x2500, 0x2502, 0x2504]:
 165         print "%04x" % client.MSP430peek(i);
 166
 167 #client.MSP430releasecpu();
 168 #client.MSP430stop();