From: Mukund Sivaraman <muks@isc.org>
Date: Wed, 3 Oct 2012 04:28:04 +0000 (+0530)
Subject: [2206] Add ZoneTableHeader::getTable() method
X-Git-Tag: trac2351_base~25
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=d25bb8d27cf235603cb8a1d789066b2aa993eccd;p=thirdparty%2Fkea.git

[2206] Add ZoneTableHeader::getTable() method
---

diff --git a/src/lib/datasrc/memory/zone_table_segment.h b/src/lib/datasrc/memory/zone_table_segment.h
index 89389a5fba..6bf0d480a3 100644
--- a/src/lib/datasrc/memory/zone_table_segment.h
+++ b/src/lib/datasrc/memory/zone_table_segment.h
@@ -34,6 +34,18 @@ namespace memory {
 /// implementation. It contains an offset pointer to the zone table (a
 /// map from domain names to zone locators) in memory.
 struct ZoneTableHeader {
+public:
+    /// \brief Returns a pointer to the underlying zone table.
+    ZoneTable* getTable() {
+        return (table.get());
+    }
+
+    /// \brief const version of getTable().
+    const ZoneTable* getTable() const {
+        return (getTable());
+    }
+
+private:
     boost::interprocess::offset_ptr<ZoneTable> table;
 };
 
diff --git a/src/lib/datasrc/tests/memory/zone_table_segment_unittest.cc b/src/lib/datasrc/tests/memory/zone_table_segment_unittest.cc
index 1dfe14341c..1313092b59 100644
--- a/src/lib/datasrc/tests/memory/zone_table_segment_unittest.cc
+++ b/src/lib/datasrc/tests/memory/zone_table_segment_unittest.cc
@@ -40,7 +40,7 @@ TEST(ZoneTableSegment, getHeader) {
     EXPECT_NE(static_cast<void*>(NULL), header);
 
     // The zone table is unset.
-    ZoneTable* table = header->table.get();
+    ZoneTable* table = header->getTable();
     EXPECT_EQ(static_cast<void*>(NULL), table);
 }