Added pattern to allow all ports

Author: haproxytechblog

README.md

@@ -54,6 +54,7 @@ For the list of allowed origins, you can specify patterns such as:
 | schema, domain name, and port | http://mydomain.com:8080 | allow only HTTP of mydomain.com from port 8080                          |
 | dot and domain name           | .mydomain.com            | allow ALL subdomains of mydomain.com                                    |
 | dot, domain name, and port    | .mydomain.com:443        | allow ALL subdomains of mydomain.com from default HTTPS source port     |
+| port is *                     | http://mydomain.com:*    | allow only HTTP of mydomain.com on ANY ports                            |
 
 ## Examples
 

lib/cors.lua

@@ -56,6 +56,11 @@ M.build_pattern = function(pattern)
       pattern = "//" .. pattern
     end
 
+    -- an asterisk for the port means allow all ports
+    if string.find(pattern, "[:]+%*$") ~= nil then
+      pattern = pattern:gsub("[:]+%*$", "[:]+[0-9]+")
+    end
+
     -- escape dots in pattern
     pattern = pattern:gsub("%.", "%%.")
 

tests/cors_tests.lua

@@ -91,6 +91,11 @@ function test_build_pattern_6()
   luaunit.assertEquals(result, "%.test%.com:8080$")
 end
 
+function test_build_pattern_7()
+  local result = cors.build_pattern("http://test.com:*")
+  luaunit.assertEquals(result, "http://test%.com[:]+[0-9]+$")
+end
+
 function test_get_allowed_origin_case_1()
   local result = cors.get_allowed_origin("http://test.com", {"http://test.com"})
   luaunit.assertEquals(result, "http://test.com")
@@ -146,5 +151,11 @@ function test_get_allowed_origin_case_11()
   luaunit.assertEquals(result, "*")
 end
 
+
+function test_get_allowed_origin_case_12()
+  local result = cors.get_allowed_origin("http://test.com:8080", {"http://test.com:*"})
+  luaunit.assertEquals(result, "http://test.com:8080")
+end
+
 -- this line must go at the end
 os.exit(luaunit.LuaUnit.run())