csr.h File Reference

#include "csrmsg.h"

Include dependency graph for csr.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
NTSTATUS NTAPI	CsrClientConnectToServer (_In_ PCWSTR ObjectDirectory, _In_ ULONG ServerId, _In_ PVOID ConnectionInfo, _Inout_ PULONG ConnectionInfoSize, _Out_ PBOOLEAN ServerToServerCall)
NTSTATUS NTAPI	CsrClientCallServer (_Inout_ PCSR_API_MESSAGE ApiMessage, _Inout_opt_ PCSR_CAPTURE_BUFFER CaptureBuffer, _In_ CSR_API_NUMBER ApiNumber, _In_ ULONG DataLength)
PCSR_CAPTURE_BUFFER NTAPI	CsrAllocateCaptureBuffer (_In_ ULONG ArgumentCount, _In_ ULONG BufferSize)
ULONG NTAPI	CsrAllocateMessagePointer (_Inout_ PCSR_CAPTURE_BUFFER CaptureBuffer, _In_ ULONG MessageLength, _Out_ PVOID *CapturedData)
VOID NTAPI	CsrCaptureMessageBuffer (_Inout_ PCSR_CAPTURE_BUFFER CaptureBuffer, _In_opt_ PVOID MessageBuffer, _In_ ULONG MessageLength, _Out_ PVOID *CapturedData)
VOID NTAPI	CsrFreeCaptureBuffer (_In_ _Frees_ptr_ PCSR_CAPTURE_BUFFER CaptureBuffer)
VOID NTAPI	CsrCaptureMessageString (_Inout_ PCSR_CAPTURE_BUFFER CaptureBuffer, _In_opt_ PCSTR String, _In_ ULONG StringLength, _In_ ULONG MaximumLength, _Out_ PSTRING CapturedString)
VOID NTAPI	CsrCaptureMessageUnicodeStringInPlace (_Inout_ PCSR_CAPTURE_BUFFER CaptureBuffer, _Inout_ PUNICODE_STRING String)
NTSTATUS NTAPI	CsrCaptureMessageMultiUnicodeStringsInPlace (_Inout_ PCSR_CAPTURE_BUFFER *CaptureBuffer, _In_ ULONG StringsCount, _In_ PUNICODE_STRING *MessageStrings)
PLARGE_INTEGER NTAPI	CsrCaptureTimeout (_In_ ULONG Milliseconds, _Out_ PLARGE_INTEGER Timeout)
VOID NTAPI	CsrProbeForRead (_In_ PVOID Address, _In_ ULONG Length, _In_ ULONG Alignment)
VOID NTAPI	CsrProbeForWrite (_In_ PVOID Address, _In_ ULONG Length, _In_ ULONG Alignment)
HANDLE NTAPI	CsrGetProcessId (VOID)
NTSTATUS NTAPI	CsrNewThread (VOID)
NTSTATUS NTAPI	CsrIdentifyAlertableThread (VOID)
NTSTATUS NTAPI	CsrSetPriorityClass (_In_ HANDLE Process, _Inout_ PULONG PriorityClass)

Function Documentation

CsrAllocateCaptureBuffer()

PCSR_CAPTURE_BUFFER NTAPI CsrAllocateCaptureBuffer	(	_In_ ULONG	ArgumentCount,
		_In_ ULONG	BufferSize
	)

Definition at line 87 of file capture.c.

