#include <linux/mm.h>
#include <linux/vmalloc.h>
+#include <wrapper/mm.h>
#include <wrapper/vmalloc.h> /* for wrapper_vmalloc_sync_all() */
#include <wrapper/ringbuffer/config.h>
#include <wrapper/ringbuffer/backend.h>
unsigned long i;
num_pages = size >> PAGE_SHIFT;
+
+ /*
+ * Verify that there is enough free pages available on the system for
+ * the current allocation request.
+ * wrapper_check_enough_free_pages uses si_mem_available() if available
+ * and returns if there should be enough free pages based on the
+ * current estimate.
+ */
+ if (!wrapper_check_enough_free_pages(num_pages))
+ goto not_enough_pages;
+
+ /*
+ * Set the current user thread as the first target of the OOM killer.
+ * If the estimate received by si_mem_available() was off, and we do
+ * end up running out of memory because of this buffer allocation, we
+ * want to kill the offending app first.
+ */
+ wrapper_set_current_oom_origin();
+
num_pages_per_subbuf = num_pages >> get_count_order(num_subbuf);
subbuf_size = chanb->subbuf_size;
num_subbuf_alloc = num_subbuf;
* will not fault.
*/
wrapper_vmalloc_sync_all();
+ wrapper_clear_current_oom_origin();
vfree(pages);
return 0;
array_error:
vfree(pages);
pages_error:
+ wrapper_clear_current_oom_origin();
+not_enough_pages:
return -ENOMEM;
}