1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/base/firmware_class.c b/drivers/base/firmware_class.c
index e97e911ebf7a..472318205236 100644
--- a/drivers/base/firmware_class.c
+++ b/drivers/base/firmware_class.c
@@ -211,18 +211,20 @@ static int
 fw_realloc_buffer(struct firmware_priv *fw_priv, int min_size)
 {
         u8 *new_data;
+        int new_size = fw_priv->alloc_size;
 
         if (min_size <= fw_priv->alloc_size)
                 return 0;
 
-        new_data = vmalloc(fw_priv->alloc_size + PAGE_SIZE);
+        new_size = ALIGN(min_size, PAGE_SIZE);
+        new_data = vmalloc(new_size);
         if (!new_data) {
                 printk(KERN_ERR "%s: unable to alloc buffer\n", __FUNCTION__);
                 /* Make sure that we don't keep incomplete data */
                 fw_load_abort(fw_priv);
                 return -ENOMEM;
         }
-        fw_priv->alloc_size += PAGE_SIZE;
+        fw_priv->alloc_size = new_size;
         if (fw_priv->fw->data) {
                 memcpy(new_data, fw_priv->fw->data, fw_priv->fw->size);
                 vfree(fw_priv->fw->data);