commit e0dfc08bc2f509de9fda0371b46988247f711a12
author Kent Overstreet <kent.overstreet@gmail.com> Tue Jun 11 21:03:23 2019 -0400
committer Kent Overstreet <kent.overstreet@linux.dev> Sun Oct 22 17:08:23 2023 -0400
tree 1777f499171ef6b440c6ec0b549d092e9d5681cc
parent 3e669816904d45d48ef2183a2ad675c24d9c941a

bcachefs: use memalloc_nofs_save() for vmalloc allocation

Signed-off-by: Kent Overstreet <kent.overstreet@linux.dev>

fs/bcachefs/btree_cache.c

diff --git a/fs/bcachefs/btree_cache.c b/fs/bcachefs/btree_cache.c
index d80ba1d..ea775d9 100644
--- a/fs/bcachefs/btree_cache.c
+++ b/fs/bcachefs/btree_cache.c
@@ -9,6 +9,7 @@
 #include "trace.h"
 
 #include <linux/prefetch.h>
+#include <linux/sched/mm.h>
 
 const char * const bch2_btree_ids[] = {
 #define x(kwd, val, name) name,
@@ -509,7 +510,9 @@ struct btree *bch2_btree_node_mem_alloc(struct bch_fs *c)
 	struct btree_cache *bc = &c->btree_cache;
 	struct btree *b;
 	u64 start_time = local_clock();
+	unsigned flags;
 
+	flags = memalloc_nofs_save();
 	mutex_lock(&bc->lock);
 
 	/*
@@ -547,6 +550,7 @@ struct btree *bch2_btree_node_mem_alloc(struct bch_fs *c)
 
 	list_del_init(&b->list);
 	mutex_unlock(&bc->lock);
+	memalloc_nofs_restore(flags);
 out:
 	b->flags		= 0;
 	b->written		= 0;