{
    PCSR_CAPTURE_BUFFER CaptureBuffer;
    ULONG OffsetsArraySize;
    ULONG MaximumSize;
 
    /* Validate the argument count. Note that on server side, CSRSRV
     * limits the count to MAXUSHORT; here we are a bit more lenient. */
    if (ArgumentCount > (MAXLONG / sizeof(ULONG_PTR)))
        return NULL;
 
    OffsetsArraySize = ArgumentCount * sizeof(ULONG_PTR);
 
    /*
     * Validate the total buffer size.
     * The total size of the header plus the pointer-offset array and the
     * provided buffer, together with the alignment padding for each argument,
     * must be less than MAXLONG aligned to 4-byte boundary.
     */
    MaximumSize = (MAXLONG & ~3) - FIELD_OFFSET(CSR_CAPTURE_BUFFER, PointerOffsetsArray);
    if (OffsetsArraySize >= MaximumSize)
        return NULL;
    MaximumSize -= OffsetsArraySize;
    if (BufferSize >= MaximumSize)
        return NULL;
    MaximumSize -= BufferSize;
    if ((ArgumentCount * 3) + 3 >= MaximumSize)
        return NULL;
 
    /* Add the size of the header and of the pointer-offset array */
    BufferSize += FIELD_OFFSET(CSR_CAPTURE_BUFFER, PointerOffsetsArray) +
                    OffsetsArraySize;
 
    /* Add the size of the alignment padding for each argument */
    BufferSize += ArgumentCount * 3;
 
    /* Align it to a 4-byte boundary */
    BufferSize = (BufferSize + 3) & ~3;
 
    /* Allocate memory from the port heap */
    CaptureBuffer = RtlAllocateHeap(CsrPortHeap, HEAP_ZERO_MEMORY, BufferSize);
    if (CaptureBuffer == NULL) return NULL;
 
    /* Initialize the header */
    CaptureBuffer->Size = BufferSize;
    CaptureBuffer->PointerCount = 0;
 
    /* Initialize the pointer-offset array */
    RtlZeroMemory(CaptureBuffer->PointerOffsetsArray, OffsetsArraySize);
 
    /* Point to the start of the free buffer */
    CaptureBuffer->BufferEnd = (PVOID)((ULONG_PTR)CaptureBuffer->PointerOffsetsArray +
                                       OffsetsArraySize);
 
    /* Return the address of the buffer */
    return CaptureBuffer;
}

Referenced by BaseCheckVDM(), CreateConsoleScreenBuffer(), CsrCaptureMessageMultiUnicodeStringsInPlace(), CsrClientConnectToServer(), DefineDosDeviceW(), DeviceEventWorker(), GetConsoleFontInfo(), GetConsoleProcessList(), GetNextVDMCommand(), GetVDMCurrentDirectories(), IntAddConsoleAlias(), IntAllocConsole(), IntAttachConsole(), IntExpungeConsoleCommandHistory(), IntGetConsoleAlias(), IntGetConsoleAliases(), IntGetConsoleAliasesLength(), IntGetConsoleAliasExes(), IntGetConsoleCommandHistory(), IntGetConsoleCommandHistoryLength(), IntGetConsoleInput(), IntGetConsoleTitle(), IntReadConsole(), IntReadConsoleOutput(), IntReadConsoleOutputCode(), IntRegisterConsoleIME(), IntSetConsoleNumberOfCommands(), IntSetConsoleTitle(), IntWriteConsole(), IntWriteConsoleInput(), IntWriteConsoleOutput(), IntWriteConsoleOutputCode(), RegisterConsoleVDM(), and SetVDMCurrentDirectories().

CsrAllocateMessagePointer()

ULONG NTAPI CsrAllocateMessagePointer	(	_Inout_ PCSR_CAPTURE_BUFFER	CaptureBuffer,
		_In_ ULONG	MessageLength,
		_Out_ PVOID *	CapturedData
	)

Definition at line 152 of file capture.c.

{
    if (MessageLength == 0)
    {
        *CapturedData = NULL;
        CapturedData = NULL;
    }
    else
    {
        /* Set the capture data at our current available buffer */
        *CapturedData = CaptureBuffer->BufferEnd;
 
        /* Validate the size */
        if (MessageLength >= MAXLONG) return 0;
 
        /* Align it to a 4-byte boundary */
        MessageLength = (MessageLength + 3) & ~3;
 
        /* Move our available buffer beyond this space */
        CaptureBuffer->BufferEnd = (PVOID)((ULONG_PTR)CaptureBuffer->BufferEnd + MessageLength);
    }
 
    /* Write down this pointer in the array and increase the count */
    CaptureBuffer->PointerOffsetsArray[CaptureBuffer->PointerCount++] = (ULONG_PTR)CapturedData;
 
    /* Return the aligned length */
    return MessageLength;
}

Referenced by CsrCaptureMessageBuffer(), CsrCaptureMessageString(), DefineDosDeviceW(), GetConsoleFontInfo(), GetConsoleProcessList(), GetNextVDMCommand(), GetVDMCurrentDirectories(), IntGetConsoleAlias(), IntGetConsoleAliases(), IntGetConsoleAliasExes(), IntGetConsoleCommandHistory(), IntGetConsoleInput(), IntGetConsoleTitle(), IntReadConsole(), IntReadConsoleOutput(), IntReadConsoleOutputCode(), and IntWriteConsoleOutput().

