mpm: added check-filesystem utility

1 files changed, 67 insertions, 0 deletions

diff --git a/mpm/python/usrp_mpm/sys_utils/filesystem_status.py b/mpm/python/usrp_mpm/sys_utils/filesystem_status.py
new file mode 100644
index 000000000..865741316
--- /dev/null
+++ b/mpm/python/usrp_mpm/sys_utils/filesystem_status.py
@@ -0,0 +1,67 @@
+#
+# Copyright 2020 Ettus Research, a National Instruments Company
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+#
+"""
+Utilities for checking the filesystem status
+"""
+
+import hashlib
+import pathlib
+import subprocess
+import time
+
+def get_uhd_version(filesystem_root='/'):
+    def parse_uhd_version(versionstring):
+        assert(versionstring[0:4] == 'UHD ')
+        array = versionstring[4:].split('-')
+        assert(len(array) == 3)
+        return {
+            'version': array[0],
+            'appendix': array[1],
+            'githash': array[2],
+        }
+    file = pathlib.Path(filesystem_root, 'usr/bin/uhd_config_info')
+    versionstring = subprocess.check_output([file, '--version']).decode('utf-8').splitlines()[0]
+    return parse_uhd_version(versionstring)
+
+def get_mender_artifact(filesystem_root='/', parse_manually=False):
+    def parse_artifact(output):
+        for line in output.splitlines():
+            if line.startswith('artifact_name='):
+                return line[14:]
+        return None
+    if filesystem_root != '/':
+        parse_manually = True
+    if parse_manually:
+        # parse mender artifact manually
+        file = pathlib.Path(filesystem_root, 'etc/mender/artifact_info')
+        if not file.exists():
+            return 'FILE NOT FOUND'
+        return parse_artifact(file.read_text())
+    else:
+        output = subprocess.check_output(['/usr/bin/mender', '-show-artifact']).decode('utf-8')
+        return output.splitlines()[0]
+
+def get_fs_version(filesystem_root='/'):
+    file = pathlib.Path(filesystem_root, 'etc/version')
+    if not file.exists():
+        return 'FILE NOT FOUND'
+    return file.read_text().splitlines()[0]
+
+def get_opkg_status_date(date_only=False, filesystem_root='/'):
+    if date_only:
+        tformat = "%Y-%m-%d"
+    else:
+        tformat = "%Y-%m-%d %H:%M:%S"
+    file = pathlib.Path(filesystem_root, 'var/lib/opkg/status')
+    if not file.exists():
+        return 'FILE NOT FOUND'
+    return time.strftime(tformat, time.gmtime(file.stat().st_mtime))
+
+def get_opkg_status_md5sum(filesystem_root='/'):
+    file = pathlib.Path(filesystem_root, 'var/lib/opkg/status')
+    if not file.exists():
+        return 'FILE NOT FOUND'
+    return hashlib.md5sum(file.read_text()).hexdigest()