2016-03-24 17:14:12 +00:00
|
|
|
/*
|
|
|
|
* This file is part of the MicroPython project, http://micropython.org/
|
|
|
|
*
|
|
|
|
* The MIT License (MIT)
|
|
|
|
*
|
|
|
|
* Copyright (c) 2016 Paul Sokolovsky
|
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
|
|
* in the Software without restriction, including without limitation the rights
|
|
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
*
|
|
|
|
* The above copyright notice and this permission notice shall be included in
|
|
|
|
* all copies or substantial portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
|
* THE SOFTWARE.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdint.h>
|
2016-03-24 22:51:51 +00:00
|
|
|
#include <string.h>
|
2016-03-24 17:14:12 +00:00
|
|
|
|
|
|
|
#include "py/nlr.h"
|
|
|
|
#include "py/obj.h"
|
2016-04-08 14:05:48 +01:00
|
|
|
#include "py/runtime.h"
|
2016-03-24 17:14:12 +00:00
|
|
|
#include "py/stream.h"
|
2016-04-10 11:42:51 +01:00
|
|
|
#include "extmod/modwebsocket.h"
|
2016-03-24 17:14:12 +00:00
|
|
|
|
|
|
|
#if MICROPY_PY_WEBSOCKET
|
|
|
|
|
2016-04-27 10:49:30 +01:00
|
|
|
enum { FRAME_HEADER, FRAME_OPT, PAYLOAD, CONTROL };
|
2016-04-09 14:14:47 +01:00
|
|
|
|
2016-04-10 11:19:26 +01:00
|
|
|
enum { BLOCKING_WRITE = 0x80 };
|
2016-03-24 17:14:12 +00:00
|
|
|
|
|
|
|
typedef struct _mp_obj_websocket_t {
|
|
|
|
mp_obj_base_t base;
|
|
|
|
mp_obj_t sock;
|
2016-03-24 22:51:51 +00:00
|
|
|
uint32_t msg_sz;
|
|
|
|
byte mask[4];
|
2016-03-24 17:14:12 +00:00
|
|
|
byte state;
|
|
|
|
byte to_recv;
|
|
|
|
byte mask_pos;
|
2016-03-24 22:51:51 +00:00
|
|
|
byte buf_pos;
|
|
|
|
byte buf[6];
|
2016-04-09 14:03:38 +01:00
|
|
|
byte opts;
|
2016-04-27 10:49:30 +01:00
|
|
|
// Copy of last data frame flags
|
2016-04-09 14:14:47 +01:00
|
|
|
byte ws_flags;
|
2016-04-27 10:49:30 +01:00
|
|
|
// Copy of current frame flags
|
|
|
|
byte last_flags;
|
2016-03-24 17:14:12 +00:00
|
|
|
} mp_obj_websocket_t;
|
|
|
|
|
2016-04-27 10:49:30 +01:00
|
|
|
STATIC mp_uint_t websocket_write(mp_obj_t self_in, const void *buf, mp_uint_t size, int *errcode);
|
|
|
|
|
2016-03-24 17:14:12 +00:00
|
|
|
STATIC mp_obj_t websocket_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
|
2016-04-09 14:03:38 +01:00
|
|
|
mp_arg_check_num(n_args, n_kw, 1, 2, false);
|
2016-03-24 17:14:12 +00:00
|
|
|
mp_obj_websocket_t *o = m_new_obj(mp_obj_websocket_t);
|
|
|
|
o->base.type = type;
|
|
|
|
o->sock = args[0];
|
|
|
|
o->state = FRAME_HEADER;
|
|
|
|
o->to_recv = 2;
|
|
|
|
o->mask_pos = 0;
|
2016-03-24 22:51:51 +00:00
|
|
|
o->buf_pos = 0;
|
2016-04-10 11:19:26 +01:00
|
|
|
o->opts = FRAME_TXT;
|
2016-04-09 14:03:38 +01:00
|
|
|
if (n_args > 1 && args[1] == mp_const_true) {
|
|
|
|
o->opts |= BLOCKING_WRITE;
|
|
|
|
}
|
2016-08-06 13:53:16 +01:00
|
|
|
return MP_OBJ_FROM_PTR(o);
|
2016-03-24 17:14:12 +00:00
|
|
|
}
|
|
|
|
|
2016-03-24 22:51:51 +00:00
|
|
|
STATIC mp_uint_t websocket_read(mp_obj_t self_in, void *buf, mp_uint_t size, int *errcode) {
|
2016-08-06 13:53:16 +01:00
|
|
|
mp_obj_websocket_t *self = MP_OBJ_TO_PTR(self_in);
|
2016-03-24 22:51:51 +00:00
|
|
|
const mp_stream_p_t *stream_p = mp_get_stream_raise(self->sock, MP_STREAM_OP_READ);
|
|
|
|
while (1) {
|
|
|
|
if (self->to_recv != 0) {
|
|
|
|
mp_uint_t out_sz = stream_p->read(self->sock, self->buf + self->buf_pos, self->to_recv, errcode);
|
2016-04-08 18:07:11 +01:00
|
|
|
if (out_sz == 0 || out_sz == MP_STREAM_ERROR) {
|
2016-03-24 22:51:51 +00:00
|
|
|
return out_sz;
|
|
|
|
}
|
|
|
|
self->buf_pos += out_sz;
|
|
|
|
self->to_recv -= out_sz;
|
|
|
|
if (self->to_recv != 0) {
|
2017-07-24 09:43:14 +01:00
|
|
|
*errcode = MP_EAGAIN;
|
2016-03-24 22:51:51 +00:00
|
|
|
return MP_STREAM_ERROR;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
switch (self->state) {
|
|
|
|
case FRAME_HEADER: {
|
2016-04-09 14:14:47 +01:00
|
|
|
// TODO: Split frame handling below is untested so far, so conservatively disable it
|
2016-03-24 22:51:51 +00:00
|
|
|
assert(self->buf[0] & 0x80);
|
2016-04-09 10:29:18 +01:00
|
|
|
|
2016-04-09 14:14:47 +01:00
|
|
|
// "Control frames MAY be injected in the middle of a fragmented message."
|
|
|
|
// So, they must be processed before data frames (and not alter
|
|
|
|
// self->ws_flags)
|
2016-04-27 10:49:30 +01:00
|
|
|
byte frame_type = self->buf[0];
|
|
|
|
self->last_flags = frame_type;
|
|
|
|
frame_type &= FRAME_OPCODE_MASK;
|
2016-04-09 14:14:47 +01:00
|
|
|
|
|
|
|
if ((self->buf[0] & FRAME_OPCODE_MASK) == FRAME_CONT) {
|
|
|
|
// Preserve previous frame type
|
|
|
|
self->ws_flags = (self->ws_flags & FRAME_OPCODE_MASK) | (self->buf[0] & ~FRAME_OPCODE_MASK);
|
|
|
|
} else {
|
|
|
|
self->ws_flags = self->buf[0];
|
|
|
|
}
|
|
|
|
|
2016-04-09 10:29:18 +01:00
|
|
|
// Reset mask in case someone will use "simplified" protocol
|
|
|
|
// without masks.
|
|
|
|
memset(self->mask, 0, sizeof(self->mask));
|
|
|
|
|
2016-03-24 22:51:51 +00:00
|
|
|
int to_recv = 0;
|
|
|
|
size_t sz = self->buf[1] & 0x7f;
|
|
|
|
if (sz == 126) {
|
|
|
|
// Msg size is next 2 bytes
|
|
|
|
to_recv += 2;
|
|
|
|
} else if (sz == 127) {
|
2016-04-27 10:49:30 +01:00
|
|
|
// Msg size is next 8 bytes
|
2016-03-24 22:51:51 +00:00
|
|
|
assert(0);
|
|
|
|
}
|
|
|
|
if (self->buf[1] & 0x80) {
|
|
|
|
// Next 4 bytes is mask
|
|
|
|
to_recv += 4;
|
|
|
|
}
|
|
|
|
|
|
|
|
self->buf_pos = 0;
|
|
|
|
self->to_recv = to_recv;
|
2017-05-29 08:08:14 +01:00
|
|
|
self->msg_sz = sz; // May be overridden by FRAME_OPT
|
2016-03-24 22:51:51 +00:00
|
|
|
if (to_recv != 0) {
|
|
|
|
self->state = FRAME_OPT;
|
|
|
|
} else {
|
2016-04-27 10:49:30 +01:00
|
|
|
if (frame_type >= FRAME_CLOSE) {
|
|
|
|
self->state = CONTROL;
|
|
|
|
} else {
|
|
|
|
self->state = PAYLOAD;
|
|
|
|
}
|
2016-03-24 22:51:51 +00:00
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
case FRAME_OPT: {
|
|
|
|
if ((self->buf_pos & 3) == 2) {
|
|
|
|
// First two bytes are message length
|
|
|
|
self->msg_sz = (self->buf[0] << 8) | self->buf[1];
|
|
|
|
}
|
|
|
|
if (self->buf_pos >= 4) {
|
|
|
|
// Last 4 bytes is mask
|
|
|
|
memcpy(self->mask, self->buf + self->buf_pos - 4, 4);
|
|
|
|
}
|
|
|
|
self->buf_pos = 0;
|
2016-04-27 10:49:30 +01:00
|
|
|
if ((self->last_flags & FRAME_OPCODE_MASK) >= FRAME_CLOSE) {
|
|
|
|
self->state = CONTROL;
|
|
|
|
} else {
|
|
|
|
self->state = PAYLOAD;
|
|
|
|
}
|
2016-03-24 22:51:51 +00:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2016-04-27 10:49:30 +01:00
|
|
|
case PAYLOAD:
|
|
|
|
case CONTROL: {
|
|
|
|
mp_uint_t out_sz = 0;
|
|
|
|
if (self->msg_sz == 0) {
|
|
|
|
// In case message had zero payload
|
|
|
|
goto no_payload;
|
|
|
|
}
|
|
|
|
|
2016-03-24 22:51:51 +00:00
|
|
|
size_t sz = MIN(size, self->msg_sz);
|
2016-04-27 10:49:30 +01:00
|
|
|
out_sz = stream_p->read(self->sock, buf, sz, errcode);
|
2016-04-13 20:14:46 +01:00
|
|
|
if (out_sz == 0 || out_sz == MP_STREAM_ERROR) {
|
2016-03-24 22:51:51 +00:00
|
|
|
return out_sz;
|
|
|
|
}
|
|
|
|
|
|
|
|
sz = out_sz;
|
|
|
|
for (byte *p = buf; sz--; p++) {
|
|
|
|
*p ^= self->mask[self->mask_pos++ & 3];
|
|
|
|
}
|
|
|
|
|
|
|
|
self->msg_sz -= out_sz;
|
|
|
|
if (self->msg_sz == 0) {
|
2016-04-27 10:49:30 +01:00
|
|
|
byte last_state;
|
|
|
|
no_payload:
|
|
|
|
last_state = self->state;
|
2016-03-24 22:51:51 +00:00
|
|
|
self->state = FRAME_HEADER;
|
|
|
|
self->to_recv = 2;
|
|
|
|
self->mask_pos = 0;
|
|
|
|
self->buf_pos = 0;
|
2016-04-27 10:49:30 +01:00
|
|
|
|
|
|
|
// Handle control frame
|
|
|
|
if (last_state == CONTROL) {
|
|
|
|
byte frame_type = self->last_flags & FRAME_OPCODE_MASK;
|
|
|
|
if (frame_type == FRAME_CLOSE) {
|
|
|
|
static char close_resp[2] = {0x88, 0};
|
|
|
|
int err;
|
|
|
|
websocket_write(self_in, close_resp, sizeof(close_resp), &err);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
//DEBUG_printf("Finished receiving ctrl message %x, ignoring\n", self->last_flags);
|
|
|
|
continue;
|
|
|
|
}
|
2016-03-24 22:51:51 +00:00
|
|
|
}
|
2016-04-27 10:49:30 +01:00
|
|
|
|
|
|
|
if (out_sz != 0) {
|
|
|
|
return out_sz;
|
|
|
|
}
|
|
|
|
// Empty (data) frame received is not EOF
|
|
|
|
continue;
|
2016-03-24 22:51:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-03-24 17:14:12 +00:00
|
|
|
STATIC mp_uint_t websocket_write(mp_obj_t self_in, const void *buf, mp_uint_t size, int *errcode) {
|
2016-08-06 13:53:16 +01:00
|
|
|
mp_obj_websocket_t *self = MP_OBJ_TO_PTR(self_in);
|
2016-04-11 12:07:57 +01:00
|
|
|
assert(size < 0x10000);
|
|
|
|
byte header[4] = {0x80 | (self->opts & FRAME_OPCODE_MASK)};
|
|
|
|
int hdr_sz;
|
|
|
|
if (size < 126) {
|
|
|
|
header[1] = size;
|
|
|
|
hdr_sz = 2;
|
|
|
|
} else {
|
|
|
|
header[1] = 126;
|
|
|
|
header[2] = size >> 8;
|
|
|
|
header[3] = size & 0xff;
|
|
|
|
hdr_sz = 4;
|
|
|
|
}
|
2016-03-24 17:14:12 +00:00
|
|
|
|
2016-04-09 14:03:38 +01:00
|
|
|
mp_obj_t dest[3];
|
|
|
|
if (self->opts & BLOCKING_WRITE) {
|
|
|
|
mp_load_method(self->sock, MP_QSTR_setblocking, dest);
|
|
|
|
dest[2] = mp_const_true;
|
|
|
|
mp_call_method_n_kw(1, 0, dest);
|
|
|
|
}
|
|
|
|
|
py/stream: Support both "exact size" and "one underlying call" operations.
Both read and write operations support variants where either a) a single
call is made to the undelying stream implementation and returned buffer
length may be less than requested, or b) calls are repeated until requested
amount of data is collected, shorter amount is returned only in case of
EOF or error.
These operations are available from the level of C support functions to be
used by other C modules to implementations of Python methods to be used in
user-facing objects.
The rationale of these changes is to allow to write concise and robust
code to work with *blocking* streams of types prone to short reads, like
serial interfaces and sockets. Particular object types may select "exact"
vs "once" types of methods depending on their needs. E.g., for sockets,
revc() and send() methods continue to be "once", while read() and write()
thus converted to "exactly" versions.
These changes don't affect non-blocking handling, e.g. trying "exact"
method on the non-blocking socket will return as much data as available
without blocking. No data available is continued to be signaled as None
return value to read() and write().
From the point of view of CPython compatibility, this model is a cross
between its io.RawIOBase and io.BufferedIOBase abstract classes. For
blocking streams, it works as io.BufferedIOBase model (guaranteeing
lack of short reads/writes), while for non-blocking - as io.RawIOBase,
returning None in case of lack of data (instead of raising expensive
exception, as required by io.BufferedIOBase). Such a cross-behavior
should be optimal for MicroPython needs.
2016-05-18 00:40:03 +01:00
|
|
|
mp_uint_t out_sz = mp_stream_write_exactly(self->sock, header, hdr_sz, errcode);
|
|
|
|
if (*errcode == 0) {
|
|
|
|
out_sz = mp_stream_write_exactly(self->sock, buf, size, errcode);
|
2016-03-24 17:14:12 +00:00
|
|
|
}
|
2016-04-09 14:03:38 +01:00
|
|
|
|
|
|
|
if (self->opts & BLOCKING_WRITE) {
|
|
|
|
dest[2] = mp_const_false;
|
|
|
|
mp_call_method_n_kw(1, 0, dest);
|
|
|
|
}
|
|
|
|
|
py/stream: Support both "exact size" and "one underlying call" operations.
Both read and write operations support variants where either a) a single
call is made to the undelying stream implementation and returned buffer
length may be less than requested, or b) calls are repeated until requested
amount of data is collected, shorter amount is returned only in case of
EOF or error.
These operations are available from the level of C support functions to be
used by other C modules to implementations of Python methods to be used in
user-facing objects.
The rationale of these changes is to allow to write concise and robust
code to work with *blocking* streams of types prone to short reads, like
serial interfaces and sockets. Particular object types may select "exact"
vs "once" types of methods depending on their needs. E.g., for sockets,
revc() and send() methods continue to be "once", while read() and write()
thus converted to "exactly" versions.
These changes don't affect non-blocking handling, e.g. trying "exact"
method on the non-blocking socket will return as much data as available
without blocking. No data available is continued to be signaled as None
return value to read() and write().
From the point of view of CPython compatibility, this model is a cross
between its io.RawIOBase and io.BufferedIOBase abstract classes. For
blocking streams, it works as io.BufferedIOBase model (guaranteeing
lack of short reads/writes), while for non-blocking - as io.RawIOBase,
returning None in case of lack of data (instead of raising expensive
exception, as required by io.BufferedIOBase). Such a cross-behavior
should be optimal for MicroPython needs.
2016-05-18 00:40:03 +01:00
|
|
|
if (*errcode != 0) {
|
|
|
|
return MP_STREAM_ERROR;
|
|
|
|
}
|
2016-04-09 14:03:38 +01:00
|
|
|
return out_sz;
|
2016-03-24 17:14:12 +00:00
|
|
|
}
|
|
|
|
|
2016-04-10 10:50:46 +01:00
|
|
|
STATIC mp_uint_t websocket_ioctl(mp_obj_t self_in, mp_uint_t request, uintptr_t arg, int *errcode) {
|
|
|
|
mp_obj_websocket_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
switch (request) {
|
|
|
|
case MP_STREAM_GET_DATA_OPTS:
|
|
|
|
return self->ws_flags & FRAME_OPCODE_MASK;
|
2016-04-10 11:19:26 +01:00
|
|
|
case MP_STREAM_SET_DATA_OPTS: {
|
|
|
|
int cur = self->opts & FRAME_OPCODE_MASK;
|
|
|
|
self->opts = (self->opts & ~FRAME_OPCODE_MASK) | (arg & FRAME_OPCODE_MASK);
|
|
|
|
return cur;
|
|
|
|
}
|
2016-04-10 10:50:46 +01:00
|
|
|
default:
|
2017-07-24 09:43:14 +01:00
|
|
|
*errcode = MP_EINVAL;
|
2016-04-10 10:50:46 +01:00
|
|
|
return MP_STREAM_ERROR;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-05-20 19:18:25 +01:00
|
|
|
STATIC mp_obj_t websocket_close(mp_obj_t self_in) {
|
|
|
|
mp_obj_websocket_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
// TODO: Send close signaling to the other side, otherwise it's
|
|
|
|
// abrupt close (connection abort).
|
|
|
|
return mp_stream_close(self->sock);
|
|
|
|
}
|
|
|
|
STATIC MP_DEFINE_CONST_FUN_OBJ_1(websocket_close_obj, websocket_close);
|
|
|
|
|
2016-08-06 14:18:03 +01:00
|
|
|
STATIC const mp_rom_map_elem_t websocket_locals_dict_table[] = {
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR_read), MP_ROM_PTR(&mp_stream_read_obj) },
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR_readinto), MP_ROM_PTR(&mp_stream_readinto_obj) },
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR_readline), MP_ROM_PTR(&mp_stream_unbuffered_readline_obj) },
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR_write), MP_ROM_PTR(&mp_stream_write_obj) },
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR_ioctl), MP_ROM_PTR(&mp_stream_ioctl_obj) },
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR_close), MP_ROM_PTR(&websocket_close_obj) },
|
2016-03-24 17:14:12 +00:00
|
|
|
};
|
|
|
|
STATIC MP_DEFINE_CONST_DICT(websocket_locals_dict, websocket_locals_dict_table);
|
|
|
|
|
|
|
|
STATIC const mp_stream_p_t websocket_stream_p = {
|
2016-03-24 22:51:51 +00:00
|
|
|
.read = websocket_read,
|
2016-03-24 17:14:12 +00:00
|
|
|
.write = websocket_write,
|
2016-04-10 10:50:46 +01:00
|
|
|
.ioctl = websocket_ioctl,
|
2016-03-24 17:14:12 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
STATIC const mp_obj_type_t websocket_type = {
|
|
|
|
{ &mp_type_type },
|
|
|
|
.name = MP_QSTR_websocket,
|
|
|
|
.make_new = websocket_make_new,
|
2016-06-18 16:19:24 +01:00
|
|
|
.protocol = &websocket_stream_p,
|
2016-08-06 14:18:03 +01:00
|
|
|
.locals_dict = (void*)&websocket_locals_dict,
|
2016-03-24 17:14:12 +00:00
|
|
|
};
|
|
|
|
|
2016-08-06 14:18:03 +01:00
|
|
|
STATIC const mp_rom_map_elem_t websocket_module_globals_table[] = {
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_websocket) },
|
|
|
|
{ MP_ROM_QSTR(MP_QSTR_websocket), MP_ROM_PTR(&websocket_type) },
|
2016-03-24 17:14:12 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
STATIC MP_DEFINE_CONST_DICT(websocket_module_globals, websocket_module_globals_table);
|
|
|
|
|
|
|
|
const mp_obj_module_t mp_module_websocket = {
|
|
|
|
.base = { &mp_type_module },
|
|
|
|
.globals = (mp_obj_dict_t*)&websocket_module_globals,
|
|
|
|
};
|
|
|
|
|
|
|
|
#endif // MICROPY_PY_WEBSOCKET
|