CsrCaptureMessageBuffer()

VOID NTAPI CsrCaptureMessageBuffer	(	_Inout_ PCSR_CAPTURE_BUFFER	CaptureBuffer,
		_In_opt_ PVOID	MessageBuffer,
		_In_ ULONG	MessageLength,
		_Out_ PVOID *	CapturedData
	)

Definition at line 189 of file capture.c.

{
    /* Simply allocate a message pointer in the buffer */
    CsrAllocateMessagePointer(CaptureBuffer, MessageLength, CapturedData);
 
    /* Check if there was any data */
    if (!MessageBuffer || !MessageLength) return;
 
    /* Copy the data into the buffer */
    RtlMoveMemory(*CapturedData, MessageBuffer, MessageLength);
}

Referenced by BaseCheckVDM(), CreateConsoleScreenBuffer(), CsrClientConnectToServer(), DeviceEventWorker(), IntAddConsoleAlias(), IntAllocConsole(), IntAttachConsole(), IntExpungeConsoleCommandHistory(), IntGetConsoleAlias(), IntGetConsoleAliases(), IntGetConsoleAliasesLength(), IntGetConsoleCommandHistory(), IntGetConsoleCommandHistoryLength(), IntRegisterConsoleIME(), IntSetConsoleNumberOfCommands(), IntSetConsoleTitle(), IntWriteConsole(), IntWriteConsoleInput(), IntWriteConsoleOutputCode(), RegisterConsoleVDM(), and SetVDMCurrentDirectories().

CsrCaptureMessageMultiUnicodeStringsInPlace()

NTSTATUS NTAPI CsrCaptureMessageMultiUnicodeStringsInPlace	(	_Inout_ PCSR_CAPTURE_BUFFER *	CaptureBuffer,
		_In_ ULONG	StringsCount,
		_In_ PUNICODE_STRING *	MessageStrings
	)

Definition at line 294 of file capture.c.

{
    ULONG Count;
 
    if (!CaptureBuffer) return STATUS_INVALID_PARAMETER;
 
    /* Allocate a new capture buffer if we don't have one already */
    if (!*CaptureBuffer)
    {
        /* Compute the required size for the capture buffer */
        ULONG Size = 0;
 
        Count = 0;
        while (Count < StringsCount)
        {
            if (MessageStrings[Count])
                Size += MessageStrings[Count]->MaximumLength;
 
            ++Count;
        }
 
        /* Allocate the capture buffer */
        *CaptureBuffer = CsrAllocateCaptureBuffer(StringsCount, Size);
        if (!*CaptureBuffer) return STATUS_NO_MEMORY;
    }
 
    /* Now capture each UNICODE string */
    Count = 0;
    while (Count < StringsCount)
    {
        if (MessageStrings[Count])
            CsrCaptureMessageUnicodeStringInPlace(*CaptureBuffer, MessageStrings[Count]);
 
        ++Count;
    }
 
    return STATUS_SUCCESS;
}

CsrCaptureMessageString()

VOID NTAPI CsrCaptureMessageString	(	_Inout_ PCSR_CAPTURE_BUFFER	CaptureBuffer,
		_In_opt_ PCSTR	String,
		_In_ ULONG	StringLength,
		_In_ ULONG	MaximumLength,
		_Out_ PSTRING	CapturedString
	)

Definition at line 222 of file capture.c.

{
    ASSERT(CapturedString != NULL);
 
    /*
     * If we don't have a string, initialize an empty one,
     * otherwise capture the given string.
     */
    if (!String)
    {
        CapturedString->Length = 0;
        CapturedString->MaximumLength = (USHORT)MaximumLength;
 
        /* Allocate a pointer for it */
        CsrAllocateMessagePointer(CaptureBuffer,
                                  MaximumLength,
                                  (PVOID*)&CapturedString->Buffer);
    }
    else
    {
        /* Cut-off the string length if needed */
        if (StringLength > MaximumLength)
            StringLength = MaximumLength;
 
        CapturedString->Length = (USHORT)StringLength;
 
        /* Allocate a buffer and get its size */
        CapturedString->MaximumLength =
            (USHORT)CsrAllocateMessagePointer(CaptureBuffer,
                                              MaximumLength,
                                              (PVOID*)&CapturedString->Buffer);
 
        /* If the string has data, copy it into the buffer */
        if (StringLength)
            RtlMoveMemory(CapturedString->Buffer, String, StringLength);
    }
 
    /* Null-terminate the string if we don't take up the whole space */
    if (CapturedString->Length < CapturedString->MaximumLength)
        CapturedString->Buffer[CapturedString->Length] = ANSI_NULL;
}

