1 files changed, 97 insertions, 0 deletions
diff --git a/storage/devicelibs/edd.py b/storage/devicelibs/edd.py
new file mode 100644
index 0000000..da03914
--- /dev/null
+++ b/storage/devicelibs/edd.py
@@ -0,0 +1,97 @@
+#
+# edd.py
+# BIOS EDD data parsing functions
+#
+# Copyright (C) 2010  Red Hat, Inc.  All rights reserved.
+#
+# 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, see <http://www.gnu.org/licenses/>.
+#
+# Author(s): Hans de Goede <hdegoede@redhat.com>
+#
+
+import os
+import struct
+
+import logging
+log = logging.getLogger("storage")
+
+def get_edd_dict(devices):
+    """Given an array of devices return a dict with the BIOS ID for them."""
+    edd_dict = {}
+
+    for biosdev in range(80, 80 + 15):
+        sysfspath = "/sys/firmware/edd/int13_dev%d" % biosdev
+        if not os.path.exists(sysfspath):
+            break # We are done
+
+        sysfspath = "/sys/firmware/edd/int13_dev%d/mbr_signature" % biosdev
+        if not os.path.exists(sysfspath):
+            log.warning("No mbrsig for biosdev: %d" % biosdev)
+            continue
+
+        try:
+            file = open(sysfspath, "r")
+            eddsig = file.read()
+            file.close()
+        except (IOError, OSError) as e:
+            log.warning("Error reading EDD mbrsig for %d: %s" %
+                        (biosdev, str(e)))
+            continue
+
+        sysfspath = "/sys/firmware/edd/int13_dev%d/sectors" % biosdev
+        try:
+            file = open(sysfspath, "r")
+            eddsize = file.read()
+            file.close()
+        except (IOError, OSError) as e:
+            eddsize = None
+
+        found = []
+        for dev in devices:
+            try:
+                fd = os.open(dev.path, os.O_RDONLY)
+                os.lseek(fd, 440, 0) 
+                mbrsig = struct.unpack('I', os.read(fd, 4))
+                os.close(fd)
+            except OSError as e:
+                log.warning("Error reading mbrsig from disk %s: %s" %
+                            (dev.name, str(e)))
+                continue
+
+            mbrsigStr = "0x%08x\n" % mbrsig
+            if mbrsigStr == eddsig:
+                if eddsize:
+                    sysfspath = "/sys%s/size" % dev.sysfsPath
+                    try:
+                        file = open(sysfspath, "r")
+                        size = file.read()
+                        file.close()
+                    except (IOError, OSError) as e:
+                        log.warning("Error getting size for: %s" % dev.name)
+                        continue
+                    if eddsize != size:
+                        continue
+                found.append(dev.name)
+
+        if not found:
+            log.error("No matching mbr signature found for biosdev %d" %
+                      biosdev)
+        elif len(found) > 1:
+            log.error("Multiple signature matches found for biosdev %d: %s" %
+                      (biosdev, str(found)))
+        else:
+            log.info("Found %s for biosdev %d" %(found[0], biosdev))
+            edd_dict[found[0]] = biosdev
+
+    return edd_dict