#ifndef MS_INCD
#define MS_INCD

#include <linux/blkdev.h>
#include "common.h"

/* MemoryStick Register */
/* Status Register 0 */
#define MS_REG_ST0_MB                           0x80    /* media busy */
#define MS_REG_ST0_FB0                          0x40    /* flush busy 0 */
#define MS_REG_ST0_BE                           0x20    /* buffer empty */
#define MS_REG_ST0_BF                           0x10    /* buffer full */
#define MS_REG_ST0_SL                           0x02    /* sleep */
#define MS_REG_ST0_WP                           0x01    /* write protected */
#define MS_REG_ST0_WP_ON                        MS_REG_ST0_WP
#define MS_REG_ST0_WP_OFF                       0x00

/* Status Register 1 */
#define MS_REG_ST1_MB		0x80    /* media busy */
#define MS_REG_ST1_FB1		0x40    /* flush busy 1 */
#define MS_REG_ST1_DTER		0x20    /* error on data(corrected) */
#define MS_REG_ST1_UCDT		0x10    /* unable to correct data */
#define MS_REG_ST1_EXER		0x08    /* error on extra(corrected) */
#define MS_REG_ST1_UCEX		0x04    /* unable to correct extra */
#define MS_REG_ST1_FGER		0x02    /* error on overwrite flag(corrected) */
#define MS_REG_ST1_UCFG		0x01    /* unable to correct overwrite flag */
#define MS_REG_ST1_DEFAULT	(MS_REG_ST1_MB   | MS_REG_ST1_FB1  | \
				MS_REG_ST1_DTER | MS_REG_ST1_UCDT | \
				MS_REG_ST1_EXER | MS_REG_ST1_UCEX | \
				MS_REG_ST1_FGER | MS_REG_ST1_UCFG)

/* System Parameter */
#define MS_REG_SYSPAR_BAMD		0x80	/* block address mode */
#define MS_REG_SYSPAR_BAND_LINEAR	MS_REG_SYSPAR_BAMD  /*   linear mode */
#define MS_REG_SYSPAR_BAND_CHIP		0x00	/*  chip mode */
#define MS_REG_SYSPAR_ATEN		0x40	/* attribute ROM enable */
#define MS_REG_SYSPAR_ATEN_ENABLE	MS_REG_SYSPAR_ATEN	/*  enable */
#define MS_REG_SYSPAR_ATEN_DISABLE	0x00	/*  disable */
#define MS_REG_SYSPAR_RESERVED                  0x2f

/* Command Parameter */
#define MS_REG_CMDPAR_CP2                       0x80
#define MS_REG_CMDPAR_CP1                       0x40
#define MS_REG_CMDPAR_CP0                       0x20
#define MS_REG_CMDPAR_BLOCK_ACCESS              0
#define MS_REG_CMDPAR_PAGE_ACCESS               MS_REG_CMDPAR_CP0
#define MS_REG_CMDPAR_EXTRA_DATA                MS_REG_CMDPAR_CP1
#define MS_REG_CMDPAR_OVERWRITE                 MS_REG_CMDPAR_CP2
#define MS_REG_CMDPAR_RESERVED                  0x1f

/* Overwrite Area */
#define MS_REG_OVR_BKST		0x80            /* block status */
#define MS_REG_OVR_BKST_OK                      MS_REG_OVR_BKST     /* OK */
#define MS_REG_OVR_BKST_NG                      0x00            /* NG */
#define MS_REG_OVR_PGST0	0x40            /* page status */
#define MS_REG_OVR_PGST1                        0x20
#define MS_REG_OVR_PGST_MASK	(MS_REG_OVR_PGST0 | MS_REG_OVR_PGST1)
#define MS_REG_OVR_PGST_OK	(MS_REG_OVR_PGST0 | MS_REG_OVR_PGST1) /* OK */
#define MS_REG_OVR_PGST_NG	MS_REG_OVR_PGST1                      /* NG */
#define MS_REG_OVR_PGST_DATA_ERROR              0x00        /* data error */
#define MS_REG_OVR_UDST                         0x10        /* update status */
#define MS_REG_OVR_UDST_UPDATING                0x00        /* updating */
#define MS_REG_OVR_UDST_NO_UPDATE               MS_REG_OVR_UDST
#define MS_REG_OVR_RESERVED                     0x08
#define MS_REG_OVR_DEFAULT                      (MS_REG_OVR_BKST_OK |      \
						MS_REG_OVR_PGST_OK |      \
						MS_REG_OVR_UDST_NO_UPDATE |   \
						MS_REG_OVR_RESERVED)
