diff options
| author | Paul Sokolovsky <pfalcon@users.sourceforge.net> | 2016-04-08 16:05:48 +0300 |
|---|---|---|
| committer | Paul Sokolovsky <pfalcon@users.sourceforge.net> | 2016-04-08 16:05:48 +0300 |
| commit | 05ba2433f69c4c64bff78f41fc7c77c87e42ac34 (patch) | |
| tree | a04ca34d1121868643d8eea88a9e5ebd8831ca7d /extmod/modwebsocket.c | |
| parent | 5e919b7ef8612046e85c17d91f12067a1e0c893f (diff) | |
| download | micropython-05ba2433f69c4c64bff78f41fc7c77c87e42ac34.tar.gz micropython-05ba2433f69c4c64bff78f41fc7c77c87e42ac34.zip | |
extmod/modwebsocket: Properly check number of args to constructor.
Diffstat (limited to 'extmod/modwebsocket.c')
| -rw-r--r-- | extmod/modwebsocket.c | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/extmod/modwebsocket.c b/extmod/modwebsocket.c index 34ed46a6ef..00ec5ee980 100644 --- a/extmod/modwebsocket.c +++ b/extmod/modwebsocket.c @@ -31,6 +31,7 @@ #include "py/nlr.h" #include "py/obj.h" +#include "py/runtime.h" #include "py/stream.h" #if MICROPY_PY_WEBSOCKET @@ -50,7 +51,7 @@ typedef struct _mp_obj_websocket_t { } mp_obj_websocket_t; 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) { - assert(n_args == 1); + mp_arg_check_num(n_args, n_kw, 1, 1, false); mp_obj_websocket_t *o = m_new_obj(mp_obj_websocket_t); o->base.type = type; o->sock = args[0]; |