diff options
author | Senthil Kumaran <orsenthil@gmail.com> | 2010-07-11 03:12:43 (GMT) |
---|---|---|
committer | Senthil Kumaran <orsenthil@gmail.com> | 2010-07-11 03:12:43 (GMT) |
commit | 2ef16328e82ae9b9715251e41b8abf4b95ae6f20 (patch) | |
tree | 41c2f9303f4eb6efc9313044b1a54728fc2160c1 | |
parent | 0d62f5bbfe1c865165fd1285bb44fa29eb3b8bde (diff) | |
download | cpython-2ef16328e82ae9b9715251e41b8abf4b95ae6f20.zip cpython-2ef16328e82ae9b9715251e41b8abf4b95ae6f20.tar.gz cpython-2ef16328e82ae9b9715251e41b8abf4b95ae6f20.tar.bz2 |
Stricter verification for file based url scheme and reliance on ftp protocol.
-rw-r--r-- | Lib/test/test_urllib2.py | 3 | ||||
-rw-r--r-- | Lib/urllib/request.py | 3 |
2 files changed, 5 insertions, 1 deletions
diff --git a/Lib/test/test_urllib2.py b/Lib/test/test_urllib2.py index 080daa4..4d117c7 100644 --- a/Lib/test/test_urllib2.py +++ b/Lib/test/test_urllib2.py @@ -731,6 +731,8 @@ class HandlerTests(unittest.TestCase): ("file://ftp.example.com///foo.txt", False), # XXXX bug: fails with OSError, should be URLError ("file://ftp.example.com/foo.txt", False), + ("file://somehost//foo/something.txt", True), + ("file://localhost//foo/something.txt", False), ]: req = Request(url) try: @@ -741,6 +743,7 @@ class HandlerTests(unittest.TestCase): else: self.assertTrue(o.req is req) self.assertEqual(req.type, "ftp") + self.assertEqual(req.type is "ftp", ftp) def test_http(self): diff --git a/Lib/urllib/request.py b/Lib/urllib/request.py index f7c7416..c6767d0 100644 --- a/Lib/urllib/request.py +++ b/Lib/urllib/request.py @@ -1188,7 +1188,8 @@ class FileHandler(BaseHandler): # Use local file or FTP depending on form of URL def file_open(self, req): url = req.selector - if url[:2] == '//' and url[2:3] != '/': + if url[:2] == '//' and url[2:3] != '/' and (req.host and + req.host != 'localhost'): req.type = 'ftp' return self.parent.open(req) else: |