Merge pull request #2 from yukityan/dev

Dev
mike96265 · Aug 1, 2019 · ac8f78c · ac8f78c
2 parents 5dfcefb + d8f7a84
commit ac8f78c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/kcp/local.py b/kcp/local.py
@@ -30,7 +30,7 @@ def ds_factory():
         tunnel = protocol.tunnel
         return tunnel.create_connection()
 
-    server = await asyncio.start_server(functools.partial(open_pipe, ds_factory=ds_factory), host=config.local_address,
+    server = await asyncio.start_server(functools.partial(open_pipe, ds_factory=ds_factory), host=config.local,
                                         port=config.local_port)
 
     logging.info("starting local at %s:%s", config.local, config.local_port)

diff --git a/kcp/protocols.py b/kcp/protocols.py
@@ -106,13 +106,15 @@ def cb(fut):
 
                 task = loop.create_task(self.create_connection(conv))
                 task.add_done_callback(cb)
+                self.accept_dict[conv] = task
 
     def close(self, exc):
         sessions = self.sessions
         for session in sessions.values():
             session.protocol.eof_received()
             session.protocol.connection_lost(exc)
-            del sessions[session.conv]
+        sessions.clear()
+        self.accept_dict.clear()
         del sessions
         self.sessions = None
 

diff --git a/kcp/server.py b/kcp/server.py
@@ -22,7 +22,7 @@ def ds_factory():
 
     protocol = ServerDataGramHandlerProtocol(functools.partial(open_pipe, ds_factory=ds_factory))
     await loop.create_datagram_endpoint(lambda: protocol, local_addr=(config.local, config.local_port))
-    logging.info("start server at %s:%s", config.local_address, config.local_port)
+    logging.info("start server at %s:%s", config.local, config.local_port)
     updater.load_config(config)
     updater.run()
     for signame in {'SIGQUIT', 'SIGTERM'}: