Index: net/dns/dns_response.cc |
diff --git a/net/dns/dns_response.cc b/net/dns/dns_response.cc |
index 625e7c041dc0ddfc0eb5c02dc650b22d58fba7d2..f14e2a81ef0938f492d48ea7749d34fc9dd70b0f 100644 |
--- a/net/dns/dns_response.cc |
+++ b/net/dns/dns_response.cc |
@@ -175,6 +175,20 @@ bool DnsResponse::InitParse(int nbytes, const DnsQuery& query) { |
return true; |
} |
+bool DnsResponse::InitParse(int nbytes) { |
+ if (nbytes >= io_buffer_->size()) |
+ return false; |
+ |
+ // Construct the parser. |
+ size_t response_offset = GetResponseOffset(); |
+ if (response_offset == 0) |
+ return false; |
+ |
+ parser_ = DnsRecordParser( |
+ io_buffer_->data(), nbytes, response_offset); |
+ return true; |
+} |
+ |
bool DnsResponse::IsValid() const { |
return parser_.IsValid(); |
} |
@@ -217,6 +231,39 @@ std::string DnsResponse::GetDottedName() const { |
return DNSDomainToString(qname()); |
} |
+size_t DnsResponse::GetResponseOffset() const { |
szym
2013/04/17 22:11:12
Use DnsRecordParser instead. ReadName with |out| s
Noam Samuel
2013/04/18 19:03:17
Done.
|
+ uint16 qdcount = base::NetToHost16(header()->qdcount); |
+ const char* data = io_buffer_->data(); |
+ size_t data_size = io_buffer_->size(); |
+ size_t pos = sizeof(dns_protocol::Header); |
+ |
+ for (uint16 i = 0; i < qdcount; i++) { |
+ while (pos < data_size && data[pos] != 0) { |
+ unsigned txt_len = static_cast<unsigned>(data[pos]); |
+ // Traversal and checks adapted from DNSDomainToString. |
+ |
+ // Handling pointers, which have both high bits set and an extra |
+ // byte trailing them to indicate a location in the buffer. |
+ if (txt_len >= 192) { |
+ pos += 2; |
+ continue; |
szym
2013/04/17 22:11:12
This is an error. At this point you should be done
Noam Samuel
2013/04/18 19:03:17
Made irrelevant.
On 2013/04/17 22:11:12, szym wro
|
+ } |
+ |
+ if (txt_len > 63) { |
+ return 0; |
+ } |
+ |
+ pos += txt_len + 1; |
+ } |
+ |
+ pos += 5; // 1 for 0-length text, 2 for qtype, 2 for qclass |
+ } |
+ |
+ if (pos > data_size) return 0; |
+ |
+ return pos; |
+} |
+ |
DnsRecordParser DnsResponse::Parser() const { |
DCHECK(parser_.IsValid()); |
// Return a copy of the parser. |