17 lines
513 B
Diff
17 lines
513 B
Diff
|
diff --git a/vncviewer/zlib.c b/vncviewer/zlib.c
|
||
|
|
index 80c4eee..76998d8 100644
|
||
|
|
--- a/vncviewer/zlib.c
|
||
|
|
+++ b/vncviewer/zlib.c
|
||
|
|
@@ -55,6 +55,11 @@ HandleZlibBPP (int rx, int ry, int rw, int rh)
|
||
|
|
raw_buffer_size = (( rw * rh ) * ( BPP / 8 ));
|
||
|
|
raw_buffer = (char*) malloc( raw_buffer_size );
|
||
|
|
|
||
|
|
+ if ( raw_buffer == NULL ) {
|
||
|
|
+ fprintf(stderr,
|
||
|
|
+ "couldn't allocate raw_buffer in HandleZlibBPP");
|
||
|
|
+ return False;
|
||
|
|
+ }
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!ReadFromRFBServer((char *)&hdr, sz_rfbZlibHeader))
|