Added max size check to recv_into

2020-10-15 18:46:42 -05:00 · 2020-10-15 18:46:42 -05:00 · fa75231ae3
commit fa75231ae3
parent f6f89565d8
1 changed files with 3 additions and 0 deletions
--- a/shared-bindings/socketpool/Socket.c
+++ b/shared-bindings/socketpool/Socket.c
@ -260,6 +260,9 @@ STATIC mp_obj_t socketpool_socket_recv_into(size_t n_args, const mp_obj_t *args)
    mp_int_t len = bufinfo.len;
    if (n_args == 3) {
        mp_int_t given_len = mp_obj_get_int(args[2]);
+        if (given_len > len) {
+            mp_raise_ValueError(translate("buffer too small for requested bytes"));
+        }
        if (given_len > 0 && given_len < len) {
            len = given_len;
        }