mtd: docg3: map erase and write functions
[firefly-linux-kernel-4.4.55.git] / drivers / mtd / devices / docg3.c
index bdcf5df982e829d99e45a97f263a68d478cf657a..1f949eac6351c32c1ce1d9b2de6ea44069550bac 100644 (file)
@@ -41,8 +41,6 @@
  *
  * As no specification is available from M-Systems/Sandisk, this drivers lacks
  * several functions available on the chip, as :
- *  - block erase
- *  - page write
  *  - IPL write
  *  - ECC fixing (lack of BCH algorith understanding)
  *  - powerdown / powerup
  *
  */
 
+/**
+ * struct docg3_oobinfo - DiskOnChip G3 OOB layout
+ * @eccbytes: 8 bytes are used (1 for Hamming ECC, 7 for BCH ECC)
+ * @eccpos: ecc positions (byte 7 is Hamming ECC, byte 8-14 are BCH ECC)
+ * @oobfree: free pageinfo bytes (byte 0 until byte 6, byte 15
+ * @oobavail: 8 available bytes remaining after ECC toll
+ */
+static struct nand_ecclayout docg3_oobinfo = {
+       .eccbytes = 8,
+       .eccpos = {7, 8, 9, 10, 11, 12, 13, 14},
+       .oobfree = {{0, 7}, {15, 1} },
+       .oobavail = 8,
+};
+
 static inline u8 doc_readb(struct docg3 *docg3, u16 reg)
 {
        u8 val = readb(docg3->base + reg);
@@ -82,7 +94,7 @@ static inline u16 doc_readw(struct docg3 *docg3, u16 reg)
 static inline void doc_writeb(struct docg3 *docg3, u8 val, u16 reg)
 {
        writeb(val, docg3->base + reg);
-       trace_docg3_io(1, 16, reg, val);
+       trace_docg3_io(1, 8, reg, val);
 }
 
 static inline void doc_writew(struct docg3 *docg3, u16 val, u16 reg)
@@ -143,7 +155,7 @@ static void doc_delay(struct docg3 *docg3, int nbNOPs)
 {
        int i;
 
-       doc_dbg("NOP x %d\n", nbNOPs);
+       doc_vdbg("NOP x %d\n", nbNOPs);
        for (i = 0; i < nbNOPs; i++)
                doc_writeb(docg3, 0, DOC_NOP);
 }
@@ -196,8 +208,8 @@ static int doc_reset_seq(struct docg3 *docg3)
 /**
  * doc_read_data_area - Read data from data area
  * @docg3: the device
- * @buf: the buffer to fill in
- * @len: the lenght to read
+ * @buf: the buffer to fill in (might be NULL is dummy reads)
+ * @len: the length to read
  * @first: first time read, DOC_READADDRESS should be set
  *
  * Reads bytes from flash data. Handles the single byte / even bytes reads.
@@ -218,8 +230,10 @@ static void doc_read_data_area(struct docg3 *docg3, void *buf, int len,
        dst16 = buf;
        for (i = 0; i < len4; i += 2) {
                data16 = doc_readw(docg3, DOC_IOSPACE_DATA);
-               *dst16 = data16;
-               dst16++;
+               if (dst16) {
+                       *dst16 = data16;
+                       dst16++;
+               }
        }
 
        if (cdr) {
@@ -229,12 +243,48 @@ static void doc_read_data_area(struct docg3 *docg3, void *buf, int len,
                dst8 = (u8 *)dst16;
                for (i = 0; i < cdr; i++) {
                        data8 = doc_readb(docg3, DOC_IOSPACE_DATA);
-                       *dst8 = data8;
-                       dst8++;
+                       if (dst8) {
+                               *dst8 = data8;
+                               dst8++;
+                       }
                }
        }
 }
 
+/**
+ * doc_write_data_area - Write data into data area
+ * @docg3: the device
+ * @buf: the buffer to get input bytes from
+ * @len: the length to write
+ *
+ * Writes bytes into flash data. Handles the single byte / even bytes writes.
+ */
+static void doc_write_data_area(struct docg3 *docg3, const void *buf, int len)
+{
+       int i, cdr, len4;
+       u16 *src16;
+       u8 *src8;
+
+       doc_dbg("doc_write_data_area(buf=%p, len=%d)\n", buf, len);
+       cdr = len & 0x3;
+       len4 = len - cdr;
+
+       doc_writew(docg3, DOC_IOSPACE_DATA, DOC_READADDRESS);
+       src16 = (u16 *)buf;
+       for (i = 0; i < len4; i += 2) {
+               doc_writew(docg3, *src16, DOC_IOSPACE_DATA);
+               src16++;
+       }
+
+       src8 = (u8 *)src16;
+       for (i = 0; i < cdr; i++) {
+               doc_writew(docg3, DOC_IOSPACE_DATA | DOC_READADDR_ONE_BYTE,
+                          DOC_READADDRESS);
+               doc_writeb(docg3, *src8, DOC_IOSPACE_DATA);
+               src8++;
+       }
+}
+
 /**
  * doc_set_data_mode - Sets the flash to reliable data mode
  * @docg3: the device
@@ -324,6 +374,37 @@ static int doc_set_extra_page_mode(struct docg3 *docg3)
                return 0;
 }
 
+/**
+ * doc_setup_addr_sector - Setup blocks/page/ofs address for one plane
+ * @docg3: the device
+ * @sector: the sector
+ */
+static void doc_setup_addr_sector(struct docg3 *docg3, int sector)
+{
+       doc_delay(docg3, 1);
+       doc_flash_address(docg3, sector & 0xff);
+       doc_flash_address(docg3, (sector >> 8) & 0xff);
+       doc_flash_address(docg3, (sector >> 16) & 0xff);
+       doc_delay(docg3, 1);
+}
+
+/**
+ * doc_setup_writeaddr_sector - Setup blocks/page/ofs address for one plane
+ * @docg3: the device
+ * @sector: the sector
+ * @ofs: the offset in the page, between 0 and (512 + 16 + 512)
+ */
+static void doc_setup_writeaddr_sector(struct docg3 *docg3, int sector, int ofs)
+{
+       ofs = ofs >> 2;
+       doc_delay(docg3, 1);
+       doc_flash_address(docg3, ofs & 0xff);
+       doc_flash_address(docg3, sector & 0xff);
+       doc_flash_address(docg3, (sector >> 8) & 0xff);
+       doc_flash_address(docg3, (sector >> 16) & 0xff);
+       doc_delay(docg3, 1);
+}
+
 /**
  * doc_seek - Set both flash planes to the specified block, page for reading
  * @docg3: the device
@@ -360,27 +441,73 @@ static int doc_read_seek(struct docg3 *docg3, int block0, int block1, int page,
        if (ret)
                goto out;
 
-       sector = (block0 << DOC_ADDR_BLOCK_SHIFT) + (page & DOC_ADDR_PAGE_MASK);
        doc_flash_sequence(docg3, DOC_SEQ_READ);
+       sector = (block0 << DOC_ADDR_BLOCK_SHIFT) + (page & DOC_ADDR_PAGE_MASK);
        doc_flash_command(docg3, DOC_CMD_PROG_BLOCK_ADDR);
-       doc_delay(docg3, 1);
-       doc_flash_address(docg3, sector & 0xff);
-       doc_flash_address(docg3, (sector >> 8) & 0xff);
-       doc_flash_address(docg3, (sector >> 16) & 0xff);
-       doc_delay(docg3, 1);
+       doc_setup_addr_sector(docg3, sector);
 
        sector = (block1 << DOC_ADDR_BLOCK_SHIFT) + (page & DOC_ADDR_PAGE_MASK);
        doc_flash_command(docg3, DOC_CMD_PROG_BLOCK_ADDR);
+       doc_setup_addr_sector(docg3, sector);
        doc_delay(docg3, 1);
-       doc_flash_address(docg3, sector & 0xff);
-       doc_flash_address(docg3, (sector >> 8) & 0xff);
-       doc_flash_address(docg3, (sector >> 16) & 0xff);
+
+out:
+       return ret;
+}
+
+/**
+ * doc_write_seek - Set both flash planes to the specified block, page for writing
+ * @docg3: the device
+ * @block0: the first plane block index
+ * @block1: the second plane block index
+ * @page: the page index within the block
+ * @ofs: offset in page to write
+ *
+ * Programs the flash even and odd planes to the specific block and page.
+ * Alternatively, programs the flash to the wear area of the specified page.
+ */
+static int doc_write_seek(struct docg3 *docg3, int block0, int block1, int page,
+                        int ofs)
+{
+       int ret = 0, sector;
+
+       doc_dbg("doc_write_seek(blocks=(%d,%d), page=%d, ofs=%d)\n",
+               block0, block1, page, ofs);
+
+       doc_set_reliable_mode(docg3);
+
+       if (ofs < 2 * DOC_LAYOUT_PAGE_SIZE) {
+               doc_flash_sequence(docg3, DOC_SEQ_SET_PLANE1);
+               doc_flash_command(docg3, DOC_CMD_READ_PLANE1);
+               doc_delay(docg3, 2);
+       } else {
+               doc_flash_sequence(docg3, DOC_SEQ_SET_PLANE2);
+               doc_flash_command(docg3, DOC_CMD_READ_PLANE2);
+               doc_delay(docg3, 2);
+       }
+
+       doc_flash_sequence(docg3, DOC_SEQ_PAGE_SETUP);
+       doc_flash_command(docg3, DOC_CMD_PROG_CYCLE1);
+
+       sector = (block0 << DOC_ADDR_BLOCK_SHIFT) + (page & DOC_ADDR_PAGE_MASK);
+       doc_setup_writeaddr_sector(docg3, sector, ofs);
+
+       doc_flash_command(docg3, DOC_CMD_PROG_CYCLE3);
        doc_delay(docg3, 2);
+       ret = doc_wait_ready(docg3);
+       if (ret)
+               goto out;
+
+       doc_flash_command(docg3, DOC_CMD_PROG_CYCLE1);
+       sector = (block1 << DOC_ADDR_BLOCK_SHIFT) + (page & DOC_ADDR_PAGE_MASK);
+       doc_setup_writeaddr_sector(docg3, sector, ofs);
+       doc_delay(docg3, 1);
 
 out:
        return ret;
 }
 
+
 /**
  * doc_read_page_ecc_init - Initialize hardware ECC engine
  * @docg3: the device
@@ -402,6 +529,58 @@ static int doc_read_page_ecc_init(struct docg3 *docg3, int len)
        return doc_wait_ready(docg3);
 }
 
+/**
+ * doc_write_page_ecc_init - Initialize hardware BCH ECC engine
+ * @docg3: the device
+ * @len: the number of bytes covered by the ECC (BCH covered)
+ *
+ * The function does initialize the hardware ECC engine to compute the Hamming
+ * ECC (on 1 byte) and the BCH Syndroms (on 7 bytes).
+ *
+ * Return 0 if succeeded, -EIO on error
+ */
+static int doc_write_page_ecc_init(struct docg3 *docg3, int len)
+{
+       doc_writew(docg3, !DOC_ECCCONF0_READ_MODE
+                  | DOC_ECCCONF0_BCH_ENABLE | DOC_ECCCONF0_HAMMING_ENABLE
+                  | (len & DOC_ECCCONF0_DATA_BYTES_MASK),
+                  DOC_ECCCONF0);
+       doc_delay(docg3, 4);
+       doc_register_readb(docg3, DOC_FLASHCONTROL);
+       return doc_wait_ready(docg3);
+}
+
+/**
+ * doc_ecc_disable - Disable Hamming and BCH ECC hardware calculator
+ * @docg3: the device
+ *
+ * Disables the hardware ECC generator and checker, for unchecked reads (as when
+ * reading OOB only or write status byte).
+ */
+static void doc_ecc_disable(struct docg3 *docg3)
+{
+       doc_writew(docg3, DOC_ECCCONF0_READ_MODE, DOC_ECCCONF0);
+       doc_delay(docg3, 4);
+}
+
+/**
+ * doc_hamming_ecc_init - Initialize hardware Hamming ECC engine
+ * @docg3: the device
+ * @nb_bytes: the number of bytes covered by the ECC (Hamming covered)
+ *
+ * This function programs the ECC hardware to compute the hamming code on the
+ * last provided N bytes to the hardware generator.
+ */
+static void doc_hamming_ecc_init(struct docg3 *docg3, int nb_bytes)
+{
+       u8 ecc_conf1;
+
+       ecc_conf1 = doc_register_readb(docg3, DOC_ECCCONF1);
+       ecc_conf1 &= ~DOC_ECCCONF1_HAMMING_BITS_MASK;
+       ecc_conf1 |= (nb_bytes & DOC_ECCCONF1_HAMMING_BITS_MASK);
+       doc_writeb(docg3, ecc_conf1, DOC_ECCCONF1);
+}
+
 /**
  * doc_read_page_prepare - Prepares reading data from a flash page
  * @docg3: the device
@@ -487,12 +666,26 @@ static int doc_read_page_getbytes(struct docg3 *docg3, int len, u_char *buf,
        return len;
 }
 
+/**
+ * doc_write_page_putbytes - Writes bytes into a prepared page
+ * @docg3: the device
+ * @len: the number of bytes to be written
+ * @buf: the buffer of input bytes
+ *
+ */
+static void doc_write_page_putbytes(struct docg3 *docg3, int len,
+                                   const u_char *buf)
+{
+       doc_write_data_area(docg3, buf, len);
+       doc_delay(docg3, 2);
+}
+
 /**
  * doc_get_hw_bch_syndroms - Get hardware calculated BCH syndroms
  * @docg3: the device
  * @syns:  the array of 7 integers where the syndroms will be stored
  */
-static void doc_get_hw_bch_syndroms(struct docg3 *docg3, int *syns)
+static void doc_get_hw_bch_syndroms(struct docg3 *docg3, u8 *syns)
 {
        int i;
 
@@ -500,6 +693,16 @@ static void doc_get_hw_bch_syndroms(struct docg3 *docg3, int *syns)
                syns[i] = doc_register_readb(docg3, DOC_BCH_SYNDROM(i));
 }
 
+/**
+ * doc_page_finish - Ends reading/writing of a flash page
+ * @docg3: the device
+ */
+static void doc_page_finish(struct docg3 *docg3)
+{
+       doc_writeb(docg3, 0, DOC_DATAEND);
+       doc_delay(docg3, 2);
+}
+
 /**
  * doc_read_page_finish - Ends reading of a flash page
  * @docg3: the device
@@ -510,8 +713,7 @@ static void doc_get_hw_bch_syndroms(struct docg3 *docg3, int *syns)
  */
 static void doc_read_page_finish(struct docg3 *docg3)
 {
-       doc_writeb(docg3, 0, DOC_DATAEND);
-       doc_delay(docg3, 2);
+       doc_page_finish(docg3);
        doc_set_device_id(docg3, 0);
 }
 
@@ -542,96 +744,109 @@ static void calc_block_sector(loff_t from, int *block0, int *block1, int *page,
 }
 
 /**
- * doc_read - Read bytes from flash
+ * doc_read_oob - Read out of band bytes from flash
  * @mtd: the device
  * @from: the offset from first block and first page, in bytes, aligned on page
  *        size
- * @len: the number of bytes to read (must be a multiple of 4)
- * @retlen: the number of bytes actually read
- * @buf: the filled in buffer
+ * @ops: the mtd oob structure
  *
- * Reads flash memory pages. This function does not read the OOB chunk, but only
- * the page data.
+ * Reads flash memory OOB area of pages.
  *
  * Returns 0 if read successfull, of -EIO, -EINVAL if an error occured
  */
-static int doc_read(struct mtd_info *mtd, loff_t from, size_t len,
-            size_t *retlen, u_char *buf)
+static int doc_read_oob(struct mtd_info *mtd, loff_t from,
+                       struct mtd_oob_ops *ops)
 {
        struct docg3 *docg3 = mtd->priv;
-       int block0, block1, page, readlen, ret, ofs = 0;
-       int syn[DOC_ECC_BCH_SIZE], eccconf1;
-       u8 oob[DOC_LAYOUT_OOB_SIZE];
+       int block0, block1, page, ret, ofs = 0;
+       u8 *oobbuf = ops->oobbuf;
+       u8 *buf = ops->datbuf;
+       size_t len, ooblen, nbdata, nboob;
+       u8 calc_ecc[DOC_ECC_BCH_SIZE], eccconf1;
+
+       if (buf)
+               len = ops->len;
+       else
+               len = 0;
+       if (oobbuf)
+               ooblen = ops->ooblen;
+       else
+               ooblen = 0;
+
+       if (oobbuf && ops->mode == MTD_OPS_PLACE_OOB)
+               oobbuf += ops->ooboffs;
+
+       doc_dbg("doc_read_oob(from=%lld, mode=%d, data=(%p:%zu), oob=(%p:%zu))\n",
+               from, ops->mode, buf, len, oobbuf, ooblen);
+       if ((len % DOC_LAYOUT_PAGE_SIZE) || (ooblen % DOC_LAYOUT_OOB_SIZE) ||
+           (from % DOC_LAYOUT_PAGE_SIZE))
+               return -EINVAL;
 
        ret = -EINVAL;
-       doc_dbg("doc_read(from=%lld, len=%zu, buf=%p)\n", from, len, buf);
-       if (from % DOC_LAYOUT_PAGE_SIZE)
-               goto err;
-       if (len % 4)
-               goto err;
-       calc_block_sector(from, &block0, &block1, &page, &ofs);
+       calc_block_sector(from + len, &block0, &block1, &page, &ofs);
        if (block1 > docg3->max_block)
                goto err;
 
-       *retlen = 0;
+       ops->oobretlen = 0;
+       ops->retlen = 0;
        ret = 0;
-       readlen = min_t(size_t, len, (size_t)DOC_LAYOUT_PAGE_SIZE);
-       while (!ret && len > 0) {
-               readlen = min_t(size_t, len, (size_t)DOC_LAYOUT_PAGE_SIZE);
+       while (!ret && (len > 0 || ooblen > 0)) {
+               calc_block_sector(from, &block0, &block1, &page, &ofs);
+               nbdata = min_t(size_t, len, (size_t)DOC_LAYOUT_PAGE_SIZE);
+               nboob = min_t(size_t, ooblen, (size_t)DOC_LAYOUT_OOB_SIZE);
                ret = doc_read_page_prepare(docg3, block0, block1, page, ofs);
                if (ret < 0)
                        goto err;
                ret = doc_read_page_ecc_init(docg3, DOC_ECC_BCH_COVERED_BYTES);
                if (ret < 0)
                        goto err_in_read;
-               ret = doc_read_page_getbytes(docg3, readlen, buf, 1);
-               if (ret < readlen)
+               ret = doc_read_page_getbytes(docg3, nbdata, buf, 1);
+               if (ret < nbdata)
                        goto err_in_read;
-               ret = doc_read_page_getbytes(docg3, DOC_LAYOUT_OOB_SIZE,
-                                            oob, 0);
-               if (ret < DOC_LAYOUT_OOB_SIZE)
+               doc_read_page_getbytes(docg3, DOC_LAYOUT_PAGE_SIZE - nbdata,
+                                      NULL, 0);
+               ret = doc_read_page_getbytes(docg3, nboob, oobbuf, 0);
+               if (ret < nboob)
                        goto err_in_read;
+               doc_read_page_getbytes(docg3, DOC_LAYOUT_OOB_SIZE - nboob,
+                                      NULL, 0);
 
-               *retlen += readlen;
-               buf += readlen;
-               len -= readlen;
-
-               ofs ^= DOC_LAYOUT_PAGE_OOB_SIZE;
-               if (ofs == 0)
-                       page += 2;
-               if (page > DOC_ADDR_PAGE_MASK) {
-                       page = 0;
-                       block0 += 2;
-                       block1 += 2;
-               }
-
-               /*
-                * There should be a BCH bitstream fixing algorithm here ...
-                * By now, a page read failure is triggered by BCH error
-                */
-               doc_get_hw_bch_syndroms(docg3, syn);
+               doc_get_hw_bch_syndroms(docg3, calc_ecc);
                eccconf1 = doc_register_readb(docg3, DOC_ECCCONF1);
 
-               doc_dbg("OOB - INFO: %02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
-                        oob[0], oob[1], oob[2], oob[3], oob[4],
-                        oob[5], oob[6]);
-               doc_dbg("OOB - HAMMING: %02x\n", oob[7]);
-               doc_dbg("OOB - BCH_ECC: %02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
-                        oob[8], oob[9], oob[10], oob[11], oob[12],
-                        oob[13], oob[14]);
-               doc_dbg("OOB - UNUSED: %02x\n", oob[15]);
+               if (nboob >= DOC_LAYOUT_OOB_SIZE) {
+                       doc_dbg("OOB - INFO: %02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
+                               oobbuf[0], oobbuf[1], oobbuf[2], oobbuf[3],
+                               oobbuf[4], oobbuf[5], oobbuf[6]);
+                       doc_dbg("OOB - HAMMING: %02x\n", oobbuf[7]);
+                       doc_dbg("OOB - BCH_ECC: %02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
+                               oobbuf[8], oobbuf[9], oobbuf[10], oobbuf[11],
+                               oobbuf[12], oobbuf[13], oobbuf[14]);
+                       doc_dbg("OOB - UNUSED: %02x\n", oobbuf[15]);
+               }
                doc_dbg("ECC checks: ECCConf1=%x\n", eccconf1);
-               doc_dbg("ECC BCH syndrom: %02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
-                       syn[0], syn[1], syn[2], syn[3], syn[4], syn[5], syn[6]);
+               doc_dbg("ECC CALC_ECC: %02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
+                       calc_ecc[0], calc_ecc[1], calc_ecc[2],
+                       calc_ecc[3], calc_ecc[4], calc_ecc[5],
+                       calc_ecc[6]);
 
                ret = -EBADMSG;
                if (block0 >= DOC_LAYOUT_BLOCK_FIRST_DATA) {
-                       if (eccconf1 & DOC_ECCCONF1_BCH_SYNDROM_ERR)
+                       if ((eccconf1 & DOC_ECCCONF1_BCH_SYNDROM_ERR) &&
+                           (eccconf1 & DOC_ECCCONF1_PAGE_IS_WRITTEN))
                                goto err_in_read;
                        if (is_prot_seq_error(docg3))
                                goto err_in_read;
                }
+
                doc_read_page_finish(docg3);
+               ops->retlen += nbdata;
+               ops->oobretlen += nboob;
+               buf += nbdata;
+               oobbuf += nboob;
+               len -= nbdata;
+               ooblen -= nboob;
+               from += DOC_LAYOUT_PAGE_SIZE;
        }
 
        return 0;
@@ -642,54 +857,33 @@ err:
 }
 
 /**
- * doc_read_oob - Read out of band bytes from flash
+ * doc_read - Read bytes from flash
  * @mtd: the device
  * @from: the offset from first block and first page, in bytes, aligned on page
  *        size
- * @ops: the mtd oob structure
+ * @len: the number of bytes to read (must be a multiple of 4)
+ * @retlen: the number of bytes actually read
+ * @buf: the filled in buffer
  *
- * Reads flash memory OOB area of pages.
+ * Reads flash memory pages. This function does not read the OOB chunk, but only
+ * the page data.
  *
  * Returns 0 if read successfull, of -EIO, -EINVAL if an error occured
  */
-static int doc_read_oob(struct mtd_info *mtd, loff_t from,
-                       struct mtd_oob_ops *ops)
+static int doc_read(struct mtd_info *mtd, loff_t from, size_t len,
+            size_t *retlen, u_char *buf)
 {
-       struct docg3 *docg3 = mtd->priv;
-       int block0, block1, page, ofs, ret;
-       u8 *buf = ops->oobbuf;
-       size_t len = ops->ooblen;
-
-       doc_dbg("doc_read_oob(from=%lld, buf=%p, len=%zu)\n", from, buf, len);
-       if (len != DOC_LAYOUT_OOB_SIZE)
-               return -EINVAL;
-
-       switch (ops->mode) {
-       case MTD_OPS_PLACE_OOB:
-               buf += ops->ooboffs;
-               break;
-       default:
-               break;
-       }
+       struct mtd_oob_ops ops;
+       size_t ret;
 
-       calc_block_sector(from, &block0, &block1, &page, &ofs);
-       if (block1 > docg3->max_block)
-               return -EINVAL;
+       memset(&ops, 0, sizeof(ops));
+       ops.datbuf = buf;
+       ops.len = len;
+       ops.mode = MTD_OPS_AUTO_OOB;
 
-       ret = doc_read_page_prepare(docg3, block0, block1, page,
-                                   ofs + DOC_LAYOUT_PAGE_SIZE);
-       if (!ret)
-               ret = doc_read_page_ecc_init(docg3, DOC_LAYOUT_OOB_SIZE);
-       if (!ret)
-               ret = doc_read_page_getbytes(docg3, DOC_LAYOUT_OOB_SIZE,
-                                            buf, 1);
-       doc_read_page_finish(docg3);
-
-       if (ret > 0)
-               ops->oobretlen = ret;
-       else
-               ops->oobretlen = 0;
-       return (ret > 0) ? 0 : ret;
+       ret = doc_read_oob(mtd, from, &ops);
+       *retlen = ops.retlen;
+       return ret;
 }
 
 static int doc_reload_bbt(struct docg3 *docg3)
@@ -781,6 +975,438 @@ static int doc_get_erase_count(struct docg3 *docg3, loff_t from)
        return max(plane1_erase_count, plane2_erase_count);
 }
 
+/**
+ * doc_get_op_status - get erase/write operation status
+ * @docg3: the device
+ *
+ * Queries the status from the chip, and returns it
+ *
+ * Returns the status (bits DOC_PLANES_STATUS_*)
+ */
+static int doc_get_op_status(struct docg3 *docg3)
+{
+       u8 status;
+
+       doc_flash_sequence(docg3, DOC_SEQ_PLANES_STATUS);
+       doc_flash_command(docg3, DOC_CMD_PLANES_STATUS);
+       doc_delay(docg3, 5);
+
+       doc_ecc_disable(docg3);
+       doc_read_data_area(docg3, &status, 1, 1);
+       return status;
+}
+
+/**
+ * doc_write_erase_wait_status - wait for write or erase completion
+ * @docg3: the device
+ *
+ * Wait for the chip to be ready again after erase or write operation, and check
+ * erase/write status.
+ *
+ * Returns 0 if erase successfull, -EIO if erase/write issue, -ETIMEOUT if
+ * timeout
+ */
+static int doc_write_erase_wait_status(struct docg3 *docg3)
+{
+       int status, ret = 0;
+
+       if (!doc_is_ready(docg3))
+               usleep_range(3000, 3000);
+       if (!doc_is_ready(docg3)) {
+               doc_dbg("Timeout reached and the chip is still not ready\n");
+               ret = -EAGAIN;
+               goto out;
+       }
+
+       status = doc_get_op_status(docg3);
+       if (status & DOC_PLANES_STATUS_FAIL) {
+               doc_dbg("Erase/Write failed on (a) plane(s), status = %x\n",
+                       status);
+               ret = -EIO;
+       }
+
+out:
+       doc_page_finish(docg3);
+       return ret;
+}
+
+/**
+ * doc_erase_block - Erase a couple of blocks
+ * @docg3: the device
+ * @block0: the first block to erase (leftmost plane)
+ * @block1: the second block to erase (rightmost plane)
+ *
+ * Erase both blocks, and return operation status
+ *
+ * Returns 0 if erase successful, -EIO if erase issue, -ETIMEOUT if chip not
+ * ready for too long
+ */
+static int doc_erase_block(struct docg3 *docg3, int block0, int block1)
+{
+       int ret, sector;
+
+       doc_dbg("doc_erase_block(blocks=(%d,%d))\n", block0, block1);
+       ret = doc_reset_seq(docg3);
+       if (ret)
+               return -EIO;
+
+       doc_set_reliable_mode(docg3);
+       doc_flash_sequence(docg3, DOC_SEQ_ERASE);
+
+       sector = block0 << DOC_ADDR_BLOCK_SHIFT;
+       doc_flash_command(docg3, DOC_CMD_PROG_BLOCK_ADDR);
+       doc_setup_addr_sector(docg3, sector);
+       sector = block1 << DOC_ADDR_BLOCK_SHIFT;
+       doc_flash_command(docg3, DOC_CMD_PROG_BLOCK_ADDR);
+       doc_setup_addr_sector(docg3, sector);
+       doc_delay(docg3, 1);
+
+       doc_flash_command(docg3, DOC_CMD_ERASECYCLE2);
+       doc_delay(docg3, 2);
+
+       if (is_prot_seq_error(docg3)) {
+               doc_err("Erase blocks %d,%d error\n", block0, block1);
+               return -EIO;
+       }
+
+       return doc_write_erase_wait_status(docg3);
+}
+
+/**
+ * doc_erase - Erase a portion of the chip
+ * @mtd: the device
+ * @info: the erase info
+ *
+ * Erase a bunch of contiguous blocks, by pairs, as a "mtd" page of 1024 is
+ * split into 2 pages of 512 bytes on 2 contiguous blocks.
+ *
+ * Returns 0 if erase successful, -EINVAL if adressing error, -EIO if erase
+ * issue
+ */
+static int doc_erase(struct mtd_info *mtd, struct erase_info *info)
+{
+       struct docg3 *docg3 = mtd->priv;
+       uint64_t len;
+       int block0, block1, page, ret, ofs = 0;
+
+       doc_dbg("doc_erase(from=%lld, len=%lld\n", info->addr, info->len);
+       doc_set_device_id(docg3, docg3->device_id);
+
+       info->state = MTD_ERASE_PENDING;
+       calc_block_sector(info->addr + info->len,
+                         &block0, &block1, &page, &ofs);
+       ret = -EINVAL;
+       if (block1 > docg3->max_block || page || ofs)
+               goto reset_err;
+
+       ret = 0;
+       calc_block_sector(info->addr, &block0, &block1, &page, &ofs);
+       doc_set_reliable_mode(docg3);
+       for (len = info->len; !ret && len > 0; len -= mtd->erasesize) {
+               info->state = MTD_ERASING;
+               ret = doc_erase_block(docg3, block0, block1);
+               block0 += 2;
+               block1 += 2;
+       }
+
+       if (ret)
+               goto reset_err;
+
+       info->state = MTD_ERASE_DONE;
+       return 0;
+
+reset_err:
+       info->state = MTD_ERASE_FAILED;
+       return ret;
+}
+
+/**
+ * doc_write_page - Write a single page to the chip
+ * @docg3: the device
+ * @to: the offset from first block and first page, in bytes, aligned on page
+ *      size
+ * @buf: buffer to get bytes from
+ * @oob: buffer to get out of band bytes from (can be NULL if no OOB should be
+ *       written)
+ * @autoecc: if 0, all 16 bytes from OOB are taken, regardless of HW Hamming or
+ *           BCH computations. If 1, only bytes 0-7 and byte 15 are taken,
+ *           remaining ones are filled with hardware Hamming and BCH
+ *           computations. Its value is not meaningfull is oob == NULL.
+ *
+ * Write one full page (ie. 1 page split on two planes), of 512 bytes, with the
+ * OOB data. The OOB ECC is automatically computed by the hardware Hamming and
+ * BCH generator if autoecc is not null.
+ *
+ * Returns 0 if write successful, -EIO if write error, -EAGAIN if timeout
+ */
+static int doc_write_page(struct docg3 *docg3, loff_t to, const u_char *buf,
+                         const u_char *oob, int autoecc)
+{
+       int block0, block1, page, ret, ofs = 0;
+       u8 syn[DOC_ECC_BCH_SIZE], hamming;
+
+       doc_dbg("doc_write_page(to=%lld)\n", to);
+       calc_block_sector(to, &block0, &block1, &page, &ofs);
+
+       doc_set_device_id(docg3, docg3->device_id);
+       ret = doc_reset_seq(docg3);
+       if (ret)
+               goto err;
+
+       /* Program the flash address block and page */
+       ret = doc_write_seek(docg3, block0, block1, page, ofs);
+       if (ret)
+               goto err;
+
+       doc_write_page_ecc_init(docg3, DOC_ECC_BCH_COVERED_BYTES);
+       doc_delay(docg3, 2);
+       doc_write_page_putbytes(docg3, DOC_LAYOUT_PAGE_SIZE, buf);
+
+       if (oob && autoecc) {
+               doc_write_page_putbytes(docg3, DOC_LAYOUT_OOB_PAGEINFO_SZ, oob);
+               doc_delay(docg3, 2);
+               oob += DOC_LAYOUT_OOB_UNUSED_OFS;
+
+               hamming = doc_register_readb(docg3, DOC_HAMMINGPARITY);
+               doc_delay(docg3, 2);
+               doc_write_page_putbytes(docg3, DOC_LAYOUT_OOB_HAMMING_SZ,
+                                       &hamming);
+               doc_delay(docg3, 2);
+
+               doc_get_hw_bch_syndroms(docg3, syn);
+               doc_write_page_putbytes(docg3, DOC_LAYOUT_OOB_BCH_SZ, syn);
+               doc_delay(docg3, 2);
+
+               doc_write_page_putbytes(docg3, DOC_LAYOUT_OOB_UNUSED_SZ, oob);
+       }
+       if (oob && !autoecc)
+               doc_write_page_putbytes(docg3, DOC_LAYOUT_OOB_SIZE, oob);
+
+       doc_delay(docg3, 2);
+       doc_page_finish(docg3);
+       doc_delay(docg3, 2);
+       doc_flash_command(docg3, DOC_CMD_PROG_CYCLE2);
+       doc_delay(docg3, 2);
+
+       /*
+        * The wait status will perform another doc_page_finish() call, but that
+        * seems to please the docg3, so leave it.
+        */
+       ret = doc_write_erase_wait_status(docg3);
+       return ret;
+err:
+       doc_read_page_finish(docg3);
+       return ret;
+}
+
+/**
+ * doc_guess_autoecc - Guess autoecc mode from mbd_oob_ops
+ * @ops: the oob operations
+ *
+ * Returns 0 or 1 if success, -EINVAL if invalid oob mode
+ */
+static int doc_guess_autoecc(struct mtd_oob_ops *ops)
+{
+       int autoecc;
+
+       switch (ops->mode) {
+       case MTD_OPS_PLACE_OOB:
+       case MTD_OPS_AUTO_OOB:
+               autoecc = 1;
+               break;
+       case MTD_OPS_RAW:
+               autoecc = 0;
+               break;
+       default:
+               autoecc = -EINVAL;
+       }
+       return autoecc;
+}
+
+/**
+ * doc_fill_autooob - Fill a 16 bytes OOB from 8 non-ECC bytes
+ * @dst: the target 16 bytes OOB buffer
+ * @oobsrc: the source 8 bytes non-ECC OOB buffer
+ *
+ */
+static void doc_fill_autooob(u8 *dst, u8 *oobsrc)
+{
+       memcpy(dst, oobsrc, DOC_LAYOUT_OOB_PAGEINFO_SZ);
+       dst[DOC_LAYOUT_OOB_UNUSED_OFS] = oobsrc[DOC_LAYOUT_OOB_PAGEINFO_SZ];
+}
+
+/**
+ * doc_backup_oob - Backup OOB into docg3 structure
+ * @docg3: the device
+ * @to: the page offset in the chip
+ * @ops: the OOB size and buffer
+ *
+ * As the docg3 should write a page with its OOB in one pass, and some userland
+ * applications do write_oob() to setup the OOB and then write(), store the OOB
+ * into a temporary storage. This is very dangerous, as 2 concurrent
+ * applications could store an OOB, and then write their pages (which will
+ * result into one having its OOB corrupted).
+ *
+ * The only reliable way would be for userland to call doc_write_oob() with both
+ * the page data _and_ the OOB area.
+ *
+ * Returns 0 if success, -EINVAL if ops content invalid
+ */
+static int doc_backup_oob(struct docg3 *docg3, loff_t to,
+                         struct mtd_oob_ops *ops)
+{
+       int ooblen = ops->ooblen, autoecc;
+
+       if (ooblen != DOC_LAYOUT_OOB_SIZE)
+               return -EINVAL;
+       autoecc = doc_guess_autoecc(ops);
+       if (autoecc < 0)
+               return autoecc;
+
+       docg3->oob_write_ofs = to;
+       docg3->oob_autoecc = autoecc;
+       if (ops->mode == MTD_OPS_AUTO_OOB) {
+               doc_fill_autooob(docg3->oob_write_buf, ops->oobbuf);
+               ops->oobretlen = 8;
+       } else {
+               memcpy(docg3->oob_write_buf, ops->oobbuf, DOC_LAYOUT_OOB_SIZE);
+               ops->oobretlen = DOC_LAYOUT_OOB_SIZE;
+       }
+       return 0;
+}
+
+/**
+ * doc_write_oob - Write out of band bytes to flash
+ * @mtd: the device
+ * @ofs: the offset from first block and first page, in bytes, aligned on page
+ *       size
+ * @ops: the mtd oob structure
+ *
+ * Either write OOB data into a temporary buffer, for the subsequent write
+ * page. The provided OOB should be 16 bytes long. If a data buffer is provided
+ * as well, issue the page write.
+ * Or provide data without OOB, and then a all zeroed OOB will be used (ECC will
+ * still be filled in if asked for).
+ *
+ * Returns 0 is successfull, EINVAL if length is not 14 bytes
+ */
+static int doc_write_oob(struct mtd_info *mtd, loff_t ofs,
+                        struct mtd_oob_ops *ops)
+{
+       struct docg3 *docg3 = mtd->priv;
+       int block0, block1, page, ret, pofs = 0, autoecc, oobdelta;
+       u8 *oobbuf = ops->oobbuf;
+       u8 *buf = ops->datbuf;
+       size_t len, ooblen;
+       u8 oob[DOC_LAYOUT_OOB_SIZE];
+
+       if (buf)
+               len = ops->len;
+       else
+               len = 0;
+       if (oobbuf)
+               ooblen = ops->ooblen;
+       else
+               ooblen = 0;
+
+       if (oobbuf && ops->mode == MTD_OPS_PLACE_OOB)
+               oobbuf += ops->ooboffs;
+
+       doc_dbg("doc_write_oob(from=%lld, mode=%d, data=(%p:%zu), oob=(%p:%zu))\n",
+               ofs, ops->mode, buf, len, oobbuf, ooblen);
+       switch (ops->mode) {
+       case MTD_OPS_PLACE_OOB:
+       case MTD_OPS_RAW:
+               oobdelta = mtd->oobsize;
+               break;
+       case MTD_OPS_AUTO_OOB:
+               oobdelta = mtd->ecclayout->oobavail;
+               break;
+       default:
+               oobdelta = 0;
+       }
+       if ((len % DOC_LAYOUT_PAGE_SIZE) || (ooblen % oobdelta) ||
+           (ofs % DOC_LAYOUT_PAGE_SIZE))
+               return -EINVAL;
+       if (len && ooblen &&
+           (len / DOC_LAYOUT_PAGE_SIZE) != (ooblen / oobdelta))
+               return -EINVAL;
+
+       ret = -EINVAL;
+       calc_block_sector(ofs + len, &block0, &block1, &page, &pofs);
+       if (block1 > docg3->max_block)
+               goto err;
+
+       ops->oobretlen = 0;
+       ops->retlen = 0;
+       ret = 0;
+       if (len == 0 && ooblen == 0)
+               return -EINVAL;
+       if (len == 0 && ooblen > 0)
+               return doc_backup_oob(docg3, ofs, ops);
+
+       autoecc = doc_guess_autoecc(ops);
+       if (autoecc < 0)
+               return autoecc;
+
+       while (!ret && len > 0) {
+               memset(oob, 0, sizeof(oob));
+               if (ofs == docg3->oob_write_ofs)
+                       memcpy(oob, docg3->oob_write_buf, DOC_LAYOUT_OOB_SIZE);
+               else if (ooblen > 0 && ops->mode == MTD_OPS_AUTO_OOB)
+                       doc_fill_autooob(oob, oobbuf);
+               else if (ooblen > 0)
+                       memcpy(oob, oobbuf, DOC_LAYOUT_OOB_SIZE);
+               ret = doc_write_page(docg3, ofs, buf, oob, autoecc);
+
+               ofs += DOC_LAYOUT_PAGE_SIZE;
+               len -= DOC_LAYOUT_PAGE_SIZE;
+               buf += DOC_LAYOUT_PAGE_SIZE;
+               if (ooblen) {
+                       oobbuf += oobdelta;
+                       ooblen -= oobdelta;
+                       ops->oobretlen += oobdelta;
+               }
+               ops->retlen += DOC_LAYOUT_PAGE_SIZE;
+       }
+err:
+       doc_set_device_id(docg3, 0);
+       return ret;
+}
+
+/**
+ * doc_write - Write a buffer to the chip
+ * @mtd: the device
+ * @to: the offset from first block and first page, in bytes, aligned on page
+ *      size
+ * @len: the number of bytes to write (must be a full page size, ie. 512)
+ * @retlen: the number of bytes actually written (0 or 512)
+ * @buf: the buffer to get bytes from
+ *
+ * Writes data to the chip.
+ *
+ * Returns 0 if write successful, -EIO if write error
+ */
+static int doc_write(struct mtd_info *mtd, loff_t to, size_t len,
+                    size_t *retlen, const u_char *buf)
+{
+       struct docg3 *docg3 = mtd->priv;
+       int ret;
+       struct mtd_oob_ops ops;
+
+       doc_dbg("doc_write(to=%lld, len=%zu)\n", to, len);
+       ops.datbuf = (char *)buf;
+       ops.len = len;
+       ops.mode = MTD_OPS_PLACE_OOB;
+       ops.oobbuf = NULL;
+       ops.ooblen = 0;
+       ops.ooboffs = 0;
+
+       ret = doc_write_oob(mtd, to, &ops);
+       *retlen = ops.retlen;
+       return ret;
+}
+
 /*
  * Debug sysfs entries
  */
@@ -852,13 +1478,15 @@ static int dbg_protection_show(struct seq_file *s, void *p)
 {
        struct docg3 *docg3 = (struct docg3 *)s->private;
        int pos = 0;
-       int protect = doc_register_readb(docg3, DOC_PROTECTION);
-       int dps0 = doc_register_readb(docg3, DOC_DPS0_STATUS);
-       int dps0_low = doc_register_readb(docg3, DOC_DPS0_ADDRLOW);
-       int dps0_high = doc_register_readb(docg3, DOC_DPS0_ADDRHIGH);
-       int dps1 = doc_register_readb(docg3, DOC_DPS1_STATUS);
-       int dps1_low = doc_register_readb(docg3, DOC_DPS1_ADDRLOW);
-       int dps1_high = doc_register_readb(docg3, DOC_DPS1_ADDRHIGH);
+       int protect, dps0, dps0_low, dps0_high, dps1, dps1_low, dps1_high;
+
+       protect = doc_register_readb(docg3, DOC_PROTECTION);
+       dps0 = doc_register_readb(docg3, DOC_DPS0_STATUS);
+       dps0_low = doc_register_readw(docg3, DOC_DPS0_ADDRLOW);
+       dps0_high = doc_register_readw(docg3, DOC_DPS0_ADDRHIGH);
+       dps1 = doc_register_readb(docg3, DOC_DPS1_STATUS);
+       dps1_low = doc_register_readw(docg3, DOC_DPS1_ADDRLOW);
+       dps1_high = doc_register_readw(docg3, DOC_DPS1_ADDRHIGH);
 
        pos += seq_printf(s, "Protection = 0x%02x (",
                         protect);
@@ -950,49 +1578,49 @@ static void __init doc_set_driver_info(int chip_id, struct mtd_info *mtd)
 
        switch (chip_id) {
        case DOC_CHIPID_G3:
-               mtd->name = "DiskOnChip G3";
+               mtd->name = kasprintf(GFP_KERNEL, "DiskOnChip G3 floor %d",
+                                     docg3->device_id);
                docg3->max_block = 2047;
                break;
        }
        mtd->type = MTD_NANDFLASH;
-       /*
-        * Once write methods are added, the correct flags will be set.
-        * mtd->flags = MTD_CAP_NANDFLASH;
-        */
-       mtd->flags = MTD_CAP_ROM;
+       mtd->flags = MTD_CAP_NANDFLASH;
        mtd->size = (docg3->max_block + 1) * DOC_LAYOUT_BLOCK_SIZE;
        mtd->erasesize = DOC_LAYOUT_BLOCK_SIZE * DOC_LAYOUT_NBPLANES;
        mtd->writesize = DOC_LAYOUT_PAGE_SIZE;
        mtd->oobsize = DOC_LAYOUT_OOB_SIZE;
        mtd->owner = THIS_MODULE;
-       mtd->erase = NULL;
+       mtd->erase = doc_erase;
        mtd->point = NULL;
        mtd->unpoint = NULL;
        mtd->read = doc_read;
-       mtd->write = NULL;
+       mtd->write = doc_write;
        mtd->read_oob = doc_read_oob;
-       mtd->write_oob = NULL;
+       mtd->write_oob = doc_write_oob;
        mtd->sync = NULL;
        mtd->block_isbad = doc_block_isbad;
+       mtd->ecclayout = &docg3_oobinfo;
 }
 
 /**
- * doc_probe - Probe the IO space for a DiskOnChip G3 chip
- * @pdev: platform device
+ * doc_probe_device - Check if a device is available
+ * @base: the io space where the device is probed
+ * @floor: the floor of the probed device
+ * @dev: the device
  *
- * Probes for a G3 chip at the specified IO space in the platform data
- * ressources.
+ * Checks whether a device at the specified IO range, and floor is available.
  *
- * Returns 0 on success, -ENOMEM, -ENXIO on error
+ * Returns a mtd_info struct if there is a device, ENODEV if none found, ENOMEM
+ * if a memory allocation failed. If floor 0 is checked, a reset of the ASIC is
+ * launched.
  */
-static int __init docg3_probe(struct platform_device *pdev)
+static struct mtd_info *doc_probe_device(void __iomem *base, int floor,
+                                        struct device *dev)
 {
-       struct device *dev = &pdev->dev;
-       struct docg3 *docg3;
-       struct mtd_info *mtd;
-       struct resource *ress;
        int ret, bbt_nbpages;
        u16 chip_id, chip_id_inv;
+       struct docg3 *docg3;
+       struct mtd_info *mtd;
 
        ret = -ENOMEM;
        docg3 = kzalloc(sizeof(struct docg3), GFP_KERNEL);
@@ -1002,69 +1630,133 @@ static int __init docg3_probe(struct platform_device *pdev)
        if (!mtd)
                goto nomem2;
        mtd->priv = docg3;
+       bbt_nbpages = DIV_ROUND_UP(docg3->max_block + 1,
+                                  8 * DOC_LAYOUT_PAGE_SIZE);
+       docg3->bbt = kzalloc(bbt_nbpages * DOC_LAYOUT_PAGE_SIZE, GFP_KERNEL);
+       if (!docg3->bbt)
+               goto nomem3;
 
-       ret = -ENXIO;
-       ress = platform_get_resource(pdev, IORESOURCE_MEM, 0);
-       if (!ress) {
-               dev_err(dev, "No I/O memory resource defined\n");
-               goto noress;
-       }
-       docg3->base = ioremap(ress->start, DOC_IOSPACE_SIZE);
-
-       docg3->dev = &pdev->dev;
-       docg3->device_id = 0;
+       docg3->dev = dev;
+       docg3->device_id = floor;
+       docg3->base = base;
        doc_set_device_id(docg3, docg3->device_id);
-       doc_set_asic_mode(docg3, DOC_ASICMODE_RESET);
+       if (!floor)
+               doc_set_asic_mode(docg3, DOC_ASICMODE_RESET);
        doc_set_asic_mode(docg3, DOC_ASICMODE_NORMAL);
 
        chip_id = doc_register_readw(docg3, DOC_CHIPID);
        chip_id_inv = doc_register_readw(docg3, DOC_CHIPID_INV);
 
-       ret = -ENODEV;
+       ret = 0;
        if (chip_id != (u16)(~chip_id_inv)) {
-               doc_info("No device found at IO addr %p\n",
-                        (void *)ress->start);
-               goto nochipfound;
+               goto nomem3;
        }
 
        switch (chip_id) {
        case DOC_CHIPID_G3:
-               doc_info("Found a G3 DiskOnChip at addr %p\n",
-                        (void *)ress->start);
+               doc_info("Found a G3 DiskOnChip at addr %p, floor %d\n",
+                        base, floor);
                break;
        default:
                doc_err("Chip id %04x is not a DiskOnChip G3 chip\n", chip_id);
-               goto nochipfound;
+               goto nomem3;
        }
 
        doc_set_driver_info(chip_id, mtd);
-       platform_set_drvdata(pdev, mtd);
 
-       ret = -ENOMEM;
-       bbt_nbpages = DIV_ROUND_UP(docg3->max_block + 1,
-                                  8 * DOC_LAYOUT_PAGE_SIZE);
-       docg3->bbt = kzalloc(bbt_nbpages * DOC_LAYOUT_PAGE_SIZE, GFP_KERNEL);
-       if (!docg3->bbt)
-               goto nochipfound;
+       doc_hamming_ecc_init(docg3, DOC_LAYOUT_OOB_PAGEINFO_SZ);
        doc_reload_bbt(docg3);
+       return mtd;
 
-       ret = mtd_device_parse_register(mtd, part_probes,
-                                       NULL, NULL, 0);
-       if (ret)
-               goto register_error;
-
-       doc_dbg_register(docg3);
-       return 0;
-
-register_error:
-       kfree(docg3->bbt);
-nochipfound:
-       iounmap(docg3->base);
-noress:
+nomem3:
        kfree(mtd);
 nomem2:
        kfree(docg3);
 nomem1:
+       return ERR_PTR(ret);
+}
+
+/**
+ * doc_release_device - Release a docg3 floor
+ * @mtd: the device
+ */
+static void doc_release_device(struct mtd_info *mtd)
+{
+       struct docg3 *docg3 = mtd->priv;
+
+       mtd_device_unregister(mtd);
+       kfree(docg3->bbt);
+       kfree(docg3);
+       kfree(mtd->name);
+       kfree(mtd);
+}
+
+/**
+ * doc_probe - Probe the IO space for a DiskOnChip G3 chip
+ * @pdev: platform device
+ *
+ * Probes for a G3 chip at the specified IO space in the platform data
+ * ressources. The floor 0 must be available.
+ *
+ * Returns 0 on success, -ENOMEM, -ENXIO on error
+ */
+static int __init docg3_probe(struct platform_device *pdev)
+{
+       struct device *dev = &pdev->dev;
+       struct mtd_info *mtd;
+       struct resource *ress;
+       void __iomem *base;
+       int ret, floor, found = 0;
+       struct mtd_info **docg3_floors;
+
+       ret = -ENXIO;
+       ress = platform_get_resource(pdev, IORESOURCE_MEM, 0);
+       if (!ress) {
+               dev_err(dev, "No I/O memory resource defined\n");
+               goto noress;
+       }
+       base = ioremap(ress->start, DOC_IOSPACE_SIZE);
+
+       ret = -ENOMEM;
+       docg3_floors = kzalloc(sizeof(*docg3_floors) * DOC_MAX_NBFLOORS,
+                              GFP_KERNEL);
+       if (!docg3_floors)
+               goto nomem;
+
+       ret = 0;
+       for (floor = 0; floor < DOC_MAX_NBFLOORS; floor++) {
+               mtd = doc_probe_device(base, floor, dev);
+               if (floor == 0 && !mtd)
+                       goto notfound;
+               if (!IS_ERR_OR_NULL(mtd))
+                       ret = mtd_device_parse_register(mtd, part_probes,
+                                                       NULL, NULL, 0);
+               else
+                       ret = PTR_ERR(mtd);
+               docg3_floors[floor] = mtd;
+               if (ret)
+                       goto err_probe;
+               if (mtd)
+                       found++;
+       }
+
+       if (!found)
+               goto notfound;
+
+       platform_set_drvdata(pdev, docg3_floors);
+       doc_dbg_register(docg3_floors[0]->priv);
+       return 0;
+
+notfound:
+       ret = -ENODEV;
+       dev_info(dev, "No supported DiskOnChip found\n");
+err_probe:
+       for (floor = 0; floor < DOC_MAX_NBFLOORS; floor++)
+               if (docg3_floors[floor])
+                       doc_release_device(docg3_floors[floor]);
+nomem:
+       iounmap(base);
+noress:
        return ret;
 }
 
@@ -1076,15 +1768,18 @@ nomem1:
  */
 static int __exit docg3_release(struct platform_device *pdev)
 {
-       struct mtd_info *mtd = platform_get_drvdata(pdev);
-       struct docg3 *docg3 = mtd->priv;
+       struct mtd_info **docg3_floors = platform_get_drvdata(pdev);
+       struct docg3 *docg3 = docg3_floors[0]->priv;
+       void __iomem *base = docg3->base;
+       int floor;
 
        doc_dbg_unregister(docg3);
-       mtd_device_unregister(mtd);
-       iounmap(docg3->base);
-       kfree(docg3->bbt);
-       kfree(docg3);
-       kfree(mtd);
+       for (floor = 0; floor < DOC_MAX_NBFLOORS; floor++)
+               if (docg3_floors[floor])
+                       doc_release_device(docg3_floors[floor]);
+
+       kfree(docg3_floors);
+       iounmap(base);
        return 0;
 }