src/remoting/host/dns_blackhole_checker.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "remoting/host/dns_blackhole_checker.h"
   6
   7 #include "net/url_request/url_fetcher.h"
   8 #include "net/url_request/url_request_context_getter.h"
   9 #include "url/gurl.h"
  10
  11 namespace remoting {
  12
  13 // Default prefix added to the base talkgadget URL.
  14 const char kDefaultHostTalkGadgetPrefix[] = "chromoting-host";
  15
  16 // The base talkgadget URL.
  17 const char kTalkGadgetUrl[] = ".talkgadget.google.com/talkgadget/"
  18                               "oauth/chrome-remote-desktop-host";
  19
  20 DnsBlackholeChecker::DnsBlackholeChecker(
  21     scoped_refptr<net::URLRequestContextGetter> url_request_context_getter,
  22     std::string talkgadget_prefix)
  23     : url_request_context_getter_(url_request_context_getter),
  24       talkgadget_prefix_(talkgadget_prefix) {
  25 }
  26
  27 DnsBlackholeChecker::~DnsBlackholeChecker() {
  28 }
  29
  30 // This is called in response to the TalkGadget http request initiated from
  31 // CheckStatus().
  32 void DnsBlackholeChecker::OnURLFetchComplete(const net::URLFetcher* source) {
  33   int response = source->GetResponseCode();
  34   bool allow = false;
  35   if (source->GetResponseCode() == 200) {
  36     LOG(INFO) << "Successfully connected to host talkgadget.";
  37     allow = true;
  38   } else {
  39     LOG(INFO) << "Unable to connect to host talkgadget (" << response << ")";
  40   }
  41   url_fetcher_.reset(NULL);
  42   callback_.Run(allow);
  43   callback_.Reset();
  44 }
  45
  46 void DnsBlackholeChecker::CheckForDnsBlackhole(
  47     const base::Callback<void(bool)>& callback) {
  48   // Make sure we're not currently in the middle of a connection check.
  49   if (!url_fetcher_.get()) {
  50     DCHECK(callback_.is_null());
  51     callback_ = callback;
  52     std::string talkgadget_url("https://");
  53     if (talkgadget_prefix_.empty()) {
  54       talkgadget_url += kDefaultHostTalkGadgetPrefix;
  55     } else {
  56       talkgadget_url += talkgadget_prefix_;
  57     }
  58     talkgadget_url += kTalkGadgetUrl;
  59     LOG(INFO) << "Verifying connection to " << talkgadget_url;
  60     url_fetcher_.reset(net::URLFetcher::Create(GURL(talkgadget_url),
  61                                                net::URLFetcher::GET, this));
  62     url_fetcher_->SetRequestContext(url_request_context_getter_.get());
  63     url_fetcher_->Start();
  64   } else {
  65     LOG(INFO) << "Pending connection check";
  66   }
  67 }
  68
  69 }  // namespace remoting