readChunk (may replace readMem) is implemented, moving memory into regs then reading...
[goodfet] / client / goodfet.cc
1 #!/usr/bin/env python
2 # GoodFET Chipcon Example
3
4 # (C) 2009 Travis Goodspeed <travis at radiantmachines.com>
5 #
6 # This code is being rewritten and refactored.  You've been warned!
7
8 import sys;
9 import binascii;
10
11 from GoodFETCC import GoodFETCC;
12 from GoodFETConsole import GoodFETConsole;
13 from intelhex import IntelHex;
14
15 if(len(sys.argv)==1):
16     print "Usage: %s verb [objects]\n" % sys.argv[0];
17     print "%s test" % sys.argv[0];
18     print "%s term" % sys.argv[0];
19     print "%s info" % sys.argv[0];
20     print "%s radioinfo" % sys.argv[0];
21     print "%s dumpcode $foo.hex [0x$start 0x$stop]" % sys.argv[0];
22     print "%s dumpdata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
23     print "%s erase" % sys.argv[0];
24     print "%s writedata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
25     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
26     print "%s peekdata 0x$start [0x$stop]" % sys.argv[0];
27     print "%s pokedata 0x$adr 0x$val" % sys.argv[0];
28     print "%s peek 0x$iram" % sys.argv[0];
29     print "%s poke 0x$iram 0x$val" % sys.argv[0];
30     print "%s peekcode 0x$start [0x$stop]" % sys.argv[0];
31     sys.exit();
32
33 #Initailize FET and set baud rate
34 #client=GoodFET.GoodFETCC.GoodFETCC();
35 client=GoodFETCC();
36 client.serInit()
37
38 #Connect to target
39 client.setup();
40 client.start();
41
42
43 if(sys.argv[1]=="explore"):
44     print "Exploring undefined commands."
45     print "Status: %s" %client.status();
46     
47     cmd=0x04; #read status
48     for foo in range(0,0x5):
49         client.CCcmd([(0x0F<<3)|(0x00)|0x03,0x09<<3]);
50         print "Status %02x: %s" % (foo,client.status());
51     for foo in range(0,3):
52         print "PC: %04x" % client.CCgetPC();
53 if(sys.argv[1]=="term"):
54     GoodFETConsole(client).run();
55 if(sys.argv[1]=="test"):
56     client.test();
57 if(sys.argv[1]=="deadtest"):
58     for i in range(1,10):
59         print "IDENT as %s" % client.CCidentstr();
60 if(sys.argv[1]=="dumpcode"):
61     f = sys.argv[2];
62     start=0x0000;
63     stop=0xFFFF;
64     if(len(sys.argv)>3):
65         start=int(sys.argv[3],16);
66     if(len(sys.argv)>4):
67         stop=int(sys.argv[4],16);
68     
69     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
70     h = IntelHex(None);
71     i=start;
72     while i<=stop:
73         h[i]=client.CCpeekcodebyte(i);
74         if(i%0x100==0):
75             print "Dumped %04x."%i;
76         i+=1;
77     h.write_hex_file(f);
78 if(sys.argv[1]=="dumpdata"):
79     f = sys.argv[2];
80     start=0xE000;
81     stop=0xFFFF;
82     if(len(sys.argv)>3):
83         start=int(sys.argv[3],16);
84     if(len(sys.argv)>4):
85         stop=int(sys.argv[4],16);
86     
87     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
88     h = IntelHex(None);
89     i=start;
90     while i<=stop:
91         h[i]=client.CCpeekdatabyte(i);
92         if(i%0x100==0):
93             print "Dumped %04x."%i;
94         i+=1;
95     h.write_hex_file(f);
96 if(sys.argv[1]=="status"):
97     print "Status: %s" %client.status();
98 if(sys.argv[1]=="info"):
99     print "%s" % client.CCidentstr();
100 if(sys.argv[1]=="radioinfo"):
101     client.CMDrs();
102 if(sys.argv[1]=="erase"):
103     print "Status: %s" % client.status();
104     client.CCchiperase();
105     print "Status: %s" %client.status();
106
107 if(sys.argv[1]=="peekinfo"):
108     print "Select info flash."
109     client.CCwr_config(1);
110     print "Config is %02x" % client.CCrd_config();
111     
112     start=0x0000;
113     if(len(sys.argv)>2):
114         start=int(sys.argv[2],16);
115     stop=start;
116     if(len(sys.argv)>3):
117         stop=int(sys.argv[3],16);
118     print "Peeking from %04x to %04x." % (start,stop);
119     while start<=stop:
120         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
121         start=start+1;
122 if(sys.argv[1]=="poke"):
123     client.CCpokeirambyte(int(sys.argv[2],16),
124                           int(sys.argv[3],16));
125 if(sys.argv[1]=="randtest"):
126     #Seed RNG
127     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
128     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
129     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
130     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
131     
132     #Dump values
133     for foo in range(1,10):
134         print "%02x" % client.CCpeekirambyte(0xBD); #RNDH
135         client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
136         client.CCreleasecpu();
137         client.CChaltcpu();
138     print "%02x" % client.CCpeekdatabyte(0xDF61); #CHIP ID
139 if(sys.argv[1]=="adctest"):
140     # ADCTest 0xDF3A 0xDF3B
141     print "ADC TEST %02x%02x" % (
142         client.CCpeekdatabyte(0xDF3A),
143         client.CCpeekdatabyte(0xDF3B));
144 if(sys.argv[1]=="config"):
145     print "Config is %02x" % client.CCrd_config();
146
147 if(sys.argv[1]=="flash"):
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      client.flash(f);
157 if(sys.argv[1]=="lock"):
158     print "Status: %s" %client.status();
159     client.CClockchip();
160     print "Status: %s" %client.status();
161 if(sys.argv[1]=="flashpage"):
162     target=0;
163     if(len(sys.argv)>2):
164         target=int(sys.argv[2],16);
165     print "Writing a page of flash from 0xF000 in XDATA"
166     client.CCflashpage(target);
167 if(sys.argv[1]=="erasebuffer"):
168     print "Erasing flash buffer.";
169     client.CCeraseflashbuffer();
170
171 if(sys.argv[1]=="writedata"):
172     f=sys.argv[2];
173     start=0;
174     stop=0xFFFF;
175     if(len(sys.argv)>3):
176         start=int(sys.argv[3],16);
177     if(len(sys.argv)>4):
178         stop=int(sys.argv[4],16);
179     
180     h = IntelHex(f);
181     
182     for i in h._buf.keys():
183         if(i>=start and i<=stop):
184             client.CCpokedatabyte(i,h[i]);
185             if(i%0x100==0):
186                 print "%04x" % i;
187 #if(sys.argv[1]=="flashtest"):
188 #    client.CCflashtest();
189 if(sys.argv[1]=="peekdata"):
190     start=0x0000;
191     if(len(sys.argv)>2):
192         start=int(sys.argv[2],16);
193     stop=start;
194     if(len(sys.argv)>3):
195         stop=int(sys.argv[3],16);
196     print "Peeking from %04x to %04x." % (start,stop);
197     while start<=stop:
198         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
199         start=start+1;
200 if(sys.argv[1]=="peek"):
201     start=0x0000;
202     if(len(sys.argv)>2):
203         start=int(sys.argv[2],16);
204     stop=start;
205     if(len(sys.argv)>3):
206         stop=int(sys.argv[3],16);
207     print "Peeking from %04x to %04x." % (start,stop);
208     while start<=stop:
209         print "%04x: %02x" % (start,client.CCpeekirambyte(start));
210         start=start+1;
211 if(sys.argv[1]=="verify"):
212     f=sys.argv[2];
213     start=0;
214     stop=0xFFFF;
215     if(len(sys.argv)>3):
216         start=int(sys.argv[3],16);
217     if(len(sys.argv)>4):
218         stop=int(sys.argv[4],16);
219     
220     h = IntelHex(f);
221     for i in h._buf.keys():
222         if(i>=start and i<stop):
223             peek=client.CCpeekcodebyte(i)
224             if(h[i]!=peek):
225                 print "ERROR at %04x, found %02x not %02x"%(i,peek,h[i]);
226             if(i%0x100==0):
227                 print "%04x" % i;
228 if(sys.argv[1]=="peekcode"):
229     start=0x0000;
230     if(len(sys.argv)>2):
231         start=int(sys.argv[2],16);
232     stop=start;
233     if(len(sys.argv)>3):
234         stop=int(sys.argv[3],16);
235     print "Peeking from %04x to %04x." % (start,stop);
236     while start<=stop:
237         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
238         start=start+1;
239 if(sys.argv[1]=="pokedata"):
240     start=0x0000;
241     val=0x00;
242     if(len(sys.argv)>2):
243         start=int(sys.argv[2],16);
244     if(len(sys.argv)>3):
245         val=int(sys.argv[3],16);
246     print "Poking %04x to become %02x." % (start,val);
247     client.CCpokedatabyte(start,val);
248
249 client.stop();