1 files changed, 126 insertions, 0 deletions
diff --git a/src/VBox/Main/include/HostDrivePartitionImpl.h b/src/VBox/Main/include/HostDrivePartitionImpl.h
new file mode 100644
index 00000000..ddd65262
--- /dev/null
+++ b/src/VBox/Main/include/HostDrivePartitionImpl.h
@@ -0,0 +1,126 @@
+/* $Id: HostDrivePartitionImpl.h $ */
+/** @file
+ * VirtualBox Main - IHostDrivePartition implementation, VBoxSVC.
+ */
+
+/*
+ * Copyright (C) 2013-2022 Oracle and/or its affiliates.
+ *
+ * This file is part of VirtualBox base platform packages, as
+ * available from https://www.virtualbox.org.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation, in version 3 of the
+ * License.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, see <https://www.gnu.org/licenses>.
+ *
+ * SPDX-License-Identifier: GPL-3.0-only
+ */
+
+#ifndef MAIN_INCLUDED_HostDrivePartitionImpl_h
+#define MAIN_INCLUDED_HostDrivePartitionImpl_h
+#ifndef RT_WITHOUT_PRAGMA_ONCE
+# pragma once
+#endif
+
+#include "HostDrivePartitionWrap.h"
+
+#include <iprt/dvm.h>
+
+class ATL_NO_VTABLE HostDrivePartition
+    : public HostDrivePartitionWrap
+{
+public:
+    DECLARE_COMMON_CLASS_METHODS(HostDrivePartition)
+
+    HRESULT FinalConstruct();
+    void FinalRelease();
+
+    /** @name Public initializer/uninitializer for internal purposes only.
+     * @{ */
+    HRESULT initFromDvmVol(RTDVMVOLUME hVol);
+    void uninit();
+    /** @} */
+
+private:
+    /** @name wrapped IHostDrivePartition properties
+     * @{  */
+    /* Common: */
+    virtual HRESULT getNumber(ULONG *aNumber) RT_OVERRIDE               { *aNumber        = m.number;       return S_OK; }
+    virtual HRESULT getSize(LONG64 *aSize) RT_OVERRIDE                  { *aSize          = m.cbVol;        return S_OK; }
+    virtual HRESULT getStart(LONG64 *aStart) RT_OVERRIDE                { *aStart         = m.offStart;     return S_OK; }
+    virtual HRESULT getType(PartitionType_T *aType) RT_OVERRIDE         { *aType          = m.enmType;      return S_OK; }
+    virtual HRESULT getActive(BOOL *aActive) RT_OVERRIDE                { *aActive        = m.active;       return S_OK; }
+    /* MBR: */
+    virtual HRESULT getTypeMBR(ULONG *aTypeMBR) RT_OVERRIDE             { *aTypeMBR       = m.bMBRType;     return S_OK; }
+    virtual HRESULT getStartCylinder(ULONG *aStartCylinder) RT_OVERRIDE { *aStartCylinder = m.firstCylinder; return S_OK; }
+    virtual HRESULT getStartHead(ULONG *aStartHead) RT_OVERRIDE         { *aStartHead     = m.firstHead;    return S_OK; }
+    virtual HRESULT getStartSector(ULONG *aStartSector) RT_OVERRIDE     { *aStartSector   = m.firstSector;  return S_OK; }
+    virtual HRESULT getEndCylinder(ULONG *aEndCylinder) RT_OVERRIDE     { *aEndCylinder   = m.lastCylinder; return S_OK; }
+    virtual HRESULT getEndHead(ULONG *aEndHead) RT_OVERRIDE             { *aEndHead       = m.lastHead;     return S_OK; }
+    virtual HRESULT getEndSector(ULONG *aEndSector) RT_OVERRIDE         { *aEndSector     = m.lastSector;   return S_OK; }
+    /* GPT: */
+    virtual HRESULT getTypeUuid(com::Guid &aTypeUuid) RT_OVERRIDE       { aTypeUuid       = m.typeUuid;     return S_OK; }
+    virtual HRESULT getUuid(com::Guid &aUuid) RT_OVERRIDE               { aUuid           = m.uuid;         return S_OK; }
+    virtual HRESULT getName(com::Utf8Str &aName) RT_OVERRIDE            { return aName.assignEx(m.name); };
+    /** @} */
+
+    /** Data. */
+    struct Data
+    {
+        Data()
+            : number(0)
+            , cbVol(0)
+            , offStart(0)
+            , enmType(PartitionType_Unknown)
+            , active(FALSE)
+            , bMBRType(0)
+            , firstCylinder(0)
+            , firstHead(0)
+            , firstSector(0)
+            , lastCylinder(0)
+            , lastHead(0)
+            , lastSector(0)
+            , typeUuid()
+            , uuid()
+            , name()
+        {
+        }
+
+        ULONG number;
+        LONG64 cbVol;
+        LONG64 offStart;
+        PartitionType_T enmType;
+        BOOL active;
+        /** @name MBR specifics
+         * @{ */
+        uint8_t bMBRType;
+        uint16_t firstCylinder;
+        uint8_t firstHead;
+        uint8_t firstSector;
+        uint16_t lastCylinder;
+        uint8_t lastHead;
+        uint8_t lastSector;
+        /** @} */
+        /** @name GPT specifics
+         * @{ */
+        com::Guid typeUuid;
+        com::Guid uuid;
+        com::Utf8Str name;
+        /** @} */
+    };
+
+    Data m;
+};
+
+#endif /* !MAIN_INCLUDED_HostDrivePartitionImpl_h */
+
+/* vi: set tabstop=4 shiftwidth=4 expandtab: */