/* Management Flag */
#define MS_REG_MNG_SCMS0	0x20    /* serial copy management system */
#define MS_REG_MNG_SCMS1                        0x10
#define MS_REG_MNG_SCMS_MASK		(MS_REG_MNG_SCMS0 | MS_REG_MNG_SCMS1)
#define MS_REG_MNG_SCMS_COPY_OK		(MS_REG_MNG_SCMS0 | MS_REG_MNG_SCMS1)
#define MS_REG_MNG_SCMS_ONE_COPY                MS_REG_MNG_SCMS1
#define MS_REG_MNG_SCMS_NO_COPY                 0x00
#define MS_REG_MNG_ATFLG	0x08	/* address transfer table flag */
#define MS_REG_MNG_ATFLG_OTHER                  MS_REG_MNG_ATFLG    /* other */
#define MS_REG_MNG_ATFLG_ATTBL		0x00	/* address transfer table */
#define MS_REG_MNG_SYSFLG                       0x04    /* system flag */
#define MS_REG_MNG_SYSFLG_USER		MS_REG_MNG_SYSFLG   /* user block */
#define MS_REG_MNG_SYSFLG_BOOT                  0x00    /* system block */
#define MS_REG_MNG_RESERVED                     0xc3
#define MS_REG_MNG_DEFAULT		(MS_REG_MNG_SCMS_COPY_OK |	\
					 MS_REG_MNG_ATFLG_OTHER |	\
					 MS_REG_MNG_SYSFLG_USER |	\
					 MS_REG_MNG_RESERVED)

/* Error codes */
#define MS_STATUS_SUCCESS                       0x0000
#define MS_ERROR_OUT_OF_SPACE                   0x0103
#define MS_STATUS_WRITE_PROTECT                 0x0106
#define MS_ERROR_READ_DATA                      0x8002
#define MS_ERROR_FLASH_READ                     0x8003
#define MS_ERROR_FLASH_WRITE                    0x8004
#define MS_ERROR_FLASH_ERASE                    0x8005
#define MS_ERROR_FLASH_COPY                     0x8006

#define MS_STATUS_ERROR                         0xfffe
#define MS_FIFO_ERROR                           0xfffd
#define MS_UNDEFINED_ERROR                      0xfffc
#define MS_KETIMEOUT_ERROR                      0xfffb
#define MS_STATUS_INT_ERROR                     0xfffa
#define MS_NO_MEMORY_ERROR                      0xfff9
#define MS_NOCARD_ERROR                         0xfff8
#define MS_LB_NOT_USED                          0xffff
#define MS_LB_ERROR                             0xfff0
#define MS_LB_BOOT_BLOCK                        0xfff1
#define MS_LB_INITIAL_ERROR                     0xfff2
#define MS_STATUS_SUCCESS_WITH_ECC              0xfff3
#define MS_LB_ACQUIRED_ERROR                    0xfff4
#define MS_LB_NOT_USED_ERASED                   0xfff5

#define MS_LibConv2Physical(pdx, LogBlock) \
	(((LogBlock) >= (pdx)->MS_Lib.NumberOfLogBlock) ? \
	MS_STATUS_ERROR : (pdx)->MS_Lib.Log2PhyMap[LogBlock])
#define MS_LibConv2Logical(pdx, PhyBlock) \
	(((PhyBlock) >= (pdx)->MS_Lib.NumberOfPhyBlock) ? \
	MS_STATUS_ERROR : (pdx)->MS_Lib.Phy2LogMap[PhyBlock])
	/*dphy->log table */

#define MS_LIB_CTRL_RDONLY                      0
#define MS_LIB_CTRL_WRPROTECT                   1
#define MS_LibCtrlCheck(pdx, Flag)	((pdx)->MS_Lib.flags & (1 << (Flag)))

