]> git.sur5r.net Git - u-boot/blobdiff - board/dave/common/flash.c
Merge git://git.denx.de/u-boot-arm
[u-boot] / board / dave / common / flash.c
index 3cdbaa920d3add5f17c5d1f2f0bf2e99da3e144c..f05adf9f1c23eafffd9af449fd5452dbef6e24ae 100644 (file)
@@ -2,30 +2,13 @@
  * (C) Copyright 2001
  * Stefan Roese, esd gmbh germany, stefan.roese@esd-electronics.com
  *
- * See file CREDITS for list of people who contributed to this
- * project.
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License as
- * published by the Free Software Foundation; either version 2 of
- * the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
- * MA 02111-1307 USA
+ * SPDX-License-Identifier:    GPL-2.0+
  */
 
 #include <common.h>
-#include <ppc4xx.h>
 #include <asm/processor.h>
 
-flash_info_t   flash_info[CFG_MAX_FLASH_BANKS]; /* info for FLASH chips        */
+flash_info_t   flash_info[CONFIG_SYS_MAX_FLASH_BANKS]; /* info for FLASH chips */
 
 /*-----------------------------------------------------------------------
  * Functions
@@ -40,11 +23,11 @@ static void flash_get_offsets (ulong base, flash_info_t *info)
        short n;
 
        /* set up sector start address table */
-        if (((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) ||
+       if (((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) ||
            ((info->flash_id & FLASH_TYPEMASK) == FLASH_AM640U)) {
            for (i = 0; i < info->sector_count; i++)
                info->start[i] = base + (i * 0x00010000);
-        } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322B) ||
+       } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322B) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL323B) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AM320B) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL324B)) {
@@ -58,7 +41,7 @@ static void flash_get_offsets (ulong base, flash_info_t *info)
                        base += 64 << 10;
                        ++i;
                }
-        } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322T) ||
+       } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322T) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL323T) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AM320T) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL324T)) {
@@ -75,7 +58,7 @@ static void flash_get_offsets (ulong base, flash_info_t *info)
                        --i;
                        info->start[i] = base;
                }
