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 RTC_DISALLOW_COPY_AND_ASSIGN(AsyncStunTCPSocket); 46 }; 47 48 } // namespace cricket 49 50 #endif // WEBRTC_P2P_BASE_ASYNCSTUNTCPSOCKET_H_ 51