bpo-35121: prefix dot in domain for proper subdomain validation (GH-10258)

Don't send cookies of domain A without Domain attribute to domain B when domain A is a suffix match of domain B while using a cookiejar with `http.cookiejar.DefaultCookiePolicy` policy.  Patch by Karthikeyan Singaravelan.
This commit is contained in:
Xtreak 2019-03-10 07:39:48 +05:30 committed by Ned Deily
parent 738c19f4c5
commit ca7fe50635
3 changed files with 45 additions and 2 deletions

View File

@ -1148,6 +1148,11 @@ class DefaultCookiePolicy(CookiePolicy):
req_host, erhn = eff_request_host(request) req_host, erhn = eff_request_host(request)
domain = cookie.domain domain = cookie.domain
if domain and not domain.startswith("."):
dotdomain = "." + domain
else:
dotdomain = domain
# strict check of non-domain cookies: Mozilla does this, MSIE5 doesn't # strict check of non-domain cookies: Mozilla does this, MSIE5 doesn't
if (cookie.version == 0 and if (cookie.version == 0 and
(self.strict_ns_domain & self.DomainStrictNonDomain) and (self.strict_ns_domain & self.DomainStrictNonDomain) and
@ -1160,7 +1165,7 @@ class DefaultCookiePolicy(CookiePolicy):
_debug(" effective request-host name %s does not domain-match " _debug(" effective request-host name %s does not domain-match "
"RFC 2965 cookie domain %s", erhn, domain) "RFC 2965 cookie domain %s", erhn, domain)
return False return False
if cookie.version == 0 and not ("."+erhn).endswith(domain): if cookie.version == 0 and not ("."+erhn).endswith(dotdomain):
_debug(" request-host %s does not match Netscape cookie domain " _debug(" request-host %s does not match Netscape cookie domain "
"%s", req_host, domain) "%s", req_host, domain)
return False return False
@ -1174,7 +1179,11 @@ class DefaultCookiePolicy(CookiePolicy):
req_host = "."+req_host req_host = "."+req_host
if not erhn.startswith("."): if not erhn.startswith("."):
erhn = "."+erhn erhn = "."+erhn
if not (req_host.endswith(domain) or erhn.endswith(domain)): if domain and not domain.startswith("."):
dotdomain = "." + domain
else:
dotdomain = domain
if not (req_host.endswith(dotdomain) or erhn.endswith(dotdomain)):
#_debug(" request domain %s does not match cookie domain %s", #_debug(" request domain %s does not match cookie domain %s",
# req_host, domain) # req_host, domain)
return False return False

View File

@ -440,6 +440,7 @@ class CookieTests(unittest.TestCase):
("http://foo.bar.com/", ".foo.bar.com", True), ("http://foo.bar.com/", ".foo.bar.com", True),
("http://foo.bar.com/", "foo.bar.com", True), ("http://foo.bar.com/", "foo.bar.com", True),
("http://foo.bar.com/", ".bar.com", True), ("http://foo.bar.com/", ".bar.com", True),
("http://foo.bar.com/", "bar.com", True),
("http://foo.bar.com/", "com", True), ("http://foo.bar.com/", "com", True),
("http://foo.com/", "rhubarb.foo.com", False), ("http://foo.com/", "rhubarb.foo.com", False),
("http://foo.com/", ".foo.com", True), ("http://foo.com/", ".foo.com", True),
@ -450,6 +451,8 @@ class CookieTests(unittest.TestCase):
("http://foo/", "foo", True), ("http://foo/", "foo", True),
("http://foo/", "foo.local", True), ("http://foo/", "foo.local", True),
("http://foo/", ".local", True), ("http://foo/", ".local", True),
("http://barfoo.com", ".foo.com", False),
("http://barfoo.com", "foo.com", False),
]: ]:
request = urllib.request.Request(url) request = urllib.request.Request(url)
r = pol.domain_return_ok(domain, request) r = pol.domain_return_ok(domain, request)
@ -984,6 +987,33 @@ class CookieTests(unittest.TestCase):
c.add_cookie_header(req) c.add_cookie_header(req)
self.assertFalse(req.has_header("Cookie")) self.assertFalse(req.has_header("Cookie"))
c.clear()
pol.set_blocked_domains([])
req = urllib.request.Request("http://acme.com/")
res = FakeResponse(headers, "http://acme.com/")
cookies = c.make_cookies(res, req)
c.extract_cookies(res, req)
self.assertEqual(len(c), 1)
req = urllib.request.Request("http://acme.com/")
c.add_cookie_header(req)
self.assertTrue(req.has_header("Cookie"))
req = urllib.request.Request("http://badacme.com/")
c.add_cookie_header(req)
self.assertFalse(pol.return_ok(cookies[0], req))
self.assertFalse(req.has_header("Cookie"))
p = pol.set_blocked_domains(["acme.com"])
req = urllib.request.Request("http://acme.com/")
c.add_cookie_header(req)
self.assertFalse(req.has_header("Cookie"))
req = urllib.request.Request("http://badacme.com/")
c.add_cookie_header(req)
self.assertFalse(req.has_header("Cookie"))
def test_secure(self): def test_secure(self):
for ns in True, False: for ns in True, False:
for whitespace in " ", "": for whitespace in " ", "":

View File

@ -0,0 +1,4 @@
Don't send cookies of domain A without Domain attribute to domain B
when domain A is a suffix match of domain B while using a cookiejar
with :class:`http.cookiejar.DefaultCookiePolicy` policy. Patch by
Karthikeyan Singaravelan.