New disks grain

saltstack · Jun 2, 2015 · 1036764 · 1036764
1 parent c07af62
commit 1036764
Show file tree

Hide file tree

Showing 2 changed files with 129 additions and 84 deletions.
diff --git a/salt/grains/disks.py b/salt/grains/disks.py
@@ -0,0 +1,129 @@
+# -*- coding: utf-8 -*-
+'''
+    Detect disks
+'''
+from __future__ import absolute_import
+
+# Import python libs
+import glob
+import logging
+import re
+
+# Import salt libs
+import salt.utils
+
+# Solve the Chicken and egg problem where grains need to run before any
+# of the modules are loaded and are generally available for any usage.
+import salt.modules.cmdmod
+
+__salt__ = {
+    'cmd.run': salt.modules.cmdmod._run_quiet
+}
+
+log = logging.getLogger(__name__)
+
+
+def disks():
+    '''
+    Return list of disk devices
+    '''
+    if salt.utils.is_freebsd():
+        return _freebsd_disks()
+    elif salt.utils.is_linux():
+        return {'SSDs': _linux_ssds()}
+    else:
+        log.trace('Disk grain does not support OS')
+
+
+def _clean_keys(key):
+    key = key.replace(' ', '_')
+    key = key.replace('(', '')
+    key = key.replace(')', '')
+    return key
+
+
+class _camconsts:
+    PROTOCOL = 'protocol'
+    DEVICE_MODEL = 'device model'
+    FIRMWARE_REVISION = 'firmware revision'
+    SERIAL_NUMBER = 'serial number'
+    WWN = 'WWN'
+    SECTOR_SIZE = 'sector size'
+    MEDIA_RPM = 'media RPM'
+
+
+_identify_attribs = {key: _camconsts.__dict__[key] for key in
+                     _camconsts.__dict__ if not key.startswith('__')}.values()
+
+
+def _freebsd_disks():
+    ret = {'disks': {}, 'SSDs': []}
+    sysctl = salt.utils.which('sysctl')
+    devices = __salt__['cmd.run']('{0} -n kern.disks'.format(sysctl))
+    SSD_TOKEN = 'non-rotating'
+
+    for device in devices.split(' '):
+        cam = _freebsd_camcontrol(device)
+        ret['disks'][device] = cam
+        if cam.get(_clean_keys(_camconsts.MEDIA_RPM)) == SSD_TOKEN:
+            ret['SSDs'].append(device)
+
+    return ret
+
+
+def _freebsd_camcontrol(device):
+    camcontrol = salt.utils.which('camcontrol')
+
+    ret = {}
+
+    def parse_identify_attribs(line):
+        for attrib in _identify_attribs:
+            search = re.search('^{}\s+(.*)'.format(attrib), line)
+            if search:
+                ret[_clean_keys(attrib)] = search.group(1)
+
+    identify = __salt__['cmd.run']('{0} identify {1}'.format(camcontrol,
+                                                             device))
+    for line in identify.split('\n'):
+        parse_identify_attribs(line)
+
+    def parse_inquiry(inquiry):
+        if not ret.get(_clean_keys(_camconsts.DEVICE_MODEL)):
+            model = re.search('\s<(.+?)>', inquiry)
+            if model:
+                ret[_clean_keys(_camconsts.DEVICE_MODEL)] = model.group(1)
+        if not ret.get(_clean_keys(_camconsts.SERIAL_NUMBER)):
+            sn = re.search('\sSerial Number\s+(\w+)\s', inquiry)
+            if sn:
+                ret[_clean_keys(_camconsts.SERIAL_NUMBER)] = sn.group(1)
+
+    inquiry = __salt__['cmd.run']('{0} inquiry {1}'.format(camcontrol, device))
+    parse_inquiry(inquiry)
+
+    return ret
+
+
+def _linux_ssds():
+    '''
+    Return list of disk devices that are SSD (non-rotational)
+    '''
+    ssd_devices = []
+
+    for entry in glob.glob('/sys/block/*/queue/rotational'):
+        with salt.utils.fopen(entry) as entry_fp:
+            device = entry.split('/')[3]
+            flag = entry_fp.read(1)
+            if flag == '0':
+                ssd_devices.append(device)
+                log.trace('Device {0} reports itself as an SSD'.format(device))
+            elif flag == '1':
+                log.trace('Device {0} does not report itself as an SSD'
+                          .format(device))
+            else:
+                log.trace('Unable to identify device {0} as an SSD or not.'
+                          ' It does not report 0 or 1'.format(device))
+    return ssd_devices
+
+
+if __name__ == '__main__':
+    print disks()
diff --git a/salt/grains/ssds.py b/salt/grains/ssds.py