BIND 10 trac2427, updated. 3adc5a448824ec9a819c52c89869855cf1de8d0c [2427] Unify origin handling

[BIND 10 source code commits]

The branch, trac2427 has been updated
       via  3adc5a448824ec9a819c52c89869855cf1de8d0c (commit)
      from  694dda8c2b939cf29d8949693f40dd12439fab0c (commit)

Those revisions listed above that are new to this repository have
not appeared on any other notification email; so we list those
revisions in full, below.

- Log -----------------------------------------------------------------
commit 3adc5a448824ec9a819c52c89869855cf1de8d0c
Author: Michal 'vorner' Vaner <michal.vaner at nic.cz>
Date:   Mon Dec 17 13:47:24 2012 +0100

    [2427] Unify origin handling
    
    Between the $INCLUDE and $ORIGIN. Mostly to make sure it is consistent.

-----------------------------------------------------------------------

Summary of changes:
 src/lib/dns/master_loader.cc                |   53 ++++++++++++++-------------
 src/lib/dns/tests/master_loader_unittest.cc |    2 +-
 2 files changed, 28 insertions(+), 27 deletions(-)

-----------------------------------------------------------------------
diff --git a/src/lib/dns/master_loader.cc b/src/lib/dns/master_loader.cc
index 00f0d7a..2548a48 100644
--- a/src/lib/dns/master_loader.cc
+++ b/src/lib/dns/master_loader.cc
@@ -133,50 +133,51 @@ public:
 
     bool loadIncremental(size_t count_limit);
 
-    void doInclude() {
-        // First, get the filename to include
-        const string
-            filename(lexer_.getNextToken(MasterToken::QSTRING).getString());
-
-        // There could be an origin (or maybe not). So try looking
-        const MasterToken name_tok(lexer_.getNextToken(MasterToken::QSTRING,
-                                                       true));
+    void doOrigin(bool is_optional) {
+        // Parse and create the new origin. It is relative to the previous
+        // one.
+        const MasterToken&
+            name_tok(lexer_.getNextToken(MasterToken::QSTRING, is_optional));
 
         if (name_tok.getType() == MasterToken::QSTRING ||
             name_tok.getType() == MasterToken::STRING) {
-            // There's an optional name, meaning origin. Extract it
-            // and store.
+
             const MasterToken::StringRegion&
                 name_string(name_tok.getStringRegion());
             active_origin_ = Name(name_string.beg, name_string.len,
                                   &active_origin_);
         } else {
-            // We return the newline there. This is because after we pop
-            // the source, we want to call eatUntilEOL and this would
-            // eat to the next one.
+            // If it is not optional, we must not get anything but
+            // a string token.
+            assert(is_optional);
+
+            // We return the newline there. This is because we want to
+            // behave the same if there is or isn't the name, leaving the
+            // newline there.
             lexer_.ungetToken();
         }
-
-        pushSource(filename);
     }
 
-    void doOrigin() {
-        // Parse and create the new origin. It is relative to the previous
-        // one.
-        const MasterToken::StringRegion&
-            name_string(lexer_.getNextToken(MasterToken::QSTRING).
-                        getStringRegion());
-        active_origin_ = Name(name_string.beg, name_string.len,
-                              &active_origin_);
-        // Make sure there's the EOLN
-        eatUntilEOL(true);
+    void doInclude() {
+        // First, get the filename to include
+        const string
+            filename(lexer_.getNextToken(MasterToken::QSTRING).getString());
+
+        // There optionally can be an origin, that applies before the include.
+        doOrigin(true);
+
+        pushSource(filename);
     }
 
     void handleDirective(const char* directive, size_t length) {
         if (iequals(directive, "INCLUDE")) {
             doInclude();
         } else if (iequals(directive, "ORIGIN")) {
-            doOrigin();
+            doOrigin(false);
+            // The doOrigin doesn't do the cleanup of the line. This is
+            // because it's shared with the doInclude and that one can't do
+            // it.
+            eatUntilEOL(true);
         } else if (iequals(directive, "TTL")) {
             // TODO: Implement
             isc_throw(isc::NotImplemented,
diff --git a/src/lib/dns/tests/master_loader_unittest.cc b/src/lib/dns/tests/master_loader_unittest.cc
index 55588a2..9c75f3c 100644
--- a/src/lib/dns/tests/master_loader_unittest.cc
+++ b/src/lib/dns/tests/master_loader_unittest.cc
@@ -478,7 +478,7 @@ TEST_F(MasterLoaderTest, includeOriginRestore) {
     // Successfully load the data
     loader_->load();
     EXPECT_TRUE(loader_->loadedSucessfully());
-    EXPECT_TRUE(errors_.empty());
+    EXPECT_TRUE(errors_.empty()) << errors_[0];
     EXPECT_TRUE(warnings_.empty());
     // And check it's the correct data
     checkARR("www.example.org");