[linux.git] / net / core / hwbm.c

/* Support for hardware buffer manager.
 *
 * Copyright (C) 2016 Marvell
 *
 * Gregory CLEMENT <[email protected]>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 */
#include <linux/kernel.h>
#include <linux/printk.h>
#include <linux/skbuff.h>
#include <net/hwbm.h>

void hwbm_buf_free(struct hwbm_pool *bm_pool, void *buf)
{
	if (likely(bm_pool->frag_size <= PAGE_SIZE))
		skb_free_frag(buf);
	else
		kfree(buf);
}
EXPORT_SYMBOL_GPL(hwbm_buf_free);

/* Refill processing for HW buffer management */
int hwbm_pool_refill(struct hwbm_pool *bm_pool, gfp_t gfp)
{
	int frag_size = bm_pool->frag_size;
	void *buf;

	if (likely(frag_size <= PAGE_SIZE))
		buf = netdev_alloc_frag(frag_size);
	else
		buf = kmalloc(frag_size, gfp);

	if (!buf)
		return -ENOMEM;

	if (bm_pool->construct)
		if (bm_pool->construct(bm_pool, buf)) {
			hwbm_buf_free(bm_pool, buf);
			return -ENOMEM;
		}

	return 0;
}
EXPORT_SYMBOL_GPL(hwbm_pool_refill);

int hwbm_pool_add(struct hwbm_pool *bm_pool, unsigned int buf_num, gfp_t gfp)
{
	int err, i;
	unsigned long flags;

	spin_lock_irqsave(&bm_pool->lock, flags);
	if (bm_pool->buf_num == bm_pool->size) {
		pr_warn("pool already filled\n");
		spin_unlock_irqrestore(&bm_pool->lock, flags);
		return bm_pool->buf_num;
	}

	if (buf_num + bm_pool->buf_num > bm_pool->size) {
		pr_warn("cannot allocate %d buffers for pool\n",
			buf_num);
		spin_unlock_irqrestore(&bm_pool->lock, flags);
		return 0;
	}

	if ((buf_num + bm_pool->buf_num) < bm_pool->buf_num) {
		pr_warn("Adding %d buffers to the %d current buffers will overflow\n",
			buf_num,  bm_pool->buf_num);
		spin_unlock_irqrestore(&bm_pool->lock, flags);
		return 0;
	}

	for (i = 0; i < buf_num; i++) {
		err = hwbm_pool_refill(bm_pool, gfp);
		if (err < 0)
			break;
	}

	/* Update BM driver with number of buffers added to pool */
	bm_pool->buf_num += i;

	pr_debug("hwpm pool: %d of %d buffers added\n", i, buf_num);
	spin_unlock_irqrestore(&bm_pool->lock, flags);

	return i;
}
EXPORT_SYMBOL_GPL(hwbm_pool_add);
Commit	Line	Data
8cb2d8bf GC	1	/* Support for hardware buffer manager.
	2	*
	3	* Copyright (C) 2016 Marvell
	4	*
	5	* Gregory CLEMENT <[email protected]>
	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
	17	void 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	}
	24	EXPORT_SYMBOL_GPL(hwbm_buf_free);
	25
	26	/* Refill processing for HW buffer management */
	27	int 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	}
	48	EXPORT_SYMBOL_GPL(hwbm_pool_refill);
	49
	50	int 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");
b388fc74	58	spin_unlock_irqrestore(&bm_pool->lock, flags);
8cb2d8bf GC	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);
b388fc74	65	spin_unlock_irqrestore(&bm_pool->lock, flags);
8cb2d8bf GC	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);
b388fc74	72	spin_unlock_irqrestore(&bm_pool->lock, flags);
8cb2d8bf GC	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	}
	90	EXPORT_SYMBOL_GPL(hwbm_pool_add);