SPI Flash writing is now performed in chunks,
[goodfet] / 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();