aa64233350db45ac884fad8524ea5e4989a193fc
[reactos.git] / base / system / services / rpcserver.c
1 /*
2 * PROJECT: ReactOS Service Control Manager
3 * LICENSE: GPL - See COPYING in the top level directory
4 * FILE: base/system/services/rpcserver.c
5 * PURPOSE: RPC server interface for the advapi32 calls
6 * COPYRIGHT: Copyright 2005-2006 Eric Kohl
7 * Copyright 2006-2007 Hervé Poussineau <hpoussin@reactos.org>
8 * Copyright 2007 Ged Murphy <gedmurphy@reactos.org>
9 */
10
11 /* INCLUDES ****************************************************************/
12
13 #include "services.h"
14
15 #include <winnls.h>
16 #include <strsafe.h>
17
18 #define NDEBUG
19 #include <debug.h>
20
21 /* GLOBALS *****************************************************************/
22
23 #define MANAGER_TAG 0x72674D68 /* 'hMgr' */
24 #define SERVICE_TAG 0x63765368 /* 'hSvc' */
25 #define INVALID_TAG 0xAABBCCDD
26
27 typedef struct _SCMGR_HANDLE
28 {
29 DWORD Tag;
30 DWORD DesiredAccess;
31 } SCMGR_HANDLE;
32
33
34 typedef struct _MANAGER_HANDLE
35 {
36 SCMGR_HANDLE Handle;
37 WCHAR DatabaseName[1];
38 } MANAGER_HANDLE, *PMANAGER_HANDLE;
39
40
41 typedef struct _SERVICE_HANDLE
42 {
43 SCMGR_HANDLE Handle;
44 PSERVICE ServiceEntry;
45 } SERVICE_HANDLE, *PSERVICE_HANDLE;
46
47
48 #define SC_MANAGER_READ \
49 (STANDARD_RIGHTS_READ | \
50 SC_MANAGER_QUERY_LOCK_STATUS | \
51 SC_MANAGER_ENUMERATE_SERVICE)
52
53 #define SC_MANAGER_WRITE \
54 (STANDARD_RIGHTS_WRITE | \
55 SC_MANAGER_MODIFY_BOOT_CONFIG | \
56 SC_MANAGER_CREATE_SERVICE)
57
58 #define SC_MANAGER_EXECUTE \
59 (STANDARD_RIGHTS_EXECUTE | \
60 SC_MANAGER_LOCK | \
61 SC_MANAGER_ENUMERATE_SERVICE | \
62 SC_MANAGER_CONNECT | \
63 SC_MANAGER_CREATE_SERVICE)
64
65
66 #define SERVICE_READ \
67 (STANDARD_RIGHTS_READ | \
68 SERVICE_INTERROGATE | \
69 SERVICE_ENUMERATE_DEPENDENTS | \
70 SERVICE_QUERY_STATUS | \
71 SERVICE_QUERY_CONFIG)
72
73 #define SERVICE_WRITE \
74 (STANDARD_RIGHTS_WRITE | \
75 SERVICE_CHANGE_CONFIG)
76
77 #define SERVICE_EXECUTE \
78 (STANDARD_RIGHTS_EXECUTE | \
79 SERVICE_USER_DEFINED_CONTROL | \
80 SERVICE_PAUSE_CONTINUE | \
81 SERVICE_STOP | \
82 SERVICE_START)
83
84 #define TAG_ARRAY_SIZE 32
85
86 /* VARIABLES ***************************************************************/
87
88 static GENERIC_MAPPING
89 ScmManagerMapping = {SC_MANAGER_READ,
90 SC_MANAGER_WRITE,
91 SC_MANAGER_EXECUTE,
92 SC_MANAGER_ALL_ACCESS};
93
94 static GENERIC_MAPPING
95 ScmServiceMapping = {SERVICE_READ,
96 SERVICE_WRITE,
97 SERVICE_EXECUTE,
98 SERVICE_ALL_ACCESS};
99
100 DWORD g_dwServiceBits = 0;
101
102 /* FUNCTIONS ***************************************************************/
103
104 VOID
105 ScmStartRpcServer(VOID)
106 {
107 RPC_STATUS Status;
108
109 DPRINT("ScmStartRpcServer() called\n");
110
111 Status = RpcServerUseProtseqEpW(L"ncacn_np",
112 10,
113 L"\\pipe\\ntsvcs",
114 NULL);
115 if (Status != RPC_S_OK)
116 {
117 DPRINT1("RpcServerUseProtseqEpW() failed (Status %lx)\n", Status);
118 return;
119 }
120
121 Status = RpcServerRegisterIf(svcctl_v2_0_s_ifspec,
122 NULL,
123 NULL);
124 if (Status != RPC_S_OK)
125 {
126 DPRINT1("RpcServerRegisterIf() failed (Status %lx)\n", Status);
127 return;
128 }
129
130 Status = RpcServerListen(1, 20, TRUE);
131 if (Status != RPC_S_OK)
132 {
133 DPRINT1("RpcServerListen() failed (Status %lx)\n", Status);
134 return;
135 }
136
137 DPRINT("ScmStartRpcServer() done\n");
138 }
139
140
141 static DWORD
142 ScmCreateManagerHandle(LPWSTR lpDatabaseName,
143 SC_HANDLE *Handle)
144 {
145 PMANAGER_HANDLE Ptr;
146
147 if (lpDatabaseName == NULL)
148 lpDatabaseName = SERVICES_ACTIVE_DATABASEW;
149
150 if (_wcsicmp(lpDatabaseName, SERVICES_FAILED_DATABASEW) == 0)
151 {
152 DPRINT("Database %S, does not exist\n", lpDatabaseName);
153 return ERROR_DATABASE_DOES_NOT_EXIST;
154 }
155 else if (_wcsicmp(lpDatabaseName, SERVICES_ACTIVE_DATABASEW) != 0)
156 {
157 DPRINT("Invalid Database name %S.\n", lpDatabaseName);
158 return ERROR_INVALID_NAME;
159 }
160
161 Ptr = HeapAlloc(GetProcessHeap(),
162 HEAP_ZERO_MEMORY,
163 FIELD_OFFSET(MANAGER_HANDLE, DatabaseName[wcslen(lpDatabaseName) + 1]));
164 if (Ptr == NULL)
165 return ERROR_NOT_ENOUGH_MEMORY;
166
167 Ptr->Handle.Tag = MANAGER_TAG;
168
169 wcscpy(Ptr->DatabaseName, lpDatabaseName);
170
171 *Handle = (SC_HANDLE)Ptr;
172
173 return ERROR_SUCCESS;
174 }
175
176
177 static DWORD
178 ScmCreateServiceHandle(PSERVICE lpServiceEntry,
179 SC_HANDLE *Handle)
180 {
181 PSERVICE_HANDLE Ptr;
182
183 Ptr = HeapAlloc(GetProcessHeap(),
184 HEAP_ZERO_MEMORY,
185 sizeof(SERVICE_HANDLE));
186 if (Ptr == NULL)
187 return ERROR_NOT_ENOUGH_MEMORY;
188
189 Ptr->Handle.Tag = SERVICE_TAG;
190
191 Ptr->ServiceEntry = lpServiceEntry;
192
193 *Handle = (SC_HANDLE)Ptr;
194
195 return ERROR_SUCCESS;
196 }
197
198
199 static PMANAGER_HANDLE
200 ScmGetServiceManagerFromHandle(SC_RPC_HANDLE Handle)
201 {
202 PMANAGER_HANDLE pManager = NULL;
203
204 _SEH2_TRY
205 {
206 if (((PMANAGER_HANDLE)Handle)->Handle.Tag == MANAGER_TAG)
207 pManager = (PMANAGER_HANDLE)Handle;
208 }
209 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
210 {
211 DPRINT1("Exception: Invalid Service Manager handle!\n");
212 }
213 _SEH2_END;
214
215 return pManager;
216 }
217
218
219 static PSERVICE_HANDLE
220 ScmGetServiceFromHandle(SC_RPC_HANDLE Handle)
221 {
222 PSERVICE_HANDLE pService = NULL;
223
224 _SEH2_TRY
225 {
226 if (((PSERVICE_HANDLE)Handle)->Handle.Tag == SERVICE_TAG)
227 pService = (PSERVICE_HANDLE)Handle;
228 }
229 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
230 {
231 DPRINT1("Exception: Invalid Service handle!\n");
232 }
233 _SEH2_END;
234
235 return pService;
236 }
237
238
239 static DWORD
240 ScmCheckAccess(SC_HANDLE Handle,
241 DWORD dwDesiredAccess)
242 {
243 PMANAGER_HANDLE hMgr;
244
245 hMgr = (PMANAGER_HANDLE)Handle;
246 if (hMgr->Handle.Tag == MANAGER_TAG)
247 {
248 RtlMapGenericMask(&dwDesiredAccess,
249 &ScmManagerMapping);
250
251 hMgr->Handle.DesiredAccess = dwDesiredAccess;
252
253 return ERROR_SUCCESS;
254 }
255 else if (hMgr->Handle.Tag == SERVICE_TAG)
256 {
257 RtlMapGenericMask(&dwDesiredAccess,
258 &ScmServiceMapping);
259
260 hMgr->Handle.DesiredAccess = dwDesiredAccess;
261
262 return ERROR_SUCCESS;
263 }
264
265 return ERROR_INVALID_HANDLE;
266 }
267
268
269 DWORD
270 ScmAssignNewTag(PSERVICE lpService)
271 {
272 HKEY hKey = NULL;
273 DWORD dwError;
274 DWORD dwGroupTagCount = 0;
275 PDWORD pdwGroupTags = NULL;
276 DWORD dwFreeTag = 0;
277 DWORD dwTagUsedBase = 1;
278 BOOLEAN TagUsed[TAG_ARRAY_SIZE];
279 INT nTagOffset;
280 DWORD i;
281 DWORD cbDataSize;
282 PLIST_ENTRY ServiceEntry;
283 PSERVICE CurrentService;
284
285 ASSERT(lpService != NULL);
286 ASSERT(lpService->lpGroup != NULL);
287
288 dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
289 L"System\\CurrentControlSet\\Control\\GroupOrderList",
290 0,
291 KEY_READ,
292 &hKey);
293
294 if (dwError != ERROR_SUCCESS)
295 goto findFreeTag;
296
297 /* query value length */
298 cbDataSize = 0;
299 dwError = RegQueryValueExW(hKey,
300 lpService->lpGroup->szGroupName,
301 NULL,
302 NULL,
303 NULL,
304 &cbDataSize);
305
306 if (dwError != ERROR_SUCCESS && dwError != ERROR_MORE_DATA)
307 goto findFreeTag;
308
309 pdwGroupTags = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, cbDataSize);
310 if (!pdwGroupTags)
311 {
312 dwError = ERROR_NOT_ENOUGH_MEMORY;
313 goto cleanup;
314 }
315
316 dwError = RegQueryValueExW(hKey,
317 lpService->lpGroup->szGroupName,
318 NULL,
319 NULL,
320 (LPBYTE)pdwGroupTags,
321 &cbDataSize);
322
323 if (dwError != ERROR_SUCCESS)
324 goto findFreeTag;
325
326 if (cbDataSize < sizeof(pdwGroupTags[0]))
327 goto findFreeTag;
328
329 dwGroupTagCount = min(pdwGroupTags[0], cbDataSize / sizeof(pdwGroupTags[0]) - 1);
330
331 findFreeTag:
332 do
333 {
334 /* mark all tags as unused */
335 for (i = 0; i < TAG_ARRAY_SIZE; i++)
336 TagUsed[i] = FALSE;
337
338 /* mark tags in GroupOrderList as used */
339 for (i = 1; i <= dwGroupTagCount; i++)
340 {
341 nTagOffset = pdwGroupTags[i] - dwTagUsedBase;
342 if (nTagOffset >= 0 && nTagOffset < TAG_ARRAY_SIZE)
343 TagUsed[nTagOffset] = TRUE;
344 }
345
346 /* mark tags in service list as used */
347 ServiceEntry = lpService->ServiceListEntry.Flink;
348 while (ServiceEntry != &lpService->ServiceListEntry)
349 {
350 ASSERT(ServiceEntry != NULL);
351 CurrentService = CONTAINING_RECORD(ServiceEntry, SERVICE, ServiceListEntry);
352 if (CurrentService->lpGroup == lpService->lpGroup)
353 {
354 nTagOffset = CurrentService->dwTag - dwTagUsedBase;
355 if (nTagOffset >= 0 && nTagOffset < TAG_ARRAY_SIZE)
356 TagUsed[nTagOffset] = TRUE;
357 }
358
359 ServiceEntry = ServiceEntry->Flink;
360 }
361
362 /* find unused tag, if any */
363 for (i = 0; i < TAG_ARRAY_SIZE; i++)
364 {
365 if (!TagUsed[i])
366 {
367 dwFreeTag = dwTagUsedBase + i;
368 break;
369 }
370 }
371
372 dwTagUsedBase += TAG_ARRAY_SIZE;
373 } while (!dwFreeTag);
374
375 cleanup:
376 if (pdwGroupTags)
377 HeapFree(GetProcessHeap(), 0, pdwGroupTags);
378
379 if (hKey)
380 RegCloseKey(hKey);
381
382 if (dwFreeTag)
383 {
384 lpService->dwTag = dwFreeTag;
385 DPRINT("Assigning new tag %lu to service %S in group %S\n",
386 lpService->dwTag, lpService->lpServiceName, lpService->lpGroup->szGroupName);
387 dwError = ERROR_SUCCESS;
388 }
389 else
390 {
391 DPRINT1("Failed to assign new tag to service %S, error=%lu\n",
392 lpService->lpServiceName, dwError);
393 }
394
395 return dwError;
396 }
397
398
399 /* Create a path suitable for the bootloader out of the full path */
400 DWORD
401 ScmConvertToBootPathName(wchar_t *CanonName, wchar_t **RelativeName)
402 {
403 SIZE_T ServiceNameLen, ExpandedLen;
404 DWORD BufferSize;
405 WCHAR Dest;
406 WCHAR *Expanded;
407 UNICODE_STRING NtPathName, SystemRoot, LinkTarget;
408 OBJECT_ATTRIBUTES ObjectAttributes;
409 NTSTATUS Status;
410 HANDLE SymbolicLinkHandle;
411
412 DPRINT("ScmConvertToBootPathName %S\n", CanonName);
413
414 if (!RelativeName)
415 return ERROR_INVALID_PARAMETER;
416
417 *RelativeName = NULL;
418
419 ServiceNameLen = wcslen(CanonName);
420
421 /* First check, if it's already good */
422 if (ServiceNameLen > 12 &&
423 !_wcsnicmp(L"\\SystemRoot\\", CanonName, 12))
424 {
425 *RelativeName = HeapAlloc(GetProcessHeap(),
426 HEAP_ZERO_MEMORY,
427 (ServiceNameLen + 1) * sizeof(WCHAR));
428 if (*RelativeName == NULL)
429 {
430 DPRINT("Error allocating memory for boot driver name!\n");
431 return ERROR_NOT_ENOUGH_MEMORY;
432 }
433
434 /* Copy it */
435 wcscpy(*RelativeName, CanonName);
436
437 DPRINT("Bootdriver name %S\n", *RelativeName);
438 return ERROR_SUCCESS;
439 }
440
441 /* If it has %SystemRoot% prefix, substitute it to \System*/
442 if (ServiceNameLen > 13 &&
443 !_wcsnicmp(L"%SystemRoot%\\", CanonName, 13))
444 {
445 /* There is no +sizeof(wchar_t) because the name is less by 1 wchar */
446 *RelativeName = HeapAlloc(GetProcessHeap(),
447 HEAP_ZERO_MEMORY,
448 ServiceNameLen * sizeof(WCHAR));
449
450 if (*RelativeName == NULL)
451 {
452 DPRINT("Error allocating memory for boot driver name!\n");
453 return ERROR_NOT_ENOUGH_MEMORY;
454 }
455
456 /* Copy it */
457 wcscpy(*RelativeName, L"\\SystemRoot\\");
458 wcscat(*RelativeName, CanonName + 13);
459
460 DPRINT("Bootdriver name %S\n", *RelativeName);
461 return ERROR_SUCCESS;
462 }
463
464 /* Get buffer size needed for expanding env strings */
465 BufferSize = ExpandEnvironmentStringsW(L"%SystemRoot%\\", &Dest, 1);
466 if (BufferSize <= 1)
467 {
468 DPRINT("Error during a call to ExpandEnvironmentStringsW()\n");
469 return ERROR_INVALID_ENVIRONMENT;
470 }
471
472 /* Allocate memory, since the size is known now */
473 Expanded = HeapAlloc(GetProcessHeap(),
474 HEAP_ZERO_MEMORY,
475 (BufferSize + 1) * sizeof(WCHAR));
476 if (!Expanded)
477 {
478 DPRINT("Error allocating memory for boot driver name!\n");
479 return ERROR_NOT_ENOUGH_MEMORY;
480 }
481
482 /* Expand it */
483 if (ExpandEnvironmentStringsW(L"%SystemRoot%\\", Expanded, BufferSize) >
484 BufferSize)
485 {
486 DPRINT("Error during a call to ExpandEnvironmentStringsW()\n");
487 HeapFree(GetProcessHeap(), 0, Expanded);
488 return ERROR_NOT_ENOUGH_MEMORY;
489 }
490
491 /* Convert to NT-style path */
492 if (!RtlDosPathNameToNtPathName_U(Expanded, &NtPathName, NULL, NULL))
493 {
494 DPRINT("Error during a call to RtlDosPathNameToNtPathName_U()\n");
495 return ERROR_INVALID_ENVIRONMENT;
496 }
497
498 DPRINT("Converted to NT-style %wZ\n", &NtPathName);
499
500 /* No need to keep the dos-path anymore */
501 HeapFree(GetProcessHeap(), 0, Expanded);
502
503 /* Copy it to the allocated place */
504 Expanded = HeapAlloc(GetProcessHeap(),
505 HEAP_ZERO_MEMORY,
506 NtPathName.Length + sizeof(UNICODE_NULL));
507 if (!Expanded)
508 {
509 DPRINT("Error allocating memory for boot driver name!\n");
510 RtlFreeUnicodeString(&NtPathName);
511 return ERROR_NOT_ENOUGH_MEMORY;
512 }
513
514 ExpandedLen = NtPathName.Length / sizeof(WCHAR);
515 wcsncpy(Expanded, NtPathName.Buffer, ExpandedLen);
516 Expanded[ExpandedLen] = UNICODE_NULL;
517 RtlFreeUnicodeString(&NtPathName);
518
519 if (ServiceNameLen > ExpandedLen &&
520 !_wcsnicmp(Expanded, CanonName, ExpandedLen))
521 {
522 HeapFree(GetProcessHeap(), 0, Expanded);
523
524 /* Only \SystemRoot\ is missing */
525 *RelativeName = HeapAlloc(GetProcessHeap(),
526 HEAP_ZERO_MEMORY,
527 (ServiceNameLen - ExpandedLen) * sizeof(WCHAR) + 13*sizeof(WCHAR));
528 if (*RelativeName == NULL)
529 {
530 DPRINT("Error allocating memory for boot driver name!\n");
531 return ERROR_NOT_ENOUGH_MEMORY;
532 }
533
534 wcscpy(*RelativeName, L"\\SystemRoot\\");
535 wcscat(*RelativeName, CanonName + ExpandedLen);
536
537 return ERROR_SUCCESS;
538 }
539
540 /* No longer need this */
541 HeapFree(GetProcessHeap(), 0, Expanded);
542
543 /* The most complex case starts here */
544 RtlInitUnicodeString(&SystemRoot, L"\\SystemRoot");
545 InitializeObjectAttributes(&ObjectAttributes,
546 &SystemRoot,
547 OBJ_CASE_INSENSITIVE,
548 NULL,
549 NULL);
550
551 /* Open this symlink */
552 Status = NtOpenSymbolicLinkObject(&SymbolicLinkHandle, SYMBOLIC_LINK_QUERY, &ObjectAttributes);
553 if (NT_SUCCESS(Status))
554 {
555 DPRINT("Opened symbolic link object\n");
556
557 RtlInitEmptyUnicodeString(&LinkTarget, NULL, 0);
558 Status = NtQuerySymbolicLinkObject(SymbolicLinkHandle, &LinkTarget, &BufferSize);
559 if (NT_SUCCESS(Status) || Status == STATUS_BUFFER_TOO_SMALL)
560 {
561 /* Check if required buffer size is sane */
562 if (BufferSize > UNICODE_STRING_MAX_BYTES - sizeof(UNICODE_NULL))
563 {
564 DPRINT("Too large buffer required\n");
565
566 NtClose(SymbolicLinkHandle);
567 return ERROR_NOT_ENOUGH_MEMORY;
568 }
569
570 /* Alloc the string */
571 LinkTarget.Length = (USHORT)BufferSize;
572 LinkTarget.MaximumLength = LinkTarget.Length + sizeof(UNICODE_NULL);
573 LinkTarget.Buffer = HeapAlloc(GetProcessHeap(),
574 HEAP_ZERO_MEMORY,
575 LinkTarget.MaximumLength);
576 if (!LinkTarget.Buffer)
577 {
578 DPRINT("Unable to alloc buffer\n");
579 NtClose(SymbolicLinkHandle);
580 return ERROR_NOT_ENOUGH_MEMORY;
581 }
582
583 /* Do a real query now */
584 Status = NtQuerySymbolicLinkObject(SymbolicLinkHandle, &LinkTarget, &BufferSize);
585 NtClose(SymbolicLinkHandle);
586 if (NT_SUCCESS(Status))
587 {
588 DPRINT("LinkTarget: %wZ\n", &LinkTarget);
589
590 ExpandedLen = LinkTarget.Length / sizeof(WCHAR);
591 if ((ServiceNameLen > ExpandedLen) &&
592 !_wcsnicmp(LinkTarget.Buffer, CanonName, ExpandedLen))
593 {
594 *RelativeName = HeapAlloc(GetProcessHeap(),
595 HEAP_ZERO_MEMORY,
596 (ServiceNameLen - ExpandedLen) * sizeof(WCHAR) + 13*sizeof(WCHAR));
597
598 if (*RelativeName == NULL)
599 {
600 DPRINT("Unable to alloc buffer\n");
601 return ERROR_NOT_ENOUGH_MEMORY;
602 }
603
604 /* Copy it over, substituting the first part
605 with SystemRoot */
606 wcscpy(*RelativeName, L"\\SystemRoot\\");
607 wcscat(*RelativeName, CanonName+ExpandedLen+1);
608
609 /* Return success */
610 return ERROR_SUCCESS;
611 }
612 else
613 {
614 return ERROR_INVALID_PARAMETER;
615 }
616 }
617 else
618 {
619 DPRINT("Error, Status = %08X\n", Status);
620 return ERROR_INVALID_PARAMETER;
621 }
622 }
623 else
624 {
625 DPRINT("Error, Status = %08X\n", Status);
626 NtClose(SymbolicLinkHandle);
627 return ERROR_INVALID_PARAMETER;
628 }
629 }
630 else
631 {
632 /* Failure */
633 DPRINT("Error, Status = %08X\n", Status);
634 return ERROR_INVALID_PARAMETER;
635 }
636 }
637
638
639 DWORD
640 ScmCanonDriverImagePath(DWORD dwStartType,
641 const wchar_t *lpServiceName,
642 wchar_t **lpCanonName)
643 {
644 DWORD Result;
645 SIZE_T ServiceNameLen;
646 UNICODE_STRING NtServiceName;
647 WCHAR *RelativeName;
648 const WCHAR *SourceName = lpServiceName;
649
650 /* Calculate the length of the service's name */
651 ServiceNameLen = wcslen(lpServiceName);
652
653 /* 12 is wcslen(L"\\SystemRoot\\") */
654 if (ServiceNameLen > 12 &&
655 !_wcsnicmp(L"\\SystemRoot\\", lpServiceName, 12))
656 {
657 /* SystemRoot prefix is already included */
658 *lpCanonName = HeapAlloc(GetProcessHeap(),
659 HEAP_ZERO_MEMORY,
660 (ServiceNameLen + 1) * sizeof(WCHAR));
661
662 if (*lpCanonName == NULL)
663 {
664 DPRINT("Error allocating memory for canonized service name!\n");
665 return ERROR_NOT_ENOUGH_MEMORY;
666 }
667
668 /* If it's a boot-time driver, it must be systemroot relative */
669 if (dwStartType == SERVICE_BOOT_START)
670 SourceName += 12;
671
672 /* Copy it */
673 wcscpy(*lpCanonName, SourceName);
674
675 DPRINT("Canonicalized name %S\n", *lpCanonName);
676 return NO_ERROR;
677 }
678
679 /* Check if it has %SystemRoot% (len=13) */
680 if (ServiceNameLen > 13 &&
681 !_wcsnicmp(L"%SystemRoot%\\", lpServiceName, 13))
682 {
683 /* Substitute %SystemRoot% with \\SystemRoot\\ */
684 *lpCanonName = HeapAlloc(GetProcessHeap(),
685 HEAP_ZERO_MEMORY,
686 (ServiceNameLen + 1) * sizeof(WCHAR));
687
688 if (*lpCanonName == NULL)
689 {
690 DPRINT("Error allocating memory for canonized service name!\n");
691 return ERROR_NOT_ENOUGH_MEMORY;
692 }
693
694 /* If it's a boot-time driver, it must be systemroot relative */
695 if (dwStartType == SERVICE_BOOT_START)
696 wcscpy(*lpCanonName, L"\\SystemRoot\\");
697
698 wcscat(*lpCanonName, lpServiceName + 13);
699
700 DPRINT("Canonicalized name %S\n", *lpCanonName);
701 return NO_ERROR;
702 }
703
704 /* Check if it's a relative path name */
705 if (lpServiceName[0] != L'\\' && lpServiceName[1] != L':')
706 {
707 *lpCanonName = HeapAlloc(GetProcessHeap(),
708 HEAP_ZERO_MEMORY,
709 (ServiceNameLen + 1) * sizeof(WCHAR));
710
711 if (*lpCanonName == NULL)
712 {
713 DPRINT("Error allocating memory for canonized service name!\n");
714 return ERROR_NOT_ENOUGH_MEMORY;
715 }
716
717 /* Just copy it over without changing */
718 wcscpy(*lpCanonName, lpServiceName);
719
720 return NO_ERROR;
721 }
722
723 /* It seems to be a DOS path, convert it */
724 if (!RtlDosPathNameToNtPathName_U(lpServiceName, &NtServiceName, NULL, NULL))
725 {
726 DPRINT("RtlDosPathNameToNtPathName_U() failed!\n");
727 return ERROR_INVALID_PARAMETER;
728 }
729
730 *lpCanonName = HeapAlloc(GetProcessHeap(),
731 HEAP_ZERO_MEMORY,
732 NtServiceName.Length + sizeof(WCHAR));
733
734 if (*lpCanonName == NULL)
735 {
736 DPRINT("Error allocating memory for canonized service name!\n");
737 RtlFreeUnicodeString(&NtServiceName);
738 return ERROR_NOT_ENOUGH_MEMORY;
739 }
740
741 /* Copy the string */
742 wcsncpy(*lpCanonName, NtServiceName.Buffer, NtServiceName.Length / sizeof(WCHAR));
743
744 /* The unicode string is not needed anymore */
745 RtlFreeUnicodeString(&NtServiceName);
746
747 if (dwStartType != SERVICE_BOOT_START)
748 {
749 DPRINT("Canonicalized name %S\n", *lpCanonName);
750 return NO_ERROR;
751 }
752
753 /* The service is boot-started, so must be relative */
754 Result = ScmConvertToBootPathName(*lpCanonName, &RelativeName);
755 if (Result)
756 {
757 /* There is a problem, free name and return */
758 HeapFree(GetProcessHeap(), 0, *lpCanonName);
759 DPRINT("Error converting named!\n");
760 return Result;
761 }
762
763 ASSERT(RelativeName);
764
765 /* Copy that string */
766 wcscpy(*lpCanonName, RelativeName + 12);
767
768 /* Free the allocated buffer */
769 HeapFree(GetProcessHeap(), 0, RelativeName);
770
771 DPRINT("Canonicalized name %S\n", *lpCanonName);
772
773 /* Success */
774 return NO_ERROR;
775 }
776
777
778 /* Internal recursive function */
779 /* Need to search for every dependency on every service */
780 static DWORD
781 Int_EnumDependentServicesW(HKEY hServicesKey,
782 PSERVICE lpService,
783 DWORD dwServiceState,
784 PSERVICE *lpServices,
785 LPDWORD pcbBytesNeeded,
786 LPDWORD lpServicesReturned)
787 {
788 DWORD dwError = ERROR_SUCCESS;
789 WCHAR szNameBuf[MAX_PATH];
790 WCHAR szValueBuf[MAX_PATH];
791 WCHAR *lpszNameBuf = szNameBuf;
792 WCHAR *lpszValueBuf = szValueBuf;
793 DWORD dwSize;
794 DWORD dwNumSubKeys;
795 DWORD dwIteration;
796 PSERVICE lpCurrentService;
797 HKEY hServiceEnumKey;
798 DWORD dwCurrentServiceState = SERVICE_ACTIVE;
799 DWORD dwDependServiceStrPtr = 0;
800 DWORD dwRequiredSize = 0;
801
802 /* Get the number of service keys */
803 dwError = RegQueryInfoKeyW(hServicesKey,
804 NULL,
805 NULL,
806 NULL,
807 &dwNumSubKeys,
808 NULL,
809 NULL,
810 NULL,
811 NULL,
812 NULL,
813 NULL,
814 NULL);
815 if (dwError != ERROR_SUCCESS)
816 {
817 DPRINT("ERROR! Unable to get number of services keys.\n");
818 return dwError;
819 }
820
821 /* Iterate the service keys to see if another service depends on the this service */
822 for (dwIteration = 0; dwIteration < dwNumSubKeys; dwIteration++)
823 {
824 dwSize = MAX_PATH;
825 dwError = RegEnumKeyExW(hServicesKey,
826 dwIteration,
827 lpszNameBuf,
828 &dwSize,
829 NULL,
830 NULL,
831 NULL,
832 NULL);
833 if (dwError != ERROR_SUCCESS)
834 return dwError;
835
836 /* Open the Service key */
837 dwError = RegOpenKeyExW(hServicesKey,
838 lpszNameBuf,
839 0,
840 KEY_READ,
841 &hServiceEnumKey);
842 if (dwError != ERROR_SUCCESS)
843 return dwError;
844
845 dwSize = MAX_PATH * sizeof(WCHAR);
846
847 /* Check for the DependOnService Value */
848 dwError = RegQueryValueExW(hServiceEnumKey,
849 L"DependOnService",
850 NULL,
851 NULL,
852 (LPBYTE)lpszValueBuf,
853 &dwSize);
854
855 /* FIXME: Handle load order. */
856
857 /* If the service found has a DependOnService value */
858 if (dwError == ERROR_SUCCESS)
859 {
860 dwDependServiceStrPtr = 0;
861
862 /* Can be more than one Dependencies in the DependOnService string */
863 while (wcslen(lpszValueBuf + dwDependServiceStrPtr) > 0)
864 {
865 if (_wcsicmp(lpszValueBuf + dwDependServiceStrPtr, lpService->lpServiceName) == 0)
866 {
867 /* Get the current enumed service pointer */
868 lpCurrentService = ScmGetServiceEntryByName(lpszNameBuf);
869
870 /* Check for valid Service */
871 if (!lpCurrentService)
872 {
873 /* This should never happen! */
874 DPRINT("This should not happen at this point, report to Developer\n");
875 return ERROR_NOT_FOUND;
876 }
877
878 /* Determine state the service is in */
879 if (lpCurrentService->Status.dwCurrentState == SERVICE_STOPPED)
880 dwCurrentServiceState = SERVICE_INACTIVE;
881
882 /* If the ServiceState matches that requested or searching for SERVICE_STATE_ALL */
883 if ((dwCurrentServiceState == dwServiceState) ||
884 (dwServiceState == SERVICE_STATE_ALL))
885 {
886 /* Calculate the required size */
887 dwRequiredSize += sizeof(SERVICE_STATUS);
888 dwRequiredSize += (DWORD)((wcslen(lpCurrentService->lpServiceName) + 1) * sizeof(WCHAR));
889 dwRequiredSize += (DWORD)((wcslen(lpCurrentService->lpDisplayName) + 1) * sizeof(WCHAR));
890
891 /* Add the size for service name and display name pointers */
892 dwRequiredSize += (2 * sizeof(PVOID));
893
894 /* increase the BytesNeeded size */
895 *pcbBytesNeeded = *pcbBytesNeeded + dwRequiredSize;
896
897 /* Don't fill callers buffer yet, as MSDN read that the last service with dependency
898 comes first */
899
900 /* Recursive call to check for its dependencies */
901 Int_EnumDependentServicesW(hServicesKey,
902 lpCurrentService,
903 dwServiceState,
904 lpServices,
905 pcbBytesNeeded,
906 lpServicesReturned);
907
908 /* If the lpServices is valid set the service pointer */
909 if (lpServices)
910 lpServices[*lpServicesReturned] = lpCurrentService;
911
912 *lpServicesReturned = *lpServicesReturned + 1;
913 }
914 }
915
916 dwDependServiceStrPtr += (DWORD)(wcslen(lpszValueBuf + dwDependServiceStrPtr) + 1);
917 }
918 }
919 else if (*pcbBytesNeeded)
920 {
921 dwError = ERROR_SUCCESS;
922 }
923
924 RegCloseKey(hServiceEnumKey);
925 }
926
927 return dwError;
928 }
929
930
931 /* Function 0 */
932 DWORD
933 WINAPI
934 RCloseServiceHandle(
935 LPSC_RPC_HANDLE hSCObject)
936 {
937 PMANAGER_HANDLE hManager;
938 PSERVICE_HANDLE hService;
939 PSERVICE lpService;
940 HKEY hServicesKey;
941 DWORD dwError;
942 DWORD pcbBytesNeeded = 0;
943 DWORD dwServicesReturned = 0;
944
945 DPRINT("RCloseServiceHandle() called\n");
946
947 DPRINT("hSCObject = %p\n", *hSCObject);
948
949 if (*hSCObject == 0)
950 return ERROR_INVALID_HANDLE;
951
952 hManager = ScmGetServiceManagerFromHandle(*hSCObject);
953 hService = ScmGetServiceFromHandle(*hSCObject);
954
955 if (hManager != NULL)
956 {
957 DPRINT("Found manager handle\n");
958
959 /* Make sure we don't access stale memory if someone tries to use this handle again. */
960 hManager->Handle.Tag = INVALID_TAG;
961
962 HeapFree(GetProcessHeap(), 0, hManager);
963 hManager = NULL;
964
965 *hSCObject = NULL;
966
967 DPRINT("RCloseServiceHandle() done\n");
968 return ERROR_SUCCESS;
969 }
970 else if (hService != NULL)
971 {
972 DPRINT("Found service handle\n");
973
974 /* Lock the service database exclusively */
975 ScmLockDatabaseExclusive();
976
977 /* Get the pointer to the service record */
978 lpService = hService->ServiceEntry;
979
980 /* Make sure we don't access stale memory if someone tries to use this handle again. */
981 hService->Handle.Tag = INVALID_TAG;
982
983 /* Free the handle */
984 HeapFree(GetProcessHeap(), 0, hService);
985 hService = NULL;
986
987 ASSERT(lpService->dwRefCount > 0);
988
989 lpService->dwRefCount--;
990 DPRINT("CloseServiceHandle - lpService->dwRefCount %u\n",
991 lpService->dwRefCount);
992
993 if (lpService->dwRefCount == 0)
994 {
995 /* If this service has been marked for deletion */
996 if (lpService->bDeleted &&
997 lpService->Status.dwCurrentState == SERVICE_STOPPED)
998 {
999 /* Open the Services Reg key */
1000 dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
1001 L"System\\CurrentControlSet\\Services",
1002 0,
1003 KEY_SET_VALUE | KEY_READ,
1004 &hServicesKey);
1005 if (dwError != ERROR_SUCCESS)
1006 {
1007 DPRINT("Failed to open services key\n");
1008 ScmUnlockDatabase();
1009 return dwError;
1010 }
1011
1012 /* Call the internal function with NULL, just to get bytes we need */
1013 Int_EnumDependentServicesW(hServicesKey,
1014 lpService,
1015 SERVICE_ACTIVE,
1016 NULL,
1017 &pcbBytesNeeded,
1018 &dwServicesReturned);
1019
1020 /* If pcbBytesNeeded returned a value then there are services running that are dependent on this service */
1021 if (pcbBytesNeeded)
1022 {
1023 DPRINT("Deletion failed due to running dependencies.\n");
1024 RegCloseKey(hServicesKey);
1025 ScmUnlockDatabase();
1026 return ERROR_SUCCESS;
1027 }
1028
1029 /* There are no references and no running dependencies,
1030 it is now safe to delete the service */
1031
1032 /* Delete the Service Key */
1033 dwError = ScmDeleteRegKey(hServicesKey,
1034 lpService->lpServiceName);
1035
1036 RegCloseKey(hServicesKey);
1037
1038 if (dwError != ERROR_SUCCESS)
1039 {
1040 DPRINT("Failed to Delete the Service Registry key\n");
1041 ScmUnlockDatabase();
1042 return dwError;
1043 }
1044
1045 /* Delete the Service */
1046 ScmDeleteServiceRecord(lpService);
1047 }
1048 }
1049
1050 ScmUnlockDatabase();
1051
1052 *hSCObject = NULL;
1053
1054 DPRINT("RCloseServiceHandle() done\n");
1055 return ERROR_SUCCESS;
1056 }
1057
1058 DPRINT("Invalid handle tag (Tag %lx)\n", hManager->Handle.Tag);
1059
1060 return ERROR_INVALID_HANDLE;
1061 }
1062
1063
1064 /* Function 1 */
1065 DWORD
1066 WINAPI
1067 RControlService(
1068 SC_RPC_HANDLE hService,
1069 DWORD dwControl,
1070 LPSERVICE_STATUS lpServiceStatus)
1071 {
1072 PSERVICE_HANDLE hSvc;
1073 PSERVICE lpService;
1074 ACCESS_MASK DesiredAccess;
1075 DWORD dwError = ERROR_SUCCESS;
1076 DWORD pcbBytesNeeded = 0;
1077 DWORD dwServicesReturned = 0;
1078 DWORD dwControlsAccepted;
1079 DWORD dwCurrentState;
1080 HKEY hServicesKey = NULL;
1081 LPCWSTR lpLogStrings[2];
1082 WCHAR szLogBuffer[80];
1083 UINT uID;
1084
1085 DPRINT("RControlService() called\n");
1086
1087 if (ScmShutdown)
1088 return ERROR_SHUTDOWN_IN_PROGRESS;
1089
1090 /* Check the service handle */
1091 hSvc = ScmGetServiceFromHandle(hService);
1092 if (hSvc == NULL)
1093 {
1094 DPRINT1("Invalid service handle!\n");
1095 return ERROR_INVALID_HANDLE;
1096 }
1097
1098 /* Check the service entry point */
1099 lpService = hSvc->ServiceEntry;
1100 if (lpService == NULL)
1101 {
1102 DPRINT1("lpService == NULL!\n");
1103 return ERROR_INVALID_HANDLE;
1104 }
1105
1106 /* Check access rights */
1107 switch (dwControl)
1108 {
1109 case SERVICE_CONTROL_STOP:
1110 DesiredAccess = SERVICE_STOP;
1111 break;
1112
1113 case SERVICE_CONTROL_PAUSE:
1114 case SERVICE_CONTROL_CONTINUE:
1115 case SERVICE_CONTROL_PARAMCHANGE:
1116 case SERVICE_CONTROL_NETBINDADD:
1117 case SERVICE_CONTROL_NETBINDREMOVE:
1118 case SERVICE_CONTROL_NETBINDENABLE:
1119 case SERVICE_CONTROL_NETBINDDISABLE:
1120 DesiredAccess = SERVICE_PAUSE_CONTINUE;
1121 break;
1122
1123 case SERVICE_CONTROL_INTERROGATE:
1124 DesiredAccess = SERVICE_INTERROGATE;
1125 break;
1126
1127 default:
1128 if (dwControl >= 128 && dwControl <= 255)
1129 DesiredAccess = SERVICE_USER_DEFINED_CONTROL;
1130 else
1131 return ERROR_INVALID_PARAMETER;
1132 break;
1133 }
1134
1135 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
1136 DesiredAccess))
1137 return ERROR_ACCESS_DENIED;
1138
1139 /* Return the current service status information */
1140 RtlCopyMemory(lpServiceStatus,
1141 &lpService->Status,
1142 sizeof(SERVICE_STATUS));
1143
1144 if (dwControl == SERVICE_CONTROL_STOP)
1145 {
1146 /* Check if the service has dependencies running as windows
1147 doesn't stop a service that does */
1148
1149 /* Open the Services Reg key */
1150 dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
1151 L"System\\CurrentControlSet\\Services",
1152 0,
1153 KEY_READ,
1154 &hServicesKey);
1155 if (dwError != ERROR_SUCCESS)
1156 {
1157 DPRINT("Failed to open services key\n");
1158 return dwError;
1159 }
1160
1161 /* Call the internal function with NULL, just to get bytes we need */
1162 Int_EnumDependentServicesW(hServicesKey,
1163 lpService,
1164 SERVICE_ACTIVE,
1165 NULL,
1166 &pcbBytesNeeded,
1167 &dwServicesReturned);
1168
1169 RegCloseKey(hServicesKey);
1170
1171 /* If pcbBytesNeeded is not zero then there are services running that
1172 are dependent on this service */
1173 if (pcbBytesNeeded != 0)
1174 {
1175 DPRINT("Service has running dependencies. Failed to stop service.\n");
1176 return ERROR_DEPENDENT_SERVICES_RUNNING;
1177 }
1178 }
1179
1180 if (lpService->Status.dwServiceType & SERVICE_DRIVER)
1181 {
1182 /* Send control code to the driver */
1183 dwError = ScmControlDriver(lpService,
1184 dwControl,
1185 lpServiceStatus);
1186 }
1187 else
1188 {
1189 dwControlsAccepted = lpService->Status.dwControlsAccepted;
1190 dwCurrentState = lpService->Status.dwCurrentState;
1191
1192 /* Return ERROR_SERVICE_NOT_ACTIVE if the service has not been started */
1193 if (lpService->lpImage == NULL || dwCurrentState == SERVICE_STOPPED)
1194 return ERROR_SERVICE_NOT_ACTIVE;
1195
1196 /* Check the current state before sending a control request */
1197 switch (dwCurrentState)
1198 {
1199 case SERVICE_STOP_PENDING:
1200 case SERVICE_STOPPED:
1201 return ERROR_SERVICE_CANNOT_ACCEPT_CTRL;
1202
1203 case SERVICE_START_PENDING:
1204 switch (dwControl)
1205 {
1206 case SERVICE_CONTROL_STOP:
1207 break;
1208
1209 case SERVICE_CONTROL_INTERROGATE:
1210 RtlCopyMemory(lpServiceStatus,
1211 &lpService->Status,
1212 sizeof(SERVICE_STATUS));
1213 return ERROR_SUCCESS;
1214
1215 default:
1216 return ERROR_SERVICE_CANNOT_ACCEPT_CTRL;
1217 }
1218 break;
1219 }
1220
1221 /* Check if the control code is acceptable to the service */
1222 switch (dwControl)
1223 {
1224 case SERVICE_CONTROL_STOP:
1225 if ((dwControlsAccepted & SERVICE_ACCEPT_STOP) == 0)
1226 return ERROR_INVALID_SERVICE_CONTROL;
1227 break;
1228
1229 case SERVICE_CONTROL_PAUSE:
1230 case SERVICE_CONTROL_CONTINUE:
1231 if ((dwControlsAccepted & SERVICE_ACCEPT_PAUSE_CONTINUE) == 0)
1232 return ERROR_INVALID_SERVICE_CONTROL;
1233 break;
1234
1235 case SERVICE_CONTROL_PARAMCHANGE:
1236 if ((dwControlsAccepted & SERVICE_ACCEPT_PARAMCHANGE) == 0)
1237 return ERROR_INVALID_SERVICE_CONTROL;
1238 break;
1239
1240 case SERVICE_CONTROL_NETBINDADD:
1241 case SERVICE_CONTROL_NETBINDREMOVE:
1242 case SERVICE_CONTROL_NETBINDENABLE:
1243 case SERVICE_CONTROL_NETBINDDISABLE:
1244 if ((dwControlsAccepted & SERVICE_ACCEPT_NETBINDCHANGE) == 0)
1245 return ERROR_INVALID_SERVICE_CONTROL;
1246 break;
1247 }
1248
1249 /* Send control code to the service */
1250 dwError = ScmControlService(lpService->lpImage->hControlPipe,
1251 lpService->lpServiceName,
1252 (SERVICE_STATUS_HANDLE)lpService,
1253 dwControl);
1254
1255 /* Return service status information */
1256 RtlCopyMemory(lpServiceStatus,
1257 &lpService->Status,
1258 sizeof(SERVICE_STATUS));
1259 }
1260
1261 if (dwError == ERROR_SUCCESS)
1262 {
1263 if (dwControl == SERVICE_CONTROL_STOP ||
1264 dwControl == SERVICE_CONTROL_PAUSE ||
1265 dwControl == SERVICE_CONTROL_CONTINUE)
1266 {
1267 /* Log a successful send control */
1268
1269 switch (dwControl)
1270 {
1271 case SERVICE_CONTROL_STOP:
1272 uID = IDS_SERVICE_STOP;
1273 break;
1274
1275 case SERVICE_CONTROL_PAUSE:
1276 uID = IDS_SERVICE_PAUSE;
1277 break;
1278
1279 case SERVICE_CONTROL_CONTINUE:
1280 uID = IDS_SERVICE_RESUME;
1281 break;
1282 }
1283 LoadStringW(GetModuleHandle(NULL), uID, szLogBuffer, ARRAYSIZE(szLogBuffer));
1284
1285 lpLogStrings[0] = lpService->lpDisplayName;
1286 lpLogStrings[1] = szLogBuffer;
1287
1288 ScmLogEvent(EVENT_SERVICE_CONTROL_SUCCESS,
1289 EVENTLOG_INFORMATION_TYPE,
1290 2,
1291 lpLogStrings);
1292 }
1293 }
1294
1295 return dwError;
1296 }
1297
1298
1299 /* Function 2 */
1300 DWORD
1301 WINAPI
1302 RDeleteService(
1303 SC_RPC_HANDLE hService)
1304 {
1305 PSERVICE_HANDLE hSvc;
1306 PSERVICE lpService;
1307 DWORD dwError;
1308
1309 DPRINT("RDeleteService() called\n");
1310
1311 if (ScmShutdown)
1312 return ERROR_SHUTDOWN_IN_PROGRESS;
1313
1314 hSvc = ScmGetServiceFromHandle(hService);
1315 if (hSvc == NULL)
1316 {
1317 DPRINT1("Invalid service handle!\n");
1318 return ERROR_INVALID_HANDLE;
1319 }
1320
1321 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
1322 DELETE))
1323 return ERROR_ACCESS_DENIED;
1324
1325 lpService = hSvc->ServiceEntry;
1326 if (lpService == NULL)
1327 {
1328 DPRINT("lpService == NULL!\n");
1329 return ERROR_INVALID_HANDLE;
1330 }
1331
1332 /* Lock the service database exclusively */
1333 ScmLockDatabaseExclusive();
1334
1335 if (lpService->bDeleted)
1336 {
1337 DPRINT("The service has already been marked for delete!\n");
1338 dwError = ERROR_SERVICE_MARKED_FOR_DELETE;
1339 goto Done;
1340 }
1341
1342 /* Mark service for delete */
1343 lpService->bDeleted = TRUE;
1344
1345 dwError = ScmMarkServiceForDelete(lpService);
1346
1347 Done:
1348 /* Unlock the service database */
1349 ScmUnlockDatabase();
1350
1351 DPRINT("RDeleteService() done\n");
1352
1353 return dwError;
1354 }
1355
1356
1357 /* Function 3 */
1358 DWORD
1359 WINAPI
1360 RLockServiceDatabase(
1361 SC_RPC_HANDLE hSCManager,
1362 LPSC_RPC_LOCK lpLock)
1363 {
1364 PMANAGER_HANDLE hMgr;
1365
1366 DPRINT("RLockServiceDatabase() called\n");
1367
1368 *lpLock = NULL;
1369
1370 hMgr = ScmGetServiceManagerFromHandle(hSCManager);
1371 if (hMgr == NULL)
1372 {
1373 DPRINT1("Invalid service manager handle!\n");
1374 return ERROR_INVALID_HANDLE;
1375 }
1376
1377 if (!RtlAreAllAccessesGranted(hMgr->Handle.DesiredAccess,
1378 SC_MANAGER_LOCK))
1379 return ERROR_ACCESS_DENIED;
1380
1381 return ScmAcquireServiceStartLock(FALSE, lpLock);
1382 }
1383
1384
1385 /* Function 4 */
1386 DWORD
1387 WINAPI
1388 RQueryServiceObjectSecurity(
1389 SC_RPC_HANDLE hService,
1390 SECURITY_INFORMATION dwSecurityInformation,
1391 LPBYTE lpSecurityDescriptor,
1392 DWORD cbBufSize,
1393 LPBOUNDED_DWORD_256K pcbBytesNeeded)
1394 {
1395 PSERVICE_HANDLE hSvc;
1396 PSERVICE lpService;
1397 ULONG DesiredAccess = 0;
1398 NTSTATUS Status;
1399 DWORD dwBytesNeeded;
1400 DWORD dwError;
1401
1402 DPRINT("RQueryServiceObjectSecurity() called\n");
1403
1404 hSvc = ScmGetServiceFromHandle(hService);
1405 if (hSvc == NULL)
1406 {
1407 DPRINT1("Invalid service handle!\n");
1408 return ERROR_INVALID_HANDLE;
1409 }
1410
1411 if (dwSecurityInformation & (DACL_SECURITY_INFORMATION |
1412 GROUP_SECURITY_INFORMATION |
1413 OWNER_SECURITY_INFORMATION))
1414 DesiredAccess |= READ_CONTROL;
1415
1416 if (dwSecurityInformation & SACL_SECURITY_INFORMATION)
1417 DesiredAccess |= ACCESS_SYSTEM_SECURITY;
1418
1419 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
1420 DesiredAccess))
1421 {
1422 DPRINT("Insufficient access rights! 0x%lx\n", hSvc->Handle.DesiredAccess);
1423 return ERROR_ACCESS_DENIED;
1424 }
1425
1426 lpService = hSvc->ServiceEntry;
1427 if (lpService == NULL)
1428 {
1429 DPRINT("lpService == NULL!\n");
1430 return ERROR_INVALID_HANDLE;
1431 }
1432
1433 /* Lock the service database */
1434 ScmLockDatabaseShared();
1435
1436 /* Retrieve the security descriptor */
1437 Status = RtlQuerySecurityObject(lpService->pSecurityDescriptor,
1438 dwSecurityInformation,
1439 (PSECURITY_DESCRIPTOR)lpSecurityDescriptor,
1440 cbBufSize,
1441 &dwBytesNeeded);
1442
1443 /* Unlock the service database */
1444 ScmUnlockDatabase();
1445
1446 if (NT_SUCCESS(Status))
1447 {
1448 *pcbBytesNeeded = dwBytesNeeded;
1449 dwError = STATUS_SUCCESS;
1450 }
1451 else if (Status == STATUS_BUFFER_TOO_SMALL)
1452 {
1453 *pcbBytesNeeded = dwBytesNeeded;
1454 dwError = ERROR_INSUFFICIENT_BUFFER;
1455 }
1456 else if (Status == STATUS_BAD_DESCRIPTOR_FORMAT)
1457 {
1458 dwError = ERROR_GEN_FAILURE;
1459 }
1460 else
1461 {
1462 dwError = RtlNtStatusToDosError(Status);
1463 }
1464
1465 return dwError;
1466 }
1467
1468
1469 /* Function 5 */
1470 DWORD
1471 WINAPI
1472 RSetServiceObjectSecurity(
1473 SC_RPC_HANDLE hService,
1474 DWORD dwSecurityInformation,
1475 LPBYTE lpSecurityDescriptor,
1476 DWORD dwSecurityDescriptorSize)
1477 {
1478 PSERVICE_HANDLE hSvc;
1479 PSERVICE lpService;
1480 ACCESS_MASK DesiredAccess = 0;
1481 HANDLE hToken = NULL;
1482 HKEY hServiceKey = NULL;
1483 BOOL bDatabaseLocked = FALSE;
1484 NTSTATUS Status;
1485 DWORD dwError;
1486
1487 DPRINT("RSetServiceObjectSecurity() called\n");
1488
1489 hSvc = ScmGetServiceFromHandle(hService);
1490 if (hSvc == NULL)
1491 {
1492 DPRINT1("Invalid service handle!\n");
1493 return ERROR_INVALID_HANDLE;
1494 }
1495
1496 if (dwSecurityInformation == 0 ||
1497 dwSecurityInformation & ~(OWNER_SECURITY_INFORMATION | GROUP_SECURITY_INFORMATION
1498 | DACL_SECURITY_INFORMATION | SACL_SECURITY_INFORMATION))
1499 {
1500 return ERROR_INVALID_PARAMETER;
1501 }
1502
1503 if (!RtlValidSecurityDescriptor((PSECURITY_DESCRIPTOR)lpSecurityDescriptor))
1504 return ERROR_INVALID_PARAMETER;
1505
1506 if (dwSecurityInformation & SACL_SECURITY_INFORMATION)
1507 DesiredAccess |= ACCESS_SYSTEM_SECURITY;
1508
1509 if (dwSecurityInformation & DACL_SECURITY_INFORMATION)
1510 DesiredAccess |= WRITE_DAC;
1511
1512 if (dwSecurityInformation & (OWNER_SECURITY_INFORMATION | GROUP_SECURITY_INFORMATION))
1513 DesiredAccess |= WRITE_OWNER;
1514
1515 if ((dwSecurityInformation & OWNER_SECURITY_INFORMATION) &&
1516 (((PISECURITY_DESCRIPTOR)lpSecurityDescriptor)->Owner == NULL))
1517 {
1518 return ERROR_INVALID_PARAMETER;
1519 }
1520
1521 if ((dwSecurityInformation & GROUP_SECURITY_INFORMATION) &&
1522 (((PISECURITY_DESCRIPTOR)lpSecurityDescriptor)->Group == NULL))
1523 {
1524 return ERROR_INVALID_PARAMETER;
1525 }
1526
1527 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
1528 DesiredAccess))
1529 {
1530 DPRINT1("Insufficient access rights! 0x%lx\n", hSvc->Handle.DesiredAccess);
1531 return ERROR_ACCESS_DENIED;
1532 }
1533
1534 lpService = hSvc->ServiceEntry;
1535 if (lpService == NULL)
1536 {
1537 DPRINT1("lpService == NULL!\n");
1538 return ERROR_INVALID_HANDLE;
1539 }
1540
1541 if (lpService->bDeleted)
1542 return ERROR_SERVICE_MARKED_FOR_DELETE;
1543
1544 #if 0
1545 RpcImpersonateClient(NULL);
1546
1547 Status = NtOpenThreadToken(NtCurrentThread(),
1548 8,
1549 TRUE,
1550 &hToken);
1551 if (!NT_SUCCESS(Status))
1552 return RtlNtStatusToDosError(Status);
1553
1554 RpcRevertToSelf();
1555 #endif
1556
1557 /* Build the new security descriptor */
1558 Status = RtlSetSecurityObject(dwSecurityInformation,
1559 (PSECURITY_DESCRIPTOR)lpSecurityDescriptor,
1560 &lpService->pSecurityDescriptor,
1561 &ScmServiceMapping,
1562 hToken);
1563 if (!NT_SUCCESS(Status))
1564 {
1565 dwError = RtlNtStatusToDosError(Status);
1566 goto Done;
1567 }
1568
1569 /* Lock the service database exclusive */
1570 ScmLockDatabaseExclusive();
1571 bDatabaseLocked = TRUE;
1572
1573 /* Open the service key */
1574 dwError = ScmOpenServiceKey(lpService->lpServiceName,
1575 READ_CONTROL | KEY_CREATE_SUB_KEY | KEY_SET_VALUE,
1576 &hServiceKey);
1577 if (dwError != ERROR_SUCCESS)
1578 goto Done;
1579
1580 /* Store the new security descriptor */
1581 dwError = ScmWriteSecurityDescriptor(hServiceKey,
1582 lpService->pSecurityDescriptor);
1583
1584 RegFlushKey(hServiceKey);
1585
1586 Done:
1587 if (hServiceKey != NULL)
1588 RegCloseKey(hServiceKey);
1589
1590 /* Unlock service database */
1591 if (bDatabaseLocked == TRUE)
1592 ScmUnlockDatabase();
1593
1594 if (hToken != NULL)
1595 NtClose(hToken);
1596
1597 DPRINT("RSetServiceObjectSecurity() done (Error %lu)\n", dwError);
1598
1599 return dwError;
1600 }
1601
1602
1603 /* Function 6 */
1604 DWORD
1605 WINAPI
1606 RQueryServiceStatus(
1607 SC_RPC_HANDLE hService,
1608 LPSERVICE_STATUS lpServiceStatus)
1609 {
1610 PSERVICE_HANDLE hSvc;
1611 PSERVICE lpService;
1612
1613 DPRINT("RQueryServiceStatus() called\n");
1614
1615 if (ScmShutdown)
1616 return ERROR_SHUTDOWN_IN_PROGRESS;
1617
1618 hSvc = ScmGetServiceFromHandle(hService);
1619 if (hSvc == NULL)
1620 {
1621 DPRINT1("Invalid service handle!\n");
1622 return ERROR_INVALID_HANDLE;
1623 }
1624
1625 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
1626 SERVICE_QUERY_STATUS))
1627 {
1628 DPRINT("Insufficient access rights! 0x%lx\n", hSvc->Handle.DesiredAccess);
1629 return ERROR_ACCESS_DENIED;
1630 }
1631
1632 lpService = hSvc->ServiceEntry;
1633 if (lpService == NULL)
1634 {
1635 DPRINT("lpService == NULL!\n");
1636 return ERROR_INVALID_HANDLE;
1637 }
1638
1639 /* Lock the service database shared */
1640 ScmLockDatabaseShared();
1641
1642 /* Return service status information */
1643 RtlCopyMemory(lpServiceStatus,
1644 &lpService->Status,
1645 sizeof(SERVICE_STATUS));
1646
1647 /* Unlock the service database */
1648 ScmUnlockDatabase();
1649
1650 return ERROR_SUCCESS;
1651 }
1652
1653
1654 static BOOL
1655 ScmIsValidServiceState(DWORD dwCurrentState)
1656 {
1657 switch (dwCurrentState)
1658 {
1659 case SERVICE_STOPPED:
1660 case SERVICE_START_PENDING:
1661 case SERVICE_STOP_PENDING:
1662 case SERVICE_RUNNING:
1663 case SERVICE_CONTINUE_PENDING:
1664 case SERVICE_PAUSE_PENDING:
1665 case SERVICE_PAUSED:
1666 return TRUE;
1667
1668 default:
1669 return FALSE;
1670 }
1671 }
1672
1673
1674 /* Function 7 */
1675 DWORD
1676 WINAPI
1677 RSetServiceStatus(
1678 RPC_SERVICE_STATUS_HANDLE hServiceStatus,
1679 LPSERVICE_STATUS lpServiceStatus)
1680 {
1681 PSERVICE lpService;
1682 DWORD dwPreviousState;
1683 DWORD dwPreviousType;
1684 LPCWSTR lpLogStrings[2];
1685 WCHAR szLogBuffer[80];
1686 UINT uID;
1687
1688 DPRINT("RSetServiceStatus() called\n");
1689 DPRINT("hServiceStatus = %lu\n", hServiceStatus);
1690 DPRINT("dwServiceType = 0x%lx\n", lpServiceStatus->dwServiceType);
1691 DPRINT("dwCurrentState = %lu\n", lpServiceStatus->dwCurrentState);
1692 DPRINT("dwControlsAccepted = %lu\n", lpServiceStatus->dwControlsAccepted);
1693 DPRINT("dwWin32ExitCode = %lu\n", lpServiceStatus->dwWin32ExitCode);
1694 DPRINT("dwServiceSpecificExitCode = %lu\n", lpServiceStatus->dwServiceSpecificExitCode);
1695 DPRINT("dwCheckPoint = %lu\n", lpServiceStatus->dwCheckPoint);
1696 DPRINT("dwWaitHint = %lu\n", lpServiceStatus->dwWaitHint);
1697
1698 if (hServiceStatus == 0)
1699 {
1700 DPRINT("hServiceStatus == NULL!\n");
1701 return ERROR_INVALID_HANDLE;
1702 }
1703
1704 lpService = (PSERVICE)hServiceStatus;
1705
1706 /* Check current state */
1707 if (!ScmIsValidServiceState(lpServiceStatus->dwCurrentState))
1708 {
1709 DPRINT("Invalid service state!\n");
1710 return ERROR_INVALID_DATA;
1711 }
1712
1713 /* Check service type */
1714 if (!(lpServiceStatus->dwServiceType & SERVICE_WIN32) &&
1715 (lpServiceStatus->dwServiceType & SERVICE_DRIVER))
1716 {
1717 DPRINT("Invalid service type!\n");
1718 return ERROR_INVALID_DATA;
1719 }
1720
1721 /* Check accepted controls */
1722 if (lpServiceStatus->dwControlsAccepted & ~0xFF)
1723 {
1724 DPRINT("Invalid controls accepted!\n");
1725 return ERROR_INVALID_DATA;
1726 }
1727
1728 /* Set the wait hint and check point only if the service is in a pending state,
1729 otherwise they should be 0 */
1730 if (lpServiceStatus->dwCurrentState == SERVICE_STOPPED ||
1731 lpServiceStatus->dwCurrentState == SERVICE_PAUSED ||
1732 lpServiceStatus->dwCurrentState == SERVICE_RUNNING)
1733 {
1734 lpServiceStatus->dwWaitHint = 0;
1735 lpServiceStatus->dwCheckPoint = 0;
1736 }
1737
1738 /* Lock the service database exclusively */
1739 ScmLockDatabaseExclusive();
1740
1741 /* Save the current service state */
1742 dwPreviousState = lpService->Status.dwCurrentState;
1743
1744 /* Save the current service type */
1745 dwPreviousType = lpService->Status.dwServiceType;
1746
1747 /* Update the service status */
1748 RtlCopyMemory(&lpService->Status,
1749 lpServiceStatus,
1750 sizeof(SERVICE_STATUS));
1751
1752 /* Restore the previous service type */
1753 lpService->Status.dwServiceType = dwPreviousType;
1754
1755 /* Dereference a stopped service */
1756 if ((lpServiceStatus->dwServiceType & SERVICE_WIN32) &&
1757 (lpServiceStatus->dwCurrentState == SERVICE_STOPPED))
1758 {
1759 /* Decrement the image run counter */
1760 lpService->lpImage->dwImageRunCount--;
1761
1762 /* If we just stopped the last running service... */
1763 if (lpService->lpImage->dwImageRunCount == 0)
1764 {
1765 /* Stop the dispatcher thread */
1766 ScmControlService(lpService->lpImage->hControlPipe,
1767 L"",
1768 (SERVICE_STATUS_HANDLE)lpService,
1769 SERVICE_CONTROL_STOP);
1770
1771 /* Remove the service image */
1772 ScmRemoveServiceImage(lpService->lpImage);
1773 lpService->lpImage = NULL;
1774 }
1775 }
1776
1777 /* Unlock the service database */
1778 ScmUnlockDatabase();
1779
1780 if ((lpServiceStatus->dwCurrentState == SERVICE_STOPPED) &&
1781 (dwPreviousState != SERVICE_STOPPED) &&
1782 (lpServiceStatus->dwWin32ExitCode != ERROR_SUCCESS))
1783 {
1784 /* Log a failed service stop */
1785 StringCchPrintfW(szLogBuffer, ARRAYSIZE(szLogBuffer),
1786 L"%lu", lpServiceStatus->dwWin32ExitCode);
1787 lpLogStrings[0] = lpService->lpDisplayName;
1788 lpLogStrings[1] = szLogBuffer;
1789
1790 ScmLogEvent(EVENT_SERVICE_EXIT_FAILED,
1791 EVENTLOG_ERROR_TYPE,
1792 2,
1793 lpLogStrings);
1794 }
1795 else if (lpServiceStatus->dwCurrentState != dwPreviousState &&
1796 (lpServiceStatus->dwCurrentState == SERVICE_STOPPED ||
1797 lpServiceStatus->dwCurrentState == SERVICE_RUNNING ||
1798 lpServiceStatus->dwCurrentState == SERVICE_PAUSED))
1799 {
1800 /* Log a successful service status change */
1801 switch(lpServiceStatus->dwCurrentState)
1802 {
1803 case SERVICE_STOPPED:
1804 uID = IDS_SERVICE_STOPPED;
1805 break;
1806
1807 case SERVICE_RUNNING:
1808 uID = IDS_SERVICE_RUNNING;
1809 break;
1810
1811 case SERVICE_PAUSED:
1812 uID = IDS_SERVICE_PAUSED;
1813 break;
1814 }
1815
1816 LoadStringW(GetModuleHandle(NULL), uID, szLogBuffer, ARRAYSIZE(szLogBuffer));
1817 lpLogStrings[0] = lpService->lpDisplayName;
1818 lpLogStrings[1] = szLogBuffer;
1819
1820 ScmLogEvent(EVENT_SERVICE_STATUS_SUCCESS,
1821 EVENTLOG_INFORMATION_TYPE,
1822 2,
1823 lpLogStrings);
1824 }
1825
1826 DPRINT("Set %S to %lu\n", lpService->lpDisplayName, lpService->Status.dwCurrentState);
1827 DPRINT("RSetServiceStatus() done\n");
1828
1829 return ERROR_SUCCESS;
1830 }
1831
1832
1833 /* Function 8 */
1834 DWORD
1835 WINAPI
1836 RUnlockServiceDatabase(
1837 LPSC_RPC_LOCK Lock)
1838 {
1839 DPRINT("RUnlockServiceDatabase(%p)\n", Lock);
1840 return ScmReleaseServiceStartLock(Lock);
1841 }
1842
1843
1844 /* Function 9 */
1845 DWORD
1846 WINAPI
1847 RNotifyBootConfigStatus(
1848 SVCCTL_HANDLEW lpMachineName,
1849 DWORD BootAcceptable)
1850 {
1851 DPRINT1("RNotifyBootConfigStatus(%p %lu) called\n", lpMachineName, BootAcceptable);
1852 return ERROR_SUCCESS;
1853
1854 // UNIMPLEMENTED;
1855 // return ERROR_CALL_NOT_IMPLEMENTED;
1856 }
1857
1858
1859 /* Function 10 */
1860 DWORD
1861 WINAPI
1862 RI_ScSetServiceBitsW(
1863 RPC_SERVICE_STATUS_HANDLE hServiceStatus,
1864 DWORD dwServiceBits,
1865 int bSetBitsOn,
1866 int bUpdateImmediately,
1867 wchar_t *lpString)
1868 {
1869 PSERVICE pService;
1870
1871 DPRINT("RI_ScSetServiceBitsW(%p %lx %d %d %S)\n",
1872 hServiceStatus, dwServiceBits, bSetBitsOn,
1873 bUpdateImmediately, lpString);
1874
1875 if (ScmShutdown)
1876 return ERROR_SHUTDOWN_IN_PROGRESS;
1877
1878 if (lpString != NULL)
1879 return ERROR_INVALID_PARAMETER;
1880
1881 if (hServiceStatus == 0)
1882 {
1883 DPRINT("hServiceStatus == NULL!\n");
1884 return ERROR_INVALID_HANDLE;
1885 }
1886
1887 // FIXME: Validate the status handle
1888 pService = (PSERVICE)hServiceStatus;
1889
1890 if (bSetBitsOn)
1891 {
1892 DPRINT("Old service bits: %08lx\n", pService->dwServiceBits);
1893 DPRINT("Old global service bits: %08lx\n", g_dwServiceBits);
1894 pService->dwServiceBits |= dwServiceBits;
1895 g_dwServiceBits |= dwServiceBits;
1896 DPRINT("New service bits: %08lx\n", pService->dwServiceBits);
1897 DPRINT("New global service bits: %08lx\n", g_dwServiceBits);
1898 }
1899 else
1900 {
1901 DPRINT("Old service bits: %08lx\n", pService->dwServiceBits);
1902 DPRINT("Old global service bits: %08lx\n", g_dwServiceBits);
1903 pService->dwServiceBits &= ~dwServiceBits;
1904 g_dwServiceBits &= ~dwServiceBits;
1905 DPRINT("New service bits: %08lx\n", pService->dwServiceBits);
1906 DPRINT("New global service bits: %08lx\n", g_dwServiceBits);
1907 }
1908
1909 return ERROR_SUCCESS;
1910 }
1911
1912
1913 /* Function 11 */
1914 DWORD
1915 WINAPI
1916 RChangeServiceConfigW(
1917 SC_RPC_HANDLE hService,
1918 DWORD dwServiceType,
1919 DWORD dwStartType,
1920 DWORD dwErrorControl,
1921 LPWSTR lpBinaryPathName,
1922 LPWSTR lpLoadOrderGroup,
1923 LPDWORD lpdwTagId,
1924 LPBYTE lpDependencies,
1925 DWORD dwDependSize,
1926 LPWSTR lpServiceStartName,
1927 LPBYTE lpPassword,
1928 DWORD dwPwSize,
1929 LPWSTR lpDisplayName)
1930 {
1931 DWORD dwError = ERROR_SUCCESS;
1932 PSERVICE_HANDLE hSvc;
1933 PSERVICE lpService = NULL;
1934 HKEY hServiceKey = NULL;
1935 LPWSTR lpDisplayNameW = NULL;
1936 LPWSTR lpImagePathW = NULL;
1937
1938 DPRINT("RChangeServiceConfigW() called\n");
1939 DPRINT("dwServiceType = 0x%lx\n", dwServiceType);
1940 DPRINT("dwStartType = %lu\n", dwStartType);
1941 DPRINT("dwErrorControl = %lu\n", dwErrorControl);
1942 DPRINT("lpBinaryPathName = %S\n", lpBinaryPathName);
1943 DPRINT("lpLoadOrderGroup = %S\n", lpLoadOrderGroup);
1944 DPRINT("lpDisplayName = %S\n", lpDisplayName);
1945
1946 if (ScmShutdown)
1947 return ERROR_SHUTDOWN_IN_PROGRESS;
1948
1949 hSvc = ScmGetServiceFromHandle(hService);
1950 if (hSvc == NULL)
1951 {
1952 DPRINT1("Invalid service handle!\n");
1953 return ERROR_INVALID_HANDLE;
1954 }
1955
1956 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
1957 SERVICE_CHANGE_CONFIG))
1958 {
1959 DPRINT("Insufficient access rights! 0x%lx\n", hSvc->Handle.DesiredAccess);
1960 return ERROR_ACCESS_DENIED;
1961 }
1962
1963 /* Check for invalid service type value */
1964 if ((dwServiceType != SERVICE_NO_CHANGE) &&
1965 (dwServiceType != SERVICE_KERNEL_DRIVER) &&
1966 (dwServiceType != SERVICE_FILE_SYSTEM_DRIVER) &&
1967 ((dwServiceType & ~SERVICE_INTERACTIVE_PROCESS) != SERVICE_WIN32_OWN_PROCESS) &&
1968 ((dwServiceType & ~SERVICE_INTERACTIVE_PROCESS) != SERVICE_WIN32_SHARE_PROCESS))
1969 {
1970 return ERROR_INVALID_PARAMETER;
1971 }
1972
1973 /* Check for invalid start type value */
1974 if ((dwStartType != SERVICE_NO_CHANGE) &&
1975 (dwStartType != SERVICE_BOOT_START) &&
1976 (dwStartType != SERVICE_SYSTEM_START) &&
1977 (dwStartType != SERVICE_AUTO_START) &&
1978 (dwStartType != SERVICE_DEMAND_START) &&
1979 (dwStartType != SERVICE_DISABLED))
1980 {
1981 return ERROR_INVALID_PARAMETER;
1982 }
1983
1984 /* Only drivers can be boot start or system start services */
1985 if ((dwStartType == SERVICE_BOOT_START) ||
1986 (dwStartType == SERVICE_SYSTEM_START))
1987 {
1988 if ((dwServiceType != SERVICE_KERNEL_DRIVER) &&
1989 (dwServiceType != SERVICE_FILE_SYSTEM_DRIVER))
1990 return ERROR_INVALID_PARAMETER;
1991 }
1992
1993 /* Check for invalid error control value */
1994 if ((dwErrorControl != SERVICE_NO_CHANGE) &&
1995 (dwErrorControl != SERVICE_ERROR_IGNORE) &&
1996 (dwErrorControl != SERVICE_ERROR_NORMAL) &&
1997 (dwErrorControl != SERVICE_ERROR_SEVERE) &&
1998 (dwErrorControl != SERVICE_ERROR_CRITICAL))
1999 {
2000 return ERROR_INVALID_PARAMETER;
2001 }
2002
2003 if (lpdwTagId && (!lpLoadOrderGroup || !*lpLoadOrderGroup))
2004 {
2005 return ERROR_INVALID_PARAMETER;
2006 }
2007
2008 lpService = hSvc->ServiceEntry;
2009 if (lpService == NULL)
2010 {
2011 DPRINT("lpService == NULL!\n");
2012 return ERROR_INVALID_HANDLE;
2013 }
2014
2015 /* Lock the service database exclusively */
2016 ScmLockDatabaseExclusive();
2017
2018 if (lpService->bDeleted)
2019 {
2020 DPRINT("The service has already been marked for delete!\n");
2021 dwError = ERROR_SERVICE_MARKED_FOR_DELETE;
2022 goto done;
2023 }
2024
2025 /* Open the service key */
2026 dwError = ScmOpenServiceKey(lpService->szServiceName,
2027 KEY_SET_VALUE,
2028 &hServiceKey);
2029 if (dwError != ERROR_SUCCESS)
2030 goto done;
2031
2032 /* Write service data to the registry */
2033
2034 /* Set the display name */
2035 if (lpDisplayName != NULL && *lpDisplayName != 0)
2036 {
2037 RegSetValueExW(hServiceKey,
2038 L"DisplayName",
2039 0,
2040 REG_SZ,
2041 (LPBYTE)lpDisplayName,
2042 (DWORD)((wcslen(lpDisplayName) + 1) * sizeof(WCHAR)));
2043
2044 /* Update the display name */
2045 lpDisplayNameW = HeapAlloc(GetProcessHeap(),
2046 HEAP_ZERO_MEMORY,
2047 (wcslen(lpDisplayName) + 1) * sizeof(WCHAR));
2048 if (lpDisplayNameW == NULL)
2049 {
2050 dwError = ERROR_NOT_ENOUGH_MEMORY;
2051 goto done;
2052 }
2053
2054 wcscpy(lpDisplayNameW, lpDisplayName);
2055 if (lpService->lpDisplayName != lpService->lpServiceName)
2056 HeapFree(GetProcessHeap(), 0, lpService->lpDisplayName);
2057
2058 lpService->lpDisplayName = lpDisplayNameW;
2059 }
2060
2061 if (dwServiceType != SERVICE_NO_CHANGE)
2062 {
2063 /* Set the service type */
2064 dwError = RegSetValueExW(hServiceKey,
2065 L"Type",
2066 0,
2067 REG_DWORD,
2068 (LPBYTE)&dwServiceType,
2069 sizeof(DWORD));
2070 if (dwError != ERROR_SUCCESS)
2071 goto done;
2072
2073 lpService->Status.dwServiceType = dwServiceType;
2074 }
2075
2076 if (dwStartType != SERVICE_NO_CHANGE)
2077 {
2078 /* Set the start value */
2079 dwError = RegSetValueExW(hServiceKey,
2080 L"Start",
2081 0,
2082 REG_DWORD,
2083 (LPBYTE)&dwStartType,
2084 sizeof(DWORD));
2085 if (dwError != ERROR_SUCCESS)
2086 goto done;
2087
2088 lpService->dwStartType = dwStartType;
2089 }
2090
2091 if (dwErrorControl != SERVICE_NO_CHANGE)
2092 {
2093 /* Set the error control value */
2094 dwError = RegSetValueExW(hServiceKey,
2095 L"ErrorControl",
2096 0,
2097 REG_DWORD,
2098 (LPBYTE)&dwErrorControl,
2099 sizeof(DWORD));
2100 if (dwError != ERROR_SUCCESS)
2101 goto done;
2102
2103 lpService->dwErrorControl = dwErrorControl;
2104 }
2105
2106 if (lpBinaryPathName != NULL && *lpBinaryPathName != 0)
2107 {
2108 /* Set the image path */
2109 lpImagePathW = lpBinaryPathName;
2110
2111 if (lpService->Status.dwServiceType & SERVICE_DRIVER)
2112 {
2113 dwError = ScmCanonDriverImagePath(lpService->dwStartType,
2114 lpBinaryPathName,
2115 &lpImagePathW);
2116
2117 if (dwError != ERROR_SUCCESS)
2118 goto done;
2119 }
2120
2121 dwError = RegSetValueExW(hServiceKey,
2122 L"ImagePath",
2123 0,
2124 REG_EXPAND_SZ,
2125 (LPBYTE)lpImagePathW,
2126 (DWORD)((wcslen(lpImagePathW) + 1) * sizeof(WCHAR)));
2127
2128 if (lpImagePathW != lpBinaryPathName)
2129 HeapFree(GetProcessHeap(), 0, lpImagePathW);
2130
2131 if (dwError != ERROR_SUCCESS)
2132 goto done;
2133 }
2134
2135 /* Set the group name */
2136 if (lpLoadOrderGroup != NULL && *lpLoadOrderGroup != 0)
2137 {
2138 dwError = RegSetValueExW(hServiceKey,
2139 L"Group",
2140 0,
2141 REG_SZ,
2142 (LPBYTE)lpLoadOrderGroup,
2143 (DWORD)((wcslen(lpLoadOrderGroup) + 1) * sizeof(WCHAR)));
2144 if (dwError != ERROR_SUCCESS)
2145 goto done;
2146
2147 dwError = ScmSetServiceGroup(lpService,
2148 lpLoadOrderGroup);
2149 if (dwError != ERROR_SUCCESS)
2150 goto done;
2151 }
2152
2153 /* Set the tag */
2154 if (lpdwTagId != NULL)
2155 {
2156 dwError = ScmAssignNewTag(lpService);
2157 if (dwError != ERROR_SUCCESS)
2158 goto done;
2159
2160 dwError = RegSetValueExW(hServiceKey,
2161 L"Tag",
2162 0,
2163 REG_DWORD,
2164 (LPBYTE)&lpService->dwTag,
2165 sizeof(DWORD));
2166 if (dwError != ERROR_SUCCESS)
2167 goto done;
2168
2169 *lpdwTagId = lpService->dwTag;
2170 }
2171
2172 /* Write dependencies */
2173 if (lpDependencies != NULL && *lpDependencies != 0)
2174 {
2175 dwError = ScmWriteDependencies(hServiceKey,
2176 (LPWSTR)lpDependencies,
2177 dwDependSize);
2178 if (dwError != ERROR_SUCCESS)
2179 goto done;
2180 }
2181
2182 /* Start name and password are only used by Win32 services */
2183 if (lpService->Status.dwServiceType & SERVICE_WIN32)
2184 {
2185 /* Write service start name */
2186 if (lpServiceStartName != NULL && *lpServiceStartName != 0)
2187 {
2188 dwError = RegSetValueExW(hServiceKey,
2189 L"ObjectName",
2190 0,
2191 REG_SZ,
2192 (LPBYTE)lpServiceStartName,
2193 (DWORD)((wcslen(lpServiceStartName) + 1) * sizeof(WCHAR)));
2194 if (dwError != ERROR_SUCCESS)
2195 goto done;
2196 }
2197
2198 if (lpPassword != NULL)
2199 {
2200 if (*(LPWSTR)lpPassword != 0)
2201 {
2202 /* FIXME: Decrypt the password */
2203
2204 /* Write the password */
2205 dwError = ScmSetServicePassword(lpService->szServiceName,
2206 (LPCWSTR)lpPassword);
2207 if (dwError != ERROR_SUCCESS)
2208 goto done;
2209 }
2210 else
2211 {
2212 /* Delete the password */
2213 dwError = ScmSetServicePassword(lpService->szServiceName,
2214 NULL);
2215 if (dwError == ERROR_FILE_NOT_FOUND)
2216 dwError = ERROR_SUCCESS;
2217
2218 if (dwError != ERROR_SUCCESS)
2219 goto done;
2220 }
2221 }
2222 }
2223
2224 done:
2225 if (hServiceKey != NULL)
2226 RegCloseKey(hServiceKey);
2227
2228 /* Unlock the service database */
2229 ScmUnlockDatabase();
2230
2231 DPRINT("RChangeServiceConfigW() done (Error %lu)\n", dwError);
2232
2233 return dwError;
2234 }
2235
2236
2237 /* Function 12 */
2238 DWORD
2239 WINAPI
2240 RCreateServiceW(
2241 SC_RPC_HANDLE hSCManager,
2242 LPCWSTR lpServiceName,
2243 LPCWSTR lpDisplayName,
2244 DWORD dwDesiredAccess,
2245 DWORD dwServiceType,
2246 DWORD dwStartType,
2247 DWORD dwErrorControl,
2248 LPCWSTR lpBinaryPathName,
2249 LPCWSTR lpLoadOrderGroup,
2250 LPDWORD lpdwTagId,
2251 LPBYTE lpDependencies,
2252 DWORD dwDependSize,
2253 LPCWSTR lpServiceStartName,
2254 LPBYTE lpPassword,
2255 DWORD dwPwSize,
2256 LPSC_RPC_HANDLE lpServiceHandle)
2257 {
2258 PMANAGER_HANDLE hManager;
2259 DWORD dwError = ERROR_SUCCESS;
2260 PSERVICE lpService = NULL;
2261 SC_HANDLE hServiceHandle = NULL;
2262 LPWSTR lpImagePath = NULL;
2263 HKEY hServiceKey = NULL;
2264 LPWSTR lpObjectName;
2265
2266 DPRINT("RCreateServiceW() called\n");
2267 DPRINT("lpServiceName = %S\n", lpServiceName);
2268 DPRINT("lpDisplayName = %S\n", lpDisplayName);
2269 DPRINT("dwDesiredAccess = %lx\n", dwDesiredAccess);
2270 DPRINT("dwServiceType = 0x%lx\n", dwServiceType);
2271 DPRINT("dwStartType = %lu\n", dwStartType);
2272 DPRINT("dwErrorControl = %lu\n", dwErrorControl);
2273 DPRINT("lpBinaryPathName = %S\n", lpBinaryPathName);
2274 DPRINT("lpLoadOrderGroup = %S\n", lpLoadOrderGroup);
2275 DPRINT("lpdwTagId = %p\n", lpdwTagId);
2276
2277 if (ScmShutdown)
2278 return ERROR_SHUTDOWN_IN_PROGRESS;
2279
2280 hManager = ScmGetServiceManagerFromHandle(hSCManager);
2281 if (hManager == NULL)
2282 {
2283 DPRINT1("Invalid service manager handle!\n");
2284 return ERROR_INVALID_HANDLE;
2285 }
2286
2287 /* Check access rights */
2288 if (!RtlAreAllAccessesGranted(hManager->Handle.DesiredAccess,
2289 SC_MANAGER_CREATE_SERVICE))
2290 {
2291 DPRINT("Insufficient access rights! 0x%lx\n",
2292 hManager->Handle.DesiredAccess);
2293 return ERROR_ACCESS_DENIED;
2294 }
2295
2296 if (*lpServiceName == 0)
2297 return ERROR_INVALID_NAME;
2298
2299 if (*lpBinaryPathName == 0)
2300 return ERROR_INVALID_PARAMETER;
2301
2302 /* Check for invalid service type value */
2303 if ((dwServiceType != SERVICE_KERNEL_DRIVER) &&
2304 (dwServiceType != SERVICE_FILE_SYSTEM_DRIVER) &&
2305 ((dwServiceType & ~SERVICE_INTERACTIVE_PROCESS) != SERVICE_WIN32_OWN_PROCESS) &&
2306 ((dwServiceType & ~SERVICE_INTERACTIVE_PROCESS) != SERVICE_WIN32_SHARE_PROCESS))
2307 {
2308 return ERROR_INVALID_PARAMETER;
2309 }
2310
2311 /* Check for invalid start type value */
2312 if ((dwStartType != SERVICE_BOOT_START) &&
2313 (dwStartType != SERVICE_SYSTEM_START) &&
2314 (dwStartType != SERVICE_AUTO_START) &&
2315 (dwStartType != SERVICE_DEMAND_START) &&
2316 (dwStartType != SERVICE_DISABLED))
2317 {
2318 return ERROR_INVALID_PARAMETER;
2319 }
2320
2321 /* Only drivers can be boot start or system start services */
2322 if ((dwStartType == SERVICE_BOOT_START) ||
2323 (dwStartType == SERVICE_SYSTEM_START))
2324 {
2325 if ((dwServiceType != SERVICE_KERNEL_DRIVER) &&
2326 (dwServiceType != SERVICE_FILE_SYSTEM_DRIVER))
2327 {
2328 return ERROR_INVALID_PARAMETER;
2329 }
2330 }
2331
2332 /* Check for invalid error control value */
2333 if ((dwErrorControl != SERVICE_ERROR_IGNORE) &&
2334 (dwErrorControl != SERVICE_ERROR_NORMAL) &&
2335 (dwErrorControl != SERVICE_ERROR_SEVERE) &&
2336 (dwErrorControl != SERVICE_ERROR_CRITICAL))
2337 {
2338 return ERROR_INVALID_PARAMETER;
2339 }
2340
2341 if ((dwServiceType == (SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS)) &&
2342 (lpServiceStartName))
2343 {
2344 /* We allow LocalSystem to run interactive. */
2345 if (wcsicmp(lpServiceStartName, L"LocalSystem"))
2346 {
2347 return ERROR_INVALID_PARAMETER;
2348 }
2349 }
2350
2351 if (lpdwTagId && (!lpLoadOrderGroup || !*lpLoadOrderGroup))
2352 {
2353 return ERROR_INVALID_PARAMETER;
2354 }
2355
2356 /* Lock the service database exclusively */
2357 ScmLockDatabaseExclusive();
2358
2359 lpService = ScmGetServiceEntryByName(lpServiceName);
2360 if (lpService)
2361 {
2362 /* Unlock the service database */
2363 ScmUnlockDatabase();
2364
2365 /* Check if it is marked for deletion */
2366 if (lpService->bDeleted)
2367 return ERROR_SERVICE_MARKED_FOR_DELETE;
2368
2369 /* Return service-exists error */
2370 return ERROR_SERVICE_EXISTS;
2371 }
2372
2373 if (lpDisplayName != NULL &&
2374 ScmGetServiceEntryByDisplayName(lpDisplayName) != NULL)
2375 {
2376 /* Unlock the service database */
2377 ScmUnlockDatabase();
2378
2379 return ERROR_DUPLICATE_SERVICE_NAME;
2380 }
2381
2382 if (dwServiceType & SERVICE_DRIVER)
2383 {
2384 dwError = ScmCanonDriverImagePath(dwStartType,
2385 lpBinaryPathName,
2386 &lpImagePath);
2387 if (dwError != ERROR_SUCCESS)
2388 goto done;
2389 }
2390 else
2391 {
2392 if (dwStartType == SERVICE_BOOT_START ||
2393 dwStartType == SERVICE_SYSTEM_START)
2394 {
2395 /* Unlock the service database */
2396 ScmUnlockDatabase();
2397
2398 return ERROR_INVALID_PARAMETER;
2399 }
2400 }
2401
2402 /* Allocate a new service entry */
2403 dwError = ScmCreateNewServiceRecord(lpServiceName,
2404 &lpService,
2405 dwServiceType,
2406 dwStartType);
2407 if (dwError != ERROR_SUCCESS)
2408 goto done;
2409
2410 /* Fill the new service entry */
2411 lpService->dwErrorControl = dwErrorControl;
2412
2413 /* Fill the display name */
2414 if (lpDisplayName != NULL &&
2415 *lpDisplayName != 0 &&
2416 _wcsicmp(lpService->lpDisplayName, lpDisplayName) != 0)
2417 {
2418 lpService->lpDisplayName = HeapAlloc(GetProcessHeap(),
2419 HEAP_ZERO_MEMORY,
2420 (wcslen(lpDisplayName) + 1) * sizeof(WCHAR));
2421 if (lpService->lpDisplayName == NULL)
2422 {
2423 dwError = ERROR_NOT_ENOUGH_MEMORY;
2424 goto done;
2425 }
2426 wcscpy(lpService->lpDisplayName, lpDisplayName);
2427 }
2428
2429 /* Assign the service to a group */
2430 if (lpLoadOrderGroup != NULL && *lpLoadOrderGroup != 0)
2431 {
2432 dwError = ScmSetServiceGroup(lpService,
2433 lpLoadOrderGroup);
2434 if (dwError != ERROR_SUCCESS)
2435 goto done;
2436 }
2437
2438 /* Assign a new tag */
2439 if (lpdwTagId != NULL)
2440 {
2441 dwError = ScmAssignNewTag(lpService);
2442 if (dwError != ERROR_SUCCESS)
2443 goto done;
2444 }
2445
2446 /* Assign the default security descriptor */
2447 if (dwServiceType & SERVICE_WIN32)
2448 {
2449 dwError = ScmCreateDefaultServiceSD(&lpService->pSecurityDescriptor);
2450 if (dwError != ERROR_SUCCESS)
2451 goto done;
2452 }
2453
2454 /* Write service data to the registry */
2455 /* Create the service key */
2456 dwError = ScmCreateServiceKey(lpServiceName,
2457 KEY_WRITE,
2458 &hServiceKey);
2459 if (dwError != ERROR_SUCCESS)
2460 goto done;
2461
2462 /* Set the display name */
2463 if (lpDisplayName != NULL && *lpDisplayName != 0)
2464 {
2465 RegSetValueExW(hServiceKey,
2466 L"DisplayName",
2467 0,
2468 REG_SZ,
2469 (LPBYTE)lpDisplayName,
2470 (DWORD)((wcslen(lpDisplayName) + 1) * sizeof(WCHAR)));
2471 }
2472
2473 /* Set the service type */
2474 dwError = RegSetValueExW(hServiceKey,
2475 L"Type",
2476 0,
2477 REG_DWORD,
2478 (LPBYTE)&dwServiceType,
2479 sizeof(DWORD));
2480 if (dwError != ERROR_SUCCESS)
2481 goto done;
2482
2483 /* Set the start value */
2484 dwError = RegSetValueExW(hServiceKey,
2485 L"Start",
2486 0,
2487 REG_DWORD,
2488 (LPBYTE)&dwStartType,
2489 sizeof(DWORD));
2490 if (dwError != ERROR_SUCCESS)
2491 goto done;
2492
2493 /* Set the error control value */
2494 dwError = RegSetValueExW(hServiceKey,
2495 L"ErrorControl",
2496 0,
2497 REG_DWORD,
2498 (LPBYTE)&dwErrorControl,
2499 sizeof(DWORD));
2500 if (dwError != ERROR_SUCCESS)
2501 goto done;
2502
2503 /* Set the image path */
2504 if (dwServiceType & SERVICE_WIN32)
2505 {
2506 dwError = RegSetValueExW(hServiceKey,
2507 L"ImagePath",
2508 0,
2509 REG_EXPAND_SZ,
2510 (LPBYTE)lpBinaryPathName,
2511 (DWORD)((wcslen(lpBinaryPathName) + 1) * sizeof(WCHAR)));
2512 if (dwError != ERROR_SUCCESS)
2513 goto done;
2514 }
2515 else if (dwServiceType & SERVICE_DRIVER)
2516 {
2517 dwError = RegSetValueExW(hServiceKey,
2518 L"ImagePath",
2519 0,
2520 REG_EXPAND_SZ,
2521 (LPBYTE)lpImagePath,
2522 (DWORD)((wcslen(lpImagePath) + 1) * sizeof(WCHAR)));
2523 if (dwError != ERROR_SUCCESS)
2524 goto done;
2525 }
2526
2527 /* Set the group name */
2528 if (lpLoadOrderGroup != NULL && *lpLoadOrderGroup != 0)
2529 {
2530 dwError = RegSetValueExW(hServiceKey,
2531 L"Group",
2532 0,
2533 REG_SZ,
2534 (LPBYTE)lpLoadOrderGroup,
2535 (DWORD)((wcslen(lpLoadOrderGroup) + 1) * sizeof(WCHAR)));
2536 if (dwError != ERROR_SUCCESS)
2537 goto done;
2538 }
2539
2540 /* Set the service tag */
2541 if (lpdwTagId != NULL)
2542 {
2543 dwError = RegSetValueExW(hServiceKey,
2544 L"Tag",
2545 0,
2546 REG_DWORD,
2547 (LPBYTE)&lpService->dwTag,
2548 sizeof(DWORD));
2549 if (dwError != ERROR_SUCCESS)
2550 goto done;
2551 }
2552
2553 /* Write dependencies */
2554 if (lpDependencies != NULL && *lpDependencies != 0)
2555 {
2556 dwError = ScmWriteDependencies(hServiceKey,
2557 (LPCWSTR)lpDependencies,
2558 dwDependSize);
2559 if (dwError != ERROR_SUCCESS)
2560 goto done;
2561 }
2562
2563 /* Start name and password are only used by Win32 services */
2564 if (dwServiceType & SERVICE_WIN32)
2565 {
2566 /* Write service start name */
2567 lpObjectName = (lpServiceStartName != NULL) ? (LPWSTR)lpServiceStartName : L"LocalSystem";
2568 dwError = RegSetValueExW(hServiceKey,
2569 L"ObjectName",
2570 0,
2571 REG_SZ,
2572 (LPBYTE)lpObjectName,
2573 (DWORD)((wcslen(lpObjectName) + 1) * sizeof(WCHAR)));
2574 if (dwError != ERROR_SUCCESS)
2575 goto done;
2576
2577 if (lpPassword != NULL && *(LPWSTR)lpPassword != 0)
2578 {
2579 /* FIXME: Decrypt the password */
2580
2581 /* Write the password */
2582 dwError = ScmSetServicePassword(lpServiceName,
2583 (LPCWSTR)lpPassword);
2584 if (dwError != ERROR_SUCCESS)
2585 goto done;
2586 }
2587
2588 /* Write the security descriptor */
2589 dwError = ScmWriteSecurityDescriptor(hServiceKey,
2590 lpService->pSecurityDescriptor);
2591 if (dwError != ERROR_SUCCESS)
2592 goto done;
2593 }
2594
2595 dwError = ScmCreateServiceHandle(lpService,
2596 &hServiceHandle);
2597 if (dwError != ERROR_SUCCESS)
2598 goto done;
2599
2600 dwError = ScmCheckAccess(hServiceHandle,
2601 dwDesiredAccess);
2602 if (dwError != ERROR_SUCCESS)
2603 goto done;
2604
2605 lpService->dwRefCount = 1;
2606 DPRINT("CreateService - lpService->dwRefCount %u\n", lpService->dwRefCount);
2607
2608 done:
2609 /* Unlock the service database */
2610 ScmUnlockDatabase();
2611
2612 if (hServiceKey != NULL)
2613 RegCloseKey(hServiceKey);
2614
2615 if (dwError == ERROR_SUCCESS)
2616 {
2617 DPRINT("hService %p\n", hServiceHandle);
2618 *lpServiceHandle = (SC_RPC_HANDLE)hServiceHandle;
2619
2620 if (lpdwTagId != NULL)
2621 *lpdwTagId = lpService->dwTag;
2622 }
2623 else
2624 {
2625 if (lpService != NULL &&
2626 lpService->lpServiceName != NULL)
2627 {
2628 /* Release the display name buffer */
2629 HeapFree(GetProcessHeap(), 0, lpService->lpDisplayName);
2630 }
2631
2632 if (hServiceHandle)
2633 {
2634 /* Remove the service handle */
2635 HeapFree(GetProcessHeap(), 0, hServiceHandle);
2636 }
2637
2638 if (lpService != NULL)
2639 {
2640 /* FIXME: remove the service entry */
2641 }
2642 }
2643
2644 if (lpImagePath != NULL)
2645 HeapFree(GetProcessHeap(), 0, lpImagePath);
2646
2647 DPRINT("RCreateServiceW() done (Error %lu)\n", dwError);
2648
2649 return dwError;
2650 }
2651
2652
2653 /* Function 13 */
2654 DWORD
2655 WINAPI
2656 REnumDependentServicesW(
2657 SC_RPC_HANDLE hService,
2658 DWORD dwServiceState,
2659 LPBYTE lpServices,
2660 DWORD cbBufSize,
2661 LPBOUNDED_DWORD_256K pcbBytesNeeded,
2662 LPBOUNDED_DWORD_256K lpServicesReturned)
2663 {
2664 DWORD dwError = ERROR_SUCCESS;
2665 DWORD dwServicesReturned = 0;
2666 DWORD dwServiceCount;
2667 HKEY hServicesKey = NULL;
2668 PSERVICE_HANDLE hSvc;
2669 PSERVICE lpService = NULL;
2670 PSERVICE *lpServicesArray = NULL;
2671 LPENUM_SERVICE_STATUSW lpServicesPtr = NULL;
2672 LPWSTR lpStr;
2673
2674 *pcbBytesNeeded = 0;
2675 *lpServicesReturned = 0;
2676
2677 DPRINT("REnumDependentServicesW() called\n");
2678
2679 hSvc = ScmGetServiceFromHandle(hService);
2680 if (hSvc == NULL)
2681 {
2682 DPRINT1("Invalid service handle!\n");
2683 return ERROR_INVALID_HANDLE;
2684 }
2685
2686 lpService = hSvc->ServiceEntry;
2687
2688 /* Check access rights */
2689 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
2690 SC_MANAGER_ENUMERATE_SERVICE))
2691 {
2692 DPRINT("Insufficient access rights! 0x%lx\n",
2693 hSvc->Handle.DesiredAccess);
2694 return ERROR_ACCESS_DENIED;
2695 }
2696
2697 /* Open the Services Reg key */
2698 dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
2699 L"System\\CurrentControlSet\\Services",
2700 0,
2701 KEY_READ,
2702 &hServicesKey);
2703 if (dwError != ERROR_SUCCESS)
2704 return dwError;
2705
2706 /* First determine the bytes needed and get the number of dependent services */
2707 dwError = Int_EnumDependentServicesW(hServicesKey,
2708 lpService,
2709 dwServiceState,
2710 NULL,
2711 pcbBytesNeeded,
2712 &dwServicesReturned);
2713 if (dwError != ERROR_SUCCESS)
2714 goto Done;
2715
2716 /* If buffer size is less than the bytes needed or pointer is null */
2717 if ((!lpServices) || (cbBufSize < *pcbBytesNeeded))
2718 {
2719 dwError = ERROR_MORE_DATA;
2720 goto Done;
2721 }
2722
2723 /* Allocate memory for array of service pointers */
2724 lpServicesArray = HeapAlloc(GetProcessHeap(),
2725 HEAP_ZERO_MEMORY,
2726 (dwServicesReturned + 1) * sizeof(PSERVICE));
2727 if (!lpServicesArray)
2728 {
2729 DPRINT1("Could not allocate a buffer!!\n");
2730 dwError = ERROR_NOT_ENOUGH_MEMORY;
2731 goto Done;
2732 }
2733
2734 dwServicesReturned = 0;
2735 *pcbBytesNeeded = 0;
2736
2737 dwError = Int_EnumDependentServicesW(hServicesKey,
2738 lpService,
2739 dwServiceState,
2740 lpServicesArray,
2741 pcbBytesNeeded,
2742 &dwServicesReturned);
2743 if (dwError != ERROR_SUCCESS)
2744 {
2745 goto Done;
2746 }
2747
2748 lpServicesPtr = (LPENUM_SERVICE_STATUSW)lpServices;
2749 lpStr = (LPWSTR)(lpServices + (dwServicesReturned * sizeof(ENUM_SERVICE_STATUSW)));
2750
2751 /* Copy EnumDepenedentService to Buffer */
2752 for (dwServiceCount = 0; dwServiceCount < dwServicesReturned; dwServiceCount++)
2753 {
2754 lpService = lpServicesArray[dwServiceCount];
2755
2756 /* Copy status info */
2757 memcpy(&lpServicesPtr->ServiceStatus,
2758 &lpService->Status,
2759 sizeof(SERVICE_STATUS));
2760
2761 /* Copy display name */
2762 wcscpy(lpStr, lpService->lpDisplayName);
2763 lpServicesPtr->lpDisplayName = (LPWSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServices);
2764 lpStr += (wcslen(lpService->lpDisplayName) + 1);
2765
2766 /* Copy service name */
2767 wcscpy(lpStr, lpService->lpServiceName);
2768 lpServicesPtr->lpServiceName = (LPWSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServices);
2769 lpStr += (wcslen(lpService->lpServiceName) + 1);
2770
2771 lpServicesPtr++;
2772 }
2773
2774 *lpServicesReturned = dwServicesReturned;
2775
2776 Done:
2777 if (lpServicesArray != NULL)
2778 HeapFree(GetProcessHeap(), 0, lpServicesArray);
2779
2780 RegCloseKey(hServicesKey);
2781
2782 DPRINT("REnumDependentServicesW() done (Error %lu)\n", dwError);
2783
2784 return dwError;
2785 }
2786
2787
2788 /* Function 14 */
2789 DWORD
2790 WINAPI
2791 REnumServicesStatusW(
2792 SC_RPC_HANDLE hSCManager,
2793 DWORD dwServiceType,
2794 DWORD dwServiceState,
2795 LPBYTE lpBuffer,
2796 DWORD dwBufSize,
2797 LPBOUNDED_DWORD_256K pcbBytesNeeded,
2798 LPBOUNDED_DWORD_256K lpServicesReturned,
2799 LPBOUNDED_DWORD_256K lpResumeHandle)
2800 {
2801 /* Enumerate all the services, not regarding of their group */
2802 return REnumServiceGroupW(hSCManager,
2803 dwServiceType,
2804 dwServiceState,
2805 lpBuffer,
2806 dwBufSize,
2807 pcbBytesNeeded,
2808 lpServicesReturned,
2809 lpResumeHandle,
2810 NULL);
2811 }
2812
2813
2814 /* Function 15 */
2815 DWORD
2816 WINAPI
2817 ROpenSCManagerW(
2818 LPWSTR lpMachineName,
2819 LPWSTR lpDatabaseName,
2820 DWORD dwDesiredAccess,
2821 LPSC_RPC_HANDLE lpScHandle)
2822 {
2823 DWORD dwError;
2824 SC_HANDLE hHandle;
2825
2826 DPRINT("ROpenSCManagerW() called\n");
2827 DPRINT("lpMachineName = %p\n", lpMachineName);
2828 DPRINT("lpMachineName: %S\n", lpMachineName);
2829 DPRINT("lpDataBaseName = %p\n", lpDatabaseName);
2830 DPRINT("lpDataBaseName: %S\n", lpDatabaseName);
2831 DPRINT("dwDesiredAccess = %x\n", dwDesiredAccess);
2832
2833 if (ScmShutdown)
2834 return ERROR_SHUTDOWN_IN_PROGRESS;
2835
2836 if (!lpScHandle)
2837 return ERROR_INVALID_PARAMETER;
2838
2839 dwError = ScmCreateManagerHandle(lpDatabaseName,
2840 &hHandle);
2841 if (dwError != ERROR_SUCCESS)
2842 {
2843 DPRINT("ScmCreateManagerHandle() failed (Error %lu)\n", dwError);
2844 return dwError;
2845 }
2846
2847 /* Check the desired access */
2848 dwError = ScmCheckAccess(hHandle,
2849 dwDesiredAccess | SC_MANAGER_CONNECT);
2850 if (dwError != ERROR_SUCCESS)
2851 {
2852 DPRINT("ScmCheckAccess() failed (Error %lu)\n", dwError);
2853 HeapFree(GetProcessHeap(), 0, hHandle);
2854 return dwError;
2855 }
2856
2857 *lpScHandle = (SC_RPC_HANDLE)hHandle;
2858 DPRINT("*hScm = %p\n", *lpScHandle);
2859
2860 DPRINT("ROpenSCManagerW() done\n");
2861
2862 return ERROR_SUCCESS;
2863 }
2864
2865
2866 /* Function 16 */
2867 DWORD
2868 WINAPI
2869 ROpenServiceW(
2870 SC_RPC_HANDLE hSCManager,
2871 LPWSTR lpServiceName,
2872 DWORD dwDesiredAccess,
2873 LPSC_RPC_HANDLE lpServiceHandle)
2874 {
2875 PSERVICE lpService;
2876 PMANAGER_HANDLE hManager;
2877 SC_HANDLE hHandle;
2878 DWORD dwError = ERROR_SUCCESS;
2879
2880 DPRINT("ROpenServiceW() called\n");
2881 DPRINT("hSCManager = %p\n", hSCManager);
2882 DPRINT("lpServiceName = %p\n", lpServiceName);
2883 DPRINT("lpServiceName: %S\n", lpServiceName);
2884 DPRINT("dwDesiredAccess = %x\n", dwDesiredAccess);
2885
2886 if (ScmShutdown)
2887 return ERROR_SHUTDOWN_IN_PROGRESS;
2888
2889 hManager = ScmGetServiceManagerFromHandle(hSCManager);
2890 if (hManager == NULL)
2891 {
2892 DPRINT1("Invalid service manager handle!\n");
2893 return ERROR_INVALID_HANDLE;
2894 }
2895
2896 if (!lpServiceHandle)
2897 return ERROR_INVALID_PARAMETER;
2898
2899 if (!lpServiceName)
2900 return ERROR_INVALID_ADDRESS;
2901
2902 /* Lock the service database exclusive */
2903 ScmLockDatabaseExclusive();
2904
2905 /* Get service database entry */
2906 lpService = ScmGetServiceEntryByName(lpServiceName);
2907 if (lpService == NULL)
2908 {
2909 DPRINT("Could not find a service!\n");
2910 dwError = ERROR_SERVICE_DOES_NOT_EXIST;
2911 goto Done;
2912 }
2913
2914 /* Create a service handle */
2915 dwError = ScmCreateServiceHandle(lpService,
2916 &hHandle);
2917 if (dwError != ERROR_SUCCESS)
2918 {
2919 DPRINT("ScmCreateServiceHandle() failed (Error %lu)\n", dwError);
2920 goto Done;
2921 }
2922
2923 /* Check the desired access */
2924 dwError = ScmCheckAccess(hHandle,
2925 dwDesiredAccess);
2926 if (dwError != ERROR_SUCCESS)
2927 {
2928 DPRINT("ScmCheckAccess() failed (Error %lu)\n", dwError);
2929 HeapFree(GetProcessHeap(), 0, hHandle);
2930 goto Done;
2931 }
2932
2933 lpService->dwRefCount++;
2934 DPRINT("OpenService - lpService->dwRefCount %u\n",lpService->dwRefCount);
2935
2936 *lpServiceHandle = (SC_RPC_HANDLE)hHandle;
2937 DPRINT("*hService = %p\n", *lpServiceHandle);
2938
2939 Done:
2940 /* Unlock the service database */
2941 ScmUnlockDatabase();
2942
2943 DPRINT("ROpenServiceW() done\n");
2944
2945 return dwError;
2946 }
2947
2948
2949 /* Function 17 */
2950 DWORD
2951 WINAPI
2952 RQueryServiceConfigW(
2953 SC_RPC_HANDLE hService,
2954 LPBYTE lpBuf, //LPQUERY_SERVICE_CONFIGW lpServiceConfig,
2955 DWORD cbBufSize,
2956 LPBOUNDED_DWORD_8K pcbBytesNeeded)
2957 {
2958 LPQUERY_SERVICE_CONFIGW lpServiceConfig = (LPQUERY_SERVICE_CONFIGW)lpBuf;
2959 DWORD dwError = ERROR_SUCCESS;
2960 PSERVICE_HANDLE hSvc;
2961 PSERVICE lpService = NULL;
2962 HKEY hServiceKey = NULL;
2963 LPWSTR lpImagePath = NULL;
2964 LPWSTR lpServiceStartName = NULL;
2965 LPWSTR lpDependencies = NULL;
2966 DWORD dwDependenciesLength = 0;
2967 DWORD dwRequiredSize;
2968 LPWSTR lpStr;
2969
2970 DPRINT("RQueryServiceConfigW() called\n");
2971
2972 if (ScmShutdown)
2973 return ERROR_SHUTDOWN_IN_PROGRESS;
2974
2975 hSvc = ScmGetServiceFromHandle(hService);
2976 if (hSvc == NULL)
2977 {
2978 DPRINT1("Invalid service handle!\n");
2979 return ERROR_INVALID_HANDLE;
2980 }
2981
2982 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
2983 SERVICE_QUERY_CONFIG))
2984 {
2985 DPRINT("Insufficient access rights! 0x%lx\n", hSvc->Handle.DesiredAccess);
2986 return ERROR_ACCESS_DENIED;
2987 }
2988
2989 lpService = hSvc->ServiceEntry;
2990 if (lpService == NULL)
2991 {
2992 DPRINT("lpService == NULL!\n");
2993 return ERROR_INVALID_HANDLE;
2994 }
2995
2996 /* Lock the service database shared */
2997 ScmLockDatabaseShared();
2998
2999 dwError = ScmOpenServiceKey(lpService->lpServiceName,
3000 KEY_READ,
3001 &hServiceKey);
3002 if (dwError != ERROR_SUCCESS)
3003 goto Done;
3004
3005 /* Read the image path */
3006 dwError = ScmReadString(hServiceKey,
3007 L"ImagePath",
3008 &lpImagePath);
3009 if (dwError != ERROR_SUCCESS)
3010 goto Done;
3011
3012 /* Read the service start name */
3013 ScmReadString(hServiceKey,
3014 L"ObjectName",
3015 &lpServiceStartName);
3016
3017 /* Read the dependencies */
3018 ScmReadDependencies(hServiceKey,
3019 &lpDependencies,
3020 &dwDependenciesLength);
3021
3022 dwRequiredSize = sizeof(QUERY_SERVICE_CONFIGW);
3023
3024 if (lpImagePath != NULL)
3025 dwRequiredSize += (DWORD)((wcslen(lpImagePath) + 1) * sizeof(WCHAR));
3026 else
3027 dwRequiredSize += 2 * sizeof(WCHAR);
3028
3029 if ((lpService->lpGroup != NULL) && (lpService->lpGroup->lpGroupName != NULL))
3030 dwRequiredSize += (DWORD)((wcslen(lpService->lpGroup->lpGroupName) + 1) * sizeof(WCHAR));
3031 else
3032 dwRequiredSize += 2 * sizeof(WCHAR);
3033
3034 if (lpDependencies != NULL)
3035 dwRequiredSize += dwDependenciesLength * sizeof(WCHAR);
3036 else
3037 dwRequiredSize += 2 * sizeof(WCHAR);
3038
3039 if (lpServiceStartName != NULL)
3040 dwRequiredSize += (DWORD)((wcslen(lpServiceStartName) + 1) * sizeof(WCHAR));
3041 else
3042 dwRequiredSize += 2 * sizeof(WCHAR);
3043
3044 if (lpService->lpDisplayName != NULL)
3045 dwRequiredSize += (DWORD)((wcslen(lpService->lpDisplayName) + 1) * sizeof(WCHAR));
3046 else
3047 dwRequiredSize += 2 * sizeof(WCHAR);
3048
3049 if (lpServiceConfig == NULL || cbBufSize < dwRequiredSize)
3050 {
3051 dwError = ERROR_INSUFFICIENT_BUFFER;
3052 }
3053 else
3054 {
3055 lpServiceConfig->dwServiceType = lpService->Status.dwServiceType;
3056 lpServiceConfig->dwStartType = lpService->dwStartType;
3057 lpServiceConfig->dwErrorControl = lpService->dwErrorControl;
3058 lpServiceConfig->dwTagId = lpService->dwTag;
3059
3060 lpStr = (LPWSTR)(lpServiceConfig + 1);
3061
3062 /* Append the image path */
3063 if (lpImagePath != NULL)
3064 {
3065 wcscpy(lpStr, lpImagePath);
3066 }
3067 else
3068 {
3069 *lpStr = 0;
3070 }
3071
3072 lpServiceConfig->lpBinaryPathName = (LPWSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServiceConfig);
3073 lpStr += (wcslen(lpStr) + 1);
3074
3075 /* Append the group name */
3076 if ((lpService->lpGroup != NULL) && (lpService->lpGroup->lpGroupName != NULL))
3077 {
3078 wcscpy(lpStr, lpService->lpGroup->lpGroupName);
3079 }
3080 else
3081 {
3082 *lpStr = 0;
3083 }
3084
3085 lpServiceConfig->lpLoadOrderGroup = (LPWSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServiceConfig);
3086 lpStr += (wcslen(lpStr) + 1);
3087
3088 /* Append Dependencies */
3089 if (lpDependencies != NULL)
3090 {
3091 memcpy(lpStr,
3092 lpDependencies,
3093 dwDependenciesLength * sizeof(WCHAR));
3094 }
3095 else
3096 {
3097 *lpStr = 0;
3098 }
3099
3100 lpServiceConfig->lpDependencies = (LPWSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServiceConfig);
3101 if (lpDependencies != NULL)
3102 lpStr += dwDependenciesLength;
3103 else
3104 lpStr += (wcslen(lpStr) + 1);
3105
3106 /* Append the service start name */
3107 if (lpServiceStartName != NULL)
3108 {
3109 wcscpy(lpStr, lpServiceStartName);
3110 }
3111 else
3112 {
3113 *lpStr = 0;
3114 }
3115
3116 lpServiceConfig->lpServiceStartName = (LPWSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServiceConfig);
3117 lpStr += (wcslen(lpStr) + 1);
3118
3119 /* Append the display name */
3120 if (lpService->lpDisplayName != NULL)
3121 {
3122 wcscpy(lpStr, lpService->lpDisplayName);
3123 }
3124 else
3125 {
3126 *lpStr = 0;
3127 }
3128
3129 lpServiceConfig->lpDisplayName = (LPWSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServiceConfig);
3130 }
3131
3132 if (pcbBytesNeeded != NULL)
3133 *pcbBytesNeeded = dwRequiredSize;
3134
3135 Done:
3136 /* Unlock the service database */
3137 ScmUnlockDatabase();
3138
3139 if (lpImagePath != NULL)
3140 HeapFree(GetProcessHeap(), 0, lpImagePath);
3141
3142 if (lpServiceStartName != NULL)
3143 HeapFree(GetProcessHeap(), 0, lpServiceStartName);
3144
3145 if (lpDependencies != NULL)
3146 HeapFree(GetProcessHeap(), 0, lpDependencies);
3147
3148 if (hServiceKey != NULL)
3149 RegCloseKey(hServiceKey);
3150
3151 DPRINT("RQueryServiceConfigW() done\n");
3152
3153 return dwError;
3154 }
3155
3156
3157 /* Function 18 */
3158 DWORD
3159 WINAPI
3160 RQueryServiceLockStatusW(
3161 SC_RPC_HANDLE hSCManager,
3162 LPBYTE lpBuf, // LPQUERY_SERVICE_LOCK_STATUSW lpLockStatus,
3163 DWORD cbBufSize,
3164 LPBOUNDED_DWORD_4K pcbBytesNeeded)
3165 {
3166 LPQUERY_SERVICE_LOCK_STATUSW lpLockStatus = (LPQUERY_SERVICE_LOCK_STATUSW)lpBuf;
3167 PMANAGER_HANDLE hMgr;
3168 DWORD dwRequiredSize;
3169
3170 if (!lpLockStatus || !pcbBytesNeeded)
3171 return ERROR_INVALID_PARAMETER;
3172
3173 hMgr = ScmGetServiceManagerFromHandle(hSCManager);
3174 if (hMgr == NULL)
3175 {
3176 DPRINT1("Invalid service manager handle!\n");
3177 return ERROR_INVALID_HANDLE;
3178 }
3179
3180 if (!RtlAreAllAccessesGranted(hMgr->Handle.DesiredAccess,
3181 SC_MANAGER_QUERY_LOCK_STATUS))
3182 {
3183 DPRINT("Insufficient access rights! 0x%lx\n", hMgr->Handle.DesiredAccess);
3184 return ERROR_ACCESS_DENIED;
3185 }
3186
3187 /* FIXME: we need to compute instead the real length of the owner name */
3188 dwRequiredSize = sizeof(QUERY_SERVICE_LOCK_STATUSW) + sizeof(WCHAR);
3189 *pcbBytesNeeded = dwRequiredSize;
3190
3191 if (cbBufSize < dwRequiredSize)
3192 return ERROR_INSUFFICIENT_BUFFER;
3193
3194 ScmQueryServiceLockStatusW(lpLockStatus);
3195
3196 return ERROR_SUCCESS;
3197 }
3198
3199
3200 /* Function 19 */
3201 DWORD
3202 WINAPI
3203 RStartServiceW(
3204 SC_RPC_HANDLE hService,
3205 DWORD argc,
3206 LPSTRING_PTRSW argv)
3207 {
3208 DWORD dwError = ERROR_SUCCESS;
3209 PSERVICE_HANDLE hSvc;
3210 PSERVICE lpService = NULL;
3211
3212 #ifndef NDEBUG
3213 DWORD i;
3214
3215 DPRINT("RStartServiceW(%p %lu %p) called\n", hService, argc, argv);
3216 DPRINT(" argc: %lu\n", argc);
3217 if (argv != NULL)
3218 {
3219 for (i = 0; i < argc; i++)
3220 {
3221 DPRINT(" argv[%lu]: %S\n", i, argv[i].StringPtr);
3222 }
3223 }
3224 #endif
3225
3226 if (ScmShutdown)
3227 return ERROR_SHUTDOWN_IN_PROGRESS;
3228
3229 hSvc = ScmGetServiceFromHandle(hService);
3230 if (hSvc == NULL)
3231 {
3232 DPRINT1("Invalid service handle!\n");
3233 return ERROR_INVALID_HANDLE;
3234 }
3235
3236 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
3237 SERVICE_START))
3238 {
3239 DPRINT("Insufficient access rights! 0x%lx\n", hSvc->Handle.DesiredAccess);
3240 return ERROR_ACCESS_DENIED;
3241 }
3242
3243 lpService = hSvc->ServiceEntry;
3244 if (lpService == NULL)
3245 {
3246 DPRINT("lpService == NULL!\n");
3247 return ERROR_INVALID_HANDLE;
3248 }
3249
3250 if (lpService->dwStartType == SERVICE_DISABLED)
3251 return ERROR_SERVICE_DISABLED;
3252
3253 if (lpService->bDeleted)
3254 return ERROR_SERVICE_MARKED_FOR_DELETE;
3255
3256 /* Start the service */
3257 dwError = ScmStartService(lpService, argc, (LPWSTR*)argv);
3258
3259 return dwError;
3260 }
3261
3262
3263 /* Function 20 */
3264 DWORD
3265 WINAPI
3266 RGetServiceDisplayNameW(
3267 SC_RPC_HANDLE hSCManager,
3268 LPCWSTR lpServiceName,
3269 LPWSTR lpDisplayName,
3270 DWORD *lpcchBuffer)
3271 {
3272 // PMANAGER_HANDLE hManager;
3273 PSERVICE lpService;
3274 DWORD dwLength;
3275 DWORD dwError;
3276
3277 DPRINT("RGetServiceDisplayNameW() called\n");
3278 DPRINT("hSCManager = %p\n", hSCManager);
3279 DPRINT("lpServiceName: %S\n", lpServiceName);
3280 DPRINT("lpDisplayName: %p\n", lpDisplayName);
3281 DPRINT("*lpcchBuffer: %lu\n", *lpcchBuffer);
3282
3283 // hManager = (PMANAGER_HANDLE)hSCManager;
3284 // if (hManager->Handle.Tag != MANAGER_TAG)
3285 // {
3286 // DPRINT("Invalid manager handle!\n");
3287 // return ERROR_INVALID_HANDLE;
3288 // }
3289
3290 /* Get service database entry */
3291 lpService = ScmGetServiceEntryByName(lpServiceName);
3292 if (lpService == NULL)
3293 {
3294 DPRINT("Could not find a service!\n");
3295
3296 /* If the service could not be found and lpcchBuffer is less than 2, windows
3297 puts null in lpDisplayName and puts 2 in lpcchBuffer */
3298 if (*lpcchBuffer < sizeof(WCHAR))
3299 {
3300 *lpcchBuffer = sizeof(WCHAR);
3301 if (lpDisplayName != NULL)
3302 {
3303 *lpDisplayName = 0;
3304 }
3305 }
3306
3307 return ERROR_SERVICE_DOES_NOT_EXIST;
3308 }
3309
3310 if (!lpService->lpDisplayName)
3311 {
3312 dwLength = (DWORD)wcslen(lpService->lpServiceName);
3313
3314 if (lpDisplayName != NULL &&
3315 *lpcchBuffer > dwLength)
3316 {
3317 wcscpy(lpDisplayName, lpService->lpServiceName);
3318 }
3319 }
3320 else
3321 {
3322 dwLength = (DWORD)wcslen(lpService->lpDisplayName);
3323
3324 if (lpDisplayName != NULL &&
3325 *lpcchBuffer > dwLength)
3326 {
3327 wcscpy(lpDisplayName, lpService->lpDisplayName);
3328 }
3329 }
3330
3331 dwError = (*lpcchBuffer > dwLength) ? ERROR_SUCCESS : ERROR_INSUFFICIENT_BUFFER;
3332
3333 *lpcchBuffer = dwLength;
3334
3335 return dwError;
3336 }
3337
3338
3339 /* Function 21 */
3340 DWORD
3341 WINAPI
3342 RGetServiceKeyNameW(
3343 SC_RPC_HANDLE hSCManager,
3344 LPCWSTR lpDisplayName,
3345 LPWSTR lpServiceName,
3346 DWORD *lpcchBuffer)
3347 {
3348 // PMANAGER_HANDLE hManager;
3349 PSERVICE lpService;
3350 DWORD dwLength;
3351 DWORD dwError;
3352
3353 DPRINT("RGetServiceKeyNameW() called\n");
3354 DPRINT("hSCManager = %p\n", hSCManager);
3355 DPRINT("lpDisplayName: %S\n", lpDisplayName);
3356 DPRINT("lpServiceName: %p\n", lpServiceName);
3357 DPRINT("*lpcchBuffer: %lu\n", *lpcchBuffer);
3358
3359 // hManager = (PMANAGER_HANDLE)hSCManager;
3360 // if (hManager->Handle.Tag != MANAGER_TAG)
3361 // {
3362 // DPRINT("Invalid manager handle!\n");
3363 // return ERROR_INVALID_HANDLE;
3364 // }
3365
3366 /* Get service database entry */
3367 lpService = ScmGetServiceEntryByDisplayName(lpDisplayName);
3368 if (lpService == NULL)
3369 {
3370 DPRINT("Could not find a service!\n");
3371
3372 /* If the service could not be found and lpcchBuffer is less than 2, windows
3373 puts null in lpDisplayName and puts 2 in lpcchBuffer */
3374 if (*lpcchBuffer < sizeof(WCHAR))
3375 {
3376 *lpcchBuffer = sizeof(WCHAR);
3377 if (lpServiceName != NULL)
3378 {
3379 *lpServiceName = 0;
3380 }
3381 }
3382
3383 return ERROR_SERVICE_DOES_NOT_EXIST;
3384 }
3385
3386 dwLength = (DWORD)wcslen(lpService->lpServiceName);
3387
3388 if (lpServiceName != NULL &&
3389 *lpcchBuffer > dwLength)
3390 {
3391 wcscpy(lpServiceName, lpService->lpServiceName);
3392 *lpcchBuffer = dwLength;
3393 return ERROR_SUCCESS;
3394 }
3395
3396 dwError = (*lpcchBuffer > dwLength) ? ERROR_SUCCESS : ERROR_INSUFFICIENT_BUFFER;
3397
3398 *lpcchBuffer = dwLength;
3399
3400 return dwError;
3401 }
3402
3403
3404 /* Function 22 */
3405 DWORD
3406 WINAPI
3407 RI_ScSetServiceBitsA(
3408 RPC_SERVICE_STATUS_HANDLE hServiceStatus,
3409 DWORD dwServiceBits,
3410 int bSetBitsOn,
3411 int bUpdateImmediately,
3412 char *lpString)
3413 {
3414 if (ScmShutdown)
3415 return ERROR_SHUTDOWN_IN_PROGRESS;
3416
3417 if (lpString != NULL)
3418 return ERROR_INVALID_PARAMETER;
3419
3420 return RI_ScSetServiceBitsW(hServiceStatus,
3421 dwServiceBits,
3422 bSetBitsOn,
3423 bUpdateImmediately,
3424 NULL);
3425 }
3426
3427
3428 /* Function 23 */
3429 DWORD
3430 WINAPI
3431 RChangeServiceConfigA(
3432 SC_RPC_HANDLE hService,
3433 DWORD dwServiceType,
3434 DWORD dwStartType,
3435 DWORD dwErrorControl,
3436 LPSTR lpBinaryPathName,
3437 LPSTR lpLoadOrderGroup,
3438 LPDWORD lpdwTagId,
3439 LPBYTE lpDependencies,
3440 DWORD dwDependSize,
3441 LPSTR lpServiceStartName,
3442 LPBYTE lpPassword,
3443 DWORD dwPwSize,
3444 LPSTR lpDisplayName)
3445 {
3446 DWORD dwError = ERROR_SUCCESS;
3447 LPWSTR lpBinaryPathNameW = NULL;
3448 LPWSTR lpLoadOrderGroupW = NULL;
3449 LPWSTR lpDependenciesW = NULL;
3450 LPWSTR lpServiceStartNameW = NULL;
3451 LPWSTR lpDisplayNameW = NULL;
3452 DWORD dwDependenciesLength = 0;
3453 SIZE_T cchLength;
3454 int len;
3455 LPCSTR lpStr;
3456
3457 if (lpBinaryPathName)
3458 {
3459 len = MultiByteToWideChar(CP_ACP, 0, lpBinaryPathName, -1, NULL, 0);
3460 lpBinaryPathNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3461 if (!lpBinaryPathNameW)
3462 {
3463 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3464 goto cleanup;
3465 }
3466 MultiByteToWideChar(CP_ACP, 0, lpBinaryPathName, -1, lpBinaryPathNameW, len);
3467 }
3468
3469 if (lpLoadOrderGroup)
3470 {
3471 len = MultiByteToWideChar(CP_ACP, 0, lpLoadOrderGroup, -1, NULL, 0);
3472 lpLoadOrderGroupW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3473 if (!lpLoadOrderGroupW)
3474 {
3475 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3476 goto cleanup;
3477 }
3478 MultiByteToWideChar(CP_ACP, 0, lpLoadOrderGroup, -1, lpLoadOrderGroupW, len);
3479 }
3480
3481 if (lpDependencies)
3482 {
3483 lpStr = (LPCSTR)lpDependencies;
3484 while (*lpStr)
3485 {
3486 cchLength = strlen(lpStr) + 1;
3487 dwDependenciesLength += (DWORD)cchLength;
3488 lpStr = lpStr + cchLength;
3489 }
3490 dwDependenciesLength++;
3491
3492 lpDependenciesW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwDependenciesLength * sizeof(WCHAR));
3493 if (!lpDependenciesW)
3494 {
3495 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3496 goto cleanup;
3497 }
3498 MultiByteToWideChar(CP_ACP, 0, (LPCSTR)lpDependencies, dwDependenciesLength, lpDependenciesW, dwDependenciesLength);
3499 }
3500
3501 if (lpServiceStartName)
3502 {
3503 len = MultiByteToWideChar(CP_ACP, 0, lpServiceStartName, -1, NULL, 0);
3504 lpServiceStartNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3505 if (!lpServiceStartNameW)
3506 {
3507 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3508 goto cleanup;
3509 }
3510 MultiByteToWideChar(CP_ACP, 0, lpServiceStartName, -1, lpServiceStartNameW, len);
3511 }
3512
3513 if (lpDisplayName)
3514 {
3515 len = MultiByteToWideChar(CP_ACP, 0, lpDisplayName, -1, NULL, 0);
3516 lpDisplayNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3517 if (!lpDisplayNameW)
3518 {
3519 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3520 goto cleanup;
3521 }
3522 MultiByteToWideChar(CP_ACP, 0, lpDisplayName, -1, lpDisplayNameW, len);
3523 }
3524
3525 dwError = RChangeServiceConfigW(hService,
3526 dwServiceType,
3527 dwStartType,
3528 dwErrorControl,
3529 lpBinaryPathNameW,
3530 lpLoadOrderGroupW,
3531 lpdwTagId,
3532 (LPBYTE)lpDependenciesW,
3533 dwDependenciesLength,
3534 lpServiceStartNameW,
3535 lpPassword,
3536 dwPwSize,
3537 lpDisplayNameW);
3538
3539 cleanup:
3540 if (lpBinaryPathNameW != NULL)
3541 HeapFree(GetProcessHeap(), 0, lpBinaryPathNameW);
3542
3543 if (lpLoadOrderGroupW != NULL)
3544 HeapFree(GetProcessHeap(), 0, lpLoadOrderGroupW);
3545
3546 if (lpDependenciesW != NULL)
3547 HeapFree(GetProcessHeap(), 0, lpDependenciesW);
3548
3549 if (lpServiceStartNameW != NULL)
3550 HeapFree(GetProcessHeap(), 0, lpServiceStartNameW);
3551
3552 if (lpDisplayNameW != NULL)
3553 HeapFree(GetProcessHeap(), 0, lpDisplayNameW);
3554
3555 return dwError;
3556 }
3557
3558
3559 /* Function 24 */
3560 DWORD
3561 WINAPI
3562 RCreateServiceA(
3563 SC_RPC_HANDLE hSCManager,
3564 LPSTR lpServiceName,
3565 LPSTR lpDisplayName,
3566 DWORD dwDesiredAccess,
3567 DWORD dwServiceType,
3568 DWORD dwStartType,
3569 DWORD dwErrorControl,
3570 LPSTR lpBinaryPathName,
3571 LPSTR lpLoadOrderGroup,
3572 LPDWORD lpdwTagId,
3573 LPBYTE lpDependencies,
3574 DWORD dwDependSize,
3575 LPSTR lpServiceStartName,
3576 LPBYTE lpPassword,
3577 DWORD dwPwSize,
3578 LPSC_RPC_HANDLE lpServiceHandle)
3579 {
3580 DWORD dwError = ERROR_SUCCESS;
3581 LPWSTR lpServiceNameW = NULL;
3582 LPWSTR lpDisplayNameW = NULL;
3583 LPWSTR lpBinaryPathNameW = NULL;
3584 LPWSTR lpLoadOrderGroupW = NULL;
3585 LPWSTR lpDependenciesW = NULL;
3586 LPWSTR lpServiceStartNameW = NULL;
3587 DWORD dwDependenciesLength = 0;
3588 SIZE_T cchLength;
3589 int len;
3590 LPCSTR lpStr;
3591
3592 if (lpServiceName)
3593 {
3594 len = MultiByteToWideChar(CP_ACP, 0, lpServiceName, -1, NULL, 0);
3595 lpServiceNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3596 if (!lpServiceNameW)
3597 {
3598 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3599 goto cleanup;
3600 }
3601 MultiByteToWideChar(CP_ACP, 0, lpServiceName, -1, lpServiceNameW, len);
3602 }
3603
3604 if (lpDisplayName)
3605 {
3606 len = MultiByteToWideChar(CP_ACP, 0, lpDisplayName, -1, NULL, 0);
3607 lpDisplayNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3608 if (!lpDisplayNameW)
3609 {
3610 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3611 goto cleanup;
3612 }
3613 MultiByteToWideChar(CP_ACP, 0, lpDisplayName, -1, lpDisplayNameW, len);
3614 }
3615
3616 if (lpBinaryPathName)
3617 {
3618 len = MultiByteToWideChar(CP_ACP, 0, lpBinaryPathName, -1, NULL, 0);
3619 lpBinaryPathNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3620 if (!lpBinaryPathNameW)
3621 {
3622 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3623 goto cleanup;
3624 }
3625 MultiByteToWideChar(CP_ACP, 0, lpBinaryPathName, -1, lpBinaryPathNameW, len);
3626 }
3627
3628 if (lpLoadOrderGroup)
3629 {
3630 len = MultiByteToWideChar(CP_ACP, 0, lpLoadOrderGroup, -1, NULL, 0);
3631 lpLoadOrderGroupW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3632 if (!lpLoadOrderGroupW)
3633 {
3634 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3635 goto cleanup;
3636 }
3637 MultiByteToWideChar(CP_ACP, 0, lpLoadOrderGroup, -1, lpLoadOrderGroupW, len);
3638 }
3639
3640 if (lpDependencies)
3641 {
3642 lpStr = (LPCSTR)lpDependencies;
3643 while (*lpStr)
3644 {
3645 cchLength = strlen(lpStr) + 1;
3646 dwDependenciesLength += (DWORD)cchLength;
3647 lpStr = lpStr + cchLength;
3648 }
3649 dwDependenciesLength++;
3650
3651 lpDependenciesW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwDependenciesLength * sizeof(WCHAR));
3652 if (!lpDependenciesW)
3653 {
3654 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3655 goto cleanup;
3656 }
3657 MultiByteToWideChar(CP_ACP, 0, (LPCSTR)lpDependencies, dwDependenciesLength, lpDependenciesW, dwDependenciesLength);
3658 }
3659
3660 if (lpServiceStartName)
3661 {
3662 len = MultiByteToWideChar(CP_ACP, 0, lpServiceStartName, -1, NULL, 0);
3663 lpServiceStartNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len * sizeof(WCHAR));
3664 if (!lpServiceStartNameW)
3665 {
3666 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
3667 goto cleanup;
3668 }
3669 MultiByteToWideChar(CP_ACP, 0, lpServiceStartName, -1, lpServiceStartNameW, len);
3670 }
3671
3672 dwError = RCreateServiceW(hSCManager,
3673 lpServiceNameW,
3674 lpDisplayNameW,
3675 dwDesiredAccess,
3676 dwServiceType,
3677 dwStartType,
3678 dwErrorControl,
3679 lpBinaryPathNameW,
3680 lpLoadOrderGroupW,
3681 lpdwTagId,
3682 (LPBYTE)lpDependenciesW,
3683 dwDependenciesLength,
3684 lpServiceStartNameW,
3685 lpPassword,
3686 dwPwSize,
3687 lpServiceHandle);
3688
3689 cleanup:
3690 if (lpServiceNameW !=NULL)
3691 HeapFree(GetProcessHeap(), 0, lpServiceNameW);
3692
3693 if (lpDisplayNameW != NULL)
3694 HeapFree(GetProcessHeap(), 0, lpDisplayNameW);
3695
3696 if (lpBinaryPathNameW != NULL)
3697 HeapFree(GetProcessHeap(), 0, lpBinaryPathNameW);
3698
3699 if (lpLoadOrderGroupW != NULL)
3700 HeapFree(GetProcessHeap(), 0, lpLoadOrderGroupW);
3701
3702 if (lpDependenciesW != NULL)
3703 HeapFree(GetProcessHeap(), 0, lpDependenciesW);
3704
3705 if (lpServiceStartNameW != NULL)
3706 HeapFree(GetProcessHeap(), 0, lpServiceStartNameW);
3707
3708 return dwError;
3709 }
3710
3711
3712 /* Function 25 */
3713 DWORD
3714 WINAPI
3715 REnumDependentServicesA(
3716 SC_RPC_HANDLE hService,
3717 DWORD dwServiceState,
3718 LPBYTE lpServices,
3719 DWORD cbBufSize,
3720 LPBOUNDED_DWORD_256K pcbBytesNeeded,
3721 LPBOUNDED_DWORD_256K lpServicesReturned)
3722 {
3723 DWORD dwError = ERROR_SUCCESS;
3724 DWORD dwServicesReturned = 0;
3725 DWORD dwServiceCount;
3726 HKEY hServicesKey = NULL;
3727 PSERVICE_HANDLE hSvc;
3728 PSERVICE lpService = NULL;
3729 PSERVICE *lpServicesArray = NULL;
3730 LPENUM_SERVICE_STATUSA lpServicesPtr = NULL;
3731 LPSTR lpStr;
3732
3733 *pcbBytesNeeded = 0;
3734 *lpServicesReturned = 0;
3735
3736 DPRINT("REnumDependentServicesA() called\n");
3737
3738 hSvc = ScmGetServiceFromHandle(hService);
3739 if (hSvc == NULL)
3740 {
3741 DPRINT1("Invalid service handle!\n");
3742 return ERROR_INVALID_HANDLE;
3743 }
3744
3745 lpService = hSvc->ServiceEntry;
3746
3747 /* Check access rights */
3748 if (!RtlAreAllAccessesGranted(hSvc->Handle.DesiredAccess,
3749 SC_MANAGER_ENUMERATE_SERVICE))
3750 {
3751 DPRINT("Insufficient access rights! 0x%lx\n",
3752 hSvc->Handle.DesiredAccess);
3753 return ERROR_ACCESS_DENIED;
3754 }
3755
3756 /* Open the Services Reg key */
3757 dwError = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
3758 L"System\\CurrentControlSet\\Services",
3759 0,
3760 KEY_READ,
3761 &hServicesKey);
3762
3763 if (dwError != ERROR_SUCCESS)
3764 return dwError;
3765
3766 /* NOTE: Windows calculates the pcbBytesNeeded based on WCHAR strings for
3767 both EnumDependentServicesA and EnumDependentServicesW. So returned pcbBytesNeeded
3768 are the same for both. Verified in WINXP. */
3769
3770 /* First determine the bytes needed and get the number of dependent services*/
3771 dwError = Int_EnumDependentServicesW(hServicesKey,
3772 lpService,
3773 dwServiceState,
3774 NULL,
3775 pcbBytesNeeded,
3776 &dwServicesReturned);
3777 if (dwError != ERROR_SUCCESS)
3778 goto Done;
3779
3780 /* If buffer size is less than the bytes needed or pointer is null*/
3781 if ((!lpServices) || (cbBufSize < *pcbBytesNeeded))
3782 {
3783 dwError = ERROR_MORE_DATA;
3784 goto Done;
3785 }
3786
3787 /* Allocate memory for array of service pointers */
3788 lpServicesArray = HeapAlloc(GetProcessHeap(),
3789 HEAP_ZERO_MEMORY,
3790 (dwServicesReturned + 1) * sizeof(PSERVICE));
3791 if (!lpServicesArray)
3792 {
3793 DPRINT("Could not allocate a buffer!!\n");
3794 dwError = ERROR_NOT_ENOUGH_MEMORY;
3795 goto Done;
3796 }
3797
3798 dwServicesReturned = 0;
3799 *pcbBytesNeeded = 0;
3800
3801 dwError = Int_EnumDependentServicesW(hServicesKey,
3802 lpService,
3803 dwServiceState,
3804 lpServicesArray,
3805 pcbBytesNeeded,
3806 &dwServicesReturned);
3807 if (dwError != ERROR_SUCCESS)
3808 {
3809 goto Done;
3810 }
3811
3812 lpServicesPtr = (LPENUM_SERVICE_STATUSA)lpServices;
3813 lpStr = (LPSTR)(lpServices + (dwServicesReturned * sizeof(ENUM_SERVICE_STATUSA)));
3814
3815 /* Copy EnumDepenedentService to Buffer */
3816 for (dwServiceCount = 0; dwServiceCount < dwServicesReturned; dwServiceCount++)
3817 {
3818 lpService = lpServicesArray[dwServiceCount];
3819
3820 /* Copy the status info */
3821 memcpy(&lpServicesPtr->ServiceStatus,
3822 &lpService->Status,
3823 sizeof(SERVICE_STATUS));
3824
3825 /* Copy display name */
3826 WideCharToMultiByte(CP_ACP,
3827 0,
3828 lpService->lpDisplayName,
3829 -1,
3830 lpStr,
3831 (int)wcslen(lpService->lpDisplayName),
3832 0,
3833 0);
3834 lpServicesPtr->lpDisplayName = (LPSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServices);
3835 lpStr += strlen(lpStr) + 1;
3836
3837 /* Copy service name */
3838 WideCharToMultiByte(CP_ACP,
3839 0,
3840 lpService->lpServiceName,
3841 -1,
3842 lpStr,
3843 (int)wcslen(lpService->lpServiceName),
3844 0,
3845 0);
3846 lpServicesPtr->lpServiceName = (LPSTR)((ULONG_PTR)lpStr - (ULONG_PTR)lpServices);
3847 lpStr += strlen(lpStr) + 1;
3848
3849 lpServicesPtr++;
3850 }
3851
3852 *lpServicesReturned = dwServicesReturned;
3853
3854 Done:
3855 if (lpServicesArray)
3856 HeapFree(GetProcessHeap(), 0, lpServicesArray);
3857
3858 RegCloseKey(hServicesKey);
3859
3860 DPRINT("REnumDependentServicesA() done (Error %lu)\n", dwError);
3861
3862 return dwError;
3863 }
3864
3865
3866 /* Function 26 */
3867 DWORD
3868 WINAPI
3869 REnumServicesStatusA(
3870 SC_RPC_HANDLE hSCManager,
3871 DWORD dwServiceType,
3872 DWORD dwServiceState,
3873 LPBYTE lpBuffer,
3874 DWORD dwBufSize,
3875 LPBOUNDED_DWORD_256K pcbBytesNeeded,
3876 LPBOUNDED_DWORD_256K lpServicesReturned,
3877 LPBOUNDED_DWORD_256K lpResumeHandle)
3878 {
3879 LPENUM_SERVICE_STATUSW lpStatusPtrW = NULL;
3880 LPENUM_SERVICE_STATUSW lpStatusPtrIncrW;
3881 LPENUM_SERVICE_STATUSA lpStatusPtrA = NULL;
3882 LPWSTR lpStringPtrW;
3883 LPSTR lpStringPtrA;
3884 DWORD dwError;
3885 DWORD dwServiceCount;
3886
3887 DPRINT("REnumServicesStatusA() called\n");
3888
3889 if (pcbBytesNeeded == NULL || lpServicesReturned == NULL)
3890 {
3891 return ERROR_INVALID_ADDRESS;
3892 }
3893
3894 if ((dwBufSize > 0) && (lpBuffer))
3895 {
3896 lpStatusPtrW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwBufSize);
3897 if (!lpStatusPtrW)
3898 {
3899 DPRINT("Failed to allocate buffer!\n");
3900 return ERROR_NOT_ENOUGH_MEMORY;
3901 }
3902 }
3903
3904 dwError = REnumServicesStatusW(hSCManager,
3905 dwServiceType,
3906 dwServiceState,
3907 (LPBYTE)lpStatusPtrW,
3908 dwBufSize,
3909 pcbBytesNeeded,