Cleaning up the Chipcon client.
[goodfet] / 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