git.rot13.org Git - goodfet/blob - client/gplay-arm.py

   1 #!/usr/bin/env ipython
   2 import sys, struct, binascii
   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     client.serInit()
  13     #
  14     #Connect to target
  15     client.setup()
  16     client.start()
  17     print "STARTUP: "+repr(client.data)
  18     #
  19
  20 def test1():
  21     global data
  22     print "\n\nTesting JTAG for ARM\n"
  23     client.writecmd(0x33,0xd0,4,[0x40,0x40,0x40,0x40]); print "loopback:   \t %s"%repr(client.data)                  # loopback
  24     data.append(client.data)
  25     client.writecmd(0x33,0xd1,2,[1,0]); print "scanchain1:\t %s"%repr(client.data)               # set scan chain
  26     data.append(client.data)
  27     client.writecmd(0x33,0xd2,0,[]); print "debug state:\t %s"%repr(client.data)                  # get dbg state
  28     data.append(client.data)
  29     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
  30     data.append(client.data)
  31     client.writecmd(0x33,0xd3,0,[0,0,0x8e,0xe5]); print "exec_stuff: \t %s"%repr(client.data)     # execute instruction
  32     data.append(client.data)
  33     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
  34     data.append(client.data)
  35     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
  36     data.append(client.data)
  37     client.writecmd(0x33,0xd3,0,[0,0,0xa0,0xe1]); print "exec_nop: \t %s"%repr(client.data)     # execute instruction
  38     data.append(client.data)
  39     client.writecmd(0x33,0xd6,0,[]); print "shift_dr_32: \t %s"%repr(client.data)                  # dr_shift32
  40     data.append(client.data)
  41     client.writecmd(0x33,0xd5,8,[3, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40]); print "set_register:\t %s"%repr(client.data)                  # set register
  42     data.append(client.data)
  43     client.writecmd(0x33,0xd4,1,[3]); print "get_register:\t %s"%repr(client.data)                  # get register
  44     data.append(client.data)
  45     client.writecmd(0x33,0xd7,0,[]); print "chain1:      \t %s"%repr(client.data)                  # chain1
  46     data.append(client.data)
  47     client.writecmd(0x33,0xd8,0,[]); print "read_chain2: \t %s"%repr(client.data)                  # read chain2
  48     data.append(client.data)
  49     client.writecmd(0x33,0xd9,0,[]); print "idcode:      \t %s"%repr(client.data)                  # read idcode
  50     data.append(client.data)
  51     client.writecmd(0x33,0xf0,2,[4,4,1,1]); print "f0:       \t %s"%repr(client.data)   # read idcode
  52     data.append(client.data)
  53     client.writecmd(0x33,0xdb,8,[0x0,4,4,4,4,4,4,4]); print "verb(0):     \t %s"%repr(client.data)
  54     data.append(client.data)
  55     client.writecmd(0x33,0xdb,8,[0x2,4,4,4,4,4,4,4]); print "verb(2):     \t %s"%repr(client.data)
  56     data.append(client.data)
  57     client.writecmd(0x33,0xdb,8,[0x3,4,4,4,4,4,4,4]); print "verb(3):     \t %s"%repr(client.data)
  58     data.append(client.data)
  59     client.writecmd(0x33,0xdb,8,[0x4,4,4,4,4,4,4,4]); print "verb(4):     \t %s"%repr(client.data)
  60     data.append(client.data)
  61     client.writecmd(0x33,0xdb,8,[0x5,4,4,4,4,4,4,4]); print "verb(5):     \t %s"%repr(client.data)
  62     data.append(client.data)
  63     client.writecmd(0x33,0xdb,8,[0x7,4,4,4,4,4,4,4]); print "verb(7):     \t %s"%repr(client.data)
  64     data.append(client.data)
  65     client.writecmd(0x33,0xdb,8,[0x9,4,4,4,4,4,4,4]); print "verb(9):     \t %s"%repr(client.data)
  66     data.append(client.data)
  67     client.writecmd(0x33,0xdb,8,[0xc,4,4,4,4,4,4,4]); print "verb(c):     \t %s"%repr(client.data)
  68     data.append(client.data)
  69     client.writecmd(0x33,0xdb,8,[0xe,0,0,0,0,0,0xa0,0xe1]); print "verb(e):     \t %s"%repr(client.data)
  70     data.append(client.data)
  71     client.writecmd(0x33,0xdb,8,[0xf,4,4,4,4,4,4,4]); print "verb(f):     \t %s"%repr(client.data)
  72     data.append(client.data)
  73
  74 def test2():
  75     global data
  76     print "\n\nTesting JTAG for ARM\n"
  77     print "IDCODE:      %x"%client.ARMident()
  78     print "Debug State: %x"%client.ARMget_dbgstate ()
  79     print "Debug State: %x"%client.ARMget_dbgstate ()
  80     print "Debug State: %x"%client.ARMget_dbgstate ()
  81     print "Debug CTRL:  %x"%client.ARMget_dbgctrl()
  82     client.writecmd(0x33,0xda,0,[])
  83     print "TEST CHAIN0: %s"%repr(client.data)
  84     print "Debug State: %x"%client.ARMget_dbgstate ()
  85     print "IDCODE:      %x"%client.ARMident()
  86     #print "Debug State: %x"%client.ARMget_dbgstate ()
  87     client.writecmd(0x33,0xd0,4,[0xf7,0xf7,0xf7,0xf7])
  88     print "Loopback:   \t %s"%repr(client.data)                  # loopback
  89     print "Debug State: %x"%client.ARMget_dbgstate ()
  90     print "IDCODE:      %x"%client.ARMident()
  91     print "GetPC: %x"%client.ARMgetPC()
  92     print "IDCODE:      %x"%client.ARMident()
  93     print "Debug State: %x"%client.ARMget_dbgstate ()
  94     print "IDCODE:      %x"%client.ARMident()
  95     print "set_register(3,0x41414141):  %x"%client.ARMset_register(3,0x41414141)
  96     print "IDCODE:      %x"%client.ARMident()
  97     print "Debug State: %x"%client.ARMget_dbgstate ()
  98     print "IDCODE:      %x"%client.ARMident()
  99     print "get_register(3):             %x"%client.ARMget_register(3)
 100     print "IDCODE:      %x"%client.ARMident()
 101     print "Debug State: %x"%client.ARMget_dbgstate ()
 102     print "IDCODE:      %x"%client.ARMident()
 103
 104 def test3():
 105     print "IDCODE:      %x"%client.ARMident()
 106     print "Debug State: %x"%client.ARMget_dbgstate ()
 107     client.writecmd(0x33,0xd0,4,[0xf7,0xf7,0xf7,0xf7])
 108     print "Loopback:   \t %s"%repr(client.data)                  # loopback
 109     client.writecmd(0x33,0xd5,8,[0,0,0,0,0xf7,0xf7,0xf7,0xf7])
 110     print "test_set_reg:   \t %s"%repr(client.data)
 111     client.writecmd(0x33,0xd4,1,[0])
 112     print "test_get_reg:   \t %s"%repr(client.data)
 113     print "set_register(3,0x41414141):  %x"%client.ARMset_register(3,0x41414141)
 114     print "get_register(3):             %x"%client.ARMget_register(3)
 115     client.writecmd(0x33,0xd4,1,[0])
 116     print "test_get_reg:   \t %s"%repr(client.data)
 117
 118 init()
 119 print "Don't forget to 'client.stop()' if you want to exit cleanly"
 120
 121
 122
 123 """
 124   case 0xD0: // loopback test
 125     cmddatalong[0] = 0x12345678;
 126   case 0xD1: // Set Scan Chain
 127     cmddatalong[0] = jtagarm7tdmi_scan_n(cmddataword[0]);
 128   case 0xD2: //
 129     cmddatalong[0] = jtagarm7tdmi_get_dbgstate();
 130   case 0xD3:
 131     cmddatalong[0] = jtagarm7tdmi_exec(cmddatalong[0]);
 132   case 0xD4:
 133     cmddatalong[0] = jtagarm7tdmi_get_register(cmddata[0]);
 134   case 0xD5:
 135     cmddatalong[0] = jtagarm7tdmi_set_register(cmddata[0], cmddatalong[1]);
 136   case 0xD6:
 137     cmddatalong[0] = jtagarm7tdmi_dr_shift32(cmddatalong[0]);
 138   case 0xD7:
 139     cmddatalong[0] = jtagarm7tdmi_chain1(cmddatalong[0], 0);
 140   case 0xD8:
 141     cmddatalong[0] = jtagarm7tdmi_chain2_read(cmddata[0], 32);
 142 """
 143
 144 """
 145 if(sys.argv[1]=="test"):
 146     client.CCtest();
 147 if(sys.argv[1]=="deadtest"):
 148     for i in range(1,10):
 149         print "IDENT as %s" % client.CCidentstr();
 150 if(sys.argv[1]=="dumpcode"):
 151     f = sys.argv[2];
 152     start=0x0000;
 153     stop=0xFFFF;
 154     if(len(sys.argv)>3):
 155         start=int(sys.argv[3],16);
 156     if(len(sys.argv)>4):
 157         stop=int(sys.argv[4],16);
 158
 159     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
 160     h = IntelHex(None);
 161     i=start;
 162     while i<=stop:
 163         h[i]=client.CCpeekcodebyte(i);
 164         if(i%0x100==0):
 165             print "Dumped %04x."%i;
 166         i+=1;
 167     h.write_hex_file(f);
 168 if(sys.argv[1]=="dumpdata"):
 169     f = sys.argv[2];
 170     start=0xE000;
 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     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
 178     h = IntelHex(None);
 179     i=start;
 180     while i<=stop:
 181         h[i]=client.CCpeekdatabyte(i);
 182         if(i%0x100==0):
 183             print "Dumped %04x."%i;
 184         i+=1;
 185     h.write_hex_file(f);
 186 if(sys.argv[1]=="status"):
 187     print "Status: %s" %client.CCstatusstr();
 188 if(sys.argv[1]=="erase"):
 189     print "Status: %s" % client.CCstatusstr();
 190     client.CCchiperase();
 191     print "Status: %s" %client.CCstatusstr();
 192
 193 if(sys.argv[1]=="peekinfo"):
 194     print "Select info flash."
 195     client.CCwr_config(1);
 196     print "Config is %02x" % client.CCrd_config();
 197
 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.CCpeekcodebyte(start));
 207         start=start+1;
 208 if(sys.argv[1]=="poke"):
 209     client.CCpokeirambyte(int(sys.argv[2],16),
 210                           int(sys.argv[3],16));
 211 if(sys.argv[1]=="randtest"):
 212     #Seed RNG
 213     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 214     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 215     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 216     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 217
 218     #Dump values
 219     for foo in range(1,10):
 220         print "%02x" % client.CCpeekirambyte(0xBD); #RNDH
 221         client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 222         client.CCreleasecpu();
 223         client.CChaltcpu();
 224     print "%02x" % client.CCpeekdatabyte(0xDF61); #CHIP ID
 225 if(sys.argv[1]=="adctest"):
 226     # ADCTest 0xDF3A 0xDF3B
 227     print "ADC TEST %02x%02x" % (
 228         client.CCpeekdatabyte(0xDF3A),
 229         client.CCpeekdatabyte(0xDF3B));
 230 if(sys.argv[1]=="config"):
 231     print "Config is %02x" % client.CCrd_config();
 232
 233 if(sys.argv[1]=="flash"):
 234      f=sys.argv[2];
 235      start=0;
 236      stop=0xFFFF;
 237      if(len(sys.argv)>3):
 238          start=int(sys.argv[3],16);
 239      if(len(sys.argv)>4):
 240          stop=int(sys.argv[4],16);
 241
 242      h = IntelHex(f);
 243      page = 0x0000;
 244      pagelen = 2048; #2kB pages in 32-bit words
 245      bcount = 0;
 246
 247      print "Wiping Flash."
 248      #Wipe all of flash.
 249      #client.CCchiperase();
 250      #Wipe the RAM buffer for the next flash page.
 251      #client.CCeraseflashbuffer();
 252      for i in h._buf.keys():
 253          while(i>page+pagelen):
 254              if bcount>0:
 255                  client.CCflashpage(page);
 256                  #client.CCeraseflashbuffer();
 257                  bcount=0;
 258                  print "Flashed page at %06x" % page
 259              page+=pagelen;
 260
 261          #Place byte into buffer.
 262          client.CCpokedatabyte(0xF000+i-page,
 263                                h[i]);
 264          bcount+=1;
 265          if(i%0x100==0):
 266                 print "Buffering %04x toward %06x" % (i,page);
 267      #last page
 268      client.CCflashpage(page);
 269      print "Flashed final page at %06x" % page;
 270
 271 if(sys.argv[1]=="lock"):
 272     print "Status: %s" %client.CCstatusstr();
 273     client.CClockchip();
 274     print "Status: %s" %client.CCstatusstr();
 275 if(sys.argv[1]=="flashpage"):
 276     target=0;
 277     if(len(sys.argv)>2):
 278         target=int(sys.argv[2],16);
 279     print "Writing a page of flash from 0xF000 in XDATA"
 280     client.CCflashpage(target);
 281 if(sys.argv[1]=="erasebuffer"):
 282     print "Erasing flash buffer.";
 283     client.CCeraseflashbuffer();
 284
 285 if(sys.argv[1]=="writedata"):
 286     f=sys.argv[2];
 287     start=0;
 288     stop=0xFFFF;
 289     if(len(sys.argv)>3):
 290         start=int(sys.argv[3],16);
 291     if(len(sys.argv)>4):
 292         stop=int(sys.argv[4],16);
 293
 294     h = IntelHex(f);
 295
 296     for i in h._buf.keys():
 297         if(i>=start and i<=stop):
 298             client.CCpokedatabyte(i,h[i]);
 299             if(i%0x100==0):
 300                 print "%04x" % i;
 301 #if(sys.argv[1]=="flashtest"):
 302 #    client.CCflashtest();
 303 if(sys.argv[1]=="peekdata"):
 304     start=0x0000;
 305     if(len(sys.argv)>2):
 306         start=int(sys.argv[2],16);
 307     stop=start;
 308     if(len(sys.argv)>3):
 309         stop=int(sys.argv[3],16);
 310     print "Peeking from %04x to %04x." % (start,stop);
 311     while start<=stop:
 312         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
 313         start=start+1;
 314 if(sys.argv[1]=="peek"):
 315     start=0x0000;
 316     if(len(sys.argv)>2):
 317         start=int(sys.argv[2],16);
 318     stop=start;
 319     if(len(sys.argv)>3):
 320         stop=int(sys.argv[3],16);
 321     print "Peeking from %04x to %04x." % (start,stop);
 322     while start<=stop:
 323         print "%04x: %02x" % (start,client.CCpeekirambyte(start));
 324         start=start+1;
 325
 326 if(sys.argv[1]=="peekcode"):
 327     start=0x0000;
 328     if(len(sys.argv)>2):
 329         start=int(sys.argv[2],16);
 330     stop=start;
 331     if(len(sys.argv)>3):
 332         stop=int(sys.argv[3],16);
 333     print "Peeking from %04x to %04x." % (start,stop);
 334     while start<=stop:
 335         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 336         start=start+1;
 337 if(sys.argv[1]=="pokedata"):
 338     start=0x0000;
 339     val=0x00;
 340     if(len(sys.argv)>2):
 341         start=int(sys.argv[2],16);
 342     if(len(sys.argv)>3):
 343         val=int(sys.argv[3],16);
 344     print "Poking %04x to become %02x." % (start,val);
 345     client.CCpokedatabyte(start,val);
 346
 347 client.stop();
 348 """