[ros-diffs] 01/01: [SRVSVC][WKSSVC] Server and workstation service set their service bit when they are running

[Eric Kohl]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=027904b441ef22bb6cea3…

commit 027904b441ef22bb6cea3dd7dae835c5bf5a9103
Author:     Eric Kohl <eric.kohl(a)reactos.org>
AuthorDate: Tue May 1 16:37:48 2018 +0200
Commit:     Eric Kohl <eric.kohl(a)reactos.org>
CommitDate: Tue May 1 16:37:48 2018 +0200

    [SRVSVC][WKSSVC] Server and workstation service set their service bit when they are
running
---
 base/services/srvsvc/precomp.h | 1 +
 base/services/srvsvc/srvsvc.c  | 6 ++++++
 base/services/wkssvc/precomp.h | 1 +
 base/services/wkssvc/wkssvc.c  | 6 ++++++
 4 files changed, 14 insertions(+)

diff --git a/base/services/srvsvc/precomp.h b/base/services/srvsvc/precomp.h
index c933646c03..a0fbcbc4bd 100644
--- 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 c37072f13e..5182987d50 100644
--- 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 172411794d..6e66ec7679 100644
--- 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 198ce5a470..5200570256 100644
--- 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;
 }