From ecbaaa39a1bf8d9b1321b502e54ee16b6f08bbb2 Mon Sep 17 00:00:00 2001
From: "Joshua M. Boniface" <joshua@boniface.me>
Date: Sun, 24 Nov 2024 18:45:01 -0500
Subject: [PATCH] Update Ceph agent to the latest version

---
 .../files/usr/lib/check_mk_agent/plugins/ceph | 141 +++++++++++++++---
 1 file changed, 121 insertions(+), 20 deletions(-)

diff --git a/roles/base/files/usr/lib/check_mk_agent/plugins/ceph b/roles/base/files/usr/lib/check_mk_agent/plugins/ceph
index 7301111..c89d396 100755
--- a/roles/base/files/usr/lib/check_mk_agent/plugins/ceph
+++ b/roles/base/files/usr/lib/check_mk_agent/plugins/ceph
@@ -1,24 +1,125 @@
-#!/bin/bash
+#!/usr/bin/env python3
+# -*- encoding: utf-8; py-indent-offset: 4 -*-
 
-# Ceph check for Check_MK
-# Installed by PVC ansible
+# (c) 2021 Heinlein Support GmbH
+#          Robert Sander <r.sander@heinlein-support.de>
 
-CMK_VERSION="2.1.0"
+# This is free software;  you can redistribute it and/or modify it
+# under the  terms of the  GNU General Public License  as published by
+# the Free Software Foundation in version 2.  This file is distributed
+# in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
+# out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
+# PARTICULAR PURPOSE. See the  GNU General Public License for more de-
+# ails.  You should have  received  a copy of the  GNU  General Public
+# License along with GNU Make; see the file  COPYING.  If  not,  write
+# to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
+# Boston, MA 02110-1301 USA.
 
-USER=client.admin
-KEYRING=/etc/ceph/ceph.client.admin.keyring
+import json
+import rados
+import os, os.path
+import subprocess
+import socket
 
-if [ -n "$USER" ] && [ -n "$KEYRING" ]; then
-    CEPH_CMD="ceph -n $USER --keyring=$KEYRING"
-    echo "<<<ceph_status>>>"
-    $CEPH_CMD -s -f json-pretty
-    if OUT="$($CEPH_CMD df detail --format json)"; then
-        echo "<<<ceph_df_json:sep(0)>>>"
-        $CEPH_CMD version --format json
-        echo "$OUT"
-    else
-        # fallback for old versions if json output is not available
-        echo "<<<ceph_df>>>"
-        $CEPH_CMD df detail
-    fi
-fi
+class RadosCMD(rados.Rados):
+    def command_mon(self, cmd, params=None):
+        data = {'prefix': cmd, 'format': 'json'}
+        if params:
+            data.update(params)
+        return self.mon_command(json.dumps(data), b'', timeout=5)
+    def command_mgr(self, cmd):
+        return self.mgr_command(json.dumps({'prefix': cmd, 'format': 'json'}), b'', timeout=5)
+    def command_osd(self, osdid, cmd):
+        return self.osd_command(osdid, json.dumps({'prefix': cmd, 'format': 'json'}), b'', timeout=5)
+    def command_pg(self, pgid, cmd):
+        return self.pg_command(pgid, json.dumps({'prefix': cmd, 'format': 'json'}), b'', timeout=5)
+
+ceph_config='/etc/ceph/ceph.conf'
+ceph_client='client.admin'
+try:
+    with open(os.path.join(os.environ['MK_CONFDIR'], 'ceph.cfg'), 'r') as config:
+        for line in config.readlines():
+            if '=' in line:
+                key, value = line.strip().split('=')
+                if key == 'CONFIG':
+                    ceph_config = value
+                if key == 'CLIENT':
+                    ceph_client = value
+except FileNotFoundError:
+    pass
+
+cluster = RadosCMD(conffile=ceph_config, name=ceph_client)
+cluster.connect()
+
+hostname = socket.gethostname().split('.', 1)[0]
+fqdn = socket.getfqdn()
+
+res = cluster.command_mon("status")
+if res[0] == 0:
+    status = json.loads(res[1])
+    mons = status.get('quorum_names', [])
+    fsid = status.get("fsid", "")
+
+    if hostname in mons or fqdn in mons:
+    # only on MON hosts
+
+        print("<<<cephstatus:sep(0)>>>")
+        print(json.dumps(status))
+
+        res = cluster.command_mon("df", params={'detail': 'detail'})
+        if res[0] == 0:
+            print("<<<cephdf:sep(0)>>>")
+            print(json.dumps(json.loads(res[1])))
+
+localosds = []
+res = cluster.command_mon("osd metadata")
+if res[0] == 0:
+    print("<<<cephosdbluefs:sep(0)>>>")
+    out = {'end': {}}
+    for osd in json.loads(res[1]):
+        if osd.get('hostname') in [hostname, fqdn]:
+            localosds.append(osd['id'])
+            if "container_hostname" in osd:
+                adminsocket = "/run/ceph/%s/ceph-osd.%d.asok" % (fsid, osd['id'])
+            else:
+                adminsocket = "/run/ceph/ceph-osd.%d.asok" % osd['id']
+            if os.path.exists(adminsocket):
+                chunks = []
+                try:
+                    sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+                    sock.connect(adminsocket)
+                    sock.sendall(b'{"prefix": "perf dump"}\n')
+                    sock.shutdown(socket.SHUT_WR)
+                    while len(chunks) == 0 or chunks[-1] != b'':
+                        chunks.append(sock.recv(4096))
+                    sock.close()
+                    chunks[0] = chunks[0][4:]
+                except:
+                    chunks = [b'{"bluefs": {}}']
+                out[osd['id']] = {'bluefs': json.loads(b"".join(chunks))['bluefs']}
+    print(json.dumps(out))
+
+osddf_raw = cluster.command_mon("osd df")
+osdperf_raw = cluster.command_mon("osd perf")
+if osddf_raw[0] == 0 and osdperf_raw[0] == 0:
+    osddf = json.loads(osddf_raw[1])
+    osdperf = json.loads(osdperf_raw[1])
+    osds = []
+    for osd in osddf['nodes']:
+        if osd['id'] in localosds:
+            osds.append(osd)
+    summary = osddf['summary']
+    perfs = []
+    if 'osd_perf_infos' in osdperf:
+        for osd in osdperf['osd_perf_infos']:
+            if osd['id'] in localosds:
+                perfs.append(osd)
+    if 'osdstats' in osdperf and 'osd_perf_infos' in osdperf['osdstats']:
+        for osd in osdperf['osdstats']['osd_perf_infos']:
+            if osd['id'] in localosds:
+                perfs.append(osd)
+
+    print("<<<cephosd:sep(0)>>>")
+    out = {'df': {'nodes': osds},
+           'perf': {'osd_perf_infos': perfs}}
+    print(json.dumps(out))