implement the object types SE_REGISTRY_KEY, SE_FILE_OBJECT, SE_SERVICE and SE_WINDOW_OBJECT for GetSecurityInfo()

svn path=/trunk/; revision=20968
This commit is contained in:
Thomas Bluemel 2006-01-22 03:27:21 +00:00
parent 0b0554b8ee
commit 68c3c9cf63
2 changed files with 72 additions and 47 deletions

View file

@ -49,7 +49,7 @@ AccRewriteGetHandleRights(HANDLE handle,
PSECURITY_DESCRIPTOR* ppSecurityDescriptor)
{
PSECURITY_DESCRIPTOR pSD = NULL;
ULONG RequiredSize, SDSize = 0;
ULONG SDSize = 0;
NTSTATUS Status;
DWORD LastErr;
DWORD Ret = ERROR_SUCCESS;
@ -57,59 +57,83 @@ AccRewriteGetHandleRights(HANDLE handle,
/* save the last error code */
LastErr = GetLastError();
AllocBuffer:
/* allocate a buffer large enough to hold the
security descriptor we need to return */
SDSize += 0x100;
if (pSD != NULL)
do
{
pSD = LocalAlloc(LMEM_FIXED,
(SIZE_T)SDSize);
}
else
{
pSD = LocalReAlloc((HLOCAL)pSD,
(SIZE_T)SDSize,
LMEM_MOVEABLE);
}
if (pSD == NULL)
{
Ret = GetLastError();
goto Cleanup;
}
/* perform the actual query depending on the object type */
switch (ObjectType)
{
case SE_KERNEL_OBJECT:
/* allocate a buffer large enough to hold the
security descriptor we need to return */
SDSize += 0x100;
if (pSD != NULL)
{
Status = NtQuerySecurityObject(handle,
SecurityInfo,
pSD,
SDSize,
&RequiredSize);
if (Status == STATUS_BUFFER_TOO_SMALL)
pSD = LocalAlloc(LMEM_FIXED,
(SIZE_T)SDSize);
}
else
{
pSD = LocalReAlloc((HLOCAL)pSD,
(SIZE_T)SDSize,
LMEM_MOVEABLE);
}
if (pSD == NULL)
{
Ret = GetLastError();
break;
}
/* perform the actual query depending on the object type */
switch (ObjectType)
{
case SE_REGISTRY_KEY:
{
/* not enough memory, increase the size of
the buffer and try again */
ASSERT(RequiredSize > SDSize);
SDSize = RequiredSize;
goto AllocBuffer;
Ret = RegGetKeySecurity((HKEY)handle,
SecurityInfo,
pSD,
&SDSize);
break;
}
case SE_FILE_OBJECT:
case SE_KERNEL_OBJECT:
{
Status = NtQuerySecurityObject(handle,
SecurityInfo,
pSD,
SDSize,
&SDSize);
Ret = RtlNtStatusToDosError(Status);
break;
}
case SE_SERVICE:
{
Ret = QueryServiceObjectSecurity((SC_HANDLE)handle,
SecurityInfo,
pSD,
SDSize,
&SDSize);
break;
}
case SE_WINDOW_OBJECT:
{
Ret = GetUserObjectSecurity(handle,
&SecurityInfo,
pSD,
SDSize,
&SDSize);
break;
}
default:
{
UNIMPLEMENTED;
Ret = ERROR_CALL_NOT_IMPLEMENTED;
break;
}
Ret = RtlNtStatusToDosError(Status);
break;
}
default:
{
UNIMPLEMENTED;
Ret = ERROR_CALL_NOT_IMPLEMENTED;
break;
}
}
} while (Ret == ERROR_INSUFFICIENT_BUFFER);
Cleanup:
if (Ret == ERROR_SUCCESS)
{
*ppSecurityDescriptor = pSD;

View file

@ -3,6 +3,7 @@
#include <windows.h>
#include <ndk/ntndk.h>
#include <accctrl.h>
#include <winsvc.h>
#ifndef HAS_FN_PROGRESSW
#define FN_PROGRESSW FN_PROGRESS