From 8da269de88b43f1e8464a6bce83b1fca65d406bb Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Sat, 28 Feb 2015 20:32:33 +0100
Subject: [PATCH] Set XMLStream.socket after the SSL connection is made too

Fixes SCRAM-SHA-1-PLUS.
---
 slixmpp/xmlstream/xmlstream.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/slixmpp/xmlstream/xmlstream.py b/slixmpp/xmlstream/xmlstream.py
index aec89c7f..2b165060 100644
--- a/slixmpp/xmlstream/xmlstream.py
+++ b/slixmpp/xmlstream/xmlstream.py
@@ -504,6 +504,7 @@ class XMLStream(asyncio.BaseProtocol):
                 der_cert = transp.get_extra_info("socket").getpeercert(True)
                 pem_cert = ssl.DER_cert_to_PEM_cert(der_cert)
                 self.event('ssl_cert', pem_cert)
+                self.socket = self.transport.get_extra_info("socket")
 
         asyncio.async(ssl_coro())