| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/dns/dns_transaction.h" | 5 #include "net/dns/dns_transaction.h" |
| 6 | 6 |
| 7 #include <deque> | 7 #include <deque> |
| 8 #include <memory> | 8 #include <memory> |
| 9 #include <string> | 9 #include <string> |
| 10 #include <utility> | 10 #include <utility> |
| 11 #include <vector> | 11 #include <vector> |
| 12 | 12 |
| 13 #include "base/big_endian.h" | 13 #include "base/big_endian.h" |
| 14 #include "base/bind.h" | 14 #include "base/bind.h" |
| 15 #include "base/location.h" | 15 #include "base/location.h" |
| 16 #include "base/macros.h" | 16 #include "base/macros.h" |
| 17 #include "base/memory/ptr_util.h" | 17 #include "base/memory/ptr_util.h" |
| 18 #include "base/memory/ref_counted.h" | 18 #include "base/memory/ref_counted.h" |
| 19 #include "base/memory/weak_ptr.h" | 19 #include "base/memory/weak_ptr.h" |
| 20 #include "base/metrics/histogram_macros.h" | 20 #include "base/metrics/histogram_macros.h" |
| 21 #include "base/profiler/scoped_tracker.h" | 21 #include "base/profiler/scoped_tracker.h" |
| 22 #include "base/rand_util.h" | 22 #include "base/rand_util.h" |
| 23 #include "base/single_thread_task_runner.h" | 23 #include "base/single_thread_task_runner.h" |
| 24 #include "base/stl_util.h" | 24 #include "base/stl_util.h" |
| 25 #include "base/strings/string_piece.h" | 25 #include "base/strings/string_piece.h" |
| 26 #include "base/threading/non_thread_safe.h" | 26 #include "base/threading/thread_checker.h" |
| 27 #include "base/threading/thread_task_runner_handle.h" | 27 #include "base/threading/thread_task_runner_handle.h" |
| 28 #include "base/timer/timer.h" | 28 #include "base/timer/timer.h" |
| 29 #include "base/values.h" | 29 #include "base/values.h" |
| 30 #include "net/base/completion_callback.h" | 30 #include "net/base/completion_callback.h" |
| 31 #include "net/base/io_buffer.h" | 31 #include "net/base/io_buffer.h" |
| 32 #include "net/base/ip_address.h" | 32 #include "net/base/ip_address.h" |
| 33 #include "net/base/ip_endpoint.h" | 33 #include "net/base/ip_endpoint.h" |
| 34 #include "net/base/net_errors.h" | 34 #include "net/base/net_errors.h" |
| 35 #include "net/dns/dns_protocol.h" | 35 #include "net/dns/dns_protocol.h" |
| 36 #include "net/dns/dns_query.h" | 36 #include "net/dns/dns_query.h" |
| (...skipping 515 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 552 | 552 |
| 553 // ---------------------------------------------------------------------------- | 553 // ---------------------------------------------------------------------------- |
| 554 | 554 |
| 555 // Implements DnsTransaction. Configuration is supplied by DnsSession. | 555 // Implements DnsTransaction. Configuration is supplied by DnsSession. |
| 556 // The suffix list is built according to the DnsConfig from the session. | 556 // The suffix list is built according to the DnsConfig from the session. |
| 557 // The timeout for each DnsUDPAttempt is given by DnsSession::NextTimeout. | 557 // The timeout for each DnsUDPAttempt is given by DnsSession::NextTimeout. |
| 558 // The first server to attempt on each query is given by | 558 // The first server to attempt on each query is given by |
| 559 // DnsSession::NextFirstServerIndex, and the order is round-robin afterwards. | 559 // DnsSession::NextFirstServerIndex, and the order is round-robin afterwards. |
| 560 // Each server is attempted DnsConfig::attempts times. | 560 // Each server is attempted DnsConfig::attempts times. |
| 561 class DnsTransactionImpl : public DnsTransaction, | 561 class DnsTransactionImpl : public DnsTransaction, |
| 562 public base::NonThreadSafe, | |
| 563 public base::SupportsWeakPtr<DnsTransactionImpl> { | 562 public base::SupportsWeakPtr<DnsTransactionImpl> { |
| 564 public: | 563 public: |
| 565 DnsTransactionImpl(DnsSession* session, | 564 DnsTransactionImpl(DnsSession* session, |
| 566 const std::string& hostname, | 565 const std::string& hostname, |
| 567 uint16_t qtype, | 566 uint16_t qtype, |
| 568 const DnsTransactionFactory::CallbackType& callback, | 567 const DnsTransactionFactory::CallbackType& callback, |
| 569 const NetLogWithSource& net_log) | 568 const NetLogWithSource& net_log) |
| 570 : session_(session), | 569 : session_(session), |
| 571 hostname_(hostname), | 570 hostname_(hostname), |
| 572 qtype_(qtype), | 571 qtype_(qtype), |
| 573 callback_(callback), | 572 callback_(callback), |
| 574 net_log_(net_log), | 573 net_log_(net_log), |
| 575 qnames_initial_size_(0), | 574 qnames_initial_size_(0), |
| 576 attempts_count_(0), | 575 attempts_count_(0), |
| 577 had_tcp_attempt_(false), | 576 had_tcp_attempt_(false), |
| 578 first_server_index_(0) { | 577 first_server_index_(0) { |
| 579 DCHECK(session_.get()); | 578 DCHECK(session_.get()); |
| 580 DCHECK(!hostname_.empty()); | 579 DCHECK(!hostname_.empty()); |
| 581 DCHECK(!callback_.is_null()); | 580 DCHECK(!callback_.is_null()); |
| 582 DCHECK(!IsIPLiteral(hostname_)); | 581 DCHECK(!IsIPLiteral(hostname_)); |
| 583 } | 582 } |
| 584 | 583 |
| 585 ~DnsTransactionImpl() override { | 584 ~DnsTransactionImpl() override { |
| 585 DCHECK_CALLED_ON_VALID_THREAD(thread_checker_); |
| 586 if (!callback_.is_null()) { | 586 if (!callback_.is_null()) { |
| 587 net_log_.EndEventWithNetErrorCode(NetLogEventType::DNS_TRANSACTION, | 587 net_log_.EndEventWithNetErrorCode(NetLogEventType::DNS_TRANSACTION, |
| 588 ERR_ABORTED); | 588 ERR_ABORTED); |
| 589 } // otherwise logged in DoCallback or Start | 589 } // otherwise logged in DoCallback or Start |
| 590 } | 590 } |
| 591 | 591 |
| 592 const std::string& GetHostname() const override { | 592 const std::string& GetHostname() const override { |
| 593 DCHECK(CalledOnValidThread()); | 593 DCHECK_CALLED_ON_VALID_THREAD(thread_checker_); |
| 594 return hostname_; | 594 return hostname_; |
| 595 } | 595 } |
| 596 | 596 |
| 597 uint16_t GetType() const override { | 597 uint16_t GetType() const override { |
| 598 DCHECK(CalledOnValidThread()); | 598 DCHECK_CALLED_ON_VALID_THREAD(thread_checker_); |
| 599 return qtype_; | 599 return qtype_; |
| 600 } | 600 } |
| 601 | 601 |
| 602 void Start() override { | 602 void Start() override { |
| 603 DCHECK(!callback_.is_null()); | 603 DCHECK(!callback_.is_null()); |
| 604 DCHECK(attempts_.empty()); | 604 DCHECK(attempts_.empty()); |
| 605 net_log_.BeginEvent(NetLogEventType::DNS_TRANSACTION, | 605 net_log_.BeginEvent(NetLogEventType::DNS_TRANSACTION, |
| 606 base::Bind(&NetLogStartCallback, &hostname_, qtype_)); | 606 base::Bind(&NetLogStartCallback, &hostname_, qtype_)); |
| 607 AttemptResult result(PrepareSearch(), NULL); | 607 AttemptResult result(PrepareSearch(), NULL); |
| 608 if (result.rv == OK) { | 608 if (result.rv == OK) { |
| (...skipping 354 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 963 std::vector<std::unique_ptr<DnsAttempt>> attempts_; | 963 std::vector<std::unique_ptr<DnsAttempt>> attempts_; |
| 964 // Count of attempts, not reset when |attempts_| vector is cleared. | 964 // Count of attempts, not reset when |attempts_| vector is cleared. |
| 965 int attempts_count_; | 965 int attempts_count_; |
| 966 bool had_tcp_attempt_; | 966 bool had_tcp_attempt_; |
| 967 | 967 |
| 968 // Index of the first server to try on each search query. | 968 // Index of the first server to try on each search query. |
| 969 int first_server_index_; | 969 int first_server_index_; |
| 970 | 970 |
| 971 base::OneShotTimer timer_; | 971 base::OneShotTimer timer_; |
| 972 | 972 |
| 973 THREAD_CHECKER(thread_checker_); |
| 974 |
| 973 DISALLOW_COPY_AND_ASSIGN(DnsTransactionImpl); | 975 DISALLOW_COPY_AND_ASSIGN(DnsTransactionImpl); |
| 974 }; | 976 }; |
| 975 | 977 |
| 976 // ---------------------------------------------------------------------------- | 978 // ---------------------------------------------------------------------------- |
| 977 | 979 |
| 978 // Implementation of DnsTransactionFactory that returns instances of | 980 // Implementation of DnsTransactionFactory that returns instances of |
| 979 // DnsTransactionImpl. | 981 // DnsTransactionImpl. |
| 980 class DnsTransactionFactoryImpl : public DnsTransactionFactory { | 982 class DnsTransactionFactoryImpl : public DnsTransactionFactory { |
| 981 public: | 983 public: |
| 982 explicit DnsTransactionFactoryImpl(DnsSession* session) { | 984 explicit DnsTransactionFactoryImpl(DnsSession* session) { |
| (...skipping 16 matching lines...) Expand all Loading... |
| 999 } // namespace | 1001 } // namespace |
| 1000 | 1002 |
| 1001 // static | 1003 // static |
| 1002 std::unique_ptr<DnsTransactionFactory> DnsTransactionFactory::CreateFactory( | 1004 std::unique_ptr<DnsTransactionFactory> DnsTransactionFactory::CreateFactory( |
| 1003 DnsSession* session) { | 1005 DnsSession* session) { |
| 1004 return std::unique_ptr<DnsTransactionFactory>( | 1006 return std::unique_ptr<DnsTransactionFactory>( |
| 1005 new DnsTransactionFactoryImpl(session)); | 1007 new DnsTransactionFactoryImpl(session)); |
| 1006 } | 1008 } |
| 1007 | 1009 |
| 1008 } // namespace net | 1010 } // namespace net |
| OLD | NEW |