Changeset 1bfe6f0 in nscp for include

Timestamp:

01/17/10 12:29:46 (3 years ago)

Author:

Michael Medin <michael@…>

Branches:

master, 0.4.0, 0.4.1, 0.4.2

Children:

6822839

Parents:

cc2efd6

Message:

Finnished Scheduler and added basic "Notify" support (no sinks yet)

Location:

include

Files:

• NSCHelper.cpp (modified) (4 diffs)
• NSCHelper.h (modified) (3 diffs)
• strEx.h (modified) (1 diff)

include/NSCHelper.cpp

@@ -83 +83 @@
 
 
+#define REPORT_ERROR  0x01
+#define REPORT_WARNING  0x02
+#define REPORT_UNKNOWN  0x04
+#define REPORT_OK   0x08
+
+unsigned int NSCHelper::report::parse(std::wstring str) {
+  unsigned int report = 0;
+  strEx::splitList lst = strEx::splitEx(str, _T(","));
+  for (strEx::splitList::const_iterator key = lst.begin(); key != lst.end(); ++key) {
+    if (*key == _T("all")) {
+      report |= REPORT_ERROR|REPORT_OK|REPORT_UNKNOWN|REPORT_WARNING;
+    } else if (*key == _T("error") || *key == _T("err") || *key == _T("critical") || *key == _T("crit")) {
+      report |= REPORT_ERROR;
+    } else if (*key == _T("warning") || *key == _T("warn")) {
+      report |= REPORT_WARNING;
+    } else if (*key == _T("unknown")) {
+      report |= REPORT_UNKNOWN;
+    } else if (*key == _T("ok")) {
+      report |= REPORT_OK;
+    }
+  }
+  return report;
+}
+bool NSCHelper::report::matches(unsigned int report, NSCAPI::nagiosReturn code) {
+  return (
+    (code == NSCAPI::returnOK && (report&REPORT_OK)==REPORT_OK) ||
+    (code == NSCAPI::returnCRIT && ((report&REPORT_ERROR)==REPORT_ERROR) ) ||
+    (code == NSCAPI::returnWARN && ((report&REPORT_WARNING)==REPORT_WARNING) ) ||
+    (code == NSCAPI::returnUNKNOWN && ((report&REPORT_UNKNOWN)==REPORT_UNKNOWN) ) ||
+    ( (code != NSCAPI::returnOK) && (code != NSCAPI::returnCRIT) && (code != NSCAPI::returnWARN) && (code != NSCAPI::returnUNKNOWN) )
+    );
+}
+std::wstring NSCHelper::report::to_string(unsigned int report) {
+  std::wstring ret;
+  if ((report&REPORT_OK)!=0) {
+    if (!ret.empty()) ret += _T(",");
+    ret += _T("ok");
+  }
+  if ((report&REPORT_WARNING)!=0) {
+    if (!ret.empty()) ret += _T(",");
+    ret += _T("warning");
+  }
+  if ((report&REPORT_ERROR)!=0) {
+    if (!ret.empty()) ret += _T(",");
+    ret += _T("critical");
+  }
+  if ((report&REPORT_UNKNOWN)!=0) {
+    if (!ret.empty()) ret += _T(",");
+    ret += _T("unknown");
+  }
+  return ret;
+}
+
+
 /**
  * Translate a message type into a human readable string.
@@ -151 +205 @@
   lpNSAPIExit fNSAPIExit = NULL;
   lpNSAPIInject fNSAPIInject = NULL;
+  lpNSAPINotify fNSAPINotify = NULL;
   lpNSAPICheckLogMessages fNSAPICheckLogMessages = NULL;
   lpNSAPIEncrypt fNSAPIEncrypt = NULL;
@@ -225 +280 @@
   return fNSAPIInject(command, argLen, argument, returnMessageBuffer, returnMessageBufferLen, returnPerfBuffer, returnPerfBufferLen);
 }
+
+NSCAPI::errorReturn NSCModuleHelper::NotifyChannel(std::wstring channel, std::wstring command, NSCAPI::nagiosReturn code, std::wstring message, std::wstring perf) {
+  if (!fNSAPINotify)
+    throw NSCMHExcpetion(_T("NSCore has not been initiated..."));
+  return fNSAPINotify(channel.c_str(), command.c_str(), code, message.c_str(), perf.c_str());
+}
+
 /**
  * Inject a request command in the core (this will then be sent to the plug-in stack for processing)
@@ -734 +796 @@
   //NSCModuleHelper::fNSAPIExit = (NSCModuleHelper::lpNSAPIExit)f(_T("NSAPIExit"));
   NSCModuleHelper::fNSAPIInject = (NSCModuleHelper::lpNSAPIInject)f(_T("NSAPIInject"));
+  NSCModuleHelper::fNSAPINotify = (NSCModuleHelper::lpNSAPINotify)f(_T("NSAPINotify"));
   NSCModuleHelper::fNSAPIGetBasePath = (NSCModuleHelper::lpNSAPIGetBasePath)f(_T("NSAPIGetBasePath"));
   NSCModuleHelper::fNSAPICheckLogMessages = (NSCModuleHelper::lpNSAPICheckLogMessages)f(_T("NSAPICheckLogMessages"));

include/NSCHelper.h

@@ -94 +94 @@
     if (currentReturnCode != NSCAPI::returnCRIT)
       currentReturnCode = NSCAPI::returnWARN;
+  }
+
+  namespace report {
+    unsigned int parse(std::wstring str);
+    bool matches(unsigned int report, NSCAPI::nagiosReturn code);
+    std::wstring to_string(unsigned int report);
   }
 };
@@ -124 +130 @@
   typedef NSCAPI::errorReturn (*lpNSAPIExit)(void);
   typedef NSCAPI::nagiosReturn (*lpNSAPIInject)(const wchar_t*, const unsigned int, wchar_t **, wchar_t *, unsigned int, wchar_t *, unsigned int);
-  
+
+  typedef NSCAPI::errorReturn (*lpNSAPINotify)(const wchar_t*, const wchar_t*, NSCAPI::nagiosReturn, const wchar_t*, const wchar_t*);
+
   typedef NSCAPI::boolReturn (*lpNSAPICheckLogMessages)(int);
   typedef NSCAPI::errorReturn (*lpNSAPIEncrypt)(unsigned int, const wchar_t*, unsigned int, wchar_t*, unsigned int *);
@@ -158 +166 @@
   NSCAPI::nagiosReturn InjectCommand(const wchar_t* command, const unsigned int argLen, wchar_t **argument, std::wstring & message, std::wstring & perf);
   NSCAPI::nagiosReturn InjectCommand(const wchar_t* command, std::list<std::wstring> argument, std::wstring & message, std::wstring & perf);
+  NSCAPI::errorReturn NotifyChannel(std::wstring channel, std::wstring command, NSCAPI::nagiosReturn code, std::wstring message, std::wstring perf);
   NSCAPI::nagiosReturn InjectSplitAndCommand(const wchar_t* command, wchar_t* buffer, wchar_t splitChar, std::wstring & message, std::wstring & perf);
   NSCAPI::nagiosReturn InjectSplitAndCommand(const std::wstring command, const std::wstring buffer, wchar_t splitChar, std::wstring & message, std::wstring & perf, bool escape = false);

include/strEx.h

@@ -382 +382 @@
     return value * smallest_unit;
   }
+  inline unsigned stoui_as_time_sec(std::wstring time, unsigned int smallest_unit = 1) {
+    std::wstring::size_type p = time.find_first_of(_T("sSmMhHdDwW"));
+    std::wstring::size_type pend = time.find_first_not_of(_T("0123456789"));
+    unsigned int value = boost::lexical_cast<unsigned int>(pend==std::wstring::npos?time:time.substr(0,pend).c_str());
+    if (p == std::wstring::npos)
+      return value * smallest_unit;
+    else if ( (time[p] == 's') || (time[p] == 'S') )
+      return value;
+    else if ( (time[p] == 'm') || (time[p] == 'M') )
+      return value * 60;
+    else if ( (time[p] == 'h') || (time[p] == 'H') )
+      return value * 60 * 60;
+    else if ( (time[p] == 'd') || (time[p] == 'D') )
+      return value * 24 * 60 * 60;
+    else if ( (time[p] == 'w') || (time[p] == 'W') )
+      return value * 7 * 24 * 60 * 60;
+    return value * smallest_unit;
+  }
 
   inline unsigned long long stoi64_as_time(std::wstring time, unsigned int smallest_unit = 1000) {