-        } else {
+       } else {
            if (info->flash_id & FLASH_BTYPE) {
                /* set sector offsets for bottom boot block type        */
                info->start[0] = base + 0x00000000;
@@ -103,10 +86,10 @@ static void flash_get_offsets (ulong base, flash_info_t *info)
 void flash_print_info  (flash_info_t *info)
 {
        int i;
-        int k;
-        int size;
-        int erased;
-        volatile unsigned long *flash;
+       int k;
+       int size;
+       int erased;
+       volatile unsigned long *flash;
 
        if (info->flash_id == FLASH_UNKNOWN) {
                printf ("missing or unknown FLASH type\n");
@@ -163,29 +146,29 @@ void flash_print_info  (flash_info_t *info)
 
        printf ("  Sector Start Addresses:");
        for (i=0; i<info->sector_count; ++i) {
-#ifdef CFG_FLASH_EMPTY_INFO
-                /*
-                 * Check if whole sector is erased
-                 */
-                if (i != (info->sector_count-1))
-                  size = info->start[i+1] - info->start[i];
-                else
-                  size = info->start[0] + info->size - info->start[i];
-                erased = 1;
-                flash = (volatile unsigned long *)info->start[i];
-                size = size >> 2;        /* divide by 4 for longword access */
-                for (k=0; k<size; k++)
-                  {
-                    if (*flash++ != 0xffffffff)
-                      {
-                        erased = 0;
-                        break;
-                      }
-                  }
+#ifdef CONFIG_SYS_FLASH_EMPTY_INFO
+               /*
+                * Check if whole sector is erased
+                */
+               if (i != (info->sector_count-1))
+                 size = info->start[i+1] - info->start[i];
+               else
+                 size = info->start[0] + info->size - info->start[i];
+               erased = 1;
+               flash = (volatile unsigned long *)info->start[i];
+               size = size >> 2;        /* divide by 4 for longword access */
+               for (k=0; k<size; k++)
+                 {
+                   if (*flash++ != 0xffffffff)
+                     {
+                       erased = 0;
+                       break;
+                     }
+                 }
 
                if ((i % 5) == 0)
                        printf ("\n   ");
-                /* print empty and read-only info */
+               /* print empty and read-only info */
                printf (" %08lX%s%s",
                        info->start[i],
                        erased ? " E" : "  ",
@@ -217,28 +200,30 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
 {
        short i;
        short n;
-       CFG_FLASH_WORD_SIZE value;
+       CONFIG_SYS_FLASH_WORD_SIZE value;
        ulong base = (ulong)addr;
-        volatile CFG_FLASH_WORD_SIZE *addr2 = (CFG_FLASH_WORD_SIZE *)addr;
+       volatile CONFIG_SYS_FLASH_WORD_SIZE *addr2 = (CONFIG_SYS_FLASH_WORD_SIZE *)addr;
+
+       debug("[%s, %d] Entering ...\n", __FUNCTION__, __LINE__);
 
        /* Write auto select command: read Manufacturer ID */
-       addr2[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00AA00AA;
-       addr2[CFG_FLASH_ADDR1] = (CFG_FLASH_WORD_SIZE)0x00550055;
-       addr2[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00900090;
+       addr2[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00AA00AA;
+       addr2[CONFIG_SYS_FLASH_ADDR1] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00550055;
+       addr2[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00900090;
 
-       value = addr2[CFG_FLASH_READ0];
+       value = addr2[CONFIG_SYS_FLASH_READ0];
 
        switch (value) {
-       case (CFG_FLASH_WORD_SIZE)AMD_MANUFACT:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_MANUFACT:
                info->flash_id = FLASH_MAN_AMD;
                break;
-       case (CFG_FLASH_WORD_SIZE)FUJ_MANUFACT:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)FUJ_MANUFACT:
                info->flash_id = FLASH_MAN_FUJ;
                break;
-       case (CFG_FLASH_WORD_SIZE)SST_MANUFACT:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)SST_MANUFACT:
                info->flash_id = FLASH_MAN_SST;
                break;
-       case (CFG_FLASH_WORD_SIZE)STM_MANUFACT:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)STM_MANUFACT:
                info->flash_id = FLASH_MAN_STM;
                break;
        default:
@@ -248,92 +233,92 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
                return (0);                     /* no or unknown flash  */
        }
 
-       value = addr2[CFG_FLASH_READ1];         /* device ID            */
+       value = addr2[CONFIG_SYS_FLASH_READ1];          /* device ID            */
 
        switch (value) {
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV400T:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV400T:
                info->flash_id += FLASH_AM400T;
                info->sector_count = 11;
                info->size = 0x00080000;
                break;                          /* => 0.5 MB            */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV400B:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV400B:
                info->flash_id += FLASH_AM400B;
                info->sector_count = 11;
                info->size = 0x00080000;
                break;                          /* => 0.5 MB            */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV800T:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV800T:
                info->flash_id += FLASH_AM800T;
                info->sector_count = 19;
                info->size = 0x00100000;
                break;                          /* => 1 MB              */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV800B:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV800B:
                info->flash_id += FLASH_AM800B;
                info->sector_count = 19;
                info->size = 0x00100000;
                break;                          /* => 1 MB              */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV160T:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV160T:
                info->flash_id += FLASH_AM160T;
                info->sector_count = 35;
                info->size = 0x00200000;
                break;                          /* => 2 MB              */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV160B:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV160B:
                info->flash_id += FLASH_AM160B;
                info->sector_count = 35;
                info->size = 0x00200000;
                break;                          /* => 2 MB              */
 
-       case (CFG_FLASH_WORD_SIZE)STM_ID_29W320DT:
-               info->flash_id += FLASH_STMW320DT;
-               info->sector_count = 67;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)STM_ID_29W320DT:
+               info->flash_id += FLASH_STMW320DT;
+               info->sector_count = 67;
                info->size = 0x00400000;  break;        /* => 4 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV320T:
-               info->flash_id += FLASH_AM320T;
-               info->sector_count = 71;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV320T:
+               info->flash_id += FLASH_AM320T;
+               info->sector_count = 71;
                info->size = 0x00400000;  break;        /* => 4 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV320B:
-               info->flash_id += FLASH_AM320B;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV320B:
+               info->flash_id += FLASH_AM320B;
                info->sector_count = 71;
                info->size = 0x00400000;  break;        /* => 4 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_DL322T:
-               info->flash_id += FLASH_AMDL322T;
-               info->sector_count = 71;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_DL322T:
+               info->flash_id += FLASH_AMDL322T;
+               info->sector_count = 71;
                info->size = 0x00400000;  break;        /* => 4 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_DL322B:
-               info->flash_id += FLASH_AMDL322B;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_DL322B:
+               info->flash_id += FLASH_AMDL322B;
                info->sector_count = 71;
                info->size = 0x00400000;  break;        /* => 4 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_DL323T:
-               info->flash_id += FLASH_AMDL323T;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_DL323T:
+               info->flash_id += FLASH_AMDL323T;
                info->sector_count = 71;
                info->size = 0x00400000;  break;        /* => 4 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_DL323B:
-               info->flash_id += FLASH_AMDL323B;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_DL323B:
+               info->flash_id += FLASH_AMDL323B;
                info->sector_count = 71;
                info->size = 0x00400000;  break;        /* => 4 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)AMD_ID_LV640U:
-               info->flash_id += FLASH_AM640U;
+       case (CONFIG_SYS_FLASH_WORD_SIZE)AMD_ID_LV640U:
+               info->flash_id += FLASH_AM640U;
                info->sector_count = 128;
                info->size = 0x00800000;  break;        /* => 8 MB      */
 
-       case (CFG_FLASH_WORD_SIZE)SST_ID_xF800A:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)SST_ID_xF800A:
                info->flash_id += FLASH_SST800A;
                info->sector_count = 16;
                info->size = 0x00100000;
                break;                          /* => 1 MB              */
 
-       case (CFG_FLASH_WORD_SIZE)SST_ID_xF160A:
+       case (CONFIG_SYS_FLASH_WORD_SIZE)SST_ID_xF160A:
                info->flash_id += FLASH_SST160A;
                info->sector_count = 32;
                info->size = 0x00200000;
@@ -346,11 +331,11 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
        }
 
        /* set up sector start address table */
-        if (((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) ||
+       if (((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) ||
            ((info->flash_id & FLASH_TYPEMASK) == FLASH_AM640U)) {
            for (i = 0; i < info->sector_count; i++)
                info->start[i] = base + (i * 0x00010000);
-        } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322B) ||
+       } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322B) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL323B) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AM320B) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL324B)) {
@@ -364,7 +349,7 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
                        base += 64 << 10;
                        ++i;
                }
-        } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322T) ||
+       } else if (((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL322T) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL323T) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AM320T) ||
                   ((info->flash_id & FLASH_TYPEMASK) == FLASH_AMDL324T)) {
@@ -381,13 +366,13 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
                        --i;
                        info->start[i] = base;
                }
