This adds in the rlimit checking for the s390 module allocator.

This has not been tested.

Signed-off-by: Rick Edgecombe <rick.p.edgeco...@intel.com>
---
 arch/s390/kernel/module.c | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/arch/s390/kernel/module.c b/arch/s390/kernel/module.c
index d298d3cb46d0..6c2356a72b63 100644
--- a/arch/s390/kernel/module.c
+++ b/arch/s390/kernel/module.c
@@ -32,12 +32,22 @@
 
 void *module_alloc(unsigned long size)
 {
+       void *p;
+
        if (PAGE_ALIGN(size) > MODULES_LEN)
                return NULL;
-       return __vmalloc_node_range(size, 1, MODULES_VADDR, MODULES_END,
+
+       if (check_inc_mod_rlimit(size))
+               return NULL;
+
+       p = __vmalloc_node_range(size, 1, MODULES_VADDR, MODULES_END,
                                    GFP_KERNEL, PAGE_KERNEL_EXEC,
                                    0, NUMA_NO_NODE,
                                    __builtin_return_address(0));
+
+       update_mod_rlimit(p, size);
+
+       return p;
 }
 
 void module_arch_freeing_init(struct module *mod)
-- 
2.17.1

Reply via email to