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

   1 #!/usr/bin/env python
   2
   3 #GoodFET SPI Flash Client
   4 #by Travis Goodspeed
   5
   6 #N.B.,
   7 #Might be Winbond W25x80-specific.
   8
   9 import sys;
  10 import binascii;
  11 import array;
  12
  13 from GoodFET import GoodFET;
  14 from intelhex import IntelHex;
  15
  16 if(len(sys.argv)==1):
  17     print "Usage: %s verb [objects]\n" % sys.argv[0];
  18     print "%s info" % sys.argv[0];
  19     print "%s dump $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  20     print "%s erase" % sys.argv[0];
  21     print "%s write $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  22     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  23     print "%s peek 0x$start [0x$stop]" % sys.argv[0];
  24     print "%s poke 0x$adr 0x$val" % sys.argv[0];
  25     sys.exit();
  26
  27 #Initailize FET and set baud rate
  28 client=GoodFET();
  29 client.serInit("/dev/ttyUSB0")
  30
  31
  32 client.SPIsetup();
  33
  34 #Dummy read.
  35 #Might read as all ones if chip has a startup delay.
  36 client.SPIjedec();
  37
  38 if(sys.argv[1]=="test"):
  39     result="";
  40     dropped=0;
  41     for i in range(40):
  42         data=client.SPIjedec();
  43         if ord(data[1])==0xFF:
  44             result+="-";
  45             dropped=dropped+1;
  46         else:
  47             result+="+";
  48     print "Connection Test: (- is bad)\n%s" % result;
  49     print "%i misreads" % dropped;
  50     if(dropped==40):
  51         print "No successful reads.  Is the chip wired correctly?";
  52     elif(dropped>0):
  53         print "Some success, some failures.  Is a wire loose?";
  54     else:
  55         print "All reads succeeded.  Wiring is probably good.";
  56
  57 if(sys.argv[1]=="info"):
  58     data=client.SPIjedec();
  59     print "Ident as %s\nManufacturer: %02x %s\nType: %02x\nCapacity: %02x" % (
  60         client.SPIjedecstr(),
  61         ord(data[1]),client.SPIjedecmanstr(),
  62         ord(data[2]),
  63         ord(data[3]));
  64
  65 if(sys.argv[1]=="dump"):
  66     f = sys.argv[2];
  67     start=0x0000;
  68     stop=0x100000; #TODO, adjust this by the JEDEC size parameter.
  69     if(len(sys.argv)>3):
  70         start=int(sys.argv[3],16);
  71     if(len(sys.argv)>4):
  72         stop=int(sys.argv[4],16);
  73
  74     print "Dumping code from %06x to %06x as %s." % (start,stop,f);
  75     file = open(f, mode='wb')
  76
  77     i=start;
  78     while i<=stop:
  79         data=client.SPIpeekblock(i);
  80
  81
  82         if(i%0x100==0):
  83             print "Dumped %06x."%i;
  84         for j in data:
  85             file.write(j);
  86             i+=1;
  87     file.close()
  88 if(sys.argv[1]=="flash"):
  89     f = sys.argv[2];
  90     start=0x0000;
  91     stop=0x100000; #TODO, adjust this by the JEDEC size parameter.
  92     if(len(sys.argv)>3):
  93         start=int(sys.argv[3],16);
  94     if(len(sys.argv)>4):
  95         stop=int(sys.argv[4],16);
  96
  97     print "Flashing code from %06x to %06x with %s." % (start,stop,f);
  98     file = open(f, mode='rb')
  99
 100     i=start;
 101     chars=list(file.read());
 102     chunksize=0x80;
 103
 104     while i<=stop:
 105         bytes=range(0,chunksize);
 106         for j in range(0,chunksize):
 107             bytes[j]=ord(chars[i+j]);
 108         #client.SPIpokebyte(i,ord(chars[i]));
 109         client.SPIpokebytes(i,bytes);
 110
 111         i+=chunksize;
 112         if(i%0x100==0):
 113             print "Flashed %06x."%i;
 114     file.close()
 115
 116
 117 if(sys.argv[1]=="erase"):
 118   client.SPIchiperase();
 119
 120 if(sys.argv[1]=="peek"):
 121     start=0x0000;
 122     if(len(sys.argv)>2):
 123         start=int(sys.argv[2],16);
 124     stop=start;
 125     if(len(sys.argv)>3):
 126         stop=int(sys.argv[3],16);
 127     print "Peeking from %06x to %06x." % (start,stop);
 128     while start<=stop:
 129         print "%06x: %02x" % (start,client.SPIpeek(start));
 130         start=start+1;
 131
 132 if(sys.argv[1]=="poke"):
 133     start=0x0000;
 134     val=0x00;
 135     if(len(sys.argv)>2):
 136         start=int(sys.argv[2],16);
 137     if(len(sys.argv)>3):
 138         val=int(sys.argv[3],16);
 139     print "Poking %06x to become %02x." % (start,val);
 140
 141     while client.SPIpeek(start)!=val:
 142         client.SPIpokebyte(start,val);
 143         print "Poked to %02x" % client.SPIpeek(start);
 144