1// SPDX-License-Identifier: GPL-2.0-or-later
 2/* Support for hardware buffer manager.
 3 *
 4 * Copyright (C) 2016 Marvell
 5 *
 6 * Gregory CLEMENT <gregory.clement@free-electrons.com>
 
 
 
 
 
 7 */
 8#include <linux/kernel.h>
 9#include <linux/printk.h>
10#include <linux/skbuff.h>
11#include <net/hwbm.h>
12
13void hwbm_buf_free(struct hwbm_pool *bm_pool, void *buf)
14{
15	if (likely(bm_pool->frag_size <= PAGE_SIZE))
16		skb_free_frag(buf);
17	else
18		kfree(buf);
19}
20EXPORT_SYMBOL_GPL(hwbm_buf_free);
21
22/* Refill processing for HW buffer management */
23int hwbm_pool_refill(struct hwbm_pool *bm_pool, gfp_t gfp)
24{
25	int frag_size = bm_pool->frag_size;
26	void *buf;
27
28	if (likely(frag_size <= PAGE_SIZE))
29		buf = netdev_alloc_frag(frag_size);
30	else
31		buf = kmalloc(frag_size, gfp);
32
33	if (!buf)
34		return -ENOMEM;
35
36	if (bm_pool->construct)
37		if (bm_pool->construct(bm_pool, buf)) {
38			hwbm_buf_free(bm_pool, buf);
39			return -ENOMEM;
40		}
41
42	return 0;
43}
44EXPORT_SYMBOL_GPL(hwbm_pool_refill);
45
46int hwbm_pool_add(struct hwbm_pool *bm_pool, unsigned int buf_num)
47{
48	int err, i;
 
49
50	mutex_lock(&bm_pool->buf_lock);
51	if (bm_pool->buf_num == bm_pool->size) {
52		pr_warn("pool already filled\n");
53		mutex_unlock(&bm_pool->buf_lock);
54		return bm_pool->buf_num;
55	}
56
57	if (buf_num + bm_pool->buf_num > bm_pool->size) {
58		pr_warn("cannot allocate %d buffers for pool\n",
59			buf_num);
60		mutex_unlock(&bm_pool->buf_lock);
61		return 0;
62	}
63
64	if ((buf_num + bm_pool->buf_num) < bm_pool->buf_num) {
65		pr_warn("Adding %d buffers to the %d current buffers will overflow\n",
66			buf_num,  bm_pool->buf_num);
67		mutex_unlock(&bm_pool->buf_lock);
68		return 0;
69	}
70
71	for (i = 0; i < buf_num; i++) {
72		err = hwbm_pool_refill(bm_pool, GFP_KERNEL);
73		if (err < 0)
74			break;
75	}
76
77	/* Update BM driver with number of buffers added to pool */
78	bm_pool->buf_num += i;
79
80	pr_debug("hwpm pool: %d of %d buffers added\n", i, buf_num);
81	mutex_unlock(&bm_pool->buf_lock);
82
83	return i;
84}
85EXPORT_SYMBOL_GPL(hwbm_pool_add);

 
 1/* Support for hardware buffer manager.
 2 *
 3 * Copyright (C) 2016 Marvell
 4 *
 5 * Gregory CLEMENT <gregory.clement@free-electrons.com>
 6 *
 7 *  This program is free software; you can redistribute it and/or modify
 8 *  it under the terms of the GNU General Public License as published by
 9 *  the Free Software Foundation; either version 2 of the License, or
10 *  (at your option) any later version.
11 */
12#include <linux/kernel.h>
13#include <linux/printk.h>
14#include <linux/skbuff.h>
15#include <net/hwbm.h>
16
17void hwbm_buf_free(struct hwbm_pool *bm_pool, void *buf)
18{
19	if (likely(bm_pool->frag_size <= PAGE_SIZE))
20		skb_free_frag(buf);
21	else
22		kfree(buf);
23}
24EXPORT_SYMBOL_GPL(hwbm_buf_free);
25
26/* Refill processing for HW buffer management */
27int hwbm_pool_refill(struct hwbm_pool *bm_pool, gfp_t gfp)
28{
29	int frag_size = bm_pool->frag_size;
30	void *buf;
31
32	if (likely(frag_size <= PAGE_SIZE))
33		buf = netdev_alloc_frag(frag_size);
34	else
35		buf = kmalloc(frag_size, gfp);
36
37	if (!buf)
38		return -ENOMEM;
39
40	if (bm_pool->construct)
41		if (bm_pool->construct(bm_pool, buf)) {
42			hwbm_buf_free(bm_pool, buf);
43			return -ENOMEM;
44		}
45
46	return 0;
47}
48EXPORT_SYMBOL_GPL(hwbm_pool_refill);
49
50int hwbm_pool_add(struct hwbm_pool *bm_pool, unsigned int buf_num, gfp_t gfp)
51{
52	int err, i;
53	unsigned long flags;
54
55	spin_lock_irqsave(&bm_pool->lock, flags);
56	if (bm_pool->buf_num == bm_pool->size) {
57		pr_warn("pool already filled\n");
58		spin_unlock_irqrestore(&bm_pool->lock, flags);
59		return bm_pool->buf_num;
60	}
61
62	if (buf_num + bm_pool->buf_num > bm_pool->size) {
63		pr_warn("cannot allocate %d buffers for pool\n",
64			buf_num);
65		spin_unlock_irqrestore(&bm_pool->lock, flags);
66		return 0;
67	}
68
69	if ((buf_num + bm_pool->buf_num) < bm_pool->buf_num) {
70		pr_warn("Adding %d buffers to the %d current buffers will overflow\n",
71			buf_num,  bm_pool->buf_num);
72		spin_unlock_irqrestore(&bm_pool->lock, flags);
73		return 0;
74	}
75
76	for (i = 0; i < buf_num; i++) {
77		err = hwbm_pool_refill(bm_pool, gfp);
78		if (err < 0)
79			break;
80	}
81
82	/* Update BM driver with number of buffers added to pool */
83	bm_pool->buf_num += i;
84
85	pr_debug("hwpm pool: %d of %d buffers added\n", i, buf_num);
86	spin_unlock_irqrestore(&bm_pool->lock, flags);
87
88	return i;
89}
90EXPORT_SYMBOL_GPL(hwbm_pool_add);