From da8208c3fb5bada3206d07fcbe087697adfed706 Mon Sep 17 00:00:00 2001 From: wanxiaoqing Date: Mon, 18 Dec 2023 18:10:55 +0800 Subject: [PATCH] =?UTF-8?q?wukong=E7=A8=B3=E5=AE=9A=E6=80=A7=E9=87=8D?= =?UTF-8?q?=E5=90=AF=E9=97=AE=E9=A2=98=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: wanxiaoqing --- include/linux/interval_tree_generic.h | 1 + lib/interval_tree.c | 1 + 2 files changed, 2 insertions(+) diff --git a/include/linux/interval_tree_generic.h b/include/linux/interval_tree_generic.h index aaa8a0767aa3..6e27d315e6cf 100644 --- a/include/linux/interval_tree_generic.h +++ b/include/linux/interval_tree_generic.h @@ -85,6 +85,7 @@ ITPREFIX ## _subtree_search(ITSTRUCT *node, ITTYPE start, ITTYPE last) \ * Loop invariant: start <= node->ITSUBTREE \ * (Cond2 is satisfied by one of the subtree nodes) \ */ \ + cond_resched(); \ if (node->ITRB.rb_left) { \ ITSTRUCT *left = rb_entry(node->ITRB.rb_left, \ ITSTRUCT, ITRB); \ diff --git a/lib/interval_tree.c b/lib/interval_tree.c index 593ce56ece50..77864645523e 100644 --- a/lib/interval_tree.c +++ b/lib/interval_tree.c @@ -3,6 +3,7 @@ #include #include #include +#include #define START(node) ((node)->start) #define LAST(node) ((node)->last) -- Gitee