X-Git-Url: http://dolda2000.com/gitweb/?p=wrw.git;a=blobdiff_plain;f=wrw%2Freq.py;h=010e907a5d203c86281f0a61e298498a8504518b;hp=b08afae7278c52b263ab277b91c38d860ceadec9;hb=d30502c8fe37bedb30ad9f3ddecd5191c5b077fb;hpb=0bf1ac2ff3015d0072a336e670bd952d0e73535e diff --git a/wrw/req.py b/wrw/req.py index b08afae..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)