wireprotoframing.py
1169 lines
| 39.5 KiB
| text/x-python
|
PythonLexer
/ mercurial / wireprotoframing.py
Gregory Szorc
|
r37069 | # wireprotoframing.py - unified framing protocol for wire protocol | ||
# | ||||
# Copyright 2018 Gregory Szorc <gregory.szorc@gmail.com> | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
# This file contains functionality to support the unified frame-based wire | ||||
# protocol. For details about the protocol, see | ||||
# `hg help internals.wireprotocol`. | ||||
from __future__ import absolute_import | ||||
Gregory Szorc
|
r37561 | import collections | ||
Gregory Szorc
|
r37069 | import struct | ||
Gregory Szorc
|
r37070 | from .i18n import _ | ||
Gregory Szorc
|
r37079 | from .thirdparty import ( | ||
attr, | ||||
) | ||||
Gregory Szorc
|
r37069 | from . import ( | ||
Yuya Nishihara
|
r37492 | encoding, | ||
Gregory Szorc
|
r37070 | error, | ||
Gregory Szorc
|
r37069 | util, | ||
) | ||||
Yuya Nishihara
|
r37102 | from .utils import ( | ||
Gregory Szorc
|
r39477 | cborutil, | ||
Yuya Nishihara
|
r37102 | stringutil, | ||
) | ||||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37304 | FRAME_HEADER_SIZE = 8 | ||
Gregory Szorc
|
r37069 | DEFAULT_MAX_FRAME_SIZE = 32768 | ||
Gregory Szorc
|
r37304 | STREAM_FLAG_BEGIN_STREAM = 0x01 | ||
STREAM_FLAG_END_STREAM = 0x02 | ||||
STREAM_FLAG_ENCODING_APPLIED = 0x04 | ||||
STREAM_FLAGS = { | ||||
b'stream-begin': STREAM_FLAG_BEGIN_STREAM, | ||||
b'stream-end': STREAM_FLAG_END_STREAM, | ||||
b'encoded': STREAM_FLAG_ENCODING_APPLIED, | ||||
} | ||||
Gregory Szorc
|
r37308 | FRAME_TYPE_COMMAND_REQUEST = 0x01 | ||
Gregory Szorc
|
r37741 | FRAME_TYPE_COMMAND_DATA = 0x02 | ||
Gregory Szorc
|
r37742 | FRAME_TYPE_COMMAND_RESPONSE = 0x03 | ||
Gregory Szorc
|
r37073 | FRAME_TYPE_ERROR_RESPONSE = 0x05 | ||
Gregory Szorc
|
r37078 | FRAME_TYPE_TEXT_OUTPUT = 0x06 | ||
Gregory Szorc
|
r37307 | FRAME_TYPE_PROGRESS = 0x07 | ||
Gregory Szorc
|
r37304 | FRAME_TYPE_STREAM_SETTINGS = 0x08 | ||
Gregory Szorc
|
r37069 | |||
FRAME_TYPES = { | ||||
Gregory Szorc
|
r37308 | b'command-request': FRAME_TYPE_COMMAND_REQUEST, | ||
Gregory Szorc
|
r37069 | b'command-data': FRAME_TYPE_COMMAND_DATA, | ||
Gregory Szorc
|
r37742 | b'command-response': FRAME_TYPE_COMMAND_RESPONSE, | ||
Gregory Szorc
|
r37073 | b'error-response': FRAME_TYPE_ERROR_RESPONSE, | ||
Gregory Szorc
|
r37078 | b'text-output': FRAME_TYPE_TEXT_OUTPUT, | ||
Gregory Szorc
|
r37307 | b'progress': FRAME_TYPE_PROGRESS, | ||
Gregory Szorc
|
r37304 | b'stream-settings': FRAME_TYPE_STREAM_SETTINGS, | ||
Gregory Szorc
|
r37069 | } | ||
Gregory Szorc
|
r37308 | FLAG_COMMAND_REQUEST_NEW = 0x01 | ||
FLAG_COMMAND_REQUEST_CONTINUATION = 0x02 | ||||
FLAG_COMMAND_REQUEST_MORE_FRAMES = 0x04 | ||||
FLAG_COMMAND_REQUEST_EXPECT_DATA = 0x08 | ||||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37308 | FLAGS_COMMAND_REQUEST = { | ||
b'new': FLAG_COMMAND_REQUEST_NEW, | ||||
b'continuation': FLAG_COMMAND_REQUEST_CONTINUATION, | ||||
b'more': FLAG_COMMAND_REQUEST_MORE_FRAMES, | ||||
b'have-data': FLAG_COMMAND_REQUEST_EXPECT_DATA, | ||||
Gregory Szorc
|
r37069 | } | ||
FLAG_COMMAND_DATA_CONTINUATION = 0x01 | ||||
FLAG_COMMAND_DATA_EOS = 0x02 | ||||
FLAGS_COMMAND_DATA = { | ||||
b'continuation': FLAG_COMMAND_DATA_CONTINUATION, | ||||
b'eos': FLAG_COMMAND_DATA_EOS, | ||||
} | ||||
Gregory Szorc
|
r37742 | FLAG_COMMAND_RESPONSE_CONTINUATION = 0x01 | ||
FLAG_COMMAND_RESPONSE_EOS = 0x02 | ||||
Gregory Szorc
|
r37073 | |||
Gregory Szorc
|
r37742 | FLAGS_COMMAND_RESPONSE = { | ||
b'continuation': FLAG_COMMAND_RESPONSE_CONTINUATION, | ||||
b'eos': FLAG_COMMAND_RESPONSE_EOS, | ||||
Gregory Szorc
|
r37073 | } | ||
Gregory Szorc
|
r37069 | # Maps frame types to their available flags. | ||
FRAME_TYPE_FLAGS = { | ||||
Gregory Szorc
|
r37308 | FRAME_TYPE_COMMAND_REQUEST: FLAGS_COMMAND_REQUEST, | ||
Gregory Szorc
|
r37069 | FRAME_TYPE_COMMAND_DATA: FLAGS_COMMAND_DATA, | ||
Gregory Szorc
|
r37742 | FRAME_TYPE_COMMAND_RESPONSE: FLAGS_COMMAND_RESPONSE, | ||
Gregory Szorc
|
r37744 | FRAME_TYPE_ERROR_RESPONSE: {}, | ||
Gregory Szorc
|
r37078 | FRAME_TYPE_TEXT_OUTPUT: {}, | ||
Gregory Szorc
|
r37307 | FRAME_TYPE_PROGRESS: {}, | ||
Gregory Szorc
|
r37304 | FRAME_TYPE_STREAM_SETTINGS: {}, | ||
Gregory Szorc
|
r37069 | } | ||
Gregory Szorc
|
r37308 | ARGUMENT_RECORD_HEADER = struct.Struct(r'<HH') | ||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37314 | def humanflags(mapping, value): | ||
"""Convert a numeric flags value to a human value, using a mapping table.""" | ||||
Yuya Nishihara
|
r37491 | namemap = {v: k for k, v in mapping.iteritems()} | ||
Gregory Szorc
|
r37314 | flags = [] | ||
Yuya Nishihara
|
r37491 | val = 1 | ||
while value >= val: | ||||
Gregory Szorc
|
r37314 | if value & val: | ||
Yuya Nishihara
|
r37491 | flags.append(namemap.get(val, '<unknown 0x%02x>' % val)) | ||
val <<= 1 | ||||
Gregory Szorc
|
r37314 | |||
return b'|'.join(flags) | ||||
Gregory Szorc
|
r37079 | @attr.s(slots=True) | ||
class frameheader(object): | ||||
"""Represents the data in a frame header.""" | ||||
length = attr.ib() | ||||
requestid = attr.ib() | ||||
Gregory Szorc
|
r37304 | streamid = attr.ib() | ||
streamflags = attr.ib() | ||||
Gregory Szorc
|
r37079 | typeid = attr.ib() | ||
flags = attr.ib() | ||||
Gregory Szorc
|
r37314 | @attr.s(slots=True, repr=False) | ||
Gregory Szorc
|
r37079 | class frame(object): | ||
"""Represents a parsed frame.""" | ||||
requestid = attr.ib() | ||||
Gregory Szorc
|
r37304 | streamid = attr.ib() | ||
streamflags = attr.ib() | ||||
Gregory Szorc
|
r37079 | typeid = attr.ib() | ||
flags = attr.ib() | ||||
payload = attr.ib() | ||||
Yuya Nishihara
|
r37492 | @encoding.strmethod | ||
Gregory Szorc
|
r37314 | def __repr__(self): | ||
Yuya Nishihara
|
r37491 | typename = '<unknown 0x%02x>' % self.typeid | ||
Gregory Szorc
|
r37314 | for name, value in FRAME_TYPES.iteritems(): | ||
if value == self.typeid: | ||||
typename = name | ||||
break | ||||
return ('frame(size=%d; request=%d; stream=%d; streamflags=%s; ' | ||||
'type=%s; flags=%s)' % ( | ||||
len(self.payload), self.requestid, self.streamid, | ||||
humanflags(STREAM_FLAGS, self.streamflags), typename, | ||||
Yuya Nishihara
|
r37490 | humanflags(FRAME_TYPE_FLAGS.get(self.typeid, {}), self.flags))) | ||
Gregory Szorc
|
r37314 | |||
Gregory Szorc
|
r37304 | def makeframe(requestid, streamid, streamflags, typeid, flags, payload): | ||
Gregory Szorc
|
r37069 | """Assemble a frame into a byte array.""" | ||
# TODO assert size of payload. | ||||
frame = bytearray(FRAME_HEADER_SIZE + len(payload)) | ||||
Gregory Szorc
|
r37075 | # 24 bits length | ||
# 16 bits request id | ||||
Gregory Szorc
|
r37304 | # 8 bits stream id | ||
# 8 bits stream flags | ||||
Gregory Szorc
|
r37075 | # 4 bits type | ||
# 4 bits flags | ||||
Gregory Szorc
|
r37069 | l = struct.pack(r'<I', len(payload)) | ||
frame[0:3] = l[0:3] | ||||
Gregory Szorc
|
r37304 | struct.pack_into(r'<HBB', frame, 3, requestid, streamid, streamflags) | ||
frame[7] = (typeid << 4) | flags | ||||
frame[8:] = payload | ||||
Gregory Szorc
|
r37069 | |||
return frame | ||||
def makeframefromhumanstring(s): | ||||
Gregory Szorc
|
r37075 | """Create a frame from a human readable string | ||
Strings have the form: | ||||
Gregory Szorc
|
r37304 | <request-id> <stream-id> <stream-flags> <type> <flags> <payload> | ||
Gregory Szorc
|
r37069 | |||
This can be used by user-facing applications and tests for creating | ||||
frames easily without having to type out a bunch of constants. | ||||
Gregory Szorc
|
r37304 | Request ID and stream IDs are integers. | ||
Gregory Szorc
|
r37075 | |||
Gregory Szorc
|
r37304 | Stream flags, frame type, and flags can be specified by integer or | ||
named constant. | ||||
Gregory Szorc
|
r37075 | |||
Gregory Szorc
|
r37069 | Flags can be delimited by `|` to bitwise OR them together. | ||
Gregory Szorc
|
r37306 | |||
If the payload begins with ``cbor:``, the following string will be | ||||
Yuya Nishihara
|
r37494 | evaluated as Python literal and the resulting object will be fed into | ||
Gregory Szorc
|
r37306 | a CBOR encoder. Otherwise, the payload is interpreted as a Python | ||
byte string literal. | ||||
Gregory Szorc
|
r37069 | """ | ||
Gregory Szorc
|
r37304 | fields = s.split(b' ', 5) | ||
requestid, streamid, streamflags, frametype, frameflags, payload = fields | ||||
Gregory Szorc
|
r37075 | |||
requestid = int(requestid) | ||||
Gregory Szorc
|
r37304 | streamid = int(streamid) | ||
finalstreamflags = 0 | ||||
for flag in streamflags.split(b'|'): | ||||
if flag in STREAM_FLAGS: | ||||
finalstreamflags |= STREAM_FLAGS[flag] | ||||
else: | ||||
finalstreamflags |= int(flag) | ||||
Gregory Szorc
|
r37069 | |||
if frametype in FRAME_TYPES: | ||||
frametype = FRAME_TYPES[frametype] | ||||
else: | ||||
frametype = int(frametype) | ||||
finalflags = 0 | ||||
validflags = FRAME_TYPE_FLAGS[frametype] | ||||
for flag in frameflags.split(b'|'): | ||||
if flag in validflags: | ||||
finalflags |= validflags[flag] | ||||
else: | ||||
finalflags |= int(flag) | ||||
Gregory Szorc
|
r37306 | if payload.startswith(b'cbor:'): | ||
Gregory Szorc
|
r39477 | payload = b''.join(cborutil.streamencode( | ||
stringutil.evalpythonliteral(payload[5:]))) | ||||
Gregory Szorc
|
r37306 | |||
else: | ||||
payload = stringutil.unescapestr(payload) | ||||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37304 | return makeframe(requestid=requestid, streamid=streamid, | ||
streamflags=finalstreamflags, typeid=frametype, | ||||
Gregory Szorc
|
r37080 | flags=finalflags, payload=payload) | ||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37070 | def parseheader(data): | ||
"""Parse a unified framing protocol frame header from a buffer. | ||||
The header is expected to be in the buffer at offset 0 and the | ||||
buffer is expected to be large enough to hold a full header. | ||||
""" | ||||
# 24 bits payload length (little endian) | ||||
Gregory Szorc
|
r37304 | # 16 bits request ID | ||
# 8 bits stream ID | ||||
# 8 bits stream flags | ||||
Gregory Szorc
|
r37070 | # 4 bits frame type | ||
# 4 bits frame flags | ||||
# ... payload | ||||
framelength = data[0] + 256 * data[1] + 16384 * data[2] | ||||
Gregory Szorc
|
r37304 | requestid, streamid, streamflags = struct.unpack_from(r'<HBB', data, 3) | ||
typeflags = data[7] | ||||
Gregory Szorc
|
r37070 | |||
frametype = (typeflags & 0xf0) >> 4 | ||||
frameflags = typeflags & 0x0f | ||||
Gregory Szorc
|
r37304 | return frameheader(framelength, requestid, streamid, streamflags, | ||
frametype, frameflags) | ||||
Gregory Szorc
|
r37070 | |||
def readframe(fh): | ||||
"""Read a unified framing protocol frame from a file object. | ||||
Returns a 3-tuple of (type, flags, payload) for the decoded frame or | ||||
None if no frame is available. May raise if a malformed frame is | ||||
seen. | ||||
""" | ||||
header = bytearray(FRAME_HEADER_SIZE) | ||||
readcount = fh.readinto(header) | ||||
if readcount == 0: | ||||
return None | ||||
if readcount != FRAME_HEADER_SIZE: | ||||
raise error.Abort(_('received incomplete frame: got %d bytes: %s') % | ||||
(readcount, header)) | ||||
Gregory Szorc
|
r37079 | h = parseheader(header) | ||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37079 | payload = fh.read(h.length) | ||
if len(payload) != h.length: | ||||
Gregory Szorc
|
r37070 | raise error.Abort(_('frame length error: expected %d; got %d') % | ||
Gregory Szorc
|
r37079 | (h.length, len(payload))) | ||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37304 | return frame(h.requestid, h.streamid, h.streamflags, h.typeid, h.flags, | ||
payload) | ||||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37308 | def createcommandframes(stream, requestid, cmd, args, datafh=None, | ||
maxframesize=DEFAULT_MAX_FRAME_SIZE): | ||||
Gregory Szorc
|
r37069 | """Create frames necessary to transmit a request to run a command. | ||
This is a generator of bytearrays. Each item represents a frame | ||||
ready to be sent over the wire to a peer. | ||||
""" | ||||
Gregory Szorc
|
r37308 | data = {b'name': cmd} | ||
Gregory Szorc
|
r37069 | if args: | ||
Gregory Szorc
|
r37308 | data[b'args'] = args | ||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r39477 | data = b''.join(cborutil.streamencode(data)) | ||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37308 | offset = 0 | ||
while True: | ||||
flags = 0 | ||||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37308 | # Must set new or continuation flag. | ||
if not offset: | ||||
flags |= FLAG_COMMAND_REQUEST_NEW | ||||
else: | ||||
flags |= FLAG_COMMAND_REQUEST_CONTINUATION | ||||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37308 | # Data frames is set on all frames. | ||
if datafh: | ||||
flags |= FLAG_COMMAND_REQUEST_EXPECT_DATA | ||||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37308 | payload = data[offset:offset + maxframesize] | ||
offset += len(payload) | ||||
if len(payload) == maxframesize and offset < len(data): | ||||
flags |= FLAG_COMMAND_REQUEST_MORE_FRAMES | ||||
Gregory Szorc
|
r37303 | yield stream.makeframe(requestid=requestid, | ||
Gregory Szorc
|
r37308 | typeid=FRAME_TYPE_COMMAND_REQUEST, | ||
Gregory Szorc
|
r37303 | flags=flags, | ||
payload=payload) | ||||
Gregory Szorc
|
r37069 | |||
Gregory Szorc
|
r37308 | if not (flags & FLAG_COMMAND_REQUEST_MORE_FRAMES): | ||
break | ||||
Gregory Szorc
|
r37069 | if datafh: | ||
while True: | ||||
data = datafh.read(DEFAULT_MAX_FRAME_SIZE) | ||||
done = False | ||||
if len(data) == DEFAULT_MAX_FRAME_SIZE: | ||||
flags = FLAG_COMMAND_DATA_CONTINUATION | ||||
else: | ||||
flags = FLAG_COMMAND_DATA_EOS | ||||
assert datafh.read(1) == b'' | ||||
done = True | ||||
Gregory Szorc
|
r37303 | yield stream.makeframe(requestid=requestid, | ||
typeid=FRAME_TYPE_COMMAND_DATA, | ||||
flags=flags, | ||||
payload=data) | ||||
Gregory Szorc
|
r37069 | |||
if done: | ||||
break | ||||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37742 | def createcommandresponseframesfrombytes(stream, requestid, data, | ||
maxframesize=DEFAULT_MAX_FRAME_SIZE): | ||||
Gregory Szorc
|
r37073 | """Create a raw frame to send a bytes response from static bytes input. | ||
Returns a generator of bytearrays. | ||||
""" | ||||
Gregory Szorc
|
r37743 | # Automatically send the overall CBOR response map. | ||
Gregory Szorc
|
r39477 | overall = b''.join(cborutil.streamencode({b'status': b'ok'})) | ||
Gregory Szorc
|
r37743 | if len(overall) > maxframesize: | ||
raise error.ProgrammingError('not yet implemented') | ||||
Gregory Szorc
|
r37073 | |||
Gregory Szorc
|
r37743 | # Simple case where we can fit the full response in a single frame. | ||
if len(overall) + len(data) <= maxframesize: | ||||
Gregory Szorc
|
r37742 | flags = FLAG_COMMAND_RESPONSE_EOS | ||
Gregory Szorc
|
r37303 | yield stream.makeframe(requestid=requestid, | ||
Gregory Szorc
|
r37742 | typeid=FRAME_TYPE_COMMAND_RESPONSE, | ||
Gregory Szorc
|
r37503 | flags=flags, | ||
Gregory Szorc
|
r37743 | payload=overall + data) | ||
Gregory Szorc
|
r37073 | return | ||
Gregory Szorc
|
r37743 | # It's easier to send the overall CBOR map in its own frame than to track | ||
# offsets. | ||||
yield stream.makeframe(requestid=requestid, | ||||
typeid=FRAME_TYPE_COMMAND_RESPONSE, | ||||
flags=FLAG_COMMAND_RESPONSE_CONTINUATION, | ||||
payload=overall) | ||||
Gregory Szorc
|
r37073 | offset = 0 | ||
while True: | ||||
chunk = data[offset:offset + maxframesize] | ||||
offset += len(chunk) | ||||
done = offset == len(data) | ||||
if done: | ||||
Gregory Szorc
|
r37742 | flags = FLAG_COMMAND_RESPONSE_EOS | ||
Gregory Szorc
|
r37073 | else: | ||
Gregory Szorc
|
r37742 | flags = FLAG_COMMAND_RESPONSE_CONTINUATION | ||
Gregory Szorc
|
r37073 | |||
Gregory Szorc
|
r37303 | yield stream.makeframe(requestid=requestid, | ||
Gregory Szorc
|
r37742 | typeid=FRAME_TYPE_COMMAND_RESPONSE, | ||
Gregory Szorc
|
r37303 | flags=flags, | ||
payload=chunk) | ||||
Gregory Szorc
|
r37073 | |||
if done: | ||||
break | ||||
Gregory Szorc
|
r37746 | def createbytesresponseframesfromgen(stream, requestid, gen, | ||
maxframesize=DEFAULT_MAX_FRAME_SIZE): | ||||
Gregory Szorc
|
r39477 | overall = b''.join(cborutil.streamencode({b'status': b'ok'})) | ||
Gregory Szorc
|
r37746 | |||
yield stream.makeframe(requestid=requestid, | ||||
typeid=FRAME_TYPE_COMMAND_RESPONSE, | ||||
flags=FLAG_COMMAND_RESPONSE_CONTINUATION, | ||||
payload=overall) | ||||
cb = util.chunkbuffer(gen) | ||||
flags = 0 | ||||
while True: | ||||
chunk = cb.read(maxframesize) | ||||
if not chunk: | ||||
break | ||||
yield stream.makeframe(requestid=requestid, | ||||
typeid=FRAME_TYPE_COMMAND_RESPONSE, | ||||
flags=flags, | ||||
payload=chunk) | ||||
flags |= FLAG_COMMAND_RESPONSE_CONTINUATION | ||||
flags ^= FLAG_COMMAND_RESPONSE_CONTINUATION | ||||
flags |= FLAG_COMMAND_RESPONSE_EOS | ||||
yield stream.makeframe(requestid=requestid, | ||||
typeid=FRAME_TYPE_COMMAND_RESPONSE, | ||||
flags=flags, | ||||
payload=b'') | ||||
def createcommanderrorresponse(stream, requestid, message, args=None): | ||||
Gregory Szorc
|
r39522 | # TODO should this be using a list of {'msg': ..., 'args': {}} so atom | ||
# formatting works consistently? | ||||
Gregory Szorc
|
r37746 | m = { | ||
b'status': b'error', | ||||
b'error': { | ||||
b'message': message, | ||||
} | ||||
} | ||||
if args: | ||||
m[b'error'][b'args'] = args | ||||
Gregory Szorc
|
r39477 | overall = b''.join(cborutil.streamencode(m)) | ||
Gregory Szorc
|
r37746 | |||
yield stream.makeframe(requestid=requestid, | ||||
typeid=FRAME_TYPE_COMMAND_RESPONSE, | ||||
flags=FLAG_COMMAND_RESPONSE_EOS, | ||||
payload=overall) | ||||
Gregory Szorc
|
r37744 | def createerrorframe(stream, requestid, msg, errtype): | ||
Gregory Szorc
|
r37073 | # TODO properly handle frame size limits. | ||
assert len(msg) <= DEFAULT_MAX_FRAME_SIZE | ||||
Gregory Szorc
|
r39477 | payload = b''.join(cborutil.streamencode({ | ||
Gregory Szorc
|
r37744 | b'type': errtype, | ||
b'message': [{b'msg': msg}], | ||||
Gregory Szorc
|
r39477 | })) | ||
Gregory Szorc
|
r37073 | |||
Gregory Szorc
|
r37303 | yield stream.makeframe(requestid=requestid, | ||
typeid=FRAME_TYPE_ERROR_RESPONSE, | ||||
Gregory Szorc
|
r37744 | flags=0, | ||
payload=payload) | ||||
Gregory Szorc
|
r37073 | |||
Gregory Szorc
|
r37335 | def createtextoutputframe(stream, requestid, atoms, | ||
maxframesize=DEFAULT_MAX_FRAME_SIZE): | ||||
Gregory Szorc
|
r37078 | """Create a text output frame to render text to people. | ||
``atoms`` is a 3-tuple of (formatting string, args, labels). | ||||
The formatting string contains ``%s`` tokens to be replaced by the | ||||
corresponding indexed entry in ``args``. ``labels`` is an iterable of | ||||
formatters to be applied at rendering time. In terms of the ``ui`` | ||||
class, each atom corresponds to a ``ui.write()``. | ||||
""" | ||||
Gregory Szorc
|
r37335 | atomdicts = [] | ||
Gregory Szorc
|
r37078 | |||
for (formatting, args, labels) in atoms: | ||||
# TODO look for localstr, other types here? | ||||
if not isinstance(formatting, bytes): | ||||
raise ValueError('must use bytes formatting strings') | ||||
for arg in args: | ||||
if not isinstance(arg, bytes): | ||||
raise ValueError('must use bytes for arguments') | ||||
for label in labels: | ||||
if not isinstance(label, bytes): | ||||
raise ValueError('must use bytes for labels') | ||||
Gregory Szorc
|
r37335 | # Formatting string must be ASCII. | ||
formatting = formatting.decode(r'ascii', r'replace').encode(r'ascii') | ||||
Gregory Szorc
|
r37078 | |||
# Arguments must be UTF-8. | ||||
args = [a.decode(r'utf-8', r'replace').encode(r'utf-8') for a in args] | ||||
# Labels must be ASCII. | ||||
labels = [l.decode(r'ascii', r'strict').encode(r'ascii') | ||||
for l in labels] | ||||
Gregory Szorc
|
r37335 | atom = {b'msg': formatting} | ||
if args: | ||||
atom[b'args'] = args | ||||
if labels: | ||||
atom[b'labels'] = labels | ||||
Gregory Szorc
|
r37078 | |||
Gregory Szorc
|
r37335 | atomdicts.append(atom) | ||
Gregory Szorc
|
r37078 | |||
Gregory Szorc
|
r39477 | payload = b''.join(cborutil.streamencode(atomdicts)) | ||
Gregory Szorc
|
r37078 | |||
Gregory Szorc
|
r37335 | if len(payload) > maxframesize: | ||
Gregory Szorc
|
r37078 | raise ValueError('cannot encode data in a single frame') | ||
Gregory Szorc
|
r37303 | yield stream.makeframe(requestid=requestid, | ||
typeid=FRAME_TYPE_TEXT_OUTPUT, | ||||
flags=0, | ||||
Gregory Szorc
|
r37335 | payload=payload) | ||
Gregory Szorc
|
r37303 | |||
class stream(object): | ||||
"""Represents a logical unidirectional series of frames.""" | ||||
Gregory Szorc
|
r37304 | def __init__(self, streamid, active=False): | ||
self.streamid = streamid | ||||
Gregory Szorc
|
r37673 | self._active = active | ||
Gregory Szorc
|
r37304 | |||
Gregory Szorc
|
r37303 | def makeframe(self, requestid, typeid, flags, payload): | ||
"""Create a frame to be sent out over this stream. | ||||
Only returns the frame instance. Does not actually send it. | ||||
""" | ||||
Gregory Szorc
|
r37304 | streamflags = 0 | ||
if not self._active: | ||||
streamflags |= STREAM_FLAG_BEGIN_STREAM | ||||
self._active = True | ||||
return makeframe(requestid, self.streamid, streamflags, typeid, flags, | ||||
payload) | ||||
def ensureserverstream(stream): | ||||
if stream.streamid % 2: | ||||
raise error.ProgrammingError('server should only write to even ' | ||||
'numbered streams; %d is not even' % | ||||
stream.streamid) | ||||
Gregory Szorc
|
r37078 | |||
Gregory Szorc
|
r37070 | class serverreactor(object): | ||
"""Holds state of a server handling frame-based protocol requests. | ||||
This class is the "brain" of the unified frame-based protocol server | ||||
component. While the protocol is stateless from the perspective of | ||||
requests/commands, something needs to track which frames have been | ||||
received, what frames to expect, etc. This class is that thing. | ||||
Instances are modeled as a state machine of sorts. Instances are also | ||||
reactionary to external events. The point of this class is to encapsulate | ||||
the state of the connection and the exchange of frames, not to perform | ||||
work. Instead, callers tell this class when something occurs, like a | ||||
frame arriving. If that activity is worthy of a follow-up action (say | ||||
*run a command*), the return value of that handler will say so. | ||||
I/O and CPU intensive operations are purposefully delegated outside of | ||||
this class. | ||||
Consumers are expected to tell instances when events occur. They do so by | ||||
calling the various ``on*`` methods. These methods return a 2-tuple | ||||
describing any follow-up action(s) to take. The first element is the | ||||
name of an action to perform. The second is a data structure (usually | ||||
a dict) specific to that action that contains more information. e.g. | ||||
if the server wants to send frames back to the client, the data structure | ||||
will contain a reference to those frames. | ||||
Valid actions that consumers can be instructed to take are: | ||||
Gregory Szorc
|
r37073 | sendframes | ||
Indicates that frames should be sent to the client. The ``framegen`` | ||||
key contains a generator of frames that should be sent. The server | ||||
assumes that all frames are sent to the client. | ||||
Gregory Szorc
|
r37070 | error | ||
Indicates that an error occurred. Consumer should probably abort. | ||||
runcommand | ||||
Indicates that the consumer should run a wire protocol command. Details | ||||
of the command to run are given in the data structure. | ||||
wantframe | ||||
Indicates that nothing of interest happened and the server is waiting on | ||||
more frames from the client before anything interesting can be done. | ||||
Gregory Szorc
|
r37074 | |||
noop | ||||
Indicates no additional action is required. | ||||
Gregory Szorc
|
r37076 | |||
Known Issues | ||||
------------ | ||||
There are no limits to the number of partially received commands or their | ||||
size. A malicious client could stream command request data and exhaust the | ||||
server's memory. | ||||
Partially received commands are not acted upon when end of input is | ||||
reached. Should the server error if it receives a partial request? | ||||
Should the client send a message to abort a partially transmitted request | ||||
to facilitate graceful shutdown? | ||||
Active requests that haven't been responded to aren't tracked. This means | ||||
that if we receive a command and instruct its dispatch, another command | ||||
with its request ID can come in over the wire and there will be a race | ||||
between who responds to what. | ||||
Gregory Szorc
|
r37070 | """ | ||
Gregory Szorc
|
r37074 | def __init__(self, deferoutput=False): | ||
"""Construct a new server reactor. | ||||
``deferoutput`` can be used to indicate that no output frames should be | ||||
instructed to be sent until input has been exhausted. In this mode, | ||||
events that would normally generate output frames (such as a command | ||||
response being ready) will instead defer instructing the consumer to | ||||
send those frames. This is useful for half-duplex transports where the | ||||
sender cannot receive until all data has been transmitted. | ||||
""" | ||||
self._deferoutput = deferoutput | ||||
Gregory Szorc
|
r37070 | self._state = 'idle' | ||
Gregory Szorc
|
r37305 | self._nextoutgoingstreamid = 2 | ||
Gregory Szorc
|
r37074 | self._bufferedframegens = [] | ||
Gregory Szorc
|
r37304 | # stream id -> stream instance for all active streams from the client. | ||
self._incomingstreams = {} | ||||
Gregory Szorc
|
r37305 | self._outgoingstreams = {} | ||
Gregory Szorc
|
r37076 | # request id -> dict of commands that are actively being received. | ||
self._receivingcommands = {} | ||||
Gregory Szorc
|
r37081 | # Request IDs that have been received and are actively being processed. | ||
# Once all output for a request has been sent, it is removed from this | ||||
# set. | ||||
self._activecommands = set() | ||||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37079 | def onframerecv(self, frame): | ||
Gregory Szorc
|
r37070 | """Process a frame that has been received off the wire. | ||
Returns a dict with an ``action`` key that details what action, | ||||
if any, the consumer should take next. | ||||
""" | ||||
Gregory Szorc
|
r37304 | if not frame.streamid % 2: | ||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('received frame with even numbered stream ID: %d') % | ||||
frame.streamid) | ||||
if frame.streamid not in self._incomingstreams: | ||||
if not frame.streamflags & STREAM_FLAG_BEGIN_STREAM: | ||||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('received frame on unknown inactive stream without ' | ||||
'beginning of stream flag set')) | ||||
self._incomingstreams[frame.streamid] = stream(frame.streamid) | ||||
if frame.streamflags & STREAM_FLAG_ENCODING_APPLIED: | ||||
# TODO handle decoding frames | ||||
self._state = 'errored' | ||||
raise error.ProgrammingError('support for decoding stream payloads ' | ||||
'not yet implemented') | ||||
if frame.streamflags & STREAM_FLAG_END_STREAM: | ||||
del self._incomingstreams[frame.streamid] | ||||
Gregory Szorc
|
r37070 | handlers = { | ||
'idle': self._onframeidle, | ||||
Gregory Szorc
|
r37076 | 'command-receiving': self._onframecommandreceiving, | ||
Gregory Szorc
|
r37070 | 'errored': self._onframeerrored, | ||
} | ||||
meth = handlers.get(self._state) | ||||
if not meth: | ||||
raise error.ProgrammingError('unhandled state: %s' % self._state) | ||||
Gregory Szorc
|
r37079 | return meth(frame) | ||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37742 | def oncommandresponseready(self, stream, requestid, data): | ||
Gregory Szorc
|
r37073 | """Signal that a bytes response is ready to be sent to the client. | ||
The raw bytes response is passed as an argument. | ||||
""" | ||||
Gregory Szorc
|
r37304 | ensureserverstream(stream) | ||
Gregory Szorc
|
r37081 | def sendframes(): | ||
Gregory Szorc
|
r37742 | for frame in createcommandresponseframesfrombytes(stream, requestid, | ||
data): | ||||
Gregory Szorc
|
r37081 | yield frame | ||
self._activecommands.remove(requestid) | ||||
result = sendframes() | ||||
Gregory Szorc
|
r37074 | |||
if self._deferoutput: | ||||
Gregory Szorc
|
r37081 | self._bufferedframegens.append(result) | ||
Gregory Szorc
|
r37074 | return 'noop', {} | ||
else: | ||||
return 'sendframes', { | ||||
Gregory Szorc
|
r37081 | 'framegen': result, | ||
Gregory Szorc
|
r37074 | } | ||
Gregory Szorc
|
r37746 | def oncommandresponsereadygen(self, stream, requestid, gen): | ||
"""Signal that a bytes response is ready, with data as a generator.""" | ||||
ensureserverstream(stream) | ||||
def sendframes(): | ||||
for frame in createbytesresponseframesfromgen(stream, requestid, | ||||
gen): | ||||
yield frame | ||||
self._activecommands.remove(requestid) | ||||
return self._handlesendframes(sendframes()) | ||||
Gregory Szorc
|
r37074 | def oninputeof(self): | ||
"""Signals that end of input has been received. | ||||
No more frames will be received. All pending activity should be | ||||
completed. | ||||
""" | ||||
Gregory Szorc
|
r37076 | # TODO should we do anything about in-flight commands? | ||
Gregory Szorc
|
r37074 | if not self._deferoutput or not self._bufferedframegens: | ||
return 'noop', {} | ||||
# If we buffered all our responses, emit those. | ||||
def makegen(): | ||||
for gen in self._bufferedframegens: | ||||
for frame in gen: | ||||
yield frame | ||||
Gregory Szorc
|
r37073 | return 'sendframes', { | ||
Gregory Szorc
|
r37074 | 'framegen': makegen(), | ||
Gregory Szorc
|
r37073 | } | ||
Gregory Szorc
|
r37746 | def _handlesendframes(self, framegen): | ||
if self._deferoutput: | ||||
self._bufferedframegens.append(framegen) | ||||
return 'noop', {} | ||||
else: | ||||
return 'sendframes', { | ||||
'framegen': framegen, | ||||
} | ||||
Gregory Szorc
|
r37744 | def onservererror(self, stream, requestid, msg): | ||
Gregory Szorc
|
r37304 | ensureserverstream(stream) | ||
Gregory Szorc
|
r37746 | def sendframes(): | ||
for frame in createerrorframe(stream, requestid, msg, | ||||
errtype='server'): | ||||
yield frame | ||||
self._activecommands.remove(requestid) | ||||
return self._handlesendframes(sendframes()) | ||||
def oncommanderror(self, stream, requestid, message, args=None): | ||||
"""Called when a command encountered an error before sending output.""" | ||||
ensureserverstream(stream) | ||||
def sendframes(): | ||||
for frame in createcommanderrorresponse(stream, requestid, message, | ||||
args): | ||||
yield frame | ||||
self._activecommands.remove(requestid) | ||||
return self._handlesendframes(sendframes()) | ||||
Gregory Szorc
|
r37073 | |||
Gregory Szorc
|
r37305 | def makeoutputstream(self): | ||
"""Create a stream to be used for sending data to the client.""" | ||||
streamid = self._nextoutgoingstreamid | ||||
self._nextoutgoingstreamid += 2 | ||||
s = stream(streamid) | ||||
self._outgoingstreams[streamid] = s | ||||
return s | ||||
Gregory Szorc
|
r37070 | def _makeerrorresult(self, msg): | ||
return 'error', { | ||||
'message': msg, | ||||
} | ||||
Gregory Szorc
|
r37076 | def _makeruncommandresult(self, requestid): | ||
entry = self._receivingcommands[requestid] | ||||
Gregory Szorc
|
r37308 | |||
if not entry['requestdone']: | ||||
self._state = 'errored' | ||||
raise error.ProgrammingError('should not be called without ' | ||||
'requestdone set') | ||||
Gregory Szorc
|
r37076 | del self._receivingcommands[requestid] | ||
if self._receivingcommands: | ||||
self._state = 'command-receiving' | ||||
else: | ||||
self._state = 'idle' | ||||
Gregory Szorc
|
r37308 | # Decode the payloads as CBOR. | ||
entry['payload'].seek(0) | ||||
Gregory Szorc
|
r39477 | request = cborutil.decodeall(entry['payload'].getvalue())[0] | ||
Gregory Szorc
|
r37308 | |||
if b'name' not in request: | ||||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('command request missing "name" field')) | ||||
if b'args' not in request: | ||||
request[b'args'] = {} | ||||
Gregory Szorc
|
r37081 | assert requestid not in self._activecommands | ||
self._activecommands.add(requestid) | ||||
Gregory Szorc
|
r37070 | return 'runcommand', { | ||
Gregory Szorc
|
r37076 | 'requestid': requestid, | ||
Gregory Szorc
|
r37308 | 'command': request[b'name'], | ||
'args': request[b'args'], | ||||
Gregory Szorc
|
r37076 | 'data': entry['data'].getvalue() if entry['data'] else None, | ||
Gregory Szorc
|
r37070 | } | ||
def _makewantframeresult(self): | ||||
return 'wantframe', { | ||||
'state': self._state, | ||||
} | ||||
Gregory Szorc
|
r37308 | def _validatecommandrequestframe(self, frame): | ||
new = frame.flags & FLAG_COMMAND_REQUEST_NEW | ||||
continuation = frame.flags & FLAG_COMMAND_REQUEST_CONTINUATION | ||||
if new and continuation: | ||||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('received command request frame with both new and ' | ||||
'continuation flags set')) | ||||
if not new and not continuation: | ||||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('received command request frame with neither new nor ' | ||||
'continuation flags set')) | ||||
Gregory Szorc
|
r37079 | def _onframeidle(self, frame): | ||
Gregory Szorc
|
r37070 | # The only frame type that should be received in this state is a | ||
# command request. | ||||
Gregory Szorc
|
r37308 | if frame.typeid != FRAME_TYPE_COMMAND_REQUEST: | ||
Gregory Szorc
|
r37070 | self._state = 'errored' | ||
return self._makeerrorresult( | ||||
Gregory Szorc
|
r37308 | _('expected command request frame; got %d') % frame.typeid) | ||
res = self._validatecommandrequestframe(frame) | ||||
if res: | ||||
return res | ||||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37079 | if frame.requestid in self._receivingcommands: | ||
Gregory Szorc
|
r37076 | self._state = 'errored' | ||
return self._makeerrorresult( | ||||
Gregory Szorc
|
r37079 | _('request with ID %d already received') % frame.requestid) | ||
Gregory Szorc
|
r37076 | |||
Gregory Szorc
|
r37081 | if frame.requestid in self._activecommands: | ||
self._state = 'errored' | ||||
Gregory Szorc
|
r37308 | return self._makeerrorresult( | ||
_('request with ID %d is already active') % frame.requestid) | ||||
new = frame.flags & FLAG_COMMAND_REQUEST_NEW | ||||
moreframes = frame.flags & FLAG_COMMAND_REQUEST_MORE_FRAMES | ||||
expectingdata = frame.flags & FLAG_COMMAND_REQUEST_EXPECT_DATA | ||||
Gregory Szorc
|
r37081 | |||
Gregory Szorc
|
r37308 | if not new: | ||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('received command request frame without new flag set')) | ||||
payload = util.bytesio() | ||||
payload.write(frame.payload) | ||||
Gregory Szorc
|
r37076 | |||
Gregory Szorc
|
r37079 | self._receivingcommands[frame.requestid] = { | ||
Gregory Szorc
|
r37308 | 'payload': payload, | ||
Gregory Szorc
|
r37076 | 'data': None, | ||
Gregory Szorc
|
r37308 | 'requestdone': not moreframes, | ||
'expectingdata': bool(expectingdata), | ||||
Gregory Szorc
|
r37076 | } | ||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37308 | # This is the final frame for this request. Dispatch it. | ||
if not moreframes and not expectingdata: | ||||
Gregory Szorc
|
r37079 | return self._makeruncommandresult(frame.requestid) | ||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37308 | assert moreframes or expectingdata | ||
self._state = 'command-receiving' | ||||
return self._makewantframeresult() | ||||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37079 | def _onframecommandreceiving(self, frame): | ||
Gregory Szorc
|
r37308 | if frame.typeid == FRAME_TYPE_COMMAND_REQUEST: | ||
# Process new command requests as such. | ||||
if frame.flags & FLAG_COMMAND_REQUEST_NEW: | ||||
return self._onframeidle(frame) | ||||
res = self._validatecommandrequestframe(frame) | ||||
if res: | ||||
return res | ||||
Gregory Szorc
|
r37076 | |||
# All other frames should be related to a command that is currently | ||||
Gregory Szorc
|
r37081 | # receiving but is not active. | ||
if frame.requestid in self._activecommands: | ||||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('received frame for request that is still active: %d') % | ||||
frame.requestid) | ||||
Gregory Szorc
|
r37079 | if frame.requestid not in self._receivingcommands: | ||
Gregory Szorc
|
r37070 | self._state = 'errored' | ||
Gregory Szorc
|
r37076 | return self._makeerrorresult( | ||
_('received frame for request that is not receiving: %d') % | ||||
Gregory Szorc
|
r37079 | frame.requestid) | ||
Gregory Szorc
|
r37076 | |||
Gregory Szorc
|
r37079 | entry = self._receivingcommands[frame.requestid] | ||
Gregory Szorc
|
r37076 | |||
Gregory Szorc
|
r37308 | if frame.typeid == FRAME_TYPE_COMMAND_REQUEST: | ||
moreframes = frame.flags & FLAG_COMMAND_REQUEST_MORE_FRAMES | ||||
expectingdata = bool(frame.flags & FLAG_COMMAND_REQUEST_EXPECT_DATA) | ||||
if entry['requestdone']: | ||||
self._state = 'errored' | ||||
return self._makeerrorresult( | ||||
_('received command request frame when request frames ' | ||||
'were supposedly done')) | ||||
if expectingdata != entry['expectingdata']: | ||||
Gregory Szorc
|
r37076 | self._state = 'errored' | ||
Gregory Szorc
|
r37308 | return self._makeerrorresult( | ||
_('mismatch between expect data flag and previous frame')) | ||||
entry['payload'].write(frame.payload) | ||||
Gregory Szorc
|
r37076 | |||
Gregory Szorc
|
r37308 | if not moreframes: | ||
entry['requestdone'] = True | ||||
if not moreframes and not expectingdata: | ||||
return self._makeruncommandresult(frame.requestid) | ||||
return self._makewantframeresult() | ||||
Gregory Szorc
|
r37076 | |||
Gregory Szorc
|
r37079 | elif frame.typeid == FRAME_TYPE_COMMAND_DATA: | ||
Gregory Szorc
|
r37076 | if not entry['expectingdata']: | ||
self._state = 'errored' | ||||
return self._makeerrorresult(_( | ||||
'received command data frame for request that is not ' | ||||
Gregory Szorc
|
r37079 | 'expecting data: %d') % frame.requestid) | ||
Gregory Szorc
|
r37076 | |||
if entry['data'] is None: | ||||
entry['data'] = util.bytesio() | ||||
Gregory Szorc
|
r37079 | return self._handlecommanddataframe(frame, entry) | ||
Gregory Szorc
|
r37308 | else: | ||
Gregory Szorc
|
r37070 | self._state = 'errored' | ||
Gregory Szorc
|
r37308 | return self._makeerrorresult(_( | ||
'received unexpected frame type: %d') % frame.typeid) | ||||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37079 | def _handlecommanddataframe(self, frame, entry): | ||
assert frame.typeid == FRAME_TYPE_COMMAND_DATA | ||||
Gregory Szorc
|
r37070 | |||
# TODO support streaming data instead of buffering it. | ||||
Gregory Szorc
|
r37079 | entry['data'].write(frame.payload) | ||
Gregory Szorc
|
r37070 | |||
Gregory Szorc
|
r37079 | if frame.flags & FLAG_COMMAND_DATA_CONTINUATION: | ||
Gregory Szorc
|
r37070 | return self._makewantframeresult() | ||
Gregory Szorc
|
r37079 | elif frame.flags & FLAG_COMMAND_DATA_EOS: | ||
Gregory Szorc
|
r37076 | entry['data'].seek(0) | ||
Gregory Szorc
|
r37079 | return self._makeruncommandresult(frame.requestid) | ||
Gregory Szorc
|
r37070 | else: | ||
self._state = 'errored' | ||||
return self._makeerrorresult(_('command data frame without ' | ||||
'flags')) | ||||
Gregory Szorc
|
r37079 | def _onframeerrored(self, frame): | ||
Gregory Szorc
|
r37070 | return self._makeerrorresult(_('server already errored')) | ||
Gregory Szorc
|
r37561 | |||
class commandrequest(object): | ||||
"""Represents a request to run a command.""" | ||||
def __init__(self, requestid, name, args, datafh=None): | ||||
self.requestid = requestid | ||||
self.name = name | ||||
self.args = args | ||||
self.datafh = datafh | ||||
self.state = 'pending' | ||||
class clientreactor(object): | ||||
"""Holds state of a client issuing frame-based protocol requests. | ||||
This is like ``serverreactor`` but for client-side state. | ||||
Each instance is bound to the lifetime of a connection. For persistent | ||||
connection transports using e.g. TCP sockets and speaking the raw | ||||
framing protocol, there will be a single instance for the lifetime of | ||||
the TCP socket. For transports where there are multiple discrete | ||||
interactions (say tunneled within in HTTP request), there will be a | ||||
separate instance for each distinct interaction. | ||||
""" | ||||
def __init__(self, hasmultiplesend=False, buffersends=True): | ||||
"""Create a new instance. | ||||
``hasmultiplesend`` indicates whether multiple sends are supported | ||||
by the transport. When True, it is possible to send commands immediately | ||||
instead of buffering until the caller signals an intent to finish a | ||||
send operation. | ||||
``buffercommands`` indicates whether sends should be buffered until the | ||||
last request has been issued. | ||||
""" | ||||
self._hasmultiplesend = hasmultiplesend | ||||
self._buffersends = buffersends | ||||
self._canissuecommands = True | ||||
self._cansend = True | ||||
self._nextrequestid = 1 | ||||
# We only support a single outgoing stream for now. | ||||
self._outgoingstream = stream(1) | ||||
self._pendingrequests = collections.deque() | ||||
self._activerequests = {} | ||||
Gregory Szorc
|
r37562 | self._incomingstreams = {} | ||
Gregory Szorc
|
r37561 | |||
def callcommand(self, name, args, datafh=None): | ||||
"""Request that a command be executed. | ||||
Receives the command name, a dict of arguments to pass to the command, | ||||
and an optional file object containing the raw data for the command. | ||||
Returns a 3-tuple of (request, action, action data). | ||||
""" | ||||
if not self._canissuecommands: | ||||
raise error.ProgrammingError('cannot issue new commands') | ||||
requestid = self._nextrequestid | ||||
self._nextrequestid += 2 | ||||
request = commandrequest(requestid, name, args, datafh=datafh) | ||||
if self._buffersends: | ||||
self._pendingrequests.append(request) | ||||
return request, 'noop', {} | ||||
else: | ||||
if not self._cansend: | ||||
raise error.ProgrammingError('sends cannot be performed on ' | ||||
'this instance') | ||||
if not self._hasmultiplesend: | ||||
self._cansend = False | ||||
self._canissuecommands = False | ||||
return request, 'sendframes', { | ||||
'framegen': self._makecommandframes(request), | ||||
} | ||||
def flushcommands(self): | ||||
"""Request that all queued commands be sent. | ||||
If any commands are buffered, this will instruct the caller to send | ||||
them over the wire. If no commands are buffered it instructs the client | ||||
to no-op. | ||||
If instances aren't configured for multiple sends, no new command | ||||
requests are allowed after this is called. | ||||
""" | ||||
if not self._pendingrequests: | ||||
return 'noop', {} | ||||
if not self._cansend: | ||||
raise error.ProgrammingError('sends cannot be performed on this ' | ||||
'instance') | ||||
# If the instance only allows sending once, mark that we have fired | ||||
# our one shot. | ||||
if not self._hasmultiplesend: | ||||
self._canissuecommands = False | ||||
self._cansend = False | ||||
def makeframes(): | ||||
while self._pendingrequests: | ||||
request = self._pendingrequests.popleft() | ||||
for frame in self._makecommandframes(request): | ||||
yield frame | ||||
return 'sendframes', { | ||||
'framegen': makeframes(), | ||||
} | ||||
def _makecommandframes(self, request): | ||||
"""Emit frames to issue a command request. | ||||
As a side-effect, update request accounting to reflect its changed | ||||
state. | ||||
""" | ||||
self._activerequests[request.requestid] = request | ||||
request.state = 'sending' | ||||
res = createcommandframes(self._outgoingstream, | ||||
request.requestid, | ||||
request.name, | ||||
request.args, | ||||
request.datafh) | ||||
for frame in res: | ||||
yield frame | ||||
request.state = 'sent' | ||||
Gregory Szorc
|
r37562 | |||
def onframerecv(self, frame): | ||||
"""Process a frame that has been received off the wire. | ||||
Returns a 2-tuple of (action, meta) describing further action the | ||||
caller needs to take as a result of receiving this frame. | ||||
""" | ||||
if frame.streamid % 2: | ||||
return 'error', { | ||||
'message': ( | ||||
_('received frame with odd numbered stream ID: %d') % | ||||
frame.streamid), | ||||
} | ||||
if frame.streamid not in self._incomingstreams: | ||||
if not frame.streamflags & STREAM_FLAG_BEGIN_STREAM: | ||||
return 'error', { | ||||
'message': _('received frame on unknown stream ' | ||||
'without beginning of stream flag set'), | ||||
} | ||||
Gregory Szorc
|
r37674 | self._incomingstreams[frame.streamid] = stream(frame.streamid) | ||
Gregory Szorc
|
r37562 | if frame.streamflags & STREAM_FLAG_ENCODING_APPLIED: | ||
raise error.ProgrammingError('support for decoding stream ' | ||||
'payloads not yet implemneted') | ||||
if frame.streamflags & STREAM_FLAG_END_STREAM: | ||||
del self._incomingstreams[frame.streamid] | ||||
if frame.requestid not in self._activerequests: | ||||
return 'error', { | ||||
'message': (_('received frame for inactive request ID: %d') % | ||||
frame.requestid), | ||||
} | ||||
request = self._activerequests[frame.requestid] | ||||
request.state = 'receiving' | ||||
handlers = { | ||||
Gregory Szorc
|
r37742 | FRAME_TYPE_COMMAND_RESPONSE: self._oncommandresponseframe, | ||
Gregory Szorc
|
r37744 | FRAME_TYPE_ERROR_RESPONSE: self._onerrorresponseframe, | ||
Gregory Szorc
|
r37562 | } | ||
meth = handlers.get(frame.typeid) | ||||
if not meth: | ||||
raise error.ProgrammingError('unhandled frame type: %d' % | ||||
frame.typeid) | ||||
return meth(request, frame) | ||||
Gregory Szorc
|
r37742 | def _oncommandresponseframe(self, request, frame): | ||
if frame.flags & FLAG_COMMAND_RESPONSE_EOS: | ||||
Gregory Szorc
|
r37562 | request.state = 'received' | ||
del self._activerequests[request.requestid] | ||||
return 'responsedata', { | ||||
'request': request, | ||||
Gregory Szorc
|
r37742 | 'expectmore': frame.flags & FLAG_COMMAND_RESPONSE_CONTINUATION, | ||
'eos': frame.flags & FLAG_COMMAND_RESPONSE_EOS, | ||||
Gregory Szorc
|
r37562 | 'data': frame.payload, | ||
} | ||||
Gregory Szorc
|
r37744 | |||
def _onerrorresponseframe(self, request, frame): | ||||
request.state = 'errored' | ||||
del self._activerequests[request.requestid] | ||||
# The payload should be a CBOR map. | ||||
Gregory Szorc
|
r39477 | m = cborutil.decodeall(frame.payload)[0] | ||
Gregory Szorc
|
r37744 | |||
return 'error', { | ||||
'request': request, | ||||
'type': m['type'], | ||||
'message': m['message'], | ||||
} | ||||