Sane SPI Flash addressing for poke date, will later convert all functions.
[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     sys.exit();
19
20 #Initialize FET and set baud rate
21 client=GoodFETMSP430();
22 client.serInit()
23
24 #Connect to target
25 client.MSP430setup();
26 #print "setup"
27
28 #Identify model number.
29 client.MSP430start();
30 client.MSP430haltcpu();
31 #print "started"
32
33 if(sys.argv[1]=="info"):
34     #print "Model    %08x " % client.MSP430deviceid();
35     #print "Core     %04x " % client.MSP430coreid();
36     #print "Identity %04x" % client.MSP430ident();
37     print "Identifies as %s (%04x)" % (
38         client.MSP430identstr(),
39         client.MSP430ident());
40 if(sys.argv[1]=="test"):
41     client.MSP430test();
42 if(sys.argv[1]=="dump"):
43     f = sys.argv[2];
44     start=0x0200;
45     stop=0xFFFF;
46     if(len(sys.argv)>3):
47         start=int(sys.argv[3],16);
48     if(len(sys.argv)>4):
49         stop=int(sys.argv[4],16);
50     
51     print "Dumping from %04x to %04x as %s." % (start,stop,f);
52     #h = IntelHex16bit(None);
53     h = IntelHex(None);
54     i=start;
55     while i<=stop:
56         data=client.MSP430peekblock(i);
57         print "Dumped %06x."%i;
58         for j in data:
59             if i<=stop: h[i]=ord(j);
60             i+=1;
61     h.write_hex_file(f);
62 if(sys.argv[1]=="erase"):
63     client.MSP430masserase();
64 if(sys.argv[1]=="ivt"):
65     client.MSP430dumpmem(0xFFE0,0xFFFF);
66 if(sys.argv[1]=="flash"):
67     f=sys.argv[2];
68     start=0;
69     stop=0xFFFF;
70     if(len(sys.argv)>3):
71         start=int(sys.argv[3],16);
72     if(len(sys.argv)>4):
73         stop=int(sys.argv[4],16);
74     
75     h = IntelHex16bit(f);
76     
77     client.MSP430masserase();
78     for i in h._buf.keys():
79         #print "%04x: %04x"%(i,h[i>>1]);
80         if(i>=start and i<stop  and i&1==0):
81             val=h[i>>1];
82             val2=client.MSP430pokeflash(i,val);
83             if(val!=val2):
84                 print "Write error at %04x, %04x!=%04x." %(
85                     i, val, val2);
86             if(i%0x100==0):
87                 print "%04x" % i;
88 if(sys.argv[1]=="verify"):
89     f=sys.argv[2];
90     start=0;
91     stop=0xFFFF;
92     if(len(sys.argv)>3):
93         start=int(sys.argv[3],16);
94     if(len(sys.argv)>4):
95         stop=int(sys.argv[4],16);
96     
97     h = IntelHex16bit(f);
98     for i in h._buf.keys():
99         if(i>=start and i<stop and i&1==0):
100             peek=client.MSP430peek(i)
101             if(h[i>>1]!=peek):
102                 print "ERROR at %04x, found %04x not %04x"%(i,peek,h[i>>1]);
103             if(i%0x100==0):
104                 print "%04x" % i;
105 if(sys.argv[1]=="peek"):
106     start=0x0000;
107     if(len(sys.argv)>2):
108         start=int(sys.argv[2],16);
109     stop=start;
110     if(len(sys.argv)>3):
111         stop=int(sys.argv[3],16);
112     print "Peeking from %04x to %04x." % (start,stop);
113     while start<=stop:
114         print "%04x: %04x" % (start,client.MSP430peek(start));
115         start=start+2;
116
117 if(sys.argv[1]=="poke"):
118     start=0x0000;
119     val=0x00;
120     if(len(sys.argv)>2):
121         start=int(sys.argv[2],16);
122     if(len(sys.argv)>3):
123         val=int(sys.argv[3],16);
124     print "Poking %06x to become %04x." % (start,val);
125     
126     while client.MSP430peek(start)!=val:
127         client.MSP430pokeflash(start,val);
128         print "Poked to %04x" % client.MSP430peek(start);
129
130
131
132 if(sys.argv[1]=="whatever"):
133     for i in [0x24FF, 0x2500, 0x2502, 0x2504]:
134         print "%04x" % client.MSP430peek(i);
135
136 client.MSP430releasecpu();
137 client.MSP430stop();