Merge pull request #27 from icgood/cancelled

Fix exception when dnsbl is cancelled

Author: icgood

LICENSE.md

@@ -1,6 +1,6 @@
 ## The MIT License (MIT)
 
-Copyright (c) 2020 Ian Good
+Copyright (c) 2021 Ian Good
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

doc/source/conf.py

@@ -22,7 +22,7 @@
 # -- Project information -----------------------------------------------------
 
 project = 'proxy-protocol'
-copyright = '2020, Ian Good'
+copyright = '2021, Ian Good'
 author = 'Ian Good'
 
 # The short X.Y version

proxyprotocol/server/protocol.py

@@ -193,8 +193,9 @@ def connection_made(self, transport: BaseTransport) -> None:
         self.dnsbl_task.add_done_callback(self._write_header)
 
     def _write_header(self, task: Task[Any]) -> None:
-        header = self.build_pp_header()
-        self.write(memoryview(header))
+        if not task.cancelled():
+            header = self.build_pp_header()
+            self.write(memoryview(header))
 
     def proxy_data(self, data: memoryview) -> None:
         self.downstream.write(data)

setup.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2020 Ian C. Good
+# Copyright (c) 2021 Ian C. Good
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -28,7 +28,7 @@
     license = f.read()
 
 setup(name='proxy-protocol',
-      version='0.7.1',
+      version='0.7.2',
       author='Ian Good',
       author_email='ian@icgood.net',
       description='PROXY protocol library with asyncio server implementation',