git.rot13.org Git - goodfet/blob - client/goodfet.arm7

   1 #!/usr/bin/env python
   2
   3 import sys;
   4 import struct
   5 import binascii;
   6
   7 from GoodFETARM7 import GoodFETARM
   8 from intelhex import IntelHex16bit, IntelHex
   9
  10
  11 if(len(sys.argv)==1):
  12     print "Usage: %s verb [objects]\n" % sys.argv[0]
  13     # halt, resume... but need for flash and dump to leave it as it was
  14     print "%s info" % sys.argv[0]
  15     print "%s dump $foo.hex [0x$start 0x$stop]" % sys.argv[0]
  16     print "%s erase" % sys.argv[0]
  17     print "%s eraseinfo" % sys.argv[0]
  18     print "%s flash $foo.hex [0x$start 0x$stop]" % sys.argv[0]
  19     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0]
  20     print "%s poke 0x$adr 0x$val" % sys.argv[0]
  21     print "%s peek 0x$start [0x$stop]" % sys.argv[0]
  22     print "%s reset" % sys.argv[0]
  23     sys.exit()
  24
  25 #Initialize FET and set baud rate
  26 client=GoodFETARM();
  27 client.serInit()
  28
  29 client.setup();
  30 client.start();
  31
  32 if(sys.argv[1]=="info"):
  33     pass
  34
  35 if(sys.argv[1]=="dump"):
  36     f = sys.argv[2];
  37     start=0x00000000;
  38     stop=0xFFFFFFFF;
  39     if(len(sys.argv)>3):
  40         start=int(sys.argv[3],16);
  41     if(len(sys.argv)>4):
  42         stop=int(sys.argv[4],16);
  43
  44     print "Dumping from %04x to %04x as %s." % (start,stop,f);
  45     #h = IntelHex16bit(None);
  46     # FIXME: get mcu state and return it to that state
  47     client.halt()
  48
  49     h = IntelHex(None);
  50     i=start;
  51     while i<=stop:
  52         #data=client.ARMreadMem(i, 48);
  53         data=client.ARMreadChunk(i, 48, verbose=0);
  54         print "Dumped %06x."%i;
  55         for dword in data:
  56             if i<=stop and dword != 0xdeadbeef:
  57                 h.puts( i, struct.pack("<I", dword) )
  58             i+=4;
  59     # FIXME: get mcu state and return it to that state
  60     client.resume()
  61     h.write_hex_file(f);
  62
  63 '''
  64 if(sys.argv[1]=="erase"):
  65     print "Erasing main flash memory."
  66     client.ARMmasserase();
  67
  68 if(sys.argv[1]=="eraseinfo"):
  69     print "Erasing info memory."
  70     client.ARMinfoerase();
  71
  72
  73 '''
  74 if(sys.argv[1]=="ivt"):
  75     client.ARMreprChunk(0xFFC0,0xFFFF);
  76
  77 if(sys.argv[1]=="regs"):
  78     for i in range(0,16):
  79         print "r%i=%04x" % (i,client.ARMget_register(i));
  80
  81 if(sys.argv[1]=="flash"):
  82     f=sys.argv[2];
  83     start=0;
  84     stop=0x10000;
  85     if(len(sys.argv)>3):
  86         start=int(sys.argv[3],16);
  87     if(len(sys.argv)>4):
  88         stop=int(sys.argv[4],16);
  89
  90     client.halt()
  91     h = IntelHex16bit(f);
  92
  93     #Should this be default?
  94     #Makes flashing multiple images inconvenient.
  95     #client.ARMmasserase();
  96
  97     count=0; #Bytes in commit.
  98     first=0;
  99     vals=[];
 100     last=0;  #Last address committed.
 101     for i in h._buf.keys():
 102         if((count>0x40 or last+2!=i) and count>0 and i&1==0):
 103             #print "%i, %x, %x" % (len(vals), last, i);
 104             client.ARMpokeflashblock(first,vals);
 105             count=0;
 106             first=0;
 107             last=0;
 108             vals=[];
 109         if(i>=start and i<stop  and i&1==0):
 110             val=h[i>>1];
 111             if(count==0):
 112                 first=i;
 113             last=i;
 114             count+=2;
 115             vals+=[val&0xff,(val&0xff00)>>8];
 116             if(i%0x100==0):
 117                 print "%04x" % i;
 118     if count>0: #last commit, ivt
 119         client.ARMpokeflashblock(first,vals);
 120     client.resume()
 121
 122 if(sys.argv[1]=="verify"):
 123     f=sys.argv[2];
 124     start=0;
 125     stop=0xFFFF;
 126     if(len(sys.argv)>3):
 127         start=int(sys.argv[3],16);
 128     if(len(sys.argv)>4):
 129         stop=int(sys.argv[4],16);
 130
 131     client.halt()
 132     h = IntelHex16bit(f);
 133     for i in h._buf.keys():
 134         if(i>=start and i<stop and i&1==0):
 135             peek=client.ARMreadMem(i)
 136             if(h[i>>1]!=peek):
 137                 print "ERROR at %04x, found %04x not %04x"%(i,peek,h[i>>1]);
 138             if(i%0x100==0):
 139                 print "%04x" % i;
 140     client.resume()
 141
 142
 143 if(sys.argv[1]=="peek"):
 144     start=0x0000;
 145     if(len(sys.argv)>2):
 146         start=int(sys.argv[2],16);
 147     stop=start;
 148     if(len(sys.argv)>3):
 149         stop=int(sys.argv[3],16);
 150
 151     print "Peeking from %04x to %04x." % (start,stop);
 152     client.halt()
 153     for dword in client.ARMreadChunk(start, (stop-start)/4, verbose=0):
 154         print "%.4x: %.8x" % (start, dword)
 155         start += 4
 156     client.resume()
 157
 158 if(sys.argv[1]=="poke"):
 159     start=0x0000;
 160     val=0x00;
 161     if(len(sys.argv)>2):
 162         start=int(sys.argv[2],16);
 163     if(len(sys.argv)>3):
 164         val=int(sys.argv[3],16);
 165
 166     print "Poking %06x to become %04x." % (start,val);
 167     client.halt()
 168     #???while client.ARMreadMem(start)[0]&(~val)>0:
 169     client.ARMwriteChunk(start, [val])
 170     print "Poked to %.8x" % client.ARMreadMem(start)[0]
 171     client.resume()
 172
 173
 174 if(sys.argv[1]=="reset"):
 175     #Set PC to RESET vector's value.
 176
 177     client.ARMsetPC(0x00000000);
 178     client.ARMreleasecpu();
 179
 180 #client.ARMreleasecpu();
 181 #client.ARMstop();