Referenced by CsrCaptureMessageUnicodeStringInPlace().

CsrCaptureMessageUnicodeStringInPlace()

VOID NTAPI CsrCaptureMessageUnicodeStringInPlace	(	_Inout_ PCSR_CAPTURE_BUFFER	CaptureBuffer,
		_Inout_ PUNICODE_STRING	String
	)

Definition at line 271 of file capture.c.

{
    ASSERT(String != NULL);
 
    /* This is a way to capture the UNICODE string, since (Maximum)Length are also in bytes */
    CsrCaptureMessageString(CaptureBuffer,
                            (PCSTR)String->Buffer,
                            String->Length,
                            String->MaximumLength,
                            (PSTRING)String);
 
    /* Null-terminate the string if we don't take up the whole space */
    if (String->Length + sizeof(WCHAR) <= String->MaximumLength)
        String->Buffer[String->Length / sizeof(WCHAR)] = UNICODE_NULL;
}

Referenced by CsrCaptureMessageMultiUnicodeStringsInPlace().

CsrCaptureTimeout()

PLARGE_INTEGER NTAPI CsrCaptureTimeout	(	_In_ ULONG	Milliseconds,
		_Out_ PLARGE_INTEGER	Timeout
	)

Definition at line 341 of file capture.c.

{
    /* Validate the time */
    if (Milliseconds == -1) return NULL;
 
    /* Convert to relative ticks */
    Timeout->QuadPart = Milliseconds * -10000LL;
    return Timeout;
}

CsrClientCallServer()

NTSTATUS NTAPI CsrClientCallServer	(	_Inout_ PCSR_API_MESSAGE	ApiMessage,
		_Inout_opt_ PCSR_CAPTURE_BUFFER	CaptureBuffer,
		_In_ CSR_API_NUMBER	ApiNumber,
		_In_ ULONG	DataLength
	)

Definition at line 368 of file connect.c.

