unit X2Log.Observer.NamedPipe;

interface
uses
  X2Log.Intf,
  X2Log.Observer.Custom,
  X2Log.Observer.CustomThreaded;


type
  TX2LogNamedPipeObserver = class(TX2LogCustomThreadedObserver)
  private
    FPipeName: string;
  protected
    function CreateWorkerThread: TX2LogObserverWorkerThread; override;
  public
    constructor Create(const APipeName: string; ALogLevels: TX2LogLevels = X2LogLevelsDefault);
  end;


implementation
uses
  System.Classes,
  System.Generics.Collections,
  System.SyncObjs,
  System.SysUtils,
  System.Types,
  Winapi.Windows,

  X2Log.Details.Registry,
  X2Log.Intf.NamedPipe,
  X2Log.Util.Stream;


type
  EX2LogSilentException = class(Exception);
  EX2LogPipeDisconnected = class(EX2LogSilentException);

  TX2LogNamedPipeClientState = (Listening, Connected, Writing);

  TX2LogNamedPipeClient = class(TObject)
  private
    FOverlapped: TOverlapped;
    FPipeHandle: THandle;
    FState: TX2LogNamedPipeClientState;
    FOverlappedEvent: TEvent;
    FWriteQueue: TObjectQueue<TX2LogQueueEntry>;
    FWriteBuffer: TMemoryStream;
  protected
    function DoSend(AEntry: TX2LogQueueEntry): Boolean;
    procedure ClearWriteBuffer;

    property PipeHandle: THandle read FPipeHandle;
    property WriteBuffer: TMemoryStream read FWriteBuffer;
    property WriteQueue: TObjectQueue<TX2LogQueueEntry> read FWriteQueue;
  public
    constructor Create(APipe: THandle);
    destructor Destroy; override;

    procedure Send(AEntry: TX2LogQueueEntry);
    procedure SendNext;

    procedure Disconnect;

    property Pipe: THandle read FPipeHandle;
    property Overlapped: TOverlapped read FOverlapped;
    property OverlappedEvent: TEvent read FOverlappedEvent;
    property State: TX2LogNamedPipeClientState read FState write FState;
  end;


  TX2LogNamedPipeWorkerThread = class(TX2LogObserverWorkerThread)
  private
    FClients: TObjectList<TX2LogNamedPipeClient>;
    FPipeName: string;
  protected
    procedure WaitForEntry; override;
    procedure ProcessEntry(AEntry: TX2LogQueueEntry); override;
    procedure ProcessClientEvent(AClientIndex: Integer);

    procedure AddListener;
    procedure RemoveClient(AClientIndex: Integer);
    procedure Setup; override;
    procedure Cleanup; override;

    property Clients: TObjectList<TX2LogNamedPipeClient> read FClients;
    property PipeName: string read FPipeName;
  public
    constructor Create(const APipeName: string);
    destructor Destroy; override;
  end;



  { Someone went through a lot of trouble to win at Scrabble... }
  function ConvertStringSecurityDescriptorToSecurityDescriptorW(StringSecurityDescriptor: PWideChar;
                                                                StringSDRevision: DWORD;
                                                                SecurityDescriptor: PSECURITY_DESCRIPTOR;
                                                                SecurityDescriptorSize: PULONG): BOOL; stdcall; external advapi32;

const
  SDDL_REVISION_1 = 1;


  PipeNamePrefix = '\\.\pipe\X2Log.';



{ TX2LogNamedPipeObserver }
constructor TX2LogNamedPipeObserver.Create(const APipeName: string; ALogLevels: TX2LogLevels);
begin
  FPipeName := APipeName;

  inherited Create(ALogLevels);
end;


function TX2LogNamedPipeObserver.CreateWorkerThread: TX2LogObserverWorkerThread;
begin
  Result := TX2LogNamedPipeWorkerThread.Create(FPipeName);
end;


{ TX2LogNamedPipeClient }
constructor TX2LogNamedPipeClient.Create(APipe: THandle);
begin
  inherited Create;

  FPipeHandle := APipe;
  FState := Listening;

  FOverlappedEvent := TEvent.Create(nil, False, False, '');
  FOverlapped.hEvent := FOverlappedEvent.Handle;
end;


destructor TX2LogNamedPipeClient.Destroy;
begin
  Disconnect;

  FreeAndNil(FOverlappedEvent);
  ClearWriteBuffer;

  inherited Destroy;
end;


procedure TX2LogNamedPipeClient.Send(AEntry: TX2LogQueueEntry);
begin
  if not Assigned(WriteBuffer) then
    DoSend(AEntry)
  else
  begin
    if not Assigned(WriteQueue) then
      FWriteQueue := TObjectQueue<TX2LogQueueEntry>.Create(True);

    WriteQueue.Enqueue(TX2LogQueueEntry.Create(AEntry));
  end;
end;


procedure TX2LogNamedPipeClient.SendNext;
var
  entry: TX2LogQueueEntry;

