From 2288076cbcaa6c15781cd51e1e7c614423e8328c Mon Sep 17 00:00:00 2001
From: Denis Bilenko <denis.bilenko@gmail.com>
Date: Sun, 24 May 2009 17:37:00 +0700
Subject: [PATCH] import fix for GreenSSL from
 http://bitbucket.org/fzzzy/eventlet/changeset/4a429ca57699/

---
 eventlet/greenio.py          |  5 +++-
 greentest/test_socket_ssl.py | 47 ++++++++++++++++++++++++++++++++++++
 2 files changed, 51 insertions(+), 1 deletion(-)

diff --git a/eventlet/greenio.py b/eventlet/greenio.py
index c7796aa..794f64b 100644
--- a/eventlet/greenio.py
+++ b/eventlet/greenio.py
@@ -539,7 +539,10 @@ class GreenSSL(GreenSocket):
     read = read
 
     def write(self, data):
-        return self.sendall(data)
+        try:
+            return self.sendall(data)
+        except util.SSL.Error, ex:
+            raise socket.sslerror(str(ex))
 
     def server(self):
         return self.fd.server()
diff --git a/greentest/test_socket_ssl.py b/greentest/test_socket_ssl.py
index 9f4a1fc..818fdc4 100644
--- a/greentest/test_socket_ssl.py
+++ b/greentest/test_socket_ssl.py
@@ -118,11 +118,58 @@ def test_rude_shutdown():
     connector()
     t.join()
 
+
+def test_rude_shutdown__write():
+    if test_support.verbose:
+        print "test_rude_shutdown__variant ..."
+
+    from eventlet.green import threading
+
+    # Some random port to connect to.
+    PORT = [9934]
+
+    listener_ready = threading.Event()
+    listener_gone = threading.Event()
+
+    # `listener` runs in a thread.  It opens a socket listening on PORT, and
+    # sits in an accept() until the main thread connects.  Then it rudely
+    # closes the socket, and sets Event `listener_gone` to let the main thread
+    # know the socket is gone.
+    def listener():
+        s = socket.socket()
+        PORT[0] = test_support.bind_port(s, '', PORT[0])
+        s.listen(5)
+        listener_ready.set()
+        s.accept()
+        s = None # reclaim the socket object, which also closes it
+        listener_gone.set()
+
+    def connector():
+        listener_ready.wait()
+        s = socket.socket()
+        s.connect(('localhost', PORT[0]))
+        listener_gone.wait()
+        try:
+            ssl_sock = socket.ssl(s)
+            ssl_sock.write("hello")
+        except socket.sslerror:
+            pass
+        else:
+            raise test_support.TestFailed(
+                      'connecting to closed SSL socket should have failed')
+
+    t = threading.Thread(target=listener)
+    t.start()
+    connector()
+    t.join()
+
+
 class Test(unittest.TestCase):
 
     test_basic = lambda self: test_basic()
     test_timeout = lambda self: test_timeout()
     test_rude_shutdown = lambda self: test_rude_shutdown()
+    test_rude_shutdown__write = lambda self: test_rude_shutdown__write()
 
 def test_main():
     if not hasattr(socket, "ssl"):