lnet/libcfs/linux/linux-mem.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * Copyright (C) 2001, 2002 Cluster File Systems, Inc.
   5  *
   6  *   This file is part of Lustre, http://www.lustre.org.
   7  *
   8  *   Lustre is free software; you can redistribute it and/or
   9  *   modify it under the terms of version 2 of the GNU General Public
  10  *   License as published by the Free Software Foundation.
  11  *
  12  *   Lustre is distributed in the hope that it will be useful,
  13  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *   GNU General Public License for more details.
  16  *
  17  *   You should have received a copy of the GNU General Public License
  18  *   along with Lustre; if not, write to the Free Software
  19  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20  */
  21 #define DEBUG_SUBSYSTEM S_PORTALS
  22
  23 #include <linux/mm.h>
  24 #include <linux/vmalloc.h>
  25 #include <linux/slab.h>
  26 #include <linux/highmem.h>
  27 #include <libcfs/libcfs.h>
  28
  29 void *
  30 cfs_alloc(size_t nr_bytes, u_int32_t flags)
  31 {
  32         void *ptr = NULL;
  33         unsigned int mflags = 0;
  34
  35 #if (LINUX_VERSION_CODE < KERNEL_VERSION(2,5,0))
  36         if (flags & CFS_ALLOC_ATOMIC)
  37                 mflags |= __GFP_HIGH;
  38         else if (flags & CFS_ALLOC_WAIT)
  39                 mflags |= __GFP_WAIT;
  40         else
  41                 mflags |= (__GFP_HIGH | __GFP_WAIT);
  42
  43         if (flags & CFS_ALLOC_FS)
  44                 mflags |= __GFP_FS;
  45         if (flags & CFS_ALLOC_IO)
  46                 mflags |= __GFP_IO | __GFP_HIGHIO;
  47 #else
  48         if (flags & CFS_ALLOC_ATOMIC)
  49                 mflags |= __GFP_HIGH;
  50         else
  51                 mflags |= __GFP_WAIT;
  52         if (flags & CFS_ALLOC_FS)
  53                 mflags |= __GFP_FS;
  54         if (flags & CFS_ALLOC_IO)
  55                 mflags |= __GFP_IO;
  56 #endif
  57
  58         ptr = kmalloc(nr_bytes, mflags);
  59         if (ptr != NULL && (flags & CFS_ALLOC_ZERO))
  60                 memset(ptr, 0, nr_bytes);
  61         return ptr;
  62 }
  63
  64 void
  65 cfs_free(void *addr)
  66 {
  67         kfree(addr);
  68 }
  69
  70 void *
  71 cfs_alloc_large(size_t nr_bytes)
  72 {
  73         return vmalloc(nr_bytes);
  74 }
  75
  76 void
  77 cfs_free_large(void *addr)
  78 {
  79         vfree(addr);
  80 }
  81
  82 cfs_page_t *
  83 cfs_alloc_pages(unsigned int flags, unsigned int order)
  84 {
  85         unsigned int mflags = 0;
  86
  87 #if (LINUX_VERSION_CODE < KERNEL_VERSION(2,5,0))
  88         if (flags & CFS_ALLOC_ATOMIC)
  89                 mflags |= __GFP_HIGH;
  90         else if (flags & CFS_ALLOC_WAIT)
  91                 mflags |= __GFP_WAIT;
  92         else
  93                 mflags |= (__GFP_HIGH | __GFP_WAIT);
  94
  95         if (flags & CFS_ALLOC_FS)
  96                 mflags |= __GFP_FS;
  97         if (flags & CFS_ALLOC_IO)
  98                 mflags |= __GFP_IO | __GFP_HIGHIO;
  99         if (flags & CFS_ALLOC_HIGH)
 100                 mflags |=  __GFP_HIGHMEM;
 101 #else
 102         if (flags & CFS_ALLOC_ATOMIC)
 103                 mflags |= __GFP_HIGH;
 104         else
 105                 mflags |= __GFP_WAIT;
 106         if (flags & CFS_ALLOC_FS)
 107                 mflags |= __GFP_FS;
 108         if (flags & CFS_ALLOC_IO)
 109                 mflags |= __GFP_IO;
 110         if (flags & CFS_ALLOC_HIGH)
 111                 mflags |=  __GFP_HIGHMEM;
 112 #endif
 113
 114         return alloc_pages(mflags, order);
 115 }
 116
 117 cfs_mem_cache_t *
 118 cfs_mem_cache_create (const char *name, size_t size, size_t offset,
 119                       unsigned long flags, void (*ctor)(void*, kmem_cache_t *, unsigned long),
 120                       void (*dtor)(void*, cfs_mem_cache_t *, unsigned long))
 121 {
 122         return kmem_cache_create(name, size, offset, flags, ctor, dtor);
 123 }
 124
 125 int
 126 cfs_mem_cache_destroy (cfs_mem_cache_t * cachep)
 127 {
 128         return kmem_cache_destroy(cachep);
 129 }
 130
 131 void *
 132 cfs_mem_cache_alloc(cfs_mem_cache_t *cachep, int flags)
 133 {
 134         unsigned int mflags = 0;
 135 #if (LINUX_VERSION_CODE < KERNEL_VERSION(2,5,0))
 136         if (flags & CFS_SLAB_ATOMIC)
 137                 mflags |= __GFP_HIGH;
 138         else if (flags & CFS_ALLOC_WAIT)
 139                 mflags |= __GFP_WAIT;
 140         else
 141                 mflags |= (__GFP_HIGH | __GFP_WAIT);
 142
 143         if (flags & CFS_SLAB_FS)
 144                 mflags |= __GFP_FS;
 145         if (flags & CFS_SLAB_IO)
 146                 mflags |= __GFP_IO | __GFP_HIGHIO;
 147 #else
 148         if (flags & CFS_SLAB_ATOMIC)
 149                 mflags |= __GFP_HIGH;
 150         else
 151                 mflags |= __GFP_WAIT;
 152         if (flags & CFS_SLAB_FS)
 153                 mflags |= __GFP_FS;
 154         if (flags & CFS_SLAB_IO)
 155                 mflags |= __GFP_IO;
 156 #endif
 157
 158         return kmem_cache_alloc(cachep, mflags);
 159 }
 160
 161 void
 162 cfs_mem_cache_free(cfs_mem_cache_t *cachep, void *objp)
 163 {
 164         return kmem_cache_free(cachep, objp);
 165 }
 166
 167 EXPORT_SYMBOL(cfs_alloc);
 168 EXPORT_SYMBOL(cfs_free);
 169 EXPORT_SYMBOL(cfs_alloc_large);
 170 EXPORT_SYMBOL(cfs_free_large);
 171 EXPORT_SYMBOL(cfs_alloc_pages);
 172 EXPORT_SYMBOL(cfs_mem_cache_create);
 173 EXPORT_SYMBOL(cfs_mem_cache_destroy);
 174 EXPORT_SYMBOL(cfs_mem_cache_alloc);
 175 EXPORT_SYMBOL(cfs_mem_cache_free);