autodetectproxy.cc

本人收集整理的一份c/c++跨平台网络库

#include "autodetectproxy.h"
#include "httpcommon.h"
//#include "xmpp/xmppclientsettings.h"
#include "proxydetect.h"

using namespace utils_base;

enum { MSG_TIMEOUT = SignalThread::ST_MSG_FIRST_AVAILABLE };

const utils_base::ProxyType TEST_ORDER[] = {
  utils_base::PROXY_HTTPS, /*talk_base::PROXY_SOCKS5,*/ utils_base::PROXY_UNKNOWN
};

AutoDetectProxy::AutoDetectProxy(const std::string& user_agent)
: agent_(user_agent), socket_(NULL), next_(0)
{
}

AutoDetectProxy::~AutoDetectProxy() {
  delete socket_;
}

void AutoDetectProxy::DoWork() {
  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";
  }
  utils_base::Url<char> url(proxy_.address.IPAsString());
  if (url.valid()) {
    LOG(LS_WARNING) << "AutoDetectProxy removing http prefix on proxy host";
    proxy_.address.SetIP(url.server());
  }
  if (proxy_.type == utils_base::PROXY_UNKNOWN) {
    //LOG(LS_INFO) << "Proxy classification start";
    Next();
    // Process I/O until Stop()
    Thread::Current()->ProcessMessages(kForever);
  }
}

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_] >= utils_base::PROXY_UNKNOWN) {
    Complete(utils_base::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(utils_base::ProxyType type) {
  Thread::Current()->Clear(this, MSG_TIMEOUT);
  socket_->Close();

  proxy_.type = type;
  utils_base::LoggingSeverity sev
    = (proxy_.type == utils_base::PROXY_UNKNOWN)
      ? utils_base::LS_ERROR : utils_base::LS_VERBOSE;
  LOG_V(sev) << "AutoDetectProxy detected " << proxy_.address.ToString()
             << " as type " << proxy_.type;

  Thread::Current()->MessageQueue::Stop();
}

void AutoDetectProxy::OnConnectEvent(utils_base::AsyncSocket * socket) {
  std::string probe;

  switch (TEST_ORDER[next_]) {
  case utils_base::PROXY_HTTPS:
    probe.assign("\005\001"
                 "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");
    //probe = "CONNECT www.google.com:443 HTTP/1.0\r\n\r\n";
    break;
  case utils_base::PROXY_SOCKS5:
    probe.assign("\005\001\000", 3);
    break;
  }

  LOG(LS_VERBOSE) << "AutoDetectProxy probing type " << TEST_ORDER[next_]
                  << " sending " << (unsigned int)probe.size() << " bytes";
  socket_->Send(probe.data(), probe.size());
}

void AutoDetectProxy::OnReadEvent(utils_base::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 utils_base::PROXY_HTTPS:
    if ((len >= 2) && (data[0] == '\x05')) {
      Complete(utils_base::PROXY_SOCKS5);
      return;
    }
    if ((len >= 5) && (strncmp(data, "HTTP/", 5) == 0)) {
      Complete(utils_base::PROXY_HTTPS);
      return;
    }
    break;
  case utils_base::PROXY_SOCKS5:
    if ((len >= 2) && (data[0] == '\x05')) {
      Complete(utils_base::PROXY_SOCKS5);
      return;
    }
    break;
  }

  ++next_;
  Next();
}

void AutoDetectProxy::OnCloseEvent(utils_base::AsyncSocket * socket, int error) {
  LOG(LS_VERBOSE) << "AutoDetectProxy closed with error: " << error;
  ++next_;
  Next();
}