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_query.h" | 5 #include "net/dns/dns_query.h" |
6 | 6 |
7 #include <limits> | 7 #include <limits> |
8 | 8 |
9 #include "base/big_endian.h" | 9 #include "base/big_endian.h" |
10 #include "base/sys_byteorder.h" | 10 #include "base/sys_byteorder.h" |
11 #include "net/base/dns_util.h" | 11 #include "net/base/dns_util.h" |
12 #include "net/base/io_buffer.h" | 12 #include "net/base/io_buffer.h" |
13 #include "net/dns/dns_protocol.h" | 13 #include "net/dns/dns_protocol.h" |
14 | 14 |
15 namespace net { | 15 namespace net { |
16 | 16 |
17 // DNS query consists of a 12-byte header followed by a question section. | 17 // DNS query consists of a 12-byte header followed by a question section. |
18 // For details, see RFC 1035 section 4.1.1. This header template sets RD | 18 // For details, see RFC 1035 section 4.1.1. This header template sets RD |
19 // bit, which directs the name server to pursue query recursively, and sets | 19 // bit, which directs the name server to pursue query recursively, and sets |
20 // the QDCOUNT to 1, meaning the question section has a single entry. | 20 // the QDCOUNT to 1, meaning the question section has a single entry. |
21 DnsQuery::DnsQuery(uint16 id, const base::StringPiece& qname, uint16 qtype) | 21 DnsQuery::DnsQuery(uint16 id, const base::StringPiece& qname, uint16 qtype) |
22 : qname_size_(qname.size()) { | 22 : qname_size_(qname.size()) { |
23 DCHECK(!DNSDomainToString(qname).empty()); | 23 DCHECK(!DNSDomainToString(qname).empty()); |
24 // QNAME + QTYPE + QCLASS | 24 // QNAME + QTYPE + QCLASS |
25 size_t question_size = qname_size_ + sizeof(uint16) + sizeof(uint16); | 25 size_t question_size = qname_size_ + sizeof(uint16) + sizeof(uint16); |
26 io_buffer_ = new IOBufferWithSize(sizeof(dns_protocol::Header) + | 26 io_buffer_ = |
27 question_size); | 27 new IOBufferWithSize(sizeof(dns_protocol::Header) + question_size); |
28 dns_protocol::Header* header = | 28 dns_protocol::Header* header = |
29 reinterpret_cast<dns_protocol::Header*>(io_buffer_->data()); | 29 reinterpret_cast<dns_protocol::Header*>(io_buffer_->data()); |
30 memset(header, 0, sizeof(dns_protocol::Header)); | 30 memset(header, 0, sizeof(dns_protocol::Header)); |
31 header->id = base::HostToNet16(id); | 31 header->id = base::HostToNet16(id); |
32 header->flags = base::HostToNet16(dns_protocol::kFlagRD); | 32 header->flags = base::HostToNet16(dns_protocol::kFlagRD); |
33 header->qdcount = base::HostToNet16(1); | 33 header->qdcount = base::HostToNet16(1); |
34 | 34 |
35 // Write question section after the header. | 35 // Write question section after the header. |
36 base::BigEndianWriter writer(reinterpret_cast<char*>(header + 1), | 36 base::BigEndianWriter writer(reinterpret_cast<char*>(header + 1), |
37 question_size); | 37 question_size); |
(...skipping 28 matching lines...) Expand all Loading... |
66 } | 66 } |
67 | 67 |
68 base::StringPiece DnsQuery::question() const { | 68 base::StringPiece DnsQuery::question() const { |
69 return base::StringPiece(io_buffer_->data() + sizeof(dns_protocol::Header), | 69 return base::StringPiece(io_buffer_->data() + sizeof(dns_protocol::Header), |
70 qname_size_ + sizeof(uint16) + sizeof(uint16)); | 70 qname_size_ + sizeof(uint16) + sizeof(uint16)); |
71 } | 71 } |
72 | 72 |
73 DnsQuery::DnsQuery(const DnsQuery& orig, uint16 id) { | 73 DnsQuery::DnsQuery(const DnsQuery& orig, uint16 id) { |
74 qname_size_ = orig.qname_size_; | 74 qname_size_ = orig.qname_size_; |
75 io_buffer_ = new IOBufferWithSize(orig.io_buffer()->size()); | 75 io_buffer_ = new IOBufferWithSize(orig.io_buffer()->size()); |
76 memcpy(io_buffer_.get()->data(), orig.io_buffer()->data(), | 76 memcpy(io_buffer_.get()->data(), |
| 77 orig.io_buffer()->data(), |
77 io_buffer_.get()->size()); | 78 io_buffer_.get()->size()); |
78 dns_protocol::Header* header = | 79 dns_protocol::Header* header = |
79 reinterpret_cast<dns_protocol::Header*>(io_buffer_->data()); | 80 reinterpret_cast<dns_protocol::Header*>(io_buffer_->data()); |
80 header->id = base::HostToNet16(id); | 81 header->id = base::HostToNet16(id); |
81 } | 82 } |
82 | 83 |
83 void DnsQuery::set_flags(uint16 flags) { | 84 void DnsQuery::set_flags(uint16 flags) { |
84 dns_protocol::Header* header = | 85 dns_protocol::Header* header = |
85 reinterpret_cast<dns_protocol::Header*>(io_buffer_->data()); | 86 reinterpret_cast<dns_protocol::Header*>(io_buffer_->data()); |
86 header->flags = flags; | 87 header->flags = flags; |
87 } | 88 } |
88 | 89 |
89 } // namespace net | 90 } // namespace net |
OLD | NEW |