1 /*
2 * Copyright 2010 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 #include "rtc_base/socket_stream.h"
12
13 #include "rtc_base/checks.h"
14 #include "rtc_base/socket.h"
15
16 namespace rtc {
17
SocketStream(Socket * socket)18 SocketStream::SocketStream(Socket* socket) : socket_(nullptr) {
19 Attach(socket);
20 }
21
~SocketStream()22 SocketStream::~SocketStream() {
23 delete socket_;
24 }
25
Attach(Socket * socket)26 void SocketStream::Attach(Socket* socket) {
27 if (socket_)
28 delete socket_;
29 socket_ = socket;
30 if (socket_) {
31 socket_->SignalConnectEvent.connect(this, &SocketStream::OnConnectEvent);
32 socket_->SignalReadEvent.connect(this, &SocketStream::OnReadEvent);
33 socket_->SignalWriteEvent.connect(this, &SocketStream::OnWriteEvent);
34 socket_->SignalCloseEvent.connect(this, &SocketStream::OnCloseEvent);
35 }
36 }
37
Detach()38 Socket* SocketStream::Detach() {
39 Socket* socket = socket_;
40 if (socket_) {
41 socket_->SignalConnectEvent.disconnect(this);
42 socket_->SignalReadEvent.disconnect(this);
43 socket_->SignalWriteEvent.disconnect(this);
44 socket_->SignalCloseEvent.disconnect(this);
45 socket_ = nullptr;
46 }
47 return socket;
48 }
49
GetState() const50 StreamState SocketStream::GetState() const {
51 RTC_DCHECK(socket_ != nullptr);
52 switch (socket_->GetState()) {
53 case Socket::CS_CONNECTED:
54 return SS_OPEN;
55 case Socket::CS_CONNECTING:
56 return SS_OPENING;
57 case Socket::CS_CLOSED:
58 default:
59 return SS_CLOSED;
60 }
61 }
62
Read(rtc::ArrayView<uint8_t> buffer,size_t & read,int & error)63 StreamResult SocketStream::Read(rtc::ArrayView<uint8_t> buffer,
64 size_t& read,
65 int& error) {
66 RTC_DCHECK(socket_ != nullptr);
67 int result = socket_->Recv(buffer.data(), buffer.size(), nullptr);
68 if (result < 0) {
69 if (socket_->IsBlocking())
70 return SR_BLOCK;
71 error = socket_->GetError();
72 return SR_ERROR;
73 }
74 if ((result > 0) || (buffer.size() == 0)) {
75 read = result;
76 return SR_SUCCESS;
77 }
78 return SR_EOS;
79 }
80
Write(rtc::ArrayView<const uint8_t> data,size_t & written,int & error)81 StreamResult SocketStream::Write(rtc::ArrayView<const uint8_t> data,
82 size_t& written,
83 int& error) {
84 RTC_DCHECK(socket_ != nullptr);
85 int result = socket_->Send(data.data(), data.size());
86 if (result < 0) {
87 if (socket_->IsBlocking())
88 return SR_BLOCK;
89 error = socket_->GetError();
90 return SR_ERROR;
91 }
92 written = result;
93 return SR_SUCCESS;
94 }
95
Close()96 void SocketStream::Close() {
97 RTC_DCHECK(socket_ != nullptr);
98 socket_->Close();
99 }
100
OnConnectEvent(Socket * socket)101 void SocketStream::OnConnectEvent(Socket* socket) {
102 RTC_DCHECK(socket == socket_);
103 SignalEvent(this, SE_OPEN | SE_READ | SE_WRITE, 0);
104 }
105
OnReadEvent(Socket * socket)106 void SocketStream::OnReadEvent(Socket* socket) {
107 RTC_DCHECK(socket == socket_);
108 SignalEvent(this, SE_READ, 0);
109 }
110
OnWriteEvent(Socket * socket)111 void SocketStream::OnWriteEvent(Socket* socket) {
112 RTC_DCHECK(socket == socket_);
113 SignalEvent(this, SE_WRITE, 0);
114 }
115
OnCloseEvent(Socket * socket,int err)116 void SocketStream::OnCloseEvent(Socket* socket, int err) {
117 RTC_DCHECK(socket == socket_);
118 SignalEvent(this, SE_CLOSE, err);
119 }
120
121 } // namespace rtc
122