|
@@ -298,8 +298,11 @@ static void *qp_alloc_queue(u64 size, u32 flags)
|
|
|
size_t pas_size;
|
|
|
size_t vas_size;
|
|
|
size_t queue_size = sizeof(*queue) + sizeof(*queue->kernel_if);
|
|
|
- const u64 num_pages = DIV_ROUND_UP(size, PAGE_SIZE) + 1;
|
|
|
+ u64 num_pages;
|
|
|
|
|
|
+ if (size > SIZE_MAX - PAGE_SIZE)
|
|
|
+ return NULL;
|
|
|
+ num_pages = DIV_ROUND_UP(size, PAGE_SIZE) + 1;
|
|
|
if (num_pages >
|
|
|
(SIZE_MAX - queue_size) /
|
|
|
(sizeof(*queue->kernel_if->u.g.pas) +
|
|
@@ -624,9 +627,12 @@ static struct vmci_queue *qp_host_alloc_queue(u64 size)
|
|
|
{
|
|
|
struct vmci_queue *queue;
|
|
|
size_t queue_page_size;
|
|
|
- const u64 num_pages = DIV_ROUND_UP(size, PAGE_SIZE) + 1;
|
|
|
+ u64 num_pages;
|
|
|
const size_t queue_size = sizeof(*queue) + sizeof(*(queue->kernel_if));
|
|
|
|
|
|
+ if (size > SIZE_MAX - PAGE_SIZE)
|
|
|
+ return NULL;
|
|
|
+ num_pages = DIV_ROUND_UP(size, PAGE_SIZE) + 1;
|
|
|
if (num_pages > (SIZE_MAX - queue_size) /
|
|
|
sizeof(*queue->kernel_if->u.h.page))
|
|
|
return NULL;
|