VirtualBox

source: vbox/trunk/src/VBox/Devices/Storage/DrvHostBase-win.cpp@ 64276

Last change on this file since 64276 was 64253, checked in by vboxsync, 8 years ago

Windows build fix
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 9.4 KB
Line 
1/* $Id: DrvHostBase-win.cpp 64253 2016-10-13 14:22:17Z vboxsync $ */
2/** @file
3 * DrvHostBase - Host base drive access driver, Windows specifics.
4 */
5
6/*
7 * Copyright (C) 2006-2016 Oracle Corporation
8 *
9 * This file is part of VirtualBox Open Source Edition (OSE), as
10 * available from http://www.virtualbox.org. This file is free software;
11 * you can redistribute it and/or modify it under the terms of the GNU
12 * General Public License (GPL) as published by the Free Software
13 * Foundation, in version 2 as it comes in the "COPYING" file of the
14 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16 */
17#define LOG_GROUP LOG_GROUP_DRV_HOST_BASE
18#pragma warning(disable : 4163)
19#define _interlockedbittestandset they_messed_it_up_in_winnt_h_this_time_sigh__interlockedbittestandset
20#define _interlockedbittestandreset they_messed_it_up_in_winnt_h_this_time_sigh__interlockedbittestandreset
21#define _interlockedbittestandset64 they_messed_it_up_in_winnt_h_this_time_sigh__interlockedbittestandset64
22#define _interlockedbittestandreset64 they_messed_it_up_in_winnt_h_this_time_sigh__interlockedbittestandreset64
23
24#define WIN32_NO_STATUS
25#include <iprt/win/windows.h>
26#include <dbt.h>
27#undef WIN32_NO_STATUS
28
29#include <winioctl.h>
30#include <ntddscsi.h>
31#pragma warning(default : 4163)
32#undef _interlockedbittestandset
33#undef _interlockedbittestandreset
34#undef _interlockedbittestandset64
35#undef _interlockedbittestandreset64
36#include <ntstatus.h>
37
38/* from ntdef.h */
39typedef LONG NTSTATUS;
40
41/* from ntddk.h */
42typedef struct _IO_STATUS_BLOCK {
43 union {
44 NTSTATUS Status;
45 PVOID Pointer;
46 };
47 ULONG_PTR Information;
48} IO_STATUS_BLOCK, *PIO_STATUS_BLOCK;
49
50
51/* from ntinternals.com */
52typedef enum _FS_INFORMATION_CLASS {
53 FileFsVolumeInformation=1,
54 FileFsLabelInformation,
55 FileFsSizeInformation,
56 FileFsDeviceInformation,
57 FileFsAttributeInformation,
58 FileFsControlInformation,
59 FileFsFullSizeInformation,
60 FileFsObjectIdInformation,
61 FileFsMaximumInformation
62} FS_INFORMATION_CLASS, *PFS_INFORMATION_CLASS;
63
64typedef struct _FILE_FS_SIZE_INFORMATION {
65 LARGE_INTEGER TotalAllocationUnits;
66 LARGE_INTEGER AvailableAllocationUnits;
67 ULONG SectorsPerAllocationUnit;
68 ULONG BytesPerSector;
69} FILE_FS_SIZE_INFORMATION, *PFILE_FS_SIZE_INFORMATION;
70
71extern "C"
72NTSTATUS __stdcall NtQueryVolumeInformationFile(
73 /*IN*/ HANDLE FileHandle,
74 /*OUT*/ PIO_STATUS_BLOCK IoStatusBlock,
75 /*OUT*/ PVOID FileSystemInformation,
76 /*IN*/ ULONG Length,
77 /*IN*/ FS_INFORMATION_CLASS FileSystemInformationClass );
78
79#include <iprt/file.h>
80#include <VBox/scsi.h>
81
82#include "DrvHostBase.h"
83
84DECLHIDDEN(int) drvHostBaseScsiCmdOs(PDRVHOSTBASE pThis, const uint8_t *pbCmd, size_t cbCmd, PDMMEDIATXDIR enmTxDir,
85 void *pvBuf, uint32_t *pcbBuf, uint8_t *pbSense, size_t cbSense, uint32_t cTimeoutMillies)
86{
87 /*
88 * Minimal input validation.
89 */
90 Assert(enmTxDir == PDMMEDIATXDIR_NONE || enmTxDir == PDMMEDIATXDIR_FROM_DEVICE || enmTxDir == PDMMEDIATXDIR_TO_DEVICE);
91 Assert(!pvBuf || pcbBuf);
92 Assert(pvBuf || enmTxDir == PDMMEDIATXDIR_NONE);
93 Assert(pbSense || !cbSense);
94 AssertPtr(pbCmd);
95 Assert(cbCmd <= 16 && cbCmd >= 1); RT_NOREF(cbCmd);
96
97 int rc = VERR_GENERAL_FAILURE;
98 int direction;
99 struct _REQ
100 {
101 SCSI_PASS_THROUGH_DIRECT spt;
102 uint8_t aSense[64];
103 } Req;
104 DWORD cbReturned = 0;
105
106 switch (enmTxDir)
107 {
108 case PDMMEDIATXDIR_NONE:
109 direction = SCSI_IOCTL_DATA_UNSPECIFIED;
110 break;
111 case PDMMEDIATXDIR_FROM_DEVICE:
112 Assert(*pcbBuf != 0);
113 /* Make sure that the buffer is clear for commands reading
114 * data. The actually received data may be shorter than what
115 * we expect, and due to the unreliable feedback about how much
116 * data the ioctl actually transferred, it's impossible to
117 * prevent that. Returning previous buffer contents may cause
118 * security problems inside the guest OS, if users can issue
119 * commands to the CDROM device. */
120 memset(pvBuf, '\0', *pcbBuf);
121 direction = SCSI_IOCTL_DATA_IN;
122 break;
123 case PDMMEDIATXDIR_TO_DEVICE:
124 direction = SCSI_IOCTL_DATA_OUT;
125 break;
126 default:
127 AssertMsgFailed(("enmTxDir invalid!\n"));
128 direction = SCSI_IOCTL_DATA_UNSPECIFIED;
129 }
130 memset(&Req, '\0', sizeof(Req));
131 Req.spt.Length = sizeof(Req.spt);
132 Req.spt.CdbLength = 12;
133 memcpy(Req.spt.Cdb, pbCmd, Req.spt.CdbLength);
134 Req.spt.DataBuffer = pvBuf;
135 Req.spt.DataTransferLength = *pcbBuf;
136 Req.spt.DataIn = direction;
137 Req.spt.TimeOutValue = (cTimeoutMillies + 999) / 1000; /* Convert to seconds */
138 Assert(cbSense <= sizeof(Req.aSense));
139 Req.spt.SenseInfoLength = (UCHAR)RT_MIN(sizeof(Req.aSense), cbSense);
140 Req.spt.SenseInfoOffset = RT_OFFSETOF(struct _REQ, aSense);
141 if (DeviceIoControl((HANDLE)RTFileToNative(pThis->hFileDevice), IOCTL_SCSI_PASS_THROUGH_DIRECT,
142 &Req, sizeof(Req), &Req, sizeof(Req), &cbReturned, NULL))
143 {
144 if (cbReturned > RT_OFFSETOF(struct _REQ, aSense))
145 memcpy(pbSense, Req.aSense, cbSense);
146 else
147 memset(pbSense, '\0', cbSense);
148 /* Windows shares the property of not properly reflecting the actually
149 * transferred data size. See above. Assume that everything worked ok.
150 * Except if there are sense information. */
151 rc = (pbSense[2] & 0x0f) == SCSI_SENSE_NONE
152 ? VINF_SUCCESS
153 : VERR_DEV_IO_ERROR;
154 }
155 else
156 rc = RTErrConvertFromWin32(GetLastError());
157 Log2(("%s: scsistatus=%d bytes returned=%d tlength=%d\n", __FUNCTION__, Req.spt.ScsiStatus, cbReturned, Req.spt.DataTransferLength));
158
159 return rc;
160}
161
162DECLHIDDEN(int) drvHostBaseGetMediaSizeOs(PDRVHOSTBASE pThis, uint64_t *pcb)
163{
164 int rc = VERR_GENERAL_FAILURE;
165
166 if (PDMMEDIATYPE_IS_FLOPPY(pThis->enmType))
167 {
168 DISK_GEOMETRY geom;
169 DWORD cbBytesReturned;
170 int cbSectors;
171
172 memset(&geom, 0, sizeof(geom));
173 rc = DeviceIoControl((HANDLE)RTFileToNative(pThis->hFileDevice), IOCTL_DISK_GET_DRIVE_GEOMETRY,
174 NULL, 0, &geom, sizeof(geom), &cbBytesReturned, NULL);
175 if (rc) {
176 cbSectors = geom.Cylinders.QuadPart * geom.TracksPerCylinder * geom.SectorsPerTrack;
177 *pcb = cbSectors * geom.BytesPerSector;
178 rc = VINF_SUCCESS;
179 }
180 else
181 {
182 DWORD dwLastError;
183
184 dwLastError = GetLastError();
185 rc = RTErrConvertFromWin32(dwLastError);
186 Log(("DrvHostFloppy: IOCTL_DISK_GET_DRIVE_GEOMETRY(%s) failed, LastError=%d rc=%Rrc\n",
187 pThis->pszDevice, dwLastError, rc));
188 return rc;
189 }
190 }
191 else
192 {
193 /* use NT api, retry a few times if the media is being verified. */
194 IO_STATUS_BLOCK IoStatusBlock = {0};
195 FILE_FS_SIZE_INFORMATION FsSize= {0};
196 NTSTATUS rcNt = NtQueryVolumeInformationFile((HANDLE)RTFileToNative(pThis->hFileDevice), &IoStatusBlock,
197 &FsSize, sizeof(FsSize), FileFsSizeInformation);
198 int cRetries = 5;
199 while (rcNt == STATUS_VERIFY_REQUIRED && cRetries-- > 0)
200 {
201 RTThreadSleep(10);
202 rcNt = NtQueryVolumeInformationFile((HANDLE)RTFileToNative(pThis->hFileDevice), &IoStatusBlock,
203 &FsSize, sizeof(FsSize), FileFsSizeInformation);
204 }
205 if (rcNt >= 0)
206 {
207 *pcb = FsSize.TotalAllocationUnits.QuadPart * FsSize.BytesPerSector;
208 return VINF_SUCCESS;
209 }
210
211 /* convert nt status code to VBox status code. */
212 /** @todo Make conversion function!. */
213 switch (rcNt)
214 {
215 case STATUS_NO_MEDIA_IN_DEVICE: rc = VERR_MEDIA_NOT_PRESENT; break;
216 case STATUS_VERIFY_REQUIRED: rc = VERR_TRY_AGAIN; break;
217 }
218 LogFlow(("drvHostBaseGetMediaSize: NtQueryVolumeInformationFile -> %#lx\n", rcNt, rc));
219 }
220 return rc;
221}
222
223
224DECLHIDDEN(int) drvHostBaseReadOs(PDRVHOSTBASE pThis, uint64_t off, void *pvBuf, size_t cbRead)
225{
226 return RTFileReadAt(pThis->hFileDevice, off, pvBuf, cbRead, NULL);
227}
228
229
230DECLHIDDEN(int) drvHostBaseWriteOs(PDRVHOSTBASE pThis, uint64_t off, const void *pvBuf, size_t cbWrite)
231{
232 return RTFileWriteAt(pThis->hFileDevice, off, pvBuf, cbWrite, NULL);
233}
234
235
236DECLHIDDEN(int) drvHostBaseFlushOs(PDRVHOSTBASE pThis)
237{
238 return RTFileFlush(pThis->hFileDevice);
239}
240
241
242DECLHIDDEN(int) drvHostBasePollerWakeupOs(PDRVHOSTBASE pThis)
243{
244 if (pThis->hwndDeviceChange)
245 PostMessage(pThis->hwndDeviceChange, WM_CLOSE, 0, 0); /* default win proc will destroy the window */
246
247 return VINF_SUCCESS;
248}
249
250
251DECLHIDDEN(void) drvHostBaseDestructOs(PDRVHOSTBASE pThis)
252{
253 if (pThis->hwndDeviceChange)
254 {
255 if (SetWindowLongPtr(pThis->hwndDeviceChange, GWLP_USERDATA, 0) == (LONG_PTR)pThis)
256 PostMessage(pThis->hwndDeviceChange, WM_CLOSE, 0, 0); /* default win proc will destroy the window */
257 pThis->hwndDeviceChange = NULL;
258 }
259
260 if (pThis->hFileDevice != NIL_RTFILE)
261 {
262 int rc = RTFileClose(pThis->hFileDevice);
263 AssertRC(rc);
264 pThis->hFileDevice = NIL_RTFILE;
265 }
266}
267
Note: See TracBrowser for help on using the repository browser.

© 2025 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette