diff options
Diffstat (limited to 'libpathod/rparse.py')
| -rw-r--r-- | libpathod/rparse.py | 31 |
1 files changed, 20 insertions, 11 deletions
diff --git a/libpathod/rparse.py b/libpathod/rparse.py index 767cb861..e56135ea 100644 --- a/libpathod/rparse.py +++ b/libpathod/rparse.py @@ -541,7 +541,7 @@ class Message: l += len(i[2]) return l - def serve(self, fp, check, is_request): + def serve(self, fp, check, request_host): """ fp: The file pointer to write to. @@ -550,9 +550,11 @@ class Message: otherwise the return is treated as an error message to be sent to the client, and service stops. - is_request: Is this a request? If False, we assume it's a response. - Used to decide what standard modifications to make if raw is not - set. + request_host: If this a request, this is the connecting host. If + None, we assume it's a response. Used to decide what standard + modifications to make if raw is not set. + + Calling this function may modify the object. """ started = time.time() if not self.raw: @@ -563,8 +565,15 @@ class Message: LiteralGenerator(str(len(self.body))), ) ) - if is_request: - pass + if request_host: + if not utils.get_header("Host", self.headers): + self.headers.append( + ( + LiteralGenerator("Host"), + LiteralGenerator(request_host) + ) + ) + else: if not utils.get_header("Date", self.headers): self.headers.append( @@ -706,8 +715,8 @@ class CraftedRequest(Request): for i in tokens: i.accept(settings, self) - def serve(self, fp, check=None): - d = Request.serve(self, fp, check, True) + def serve(self, fp, check, host): + d = Request.serve(self, fp, check, host) d["spec"] = self.spec return d @@ -719,8 +728,8 @@ class CraftedResponse(Response): for i in tokens: i.accept(settings, self) - def serve(self, fp, check=None): - d = Response.serve(self, fp, check, False) + def serve(self, fp, check): + d = Response.serve(self, fp, check, None) d["spec"] = self.spec return d @@ -738,7 +747,7 @@ class PathodErrorResponse(Response): ] def serve(self, fp, check=None): - d = Response.serve(self, fp, check, False) + d = Response.serve(self, fp, check, None) d["internal"] = True return d |