{
    NTSTATUS Status;
 
    /* Make sure the length is valid */
    if (DataLength > (MAXSHORT - sizeof(CSR_API_MESSAGE)))
    {
        DPRINT1("DataLength too big: %lu\n", DataLength);
        return STATUS_INVALID_PARAMETER;
    }
 
    /* Fill out the Port Message Header */
    ApiMessage->Header.u2.ZeroInit = 0;
    /* DataLength = user_data_size + anything between
     * header and data, including intermediate padding */
    ApiMessage->Header.u1.s1.DataLength = (CSHORT)DataLength +
        FIELD_OFFSET(CSR_API_MESSAGE, Data) - sizeof(ApiMessage->Header);
    /* TotalLength = header_size + DataLength + any structure trailing padding */
    ApiMessage->Header.u1.s1.TotalLength = (CSHORT)DataLength +
        sizeof(CSR_API_MESSAGE) - sizeof(ApiMessage->Data);
 
    /* Fill out the CSR Header */
    ApiMessage->ApiNumber = ApiNumber;
    ApiMessage->CsrCaptureData = NULL;
 
    DPRINT("API: %lx, u1.s1.DataLength: %x, u1.s1.TotalLength: %x\n",
           ApiNumber,
           ApiMessage->Header.u1.s1.DataLength,
           ApiMessage->Header.u1.s1.TotalLength);
 
    /* Check if we are already inside a CSR Server */
    if (!InsideCsrProcess)
    {
        ULONG PointerCount;
        PULONG_PTR OffsetPointer;
 
        /* Check if we got a Capture Buffer */
        if (CaptureBuffer)
        {
            /*
             * We have to convert from our local (client) view
             * to the remote (server) view.
             */
            ApiMessage->CsrCaptureData = (PCSR_CAPTURE_BUFFER)
                ((ULONG_PTR)CaptureBuffer + CsrPortMemoryDelta);
 
            /* Lock the buffer */
            CaptureBuffer->BufferEnd = NULL;
 
            /*
             * Each client pointer inside the CSR message is converted into
             * a server pointer, and each pointer to these message pointers
             * is converted into an offset.
             */
            PointerCount  = CaptureBuffer->PointerCount;
            OffsetPointer = CaptureBuffer->PointerOffsetsArray;
            while (PointerCount--)
            {
                if (*OffsetPointer != 0)
                {
                    *(PULONG_PTR)*OffsetPointer += CsrPortMemoryDelta;
                    *OffsetPointer -= (ULONG_PTR)ApiMessage;
                }
                ++OffsetPointer;
            }
        }
 
        /* Send the LPC Message */
        Status = NtRequestWaitReplyPort(CsrApiPort,
                                        &ApiMessage->Header,
                                        &ApiMessage->Header);
 
        /* Check if we got a Capture Buffer */
        if (CaptureBuffer)
        {
            /*
             * We have to convert back from the remote (server) view
             * to our local (client) view.
             */
            ApiMessage->CsrCaptureData = (PCSR_CAPTURE_BUFFER)
                ((ULONG_PTR)ApiMessage->CsrCaptureData - CsrPortMemoryDelta);
 
            /*
             * Convert back the offsets into pointers to CSR message
             * pointers, and convert back these message server pointers
             * into client pointers.
             */
            PointerCount  = CaptureBuffer->PointerCount;
            OffsetPointer = CaptureBuffer->PointerOffsetsArray;
            while (PointerCount--)
            {
                if (*OffsetPointer != 0)
                {
                    *OffsetPointer += (ULONG_PTR)ApiMessage;
                    *(PULONG_PTR)*OffsetPointer -= CsrPortMemoryDelta;
                }
                ++OffsetPointer;
            }
        }
 
        /* Check for success */
        if (!NT_SUCCESS(Status))
        {
            /* We failed. Overwrite the return value with the failure. */
            DPRINT1("LPC Failed: %lx\n", Status);
            ApiMessage->Status = Status;
        }
    }
    else
    {
        /* This is a server-to-server call */
        DPRINT("Server-to-server call\n");
 
        /* Save our CID; we check this equality inside CsrValidateMessageBuffer */
        ApiMessage->Header.ClientId = NtCurrentTeb()->ClientId;
 
        /* Do a direct call */
        Status = CsrServerApiRoutine(ApiMessage, ApiMessage);
 
        /* Check for success */
        if (!NT_SUCCESS(Status))
        {
            /* We failed. Overwrite the return value with the failure. */
            ApiMessage->Status = Status;
        }
    }
 
    /* Return the CSR Result */
    DPRINT("Got back: 0x%lx\n", ApiMessage->Status);
    return ApiMessage->Status;
}

Referenced by BaseCheckForVDM(), BaseCheckVDM(), BasepNotifyCsrOfThread(), BaseUpdateVDMEntry(), CloseConsoleHandle(), ConsoleMenuControl(), CreateConsoleScreenBuffer(), CreateProcessInternalW(), CsrClientConnectToServer(), CsrIdentifyAlertableThread(), CsrSetPriorityClass(), DefineDosDeviceW(), DeviceEventWorker(), DuplicateConsoleHandle(), EndTask(), ExitProcess(), ExitVDM(), ExitWindowsWorker(), FlushConsoleInputBuffer(), FreeConsole(), GenerateConsoleCtrlEvent(), GetConsoleCP(), GetConsoleCursorInfo(), GetConsoleDisplayMode(), GetConsoleFontInfo(), GetConsoleFontSize(), GetConsoleHandleInformation(), GetConsoleHardwareState(), GetConsoleHistoryInfo(), GetConsoleMode(), GetConsoleOutputCP(), GetConsoleProcessList(), GetConsoleScreenBufferInfo(), GetConsoleSelectionInfo(), GetConsoleWindow(), GetCurrentConsoleFont(), GetLargestConsoleWindowSize(), GetNextVDMCommand(), GetNumberOfConsoleFonts(), GetNumberOfConsoleInputEvents(), GetNumberOfConsoleMouseButtons(), GetProcessShutdownParameters(), GetTempFileNameW(), GetThreadDesktop(), GetVDMCurrentDirectories(), IntAddConsoleAlias(), IntAllocConsole(), IntAttachConsole(), IntExpungeConsoleCommandHistory(), IntFillConsoleOutputCode(), IntGetConsoleAlias(), IntGetConsoleAliases(), IntGetConsoleAliasesLength(), IntGetConsoleAliasExes(), IntGetConsoleAliasExesLength(), IntGetConsoleCommandHistory(), IntGetConsoleCommandHistoryLength(), IntGetConsoleInput(), IntGetConsoleKeyboardLayoutName(), IntGetConsoleTitle(), IntReadConsole(), IntReadConsoleOutput(), IntReadConsoleOutputCode(), IntRegisterConsoleIME(), IntScrollConsoleScreenBuffer(), IntSetConsoleNumberOfCommands(), IntSetConsoleTitle(), IntUnregisterConsoleIME(), IntWriteConsole(), IntWriteConsoleInput(), IntWriteConsoleOutput(), IntWriteConsoleOutputCode(), InvalidateConsoleDIBits(), Logon(), NotifySoundSentry(), OpenConsoleW(), RegisterConsoleVDM(), RegisterLogonProcess(), RegisterServicesProcess(), SetConsoleActiveScreenBuffer(), SetConsoleCommandHistoryMode(), SetConsoleCP(), SetConsoleCursor(), SetConsoleCursorInfo(), SetConsoleCursorPosition(), SetConsoleDisplayMode(), SetConsoleFont(), SetConsoleHandleInformation(), SetConsoleHardwareState(), SetConsoleHistoryInfo(), SetConsoleIcon(), SetConsoleMenuClose(), SetConsoleMode(), SetConsoleOutputCP(), SetConsolePalette(), SetConsoleScreenBufferSize(), SetConsoleTextAttribute(), SetConsoleWindowInfo(), SetLastConsoleEventActive(), SetProcessShutdownParameters(), SetTEBLangID(), SetVDMCurrentDirectories(), ShowConsoleCursor(), and VerifyConsoleIoHandle().

