Changeset bb8b6d1 in nscp for include

Timestamp:

09/19/08 06:48:41 (5 years ago)

Author:

Michael Medin <michael@…>

Branches:

master, 0.4.0, 0.4.1, 0.4.2, stable

Children:

c09c225

Parents:

371df23

Message:

2008-09-18 MickeM - 0.3.4

• Changed so "missing services" are treated as stopped.

CheckServiceState missing=stopped ShowAll

OK: missing: not found

• Fixed issue with : in service name. + Added some "reasonable default cheks" under [External Alias] for the CheckExternalScripts.dll module.

Location:

include

Files:

• EnumNtSrv.cpp (modified) (1 diff)
• EnumNtSrv.h (modified) (1 diff)
• checkHelpers.hpp (modified) (3 diffs)
• config.h (modified) (1 diff)

include/EnumNtSrv.cpp

@@ -189 +189 @@
   SC_HANDLE sh = ::OpenService(scman,name.c_str(),SERVICE_QUERY_STATUS);
   if (!sh) {
+    std::wstring short_name;
     DWORD bufLen = SC_BUF_LEN;
     TCHAR *buf = new TCHAR[bufLen+1];
     if (GetServiceKeyName(scman, name.c_str(), buf, &bufLen) == 0) {
+      short_name = name;
+    } else {
+      short_name = buf;
+    }
+    delete [] buf;
+    sh = ::OpenService(scman,short_name.c_str(),SERVICE_QUERY_STATUS);
+    if (sh == NULL) {
+      DWORD dwErr = GetLastError();
       ::CloseServiceHandle(scman);
-      delete [] buf;
-      throw NTServiceException(name, _T("GetServiceKeyName: Could not translate service name: ") + error::lookup::last_error());
-    }
-    /*
-    Why does this not work? (a bug in the API? says it should return the correct size?)
-    if (bufLen >= SC_BUF_LEN) {
-      ::CloseServiceHandle(scman);
-      throw NTServiceException(name, "Service name to long to handle", GetLastError());
-    }
-    buf[bufLen] = 0;
-    */
-    sh = ::OpenService(scman,buf,SERVICE_QUERY_STATUS);
-    delete [] buf;
-    if (sh == NULL) {
-      ::CloseServiceHandle(scman);
-      throw NTServiceException(name, _T("OpenService: Could not open Service: ") + error::lookup::last_error());
+      if (dwErr == ERROR_SERVICE_DOES_NOT_EXIST) {
+        info.m_dwCurrentState = MY_SERVICE_NOT_FOUND;
+        info.m_dwServiceType = MY_SERVICE_NOT_FOUND;
+        return info;
+      } else {
+        throw NTServiceException(name, _T("OpenService: Could not open Service: ") + error::lookup::last_error(dwErr));
+      }
     }
   }

include/EnumNtSrv.h

@@ -47 +47 @@
   }
 };
+#define MY_SERVICE_NOT_FOUND                        0xffff0000
 
 class TNtServiceInfo {

include/checkHelpers.hpp

@@ -298 +298 @@
 
   typedef unsigned long state_type;
-  const int state_none  = 0x00;
-  const int state_started = 0x01;
-  const int state_stopped = 0x02;
+  const int state_none    = 0x00;
+  const int state_started   = 0x01;
+  const int state_stopped   = 0x02;
+  const int state_not_found = 0x06;
 
   class state_handler {
@@ -314 +315 @@
         else if (*it == _T("ignored"))
           ret |= state_none;
+        else if (*it == _T("not found"))
+          ret |= state_not_found;
       }
       return ret;
@@ -324 +327 @@
       else if (value == state_none)
         return _T("none");
+      else if (value == state_not_found)
+        return _T("not found");
       return _T("unknown");
     }

include/config.h

@@ -26 +26 @@
 
 // Version
-//#define SZBETATAG _T(" ")
-#define SZBETATAG _T(" RC ")  
+#define SZBETATAG _T(" ")
+//#define SZBETATAG _T(" RC ")  
 //#define SZBETATAG _T(" BETA ") 
 #define SZVERSION STRPRODUCTVER SZBETATAG STRPRODUCTDATE