drivers/staging/lustre/lnet/libcfs/linux/linux-mem.c

   1 /*
   2  * This program is free software; you can redistribute it and/or modify
   3  * it under the terms of the GNU General Public License version 2 only,
   4  * as published by the Free Software Foundation.
   5  *
   6  * This program is distributed in the hope that it will be useful, but
   7  * WITHOUT ANY WARRANTY; without even the implied warranty of
   8  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   9  * General Public License version 2 for more details (a copy is included
  10  * in the LICENSE file that accompanied this code).
  11  *
  12  * You should have received a copy of the GNU General Public License
  13  * version 2 along with this program; If not, see
  14  * http://www.gnu.org/licenses/gpl-2.0.html
  15  *
  16  */
  17 /*
  18  * This file creates a memory allocation primitive for Lustre, that
  19  * allows to fallback to vmalloc allocations should regular kernel allocations
  20  * fail due to size or system memory fragmentation.
  21  *
  22  * Author: Oleg Drokin <green@linuxhacker.ru>
  23  *
  24  */
  25 /*
  26  * This file is part of Lustre, http://www.lustre.org/
  27  * Lustre is a trademark of Seagate Technology.
  28  */
  29 #include <linux/slab.h>
  30 #include <linux/vmalloc.h>
  31
  32 #include "../../../include/linux/libcfs/libcfs.h"
  33
  34 void *libcfs_kvzalloc(size_t size, gfp_t flags)
  35 {
  36         void *ret;
  37
  38         ret = kzalloc(size, flags | __GFP_NOWARN);
  39         if (!ret)
  40                 ret = __vmalloc(size, flags | __GFP_ZERO, PAGE_KERNEL);
  41         return ret;
  42 }
  43 EXPORT_SYMBOL(libcfs_kvzalloc);
  44
  45 void *libcfs_kvzalloc_cpt(struct cfs_cpt_table *cptab, int cpt, size_t size,
  46                           gfp_t flags)
  47 {
  48         void *ret;
  49
  50         ret = kzalloc_node(size, flags | __GFP_NOWARN,
  51                            cfs_cpt_spread_node(cptab, cpt));
  52         if (!ret) {
  53                 WARN_ON(!(flags & (__GFP_FS | __GFP_HIGH)));
  54                 ret = vmalloc_node(size, cfs_cpt_spread_node(cptab, cpt));
  55         }
  56
  57         return ret;
  58 }
  59 EXPORT_SYMBOL(libcfs_kvzalloc_cpt);