dissect.hypervisor 3.15.dev1__tar.gz → 3.15.dev2__tar.gz

{dissect_hypervisor-3.15.dev1/dissect.hypervisor.egg-info → dissect_hypervisor-3.15.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dissect.hypervisor
-Version: 3.15.dev1
+Version: 3.15.dev2
 Summary: A Dissect module implementing parsers for various hypervisor disk, backup and configuration files
 Author-email: Dissect Team <dissect@fox-it.com>
 License: Affero General Public License v3

dissect_hypervisor-3.15.dev2/dissect/hypervisor/descriptor/vbox.py

@@ -0,0 +1,17 @@
+from typing import IO, Iterator
+from xml.etree.ElementTree import Element
+
+from defusedxml import ElementTree
+
+
+class VBox:
+    VBOX_XML_NAMESPACE = "{http://www.virtualbox.org/}"
+
+    def __init__(self, fh: IO):
+        self._xml: Element = ElementTree.fromstring(fh.read())
+
+    def disks(self) -> Iterator[str]:
+        for hdd_elem in self._xml.findall(f".//{self.VBOX_XML_NAMESPACE}HardDisk[@location][@type='Normal']"):
+            # Allow format specifier to be case-insensitive (i.e. VDI, vdi)
+            if (format := hdd_elem.get("format")) and format.lower() == "vdi":
+                yield hdd_elem.attrib["location"]

{dissect_hypervisor-3.15.dev1 → dissect_hypervisor-3.15.dev2/dissect.hypervisor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dissect.hypervisor
-Version: 3.15.dev1
+Version: 3.15.dev2
 Summary: A Dissect module implementing parsers for various hypervisor disk, backup and configuration files
 Author-email: Dissect Team <dissect@fox-it.com>
 License: Affero General Public License v3

dissect_hypervisor-3.15.dev2/tests/test_vbox.py

@@ -0,0 +1,41 @@
+from io import StringIO
+
+from dissect.hypervisor.descriptor.vbox import VBox
+
+
+def test_vbox() -> None:
+    xml = """
+    <?xml version="1.0"?>
+    <VirtualBox xmlns="http://www.virtualbox.org/">
+        <Machine>
+            <MediaRegistry>
+                <HardDisks>
+                    <HardDisk location="os2warp4.vdi" format="VDI" type="Normal" />
+                </HardDisks>
+            </MediaRegistry>
+        </Machine>
+    </VirtualBox>
+    """
+
+    with StringIO(xml.strip()) as fh:
+        vbox = VBox(fh)
+        assert next(vbox.disks()) == "os2warp4.vdi"
+
+
+def test_vbox_lowercase_disk_format() -> None:
+    xml = """
+    <?xml version="1.0"?>
+    <VirtualBox xmlns="http://www.virtualbox.org/">
+        <Machine>
+            <MediaRegistry>
+                <HardDisks>
+                    <HardDisk location="WinDev2407Eval-disk001.vdi" format="vdi" type="Normal" />
+                </HardDisks>
+            </MediaRegistry>
+        </Machine>
+    </VirtualBox>
+    """
+
+    with StringIO(xml.strip()) as fh:
+        vbox = VBox(fh)
+        assert next(vbox.disks()) == "WinDev2407Eval-disk001.vdi"

dissect_hypervisor-3.15.dev1/dissect/hypervisor/descriptor/vbox.py

@@ -1,17 +0,0 @@
-from typing import IO, Iterator
-from xml.etree.ElementTree import Element
-
-from defusedxml import ElementTree
-
-
-class VBox:
-    VBOX_XML_NAMESPACE = "{http://www.virtualbox.org/}"
-
-    def __init__(self, fh: IO):
-        self._xml: Element = ElementTree.fromstring(fh.read())
-
-    def disks(self) -> Iterator[str]:
-        for hdd_elem in self._xml.findall(
-            f".//{self.VBOX_XML_NAMESPACE}HardDisk[@location][@format='VDI'][@type='Normal']"
-        ):
-            yield hdd_elem.attrib["location"]

dissect_hypervisor-3.15.dev1/tests/test_vbox.py

@@ -1,22 +0,0 @@
-from io import StringIO
-
-from dissect.hypervisor.descriptor.vbox import VBox
-
-
-def test_vbox():
-    xml = """
-    <?xml version="1.0"?>
-    <VirtualBox xmlns="http://www.virtualbox.org/">
-        <Machine>
-            <MediaRegistry>
-                <HardDisks>
-                    <HardDisk location="os2warp4.vdi" format="VDI" type="Normal" />
-                </HardDisks>
-            </MediaRegistry>
-        </Machine>
-    </VirtualBox>
-    """
-
-    with StringIO(xml.strip()) as fh:
-        vbox = VBox(fh)
-        assert next(vbox.disks()) == "os2warp4.vdi"