Check that display was opened successfully.

AndrewFromMelbourne · Jan 21, 2016 · 808fb02 · 808fb02
1 parent caa4068
commit 808fb02
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/raspi2png.c b/raspi2png.c
@@ -229,6 +229,12 @@ main(
     DISPMANX_DISPLAY_HANDLE_T displayHandle
         = vc_dispmanx_display_open(displayNumber);
 
+    if (displayHandle == 0)
+    {
+        fprintf(stderr, "%s: unable to open display\n", program);
+        exit(EXIT_FAILURE);
+    }
+
     DISPMANX_MODEINFO_T modeInfo;
     result = vc_dispmanx_display_get_info(displayHandle, &modeInfo);