jp9000/obs-studio
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

win-dshow: Check return values for memory allocation functions

Browse Source 
Since some of these run inside the virtual cam module, we should be a
good guest and not crash the host process if we run out of memory.
This commit is contained in:
Richard Stanway 2021-03-23 03:07:22 +01:00
parent 670156db8b
commit 45643adb03
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
plugins/win-dshow/shared-memory-queue.c
View File

@ -90,6 +90,10 @@ video_queue_t *video_queue_create(uint32_t cx, uint32_t cy, uint64_t interval)
vq.header = (struct queue_header *)MapViewOfFile(
vq.handle, FILE_MAP_ALL_ACCESS, 0, 0, 0);
if (!vq.header) {
CloseHandle(vq.handle);
return NULL;
}
memcpy(vq.header, &header, sizeof(header));
for (size_t i = 0; i < 3; i++) {
@ -98,6 +102,10 @@ video_queue_t *video_queue_create(uint32_t cx, uint32_t cy, uint64_t interval)
vq.frame[i] = ((uint8_t *)vq.header) + off + FRAME_HEADER_SIZE;
}
pvq = malloc(sizeof(vq));
if (!pvq) {
CloseHandle(vq.handle);
return NULL;
}
memcpy(pvq, &vq, sizeof(vq));
return pvq;
}
@ -113,8 +121,16 @@ video_queue_t *video_queue_open()
vq.header = (struct queue_header *)MapViewOfFile(
vq.handle, FILE_MAP_READ, 0, 0, 0);
if (!vq.header) {
CloseHandle(vq.handle);
return NULL;
}
struct video_queue *pvq = malloc(sizeof(vq));
if (!pvq) {
CloseHandle(vq.handle);
return NULL;
}
memcpy(pvq, &vq, sizeof(vq));
return pvq;
}