libcfs/include/libcfs/linux/linux-mem.h

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.gnu.org/licenses/gpl-2.0.html
  19  *
  20  * GPL HEADER END
  21  */
  22 /*
  23  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * Copyright (c) 2011, 2014, Intel Corporation.
  27  */
  28 /*
  29  * This file is part of Lustre, http://www.lustre.org/
  30  * Lustre is a trademark of Sun Microsystems, Inc.
  31  *
  32  * libcfs/include/libcfs/linux/linux-mem.h
  33  *
  34  * Basic library routines.
  35  */
  36
  37 #ifndef __LIBCFS_LINUX_CFS_MEM_H__
  38 #define __LIBCFS_LINUX_CFS_MEM_H__
  39
  40 #include <linux/mm.h>
  41 #include <linux/vmalloc.h>
  42 #include <linux/pagemap.h>
  43 #include <linux/slab.h>
  44 #ifdef HAVE_MM_INLINE
  45 # include <linux/mm_inline.h>
  46 #endif
  47 #include <linux/sched.h>
  48 #ifdef HAVE_SCHED_HEADERS
  49 #include <linux/sched/mm.h>
  50 #endif
  51
  52 #ifdef HAVE_TOTALRAM_PAGES_AS_FUNC
  53  #ifndef cfs_totalram_pages
  54   #define cfs_totalram_pages() totalram_pages()
  55  #endif
  56 #else
  57  #ifndef cfs_totalram_pages
  58   #define cfs_totalram_pages() totalram_pages
  59  #endif
  60 #endif
  61
  62 #ifndef HAVE_MEMALLOC_RECLAIM
  63 static inline unsigned int memalloc_noreclaim_save(void)
  64 {
  65         unsigned int flags = current->flags & PF_MEMALLOC;
  66
  67         current->flags |= PF_MEMALLOC;
  68         return flags;
  69 }
  70
  71 static inline void memalloc_noreclaim_restore(unsigned int flags)
  72 {
  73         current->flags = (current->flags & ~PF_MEMALLOC) | flags;
  74 }
  75 #endif /* !HAVE_MEMALLOC_RECLAIM */
  76
  77 /*
  78  * Shrinker
  79  */
  80 # define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask)  \
  81                        struct shrinker *shrinker, \
  82                        struct shrink_control *sc
  83 # define shrink_param(sc, var) ((sc)->var)
  84
  85 #ifdef HAVE_SHRINKER_COUNT
  86 struct shrinker_var {
  87         unsigned long (*count)(struct shrinker *,
  88                                struct shrink_control *sc);
  89         unsigned long (*scan)(struct shrinker *,
  90                               struct shrink_control *sc);
  91 };
  92 # define DEF_SHRINKER_VAR(name, shrink, count_obj, scan_obj) \
  93             struct shrinker_var name = { .count = count_obj, .scan = scan_obj }
  94 #else
  95 struct shrinker_var {
  96         int (*shrink)(SHRINKER_ARGS(sc, nr_to_scan, gfp_mask));
  97 };
  98 # define DEF_SHRINKER_VAR(name, shrinker, count, scan) \
  99             struct shrinker_var name = { .shrink = shrinker }
 100 # define SHRINK_STOP (~0UL)
 101 #endif
 102
 103 static inline
 104 struct shrinker *set_shrinker(int seek, struct shrinker_var *var)
 105 {
 106         struct shrinker *s;
 107
 108         s = kzalloc(sizeof(*s), GFP_KERNEL);
 109         if (s == NULL)
 110                 return (NULL);
 111
 112 #ifdef HAVE_SHRINKER_COUNT
 113         s->count_objects = var->count;
 114         s->scan_objects = var->scan;
 115 #else
 116         s->shrink = var->shrink;
 117 #endif
 118         s->seeks = seek;
 119
 120         register_shrinker(s);
 121
 122         return s;
 123 }
 124
 125 static inline
 126 void remove_shrinker(struct shrinker *shrinker)
 127 {
 128         if (shrinker == NULL)
 129                 return;
 130
 131         unregister_shrinker(shrinker);
 132         kfree(shrinker);
 133 }
 134
 135 #endif /* __LINUX_CFS_MEM_H__ */