]> git.karo-electronics.de Git - mv-sheeva.git/blobdiff - drivers/staging/keucr/ms.c
Merge branch 'master' into csb1725
[mv-sheeva.git] / drivers / staging / keucr / ms.c
diff --git a/drivers/staging/keucr/ms.c b/drivers/staging/keucr/ms.c
new file mode 100644 (file)
index 0000000..9a3fdb4
--- /dev/null
@@ -0,0 +1,956 @@
+#include <linux/slab.h>
+#include "usb.h"
+#include "scsiglue.h"
+#include "transport.h"
+#include "ms.h"
+
+//----- MS_ReaderCopyBlock() ------------------------------------------
+int MS_ReaderCopyBlock(struct us_data *us, WORD oldphy, WORD newphy, WORD PhyBlockAddr, BYTE PageNum, PBYTE buf, WORD len)
+{
+       struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *) us->iobuf;
+       int     result;
+
+       //printk("MS_ReaderCopyBlock --- PhyBlockAddr = %x, PageNum = %x\n", PhyBlockAddr, PageNum);
+       result = ENE_LoadBinCode(us, MS_RW_PATTERN);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       memset(bcb, 0, sizeof(struct bulk_cb_wrap));
+       bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
+       bcb->DataTransferLength = 0x200*len;
+       bcb->Flags                      = 0x00;
+       bcb->CDB[0]                     = 0xF0;
+       bcb->CDB[1]                     = 0x08;
+       bcb->CDB[4]                     = (BYTE)(oldphy);
+       bcb->CDB[3]                     = (BYTE)(oldphy>>8);
+       bcb->CDB[2]                     = (BYTE)(oldphy>>16);
+       bcb->CDB[7]                     = (BYTE)(newphy);
+       bcb->CDB[6]                     = (BYTE)(newphy>>8);
+       bcb->CDB[5]                     = (BYTE)(newphy>>16);
+       bcb->CDB[9]                     = (BYTE)(PhyBlockAddr);
+       bcb->CDB[8]                     = (BYTE)(PhyBlockAddr>>8);
+       bcb->CDB[10]            = PageNum;
+
+       result = ENE_SendScsiCmd(us, FDIR_WRITE, buf, 0);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       return USB_STOR_TRANSPORT_GOOD;
+}
+
+//----- MS_ReaderReadPage() ------------------------------------------
+int MS_ReaderReadPage(struct us_data *us, DWORD PhyBlockAddr, BYTE PageNum, PDWORD PageBuf, MS_LibTypeExtdat *ExtraDat)
+{
+       struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *) us->iobuf;
+       int     result;
+       BYTE    ExtBuf[4];
+       DWORD   bn = PhyBlockAddr * 0x20 + PageNum;
+
+       //printk("MS --- MS_ReaderReadPage,  PhyBlockAddr = %x, PageNum = %x\n", PhyBlockAddr, PageNum);
+
+       result = ENE_LoadBinCode(us, MS_RW_PATTERN);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       // Read Page Data
+       memset(bcb, 0, sizeof(struct bulk_cb_wrap));
+       bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
+       bcb->DataTransferLength = 0x200;
+       bcb->Flags                      = 0x80;
+       bcb->CDB[0]                     = 0xF1;
+       bcb->CDB[1]                     = 0x02;
+       bcb->CDB[5]                     = (BYTE)(bn);
+       bcb->CDB[4]                     = (BYTE)(bn>>8);
+       bcb->CDB[3]                     = (BYTE)(bn>>16);
+       bcb->CDB[2]                     = (BYTE)(bn>>24);
+       
+       result = ENE_SendScsiCmd(us, FDIR_READ, PageBuf, 0);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       // Read Extra Data
+       memset(bcb, 0, sizeof(struct bulk_cb_wrap));
+       bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
+       bcb->DataTransferLength = 0x4;
+       bcb->Flags                      = 0x80;
+       bcb->CDB[0]                     = 0xF1;
+       bcb->CDB[1]                     = 0x03;
+       bcb->CDB[5]                     = (BYTE)(PageNum);
+       bcb->CDB[4]                     = (BYTE)(PhyBlockAddr);
+       bcb->CDB[3]                     = (BYTE)(PhyBlockAddr>>8);
+       bcb->CDB[2]                     = (BYTE)(PhyBlockAddr>>16);
+       bcb->CDB[6]                     = 0x01;
+
+       result = ENE_SendScsiCmd(us, FDIR_READ, &ExtBuf, 0);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       ExtraDat->reserved = 0;
+       ExtraDat->intr     = 0x80;  // Not yet, ¥ý°²³], µ¥ fireware support
+       ExtraDat->status0  = 0x10;  // Not yet, ¥ý°²³], µ¥ fireware support
+       ExtraDat->status1  = 0x00;  // Not yet, ¥ý°²³], µ¥ fireware support
+       ExtraDat->ovrflg   = ExtBuf[0];
+       ExtraDat->mngflg   = ExtBuf[1];
+       ExtraDat->logadr   = MemStickLogAddr(ExtBuf[2], ExtBuf[3]);
+
+       return USB_STOR_TRANSPORT_GOOD;
+}
+
+//----- MS_ReaderEraseBlock() ----------------------------------------
+int MS_ReaderEraseBlock(struct us_data *us, DWORD PhyBlockAddr)
+{
+       struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *) us->iobuf;
+       int     result;
+       DWORD   bn = PhyBlockAddr;
+
+       //printk("MS --- MS_ReaderEraseBlock,  PhyBlockAddr = %x\n", PhyBlockAddr);
+       result = ENE_LoadBinCode(us, MS_RW_PATTERN);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       memset(bcb, 0, sizeof(struct bulk_cb_wrap));
+       bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
+       bcb->DataTransferLength = 0x200;
+       bcb->Flags                      = 0x80;
+       bcb->CDB[0]                     = 0xF2;
+       bcb->CDB[1]                     = 0x06;
+       bcb->CDB[4]                     = (BYTE)(bn);
+       bcb->CDB[3]                     = (BYTE)(bn>>8);
+       bcb->CDB[2]                     = (BYTE)(bn>>16);
+       
+       result = ENE_SendScsiCmd(us, FDIR_READ, NULL, 0);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       return USB_STOR_TRANSPORT_GOOD;
+}
+
+//----- MS_CardInit() ------------------------------------------------
+int MS_CardInit(struct us_data *us)
+{
+       DWORD                   result=0;
+       WORD                    TmpBlock;
+       PBYTE                   PageBuffer0 = NULL, PageBuffer1 = NULL;
+       MS_LibTypeExtdat        extdat;
+       WORD                    btBlk1st, btBlk2nd;
+       DWORD                   btBlk1stErred;
+
+       printk("MS_CardInit start\n");
+
+       MS_LibFreeAllocatedArea(us);
+
+       if (((PageBuffer0 = kmalloc(MS_BYTES_PER_PAGE, GFP_KERNEL)) == NULL) ||
+           ((PageBuffer1 = kmalloc(MS_BYTES_PER_PAGE, GFP_KERNEL)) == NULL))
+       {
+               result = MS_NO_MEMORY_ERROR;
+               goto exit;
+       }
+
+       btBlk1st = btBlk2nd = MS_LB_NOT_USED;
+       btBlk1stErred = 0;
+
+       for (TmpBlock=0; TmpBlock < MS_MAX_INITIAL_ERROR_BLOCKS+2; TmpBlock++)
+       {
+               switch (MS_ReaderReadPage(us, TmpBlock, 0, (DWORD *)PageBuffer0, &extdat))
+               {
+                       case MS_STATUS_SUCCESS:
+                       break;
+                       case MS_STATUS_INT_ERROR:
+                       break;
+                       case MS_STATUS_ERROR:
+                       default:
+                       continue;
+               }
+
+               if ((extdat.ovrflg & MS_REG_OVR_BKST) == MS_REG_OVR_BKST_NG)
+                       continue;
+
+               if (((extdat.mngflg & MS_REG_MNG_SYSFLG) == MS_REG_MNG_SYSFLG_USER) ||
+                       (BigEndianWORD(((MemStickBootBlockPage0 *)PageBuffer0)->header.wBlockID) != MS_BOOT_BLOCK_ID) ||
+                       (BigEndianWORD(((MemStickBootBlockPage0 *)PageBuffer0)->header.wFormatVersion) != MS_BOOT_BLOCK_FORMAT_VERSION) ||
+                       (((MemStickBootBlockPage0 *)PageBuffer0)->header.bNumberOfDataEntry != MS_BOOT_BLOCK_DATA_ENTRIES))
+                               continue;
+
+               if (btBlk1st != MS_LB_NOT_USED)
+               {
+                       btBlk2nd = TmpBlock;
+                       break;
+               }
+
+               btBlk1st = TmpBlock;
+               memcpy(PageBuffer1, PageBuffer0, MS_BYTES_PER_PAGE);
+               if (extdat.status1 & (MS_REG_ST1_DTER | MS_REG_ST1_EXER | MS_REG_ST1_FGER))
+                       btBlk1stErred = 1;
+       }
+
+       if (btBlk1st == MS_LB_NOT_USED)
+       {
+               result = MS_STATUS_ERROR;
+               goto exit;
+       }
+
+       // write protect
+       if ((extdat.status0 & MS_REG_ST0_WP) == MS_REG_ST0_WP_ON)
+               MS_LibCtrlSet(us, MS_LIB_CTRL_WRPROTECT);
+
+       result = MS_STATUS_ERROR;
+       // 1st Boot Block
+       if (btBlk1stErred == 0)
+               result = MS_LibProcessBootBlock(us, btBlk1st, PageBuffer1);   // 1st
+       // 2nd Boot Block
+       if (result && (btBlk2nd != MS_LB_NOT_USED))
+               result = MS_LibProcessBootBlock(us, btBlk2nd, PageBuffer0);
+
+       if (result)
+       {
+               result = MS_STATUS_ERROR;
+               goto exit;
+       }
+
+       for (TmpBlock = 0; TmpBlock < btBlk1st; TmpBlock++)
+               us->MS_Lib.Phy2LogMap[TmpBlock] = MS_LB_INITIAL_ERROR;
+
+       us->MS_Lib.Phy2LogMap[btBlk1st] = MS_LB_BOOT_BLOCK;
+
+       if (btBlk2nd != MS_LB_NOT_USED)
+       {
+               for (TmpBlock = btBlk1st + 1; TmpBlock < btBlk2nd; TmpBlock++)
+                       us->MS_Lib.Phy2LogMap[TmpBlock] = MS_LB_INITIAL_ERROR;
+               us->MS_Lib.Phy2LogMap[btBlk2nd] = MS_LB_BOOT_BLOCK;
+       }
+
+       result = MS_LibScanLogicalBlockNumber(us, btBlk1st);
+       if (result)
+               goto exit;
+
+       for (TmpBlock=MS_PHYSICAL_BLOCKS_PER_SEGMENT; TmpBlock<us->MS_Lib.NumberOfPhyBlock; TmpBlock+=MS_PHYSICAL_BLOCKS_PER_SEGMENT)
+       {
+               if (MS_CountFreeBlock(us, TmpBlock) == 0)
+               {
+                       MS_LibCtrlSet(us, MS_LIB_CTRL_WRPROTECT);
+                       break;
+               }
+       }
+
+       // write
+       if (MS_LibAllocWriteBuf(us))
+       {
+               result = MS_NO_MEMORY_ERROR;
+               goto exit;
+       }
+
+       result = MS_STATUS_SUCCESS;
+
+exit:
+       if (PageBuffer1)                kfree(PageBuffer1);
+       if (PageBuffer0)                kfree(PageBuffer0);
+
+       printk("MS_CardInit end\n");
+       return result;
+}
+
+//----- MS_LibCheckDisableBlock() ------------------------------------
+int MS_LibCheckDisableBlock(struct us_data *us, WORD PhyBlock)
+{
+       PWORD                   PageBuf=NULL;
+       DWORD                   result=MS_STATUS_SUCCESS;
+       DWORD                   blk, index=0;
+       MS_LibTypeExtdat        extdat;
+
+       if (((PageBuf = kmalloc(MS_BYTES_PER_PAGE, GFP_KERNEL)) == NULL))
+       {
+               result = MS_NO_MEMORY_ERROR;
+               goto exit;
+       }
+
+       MS_ReaderReadPage(us, PhyBlock, 1, (DWORD *)PageBuf, &extdat);
+       do
+       {
+               blk = BigEndianWORD(PageBuf[index]);
+               if (blk == MS_LB_NOT_USED)
+                       break;
+               if (blk == us->MS_Lib.Log2PhyMap[0])
+               {
+                       result = MS_ERROR_FLASH_READ;
+                       break;
+               }
+               index++;
+       } while(1);
+
+exit:
+       if (PageBuf)    kfree(PageBuf);
+       return result;
+}
+
+//----- MS_LibFreeAllocatedArea() ------------------------------------
+void MS_LibFreeAllocatedArea(struct us_data *us)
+{
+       MS_LibFreeWriteBuf(us);
+       MS_LibFreeLogicalMap(us);
+
+       us->MS_Lib.flags                        = 0;
+       us->MS_Lib.BytesPerSector       = 0;
+       us->MS_Lib.SectorsPerCylinder   = 0;
+
+       us->MS_Lib.cardType             = 0;
+       us->MS_Lib.blockSize            = 0;
+       us->MS_Lib.PagesPerBlock        = 0;
+
+       us->MS_Lib.NumberOfPhyBlock     = 0;
+       us->MS_Lib.NumberOfLogBlock     = 0;
+}
+
+//----- MS_LibFreeWriteBuf() -----------------------------------------
+void MS_LibFreeWriteBuf(struct us_data *us)
+{
+       us->MS_Lib.wrtblk = (WORD)-1; //set to -1
+       MS_LibClearPageMap(us); // memset((fdoExt)->MS_Lib.pagemap, 0, sizeof((fdoExt)->MS_Lib.pagemap))
+
+       if (us->MS_Lib.blkpag)
+       {
+               kfree((BYTE *)(us->MS_Lib.blkpag));  // Arnold test ...
+               us->MS_Lib.blkpag = NULL;
+       }
+
+       if (us->MS_Lib.blkext)
+       {
+               kfree((BYTE *)(us->MS_Lib.blkext));  // Arnold test ...
+               us->MS_Lib.blkext = NULL;
+       }
+}
+
+//----- MS_LibFreeLogicalMap() ---------------------------------------
+int MS_LibFreeLogicalMap(struct us_data *us)
+{
+       if (us->MS_Lib.Phy2LogMap)
+       {
+               kfree(us->MS_Lib.Phy2LogMap);
+               us->MS_Lib.Phy2LogMap = NULL;
+       }
+
+       if (us->MS_Lib.Log2PhyMap)
+       {
+               kfree(us->MS_Lib.Log2PhyMap);
+               us->MS_Lib.Log2PhyMap = NULL;
+       }
+
+    return 0;
+}
+
+//----- MS_LibProcessBootBlock() -------------------------------------
+int MS_LibProcessBootBlock(struct us_data *us, WORD PhyBlock, BYTE *PageData)
+{
+       MemStickBootBlockSysEnt  *SysEntry;
+       MemStickBootBlockSysInf  *SysInfo;
+       DWORD                    i, result;
+       BYTE                     PageNumber;
+       BYTE                     *PageBuffer;
+       MS_LibTypeExtdat         ExtraData;
+
+       if ((PageBuffer = (BYTE *)kmalloc(MS_BYTES_PER_PAGE, GFP_KERNEL))==NULL)
+               return (DWORD)-1;
+
+       result = (DWORD)-1;
+
+       SysInfo= &(((MemStickBootBlockPage0 *)PageData)->sysinf);
+
+       if ((SysInfo->bMsClass != MS_SYSINF_MSCLASS_TYPE_1)                                   ||
+               (BigEndianWORD(SysInfo->wPageSize) != MS_SYSINF_PAGE_SIZE)                       ||
+               ((SysInfo->bSecuritySupport & MS_SYSINF_SECURITY) == MS_SYSINF_SECURITY_SUPPORT) ||
+               (SysInfo->bReserved1 != MS_SYSINF_RESERVED1)                                     ||
+               (SysInfo->bReserved2 != MS_SYSINF_RESERVED2)                                     ||
+               (SysInfo->bFormatType!= MS_SYSINF_FORMAT_FAT)                                    ||
+               (SysInfo->bUsage != MS_SYSINF_USAGE_GENERAL))
+               goto exit;
+
+       switch (us->MS_Lib.cardType = SysInfo->bCardType)
+       {
+               case MS_SYSINF_CARDTYPE_RDONLY:
+                       MS_LibCtrlSet(us, MS_LIB_CTRL_RDONLY);
+                       break;
+               case MS_SYSINF_CARDTYPE_RDWR:
+                       MS_LibCtrlReset(us, MS_LIB_CTRL_RDONLY);
+                       break;
+               case MS_SYSINF_CARDTYPE_HYBRID:
+               default:
+                       goto exit;
+       }
+
+       us->MS_Lib.blockSize        = BigEndianWORD(SysInfo->wBlockSize);
+       us->MS_Lib.NumberOfPhyBlock = BigEndianWORD(SysInfo->wBlockNumber);
+       us->MS_Lib.NumberOfLogBlock = BigEndianWORD(SysInfo->wTotalBlockNumber)- 2;
+       us->MS_Lib.PagesPerBlock    = us->MS_Lib.blockSize * SIZE_OF_KIRO / MS_BYTES_PER_PAGE;
+       us->MS_Lib.NumberOfSegment  = us->MS_Lib.NumberOfPhyBlock / MS_PHYSICAL_BLOCKS_PER_SEGMENT;
+       us->MS_Model                = BigEndianWORD(SysInfo->wMemorySize);
+
+       if (MS_LibAllocLogicalMap(us))                  //Allocate to all number of logicalblock and physicalblock
+               goto exit;
+
+       MS_LibSetBootBlockMark(us, PhyBlock);           //Mark the book block
+
+       SysEntry = &(((MemStickBootBlockPage0 *)PageData)->sysent);
+
+       for (i=0; i<MS_NUMBER_OF_SYSTEM_ENTRY; i++)
+       {
+               DWORD  EntryOffset, EntrySize;
+
+               if ((EntryOffset = BigEndianDWORD(SysEntry->entry[i].dwStart)) == 0xffffff)
+                       continue;
+
+               if ((EntrySize = BigEndianDWORD(SysEntry->entry[i].dwSize)) == 0)
+                       continue;
+
+               if (EntryOffset + MS_BYTES_PER_PAGE + EntrySize > us->MS_Lib.blockSize * (DWORD)SIZE_OF_KIRO)
+                       continue;
+
+               if (i == 0)
+               {
+                       BYTE  PrevPageNumber = 0;
+                       WORD  phyblk;
+
+                       if (SysEntry->entry[i].bType != MS_SYSENT_TYPE_INVALID_BLOCK)
+                               goto exit;
+
+                       while (EntrySize > 0)
+                       {
+                               if ((PageNumber = (BYTE)(EntryOffset / MS_BYTES_PER_PAGE + 1)) != PrevPageNumber)
+                               {
+                                       switch (MS_ReaderReadPage(us, PhyBlock, PageNumber, (DWORD *)PageBuffer, &ExtraData))
+                                       {
+                                               case MS_STATUS_SUCCESS:
+                                                       break;
+                                               case MS_STATUS_WRITE_PROTECT:
+                                               case MS_ERROR_FLASH_READ:
+                                               case MS_STATUS_ERROR:
+                                               default:
+                                                       goto exit;
+                                       }
+
+                                       PrevPageNumber = PageNumber;
+                               }
+
+                               if ((phyblk = BigEndianWORD(*(WORD *)(PageBuffer + (EntryOffset % MS_BYTES_PER_PAGE)))) < 0x0fff)
+                                       MS_LibSetInitialErrorBlock(us, phyblk);
+
+                               EntryOffset += 2;
+                               EntrySize -= 2;
+                       }
+               }
+               else if (i == 1)
+               {  // CIS/IDI
+                       MemStickBootBlockIDI  *idi;
+
+                       if (SysEntry->entry[i].bType != MS_SYSENT_TYPE_CIS_IDI)
+                               goto exit;
+
+                       switch (MS_ReaderReadPage(us, PhyBlock, (BYTE)(EntryOffset / MS_BYTES_PER_PAGE + 1), (DWORD *)PageBuffer, &ExtraData))
+                       {
+                               case MS_STATUS_SUCCESS:
+                                       break;
+                               case MS_STATUS_WRITE_PROTECT:
+                               case MS_ERROR_FLASH_READ:
+                               case MS_STATUS_ERROR:
+                               default:
+                                       goto exit;
+                       }
+
+                       idi = &((MemStickBootBlockCIS_IDI *)(PageBuffer + (EntryOffset % MS_BYTES_PER_PAGE)))->idi.idi;
+                       if (LittleEndianWORD(idi->wIDIgeneralConfiguration) != MS_IDI_GENERAL_CONF)
+                               goto exit;
+
+                       us->MS_Lib.BytesPerSector = LittleEndianWORD(idi->wIDIbytesPerSector);
+                       if (us->MS_Lib.BytesPerSector != MS_BYTES_PER_PAGE)
+                               goto exit;
+               }
+       } // End for ..
+
+       result = 0;
+
+exit:
+       if (result)             MS_LibFreeLogicalMap(us);
+       if (PageBuffer) kfree(PageBuffer);
+
+       result = 0;
+       return result;
+}
+
+//----- MS_LibAllocLogicalMap() --------------------------------------
+int MS_LibAllocLogicalMap(struct us_data *us)
+{
+       DWORD  i;
+
+
+       us->MS_Lib.Phy2LogMap = (WORD *)kmalloc(us->MS_Lib.NumberOfPhyBlock * sizeof(WORD), GFP_KERNEL);
+       us->MS_Lib.Log2PhyMap = (WORD *)kmalloc(us->MS_Lib.NumberOfLogBlock * sizeof(WORD), GFP_KERNEL);
+
+       if ((us->MS_Lib.Phy2LogMap == NULL) || (us->MS_Lib.Log2PhyMap == NULL))
+       {
+               MS_LibFreeLogicalMap(us);
+               return (DWORD)-1;
+       }
+
+       for (i = 0; i < us->MS_Lib.NumberOfPhyBlock; i++)
+               us->MS_Lib.Phy2LogMap[i] = MS_LB_NOT_USED;
+
+       for (i = 0; i < us->MS_Lib.NumberOfLogBlock; i++)
+       us->MS_Lib.Log2PhyMap[i] = MS_LB_NOT_USED;
+
+       return 0;
+}
+
+//----- MS_LibSetBootBlockMark() -------------------------------------
+int MS_LibSetBootBlockMark(struct us_data *us, WORD phyblk)
+{
+    return MS_LibSetLogicalBlockMark(us, phyblk, MS_LB_BOOT_BLOCK);
+}
+
+//----- MS_LibSetLogicalBlockMark() ----------------------------------
+int MS_LibSetLogicalBlockMark(struct us_data *us, WORD phyblk, WORD mark)
+{
+    if (phyblk >= us->MS_Lib.NumberOfPhyBlock)
+        return (DWORD)-1;
+
+    us->MS_Lib.Phy2LogMap[phyblk] = mark;
+
+    return 0;
+}
+
+//----- MS_LibSetInitialErrorBlock() ---------------------------------
+int MS_LibSetInitialErrorBlock(struct us_data *us, WORD phyblk)
+{
+    return MS_LibSetLogicalBlockMark(us, phyblk, MS_LB_INITIAL_ERROR);
+}
+
+//----- MS_LibScanLogicalBlockNumber() -------------------------------
+int MS_LibScanLogicalBlockNumber(struct us_data *us, WORD btBlk1st)
+{
+       WORD                    PhyBlock, newblk, i;
+       WORD                    LogStart, LogEnde;
+       MS_LibTypeExtdat        extdat;
+       BYTE                    buf[0x200];
+       DWORD                   count=0, index=0;
+
+       for (PhyBlock = 0; PhyBlock < us->MS_Lib.NumberOfPhyBlock;)
+       {
+               MS_LibPhy2LogRange(PhyBlock, &LogStart, &LogEnde);
+
+               for (i=0; i<MS_PHYSICAL_BLOCKS_PER_SEGMENT; i++, PhyBlock++)
+               {
+                       switch (MS_LibConv2Logical(us, PhyBlock))
+                       {
+                               case MS_STATUS_ERROR:
+                                       continue;
+                               default:
+                                       break;
+                       }
+
+                       if (count == PhyBlock)
+                       {
+                               MS_LibReadExtraBlock(us, PhyBlock, 0, 0x80, &buf);
+                               count += 0x80;
+                       }
+                       index = (PhyBlock % 0x80) * 4;
+
+                       extdat.ovrflg = buf[index];
+                       extdat.mngflg = buf[index+1];
+                       extdat.logadr = MemStickLogAddr(buf[index+2], buf[index+3]);
+
+                       if ((extdat.ovrflg & MS_REG_OVR_BKST) != MS_REG_OVR_BKST_OK)
+                       {
+                               MS_LibSetAcquiredErrorBlock(us, PhyBlock);
+                               continue;
+                       }
+
+                       if ((extdat.mngflg & MS_REG_MNG_ATFLG) == MS_REG_MNG_ATFLG_ATTBL)
+                       {
+                               MS_LibErasePhyBlock(us, PhyBlock);
+                               continue;
+                       }
+
+                       if (extdat.logadr != MS_LB_NOT_USED)
+                       {
+                               if ((extdat.logadr < LogStart) || (LogEnde <= extdat.logadr))
+                               {
+                                       MS_LibErasePhyBlock(us, PhyBlock);
+                                       continue;
+                               }
+
+                               if ((newblk = MS_LibConv2Physical(us, extdat.logadr)) != MS_LB_NOT_USED)
+                               {
+                                       if (extdat.logadr==0)
+                                       {
+                                               MS_LibSetLogicalPair(us, extdat.logadr, PhyBlock);
+                                               if ( MS_LibCheckDisableBlock(us, btBlk1st) )
+                                               {
+                                                       MS_LibSetLogicalPair(us, extdat.logadr, newblk);
+                                                       continue;
+                                               }
+                                       }
+
+                                       MS_LibReadExtra(us, newblk, 0, &extdat);
+                                       if ((extdat.ovrflg & MS_REG_OVR_UDST) == MS_REG_OVR_UDST_UPDATING)
+                                       {
+                                               MS_LibErasePhyBlock(us, PhyBlock);
+                                               continue;
+                                       }
+                                       else
+                                               MS_LibErasePhyBlock(us, newblk);
+                               }
+
+                               MS_LibSetLogicalPair(us, extdat.logadr, PhyBlock);
+                       }
+               }
+       } //End for ...
+
+       return MS_STATUS_SUCCESS;
+}
+
+//----- MS_LibAllocWriteBuf() ----------------------------------------
+int MS_LibAllocWriteBuf(struct us_data *us)
+{
+       us->MS_Lib.wrtblk = (WORD)-1;
+
+       us->MS_Lib.blkpag = (BYTE *)kmalloc(us->MS_Lib.PagesPerBlock * us->MS_Lib.BytesPerSector, GFP_KERNEL);
+       us->MS_Lib.blkext = (MS_LibTypeExtdat *)kmalloc(us->MS_Lib.PagesPerBlock * sizeof(MS_LibTypeExtdat), GFP_KERNEL);
+
+       if ((us->MS_Lib.blkpag == NULL) || (us->MS_Lib.blkext == NULL))
+       {
+               MS_LibFreeWriteBuf(us);
+               return (DWORD)-1;
+       }
+
+       MS_LibClearWriteBuf(us);
+
+       return 0;
+}
+
+//----- MS_LibClearWriteBuf() ----------------------------------------
+void MS_LibClearWriteBuf(struct us_data *us)
+{
+       int i;
+
+       us->MS_Lib.wrtblk = (WORD)-1;
+       MS_LibClearPageMap(us);
+
+       if (us->MS_Lib.blkpag)
+               memset(us->MS_Lib.blkpag, 0xff, us->MS_Lib.PagesPerBlock * us->MS_Lib.BytesPerSector);
+
+       if (us->MS_Lib.blkext)
+       {
+               for (i = 0; i < us->MS_Lib.PagesPerBlock; i++)
+               {
+                       us->MS_Lib.blkext[i].status1 = MS_REG_ST1_DEFAULT;
+                       us->MS_Lib.blkext[i].ovrflg = MS_REG_OVR_DEFAULT;
+                       us->MS_Lib.blkext[i].mngflg = MS_REG_MNG_DEFAULT;
+                       us->MS_Lib.blkext[i].logadr = MS_LB_NOT_USED;
+               }
+       }
+}
+
+//----- MS_LibPhy2LogRange() -----------------------------------------
+void MS_LibPhy2LogRange(WORD PhyBlock, WORD *LogStart, WORD *LogEnde)
+{
+       PhyBlock /= MS_PHYSICAL_BLOCKS_PER_SEGMENT;
+
+       if (PhyBlock)
+       {
+               *LogStart = MS_LOGICAL_BLOCKS_IN_1ST_SEGMENT + (PhyBlock - 1) * MS_LOGICAL_BLOCKS_PER_SEGMENT;//496
+               *LogEnde = *LogStart + MS_LOGICAL_BLOCKS_PER_SEGMENT;//496
+       }
+       else
+       {
+               *LogStart = 0;
+               *LogEnde = MS_LOGICAL_BLOCKS_IN_1ST_SEGMENT;//494
+       }
+}
+
+//----- MS_LibReadExtraBlock() --------------------------------------------
+int MS_LibReadExtraBlock(struct us_data *us, DWORD PhyBlock, BYTE PageNum, BYTE blen, void *buf)
+{
+       struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *) us->iobuf;
+       int     result;
+
+       //printk("MS_LibReadExtraBlock --- PhyBlock = %x, PageNum = %x, blen = %x\n", PhyBlock, PageNum, blen);
+
+       // Read Extra Data
+       memset(bcb, 0, sizeof(struct bulk_cb_wrap));
+       bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
+       bcb->DataTransferLength = 0x4 * blen;
+       bcb->Flags                      = 0x80;
+       bcb->CDB[0]                     = 0xF1;
+       bcb->CDB[1]                     = 0x03;
+       bcb->CDB[5]                     = (BYTE)(PageNum);
+       bcb->CDB[4]                     = (BYTE)(PhyBlock);
+       bcb->CDB[3]                     = (BYTE)(PhyBlock>>8);
+       bcb->CDB[2]                     = (BYTE)(PhyBlock>>16);
+       bcb->CDB[6]                     = blen;
+
+       result = ENE_SendScsiCmd(us, FDIR_READ, buf, 0);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       return USB_STOR_TRANSPORT_GOOD;
+}
+
+//----- MS_LibReadExtra() --------------------------------------------
+int MS_LibReadExtra(struct us_data *us, DWORD PhyBlock, BYTE PageNum, MS_LibTypeExtdat *ExtraDat)
+{
+       struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *) us->iobuf;
+       int     result;
+       BYTE    ExtBuf[4];
+
+       //printk("MS_LibReadExtra --- PhyBlock = %x, PageNum = %x\n", PhyBlock, PageNum);
+       memset(bcb, 0, sizeof(struct bulk_cb_wrap));
+       bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
+       bcb->DataTransferLength = 0x4;
+       bcb->Flags                      = 0x80;
+       bcb->CDB[0]                     = 0xF1;
+       bcb->CDB[1]                     = 0x03;
+       bcb->CDB[5]                     = (BYTE)(PageNum);
+       bcb->CDB[4]                     = (BYTE)(PhyBlock);
+       bcb->CDB[3]                     = (BYTE)(PhyBlock>>8);
+       bcb->CDB[2]                     = (BYTE)(PhyBlock>>16);
+       bcb->CDB[6]                     = 0x01;
+       
+       result = ENE_SendScsiCmd(us, FDIR_READ, &ExtBuf, 0);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       ExtraDat->reserved = 0;
+       ExtraDat->intr     = 0x80;  // Not yet, waiting for fireware support
+       ExtraDat->status0  = 0x10;  // Not yet, waiting for fireware support
+       ExtraDat->status1  = 0x00;  // Not yet, waiting for fireware support
+       ExtraDat->ovrflg   = ExtBuf[0];
+       ExtraDat->mngflg   = ExtBuf[1];
+       ExtraDat->logadr   = MemStickLogAddr(ExtBuf[2], ExtBuf[3]);
+       
+       return USB_STOR_TRANSPORT_GOOD;
+}
+
+//----- MS_LibSetAcquiredErrorBlock() --------------------------------
+int MS_LibSetAcquiredErrorBlock(struct us_data *us, WORD phyblk)
+{
+       WORD log;
+
+       if (phyblk >= us->MS_Lib.NumberOfPhyBlock)
+               return (DWORD)-1;
+
+       if ((log = us->MS_Lib.Phy2LogMap[phyblk]) < us->MS_Lib.NumberOfLogBlock)
+               us->MS_Lib.Log2PhyMap[log] = MS_LB_NOT_USED;
+
+       if (us->MS_Lib.Phy2LogMap[phyblk] != MS_LB_INITIAL_ERROR)
+               us->MS_Lib.Phy2LogMap[phyblk] = MS_LB_ACQUIRED_ERROR;
+
+       return 0;
+}
+
+//----- MS_LibErasePhyBlock() ----------------------------------------
+int MS_LibErasePhyBlock(struct us_data *us, WORD phyblk)
+{
+       WORD  log;
+
+       if (phyblk >= us->MS_Lib.NumberOfPhyBlock)
+               return MS_STATUS_ERROR;
+
+       if ((log = us->MS_Lib.Phy2LogMap[phyblk]) < us->MS_Lib.NumberOfLogBlock)
+               us->MS_Lib.Log2PhyMap[log] = MS_LB_NOT_USED;
+
+       us->MS_Lib.Phy2LogMap[phyblk] = MS_LB_NOT_USED;
+
+       if (MS_LibIsWritable(us))
+       {
+               switch (MS_ReaderEraseBlock(us, phyblk))
+               {
+                       case MS_STATUS_SUCCESS:
+                               us->MS_Lib.Phy2LogMap[phyblk] = MS_LB_NOT_USED_ERASED;
+                               return MS_STATUS_SUCCESS;
+                       case MS_ERROR_FLASH_ERASE:
+                       case MS_STATUS_INT_ERROR :
+                               MS_LibErrorPhyBlock(us, phyblk);
+                               return MS_ERROR_FLASH_ERASE;
+                       case MS_STATUS_ERROR:
+                       default:
+                               MS_LibCtrlSet(us, MS_LIB_CTRL_RDONLY);
+                               MS_LibSetAcquiredErrorBlock(us, phyblk);
+                               return MS_STATUS_ERROR;
+               }
+       }
+
+       MS_LibSetAcquiredErrorBlock(us, phyblk);
+
+       return MS_STATUS_SUCCESS;
+}
+
+//----- MS_LibErrorPhyBlock() ----------------------------------------
+int MS_LibErrorPhyBlock(struct us_data *us, WORD phyblk)
+{
+    if (phyblk >= us->MS_Lib.NumberOfPhyBlock)
+        return MS_STATUS_ERROR;
+
+    MS_LibSetAcquiredErrorBlock(us, phyblk);
+
+    if (MS_LibIsWritable(us))
+        return MS_LibOverwriteExtra(us, phyblk, 0, (BYTE)(~MS_REG_OVR_BKST));
+
+
+    return MS_STATUS_SUCCESS;
+}
+
+//----- MS_LibOverwriteExtra() ---------------------------------------
+int MS_LibOverwriteExtra(struct us_data *us, DWORD PhyBlockAddr, BYTE PageNum, BYTE OverwriteFlag)
+{
+       struct bulk_cb_wrap *bcb = (struct bulk_cb_wrap *) us->iobuf;
+       int     result;
+
+       //printk("MS --- MS_LibOverwriteExtra,  PhyBlockAddr = %x, PageNum = %x\n", PhyBlockAddr, PageNum);
+       result = ENE_LoadBinCode(us, MS_RW_PATTERN);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       memset(bcb, 0, sizeof(struct bulk_cb_wrap));
+       bcb->Signature = cpu_to_le32(US_BULK_CB_SIGN);
+       bcb->DataTransferLength = 0x4;
+       bcb->Flags                      = 0x80;
+       bcb->CDB[0]                     = 0xF2;
+       bcb->CDB[1]                     = 0x05;
+       bcb->CDB[5]                     = (BYTE)(PageNum);
+       bcb->CDB[4]                     = (BYTE)(PhyBlockAddr);
+       bcb->CDB[3]                     = (BYTE)(PhyBlockAddr>>8);
+       bcb->CDB[2]                     = (BYTE)(PhyBlockAddr>>16);
+       bcb->CDB[6]                     = OverwriteFlag;
+       bcb->CDB[7]                     = 0xFF;
+       bcb->CDB[8]                     = 0xFF;
+       bcb->CDB[9]                     = 0xFF;
+       
+       result = ENE_SendScsiCmd(us, FDIR_READ, NULL, 0);
+       if (result != USB_STOR_XFER_GOOD)
+               return USB_STOR_TRANSPORT_ERROR;
+
+       return USB_STOR_TRANSPORT_GOOD;
+}
+
+//----- MS_LibForceSetLogicalPair() ----------------------------------
+int MS_LibForceSetLogicalPair(struct us_data *us, WORD logblk, WORD phyblk)
+{
+       if (logblk == MS_LB_NOT_USED)
+               return 0;
+
+       if ((logblk >= us->MS_Lib.NumberOfLogBlock) || (phyblk >= us->MS_Lib.NumberOfPhyBlock))
+               return (DWORD)-1;
+
+       us->MS_Lib.Phy2LogMap[phyblk] = logblk;
+       us->MS_Lib.Log2PhyMap[logblk] = phyblk;
+
+       return 0;
+}
+
+//----- MS_LibSetLogicalPair() ---------------------------------------
+int MS_LibSetLogicalPair(struct us_data *us, WORD logblk, WORD phyblk)
+{
+       if ((logblk >= us->MS_Lib.NumberOfLogBlock) || (phyblk >= us->MS_Lib.NumberOfPhyBlock))
+               return (DWORD)-1;
+
+       us->MS_Lib.Phy2LogMap[phyblk] = logblk;
+       us->MS_Lib.Log2PhyMap[logblk] = phyblk;
+
+       return 0;
+}
+
+//----- MS_CountFreeBlock() ------------------------------------------
+int MS_CountFreeBlock(struct us_data *us, WORD PhyBlock)
+{
+       DWORD Ende, Count;
+
+       Ende = PhyBlock + MS_PHYSICAL_BLOCKS_PER_SEGMENT;
+       for (Count = 0; PhyBlock < Ende; PhyBlock++)
+       {
+               switch (us->MS_Lib.Phy2LogMap[PhyBlock])
+               {
+                       case MS_LB_NOT_USED:
+                       case MS_LB_NOT_USED_ERASED:
+                               Count++;
+                       default:
+                               break;
+               }
+       }
+
+       return Count;
+}
+
+//----- MS_LibSearchBlockFromPhysical() ------------------------------
+int MS_LibSearchBlockFromPhysical(struct us_data *us, WORD phyblk)
+{
+       WORD                    Newblk;
+       WORD                    blk;
+       MS_LibTypeExtdat        extdat;
+
+       if (phyblk >= us->MS_Lib.NumberOfPhyBlock)
+               return MS_LB_ERROR;
+
+       for (blk = phyblk + 1; blk != phyblk; blk++)
+       {
+               if ((blk & MS_PHYSICAL_BLOCKS_PER_SEGMENT_MASK) == 0)
+                       blk -= MS_PHYSICAL_BLOCKS_PER_SEGMENT;
+
+               Newblk = us->MS_Lib.Phy2LogMap[blk];
+               if (us->MS_Lib.Phy2LogMap[blk] == MS_LB_NOT_USED_ERASED)
+                       return blk;
+               else if (us->MS_Lib.Phy2LogMap[blk] == MS_LB_NOT_USED)
+               {
+                       switch (MS_LibReadExtra(us, blk, 0, &extdat))
+                       {
+                               case MS_STATUS_SUCCESS :
+                               case MS_STATUS_SUCCESS_WITH_ECC:
+                                       break;
+                               case MS_NOCARD_ERROR:
+                                       return MS_NOCARD_ERROR;
+                               case MS_STATUS_INT_ERROR:
+                                       return MS_LB_ERROR;
+                               case MS_ERROR_FLASH_READ:
+                               default:
+                                       MS_LibSetAcquiredErrorBlock(us, blk);     // MS_LibErrorPhyBlock(fdoExt, blk);
+                                       continue;
+                       } // End switch
+
+                       if ((extdat.ovrflg & MS_REG_OVR_BKST) != MS_REG_OVR_BKST_OK)
+                       {
+                               MS_LibSetAcquiredErrorBlock(us, blk);
+                               continue;
+                       }
+
+                       switch (MS_LibErasePhyBlock(us, blk))
+                       {
+                               case MS_STATUS_SUCCESS:
+                                       return blk;
+                               case MS_STATUS_ERROR:
+                                       return MS_LB_ERROR;
+                               case MS_ERROR_FLASH_ERASE:
+                               default:
+                                       MS_LibErrorPhyBlock(us, blk);
+                                       break;
+                       }
+               }
+       } // End for
+
+       return MS_LB_ERROR;
+}
+
+//----- MS_LibSearchBlockFromLogical() -------------------------------
+int MS_LibSearchBlockFromLogical(struct us_data *us, WORD logblk)
+{
+       WORD phyblk;
+
+       if ((phyblk=MS_LibConv2Physical(us, logblk)) >= MS_LB_ERROR)
+       {
+               if (logblk >= us->MS_Lib.NumberOfLogBlock)
+                       return MS_LB_ERROR;
+
+               phyblk = (logblk + MS_NUMBER_OF_BOOT_BLOCK) / MS_LOGICAL_BLOCKS_PER_SEGMENT;
+               phyblk *= MS_PHYSICAL_BLOCKS_PER_SEGMENT;
+               phyblk += MS_PHYSICAL_BLOCKS_PER_SEGMENT - 1;
+       }
+
+       return MS_LibSearchBlockFromPhysical(us, phyblk);
+}
+
+