Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20;
From:   "Liam R. Howlett" <Liam.Howlett@oracle.com>
To:     Andrew Morton <akpm@linux-foundation.org>
Cc:     linux-mm@kvack.org, linux-kernel@vger.kernel.org,
        Rick Edgecombe <rick.p.edgecombe@intel.com>,
        "Liam R. Howlett" <Liam.Howlett@oracle.com>
Subject: [PATCH 1/3] maple_tree: Make maple state reusable after mas_empty_area_rev()
Date:   Fri, 14 Apr 2023 10:57:26 -0400
Message-Id: <20230414145728.4067069-1-Liam.Howlett@oracle.com>
Content-Transfer-Encoding: 8bit
Content-Type: text/plain
MIME-Version: 1.0
X-MS-Exchange-AntiSpam-MessageData-ChunkCount: 1
X-MS-Exchange-AntiSpam-MessageData-0: =?us-ascii?Q?/0jrIv6+c4O7TXVAjADSs7LjBSNzsJ0t2l9jzZR+uyKWGZX/6ZhRS1YPD5W0?=
 =?us-ascii?Q?J71EZQrVJ0Pi4+BSNRfB8DgN8wQkV6fEMl5QfQcIBoV8aG0VFEGBx8Qpq1Rg?=
 =?us-ascii?Q?SDftVO7T9I05SDtU1ZHuIbbhEvR9qv1l+NRgh6nOmhLFbsGA9jDUeCrWVqz5?=
 =?us-ascii?Q?QGTMSBbQIIhaDUHPYC5Oz5+gFI/Ka91szsC7IV+MnR7HSX2dBOqklCi4qhqf?=
 =?us-ascii?Q?nMBzEu6gtDHeXaq0L1XIb1OhAK5ZZA7xkNTSlIAboxP3z0o21k+dzOF6N4Z0?=
 =?us-ascii?Q?aa/7BOXQfF4pJhlk1FhDqPm7/b3DxLC2E2WmBpdGSGGmbLZdHJDkcT7Ihebg?=
 =?us-ascii?Q?sNPdOma+D+TxOjjw/6KOf9DY+AIKOdHZ2lr/o9GSpYhjL1rHZk7YLHMPEHDV?=
 =?us-ascii?Q?OtwTL6/soeumuz0dc5JjIQ4oqOTbsnTNzNWiBwDhGmI7fAy7Dn2t/2qBgLFw?=
 =?us-ascii?Q?os2GzkLYFMFg+QP3L2m8mFxbVSmVn56a7yYIaoO7SBpbPX022ibAxQgkQ2Rb?=
 =?us-ascii?Q?MQ9DaIqPk3A9vc1aBTRdDWr2+vaQo5NE9Q9ynjt2y6bUZGmnsX9uWepmos85?=
 =?us-ascii?Q?D6fy6iZezr4O1ClH+1vmUI8tk6dUJR2M5CCYVZZK67cxQcSuoB2Jnl/lfuGi?=
 =?us-ascii?Q?/me9GW1HJBA+PCTor9RHb/nJz7R09OIdLztkcHrO0F8GJK/KjSeNQOZo+KbK?=
 =?us-ascii?Q?HpbFJeQ05sdq27ulf+zxNWRGrBa8vuwMKifzoNZgZ4W69sR1wHu0acS4Ox4a?=
 =?us-ascii?Q?dr/EDHWFU1p0UniV+3iNN4Q0crSlKeXzTM9BIKFKMIgV7lPpQ5HorJUFMruw?=
 =?us-ascii?Q?0DfOcw41PKhOJOZeyrTXLyq4wuzf5OMNDusBk7et6SqgsK0taLPSTsx7zIbj?=
 =?us-ascii?Q?WE84DrEDuoQROAGzY8NvqsQrMj2k6yc28Sz6nMc4goGqnPnjRUEEa0xMlzxl?=
 =?us-ascii?Q?ui+QkTwj5SF1ZPdK6jEOJMUynZvLyUIQG9A0xV+iprTfWze+fNrPhGlOjK1F?=
 =?us-ascii?Q?8hAObtcJ4JYfp0TCDvTOCY68sU/cAIQpPePxZ9PtYOJtJvv5/FAVzqjR9bSo?=
 =?us-ascii?Q?A9eTSybiMihTJ7Z+H6D7DcLqqe8jfqT5XWr1HLFwBwtS+OCxbqJAZkHXlCQ1?=
 =?us-ascii?Q?y2NLrdLyhyF83ClqsP3c0sMdIiNAAf14FviwTylX5tHER5fMshW8RS0Lnm6m?=
 =?us-ascii?Q?ivwOxBlofvf1PWLDDRkgyVY/KhYNIMjSmPr8loHbMu279xUsk326l3UQapIX?=
 =?us-ascii?Q?XTFkqPRxC6N5LaBiowt6aIoR+kPVaQrlnm4miVPC2V7d+XC9JmlkUPUZqOGP?=
 =?us-ascii?Q?fx+W/XRDSzo2mIze5KM/E8FXbuvtJn98E3VNP7Kh3atjyp7mrE+k3VlfUXIF?=
 =?us-ascii?Q?3lk9DJDOz0c5TZCFpOuammf1acfd8gFDHlW8B1QeG9muoflvUYiuTrLg09M+?=
 =?us-ascii?Q?NBDhp8W62RK0juozjJV4WG27QwWxmOrnJhTvJ/yDq28i+B1v6e6SRgrCLEt2?=
 =?us-ascii?Q?8xQ4rLMHgFXhAzwpuj2dwkonrofgumfG8DP3JJkX9iCAaiyem81cLotdce6f?=
 =?us-ascii?Q?clGfn9SWYKqJHkG2hZpaNmizhGDnHectjmSQFdZtFHqgvh2WdoE5CHfZYZ4r?=
 =?us-ascii?Q?EA=3D=3D?=
