policies: Add Debian/Ubuntu policies

Adds support for Debian and Ubuntu based
distributions.

Signed-off-by: Adam Stokes <adam.stokes@canonical.com>

3 files changed, 194 insertions, 0 deletions

diff --git a/sos/plugins/__init__.py b/sos/plugins/__init__.py
index 576196de..a1d38f45 100644
--- a/sos/plugins/__init__.py
+++ b/sos/plugins/__init__.py
@@ -610,6 +610,14 @@ class RedHatPlugin(object):
     """Tagging class to indicate that this plugin works with Red Hat Linux"""
     pass
 
+class UbuntuPlugin(object):
+    """Tagging class to indicate that this plugin works with Ubuntu Linux"""
+    pass
+
+class DebianPlugin(object):
+    """Tagging class to indicate that this plugin works with Debian Linux"""
+    pass
+
 class IndependentPlugin(object):
     """Tagging class that indicates this plugin can run on any platform"""
     pass
diff --git a/sos/policies/debian.py b/sos/policies/debian.py
new file mode 100644
index 00000000..77a5487c
--- /dev/null
+++ b/sos/policies/debian.py
@@ -0,0 +1,154 @@
+from sos import _sos as _
+from sos.plugins import DebianPlugin, IndependentPlugin
+from sos.policies import PackageManager, Policy
+from sos.utilities import shell_out
+
+import os
+import sys
+import re
+
+class DebianPackageManager(PackageManager):
+
+    def _get_deb_list(self):
+        pkg_list = shell_out(["dpkg-query",
+            "-W",
+            "-f=",
+        "'${Package}|${Version}\\n' \*"]).splitlines()
+        self._debs = {}
+        for pkg in pkg_list:
+            name, version = pkg.split("|")
+            self._debs[name] = {
+                    'name': name,
+                    'version': version
+                    }
+
+    def allPkgsByName(self, name):
+        return fnmatch.filter(self.allPkgs().keys(), name)
+
+    def allPkgsByNameRegex(self, regex_name, flags=None):
+        reg = re.compile(regex_name, flags)
+        return [pkg for pkg in self.allPkgs().keys() if reg.match(pkg)]
+
+    def pkgByName(self, name):
+        try:
+            self.AllPkgsByName(name)[-1]
+        except Exception:
+            return None
+
+    def allPkgs(self):
+        if not self._debs:
+            self._debs = self._get_deb_list()
+        return self._debs
+
+    def pkgNVRA(self, pkg):
+        fields = pkg.split("-")
+        version, release, arch = fields[-3:]
+        name = "-".join(fields[:-3])
+        return (name, version, release, arch)
+        
+class DebianPolicy(Policy):
+    def __init__(self):
+        super(DebianPolicy, self).__init__()
+        self.reportName = ""
+        self.ticketNumber = ""
+        self.package_manager = DebianPackageManager()
+
+    def validatePlugin(self, plugin_class):
+        "Checks that the plugin will execute given the environment"
+        return issubclass(plugin_class, DebianPlugin) or issubclass(plugin_class, IndependentPlugin)
+
+    @classmethod
+    def check(self):
+        """This method checks to see if we are running on Debian.
+           It returns True or False."""
+        if os.path.isfile('/etc/debian_version'):
+            return True
+        return False
+
+    def preferedArchive(self):
+        from sos.utilities import TarFileArchive
+        return TarFileArchive
+
+    def getPreferredHashAlgorithm(self):
+        checksum = "md5"
+        try:
+            fp = open("/proc/sys/crypto/fips_enabled", "r")
+        except:
+            return checksum
+
+        fips_enabled = fp.read()
+        if fips_enabled.find("1") >= 0:
+            checksum = "sha256"
+        fp.close()
+        return checksum
+
+    def pkgByName(self, name):
+        return self.package_manager.pkgByName(name)
+
+    def runlevelDefault(self):
+        try:
+            with open("/etc/inittab") as fp:
+                pattern = r"id:(\d{1}):initdefault:"
+                text = fp.read()
+                return int(re.findall(pattern, text)[0])
+        except:
+            return 3
+
+    def kernelVersion(self):
+        return self.release
+
+    def hostName(self):
+        return self.hostname
+
+    def debianVersion(self):
+        try:
+            fp = open("/etc/debian_version").read()
+            if "wheezy/sid" in fp:
+                return 6
+            fp.close()
+        except:
+            pass
+        return False
+
+    def isKernelSMP(self):
+        return self.smp
+
+    def getArch(self):
+        return self.machine
+
+    def preWork(self):
+        # this method will be called before the gathering begins
+
+        localname = self.hostName()
+
+        if not self.commons['cmdlineopts'].batch and not self.commons['cmdlineopts'].silent:
+            try:
+                self.reportName = raw_input(_("Please enter your first initial and last name [%s]: ") % localname)
+                self.reportName = re.sub(r"[^a-zA-Z.0-9]", "", self.reportName)
+
+                self.ticketNumber = raw_input(_("Please enter the case number that you are generating this report for: "))
+                self.ticketNumber = re.sub(r"[^0-9]", "", self.ticketNumber)
+                self._print()
+            except:
+                self._print()
+                sys.exit(0)
+
+        if len(self.reportName) == 0:
+            self.reportName = localname
+
+        if self.commons['cmdlineopts'].customerName:
+            self.reportName = self.commons['cmdlineopts'].customerName
+            self.reportName = re.sub(r"[^a-zA-Z.0-9]", "", self.reportName)
+
+        if self.commons['cmdlineopts'].ticketNumber:
+            self.ticketNumber = self.commons['cmdlineopts'].ticketNumber
+            self.ticketNumber = re.sub(r"[^0-9]", "", self.ticketNumber)
+
+        return
+
+    def packageResults(self, archive_filename):
+        self._print(_("Creating compressed archive..."))
+
+    def get_msg(self):
+        msg_dict = {"distro": "Debian"}
+        return self.msg % msg_dict
diff --git a/sos/policies/ubuntu.py b/sos/policies/ubuntu.py
new file mode 100644
index 00000000..c841b22b
--- /dev/null
+++ b/sos/policies/ubuntu.py
@@ -0,0 +1,32 @@
+from sos import _sos as _
+from sos.plugins import UbuntuPlugin, IndependentPlugin
+from sos.policies.debian import DebianPolicy, DebianPackageManager
+from sos.utilities import shell_out
+
+import os
+        
+class UbuntuPolicy(DebianPolicy):
+    def __init__(self):
+        super(UbuntuPolicy, self).__init__()
+
+    def validatePlugin(self, plugin_class):
+        "Checks that the plugin will execute given the environment"
+        return issubclass(plugin_class, UbuntuPlugin) or issubclass(plugin_class, IndependentPlugin)
+
+    @classmethod
+    def check(self):
+        """This method checks to see if we are running on Ubuntu.
+           It returns True or False."""
+        if os.path.isfile('/etc/lsb-release'):
+            try:
+                fp = open('/etc/lsb-release', 'r')
+                if "Ubuntu" in fp.read():
+                    fp.close()
+                    return True
+            except:
+                return False
+        return False
+
+    def get_msg(self):
+        msg_dict = {"distro": "Ubuntu"}
+        return self.msg % msg_dict