mirror of https://github.com/python/cpython
Merged revisions 76726-76727 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk This merge changes the test harness to manually do the socket shutdown that was made automatic in trunk by enhancement patch r73638 (issue 6267). Patch modification by Scott Dial. ........ r76726 | r.david.murray | 2009-12-09 10:15:31 -0500 (Wed, 09 Dec 2009) | 6 lines Issue 5949: fixed IMAP4_SSL hang when the IMAP server response is missing proper end-of-line termination. Patch and tests by Scott Dial. The new tests include a test harness which will make it easier to add additional tests. ........ r76727 | r.david.murray | 2009-12-09 11:41:39 -0500 (Wed, 09 Dec 2009) | 2 lines Skip new imaplib SSL tests if ssl is not available. ........
This commit is contained in:
R. David Murray
parent
7c9a43e02e
commit
07ca761f62
|
|
@ -1001,6 +1001,8 @@ class IMAP4:
|
|||
raise self.abort('socket error: EOF')
|
||||
|
||||
# Protocol mandates all lines terminated by CRLF
|
||||
if not line.endswith('\r\n'):
|
||||
raise self.abort('socket error: unterminated line')
|
||||
|
||||
line = line[:-2]
|
||||
if __debug__:
|
||||
|
|
@ -1167,7 +1169,7 @@ else:
|
|||
while 1:
|
||||
char = self.sslobj.read(1)
|
||||
line.append(char)
|
||||
if char == "\n": return ''.join(line)
|
||||
if char in ("\n", ""): return ''.join(line)
|
||||
|
||||
|
||||
def send(self, data):
|
||||
|
|
|
|||
|
|
@ -1,11 +1,31 @@
|
|||
from test import test_support as support
|
||||
# If we end up with a significant number of tests that don't require
|
||||
# threading, this test module should be split. Right now we skip
|
||||
# them all if we don't have threading.
|
||||
threading = support.import_module('threading')
|
||||
|
||||
from contextlib import contextmanager
|
||||
import imaplib
|
||||
import os.path
|
||||
import select
|
||||
import socket
|
||||
import SocketServer
|
||||
import sys
|
||||
import time
|
||||
|
||||
from test import test_support
|
||||
from test_support import verbose
|
||||
import unittest
|
||||
|
||||
try:
|
||||
import ssl
|
||||
except ImportError:
|
||||
ssl = None
|
||||
|
||||
CERTFILE = None
|
||||
|
||||
|
||||
class TestImaplib(unittest.TestCase):
|
||||
|
||||
def test_that_Time2Internaldate_returns_a_result(self):
|
||||
# We can check only that it successfully produces a result,
|
||||
# not the correctness of the result itself, since the result
|
||||
|
|
@ -17,9 +37,176 @@ class TestImaplib(unittest.TestCase):
|
|||
imaplib.Time2Internaldate(t)
|
||||
|
||||
|
||||
def test_main():
|
||||
test_support.run_unittest(TestImaplib)
|
||||
if ssl:
|
||||
|
||||
class SecureTCPServer(SocketServer.TCPServer):
|
||||
|
||||
def get_request(self):
|
||||
newsocket, fromaddr = self.socket.accept()
|
||||
connstream = ssl.wrap_socket(newsocket,
|
||||
server_side=True,
|
||||
certfile=CERTFILE)
|
||||
return connstream, fromaddr
|
||||
|
||||
IMAP4_SSL = imaplib.IMAP4_SSL
|
||||
|
||||
else:
|
||||
|
||||
class SecureTCPServer:
|
||||
pass
|
||||
|
||||
IMAP4_SSL = None
|
||||
|
||||
class TimeoutStreamRequestHandler(SocketServer.StreamRequestHandler):
|
||||
|
||||
timeout = 1
|
||||
|
||||
def setup(self):
|
||||
self.connection = self.request
|
||||
if self.timeout is not None:
|
||||
self.connection.settimeout(self.timeout)
|
||||
self.rfile = self.connection.makefile('rb', self.rbufsize)
|
||||
self.wfile = self.connection.makefile('wb', self.wbufsize)
|
||||
|
||||
|
||||
class SimpleIMAPHandler(TimeoutStreamRequestHandler):
|
||||
|
||||
def _send(self, message):
|
||||
if verbose: print "SENT:", message.strip()
|
||||
self.wfile.write(message)
|
||||
|
||||
def handle(self):
|
||||
# Send a welcome message.
|
||||
self._send('* OK IMAP4rev1\r\n')
|
||||
while 1:
|
||||
# Gather up input until we receive a line terminator or we timeout.
|
||||
# Accumulate read(1) because it's simpler to handle the differences
|
||||
# between naked sockets and SSL sockets.
|
||||
line = ''
|
||||
while 1:
|
||||
try:
|
||||
part = self.rfile.read(1)
|
||||
if part == '':
|
||||
# Naked sockets return empty strings..
|
||||
return
|
||||
line += part
|
||||
except IOError:
|
||||
# ..but SSLSockets throw exceptions.
|
||||
return
|
||||
if line.endswith('\r\n'):
|
||||
break
|
||||
|
||||
if verbose: print 'GOT:', line.strip()
|
||||
splitline = line.split()
|
||||
tag = splitline[0]
|
||||
cmd = splitline[1]
|
||||
args = splitline[2:]
|
||||
|
||||
if hasattr(self, 'cmd_%s' % (cmd,)):
|
||||
getattr(self, 'cmd_%s' % (cmd,))(tag, args)
|
||||
else:
|
||||
self._send('%s BAD %s unknown\r\n' % (tag, cmd))
|
||||
|
||||
def cmd_CAPABILITY(self, tag, args):
|
||||
self._send('* CAPABILITY IMAP4rev1\r\n')
|
||||
self._send('%s OK CAPABILITY completed\r\n' % (tag,))
|
||||
|
||||
class BaseThreadedNetworkedTests(unittest.TestCase):
|
||||
|
||||
def make_server(self, addr, hdlr):
|
||||
|
||||
class MyServer(self.server_class):
|
||||
def handle_error(self, request, client_address):
|
||||
self.close_request(request)
|
||||
self.server_close()
|
||||
raise
|
||||
|
||||
if verbose: print "creating server"
|
||||
server = MyServer(addr, hdlr)
|
||||
self.assertEquals(server.server_address, server.socket.getsockname())
|
||||
|
||||
if verbose:
|
||||
print "server created"
|
||||
print "ADDR =", addr
|
||||
print "CLASS =", self.server_class
|
||||
print "HDLR =", server.RequestHandlerClass
|
||||
|
||||
t = threading.Thread(
|
||||
name='%s serving' % self.server_class,
|
||||
target=server.serve_forever,
|
||||
# Short poll interval to make the test finish quickly.
|
||||
# Time between requests is short enough that we won't wake
|
||||
# up spuriously too many times.
|
||||
kwargs={'poll_interval':0.01})
|
||||
t.daemon = True # In case this function raises.
|
||||
t.start()
|
||||
if verbose: print "server running"
|
||||
return server, t
|
||||
|
||||
def reap_server(self, server, thread):
|
||||
if verbose: print "waiting for server"
|
||||
server.shutdown()
|
||||
thread.join()
|
||||
if verbose: print "done"
|
||||
|
||||
@contextmanager
|
||||
def reaped_server(self, hdlr):
|
||||
server, thread = self.make_server((support.HOST, 0), hdlr)
|
||||
try:
|
||||
yield server
|
||||
finally:
|
||||
self.reap_server(server, thread)
|
||||
|
||||
def test_connect(self):
|
||||
with self.reaped_server(SimpleIMAPHandler) as server:
|
||||
client = self.imap_class(*server.server_address)
|
||||
client.shutdown()
|
||||
|
||||
def test_issue5949(self):
|
||||
|
||||
class EOFHandler(TimeoutStreamRequestHandler):
|
||||
def handle(self):
|
||||
# EOF without sending a complete welcome message.
|
||||
self.wfile.write('* OK')
|
||||
# explicitly shutdown. socket.close() merely releases
|
||||
# the socket and waits for GC to perform the actual close.
|
||||
self.request.shutdown(socket.SHUT_WR)
|
||||
|
||||
with self.reaped_server(EOFHandler) as server:
|
||||
self.assertRaises(imaplib.IMAP4.abort,
|
||||
self.imap_class, *server.server_address)
|
||||
|
||||
class ThreadedNetworkedTests(BaseThreadedNetworkedTests):
|
||||
|
||||
server_class = SocketServer.TCPServer
|
||||
imap_class = imaplib.IMAP4
|
||||
|
||||
|
||||
class ThreadedNetworkedTestsSSL(BaseThreadedNetworkedTests):
|
||||
|
||||
server_class = SecureTCPServer
|
||||
imap_class = IMAP4_SSL
|
||||
|
||||
|
||||
def test_main():
|
||||
tests = [TestImaplib]
|
||||
|
||||
if support.is_resource_enabled('network'):
|
||||
if ssl:
|
||||
global CERTFILE
|
||||
CERTFILE = os.path.join(os.path.dirname(__file__) or os.curdir,
|
||||
"keycert.pem")
|
||||
if not os.path.exists(CERTFILE):
|
||||
raise support.TestFailed("Can't read certificate files!")
|
||||
tests.append(ThreadedNetworkedTestsSSL)
|
||||
tests.append(ThreadedNetworkedTests)
|
||||
|
||||
threadinfo = support.threading_setup()
|
||||
|
||||
support.run_unittest(*tests)
|
||||
|
||||
support.threading_cleanup(*threadinfo)
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
support.use_resources = ['network']
|
||||
test_main()
|
||||
|
|
|
|||
|
|
@ -33,6 +33,9 @@ Core and Builtins
|
|||
Library
|
||||
-------
|
||||
|
||||
- Issue #5949: fixed IMAP4_SSL hang when the IMAP server response is
|
||||
missing proper end-of-line termination.
|
||||
|
||||
- Fix variations of extending deques: d.extend(d) d.extendleft(d) d+=d
|
||||
|
||||
- Issue #1923: Fixed the removal of meaningful spaces when PKG-INFO is
|
||||
|
|
|
|||
Loading…
Reference in New Issue