| hg-reviewers |
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| def extsetup(ui): | def extsetup(ui): | ||||
| # Change the base HTTP server class so various events can be performed. | # Change the base HTTP server class so various events can be performed. | ||||
| # See SocketServer.BaseServer for how the specially named methods work. | # See SocketServer.BaseServer for how the specially named methods work. | ||||
| class badserver(server.MercurialHTTPServer): | class badserver(server.MercurialHTTPServer): | ||||
| def __init__(self, ui, *args, **kwargs): | def __init__(self, ui, *args, **kwargs): | ||||
| self._ui = ui | self._ui = ui | ||||
| super(badserver, self).__init__(ui, *args, **kwargs) | super(badserver, self).__init__(ui, *args, **kwargs) | ||||
| recvbytes = self._ui.config('badserver', 'closeafterrecvbytes') | recvbytes = self._ui.config(b'badserver', b'closeafterrecvbytes') | ||||
| recvbytes = recvbytes.split(',') | recvbytes = recvbytes.split(',') | ||||
| self.closeafterrecvbytes = [int(v) for v in recvbytes if v] | self.closeafterrecvbytes = [int(v) for v in recvbytes if v] | ||||
| sendbytes = self._ui.config('badserver', 'closeaftersendbytes') | sendbytes = self._ui.config(b'badserver', b'closeaftersendbytes') | ||||
| sendbytes = sendbytes.split(',') | sendbytes = sendbytes.split(',') | ||||
| self.closeaftersendbytes = [int(v) for v in sendbytes if v] | self.closeaftersendbytes = [int(v) for v in sendbytes if v] | ||||
| # Need to inherit object so super() works. | # Need to inherit object so super() works. | ||||
| class badrequesthandler(self.RequestHandlerClass, object): | class badrequesthandler(self.RequestHandlerClass, object): | ||||
| def send_header(self, name, value): | def send_header(self, name, value): | ||||
| # Make headers deterministic to facilitate testing. | # Make headers deterministic to facilitate testing. | ||||
| if name.lower() == 'date': | if name.lower() == 'date': | ||||
| value = 'Fri, 14 Apr 2017 00:00:00 GMT' | value = 'Fri, 14 Apr 2017 00:00:00 GMT' | ||||
| elif name.lower() == 'server': | elif name.lower() == 'server': | ||||
| value = 'badhttpserver' | value = 'badhttpserver' | ||||
| return super(badrequesthandler, self).send_header(name, | return super(badrequesthandler, self).send_header(name, | ||||
| value) | value) | ||||
| self.RequestHandlerClass = badrequesthandler | self.RequestHandlerClass = badrequesthandler | ||||
| # Called to accept() a pending socket. | # Called to accept() a pending socket. | ||||
| def get_request(self): | def get_request(self): | ||||
| if self._ui.configbool('badserver', 'closebeforeaccept'): | if self._ui.configbool(b'badserver', b'closebeforeaccept'): | ||||
| self.socket.close() | self.socket.close() | ||||
| # Tells the server to stop processing more requests. | # Tells the server to stop processing more requests. | ||||
| self.__shutdown_request = True | self.__shutdown_request = True | ||||
| # Simulate failure to stop processing this request. | # Simulate failure to stop processing this request. | ||||
| raise socket.error('close before accept') | raise socket.error('close before accept') | ||||