Merge branch 'master' of master.kernel.org:/pub/scm/linux/kernel/git/davem/net-2.6
[safe/jmp/linux-2.6] / drivers / staging / hv / StorVsc.c
index a235613..2f7c425 100644 (file)
@@ -1,5 +1,4 @@
 /*
- *
  * Copyright (c) 2009, Microsoft Corporation.
  *
  * This program is free software; you can redistribute it and/or modify it
  * Authors:
  *   Haiyang Zhang <haiyangz@microsoft.com>
  *   Hank Janssen  <hjanssen@microsoft.com>
- *
  */
-
 #include <linux/kernel.h>
 #include <linux/string.h>
 #include <linux/mm.h>
 #include <linux/delay.h>
 #include "osd.h"
-#include "include/logging.h"
-#include "include/StorVscApi.h"
-#include "include/VmbusPacketFormat.h"
+#include "logging.h"
+#include "StorVscApi.h"
+#include "VmbusPacketFormat.h"
 #include "vstorage.h"
 
 
-
-/* #defines */
-
-
-
-/* Data types */
-
-
-typedef struct _STORVSC_REQUEST_EXTENSION {
-       /* LIST_ENTRY                                           ListEntry; */
+struct storvsc_request_extension {
+       /* LIST_ENTRY ListEntry; */
 
        struct hv_storvsc_request *Request;
        struct hv_device *Device;
@@ -49,15 +38,15 @@ typedef struct _STORVSC_REQUEST_EXTENSION {
        /* Synchronize the request/response if needed */
        struct osd_waitevent *WaitEvent;
 
-       VSTOR_PACKET                                    VStorPacket;
-} STORVSC_REQUEST_EXTENSION;
-
+       struct vstor_packet VStorPacket;
+};
 
 /* A storvsc device is a device object that contains a vmbus channel */
-typedef struct _STORVSC_DEVICE{
+struct storvsc_device {
        struct hv_device *Device;
 
-       atomic_t RefCount; /* 0 indicates the device is being destroyed */
+       /* 0 indicates the device is being destroyed */
+       atomic_t RefCount;
 
        atomic_t NumOutstandingRequests;
 
@@ -66,25 +55,20 @@ typedef struct _STORVSC_DEVICE{
         * controller. In reality, the pathid, targetid is always 0
         * and the port is set by us
         */
-       unsigned int                                            PortNumber;
-    unsigned char                                              PathId;
-    unsigned char                                              TargetId;
+       unsigned int PortNumber;
+       unsigned char PathId;
+       unsigned char TargetId;
 
-       /* LIST_ENTRY                                   OutstandingRequestList; */
-       /* HANDLE                                               OutstandingRequestLock; */
+       /* LIST_ENTRY OutstandingRequestList; */
+       /* HANDLE OutstandingRequestLock; */
 
        /* Used for vsc/vsp channel reset process */
-       STORVSC_REQUEST_EXTENSION       InitRequest;
-
-       STORVSC_REQUEST_EXTENSION       ResetRequest;
-
-} STORVSC_DEVICE;
-
-
+       struct storvsc_request_extension InitRequest;
+       struct storvsc_request_extension ResetRequest;
+};
 
-/* Globals */
 
-static const char* gDriverName="storvsc";
+static const char *gDriverName = "storvsc";
 
 /* {ba6163d9-04a1-4d29-b605-72e2ffb1dc7f} */
 static const struct hv_guid gStorVscDeviceType = {
@@ -95,64 +79,11 @@ static const struct hv_guid gStorVscDeviceType = {
 };
 
 
-/* Internal routines */
-
-static int
-StorVscOnDeviceAdd(
-       struct hv_device *Device,
-       void                    *AdditionalInfo
-       );
-
-static int
-StorVscOnDeviceRemove(
-       struct hv_device *Device
-       );
-
-static int
-StorVscOnIORequest(
-       struct hv_device *Device,
-       struct hv_storvsc_request *Request
-       );
-
-static int
-StorVscOnHostReset(
-       struct hv_device *Device
-       );
-
-static void
-StorVscOnCleanup(
-       struct hv_driver *Device
-       );
-
-static void
-StorVscOnChannelCallback(
-       void * Context
-       );
-
-static void
-StorVscOnIOCompletion(
-       struct hv_device *Device,
-       VSTOR_PACKET    *VStorPacket,
-       STORVSC_REQUEST_EXTENSION *RequestExt
-       );
-
-static void
-StorVscOnReceive(
-       struct hv_device *Device,
-       VSTOR_PACKET    *VStorPacket,
-       STORVSC_REQUEST_EXTENSION *RequestExt
-       );
-
-static int
-StorVscConnectToVsp(
-       struct hv_device *Device
-       );
-
-static inline STORVSC_DEVICE* AllocStorDevice(struct hv_device *Device)
+static inline struct storvsc_device *AllocStorDevice(struct hv_device *Device)
 {
-       STORVSC_DEVICE *storDevice;
+       struct storvsc_device *storDevice;
 
-       storDevice = kzalloc(sizeof(STORVSC_DEVICE), GFP_KERNEL);
+       storDevice = kzalloc(sizeof(struct storvsc_device), GFP_KERNEL);
        if (!storDevice)
                return NULL;
 
@@ -166,18 +97,18 @@ static inline STORVSC_DEVICE* AllocStorDevice(struct hv_device *Device)
        return storDevice;
 }
 
-static inline void FreeStorDevice(STORVSC_DEVICE *Device)
+static inline void FreeStorDevice(struct storvsc_device *Device)
 {
-       ASSERT( atomic_read(&Device->RefCount) == 0);
+       ASSERT(atomic_read(&Device->RefCount) == 0);
        kfree(Device);
 }
 
 /* Get the stordevice object iff exists and its refcount > 1 */
-static inline STORVSC_DEVICE* GetStorDevice(struct hv_device *Device)
+static inline struct storvsc_device *GetStorDevice(struct hv_device *Device)
 {
-       STORVSC_DEVICE *storDevice;
+       struct storvsc_device *storDevice;
 
-       storDevice = (STORVSC_DEVICE*)Device->Extension;
+       storDevice = (struct storvsc_device *)Device->Extension;
        if (storDevice && atomic_read(&storDevice->RefCount) > 1)
                atomic_inc(&storDevice->RefCount);
        else
@@ -187,11 +118,11 @@ static inline STORVSC_DEVICE* GetStorDevice(struct hv_device *Device)
 }
 
 /* Get the stordevice object iff exists and its refcount > 0 */
-static inline STORVSC_DEVICE* MustGetStorDevice(struct hv_device *Device)
+static inline struct storvsc_device *MustGetStorDevice(struct hv_device *Device)
 {
-       STORVSC_DEVICE *storDevice;
+       struct storvsc_device *storDevice;
 
-       storDevice = (STORVSC_DEVICE*)Device->Extension;
+       storDevice = (struct storvsc_device *)Device->Extension;
        if (storDevice && atomic_read(&storDevice->RefCount))
                atomic_inc(&storDevice->RefCount);
        else
@@ -202,9 +133,9 @@ static inline STORVSC_DEVICE* MustGetStorDevice(struct hv_device *Device)
 
 static inline void PutStorDevice(struct hv_device *Device)
 {
-       STORVSC_DEVICE *storDevice;
+       struct storvsc_device *storDevice;
 
-       storDevice = (STORVSC_DEVICE*)Device->Extension;
+       storDevice = (struct storvsc_device *)Device->Extension;
        ASSERT(storDevice);
 
        atomic_dec(&storDevice->RefCount);
@@ -212,165 +143,48 @@ static inline void PutStorDevice(struct hv_device *Device)
 }
 
 /* Drop ref count to 1 to effectively disable GetStorDevice() */
-static inline STORVSC_DEVICE* ReleaseStorDevice(struct hv_device *Device)
+static inline struct storvsc_device *ReleaseStorDevice(struct hv_device *Device)
 {
-       STORVSC_DEVICE *storDevice;
+       struct storvsc_device *storDevice;
 
-       storDevice = (STORVSC_DEVICE*)Device->Extension;
+       storDevice = (struct storvsc_device *)Device->Extension;
        ASSERT(storDevice);
 
        /* Busy wait until the ref drop to 2, then set it to 1 */
        while (atomic_cmpxchg(&storDevice->RefCount, 2, 1) != 2)
-       {
                udelay(100);
-       }
 
        return storDevice;
 }
 
 /* Drop ref count to 0. No one can use StorDevice object. */
-static inline STORVSC_DEVICE* FinalReleaseStorDevice(struct hv_device *Device)
+static inline struct storvsc_device *FinalReleaseStorDevice(
+                       struct hv_device *Device)
 {
-       STORVSC_DEVICE *storDevice;
+       struct storvsc_device *storDevice;
 
-       storDevice = (STORVSC_DEVICE*)Device->Extension;
+       storDevice = (struct storvsc_device *)Device->Extension;
        ASSERT(storDevice);
 
        /* Busy wait until the ref drop to 1, then set it to 0 */
        while (atomic_cmpxchg(&storDevice->RefCount, 1, 0) != 1)
-       {
                udelay(100);
-       }
 
        Device->Extension = NULL;
        return storDevice;
 }
 
-/*++;
-
-
-Name:
-       StorVscInitialize()
-
-Description:
-       Main entry point
-
---*/
-int
-StorVscInitialize(
-       struct hv_driver *Driver
-       )
-{
-       STORVSC_DRIVER_OBJECT* storDriver = (STORVSC_DRIVER_OBJECT*)Driver;
-       int ret=0;
-
-       DPRINT_ENTER(STORVSC);
-
-       DPRINT_DBG(STORVSC, "sizeof(STORVSC_REQUEST)=%zd sizeof(STORVSC_REQUEST_EXTENSION)=%zd sizeof(VSTOR_PACKET)=%zd, sizeof(VMSCSI_REQUEST)=%zd",
-               sizeof(struct hv_storvsc_request), sizeof(STORVSC_REQUEST_EXTENSION), sizeof(VSTOR_PACKET), sizeof(VMSCSI_REQUEST));
-
-       /* Make sure we are at least 2 pages since 1 page is used for control */
-       ASSERT(storDriver->RingBufferSize >= (PAGE_SIZE << 1));
-
-       Driver->name = gDriverName;
-       memcpy(&Driver->deviceType, &gStorVscDeviceType, sizeof(struct hv_guid));
-
-       storDriver->RequestExtSize                      = sizeof(STORVSC_REQUEST_EXTENSION);
-
-       /*
-        * Divide the ring buffer data size (which is 1 page less
-        * than the ring buffer size since that page is reserved for
-        * the ring buffer indices) by the max request size (which is
-        * VMBUS_CHANNEL_PACKET_MULITPAGE_BUFFER + VSTOR_PACKET + u64)
-        */
-       storDriver->MaxOutstandingRequestsPerChannel =
-               ((storDriver->RingBufferSize - PAGE_SIZE) / ALIGN_UP(MAX_MULTIPAGE_BUFFER_PACKET + sizeof(VSTOR_PACKET) + sizeof(u64),sizeof(u64)));
-
-       DPRINT_INFO(STORVSC, "max io %u, currently %u\n", storDriver->MaxOutstandingRequestsPerChannel, STORVSC_MAX_IO_REQUESTS);
-
-       /* Setup the dispatch table */
-       storDriver->Base.OnDeviceAdd                    = StorVscOnDeviceAdd;
-       storDriver->Base.OnDeviceRemove         = StorVscOnDeviceRemove;
-       storDriver->Base.OnCleanup                      = StorVscOnCleanup;
-
-       storDriver->OnIORequest                         = StorVscOnIORequest;
-       storDriver->OnHostReset                         = StorVscOnHostReset;
-
-       DPRINT_EXIT(STORVSC);
-
-       return ret;
-}
-
-/*++
-
-Name:
-       StorVscOnDeviceAdd()
-
-Description:
-       Callback when the device belonging to this driver is added
-
---*/
-static int
-StorVscOnDeviceAdd(
-       struct hv_device *Device,
-       void                    *AdditionalInfo
-       )
-{
-       int ret=0;
-       STORVSC_DEVICE *storDevice;
-       /* VMSTORAGE_CHANNEL_PROPERTIES *props; */
-       STORVSC_DEVICE_INFO *deviceInfo = (STORVSC_DEVICE_INFO*)AdditionalInfo;
-
-       DPRINT_ENTER(STORVSC);
-
-       storDevice = AllocStorDevice(Device);
-       if (!storDevice)
-       {
-               ret = -1;
-               goto Cleanup;
-       }
-
-       /* Save the channel properties to our storvsc channel */
-       /* props = (VMSTORAGE_CHANNEL_PROPERTIES*) channel->offerMsg.Offer.u.Standard.UserDefined; */
-
-       /* FIXME: */
-       /*
-        * If we support more than 1 scsi channel, we need to set the
-        * port number here to the scsi channel but how do we get the
-        * scsi channel prior to the bus scan
-        */
-
-       /* storChannel->PortNumber = 0;
-       storChannel->PathId = props->PathId;
-       storChannel->TargetId = props->TargetId; */
-
-       storDevice->PortNumber = deviceInfo->PortNumber;
-       /* Send it back up */
-       ret = StorVscConnectToVsp(Device);
-
-       /* deviceInfo->PortNumber = storDevice->PortNumber; */
-       deviceInfo->PathId = storDevice->PathId;
-       deviceInfo->TargetId = storDevice->TargetId;
-
-       DPRINT_DBG(STORVSC, "assigned port %u, path %u target %u\n", storDevice->PortNumber, storDevice->PathId, storDevice->TargetId);
-
-Cleanup:
-       DPRINT_EXIT(STORVSC);
-
-       return ret;
-}
-
 static int StorVscChannelInit(struct hv_device *Device)
 {
-       int ret=0;
-       STORVSC_DEVICE *storDevice;
-       STORVSC_REQUEST_EXTENSION *request;
-       VSTOR_PACKET *vstorPacket;
+       struct storvsc_device *storDevice;
+       struct storvsc_request_extension *request;
+       struct vstor_packet *vstorPacket;
+       int ret;
 
        storDevice = GetStorDevice(Device);
-       if (!storDevice)
-       {
-               DPRINT_ERR(STORVSC, "unable to get stor device...device being destroyed?");
+       if (!storDevice) {
+               DPRINT_ERR(STORVSC, "unable to get stor device..."
+                          "device being destroyed?");
                DPRINT_EXIT(STORVSC);
                return -1;
        }
@@ -378,9 +192,11 @@ static int StorVscChannelInit(struct hv_device *Device)
        request = &storDevice->InitRequest;
        vstorPacket = &request->VStorPacket;
 
-       /* Now, initiate the vsc/vsp initialization protocol on the open channel */
-
-       memset(request, sizeof(STORVSC_REQUEST_EXTENSION), 0);
+       /*
+        * Now, initiate the vsc/vsp initialization protocol on the open
+        * channel
+        */
+       memset(request, 0, sizeof(struct storvsc_request_extension));
        request->WaitEvent = osd_WaitEventCreate();
 
        vstorPacket->Operation = VStorOperationBeginInitialization;
@@ -393,127 +209,134 @@ static int StorVscChannelInit(struct hv_device *Device)
        DPRINT_INFO(STORVSC, "BEGIN_INITIALIZATION_OPERATION...");
 
        ret = Device->Driver->VmbusChannelInterface.SendPacket(Device,
-                                                                                                                       vstorPacket,
-                                                                                                                       sizeof(VSTOR_PACKET),
-                                                                                                                       (unsigned long)request,
-                                                                                                                       VmbusPacketTypeDataInBand,
-                                                                                                                       VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
-       if ( ret != 0)
-       {
-               DPRINT_ERR(STORVSC, "unable to send BEGIN_INITIALIZATION_OPERATION");
+                               vstorPacket,
+                               sizeof(struct vstor_packet),
+                               (unsigned long)request,
+                               VmbusPacketTypeDataInBand,
+                               VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
+       if (ret != 0) {
+               DPRINT_ERR(STORVSC,
+                          "unable to send BEGIN_INITIALIZATION_OPERATION");
                goto Cleanup;
        }
 
        osd_WaitEventWait(request->WaitEvent);
 
-       if (vstorPacket->Operation != VStorOperationCompleteIo || vstorPacket->Status != 0)
-       {
-               DPRINT_ERR(STORVSC, "BEGIN_INITIALIZATION_OPERATION failed (op %d status 0x%x)", vstorPacket->Operation, vstorPacket->Status);
+       if (vstorPacket->Operation != VStorOperationCompleteIo ||
+           vstorPacket->Status != 0) {
+               DPRINT_ERR(STORVSC, "BEGIN_INITIALIZATION_OPERATION failed "
+                          "(op %d status 0x%x)",
+                          vstorPacket->Operation, vstorPacket->Status);
                goto Cleanup;
        }
 
        DPRINT_INFO(STORVSC, "QUERY_PROTOCOL_VERSION_OPERATION...");
 
        /* reuse the packet for version range supported */
-       memset(vstorPacket, sizeof(VSTOR_PACKET), 0);
+       memset(vstorPacket, 0, sizeof(struct vstor_packet));
        vstorPacket->Operation = VStorOperationQueryProtocolVersion;
        vstorPacket->Flags = REQUEST_COMPLETION_FLAG;
 
-    vstorPacket->Version.MajorMinor = VMSTOR_PROTOCOL_VERSION_CURRENT;
-    FILL_VMSTOR_REVISION(vstorPacket->Version.Revision);
+       vstorPacket->Version.MajorMinor = VMSTOR_PROTOCOL_VERSION_CURRENT;
+       FILL_VMSTOR_REVISION(vstorPacket->Version.Revision);
 
        ret = Device->Driver->VmbusChannelInterface.SendPacket(Device,
-                                                                                                                       vstorPacket,
-                                                                                                                       sizeof(VSTOR_PACKET),
-                                                                                                                       (unsigned long)request,
-                                                                                                                       VmbusPacketTypeDataInBand,
-                                                                                                                       VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
-       if ( ret != 0)
-       {
-               DPRINT_ERR(STORVSC, "unable to send BEGIN_INITIALIZATION_OPERATION");
+                               vstorPacket,
+                               sizeof(struct vstor_packet),
+                               (unsigned long)request,
+                               VmbusPacketTypeDataInBand,
+                               VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
+       if (ret != 0) {
+               DPRINT_ERR(STORVSC,
+                          "unable to send BEGIN_INITIALIZATION_OPERATION");
                goto Cleanup;
        }
 
        osd_WaitEventWait(request->WaitEvent);
 
        /* TODO: Check returned version */
-       if (vstorPacket->Operation != VStorOperationCompleteIo || vstorPacket->Status != 0)
-       {
-               DPRINT_ERR(STORVSC, "QUERY_PROTOCOL_VERSION_OPERATION failed (op %d status 0x%x)", vstorPacket->Operation, vstorPacket->Status);
+       if (vstorPacket->Operation != VStorOperationCompleteIo ||
+           vstorPacket->Status != 0) {
+               DPRINT_ERR(STORVSC, "QUERY_PROTOCOL_VERSION_OPERATION failed "
+                          "(op %d status 0x%x)",
+                          vstorPacket->Operation, vstorPacket->Status);
                goto Cleanup;
        }
 
        /* Query channel properties */
        DPRINT_INFO(STORVSC, "QUERY_PROPERTIES_OPERATION...");
 
-       memset(vstorPacket, sizeof(VSTOR_PACKET), 0);
-    vstorPacket->Operation = VStorOperationQueryProperties;
+       memset(vstorPacket, 0, sizeof(struct vstor_packet));
+       vstorPacket->Operation = VStorOperationQueryProperties;
        vstorPacket->Flags = REQUEST_COMPLETION_FLAG;
-    vstorPacket->StorageChannelProperties.PortNumber = storDevice->PortNumber;
+       vstorPacket->StorageChannelProperties.PortNumber =
+                                       storDevice->PortNumber;
 
        ret = Device->Driver->VmbusChannelInterface.SendPacket(Device,
-                                                                                                                       vstorPacket,
-                                                                                                                       sizeof(VSTOR_PACKET),
-                                                                                                                       (unsigned long)request,
-                                                                                                                       VmbusPacketTypeDataInBand,
-                                                                                                                       VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
-
-       if ( ret != 0)
-       {
-               DPRINT_ERR(STORVSC, "unable to send QUERY_PROPERTIES_OPERATION");
+                               vstorPacket,
+                               sizeof(struct vstor_packet),
+                               (unsigned long)request,
+                               VmbusPacketTypeDataInBand,
+                               VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
+
+       if (ret != 0) {
+               DPRINT_ERR(STORVSC,
+                          "unable to send QUERY_PROPERTIES_OPERATION");
                goto Cleanup;
        }
 
        osd_WaitEventWait(request->WaitEvent);
 
        /* TODO: Check returned version */
-       if (vstorPacket->Operation != VStorOperationCompleteIo || vstorPacket->Status != 0)
-       {
-               DPRINT_ERR(STORVSC, "QUERY_PROPERTIES_OPERATION failed (op %d status 0x%x)", vstorPacket->Operation, vstorPacket->Status);
+       if (vstorPacket->Operation != VStorOperationCompleteIo ||
+           vstorPacket->Status != 0) {
+               DPRINT_ERR(STORVSC, "QUERY_PROPERTIES_OPERATION failed "
+                          "(op %d status 0x%x)",
+                          vstorPacket->Operation, vstorPacket->Status);
                goto Cleanup;
        }
 
-       /* storDevice->PortNumber = vstorPacket->StorageChannelProperties.PortNumber; */
        storDevice->PathId = vstorPacket->StorageChannelProperties.PathId;
        storDevice->TargetId = vstorPacket->StorageChannelProperties.TargetId;
 
-       DPRINT_DBG(STORVSC, "channel flag 0x%x, max xfer len 0x%x", vstorPacket->StorageChannelProperties.Flags, vstorPacket->StorageChannelProperties.MaxTransferBytes);
+       DPRINT_DBG(STORVSC, "channel flag 0x%x, max xfer len 0x%x",
+                  vstorPacket->StorageChannelProperties.Flags,
+                  vstorPacket->StorageChannelProperties.MaxTransferBytes);
 
        DPRINT_INFO(STORVSC, "END_INITIALIZATION_OPERATION...");
 
-       memset(vstorPacket, sizeof(VSTOR_PACKET), 0);
-    vstorPacket->Operation = VStorOperationEndInitialization;
+       memset(vstorPacket, 0, sizeof(struct vstor_packet));
+       vstorPacket->Operation = VStorOperationEndInitialization;
        vstorPacket->Flags = REQUEST_COMPLETION_FLAG;
 
        ret = Device->Driver->VmbusChannelInterface.SendPacket(Device,
-                                                                                                                       vstorPacket,
-                                                                                                                       sizeof(VSTOR_PACKET),
-                                                                                                                       (unsigned long)request,
-                                                                                                                       VmbusPacketTypeDataInBand,
-                                                                                                                       VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
-
-       if ( ret != 0)
-       {
-               DPRINT_ERR(STORVSC, "unable to send END_INITIALIZATION_OPERATION");
+                               vstorPacket,
+                               sizeof(struct vstor_packet),
+                               (unsigned long)request,
+                               VmbusPacketTypeDataInBand,
+                               VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
+
+       if (ret != 0) {
+               DPRINT_ERR(STORVSC,
+                          "unable to send END_INITIALIZATION_OPERATION");
                goto Cleanup;
        }
 
        osd_WaitEventWait(request->WaitEvent);
 
-       if (vstorPacket->Operation != VStorOperationCompleteIo || vstorPacket->Status != 0)
-       {
-               DPRINT_ERR(STORVSC, "END_INITIALIZATION_OPERATION failed (op %d status 0x%x)", vstorPacket->Operation, vstorPacket->Status);
+       if (vstorPacket->Operation != VStorOperationCompleteIo ||
+           vstorPacket->Status != 0) {
+               DPRINT_ERR(STORVSC, "END_INITIALIZATION_OPERATION failed "
+                          "(op %d status 0x%x)",
+                          vstorPacket->Operation, vstorPacket->Status);
                goto Cleanup;
        }
 
        DPRINT_INFO(STORVSC, "**** storage channel up and running!! ****");
 
 Cleanup:
-       if (request->WaitEvent)
-       {
-               kfree(request->WaitEvent);
-               request->WaitEvent = NULL;
-       }
+       kfree(request->WaitEvent);
+       request->WaitEvent = NULL;
 
        PutStorDevice(Device);
 
@@ -521,33 +344,185 @@ Cleanup:
        return ret;
 }
 
+static void StorVscOnIOCompletion(struct hv_device *Device,
+                                 struct vstor_packet *VStorPacket,
+                                 struct storvsc_request_extension *RequestExt)
+{
+       struct hv_storvsc_request *request;
+       struct storvsc_device *storDevice;
+
+       DPRINT_ENTER(STORVSC);
+
+       storDevice = MustGetStorDevice(Device);
+       if (!storDevice) {
+               DPRINT_ERR(STORVSC, "unable to get stor device..."
+                          "device being destroyed?");
+               DPRINT_EXIT(STORVSC);
+               return;
+       }
+
+       DPRINT_DBG(STORVSC, "IO_COMPLETE_OPERATION - request extension %p "
+                  "completed bytes xfer %u", RequestExt,
+                  VStorPacket->VmSrb.DataTransferLength);
+
+       ASSERT(RequestExt != NULL);
+       ASSERT(RequestExt->Request != NULL);
+
+       request = RequestExt->Request;
+
+       ASSERT(request->OnIOCompletion != NULL);
+
+       /* Copy over the status...etc */
+       request->Status = VStorPacket->VmSrb.ScsiStatus;
+
+       if (request->Status != 0 || VStorPacket->VmSrb.SrbStatus != 1) {
+               DPRINT_WARN(STORVSC,
+                           "cmd 0x%x scsi status 0x%x srb status 0x%x\n",
+                           request->Cdb[0], VStorPacket->VmSrb.ScsiStatus,
+                           VStorPacket->VmSrb.SrbStatus);
+       }
+
+       if ((request->Status & 0xFF) == 0x02) {
+               /* CHECK_CONDITION */
+               if (VStorPacket->VmSrb.SrbStatus & 0x80) {
+                       /* autosense data available */
+                       DPRINT_WARN(STORVSC, "storvsc pkt %p autosense data "
+                                   "valid - len %d\n", RequestExt,
+                                   VStorPacket->VmSrb.SenseInfoLength);
+
+                       ASSERT(VStorPacket->VmSrb.SenseInfoLength <=
+                               request->SenseBufferSize);
+                       memcpy(request->SenseBuffer,
+                              VStorPacket->VmSrb.SenseData,
+                              VStorPacket->VmSrb.SenseInfoLength);
+
+                       request->SenseBufferSize =
+                                       VStorPacket->VmSrb.SenseInfoLength;
+               }
+       }
+
+       /* TODO: */
+       request->BytesXfer = VStorPacket->VmSrb.DataTransferLength;
+
+       request->OnIOCompletion(request);
+
+       atomic_dec(&storDevice->NumOutstandingRequests);
+
+       PutStorDevice(Device);
+
+       DPRINT_EXIT(STORVSC);
+}
+
+static void StorVscOnReceive(struct hv_device *Device,
+                            struct vstor_packet *VStorPacket,
+                            struct storvsc_request_extension *RequestExt)
+{
+       switch (VStorPacket->Operation) {
+       case VStorOperationCompleteIo:
+               DPRINT_DBG(STORVSC, "IO_COMPLETE_OPERATION");
+               StorVscOnIOCompletion(Device, VStorPacket, RequestExt);
+               break;
+       case VStorOperationRemoveDevice:
+               DPRINT_INFO(STORVSC, "REMOVE_DEVICE_OPERATION");
+               /* TODO: */
+               break;
+
+       default:
+               DPRINT_INFO(STORVSC, "Unknown operation received - %d",
+                           VStorPacket->Operation);
+               break;
+       }
+}
 
-static int
-StorVscConnectToVsp(
-       struct hv_device *Device
-       )
+static void StorVscOnChannelCallback(void *context)
 {
-       int ret=0;
-    VMSTORAGE_CHANNEL_PROPERTIES props;
+       struct hv_device *device = (struct hv_device *)context;
+       struct storvsc_device *storDevice;
+       u32 bytesRecvd;
+       u64 requestId;
+       unsigned char packet[ALIGN_UP(sizeof(struct vstor_packet), 8)];
+       struct storvsc_request_extension *request;
+       int ret;
+
+       DPRINT_ENTER(STORVSC);
+
+       ASSERT(device);
+
+       storDevice = MustGetStorDevice(device);
+       if (!storDevice) {
+               DPRINT_ERR(STORVSC, "unable to get stor device..."
+                          "device being destroyed?");
+               DPRINT_EXIT(STORVSC);
+               return;
+       }
+
+       do {
+               ret = device->Driver->VmbusChannelInterface.RecvPacket(device,
+                               packet,
+                               ALIGN_UP(sizeof(struct vstor_packet), 8),
+                               &bytesRecvd, &requestId);
+               if (ret == 0 && bytesRecvd > 0) {
+                       DPRINT_DBG(STORVSC, "receive %d bytes - tid %llx",
+                                  bytesRecvd, requestId);
+
+                       /* ASSERT(bytesRecvd == sizeof(struct vstor_packet)); */
+
+                       request = (struct storvsc_request_extension *)
+                                       (unsigned long)requestId;
+                       ASSERT(request);
 
-       STORVSC_DRIVER_OBJECT *storDriver = (STORVSC_DRIVER_OBJECT*) Device->Driver;;
+                       /* if (vstorPacket.Flags & SYNTHETIC_FLAG) */
+                       if ((request == &storDevice->InitRequest) ||
+                           (request == &storDevice->ResetRequest)) {
+                               /* DPRINT_INFO(STORVSC,
+                                *             "reset completion - operation "
+                                *             "%u status %u",
+                                *             vstorPacket.Operation,
+                                *             vstorPacket.Status); */
 
-       memset(&props, sizeof(VMSTORAGE_CHANNEL_PROPERTIES), 0);
+                               memcpy(&request->VStorPacket, packet,
+                                      sizeof(struct vstor_packet));
+
+                               osd_WaitEventSet(request->WaitEvent);
+                       } else {
+                               StorVscOnReceive(device,
+                                               (struct vstor_packet *)packet,
+                                               request);
+                       }
+               } else {
+                       /* DPRINT_DBG(STORVSC, "nothing else to read..."); */
+                       break;
+               }
+       } while (1);
+
+       PutStorDevice(device);
+
+       DPRINT_EXIT(STORVSC);
+       return;
+}
+
+static int StorVscConnectToVsp(struct hv_device *Device)
+{
+       struct vmstorage_channel_properties props;
+       struct storvsc_driver_object *storDriver;
+       int ret;
+
+       storDriver = (struct storvsc_driver_object *)Device->Driver;
+       memset(&props, 0, sizeof(struct vmstorage_channel_properties));
 
        /* Open the channel */
        ret = Device->Driver->VmbusChannelInterface.Open(Device,
-               storDriver->RingBufferSize,
-               storDriver->RingBufferSize,
-               (void *)&props,
-               sizeof(VMSTORAGE_CHANNEL_PROPERTIES),
-               StorVscOnChannelCallback,
-               Device
-               );
-
-       DPRINT_DBG(STORVSC, "storage props: path id %d, tgt id %d, max xfer %d", props.PathId, props.TargetId, props.MaxTransferBytes);
-
-       if (ret != 0)
-       {
+                       storDriver->RingBufferSize,
+                       storDriver->RingBufferSize,
+                       (void *)&props,
+                       sizeof(struct vmstorage_channel_properties),
+                       StorVscOnChannelCallback,
+                       Device);
+
+       DPRINT_DBG(STORVSC, "storage props: path id %d, tgt id %d, max xfer %d",
+                  props.PathId, props.TargetId, props.MaxTransferBytes);
+
+       if (ret != 0) {
                DPRINT_ERR(STORVSC, "unable to open channel: %d", ret);
                return -1;
        }
@@ -557,27 +532,69 @@ StorVscConnectToVsp(
        return ret;
 }
 
+/**
+ * StorVscOnDeviceAdd - Callback when the device belonging to this driver is added
+ */
+static int StorVscOnDeviceAdd(struct hv_device *Device, void *AdditionalInfo)
+{
+       struct storvsc_device *storDevice;
+       /* struct vmstorage_channel_properties *props; */
+       struct storvsc_device_info *deviceInfo;
+       int ret = 0;
+
+       DPRINT_ENTER(STORVSC);
+
+       deviceInfo = (struct storvsc_device_info *)AdditionalInfo;
+       storDevice = AllocStorDevice(Device);
+       if (!storDevice) {
+               ret = -1;
+               goto Cleanup;
+       }
+
+       /* Save the channel properties to our storvsc channel */
+       /* props = (struct vmstorage_channel_properties *)
+        *              channel->offerMsg.Offer.u.Standard.UserDefined; */
+
+       /* FIXME: */
+       /*
+        * If we support more than 1 scsi channel, we need to set the
+        * port number here to the scsi channel but how do we get the
+        * scsi channel prior to the bus scan
+        */
+
+       /* storChannel->PortNumber = 0;
+       storChannel->PathId = props->PathId;
+       storChannel->TargetId = props->TargetId; */
+
+       storDevice->PortNumber = deviceInfo->PortNumber;
+       /* Send it back up */
+       ret = StorVscConnectToVsp(Device);
+
+       /* deviceInfo->PortNumber = storDevice->PortNumber; */
+       deviceInfo->PathId = storDevice->PathId;
+       deviceInfo->TargetId = storDevice->TargetId;
 
-/*++
+       DPRINT_DBG(STORVSC, "assigned port %u, path %u target %u\n",
+                  storDevice->PortNumber, storDevice->PathId,
+                  storDevice->TargetId);
 
-Name:
-       StorVscOnDeviceRemove()
+Cleanup:
+       DPRINT_EXIT(STORVSC);
 
-Description:
-       Callback when the our device is being removed
+       return ret;
+}
 
---*/
-static int
-StorVscOnDeviceRemove(
-       struct hv_device *Device
-       )
+/**
+ * StorVscOnDeviceRemove - Callback when the our device is being removed
+ */
+static int StorVscOnDeviceRemove(struct hv_device *Device)
 {
-       STORVSC_DEVICE *storDevice;
-       int ret=0;
+       struct storvsc_device *storDevice;
 
        DPRINT_ENTER(STORVSC);
 
-       DPRINT_INFO(STORVSC, "disabling storage device (%p)...", Device->Extension);
+       DPRINT_INFO(STORVSC, "disabling storage device (%p)...",
+                   Device->Extension);
 
        storDevice = ReleaseStorDevice(Device);
 
@@ -586,14 +603,14 @@ StorVscOnDeviceRemove(
         * only allow inbound traffic (responses) to proceed so that
         * outstanding requests can be completed.
         */
-       while (atomic_read(&storDevice->NumOutstandingRequests))
-       {
-               DPRINT_INFO(STORVSC, "waiting for %d requests to complete...", atomic_read(&storDevice->NumOutstandingRequests));
-
+       while (atomic_read(&storDevice->NumOutstandingRequests)) {
+               DPRINT_INFO(STORVSC, "waiting for %d requests to complete...",
+                           atomic_read(&storDevice->NumOutstandingRequests));
                udelay(100);
        }
 
-       DPRINT_INFO(STORVSC, "removing storage device (%p)...", Device->Extension);
+       DPRINT_INFO(STORVSC, "removing storage device (%p)...",
+                   Device->Extension);
 
        storDevice = FinalReleaseStorDevice(Device);
 
@@ -605,46 +622,24 @@ StorVscOnDeviceRemove(
        FreeStorDevice(storDevice);
 
        DPRINT_EXIT(STORVSC);
-       return ret;
+       return 0;
 }
 
-/* ***************
-static void
-StorVscOnTargetRescan(
-void *Context
-)
+static int StorVscOnHostReset(struct hv_device *Device)
 {
-struct hv_device *device=(struct hv_device *)Context;
-STORVSC_DRIVER_OBJECT *storDriver;
-
-DPRINT_ENTER(STORVSC);
-
-storDriver = (STORVSC_DRIVER_OBJECT*) device->Driver;
-storDriver->OnHostRescan(device);
-
-DPRINT_EXIT(STORVSC);
-}
-*********** */
-
-static int
-StorVscOnHostReset(
-       struct hv_device *Device
-       )
-{
-       int ret=0;
-
-       STORVSC_DEVICE *storDevice;
-       STORVSC_REQUEST_EXTENSION *request;
-       VSTOR_PACKET *vstorPacket;
+       struct storvsc_device *storDevice;
+       struct storvsc_request_extension *request;
+       struct vstor_packet *vstorPacket;
+       int ret;
 
        DPRINT_ENTER(STORVSC);
 
        DPRINT_INFO(STORVSC, "resetting host adapter...");
 
        storDevice = GetStorDevice(Device);
-       if (!storDevice)
-       {
-               DPRINT_ERR(STORVSC, "unable to get stor device...device being destroyed?");
+       if (!storDevice) {
+               DPRINT_ERR(STORVSC, "unable to get stor device..."
+                          "device being destroyed?");
                DPRINT_EXIT(STORVSC);
                return -1;
        }
@@ -654,19 +649,19 @@ StorVscOnHostReset(
 
        request->WaitEvent = osd_WaitEventCreate();
 
-    vstorPacket->Operation = VStorOperationResetBus;
-    vstorPacket->Flags = REQUEST_COMPLETION_FLAG;
-    vstorPacket->VmSrb.PathId = storDevice->PathId;
+       vstorPacket->Operation = VStorOperationResetBus;
+       vstorPacket->Flags = REQUEST_COMPLETION_FLAG;
+       vstorPacket->VmSrb.PathId = storDevice->PathId;
 
        ret = Device->Driver->VmbusChannelInterface.SendPacket(Device,
-                                                                                                                       vstorPacket,
-                                                                                                                       sizeof(VSTOR_PACKET),
-                                                                                                                       (unsigned long)&storDevice->ResetRequest,
-                                                                                                                       VmbusPacketTypeDataInBand,
-                                                                                                                       VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
-       if (ret != 0)
-       {
-               DPRINT_ERR(STORVSC, "Unable to send reset packet %p ret %d", vstorPacket, ret);
+                               vstorPacket,
+                               sizeof(struct vstor_packet),
+                               (unsigned long)&storDevice->ResetRequest,
+                               VmbusPacketTypeDataInBand,
+                               VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
+       if (ret != 0) {
+               DPRINT_ERR(STORVSC, "Unable to send reset packet %p ret %d",
+                          vstorPacket, ret);
                goto Cleanup;
        }
 
@@ -687,63 +682,60 @@ Cleanup:
        return ret;
 }
 
-/*++
-
-Name:
-       StorVscOnIORequest()
-
-Description:
-       Callback to initiate an I/O request
-
---*/
-static int
-StorVscOnIORequest(
-       struct hv_device *Device,
-       struct hv_storvsc_request *Request
-       )
+/**
+ * StorVscOnIORequest - Callback to initiate an I/O request
+ */
+static int StorVscOnIORequest(struct hv_device *Device,
+                             struct hv_storvsc_request *Request)
 {
-       STORVSC_DEVICE *storDevice;
-       STORVSC_REQUEST_EXTENSION* requestExtension = (STORVSC_REQUEST_EXTENSION*) Request->Extension;
-       VSTOR_PACKET* vstorPacket =&requestExtension->VStorPacket;
-       int ret=0;
+       struct storvsc_device *storDevice;
+       struct storvsc_request_extension *requestExtension;
+       struct vstor_packet *vstorPacket;
+       int ret = 0;
 
        DPRINT_ENTER(STORVSC);
 
+       requestExtension =
+               (struct storvsc_request_extension *)Request->Extension;
+       vstorPacket = &requestExtension->VStorPacket;
        storDevice = GetStorDevice(Device);
 
-       DPRINT_DBG(STORVSC, "enter - Device %p, DeviceExt %p, Request %p, Extension %p",
-               Device, storDevice, Request, requestExtension);
+       DPRINT_DBG(STORVSC, "enter - Device %p, DeviceExt %p, Request %p, "
+                  "Extension %p", Device, storDevice, Request,
+                  requestExtension);
 
        DPRINT_DBG(STORVSC, "req %p len %d bus %d, target %d, lun %d cdblen %d",
-               Request, Request->DataBuffer.Length, Request->Bus, Request->TargetId, Request->LunId, Request->CdbLen);
+                  Request, Request->DataBuffer.Length, Request->Bus,
+                  Request->TargetId, Request->LunId, Request->CdbLen);
 
-       if (!storDevice)
-       {
-               DPRINT_ERR(STORVSC, "unable to get stor device...device being destroyed?");
+       if (!storDevice) {
+               DPRINT_ERR(STORVSC, "unable to get stor device..."
+                          "device being destroyed?");
                DPRINT_EXIT(STORVSC);
                return -2;
        }
 
-       /* print_hex_dump_bytes("", DUMP_PREFIX_NONE, Request->Cdb, Request->CdbLen); */
+       /* print_hex_dump_bytes("", DUMP_PREFIX_NONE, Request->Cdb,
+        *                      Request->CdbLen); */
 
        requestExtension->Request = Request;
        requestExtension->Device  = Device;
 
-       memset(vstorPacket, 0 , sizeof(VSTOR_PACKET));
+       memset(vstorPacket, 0 , sizeof(struct vstor_packet));
 
        vstorPacket->Flags |= REQUEST_COMPLETION_FLAG;
 
-    vstorPacket->VmSrb.Length = sizeof(VMSCSI_REQUEST);
+       vstorPacket->VmSrb.Length = sizeof(struct vmscsi_request);
 
        vstorPacket->VmSrb.PortNumber = Request->Host;
-    vstorPacket->VmSrb.PathId = Request->Bus;
-    vstorPacket->VmSrb.TargetId = Request->TargetId;
-    vstorPacket->VmSrb.Lun = Request->LunId;
+       vstorPacket->VmSrb.PathId = Request->Bus;
+       vstorPacket->VmSrb.TargetId = Request->TargetId;
+       vstorPacket->VmSrb.Lun = Request->LunId;
 
        vstorPacket->VmSrb.SenseInfoLength = SENSE_BUFFER_SIZE;
 
        /* Copy over the scsi command descriptor block */
-    vstorPacket->VmSrb.CdbLength = Request->CdbLen;
+       vstorPacket->VmSrb.CdbLength = Request->CdbLen;
        memcpy(&vstorPacket->VmSrb.Cdb, Request->Cdb, Request->CdbLen);
 
        vstorPacket->VmSrb.DataIn = Request->Type;
@@ -751,36 +743,35 @@ StorVscOnIORequest(
 
        vstorPacket->Operation = VStorOperationExecuteSRB;
 
-       DPRINT_DBG(STORVSC, "srb - len %d port %d, path %d, target %d, lun %d senselen %d cdblen %d",
-               vstorPacket->VmSrb.Length,
-               vstorPacket->VmSrb.PortNumber,
-               vstorPacket->VmSrb.PathId,
-               vstorPacket->VmSrb.TargetId,
-               vstorPacket->VmSrb.Lun,
-               vstorPacket->VmSrb.SenseInfoLength,
-               vstorPacket->VmSrb.CdbLength);
-
-       if (requestExtension->Request->DataBuffer.Length)
-       {
-               ret = Device->Driver->VmbusChannelInterface.SendPacketMultiPageBuffer(Device,
+       DPRINT_DBG(STORVSC, "srb - len %d port %d, path %d, target %d, "
+                  "lun %d senselen %d cdblen %d",
+                  vstorPacket->VmSrb.Length,
+                  vstorPacket->VmSrb.PortNumber,
+                  vstorPacket->VmSrb.PathId,
+                  vstorPacket->VmSrb.TargetId,
+                  vstorPacket->VmSrb.Lun,
+                  vstorPacket->VmSrb.SenseInfoLength,
+                  vstorPacket->VmSrb.CdbLength);
+
+       if (requestExtension->Request->DataBuffer.Length) {
+               ret = Device->Driver->VmbusChannelInterface.
+                       SendPacketMultiPageBuffer(Device,
                                &requestExtension->Request->DataBuffer,
                                vstorPacket,
-                               sizeof(VSTOR_PACKET),
+                               sizeof(struct vstor_packet),
                                (unsigned long)requestExtension);
-       }
-       else
-       {
+       } else {
                ret = Device->Driver->VmbusChannelInterface.SendPacket(Device,
-                                                                                                                       vstorPacket,
-                                                                                                                       sizeof(VSTOR_PACKET),
-                                                                                                                       (unsigned long)requestExtension,
-                                                                                                                       VmbusPacketTypeDataInBand,
-                                                                                                                       VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
+                               vstorPacket,
+                               sizeof(struct vstor_packet),
+                               (unsigned long)requestExtension,
+                               VmbusPacketTypeDataInBand,
+                               VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED);
        }
 
-       if (ret != 0)
-       {
-               DPRINT_DBG(STORVSC, "Unable to send packet %p ret %d", vstorPacket, ret);
+       if (ret != 0) {
+               DPRINT_DBG(STORVSC, "Unable to send packet %p ret %d",
+                          vstorPacket, ret);
        }
 
        atomic_inc(&storDevice->NumOutstandingRequests);
@@ -791,193 +782,69 @@ StorVscOnIORequest(
        return ret;
 }
 
-/*++
-
-Name:
-       StorVscOnCleanup()
-
-Description:
-       Perform any cleanup when the driver is removed
-
---*/
-static void
-StorVscOnCleanup(
-       struct hv_driver *Driver
-       )
+/**
+ * StorVscOnCleanup - Perform any cleanup when the driver is removed
+ */
+static void StorVscOnCleanup(struct hv_driver *Driver)
 {
        DPRINT_ENTER(STORVSC);
        DPRINT_EXIT(STORVSC);
 }
 
-
-static void
-StorVscOnIOCompletion(
-       struct hv_device *Device,
-       VSTOR_PACKET    *VStorPacket,
-       STORVSC_REQUEST_EXTENSION *RequestExt
-       )
+/**
+ * StorVscInitialize - Main entry point
+ */
+int StorVscInitialize(struct hv_driver *Driver)
 {
-       struct hv_storvsc_request *request;
-       STORVSC_DEVICE *storDevice;
+       struct storvsc_driver_object *storDriver;
 
        DPRINT_ENTER(STORVSC);
 
-       storDevice = MustGetStorDevice(Device);
-       if (!storDevice)
-       {
-               DPRINT_ERR(STORVSC, "unable to get stor device...device being destroyed?");
-               DPRINT_EXIT(STORVSC);
-               return;
-       }
-
-       DPRINT_DBG(STORVSC, "IO_COMPLETE_OPERATION - request extension %p completed bytes xfer %u",
-               RequestExt, VStorPacket->VmSrb.DataTransferLength);
-
-       ASSERT(RequestExt != NULL);
-       ASSERT(RequestExt->Request != NULL);
+       storDriver = (struct storvsc_driver_object *)Driver;
 
-       request = RequestExt->Request;
+       DPRINT_DBG(STORVSC, "sizeof(STORVSC_REQUEST)=%zd "
+                  "sizeof(struct storvsc_request_extension)=%zd "
+                  "sizeof(struct vstor_packet)=%zd, "
+                  "sizeof(struct vmscsi_request)=%zd",
+                  sizeof(struct hv_storvsc_request),
+                  sizeof(struct storvsc_request_extension),
+                  sizeof(struct vstor_packet),
+                  sizeof(struct vmscsi_request));
 
-       ASSERT(request->OnIOCompletion != NULL);
-
-       /* Copy over the status...etc */
-       request->Status = VStorPacket->VmSrb.ScsiStatus;
-
-       if (request->Status != 0 || VStorPacket->VmSrb.SrbStatus != 1)
-       {
-               DPRINT_WARN(STORVSC, "cmd 0x%x scsi status 0x%x srb status 0x%x\n",
-                       request->Cdb[0],
-                       VStorPacket->VmSrb.ScsiStatus,
-                       VStorPacket->VmSrb.SrbStatus);
-       }
-
-       if ((request->Status & 0xFF) == 0x02) /* CHECK_CONDITION */
-       {
-               if (VStorPacket->VmSrb.SrbStatus & 0x80) /* autosense data available */
-               {
-                       DPRINT_WARN(STORVSC, "storvsc pkt %p autosense data valid - len %d\n",
-                               RequestExt, VStorPacket->VmSrb.SenseInfoLength);
+       /* Make sure we are at least 2 pages since 1 page is used for control */
+       ASSERT(storDriver->RingBufferSize >= (PAGE_SIZE << 1));
 
-                       ASSERT(VStorPacket->VmSrb.SenseInfoLength <=  request->SenseBufferSize);
-                       memcpy(request->SenseBuffer,
-                               VStorPacket->VmSrb.SenseData,
-                               VStorPacket->VmSrb.SenseInfoLength);
+       Driver->name = gDriverName;
+       memcpy(&Driver->deviceType, &gStorVscDeviceType,
+              sizeof(struct hv_guid));
 
-                       request->SenseBufferSize = VStorPacket->VmSrb.SenseInfoLength;
-               }
-       }
+       storDriver->RequestExtSize = sizeof(struct storvsc_request_extension);
 
-       /* TODO: */
-       request->BytesXfer = VStorPacket->VmSrb.DataTransferLength;
+       /*
+        * Divide the ring buffer data size (which is 1 page less
+        * than the ring buffer size since that page is reserved for
+        * the ring buffer indices) by the max request size (which is
+        * VMBUS_CHANNEL_PACKET_MULITPAGE_BUFFER + struct vstor_packet + u64)
+        */
+       storDriver->MaxOutstandingRequestsPerChannel =
+               ((storDriver->RingBufferSize - PAGE_SIZE) /
+                 ALIGN_UP(MAX_MULTIPAGE_BUFFER_PACKET +
+                          sizeof(struct vstor_packet) + sizeof(u64),
+                          sizeof(u64)));
 
-       request->OnIOCompletion(request);
+       DPRINT_INFO(STORVSC, "max io %u, currently %u\n",
+                   storDriver->MaxOutstandingRequestsPerChannel,
+                   STORVSC_MAX_IO_REQUESTS);
 
-       atomic_dec(&storDevice->NumOutstandingRequests);
+       /* Setup the dispatch table */
+       storDriver->Base.OnDeviceAdd    = StorVscOnDeviceAdd;
+       storDriver->Base.OnDeviceRemove = StorVscOnDeviceRemove;
+       storDriver->Base.OnCleanup      = StorVscOnCleanup;
 
-       PutStorDevice(Device);
+       storDriver->OnIORequest         = StorVscOnIORequest;
+       storDriver->OnHostReset         = StorVscOnHostReset;
 
        DPRINT_EXIT(STORVSC);
-}
-
-
-static void
-StorVscOnReceive(
-       struct hv_device *Device,
-       VSTOR_PACKET    *VStorPacket,
-       STORVSC_REQUEST_EXTENSION *RequestExt
-       )
-{
-       switch(VStorPacket->Operation)
-       {
-               case VStorOperationCompleteIo:
-
-                       DPRINT_DBG(STORVSC, "IO_COMPLETE_OPERATION");
-                       StorVscOnIOCompletion(Device, VStorPacket, RequestExt);
-                       break;
-
-               /* case ENUMERATE_DEVICE_OPERATION: */
-
-               /* DPRINT_INFO(STORVSC, "ENUMERATE_DEVICE_OPERATION"); */
-
-               /* StorVscOnTargetRescan(Device); */
-               /* break; */
-
-       case VStorOperationRemoveDevice:
 
-                       DPRINT_INFO(STORVSC, "REMOVE_DEVICE_OPERATION");
-                       /* TODO: */
-                       break;
-
-               default:
-                       DPRINT_INFO(STORVSC, "Unknown operation received - %d", VStorPacket->Operation);
-                       break;
-       }
-}
-
-static void
-StorVscOnChannelCallback(
-       void * Context
-       )
-{
-       int ret=0;
-       struct hv_device *device = (struct hv_device*)Context;
-       STORVSC_DEVICE *storDevice;
-       u32 bytesRecvd;
-       u64 requestId;
-       unsigned char packet[ALIGN_UP(sizeof(VSTOR_PACKET),8)];
-       STORVSC_REQUEST_EXTENSION *request;
-
-       DPRINT_ENTER(STORVSC);
-
-       ASSERT(device);
-
-       storDevice = MustGetStorDevice(device);
-       if (!storDevice)
-       {
-               DPRINT_ERR(STORVSC, "unable to get stor device...device being destroyed?");
-               DPRINT_EXIT(STORVSC);
-               return;
-       }
-
-       do
-       {
-               ret = device->Driver->VmbusChannelInterface.RecvPacket(device,
-                                                                                                                               packet,
-                                                                                                                               ALIGN_UP(sizeof(VSTOR_PACKET),8),
-                                                                                                                               &bytesRecvd,
-                                                                                                                               &requestId);
-               if (ret == 0 && bytesRecvd > 0)
-               {
-                       DPRINT_DBG(STORVSC, "receive %d bytes - tid %llx", bytesRecvd, requestId);
-
-                       /* ASSERT(bytesRecvd == sizeof(VSTOR_PACKET)); */
-
-                       request = (STORVSC_REQUEST_EXTENSION*)(unsigned long)requestId;
-                       ASSERT(request);
-
-                       /* if (vstorPacket.Flags & SYNTHETIC_FLAG) */
-                       if ((request == &storDevice->InitRequest) || (request == &storDevice->ResetRequest))
-                       {
-                               /* DPRINT_INFO(STORVSC, "reset completion - operation %u status %u", vstorPacket.Operation, vstorPacket.Status); */
-
-                               memcpy(&request->VStorPacket, packet, sizeof(VSTOR_PACKET));
-
-                               osd_WaitEventSet(request->WaitEvent);
-                       }
-                       else
-                       {
-                               StorVscOnReceive(device, (VSTOR_PACKET*)packet, request);
-                       }
-               }
-               else
-               {
-                       /* DPRINT_DBG(STORVSC, "nothing else to read..."); */
-                       break;
-               }
-       } while (1);
-
-       PutStorDevice(device);
-
-       DPRINT_EXIT(STORVSC);
-       return;
+       return 0;
 }