Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(226)

Side by Side Diff: webrtc/base/proxyserver.cc

Issue 2620303003: Replace ASSERT by RTC_DCHECK in all non-test code. (Closed)
Patch Set: Address final nits. Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « webrtc/base/proxydetect.cc ('k') | webrtc/base/rollingaccumulator.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 /* 1 /*
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 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 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 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include "webrtc/base/proxyserver.h" 11 #include "webrtc/base/proxyserver.h"
12 12
13 #include <algorithm> 13 #include <algorithm>
14
15 #include "webrtc/base/checks.h"
14 #include "webrtc/base/socketfactory.h" 16 #include "webrtc/base/socketfactory.h"
15 17
16 namespace rtc { 18 namespace rtc {
17 19
18 // ProxyServer 20 // ProxyServer
19 ProxyServer::ProxyServer( 21 ProxyServer::ProxyServer(
20 SocketFactory* int_factory, const SocketAddress& int_addr, 22 SocketFactory* int_factory, const SocketAddress& int_addr,
21 SocketFactory* ext_factory, const SocketAddress& ext_ip) 23 SocketFactory* ext_factory, const SocketAddress& ext_ip)
22 : ext_factory_(ext_factory), ext_ip_(ext_ip.ipaddr(), 0), // strip off port 24 : ext_factory_(ext_factory), ext_ip_(ext_ip.ipaddr(), 0), // strip off port
23 server_socket_(int_factory->CreateAsyncSocket(int_addr.family(), 25 server_socket_(int_factory->CreateAsyncSocket(int_addr.family(),
24 SOCK_STREAM)) { 26 SOCK_STREAM)) {
25 ASSERT(server_socket_.get() != NULL); 27 RTC_DCHECK(server_socket_.get() != NULL);
26 ASSERT(int_addr.family() == AF_INET || int_addr.family() == AF_INET6); 28 RTC_DCHECK(int_addr.family() == AF_INET || int_addr.family() == AF_INET6);
27 server_socket_->Bind(int_addr); 29 server_socket_->Bind(int_addr);
28 server_socket_->Listen(5); 30 server_socket_->Listen(5);
29 server_socket_->SignalReadEvent.connect(this, &ProxyServer::OnAcceptEvent); 31 server_socket_->SignalReadEvent.connect(this, &ProxyServer::OnAcceptEvent);
30 } 32 }
31 33
32 ProxyServer::~ProxyServer() { 34 ProxyServer::~ProxyServer() {
33 for (BindingList::iterator it = bindings_.begin(); 35 for (BindingList::iterator it = bindings_.begin();
34 it != bindings_.end(); ++it) { 36 it != bindings_.end(); ++it) {
35 delete (*it); 37 delete (*it);
36 } 38 }
37 } 39 }
38 40
39 SocketAddress ProxyServer::GetServerAddress() { 41 SocketAddress ProxyServer::GetServerAddress() {
40 return server_socket_->GetLocalAddress(); 42 return server_socket_->GetLocalAddress();
41 } 43 }
42 44
43 void ProxyServer::OnAcceptEvent(AsyncSocket* socket) { 45 void ProxyServer::OnAcceptEvent(AsyncSocket* socket) {
44 ASSERT(socket != NULL && socket == server_socket_.get()); 46 RTC_DCHECK(socket != NULL && socket == server_socket_.get());
45 AsyncSocket* int_socket = socket->Accept(NULL); 47 AsyncSocket* int_socket = socket->Accept(NULL);
46 AsyncProxyServerSocket* wrapped_socket = WrapSocket(int_socket); 48 AsyncProxyServerSocket* wrapped_socket = WrapSocket(int_socket);
47 AsyncSocket* ext_socket = ext_factory_->CreateAsyncSocket(ext_ip_.family(), 49 AsyncSocket* ext_socket = ext_factory_->CreateAsyncSocket(ext_ip_.family(),
48 SOCK_STREAM); 50 SOCK_STREAM);
49 if (ext_socket) { 51 if (ext_socket) {
50 ext_socket->Bind(ext_ip_); 52 ext_socket->Bind(ext_ip_);
51 bindings_.push_back(new ProxyBinding(wrapped_socket, ext_socket)); 53 bindings_.push_back(new ProxyBinding(wrapped_socket, ext_socket));
52 } else { 54 } else {
53 LOG(LS_ERROR) << "Unable to create external socket on proxy accept event"; 55 LOG(LS_ERROR) << "Unable to create external socket on proxy accept event";
54 } 56 }
(...skipping 20 matching lines...) Expand all
75 &ProxyBinding::OnExternalConnect); 77 &ProxyBinding::OnExternalConnect);
76 ext_socket_->SignalReadEvent.connect(this, &ProxyBinding::OnExternalRead); 78 ext_socket_->SignalReadEvent.connect(this, &ProxyBinding::OnExternalRead);
77 ext_socket_->SignalWriteEvent.connect(this, &ProxyBinding::OnExternalWrite); 79 ext_socket_->SignalWriteEvent.connect(this, &ProxyBinding::OnExternalWrite);
78 ext_socket_->SignalCloseEvent.connect(this, &ProxyBinding::OnExternalClose); 80 ext_socket_->SignalCloseEvent.connect(this, &ProxyBinding::OnExternalClose);
79 } 81 }
80 82
81 ProxyBinding::~ProxyBinding() = default; 83 ProxyBinding::~ProxyBinding() = default;
82 84
83 void ProxyBinding::OnConnectRequest(AsyncProxyServerSocket* socket, 85 void ProxyBinding::OnConnectRequest(AsyncProxyServerSocket* socket,
84 const SocketAddress& addr) { 86 const SocketAddress& addr) {
85 ASSERT(!connected_ && ext_socket_.get() != NULL); 87 RTC_DCHECK(!connected_ && ext_socket_.get() != NULL);
86 ext_socket_->Connect(addr); 88 ext_socket_->Connect(addr);
87 // TODO: handle errors here 89 // TODO: handle errors here
88 } 90 }
89 91
90 void ProxyBinding::OnInternalRead(AsyncSocket* socket) { 92 void ProxyBinding::OnInternalRead(AsyncSocket* socket) {
91 Read(int_socket_.get(), &out_buffer_); 93 Read(int_socket_.get(), &out_buffer_);
92 Write(ext_socket_.get(), &out_buffer_); 94 Write(ext_socket_.get(), &out_buffer_);
93 } 95 }
94 96
95 void ProxyBinding::OnInternalWrite(AsyncSocket* socket) { 97 void ProxyBinding::OnInternalWrite(AsyncSocket* socket) {
96 Write(int_socket_.get(), &in_buffer_); 98 Write(int_socket_.get(), &in_buffer_);
97 } 99 }
98 100
99 void ProxyBinding::OnInternalClose(AsyncSocket* socket, int err) { 101 void ProxyBinding::OnInternalClose(AsyncSocket* socket, int err) {
100 Destroy(); 102 Destroy();
101 } 103 }
102 104
103 void ProxyBinding::OnExternalConnect(AsyncSocket* socket) { 105 void ProxyBinding::OnExternalConnect(AsyncSocket* socket) {
104 ASSERT(socket != NULL); 106 RTC_DCHECK(socket != NULL);
105 connected_ = true; 107 connected_ = true;
106 int_socket_->SendConnectResult(0, socket->GetRemoteAddress()); 108 int_socket_->SendConnectResult(0, socket->GetRemoteAddress());
107 } 109 }
108 110
109 void ProxyBinding::OnExternalRead(AsyncSocket* socket) { 111 void ProxyBinding::OnExternalRead(AsyncSocket* socket) {
110 Read(ext_socket_.get(), &in_buffer_); 112 Read(ext_socket_.get(), &in_buffer_);
111 Write(int_socket_.get(), &in_buffer_); 113 Write(int_socket_.get(), &in_buffer_);
112 } 114 }
113 115
114 void ProxyBinding::OnExternalWrite(AsyncSocket* socket) { 116 void ProxyBinding::OnExternalWrite(AsyncSocket* socket) {
115 Write(ext_socket_.get(), &out_buffer_); 117 Write(ext_socket_.get(), &out_buffer_);
116 } 118 }
117 119
118 void ProxyBinding::OnExternalClose(AsyncSocket* socket, int err) { 120 void ProxyBinding::OnExternalClose(AsyncSocket* socket, int err) {
119 if (!connected_) { 121 if (!connected_) {
120 int_socket_->SendConnectResult(err, SocketAddress()); 122 int_socket_->SendConnectResult(err, SocketAddress());
121 } 123 }
122 Destroy(); 124 Destroy();
123 } 125 }
124 126
125 void ProxyBinding::Read(AsyncSocket* socket, FifoBuffer* buffer) { 127 void ProxyBinding::Read(AsyncSocket* socket, FifoBuffer* buffer) {
126 // Only read if the buffer is empty. 128 // Only read if the buffer is empty.
127 ASSERT(socket != NULL); 129 RTC_DCHECK(socket != NULL);
128 size_t size; 130 size_t size;
129 int read; 131 int read;
130 if (buffer->GetBuffered(&size) && size == 0) { 132 if (buffer->GetBuffered(&size) && size == 0) {
131 void* p = buffer->GetWriteBuffer(&size); 133 void* p = buffer->GetWriteBuffer(&size);
132 read = socket->Recv(p, size, nullptr); 134 read = socket->Recv(p, size, nullptr);
133 buffer->ConsumeWriteBuffer(std::max(read, 0)); 135 buffer->ConsumeWriteBuffer(std::max(read, 0));
134 } 136 }
135 } 137 }
136 138
137 void ProxyBinding::Write(AsyncSocket* socket, FifoBuffer* buffer) { 139 void ProxyBinding::Write(AsyncSocket* socket, FifoBuffer* buffer) {
138 ASSERT(socket != NULL); 140 RTC_DCHECK(socket != NULL);
139 size_t size; 141 size_t size;
140 int written; 142 int written;
141 const void* p = buffer->GetReadData(&size); 143 const void* p = buffer->GetReadData(&size);
142 written = socket->Send(p, size); 144 written = socket->Send(p, size);
143 buffer->ConsumeReadData(std::max(written, 0)); 145 buffer->ConsumeReadData(std::max(written, 0));
144 } 146 }
145 147
146 void ProxyBinding::Destroy() { 148 void ProxyBinding::Destroy() {
147 SignalDestroyed(this); 149 SignalDestroyed(this);
148 } 150 }
149 151
150 AsyncProxyServerSocket* SocksProxyServer::WrapSocket(AsyncSocket* socket) { 152 AsyncProxyServerSocket* SocksProxyServer::WrapSocket(AsyncSocket* socket) {
151 return new AsyncSocksProxyServerSocket(socket); 153 return new AsyncSocksProxyServerSocket(socket);
152 } 154 }
153 155
154 } // namespace rtc 156 } // namespace rtc
OLDNEW
« no previous file with comments | « webrtc/base/proxydetect.cc ('k') | webrtc/base/rollingaccumulator.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698