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.sun.com/software/products/lustre/docs/GPLv2.pdf
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
libcfs_kvzalloc(size_t size,gfp_t flags)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
libcfs_kvzalloc_cpt(struct cfs_cpt_table * cptab,int cpt,size_t size,gfp_t flags)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);
60