#!/usr/bin/env python # The example SPDY server. You need Python 3.3 or later because we # use TLS NPN. Put private key and certificate file in the current # working directory. import socket import threading import socketserver import ssl import io import spdylay # private key file KEY_FILE='server.key' # certificate file CERT_FILE='server.crt' def send_cb(session, data): ssctrl = session.user_data ssctrl.sock.sendall(data) return len(data) def read_cb(session, stream_id, length, source): return source.read(length) def on_ctrl_recv_cb(session, frame): ssctrl = session.user_data if frame.frame_type == spdylay.SYN_STREAM: stctrl = StreamCtrl(frame.stream_id) stctrl.headers.extend(frame.nv) ssctrl.streams[frame.stream_id] = stctrl def on_stream_close_cb(session, stream_id, status_code): ssctrl = session.user_data if stream_id in ssctrl.streams: del ssctrl.streams[stream_id] def on_request_recv_cb(session, stream_id): ssctrl = session.user_data if stream_id in ssctrl.streams: stctrl = ssctrl.streams[stream_id] for name, value in stctrl.headers: if name == b'user-agent': user_agent = value.decode('utf-8') break else: user_agent = '' html = '''\ SPDY FTW

SPDY FTW

The age of HTTP/1.1 is over. The time of SPDY has come.

Your browser {} supports SPDY.

'''.format(user_agent) data_prd = spdylay.DataProvider(io.BytesIO(bytes(html, 'utf-8')), read_cb) stctrl.data_prd = data_prd nv = [(b':status', b'200 OK'), (b':version', b'HTTP/1.1'), (b'server', b'python+spdylay')] session.submit_response(stream_id, nv, data_prd) class StreamCtrl: def __init__(self, stream_id): self.stream_id = stream_id self.data_prd = None self.headers = [] class SessionCtrl: def __init__(self, sock): self.sock = sock self.streams = {} class ThreadedSPDYRequestHandler(socketserver.BaseRequestHandler): def handle(self): ctx = ssl.SSLContext(ssl.PROTOCOL_SSLv23) ctx.load_cert_chain(CERT_FILE, KEY_FILE) ctx.set_npn_protocols(['spdy/3', 'spdy/2']) sock = ctx.wrap_socket(self.request, server_side=True) if sock.selected_npn_protocol() == 'spdy/3': version = spdylay.PROTO_SPDY3 elif sock.selected_npn_protocol() == 'spdy/2': version = spdylay.PROTO_SPDY2 else: return ssctrl = SessionCtrl(sock) session = spdylay.Session(spdylay.SERVER, version, send_cb=send_cb, on_ctrl_recv_cb=on_ctrl_recv_cb, on_stream_close_cb=on_stream_close_cb, on_request_recv_cb=on_request_recv_cb, user_data=ssctrl) session.submit_settings(\ spdylay.FLAG_SETTINGS_NONE, [(spdylay.SETTINGS_MAX_CONCURRENT_STREAMS, spdylay.ID_FLAG_SETTINGS_NONE, 100)]) while session.want_read() or session.want_write(): data = sock.recv(4096) if data: session.recv(data) session.send() else: break class ThreadedSPDYServer(socketserver.ThreadingMixIn, socketserver.TCPServer): def __init__(self, svaddr, handler): self.allow_reuse_address = True socketserver.TCPServer.__init__(self, svaddr, handler) if __name__ == "__main__": # Port 0 means to select an arbitrary unused port HOST, PORT = "localhost", 3000 server = ThreadedSPDYServer((HOST, PORT), ThreadedSPDYRequestHandler) ip, port = server.server_address # Start a thread with the server -- that thread will then start one # more thread for each request server_thread = threading.Thread(target=server.serve_forever) # Exit the server thread when the main thread terminates #server_thread.daemon = True server_thread.start()