Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-07-04 17:01:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

only allocate a buffer in StartServiceA when dwNumServiceArgs > 0

Browse source 
adjust the image path in services/rpcserver

svn path=/trunk/; revision=31761
This commit is contained in:
Christoph von Wittich 2008-01-13 14:22:55 +00:00
parent 16f2303c0b
commit ea6661ca60
2 changed files with 35 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

13
reactos/base/system/services/rpcserver.c
View file

@ -1058,18 +1058,25 @@ ScmrCreateServiceW(handle_t BindingHandle,
if (dwServiceType & SERVICE_DRIVER) if (dwServiceType & SERVICE_DRIVER)
{ {
/* FIXME: Adjust the image path */
lpImagePath = (WCHAR*) HeapAlloc(GetProcessHeap(), lpImagePath = (WCHAR*) HeapAlloc(GetProcessHeap(),
HEAP_ZERO_MEMORY, HEAP_ZERO_MEMORY,
(wcslen(lpBinaryPathName) + 1) * sizeof(WCHAR)); (wcslen(lpBinaryPathName) + 5) * sizeof(WCHAR));
if (lpImagePath == NULL) if (lpImagePath == NULL)
{ {
dwError = ERROR_NOT_ENOUGH_MEMORY; dwError = ERROR_NOT_ENOUGH_MEMORY;
goto done; goto done;
} }
if (lpBinaryPathName[1] == L':')
{
wcscpy(lpImagePath, L"\\??\\");
wcscat(lpImagePath, lpBinaryPathName);
}
else
{
wcscpy(lpImagePath, lpBinaryPathName); wcscpy(lpImagePath, lpBinaryPathName);
} }
}
/* Allocate a new service entry */ /* Allocate a new service entry */
dwError = ScmCreateNewServiceRecord(lpServiceName, dwError = ScmCreateNewServiceRecord(lpServiceName,

9
reactos/dll/win32/advapi32/service/scm.c
View file

@ -1998,13 +1998,14 @@ StartServiceA(SC_HANDLE hService,
DWORD dwNumServiceArgs, DWORD dwNumServiceArgs,
LPCSTR *lpServiceArgVectors) LPCSTR *lpServiceArgVectors)
{ {
LPSTR lpBuffer; LPSTR lpBuffer = NULL;
LPSTR lpStr; LPSTR lpStr;
DWORD dwError; DWORD dwError;
DWORD dwBufSize; DWORD dwBufSize = 0;
DWORD i; DWORD i;
dwBufSize = 0; if (dwNumServiceArgs > 0)
{
for (i = 0; i < dwNumServiceArgs; i++) for (i = 0; i < dwNumServiceArgs; i++)
{ {
dwBufSize += (strlen(lpServiceArgVectors[i]) + 1); dwBufSize += (strlen(lpServiceArgVectors[i]) + 1);
@ -2026,6 +2027,7 @@ StartServiceA(SC_HANDLE hService,
lpStr += (strlen(lpServiceArgVectors[i]) + 1); lpStr += (strlen(lpServiceArgVectors[i]) + 1);
} }
*lpStr = 0; *lpStr = 0;
}
dwError = ScmrStartServiceA(BindingHandle, dwError = ScmrStartServiceA(BindingHandle,
(unsigned int)hService, (unsigned int)hService,
@ -2033,6 +2035,7 @@ StartServiceA(SC_HANDLE hService,
(unsigned char *)lpBuffer, (unsigned char *)lpBuffer,
dwBufSize); dwBufSize);
if (lpBuffer != NULL)
HeapFree(GetProcessHeap(), 0, lpBuffer); HeapFree(GetProcessHeap(), 0, lpBuffer);
if (dwError != ERROR_SUCCESS) if (dwError != ERROR_SUCCESS)