CsrClientConnectToServer()

NTSTATUS NTAPI CsrClientConnectToServer	(	_In_ PCWSTR	ObjectDirectory,
		_In_ ULONG	ServerId,
		_In_ PVOID	ConnectionInfo,
		_Inout_ PULONG	ConnectionInfoSize,
		_Out_ PBOOLEAN	ServerToServerCall
	)

Definition at line 198 of file connect.c.

{
    NTSTATUS Status;
    PIMAGE_NT_HEADERS NtHeader;
 
    DPRINT("CsrClientConnectToServer: %lx %p\n", ServerId, ConnectionInfo);
 
    /* Validate the Connection Info */
    if (ConnectionInfo && (!ConnectionInfoSize || !*ConnectionInfoSize))
    {
        DPRINT1("Connection info given, but no length\n");
        return STATUS_INVALID_PARAMETER;
    }
 
    /* Check if we're inside a CSR Process */
    if (InsideCsrProcess)
    {
        /* Tell the client that we're already inside CSR */
        if (ServerToServerCall) *ServerToServerCall = TRUE;
        return STATUS_SUCCESS;
    }
 
    /*
     * We might be in a CSR Process but not know it, if this is the first call.
     * So let's find out.
     */
    if (!(NtHeader = RtlImageNtHeader(NtCurrentPeb()->ImageBaseAddress)))
    {
        /* The image isn't valid */
        DPRINT1("Invalid image\n");
        return STATUS_INVALID_IMAGE_FORMAT;
    }
    InsideCsrProcess = (NtHeader->OptionalHeader.Subsystem == IMAGE_SUBSYSTEM_NATIVE);
 
    /* Now we can check if we are inside or not */
    if (InsideCsrProcess)
    {
        UNICODE_STRING CsrSrvName;
        HANDLE hCsrSrv;
        ANSI_STRING CsrServerRoutineName;
 
        /* We're inside, so let's find csrsrv */
        RtlInitUnicodeString(&CsrSrvName, L"csrsrv");
        Status = LdrGetDllHandle(NULL,
                                 NULL,
                                 &CsrSrvName,
                                 &hCsrSrv);
 
        /* Now get the Server to Server routine */
        RtlInitAnsiString(&CsrServerRoutineName, "CsrCallServerFromServer");
        Status = LdrGetProcedureAddress(hCsrSrv,
                                        &CsrServerRoutineName,
                                        0L,
                                        (PVOID*)&CsrServerApiRoutine);
 
        /* Use the local heap as port heap */
        CsrPortHeap = RtlGetProcessHeap();
 
        /* Tell the caller we're inside the server */
        if (ServerToServerCall) *ServerToServerCall = InsideCsrProcess;
        return STATUS_SUCCESS;
    }
 
    /* Now check if connection info is given */
    if (ConnectionInfo)
    {
        CSR_API_MESSAGE ApiMessage;
        PCSR_CLIENT_CONNECT ClientConnect = &ApiMessage.Data.CsrClientConnect;
        PCSR_CAPTURE_BUFFER CaptureBuffer;
 
        /* Well, we're definitely in a client now */
        InsideCsrProcess = FALSE;
 
        /* Do we have a connection to CSR yet? */
        if (!CsrApiPort)
        {
            /* No, set it up now */
            Status = CsrpConnectToServer(ObjectDirectory);
            if (!NT_SUCCESS(Status))
            {
                /* Failed */
                DPRINT1("Failure to connect to CSR\n");
                return Status;
            }
        }
 
        /* Setup the connect message header */
        ClientConnect->ServerId = ServerId;
        ClientConnect->ConnectionInfoSize = *ConnectionInfoSize;
 
        /* Setup a buffer for the connection info */
        CaptureBuffer = CsrAllocateCaptureBuffer(1, ClientConnect->ConnectionInfoSize);
        if (CaptureBuffer == NULL)
        {
            return STATUS_INSUFFICIENT_RESOURCES;
        }
 
        /* Capture the connection info data */
        CsrCaptureMessageBuffer(CaptureBuffer,
                                ConnectionInfo,
                                ClientConnect->ConnectionInfoSize,
                                &ClientConnect->ConnectionInfo);
 
        /* Return the allocated length */
        *ConnectionInfoSize = ClientConnect->ConnectionInfoSize;
 
        /* Call CSR */
        Status = CsrClientCallServer(&ApiMessage,
                                     CaptureBuffer,
                                     CSR_CREATE_API_NUMBER(CSRSRV_SERVERDLL_INDEX, CsrpClientConnect),
                                     sizeof(*ClientConnect));
 
        /* Copy the updated connection info data back into the user buffer */
        RtlMoveMemory(ConnectionInfo,
                      ClientConnect->ConnectionInfo,
                      *ConnectionInfoSize);
 
        /* Free the capture buffer */
        CsrFreeCaptureBuffer(CaptureBuffer);
    }
    else
    {
        /* No connection info, just return */
        Status = STATUS_SUCCESS;
    }
 
    /* Let the caller know if this was server to server */
    DPRINT("Status was: 0x%lx. Are we in server: 0x%x\n", Status, InsideCsrProcess);
    if (ServerToServerCall) *ServerToServerCall = InsideCsrProcess;
 
    return Status;
}

