Miroirs/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp synced 2025-02-13 20:48:33 +01:00
Code Issues Projects Releases Packages Wiki Activity

try this

Browse source
This commit is contained in:
coletdjnz 2024-04-04 19:34:28 +13:00
parent 41add1d7af
commit 6c3140a8c1
No known key found for this signature in database
GPG key ID: 91984263BB39894A
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
test/test_http_proxy.py
View file

@ -156,9 +156,8 @@ class HTTPConnectProxyHandler(BaseHTTPRequestHandler, HTTPProxyAuthMixin):
'path': self.path, 'path': self.path,
'proxy': ':'.join(str(y) for y in self.connection.getsockname()), 'proxy': ':'.join(str(y) for y in self.connection.getsockname()),
} }
request = self.request
self.request_handler(self.request, self.client_address, self.server, proxy_info=proxy_info) self.request_handler(self.request, self.client_address, self.server, proxy_info=proxy_info)
self.server.close_request(request) self.server.close_request(self.request)
class HTTPSConnectProxyHandler(HTTPConnectProxyHandler): class HTTPSConnectProxyHandler(HTTPConnectProxyHandler):
@ -167,8 +166,13 @@ class HTTPSConnectProxyHandler(HTTPConnectProxyHandler):
sslctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER) sslctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
sslctx.load_cert_chain(certfn, None) sslctx.load_cert_chain(certfn, None)
request = sslctx.wrap_socket(request, server_side=True) request = sslctx.wrap_socket(request, server_side=True)
self._original_request = request
super().__init__(request, *args, **kwargs) super().__init__(request, *args, **kwargs)
def do_CONNECT(self):
super().do_CONNECT()
self.server.close_request(self._original_request)
@contextlib.contextmanager @contextlib.contextmanager
def proxy_server(proxy_server_class, request_handler, bind_ip=None, **proxy_server_kwargs): def proxy_server(proxy_server_class, request_handler, bind_ip=None, **proxy_server_kwargs):