More EM260 code. RNG works only after another board has initialized the chip.
[goodfet] / client / gplay-arm.py
1 #!/usr/bin/env ipython
2 import sys, struct, binascii,time
3 from GoodFETARM import *
4 from intelhex import IntelHex
5
6
7 data = []
8
9 client=GoodFETARM();
10 def init():
11     #Initailize FET and set baud rate
12     print >>sys.stderr,"Connecting to goodfet..."
13     client.serInit()
14     #
15     #Connect to target
16     print >>sys.stderr,"Setting up JTAG ARM on goodfet..."
17     client.setup()
18     print >>sys.stderr,"Starting JTAG ARM on goodfet..."
19     client.start()
20     print "STARTUP: "+repr(client.data)
21     #
22
23 def print_registers():
24     return [ hex(client.ARMget_register(x)) for x in range(15) ]
25
26 def regwratchet(num,hi=13,lo=0):
27     for x in xrange(lo,hi+1):
28         client.ARMset_register(x,num)
29
30 def test1():
31     global data
32     print "\n\nTesting JTAG for ARM\n"
33     client.writecmd(0x33,0xd0,4,[0x40,0x40,0x40,0x40]); print "loopback:   \t %s"%repr(client.data)                  # loopback
34     data.append(client.data)
35     client.writecmd(0x33,0xd1,2,[1,0]); print "scanchain1:\t %s"%repr(client.data)               # set scan chain
36     data.append(client.data)
37     client.writecmd(0x33,0xd2,0,[]); print "debug state:\t %s"%repr(client.data)                  # get dbg state
38     data.append(client.data)
39     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
40     data.append(client.data)
41     client.writecmd(0x33,0xd3,0,[0,0,0x8e,0xe5]); print "exec_stuff: \t %s"%repr(client.data)     # execute instruction
42     data.append(client.data)
43     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
44     data.append(client.data)
45     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
46     data.append(client.data)
47     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
48     data.append(client.data)
49     client.writecmd(0x33,0xd6,0,[]); print "shift_dr_32: \t %s"%repr(client.data)                  # dr_shift32
50     data.append(client.data)
51     client.writecmd(0x33,0xd5,8,[3, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40]); print "set_register:\t %s"%repr(client.data)                  # set register
52     data.append(client.data)
53     client.writecmd(0x33,0xd4,1,[3]); print "get_register:\t %s"%repr(client.data)                  # get register
54     data.append(client.data)
55     client.writecmd(0x33,0xd7,0,[]); print "chain1:      \t %s"%repr(client.data)                  # chain1
56     data.append(client.data)
57     client.writecmd(0x33,0xd8,0,[]); print "read_chain2: \t %s"%repr(client.data)                  # read chain2
58     data.append(client.data)
59     client.writecmd(0x33,0xd9,0,[]); print "idcode:      \t %s"%repr(client.data)                  # read idcode
60     data.append(client.data)
61     client.writecmd(0x33,0xf0,2,[4,4,1,1]); print "f0:       \t %s"%repr(client.data)   # read idcode
62     data.append(client.data)
63     client.writecmd(0x33,0xdb,8,[0x0,4,4,4,4,4,4,4]); print "verb(0):     \t %s"%repr(client.data)
64     data.append(client.data)
65     client.writecmd(0x33,0xdb,8,[0x2,4,4,4,4,4,4,4]); print "verb(2):     \t %s"%repr(client.data)
66     data.append(client.data)
67     client.writecmd(0x33,0xdb,8,[0x3,4,4,4,4,4,4,4]); print "verb(3):     \t %s"%repr(client.data)
68     data.append(client.data)
69     client.writecmd(0x33,0xdb,8,[0x4,4,4,4,4,4,4,4]); print "verb(4):     \t %s"%repr(client.data)
70     data.append(client.data)
71     client.writecmd(0x33,0xdb,8,[0x5,4,4,4,4,4,4,4]); print "verb(5):     \t %s"%repr(client.data)
72     data.append(client.data)
73     client.writecmd(0x33,0xdb,8,[0x7,4,4,4,4,4,4,4]); print "verb(7):     \t %s"%repr(client.data)
74     data.append(client.data)
75     client.writecmd(0x33,0xdb,8,[0x9,4,4,4,4,4,4,4]); print "verb(9):     \t %s"%repr(client.data)
76     data.append(client.data)
77     client.writecmd(0x33,0xdb,8,[0xc,4,4,4,4,4,4,4]); print "verb(c):     \t %s"%repr(client.data)
78     data.append(client.data)
79     client.writecmd(0x33,0xdb,8,[0xe,0,0,0,0,0,0xa0,0xe1]); print "verb(e):     \t %s"%repr(client.data)
80     data.append(client.data)
81     client.writecmd(0x33,0xdb,8,[0xf,4,4,4,4,4,4,4]); print "verb(f):     \t %s"%repr(client.data)
82     data.append(client.data)
83
84 def test2():
85     global data
86     print "\n\nTesting JTAG for ARM\n"
87     print "IDCODE:      %x"%client.ARMident()
88     print "Debug State: %x"%client.ARMget_dbgstate ()
89     print "Debug State: %x"%client.ARMget_dbgstate ()
90     print "Debug State: %x"%client.ARMget_dbgstate ()
91     print "Debug CTRL:  %x"%client.ARMget_dbgctrl()
92     client.writecmd(0x33,0xda,0,[])
93     print "TEST CHAIN0: %s"%repr(client.data)
94     print "Debug State: %x"%client.ARMget_dbgstate ()
95     print "IDCODE:      %x"%client.ARMident()
96     print "Debug State: %x"%client.ARMget_dbgstate ()
97     client.writecmd(0x33,0xd0,4,[0xf7,0xf7,0xf7,0xf7])
98     print "Loopback:   \t %s"%repr(client.data)                  # loopback
99     print "Debug State: %x"%client.ARMget_dbgstate ()
100     print "IDCODE:      %x"%client.ARMident()
101     print "GetPC: %x"%client.ARMgetPC()
102     print "IDCODE:      %x"%client.ARMident()
103     print "Debug State: %x"%client.ARMget_dbgstate ()
104     print "IDCODE:      %x"%client.ARMident()
105     print "set_register(3,0x41414141):  %x"%client.ARMset_register(3,0x41414141)
106     print "IDCODE:      %x"%client.ARMident()
107     print "Debug State: %x"%client.ARMget_dbgstate ()
108     print "IDCODE:      %x"%client.ARMident()
109     print "get_register(3):             %x"%client.ARMget_register(3)
110     print "IDCODE:      %x"%client.ARMident()
111     print "Debug State: %x"%client.ARMget_dbgstate ()
112     print "IDCODE:      %x"%client.ARMident()
113
114 def test3():
115     print "IDCODE:      %x"%client.ARMident()
116     print "Debug State: %x"%client.ARMget_dbgstate ()
117     client.writecmd(0x33,0xd0,4,[0xf7,0xf7,0xf7,0xf7])
118     print "Loopback:   \t %s"%repr(client.data)                  # loopback
119     client.writecmd(0x33,0xd5,8,[0,0,0,0,0xf7,0xf7,0xf7,0xf7])
120     print "test_set_reg:   \t %s"%repr(client.data)            
121     client.writecmd(0x33,0xd4,1,[0])
122     print "test_get_reg:   \t %s"%repr(client.data)           
123     print "set_register(3,0x41414141):  %x"%client.ARMset_register(3,0x41414141)
124     print "get_register(3):             %x"%client.ARMget_register(3)
125     client.writecmd(0x33,0xd4,1,[0])
126     print "test_get_reg:   \t %s"%repr(client.data)           
127
128 def test4():
129         print "IDCODE:      %x"%client.ARMident()
130         print "Debug State: %x"%client.ARMget_dbgstate ()
131         print "set_register(0,0x4141):  %x"%client.ARMset_register(0,0x4141)
132         print "get_register(0):             %x"%client.ARMget_register(0)
133         print "set_register(1,0x4141):  %x"%client.ARMset_register(1,0x4142)
134         print "get_register(1):             %x"%client.ARMget_register(1)
135         print "set_register(2,0x4141):  %x"%client.ARMset_register(2,0x4143)
136         print "get_register(2):             %x"%client.ARMget_register(2)
137         print "set_register(3,0x4141):  %x"%client.ARMset_register(3,0x4144)
138         print "get_register(3):             %x"%client.ARMget_register(3)
139         print "set_register(4,0x4141):  %x"%client.ARMset_register(4,0x4145)
140         print "get_register(4):             %x"%client.ARMget_register(4)
141         print "set_register(5,0x4141):  %x"%client.ARMset_register(5,0x4146)
142         print "get_register(5):             %x"%client.ARMget_register(5)
143         print "set_register(6,0x4141):  %x"%client.ARMset_register(6,0x4147)
144         print "get_register(6):             %x"%client.ARMget_register(6)
145         print "set_register(7,0x4141):  %x"%client.ARMset_register(7,0x4148)
146         print "get_register(7):             %x"%client.ARMget_register(7)
147         print "set_register(8,0x4141):  %x"%client.ARMset_register(8,0x4149)
148         print "get_register(8):             %x"%client.ARMget_register(8)
149         print "set_register(9,0x4141):  %x"%client.ARMset_register(9,0x4140)
150         print "get_register(9):             %x"%client.ARMget_register(9)
151         print "set_register(10,0x4141):  %x"%client.ARMset_register(10,0x4151)
152         print "get_register(10):             %x"%client.ARMget_register(10)
153         print "set_register(11,0x4141):  %x"%client.ARMset_register(11,0x4152)
154         print "get_register(11):             %x"%client.ARMget_register(11)
155         print "set_register(12,0x4141):  %x"%client.ARMset_register(12,0x4153)
156         print "get_register(12):             %x"%client.ARMget_register(12)
157         print "set_register(13,0x4141):  %x"%client.ARMset_register(13,0x4154)
158         print "get_register(13):             %x"%client.ARMget_register(13)
159         print "set_register(14,0x4141):  %x"%client.ARMset_register(14,0x4155)
160         print "get_register(14):             %x"%client.ARMget_register(14)
161         #print "set_register(15,0x4141):  %x"%client.ARMset_register(15,0x41414156)
162         #print "get_register(15):             %x"%client.ARMget_register(15)
163
164
165 seed = 0
166 def test5(start=0,end=15):
167     global results,seed
168     results = [[] for x in range(end)]
169     while True:
170         #print "IDCODE:      %x"%client.ARMident()
171         #print "Debug State: %x"%client.ARMget_dbgstate ()
172         for x in range(start,end):
173             num = client.ARMset_register(x,seed)
174             print "set_register(%d,0x%x):  %x"%(x,seed,num)
175             num = client.ARMget_register(x)
176             print "get_register(%d):             %x"%(x,num)
177             results[x].append(num)
178             if (num != seed):
179                 for y in range(13):
180                     num = client.ARMset_register(x,seed)
181                     print "set_register(%d,0x%x):  %x"%(x,seed,num)
182                     num = client.ARMget_register(x)
183                     print "get_register(%d):             %x"%(x,num)
184                     results[x].append(num)
185             seed += 1
186             client.ARMident()
187             client.ARMident()
188             print "Debug State: %x"%client.ARMget_dbgstate ()
189
190 def test6(start=0,end=15):
191     global results,seed
192     results = [[] for x in range(end)]
193     while True:
194         #print "IDCODE:      %x"%client.ARMident()
195         #print "Debug State: %x"%client.ARMget_dbgstate ()
196         for x in range(start,end):
197             num = client.ARMset_register(x,seed+x)
198             print "set_register(%d,0x%x):  %x"%(x,seed+x,num)
199             client.ARMident()
200             client.ARMident()
201         for x in range(start,end):
202             num = client.ARMget_register(x)
203             print "get_register(%d):             %x"%(x,num)
204             results[x].append(num)
205             if (num != seed+x):
206                 for y in range(13):
207                     num = client.ARMset_register(x,seed+x)
208                     print "set_register(%d,0x%x):  %x"%(x,seed+x,num)
209                     num = client.ARMget_register(x)
210                     print "get_register(%d):             %x"%(x,num)
211                     results[x].append(num)
212             client.ARMident()
213             client.ARMident()
214             print "Debug State: %x"%client.ARMget_dbgstate ()
215         seed += 1
216
217 def test7(start=0,end=14):
218     global results,seed
219     results = [[] for x in range(end)]
220     while True:
221         #print "IDCODE:      %x"%client.ARMident()
222         #print "Debug State: %x"%client.ARMget_dbgstate ()
223         for x in range(end,start-1, -1):
224             num = client.ARMset_register(x,seed+x)
225             time.sleep(1)
226             print "set_register(%d,0x%x):  %x"%(x,seed+x,num)
227         for y in range(10):
228           for x in range(start,endi+1):
229             num = client.ARMget_register(x)
230             time.sleep(1)
231             print "get_register(%d):             %x"%(x,num)
232             results[x].append(num)
233         seed += 1
234
235 def readtest(start=0,end=14):
236     global results,seed
237     results = [[] for x in range(end)]
238     while True:
239         #print "IDCODE:      %x"%client.ARMident()
240         #print "Debug State: %x"%client.ARMget_dbgstate ()
241         for x in range(end,start, -1):
242             num = client.ARMget_register(x)
243             time.sleep(1)
244             print "get_register(%d):  %x"%(x,num)
245         for y in range(10):
246           for x in range(start,end,2):
247             num = client.ARMget_register(x)
248             time.sleep(1)
249             print "get_register(%d):             %x"%(x,num)
250             results[x].append(num)
251         seed += 1
252
253 init()
254 print "Don't forget to 'client.stop()' if you want to exit cleanly"
255
256
257 def printResults():
258     for y in range(len(results)):
259             x=results[y]
260             print "%.2x=%s"%(y,repr(["%x"%t for t in x]))
261
262 """
263   case 0xD0: // loopback test
264     cmddatalong[0] = 0x12345678;
265   case 0xD1: // Set Scan Chain
266     cmddatalong[0] = jtagarm7tdmi_scan_n(cmddataword[0]);
267   case 0xD2: //
268     cmddatalong[0] = jtagarm7tdmi_get_dbgstate();
269   case 0xD3:
270     cmddatalong[0] = jtagarm7tdmi_exec(cmddatalong[0]);
271   case 0xD4:
272     cmddatalong[0] = jtagarm7tdmi_get_register(cmddata[0]);
273   case 0xD5:
274     cmddatalong[0] = jtagarm7tdmi_set_register(cmddata[0], cmddatalong[1]);
275   case 0xD6:
276     cmddatalong[0] = jtagarm7tdmi_dr_shift32(cmddatalong[0]);
277   case 0xD7:
278     cmddatalong[0] = jtagarm7tdmi_chain1(cmddatalong[0], 0);
279   case 0xD8:
280     cmddatalong[0] = jtagarm7tdmi_chain2_read(cmddata[0], 32);
281 """
282
283 """
284 if(sys.argv[1]=="test"):
285     client.CCtest();
286 if(sys.argv[1]=="deadtest"):
287     for i in range(1,10):
288         print "IDENT as %s" % client.CCidentstr();
289 if(sys.argv[1]=="dumpcode"):
290     f = sys.argv[2];
291     start=0x0000;
292     stop=0xFFFF;
293     if(len(sys.argv)>3):
294         start=int(sys.argv[3],16);
295     if(len(sys.argv)>4):
296         stop=int(sys.argv[4],16);
297     
298     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
299     h = IntelHex(None);
300     i=start;
301     while i<=stop:
302         h[i]=client.CCpeekcodebyte(i);
303         if(i%0x100==0):
304             print "Dumped %04x."%i;
305         i+=1;
306     h.write_hex_file(f);
307 if(sys.argv[1]=="dumpdata"):
308     f = sys.argv[2];
309     start=0xE000;
310     stop=0xFFFF;
311     if(len(sys.argv)>3):
312         start=int(sys.argv[3],16);
313     if(len(sys.argv)>4):
314         stop=int(sys.argv[4],16);
315     
316     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
317     h = IntelHex(None);
318     i=start;
319     while i<=stop:
320         h[i]=client.CCpeekdatabyte(i);
321         if(i%0x100==0):
322             print "Dumped %04x."%i;
323         i+=1;
324     h.write_hex_file(f);
325 if(sys.argv[1]=="status"):
326     print "Status: %s" %client.CCstatusstr();
327 if(sys.argv[1]=="erase"):
328     print "Status: %s" % client.CCstatusstr();
329     client.CCchiperase();
330     print "Status: %s" %client.CCstatusstr();
331
332 if(sys.argv[1]=="peekinfo"):
333     print "Select info flash."
334     client.CCwr_config(1);
335     print "Config is %02x" % client.CCrd_config();
336     
337     start=0x0000;
338     if(len(sys.argv)>2):
339         start=int(sys.argv[2],16);
340     stop=start;
341     if(len(sys.argv)>3):
342         stop=int(sys.argv[3],16);
343     print "Peeking from %04x to %04x." % (start,stop);
344     while start<=stop:
345         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
346         start=start+1;
347 if(sys.argv[1]=="poke"):
348     client.CCpokeirambyte(int(sys.argv[2],16),
349                           int(sys.argv[3],16));
350 if(sys.argv[1]=="randtest"):
351     #Seed RNG
352     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
353     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
354     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
355     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
356     
357     #Dump values
358     for foo in range(1,10):
359         print "%02x" % client.CCpeekirambyte(0xBD); #RNDH
360         client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
361         client.CCreleasecpu();
362         client.CChaltcpu();
363     print "%02x" % client.CCpeekdatabyte(0xDF61); #CHIP ID
364 if(sys.argv[1]=="adctest"):
365     # ADCTest 0xDF3A 0xDF3B
366     print "ADC TEST %02x%02x" % (
367         client.CCpeekdatabyte(0xDF3A),
368         client.CCpeekdatabyte(0xDF3B));
369 if(sys.argv[1]=="config"):
370     print "Config is %02x" % client.CCrd_config();
371
372 if(sys.argv[1]=="flash"):
373      f=sys.argv[2];
374      start=0;
375      stop=0xFFFF;
376      if(len(sys.argv)>3):
377          start=int(sys.argv[3],16);
378      if(len(sys.argv)>4):
379          stop=int(sys.argv[4],16);
380    
381      h = IntelHex(f);
382      page = 0x0000;
383      pagelen = 2048; #2kB pages in 32-bit words
384      bcount = 0;
385      
386      print "Wiping Flash."
387      #Wipe all of flash.
388      #client.CCchiperase();
389      #Wipe the RAM buffer for the next flash page.
390      #client.CCeraseflashbuffer();
391      for i in h._buf.keys():
392          while(i>page+pagelen):
393              if bcount>0:
394                  client.CCflashpage(page);
395                  #client.CCeraseflashbuffer();
396                  bcount=0;
397                  print "Flashed page at %06x" % page
398              page+=pagelen;
399              
400          #Place byte into buffer.
401          client.CCpokedatabyte(0xF000+i-page,
402                                h[i]);
403          bcount+=1;
404          if(i%0x100==0):
405                 print "Buffering %04x toward %06x" % (i,page);
406      #last page
407      client.CCflashpage(page);
408      print "Flashed final page at %06x" % page;
409      
410 if(sys.argv[1]=="lock"):
411     print "Status: %s" %client.CCstatusstr();
412     client.CClockchip();
413     print "Status: %s" %client.CCstatusstr();
414 if(sys.argv[1]=="flashpage"):
415     target=0;
416     if(len(sys.argv)>2):
417         target=int(sys.argv[2],16);
418     print "Writing a page of flash from 0xF000 in XDATA"
419     client.CCflashpage(target);
420 if(sys.argv[1]=="erasebuffer"):
421     print "Erasing flash buffer.";
422     client.CCeraseflashbuffer();
423
424 if(sys.argv[1]=="writedata"):
425     f=sys.argv[2];
426     start=0;
427     stop=0xFFFF;
428     if(len(sys.argv)>3):
429         start=int(sys.argv[3],16);
430     if(len(sys.argv)>4):
431         stop=int(sys.argv[4],16);
432     
433     h = IntelHex(f);
434     
435     for i in h._buf.keys():
436         if(i>=start and i<=stop):
437             client.CCpokedatabyte(i,h[i]);
438             if(i%0x100==0):
439                 print "%04x" % i;
440 #if(sys.argv[1]=="flashtest"):
441 #    client.CCflashtest();
442 if(sys.argv[1]=="peekdata"):
443     start=0x0000;
444     if(len(sys.argv)>2):
445         start=int(sys.argv[2],16);
446     stop=start;
447     if(len(sys.argv)>3):
448         stop=int(sys.argv[3],16);
449     print "Peeking from %04x to %04x." % (start,stop);
450     while start<=stop:
451         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
452         start=start+1;
453 if(sys.argv[1]=="peek"):
454     start=0x0000;
455     if(len(sys.argv)>2):
456         start=int(sys.argv[2],16);
457     stop=start;
458     if(len(sys.argv)>3):
459         stop=int(sys.argv[3],16);
460     print "Peeking from %04x to %04x." % (start,stop);
461     while start<=stop:
462         print "%04x: %02x" % (start,client.CCpeekirambyte(start));
463         start=start+1;
464
465 if(sys.argv[1]=="peekcode"):
466     start=0x0000;
467     if(len(sys.argv)>2):
468         start=int(sys.argv[2],16);
469     stop=start;
470     if(len(sys.argv)>3):
471         stop=int(sys.argv[3],16);
472     print "Peeking from %04x to %04x." % (start,stop);
473     while start<=stop:
474         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
475         start=start+1;
476 if(sys.argv[1]=="pokedata"):
477     start=0x0000;
478     val=0x00;
479     if(len(sys.argv)>2):
480         start=int(sys.argv[2],16);
481     if(len(sys.argv)>3):
482         val=int(sys.argv[3],16);
483     print "Poking %04x to become %02x." % (start,val);
484     client.CCpokedatabyte(start,val);
485
486 client.stop();
487 """