1 /*
2  *  Copyright 2005 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_BASE_SOCKETSTREAM_H_
12 #define WEBRTC_BASE_SOCKETSTREAM_H_
13 
14 #include "webrtc/base/asyncsocket.h"
15 #include "webrtc/base/common.h"
16 #include "webrtc/base/stream.h"
17 
18 namespace rtc {
19 
20 ///////////////////////////////////////////////////////////////////////////////
21 
22 class SocketStream : public StreamInterface, public sigslot::has_slots<> {
23  public:
24   explicit SocketStream(AsyncSocket* socket);
25   ~SocketStream() override;
26 
27   void Attach(AsyncSocket* socket);
28   AsyncSocket* Detach();
29 
GetSocket()30   AsyncSocket* GetSocket() { return socket_; }
31 
32   StreamState GetState() const override;
33 
34   StreamResult Read(void* buffer,
35                     size_t buffer_len,
36                     size_t* read,
37                     int* error) override;
38 
39   StreamResult Write(const void* data,
40                      size_t data_len,
41                      size_t* written,
42                      int* error) override;
43 
44   void Close() override;
45 
46  private:
47   void OnConnectEvent(AsyncSocket* socket);
48   void OnReadEvent(AsyncSocket* socket);
49   void OnWriteEvent(AsyncSocket* socket);
50   void OnCloseEvent(AsyncSocket* socket, int err);
51 
52   AsyncSocket* socket_;
53 
54   RTC_DISALLOW_COPY_AND_ASSIGN(SocketStream);
55 };
56 
57 ///////////////////////////////////////////////////////////////////////////////
58 
59 }  // namespace rtc
60 
61 #endif  // WEBRTC_BASE_SOCKETSTREAM_H_
62