Referenced by ConnectConsole(), and DllMain().

CsrFreeCaptureBuffer()

VOID NTAPI CsrFreeCaptureBuffer

(

_In_ _Frees_ptr_ PCSR_CAPTURE_BUFFER

CaptureBuffer

)

Definition at line 210 of file capture.c.

{
    /* Free it from the heap */
    RtlFreeHeap(CsrPortHeap, 0, CaptureBuffer);
}

Referenced by BaseCheckVDM(), CreateConsoleScreenBuffer(), CreateProcessInternalW(), CsrClientConnectToServer(), DefineDosDeviceW(), DeviceEventWorker(), GetConsoleFontInfo(), GetConsoleProcessList(), GetNextVDMCommand(), GetVDMCurrentDirectories(), IntAddConsoleAlias(), IntAllocConsole(), IntAttachConsole(), IntExpungeConsoleCommandHistory(), IntGetConsoleAlias(), IntGetConsoleAliases(), IntGetConsoleAliasesLength(), IntGetConsoleAliasExes(), IntGetConsoleCommandHistory(), IntGetConsoleCommandHistoryLength(), IntGetConsoleInput(), IntGetConsoleTitle(), IntReadConsole(), IntReadConsoleOutput(), IntReadConsoleOutputCode(), IntRegisterConsoleIME(), IntSetConsoleNumberOfCommands(), IntSetConsoleTitle(), IntWriteConsole(), IntWriteConsoleInput(), IntWriteConsoleOutput(), IntWriteConsoleOutputCode(), RegisterConsoleVDM(), and SetVDMCurrentDirectories().

CsrGetProcessId()

HANDLE NTAPI CsrGetProcessId

(

VOID

)

Definition at line 509 of file connect.c.

{
    return CsrProcessId;
}

Referenced by ProcessIdToHandle().

