171 lines
5 KiB
C++
171 lines
5 KiB
C++
|
//----------------------------------------------------------------------------
|
||
|
//
|
||
|
// General utility routines.
|
||
|
//
|
||
|
// Copyright (C) Microsoft Corporation, 2000-2001.
|
||
|
//
|
||
|
//----------------------------------------------------------------------------
|
||
|
|
||
|
#ifndef __CMNUTIL_HPP__
|
||
|
#define __CMNUTIL_HPP__
|
||
|
|
||
|
#define DIMAT(Array, EltType) (sizeof(Array) / sizeof(EltType))
|
||
|
#define DIMA(Array) DIMAT(Array, (Array)[0])
|
||
|
|
||
|
#define CSRSS_PROCESS_ID ((ULONG)-1)
|
||
|
|
||
|
// Converts an NTSTATUS into an HRESULT with success check.
|
||
|
#define CONV_NT_STATUS(Status) \
|
||
|
(NT_SUCCESS(Status) ? S_OK : HRESULT_FROM_NT(Status))
|
||
|
|
||
|
// Converts a Win32 status into an HRESULT with a guarantee it's
|
||
|
// an error code. This avoids problems with routines which
|
||
|
// don't set a last error.
|
||
|
#define WIN32_STATUS(Err) ((Err) == 0 ? E_FAIL : HRESULT_FROM_WIN32(Err))
|
||
|
#define WIN32_LAST_STATUS() WIN32_STATUS(GetLastError())
|
||
|
|
||
|
// Converts a BOOL into an HRESULT with success check.
|
||
|
#define CONV_W32_STATUS(Status) \
|
||
|
((Status) ? S_OK : WIN32_LAST_STATUS())
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
//
|
||
|
// Assertions.
|
||
|
//
|
||
|
//----------------------------------------------------------------------------
|
||
|
|
||
|
#if DBG
|
||
|
|
||
|
void DbgAssertionFailed(PCSTR File, int Line, PCSTR Str);
|
||
|
|
||
|
#define DBG_ASSERT(Expr) \
|
||
|
if (!(Expr)) \
|
||
|
{ \
|
||
|
DbgAssertionFailed(__FILE__, __LINE__, #Expr); \
|
||
|
} \
|
||
|
else 0
|
||
|
|
||
|
#else
|
||
|
|
||
|
#define DBG_ASSERT(Expr)
|
||
|
|
||
|
#endif
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
//
|
||
|
// COM help.
|
||
|
//
|
||
|
//----------------------------------------------------------------------------
|
||
|
|
||
|
// Wrapper that can be locally implemented if necessary to
|
||
|
// remove usage of ole32.dll on platforms where IsEqualIID
|
||
|
// is not inlined.
|
||
|
#define DbgIsEqualIID(Id1, Id2) \
|
||
|
IsEqualIID(Id1, Id2)
|
||
|
|
||
|
// Safely releases and NULLs an interface pointer.
|
||
|
#define RELEASE(pUnk) \
|
||
|
((pUnk) != NULL ? ((pUnk)->Release(), (pUnk) = NULL) : NULL)
|
||
|
|
||
|
// Transfers an interface pointer into a holder that may or
|
||
|
// may not already hold and interface.
|
||
|
#define TRANSFER(pOld, pNew) \
|
||
|
(((pNew) != NULL ? (pNew)->AddRef() : 0), \
|
||
|
((pOld) != NULL ? (pOld)->Release() : 0), \
|
||
|
(pOld) = (pNew))
|
||
|
|
||
|
//----------------------------------------------------------------------------
|
||
|
//
|
||
|
// Utility functions.
|
||
|
//
|
||
|
//----------------------------------------------------------------------------
|
||
|
|
||
|
extern PSECURITY_DESCRIPTOR g_AllAccessSecDesc;
|
||
|
extern SECURITY_ATTRIBUTES g_AllAccessSecAttr;
|
||
|
|
||
|
PSTR FormatStatusCode(HRESULT Status);
|
||
|
|
||
|
#define FormatStatus(Status) FormatStatusArgs(Status, NULL)
|
||
|
PSTR FormatStatusArgs(HRESULT Status, PVOID Arguments);
|
||
|
|
||
|
BOOL InstallAsAeDebug(PCSTR Append);
|
||
|
|
||
|
HANDLE CreatePidEvent(ULONG Pid, ULONG CreateOrOpen);
|
||
|
BOOL SetPidEvent(ULONG Pid, ULONG CreateOrOpen);
|
||
|
|
||
|
HRESULT EnableDebugPrivilege(void);
|
||
|
|
||
|
HRESULT FillDataBuffer(PVOID Data, ULONG DataLen,
|
||
|
PVOID Buffer, ULONG BufferLen, PULONG BufferUsed);
|
||
|
HRESULT FillStringBuffer(PCSTR String, ULONG StringLenIn,
|
||
|
PSTR Buffer, ULONG BufferLen, PULONG StringLenOut);
|
||
|
HRESULT AppendToStringBuffer(HRESULT Status, PCSTR String, BOOL First,
|
||
|
PSTR* Buffer, ULONG* BufferLen, PULONG LenOut);
|
||
|
|
||
|
void Win32ToNtTimeout(ULONG Win32Timeout, PLARGE_INTEGER NtTimeout);
|
||
|
|
||
|
HRESULT InitializeAllAccessSecObj(void);
|
||
|
void DeleteAllAccessSecObj(void);
|
||
|
|
||
|
#define BASE_YEAR_ADJUSTMENT 11644473600
|
||
|
|
||
|
// Convert to seconds and then from base year 1601 to base year 1970.
|
||
|
#define FileTimeToTimeDateStamp(FileTime) \
|
||
|
(ULONG)(((FileTime) / 10000000) - BASE_YEAR_ADJUSTMENT)
|
||
|
|
||
|
// Adjust date back to 1601 from 1970 and convert to 100 nanoseconds
|
||
|
#define TimeDateStampToFileTime(TimeDate) \
|
||
|
(((ULONG64)(TimeDate) + BASE_YEAR_ADJUSTMENT) * 10000000)
|
||
|
|
||
|
// Convert to seconds
|
||
|
#define FileTimeToTime(FileTime) \
|
||
|
(ULONG)((FileTime) / 10000000)
|
||
|
|
||
|
// Convert to seconds .
|
||
|
#define TimeToFileTime(TimeDate) \
|
||
|
((ULONG64)(TimeDate) * 10000000)
|
||
|
|
||
|
HRESULT QueryVersionDataBuffer(PVOID VerData, PCSTR Item,
|
||
|
PVOID Buffer, ULONG BufferSize,
|
||
|
PULONG DataSize);
|
||
|
PVOID GetAllFileVersionInfo(PSTR VerFile);
|
||
|
BOOL GetFileStringFileInfo(PSTR VerFile, PCSTR SubItem,
|
||
|
PSTR Buffer, ULONG BufferSize);
|
||
|
|
||
|
HRESULT ExpandDumpCab(PCSTR CabFile, ULONG FileFlags,
|
||
|
PSTR DmpFile, INT_PTR* DmpFh);
|
||
|
|
||
|
enum FILE_IO_TYPE
|
||
|
{
|
||
|
FIO_WIN32,
|
||
|
FIO_WININET,
|
||
|
};
|
||
|
|
||
|
class PathFile
|
||
|
{
|
||
|
public:
|
||
|
PathFile(FILE_IO_TYPE IoType)
|
||
|
{
|
||
|
m_IoType = IoType;
|
||
|
}
|
||
|
virtual ~PathFile(void);
|
||
|
|
||
|
virtual HRESULT Open(PCSTR Path, ULONG SymOpt) = 0;
|
||
|
virtual HRESULT QueryDataAvailable(PULONG Avail) = 0;
|
||
|
virtual HRESULT GetLastWriteTime(PFILETIME Time) = 0;
|
||
|
virtual HRESULT Read(PVOID Buffer, ULONG BufferLen, PULONG Done) = 0;
|
||
|
|
||
|
FILE_IO_TYPE m_IoType;
|
||
|
};
|
||
|
|
||
|
BOOL IsUrlPathComponent(PCSTR Path);
|
||
|
|
||
|
#ifndef NT_NATIVE
|
||
|
|
||
|
BOOL PathFileExists(PCSTR Path, ULONG SymOpt, FILE_IO_TYPE* IoType);
|
||
|
HRESULT OpenPathFile(PCSTR Path, ULONG SymOpt, PathFile** File);
|
||
|
|
||
|
#endif
|
||
|
|
||
|
#endif // #ifndef __CMNUTIL_HPP__
|