2 # GoodFET SPI and SPIFlash Client Library
4 # (C) 2009 Travis Goodspeed <travis at radiantmachines.com>
6 # This code is being rewritten and refactored. You've been warned!
8 import sys, time, string, cStringIO, struct, glob, serial, os;
10 from GoodFET import GoodFET;
12 class GoodFETSPI(GoodFET):
14 """Move the FET into the SPI application."""
15 self.writecmd(0x01,0x10,0,self.data); #SPI/SETUP
17 def SPItrans8(self,byte):
18 """Read and write 8 bits by SPI."""
19 data=self.SPItrans([byte]);
22 def SPItrans(self,data):
23 """Exchange data by SPI."""
25 self.writecmd(0x01,0x00,len(data),data);
28 class GoodFETSPI25C(GoodFETSPI):
38 """Read a byte from the given address."""
39 data=self.SPItrans([self.READ,(adr>>8)&0xFF,adr&0xFF,0x00]);
43 class GoodFETSPIFlash(GoodFETSPI):
44 JEDECmanufacturers={0xFF: "MISSING",
52 JEDECdevices={0xFFFFFF: "MISSING",
58 0xC22017: "MX25L6405D",
59 0xC22016: "MX25L3205D",
60 0xC22015: "MX25L1605D",
61 0xC22014: "MX25L8005",
62 0xC22013: "MX25L4005",
66 JEDECsizes={0x17: 0x800000,
78 """Grab an SPI Flash ROM's JEDEC bytes."""
80 data=self.SPItrans(data);
82 self.JEDECmanufacturer=ord(data[1]);
83 self.JEDECtype=ord(data[2]);
84 self.JEDECcapacity=ord(data[3]);
85 self.JEDECsize=self.JEDECsizes.get(self.JEDECcapacity);
86 if self.JEDECsize==None:
88 self.JEDECdevice=(ord(data[1])<<16)+(ord(data[2])<<8)+ord(data[3]);
90 def SPIpeek(self,adr):
91 """Grab a byte from an SPI Flash ROM."""
98 return ord(self.data[4]);
99 def SPIpeekblock(self,adr):
100 """Grab a few block from an SPI Flash ROM. Block size is unknown"""
101 data=[(adr&0xFF0000)>>16,
105 self.writecmd(0x01,0x02,3,data);
108 def SPIpokebyte(self,adr,val):
109 self.SPIpokebytes(adr,[val]);
110 def SPIpokebytes(self,adr,data):
111 #Used to be 24 bits, BE, not 32 bits, LE.
112 adranddata=[adr&0xFF,
117 #print "%06x: poking %i bytes" % (adr,len(data));
118 self.writecmd(0x01,0x03,
119 len(adranddata),adranddata);
121 def SPIchiperase(self):
122 """Mass erase an SPI Flash ROM."""
123 self.writecmd(0x01,0x81);
124 def SPIwriteenable(self):
125 """SPI Flash Write Enable"""
129 def SPIjedecmanstr(self):
130 """Grab the JEDEC manufacturer string. Call after SPIjedec()."""
131 man=self.JEDECmanufacturers.get(self.JEDECmanufacturer)
136 def SPIjedecstr(self):
137 """Grab the JEDEC manufacturer string. Call after SPIjedec()."""
138 man=self.JEDECmanufacturers.get(self.JEDECmanufacturer);
141 device=self.JEDECdevices.get(self.JEDECdevice);
144 return "%s %s" % (man,device);