diff options
Diffstat (limited to 'third_party/libjingle/source/talk/base/autodetectproxy.cc')
| -rw-r--r-- | third_party/libjingle/source/talk/base/autodetectproxy.cc | 187 |
1 files changed, 187 insertions, 0 deletions
diff --git a/third_party/libjingle/source/talk/base/autodetectproxy.cc b/third_party/libjingle/source/talk/base/autodetectproxy.cc new file mode 100644 index 0000000..2b06430 --- /dev/null +++ b/third_party/libjingle/source/talk/base/autodetectproxy.cc @@ -0,0 +1,187 @@ +/* + * libjingle + * Copyright 2004--2005, Google Inc. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * 3. The name of the author may not be used to endorse or promote products + * derived from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED + * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO + * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, + * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; + * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, + * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR + * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF + * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "talk/base/autodetectproxy.h" +#include "talk/base/httpcommon.h" +#include "talk/base/httpcommon-inl.h" +#include "talk/base/proxydetect.h" + +namespace talk_base { + +enum { MSG_TIMEOUT = SignalThread::ST_MSG_FIRST_AVAILABLE }; + +static const ProxyType TEST_ORDER[] = { + PROXY_HTTPS, PROXY_SOCKS5, PROXY_UNKNOWN +}; + +AutoDetectProxy::AutoDetectProxy(const std::string& user_agent) + : agent_(user_agent), socket_(NULL), next_(0) { +} + +AutoDetectProxy::~AutoDetectProxy() { +} + +void AutoDetectProxy::DoWork() { + // TODO(oja): Try connecting to server_url without proxy first here? + if (!server_url_.empty()) { + LOG(LS_INFO) << "GetProxySettingsForUrl(" << server_url_ << ") - start"; + GetProxySettingsForUrl(agent_.c_str(), server_url_.c_str(), proxy_, true); + LOG(LS_INFO) << "GetProxySettingsForUrl - stop"; + } + Url<char> url(proxy_.address.IPAsString()); + if (url.valid()) { + LOG(LS_WARNING) << "AutoDetectProxy removing http prefix on proxy host"; + proxy_.address.SetIP(url.host()); + } + LOG(LS_INFO) << "AutoDetectProxy found proxy at " << proxy_.address; + if (proxy_.type == PROXY_UNKNOWN) { + LOG(LS_INFO) << "AutoDetectProxy initiating proxy classification"; + Next(); + // Process I/O until Stop() + Thread::Current()->ProcessMessages(kForever); + // Clean up the autodetect socket, from the thread that created it + delete socket_; + } + // TODO(oja): If we found a proxy, try to use it to verify that it + // works by sending a request to server_url. This could either be + // done here or by the HttpPortAllocator. +} + +void AutoDetectProxy::OnMessage(Message *msg) { + if (MSG_TIMEOUT == msg->message_id) { + OnCloseEvent(socket_, ETIMEDOUT); + } else { + SignalThread::OnMessage(msg); + } +} + +void AutoDetectProxy::Next() { + if (TEST_ORDER[next_] >= PROXY_UNKNOWN) { + Complete(PROXY_UNKNOWN); + return; + } + + LOG(LS_VERBOSE) << "AutoDetectProxy connecting to " + << proxy_.address.ToString(); + + if (socket_) { + Thread::Current()->Clear(this, MSG_TIMEOUT); + socket_->Close(); + Thread::Current()->Dispose(socket_); + socket_ = NULL; + } + + socket_ = Thread::Current()->socketserver()->CreateAsyncSocket(SOCK_STREAM); + socket_->SignalConnectEvent.connect(this, &AutoDetectProxy::OnConnectEvent); + socket_->SignalReadEvent.connect(this, &AutoDetectProxy::OnReadEvent); + socket_->SignalCloseEvent.connect(this, &AutoDetectProxy::OnCloseEvent); + socket_->Connect(proxy_.address); + + // Timeout after 2 seconds + Thread::Current()->PostDelayed(2000, this, MSG_TIMEOUT); +} + +void AutoDetectProxy::Complete(ProxyType type) { + Thread::Current()->Clear(this, MSG_TIMEOUT); + socket_->Close(); + + proxy_.type = type; + LoggingSeverity sev = (proxy_.type == PROXY_UNKNOWN) ? LS_ERROR : LS_INFO; + LOG_V(sev) << "AutoDetectProxy detected " << proxy_.address.ToString() + << " as type " << proxy_.type; + + Thread::Current()->Quit(); +} + +void AutoDetectProxy::OnConnectEvent(AsyncSocket * socket) { + std::string probe; + + switch (TEST_ORDER[next_]) { + case PROXY_HTTPS: + probe.assign("CONNECT www.google.com:443 HTTP/1.0\r\n" + "User-Agent: "); + probe.append(agent_); + probe.append("\r\n" + "Host: www.google.com\r\n" + "Content-Length: 0\r\n" + "Proxy-Connection: Keep-Alive\r\n" + "\r\n"); + break; + case PROXY_SOCKS5: + probe.assign("\005\001\000", 3); + break; + default: + ASSERT(false); + return; + } + + LOG(LS_VERBOSE) << "AutoDetectProxy probing type " << TEST_ORDER[next_] + << " sending " << probe.size() << " bytes"; + socket_->Send(probe.data(), probe.size()); +} + +void AutoDetectProxy::OnReadEvent(AsyncSocket * socket) { + char data[257]; + int len = socket_->Recv(data, 256); + if (len > 0) { + data[len] = 0; + LOG(LS_VERBOSE) << "AutoDetectProxy read " << len << " bytes"; + } + + switch (TEST_ORDER[next_]) { + case PROXY_HTTPS: + if ((len >= 2) && (data[0] == '\x05')) { + Complete(PROXY_SOCKS5); + return; + } + if ((len >= 5) && (strncmp(data, "HTTP/", 5) == 0)) { + Complete(PROXY_HTTPS); + return; + } + break; + case PROXY_SOCKS5: + if ((len >= 2) && (data[0] == '\x05')) { + Complete(PROXY_SOCKS5); + return; + } + break; + default: + ASSERT(false); + return; + } + + ++next_; + Next(); +} + +void AutoDetectProxy::OnCloseEvent(AsyncSocket * socket, int error) { + LOG(LS_VERBOSE) << "AutoDetectProxy closed with error: " << error; + ++next_; + Next(); +} + +} // namespace talk_base |