X-MS-Exchange-AntiSpam-ExternalHop-MessageData-ChunkCount: 1
X-MS-Exchange-AntiSpam-ExternalHop-MessageData-0: LITFnAL26wLrygZp3enNx49OcXLbYz+sOmuIIdBXXm/PL6lAdlbrLiT34YOsRJkQBzm+b0mkka9jI7Nw0CMjo5E/BA9/GpJ+cr2R4wPSxUSOUIKpyuzI3JtTIh+ePEuPMiSb6l31Utcko0H1bfDpTrZKzcA/foiMPi6zXdkUe4bWE1vXrKRaBvoxBkySSQM2JT8fihuNdc40nbO1Glkhzj9AwRoXVEnHdO/tRPkt1NZ9aK3OHRS5gajr10BEBi9T3ZAx04e5mqP5Yd+aX/mV/u3S2Q3bOdo+wmna+3ojFzNM3aybvkpZ+iX9/fq5vQ8zY6ArjXafHRvj6b+12V2aOwhxO9W1JbiZoSQYbtJRJJxk+zqSPmwHohY0bgfMACcx+JFfSQplhAJH7LFC0jmKxw7iqyN2cq6JLs82Y1Wzk41wrosVzaiosZma9DgO0urclUHLXnUIkFYAf4oLIVj+8tfQXy2yQXZJyT+cT8CJ2zvzi9e4GvRsTKy3RJCO1sB8zHR2xCnjsDVcfoboM3bToWNlToQp8M+BI4ZCkbB2hkQJAvssIunitG9YPFhc9V2A5BNpQDKF9af1GRbQej4RlaZUs5ipPE7pV6Uw2mRZU0imkakFxem1UFeOv4iCIBau92DS71PcANPCSer4MIhx2KMp+Hf+DeiYEwZoFJJxdvA6QTguShOceh7Ue/oMxhwWpPmI/LvlMHX4gTnLfLVD7p6AIN9OAX3C6iKSRhSN9i29/LEKZ/FmxmipjPT9jjpE3Li2hwN9GCCYxV9urDTT+uttdr6ecmU73SPXaZdHoZ01898LRBbVOSfv/cvAAUtMqdmAiZD4N7jEQlrmZj4RrY6N9wIjNjaXNHSPcYRBw3eqNpidPAp5+BH/Jq/qCVD1
X-OriginatorOrg: oracle.com
X-MS-Exchange-CrossTenant-Network-Message-Id: e9cc4981-1268-4037-54fd-08db3cf89e0f
X-MS-Exchange-CrossTenant-AuthSource: SN6PR10MB3022.namprd10.prod.outlook.com
X-MS-Exchange-CrossTenant-AuthAs: Internal
X-MS-Exchange-CrossTenant-OriginalArrivalTime: 14 Apr 2023 14:57:50.5296
 (UTC)
X-MS-Exchange-CrossTenant-FromEntityHeader: Hosted
X-MS-Exchange-CrossTenant-Id: 4e2c6054-71cb-48f1-bd6c-3a9705aca71b
X-MS-Exchange-CrossTenant-MailboxType: HOSTED
X-MS-Exchange-CrossTenant-UserPrincipalName: /83t4Yx5kz1l4AlssjddbY6v4Ib/xrBNrPreoq7Ms7O8IKo6aEmYg1Yikg+l7xmXPZvHHpdivtJ7DWWZ4ZPAgQ==
X-MS-Exchange-Transport-CrossTenantHeadersStamped: PH0PR10MB4582
X-Proofpoint-Virus-Version: vendor=baseguard
 engine=ICAP:2.0.254,Aquarius:18.0.942,Hydra:6.0.573,FMLib:17.11.170.22
 definitions=2023-04-14_08,2023-04-14_01,2023-02-09_01
