Exemplo n.º 1
0
    def test_ssl_connect2 (self):
        def accept_once (listenfd):
            try:
                conn, addr = listenfd.accept()
                conn.write('hello\r\n')
                shutdown_safe(conn)
                conn.close()
            finally:
                shutdown_safe(listenfd)
                listenfd.close()

        server = ssl_listener(('0.0.0.0', 0),
                              self.certificate_file,
                              self.private_key_file)
        greenthread.spawn(accept_once, server)

        raw_client = connect(('127.0.0.1', server.getsockname()[1]))
        client = util.wrap_ssl(raw_client)
        fd = socket._fileobject(client, 'rb', 8192)

        assert fd.readline() == 'hello\r\n'
        try:
            self.assertEquals('', fd.read(10))
        except SSL.ZeroReturnError:
            # if it's a GreenSSL object it'll do this
            pass
        shutdown_safe(client)
        client.close()
Exemplo n.º 2
0
 def serve ():
     sock, addr = listener.accept()
     self.assertEquals(sock.recv(6), 'before')
     sock_ssl = util.wrap_ssl(sock, certificate_file, private_key_file,
                              server_side = True)
     sock_ssl.do_handshake()
     self.assertEquals(sock_ssl.read(6), 'during')
     sock2 = sock_ssl.unwrap()
     self.assertEquals(sock2.recv(5), 'after')
     sock2.close()
Exemplo n.º 3
0
    def test_duplex_response (self):
        def serve (listener):
            sock, addr = listener.accept()
            stuff = sock.read(8192)
            sock.write('response')

        sock = listen_ssl_socket()

        server_coro = evy.spawn(serve, sock)

        client = util.wrap_ssl(connect(('127.0.0.1', sock.getsockname()[1])))
        client.write('line 1\r\nline 2\r\n\r\n')
        self.assertEquals(client.read(8192), 'response')
        server_coro.wait()
Exemplo n.º 4
0
    def test_ssl_connect (self):
        def serve (listener):
            sock, addr = listener.accept()
            stuff = sock.read(8192)

        sock = listen_ssl_socket()
        server_coro = evy.spawn(serve, sock)

        raw_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        ssl_client = util.wrap_ssl(raw_client)
        ssl_client.connect(('127.0.0.1', sock.getsockname()[1]))
        ssl_client.write('abc')
        shutdown_safe(ssl_client)
        ssl_client.close()
        server_coro.wait()
Exemplo n.º 5
0
    def test_ssl_close (self):
        def serve (listener):
            sock, addr = listener.accept()
            stuff = sock.read(8192)
            try:
                self.assertEquals("", sock.read(8192))
            except SSL.ZeroReturnError:
                pass

        sock = listen_ssl_socket()

        server_coro = evy.spawn(serve, sock)

        raw_client = connect(('127.0.0.1', sock.getsockname()[1]))
        client = util.wrap_ssl(raw_client)
        client.write('X')
        shutdown_safe(client)
        client.close()
        server_coro.wait()
Exemplo n.º 6
0
    def test_ssl_unwrap (self):
        def serve ():
            sock, addr = listener.accept()
            self.assertEquals(sock.recv(6), 'before')
            sock_ssl = util.wrap_ssl(sock, certificate_file, private_key_file,
                                     server_side = True)
            sock_ssl.do_handshake()
            self.assertEquals(sock_ssl.read(6), 'during')
            sock2 = sock_ssl.unwrap()
            self.assertEquals(sock2.recv(5), 'after')
            sock2.close()

        listener = listen(('127.0.0.1', 0))
        server_coro = evy.spawn(serve)
        client = connect((listener.getsockname()))
        client.send('before')
        client_ssl = util.wrap_ssl(client)
        client_ssl.do_handshake()
        client_ssl.write('during')
        client2 = client_ssl.unwrap()
        client2.send('after')
        server_coro.wait()
Exemplo n.º 7
0
def listen_ssl_socket (address = ('127.0.0.1', 0)):
    sock = util.wrap_ssl(socket.socket(), certificate_file, private_key_file, True)
    sock.bind(address)
    sock.listen(50)

    return sock
Exemplo n.º 8
0
        def ssl (sock, certificate = None, private_key = None):
            from evy import util

            wrapped = util.wrap_ssl(sock, certificate, private_key)
            return GreenSSLObject(wrapped)