mirror of
https://github.com/prometheus/node_exporter
synced 2024-12-19 12:54:47 +00:00
379 lines
10 KiB
Python
Executable File
379 lines
10 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import argparse
|
|
import collections
|
|
import csv
|
|
import datetime
|
|
import decimal
|
|
import re
|
|
import shlex
|
|
import subprocess
|
|
|
|
device_info_re = re.compile(r'^(?P<k>[^:]+?)(?:(?:\sis|):)\s*(?P<v>.*)$')
|
|
|
|
ata_error_count_re = re.compile(
|
|
r'^Error (\d+) \[\d+\] occurred', re.MULTILINE)
|
|
|
|
self_test_re = re.compile(r'^SMART.*(PASSED|OK)$', re.MULTILINE)
|
|
|
|
device_info_map = {
|
|
'Vendor': 'vendor',
|
|
'Product': 'product',
|
|
'Revision': 'revision',
|
|
'Logical Unit id': 'lun_id',
|
|
'Model Family': 'model_family',
|
|
'Device Model': 'device_model',
|
|
'Serial Number': 'serial_number',
|
|
'Firmware Version': 'firmware_version',
|
|
}
|
|
|
|
smart_attributes_whitelist = {
|
|
'airflow_temperature_cel',
|
|
'command_timeout',
|
|
'current_pending_sector',
|
|
'end_to_end_error',
|
|
'erase_fail_count_total',
|
|
'g_sense_error_rate',
|
|
'hardware_ecc_recovered',
|
|
'host_reads_mib',
|
|
'host_reads_32mib',
|
|
'host_writes_mib',
|
|
'host_writes_32mib',
|
|
'load_cycle_count',
|
|
'media_wearout_indicator',
|
|
'wear_leveling_count',
|
|
'nand_writes_1gib',
|
|
'offline_uncorrectable',
|
|
'power_cycle_count',
|
|
'power_on_hours',
|
|
'program_fail_count',
|
|
'raw_read_error_rate',
|
|
'reallocated_event_count',
|
|
'reallocated_sector_ct',
|
|
'reported_uncorrect',
|
|
'sata_downshift_count',
|
|
'seek_error_rate',
|
|
'spin_retry_count',
|
|
'spin_up_time',
|
|
'start_stop_count',
|
|
'temperature_case',
|
|
'temperature_celsius',
|
|
'temperature_internal',
|
|
'total_lbas_read',
|
|
'total_lbas_written',
|
|
'udma_crc_error_count',
|
|
'unsafe_shutdown_count',
|
|
'workld_host_reads_perc',
|
|
'workld_media_wear_indic',
|
|
'workload_minutes',
|
|
}
|
|
|
|
Metric = collections.namedtuple('Metric', 'name labels value')
|
|
|
|
SmartAttribute = collections.namedtuple('SmartAttribute', [
|
|
'id', 'name', 'flag', 'value', 'worst', 'threshold', 'type', 'updated',
|
|
'when_failed', 'raw_value',
|
|
])
|
|
|
|
|
|
class Device(collections.namedtuple('DeviceBase', 'path opts')):
|
|
"""Representation of a device as found by smartctl --scan output."""
|
|
|
|
@property
|
|
def type(self):
|
|
return self.opts.type
|
|
|
|
@property
|
|
def base_labels(self):
|
|
return {'disk': self.path}
|
|
|
|
def smartctl_select(self):
|
|
return ['--device', self.type, self.path]
|
|
|
|
|
|
def metric_key(metric, prefix=''):
|
|
return '{prefix}{metric.name}'.format(prefix=prefix, metric=metric)
|
|
|
|
|
|
def metric_format(metric, prefix=''):
|
|
key = metric_key(metric, prefix)
|
|
labels = ','.join(
|
|
'{k}="{v}"'.format(k=k, v=v) for k, v in metric.labels.items())
|
|
value = decimal.Decimal(metric.value)
|
|
|
|
return '{key}{{{labels}}} {value}'.format(
|
|
key=key, labels=labels, value=value)
|
|
|
|
|
|
def metric_print_meta(metric, prefix=''):
|
|
key = metric_key(metric, prefix)
|
|
print('# HELP {key} SMART metric {metric.name}'.format(
|
|
key=key, metric=metric))
|
|
print('# TYPE {key} gauge'.format(key=key, metric=metric))
|
|
|
|
|
|
def metric_print(metric, prefix=''):
|
|
print(metric_format(metric, prefix))
|
|
|
|
|
|
def smart_ctl(*args, check=True):
|
|
"""Wrapper around invoking the smartctl binary.
|
|
|
|
Returns:
|
|
(str) Data piped to stdout by the smartctl subprocess.
|
|
"""
|
|
try:
|
|
return subprocess.run(
|
|
['smartctl', *args], stdout=subprocess.PIPE, check=check
|
|
).stdout.decode('utf-8')
|
|
except subprocess.CalledProcessError as e:
|
|
return e.output.decode('utf-8')
|
|
|
|
def smart_ctl_version():
|
|
return smart_ctl('-V').split('\n')[0].split()[1]
|
|
|
|
|
|
def find_devices():
|
|
"""Find SMART devices.
|
|
|
|
Yields:
|
|
(Device) Single device found by smartctl.
|
|
"""
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-d', '--device', dest='type')
|
|
|
|
devices = smart_ctl('--scan-open')
|
|
|
|
for device in devices.split('\n'):
|
|
device = device.strip()
|
|
if not device:
|
|
continue
|
|
|
|
tokens = shlex.split(device, comments=True)
|
|
if not tokens:
|
|
continue
|
|
|
|
yield Device(tokens[0], parser.parse_args(tokens[1:]))
|
|
|
|
|
|
def device_is_active(device):
|
|
"""Returns whenever the given device is currently active or not.
|
|
|
|
Args:
|
|
device: (Device) Device in question.
|
|
|
|
Returns:
|
|
(bool) True if the device is active and False otherwise.
|
|
"""
|
|
try:
|
|
smart_ctl('--nocheck', 'standby', *device.smartctl_select())
|
|
except subprocess.CalledProcessError:
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def device_info(device):
|
|
"""Query device for basic model information.
|
|
|
|
Args:
|
|
device: (Device) Device in question.
|
|
|
|
Returns:
|
|
(generator): Generator yielding:
|
|
|
|
key (str): Key describing the value.
|
|
value (str): Actual value.
|
|
"""
|
|
info_lines = smart_ctl(
|
|
'--info', *device.smartctl_select()
|
|
).strip().split('\n')[3:]
|
|
|
|
matches = (device_info_re.match(l) for l in info_lines)
|
|
return (m.groups() for m in matches if m is not None)
|
|
|
|
|
|
def device_smart_capabilities(device):
|
|
"""Returns SMART capabilities of the given device.
|
|
|
|
Args:
|
|
device: (Device) Device in question.
|
|
|
|
Returns:
|
|
(tuple): tuple containing:
|
|
|
|
(bool): True whenever SMART is available, False otherwise.
|
|
(bool): True whenever SMART is enabled, False otherwise.
|
|
"""
|
|
groups = device_info(device)
|
|
|
|
state = {
|
|
g[1].split(' ', 1)[0]
|
|
for g in groups if g[0] == 'SMART support'}
|
|
|
|
smart_available = 'Available' in state
|
|
smart_enabled = 'Enabled' in state
|
|
|
|
return smart_available, smart_enabled
|
|
|
|
|
|
def collect_device_info(device):
|
|
"""Collect basic device information.
|
|
|
|
Args:
|
|
device: (Device) Device in question.
|
|
|
|
Yields:
|
|
(Metric) metrics describing general device information.
|
|
"""
|
|
values = dict(device_info(device))
|
|
yield Metric('device_info', {
|
|
**device.base_labels,
|
|
**{v: values[k] for k, v in device_info_map.items() if k in values}
|
|
}, True)
|
|
|
|
|
|
def collect_device_health_self_assessment(device):
|
|
"""Collect metric about the device health self assessment.
|
|
|
|
Args:
|
|
device: (Device) Device in question.
|
|
|
|
Yields:
|
|
(Metric) Device health self assessment.
|
|
"""
|
|
out = smart_ctl('--health', *device.smartctl_select())
|
|
|
|
if self_test_re.search(out):
|
|
self_assessment_passed = True
|
|
else:
|
|
self_assessment_passed = False
|
|
|
|
yield Metric(
|
|
'device_smart_healthy', device.base_labels, self_assessment_passed)
|
|
|
|
|
|
def collect_ata_metrics(device):
|
|
# Fetch SMART attributes for the given device.
|
|
attributes = smart_ctl(
|
|
'--attributes', *device.smartctl_select()
|
|
)
|
|
|
|
# replace multiple occurrences of whitespace with a single whitespace
|
|
# so that the CSV Parser recognizes individual columns properly.
|
|
attributes = re.sub(r'[\t\x20]+', ' ', attributes)
|
|
|
|
# Turn smartctl output into a list of lines and skip to the table of
|
|
# SMART attributes.
|
|
attribute_lines = attributes.strip().split('\n')[7:]
|
|
|
|
reader = csv.DictReader(
|
|
(l.strip() for l in attribute_lines),
|
|
fieldnames=SmartAttribute._fields[:-1],
|
|
restkey=SmartAttribute._fields[-1], delimiter=' ')
|
|
for entry in reader:
|
|
# We're only interested in the SMART attributes that are
|
|
# whitelisted here.
|
|
entry['name'] = entry['name'].lower()
|
|
if entry['name'] not in smart_attributes_whitelist:
|
|
continue
|
|
|
|
# Ensure that only the numeric parts are fetched from the raw_value.
|
|
# Attributes such as 194 Temperature_Celsius reported by my SSD
|
|
# are in the format of "36 (Min/Max 24/40)" which can't be expressed
|
|
# properly as a prometheus metric.
|
|
m = re.match('^(\d+)', ' '.join(entry['raw_value']))
|
|
if not m:
|
|
continue
|
|
entry['raw_value'] = m.group(1)
|
|
|
|
if entry['name'] in smart_attributes_whitelist:
|
|
labels = {
|
|
'name': entry['name'],
|
|
**device.base_labels,
|
|
}
|
|
|
|
for col in 'value', 'worst', 'threshold':
|
|
yield Metric(
|
|
'attr_{col}'.format(name=entry["name"], col=col),
|
|
labels, entry[col])
|
|
|
|
|
|
def collect_ata_error_count(device):
|
|
"""Inspect the device error log and report the amount of entries.
|
|
|
|
Args:
|
|
device: (Device) Device in question.
|
|
|
|
Yields:
|
|
(Metric) Device error count.
|
|
"""
|
|
error_log = smart_ctl(
|
|
'-l', 'xerror,1', *device.smartctl_select(), check=False)
|
|
|
|
m = ata_error_count_re.search(error_log)
|
|
|
|
error_count = m.group(1) if m is not None else 0
|
|
|
|
yield Metric('device_errors', device.base_labels, error_count)
|
|
|
|
|
|
def collect_disks_smart_metrics():
|
|
now = int(datetime.datetime.utcnow().timestamp())
|
|
|
|
for device in find_devices():
|
|
yield Metric('smartctl_run', device.base_labels, now)
|
|
|
|
is_active = device_is_active(device)
|
|
|
|
yield Metric('device_active', device.base_labels, is_active)
|
|
|
|
# Skip further metrics collection to prevent the disk from
|
|
# spinning up.
|
|
if not is_active:
|
|
continue
|
|
|
|
yield from collect_device_info(device)
|
|
|
|
smart_available, smart_enabled = device_smart_capabilities(device)
|
|
|
|
yield Metric(
|
|
'device_smart_available', device.base_labels, smart_available)
|
|
yield Metric(
|
|
'device_smart_enabled', device.base_labels, smart_enabled)
|
|
|
|
# Skip further metrics collection here if SMART is disabled
|
|
# on the device. Further smartctl invocations would fail
|
|
# anyways.
|
|
if not smart_available:
|
|
continue
|
|
|
|
yield from collect_device_health_self_assessment(device)
|
|
|
|
if device.type.startswith('sat'):
|
|
yield from collect_ata_metrics(device)
|
|
|
|
yield from collect_ata_error_count(device)
|
|
|
|
|
|
def main():
|
|
version_metric = Metric('smartctl_version', {
|
|
'version': smart_ctl_version()
|
|
}, True)
|
|
metric_print_meta(version_metric, 'smartmon_')
|
|
metric_print(version_metric, 'smartmon_')
|
|
|
|
metrics = list(collect_disks_smart_metrics())
|
|
metrics.sort(key=lambda i: i.name)
|
|
|
|
previous_name = None
|
|
for m in metrics:
|
|
if m.name != previous_name:
|
|
metric_print_meta(m, 'smartmon_')
|
|
|
|
previous_name = m.name
|
|
|
|
metric_print(m, 'smartmon_')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|