CsrIdentifyAlertableThread()

NTSTATUS NTAPI CsrIdentifyAlertableThread

(

VOID

)

Definition at line 37 of file api.c.

{
#if (NTDDI_VERSION < NTDDI_WS03)
    NTSTATUS Status;
    CSR_API_MESSAGE ApiMessage;
    PCSR_IDENTIFY_ALERTABLE_THREAD IdentifyAlertableThread;
 
    /* Set up the data for CSR */
    IdentifyAlertableThread = &ApiMessage.Data.IdentifyAlertableThread;
    IdentifyAlertableThread->Cid = NtCurrentTeb()->ClientId;
 
    /* Call it */
    Status = CsrClientCallServer(&ApiMessage,
                                 NULL,
                                 CSR_CREATE_API_NUMBER(CSRSRV_SERVERDLL_INDEX, CsrpIdentifyAlertableThread),
                                 sizeof(*IdentifyAlertableThread));
 
    /* Return to caller */
    return Status;
#else
    /* Deprecated */
    return STATUS_SUCCESS;
#endif
}

CsrNewThread()

NTSTATUS NTAPI CsrNewThread

(

VOID

)

Definition at line 26 of file api.c.

{
    /* Register the termination port to CSR's */
    return NtRegisterThreadTerminatePort(CsrApiPort);
}

Referenced by BaseThreadStartup(), and DllMain().

CsrProbeForRead()

VOID NTAPI CsrProbeForRead	(	_In_ PVOID	Address,
		_In_ ULONG	Length,
		_In_ ULONG	Alignment
	)

Definition at line 23 of file capture.c.

{
    volatile UCHAR *Pointer;
    UCHAR Data;
 
    /* Validate length */
    if (Length == 0) return;
 
    /* Validate alignment */
    if ((ULONG_PTR)Address & (Alignment - 1))
    {
        /* Raise exception if it doesn't match */
        RtlRaiseStatus(STATUS_DATATYPE_MISALIGNMENT);
    }
 
    /* Probe first byte */
    Pointer = Address;
    Data = *Pointer;
 
    /* Probe last byte */
    Pointer = (PUCHAR)Address + Length - 1;
    Data = *Pointer;
    (void)Data;
}

CsrProbeForWrite()

VOID NTAPI CsrProbeForWrite	(	_In_ PVOID	Address,
		_In_ ULONG	Length,
		_In_ ULONG	Alignment
	)

Definition at line 56 of file capture.c.

{
    volatile UCHAR *Pointer;
 
    /* Validate length */
    if (Length == 0) return;
 
    /* Validate alignment */
    if ((ULONG_PTR)Address & (Alignment - 1))
    {
        /* Raise exception if it doesn't match */
        RtlRaiseStatus(STATUS_DATATYPE_MISALIGNMENT);
    }
 
    /* Probe first byte */
    Pointer = Address;
    *Pointer = *Pointer;
 
    /* Probe last byte */
    Pointer = (PUCHAR)Address + Length - 1;
    *Pointer = *Pointer;
}

CsrSetPriorityClass()

NTSTATUS NTAPI CsrSetPriorityClass	(	_In_ HANDLE	Process,
		_Inout_ PULONG	PriorityClass
	)

Definition at line 67 of file api.c.

{
#if (NTDDI_VERSION < NTDDI_WS03)
    NTSTATUS Status;
    CSR_API_MESSAGE ApiMessage;
    PCSR_SET_PRIORITY_CLASS SetPriorityClass = &ApiMessage.Data.SetPriorityClass;
 
    /* Set up the data for CSR */
    SetPriorityClass->hProcess = Process;
    SetPriorityClass->PriorityClass = *PriorityClass;
 
    /* Call it */
    Status = CsrClientCallServer(&ApiMessage,
                                 NULL,
                                 CSR_CREATE_API_NUMBER(CSRSRV_SERVERDLL_INDEX, CsrpSetPriorityClass),
                                 sizeof(*SetPriorityClass));
 
    /* Return what we got, if requested */
    if (*PriorityClass) *PriorityClass = SetPriorityClass->PriorityClass;
 
    /* Return to caller */
    return Status;
#else
    UNREFERENCED_PARAMETER(Process);
    UNREFERENCED_PARAMETER(PriorityClass);
 
    /* Deprecated */
    return STATUS_INVALID_PARAMETER;
#endif
}