From bf2470cec69ab00f7213aa136f5a2d486c9da17a Mon Sep 17 00:00:00 2001
From: Christian Beier <dontmind@freeshell.org>
Date: Thu, 17 Mar 2011 13:11:59 +0100
Subject: Check rfbGetScreen() return value everywhere.

This fixes a segfault when a server is invoked with the
'-help' commandline argument.
---
 VisualNaCro/nacro.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'VisualNaCro')

diff --git a/VisualNaCro/nacro.c b/VisualNaCro/nacro.c
index 2fc9e36..9c492b0 100644
--- a/VisualNaCro/nacro.c
+++ b/VisualNaCro/nacro.c
@@ -121,6 +121,8 @@ static rfbBool malloc_frame_buffer(rfbClient* cl)
 		res->client->frameBuffer=malloc(w*4*h);
 		
 		res->server=rfbGetScreen(NULL,NULL,w,h,8,3,4);
+		if(!res->server)
+                  return FALSE;
 		res->server->screenData=res;
 		res->server->port=res->listen_port;
 		res->server->frameBuffer=res->client->frameBuffer;
-- 
cgit v1.2.1