From 2e89cf3465763ba9d048e8cac15cae14be201d27 Mon Sep 17 00:00:00 2001 From: Senthil Kumaran Date: Sun, 14 Nov 2010 03:14:52 +0000 Subject: [PATCH] Merged revisions 86450 via svnmerge from svn+ssh://pythondev@svn.python.org/python/branches/py3k ........ r86450 | senthil.kumaran | 2010-11-13 20:27:49 +0800 (Sat, 13 Nov 2010) | 3 lines Fix Issue5111 - Wrap the Ipv6 host with [] in the Host header ........ --- Lib/http/client.py | 7 +++++++ Lib/test/test_httplib.py | 19 +++++++++++++++++++ Misc/NEWS | 2 ++ 3 files changed, 28 insertions(+) diff --git a/Lib/http/client.py b/Lib/http/client.py index f273b03c6f6..bd092c215ef 100644 --- a/Lib/http/client.py +++ b/Lib/http/client.py @@ -873,6 +873,13 @@ class HTTPConnection: host_enc = self.host.encode("ascii") except UnicodeEncodeError: host_enc = self.host.encode("idna") + + # As per RFC 273, IPv6 address should be wrapped with [] + # when used as Host header + + if self.host.find(':') >= 0: + host_enc = b'[' + host_enc + b']' + if self.port == self.default_port: self.putheader('Host', host_enc) else: diff --git a/Lib/test/test_httplib.py b/Lib/test/test_httplib.py index 6b703dd43c3..ee283168b68 100644 --- a/Lib/test/test_httplib.py +++ b/Lib/test/test_httplib.py @@ -96,6 +96,25 @@ class HeaderTests(TestCase): conn.putheader('Content-length', 42) self.assertTrue(b'Content-length: 42' in conn._buffer) + def test_ipv6host_header(self): + # Default host header on IPv6 transaction should wrapped by [] if + # its actual IPv6 address + expected = b'GET /foo HTTP/1.1\r\nHost: [2001::]:81\r\n' \ + b'Accept-Encoding: identity\r\n\r\n' + conn = client.HTTPConnection('[2001::]:81') + sock = FakeSocket('') + conn.sock = sock + conn.request('GET', '/foo') + self.assertTrue(sock.data.startswith(expected)) + + expected = b'GET /foo HTTP/1.1\r\nHost: [2001:102A::]\r\n' \ + b'Accept-Encoding: identity\r\n\r\n' + conn = client.HTTPConnection('[2001:102A::]') + sock = FakeSocket('') + conn.sock = sock + conn.request('GET', '/foo') + self.assertTrue(sock.data.startswith(expected)) + class BasicTest(TestCase): def test_status_lines(self): diff --git a/Misc/NEWS b/Misc/NEWS index 7fb9717494c..fcf9ec7089e 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -155,6 +155,8 @@ C-API Library ------- +- Issue #5111: IPv6 Host in the Header is wrapped inside [ ]. Patch by Chandru. + - Issue #4471: Properly shutdown socket in IMAP.shutdown(). Patch by Lorenzo M. Catucci.