-        } else if ((info->flash_id & FLASH_TYPEMASK) == FLASH_STMW320DT) {
+       } else if ((info->flash_id & FLASH_TYPEMASK) == FLASH_STMW320DT) {
                /* set sector offsets for top boot block type           */
                base += info->size;
                i = info->sector_count;
-                /*  1 x 16k boot sector */
+               /*  1 x 16k boot sector */
                base -= 16 << 10;
-                --i;
+               --i;
                info->start[i] = base;
                /*  2 x 8k  boot sectors */
                for (n=0; n<2; ++n) {
@@ -395,9 +380,9 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
                        --i;
                        info->start[i] = base;
                }
-                /*  1 x 32k boot sector */
+               /*  1 x 32k boot sector */
                base -= 32 << 10;
-                --i;
+               --i;
                info->start[i] = base;
 
                while (i > 0) {                 /* 64k regular sectors  */
@@ -405,7 +390,7 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
                        --i;
                        info->start[i] = base;
                }
-        } else {
+       } else {
            if (info->flash_id & FLASH_BTYPE) {
                /* set sector offsets for bottom boot block type        */
                info->start[0] = base + 0x00000000;
@@ -431,19 +416,19 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
        for (i = 0; i < info->sector_count; i++) {
                /* read sector protection at sector address, (A7 .. A0) = 0x02 */
                /* D0 = 1 if protected */
-               addr2 = (volatile CFG_FLASH_WORD_SIZE *)(info->start[i]);
-                if ((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST)
-                  info->protect[i] = 0;
-                else
-                  info->protect[i] = addr2[CFG_FLASH_READ2] & 1;
+               addr2 = (volatile CONFIG_SYS_FLASH_WORD_SIZE *)(info->start[i]);
+               if ((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST)
+                 info->protect[i] = 0;
+               else
+                 info->protect[i] = addr2[CONFIG_SYS_FLASH_READ2] & 1;
        }
 
        /*
         * Prevent writes to uninitialized FLASH.
         */
        if (info->flash_id != FLASH_UNKNOWN) {
-               addr2 = (CFG_FLASH_WORD_SIZE *)info->start[0];
-               *addr2 = (CFG_FLASH_WORD_SIZE)0x00F000F0;       /* reset bank */
+               addr2 = (CONFIG_SYS_FLASH_WORD_SIZE *)info->start[0];
+               *addr2 = (CONFIG_SYS_FLASH_WORD_SIZE)0x00F000F0;        /* reset bank */
        }
 
        return (info->size);
@@ -455,11 +440,11 @@ static ulong flash_get_size (vu_long *addr, flash_info_t *info)
 
 int    flash_erase (flash_info_t *info, int s_first, int s_last)
 {
-       volatile CFG_FLASH_WORD_SIZE *addr = (CFG_FLASH_WORD_SIZE *)(info->start[0]);
-       volatile CFG_FLASH_WORD_SIZE *addr2;
+       volatile CONFIG_SYS_FLASH_WORD_SIZE *addr = (CONFIG_SYS_FLASH_WORD_SIZE *)(info->start[0]);
+       volatile CONFIG_SYS_FLASH_WORD_SIZE *addr2;
        int flag, prot, sect, l_sect;
        ulong start, now, last;
-        int i;
+       int i;
 
        if ((s_first < 0) || (s_first > s_last)) {
                if (info->flash_id == FLASH_UNKNOWN) {
@@ -497,26 +482,26 @@ int       flash_erase (flash_info_t *info, int s_first, int s_last)
        /* Start erase on unprotected sectors */
        for (sect = s_first; sect<=s_last; sect++) {
                if (info->protect[sect] == 0) { /* not protected */
-                   addr2 = (CFG_FLASH_WORD_SIZE *)(info->start[sect]);
-                    if ((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) {
-                        addr[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00AA00AA;
-                        addr[CFG_FLASH_ADDR1] = (CFG_FLASH_WORD_SIZE)0x00550055;
-                        addr[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00800080;
-                        addr[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00AA00AA;
-                        addr[CFG_FLASH_ADDR1] = (CFG_FLASH_WORD_SIZE)0x00550055;
-                        addr2[0] = (CFG_FLASH_WORD_SIZE)0x00500050;  /* block erase */
-                        for (i=0; i<50; i++)
-                          udelay(1000);  /* wait 1 ms */
-                    } else {
-                        if (sect == s_first) {
-                            addr[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00AA00AA;
-                            addr[CFG_FLASH_ADDR1] = (CFG_FLASH_WORD_SIZE)0x00550055;
-                            addr[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00800080;
-                            addr[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00AA00AA;
-                            addr[CFG_FLASH_ADDR1] = (CFG_FLASH_WORD_SIZE)0x00550055;
-                        }
-                        addr2[0] = (CFG_FLASH_WORD_SIZE)0x00300030;  /* sector erase */
-                    }
+                   addr2 = (CONFIG_SYS_FLASH_WORD_SIZE *)(info->start[sect]);
+                   if ((info->flash_id & FLASH_VENDMASK) == FLASH_MAN_SST) {
+                       addr[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00AA00AA;
+                       addr[CONFIG_SYS_FLASH_ADDR1] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00550055;
+                       addr[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00800080;
+                       addr[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00AA00AA;
+                       addr[CONFIG_SYS_FLASH_ADDR1] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00550055;
+                       addr2[0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00500050;  /* block erase */
+                       for (i=0; i<50; i++)
+                         udelay(1000);  /* wait 1 ms */
+                   } else {
+                       if (sect == s_first) {
+                           addr[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00AA00AA;
+                           addr[CONFIG_SYS_FLASH_ADDR1] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00550055;
+                           addr[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00800080;
+                           addr[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00AA00AA;
+                           addr[CONFIG_SYS_FLASH_ADDR1] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00550055;
+                       }
+                       addr2[0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00300030;  /* sector erase */
+                   }
                    l_sect = sect;
                }
        }
@@ -536,9 +521,9 @@ int flash_erase (flash_info_t *info, int s_first, int s_last)
 
        start = get_timer (0);
        last  = start;
-       addr = (CFG_FLASH_WORD_SIZE *)(info->start[l_sect]);
-       while ((addr[0] & (CFG_FLASH_WORD_SIZE)0x00800080) != (CFG_FLASH_WORD_SIZE)0x00800080) {
-               if ((now = get_timer(start)) > CFG_FLASH_ERASE_TOUT) {
+       addr = (CONFIG_SYS_FLASH_WORD_SIZE *)(info->start[l_sect]);
+       while ((addr[0] & (CONFIG_SYS_FLASH_WORD_SIZE)0x00800080) != (CONFIG_SYS_FLASH_WORD_SIZE)0x00800080) {
+               if ((now = get_timer(start)) > CONFIG_SYS_FLASH_ERASE_TOUT) {
                        printf ("Timeout\n");
                        return 1;
                }
@@ -551,8 +536,8 @@ int flash_erase (flash_info_t *info, int s_first, int s_last)
 
 DONE:
        /* reset to read mode */
-       addr = (CFG_FLASH_WORD_SIZE *)info->start[0];
-       addr[0] = (CFG_FLASH_WORD_SIZE)0x00F000F0;      /* reset bank */
+       addr = (CONFIG_SYS_FLASH_WORD_SIZE *)info->start[0];
+       addr[0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00F000F0;       /* reset bank */
 
        printf (" done\n");
        return 0;
@@ -578,15 +563,27 @@ int write_buff (flash_info_t *info, uchar *src, ulong addr, ulong cnt)
        if ((l = addr - wp) != 0) {
                data = 0;
                for (i=0, cp=wp; i<l; ++i, ++cp) {
+#ifdef CONFIG_B2
+                       data = data | ((*(uchar *)cp)<<(8*i));
+#else
                        data = (data << 8) | (*(uchar *)cp);
+#endif
                }
                for (; i<4 && cnt>0; ++i) {
+#ifdef CONFIG_B2
+                       data = data  | ((*src++)<<(8*i));
+#else
                        data = (data << 8) | *src++;
+#endif
                        --cnt;
                        ++cp;
                }
                for (; cnt==0 && i<4; ++i, ++cp) {
+#ifdef CONFIG_B2
+                       data = data | ((*(uchar *)cp)<<(8*i));
+#else
                        data = (data << 8) | (*(uchar *)cp);
+#endif
                }
 
                if ((rc = write_word(info, wp, data)) != 0) {
@@ -600,9 +597,14 @@ int write_buff (flash_info_t *info, uchar *src, ulong addr, ulong cnt)
         */
        while (cnt >= 4) {
                data = 0;
+#ifdef CONFIG_B2
+               data = (*(ulong*)src);
+               src += 4;
+#else
                for (i=0; i<4; ++i) {
                        data = (data << 8) | *src++;
                }
+#endif
                if ((rc = write_word(info, wp, data)) != 0) {
                        return (rc);
                }
@@ -619,11 +621,19 @@ int write_buff (flash_info_t *info, uchar *src, ulong addr, ulong cnt)
         */
        data = 0;
        for (i=0, cp=wp; i<4 && cnt>0; ++i, ++cp) {
+#ifdef CONFIG_B2
+               data = data  | ((*src++)<<(8*i));
+#else
                data = (data << 8) | *src++;
+#endif
                --cnt;
        }
        for (; i<4; ++i, ++cp) {
+#ifdef CONFIG_B2
+               data = data | ((*(uchar *)cp)<<(8*i));
+#else
                data = (data << 8) | (*(uchar *)cp);
+#endif
        }
 
        return (write_word(info, wp, data));
@@ -637,42 +647,42 @@ int write_buff (flash_info_t *info, uchar *src, ulong addr, ulong cnt)
  */
 static int write_word (flash_info_t *info, ulong dest, ulong data)
 {
-        volatile CFG_FLASH_WORD_SIZE *addr2 = (CFG_FLASH_WORD_SIZE *)(info->start[0]);
-        volatile CFG_FLASH_WORD_SIZE *dest2 = (CFG_FLASH_WORD_SIZE *)dest;
-        volatile CFG_FLASH_WORD_SIZE *data2 = (CFG_FLASH_WORD_SIZE *)&data;
+       ulong *data_ptr = &data;
+       volatile CONFIG_SYS_FLASH_WORD_SIZE *addr2 = (CONFIG_SYS_FLASH_WORD_SIZE *)(info->start[0]);
+       volatile CONFIG_SYS_FLASH_WORD_SIZE *dest2 = (CONFIG_SYS_FLASH_WORD_SIZE *)dest;
+       volatile CONFIG_SYS_FLASH_WORD_SIZE *data2 = (CONFIG_SYS_FLASH_WORD_SIZE *)data_ptr;
        ulong start;
        int flag;
-        int i;
+       int i;
 
        /* Check if Flash is (sufficiently) erased */
-       if ((*((volatile CFG_FLASH_WORD_SIZE *)dest) &
-             (CFG_FLASH_WORD_SIZE)data) != (CFG_FLASH_WORD_SIZE)data) {
+       if ((*((volatile ulong *)dest) & data) != data) {
                return (2);
        }
        /* Disable interrupts which might cause a timeout here */
        flag = disable_interrupts();
 
-        for (i=0; i<4/sizeof(CFG_FLASH_WORD_SIZE); i++)
-          {
-            addr2[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00AA00AA;
-            addr2[CFG_FLASH_ADDR1] = (CFG_FLASH_WORD_SIZE)0x00550055;
-            addr2[CFG_FLASH_ADDR0] = (CFG_FLASH_WORD_SIZE)0x00A000A0;
-
-            dest2[i] = data2[i];
-
-            /* re-enable interrupts if necessary */
-            if (flag)
-              enable_interrupts();
-
-            /* data polling for D7 */
-            start = get_timer (0);
-            while ((dest2[i] & (CFG_FLASH_WORD_SIZE)0x00800080) !=
-                   (data2[i] & (CFG_FLASH_WORD_SIZE)0x00800080)) {
-              if (get_timer(start) > CFG_FLASH_WRITE_TOUT) {
-                return (1);
-              }
-            }
-          }
+       for (i=0; i<4/sizeof(CONFIG_SYS_FLASH_WORD_SIZE); i++)
+         {
+           addr2[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00AA00AA;
+           addr2[CONFIG_SYS_FLASH_ADDR1] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00550055;
+           addr2[CONFIG_SYS_FLASH_ADDR0] = (CONFIG_SYS_FLASH_WORD_SIZE)0x00A000A0;
+
+           dest2[i] = data2[i];
+
+           /* re-enable interrupts if necessary */
+           if (flag)
+             enable_interrupts();
+
+           /* data polling for D7 */
+           start = get_timer (0);
+           while ((dest2[i] & (CONFIG_SYS_FLASH_WORD_SIZE)0x00800080) !=
+                  (data2[i] & (CONFIG_SYS_FLASH_WORD_SIZE)0x00800080)) {
+             if (get_timer(start) > CONFIG_SYS_FLASH_WRITE_TOUT) {
+               return (1);
+             }
+           }
+         }
 
        return (0);
 }