Throw error on null bitmap to prevent crash

lewisl9029 · lewisl9029 · commit ae6c27ddf12a · 2021-03-06T23:18:54.000-08:00
diff --git a/src/main.cc b/src/main.cc
@@ -758,6 +758,10 @@ Napi::Object _captureScreen(const Napi::CallbackInfo &info)
 	}
 
 	MMBitmapRef bitmap = copyMMBitmapFromDisplayInRect(MMRectMake(x, y, w, h));
+	
+	if (bitmap == NULL) {
+		throw Napi::Error::New(env, "Error: Failed to capture screen");
+	}
 
 	uint32_t bufferSize = bitmap->bytewidth * bitmap->height;
 	Napi::Buffer<char> buffer = Napi::Buffer<char>::New(env, (char *)bitmap->imageBuffer, bufferSize, finalizer);

Original file line number	Diff line number	Diff line change
`@@ -758,6 +758,10 @@ Napi::Object _captureScreen(const Napi::CallbackInfo &info)`
`758`	`758`	`}`
`759`	`759`
`760`	`760`	`MMBitmapRef bitmap = copyMMBitmapFromDisplayInRect(MMRectMake(x, y, w, h));`
	`761`	`+`
	`762`	`+ if (bitmap == NULL) {`
	`763`	`+ throw Napi::Error::New(env, "Error: Failed to capture screen");`
	`764`	`+ }`
`761`	`765`
`762`	`766`	`uint32_t bufferSize = bitmap->bytewidth * bitmap->height;`
`763`	`767`	`Napi::Buffer<char> buffer = Napi::Buffer<char>::New(env, (char *)bitmap->imageBuffer, bufferSize, finalizer);`