Added the `ratequeue' program.
[ashd.git] / python3 / ashd / scgi.py
CommitLineData
173e0e9e
FT
1class protoerr(Exception):
2 pass
3
173e0e9e
FT
4def readns(sk):
5 hln = 0
6 while True:
7 c = sk.read(1)
8 if c == b':':
9 break
10 elif c >= b'0' or c <= b'9':
11 hln = (hln * 10) + (ord(c) - ord(b'0'))
12 else:
13 raise protoerr("Invalid netstring length byte: " + c)
14 ret = sk.read(hln)
15 if sk.read(1) != b',':
16 raise protoerr("Non-terminated netstring")
17 return ret
18
19def readhead(sk):
20 parts = readns(sk).split(b'\0')[:-1]
21 if len(parts) % 2 != 0:
22 raise protoerr("Malformed headers")
23 ret = {}
24 i = 0
25 while i < len(parts):
26 ret[parts[i]] = parts[i + 1]
27 i += 2
28 return ret
29
173e0e9e
FT
30def decodehead(head, coding):
31 return {k.decode(coding): v.decode(coding) for k, v in head.items()}