s390/gup: add missing TASK_SIZE check to get_user_pages_fast()

[karo-tx-linux.git] / lib / genalloc.c

diff --git a/lib/genalloc.c b/lib/genalloc.c
index f352cc42f4f8a58b8e4c8304de32cb31b4e29b67..716f94747c817998dcc1f908d4ef4e09344f91b3 100644 (file)
--- a/lib/genalloc.c
+++ b/lib/genalloc.c
@@ -176,7 +176,7 @@ int gen_pool_add_virt(struct gen_pool *pool, unsigned long virt, phys_addr_t phy
        struct gen_pool_chunk *chunk;
        int nbits = size >> pool->min_alloc_order;
        int nbytes = sizeof(struct gen_pool_chunk) +
-                               (nbits + BITS_PER_BYTE - 1) / BITS_PER_BYTE;
+                               BITS_TO_LONGS(nbits) * sizeof(long);
 
        chunk = kmalloc_node(nbytes, GFP_KERNEL | __GFP_ZERO, nid);
        if (unlikely(chunk == NULL))