python: add buffer length check in sock_recvfrom_into()

2014-02-25 18:51:03 +01:00 · 2014-02-25 18:51:03 +01:00 · c4fec93ab9
commit c4fec93ab9
parent ec37e34f07
1 changed files with 6 additions and 0 deletions
--- a/sys/src/cmd/python/Modules/socketmodule.c
+++ b/sys/src/cmd/python/Modules/socketmodule.c
@ -2547,6 +2547,12 @@ sock_recvfrom_into(PySocketSockObject *s, PyObject *args, PyObject* kwds)
            recvlen = buflen;
 	}

+	if (buflen < recvlen) {
+		PyErr_SetString(PyExc_ValueError,
+				"buffer too small for requested bytes");
+		return NULL;
+	}
+
 	readlen = sock_recvfrom_guts(s, buf, recvlen, flags, &addr);
 	if (readlen < 0) {
 		/* Return an error */