Fix bugs in IPv6 address parsing

* Multiple colons should not be accepted in state 4.
* After parsing a digit group and encountering a colon,
  we need allow either another colon or the beginning
  of either another group or an IPv4 address.  Add state 5
  to handle this.

1 files changed, 27 insertions, 2 deletions

diff --git a/src/Uri.cpp b/src/Uri.cpp
index 30c9671..d9b2b46 100644
--- a/src/Uri.cpp
+++ b/src/Uri.cpp
@@ -317,7 +317,7 @@ namespace {
                     if (c == ':') {
                         numDigits = 0;
                         ++numGroups;
-                        state = 2;
+                        state = 5;
                     } else if (HEXDIG.Contains(c)) {
                         if (++numDigits > 4) {
                             return false;
@@ -331,6 +331,7 @@ namespace {
                     if (c == ':') {
                         numDigits = 0;
                         ++numGroups;
+                        state = 2;
                     } else if (c == '.') {
                         ipv4AddressEncountered = true;
                         break;
@@ -347,6 +348,26 @@ namespace {
                         return false;
                     }
                 } break;
+
+                case 5: { // were in a group, encountered one colon
+                    if (c == ':') {
+                        if (doubleColonEncountered) {
+                            return false;
+                        } else {
+                            doubleColonEncountered = true;
+                            state = 2;
+                        }
+                    } else if (DIGIT.Contains(c)) {
+                        potentialIpv4AddressStart = position;
+                        ++numDigits;
+                        state = 4;
+                    } else if (HEXDIG.Contains(c)) {
+                        ++numDigits;
+                        state = 3;
+                    } else {
+                        return false;
+                    }
+                } break;
             }
             if (ipv4AddressEncountered) {
                 break;
@@ -359,7 +380,11 @@ namespace {
         }
         if (
             (position == address.length())
-            && (state == 1)
+            && (
+                (state == 1)
+                || (state == 2)
+                || (state == 5)
+            )
         ) { // trailing single colon
             return false;
         }