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