util.ip: Fix protocol detection of IPv6 addresses beginning with : [Backported from...

[prosody.git] / util / ip.lua

diff --git a/util/ip.lua b/util/ip.lua
index 856bf03498261860db5796afccac66258c33beae..043303eebb311ce6a1d07bcc23b9bdf97db4e620 100644 (file)
--- a/util/ip.lua
+++ b/util/ip.lua
@@ -12,7 +12,17 @@ local ip_mt = { __index = function (ip, key) return (ip_methods[key])(ip); end,
 local hex2bits = { ["0"] = "0000", ["1"] = "0001", ["2"] = "0010", ["3"] = "0011", ["4"] = "0100", ["5"] = "0101", ["6"] = "0110", ["7"] = "0111", ["8"] = "1000", ["9"] = "1001", ["A"] = "1010", ["B"] = "1011", ["C"] = "1100", ["D"] = "1101", ["E"] = "1110", ["F"] = "1111" };
 
 local function new_ip(ipStr, proto)
-       if proto ~= "IPv4" and proto ~= "IPv6" then
+       if not proto then
+               local sep = ipStr:match("^%x+(.)");
+               if sep == ":" or (not(sep) and ipStr:sub(1,1) == ":") then
+                       proto = "IPv6"
+               elseif sep == "." then
+                       proto = "IPv4"
+               end
+               if not proto then
+                       return nil, "invalid address";
+               end
+       elseif proto ~= "IPv4" and proto ~= "IPv6" then
                return nil, "invalid protocol";
        end
        if proto == "IPv6" and ipStr:find('.', 1, true) then