mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
836 lines
22 KiB
Python
836 lines
22 KiB
Python
import operator, string, random, mmap, os, time, copy
|
|
from email.utils import formatdate
|
|
import contrib.pyparsing as pp
|
|
from netlib import http_status, tcp
|
|
|
|
import utils
|
|
|
|
BLOCKSIZE = 1024
|
|
TRUNCATE = 1024
|
|
|
|
class FileAccessDenied(Exception): pass
|
|
|
|
|
|
class ParseException(Exception):
|
|
def __init__(self, msg, s, col):
|
|
Exception.__init__(self)
|
|
self.msg = msg
|
|
self.s = s
|
|
self.col = col
|
|
|
|
def marked(self):
|
|
return "%s\n%s"%(self.s, " "*(self.col-1) + "^")
|
|
|
|
def __str__(self):
|
|
return "%s at char %s"%(self.msg, self.col)
|
|
|
|
|
|
def send_chunk(fp, val, blocksize, start, end):
|
|
"""
|
|
(start, end): Inclusive lower bound, exclusive upper bound.
|
|
"""
|
|
for i in range(start, end, blocksize):
|
|
fp.write(
|
|
val[i:min(i+blocksize, end)]
|
|
)
|
|
return end-start
|
|
|
|
|
|
def write_values(fp, vals, actions, sofar=0, skip=0, blocksize=BLOCKSIZE):
|
|
"""
|
|
vals: A list of values, which may be strings or Value objects.
|
|
actions: A list of (offset, action, arg) tuples. Action may be "pause" or "disconnect".
|
|
|
|
Both vals and actions are in reverse order, with the first items last.
|
|
|
|
Return True if connection should disconnect.
|
|
"""
|
|
sofar = 0
|
|
try:
|
|
while vals:
|
|
v = vals.pop()
|
|
offset = 0
|
|
while actions and actions[-1][0] < (sofar + len(v)):
|
|
a = actions.pop()
|
|
offset += send_chunk(fp, v, blocksize, offset, a[0]-sofar-offset)
|
|
if a[1] == "pause":
|
|
time.sleep(a[2])
|
|
elif a[1] == "disconnect":
|
|
return True
|
|
elif a[1] == "inject":
|
|
send_chunk(fp, a[2], blocksize, 0, len(a[2]))
|
|
send_chunk(fp, v, blocksize, offset, len(v))
|
|
sofar += len(v)
|
|
# Remainders
|
|
while actions:
|
|
a = actions.pop()
|
|
if a[1] == "pause":
|
|
time.sleep(a[2])
|
|
elif a[1] == "disconnect":
|
|
return True
|
|
elif a[1] == "inject":
|
|
send_chunk(fp, a[2], blocksize, 0, len(a[2]))
|
|
except tcp.NetLibDisconnect: # pragma: no cover
|
|
return True
|
|
|
|
|
|
DATATYPES = dict(
|
|
ascii_letters = string.ascii_letters,
|
|
ascii_lowercase = string.ascii_lowercase,
|
|
ascii_uppercase = string.ascii_uppercase,
|
|
digits = string.digits,
|
|
hexdigits = string.hexdigits,
|
|
octdigits = string.octdigits,
|
|
punctuation = string.punctuation,
|
|
whitespace = string.whitespace,
|
|
ascii = string.printable,
|
|
bytes = "".join(chr(i) for i in range(256))
|
|
)
|
|
|
|
|
|
#v_integer = pp.Regex(r"[+-]?\d+")\
|
|
v_integer = pp.Regex(r"\d+")\
|
|
.setName("integer")\
|
|
.setParseAction(lambda toks: int(toks[0]))
|
|
|
|
|
|
v_literal = pp.MatchFirst(
|
|
[
|
|
pp.QuotedString("\"", escChar="\\", unquoteResults=True, multiline=True),
|
|
pp.QuotedString("'", escChar="\\", unquoteResults=True, multiline=True),
|
|
]
|
|
)
|
|
|
|
v_naked_literal = pp.MatchFirst(
|
|
[
|
|
v_literal,
|
|
pp.Word("".join(i for i in pp.printables if i not in ",:\n"))
|
|
]
|
|
)
|
|
|
|
|
|
class LiteralGenerator:
|
|
def __init__(self, s):
|
|
self.s = s
|
|
|
|
def __eq__(self, other):
|
|
return self[:] == other
|
|
|
|
def __len__(self):
|
|
return len(self.s)
|
|
|
|
def __getitem__(self, x):
|
|
return self.s.__getitem__(x)
|
|
|
|
def __getslice__(self, a, b):
|
|
return self.s.__getslice__(a, b)
|
|
|
|
def __repr__(self):
|
|
return '"%s"'%self.s
|
|
|
|
|
|
class RandomGenerator:
|
|
def __init__(self, dtype, length):
|
|
self.dtype = dtype
|
|
self.length = length
|
|
|
|
def __len__(self):
|
|
return self.length
|
|
|
|
def __getitem__(self, x):
|
|
return random.choice(DATATYPES[self.dtype])
|
|
|
|
def __getslice__(self, a, b):
|
|
b = min(b, self.length)
|
|
chars = DATATYPES[self.dtype]
|
|
return "".join(random.choice(chars) for x in range(a, b))
|
|
|
|
def __repr__(self):
|
|
return "%s random from %s"%(self.length, self.dtype)
|
|
|
|
|
|
class FileGenerator:
|
|
def __init__(self, path):
|
|
self.path = path
|
|
self.fp = file(path, "r")
|
|
self.map = mmap.mmap(self.fp.fileno(), 0, prot=mmap.PROT_READ)
|
|
|
|
def __len__(self):
|
|
return len(self.map)
|
|
|
|
def __getitem__(self, x):
|
|
return self.map.__getitem__(x)
|
|
|
|
def __getslice__(self, a, b):
|
|
return self.map.__getslice__(a, b)
|
|
|
|
def __repr__(self):
|
|
return "<%s"%self.path
|
|
|
|
|
|
class _Value:
|
|
def __init__(self, val):
|
|
self.val = val.decode("string_escape")
|
|
|
|
def get_generator(self, settings):
|
|
return LiteralGenerator(self.val)
|
|
|
|
def __repr__(self):
|
|
return self.spec()
|
|
|
|
|
|
class ValueLiteral(_Value):
|
|
@classmethod
|
|
def expr(klass):
|
|
e = v_literal.copy()
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
def spec(self):
|
|
return '"%s"'%self.val.encode("string_escape")
|
|
|
|
|
|
class ValueNakedLiteral(_Value):
|
|
@classmethod
|
|
def expr(klass):
|
|
e = v_naked_literal.copy()
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
def spec(self):
|
|
return self.val.encode("string_escape")
|
|
|
|
|
|
class ValueGenerate:
|
|
def __init__(self, usize, unit, datatype):
|
|
if not unit:
|
|
unit = "b"
|
|
self.usize, self.unit, self.datatype = usize, unit, datatype
|
|
|
|
def bytes(self):
|
|
return self.usize * utils.SIZE_UNITS[self.unit]
|
|
|
|
def get_generator(self, settings):
|
|
return RandomGenerator(self.datatype, self.bytes())
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("@").suppress() + v_integer
|
|
|
|
u = reduce(operator.or_, [pp.Literal(i) for i in utils.SIZE_UNITS.keys()])
|
|
e = e + pp.Optional(u, default=None)
|
|
|
|
s = pp.Literal(",").suppress()
|
|
s += reduce(operator.or_, [pp.Literal(i) for i in DATATYPES.keys()])
|
|
e += pp.Optional(s, default="bytes")
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
def spec(self):
|
|
s = "@%s"%self.usize
|
|
if self.unit != "b":
|
|
s += self.unit
|
|
if self.datatype != "bytes":
|
|
s += ",%s"%self.datatype
|
|
return s
|
|
|
|
def __repr__(self):
|
|
return self.spec()
|
|
|
|
|
|
class ValueFile:
|
|
def __init__(self, path):
|
|
self.path = path
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("<").suppress()
|
|
e = e + v_naked_literal
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
def get_generator(self, settings):
|
|
uf = settings.get("unconstrained_file_access")
|
|
sd = settings.get("staticdir")
|
|
if not sd:
|
|
raise FileAccessDenied("File access disabled.")
|
|
sd = os.path.normpath(os.path.abspath(sd))
|
|
|
|
s = os.path.expanduser(self.path)
|
|
s = os.path.normpath(os.path.abspath(os.path.join(sd, s)))
|
|
if not uf and not s.startswith(sd):
|
|
raise FileAccessDenied("File access outside of configured directory")
|
|
if not os.path.isfile(s):
|
|
raise FileAccessDenied("File not readable")
|
|
return FileGenerator(s)
|
|
|
|
def spec(self):
|
|
return '<"%s"'%self.path.encode("string_escape")
|
|
|
|
|
|
Value = pp.MatchFirst(
|
|
[
|
|
ValueGenerate.expr(),
|
|
ValueFile.expr(),
|
|
ValueLiteral.expr()
|
|
]
|
|
)
|
|
|
|
|
|
NakedValue = pp.MatchFirst(
|
|
[
|
|
ValueGenerate.expr(),
|
|
ValueFile.expr(),
|
|
ValueLiteral.expr(),
|
|
ValueNakedLiteral.expr(),
|
|
]
|
|
)
|
|
|
|
|
|
Offset = pp.MatchFirst(
|
|
[
|
|
v_integer,
|
|
pp.Literal("r"),
|
|
pp.Literal("a")
|
|
]
|
|
)
|
|
|
|
|
|
class _Header:
|
|
def __init__(self, key, value):
|
|
self.key, self.value = key, value
|
|
|
|
def values(self, settings):
|
|
return [
|
|
self.key.get_generator(settings),
|
|
": ",
|
|
self.value.get_generator(settings),
|
|
"\r\n",
|
|
]
|
|
|
|
def accept(self, settings, r):
|
|
r.headers.append(self)
|
|
|
|
|
|
class Header(_Header):
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("h").suppress()
|
|
e += Value
|
|
e += pp.Literal("=").suppress()
|
|
e += Value
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
|
|
class ShortcutContentType(_Header):
|
|
def __init__(self, value):
|
|
_Header.__init__(self, ValueLiteral("Content-Type"), value)
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("c").suppress()
|
|
e = e + Value
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
|
|
class ShortcutLocation(_Header):
|
|
def __init__(self, value):
|
|
_Header.__init__(self, ValueLiteral("Location"), value)
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("l").suppress()
|
|
e = e + Value
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
|
|
class Body:
|
|
def __init__(self, value):
|
|
self.value = value
|
|
|
|
def accept(self, settings, r):
|
|
r.body = self.value.get_generator(settings)
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("b").suppress()
|
|
e = e + Value
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
|
|
class Raw:
|
|
def accept(self, settings, r):
|
|
r.raw = True
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("r").suppress()
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
|
|
class Path:
|
|
def __init__(self, value):
|
|
if isinstance(value, basestring):
|
|
value = ValueLiteral(value)
|
|
self.value = value
|
|
|
|
def accept(self, settings, r):
|
|
r.path = self.value.get_generator(settings)
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = NakedValue.copy()
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
|
|
|
|
class Method:
|
|
methods = [
|
|
"get",
|
|
"head",
|
|
"post",
|
|
"put",
|
|
"delete",
|
|
"options",
|
|
"trace",
|
|
"connect",
|
|
]
|
|
def __init__(self, value):
|
|
# If it's a string, we were passed one of the methods, so we upper-case
|
|
# it to be canonical. The user can specify a different case by using a
|
|
# string value literal.
|
|
if isinstance(value, basestring):
|
|
value = ValueLiteral(value.upper())
|
|
self.value = value
|
|
|
|
def accept(self, settings, r):
|
|
r.method = self.value.get_generator(settings)
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
parts = [pp.CaselessLiteral(i) for i in klass.methods]
|
|
m = pp.MatchFirst(parts)
|
|
spec = m | Value.copy()
|
|
spec = spec.setParseAction(lambda x: klass(*x))
|
|
return spec
|
|
|
|
|
|
class _Action:
|
|
"""
|
|
An action that operates on the raw data stream of the message. All
|
|
actions have one thing in common: an offset that specifies where the
|
|
action should take place.
|
|
"""
|
|
def __init__(self, offset):
|
|
self.offset = offset
|
|
|
|
def resolve_offset(self, msg, settings, request_host):
|
|
"""
|
|
Resolves offset specifications to a numeric offset. Returns a copy
|
|
of the action object.
|
|
"""
|
|
c = copy.copy(self)
|
|
l = msg.length(settings, request_host)
|
|
if c.offset == "r":
|
|
c.offset = random.randrange(l)
|
|
elif c.offset == "a":
|
|
c.offset = l + 1
|
|
return c
|
|
|
|
def __cmp__(self, other):
|
|
return cmp(self.offset, other.offset)
|
|
|
|
def __repr__(self):
|
|
return self.spec()
|
|
|
|
def accept(self, settings, r):
|
|
r.actions.append(self)
|
|
|
|
|
|
class PauseAt(_Action):
|
|
def __init__(self, offset, seconds):
|
|
_Action.__init__(self, offset)
|
|
self.seconds = seconds
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("p").suppress()
|
|
e += Offset
|
|
e += pp.Literal(",").suppress()
|
|
e += pp.MatchFirst(
|
|
[
|
|
v_integer,
|
|
pp.Literal("f")
|
|
]
|
|
)
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
def spec(self):
|
|
return "p%s,%s"%(self.offset, self.seconds)
|
|
|
|
def intermediate(self, settings):
|
|
return (self.offset, "pause", self.seconds)
|
|
|
|
|
|
class DisconnectAt(_Action):
|
|
def __init__(self, offset):
|
|
_Action.__init__(self, offset)
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("d").suppress()
|
|
e += Offset
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
def spec(self):
|
|
return "d%s"%self.offset
|
|
|
|
def intermediate(self, settings):
|
|
return (self.offset, "disconnect")
|
|
|
|
|
|
class InjectAt(_Action):
|
|
def __init__(self, offset, value):
|
|
_Action.__init__(self, offset)
|
|
self.value = value
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = pp.Literal("i").suppress()
|
|
e += Offset
|
|
e += pp.Literal(",").suppress()
|
|
e += Value
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
def spec(self):
|
|
return "i%s,%s"%(self.offset, self.value.spec())
|
|
|
|
def intermediate(self, settings):
|
|
return (
|
|
self.offset,
|
|
"inject",
|
|
self.value.get_generator(settings)
|
|
)
|
|
|
|
|
|
class Code:
|
|
def __init__(self, code, msg=None):
|
|
self.code, self.msg = code, msg
|
|
if msg is None:
|
|
self.msg = ValueLiteral(http_status.RESPONSES.get(self.code, "Unknown code"))
|
|
|
|
def accept(self, settings, r):
|
|
r.code = self.code
|
|
r.msg = self.msg.get_generator(settings)
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
e = v_integer
|
|
e = e + pp.Optional(
|
|
Value
|
|
)
|
|
return e.setParseAction(lambda x: klass(*x))
|
|
|
|
|
|
|
|
class Message:
|
|
version = "HTTP/1.1"
|
|
def __init__(self):
|
|
self.body = LiteralGenerator("")
|
|
self.headers = []
|
|
self.actions = []
|
|
self.raw = False
|
|
|
|
def length(self, settings, request_host):
|
|
"""
|
|
Calculate the length of the base message without any applied actions.
|
|
"""
|
|
l = sum(len(x) for x in self.preamble())
|
|
l += 2
|
|
for h in self.headervals(settings, request_host):
|
|
l += len(h)
|
|
l += 2
|
|
l += len(self.body)
|
|
return l
|
|
|
|
def preview_safe(self):
|
|
"""
|
|
Modify this message to be safe for previews. Returns a list of elided actions.
|
|
"""
|
|
pauses = [i for i in self.actions if isinstance(i, PauseAt)]
|
|
self.actions = [i for i in self.actions if not isinstance(i, PauseAt)]
|
|
return pauses
|
|
|
|
def effective_length(self, settings, request_host):
|
|
"""
|
|
Calculate the length of the base message with all applied actions.
|
|
"""
|
|
# Order matters here, and must match the order of application in
|
|
# write_values.
|
|
l = self.length(settings, request_host)
|
|
for i in reversed(self.ready_actions(settings, request_host)):
|
|
if i[1] == "disconnect":
|
|
return i[0]
|
|
elif i[1] == "inject":
|
|
l += len(i[2])
|
|
return l
|
|
|
|
def headervals(self, settings, request_host):
|
|
hdrs = self.headers[:]
|
|
if not self.raw:
|
|
if self.body and not utils.get_header("Content-Length", self.headers):
|
|
hdrs.append(
|
|
Header(
|
|
ValueLiteral("Content-Length"),
|
|
ValueLiteral(str(len(self.body))),
|
|
)
|
|
)
|
|
if request_host:
|
|
if not utils.get_header("Host", self.headers):
|
|
hdrs.append(
|
|
Header(
|
|
ValueLiteral("Host"),
|
|
ValueLiteral(request_host)
|
|
)
|
|
)
|
|
|
|
else:
|
|
if not utils.get_header("Date", self.headers):
|
|
hdrs.append(
|
|
Header(
|
|
ValueLiteral("Date"),
|
|
ValueLiteral(formatdate(timeval=None, localtime=False, usegmt=True))
|
|
)
|
|
)
|
|
values = []
|
|
for h in hdrs:
|
|
values.extend(h.values(settings))
|
|
return values
|
|
|
|
def ready_actions(self, settings, request_host):
|
|
actions = [i.resolve_offset(self, settings, request_host) for i in self.actions]
|
|
actions.sort()
|
|
actions.reverse()
|
|
return [i.intermediate(settings) for i in actions]
|
|
|
|
def serve(self, settings, fp, check, request_host):
|
|
"""
|
|
fp: The file pointer to write to.
|
|
|
|
check: A function called with the effective actions (after random
|
|
values have been calculated). If it returns False service proceeds,
|
|
otherwise the return is treated as an error message to be sent to
|
|
the client, and service stops.
|
|
|
|
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()
|
|
|
|
hdrs = self.headervals(settings, request_host)
|
|
|
|
vals = self.preamble()
|
|
vals.append("\r\n")
|
|
vals.extend(hdrs)
|
|
vals.append("\r\n")
|
|
if self.body:
|
|
vals.append(self.body)
|
|
vals.reverse()
|
|
actions = self.ready_actions(settings, request_host)
|
|
if check:
|
|
ret = check(self, actions)
|
|
if ret:
|
|
err = PathodErrorResponse(ret)
|
|
err.serve(settings, fp)
|
|
return dict(
|
|
disconnect = True,
|
|
error = ret
|
|
)
|
|
|
|
disconnect = write_values(fp, vals, actions[:])
|
|
duration = time.time() - started
|
|
ret = dict(
|
|
disconnect = disconnect,
|
|
started = started,
|
|
duration = duration,
|
|
)
|
|
for i in self.logattrs:
|
|
v = getattr(self, i)
|
|
# Careful not to log any VALUE specs without sanitizing them first. We truncate at 1k.
|
|
if hasattr(v, "__len__"):
|
|
v = v[:TRUNCATE]
|
|
v = v.encode("string_escape")
|
|
ret[i] = v
|
|
return ret
|
|
|
|
|
|
Sep = pp.Optional(pp.Literal(":")).suppress()
|
|
|
|
class Response(Message):
|
|
comps = (
|
|
Body,
|
|
Header,
|
|
PauseAt,
|
|
DisconnectAt,
|
|
InjectAt,
|
|
ShortcutContentType,
|
|
ShortcutLocation,
|
|
Raw
|
|
)
|
|
logattrs = ["code", "version", "body"]
|
|
def __init__(self):
|
|
Message.__init__(self)
|
|
self.code = None
|
|
self.msg = None
|
|
|
|
def preamble(self):
|
|
return [self.version, " ", str(self.code), " ", self.msg]
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
parts = [i.expr() for i in klass.comps]
|
|
atom = pp.MatchFirst(parts)
|
|
resp = pp.And(
|
|
[
|
|
Code.expr(),
|
|
pp.ZeroOrMore(Sep + atom)
|
|
]
|
|
)
|
|
return resp
|
|
|
|
def __str__(self):
|
|
parts = [
|
|
"%s %s"%(self.code, self.msg[:])
|
|
]
|
|
return "\n".join(parts)
|
|
|
|
|
|
class Request(Message):
|
|
comps = (
|
|
Body,
|
|
Header,
|
|
PauseAt,
|
|
DisconnectAt,
|
|
InjectAt,
|
|
ShortcutContentType,
|
|
Raw
|
|
)
|
|
logattrs = ["method", "path", "body"]
|
|
def __init__(self):
|
|
Message.__init__(self)
|
|
self.method = None
|
|
self.path = None
|
|
|
|
def preamble(self):
|
|
return [self.method, " ", self.path, " ", self.version]
|
|
|
|
@classmethod
|
|
def expr(klass):
|
|
parts = [i.expr() for i in klass.comps]
|
|
atom = pp.MatchFirst(parts)
|
|
resp = pp.And(
|
|
[
|
|
Method.expr(),
|
|
Sep,
|
|
Path.expr(),
|
|
pp.ZeroOrMore(Sep + atom)
|
|
]
|
|
)
|
|
return resp
|
|
|
|
def __str__(self):
|
|
parts = [
|
|
"%s %s"%(self.method[:], self.path[:])
|
|
]
|
|
return "\n".join(parts)
|
|
|
|
|
|
class CraftedRequest(Request):
|
|
def __init__(self, settings, spec, tokens):
|
|
Request.__init__(self)
|
|
self.spec, self.tokens = spec, tokens
|
|
for i in tokens:
|
|
i.accept(settings, self)
|
|
|
|
def serve(self, settings, fp, check, host):
|
|
d = Request.serve(self, settings, fp, check, host)
|
|
d["spec"] = self.spec
|
|
return d
|
|
|
|
|
|
class CraftedResponse(Response):
|
|
def __init__(self, settings, spec, tokens):
|
|
Response.__init__(self)
|
|
self.spec, self.tokens = spec, tokens
|
|
for i in tokens:
|
|
i.accept(settings, self)
|
|
|
|
def serve(self, settings, fp, check):
|
|
d = Response.serve(self, settings, fp, check, None)
|
|
d["spec"] = self.spec
|
|
return d
|
|
|
|
|
|
class PathodErrorResponse(Response):
|
|
def __init__(self, msg, body=None):
|
|
Response.__init__(self)
|
|
self.code = 800
|
|
self.msg = LiteralGenerator(msg)
|
|
self.body = LiteralGenerator("pathod error: " + (body or msg))
|
|
self.headers = [
|
|
Header(ValueLiteral("Content-Type"), ValueLiteral("text/plain")),
|
|
]
|
|
|
|
def serve(self, settings, fp, check=None):
|
|
d = Response.serve(self, settings, fp, check, None)
|
|
d["internal"] = True
|
|
return d
|
|
|
|
|
|
FILESTART = "+"
|
|
def read_file(settings, s):
|
|
uf = settings.get("unconstrained_file_access")
|
|
sd = settings.get("staticdir")
|
|
if not sd:
|
|
raise FileAccessDenied("File access disabled.")
|
|
sd = os.path.normpath(os.path.abspath(sd))
|
|
s = s[1:]
|
|
s = os.path.expanduser(s)
|
|
s = os.path.normpath(os.path.abspath(os.path.join(sd, s)))
|
|
if not uf and not s.startswith(sd):
|
|
raise FileAccessDenied("File access outside of configured directory")
|
|
if not os.path.isfile(s):
|
|
raise FileAccessDenied("File not readable")
|
|
return file(s, "r").read()
|
|
|
|
|
|
def parse_response(settings, s):
|
|
"""
|
|
May raise ParseException or FileAccessDenied
|
|
"""
|
|
try:
|
|
s.decode("ascii")
|
|
except UnicodeError:
|
|
raise ParseException("Spec must be valid ASCII.", 0, 0)
|
|
if s.startswith(FILESTART):
|
|
s = read_file(settings, s)
|
|
try:
|
|
return CraftedResponse(settings, s, Response.expr().parseString(s, parseAll=True))
|
|
except pp.ParseException, v:
|
|
raise ParseException(v.msg, v.line, v.col)
|
|
|
|
|
|
def parse_request(settings, s):
|
|
"""
|
|
May raise ParseException or FileAccessDenied
|
|
"""
|
|
try:
|
|
s.decode("ascii")
|
|
except UnicodeError:
|
|
raise ParseException("Spec must be valid ASCII.", 0, 0)
|
|
if s.startswith(FILESTART):
|
|
s = read_file(settings, s)
|
|
try:
|
|
return CraftedRequest(settings, s, Request.expr().parseString(s, parseAll=True))
|
|
except pp.ParseException, v:
|
|
raise ParseException(v.msg, v.line, v.col)
|