#define MS_LibCtrlSet(pdx, Flag)	((pdx)->MS_Lib.flags |= (1 << (Flag)))
#define MS_LibCtrlReset(pdx, Flag)	((pdx)->MS_Lib.flags &= ~(1 << (Flag)))
#define MS_LibIsWritable(pdx) \
	((MS_LibCtrlCheck((pdx), MS_LIB_CTRL_RDONLY) == 0) && \
	(MS_LibCtrlCheck(pdx, MS_LIB_CTRL_WRPROTECT) == 0))

#define MS_MAX_PAGES_PER_BLOCK                  32
#define MS_LIB_BITS_PER_BYTE                    8

#define MS_LibPageMapIdx(n) ((n) / MS_LIB_BITS_PER_BYTE)
#define MS_LibPageMapBit(n) (1 << ((n) % MS_LIB_BITS_PER_BYTE))
#define MS_LibCheckPageMapBit(pdx, n) \
	((pdx)->MS_Lib.pagemap[MS_LibPageMapIdx(n)] & MS_LibPageMapBit(n))
#define MS_LibSetPageMapBit(pdx, n) \
	((pdx)->MS_Lib.pagemap[MS_LibPageMapIdx(n)] |= MS_LibPageMapBit(n))
#define MS_LibResetPageMapBit(pdx, n) \
	((pdx)->MS_Lib.pagemap[MS_LibPageMapIdx(n)] &= ~MS_LibPageMapBit(n))
#define MS_LibClearPageMap(pdx) \
	memset((pdx)->MS_Lib.pagemap, 0, sizeof((pdx)->MS_Lib.pagemap))


#define MemStickLogAddr(logadr1, logadr0) \
	((((WORD)(logadr1)) << 8) | (logadr0))

#define MS_BYTES_PER_PAGE                       512

#define MS_MAX_INITIAL_ERROR_BLOCKS             10
#define MS_NUMBER_OF_PAGES_FOR_BOOT_BLOCK       3
#define MS_NUMBER_OF_PAGES_FOR_LPCTBL           2

#define MS_NUMBER_OF_BOOT_BLOCK                 2
#define MS_NUMBER_OF_SYSTEM_BLOCK               4
#define MS_LOGICAL_BLOCKS_PER_SEGMENT           496
#define MS_LOGICAL_BLOCKS_IN_1ST_SEGMENT        494
#define MS_PHYSICAL_BLOCKS_PER_SEGMENT          0x200   /* 512 */
#define MS_PHYSICAL_BLOCKS_PER_SEGMENT_MASK     0x1ff

#define MS_SECTOR_SIZE                          512
#define MBR_SIGNATURE                           0xAA55
#define PBR_SIGNATURE                           0xAA55

#define PARTITION_FAT_12                        1
#define PARTITION_FAT_16                        2

#define MS_BOOT_BLOCK_ID                        0x0001
#define MS_BOOT_BLOCK_FORMAT_VERSION            0x0100
#define MS_BOOT_BLOCK_DATA_ENTRIES              2

#define MS_SYSINF_MSCLASS_TYPE_1                1
#define MS_SYSINF_CARDTYPE_RDONLY               1
#define MS_SYSINF_CARDTYPE_RDWR                 2
#define MS_SYSINF_CARDTYPE_HYBRID               3
#define MS_SYSINF_SECURITY                      0x01
#define MS_SYSINF_SECURITY_NO_SUPPORT           MS_SYSINF_SECURITY
#define MS_SYSINF_SECURITY_SUPPORT              0
#define MS_SYSINF_FORMAT_MAT                    0   /* ? */
#define MS_SYSINF_FORMAT_FAT                    1
#define MS_SYSINF_USAGE_GENERAL                 0
#define MS_SYSINF_PAGE_SIZE                     MS_BYTES_PER_PAGE /* fixed */
#define MS_SYSINF_RESERVED1                     1
#define MS_SYSINF_RESERVED2                     1

#define MS_SYSENT_TYPE_INVALID_BLOCK            0x01
#define MS_SYSENT_TYPE_CIS_IDI                  0x0a    /* CIS/IDI */

#define SIZE_OF_KIRO                            1024

