[SRVSVC][WKSSVC] Server and workstation service set their service bit when they are...

diff --git a/base/services/srvsvc/precomp.h b/base/services/srvsvc/precomp.h
index c933646..a0fbcbc 100644 (file)
--- a/base/services/srvsvc/precomp.h
+++ b/base/services/srvsvc/precomp.h
@@ -9,6 +9,7 @@
 #include <winbase.h>
 #include <winreg.h>
 #include <winsvc.h>
+#include <lmserver.h>
 
 #include <srvsvc_s.h>
 

diff --git a/base/services/srvsvc/srvsvc.c b/base/services/srvsvc/srvsvc.c
index c37072f..5182987 100644 (file)
--- a/base/services/srvsvc/srvsvc.c
+++ b/base/services/srvsvc/srvsvc.c
@@ -127,6 +127,12 @@ ServiceInit(VOID)
     else
         CloseHandle(hThread);
 
+    /* Report a running server service */
+    SetServiceBits(ServiceStatusHandle,
+                   SV_TYPE_SERVER,
+                   TRUE,
+                   TRUE);
+
     return ERROR_SUCCESS;
 }
 

diff --git a/base/services/wkssvc/precomp.h b/base/services/wkssvc/precomp.h
index 1724117..6e66ec7 100644 (file)
--- a/base/services/wkssvc/precomp.h
+++ b/base/services/wkssvc/precomp.h
@@ -9,6 +9,7 @@
 #include <winbase.h>
 #include <winreg.h>
 #include <winsvc.h>
+#include <lmserver.h>
 
 #include <wkssvc_s.h>
 

diff --git a/base/services/wkssvc/wkssvc.c b/base/services/wkssvc/wkssvc.c
index 198ce5a..5200570 100644 (file)
--- a/base/services/wkssvc/wkssvc.c
+++ b/base/services/wkssvc/wkssvc.c
@@ -127,6 +127,12 @@ ServiceInit(VOID)
     else
         CloseHandle(hThread);
 
+    /* Report a running workstation service */
+    SetServiceBits(ServiceStatusHandle,
+                   SV_TYPE_WORKSTATION,
+                   TRUE,
+                   TRUE);
+
     return ERROR_SUCCESS;
 }