Rooba
/
proxy.py
mirror of https://github.com/abhinavsingh/proxy.py.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Register for signals only when running in main thread (#1087)

This commit is contained in:
Abhinav Singh 2022-02-11 00:18:40 +05:30 committed by GitHub
parent dc15dc1075
commit 0629fc43e9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
proxy/proxy.py
View File

@ -14,6 +14,7 @@ import time
import pprint import pprint
import signal import signal
import logging import logging
import threading
from typing import TYPE_CHECKING, Any, List, Optional, cast from typing import TYPE_CHECKING, Any, List, Optional, cast
from .core.ssh import SshTunnelListener, SshHttpProtocolHandler from .core.ssh import SshTunnelListener, SshHttpProtocolHandler
@ -255,6 +256,7 @@ class Proxy:
('', self.flags.tunnel_remote_port), ('', self.flags.tunnel_remote_port),
) )
# TODO: May be close listener fd as we don't need it now # TODO: May be close listener fd as we don't need it now
if threading.current_thread() == threading.main_thread():
self._register_signals() self._register_signals()
def shutdown(self) -> None: def shutdown(self) -> None: