• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  *  Copyright 2004 The WebRTC Project Authors. All rights reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #ifndef RTC_BASE_SERVER_SOCKET_ADAPTERS_H_
12 #define RTC_BASE_SERVER_SOCKET_ADAPTERS_H_
13 
14 #include "rtc_base/socket_adapters.h"
15 
16 namespace rtc {
17 
18 // Interface for implementing proxy server sockets.
19 class AsyncProxyServerSocket : public BufferedReadAdapter {
20  public:
21   AsyncProxyServerSocket(AsyncSocket* socket, size_t buffer_size);
22   ~AsyncProxyServerSocket() override;
23   sigslot::signal2<AsyncProxyServerSocket*, const SocketAddress&>
24       SignalConnectRequest;
25   virtual void SendConnectResult(int err, const SocketAddress& addr) = 0;
26 };
27 
28 // Implements a socket adapter that performs the server side of a
29 // fake SSL handshake. Used when implementing a relay server that does "ssltcp".
30 class AsyncSSLServerSocket : public BufferedReadAdapter {
31  public:
32   explicit AsyncSSLServerSocket(AsyncSocket* socket);
33 
34  protected:
35   void ProcessInput(char* data, size_t* len) override;
36   RTC_DISALLOW_COPY_AND_ASSIGN(AsyncSSLServerSocket);
37 };
38 
39 // Implements a proxy server socket for the SOCKS protocol.
40 class AsyncSocksProxyServerSocket : public AsyncProxyServerSocket {
41  public:
42   explicit AsyncSocksProxyServerSocket(AsyncSocket* socket);
43 
44  private:
45   void ProcessInput(char* data, size_t* len) override;
46   void DirectSend(const ByteBufferWriter& buf);
47 
48   void HandleHello(ByteBufferReader* request);
49   void SendHelloReply(uint8_t method);
50   void HandleAuth(ByteBufferReader* request);
51   void SendAuthReply(uint8_t result);
52   void HandleConnect(ByteBufferReader* request);
53   void SendConnectResult(int result, const SocketAddress& addr) override;
54 
55   void Error(int error);
56 
57   static const int kBufferSize = 1024;
58   enum State {
59     SS_HELLO,
60     SS_AUTH,
61     SS_CONNECT,
62     SS_CONNECT_PENDING,
63     SS_TUNNEL,
64     SS_ERROR
65   };
66   State state_;
67   RTC_DISALLOW_COPY_AND_ASSIGN(AsyncSocksProxyServerSocket);
68 };
69 
70 }  // namespace rtc
71 
72 #endif  // RTC_BASE_SERVER_SOCKET_ADAPTERS_H_
73