typedef struct _STORAGE_DESCRIPTOR_HEADER {

  ULONG  Version;

  ULONG  Size;

} STORAGE_DESCRIPTOR_HEADER, *PSTORAGE_DESCRIPTOR_HEADER;

/******************************************************************************

 * Function: get the bus type of an disk

 * input: drive name (c:)

 * output: bus type

 * return: Succeed, 0

 *         Fail, -1

 ******************************************************************************/

DWORD GetDriveTypeByBus( const CHAR *drive, WORD *type )

{

  HANDLE hDevice;               // handle to the drive to be examined

  BOOL result;                 // results flag

  DWORD readed;                   // discard results

  STORAGE_DESCRIPTOR_HEADER *pDevDescHeader;

  STORAGE_DEVICE_DESCRIPTOR *pDevDesc;

  DWORD devDescLength;

  STORAGE_PROPERTY_QUERY query;

  hDevice = CreateFile( drive, // drive to open

    GENERIC_READ | GENERIC_WRITE,     // access to the drive

    FILE_SHARE_READ | FILE_SHARE_WRITE, //share mode

    NULL,             // default security attributes

    OPEN_EXISTING,    // disposition

    ,                // file attributes

    NULL            // do not copy file attribute

    );

  if ( hDevice == INVALID_HANDLE_VALUE ) // cannot open the drive

  {

    fprintf( stderr, "CreateFile() Error: %ld\n", GetLastError( ) );

    return DWORD( - );

  }

  query.PropertyId = StorageDeviceProperty;

  query.QueryType = PropertyStandardQuery;

  pDevDescHeader = (STORAGE_DESCRIPTOR_HEADER *) malloc(

    sizeof(STORAGE_DESCRIPTOR_HEADER) );

  if ( NULL == pDevDescHeader )

  {

    return (DWORD) -;

  }

  result = DeviceIoControl( hDevice,     // device to be queried

    IOCTL_STORAGE_QUERY_PROPERTY,     // operation to perform

    &query, sizeof query,               // no input buffer

    pDevDescHeader, sizeof(STORAGE_DESCRIPTOR_HEADER),     // output buffer

    &readed,                 // # bytes returned

    NULL );      // synchronous I/O

  if ( !result )        //fail

  {

    fprintf( stderr, "IOCTL_STORAGE_QUERY_PROPERTY Error: %ld\n",

      GetLastError( ) );

    free( pDevDescHeader );

    (void) CloseHandle( hDevice );

    return DWORD( - );

  }

  devDescLength = pDevDescHeader->Size;

  pDevDesc = (STORAGE_DEVICE_DESCRIPTOR *) malloc( devDescLength );

  if ( NULL == pDevDesc )

  {

    free( pDevDescHeader );

    return (DWORD) -;

  }

  result = DeviceIoControl( hDevice,     // device to be queried

    IOCTL_STORAGE_QUERY_PROPERTY,     // operation to perform

    &query, sizeof query,               // no input buffer

    pDevDesc, devDescLength,     // output buffer

    &readed,                 // # bytes returned

    NULL );      // synchronous I/O

  if ( !result )        //fail

  {

    fprintf( stderr, "IOCTL_STORAGE_QUERY_PROPERTY Error: %ld\n",

      GetLastError( ) );

    free( pDevDescHeader );

    free( pDevDesc );

    (void) CloseHandle( hDevice );

    return DWORD( - );

  }

  //printf("%d\n", pDevDesc->BusType);

  *type = (WORD) pDevDesc->BusType;

  free( pDevDescHeader );

  free( pDevDesc );

  (void) CloseHandle( hDevice );

  return ;

}

BOOL

WINAPI

DeviceIoControl(

     _In_        (HANDLE)       hDevice,               // handle to a partition

     _In_        (DWORD) IOCTL_STORAGE_QUERY_PROPERTY, // dwIoControlCode

     _In_        (LPVOID)       lpInBuffer,            // input buffer - STORAGE_PROPERTY_QUERY structure

     _In_        (DWORD)        nInBufferSize,         // size of input buffer

     _Out_opt_   (LPVOID)       lpOutBuffer,           // output buffer - see Remarks

     _In_        (DWORD)        nOutBufferSize,        // size of output buffer

     _Out_opt_   (LPDWORD)      lpBytesReturned,       // number of bytes returned

     _Inout_opt_ (LPOVERLAPPED) lpOverlapped );        // OVERLAPPED structure

Parameters

hDevice

A handle to the disk device from which partition information is retrieved. To retrieve a device handle, call the CreateFile function.

dwIoControlCode

The control code for the operation. Use IOCTL_STORAGE_QUERY_PROPERTY for this operation.

lpInBuffer

A pointer to a buffer that contains a STORAGE_PROPERTY_QUERY data structure that specifies the details about the query. Device properties must be retrieved only from a device; attempting to retrieve device properties from an adapter will cause an error.

nInBufferSize

The size of the input buffer, in bytes. It must be large enough to contain aSTORAGE_PROPERTY_QUERY data structure.

lpOutBuffer

An optional pointer to a buffer that receives a structure that starts with the same fields as a STORAGE_DESCRIPTOR_HEADER data structure. For more information on the specific structures returned see the Remarks section.

nOutBufferSize

The size of the output buffer, in bytes. It can be zero to determine whether a property exists without retrieving its data. To do that, set this parameter to zero (0) and the QueryType member of the STORAGE_PROPERTY_QUERY input structure to PropertyExistsQuery (1). If the call to DeviceIoControl returns a nonzero value then the property exists.

lpBytesReturned

A pointer to a variable that receives the size of the data stored in the output buffer, in bytes.

If the output buffer is too small, the call fails, GetLastError returnsERROR_INSUFFICIENT_BUFFER, and lpBytesReturned is zero.

If lpOverlapped is NULL, lpBytesReturned cannot be NULL. Even when an operation returns no output data and lpOutBuffer is NULL, DeviceIoControl makes use oflpBytesReturned. After such an operation, the value of lpBytesReturned is meaningless.

If lpOverlapped is not NULL, lpBytesReturned can be NULL. If this parameter is not NULL and the operation returns data, lpBytesReturned is meaningless until the overlapped operation has completed. To retrieve the number of bytes returned, call GetOverlappedResult. If hDevice is associated with an I/O completion port, you can retrieve the number of bytes returned by callingGetQueuedCompletionStatus.

lpOverlapped

A pointer to an OVERLAPPED structure.

If hDevice was opened without specifying FILE_FLAG_OVERLAPPED, lpOverlapped is ignored.

If hDevice was opened with the FILE_FLAG_OVERLAPPED flag, the operation is performed as an overlapped (asynchronous) operation. In this case, lpOverlappedmust point to a valid OVERLAPPED structure that contains a handle to an event object. Otherwise, the function fails in unpredictable ways.

For overlapped operations, DeviceIoControl returns immediately, and the event object is signaled when the operation is complete. Otherwise, the function does not return until the operation is complete or an error occurs.

Return value

If the operation completes successfully, DeviceIoControl returns a nonzero value.

If the operation fails or is pending, DeviceIoControl returns zero. To get extended error information, call GetLastError.

Remarks

The optional output buffer returned through the lpOutBuffer parameter can be one of several structures depending on the value of the PropertyId member of theSTORAGE_PROPERTY_QUERY structure pointed to by the lpInBuffer parameter. These values are enumerated by the STORAGE_PROPERTY_ID enumeration. If the QueryTypemember of the STORAGE_PROPERTY_QUERY is set to PropertyExistsQuery then no structure is returned.