684f3be000
In debug mode, BaseEventLoop._run_once() now sets the BaseEventLoop._current_handle attribute to the handle currently executed. In release mode or when no handle is executed, the attribute is None. BaseEventLoop.default_exception_handler() displays the traceback of the current handle if available.
95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
"""Test service that accepts connections and reads all data off them."""
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
|
|
from asyncio import *
|
|
|
|
ARGS = argparse.ArgumentParser(description="TCP data sink example.")
|
|
ARGS.add_argument(
|
|
'--tls', action='store_true', dest='tls',
|
|
default=False, help='Use TLS with a self-signed cert')
|
|
ARGS.add_argument(
|
|
'--iocp', action='store_true', dest='iocp',
|
|
default=False, help='Use IOCP event loop (Windows only)')
|
|
ARGS.add_argument(
|
|
'--host', action='store', dest='host',
|
|
default='127.0.0.1', help='Host name')
|
|
ARGS.add_argument(
|
|
'--port', action='store', dest='port',
|
|
default=1111, type=int, help='Port number')
|
|
ARGS.add_argument(
|
|
'--maxsize', action='store', dest='maxsize',
|
|
default=16*1024*1024, type=int, help='Max total data size')
|
|
|
|
server = None
|
|
args = None
|
|
|
|
|
|
def dprint(*args):
|
|
print('sink:', *args, file=sys.stderr)
|
|
|
|
|
|
class Service(Protocol):
|
|
|
|
def connection_made(self, tr):
|
|
dprint('connection from', tr.get_extra_info('peername'))
|
|
dprint('my socket is', tr.get_extra_info('sockname'))
|
|
self.tr = tr
|
|
self.total = 0
|
|
|
|
def data_received(self, data):
|
|
if data == b'stop':
|
|
dprint('stopping server')
|
|
server.close()
|
|
self.tr.close()
|
|
return
|
|
self.total += len(data)
|
|
dprint('received', len(data), 'bytes; total', self.total)
|
|
if self.total > args.maxsize:
|
|
dprint('closing due to too much data')
|
|
self.tr.close()
|
|
|
|
def connection_lost(self, how):
|
|
dprint('closed', repr(how))
|
|
|
|
|
|
@coroutine
|
|
def start(loop, host, port):
|
|
global server
|
|
sslctx = None
|
|
if args.tls:
|
|
import ssl
|
|
# TODO: take cert/key from args as well.
|
|
here = os.path.join(os.path.dirname(__file__), '..', 'tests')
|
|
sslctx = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
|
|
sslctx.options |= ssl.OP_NO_SSLv2
|
|
sslctx.load_cert_chain(
|
|
certfile=os.path.join(here, 'ssl_cert.pem'),
|
|
keyfile=os.path.join(here, 'ssl_key.pem'))
|
|
|
|
server = yield from loop.create_server(Service, host, port, ssl=sslctx)
|
|
dprint('serving TLS' if sslctx else 'serving',
|
|
[s.getsockname() for s in server.sockets])
|
|
yield from server.wait_closed()
|
|
|
|
|
|
def main():
|
|
global args
|
|
args = ARGS.parse_args()
|
|
if args.iocp:
|
|
from asyncio.windows_events import ProactorEventLoop
|
|
loop = ProactorEventLoop()
|
|
set_event_loop(loop)
|
|
else:
|
|
loop = get_event_loop()
|
|
try:
|
|
loop.run_until_complete(start(loop, args.host, args.port))
|
|
finally:
|
|
loop.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|