WebSockets/IdServerIOHandlerWebsocket.pas at master · mvancanneyt/WebSockets · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
unit IdServerIOHandlerWebSocket;
interface
{$I wsdefines.pas}
uses
  System.Classes, IdServerIOHandlerStack, IdIOHandlerStack, IdGlobal,
  IdIOHandler, IdYarn, IdThread, IdSocketHandle, IdIOHandlerWebSocket,
  IdWebSocketTypes;

type
  TIdServerIOHandlerWebSocket = class(TIdServerIOHandlerStack, ISetWebSocketClosing)
  protected
    FOnWebSocketClosing: TOnWebSocketClosing;
    procedure InitComponent; override;
    procedure SetWebSocketClosing(const AValue: TOnWebSocketClosing);
  public
    function Accept(ASocket: TIdSocketHandle; AListenerThread: TIdThread;
      AYarn: TIdYarn): TIdIOHandler; override;
    function MakeClientIOHandler(ATheThread:TIdYarn): TIdIOHandler; override;
  end;

implementation

function TIdServerIOHandlerWebSocket.Accept(ASocket: TIdSocketHandle;
  AListenerThread: TIdThread; AYarn: TIdYarn): TIdIOHandler;
begin
  Result := inherited Accept(ASocket, AListenerThread, AYarn);
  if Result <> nil then
  begin
    (Result as TIdIOHandlerWebSocket).IsServerSide := True; // server must not mask, only client
    (Result as TIdIOHandlerWebSocket).UseNagle := False;
  end;
end;

procedure TIdServerIOHandlerWebSocket.InitComponent;
begin
  inherited InitComponent;
  IOHandlerSocketClass := TIdIOHandlerWebsocketServer;
end;

function TIdServerIOHandlerWebSocket.MakeClientIOHandler(
  ATheThread: TIdYarn): TIdIOHandler;
begin
  Result := inherited MakeClientIOHandler(ATheThread);
  if Result <> nil then
  begin
    (Result as TIdIOHandlerWebSocket).IsServerSide := True; // server must not mask, only client
    (Result as TIdIOHandlerWebSocket).UseNagle := False;
  end;
end;

procedure TIdServerIOHandlerWebSocket.SetWebSocketClosing(
  const AValue: TOnWebSocketClosing);
begin
  FOnWebSocketClosing := AValue;
end;

end.