#include <IOKit/IOService.h>
#include <IOKit/platform/AppleMacIODevice.h>
#include <IOKit/storage/IOBlockStorageDriver.h>
extern "C" {
#define MACH_KERNEL
#include "bus_device.h"
}

class org_mklinux_iokit_swim3_driver : public IOService
{
OSDeclareDefaultStructors(org_mklinux_iokit_swim3_driver)
public:	
        AppleMacIODevice *DAG_Rules;
        struct bus_device *busdev;
        bool	media_present;
        caddr_t vaddr, paddr;
        caddr_t dma_vaddr, dma_paddr;
	virtual bool init(OSDictionary *dictionary = 0);
	virtual void free(void);	
	virtual IOService *probe(IOService *provider, SInt32 *score);
        virtual IOService * instantiateNub ( void );
        virtual bool createNub ( IOService * provider );		
	virtual bool start(IOService *provider);	
	virtual void stop(IOService *provider);
        virtual char *		getVendorString ( void );
        virtual char *		getProductString ( void );
        virtual char *		getRevisionString ( void );
        virtual char *		getAdditionalDeviceInfoString ( void );
        virtual IOReturn doAsyncReadWrite(IOMemoryDescriptor *buffer,
                                UInt32 block,UInt32 nblks,
                                IOStorageCompletion completion);
        virtual IOReturn doEjectMedia(void);
        virtual IOReturn doFormatMedia(UInt64 byteCapacity);
	virtual UInt32 doGetFormatCapacities(UInt64 * capacities,
                                UInt32 capacitiesMaxCount);
        virtual IOReturn doLockUnlockMedia(bool doLock);
        virtual IOReturn doSyncReadWrite(IOMemoryDescriptor *buffer,
                                UInt32 block,UInt32 nblks);
        virtual IOReturn doSynchronizeCache(void);
        virtual IOReturn reportBlockSize(UInt64 *blockSize);
        virtual IOReturn reportEjectability(bool *isEjectable);
        virtual IOReturn reportLockability(bool *isLockable);
        virtual IOReturn reportMaxReadTransfer(UInt64 blockSize,UInt64 *max);
        virtual IOReturn reportMaxValidBlock(UInt64 *maxBlock);
        virtual IOReturn reportMaxWriteTransfer(UInt64 blockSize,UInt64 *max);
        virtual IOReturn reportMediaState(bool *mediaPresent,bool *changedState);
        virtual IOReturn reportPollRequirements(bool *pollRequired,
                                bool *pollIsExpensive);
        virtual IOReturn reportRemovability(bool *isRemovable);
        virtual IOReturn reportWriteProtection(bool *isWriteProtected);

};

class org_mklinux_iokit_swim3_device : public IOBlockStorageDevice
{

        OSDeclareDefaultStructors ( org_mklinux_iokit_swim3_device )
    
protected:
    
        org_mklinux_iokit_swim3_driver *     fProvider;
    
    	virtual bool		attach ( IOService * provider );
	virtual void		detach ( IOService * provider );
        
public:

        virtual IOReturn	setProperties ( OSObject * properties );
	
	virtual IOReturn	doAsyncReadWrite ( 	IOMemoryDescriptor *	buffer,
											UInt32					block,
											UInt32					nblks,
											IOStorageCompletion		completion );

	virtual IOReturn	doSyncReadWrite ( 	IOMemoryDescriptor *	buffer,
											UInt32					block,
											UInt32					nblks );
	
    virtual IOReturn	doEjectMedia ( void );
	
    virtual IOReturn	doFormatMedia ( UInt64 byteCapacity );
	
    virtual UInt32		doGetFormatCapacities ( UInt64 *	capacities,
    											UInt32		capacitiesMaxCount ) const;
	
    virtual IOReturn	doLockUnlockMedia ( bool doLock );
	
    virtual IOReturn	doSynchronizeCache ( void );
        
    virtual char *		getVendorString ( void );
    
    virtual char *		getProductString ( void );
    
    virtual char *		getRevisionString ( void );
    
    virtual char *		getAdditionalDeviceInfoString ( void );
    
    virtual IOReturn	reportBlockSize ( UInt64 * blockSize );
    
    virtual IOReturn	reportEjectability ( bool * isEjectable );
    
    virtual IOReturn	reportLockability ( bool * isLockable );
    
    virtual IOReturn	reportMediaState ( bool * mediaPresent, bool * changed );
    
    virtual IOReturn	reportPollRequirements ( 	bool * pollIsRequired,
    												bool * pollIsExpensive );
    
    virtual IOReturn	reportMaxReadTransfer ( UInt64 blockSize, UInt64 * max );
    
    virtual IOReturn	reportMaxValidBlock ( UInt64 * maxBlock );
    
    virtual IOReturn	reportMaxWriteTransfer ( UInt64 blockSize, UInt64 * max );
    
    virtual IOReturn	reportRemovability ( bool * isRemovable );
    
    virtual IOReturn	reportWriteProtection ( bool * isWriteProtected );
    
};