Debian 3.9.10debian/3.9.10debian

1 files changed, 154 insertions, 0 deletions

diff --git a/src/pmdas/solaris/zpool.c b/src/pmdas/solaris/zpool.c
new file mode 100644
index 0000000..2f402f2
--- /dev/null
+++ b/src/pmdas/solaris/zpool.c
@@ -0,0 +1,154 @@
+/*
+ * Copyright (C) 2009 Max Matveev.  All rights reserved.
+ *
+ * This program 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; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * for more details.
+ */
+
+#include <libzfs.h>
+
+#include "common.h"
+
+struct zpool_stats {
+    int vdev_stats_fresh;
+    vdev_stat_t vds;
+};
+
+static libzfs_handle_t *zh;
+static int zp_added;
+
+/*
+ * For each zpool check the name in the instance cache, if it's not there then
+ * add it to the cache. Regardless if it's the first time we've seen this one
+ * or if it was in the cache before refresh the stats
+ */
+static int
+zp_cache_pool(zpool_handle_t *zp, void *arg)
+{
+	nvlist_t *cfg = zpool_get_config(zp, NULL);
+	char *zpname = (char *)zpool_get_name(zp);
+	struct zpool_stats *zps = NULL;
+	pmInDom zpindom = indomtab[ZPOOL_INDOM].it_indom;
+        uint_t cnt = 0;
+        vdev_stat_t *vds;
+	int rv;
+	int inst;
+        nvlist_t *vdt;
+
+	if ((rv = pmdaCacheLookupName(zpindom, zpname, &inst,
+					(void **)&zps)) != PMDA_CACHE_ACTIVE) {
+	    int newpool = (zps == NULL);
+
+	    if (rv != PMDA_CACHE_INACTIVE || zps == NULL) {
+		zps = malloc(sizeof(*zps));
+		if (zps == NULL) {
+		    __pmNotifyErr(LOG_WARNING,
+				  "Cannot allocate memory to hold stats for "
+				  "zpool '%s'\n",
+				  zpname);
+		    goto done;
+		}
+	    }
+
+	    rv = pmdaCacheStore(zpindom, PMDA_CACHE_ADD, zpname, zps);
+	    if (rv < 0) {
+		__pmNotifyErr(LOG_WARNING,
+			      "Cannot add '%s' to the cache "
+			      "for instance domain %s: %s\n",
+			      zpname, pmInDomStr(zpindom), pmErrStr(rv));
+		free(zps);
+		goto done;
+	    }
+	    zp_added += newpool;
+	}
+
+	rv = nvlist_lookup_nvlist(cfg, ZPOOL_CONFIG_VDEV_TREE, &vdt);
+	if (rv != 0) {
+	    __pmNotifyErr(LOG_ERR, "Cannot get vdev tree for '%s': %d %d\n",
+			  zpname, rv, oserror());
+	    zps->vdev_stats_fresh = 0;
+	} else {
+	    /* accommodate zpool api changes ... */
+#ifdef ZPOOL_CONFIG_VDEV_STATS
+	    rv = nvlist_lookup_uint64_array(vdt, ZPOOL_CONFIG_VDEV_STATS,
+					    (uint64_t **)&vds, &cnt);
+#else
+	    rv = nvlist_lookup_uint64_array(vdt, ZPOOL_CONFIG_STATS,
+					    (uint64_t **)&vds, &cnt);
+#endif
+	    if (rv == 0) {
+		memcpy(&zps->vds, vds, sizeof(zps->vds));
+		zps->vdev_stats_fresh = 1;
+	    } else {
+		__pmNotifyErr(LOG_ERR,
+			      "Cannot get zpool stats for '%s': %d %d\n",
+			      zpname, rv, oserror());
+		zps->vdev_stats_fresh = 0;
+	    }
+	}
+
+done:
+	zpool_close(zp);
+	return 0;
+}
+
+void
+zpool_refresh(void)
+{
+    zp_added = 0;
+
+    pmdaCacheOp(indomtab[ZPOOL_INDOM].it_indom, PMDA_CACHE_INACTIVE);
+    zpool_iter(zh, zp_cache_pool, NULL);
+
+    if (zp_added) {
+	pmdaCacheOp(indomtab[ZPOOL_INDOM].it_indom, PMDA_CACHE_SAVE);
+    }
+}
+
+int
+zpool_fetch(pmdaMetric *pm, int inst, pmAtomValue *atom)
+{
+    struct zpool_stats *zps;
+    char *zpname;
+    metricdesc_t *md = pm->m_user;
+
+    if (pmdaCacheLookup(indomtab[ZPOOL_INDOM].it_indom, inst, &zpname,
+			(void **)&zps) != PMDA_CACHE_ACTIVE)
+	return PM_ERR_INST;
+
+    if (zps->vdev_stats_fresh) {
+	switch (pmid_item(md->md_desc.pmid)) {
+	case 0: /* zpool.state */
+	    atom->cp = zpool_state_to_name(zps->vds.vs_state, zps->vds.vs_aux);
+	    break;
+	case 1: /* zpool.state_int */
+	    atom->ul = (zps->vds.vs_aux << 8) | zps->vds.vs_state;
+	    break;
+	default:
+	    memcpy(&atom->ull, ((char *)&zps->vds) + md->md_offset,
+			sizeof(atom->ull));
+	}
+    }
+    return zps->vdev_stats_fresh;
+}
+
+void
+zpool_init(int first)
+{
+    if (zh)
+	return;
+
+    zh = libzfs_init();
+    if (zh) {
+	pmdaCacheOp(indomtab[ZPOOL_INDOM].it_indom, PMDA_CACHE_LOAD);
+	zpool_iter(zh, zp_cache_pool, NULL);
+	pmdaCacheOp(indomtab[ZPOOL_INDOM].it_indom, PMDA_CACHE_SAVE);
+    }
+}