begin
  ClearWriteBuffer;

  if Assigned(WriteQueue) then
  begin
    while WriteQueue.Count > 0 do
    begin
      entry := WriteQueue.Extract;
      try
        { Returns False when IO is pending }
        if not DoSend(entry) then
          break;
      finally
        FreeAndNil(entry);
      end;
    end;
  end;
end;


procedure TX2LogNamedPipeClient.Disconnect;
begin
  if PipeHandle <> INVALID_HANDLE_VALUE then
  begin
    CancelIo(PipeHandle);
    DisconnectNamedPipe(PipeHandle);
    CloseHandle(PipeHandle);

    FPipeHandle := INVALID_HANDLE_VALUE;
  end;
end;


function TX2LogNamedPipeClient.DoSend(AEntry: TX2LogQueueEntry): Boolean;
var
  header: TX2LogMessageHeader;
  bytesWritten: Cardinal;
  lastError: Cardinal;
  detailsStream: TMemoryStream;
  serializerIID: TGUID;
  serializer: IX2LogDetailsSerializer;

begin
  ClearWriteBuffer;

  FWriteBuffer := TMemoryStream.Create;

  { Header }
  header.ID := X2LogMessageHeader;
  header.Version := X2LogMessageVersion;
  header.Size := SizeOf(header);
  header.Level := AEntry.Level;
  header.DateTime := AEntry.DateTime;

  WriteBuffer.WriteBuffer(header, SizeOf(header));

  { Category }
  TStreamUtil.WriteString(WriteBuffer, AEntry.Category);

  { Message }
  TStreamUtil.WriteString(WriteBuffer, AEntry.Message);

  { Details }
  if TX2LogDetailsRegistry.GetSerializer(AEntry.Details, serializer) then
  begin
    detailsStream := TMemoryStream.Create;
    try
      serializer.Serialize(AEntry.Details, detailsStream);

      serializerIID := AEntry.Details.SerializerIID;
      WriteBuffer.WriteBuffer(serializerIID, SizeOf(TGUID));

      TStreamUtil.WriteCardinal(WriteBuffer, detailsStream.Size);
      WriteBuffer.CopyFrom(detailsStream, 0);
    finally
      FreeAndNil(detailsStream);
    end;
  end else
  begin
    serializerIID := GUID_NULL;
    WriteBuffer.WriteBuffer(serializerIID, SizeOf(TGUID));
  end;


  Result := WriteFile(Pipe, WriteBuffer.Memory^, WriteBuffer.Size, bytesWritten, @Overlapped);
  if not Result then
  begin
    lastError := GetLastError;
    if lastError in [ERROR_NO_DATA, ERROR_PIPE_NOT_CONNECTED] then
      raise EX2LogPipeDisconnected.Create('Client disconnected');

    if lastError = ERROR_IO_PENDING then
      State := Writing
    else
    begin
      ClearWriteBuffer;
      RaiseLastOSError;
    end;
  end else
  begin
    ClearWriteBuffer;
    State := Connected;
  end;
end;


procedure TX2LogNamedPipeClient.ClearWriteBuffer;
begin
  FreeAndNil(FWriteBuffer);
end;



{ TX2LogNamedPipeWorkerThread }
constructor TX2LogNamedPipeWorkerThread.Create(const APipeName: string);
begin
  FPipeName := APipeName;
  FClients := TObjectList<TX2LogNamedPipeClient>.Create(True);

  inherited Create;
end;


destructor TX2LogNamedPipeWorkerThread.Destroy;
begin
  inherited Destroy;

  FreeAndNil(FClients);
end;


procedure TX2LogNamedPipeWorkerThread.Setup;
begin
  inherited Setup;

  AddListener;
end;


procedure TX2LogNamedPipeWorkerThread.Cleanup;
var
  client: TX2LogNamedPipeClient;

begin
  for client in Clients do
    client.Disconnect;

  inherited Cleanup;
end;


procedure TX2LogNamedPipeWorkerThread.WaitForEntry;
var
  eventHandles: array of THandle;
  clientIndex: Integer;
  waitResult: Cardinal;

