able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/ds/netapi/svcdlls/logonsrv/server/nlp.c

2065 lines
47 KiB
C
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1987-1996 Microsoft Corporation
Module Name:
nlp.c
Abstract:
Private Netlogon service utility routines.
Author:
Cliff Van Dyke (cliffv) 7-Jun-1991
Environment:
User mode only.
Contains NT-specific code.
Requires ANSI C extensions: slash-slash comments, long external names.
Revision History:
08-May-1992 JohnRo
Use net config helpers for NetLogon.
Use <prefix.h> equates.
--*/
//
// Common include files.
//
#include "logonsrv.h" // Include files common to entire service
#pragma hdrstop
// Include this again to declare the globals
#define DEBUG_ALLOCATE
#include <nldebug.h> // Netlogon debugging
#undef DEBUG_ALLOCATE
//
// Include files specific to this .c file
//
#include <winerror.h> // NO_ERROR
#include <prefix.h> // PREFIX_ equates.
#include <stdarg.h> // va_list, etc.
#include <stdio.h> // vsprintf().
#include <tstr.h> // TCHAR_ equates.
LPWSTR
NlStringToLpwstr(
IN PUNICODE_STRING String
)
/*++
Routine Description:
Convert a Unicode String to an LPWSTR.
Arguments:
String - Unicode String to copy
Return Value:
LPWSTR in a NetpMemoryAllocate'd buffer.
NULL is returned if there is no memory.
--*/
{
LPWSTR Buffer;
Buffer = NetpMemoryAllocate( String->Length + sizeof(WCHAR) );
if ( Buffer != NULL ) {
RtlCopyMemory( Buffer, String->Buffer, String->Length );
Buffer[ String->Length / sizeof(WCHAR) ] = L'\0';
}
return Buffer;
}
BOOLEAN
NlAllocStringFromWStr(
IN LPWSTR InString,
OUT PUNICODE_STRING OutString
)
/*++
Routine Description:
Convert a zero terminated string into an allocated UNICODE_STRING structure.
Arguments:
InString - String to copy
OutString - String to copy into.
OutString->Buffer should be freed using MIDL_user_free
Return Value:
TRUE - success
FALSE - couldn't allocate memory
--*/
{
if ( InString == NULL ) {
OutString->Length = 0;
OutString->MaximumLength = 0;
OutString->Buffer = NULL;
} else {
OutString->Length = wcslen(InString) * sizeof(WCHAR);
OutString->MaximumLength = OutString->Length + sizeof(WCHAR);
OutString->Buffer = MIDL_user_allocate( OutString->MaximumLength );
if ( OutString->Buffer == NULL ) {
return FALSE;
}
RtlCopyMemory( OutString->Buffer,
InString,
OutString->MaximumLength );
}
return TRUE;
}
BOOLEAN
NlDuplicateUnicodeString(
IN PUNICODE_STRING InString OPTIONAL,
OUT PUNICODE_STRING OutString
)
/*++
Routine Description:
Convert a UNICODE_STRING string into an allocated UNICODE_STRING structure.
Arguments:
InString - String to copy
OutString - String to copy into.
OutString should be freed using NlFreeUnicodeString
Return Value:
TRUE - success
FALSE - couldn't allocate memory
--*/
{
if ( InString == NULL || InString->Length == 0 ) {
OutString->Length = 0;
OutString->MaximumLength = 0;
OutString->Buffer = NULL;
} else {
OutString->Length = InString->Length;
OutString->MaximumLength = OutString->Length + sizeof(WCHAR);
OutString->Buffer = MIDL_user_allocate( OutString->MaximumLength );
if ( OutString->Buffer == NULL ) {
return FALSE;
}
RtlCopyMemory( OutString->Buffer,
InString->Buffer,
OutString->Length );
OutString->Buffer[OutString->Length/sizeof(WCHAR)] = L'\0';
}
return TRUE;
}
VOID
NlFreeUnicodeString(
IN PUNICODE_STRING InString OPTIONAL
)
/*++
Routine Description:
Free the UNICODE_STRING structure allocated by NlDuplicateUnicodeString.
Arguments:
InString - String to free
Return Value:
None.
--*/
{
if ( InString != NULL ) {
if ( InString->Buffer != NULL ) {
MIDL_user_free( InString->Buffer );
}
InString->Length = 0;
InString->MaximumLength = 0;
InString->Buffer = NULL;
}
}
LPSTR
NlStringToLpstr(
IN PUNICODE_STRING String
)
/*++
Routine Description:
Convert a Unicode String to an LPSTR.
Arguments:
String - Unicode String to copy
Return Value:
LPWSTR in a NetpMemoryAllocate'd buffer.
NULL is returned if there is no memory.
--*/
{
NTSTATUS Status;
STRING OemString;
OemString.MaximumLength = (USHORT) RtlUnicodeStringToOemSize( String );
OemString.Buffer = NetpMemoryAllocate( OemString.MaximumLength );
if ( OemString.Buffer != NULL ) {
Status = RtlUnicodeStringToOemString( &OemString,
String,
(BOOLEAN) FALSE );
if ( !NT_SUCCESS( Status ) ) {
NetpMemoryFree( OemString.Buffer );
return NULL;
}
}
return OemString.Buffer;
}
VOID
NlpPutString(
IN PUNICODE_STRING OutString,
IN PUNICODE_STRING InString,
IN PUCHAR *Where
)
/*++
Routine Description:
This routine copies the InString string to the memory pointed to by
the Where parameter, and fixes the OutString string to point to that
new copy.
Parameters:
OutString - A pointer to a destination NT string
InString - A pointer to an NT string to be copied
Where - A pointer to space to put the actual string for the
OutString. The pointer is adjusted to point to the first byte
following the copied string.
Return Values:
None.
--*/
{
NlAssert( OutString != NULL );
NlAssert( InString != NULL );
NlAssert( Where != NULL && *Where != NULL);
NlAssert( *Where == ROUND_UP_POINTER( *Where, sizeof(WCHAR) ) );
#ifdef notdef
KdPrint(("NlpPutString: %ld %Z\n", InString->Length, InString ));
KdPrint((" InString: %lx %lx OutString: %lx Where: %lx\n", InString,
InString->Buffer, OutString, *Where ));
#endif
if ( InString->Length > 0 ) {
OutString->Buffer = (PWCH) *Where;
OutString->MaximumLength = (USHORT)(InString->Length + sizeof(WCHAR));
RtlCopyUnicodeString( OutString, InString );
*Where += InString->Length;
// *((WCHAR *)(*Where)) = L'\0';
*(*Where) = '\0';
*(*Where + 1) = '\0';
*Where += 2;
} else {
RtlInitUnicodeString(OutString, NULL);
}
#ifdef notdef
KdPrint((" OutString: %ld %lx\n", OutString->Length, OutString->Buffer));
#endif
return;
}
VOID
NlpWriteEventlog (
IN DWORD EventId,
IN DWORD EventType,
IN LPBYTE RawDataBuffer OPTIONAL,
IN DWORD RawDataSize,
IN LPWSTR *StringArray,
IN DWORD StringCount
)
/*++
Routine Description:
Stub routine for calling Event Log.
Arguments:
EventId - event log ID.
EventType - Type of event.
RawDataBuffer - Data to be logged with the error.
numbyte - Size in bytes of "RawDataBuffer"
StringArray - array of null-terminated strings.
StringCount - number of zero terminated strings in "StringArray". The following
flags can be OR'd in to the count:
LAST_MESSAGE_IS_NTSTATUS
LAST_MESSAGE_IS_NETSTATUS
ALLOW_DUPLICATE_EVENTS
RAISE_ALERT_TOO
Return Value:
TRUE: The message was written.
--*/
{
DWORD ErrorCode;
DWORD ActualStringCount = StringCount & NETP_STRING_COUNT_MASK;
//
// If an NT status code was passed in,
// convert it to a net status code.
//
if ( StringCount & NETP_LAST_MESSAGE_IS_NTSTATUS ) {
//
// Do the "better" error mapping when eventviewer ParameterMessageFile
// can be a list of files. Then, add netmsg.dll to the list.
//
// StringArray[ActualStringCount-1] = (LPWSTR) NetpNtStatusToApiStatus( (NTSTATUS) StringArray[ActualStringCount-1] );
if ( (NTSTATUS)(ULONG_PTR)StringArray[ActualStringCount-1] == STATUS_SYNCHRONIZATION_REQUIRED ) {
StringArray[ActualStringCount-1] = (LPWSTR) NERR_SyncRequired;
StringCount &= ~NETP_LAST_MESSAGE_IS_NTSTATUS;
StringCount |= NETP_LAST_MESSAGE_IS_NETSTATUS;
}
}
//
// Dump the event to the debug file.
//
#if NETLOGONDBG
IF_NL_DEBUG( MISC ) {
DWORD i;
NlPrint((NL_MISC, "Eventlog: %ld (%ld) ",
EventId,
EventType ));
for (i = 0; i < ActualStringCount ; i++ ) {
if ( i == ActualStringCount-1) {
if ( StringCount & NETP_LAST_MESSAGE_IS_NTSTATUS ) {
NlPrint((NL_MISC, "0x%lx ", StringArray[i] ));
} else if ( StringCount & NETP_LAST_MESSAGE_IS_NETSTATUS ) {
NlPrint((NL_MISC, "%ld ", StringArray[i] ));
} else {
NlPrint((NL_MISC, "\"%ws\" ", StringArray[i] ));
}
} else {
NlPrint((NL_MISC, "\"%ws\" ", StringArray[i] ));
}
}
if( RawDataSize ) {
if ( RawDataSize > 16 ) {
NlPrint((NL_MISC, "\n" ));
}
NlpDumpBuffer( NL_MISC, RawDataBuffer, RawDataSize );
} else {
NlPrint((NL_MISC, "\n" ));
}
}
#endif // NETLOGONDBG
//
// Write the event and avoid duplicates
//
ErrorCode = NetpEventlogWrite (
NlGlobalEventlogHandle,
EventId,
EventType,
RawDataBuffer,
RawDataSize,
StringArray,
StringCount );
if( ErrorCode != NO_ERROR ) {
if ( ErrorCode == ERROR_ALREADY_EXISTS ) {
NlPrint((NL_MISC,
"Didn't log event since it was already logged.\n" ));
} else {
NlPrint((NL_CRITICAL,
"Error writing this event in the eventlog, Status = %ld\n",
ErrorCode ));
}
goto Cleanup;
}
Cleanup:
return;
}
#if NETLOGONDBG
VOID
NlpDumpBuffer(
IN DWORD DebugFlag,
PVOID Buffer,
DWORD BufferSize
)
/*++
Routine Description:
Dumps the buffer content on to the debugger output.
Arguments:
DebugFlag: Debug flag to pass on to NlPrintRoutine
Buffer: buffer pointer.
BufferSize: size of the buffer.
Return Value:
none
--*/
{
#define NUM_CHARS 16
DWORD i, limit;
CHAR TextBuffer[NUM_CHARS + 1];
LPBYTE BufferPtr = Buffer;
BOOLEAN DumpDwords = FALSE;
//
// If we aren't debugging this functionality, just return.
//
if ( (NlGlobalParameters.DbFlag & DebugFlag) == 0 ) {
return;
}
//
// Don't want to intermingle output from different threads.
//
EnterCriticalSection( &NlGlobalLogFileCritSect );
if ( BufferSize > NUM_CHARS ) {
NlPrint((0,"\n")); // Ensure this starts on a new line
NlPrint((0,"------------------------------------\n"));
} else {
if ( BufferSize % sizeof(DWORD) == 0 ) {
DumpDwords = TRUE;
}
}
//
// Hex dump of the bytes
//
limit = ((BufferSize - 1) / NUM_CHARS + 1) * NUM_CHARS;
for (i = 0; i < limit; i++) {
if (i < BufferSize) {
if ( DumpDwords ) {
if ( i % sizeof(DWORD) == 0 ) {
DWORD ADword;
RtlCopyMemory( &ADword, &BufferPtr[i], sizeof(DWORD) );
NlPrint((0,"%08x ", ADword));
}
} else {
NlPrint((0,"%02x ", BufferPtr[i]));
}
if ( isprint(BufferPtr[i]) ) {
TextBuffer[i % NUM_CHARS] = (CHAR) BufferPtr[i];
} else {
TextBuffer[i % NUM_CHARS] = '.';
}
} else {
if ( DumpDwords ) {
TextBuffer[i % NUM_CHARS] = '\0';
} else {
if ( BufferSize > NUM_CHARS ) {
NlPrint((0," "));
TextBuffer[i % NUM_CHARS] = ' ';
} else {
TextBuffer[i % NUM_CHARS] = '\0';
}
}
}
if ((i + 1) % NUM_CHARS == 0) {
TextBuffer[NUM_CHARS] = 0;
NlPrint((0," %s\n", TextBuffer));
}
}
if ( BufferSize > NUM_CHARS ) {
NlPrint((0,"------------------------------------\n"));
} else if ( BufferSize < NUM_CHARS ) {
NlPrint((0,"\n"));
}
LeaveCriticalSection( &NlGlobalLogFileCritSect );
}
VOID
NlpDumpGuid(
IN DWORD DebugFlag,
IN GUID *Guid OPTIONAL
)
/*++
Routine Description:
Dumps a GUID to the debugger output.
Arguments:
DebugFlag: Debug flag to pass on to NlPrintRoutine
Guid: Guid to print
Return Value:
none
--*/
{
RPC_STATUS RpcStatus;
char *StringGuid;
//
// If we aren't debugging this functionality, just return.
//
if ( (NlGlobalParameters.DbFlag & DebugFlag) == 0 ) {
return;
}
if ( Guid == NULL ) {
NlPrint(( DebugFlag, "(null)" ));
} else {
RpcStatus = UuidToStringA( Guid, &StringGuid );
if ( RpcStatus != RPC_S_OK ) {
return;
}
NlPrint(( DebugFlag, "%s", StringGuid ));
RpcStringFreeA( &StringGuid );
}
}
VOID
NlpDumpPeriod(
IN DWORD DebugFlag,
IN LPSTR Comment,
IN ULONG Period
)
/*++
Routine Description:
Print an elapsed time (in milliseconds)
Arguments:
DebugFlag - Debug flag to pass on to NlPrintRoutine
Comment - Comment to print in front of the time
Period - Time period (in milliseconds)
Return Value:
None
--*/
{
//
// Convert the period to something easily readable
//
if ( Period == MAILSLOT_WAIT_FOREVER ) {
NlPrint(( DebugFlag,
"%s 'never' (0x%lx)\n",
Comment,
Period ));
} else if ( (Period / NL_MILLISECONDS_PER_DAY) != 0 ) {
NlPrint(( DebugFlag,
"%s %ld days (0x%lx)\n",
Comment,
Period / NL_MILLISECONDS_PER_DAY,
Period ));
} else if ( (Period / NL_MILLISECONDS_PER_HOUR) != 0 ) {
NlPrint(( DebugFlag,
"%s %ld hours (0x%lx)\n",
Comment,
Period / NL_MILLISECONDS_PER_HOUR,
Period ));
} else if ( (Period / NL_MILLISECONDS_PER_MINUTE) != 0 ) {
NlPrint(( DebugFlag,
"%s %ld minutes (0x%lx)\n",
Comment,
Period / NL_MILLISECONDS_PER_MINUTE,
Period ));
} else {
NlPrint(( DebugFlag,
"%s %ld seconds (0x%lx)\n",
Comment,
Period / NL_MILLISECONDS_PER_SECOND,
Period ));
}
}
VOID
NlpDumpTime(
IN DWORD DebugFlag,
IN LPSTR Comment,
IN LARGE_INTEGER ConvertTime
)
/*++
Routine Description:
Print the specified time
Arguments:
DebugFlag - Debug flag to pass on to NlPrintRoutine
Comment - Comment to print in front of the time
Time - GMT time to print (Nothing is printed if this is zero)
Return Value:
None
--*/
{
//
// If we aren't debugging this functionality, just return.
//
if ( (NlGlobalParameters.DbFlag & DebugFlag) == 0 ) {
return;
}
//
// Convert an NT GMT time to ascii,
// Do so
//
if ( ConvertTime.QuadPart != 0 ) {
LARGE_INTEGER LocalTime;
TIME_FIELDS TimeFields;
NTSTATUS Status;
Status = RtlSystemTimeToLocalTime( &ConvertTime, &LocalTime );
if ( !NT_SUCCESS( Status )) {
NlPrint(( NL_CRITICAL, "Can't convert time from GMT to Local time\n" ));
LocalTime = ConvertTime;
}
RtlTimeToTimeFields( &LocalTime, &TimeFields );
NlPrint(( DebugFlag, "%s%8.8lx %8.8lx = %ld/%ld/%ld %ld:%2.2ld:%2.2ld\n",
Comment,
ConvertTime.LowPart,
ConvertTime.HighPart,
TimeFields.Month,
TimeFields.Day,
TimeFields.Year,
TimeFields.Hour,
TimeFields.Minute,
TimeFields.Second ));
}
}
VOID
NlpDumpSid(
IN DWORD DebugFlag,
IN PSID Sid OPTIONAL
)
/*++
Routine Description:
Dumps a SID to the debugger output
Arguments:
DebugFlag - Debug flag to pass on to NlPrintRoutine
Sid - SID to output
Return Value:
none
--*/
{
//
// If we aren't debugging this functionality, just return.
//
if ( (NlGlobalParameters.DbFlag & DebugFlag) == 0 ) {
return;
}
//
// Output the SID
//
if ( Sid == NULL ) {
NlPrint((0, "(null)\n"));
} else {
UNICODE_STRING SidString;
NTSTATUS Status;
Status = RtlConvertSidToUnicodeString( &SidString, Sid, TRUE );
if ( !NT_SUCCESS(Status) ) {
NlPrint((0, "Invalid 0x%lX\n", Status ));
} else {
NlPrint((0, "%wZ\n", &SidString ));
RtlFreeUnicodeString( &SidString );
}
}
}
#endif // NETLOGONDBG
DWORD
NlpAtoX(
IN LPWSTR String
)
/*++
Routine Description:
Converts hexadecimal string to DWORD integer.
Accepts the following form of hex string
0[x-X][0-9, a-f, A-F]*
Arguments:
String: hexadecimal string.
Return Value:
Decimal value of the hex string.
--*/
{
DWORD Value = 0;
if( String == NULL )
return 0;
if( *String != TEXT('0') )
return 0;
String++;
if( *String == TCHAR_EOS )
return 0;
if( ( *String != TEXT('x') ) && ( *String != TEXT('X') ) )
return 0;
String++;
while(*String != TCHAR_EOS ) {
if( (*String >= TEXT('0')) && (*String <= TEXT('9')) ) {
Value = Value * 16 + ( *String - '0');
} else if( (*String >= TEXT('a')) && (*String <= TEXT('f')) ) {
Value = Value * 16 + ( 10 + *String - TEXT('a'));
} else if( (*String >= TEXT('A')) && (*String <= TEXT('F')) ) {
Value = Value * 16 + ( 10 + *String - TEXT('A'));
} else {
break;
}
String++;
}
return Value;
}
VOID
NlWaitForSingleObject(
IN LPSTR WaitReason,
IN HANDLE WaitHandle
)
/*++
Routine Description:
Waits an infinite amount of time for the specified handle.
Arguments:
WaitReason - Text describing what we're waiting on
WaitHandle - Handle to wait on
Return Value:
None
--*/
{
DWORD WaitStatus;
//
// Loop waiting.
//
for (;;) {
WaitStatus = WaitForSingleObject( WaitHandle,
2*60*1000 ); // Two minutes
if ( WaitStatus == WAIT_TIMEOUT ) {
NlPrint((NL_CRITICAL,
"WaitForSingleObject 2-minute timeout (Rewaiting): %s\n",
WaitReason ));
continue;
} else if ( WaitStatus == WAIT_OBJECT_0 ) {
break;
} else {
NlPrint((NL_CRITICAL,
"WaitForSingleObject error: %ld %ld %s\n",
WaitStatus,
GetLastError(),
WaitReason ));
UNREFERENCED_PARAMETER(WaitReason);
break;
}
}
}
BOOLEAN
NlWaitForSamService(
BOOLEAN NetlogonServiceCalling
)
/*++
Routine Description:
This procedure waits for the SAM service to start and to complete
all its initialization.
Arguments:
NetlogonServiceCalling:
TRUE if this is the netlogon service proper calling
FALSE if this is the changelog worker thread calling
Return Value:
TRUE : if the SAM service is successfully starts.
FALSE : if the SAM service can't start.
--*/
{
NTSTATUS Status;
DWORD WaitStatus;
UNICODE_STRING EventName;
HANDLE EventHandle;
OBJECT_ATTRIBUTES EventAttributes;
//
// open SAM event
//
RtlInitUnicodeString( &EventName, L"\\SAM_SERVICE_STARTED");
InitializeObjectAttributes( &EventAttributes, &EventName, 0, 0, NULL );
Status = NtOpenEvent( &EventHandle,
SYNCHRONIZE|EVENT_MODIFY_STATE,
&EventAttributes );
if ( !NT_SUCCESS(Status)) {
if( Status == STATUS_OBJECT_NAME_NOT_FOUND ) {
//
// SAM hasn't created this event yet, let us create it now.
// SAM opens this event to set it.
//
Status = NtCreateEvent(
&EventHandle,
SYNCHRONIZE|EVENT_MODIFY_STATE,
&EventAttributes,
NotificationEvent,
FALSE // The event is initially not signaled
);
if( Status == STATUS_OBJECT_NAME_EXISTS ||
Status == STATUS_OBJECT_NAME_COLLISION ) {
//
// second change, if the SAM created the event before we
// do.
//
Status = NtOpenEvent( &EventHandle,
SYNCHRONIZE|EVENT_MODIFY_STATE,
&EventAttributes );
}
}
if ( !NT_SUCCESS(Status)) {
//
// could not make the event handle
//
NlPrint((NL_CRITICAL,
"NlWaitForSamService couldn't make the event handle : "
"%lx\n", Status));
return( FALSE );
}
}
//
// Loop waiting.
//
for (;;) {
WaitStatus = WaitForSingleObject( EventHandle,
5*1000 ); // 5 Seconds
if ( WaitStatus == WAIT_TIMEOUT ) {
if ( NetlogonServiceCalling ) {
NlPrint((NL_CRITICAL,
"NlWaitForSamService 5-second timeout (Rewaiting)\n" ));
if (!GiveInstallHints( FALSE )) {
(VOID) NtClose( EventHandle );
return FALSE;
}
}
continue;
} else if ( WaitStatus == WAIT_OBJECT_0 ) {
break;
} else {
NlPrint((NL_CRITICAL,
"NlWaitForSamService: error %ld %ld\n",
GetLastError(),
WaitStatus ));
(VOID) NtClose( EventHandle );
return FALSE;
}
}
(VOID) NtClose( EventHandle );
return TRUE;
}
NET_API_STATUS
NlReadBinaryLog(
IN LPWSTR FileSuffix,
IN BOOL DeleteName,
OUT LPBYTE *Buffer,
OUT PULONG BufferSize
)
/*++
Routine Description:
Read a file into a buffer.
Arguments:
FileSuffix - Specifies the name of the file to write (relative to the
Windows directory)
DeleteName - If true the file will be deleted
Buffer - Returns an allocated buffer containing the file.
Buffer should be freed using LocalFree.
If the file doesn't exist, NULL will be returned (and NO_ERROR)
BufferSize - Returns size (in bytes) of buffer
Return Value:
Status of the operation.
--*/
{
NET_API_STATUS NetStatus;
LPWSTR FileName = NULL;
UINT WindowsDirectoryLength;
HANDLE FileHandle = INVALID_HANDLE_VALUE;
BOOLEAN FileNameKnown = FALSE;
ULONG BytesRead;
//
// Initialization
//
*Buffer = NULL;
*BufferSize = 0;
//
// Allocate a block to build the file name in
// (Don't put it on the stack since we don't want to commit a huge stack.)
//
FileName = LocalAlloc( LMEM_ZEROINIT, sizeof(WCHAR) * (MAX_PATH+1) );
if ( FileName == NULL ) {
NetStatus = ERROR_NOT_ENOUGH_MEMORY;
goto Cleanup;
}
//
// Build the name of the log file
//
WindowsDirectoryLength = GetSystemWindowsDirectoryW(
FileName,
sizeof(WCHAR) * (MAX_PATH+1) );
if ( WindowsDirectoryLength == 0 ) {
NetStatus = GetLastError();
NlPrint(( NL_CRITICAL,
"NlWriteBinaryLog: Unable to GetSystemWindowsDirectoryW (%ld)\n",
NetStatus ));
goto Cleanup;
}
if ( WindowsDirectoryLength + wcslen( FileSuffix ) + 1 >= MAX_PATH ) {
NlPrint((NL_CRITICAL,
"NlWriteBinaryLog: file name length is too long \n" ));
NetStatus = ERROR_INVALID_NAME;
goto Cleanup;
}
wcscat( FileName, FileSuffix );
FileNameKnown = TRUE;
//
// Open binary log file if exists
//
FileHandle = CreateFileW(
FileName,
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL, // Supply better security ??
OPEN_EXISTING, // Only open it if it exists
FILE_ATTRIBUTE_NORMAL,
NULL ); // No template
if ( FileHandle == INVALID_HANDLE_VALUE) {
NlPrint(( NL_CRITICAL,
FORMAT_LPWSTR ": Unable to open. %ld\n",
FileName,
GetLastError() ));
// This isn't fatal
NetStatus = NO_ERROR;
goto Cleanup;
}
//
// Get the size of the file.
//
*BufferSize = GetFileSize( FileHandle, NULL );
if ( *BufferSize == 0xFFFFFFFF ) {
NetStatus = GetLastError();
NlPrint((NL_CRITICAL,
"%ws: Unable to GetFileSize: %ld \n",
FileName,
NetStatus ));
*BufferSize = 0;
goto Cleanup;
}
if ( *BufferSize < 1 ) {
NlPrint(( NL_CRITICAL,
"NlReadBinaryLog: %ws: size too small: %ld.\n",
FileName,
*BufferSize ));
*BufferSize = 0;
NetStatus = NO_ERROR;
goto Cleanup;
}
//
// Allocate a buffer to read the file into.
//
*Buffer = LocalAlloc( 0, *BufferSize );
if ( *Buffer == NULL ) {
*BufferSize = 0;
NetStatus = ERROR_NOT_ENOUGH_MEMORY;
goto Cleanup;
}
//
// Read the file into the buffer.
//
if ( !ReadFile( FileHandle,
*Buffer,
*BufferSize,
&BytesRead,
NULL ) ) { // Not Overlapped
NetStatus = GetLastError();
if ( NetStatus != ERROR_FILE_NOT_FOUND ) {
NlPrint(( NL_CRITICAL,
"NlReadBinaryLog: %ws: Cannot ReadFile: %ld.\n",
FileName,
NetStatus ));
}
LocalFree( *Buffer );
*Buffer = NULL;
*BufferSize = 0;
NetStatus = NO_ERROR;
goto Cleanup;
}
if ( BytesRead != *BufferSize ) {
NlPrint(( NL_CRITICAL,
"NlReadBinaryLog: %ws: Cannot read entire File: %ld %ld.\n",
FileName,
BytesRead,
*BufferSize ));
LocalFree( *Buffer );
*Buffer = NULL;
*BufferSize = 0;
NetStatus = NO_ERROR;
goto Cleanup;
}
NetStatus = NO_ERROR;
//
// Be tidy.
//
Cleanup:
if ( FileHandle != INVALID_HANDLE_VALUE ) {
CloseHandle( FileHandle );
}
//
// If the caller asked us to delete the file,
// do so now
//
if (DeleteName && FileNameKnown) {
if ( !DeleteFile( FileName ) ) {
DWORD WinError;
WinError = GetLastError();
if ( WinError != ERROR_FILE_NOT_FOUND ) {
NlPrint((NL_CRITICAL,
"Cannot delete %ws (%ld)\n",
FileName,
WinError ));
}
// This isn't fatal
}
}
if ( FileName != NULL ) {
LocalFree( FileName );
}
return NetStatus;
}
#if NETLOGONDBG
VOID
NlOpenDebugFile(
IN BOOL ReopenFlag
)
/*++
Routine Description:
Opens or re-opens the debug file
Arguments:
ReopenFlag - TRUE to indicate the debug file is to be closed, renamed,
and recreated.
Return Value:
None
--*/
{
LPWSTR AllocatedBuffer = NULL;
LPWSTR LogFileName;
LPWSTR BakFileName;
DWORD FileAttributes;
DWORD PathLength;
DWORD WinError;
//
// Allocate a buffer for storage local to this procedure.
// (Don't put it on the stack since we don't want to commit a huge stack.)
//
AllocatedBuffer = LocalAlloc( 0, sizeof(WCHAR) *
(MAX_PATH+1 +
MAX_PATH+1 ) );
if ( AllocatedBuffer == NULL ) {
goto ErrorReturn;
}
LogFileName = AllocatedBuffer;
BakFileName = &LogFileName[MAX_PATH+1];
//
// Close the handle to the debug file, if it is currently open
//
EnterCriticalSection( &NlGlobalLogFileCritSect );
if ( NlGlobalLogFile != INVALID_HANDLE_VALUE ) {
CloseHandle( NlGlobalLogFile );
NlGlobalLogFile = INVALID_HANDLE_VALUE;
}
LeaveCriticalSection( &NlGlobalLogFileCritSect );
//
// make debug directory path first, if it is not made before.
//
if( NlGlobalDebugSharePath == NULL ) {
if ( !GetSystemWindowsDirectoryW( LogFileName, MAX_PATH+1 ) ) {
NlPrint((NL_CRITICAL, "Window Directory Path can't be retrieved, %lu.\n",
GetLastError() ));
goto ErrorReturn;
}
//
// check debug path length.
//
PathLength = wcslen(LogFileName) * sizeof(WCHAR) +
sizeof(DEBUG_DIR) + sizeof(WCHAR);
if( (PathLength + sizeof(DEBUG_FILE) > MAX_PATH+1 ) ||
(PathLength + sizeof(DEBUG_BAK_FILE) > MAX_PATH+1 ) ) {
NlPrint((NL_CRITICAL, "Debug directory path (%ws) length is too long.\n",
LogFileName));
goto ErrorReturn;
}
wcscat(LogFileName, DEBUG_DIR);
//
// copy debug directory name to global var.
//
NlGlobalDebugSharePath =
NetpMemoryAllocate( (wcslen(LogFileName) + 1) * sizeof(WCHAR) );
if( NlGlobalDebugSharePath == NULL ) {
NlPrint((NL_CRITICAL, "Can't allocated memory for debug share "
"(%ws).\n", LogFileName));
goto ErrorReturn;
}
wcscpy(NlGlobalDebugSharePath, LogFileName);
}
else {
wcscpy(LogFileName, NlGlobalDebugSharePath);
}
//
// Check this path exists.
//
FileAttributes = GetFileAttributesW( LogFileName );
if( FileAttributes == 0xFFFFFFFF ) {
WinError = GetLastError();
if( WinError == ERROR_FILE_NOT_FOUND ) {
//
// Create debug directory.
//
if( !CreateDirectoryW( LogFileName, NULL) ) {
NlPrint((NL_CRITICAL, "Can't create Debug directory (%ws), %lu.\n",
LogFileName, GetLastError() ));
goto ErrorReturn;
}
}
else {
NlPrint((NL_CRITICAL, "Can't Get File attributes(%ws), %lu.\n",
LogFileName, WinError ));
goto ErrorReturn;
}
}
else {
//
// if this is not a directory.
//
if(!(FileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
NlPrint((NL_CRITICAL, "Debug directory path (%ws) exists as file.\n",
LogFileName));
goto ErrorReturn;
}
}
//
// Create the name of the old and new log file names
//
(VOID) wcscpy( BakFileName, LogFileName );
(VOID) wcscat( LogFileName, L"\\Netlogon.log" );
(VOID) wcscat( BakFileName, L"\\Netlogon.bak" );
//
// If this is a re-open,
// delete the backup file,
// rename the current file to the backup file.
//
if ( ReopenFlag ) {
if ( !DeleteFile( BakFileName ) ) {
WinError = GetLastError();
if ( WinError != ERROR_FILE_NOT_FOUND ) {
NlPrint((NL_CRITICAL,
"Cannot delete " FORMAT_LPWSTR "(%ld)\n",
BakFileName,
WinError ));
NlPrint((NL_CRITICAL, " Try to re-open the file.\n"));
ReopenFlag = FALSE; // Don't truncate the file
}
}
}
if ( ReopenFlag ) {
if ( !MoveFile( LogFileName, BakFileName ) ) {
NlPrint((NL_CRITICAL,
"Cannot rename " FORMAT_LPWSTR " to " FORMAT_LPWSTR
" (%ld)\n",
LogFileName,
BakFileName,
GetLastError() ));
NlPrint((NL_CRITICAL,
" Try to re-open the file.\n"));
ReopenFlag = FALSE; // Don't truncate the file
}
}
//
// Open the file.
//
EnterCriticalSection( &NlGlobalLogFileCritSect );
NlGlobalLogFile = CreateFileW( LogFileName,
GENERIC_WRITE|GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
ReopenFlag ? CREATE_ALWAYS : OPEN_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL );
if ( NlGlobalLogFile == INVALID_HANDLE_VALUE ) {
NlPrint((NL_CRITICAL, "cannot open " FORMAT_LPWSTR ",\n",
LogFileName ));
LeaveCriticalSection( &NlGlobalLogFileCritSect );
goto ErrorReturn;
} else {
// Position the log file at the end
(VOID) SetFilePointer( NlGlobalLogFile,
0,
NULL,
FILE_END );
}
LeaveCriticalSection( &NlGlobalLogFileCritSect );
Cleanup:
if ( AllocatedBuffer != NULL ) {
LocalFree( AllocatedBuffer );
}
return;
ErrorReturn:
NlPrint((NL_CRITICAL,
" Debug output will be written to debug terminal.\n"));
goto Cleanup;
}
#define MAX_PRINTF_LEN 1024 // Arbitrary.
VOID
NlPrintRoutineV(
IN DWORD DebugFlag,
IN LPSTR Format,
va_list arglist
)
// Must be called with NlGlobalLogFileCritSect locked
{
static LPSTR NlGlobalLogFileOutputBuffer = NULL;
ULONG length;
int lengthTmp;
DWORD BytesWritten;
static BeginningOfLine = TRUE;
static LineCount = 0;
static TruncateLogFileInProgress = FALSE;
static LogProblemWarned = FALSE;
//
// If we aren't debugging this functionality, just return.
//
if ( DebugFlag != 0 && (NlGlobalParameters.DbFlag & DebugFlag) == 0 ) {
return;
}
//
// Allocate a buffer to build the line in.
// If there isn't already one.
//
length = 0;
if ( NlGlobalLogFileOutputBuffer == NULL ) {
NlGlobalLogFileOutputBuffer = LocalAlloc( 0, MAX_PRINTF_LEN + 1 );
if ( NlGlobalLogFileOutputBuffer == NULL ) {
return;
}
}
//
// Handle the beginning of a new line.
//
//
if ( BeginningOfLine ) {
//
// Never print empty lines.
//
if ( Format[0] == '\n' && Format[1] == '\0' ) {
return;
}
//
// If the log file is getting huge,
// truncate it.
//
if ( NlGlobalLogFile != INVALID_HANDLE_VALUE &&
!TruncateLogFileInProgress ) {
//
// Only check every 50 lines,
//
LineCount++;
if ( LineCount >= 50 ) {
DWORD FileSize;
LineCount = 0;
//
// Is the log file too big?
//
FileSize = GetFileSize( NlGlobalLogFile, NULL );
if ( FileSize == 0xFFFFFFFF ) {
(void) DbgPrint( "[NETLOGON] Cannot GetFileSize %ld\n",
GetLastError );
} else if ( FileSize > NlGlobalParameters.LogFileMaxSize ) {
TruncateLogFileInProgress = TRUE;
LeaveCriticalSection( &NlGlobalLogFileCritSect );
NlOpenDebugFile( TRUE );
NlPrint(( NL_MISC,
"Logfile truncated because it was larger than %ld bytes\n",
NlGlobalParameters.LogFileMaxSize ));
EnterCriticalSection( &NlGlobalLogFileCritSect );
TruncateLogFileInProgress = FALSE;
}
}
}
//
// If we're writing to the debug terminal,
// indicate this is a Netlogon message.
//
if ( NlGlobalLogFile == INVALID_HANDLE_VALUE ) {
length += (ULONG) sprintf( &NlGlobalLogFileOutputBuffer[length], "[NETLOGON] " );
}
//
// Put the timestamp at the begining of the line.
//
{
SYSTEMTIME SystemTime;
GetLocalTime( &SystemTime );
length += (ULONG) sprintf( &NlGlobalLogFileOutputBuffer[length],
"%02u/%02u %02u:%02u:%02u ",
SystemTime.wMonth,
SystemTime.wDay,
SystemTime.wHour,
SystemTime.wMinute,
SystemTime.wSecond );
}
//
// Indicate the type of message on the line
//
{
char *Text;
switch (DebugFlag) {
case NL_INIT:
Text = "INIT"; break;
case NL_MISC:
Text = "MISC"; break;
case NL_LOGON:
Text = "LOGON"; break;
case NL_SYNC:
case NL_PACK_VERBOSE:
case NL_REPL_TIME:
case NL_REPL_OBJ_TIME:
case NL_SYNC_MORE:
Text = "SYNC"; break;
case NL_ENCRYPT:
Text = "ENCRYPT"; break;
case NL_MAILSLOT:
case NL_MAILSLOT_TEXT:
Text = "MAILSLOT"; break;
case NL_SITE:
case NL_SITE_MORE:
Text = "SITE"; break;
case NL_CRITICAL:
Text = "CRITICAL"; break;
case NL_SESSION_SETUP:
case NL_SESSION_MORE:
case NL_CHALLENGE_RES:
case NL_INHIBIT_CANCEL:
case NL_SERVER_SESS:
Text = "SESSION"; break;
case NL_CHANGELOG:
Text = "CHANGELOG"; break;
case NL_PULSE_MORE:
Text = "PULSE"; break;
case NL_DOMAIN:
Text = "DOMAIN"; break;
case NL_DNS:
case NL_DNS_MORE:
Text = "DNS"; break;
case NL_WORKER:
Text = "WORKER"; break;
case NL_TIMESTAMP:
case NL_BREAKPOINT:
default:
Text = "UNKNOWN"; break;
case 0:
Text = NULL;
}
if ( Text != NULL ) {
length += (ULONG) sprintf( &NlGlobalLogFileOutputBuffer[length], "[%s] ", Text );
}
}
}
//
// Put a the information requested by the caller onto the line
//
lengthTmp = (ULONG) _vsnprintf( &NlGlobalLogFileOutputBuffer[length],
MAX_PRINTF_LEN - length - 1,
Format,
arglist );
if ( lengthTmp < 0 ) {
length = MAX_PRINTF_LEN - 1;
// always end the line which cannot fit into the buffer
NlGlobalLogFileOutputBuffer[length-1] = '\n';
} else {
length += lengthTmp;
}
BeginningOfLine = (length > 0 && NlGlobalLogFileOutputBuffer[length-1] == '\n' );
if ( BeginningOfLine ) {
NlGlobalLogFileOutputBuffer[length-1] = '\r';
NlGlobalLogFileOutputBuffer[length] = '\n';
NlGlobalLogFileOutputBuffer[length+1] = '\0';
length++;
}
//
// If the log file isn't open,
// just output to the debug terminal
//
if ( NlGlobalLogFile == INVALID_HANDLE_VALUE ) {
#if DBG
if ( !LogProblemWarned ) {
(void) DbgPrint( "[NETLOGON] Cannot write to log file [Invalid Handle]\n" );
LogProblemWarned = TRUE;
}
#endif // DBG
//
// Write the debug info to the log file.
//
} else {
if ( !WriteFile( NlGlobalLogFile,
NlGlobalLogFileOutputBuffer,
length,
&BytesWritten,
NULL ) ) {
#if DBG
if ( !LogProblemWarned ) {
(void) DbgPrint( "[NETLOGON] Cannot write to log file %ld\n", GetLastError() );
LogProblemWarned = TRUE;
}
#endif // DBG
}
}
} // NlPrintRoutineV
VOID
NlPrintRoutine(
IN DWORD DebugFlag,
IN LPSTR Format,
...
)
{
va_list arglist;
//
// vsprintf isn't multithreaded + we don't want to intermingle output
// from different threads.
//
EnterCriticalSection( &NlGlobalLogFileCritSect );
//
// Simply change arguments to va_list form and call NlPrintRoutineV
//
va_start(arglist, Format);
NlPrintRoutineV( DebugFlag, Format, arglist );
va_end(arglist);
LeaveCriticalSection( &NlGlobalLogFileCritSect );
} // NlPrintRoutine
VOID
NlPrintDomRoutine(
IN DWORD DebugFlag,
IN PDOMAIN_INFO DomainInfo OPTIONAL,
IN LPSTR Format,
...
)
{
va_list arglist;
//
// vsprintf isn't multithreaded + we don't want to intermingle output
// from different threads.
//
EnterCriticalSection( &NlGlobalLogFileCritSect );
//
// Prefix the printed line with the domain name
//
if ( NlGlobalServicedDomainCount > 1 ) {
if ( DomainInfo == NULL ) {
NlPrint(( DebugFlag, "%ws: ", L"[Unknown]" ));
} else if ( DomainInfo->DomUnicodeDomainName != NULL &&
*(DomainInfo->DomUnicodeDomainName) != UNICODE_NULL ) {
NlPrint(( DebugFlag, "%ws: ", DomainInfo->DomUnicodeDomainName ));
} else {
NlPrint(( DebugFlag, "%ws: ", DomainInfo->DomUnicodeDnsDomainName ));
}
}
//
// Simply change arguments to va_list form and call NlPrintRoutineV
//
va_start(arglist, Format);
NlPrintRoutineV( DebugFlag, Format, arglist );
va_end(arglist);
LeaveCriticalSection( &NlGlobalLogFileCritSect );
} // NlPrintDomRoutine
VOID
NlPrintCsRoutine(
IN DWORD DebugFlag,
IN PCLIENT_SESSION ClientSession,
IN LPSTR Format,
...
)
{
va_list arglist;
//
// vsprintf isn't multithreaded + we don't want to intermingle output
// from different threads.
//
EnterCriticalSection( &NlGlobalLogFileCritSect );
//
// If a ClientSession was actually passed,
// print information specific to the session.
//
if ( ClientSession != NULL ) {
//
// Prefix the printed line with the hosted domain name
//
if ( NlGlobalServicedDomainCount > 1 ) {
NlPrint(( DebugFlag,
"%ws: ",
ClientSession->CsDomainInfo == NULL ? L"[Unknown]" : ClientSession->CsDomainInfo->DomUnicodeDomainName ));
}
//
// Prefix the printed line with the name of the trusted domain
//
NlPrint(( DebugFlag,
"%ws: ",
ClientSession->CsDebugDomainName == NULL ? L"[Unknown]" : ClientSession->CsDebugDomainName ));
}
//
// Simply change arguments to va_list form and call NlPrintRoutineV
//
va_start(arglist, Format);
NlPrintRoutineV( DebugFlag, Format, arglist );
va_end(arglist);
LeaveCriticalSection( &NlGlobalLogFileCritSect );
} // NlPrintCsRoutine
//
// Have my own version of RtlAssert so debug versions of netlogon really assert on
// free builds.
//
VOID
NlAssertFailed(
IN PVOID FailedAssertion,
IN PVOID FileName,
IN ULONG LineNumber,
IN PCHAR Message OPTIONAL
)
{
char Response[ 2 ];
#if DBG
while (TRUE) {
#endif // DBG
NlPrint(( NL_CRITICAL, "Assertion failed: %s%s (Source File: %s, line %ld)\n",
Message ? Message : "",
FailedAssertion,
FileName,
LineNumber
));
#if DBG
DbgPrint( "\n*** Assertion failed: %s%s\n*** Source File: %s, line %ld\n\n",
Message ? Message : "",
FailedAssertion,
FileName,
LineNumber
);
DbgPrompt( "Break, Ignore, Terminate Process or Terminate Thread (bipt)? ",
Response,
sizeof( Response )
);
switch (Response[0]) {
case 'B':
case 'b':
DbgBreakPoint();
break;
case 'I':
case 'i':
return;
case 'P':
case 'p':
NtTerminateProcess( NtCurrentProcess(), STATUS_UNSUCCESSFUL );
break;
case 'T':
case 't':
NtTerminateThread( NtCurrentThread(), STATUS_UNSUCCESSFUL );
break;
}
}
DbgBreakPoint();
NtTerminateProcess( NtCurrentProcess(), STATUS_UNSUCCESSFUL );
#endif // DBG
}
#endif // NETLOGONDBG
BOOLEAN
NlpIsNtStatusResourceError(
NTSTATUS Status
)
/*++
Routine Description:
Returns TRUE if the passed in status is a resource error.
Arguments:
Status - NT status code to check
Return Value:
TRUE - if the status indicates a lack of resources
--*/
{
switch ( Status ) {
case STATUS_NO_MEMORY:
case STATUS_INSUFFICIENT_RESOURCES:
case STATUS_DISK_FULL:
return TRUE;
default:
return FALSE;
}
}
BOOLEAN
NlpDidDcFail(
NTSTATUS Status
)
/*++
Routine Description:
Call this routine with the Status code returned from a secure channel API.
This routine checks the status code to determine if it specifically is one
that indicates the DC is having problems. The caller should respond by
dropping the secure channel and picking another DC.
Arguments:
Status - NT status code to check
Return Value:
TRUE - if the status indicates the DC failed
--*/
{
//
// ???: we might consider adding the communications errors here
// (e.g., RPC_NT_CALL_FAILED and RPC_NT_SERVER_UNAVAILABLE).
// However, all current callers already handle this issue using a more generic
// mechanism. For instance, those secure channel API that take an authenticator
// will have the authenticator wrong for communications errors. The other secure
// channel API rely on the RPC exception differentiating between comm errors
// and status codes from the DC.
//
//
// Handle the "original recipe" status code indicating a secure channel problem
//
switch ( Status ) {
case STATUS_ACCESS_DENIED:
return TRUE;
default:
return NlpIsNtStatusResourceError( Status );
}
return FALSE;
}
Reference in a new issue Copy permalink