On Thu, Nov 13, 2014 at 05:09:27PM -0500, Tejun Heo wrote:
> +int ptrset_add(void *ptr, struct ptrset *set, gfp_t gfp)
> +{
> +     struct ptrset_elem *elem;
> +     struct rb_node *parent, **slot;
> +
> +     elem = kmalloc(sizeof(*elem), gfp);
> +     if (!elem && !in_interrupt())   /* see ptrset_preload() */
> +             elem = this_cpu_xchg(ptrset_preload_elem, NULL);
> +     if (!elem)
> +             return -ENOMEM;
> +     elem->ptr = ptr;
> +
> +     if (ptrset_find_slot(elem->ptr, set, &slot, &parent)) {
> +             kfree(elem);
> +             return -EEXIST;
> +     }

Maybe allocation *after* ptrset_find_slot() will be better?
This will avoid extra kmalloc/kfree if such ptr already exist (and also
will avoid ENOMEM for such cases).
--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to