20 Nov
2015
20 Nov
'15
9:40 a.m.
Module: check_mk
Branch: master
Commit: 25001acf6d4233835a07d92b432b441f4575937f
URL:
http://git.mathias-kettner.de/git/?p=check_mk.git;a=commit;h=25001acf6d4233…
Author: Sebastian Herbord <sh(a)mathias-kettner.de>
Date: Fri Nov 20 10:39:18 2015 +0100
added a small tool to help browse the wmi hierarchy
---
agents/windows/wmitest.cc | 136 ++++++++++++++++++++++++++++++++++++++++++++
agents/windows/wmitest.exe | Bin 0 -> 1448856 bytes
2 files changed, 136 insertions(+)
diff --git a/agents/windows/wmitest.cc b/agents/windows/wmitest.cc
new file mode 100644
index 0000000..0830760
--- /dev/null
+++ b/agents/windows/wmitest.cc
@@ -0,0 +1,136 @@
+#include "wmiHelper.h"
+#include "stringutil.h"
+#include <string>
+#include <algorithm>
+
+
+void print_usage(const char *exe_name)
+{
+ printf("Usage: %s action [action specific parameters]"
+ "\n\ttree - print the whole wmi namespace tree"
+ "\n\tcsv <namespace> <class> - print the whole class table
in csv format"
+ , exe_name);
+}
+
+
+void print_namespace(const std::wstring &path, int depth = 0)
+{
+ wmi::Helper helper(path.c_str());
+
+ std::string offset = std::string(depth * 2, ' ');
+ {
+ wmi::Result result = helper.query(L"SELECT name FROM __Namespace");
+ bool more = result.valid();
+ while (more) {
+ std::wstring name = result.get<std::wstring>(L"name");
+ printf("%s%ls\n", offset.c_str(), name.c_str());
+ try {
+ print_namespace(path + L"\\" + name, depth + 1);
+ } catch (const std::exception &e) {
+ printf("-- failed: %s\n", e.what());
+ }
+
+ more = result.next();
+ }
+ }
+ {
+ wmi::Result result = helper.query(L"SELECT * FROM meta_class");
+ bool more = result.valid();
+ while (more) {
+ std::wstring name = result.get<std::wstring>(L"__CLASS");
+ printf("%s> %ls\n", offset.c_str(), name.c_str());
+ more = result.next();
+ }
+ }
+}
+
+
+template <typename T>
+std::basic_string<T> join(const std::vector<std::basic_string<T>>
&input,
+ const T *sep) {
+ std::basic_ostringstream<T> stream;
+ bool first = true;
+
+
+ for (const std::basic_string<T> &val : input) {
+ if (!first) {
+ stream << sep;
+ } else {
+ first = false;
+ }
+ stream << val;
+ }
+ return stream.str();
+}
+
+
+void print_table(const std::string &ns, const std::string &pattern) {
+ wmi::Helper helper(to_utf16(ns.c_str()).c_str());
+
+ wmi::Result result = helper.query(L"SELECT * FROM meta_class");
+ bool more = result.valid();
+ if (!more) {
+ printf("Invalid result for meta_class\n");
+ }
+ while (more) {
+ std::wstring name = result.get<std::wstring>(L"__CLASS");
+ try {
+ if (globmatch(pattern.c_str(), to_utf8(name.c_str()).c_str())) {
+ printf("<<<%ls>>>\n", name.c_str());
+ wmi::Result sub_result = helper.query((std::wstring(L"SELECT * FROM
") + name).c_str());
+ bool first = true;
+ bool sub_more = sub_result.valid();
+ if (!sub_more) {
+ printf("Invalid result\n");
+ }
+ while (sub_more) {
+ if (first) {
+ // print header
+ printf("%ls\n", join(sub_result.names(),
L",").c_str());
+ first = false;
+ }
+ std::vector<std::wstring> values = sub_result.names();
+ // resolve all table keys to their value on this row.
+ std::transform(values.begin(), values.end(), values.begin(),
+ [&sub_result] (const std::wstring &name) {
+ return sub_result.get<std::wstring>(name.c_str());
+ }
+ );
+ printf("%ls\n", join(values, L",").c_str());
+
+ sub_more = sub_result.next();
+ }
+ }
+ } catch (const std::exception &e) {
+ printf("Exception: %s\n", e.what());
+ }
+ more = result.next();
+ }
+}
+
+
+int main(int argc, char **argv)
+{
+ if (argc < 2) {
+ print_usage(argv[0]);
+ return 1;
+ }
+
+ try {
+ if (strcmp(argv[1], "tree") == 0) {
+ print_namespace(L"Root");
+ }
+
+ if (strcmp(argv[1], "csv") == 0) {
+ if (argc < 4) {
+ print_usage(argv[0]);
+ return 1;
+ }
+ print_table(argv[2], argv[3]);
+ }
+
+ return 0;
+ } catch (const std::exception &e) {
+ printf("Failed: %s\n", e.what());
+ }
+}
diff --git a/agents/windows/wmitest.exe b/agents/windows/wmitest.exe
new file mode 100755
index 0000000..ce6fc2b
Binary files /dev/null and b/agents/windows/wmitest.exe differ