X-Proofpoint-Spam-Details: rule=notspam policy=default score=0 mlxlogscore=999 mlxscore=0
 malwarescore=0 suspectscore=0 bulkscore=0 spamscore=0 phishscore=0
 adultscore=0 classifier=spam adjust=0 reason=mlx scancount=1
 engine=8.12.0-2303200000 definitions=main-2304140131
X-Proofpoint-GUID: 092BZYmO4tmymAdF9uC68hBdlzX8OGAv
X-Proofpoint-ORIG-GUID: 092BZYmO4tmymAdF9uC68hBdlzX8OGAv
X-Spam-Status: No, score=-2.8 required=5.0 tests=BAYES_00,DKIM_SIGNED,
        DKIM_VALID,DKIM_VALID_AU,DKIM_VALID_EF,RCVD_IN_DNSWL_LOW,
        RCVD_IN_MSPIKE_H2,SPF_HELO_NONE,SPF_NONE,T_SCC_BODY_TEXT_LINE
        autolearn=ham autolearn_force=no version=3.4.6
X-Spam-Checker-Version: SpamAssassin 3.4.6 (2021-04-09) on
        lindbergh.monkeyblade.net
Precedence: bulk
List-ID: <linux-kernel.vger.kernel.org>
X-Mailing-List: linux-kernel@vger.kernel.org

Stop using maple state min/max for the range by passing through pointers
for those values.  This will allow the maple state to be reused without
resetting.

Also add some logic to fail out early on searching with invalid
arguments.

Fixes: 54a611b60590 ("Maple Tree: add new data structure")
Signed-off-by: Liam R. Howlett <Liam.Howlett@oracle.com>
---
 lib/maple_tree.c | 27 +++++++++++++--------------
 1 file changed, 13 insertions(+), 14 deletions(-)

diff --git a/lib/maple_tree.c b/lib/maple_tree.c
index 4df6a0ce1c1b..ed350aa293b2 100644
--- a/lib/maple_tree.c
+++ b/lib/maple_tree.c
@@ -4938,7 +4938,8 @@ static inline void *mas_prev_entry(struct ma_state *mas, unsigned long min)
  * Return: True if found in a leaf, false otherwise.
  *
  */
-static bool mas_rev_awalk(struct ma_state *mas, unsigned long size)
+static bool mas_rev_awalk(struct ma_state *mas, unsigned long size,
+		unsigned long *gap_min, unsigned long *gap_max)
 {
 	enum maple_type type = mte_node_type(mas->node);
 	struct maple_node *node = mas_mn(mas);
@@ -5003,8 +5004,8 @@ static bool mas_rev_awalk(struct ma_state *mas, unsigned long size)
 
 	if (unlikely(ma_is_leaf(type))) {
 		mas->offset = offset;
-		mas->min = min;
-		mas->max = min + gap - 1;
+		*gap_min = min;
+		*gap_max = min + gap - 1;
 		return true;
 	}
 
@@ -5280,6 +5281,9 @@ int mas_empty_area(struct ma_state *mas, unsigned long min,
 	unsigned long *pivots;
 	enum maple_type mt;
 
+	if (min >= max)
+		return -EINVAL;
+
 	if (mas_is_start(mas))
 		mas_start(mas);
 	else if (mas->offset >= 2)
@@ -5334,6 +5338,9 @@ int mas_empty_area_rev(struct ma_state *mas, unsigned long min,
 {
 	struct maple_enode *last = mas->node;
 
+	if (min >= max)
+		return -EINVAL;
+
 	if (mas_is_start(mas)) {
 		mas_start(mas);
 		mas->offset = mas_data_end(mas);
@@ -5353,7 +5360,7 @@ int mas_empty_area_rev(struct ma_state *mas, unsigned long min,
 	mas->index = min;
 	mas->last = max;
 
-	while (!mas_rev_awalk(mas, size)) {
+	while (!mas_rev_awalk(mas, size, &min, &max)) {
 		if (last == mas->node) {
 			if (!mas_rewind_node(mas))
 				return -EBUSY;
@@ -5368,17 +5375,9 @@ int mas_empty_area_rev(struct ma_state *mas, unsigned long min,
 	if (unlikely(mas->offset == MAPLE_NODE_SLOTS))
 		return -EBUSY;
 
-	/*
-	 * mas_rev_awalk() has set mas->min and mas->max to the gap values.  If
-	 * the maximum is outside the window we are searching, then use the last
-	 * location in the search.
-	 * mas->max and mas->min is the range of the gap.
-	 * mas->index and mas->last are currently set to the search range.
-	 */
-
 	/* Trim the upper limit to the max. */
-	if (mas->max <= mas->last)
-		mas->last = mas->max;
+	if (max <= mas->last)
+		mas->last = max;
 
 	mas->index = mas->last - size + 1;
 	return 0;
-- 
2.39.2