diff --git a/src/sftp.c b/src/sftp.c index bd4a4b5b..09bc630c 100644 --- a/src/sftp.c +++ b/src/sftp.c @@ -33,6 +33,7 @@ #include #include #include +#include #ifndef _WIN32 #include @@ -353,7 +354,7 @@ sftp_packet sftp_packet_read(sftp_session sftp) { buffer_get_u8(packet->payload, &packet->type); size = ntohl(size); - if (size == 0) { + if (size == 0 || size > UINT_MAX) { return packet; } size--;