def test_startTLS(self):
"""
POP3Client.startTLS starts a TLS session over its existing TCP
connection.
"""
sf = TLSServerFactory()
sf.protocol.output = [
['+OK'], # Server greeting
['+OK', 'STLS', '.'], # CAPA response
['+OK'], # STLS response
['+OK', '.'], # Second CAPA response
['+OK'] # QUIT response
]
sf.protocol.context = ServerTLSContext()
port = reactor.listenTCP(0, sf, interface='127.0.0.1')
self.addCleanup(port.stopListening)
H = port.getHost().host
P = port.getHost().port
connLostDeferred = defer.Deferred()
cp = SimpleClient(defer.Deferred(), ClientTLSContext())
def connectionLost(reason):
SimpleClient.connectionLost(cp, reason)
connLostDeferred.callback(None)
cp.connectionLost = connectionLost
cf = protocol.ClientFactory()
cf.protocol = lambda: cp
conn = reactor.connectTCP(H, P, cf)
def cbConnected(ignored):
log.msg("Connected to server; starting TLS")
return cp.startTLS()
def cbStartedTLS(ignored):
log.msg("Started TLS; disconnecting")
return cp.quit()
def cbDisconnected(ign):
log.msg("Disconnected; asserting correct input received")
self.assertEqual(sf.input, ['CAPA', 'STLS', 'CAPA', 'QUIT'])
def cleanup(result):
log.msg(
"Asserted correct input; disconnecting client and shutting down server"
)
conn.disconnect()
return connLostDeferred
cp.deferred.addCallback(cbConnected)
cp.deferred.addCallback(cbStartedTLS)
cp.deferred.addCallback(cbDisconnected)
cp.deferred.addBoth(cleanup)
return cp.deferred
def testTLS(self):
clientCTX = ClientTLSContext()
serverCTX = ServerTLSContext()
client = NoticeTLSClient(contextFactory=clientCTX)
server = DummyESMTP(contextFactory=serverCTX)
def check(ignored):
self.assertEquals(client.tls, True)
self.assertEquals(server.startedTLS, True)
return self.loopback(server, client).addCallback(check)
def testStartTLS(self):
sf = TLSServerFactory()
sf.protocol.output = [
['+OK'], # Server greeting
['+OK', 'STLS', '.'], # CAPA response
['+OK'], # STLS response
['+OK', '.'], # Second CAPA response
['+OK'] # QUIT response
]
sf.protocol.context = ServerTLSContext()
port = reactor.listenTCP(0, sf, interface='127.0.0.1')
H = port.getHost().host
P = port.getHost().port
cp = SimpleClient(defer.Deferred(), ClientTLSContext())
cf = protocol.ClientFactory()
cf.protocol = lambda: cp
conn = reactor.connectTCP(H, P, cf)
def cbConnected(ignored):
log.msg("Connected to server; starting TLS")
return cp.startTLS()
def cbStartedTLS(ignored):
log.msg("Started TLS; disconnecting")
return cp.quit()
def cbDisconnected(ign):
log.msg("Disconnected; asserting correct input received")
self.assertEquals(sf.input, ['CAPA', 'STLS', 'CAPA', 'QUIT'])
def cleanup(result):
log.msg(
"Asserted correct input; disconnecting client and shutting down server"
)
conn.disconnect()
def cbShutdown(ignored):
log.msg("Shut down server")
return result
return defer.maybeDeferred(
port.stopListening).addCallback(cbShutdown)
cp.deferred.addCallback(cbConnected)
cp.deferred.addCallback(cbStartedTLS)
cp.deferred.addCallback(cbDisconnected)
cp.deferred.addBoth(cleanup)
return cp.deferred
def main():
hostname = raw_input('POP3 Server Hostname: ')
port = raw_input('POP3 port: ')
username = raw_input('POP3 Username: '******'POP3 Password: '******'use SSL (Yes/No): ')
onConn = defer.Deferred(
).addCallback(cbServerGreeting, username, password
).addErrback(ebConnection
)
factory = SimplePOP3ClientFactory(username, onConn)
if useSSL.lower() == "yes":
conn = reactor.connectSSL(hostname, int(port), factory, ClientTLSContext())
else:
conn = reactor.connectTCP(hostname, int(port), factory)
reactor.run()
def lineReceived(self, line):
if line == b"READY":
self.transport.startTLS(ClientTLSContext(), self.factory.client)
for l in self.posttext:
self.sendLine(l)
self.transport.loseConnection()
def serverGreeting(self, challenge):
if self.greetDeferred is not None:
d, self.greetDeferred = self.greetDeferred, None
self.context = ClientTLSContext()
d.callback(self)
