ion: fix crash when alloc len is -1

If userspace passes a length between -4095 and -1 to allocate it
will pass the len != 0 check, but when len is page aligned it will
be 0.  Check len after page aligning.

Drop the warning as well, userspace shouldn't be able to trigger
a warning in the kernel.

Change-Id: I96c7142637638991f3a9af9be7cfbb50f79f3803
Signed-off-by: Colin Cross <ccross@android.com>

diff --git a/drivers/staging/android/ion/ion.c b/drivers/staging/android/ion/ion.c
index 0e379a3381051f7fcd951fae0867ccf2f347b042..a8d4735b462c55200494f4fb5cceb0b086b9167a 100644 (file)
--- a/drivers/staging/android/ion/ion.c
+++ b/drivers/staging/android/ion/ion.c
@@ -485,11 +485,11 @@ struct ion_handle *ion_alloc(struct ion_client *client, size_t len,
         * request of the caller allocate from it.  Repeat until allocate has
         * succeeded or all heaps have been tried
         */
-       if (WARN_ON(!len))
-               return ERR_PTR(-EINVAL);
-
        len = PAGE_ALIGN(len);
 
+       if (!len)
+               return ERR_PTR(-EINVAL);
+
        down_read(&dev->lock);
        plist_for_each_entry(heap, &dev->heaps, node) {
                /* if the caller didn't specify this heap id */