iio: buffer: Save a few cycles in 'iio_scan_mask_set()'

Use 'bitmap_alloc()' instead of 'bitmap_zalloc()' because the bitmap
is fully overridden by a 'bitmap_copy()' call just after its allocation.

While at it, fix the style of a NULL check.

Signed-off-by: Christophe JAILLET <christophe.jaillet@wanadoo.fr>
Link: https://lore.kernel.org/r/89d8a766eb971eda1ee362444a8711037bdb208c.1626261211.git.christophe.jaillet@wanadoo.fr
Signed-off-by: Jonathan Cameron <Jonathan.Cameron@huawei.com>

diff --git a/drivers/iio/industrialio-buffer.c b/drivers/iio/industrialio-buffer.c
index fdd623407b96949a7a7bacfce67b3935a3adddef..6d4776a7f002e4b1efa4435799a78a558bd83f19 100644 (file)
--- a/drivers/iio/industrialio-buffer.c
+++ b/drivers/iio/industrialio-buffer.c
@@ -354,8 +354,8 @@ static int iio_scan_mask_set(struct iio_dev *indio_dev,
        const unsigned long *mask;
        unsigned long *trialmask;
 
-       trialmask = bitmap_zalloc(indio_dev->masklength, GFP_KERNEL);
-       if (trialmask == NULL)
+       trialmask = bitmap_alloc(indio_dev->masklength, GFP_KERNEL);
+       if (!trialmask)
                return -ENOMEM;
        if (!indio_dev->masklength) {
                WARN(1, "Trying to set scanmask prior to registering buffer\n");