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
  12 from GoodFET import GoodFET;
  13 from intelhex import IntelHex;
  14
  15 if(len(sys.argv)==1):
  16     print "Usage: %s verb [objects]\n" % sys.argv[0];
  17     print "%s info" % sys.argv[0];
  18     print "%s dump $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  19     print "%s erase" % sys.argv[0];
  20     print "%s write $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  21     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  22     print "%s peek 0x$start [0x$stop]" % sys.argv[0];
  23     print "%s poke 0x$adr 0x$val" % sys.argv[0];
  24     sys.exit();
  25
  26 #Initailize FET and set baud rate
  27 client=GoodFET();
  28 client.serInit("/dev/ttyUSB0")
  29
  30 client.SPIsetup();
  31
  32 if(sys.argv[1]=="test"):
  33     result="";
  34     dropped=0;
  35     for i in range(40):
  36         data=client.SPIjedec();
  37         if ord(data[1])==0xFF:
  38             result+="-";
  39             dropped=dropped+1;
  40         else:
  41             result+="+";
  42     print "Connection Test: (- is bad)\n%s" % result;
  43     print "%i misreads" % dropped;
  44     if(dropped==40):
  45         print "No successful reads.  Is the chip wired correctly?";
  46     elif(dropped>0):
  47         print "Some success, some failures.  Is a wire loose?";
  48     else:
  49         print "All reads succeeded.  Wiring is probably good.";
  50
  51 if(sys.argv[1]=="info"):
  52     data=client.SPIjedec();
  53     print "Ident as %s\nManufacturer: %02x %s\nType: %02x\nCapacity: %02x" % (
  54         client.SPIjedecstr(),
  55         ord(data[1]),client.SPIjedecmanstr(),
  56         ord(data[2]),
  57         ord(data[3]));
  58
  59 if(sys.argv[1]=="dump"):
  60     f = sys.argv[2];
  61     start=0x0000;
  62     stop=0x100000; #TODO, adjust this by the JEDEC size parameter.
  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
  69
  70     print "Dumping code from %06x to %06x as %s." % (start,stop,f);
  71     file = open(f, mode='wb')
  72
  73     i=start;
  74     while i<=stop:
  75         data=client.SPIpeekblock(i);
  76
  77
  78         if(i%0x100==0):
  79             print "Dumped %06x."%i;
  80         for j in data:
  81             file.write(j);
  82             i+=1;
  83     file.close()
  84 if(sys.argv[1]=="erase"):
  85   client.SPIchiperase();
  86
  87 if(sys.argv[1]=="peek"):
  88     start=0x0000;
  89     if(len(sys.argv)>2):
  90         start=int(sys.argv[2],16);
  91     stop=start;
  92     if(len(sys.argv)>3):
  93         stop=int(sys.argv[3],16);
  94     print "Peeking from %06x to %06x." % (start,stop);
  95     while start<=stop:
  96         print "%06x: %02x" % (start,client.SPIpeek(start));
  97         start=start+1;
  98
  99 if(sys.argv[1]=="poke"):
 100     start=0x0000;
 101     val=0x00;
 102     if(len(sys.argv)>2):
 103         start=int(sys.argv[2],16);
 104     if(len(sys.argv)>3):
 105         val=int(sys.argv[3],16);
 106     print "Poking %06x to become %02x." % (start,val);
 107     client.SPIpokebyte(start,val);
 108
 109