[virt-tools-list] [PATCH 4/5] virtinst: Add --smbios command line option

[Charles Arnold]

Classes for building and installing with libvirt <sysinfo> XML
for the defining of SMBios information to a guest.

Signed-off-by: Charles Arnold <carnold at suse.com>

diff --git a/virtinst/sysinfo.py b/virtinst/sysinfo.py
new file mode 100644
index 0000000..eca7921
--- /dev/null
+++ b/virtinst/sysinfo.py
@@ -0,0 +1,158 @@
+#
+# Copyright (C) 2016 Red Hat, Inc.
+# Copyright (C) 2016 SUSE LINUX Products GmbH, Nuernberg, Germany.
+# Charles Arnold <carnold at suse.com>
+#
+# 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; either version 2 of the License, or
+# (at your option) any later version.
+#
+# 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, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+# MA 02110-1301 USA.
+"""
+Classes for building and installing with libvirt <sysinfo> XML
+"""
+
+import datetime
+
+from . import util
+from .xmlbuilder import XMLBuilder, XMLChildProperty, XMLProperty
+
+
+class _Entry(XMLBuilder):
+    """
+    Entry Information for SMBios Types
+    """
+    _XML_ROOT_NAME = "entry"
+
+    value = XMLProperty(".")
+    name = XMLProperty("./@name")
+
+
+class _SMBiosBios(XMLBuilder):
+    """
+    Type 0: BIOS Information
+    """
+    _XML_ROOT_NAME = "bios"
+
+    entry = XMLChildProperty(_Entry)
+
+    def add_entry(self, key, value):
+        keys = ["vendor", "version", "date", "release"]
+        if key not in keys:
+            return
+        # Libvirt: If supplied, date is in either mm/dd/yy or mm/dd/yyyy format
+        if key == "date":
+            try:
+                datetime.datetime.strptime(value, '%m/%d/%Y')
+            except ValueError:
+                try:
+                    datetime.datetime.strptime(value, '%m/%d/%y')
+                except ValueError:
+                    raise RuntimeError(_("SMBios date string '%s' is invalid.")
+                                % value)
+        entry = _Entry(self)
+        self.add_child(entry)
+        entry.name = key
+        entry.value = value
+
+
+class _SMBiosSystem(XMLBuilder):
+    """
+    Type 1: System Information
+    """
+    _XML_ROOT_NAME = "system"
+
+    entry = XMLChildProperty(_Entry)
+
+    def add_entry(self, key, value, guest):
+        keys = ["manufacturer", "product", "version", "serial", "uuid",
+                "sku", "family"]
+        if key not in keys:
+            return
+        # If a uuid is supplied it must match the guest UUID. This would be
+        # impossible to guess if the guest uuid is autogenerated so just
+        # overwrite the guest uuid with what is passed in assuming it passes
+        # sanity checking.
+        if key == "uuid":
+            try:
+                util.validate_uuid(value)
+            except ValueError:
+                raise ValueError(_("Invalid uuid for SMBios: %s") % value)
+
+            if util.vm_uuid_collision(self.conn, value):
+                raise ValueError(_("UUID '%s' is in use by another guest.") %
+                            value)
+
+            # Override guest uuid with passed in SMBios value (they must match)
+            guest.uuid = value
+
+        entry = _Entry(self)
+        self.add_child(entry)
+        entry.name = key
+        entry.value = value
+
+
+class _SMBiosBaseboard(XMLBuilder):
+    """
+    Type 2: Baseboard (or Module) Information
+    """
+    _XML_ROOT_NAME = "baseBoard"
+
+    entry = XMLChildProperty(_Entry)
+
+    def add_entry(self, key, value):
+        keys = ["manufacturer", "product", "version", "asset", "location"]
+        if key not in keys:
+            return
+        entry = _Entry(self)
+        self.add_child(entry)
+        entry.name = key
+        entry.value = value
+
+
+class SYSInfo(XMLBuilder):
+    """
+    Top level class for <sysinfo type='smbios'> object XML
+    """
+
+    SMBIOS_TYPE_BIOS      = "0"
+    SMBIOS_TYPE_SYSTEM    = "1"
+    SMBIOS_TYPE_BASEBOARD = "2"
+
+    _XML_ROOT_NAME = "sysinfo"
+    _XML_PROP_ORDER = ["bios", "system", "baseBoard"]
+
+    def parse(self, guest, value):
+        val = value.split(',')
+        optdict = dict(s.split('=') for s in val)
+        self.type = "smbios"
+
+        if optdict['type'] == self.SMBIOS_TYPE_BIOS:
+            bios = _SMBiosBios(self.conn)
+            self.add_child(bios)
+            for key in optdict:
+                bios.add_entry(key, optdict[key])
+        elif optdict['type'] == self.SMBIOS_TYPE_SYSTEM:
+            system = _SMBiosSystem(self.conn)
+            self.add_child(system)
+            for key in optdict:
+                system.add_entry(key, optdict[key], guest)
+        elif optdict['type'] == self.SMBIOS_TYPE_BASEBOARD:
+            baseBoard = _SMBiosBaseboard(self.conn)
+            self.add_child(baseBoard)
+            for key in optdict:
+                baseBoard.add_entry(key, optdict[key])
+
+    type = XMLProperty("./@type")
+    bios = XMLChildProperty(_SMBiosBios)
+    system = XMLChildProperty(_SMBiosSystem)
+    baseBoard = XMLChildProperty(_SMBiosBaseboard)