LU-13783 libcfs: support __vmalloc with only 2 args.

[fs/lustre-release.git] / libcfs / include / libcfs / linux / linux-mem.h

diff --git a/libcfs/include/libcfs/linux/linux-mem.h b/libcfs/include/libcfs/linux/linux-mem.h
index 6aca65b..69d027f 100644 (file)
--- a/libcfs/include/libcfs/linux/linux-mem.h
+++ b/libcfs/include/libcfs/linux/linux-mem.h
@@ -44,81 +44,97 @@
 #ifdef HAVE_MM_INLINE
 # include <linux/mm_inline.h>
 #endif
-
-/*
- * Shrinker
- */
-#ifdef HAVE_SHRINK_CONTROL
-# define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask)  \
-                       struct shrinker *shrinker, \
-                       struct shrink_control *sc
-# define shrink_param(sc, var) ((sc)->var)
-#else
-struct shrink_control {
-       gfp_t gfp_mask;
-       unsigned long nr_to_scan;
-};
-# ifdef HAVE_SHRINKER_WANT_SHRINK_PTR
-#  define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask)  \
-                        struct shrinker *shrinker, \
-                        int nr_to_scan, gfp_t gfp_mask
-# else
-#  define SHRINKER_ARGS(sc, nr_to_scan, gfp_mask)  \
-                        int nr_to_scan, gfp_t gfp_mask
-# endif
-       /* avoid conflict with spl mm_compat.h */
-# define HAVE_SHRINK_CONTROL_STRUCT 1
-# define shrink_param(sc, var) (var)
+#include <linux/sched.h>
+#ifdef HAVE_SCHED_HEADERS
+#include <linux/sched/mm.h>
 #endif
 
-#ifdef HAVE_SHRINKER_COUNT
-struct shrinker_var {
-       unsigned long (*count)(struct shrinker *,
-                              struct shrink_control *sc);
-       unsigned long (*scan)(struct shrinker *,
-                             struct shrink_control *sc);
-};
-# define DEF_SHRINKER_VAR(name, shrink, count_obj, scan_obj) \
-           struct shrinker_var name = { .count = count_obj, .scan = scan_obj }
+#ifdef HAVE_TOTALRAM_PAGES_AS_FUNC
+ #ifndef cfs_totalram_pages
+  #define cfs_totalram_pages() totalram_pages()
+ #endif
 #else
-struct shrinker_var {
-       int (*shrink)(SHRINKER_ARGS(sc, nr_to_scan, gfp_mask));
-};
-# define DEF_SHRINKER_VAR(name, shrinker, count, scan) \
-           struct shrinker_var name = { .shrink = shrinker }
-# define SHRINK_STOP (~0UL)
+ #ifndef cfs_totalram_pages
+  #define cfs_totalram_pages() totalram_pages
+ #endif
 #endif
 
-static inline
-struct shrinker *set_shrinker(int seek, struct shrinker_var *var)
+#ifndef HAVE_MEMALLOC_RECLAIM
+static inline unsigned int memalloc_noreclaim_save(void)
 {
-        struct shrinker *s;
+       unsigned int flags = current->flags & PF_MEMALLOC;
 
-       s = kzalloc(sizeof(*s), GFP_KERNEL);
-        if (s == NULL)
-                return (NULL);
+       current->flags |= PF_MEMALLOC;
+       return flags;
+}
 
-#ifdef HAVE_SHRINKER_COUNT
-       s->count_objects = var->count;
-       s->scan_objects = var->scan;
-#else
-       s->shrink = var->shrink;
+static inline void memalloc_noreclaim_restore(unsigned int flags)
+{
+       current->flags = (current->flags & ~PF_MEMALLOC) | flags;
+}
+#endif /* !HAVE_MEMALLOC_RECLAIM */
+
+#ifndef HAVE_BITMAP_ALLOC
+static inline unsigned long *bitmap_alloc(unsigned int nbits, gfp_t flags)
+{
+       return kmalloc_array(BITS_TO_LONGS(nbits), sizeof(unsigned long),
+                            flags);
+}
+
+static inline unsigned long *bitmap_zalloc(unsigned int nbits, gfp_t flags)
+{
+       return bitmap_alloc(nbits, flags | __GFP_ZERO);
+}
+
+static inline void bitmap_free(const unsigned long *bitmap)
+{
+       kfree(bitmap);
+}
+#endif /* !HAVE_BITMAP_ALLOC */
+
+/*
+ * Shrinker
+ */
+#ifndef SHRINK_STOP
+# define SHRINK_STOP (~0UL)
 #endif
-        s->seeks = seek;
 
-        register_shrinker(s);
+#ifndef HAVE_MMAP_LOCK
+static inline void mmap_write_lock(struct mm_struct *mm)
+{
+       down_write(&mm->mmap_sem);
+}
 
-        return s;
+static inline bool mmap_write_trylock(struct mm_struct *mm)
+{
+       return down_write_trylock(&mm->mmap_sem) != 0;
 }
 
-static inline
-void remove_shrinker(struct shrinker *shrinker)
+static inline void mmap_write_unlock(struct mm_struct *mm)
 {
-        if (shrinker == NULL)
-                return;
+       up_write(&mm->mmap_sem);
+}
 
-        unregister_shrinker(shrinker);
-        kfree(shrinker);
+static inline void mmap_read_lock(struct mm_struct *mm)
+{
+       down_read(&mm->mmap_sem);
 }
 
+static inline bool mmap_read_trylock(struct mm_struct *mm)
+{
+       return down_read_trylock(&mm->mmap_sem) != 0;
+}
+
+static inline void mmap_read_unlock(struct mm_struct *mm)
+{
+       up_read(&mm->mmap_sem);
+}
+#endif
+
+#ifdef HAVE_VMALLOC_2ARGS
+#define __ll_vmalloc(size, flags) __vmalloc(size, flags)
+#else
+#define __ll_vmalloc(size, flags) __vmalloc(size, flags, PAGE_KERNEL)
+#endif
+
 #endif /* __LINUX_CFS_MEM_H__ */