/* BOOT BLOCK */
#define MS_NUMBER_OF_SYSTEM_ENTRY               4

/*
 * MemStickRegisters
 */
/* Status registers (16 bytes) */
typedef struct {
	BYTE Reserved0;		/* 00 */
	BYTE INTRegister;	/* 01 */
	BYTE StatusRegister0;	/* 02 */
	BYTE StatusRegister1;	/* 03 */
	BYTE Reserved1[12];	/* 04-0F */
} MemStickStatusRegisters;

/* Parameter registers (6 bytes) */
typedef struct {
	BYTE SystemParameter;	/* 10 */
	BYTE BlockAddress2;	/* 11 */
	BYTE BlockAddress1;	/* 12 */
	BYTE BlockAddress0;	/* 13 */
	BYTE CMDParameter;	/* 14 */
	BYTE PageAddress;	/* 15 */
} MemStickParameterRegisters;

/* Extra registers (9 bytes) */
typedef struct {
	BYTE OverwriteFlag;	/* 16 */
	BYTE ManagementFlag;	/* 17 */
	BYTE LogicalAddress1;	/* 18 */
	BYTE LogicalAddress0;	/* 19 */
	BYTE ReservedArea[5];	/* 1A-1E */
} MemStickExtraDataRegisters;

/* All registers in Memory Stick (32 bytes, includes 1 byte padding) */
typedef struct {
	MemStickStatusRegisters status;
	MemStickParameterRegisters param;
	MemStickExtraDataRegisters extra;
	BYTE padding;
} MemStickRegisters, *PMemStickRegisters;

/*
 * MemStickBootBlockPage0
 */
typedef struct {
	WORD wBlockID;
	WORD wFormatVersion;
	BYTE bReserved1[184];
	BYTE bNumberOfDataEntry;
	BYTE bReserved2[179];
} MemStickBootBlockHeader;

typedef struct {
	DWORD dwStart;
	DWORD dwSize;
	BYTE bType;
	BYTE bReserved[3];
} MemStickBootBlockSysEntRec;

typedef struct {
	MemStickBootBlockSysEntRec entry[MS_NUMBER_OF_SYSTEM_ENTRY];
} MemStickBootBlockSysEnt;

typedef struct {
	BYTE bMsClass;		/* must be 1 */
	BYTE bCardType;		/* see below */
	WORD wBlockSize;	/* n KB */
	WORD wBlockNumber;	/* number of physical block */
	WORD wTotalBlockNumber;	/* number of logical block */
	WORD wPageSize;		/* must be 0x200 */
	BYTE bExtraSize;	/* 0x10 */
	BYTE bSecuritySupport;
	BYTE bAssemblyDate[8];
	BYTE bFactoryArea[4];
	BYTE bAssemblyMakerCode;
	BYTE bAssemblyMachineCode[3];
	WORD wMemoryMakerCode;
	WORD wMemoryDeviceCode;
	WORD wMemorySize;
	BYTE bReserved1;
	BYTE bReserved2;
	BYTE bVCC;
	BYTE bVPP;
	WORD wControllerChipNumber;
	WORD wControllerFunction;	/* New MS */
	BYTE bReserved3[9];		/* New MS */
	BYTE bParallelSupport;		/* New MS */
	WORD wFormatValue;		/* New MS */
	BYTE bFormatType;
	BYTE bUsage;
	BYTE bDeviceType;
	BYTE bReserved4[22];
	BYTE bFUValue3;
	BYTE bFUValue4;
	BYTE bReserved5[15];
} MemStickBootBlockSysInf;

typedef struct {
	MemStickBootBlockHeader header;
	MemStickBootBlockSysEnt sysent;
	MemStickBootBlockSysInf sysinf;
} MemStickBootBlockPage0;

/*
 * MemStickBootBlockCIS_IDI
 */
