Merge branch 'master' of fludd.seatribe.se:/usr/local/src/ashd
[ashd.git] / python / ashd / scgi.py
1 class protoerr(Exception):
2     pass
3
4 def readns(sk):
5     hln = 0
6     while True:
7         c = sk.read(1)
8         if c == ':':
9             break
10         elif c >= '0' or c <= '9':
11             hln = (hln * 10) + (ord(c) - ord('0'))
12         else:
13             raise protoerr, "Invalid netstring length byte: " + c
14     ret = sk.read(hln)
15     if sk.read(1) != ',':
16         raise protoerr, "Non-terminated netstring"
17     return ret
18
19 def readhead(sk):
20     parts = readns(sk).split('\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