Limited NSEC support to RecordParsed

net/dns/record_rdata.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/dns/record_rdata.h"
 
 #include "net/base/big_endian.h"
 #include "net/base/dns_util.h"
 #include "net/dns/dns_protocol.h"
 #include "net/dns/dns_response.h"
@@ skipping 197 matching lines @@
 uint16 TxtRecordRdata::Type() const {
   return TxtRecordRdata::kType;
 }
 
 bool TxtRecordRdata::IsEqual(const RecordRdata* other) const {
   if (other->Type() != Type()) return false;
   const TxtRecordRdata* txt_other = static_cast<const TxtRecordRdata*>(other);
   return texts_ == txt_other->texts_;
 }
 
+NsecRecordRdata::NsecRecordRdata() {
+}
+
+NsecRecordRdata::~NsecRecordRdata() {
+}
+
+// static
+scoped_ptr<NsecRecordRdata> NsecRecordRdata::Create(
+    const base::StringPiece& data,
+    const DnsRecordParser& parser) {
+  scoped_ptr<NsecRecordRdata> rdata(new NsecRecordRdata);
+
+  unsigned len_next_domain = parser.ReadName(data.data(), NULL);
+
+  if (len_next_domain == 0 || data.length() < len_next_domain + 2)
+    return scoped_ptr<NsecRecordRdata>();
+
+  unsigned window_num = data[len_next_domain];
+  unsigned len = data[len_next_domain+1];
+
+  // The window number must be zero in mDns-secific nsec records. The bitmap
+  // length must be between 1 and 32, and since we may only have one window, the
+  // data length must be exactl equal to the domain length plus bitmap size.
+  if (window_num != 0 || len == 0 || len > 32 ||
+      data.length() != len_next_domain + 2 + len)
+    return scoped_ptr<NsecRecordRdata>();
+
+  base::StringPiece bitmap_data = data.substr(len_next_domain + 2);
+
+  rdata->bitmap_.insert(rdata->bitmap_.begin(),
+                        bitmap_data.begin(),
+                        bitmap_data.end());
+
+  return rdata.Pass();
+}
+
+uint16 NsecRecordRdata::Type() const {
+  return NsecRecordRdata::kType;
+}
+
+bool NsecRecordRdata::IsEqual(const RecordRdata* other) const {
+  if (other->Type() != Type())
+    return false;
+  const NsecRecordRdata* nsec_other =
+      static_cast<const NsecRecordRdata*>(other);
+  return bitmap_ == nsec_other->bitmap_;
+}
+
+bool NsecRecordRdata::GetBit(unsigned i) {
+  unsigned byte_num = i/8;
+  if (bitmap_.size() < byte_num + 1)
+    return false;
+
+  unsigned bit_num = 7 - i % 8;
+  return (bitmap_[byte_num] & (1 << bit_num)) != 0;
+}
+
 }  // namespace net