X-Git-Url: http://dolda2000.com/gitweb/?p=wrw.git;a=blobdiff_plain;f=wrw%2Freq.py;h=010e907a5d203c86281f0a61e298498a8504518b;hp=016d2d8d6050f034639421da6be19f2d80606fce;hb=d30502c8fe37bedb30ad9f3ddecd5191c5b077fb;hpb=b0f95b219e2f9aea9f81d696adbfaaa0e41a6eaf diff --git a/wrw/req.py b/wrw/req.py index 016d2d8..010e907 100644 --- a/wrw/req.py +++ b/wrw/req.py @@ -51,6 +51,10 @@ def fixcase(str): i += 1 return str +class shortinput(IOError, EOFError): + def __init__(self): + super().__init__("Unexpected EOF") + class limitreader(object): def __init__(self, back, limit): self.bk = back @@ -68,7 +72,7 @@ class limitreader(object): while len(self.buf) < ra: ret = self.bk.read(ra - len(self.buf)) if ret == b"": - raise IOError("Unexpected EOF") + raise shortinput() self.buf.extend(ret) self.rb += len(ret) ret = bytes(self.buf[:ra]) @@ -98,7 +102,7 @@ class limitreader(object): ra = min(ra, 1024) ret = self.bk.read(ra) if ret == b"": - raise IOError("Unpexpected EOF") + raise shortinput() self.buf.extend(ret) self.rb += len(ret) @@ -230,6 +234,7 @@ class copyrequest(request): self.https = p.https self.ihead = p.ihead self.ohead = p.ohead + self.input = p.input def status(self, code, msg): return self.parent.status(code, msg)