diff --git a/libvncserver/rfbserver.c b/libvncserver/rfbserver.c index d2e2c5c..52f9fad 100644 --- a/libvncserver/rfbserver.c +++ b/libvncserver/rfbserver.c @@ -2404,6 +2404,12 @@ rfbProcessClientNormalMessage(rfbClientPtr cl) str = (char *)malloc(msg.cct.length); + if (str == NULL) { + rfbLogPerror("rfbProcessClientNormalMessage: not enough memory"); + rfbCloseClient(cl); + return; + } + if ((n = rfbReadExact(cl, str, msg.cct.length)) <= 0) { if (n != 0) rfbLogPerror("rfbProcessClientNormalMessage: read");