typedef struct {
	BYTE bCistplDEVICE[6];            /* 0 */
	BYTE bCistplDEVICE0C[6];          /* 6 */
	BYTE bCistplJEDECC[4];            /* 12 */
	BYTE bCistplMANFID[6];            /* 16 */
	BYTE bCistplVER1[32];             /* 22 */
	BYTE bCistplFUNCID[4];            /* 54 */
	BYTE bCistplFUNCE0[4];            /* 58 */
	BYTE bCistplFUNCE1[5];            /* 62 */
	BYTE bCistplCONF[7];              /* 67 */
	BYTE bCistplCFTBLENT0[10];        /* 74 */
	BYTE bCistplCFTBLENT1[8];         /* 84 */
	BYTE bCistplCFTBLENT2[12];        /* 92 */
	BYTE bCistplCFTBLENT3[8];         /* 104 */
	BYTE bCistplCFTBLENT4[17];        /* 112 */
	BYTE bCistplCFTBLENT5[8];         /* 129 */
	BYTE bCistplCFTBLENT6[17];        /* 137 */
	BYTE bCistplCFTBLENT7[8];         /* 154 */
	BYTE bCistplNOLINK[3];            /* 162 */
} MemStickBootBlockCIS;

typedef struct {
#define MS_IDI_GENERAL_CONF         0x848A
	WORD wIDIgeneralConfiguration;     /* 0 */
	WORD wIDInumberOfCylinder;         /* 1 */
	WORD wIDIreserved0;                /* 2 */
	WORD wIDInumberOfHead;             /* 3 */
	WORD wIDIbytesPerTrack;            /* 4 */
	WORD wIDIbytesPerSector;           /* 5 */
	WORD wIDIsectorsPerTrack;          /* 6 */
	WORD wIDItotalSectors[2];          /* 7-8  high,low */
	WORD wIDIreserved1[11];            /* 9-19 */
	WORD wIDIbufferType;               /* 20 */
	WORD wIDIbufferSize;               /* 21 */
	WORD wIDIlongCmdECC;               /* 22 */
	WORD wIDIfirmVersion[4];           /* 23-26 */
	WORD wIDImodelName[20];            /* 27-46 */
	WORD wIDIreserved2;                /* 47 */
	WORD wIDIlongWordSupported;        /* 48 */
	WORD wIDIdmaSupported;             /* 49 */
	WORD wIDIreserved3;                /* 50 */
	WORD wIDIpioTiming;                /* 51 */
	WORD wIDIdmaTiming;                /* 52 */
	WORD wIDItransferParameter;        /* 53 */
	WORD wIDIformattedCylinder;        /* 54 */
	WORD wIDIformattedHead;            /* 55 */
	WORD wIDIformattedSectorsPerTrack; /* 56 */
	WORD wIDIformattedTotalSectors[2]; /* 57-58 */
	WORD wIDImultiSector;              /* 59 */
	WORD wIDIlbaSectors[2];            /* 60-61 */
	WORD wIDIsingleWordDMA;            /* 62 */
	WORD wIDImultiWordDMA;             /* 63 */
	WORD wIDIreserved4[192];           /* 64-255 */
} MemStickBootBlockIDI;

typedef struct {
	union {
		MemStickBootBlockCIS cis;
		BYTE dmy[256];
	} cis;

	union {
	MemStickBootBlockIDI idi;
	BYTE dmy[256];
	} idi;

} MemStickBootBlockCIS_IDI;

/*
 * MS_LibControl
 */
typedef struct {
	BYTE reserved;
	BYTE intr;
	BYTE status0;
	BYTE status1;
	BYTE ovrflg;
	BYTE mngflg;
	WORD logadr;
} MS_LibTypeExtdat;

typedef struct {
	DWORD flags;
	DWORD BytesPerSector;
	DWORD NumberOfCylinder;
	DWORD SectorsPerCylinder;
	WORD cardType;			/* R/W, RO, Hybrid */
	WORD blockSize;
	WORD PagesPerBlock;
	WORD NumberOfPhyBlock;
	WORD NumberOfLogBlock;
	WORD NumberOfSegment;
	WORD *Phy2LogMap;		/* phy2log table */
	WORD *Log2PhyMap;		/* log2phy table */
	WORD wrtblk;
	BYTE pagemap[(MS_MAX_PAGES_PER_BLOCK + (MS_LIB_BITS_PER_BYTE-1)) /
		     MS_LIB_BITS_PER_BYTE];
	BYTE *blkpag;
	MS_LibTypeExtdat *blkext;
	BYTE copybuf[512];
} MS_LibControl;

#endif