Good stuff for ARM7TDMI. get_/set_register() seem to work ok.
[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 #Get SmartRF Studio regs if they exist.
43 #client.loadsymbols(); 
44
45 if(sys.argv[1]=="explore"):
46     print "Exploring undefined commands."
47     print "Status: %s" %client.status();
48     
49     cmd=0x04; #read status
50     for foo in range(0,0x5):
51         client.CCcmd([(0x0F<<3)|(0x00)|0x03,0x09<<3]);
52         print "Status %02x: %s" % (foo,client.status());
53     for foo in range(0,3):
54         print "PC: %04x" % client.CCgetPC();
55 if(sys.argv[1]=="term"):
56     GoodFETConsole(client).run();
57 if(sys.argv[1]=="test"):
58     client.test();
59 if(sys.argv[1]=="deadtest"):
60     for i in range(1,10):
61         print "IDENT as %s" % client.CCidentstr();
62 if(sys.argv[1]=="dumpcode"):
63     f = sys.argv[2];
64     start=0x0000;
65     stop=0xFFFF;
66     if(len(sys.argv)>3):
67         start=int(sys.argv[3],16);
68     if(len(sys.argv)>4):
69         stop=int(sys.argv[4],16);
70     
71     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
72     h = IntelHex(None);
73     i=start;
74     while i<=stop:
75         h[i]=client.CCpeekcodebyte(i);
76         if(i%0x100==0):
77             print "Dumped %04x."%i;
78         i+=1;
79     h.write_hex_file(f);
80 if(sys.argv[1]=="dumpdata"):
81     f = sys.argv[2];
82     start=0xE000;
83     stop=0xFFFF;
84     if(len(sys.argv)>3):
85         start=int(sys.argv[3],16);
86     if(len(sys.argv)>4):
87         stop=int(sys.argv[4],16);
88     
89     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
90     h = IntelHex(None);
91     i=start;
92     while i<=stop:
93         h[i]=client.CCpeekdatabyte(i);
94         if(i%0x100==0):
95             print "Dumped %04x."%i;
96         i+=1;
97     h.write_hex_file(f);
98 if(sys.argv[1]=="status"):
99     print "Status: %s" %client.status();
100 if(sys.argv[1]=="info"):
101     print "%s" % client.CCidentstr();
102 if(sys.argv[1]=="radioinfo"):
103     client.CMDrs();
104 if(sys.argv[1]=="erase"):
105     print "Status: %s" % client.status();
106     client.CCchiperase();
107     print "Status: %s" %client.status();
108
109 if(sys.argv[1]=="peekinfo"):
110     print "Select info flash."
111     client.CCwr_config(1);
112     print "Config is %02x" % client.CCrd_config();
113     
114     start=0x0000;
115     if(len(sys.argv)>2):
116         start=int(sys.argv[2],16);
117     stop=start;
118     if(len(sys.argv)>3):
119         stop=int(sys.argv[3],16);
120     print "Peeking from %04x to %04x." % (start,stop);
121     while start<=stop:
122         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
123         start=start+1;
124 if(sys.argv[1]=="poke"):
125     client.CCpokeirambyte(int(sys.argv[2],16),
126                           int(sys.argv[3],16));
127 if(sys.argv[1]=="randtest"):
128     #Seed RNG
129     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
130     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
131     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
132     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
133     
134     #Dump values
135     for foo in range(1,10):
136         print "%02x" % client.CCpeekirambyte(0xBD); #RNDH
137         client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
138         client.CCreleasecpu();
139         client.CChaltcpu();
140     print "%02x" % client.CCpeekdatabyte(0xDF61); #CHIP ID
141 if(sys.argv[1]=="adctest"):
142     # ADCTest 0xDF3A 0xDF3B
143     print "ADC TEST %02x%02x" % (
144         client.CCpeekdatabyte(0xDF3A),
145         client.CCpeekdatabyte(0xDF3B));
146 if(sys.argv[1]=="config"):
147     print "Config is %02x" % client.CCrd_config();
148
149 if(sys.argv[1]=="flash"):
150      f=sys.argv[2];
151      start=0;
152      stop=0xFFFF;
153      if(len(sys.argv)>3):
154          start=int(sys.argv[3],16);
155      if(len(sys.argv)>4):
156          stop=int(sys.argv[4],16);
157    
158      client.flash(f);
159 if(sys.argv[1]=="lock"):
160     print "Status: %s" %client.status();
161     client.CClockchip();
162     print "Status: %s" %client.status();
163 if(sys.argv[1]=="flashpage"):
164     target=0;
165     if(len(sys.argv)>2):
166         target=int(sys.argv[2],16);
167     print "Writing a page of flash from 0xF000 in XDATA"
168     client.CCflashpage(target);
169 if(sys.argv[1]=="erasebuffer"):
170     print "Erasing flash buffer.";
171     client.CCeraseflashbuffer();
172
173 if(sys.argv[1]=="writedata"):
174     f=sys.argv[2];
175     start=0;
176     stop=0xFFFF;
177     if(len(sys.argv)>3):
178         start=int(sys.argv[3],16);
179     if(len(sys.argv)>4):
180         stop=int(sys.argv[4],16);
181     
182     h = IntelHex(f);
183     
184     for i in h._buf.keys():
185         if(i>=start and i<=stop):
186             client.CCpokedatabyte(i,h[i]);
187             if(i%0x100==0):
188                 print "%04x" % i;
189 #if(sys.argv[1]=="flashtest"):
190 #    client.CCflashtest();
191 if(sys.argv[1]=="peekdata"):
192     start=0x0000;
193     if(len(sys.argv)>2):
194         start=int(sys.argv[2],16);
195     stop=start;
196     if(len(sys.argv)>3):
197         stop=int(sys.argv[3],16);
198     print "Peeking from %04x to %04x." % (start,stop);
199     while start<=stop:
200         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
201         start=start+1;
202 if(sys.argv[1]=="peek"):
203     start=0x0000;
204     if(len(sys.argv)>2):
205         start=int(sys.argv[2],16);
206     stop=start;
207     if(len(sys.argv)>3):
208         stop=int(sys.argv[3],16);
209     print "Peeking from %04x to %04x." % (start,stop);
210     while start<=stop:
211         print "%04x: %02x" % (start,client.CCpeekirambyte(start));
212         start=start+1;
213 if(sys.argv[1]=="verify"):
214     f=sys.argv[2];
215     start=0;
216     stop=0xFFFF;
217     if(len(sys.argv)>3):
218         start=int(sys.argv[3],16);
219     if(len(sys.argv)>4):
220         stop=int(sys.argv[4],16);
221     
222     h = IntelHex(f);
223     for i in h._buf.keys():
224         if(i>=start and i<stop):
225             peek=client.CCpeekcodebyte(i)
226             if(h[i]!=peek):
227                 print "ERROR at %04x, found %02x not %02x"%(i,peek,h[i]);
228             if(i%0x100==0):
229                 print "%04x" % i;
230 if(sys.argv[1]=="peekcode"):
231     start=0x0000;
232     if(len(sys.argv)>2):
233         start=int(sys.argv[2],16);
234     stop=start;
235     if(len(sys.argv)>3):
236         stop=int(sys.argv[3],16);
237     print "Peeking from %04x to %04x." % (start,stop);
238     while start<=stop:
239         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
240         start=start+1;
241 if(sys.argv[1]=="pokedata"):
242     start=0x0000;
243     val=0x00;
244     if(len(sys.argv)>2):
245         start=int(sys.argv[2],16);
246     if(len(sys.argv)>3):
247         val=int(sys.argv[3],16);
248     print "Poking %04x to become %02x." % (start,val);
249     client.CCpokedatabyte(start,val);
250
251 client.stop();