1 /*
2  * Copyright (c) 2013-2015, Mellanox Technologies. All rights reserved.
3  *
4  * This software is available to you under a choice of one of two
5  * licenses.  You may choose to be licensed under the terms of the GNU
6  * General Public License (GPL) Version 2, available from the file
7  * COPYING in the main directory of this source tree, or the
8  * OpenIB.org BSD license below:
9  *
10  *     Redistribution and use in source and binary forms, with or
11  *     without modification, are permitted provided that the following
12  *     conditions are met:
13  *
14  *      - Redistributions of source code must retain the above
15  *        copyright notice, this list of conditions and the following
16  *        disclaimer.
17  *
18  *      - Redistributions in binary form must reproduce the above
19  *        copyright notice, this list of conditions and the following
20  *        disclaimer in the documentation and/or other materials
21  *        provided with the distribution.
22  *
23  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
24  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
25  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
26  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
27  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
28  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
29  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
30  * SOFTWARE.
31  */
32 
33 #include <linux/highmem.h>
34 #include <linux/kernel.h>
35 #include <linux/delay.h>
36 #include <linux/mlx5/driver.h>
37 #include <linux/xarray.h>
38 #include "mlx5_core.h"
39 #include "lib/eq.h"
40 #include "lib/tout.h"
41 
42 enum {
43 	MLX5_PAGES_CANT_GIVE	= 0,
44 	MLX5_PAGES_GIVE		= 1,
45 	MLX5_PAGES_TAKE		= 2
46 };
47 
48 struct mlx5_pages_req {
49 	struct mlx5_core_dev *dev;
50 	u16	func_id;
51 	u8	ec_function;
52 	s32	npages;
53 	struct work_struct work;
54 	u8	release_all;
55 };
56 
57 struct fw_page {
58 	struct rb_node		rb_node;
59 	u64			addr;
60 	struct page	       *page;
61 	u32			function;
62 	unsigned long		bitmask;
63 	struct list_head	list;
64 	unsigned int free_count;
65 };
66 
67 enum {
68 	MLX5_MAX_RECLAIM_TIME_MILI	= 5000,
69 	MLX5_NUM_4K_IN_PAGE		= PAGE_SIZE / MLX5_ADAPTER_PAGE_SIZE,
70 };
71 
72 static u32 get_function(u16 func_id, bool ec_function)
73 {
74 	return (u32)func_id | (ec_function << 16);
75 }
76 
77 static u16 func_id_to_type(struct mlx5_core_dev *dev, u16 func_id, bool ec_function)
78 {
79 	if (!func_id)
80 		return mlx5_core_is_ecpf(dev) && !ec_function ? MLX5_HOST_PF : MLX5_PF;
81 
82 	if (func_id <= max(mlx5_core_max_vfs(dev), mlx5_core_max_ec_vfs(dev))) {
83 		if (ec_function)
84 			return MLX5_EC_VF;
85 		else
86 			return MLX5_VF;
87 	}
88 	return MLX5_SF;
89 }
90 
91 static u32 mlx5_get_ec_function(u32 function)
92 {
93 	return function >> 16;
94 }
95 
96 static u32 mlx5_get_func_id(u32 function)
97 {
98 	return function & 0xffff;
99 }
100 
101 static struct rb_root *page_root_per_function(struct mlx5_core_dev *dev, u32 function)
102 {
103 	struct rb_root *root;
104 	int err;
105 
106 	root = xa_load(&dev->priv.page_root_xa, function);
107 	if (root)
108 		return root;
109 
110 	root = kzalloc(sizeof(*root), GFP_KERNEL);
111 	if (!root)
112 		return ERR_PTR(-ENOMEM);
113 
114 	err = xa_insert(&dev->priv.page_root_xa, function, root, GFP_KERNEL);
115 	if (err) {
116 		kfree(root);
117 		return ERR_PTR(err);
118 	}
119 
120 	*root = RB_ROOT;
121 
122 	return root;
123 }
124 
125 static int insert_page(struct mlx5_core_dev *dev, u64 addr, struct page *page, u32 function)
126 {
127 	struct rb_node *parent = NULL;
128 	struct rb_root *root;
129 	struct rb_node **new;
130 	struct fw_page *nfp;
131 	struct fw_page *tfp;
132 	int i;
133 
134 	root = page_root_per_function(dev, function);
135 	if (IS_ERR(root))
136 		return PTR_ERR(root);
137 
138 	new = &root->rb_node;
139 
140 	while (*new) {
141 		parent = *new;
142 		tfp = rb_entry(parent, struct fw_page, rb_node);
143 		if (tfp->addr < addr)
144 			new = &parent->rb_left;
145 		else if (tfp->addr > addr)
146 			new = &parent->rb_right;
147 		else
148 			return -EEXIST;
149 	}
150 
151 	nfp = kzalloc(sizeof(*nfp), GFP_KERNEL);
152 	if (!nfp)
153 		return -ENOMEM;
154 
155 	nfp->addr = addr;
156 	nfp->page = page;
157 	nfp->function = function;
158 	nfp->free_count = MLX5_NUM_4K_IN_PAGE;
159 	for (i = 0; i < MLX5_NUM_4K_IN_PAGE; i++)
160 		set_bit(i, &nfp->bitmask);
161 
162 	rb_link_node(&nfp->rb_node, parent, new);
163 	rb_insert_color(&nfp->rb_node, root);
164 	list_add(&nfp->list, &dev->priv.free_list);
165 
166 	return 0;
167 }
168 
169 static struct fw_page *find_fw_page(struct mlx5_core_dev *dev, u64 addr,
170 				    u32 function)
171 {
172 	struct fw_page *result = NULL;
173 	struct rb_root *root;
174 	struct rb_node *tmp;
175 	struct fw_page *tfp;
176 
177 	root = xa_load(&dev->priv.page_root_xa, function);
178 	if (WARN_ON_ONCE(!root))
179 		return NULL;
180 
181 	tmp = root->rb_node;
182 
183 	while (tmp) {
184 		tfp = rb_entry(tmp, struct fw_page, rb_node);
185 		if (tfp->addr < addr) {
186 			tmp = tmp->rb_left;
187 		} else if (tfp->addr > addr) {
188 			tmp = tmp->rb_right;
189 		} else {
190 			result = tfp;
191 			break;
192 		}
193 	}
194 
195 	return result;
196 }
197 
198 static int mlx5_cmd_query_pages(struct mlx5_core_dev *dev, u16 *func_id,
199 				s32 *npages, int boot)
200 {
201 	u32 out[MLX5_ST_SZ_DW(query_pages_out)] = {};
202 	u32 in[MLX5_ST_SZ_DW(query_pages_in)] = {};
203 	int err;
204 
205 	MLX5_SET(query_pages_in, in, opcode, MLX5_CMD_OP_QUERY_PAGES);
206 	MLX5_SET(query_pages_in, in, op_mod, boot ?
207 		 MLX5_QUERY_PAGES_IN_OP_MOD_BOOT_PAGES :
208 		 MLX5_QUERY_PAGES_IN_OP_MOD_INIT_PAGES);
209 	MLX5_SET(query_pages_in, in, embedded_cpu_function, mlx5_core_is_ecpf(dev));
210 
211 	err = mlx5_cmd_exec_inout(dev, query_pages, in, out);
212 	if (err)
213 		return err;
214 
215 	*npages = MLX5_GET(query_pages_out, out, num_pages);
216 	*func_id = MLX5_GET(query_pages_out, out, function_id);
217 
218 	return err;
219 }
220 
221 static int alloc_4k(struct mlx5_core_dev *dev, u64 *addr, u32 function)
222 {
223 	struct fw_page *fp = NULL;
224 	struct fw_page *iter;
225 	unsigned n;
226 
227 	list_for_each_entry(iter, &dev->priv.free_list, list) {
228 		if (iter->function != function)
229 			continue;
230 		fp = iter;
231 	}
232 
233 	if (list_empty(&dev->priv.free_list) || !fp)
234 		return -ENOMEM;
235 
236 	n = find_first_bit(&fp->bitmask, 8 * sizeof(fp->bitmask));
237 	if (n >= MLX5_NUM_4K_IN_PAGE) {
238 		mlx5_core_warn(dev, "alloc 4k bug: fw page = 0x%llx, n = %u, bitmask: %lu, max num of 4K pages: %d\n",
239 			       fp->addr, n, fp->bitmask,  MLX5_NUM_4K_IN_PAGE);
240 		return -ENOENT;
241 	}
242 	clear_bit(n, &fp->bitmask);
243 	fp->free_count--;
244 	if (!fp->free_count)
245 		list_del(&fp->list);
246 
247 	*addr = fp->addr + n * MLX5_ADAPTER_PAGE_SIZE;
248 
249 	return 0;
250 }
251 
252 #define MLX5_U64_4K_PAGE_MASK ((~(u64)0U) << PAGE_SHIFT)
253 
254 static void free_fwp(struct mlx5_core_dev *dev, struct fw_page *fwp,
255 		     bool in_free_list)
256 {
257 	struct rb_root *root;
258 
259 	root = xa_load(&dev->priv.page_root_xa, fwp->function);
260 	if (WARN_ON_ONCE(!root))
261 		return;
262 
263 	rb_erase(&fwp->rb_node, root);
264 	if (in_free_list)
265 		list_del(&fwp->list);
266 	dma_unmap_page(mlx5_core_dma_dev(dev), fwp->addr & MLX5_U64_4K_PAGE_MASK,
267 		       PAGE_SIZE, DMA_BIDIRECTIONAL);
268 	__free_page(fwp->page);
269 	kfree(fwp);
270 }
271 
272 static void free_4k(struct mlx5_core_dev *dev, u64 addr, u32 function)
273 {
274 	struct fw_page *fwp;
275 	int n;
276 
277 	fwp = find_fw_page(dev, addr & MLX5_U64_4K_PAGE_MASK, function);
278 	if (!fwp) {
279 		mlx5_core_warn_rl(dev, "page not found\n");
280 		return;
281 	}
282 	n = (addr & ~MLX5_U64_4K_PAGE_MASK) >> MLX5_ADAPTER_PAGE_SHIFT;
283 	fwp->free_count++;
284 	set_bit(n, &fwp->bitmask);
285 	if (fwp->free_count == MLX5_NUM_4K_IN_PAGE)
286 		free_fwp(dev, fwp, fwp->free_count != 1);
287 	else if (fwp->free_count == 1)
288 		list_add(&fwp->list, &dev->priv.free_list);
289 }
290 
291 static int alloc_system_page(struct mlx5_core_dev *dev, u32 function)
292 {
293 	struct device *device = mlx5_core_dma_dev(dev);
294 	int nid = dev_to_node(device);
295 	struct page *page;
296 	u64 zero_addr = 1;
297 	u64 addr;
298 	int err;
299 
300 	page = alloc_pages_node(nid, GFP_HIGHUSER, 0);
301 	if (!page) {
302 		mlx5_core_warn(dev, "failed to allocate page\n");
303 		return -ENOMEM;
304 	}
305 map:
306 	addr = dma_map_page(device, page, 0, PAGE_SIZE, DMA_BIDIRECTIONAL);
307 	if (dma_mapping_error(device, addr)) {
308 		mlx5_core_warn(dev, "failed dma mapping page\n");
309 		err = -ENOMEM;
310 		goto err_mapping;
311 	}
312 
313 	/* Firmware doesn't support page with physical address 0 */
314 	if (addr == 0) {
315 		zero_addr = addr;
316 		goto map;
317 	}
318 
319 	err = insert_page(dev, addr, page, function);
320 	if (err) {
321 		mlx5_core_err(dev, "failed to track allocated page\n");
322 		dma_unmap_page(device, addr, PAGE_SIZE, DMA_BIDIRECTIONAL);
323 	}
324 
325 err_mapping:
326 	if (err)
327 		__free_page(page);
328 
329 	if (zero_addr == 0)
330 		dma_unmap_page(device, zero_addr, PAGE_SIZE,
331 			       DMA_BIDIRECTIONAL);
332 
333 	return err;
334 }
335 
336 static void page_notify_fail(struct mlx5_core_dev *dev, u16 func_id,
337 			     bool ec_function)
338 {
339 	u32 in[MLX5_ST_SZ_DW(manage_pages_in)] = {};
340 	int err;
341 
342 	MLX5_SET(manage_pages_in, in, opcode, MLX5_CMD_OP_MANAGE_PAGES);
343 	MLX5_SET(manage_pages_in, in, op_mod, MLX5_PAGES_CANT_GIVE);
344 	MLX5_SET(manage_pages_in, in, function_id, func_id);
345 	MLX5_SET(manage_pages_in, in, embedded_cpu_function, ec_function);
346 
347 	err = mlx5_cmd_exec_in(dev, manage_pages, in);
348 	if (err)
349 		mlx5_core_warn(dev, "page notify failed func_id(%d) err(%d)\n",
350 			       func_id, err);
351 }
352 
353 static int give_pages(struct mlx5_core_dev *dev, u16 func_id, int npages,
354 		      int event, bool ec_function)
355 {
356 	u32 function = get_function(func_id, ec_function);
357 	u32 out[MLX5_ST_SZ_DW(manage_pages_out)] = {0};
358 	int inlen = MLX5_ST_SZ_BYTES(manage_pages_in);
359 	int notify_fail = event;
360 	u16 func_type;
361 	u64 addr;
362 	int err;
363 	u32 *in;
364 	int i;
365 
366 	inlen += npages * MLX5_FLD_SZ_BYTES(manage_pages_in, pas[0]);
367 	in = kvzalloc(inlen, GFP_KERNEL);
368 	if (!in) {
369 		err = -ENOMEM;
370 		mlx5_core_warn(dev, "vzalloc failed %d\n", inlen);
371 		goto out_free;
372 	}
373 
374 	for (i = 0; i < npages; i++) {
375 retry:
376 		err = alloc_4k(dev, &addr, function);
377 		if (err) {
378 			if (err == -ENOMEM)
379 				err = alloc_system_page(dev, function);
380 			if (err) {
381 				dev->priv.fw_pages_alloc_failed += (npages - i);
382 				goto out_4k;
383 			}
384 
385 			goto retry;
386 		}
387 		MLX5_ARRAY_SET64(manage_pages_in, in, pas, i, addr);
388 	}
389 
390 	MLX5_SET(manage_pages_in, in, opcode, MLX5_CMD_OP_MANAGE_PAGES);
391 	MLX5_SET(manage_pages_in, in, op_mod, MLX5_PAGES_GIVE);
392 	MLX5_SET(manage_pages_in, in, function_id, func_id);
393 	MLX5_SET(manage_pages_in, in, input_num_entries, npages);
394 	MLX5_SET(manage_pages_in, in, embedded_cpu_function, ec_function);
395 
396 	err = mlx5_cmd_do(dev, in, inlen, out, sizeof(out));
397 	if (err == -EREMOTEIO) {
398 		notify_fail = 0;
399 		/* if triggered by FW and failed by FW ignore */
400 		if (event) {
401 			err = 0;
402 			goto out_dropped;
403 		}
404 	}
405 	err = mlx5_cmd_check(dev, err, in, out);
406 	if (err) {
407 		mlx5_core_warn(dev, "func_id 0x%x, npages %d, err %d\n",
408 			       func_id, npages, err);
409 		goto out_dropped;
410 	}
411 
412 	func_type = func_id_to_type(dev, func_id, ec_function);
413 	dev->priv.page_counters[func_type] += npages;
414 	dev->priv.fw_pages += npages;
415 
416 	mlx5_core_dbg(dev, "npages %d, ec_function %d, func_id 0x%x, err %d\n",
417 		      npages, ec_function, func_id, err);
418 
419 	kvfree(in);
420 	return 0;
421 
422 out_dropped:
423 	dev->priv.give_pages_dropped += npages;
424 out_4k:
425 	for (i--; i >= 0; i--)
426 		free_4k(dev, MLX5_GET64(manage_pages_in, in, pas[i]), function);
427 out_free:
428 	kvfree(in);
429 	if (notify_fail)
430 		page_notify_fail(dev, func_id, ec_function);
431 	return err;
432 }
433 
434 static void release_all_pages(struct mlx5_core_dev *dev, u16 func_id,
435 			      bool ec_function)
436 {
437 	u32 function = get_function(func_id, ec_function);
438 	struct rb_root *root;
439 	struct rb_node *p;
440 	int npages = 0;
441 	u16 func_type;
442 
443 	root = xa_load(&dev->priv.page_root_xa, function);
444 	if (WARN_ON_ONCE(!root))
445 		return;
446 
447 	p = rb_first(root);
448 	while (p) {
449 		struct fw_page *fwp = rb_entry(p, struct fw_page, rb_node);
450 
451 		p = rb_next(p);
452 		npages += (MLX5_NUM_4K_IN_PAGE - fwp->free_count);
453 		free_fwp(dev, fwp, fwp->free_count);
454 	}
455 
456 	func_type = func_id_to_type(dev, func_id, ec_function);
457 	dev->priv.page_counters[func_type] -= npages;
458 	dev->priv.fw_pages -= npages;
459 
460 	mlx5_core_dbg(dev, "npages %d, ec_function %d, func_id 0x%x\n",
461 		      npages, ec_function, func_id);
462 }
463 
464 static u32 fwp_fill_manage_pages_out(struct fw_page *fwp, u32 *out, u32 index,
465 				     u32 npages)
466 {
467 	u32 pages_set = 0;
468 	unsigned int n;
469 
470 	for_each_clear_bit(n, &fwp->bitmask, MLX5_NUM_4K_IN_PAGE) {
471 		MLX5_ARRAY_SET64(manage_pages_out, out, pas, index + pages_set,
472 				 fwp->addr + (n * MLX5_ADAPTER_PAGE_SIZE));
473 		pages_set++;
474 
475 		if (!--npages)
476 			break;
477 	}
478 
479 	return pages_set;
480 }
481 
482 static int reclaim_pages_cmd(struct mlx5_core_dev *dev,
483 			     u32 *in, int in_size, u32 *out, int out_size)
484 {
485 	struct rb_root *root;
486 	struct fw_page *fwp;
487 	struct rb_node *p;
488 	bool ec_function;
489 	u32 func_id;
490 	u32 npages;
491 	u32 i = 0;
492 
493 	if (!mlx5_cmd_is_down(dev))
494 		return mlx5_cmd_do(dev, in, in_size, out, out_size);
495 
496 	/* No hard feelings, we want our pages back! */
497 	npages = MLX5_GET(manage_pages_in, in, input_num_entries);
498 	func_id = MLX5_GET(manage_pages_in, in, function_id);
499 	ec_function = MLX5_GET(manage_pages_in, in, embedded_cpu_function);
500 
501 	root = xa_load(&dev->priv.page_root_xa, get_function(func_id, ec_function));
502 	if (WARN_ON_ONCE(!root))
503 		return -EEXIST;
504 
505 	p = rb_first(root);
506 	while (p && i < npages) {
507 		fwp = rb_entry(p, struct fw_page, rb_node);
508 		p = rb_next(p);
509 
510 		i += fwp_fill_manage_pages_out(fwp, out, i, npages - i);
511 	}
512 
513 	MLX5_SET(manage_pages_out, out, output_num_entries, i);
514 	return 0;
515 }
516 
517 static int reclaim_pages(struct mlx5_core_dev *dev, u16 func_id, int npages,
518 			 int *nclaimed, bool event, bool ec_function)
519 {
520 	u32 function = get_function(func_id, ec_function);
521 	int outlen = MLX5_ST_SZ_BYTES(manage_pages_out);
522 	u32 in[MLX5_ST_SZ_DW(manage_pages_in)] = {};
523 	int num_claimed;
524 	u16 func_type;
525 	u32 *out;
526 	int err;
527 	int i;
528 
529 	if (nclaimed)
530 		*nclaimed = 0;
531 
532 	outlen += npages * MLX5_FLD_SZ_BYTES(manage_pages_out, pas[0]);
533 	out = kvzalloc(outlen, GFP_KERNEL);
534 	if (!out)
535 		return -ENOMEM;
536 
537 	MLX5_SET(manage_pages_in, in, opcode, MLX5_CMD_OP_MANAGE_PAGES);
538 	MLX5_SET(manage_pages_in, in, op_mod, MLX5_PAGES_TAKE);
539 	MLX5_SET(manage_pages_in, in, function_id, func_id);
540 	MLX5_SET(manage_pages_in, in, input_num_entries, npages);
541 	MLX5_SET(manage_pages_in, in, embedded_cpu_function, ec_function);
542 
543 	mlx5_core_dbg(dev, "func 0x%x, npages %d, outlen %d\n",
544 		      func_id, npages, outlen);
545 	err = reclaim_pages_cmd(dev, in, sizeof(in), out, outlen);
546 	if (err) {
547 		npages = MLX5_GET(manage_pages_in, in, input_num_entries);
548 		dev->priv.reclaim_pages_discard += npages;
549 	}
550 	/* if triggered by FW event and failed by FW then ignore */
551 	if (event && err == -EREMOTEIO) {
552 		err = 0;
553 		goto out_free;
554 	}
555 
556 	err = mlx5_cmd_check(dev, err, in, out);
557 	if (err) {
558 		mlx5_core_err(dev, "failed reclaiming pages: err %d\n", err);
559 		goto out_free;
560 	}
561 
562 	num_claimed = MLX5_GET(manage_pages_out, out, output_num_entries);
563 	if (num_claimed > npages) {
564 		mlx5_core_warn(dev, "fw returned %d, driver asked %d => corruption\n",
565 			       num_claimed, npages);
566 		err = -EINVAL;
567 		goto out_free;
568 	}
569 
570 	for (i = 0; i < num_claimed; i++)
571 		free_4k(dev, MLX5_GET64(manage_pages_out, out, pas[i]), function);
572 
573 	if (nclaimed)
574 		*nclaimed = num_claimed;
575 
576 	func_type = func_id_to_type(dev, func_id, ec_function);
577 	dev->priv.page_counters[func_type] -= num_claimed;
578 	dev->priv.fw_pages -= num_claimed;
579 
580 out_free:
581 	kvfree(out);
582 	return err;
583 }
584 
585 static void pages_work_handler(struct work_struct *work)
586 {
587 	struct mlx5_pages_req *req = container_of(work, struct mlx5_pages_req, work);
588 	struct mlx5_core_dev *dev = req->dev;
589 	int err = 0;
590 
591 	if (req->release_all)
592 		release_all_pages(dev, req->func_id, req->ec_function);
593 	else if (req->npages < 0)
594 		err = reclaim_pages(dev, req->func_id, -1 * req->npages, NULL,
595 				    true, req->ec_function);
596 	else if (req->npages > 0)
597 		err = give_pages(dev, req->func_id, req->npages, 1, req->ec_function);
598 
599 	if (err)
600 		mlx5_core_warn(dev, "%s fail %d\n",
601 			       req->npages < 0 ? "reclaim" : "give", err);
602 
603 	kfree(req);
604 }
605 
606 enum {
607 	EC_FUNCTION_MASK = 0x8000,
608 	RELEASE_ALL_PAGES_MASK = 0x4000,
609 };
610 
611 static int req_pages_handler(struct notifier_block *nb,
612 			     unsigned long type, void *data)
613 {
614 	struct mlx5_pages_req *req;
615 	struct mlx5_core_dev *dev;
616 	struct mlx5_priv *priv;
617 	struct mlx5_eqe *eqe;
618 	bool ec_function;
619 	bool release_all;
620 	u16 func_id;
621 	s32 npages;
622 
623 	priv = mlx5_nb_cof(nb, struct mlx5_priv, pg_nb);
624 	dev  = container_of(priv, struct mlx5_core_dev, priv);
625 	eqe  = data;
626 
627 	func_id = be16_to_cpu(eqe->data.req_pages.func_id);
628 	npages  = be32_to_cpu(eqe->data.req_pages.num_pages);
629 	ec_function = be16_to_cpu(eqe->data.req_pages.ec_function) & EC_FUNCTION_MASK;
630 	release_all = be16_to_cpu(eqe->data.req_pages.ec_function) &
631 		      RELEASE_ALL_PAGES_MASK;
632 	mlx5_core_dbg(dev, "page request for func 0x%x, npages %d, release_all %d\n",
633 		      func_id, npages, release_all);
634 	req = kzalloc(sizeof(*req), GFP_ATOMIC);
635 	if (!req) {
636 		mlx5_core_warn(dev, "failed to allocate pages request\n");
637 		return NOTIFY_DONE;
638 	}
639 
640 	req->dev = dev;
641 	req->func_id = func_id;
642 	req->npages = npages;
643 	req->ec_function = ec_function;
644 	req->release_all = release_all;
645 	INIT_WORK(&req->work, pages_work_handler);
646 	queue_work(dev->priv.pg_wq, &req->work);
647 	return NOTIFY_OK;
648 }
649 
650 int mlx5_satisfy_startup_pages(struct mlx5_core_dev *dev, int boot)
651 {
652 	u16 func_id;
653 	s32 npages;
654 	int err;
655 
656 	err = mlx5_cmd_query_pages(dev, &func_id, &npages, boot);
657 	if (err)
658 		return err;
659 
660 	mlx5_core_dbg(dev, "requested %d %s pages for func_id 0x%x\n",
661 		      npages, boot ? "boot" : "init", func_id);
662 
663 	return give_pages(dev, func_id, npages, 0, mlx5_core_is_ecpf(dev));
664 }
665 
666 enum {
667 	MLX5_BLKS_FOR_RECLAIM_PAGES = 12
668 };
669 
670 static int optimal_reclaimed_pages(void)
671 {
672 	struct mlx5_cmd_prot_block *block;
673 	struct mlx5_cmd_layout *lay;
674 	int ret;
675 
676 	ret = (sizeof(lay->out) + MLX5_BLKS_FOR_RECLAIM_PAGES * sizeof(block->data) -
677 	       MLX5_ST_SZ_BYTES(manage_pages_out)) /
678 	       MLX5_FLD_SZ_BYTES(manage_pages_out, pas[0]);
679 
680 	return ret;
681 }
682 
683 static int mlx5_reclaim_root_pages(struct mlx5_core_dev *dev,
684 				   struct rb_root *root, u32 function)
685 {
686 	u64 recl_pages_to_jiffies = msecs_to_jiffies(mlx5_tout_ms(dev, RECLAIM_PAGES));
687 	unsigned long end = jiffies + recl_pages_to_jiffies;
688 
689 	while (!RB_EMPTY_ROOT(root)) {
690 		u32 ec_function = mlx5_get_ec_function(function);
691 		u32 function_id = mlx5_get_func_id(function);
692 		int nclaimed;
693 		int err;
694 
695 		err = reclaim_pages(dev, function_id, optimal_reclaimed_pages(),
696 				    &nclaimed, false, ec_function);
697 		if (err) {
698 			mlx5_core_warn(dev, "reclaim_pages err (%d) func_id=0x%x ec_func=0x%x\n",
699 				       err, function_id, ec_function);
700 			return err;
701 		}
702 
703 		if (nclaimed)
704 			end = jiffies + recl_pages_to_jiffies;
705 
706 		if (time_after(jiffies, end)) {
707 			mlx5_core_warn(dev, "FW did not return all pages. giving up...\n");
708 			break;
709 		}
710 	}
711 
712 	return 0;
713 }
714 
715 int mlx5_reclaim_startup_pages(struct mlx5_core_dev *dev)
716 {
717 	struct rb_root *root;
718 	unsigned long id;
719 	void *entry;
720 
721 	xa_for_each(&dev->priv.page_root_xa, id, entry) {
722 		root = entry;
723 		mlx5_reclaim_root_pages(dev, root, id);
724 		xa_erase(&dev->priv.page_root_xa, id);
725 		kfree(root);
726 	}
727 
728 	WARN_ON(!xa_empty(&dev->priv.page_root_xa));
729 
730 	WARN(dev->priv.fw_pages,
731 	     "FW pages counter is %d after reclaiming all pages\n",
732 	     dev->priv.fw_pages);
733 	WARN(dev->priv.page_counters[MLX5_VF],
734 	     "VFs FW pages counter is %d after reclaiming all pages\n",
735 	     dev->priv.page_counters[MLX5_VF]);
736 	WARN(dev->priv.page_counters[MLX5_HOST_PF],
737 	     "External host PF FW pages counter is %d after reclaiming all pages\n",
738 	     dev->priv.page_counters[MLX5_HOST_PF]);
739 	WARN(dev->priv.page_counters[MLX5_EC_VF],
740 	     "EC VFs FW pages counter is %d after reclaiming all pages\n",
741 	     dev->priv.page_counters[MLX5_EC_VF]);
742 
743 	return 0;
744 }
745 
746 int mlx5_pagealloc_init(struct mlx5_core_dev *dev)
747 {
748 	INIT_LIST_HEAD(&dev->priv.free_list);
749 	dev->priv.pg_wq = create_singlethread_workqueue("mlx5_page_allocator");
750 	if (!dev->priv.pg_wq)
751 		return -ENOMEM;
752 
753 	xa_init(&dev->priv.page_root_xa);
754 	mlx5_pages_debugfs_init(dev);
755 
756 	return 0;
757 }
758 
759 void mlx5_pagealloc_cleanup(struct mlx5_core_dev *dev)
760 {
761 	mlx5_pages_debugfs_cleanup(dev);
762 	xa_destroy(&dev->priv.page_root_xa);
763 	destroy_workqueue(dev->priv.pg_wq);
764 }
765 
766 void mlx5_pagealloc_start(struct mlx5_core_dev *dev)
767 {
768 	MLX5_NB_INIT(&dev->priv.pg_nb, req_pages_handler, PAGE_REQUEST);
769 	mlx5_eq_notifier_register(dev, &dev->priv.pg_nb);
770 }
771 
772 void mlx5_pagealloc_stop(struct mlx5_core_dev *dev)
773 {
774 	mlx5_eq_notifier_unregister(dev, &dev->priv.pg_nb);
775 	flush_workqueue(dev->priv.pg_wq);
776 }
777 
778 int mlx5_wait_for_pages(struct mlx5_core_dev *dev, int *pages)
779 {
780 	u64 recl_vf_pages_to_jiffies = msecs_to_jiffies(mlx5_tout_ms(dev, RECLAIM_VFS_PAGES));
781 	unsigned long end = jiffies + recl_vf_pages_to_jiffies;
782 	int prev_pages = *pages;
783 
784 	/* In case of internal error we will free the pages manually later */
785 	if (dev->state == MLX5_DEVICE_STATE_INTERNAL_ERROR) {
786 		mlx5_core_warn(dev, "Skipping wait for vf pages stage");
787 		return 0;
788 	}
789 
790 	mlx5_core_dbg(dev, "Waiting for %d pages\n", prev_pages);
791 	while (*pages) {
792 		if (time_after(jiffies, end)) {
793 			mlx5_core_warn(dev, "aborting while there are %d pending pages\n", *pages);
794 			return -ETIMEDOUT;
795 		}
796 		if (*pages < prev_pages) {
797 			end = jiffies + recl_vf_pages_to_jiffies;
798 			prev_pages = *pages;
799 		}
800 		msleep(50);
801 	}
802 
803 	mlx5_core_dbg(dev, "All pages received\n");
804 	return 0;
805 }
806