1 files changed, 132 insertions, 0 deletions
diff --git a/sos/report/plugins/ovn_central.py b/sos/report/plugins/ovn_central.py
new file mode 100644
index 00000000..3c564788
--- /dev/null
+++ b/sos/report/plugins/ovn_central.py
@@ -0,0 +1,132 @@
+# Copyright (C) 2018 Mark Michelson <mmichels@redhat.com>
+
+# This file is part of the sos project: https://github.com/sosreport/sos
+#
+# This copyrighted material is made available to anyone wishing to use,
+# modify, copy, or redistribute it subject to the terms and conditions of
+# version 2 of the GNU General Public License.
+#
+# See the LICENSE file in the source distribution for further information.
+
+from sos.plugins import Plugin, RedHatPlugin, DebianPlugin, UbuntuPlugin
+import json
+import os
+import six
+
+
+class OVNCentral(Plugin):
+    """ OVN Northd
+    """
+    plugin_name = "ovn_central"
+    profiles = ('network', 'virt')
+
+    def get_tables_from_schema(self, filename, skip=[]):
+        if self._container_name:
+            cmd = "cat %s" % filename
+            res = self.exec_cmd(cmd, timeout=None, foreground=True,
+                                container=self._container_name)
+            if res['status'] != 0:
+                self._log_error("Could not retrieve DB schema file from "
+                                "container %s" % self._container_name)
+                return
+            try:
+                db = json.loads(res['output'])
+            except Exception:
+                self._log_error("Cannot parse JSON file %s" % filename)
+                return
+        else:
+            try:
+                with open(filename, 'r') as f:
+                    try:
+                        db = json.load(f)
+                    except Exception:
+                        self._log_error(
+                            "Cannot parse JSON file %s" % filename)
+                        return
+            except IOError as ex:
+                self._log_error(
+                    "Could not open DB schema file %s: %s" % (filename, ex))
+                return
+        try:
+            return [table for table in six.iterkeys(
+                db['tables']) if table not in skip]
+        except AttributeError:
+            self._log_error("DB schema %s has no 'tables' key" % filename)
+
+    def add_database_output(self, tables, cmds, ovn_cmd):
+        if not tables:
+            return
+        for table in tables:
+            cmds.append('%s list %s' % (ovn_cmd, table))
+
+    def check_enabled(self):
+        return (self.container_exists('ovs-dbs-bundle.*') or
+                super(OVNCentral, self).check_enabled())
+
+    def setup(self):
+        self._container_name = self.get_container_by_name('ovs-dbs-bundle.*')
+
+        ovs_rundir = os.environ.get('OVS_RUNDIR')
+        for pidfile in ['ovnnb_db.pid', 'ovnsb_db.pid', 'ovn-northd.pid']:
+            self.add_copy_spec([
+                os.path.join('/var/lib/openvswitch/ovn', pidfile),
+                os.path.join('/usr/local/var/run/openvswitch', pidfile),
+                os.path.join('/run/openvswitch/', pidfile),
+            ])
+
+            if ovs_rundir:
+                self.add_copy_spec(os.path.join(ovs_rundir, pidfile))
+
+        # Some user-friendly versions of DB output
+        cmds = [
+            'ovn-nbctl show',
+            'ovn-sbctl show',
+            'ovn-sbctl lflow-list',
+            'ovn-nbctl get-ssl',
+            'ovn-nbctl get-connection',
+            'ovn-sbctl get-ssl',
+            'ovn-sbctl get-connection',
+        ]
+
+        schema_dir = '/usr/share/openvswitch'
+
+        nb_tables = self.get_tables_from_schema(os.path.join(
+            schema_dir, 'ovn-nb.ovsschema'))
+        sb_tables = self.get_tables_from_schema(os.path.join(
+            schema_dir, 'ovn-sb.ovsschema'), ['Logical_Flow'])
+
+        self.add_database_output(nb_tables, cmds, 'ovn-nbctl')
+        self.add_database_output(sb_tables, cmds, 'ovn-sbctl')
+
+        # If OVN is containerized, we need to run the above commands inside
+        # the container.
+        cmds = [
+            self.fmt_container_cmd(self._container_name, cmd) for cmd in cmds
+        ]
+
+        self.add_cmd_output(cmds, foreground=True)
+
+        self.add_copy_spec("/etc/sysconfig/ovn-northd")
+
+        ovs_dbdir = os.environ.get('OVS_DBDIR')
+        for dbfile in ['ovnnb_db.db', 'ovnsb_db.db']:
+            self.add_copy_spec([
+                os.path.join('/var/lib/openvswitch/ovn', dbfile),
+                os.path.join('/usr/local/etc/openvswitch', dbfile),
+                os.path.join('/etc/openvswitch', dbfile),
+                os.path.join('/var/lib/openvswitch', dbfile),
+            ])
+            if ovs_dbdir:
+                self.add_copy_spec(os.path.join(ovs_dbdir, dbfile))
+
+        self.add_journal(units="ovn-northd")
+
+
+class RedHatOVNCentral(OVNCentral, RedHatPlugin):
+
+    packages = ('openvswitch-ovn-central', 'ovn2.*-central', )
+
+
+class DebianOVNCentral(OVNCentral, DebianPlugin, UbuntuPlugin):
+
+    packages = ('ovn-central', )