//         with code, comments and ideas from :-
 //         Richard Purdie <richard@openedhand.com>
 
+#include <linux/cleanup.h>
 #include <linux/module.h>
 #include <linux/moduleparam.h>
 #include <linux/init.h>
        struct soc_bytes *params = (void *)kcontrol->private_value;
        int ret, len;
        unsigned int val, mask;
-       void *data;
 
        if (!component->regmap || !params->num_regs)
                return -EINVAL;
 
        len = params->num_regs * component->val_bytes;
 
-       data = kmemdup(ucontrol->value.bytes.data, len, GFP_KERNEL | GFP_DMA);
+       void *data __free(kfree) = kmemdup(ucontrol->value.bytes.data, len,
+                                          GFP_KERNEL | GFP_DMA);
        if (!data)
                return -ENOMEM;
 
        if (params->mask) {
                ret = regmap_read(component->regmap, params->base, &val);
                if (ret != 0)
-                       goto out;
+                       return ret;
 
                val &= params->mask;
 
                        ret = regmap_parse_val(component->regmap,
                                                        &mask, &mask);
                        if (ret != 0)
-                               goto out;
+                               return ret;
 
                        ((u16 *)data)[0] &= mask;
 
                        ret = regmap_parse_val(component->regmap,
                                                        &val, &val);
                        if (ret != 0)
-                               goto out;
+                               return ret;
 
                        ((u16 *)data)[0] |= val;
                        break;
                        ret = regmap_parse_val(component->regmap,
                                                        &mask, &mask);
                        if (ret != 0)
-                               goto out;
+                               return ret;
 
                        ((u32 *)data)[0] &= mask;
 
                        ret = regmap_parse_val(component->regmap,
                                                        &val, &val);
                        if (ret != 0)
-                               goto out;
+                               return ret;
 
                        ((u32 *)data)[0] |= val;
                        break;
                default:
-                       ret = -EINVAL;
-                       goto out;
+                       return -EINVAL;
                }
        }
 
-       ret = regmap_raw_write(component->regmap, params->base,
-                              data, len);
-
-out:
-       kfree(data);
-
-       return ret;
+       return regmap_raw_write(component->regmap, params->base, data, len);
 }
 EXPORT_SYMBOL_GPL(snd_soc_bytes_put);