diff --git a/README.md b/README.md index f53aa4c..0673f55 100644 --- a/README.md +++ b/README.md @@ -71,8 +71,8 @@ Start the checker using uvicorn --reload example:checker.app ``` -And browse to [http://localhost:8000/docs] to explore the web interface, which allows you to send requests to the checker. +And browse to (http://localhost:8000/docs) to explore the web interface, which allows you to send requests to the checker. ## Documentation -You can check out the full tutorial at [https://enowars.github.io/docs] \ No newline at end of file +You can check out the full tutorial at (https://enowars.github.io/docs) diff --git a/enochecker3/enochecker.py b/enochecker3/enochecker.py index 1a8c86f..1974e59 100644 --- a/enochecker3/enochecker.py +++ b/enochecker3/enochecker.py @@ -64,7 +64,7 @@ TIMEOUT_BUFFER = 2 -AsyncSocket = Tuple[asyncio.StreamReader, asyncio.StreamWriter] +AsyncSocket = AsyncIterator[Tuple[asyncio.StreamReader, asyncio.StreamWriter]] class EnocheckerException(Exception): @@ -325,9 +325,7 @@ def _get_logger_adapter( ) @contextlib.asynccontextmanager - async def _get_async_socket( - self, task: BaseCheckerTaskMessage - ) -> AsyncIterator[AsyncSocket]: + async def _get_async_socket(self, task: BaseCheckerTaskMessage) -> AsyncSocket: try: conn = await asyncio.streams.open_connection( task.address, self.service_port diff --git a/example.py b/example.py index 06816df..c178092 100644 --- a/example.py +++ b/example.py @@ -55,4 +55,4 @@ async def exploit_test(searcher: FlagSearcher, client: AsyncClient) -> Optional[ return flag if __name__ == "__main__": - checker.run() \ No newline at end of file + checker.run() diff --git a/setup.py b/setup.py index c2f2c00..aba8ccc 100755 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ setuptools.setup( name="enochecker3", - version="0.1.0", + version="0.1.1", author="ldruschk", author_email="ldruschk@posteo.de", description="FastAPI based library for building async python checkers for the EnoEngine A/D CTF Framework",