Closer to reliable AVR glitching.
[goodfet] / 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();