Skip to content

Commit ab802ba

Browse files
committed
Added safety checks.
1 parent 7cee00e commit ab802ba

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/screengrab.c

+4-1
Original file line numberDiff line numberDiff line change
@@ -31,14 +31,17 @@ MMBitmapRef copyMMBitmapFromDisplayInRect(MMRect rect)
3131
rect.size.width,
3232
rect.size.height));
3333

34+
if (!image) { return NULL; }
35+
3436
CFDataRef imageData = CGDataProviderCopyData(CGImageGetDataProvider(image));
3537

38+
if (!imageData) { return NULL; }
39+
3640
bufferSize = CFDataGetLength(imageData);
3741
buffer = malloc(bufferSize);
3842

3943
CFDataGetBytes(imageData, CFRangeMake(0,bufferSize), buffer);
4044

41-
/* Use image size because it can differ from specified size */
4245
bitmap = createMMBitmap(buffer,
4346
CGImageGetWidth(image),
4447
CGImageGetHeight(image),

0 commit comments

Comments
 (0)