def authenticate(self, f: http.HTTPFlow) -> bool: valid_credentials = self.check(f) if valid_credentials: f.metadata["proxyauth"] = valid_credentials del f.request.headers[self.which_auth_header()] return True else: f.response = self.auth_required_response() return False
def requestheaders(self, f: http.HTTPFlow) -> None: if self.enabled(): # Is this connection authenticated by a previous HTTP CONNECT? if f.client_conn in self.authenticated: f.metadata["proxyauth"] = self.authenticated[f.client_conn] return self.authenticate(f)