1 /*
2  *  Copyright 2013 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 WEBRTC_P2P_BASE_ASYNCSTUNTCPSOCKET_H_
12 #define WEBRTC_P2P_BASE_ASYNCSTUNTCPSOCKET_H_
13 
14 #include "webrtc/base/asynctcpsocket.h"
15 #include "webrtc/base/scoped_ptr.h"
16 #include "webrtc/base/socketfactory.h"
17 
18 namespace cricket {
19 
20 class AsyncStunTCPSocket : public rtc::AsyncTCPSocketBase {
21  public:
22   // Binds and connects |socket| and creates AsyncTCPSocket for
23   // it. Takes ownership of |socket|. Returns NULL if bind() or
24   // connect() fail (|socket| is destroyed in that case).
25   static AsyncStunTCPSocket* Create(
26       rtc::AsyncSocket* socket,
27       const rtc::SocketAddress& bind_address,
28       const rtc::SocketAddress& remote_address);
29 
30   AsyncStunTCPSocket(rtc::AsyncSocket* socket, bool listen);
~AsyncStunTCPSocket()31   virtual ~AsyncStunTCPSocket() {}
32 
33   virtual int Send(const void* pv, size_t cb,
34                    const rtc::PacketOptions& options);
35   virtual void ProcessInput(char* data, size_t* len);
36   virtual void HandleIncomingConnection(rtc::AsyncSocket* socket);
37 
38  private:
39   // This method returns the message hdr + length written in the header.
40   // This method also returns the number of padding bytes needed/added to the
41   // turn message. |pad_bytes| should be used only when |is_turn| is true.
42   size_t GetExpectedLength(const void* data, size_t len,
43                            int* pad_bytes);
44 
45   DISALLOW_EVIL_CONSTRUCTORS(AsyncStunTCPSocket);
46 };
47 
48 }  // namespace cricket
49 
50 #endif  // WEBRTC_P2P_BASE_ASYNCSTUNTCPSOCKET_H_
51