Quelle  darray.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0

#include <linux/log2.h>
#include <linux/slab.h>
#include <linux/vmalloc.h>
#include "darray.h"

int __bch2_darray_resize_noprof(darray_char *d, size_t element_size, size_t new_size, gfp_t gfp)
{
 if (new_size > d->size) {
  new_size = roundup_pow_of_two(new_size);

  /*
 * This is a workaround: kvmalloc() doesn't support > INT_MAX
 * allocations, but vmalloc() does.
 * The limit needs to be lifted from kvmalloc, and when it does
 * we'll go back to just using that.
 */
  size_t bytes;
  if (unlikely(check_mul_overflow(new_size, element_size, &bytes)))
   return -ENOMEM;

  void *data = likely(bytes < INT_MAX)
   ? kvmalloc_noprof(bytes, gfp)
   : vmalloc_noprof(bytes);
  if (!data)
   return -ENOMEM;

  if (d->size)
   memcpy(data, d->data, d->size * element_size);
  if (d->data != d->preallocated)
   kvfree(d->data);
  d->data = data;
  d->size = new_size;
 }

 return 0;
}