begin
  repeat
    SetLength(eventHandles, Clients.Count + 1);
    for clientIndex := 0 to Pred(Clients.Count) do
      eventHandles[clientIndex] := Clients[clientIndex].OverlappedEvent.Handle;

    eventHandles[Clients.Count] := LogQueueSignal.Handle;

    waitResult := WaitForMultipleObjects(Length(eventHandles), @eventHandles[0], False, INFINITE);

    { WAIT_OBJECT_0 = 0, no check for minimum bounds since it's an unsigned variable ("Comparison always results to true") }
    if waitResult < Cardinal(WAIT_OBJECT_0 + High(eventHandles)) then
    begin
      { Connect or write I/O completed }
      clientIndex := waitResult - WAIT_OBJECT_0;
      if (clientIndex >= 0) and (clientIndex < Clients.Count) then
        ProcessClientEvent(clientIndex);
    end else if waitResult = Cardinal(WAIT_OBJECT_0 + High(eventHandles)) then
    begin
      { Entry queued }
      break;
    end else if (waitResult >= WAIT_ABANDONED_0) and (waitResult <= Cardinal(WAIT_ABANDONED_0 + High(eventHandles))) then
    begin
      { Client event abandoned }
      clientIndex := waitResult - WAIT_ABANDONED_0;
      if (clientIndex >= 0) and (clientIndex < Clients.Count) then
        RemoveClient(clientIndex)
      else if clientIndex = Clients.Count then
        Terminate;
    end else if waitResult = WAIT_FAILED then
      RaiseLastOSError;
  until False;
end;


procedure TX2LogNamedPipeWorkerThread.ProcessEntry(AEntry: TX2LogQueueEntry);
var
  clientIndex: Integer;
  client: TX2LogNamedPipeClient;

begin
  { Broadcast to connected clients }
  for clientIndex := Pred(Clients.Count) downto 0 do
  begin
    client := Clients[clientIndex];

    if client.State <> Listening then
    try
      client.Send(AEntry);
    except
      on E:EX2LogPipeDisconnected do
        RemoveClient(clientIndex);
    end;
  end;
end;


procedure TX2LogNamedPipeWorkerThread.ProcessClientEvent(AClientIndex: Integer);
var
  client: TX2LogNamedPipeClient;
  bytesTransferred: Cardinal;

begin
  client := Clients[AClientIndex];

  case client.State of
    Listening:
      { Client connected }
      if GetOverlappedResult(client.Pipe, client.Overlapped, bytesTransferred, False) then
      begin
        client.State := Connected;
        AddListener;
      end else
        RemoveClient(AClientIndex);

    Writing:
      { Write operation completed }
      if GetOverlappedResult(client.Pipe, client.Overlapped, bytesTransferred, False) and
         (bytesTransferred > 0) then
      begin
        try
          client.SendNext;
        except
          on E:EX2LogPipeDisconnected do
            RemoveClient(AClientIndex);
        end;
      end else
        RemoveClient(AClientIndex);
  end;
end;


procedure TX2LogNamedPipeWorkerThread.AddListener;
const
  BufferSize = 4096;
  DefaultTimeout = 5000;

var
  security: TSecurityAttributes;
  pipe: THandle;
  client: TX2LogNamedPipeClient;

begin
  FillChar(security, SizeOf(security), 0);
  security.nLength := SizeOf(security);
  security.bInheritHandle := False;

  pipe := INVALID_HANDLE_VALUE;

  { Thanks to: http://www.osronline.com/showthread.cfm?link=204207
          and: http://www.netid.washington.edu/documentation/domains/sddl.aspx

      0x12019f =
         0x00100000 - SYNCHRONIZE
         0x00020000 - READ_CONTROL
         0x00000100 - FILE_WRITE_ATTRIBUTES
         0x00000080 - FILE_READ_ATTRIBUTES
         0x00000010 - FILE_WRITE_EA
         0x00000008 - FILE_READ_EA
         0x00000004 - FILE_CREATE_PIPE_INSTANCE
         0x00000002 - FILE_WRITE_DATA
         0x00000001 - FILE_READ_DATA }
  if ConvertStringSecurityDescriptorToSecurityDescriptorW('D:' +                  // Discretionary ACL
                                                          '(D;;FA;;;NU)' +        // Deny file all access (FA) to network user access (NU)
                                                          '(A;;0x12019f;;;WD)',   // Allow permissions for everyone (WD)
                                                          SDDL_REVISION_1,
                                                          @security.lpSecurityDescriptor,
                                                          nil) then
  begin
    try
      pipe := CreateNamedPipe(PChar(PipeNamePrefix + PipeName), PIPE_ACCESS_OUTBOUND or FILE_FLAG_OVERLAPPED,
                              PIPE_TYPE_MESSAGE or PIPE_READMODE_MESSAGE or PIPE_WAIT, PIPE_UNLIMITED_INSTANCES,
                              BufferSize, BufferSize, DefaultTimeout, @security);
    finally
      LocalFree(HLOCAL(security.lpSecurityDescriptor));
    end;
  end else
    RaiseLastOSError;

  if pipe <> INVALID_HANDLE_VALUE then
  begin
    client := TX2LogNamedPipeClient.Create(pipe);

    if not ConnectNamedPipe(client.Pipe, @client.Overlapped) then
    begin
      case GetLastError of
        ERROR_IO_PENDING:
          Clients.Add(client);

        ERROR_PIPE_CONNECTED:
          begin
            client.State := Connected;
            Clients.Add(client);

            AddListener;
          end;
      else
        { Error occured }
        FreeAndNil(client);
      end;
    end;
  end;
end;


procedure TX2LogNamedPipeWorkerThread.RemoveClient(AClientIndex: Integer);
begin
  Clients.Delete(AClientIndex);
end;

end.