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=0xFFFF;
  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     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
  69     h = IntelHex(None);
  70     i=start;
  71     while i<=stop:
  72         h[i>>1]=client.SPIpeek(i);
  73         if(i%0x100==0):
  74             print "Dumped %04x."%i;
  75         i+=1;
  76     h.write_hex_file(f);
  77
  78 if(sys.argv[1]=="erase"):
  79   print "Status: %s" % client.CCstatusstr();
  80   client.CCchiperase();
  81   print "Status: %s" %client.CCstatusstr();
  82
  83 if(sys.argv[1]=="peek"):
  84     start=0x0000;
  85     if(len(sys.argv)>2):
  86         start=int(sys.argv[2],16);
  87     stop=start;
  88     if(len(sys.argv)>3):
  89         stop=int(sys.argv[3],16);
  90     print "Peeking from %04x to %04x." % (start,stop);
  91     while start<=stop:
  92         print "%04x: %02x" % (start,client.SPIpeek(start));
  93         start=start+1;
  94 # if(sys.argv[1]=="poke"):
  95 #     start=0x0000;
  96 #     val=0x00;
  97 #     if(len(sys.argv)>2):
  98 #         start=int(sys.argv[2],16);
  99 #     if(len(sys.argv)>3):
 100 #         val=int(sys.argv[3],16);
 101 #     print "Poking %04x to become %02x." % (start,val);
 102 #     client.CCpokedatabyte(start,val);
 103
 104