1 | /* $Id: HostDriveImpl.cpp 96407 2022-08-22 17:43:14Z vboxsync $ */
|
---|
2 | /** @file
|
---|
3 | * VirtualBox Main - IHostDrive implementation, VBoxSVC.
|
---|
4 | */
|
---|
5 |
|
---|
6 | /*
|
---|
7 | * Copyright (C) 2013-2022 Oracle and/or its affiliates.
|
---|
8 | *
|
---|
9 | * This file is part of VirtualBox base platform packages, as
|
---|
10 | * available from https://www.virtualbox.org.
|
---|
11 | *
|
---|
12 | * This program is free software; you can redistribute it and/or
|
---|
13 | * modify it under the terms of the GNU General Public License
|
---|
14 | * as published by the Free Software Foundation, in version 3 of the
|
---|
15 | * License.
|
---|
16 | *
|
---|
17 | * This program is distributed in the hope that it will be useful, but
|
---|
18 | * WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
---|
20 | * General Public License for more details.
|
---|
21 | *
|
---|
22 | * You should have received a copy of the GNU General Public License
|
---|
23 | * along with this program; if not, see <https://www.gnu.org/licenses>.
|
---|
24 | *
|
---|
25 | * SPDX-License-Identifier: GPL-3.0-only
|
---|
26 | */
|
---|
27 |
|
---|
28 | #define LOG_GROUP LOG_GROUP_MAIN_HOSTDRIVE
|
---|
29 | #include "Global.h"
|
---|
30 | #include "HostDriveImpl.h"
|
---|
31 | #include "HostDrivePartitionImpl.h"
|
---|
32 | #include "LoggingNew.h"
|
---|
33 | #include "VirtualBoxImpl.h"
|
---|
34 |
|
---|
35 | #include <iprt/dvm.h>
|
---|
36 | #include <iprt/err.h>
|
---|
37 | #include <iprt/file.h>
|
---|
38 | #include <iprt/path.h>
|
---|
39 | #include <iprt/vfs.h>
|
---|
40 | #include <VBox/com/Guid.h>
|
---|
41 |
|
---|
42 |
|
---|
43 | /*
|
---|
44 | * HostDrive implementation.
|
---|
45 | */
|
---|
46 | DEFINE_EMPTY_CTOR_DTOR(HostDrive)
|
---|
47 |
|
---|
48 | HRESULT HostDrive::FinalConstruct()
|
---|
49 | {
|
---|
50 | return BaseFinalConstruct();
|
---|
51 | }
|
---|
52 |
|
---|
53 | void HostDrive::FinalRelease()
|
---|
54 | {
|
---|
55 | uninit();
|
---|
56 |
|
---|
57 | BaseFinalRelease();
|
---|
58 | }
|
---|
59 |
|
---|
60 | /**
|
---|
61 | * Initializes the instance.
|
---|
62 | */
|
---|
63 | HRESULT HostDrive::initFromPathAndModel(const com::Utf8Str &drivePath, const com::Utf8Str &driveModel)
|
---|
64 | {
|
---|
65 | LogFlowThisFunc(("\n"));
|
---|
66 |
|
---|
67 | AssertReturn(!drivePath.isEmpty(), E_INVALIDARG);
|
---|
68 |
|
---|
69 | /* Enclose the state transition NotReady->InInit->Ready */
|
---|
70 | AutoInitSpan autoInitSpan(this);
|
---|
71 | AssertReturn(autoInitSpan.isOk(), E_FAIL);
|
---|
72 |
|
---|
73 | m.partitioningType = PartitioningType_MBR;
|
---|
74 | m.drivePath = drivePath;
|
---|
75 | m.model = driveModel;
|
---|
76 | m.partitions.clear();
|
---|
77 |
|
---|
78 | /*
|
---|
79 | * Try open the drive so we can extract futher details,
|
---|
80 | * like the size, sector size and partitions.
|
---|
81 | */
|
---|
82 | HRESULT hrc = E_FAIL;
|
---|
83 | RTFILE hRawFile = NIL_RTFILE;
|
---|
84 | int vrc = RTFileOpen(&hRawFile, drivePath.c_str(), RTFILE_O_READ | RTFILE_O_OPEN | RTFILE_O_DENY_NONE);
|
---|
85 | if (RT_SUCCESS(vrc))
|
---|
86 | {
|
---|
87 | vrc = RTFileQuerySize(hRawFile, &m.cbDisk);
|
---|
88 | int vrc2 = RTFileQuerySectorSize(hRawFile, &m.cbSector);
|
---|
89 | if (RT_FAILURE(vrc2))
|
---|
90 | vrc = vrc2;
|
---|
91 | if (RT_SUCCESS(vrc))
|
---|
92 | {
|
---|
93 | /*
|
---|
94 | * Hand it to DVM.
|
---|
95 | */
|
---|
96 | RTVFSFILE hVfsFile = NIL_RTVFSFILE;
|
---|
97 | vrc = RTVfsFileFromRTFile(hRawFile, RTFILE_O_READ | RTFILE_O_OPEN | RTFILE_O_DENY_NONE /*fOpen*/,
|
---|
98 | true /*fLeaveOpen*/, &hVfsFile);
|
---|
99 | if (RT_SUCCESS(vrc))
|
---|
100 | {
|
---|
101 | RTDVM hVolMgr = NIL_RTDVM;
|
---|
102 | vrc = RTDvmCreate(&hVolMgr, hVfsFile, m.cbSector, 0 /*fFlags*/);
|
---|
103 | if (RT_SUCCESS(vrc))
|
---|
104 | {
|
---|
105 | vrc = RTDvmMapOpen(hVolMgr);
|
---|
106 | if (RT_SUCCESS(vrc))
|
---|
107 | {
|
---|
108 | hrc = S_OK;
|
---|
109 |
|
---|
110 | /*
|
---|
111 | * Get details.
|
---|
112 | */
|
---|
113 | switch (RTDvmMapGetFormatType(hVolMgr))
|
---|
114 | {
|
---|
115 | case RTDVMFORMATTYPE_GPT:
|
---|
116 | m.partitioningType = PartitioningType_GPT;
|
---|
117 | break;
|
---|
118 | case RTDVMFORMATTYPE_MBR:
|
---|
119 | m.partitioningType = PartitioningType_MBR;
|
---|
120 | break;
|
---|
121 | case RTDVMFORMATTYPE_BSD_LABEL:
|
---|
122 | AssertMsgFailed(("TODO\n"));
|
---|
123 | break;
|
---|
124 | default:
|
---|
125 | AssertFailed();
|
---|
126 | }
|
---|
127 |
|
---|
128 | RTUUID Uuid = RTUUID_INITIALIZE_NULL;
|
---|
129 | if (RT_SUCCESS(RTDvmMapQueryDiskUuid(hVolMgr, &Uuid)))
|
---|
130 | m.uuid = Uuid;
|
---|
131 |
|
---|
132 | /*
|
---|
133 | * Enumerate volumes and tuck them into the partitions list..
|
---|
134 | */
|
---|
135 | uint32_t const cVolumes = RTDvmMapGetValidVolumes(hVolMgr);
|
---|
136 | RTDVMVOLUME hVol = NIL_RTDVMVOLUME;
|
---|
137 | for (uint32_t i = 0; i < cVolumes; i++)
|
---|
138 | {
|
---|
139 | /* Enumeration cruft: */
|
---|
140 | RTDVMVOLUME hVolNext = NIL_RTDVMVOLUME;
|
---|
141 | if (i == 0)
|
---|
142 | vrc = RTDvmMapQueryFirstVolume(hVolMgr, &hVolNext);
|
---|
143 | else
|
---|
144 | vrc = RTDvmMapQueryNextVolume(hVolMgr, hVol, &hVolNext);
|
---|
145 | AssertRCBreakStmt(vrc, hrc = Global::vboxStatusCodeToCOM(vrc));
|
---|
146 |
|
---|
147 | uint32_t cRefs = RTDvmVolumeRelease(hVol);
|
---|
148 | Assert(cRefs != UINT32_MAX); RT_NOREF(cRefs);
|
---|
149 | hVol = hVolNext;
|
---|
150 |
|
---|
151 | /* Instantiate a new partition object and add it to the list: */
|
---|
152 | ComObjPtr<HostDrivePartition> ptrHostPartition;
|
---|
153 | hrc = ptrHostPartition.createObject();
|
---|
154 | if (SUCCEEDED(hrc))
|
---|
155 | hrc = ptrHostPartition->initFromDvmVol(hVol);
|
---|
156 | if (SUCCEEDED(hrc))
|
---|
157 | try
|
---|
158 | {
|
---|
159 | m.partitions.push_back(ptrHostPartition);
|
---|
160 | }
|
---|
161 | catch (std::bad_alloc &)
|
---|
162 | {
|
---|
163 | AssertFailedBreakStmt(hrc = E_OUTOFMEMORY);
|
---|
164 | }
|
---|
165 | }
|
---|
166 | RTDvmVolumeRelease(hVol);
|
---|
167 | }
|
---|
168 | else
|
---|
169 | hrc = Global::vboxStatusCodeToCOM(vrc);
|
---|
170 | RTDvmRelease(hVolMgr);
|
---|
171 | }
|
---|
172 | else
|
---|
173 | hrc = Global::vboxStatusCodeToCOM(vrc);
|
---|
174 | RTVfsFileRelease(hVfsFile);
|
---|
175 | }
|
---|
176 | else
|
---|
177 | hrc = Global::vboxStatusCodeToCOM(vrc);
|
---|
178 | }
|
---|
179 | else /* VERR_IO_NOT_READ / STATUS_NO_MEDIA_IN_DEVICE is likely for card readers on windows. */
|
---|
180 | hrc = Global::vboxStatusCodeToCOM(vrc);
|
---|
181 | RTFileClose(hRawFile);
|
---|
182 | }
|
---|
183 | else
|
---|
184 | {
|
---|
185 | /*
|
---|
186 | * We don't use the Global::vboxStatusCodeToCOM(vrc) here
|
---|
187 | * because RTFileOpen can return some error which causes
|
---|
188 | * the assertion and breaks original idea of returning
|
---|
189 | * the object in the limited state.
|
---|
190 | */
|
---|
191 | if ( vrc == VERR_RESOURCE_BUSY
|
---|
192 | || vrc == VERR_ACCESS_DENIED)
|
---|
193 | hrc = E_ACCESSDENIED;
|
---|
194 | else
|
---|
195 | hrc = VBOX_E_IPRT_ERROR;
|
---|
196 | }
|
---|
197 |
|
---|
198 | /* Confirm a successful initialization */
|
---|
199 | if (SUCCEEDED(hrc))
|
---|
200 | autoInitSpan.setSucceeded();
|
---|
201 | else
|
---|
202 | autoInitSpan.setLimited(hrc);
|
---|
203 | return S_OK;
|
---|
204 | }
|
---|
205 |
|
---|
206 | /**
|
---|
207 | * Uninitializes the instance.
|
---|
208 | * Called either from FinalRelease() or by the parent when it gets destroyed.
|
---|
209 | */
|
---|
210 | void HostDrive::uninit()
|
---|
211 | {
|
---|
212 | LogFlowThisFunc(("\n"));
|
---|
213 |
|
---|
214 | /* Enclose the state transition Ready->InUninit->NotReady */
|
---|
215 | AutoUninitSpan autoUninitSpan(this);
|
---|
216 | if (autoUninitSpan.uninitDone())
|
---|
217 | return;
|
---|
218 |
|
---|
219 | m.drivePath.setNull();
|
---|
220 | m.partitions.clear();
|
---|
221 | }
|
---|
222 |
|
---|
223 |
|
---|
224 | /*********************************************************************************************************************************
|
---|
225 | * IHostDrive properties *
|
---|
226 | *********************************************************************************************************************************/
|
---|
227 |
|
---|
228 | HRESULT HostDrive::getPartitioningType(PartitioningType_T *aPartitioningType)
|
---|
229 | {
|
---|
230 | *aPartitioningType = m.partitioningType;
|
---|
231 | return S_OK;
|
---|
232 | }
|
---|
233 |
|
---|
234 | HRESULT HostDrive::getDrivePath(com::Utf8Str &aDrivePath)
|
---|
235 | {
|
---|
236 | aDrivePath = m.drivePath;
|
---|
237 | return S_OK;
|
---|
238 | }
|
---|
239 |
|
---|
240 | HRESULT HostDrive::getUuid(com::Guid &aUuid)
|
---|
241 | {
|
---|
242 | aUuid = m.uuid;
|
---|
243 | return S_OK;
|
---|
244 | }
|
---|
245 |
|
---|
246 | HRESULT HostDrive::getSectorSize(ULONG *aSectorSize)
|
---|
247 | {
|
---|
248 | *aSectorSize = m.cbSector;
|
---|
249 | return S_OK;
|
---|
250 | }
|
---|
251 |
|
---|
252 | HRESULT HostDrive::getSize(LONG64 *aSize)
|
---|
253 | {
|
---|
254 | *aSize = (LONG64)m.cbDisk;
|
---|
255 | if (*aSize < 0)
|
---|
256 | *aSize = INT64_MAX;
|
---|
257 | return S_OK;
|
---|
258 | }
|
---|
259 |
|
---|
260 | HRESULT HostDrive::getModel(com::Utf8Str &aModel)
|
---|
261 | {
|
---|
262 | return aModel.assignEx(m.model);
|
---|
263 | }
|
---|
264 |
|
---|
265 | HRESULT HostDrive::getPartitions(std::vector<ComPtr<IHostDrivePartition> > &aPartitions)
|
---|
266 | {
|
---|
267 | aPartitions = m.partitions;
|
---|
268 | return S_OK;
|
---|
269 | }
|
---|
270 |
|
---|
271 |
|
---|
272 |
|
---|
273 | /* vi: set tabstop=4 shiftwidth=4 expandtab: */
|
---|