mirror of
https://github.com/prometheus-community/smartctl_exporter.git
synced 2024-12-21 02:21:55 +01:00
66 lines
2 KiB
Python
66 lines
2 KiB
Python
|
#! /usr/bin/env python3
|
||
|
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# Redact potentially sensitive information in smartctl JSON files
|
||
|
|
# This script does an in-place modification.
|
||
|
|
import json
|
||
|
|
import sys
|
||
|
|
import copy
|
||
|
|
import os
|
||
|
|
|
||
|
|
def main():
|
||
|
|
for arg in sys.argv[1:]:
|
||
|
|
print(arg)
|
||
|
|
redact_one_file(arg)
|
||
|
|
|
||
|
|
def redact_one_file(filename):
|
||
|
|
data = None
|
||
|
|
tmpname = filename+".new"
|
||
|
|
with open(filename, "r") as jsonFile:
|
||
|
|
data = json.load(jsonFile)
|
||
|
|
|
||
|
|
newdata = redact_data(data)
|
||
|
|
|
||
|
|
with open(tmpname, "w") as jsonFile:
|
||
|
|
json.dump(newdata, jsonFile, indent="\t", sort_keys=True)
|
||
|
|
|
||
|
|
os.rename(tmpname, filename)
|
||
|
|
|
||
|
|
def mutate_nested_dict(d, keys, newvalue, if_present=False):
|
||
|
|
# if_present=True: only mutate if the full key path exists.
|
||
|
|
if len(keys) == 1:
|
||
|
|
if not if_present or keys[0] in d:
|
||
|
|
d[keys[0]] = newvalue
|
||
|
|
else:
|
||
|
|
k = keys[0]
|
||
|
|
if k in d:
|
||
|
|
mutate_nested_dict(d[k], keys[1:], newvalue, if_present=if_present)
|
||
|
|
|
||
|
|
REDACTED_STRING = 'REDACTED'
|
||
|
|
REDACTED_TIME_T = 1234567890
|
||
|
|
REDACTED_ASCTIME = "Fri Feb 13 23:31:30 2009 UTC" # TODO: generate from TIME_T, with UTC
|
||
|
|
REDACTED_HEX16_STR = '0x1234567890abcdef'
|
||
|
|
REDACTED_UINT32 = 1234567890
|
||
|
|
|
||
|
|
REDACT_FIELDS = [
|
||
|
|
{'k': ['smartctl','platform_info'], 'v': REDACTED_STRING},
|
||
|
|
{'k': ['smartctl','build_info'], 'v': REDACTED_STRING},
|
||
|
|
{'k': ['serial_number'], 'v': REDACTED_STRING},
|
||
|
|
{'k': ['firmware_version'], 'v': REDACTED_STRING},
|
||
|
|
{'k': ['local_time', 'time_t'], 'v': REDACTED_TIME_T},
|
||
|
|
{'k': ['local_time', 'asctime'], 'v': REDACTED_ASCTIME},
|
||
|
|
{'k': ['logical_unit_id'], 'v': REDACTED_HEX16_STR},
|
||
|
|
{'k': ['wwn','id'], 'v': REDACTED_UINT32},
|
||
|
|
# TODO: how to redact /dev/sdX /dev/nvmeN ??
|
||
|
|
]
|
||
|
|
|
||
|
|
def redact_data(data):
|
||
|
|
newdata = copy.deepcopy(data)
|
||
|
|
for f in REDACT_FIELDS:
|
||
|
|
#newval = str(f['v'])+str(f['k']) # for debugging
|
||
|
|
newval = f['v']
|
||
|
|
mutate_nested_dict(newdata, f['k'], newval, if_present=True)
|
||
|
|
return newdata
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|