mirror of
https://github.com/mirror/processhacker
synced 2026-06-08 16:03:24 +00:00
wj32
650349dd46
* added PhSvcApiAddAccountRight git-svn-id: svn://svn.code.sf.net/p/processhacker/code@4238 21ef857c-d57f-4fe0-8362-d861dc6d29cd
547 lines
20 KiB
C
547 lines
20 KiB
C
/*
|
|
* Process Hacker -
|
|
* service properties
|
|
*
|
|
* Copyright (C) 2010-2011 wj32
|
|
*
|
|
* This file is part of Process Hacker.
|
|
*
|
|
* Process Hacker is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Process Hacker is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Process Hacker. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <phapp.h>
|
|
#include <phplug.h>
|
|
#include <phsvccl.h>
|
|
#include <windowsx.h>
|
|
|
|
typedef struct _SERVICE_PROPERTIES_CONTEXT
|
|
{
|
|
PPH_SERVICE_ITEM ServiceItem;
|
|
BOOLEAN Ready;
|
|
BOOLEAN Dirty;
|
|
|
|
BOOLEAN OldDelayedStart;
|
|
} SERVICE_PROPERTIES_CONTEXT, *PSERVICE_PROPERTIES_CONTEXT;
|
|
|
|
INT_PTR CALLBACK PhpServiceGeneralDlgProc(
|
|
__in HWND hwndDlg,
|
|
__in UINT uMsg,
|
|
__in WPARAM wParam,
|
|
__in LPARAM lParam
|
|
);
|
|
|
|
static NTSTATUS PhpOpenService(
|
|
__out PHANDLE Handle,
|
|
__in ACCESS_MASK DesiredAccess,
|
|
__in_opt PVOID Context
|
|
)
|
|
{
|
|
SC_HANDLE serviceHandle;
|
|
|
|
if (!(serviceHandle = PhOpenService(
|
|
((PPH_SERVICE_ITEM)Context)->Name->Buffer,
|
|
DesiredAccess
|
|
)))
|
|
return PhGetLastWin32ErrorAsNtStatus();
|
|
|
|
*Handle = serviceHandle;
|
|
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
VOID PhShowServiceProperties(
|
|
__in HWND ParentWindowHandle,
|
|
__in PPH_SERVICE_ITEM ServiceItem
|
|
)
|
|
{
|
|
PROPSHEETHEADER propSheetHeader = { sizeof(propSheetHeader) };
|
|
PROPSHEETPAGE propSheetPage;
|
|
HPROPSHEETPAGE pages[32];
|
|
SERVICE_PROPERTIES_CONTEXT context;
|
|
PH_STD_OBJECT_SECURITY stdObjectSecurity;
|
|
PPH_ACCESS_ENTRY accessEntries;
|
|
ULONG numberOfAccessEntries;
|
|
|
|
propSheetHeader.dwFlags =
|
|
PSH_NOAPPLYNOW |
|
|
PSH_NOCONTEXTHELP |
|
|
PSH_PROPTITLE;
|
|
propSheetHeader.hwndParent = ParentWindowHandle;
|
|
propSheetHeader.pszCaption = ServiceItem->Name->Buffer;
|
|
propSheetHeader.nPages = 0;
|
|
propSheetHeader.nStartPage = 0;
|
|
propSheetHeader.phpage = pages;
|
|
|
|
// General
|
|
|
|
memset(&context, 0, sizeof(SERVICE_PROPERTIES_CONTEXT));
|
|
context.ServiceItem = ServiceItem;
|
|
context.Ready = FALSE;
|
|
context.Dirty = FALSE;
|
|
|
|
memset(&propSheetPage, 0, sizeof(PROPSHEETPAGE));
|
|
propSheetPage.dwSize = sizeof(PROPSHEETPAGE);
|
|
propSheetPage.pszTemplate = MAKEINTRESOURCE(IDD_SRVGENERAL);
|
|
propSheetPage.pfnDlgProc = PhpServiceGeneralDlgProc;
|
|
propSheetPage.lParam = (LPARAM)&context;
|
|
pages[propSheetHeader.nPages++] = CreatePropertySheetPage(&propSheetPage);
|
|
|
|
// Security
|
|
|
|
stdObjectSecurity.OpenObject = PhpOpenService;
|
|
stdObjectSecurity.ObjectType = L"Service";
|
|
stdObjectSecurity.Context = ServiceItem;
|
|
|
|
if (PhGetAccessEntries(L"Service", &accessEntries, &numberOfAccessEntries))
|
|
{
|
|
pages[propSheetHeader.nPages++] = PhCreateSecurityPage(
|
|
ServiceItem->Name->Buffer,
|
|
PhStdGetObjectSecurity,
|
|
PhStdSetObjectSecurity,
|
|
&stdObjectSecurity,
|
|
accessEntries,
|
|
numberOfAccessEntries
|
|
);
|
|
PhFree(accessEntries);
|
|
}
|
|
|
|
if (PhPluginsEnabled)
|
|
{
|
|
PH_PLUGIN_OBJECT_PROPERTIES objectProperties;
|
|
|
|
objectProperties.Parameter = ServiceItem;
|
|
objectProperties.NumberOfPages = propSheetHeader.nPages;
|
|
objectProperties.MaximumNumberOfPages = sizeof(pages) / sizeof(HPROPSHEETPAGE);
|
|
objectProperties.Pages = pages;
|
|
|
|
PhInvokeCallback(PhGetGeneralCallback(GeneralCallbackServicePropertiesInitializing), &objectProperties);
|
|
|
|
propSheetHeader.nPages = objectProperties.NumberOfPages;
|
|
}
|
|
|
|
PropertySheet(&propSheetHeader);
|
|
}
|
|
|
|
static VOID PhpRefreshControls(
|
|
__in HWND hwndDlg
|
|
)
|
|
{
|
|
if (
|
|
WindowsVersion >= WINDOWS_VISTA &&
|
|
PhEqualString2(PHA_GET_DLGITEM_TEXT(hwndDlg, IDC_STARTTYPE), L"Auto Start", FALSE)
|
|
)
|
|
{
|
|
EnableWindow(GetDlgItem(hwndDlg, IDC_DELAYEDSTART), TRUE);
|
|
}
|
|
else
|
|
{
|
|
EnableWindow(GetDlgItem(hwndDlg, IDC_DELAYEDSTART), FALSE);
|
|
}
|
|
}
|
|
|
|
INT_PTR CALLBACK PhpServiceGeneralDlgProc(
|
|
__in HWND hwndDlg,
|
|
__in UINT uMsg,
|
|
__in WPARAM wParam,
|
|
__in LPARAM lParam
|
|
)
|
|
{
|
|
switch (uMsg)
|
|
{
|
|
case WM_INITDIALOG:
|
|
{
|
|
LPPROPSHEETPAGE propSheetPage = (LPPROPSHEETPAGE)lParam;
|
|
PSERVICE_PROPERTIES_CONTEXT context = (PSERVICE_PROPERTIES_CONTEXT)propSheetPage->lParam;
|
|
PPH_SERVICE_ITEM serviceItem = context->ServiceItem;
|
|
SC_HANDLE serviceHandle;
|
|
|
|
// HACK
|
|
PhCenterWindow(GetParent(hwndDlg), GetParent(GetParent(hwndDlg)));
|
|
|
|
SetProp(hwndDlg, PhMakeContextAtom(), (HANDLE)context);
|
|
|
|
PhAddComboBoxStrings(GetDlgItem(hwndDlg, IDC_TYPE), PhServiceTypeStrings,
|
|
sizeof(PhServiceTypeStrings) / sizeof(WCHAR *));
|
|
PhAddComboBoxStrings(GetDlgItem(hwndDlg, IDC_STARTTYPE), PhServiceStartTypeStrings,
|
|
sizeof(PhServiceStartTypeStrings) / sizeof(WCHAR *));
|
|
PhAddComboBoxStrings(GetDlgItem(hwndDlg, IDC_ERRORCONTROL), PhServiceErrorControlStrings,
|
|
sizeof(PhServiceErrorControlStrings) / sizeof(WCHAR *));
|
|
|
|
SetDlgItemText(hwndDlg, IDC_DESCRIPTION, serviceItem->DisplayName->Buffer);
|
|
ComboBox_SelectString(GetDlgItem(hwndDlg, IDC_TYPE), -1,
|
|
PhGetServiceTypeString(serviceItem->Type));
|
|
ComboBox_SelectString(GetDlgItem(hwndDlg, IDC_STARTTYPE), -1,
|
|
PhGetServiceStartTypeString(serviceItem->StartType));
|
|
ComboBox_SelectString(GetDlgItem(hwndDlg, IDC_ERRORCONTROL), -1,
|
|
PhGetServiceErrorControlString(serviceItem->ErrorControl));
|
|
|
|
serviceHandle = PhOpenService(serviceItem->Name->Buffer, SERVICE_QUERY_CONFIG);
|
|
|
|
if (serviceHandle)
|
|
{
|
|
LPQUERY_SERVICE_CONFIG config;
|
|
PPH_STRING description;
|
|
BOOLEAN delayedStart;
|
|
|
|
if (config = PhGetServiceConfig(serviceHandle))
|
|
{
|
|
SetDlgItemText(hwndDlg, IDC_GROUP, config->lpLoadOrderGroup);
|
|
SetDlgItemText(hwndDlg, IDC_BINARYPATH, config->lpBinaryPathName);
|
|
SetDlgItemText(hwndDlg, IDC_USERACCOUNT, config->lpServiceStartName);
|
|
|
|
PhFree(config);
|
|
}
|
|
|
|
if (description = PhGetServiceDescription(serviceHandle))
|
|
{
|
|
SetDlgItemText(hwndDlg, IDC_DESCRIPTION, description->Buffer);
|
|
PhDereferenceObject(description);
|
|
}
|
|
|
|
if (
|
|
WindowsVersion >= WINDOWS_VISTA &&
|
|
PhGetServiceDelayedAutoStart(serviceHandle, &delayedStart)
|
|
)
|
|
{
|
|
context->OldDelayedStart = delayedStart;
|
|
|
|
if (delayedStart)
|
|
Button_SetCheck(GetDlgItem(hwndDlg, IDC_DELAYEDSTART), BST_CHECKED);
|
|
}
|
|
|
|
CloseServiceHandle(serviceHandle);
|
|
}
|
|
|
|
SetDlgItemText(hwndDlg, IDC_PASSWORD, L"password");
|
|
Button_SetCheck(GetDlgItem(hwndDlg, IDC_PASSWORDCHECK), BST_UNCHECKED);
|
|
|
|
SetDlgItemText(hwndDlg, IDC_SERVICEDLL, L"N/A");
|
|
|
|
{
|
|
HANDLE keyHandle;
|
|
PPH_STRING keyName;
|
|
|
|
keyName = PhConcatStrings(
|
|
3,
|
|
L"System\\CurrentControlSet\\Services\\",
|
|
serviceItem->Name->Buffer,
|
|
L"\\Parameters"
|
|
);
|
|
|
|
if (NT_SUCCESS(PhOpenKey(
|
|
&keyHandle,
|
|
KEY_READ,
|
|
PH_KEY_LOCAL_MACHINE,
|
|
&keyName->sr,
|
|
0
|
|
)))
|
|
{
|
|
PPH_STRING serviceDllString;
|
|
|
|
if (serviceDllString = PhQueryRegistryString(keyHandle, L"ServiceDll"))
|
|
{
|
|
PPH_STRING expandedString;
|
|
|
|
if (expandedString = PhExpandEnvironmentStrings(&serviceDllString->sr))
|
|
{
|
|
SetDlgItemText(hwndDlg, IDC_SERVICEDLL, expandedString->Buffer);
|
|
PhDereferenceObject(expandedString);
|
|
}
|
|
|
|
PhDereferenceObject(serviceDllString);
|
|
}
|
|
|
|
NtClose(keyHandle);
|
|
}
|
|
|
|
PhDereferenceObject(keyName);
|
|
}
|
|
|
|
PhpRefreshControls(hwndDlg);
|
|
|
|
context->Ready = TRUE;
|
|
}
|
|
break;
|
|
case WM_DESTROY:
|
|
{
|
|
RemoveProp(hwndDlg, PhMakeContextAtom());
|
|
}
|
|
break;
|
|
case WM_COMMAND:
|
|
{
|
|
PSERVICE_PROPERTIES_CONTEXT context =
|
|
(PSERVICE_PROPERTIES_CONTEXT)GetProp(hwndDlg, PhMakeContextAtom());
|
|
|
|
switch (LOWORD(wParam))
|
|
{
|
|
case IDCANCEL:
|
|
{
|
|
// Workaround for property sheet + multiline edit: http://support.microsoft.com/kb/130765
|
|
|
|
SendMessage(GetParent(hwndDlg), uMsg, wParam, lParam);
|
|
}
|
|
break;
|
|
case IDC_PASSWORD:
|
|
{
|
|
if (HIWORD(wParam) == EN_CHANGE)
|
|
{
|
|
Button_SetCheck(GetDlgItem(hwndDlg, IDC_PASSWORDCHECK), BST_CHECKED);
|
|
}
|
|
}
|
|
break;
|
|
case IDC_DELAYEDSTART:
|
|
{
|
|
context->Dirty = TRUE;
|
|
}
|
|
break;
|
|
case IDC_BROWSE:
|
|
{
|
|
static PH_FILETYPE_FILTER filters[] =
|
|
{
|
|
{ L"Executable files (*.exe;*.sys)", L"*.exe;*.sys" },
|
|
{ L"All files (*.*)", L"*.*" }
|
|
};
|
|
PVOID fileDialog;
|
|
PPH_STRING fileName;
|
|
|
|
fileDialog = PhCreateOpenFileDialog();
|
|
PhSetFileDialogFilter(fileDialog, filters, sizeof(filters) / sizeof(PH_FILETYPE_FILTER));
|
|
|
|
fileName = PhGetFileName(PHA_GET_DLGITEM_TEXT(hwndDlg, IDC_BINARYPATH));
|
|
PhSetFileDialogFileName(fileDialog, fileName->Buffer);
|
|
PhDereferenceObject(fileName);
|
|
|
|
if (PhShowFileDialog(hwndDlg, fileDialog))
|
|
{
|
|
fileName = PhGetFileDialogFileName(fileDialog);
|
|
SetDlgItemText(hwndDlg, IDC_BINARYPATH, fileName->Buffer);
|
|
PhDereferenceObject(fileName);
|
|
}
|
|
|
|
PhFreeFileDialog(fileDialog);
|
|
}
|
|
break;
|
|
}
|
|
|
|
switch (HIWORD(wParam))
|
|
{
|
|
case EN_CHANGE:
|
|
case CBN_SELCHANGE:
|
|
{
|
|
PhpRefreshControls(hwndDlg);
|
|
|
|
if (context->Ready)
|
|
context->Dirty = TRUE;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
case WM_NOTIFY:
|
|
{
|
|
LPNMHDR header = (LPNMHDR)lParam;
|
|
|
|
switch (header->code)
|
|
{
|
|
case PSN_QUERYINITIALFOCUS:
|
|
{
|
|
SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, (LONG_PTR)GetDlgItem(hwndDlg, IDC_TYPE));
|
|
}
|
|
return TRUE;
|
|
case PSN_KILLACTIVE:
|
|
{
|
|
SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, FALSE);
|
|
}
|
|
return TRUE;
|
|
case PSN_APPLY:
|
|
{
|
|
NTSTATUS status;
|
|
PSERVICE_PROPERTIES_CONTEXT context =
|
|
(PSERVICE_PROPERTIES_CONTEXT)GetProp(hwndDlg, PhMakeContextAtom());
|
|
PPH_SERVICE_ITEM serviceItem = context->ServiceItem;
|
|
SC_HANDLE serviceHandle;
|
|
PPH_STRING newServiceTypeString;
|
|
PPH_STRING newServiceStartTypeString;
|
|
PPH_STRING newServiceErrorControlString;
|
|
ULONG newServiceType;
|
|
ULONG newServiceStartType;
|
|
ULONG newServiceErrorControl;
|
|
PPH_STRING newServiceGroup;
|
|
PPH_STRING newServiceBinaryPath;
|
|
PPH_STRING newServiceUserAccount;
|
|
PPH_STRING newServicePassword;
|
|
|
|
SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, PSNRET_NOERROR);
|
|
|
|
if (!context->Dirty)
|
|
{
|
|
return TRUE;
|
|
}
|
|
|
|
newServiceTypeString = PHA_DEREFERENCE(PhGetWindowText(GetDlgItem(hwndDlg, IDC_TYPE)));
|
|
newServiceStartTypeString = PHA_DEREFERENCE(PhGetWindowText(GetDlgItem(hwndDlg, IDC_STARTTYPE)));
|
|
newServiceErrorControlString = PHA_DEREFERENCE(PhGetWindowText(GetDlgItem(hwndDlg, IDC_ERRORCONTROL)));
|
|
newServiceType = PhGetServiceTypeInteger(newServiceTypeString->Buffer);
|
|
newServiceStartType = PhGetServiceStartTypeInteger(newServiceStartTypeString->Buffer);
|
|
newServiceErrorControl = PhGetServiceErrorControlInteger(newServiceErrorControlString->Buffer);
|
|
|
|
newServiceGroup = PHA_DEREFERENCE(PhGetWindowText(GetDlgItem(hwndDlg, IDC_GROUP)));
|
|
newServiceBinaryPath = PHA_DEREFERENCE(PhGetWindowText(GetDlgItem(hwndDlg, IDC_BINARYPATH)));
|
|
newServiceUserAccount = PHA_DEREFERENCE(PhGetWindowText(GetDlgItem(hwndDlg, IDC_USERACCOUNT)));
|
|
|
|
if (Button_GetCheck(GetDlgItem(hwndDlg, IDC_PASSWORDCHECK)) == BST_CHECKED)
|
|
{
|
|
newServicePassword = PhGetWindowText(GetDlgItem(hwndDlg, IDC_PASSWORD));
|
|
}
|
|
else
|
|
{
|
|
newServicePassword = NULL;
|
|
}
|
|
|
|
if (newServiceType == SERVICE_KERNEL_DRIVER && newServiceUserAccount->Length == 0)
|
|
{
|
|
newServiceUserAccount = NULL;
|
|
}
|
|
|
|
serviceHandle = PhOpenService(serviceItem->Name->Buffer, SERVICE_CHANGE_CONFIG);
|
|
|
|
if (serviceHandle)
|
|
{
|
|
if (ChangeServiceConfig(
|
|
serviceHandle,
|
|
newServiceType,
|
|
newServiceStartType,
|
|
newServiceErrorControl,
|
|
newServiceBinaryPath->Buffer,
|
|
newServiceGroup->Buffer,
|
|
NULL,
|
|
NULL,
|
|
PhGetString(newServiceUserAccount),
|
|
PhGetString(newServicePassword),
|
|
NULL
|
|
))
|
|
{
|
|
if (WindowsVersion >= WINDOWS_VISTA)
|
|
{
|
|
BOOLEAN newDelayedStart;
|
|
|
|
newDelayedStart = Button_GetCheck(GetDlgItem(hwndDlg, IDC_DELAYEDSTART)) == BST_CHECKED;
|
|
|
|
if (newDelayedStart != context->OldDelayedStart)
|
|
{
|
|
PhSetServiceDelayedAutoStart(serviceHandle, newDelayedStart);
|
|
}
|
|
}
|
|
|
|
PhMarkNeedsConfigUpdateServiceItem(serviceItem);
|
|
|
|
CloseServiceHandle(serviceHandle);
|
|
}
|
|
else
|
|
{
|
|
CloseServiceHandle(serviceHandle);
|
|
goto ErrorCase;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (GetLastError() == ERROR_ACCESS_DENIED && !PhElevated)
|
|
{
|
|
// Elevate using phsvc.
|
|
if (PhUiConnectToPhSvc(hwndDlg, FALSE))
|
|
{
|
|
if (NT_SUCCESS(status = PhSvcCallChangeServiceConfig(
|
|
serviceItem->Name->Buffer,
|
|
newServiceType,
|
|
newServiceStartType,
|
|
newServiceErrorControl,
|
|
newServiceBinaryPath->Buffer,
|
|
newServiceGroup->Buffer,
|
|
NULL,
|
|
NULL,
|
|
PhGetString(newServiceUserAccount),
|
|
PhGetString(newServicePassword),
|
|
NULL
|
|
)))
|
|
{
|
|
if (WindowsVersion >= WINDOWS_VISTA)
|
|
{
|
|
BOOLEAN newDelayedStart;
|
|
|
|
newDelayedStart = Button_GetCheck(GetDlgItem(hwndDlg, IDC_DELAYEDSTART)) == BST_CHECKED;
|
|
|
|
if (newDelayedStart != context->OldDelayedStart)
|
|
{
|
|
SERVICE_DELAYED_AUTO_START_INFO info;
|
|
|
|
info.fDelayedAutostart = newDelayedStart;
|
|
PhSvcCallChangeServiceConfig2(
|
|
serviceItem->Name->Buffer,
|
|
SERVICE_CONFIG_DELAYED_AUTO_START_INFO,
|
|
&info
|
|
);
|
|
}
|
|
}
|
|
|
|
PhMarkNeedsConfigUpdateServiceItem(serviceItem);
|
|
}
|
|
|
|
PhUiDisconnectFromPhSvc();
|
|
|
|
if (!NT_SUCCESS(status))
|
|
{
|
|
SetLastError(PhNtStatusToDosError(status));
|
|
goto ErrorCase;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// User cancelled elevation.
|
|
SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, PSNRET_INVALID);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
goto ErrorCase;
|
|
}
|
|
}
|
|
|
|
goto Cleanup;
|
|
ErrorCase:
|
|
if (PhShowMessage(
|
|
hwndDlg,
|
|
MB_ICONERROR | MB_RETRYCANCEL,
|
|
L"Unable to change service configuration: %s",
|
|
((PPH_STRING)PHA_DEREFERENCE(PhGetWin32Message(GetLastError())))->Buffer
|
|
) == IDRETRY)
|
|
{
|
|
SetWindowLongPtr(hwndDlg, DWLP_MSGRESULT, PSNRET_INVALID);
|
|
}
|
|
|
|
Cleanup:
|
|
if (newServicePassword)
|
|
{
|
|
RtlSecureZeroMemory(newServicePassword->Buffer, newServicePassword->Length);
|
|
PhDereferenceObject(newServicePassword);
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|