logport.c

来自「一个类似windows」· C语言 代码 · 共 221 行

/*
 * PROJECT:          ReactOS kernel
 * LICENSE:          GPL - See COPYING in the top level directory
 * FILE:             services/eventlog/logport.c
 * PURPOSE:          Event logging service
 * COPYRIGHT:        Copyright 2002 Eric Kohl
 *                   Copyright 2005 Saveliy Tretiakov 
 */

/* INCLUDES *****************************************************************/

#include "eventlog.h"

/* GLOBALS ******************************************************************/

HANDLE ConnectPortHandle = NULL;
HANDLE MessagePortHandle = NULL;
extern PLOGFILE SystemLog;
extern HANDLE MyHeap;
extern BOOL onLiveCD;

/* FUNCTIONS ****************************************************************/

NTSTATUS STDCALL PortThreadRoutine(PVOID Param)
{
    NTSTATUS Status = STATUS_SUCCESS;
    
    Status = InitLogPort();
    if(!NT_SUCCESS(Status))
        return Status;

    while(NT_SUCCESS(Status))
    {
        Status = ProcessPortMessage();
    }

    if(ConnectPortHandle != NULL)
        NtClose(ConnectPortHandle);

    if(MessagePortHandle != NULL)
        NtClose(MessagePortHandle);

    return Status;
}

NTSTATUS InitLogPort(VOID)
{
    OBJECT_ATTRIBUTES ObjectAttributes;
    UNICODE_STRING PortName;
    PORT_MESSAGE Request;
    NTSTATUS Status;
    
    ConnectPortHandle = NULL;
    MessagePortHandle = NULL;

    RtlInitUnicodeString(&PortName, L"\\ErrorLogPort");
    InitializeObjectAttributes(
                &ObjectAttributes,
                &PortName,
                0,
                NULL,
                NULL);

    Status = NtCreatePort(
                &ConnectPortHandle,
                &ObjectAttributes,
                0,
                0x100,
                0x2000);
                
    if(!NT_SUCCESS(Status))
    {
        DPRINT1("NtCreatePort() failed (Status %lx)\n", Status);
        goto ByeBye;
    }

    Status = NtListenPort(ConnectPortHandle, &Request);
  
    if(!NT_SUCCESS(Status))
    {
        DPRINT1("NtListenPort() failed (Status %lx)\n", Status);
        goto ByeBye;
    }

    Status = NtAcceptConnectPort(
                    &MessagePortHandle,
                    ConnectPortHandle,
                    NULL,
                    TRUE,
                    NULL,
                    NULL);
                    
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtAcceptConnectPort() failed (Status %lx)\n", Status);
        goto ByeBye;
    }

    Status = NtCompleteConnectPort(MessagePortHandle);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtCompleteConnectPort() failed (Status %lx)\n", Status);
        goto ByeBye;
    }

ByeBye:
    if (!NT_SUCCESS(Status))
    {
        if(ConnectPortHandle != NULL)
            NtClose(ConnectPortHandle);

        if(MessagePortHandle != NULL)
            NtClose(MessagePortHandle);
    }
    return Status;
}


NTSTATUS ProcessPortMessage(VOID)
{
    IO_ERROR_LPC Request;
    PIO_ERROR_LOG_MESSAGE Message;
	PEVENTLOGRECORD pRec;
    ULONG ulRecNum, ulRecSize ;
    NTSTATUS Status;
	PLOGFILE SystemLog = NULL;
    
    DPRINT1("ProcessPortMessage() called\n");

    Status = STATUS_SUCCESS;

    while(TRUE)
    {
        Status = NtReplyWaitReceivePort(
                        MessagePortHandle,
                        0,
                        NULL,
                        &Request.Header);
                        
        if(!NT_SUCCESS(Status))
        {
            DPRINT1("NtReplyWaitReceivePort() failed (Status %lx)\n", Status);
            break;
        }

        DPRINT("Received message\n");

        if(Request.Header.u2.s2.Type == LPC_PORT_CLOSED)
        {
            DPRINT("Port closed\n");
            return STATUS_SUCCESS;
        }
      
        if(Request.Header.u2.s2.Type == LPC_REQUEST)
        {
            DPRINT("Received request\n");
        }
        else if (Request.Header.u2.s2.Type == LPC_DATAGRAM)
        {
            DPRINT("Received datagram\n");
            Message = (PIO_ERROR_LOG_MESSAGE)&Request.Message;
			ulRecNum = SystemLog ? SystemLog->Header.NextRecord : 0;

			ulRecSize = LogfBuildNewRecord(NULL,
				ulRecNum,
				Message->Type,
				Message->EntryData.EventCategory,
				Message->EntryData.ErrorCode,
				(WCHAR*)(((PBYTE)Message)+Message->DriverNameOffset),
				L"MyComputer", /* FIXME */
				0,
				NULL,
				Message->EntryData.NumberOfStrings,
				(WCHAR*)(((PBYTE)Message)+Message->EntryData.StringOffset),
				Message->EntryData.DumpDataSize,
				(LPVOID)(((PBYTE)Message)
					+sizeof(IO_ERROR_LOG_PACKET)-sizeof(ULONG)));

			DPRINT("ulRecSize = %d\n", ulRecSize);

			pRec = HeapAlloc(MyHeap, 0, ulRecSize);

			if(pRec == NULL)
			{
				DPRINT("Can't allocate heap!\n");
				return STATUS_NO_MEMORY;
			}

			LogfBuildNewRecord((PBYTE)pRec,
				ulRecNum,
				Message->Type,
				Message->EntryData.EventCategory,
				Message->EntryData.ErrorCode,
				(WCHAR*)(((PBYTE)Message)+Message->DriverNameOffset),
				L"MyComputer", /* FIXME */
				0,
				NULL,
				Message->EntryData.NumberOfStrings,
				(WCHAR*)(((PBYTE)Message)+Message->EntryData.StringOffset),
				Message->EntryData.DumpDataSize,
				(LPVOID)(((PBYTE)Message)
					+sizeof(IO_ERROR_LOG_PACKET)-sizeof(ULONG)));
			
			DPRINT("\n --- EVENTLOG RECORD ---\n");
			PRINT_RECORD(pRec);
			DPRINT("\n");
			
			if(!onLiveCD && SystemLog)
			{
				if(!LogfWriteData(SystemLog, ulRecSize, (PBYTE)pRec))
					DPRINT("LogfWriteData failed!\n");
				else DPRINT("Data written to Log!\n");
			}

			HeapFree(MyHeap, 0, pRec);
        }
    }
    return Status;
}