[GitHub/mt8127/android_kernel_alcatel_ttab.git] / net / mac80211 / wme.c

/*
 * Copyright 2004, Instant802 Networks, Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include <linux/netdevice.h>
#include <linux/skbuff.h>
#include <linux/module.h>
#include <linux/if_arp.h>
#include <linux/types.h>
#include <net/ip.h>
#include <net/pkt_sched.h>

#include <net/mac80211.h>
#include "ieee80211_i.h"
#include "wme.h"

/* maximum number of hardware queues we support. */
#define TC_80211_MAX_QUEUES 16

const int ieee802_1d_to_ac[8] = { 2, 3, 3, 2, 1, 1, 0, 0 };

struct ieee80211_sched_data
{
	unsigned long qdisc_pool[BITS_TO_LONGS(TC_80211_MAX_QUEUES)];
	struct tcf_proto *filter_list;
	struct Qdisc *queues[TC_80211_MAX_QUEUES];
	struct sk_buff_head requeued[TC_80211_MAX_QUEUES];
};

static const char llc_ip_hdr[8] = {0xAA, 0xAA, 0x3, 0, 0, 0, 0x08, 0};

/* given a data frame determine the 802.1p/1d tag to use */
static inline unsigned classify_1d(struct sk_buff *skb, struct Qdisc *qd)
{
	struct iphdr *ip;
	int dscp;
	int offset;

	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct tcf_result res = { -1, 0 };

	/* if there is a user set filter list, call out to that */
	if (q->filter_list) {
		tc_classify(skb, q->filter_list, &res);
		if (res.class != -1)
			return res.class;
	}

	/* skb->priority values from 256->263 are magic values to
	 * directly indicate a specific 802.1d priority.
	 * This is used to allow 802.1d priority to be passed directly in
	 * from VLAN tags, etc. */
	if (skb->priority >= 256 && skb->priority <= 263)
		return skb->priority - 256;

	/* check there is a valid IP header present */
	offset = ieee80211_get_hdrlen_from_skb(skb);
	if (skb->len < offset + sizeof(llc_ip_hdr) + sizeof(*ip) ||
	    memcmp(skb->data + offset, llc_ip_hdr, sizeof(llc_ip_hdr)))
		return 0;

	ip = (struct iphdr *) (skb->data + offset + sizeof(llc_ip_hdr));

	dscp = ip->tos & 0xfc;
	if (dscp & 0x1c)
		return 0;
	return dscp >> 5;
}


static inline int wme_downgrade_ac(struct sk_buff *skb)
{
	switch (skb->priority) {
	case 6:
	case 7:
		skb->priority = 5; /* VO -> VI */
		return 0;
	case 4:
	case 5:
		skb->priority = 3; /* VI -> BE */
		return 0;
	case 0:
	case 3:
		skb->priority = 2; /* BE -> BK */
		return 0;
	default:
		return -1;
	}
}


/* positive return value indicates which queue to use
 * negative return value indicates to drop the frame */
static inline int classify80211(struct sk_buff *skb, struct Qdisc *qd)
{
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
	unsigned short fc = le16_to_cpu(hdr->frame_control);
	int qos;

	/* see if frame is data or non data frame */
	if (unlikely((fc & IEEE80211_FCTL_FTYPE) != IEEE80211_FTYPE_DATA)) {
		/* management frames go on AC_VO queue, but are sent
		* without QoS control fields */
		return IEEE80211_TX_QUEUE_DATA0;
	}

	if (0 /* injected */) {
		/* use AC from radiotap */
	}

	/* is this a QoS frame? */
	qos = fc & IEEE80211_STYPE_QOS_DATA;

	if (!qos) {
		skb->priority = 0; /* required for correct WPA/11i MIC */
		return ieee802_1d_to_ac[skb->priority];
	}

	/* use the data classifier to determine what 802.1d tag the
	 * data frame has */
	skb->priority = classify_1d(skb, qd);

	/* in case we are a client verify acm is not set for this ac */
	while (unlikely(local->wmm_acm & BIT(skb->priority))) {
		if (wme_downgrade_ac(skb)) {
			/* No AC with lower priority has acm=0, drop packet. */
			return -1;
		}
	}

	/* look up which queue to use for frames with this 1d tag */
	return ieee802_1d_to_ac[skb->priority];
}


static int wme_qdiscop_enqueue(struct sk_buff *skb, struct Qdisc* qd)
{
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct ieee80211_tx_packet_data *pkt_data =
		(struct ieee80211_tx_packet_data *) skb->cb;
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
	unsigned short fc = le16_to_cpu(hdr->frame_control);
	struct Qdisc *qdisc;
	int err, queue;
	struct sta_info *sta;
	u8 tid;

	if (pkt_data->flags & IEEE80211_TXPD_REQUEUE) {
		queue = pkt_data->queue;
		rcu_read_lock();
		sta = sta_info_get(local, hdr->addr1);
		tid = skb->priority & QOS_CONTROL_TAG1D_MASK;
		if (sta) {
			int ampdu_queue = sta->tid_to_tx_q[tid];
			if ((ampdu_queue < local->hw.queues) &&
			    test_bit(ampdu_queue, q->qdisc_pool)) {
				queue = ampdu_queue;
				pkt_data->flags |= IEEE80211_TXPD_AMPDU;
			} else {
				pkt_data->flags &= ~IEEE80211_TXPD_AMPDU;
			}
		}
		rcu_read_unlock();
		skb_queue_tail(&q->requeued[queue], skb);
		qd->q.qlen++;
		return 0;
	}

	queue = classify80211(skb, qd);

	/* now we know the 1d priority, fill in the QoS header if there is one
	 */
	if (WLAN_FC_IS_QOS_DATA(fc)) {
		u8 *p = skb->data + ieee80211_get_hdrlen(fc) - 2;
		u8 ack_policy = 0;
		tid = skb->priority & QOS_CONTROL_TAG1D_MASK;
		if (local->wifi_wme_noack_test)
			ack_policy |= QOS_CONTROL_ACK_POLICY_NOACK <<
					QOS_CONTROL_ACK_POLICY_SHIFT;
		/* qos header is 2 bytes, second reserved */
		*p = ack_policy | tid;
		p++;
		*p = 0;

		rcu_read_lock();

		sta = sta_info_get(local, hdr->addr1);
		if (sta) {
			int ampdu_queue = sta->tid_to_tx_q[tid];
			if ((ampdu_queue < local->hw.queues) &&
				test_bit(ampdu_queue, q->qdisc_pool)) {
				queue = ampdu_queue;
				pkt_data->flags |= IEEE80211_TXPD_AMPDU;
			} else {
				pkt_data->flags &= ~IEEE80211_TXPD_AMPDU;
			}
		}

		rcu_read_unlock();
	}

	if (unlikely(queue >= local->hw.queues)) {
#if 0
		if (net_ratelimit()) {
			printk(KERN_DEBUG "%s - queue=%d (hw does not "
			       "support) -> %d\n",
			       __func__, queue, local->hw.queues - 1);
		}
#endif
		queue = local->hw.queues - 1;
	}

	if (unlikely(queue < 0)) {
			kfree_skb(skb);
			err = NET_XMIT_DROP;
	} else {
		tid = skb->priority & QOS_CONTROL_TAG1D_MASK;
		pkt_data->queue = (unsigned int) queue;
		qdisc = q->queues[queue];
		err = qdisc->enqueue(skb, qdisc);
		if (err == NET_XMIT_SUCCESS) {
			qd->q.qlen++;
			qd->bstats.bytes += skb->len;
			qd->bstats.packets++;
			return NET_XMIT_SUCCESS;
		}
	}
	qd->qstats.drops++;
	return err;
}


/* TODO: clean up the cases where master_hard_start_xmit
 * returns non 0 - it shouldn't ever do that. Once done we
 * can remove this function */
static int wme_qdiscop_requeue(struct sk_buff *skb, struct Qdisc* qd)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct ieee80211_tx_packet_data *pkt_data =
		(struct ieee80211_tx_packet_data *) skb->cb;
	struct Qdisc *qdisc;
	int err;

	/* we recorded which queue to use earlier! */
	qdisc = q->queues[pkt_data->queue];

	if ((err = qdisc->ops->requeue(skb, qdisc)) == 0) {
		qd->q.qlen++;
		return 0;
	}
	qd->qstats.drops++;
	return err;
}


static struct sk_buff *wme_qdiscop_dequeue(struct Qdisc* qd)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct net_device *dev = qd->dev;
	struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;
	struct sk_buff *skb;
	struct Qdisc *qdisc;
	int queue;

	/* check all the h/w queues in numeric/priority order */
	for (queue = 0; queue < hw->queues; queue++) {
		/* see if there is room in this hardware queue */
		if ((test_bit(IEEE80211_LINK_STATE_XOFF,
				&local->state[queue])) ||
		    (test_bit(IEEE80211_LINK_STATE_PENDING,
				&local->state[queue])) ||
			 (!test_bit(queue, q->qdisc_pool)))
			continue;

		/* there is space - try and get a frame */
		skb = skb_dequeue(&q->requeued[queue]);
		if (skb) {
			qd->q.qlen--;
			return skb;
		}

		qdisc = q->queues[queue];
		skb = qdisc->dequeue(qdisc);
		if (skb) {
			qd->q.qlen--;
			return skb;
		}
	}
	/* returning a NULL here when all the h/w queues are full means we
	 * never need to call netif_stop_queue in the driver */
	return NULL;
}


static void wme_qdiscop_reset(struct Qdisc* qd)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;
	int queue;

	/* QUESTION: should we have some hardware flush functionality here? */

	for (queue = 0; queue < hw->queues; queue++) {
		skb_queue_purge(&q->requeued[queue]);
		qdisc_reset(q->queues[queue]);
	}
	qd->q.qlen = 0;
}


static void wme_qdiscop_destroy(struct Qdisc* qd)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;
	int queue;

	tcf_destroy_chain(q->filter_list);
	q->filter_list = NULL;

	for (queue=0; queue < hw->queues; queue++) {
		skb_queue_purge(&q->requeued[queue]);
		qdisc_destroy(q->queues[queue]);
		q->queues[queue] = &noop_qdisc;
	}
}


/* called whenever parameters are updated on existing qdisc */
static int wme_qdiscop_tune(struct Qdisc *qd, struct nlattr *opt)
{
/*	struct ieee80211_sched_data *q = qdisc_priv(qd);
*/
	/* check our options block is the right size */
	/* copy any options to our local structure */
/*	Ignore options block for now - always use static mapping
	struct tc_ieee80211_qopt *qopt = nla_data(opt);

	if (opt->nla_len < nla_attr_size(sizeof(*qopt)))
		return -EINVAL;
	memcpy(q->tag2queue, qopt->tag2queue, sizeof(qopt->tag2queue));
*/
	return 0;
}


/* called during initial creation of qdisc on device */
static int wme_qdiscop_init(struct Qdisc *qd, struct nlattr *opt)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct net_device *dev = qd->dev;
	struct ieee80211_local *local;
	int queues;
	int err = 0, i;

	/* check that device is a mac80211 device */
	if (!dev->ieee80211_ptr ||
	    dev->ieee80211_ptr->wiphy->privid != mac80211_wiphy_privid)
		return -EINVAL;

	/* check this device is an ieee80211 master type device */
	if (dev->type != ARPHRD_IEEE80211)
		return -EINVAL;

	/* check that there is no qdisc currently attached to device
	 * this ensures that we will be the root qdisc. (I can't find a better
	 * way to test this explicitly) */
	if (dev->qdisc_sleeping != &noop_qdisc)
		return -EINVAL;

	if (qd->flags & TCQ_F_INGRESS)
		return -EINVAL;

	local = wdev_priv(dev->ieee80211_ptr);
	queues = local->hw.queues;

	/* if options were passed in, set them */
	if (opt) {
		err = wme_qdiscop_tune(qd, opt);
	}

	/* create child queues */
	for (i = 0; i < queues; i++) {
		skb_queue_head_init(&q->requeued[i]);
		q->queues[i] = qdisc_create_dflt(qd->dev, &pfifo_qdisc_ops,
						 qd->handle);
		if (!q->queues[i]) {
			q->queues[i] = &noop_qdisc;
			printk(KERN_ERR "%s child qdisc %i creation failed", dev->name, i);
		}
	}

	/* reserve all legacy QoS queues */
	for (i = 0; i < min(IEEE80211_TX_QUEUE_DATA4, queues); i++)
		set_bit(i, q->qdisc_pool);

	return err;
}

static int wme_qdiscop_dump(struct Qdisc *qd, struct sk_buff *skb)
{
/*	struct ieee80211_sched_data *q = qdisc_priv(qd);
	unsigned char *p = skb->tail;
	struct tc_ieee80211_qopt opt;

	memcpy(&opt.tag2queue, q->tag2queue, TC_80211_MAX_TAG + 1);
	NLA_PUT(skb, TCA_OPTIONS, sizeof(opt), &opt);
*/	return skb->len;
/*
nla_put_failure:
	skb_trim(skb, p - skb->data);*/
	return -1;
}


static int wme_classop_graft(struct Qdisc *qd, unsigned long arg,
			     struct Qdisc *new, struct Qdisc **old)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;
	unsigned long queue = arg - 1;

	if (queue >= hw->queues)
		return -EINVAL;

	if (!new)
		new = &noop_qdisc;

	sch_tree_lock(qd);
	*old = q->queues[queue];
	q->queues[queue] = new;
	qdisc_reset(*old);
	sch_tree_unlock(qd);

	return 0;
}


static struct Qdisc *
wme_classop_leaf(struct Qdisc *qd, unsigned long arg)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;
	unsigned long queue = arg - 1;

	if (queue >= hw->queues)
		return NULL;

	return q->queues[queue];
}


static unsigned long wme_classop_get(struct Qdisc *qd, u32 classid)
{
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;
	unsigned long queue = TC_H_MIN(classid);

	if (queue - 1 >= hw->queues)
		return 0;

	return queue;
}


static unsigned long wme_classop_bind(struct Qdisc *qd, unsigned long parent,
				      u32 classid)
{
	return wme_classop_get(qd, classid);
}


static void wme_classop_put(struct Qdisc *q, unsigned long cl)
{
}


static int wme_classop_change(struct Qdisc *qd, u32 handle, u32 parent,
			      struct nlattr **tca, unsigned long *arg)
{
	unsigned long cl = *arg;
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;

	if (cl - 1 > hw->queues)
		return -ENOENT;

	/* TODO: put code to program hardware queue parameters here,
	 * to allow programming from tc command line */

	return 0;
}


/* we don't support deleting hardware queues
 * when we add WMM-SA support - TSPECs may be deleted here */
static int wme_classop_delete(struct Qdisc *qd, unsigned long cl)
{
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;

	if (cl - 1 > hw->queues)
		return -ENOENT;
	return 0;
}


static int wme_classop_dump_class(struct Qdisc *qd, unsigned long cl,
				  struct sk_buff *skb, struct tcmsg *tcm)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;

	if (cl - 1 > hw->queues)
		return -ENOENT;
	tcm->tcm_handle = TC_H_MIN(cl);
	tcm->tcm_parent = qd->handle;
	tcm->tcm_info = q->queues[cl-1]->handle; /* do we need this? */
	return 0;
}


static void wme_classop_walk(struct Qdisc *qd, struct qdisc_walker *arg)
{
	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	struct ieee80211_hw *hw = &local->hw;
	int queue;

	if (arg->stop)
		return;

	for (queue = 0; queue < hw->queues; queue++) {
		if (arg->count < arg->skip) {
			arg->count++;
			continue;
		}
		/* we should return classids for our internal queues here
		 * as well as the external ones */
		if (arg->fn(qd, queue+1, arg) < 0) {
			arg->stop = 1;
			break;
		}
		arg->count++;
	}
}


static struct tcf_proto ** wme_classop_find_tcf(struct Qdisc *qd,
						unsigned long cl)
{
	struct ieee80211_sched_data *q = qdisc_priv(qd);

	if (cl)
		return NULL;

	return &q->filter_list;
}


/* this qdisc is classful (i.e. has classes, some of which may have leaf qdiscs attached)
 * - these are the operations on the classes */
static const struct Qdisc_class_ops class_ops =
{
	.graft = wme_classop_graft,
	.leaf = wme_classop_leaf,

	.get = wme_classop_get,
	.put = wme_classop_put,
	.change = wme_classop_change,
	.delete = wme_classop_delete,
	.walk = wme_classop_walk,

	.tcf_chain = wme_classop_find_tcf,
	.bind_tcf = wme_classop_bind,
	.unbind_tcf = wme_classop_put,

	.dump = wme_classop_dump_class,
};


/* queueing discipline operations */
static struct Qdisc_ops wme_qdisc_ops __read_mostly =
{
	.next = NULL,
	.cl_ops = &class_ops,
	.id = "ieee80211",
	.priv_size = sizeof(struct ieee80211_sched_data),

	.enqueue = wme_qdiscop_enqueue,
	.dequeue = wme_qdiscop_dequeue,
	.requeue = wme_qdiscop_requeue,
	.drop = NULL, /* drop not needed since we are always the root qdisc */

	.init = wme_qdiscop_init,
	.reset = wme_qdiscop_reset,
	.destroy = wme_qdiscop_destroy,
	.change = wme_qdiscop_tune,

	.dump = wme_qdiscop_dump,
};


void ieee80211_install_qdisc(struct net_device *dev)
{
	struct Qdisc *qdisc;

	qdisc = qdisc_create_dflt(dev, &wme_qdisc_ops, TC_H_ROOT);
	if (!qdisc) {
		printk(KERN_ERR "%s: qdisc installation failed\n", dev->name);
		return;
	}

	/* same handle as would be allocated by qdisc_alloc_handle() */
	qdisc->handle = 0x80010000;

	qdisc_lock_tree(dev);
	list_add_tail(&qdisc->list, &dev->qdisc_list);
	dev->qdisc_sleeping = qdisc;
	qdisc_unlock_tree(dev);
}


int ieee80211_qdisc_installed(struct net_device *dev)
{
	return dev->qdisc_sleeping->ops == &wme_qdisc_ops;
}


int ieee80211_wme_register(void)
{
	return register_qdisc(&wme_qdisc_ops);
}


void ieee80211_wme_unregister(void)
{
	unregister_qdisc(&wme_qdisc_ops);
}

int ieee80211_ht_agg_queue_add(struct ieee80211_local *local,
			struct sta_info *sta, u16 tid)
{
	int i;
	struct ieee80211_sched_data *q =
			qdisc_priv(local->mdev->qdisc_sleeping);
	DECLARE_MAC_BUF(mac);

	/* prepare the filter and save it for the SW queue
	 * matching the recieved HW queue */

	/* try to get a Qdisc from the pool */
	for (i = IEEE80211_TX_QUEUE_BEACON; i < local->hw.queues; i++)
		if (!test_and_set_bit(i, q->qdisc_pool)) {
			ieee80211_stop_queue(local_to_hw(local), i);
			sta->tid_to_tx_q[tid] = i;

			/* IF there are already pending packets
			 * on this tid first we need to drain them
			 * on the previous queue
			 * since HT is strict in order */
#ifdef CONFIG_MAC80211_HT_DEBUG
			if (net_ratelimit())
				printk(KERN_DEBUG "allocated aggregation queue"
					" %d tid %d addr %s pool=0x%lX",
					i, tid, print_mac(mac, sta->addr),
					q->qdisc_pool[0]);
#endif /* CONFIG_MAC80211_HT_DEBUG */
			return 0;
		}

	return -EAGAIN;
}

/**
 * the caller needs to hold local->mdev->queue_lock
 */
void ieee80211_ht_agg_queue_remove(struct ieee80211_local *local,
				   struct sta_info *sta, u16 tid,
				   u8 requeue)
{
	struct ieee80211_sched_data *q =
		qdisc_priv(local->mdev->qdisc_sleeping);
	int agg_queue = sta->tid_to_tx_q[tid];

	/* return the qdisc to the pool */
	clear_bit(agg_queue, q->qdisc_pool);
	sta->tid_to_tx_q[tid] = local->hw.queues;

	if (requeue)
		ieee80211_requeue(local, agg_queue);
	else
		q->queues[agg_queue]->ops->reset(q->queues[agg_queue]);
}

void ieee80211_requeue(struct ieee80211_local *local, int queue)
{
	struct Qdisc *root_qd = local->mdev->qdisc_sleeping;
	struct ieee80211_sched_data *q = qdisc_priv(root_qd);
	struct Qdisc *qdisc = q->queues[queue];
	struct sk_buff *skb = NULL;
	u32 len = qdisc->q.qlen;

	if (!qdisc || !qdisc->dequeue)
		return;

	printk(KERN_DEBUG "requeue: qlen = %d\n", qdisc->q.qlen);
	for (len = qdisc->q.qlen; len > 0; len--) {
		skb = qdisc->dequeue(qdisc);
		root_qd->q.qlen--;
		/* packet will be classified again and */
		/* skb->packet_data->queue will be overridden if needed */
		if (skb)
			wme_qdiscop_enqueue(skb, root_qd);
	}
}
Commit	Line	Data
f0706e82 JB	1	/*
	2	* Copyright 2004, Instant802 Networks, Inc.
	3	*
	4	* This program is free software; you can redistribute it and/or modify
	5	* it under the terms of the GNU General Public License version 2 as
	6	* published by the Free Software Foundation.
	7	*/
	8
	9	#include <linux/netdevice.h>
	10	#include <linux/skbuff.h>
	11	#include <linux/module.h>
	12	#include <linux/if_arp.h>
	13	#include <linux/types.h>
	14	#include <net/ip.h>
	15	#include <net/pkt_sched.h>
	16
	17	#include <net/mac80211.h>
	18	#include "ieee80211_i.h"
	19	#include "wme.h"
	20
f0706e82	21	/* maximum number of hardware queues we support. */
9e723492 RR	22	#define TC_80211_MAX_QUEUES 16
	23
	24	const int ieee802_1d_to_ac[8] = { 2, 3, 3, 2, 1, 1, 0, 0 };
f0706e82 JB	25
	26	struct ieee80211_sched_data
	27	{
a9af2013	28	unsigned long qdisc_pool[BITS_TO_LONGS(TC_80211_MAX_QUEUES)];
f0706e82 JB	29	struct tcf_proto *filter_list;
	30	struct Qdisc *queues[TC_80211_MAX_QUEUES];
	31	struct sk_buff_head requeued[TC_80211_MAX_QUEUES];
	32	};
	33
a8bdf29c	34	static const char llc_ip_hdr[8] = {0xAA, 0xAA, 0x3, 0, 0, 0, 0x08, 0};
f0706e82 JB	35
	36	/* given a data frame determine the 802.1p/1d tag to use */
	37	static inline unsigned classify_1d(struct sk_buff skb, struct Qdisc qd)
	38	{
	39	struct iphdr *ip;
	40	int dscp;
	41	int offset;
	42
	43	struct ieee80211_sched_data *q = qdisc_priv(qd);
	44	struct tcf_result res = { -1, 0 };
	45
	46	/* if there is a user set filter list, call out to that */
	47	if (q->filter_list) {
	48	tc_classify(skb, q->filter_list, &res);
	49	if (res.class != -1)
	50	return res.class;
	51	}
	52
	53	/* skb->priority values from 256->263 are magic values to
	54	* directly indicate a specific 802.1d priority.
	55	* This is used to allow 802.1d priority to be passed directly in
	56	* from VLAN tags, etc. */
	57	if (skb->priority >= 256 && skb->priority <= 263)
	58	return skb->priority - 256;
	59
	60	/* check there is a valid IP header present */
a8bdf29c GC	61	offset = ieee80211_get_hdrlen_from_skb(skb);
	62	if (skb->len < offset + sizeof(llc_ip_hdr) + sizeof(*ip) \|\|
	63	memcmp(skb->data + offset, llc_ip_hdr, sizeof(llc_ip_hdr)))
f0706e82 JB	64	return 0;
f0706e82 JB	65
a8bdf29c	66	ip = (struct iphdr *) (skb->data + offset + sizeof(llc_ip_hdr));
f0706e82 JB	67
	68	dscp = ip->tos & 0xfc;
	69	if (dscp & 0x1c)
	70	return 0;
	71	return dscp >> 5;
	72	}
	73
	74
	75	static inline int wme_downgrade_ac(struct sk_buff *skb)
	76	{
	77	switch (skb->priority) {
	78	case 6:
	79	case 7:
	80	skb->priority = 5; /* VO -> VI */
	81	return 0;
	82	case 4:
	83	case 5:
	84	skb->priority = 3; /* VI -> BE */
	85	return 0;
	86	case 0:
	87	case 3:
	88	skb->priority = 2; /* BE -> BK */
	89	return 0;
	90	default:
	91	return -1;
	92	}
	93	}
	94
	95
	96	/* positive return value indicates which queue to use
	97	* negative return value indicates to drop the frame */
	98	static inline int classify80211(struct sk_buff skb, struct Qdisc qd)
	99	{
	100	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
f0706e82 JB	101	struct ieee80211_hdr hdr = (struct ieee80211_hdr ) skb->data;
	102	unsigned short fc = le16_to_cpu(hdr->frame_control);
	103	int qos;
f0706e82 JB	104
	105	/* see if frame is data or non data frame */
	106	if (unlikely((fc & IEEE80211_FCTL_FTYPE) != IEEE80211_FTYPE_DATA)) {
	107	/* management frames go on AC_VO queue, but are sent
	108	* without QoS control fields */
	109	return IEEE80211_TX_QUEUE_DATA0;
	110	}
	111
f9d540ee JB	112	if (0 /* injected */) {
f9d540ee JB	113	/* use AC from radiotap */
f0706e82 JB	114	}
	115
	116	/* is this a QoS frame? */
	117	qos = fc & IEEE80211_STYPE_QOS_DATA;
	118
	119	if (!qos) {
	120	skb->priority = 0; /* required for correct WPA/11i MIC */
	121	return ieee802_1d_to_ac[skb->priority];
	122	}
	123
	124	/* use the data classifier to determine what 802.1d tag the
3c3b00ca	125	* data frame has */
f0706e82 JB	126	skb->priority = classify_1d(skb, qd);
f0706e82 JB	127
3c3b00ca	128	/* in case we are a client verify acm is not set for this ac */
f0706e82 JB	129	while (unlikely(local->wmm_acm & BIT(skb->priority))) {
f0706e82 JB	130	if (wme_downgrade_ac(skb)) {
3c3b00ca	131	/* No AC with lower priority has acm=0, drop packet. */
f0706e82 JB	132	return -1;
	133	}
	134	}
	135
	136	/* look up which queue to use for frames with this 1d tag */
	137	return ieee802_1d_to_ac[skb->priority];
	138	}
	139
	140
	141	static int wme_qdiscop_enqueue(struct sk_buff skb, struct Qdisc qd)
	142	{
	143	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	144	struct ieee80211_sched_data *q = qdisc_priv(qd);
	145	struct ieee80211_tx_packet_data *pkt_data =
	146	(struct ieee80211_tx_packet_data *) skb->cb;
	147	struct ieee80211_hdr hdr = (struct ieee80211_hdr ) skb->data;
	148	unsigned short fc = le16_to_cpu(hdr->frame_control);
	149	struct Qdisc *qdisc;
	150	int err, queue;
9e723492 RR	151	struct sta_info *sta;
9e723492 RR	152	u8 tid;
f0706e82	153
e8bf9649	154	if (pkt_data->flags & IEEE80211_TXPD_REQUEUE) {
9e723492	155	queue = pkt_data->queue;
d0709a65	156	rcu_read_lock();
9e723492 RR	157	sta = sta_info_get(local, hdr->addr1);
	158	tid = skb->priority & QOS_CONTROL_TAG1D_MASK;
	159	if (sta) {
	160	int ampdu_queue = sta->tid_to_tx_q[tid];
	161	if ((ampdu_queue < local->hw.queues) &&
a9af2013	162	test_bit(ampdu_queue, q->qdisc_pool)) {
9e723492 RR	163	queue = ampdu_queue;
	164	pkt_data->flags \|= IEEE80211_TXPD_AMPDU;
	165	} else {
	166	pkt_data->flags &= ~IEEE80211_TXPD_AMPDU;
	167	}
9e723492	168	}
d0709a65	169	rcu_read_unlock();
9e723492	170	skb_queue_tail(&q->requeued[queue], skb);
f0706e82 JB	171	qd->q.qlen++;
	172	return 0;
	173	}
	174
	175	queue = classify80211(skb, qd);
	176
	177	/* now we know the 1d priority, fill in the QoS header if there is one
	178	*/
	179	if (WLAN_FC_IS_QOS_DATA(fc)) {
	180	u8 *p = skb->data + ieee80211_get_hdrlen(fc) - 2;
9e723492 RR	181	u8 ack_policy = 0;
9e723492 RR	182	tid = skb->priority & QOS_CONTROL_TAG1D_MASK;
f0706e82	183	if (local->wifi_wme_noack_test)
9e723492	184	ack_policy \|= QOS_CONTROL_ACK_POLICY_NOACK <<
f0706e82 JB	185	QOS_CONTROL_ACK_POLICY_SHIFT;
f0706e82 JB	186	/* qos header is 2 bytes, second reserved */
9e723492	187	*p = ack_policy \| tid;
f0706e82 JB	188	p++;
f0706e82 JB	189	*p = 0;
9e723492	190
d0709a65 JB	191	rcu_read_lock();
d0709a65 JB	192
9e723492 RR	193	sta = sta_info_get(local, hdr->addr1);
	194	if (sta) {
	195	int ampdu_queue = sta->tid_to_tx_q[tid];
	196	if ((ampdu_queue < local->hw.queues) &&
a9af2013	197	test_bit(ampdu_queue, q->qdisc_pool)) {
9e723492 RR	198	queue = ampdu_queue;
	199	pkt_data->flags \|= IEEE80211_TXPD_AMPDU;
	200	} else {
	201	pkt_data->flags &= ~IEEE80211_TXPD_AMPDU;
	202	}
9e723492	203	}
d0709a65 JB	204
d0709a65 JB	205	rcu_read_unlock();
f0706e82 JB	206	}
	207
	208	if (unlikely(queue >= local->hw.queues)) {
	209	#if 0
	210	if (net_ratelimit()) {
	211	printk(KERN_DEBUG "%s - queue=%d (hw does not "
	212	"support) -> %d\n",
	213	__func__, queue, local->hw.queues - 1);
	214	}
	215	#endif
	216	queue = local->hw.queues - 1;
	217	}
	218
	219	if (unlikely(queue < 0)) {
	220	kfree_skb(skb);
	221	err = NET_XMIT_DROP;
	222	} else {
9e723492	223	tid = skb->priority & QOS_CONTROL_TAG1D_MASK;
f0706e82 JB	224	pkt_data->queue = (unsigned int) queue;
	225	qdisc = q->queues[queue];
	226	err = qdisc->enqueue(skb, qdisc);
	227	if (err == NET_XMIT_SUCCESS) {
	228	qd->q.qlen++;
	229	qd->bstats.bytes += skb->len;
	230	qd->bstats.packets++;
	231	return NET_XMIT_SUCCESS;
	232	}
	233	}
	234	qd->qstats.drops++;
	235	return err;
	236	}
	237
	238
	239	/* TODO: clean up the cases where master_hard_start_xmit
	240	* returns non 0 - it shouldn't ever do that. Once done we
	241	* can remove this function */
	242	static int wme_qdiscop_requeue(struct sk_buff skb, struct Qdisc qd)
	243	{
	244	struct ieee80211_sched_data *q = qdisc_priv(qd);
	245	struct ieee80211_tx_packet_data *pkt_data =
	246	(struct ieee80211_tx_packet_data *) skb->cb;
	247	struct Qdisc *qdisc;
	248	int err;
	249
	250	/* we recorded which queue to use earlier! */
	251	qdisc = q->queues[pkt_data->queue];
	252
	253	if ((err = qdisc->ops->requeue(skb, qdisc)) == 0) {
	254	qd->q.qlen++;
	255	return 0;
	256	}
	257	qd->qstats.drops++;
	258	return err;
	259	}
	260
	261
	262	static struct sk_buff wme_qdiscop_dequeue(struct Qdisc qd)
	263	{
	264	struct ieee80211_sched_data *q = qdisc_priv(qd);
	265	struct net_device *dev = qd->dev;
	266	struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
	267	struct ieee80211_hw *hw = &local->hw;
	268	struct sk_buff *skb;
	269	struct Qdisc *qdisc;
	270	int queue;
	271
	272	/* check all the h/w queues in numeric/priority order */
	273	for (queue = 0; queue < hw->queues; queue++) {
	274	/* see if there is room in this hardware queue */
9e723492 RR	275	if ((test_bit(IEEE80211_LINK_STATE_XOFF,
	276	&local->state[queue])) \|\|
	277	(test_bit(IEEE80211_LINK_STATE_PENDING,
	278	&local->state[queue])) \|\|
a9af2013	279	(!test_bit(queue, q->qdisc_pool)))
f0706e82 JB	280	continue;
	281
	282	/* there is space - try and get a frame */
	283	skb = skb_dequeue(&q->requeued[queue]);
	284	if (skb) {
	285	qd->q.qlen--;
	286	return skb;
	287	}
	288
	289	qdisc = q->queues[queue];
	290	skb = qdisc->dequeue(qdisc);
	291	if (skb) {
	292	qd->q.qlen--;
	293	return skb;
	294	}
	295	}
	296	/* returning a NULL here when all the h/w queues are full means we
	297	* never need to call netif_stop_queue in the driver */
	298	return NULL;
	299	}
	300
	301
	302	static void wme_qdiscop_reset(struct Qdisc* qd)
	303	{
	304	struct ieee80211_sched_data *q = qdisc_priv(qd);
	305	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	306	struct ieee80211_hw *hw = &local->hw;
	307	int queue;
	308
	309	/* QUESTION: should we have some hardware flush functionality here? */
	310
	311	for (queue = 0; queue < hw->queues; queue++) {
	312	skb_queue_purge(&q->requeued[queue]);
	313	qdisc_reset(q->queues[queue]);
	314	}
	315	qd->q.qlen = 0;
	316	}
	317
	318
	319	static void wme_qdiscop_destroy(struct Qdisc* qd)
	320	{
	321	struct ieee80211_sched_data *q = qdisc_priv(qd);
	322	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	323	struct ieee80211_hw *hw = &local->hw;
	324	int queue;
	325
	326	tcf_destroy_chain(q->filter_list);
	327	q->filter_list = NULL;
	328
	329	for (queue=0; queue < hw->queues; queue++) {
	330	skb_queue_purge(&q->requeued[queue]);
	331	qdisc_destroy(q->queues[queue]);
	332	q->queues[queue] = &noop_qdisc;
	333	}
	334	}
	335
	336
	337	/* called whenever parameters are updated on existing qdisc */
1e90474c	338	static int wme_qdiscop_tune(struct Qdisc qd, struct nlattr opt)
f0706e82 JB	339	{
	340	/* struct ieee80211_sched_data *q = qdisc_priv(qd);
	341	*/
	342	/* check our options block is the right size */
	343	/* copy any options to our local structure */
	344	/* Ignore options block for now - always use static mapping
1e90474c	345	struct tc_ieee80211_qopt *qopt = nla_data(opt);
f0706e82	346
1e90474c	347	if (opt->nla_len < nla_attr_size(sizeof(*qopt)))
f0706e82 JB	348	return -EINVAL;
	349	memcpy(q->tag2queue, qopt->tag2queue, sizeof(qopt->tag2queue));
	350	*/
	351	return 0;
	352	}
	353
	354
	355	/* called during initial creation of qdisc on device */
1e90474c	356	static int wme_qdiscop_init(struct Qdisc qd, struct nlattr opt)
f0706e82 JB	357	{
	358	struct ieee80211_sched_data *q = qdisc_priv(qd);
	359	struct net_device *dev = qd->dev;
	360	struct ieee80211_local *local;
	361	int queues;
	362	int err = 0, i;
	363
	364	/* check that device is a mac80211 device */
	365	if (!dev->ieee80211_ptr \|\|
	366	dev->ieee80211_ptr->wiphy->privid != mac80211_wiphy_privid)
	367	return -EINVAL;
	368
	369	/* check this device is an ieee80211 master type device */
	370	if (dev->type != ARPHRD_IEEE80211)
	371	return -EINVAL;
	372
	373	/* check that there is no qdisc currently attached to device
	374	* this ensures that we will be the root qdisc. (I can't find a better
	375	* way to test this explicitly) */
	376	if (dev->qdisc_sleeping != &noop_qdisc)
	377	return -EINVAL;
	378
	379	if (qd->flags & TCQ_F_INGRESS)
	380	return -EINVAL;
	381
	382	local = wdev_priv(dev->ieee80211_ptr);
	383	queues = local->hw.queues;
	384
	385	/* if options were passed in, set them */
	386	if (opt) {
	387	err = wme_qdiscop_tune(qd, opt);
	388	}
	389
	390	/* create child queues */
	391	for (i = 0; i < queues; i++) {
	392	skb_queue_head_init(&q->requeued[i]);
	393	q->queues[i] = qdisc_create_dflt(qd->dev, &pfifo_qdisc_ops,
	394	qd->handle);
136e83d6	395	if (!q->queues[i]) {
f0706e82 JB	396	q->queues[i] = &noop_qdisc;
	397	printk(KERN_ERR "%s child qdisc %i creation failed", dev->name, i);
	398	}
	399	}
	400
9e723492 RR	401	/* reserve all legacy QoS queues */
9e723492 RR	402	for (i = 0; i < min(IEEE80211_TX_QUEUE_DATA4, queues); i++)
a9af2013	403	set_bit(i, q->qdisc_pool);
9e723492	404
f0706e82 JB	405	return err;
	406	}
	407
	408	static int wme_qdiscop_dump(struct Qdisc qd, struct sk_buff skb)
	409	{
	410	/* struct ieee80211_sched_data *q = qdisc_priv(qd);
	411	unsigned char *p = skb->tail;
	412	struct tc_ieee80211_qopt opt;
	413
	414	memcpy(&opt.tag2queue, q->tag2queue, TC_80211_MAX_TAG + 1);
1e90474c	415	NLA_PUT(skb, TCA_OPTIONS, sizeof(opt), &opt);
f0706e82 JB	416	*/ return skb->len;
f0706e82 JB	417	/*
1e90474c	418	nla_put_failure:
f0706e82 JB	419	skb_trim(skb, p - skb->data);*/
	420	return -1;
	421	}
	422
	423
	424	static int wme_classop_graft(struct Qdisc *qd, unsigned long arg,
	425	struct Qdisc new, struct Qdisc *old)
	426	{
	427	struct ieee80211_sched_data *q = qdisc_priv(qd);
	428	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	429	struct ieee80211_hw *hw = &local->hw;
	430	unsigned long queue = arg - 1;
	431
	432	if (queue >= hw->queues)
	433	return -EINVAL;
	434
	435	if (!new)
	436	new = &noop_qdisc;
	437
	438	sch_tree_lock(qd);
	439	*old = q->queues[queue];
	440	q->queues[queue] = new;
	441	qdisc_reset(*old);
	442	sch_tree_unlock(qd);
	443
	444	return 0;
	445	}
	446
	447
	448	static struct Qdisc *
	449	wme_classop_leaf(struct Qdisc *qd, unsigned long arg)
	450	{
	451	struct ieee80211_sched_data *q = qdisc_priv(qd);
	452	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	453	struct ieee80211_hw *hw = &local->hw;
	454	unsigned long queue = arg - 1;
	455
	456	if (queue >= hw->queues)
	457	return NULL;
	458
	459	return q->queues[queue];
	460	}
	461
	462
	463	static unsigned long wme_classop_get(struct Qdisc *qd, u32 classid)
	464	{
	465	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	466	struct ieee80211_hw *hw = &local->hw;
	467	unsigned long queue = TC_H_MIN(classid);
	468
	469	if (queue - 1 >= hw->queues)
	470	return 0;
	471
	472	return queue;
	473	}
	474
	475
	476	static unsigned long wme_classop_bind(struct Qdisc *qd, unsigned long parent,
	477	u32 classid)
	478	{
	479	return wme_classop_get(qd, classid);
	480	}
	481
	482
483	static void wme_classop_put(struct Qdisc *q, unsigned long cl)
484	{
485	}
486
487
488	static int wme_classop_change(struct Qdisc *qd, u32 handle, u32 parent,
1e90474c	489	struct nlattr *tca, unsigned long arg)
f0706e82 JB	490	{
	491	unsigned long cl = *arg;
	492	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	493	struct ieee80211_hw *hw = &local->hw;
	494
	495	if (cl - 1 > hw->queues)
	496	return -ENOENT;
	497
	498	/* TODO: put code to program hardware queue parameters here,
	499	* to allow programming from tc command line */
	500
	501	return 0;
	502	}
	503
	504
	505	/* we don't support deleting hardware queues
	506	* when we add WMM-SA support - TSPECs may be deleted here */
	507	static int wme_classop_delete(struct Qdisc *qd, unsigned long cl)
	508	{
	509	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	510	struct ieee80211_hw *hw = &local->hw;
	511
	512	if (cl - 1 > hw->queues)
	513	return -ENOENT;
	514	return 0;
	515	}
	516
	517
	518	static int wme_classop_dump_class(struct Qdisc *qd, unsigned long cl,
	519	struct sk_buff skb, struct tcmsg tcm)
	520	{
	521	struct ieee80211_sched_data *q = qdisc_priv(qd);
	522	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	523	struct ieee80211_hw *hw = &local->hw;
	524
	525	if (cl - 1 > hw->queues)
	526	return -ENOENT;
	527	tcm->tcm_handle = TC_H_MIN(cl);
	528	tcm->tcm_parent = qd->handle;
	529	tcm->tcm_info = q->queues[cl-1]->handle; /* do we need this? */
	530	return 0;
	531	}
	532
	533
	534	static void wme_classop_walk(struct Qdisc qd, struct qdisc_walker arg)
	535	{
	536	struct ieee80211_local *local = wdev_priv(qd->dev->ieee80211_ptr);
	537	struct ieee80211_hw *hw = &local->hw;
	538	int queue;
	539
	540	if (arg->stop)
	541	return;
	542
	543	for (queue = 0; queue < hw->queues; queue++) {
	544	if (arg->count < arg->skip) {
	545	arg->count++;
	546	continue;
	547	}
	548	/* we should return classids for our internal queues here
	549	* as well as the external ones */
	550	if (arg->fn(qd, queue+1, arg) < 0) {
	551	arg->stop = 1;
	552	break;
	553	}
554	arg->count++;
555	}
556	}
557
558
559	static struct tcf_proto ** wme_classop_find_tcf(struct Qdisc *qd,
560	unsigned long cl)
561	{
562	struct ieee80211_sched_data *q = qdisc_priv(qd);
563
564	if (cl)
565	return NULL;
566
567	return &q->filter_list;
568	}
569
570
571	/* this qdisc is classful (i.e. has classes, some of which may have leaf qdiscs attached)
572	* - these are the operations on the classes */
20fea08b	573	static const struct Qdisc_class_ops class_ops =
f0706e82 JB	574	{
	575	.graft = wme_classop_graft,
	576	.leaf = wme_classop_leaf,
	577
	578	.get = wme_classop_get,
	579	.put = wme_classop_put,
	580	.change = wme_classop_change,
	581	.delete = wme_classop_delete,
	582	.walk = wme_classop_walk,
	583
	584	.tcf_chain = wme_classop_find_tcf,
	585	.bind_tcf = wme_classop_bind,
	586	.unbind_tcf = wme_classop_put,
	587
	588	.dump = wme_classop_dump_class,
	589	};
	590
	591
	592	/* queueing discipline operations */
20fea08b	593	static struct Qdisc_ops wme_qdisc_ops __read_mostly =
f0706e82 JB	594	{
	595	.next = NULL,
	596	.cl_ops = &class_ops,
	597	.id = "ieee80211",
	598	.priv_size = sizeof(struct ieee80211_sched_data),
	599
	600	.enqueue = wme_qdiscop_enqueue,
	601	.dequeue = wme_qdiscop_dequeue,
	602	.requeue = wme_qdiscop_requeue,
	603	.drop = NULL, /* drop not needed since we are always the root qdisc */
	604
	605	.init = wme_qdiscop_init,
	606	.reset = wme_qdiscop_reset,
	607	.destroy = wme_qdiscop_destroy,
	608	.change = wme_qdiscop_tune,
	609
	610	.dump = wme_qdiscop_dump,
	611	};
	612
	613
	614	void ieee80211_install_qdisc(struct net_device *dev)
	615	{
	616	struct Qdisc *qdisc;
	617
	618	qdisc = qdisc_create_dflt(dev, &wme_qdisc_ops, TC_H_ROOT);
	619	if (!qdisc) {
	620	printk(KERN_ERR "%s: qdisc installation failed\n", dev->name);
	621	return;
	622	}
	623
	624	/* same handle as would be allocated by qdisc_alloc_handle() */
	625	qdisc->handle = 0x80010000;
	626
	627	qdisc_lock_tree(dev);
	628	list_add_tail(&qdisc->list, &dev->qdisc_list);
	629	dev->qdisc_sleeping = qdisc;
	630	qdisc_unlock_tree(dev);
	631	}
	632
	633
	634	int ieee80211_qdisc_installed(struct net_device *dev)
	635	{
	636	return dev->qdisc_sleeping->ops == &wme_qdisc_ops;
	637	}
	638
	639
	640	int ieee80211_wme_register(void)
	641	{
	642	return register_qdisc(&wme_qdisc_ops);
	643	}
	644
	645
	646	void ieee80211_wme_unregister(void)
	647	{
	648	unregister_qdisc(&wme_qdisc_ops);
	649	}
9e723492 RR	650
	651	int ieee80211_ht_agg_queue_add(struct ieee80211_local *local,
	652	struct sta_info *sta, u16 tid)
	653	{
	654	int i;
	655	struct ieee80211_sched_data *q =
	656	qdisc_priv(local->mdev->qdisc_sleeping);
	657	DECLARE_MAC_BUF(mac);
	658
	659	/* prepare the filter and save it for the SW queue
	660	* matching the recieved HW queue */
	661
	662	/* try to get a Qdisc from the pool */
	663	for (i = IEEE80211_TX_QUEUE_BEACON; i < local->hw.queues; i++)
a9af2013	664	if (!test_and_set_bit(i, q->qdisc_pool)) {
9e723492 RR	665	ieee80211_stop_queue(local_to_hw(local), i);
	666	sta->tid_to_tx_q[tid] = i;
	667
	668	/* IF there are already pending packets
	669	* on this tid first we need to drain them
	670	* on the previous queue
	671	* since HT is strict in order */
	672	#ifdef CONFIG_MAC80211_HT_DEBUG
	673	if (net_ratelimit())
	674	printk(KERN_DEBUG "allocated aggregation queue"
a9af2013	675	" %d tid %d addr %s pool=0x%lX",
9e723492	676	i, tid, print_mac(mac, sta->addr),
a9af2013	677	q->qdisc_pool[0]);
9e723492 RR	678	#endif /* CONFIG_MAC80211_HT_DEBUG */
	679	return 0;
	680	}
	681
	682	return -EAGAIN;
	683	}
	684
	685	/**
	686	* the caller needs to hold local->mdev->queue_lock
	687	*/
	688	void ieee80211_ht_agg_queue_remove(struct ieee80211_local *local,
	689	struct sta_info *sta, u16 tid,
	690	u8 requeue)
	691	{
	692	struct ieee80211_sched_data *q =
	693	qdisc_priv(local->mdev->qdisc_sleeping);
	694	int agg_queue = sta->tid_to_tx_q[tid];
	695
	696	/* return the qdisc to the pool */
a9af2013	697	clear_bit(agg_queue, q->qdisc_pool);
9e723492 RR	698	sta->tid_to_tx_q[tid] = local->hw.queues;
	699
	700	if (requeue)
	701	ieee80211_requeue(local, agg_queue);
	702	else
	703	q->queues[agg_queue]->ops->reset(q->queues[agg_queue]);
	704	}
	705
	706	void ieee80211_requeue(struct ieee80211_local *local, int queue)
	707	{
	708	struct Qdisc *root_qd = local->mdev->qdisc_sleeping;
	709	struct ieee80211_sched_data *q = qdisc_priv(root_qd);
	710	struct Qdisc *qdisc = q->queues[queue];
	711	struct sk_buff *skb = NULL;
	712	u32 len = qdisc->q.qlen;
	713
	714	if (!qdisc \|\| !qdisc->dequeue)
	715	return;
	716
	717	printk(KERN_DEBUG "requeue: qlen = %d\n", qdisc->q.qlen);
	718	for (len = qdisc->q.qlen; len > 0; len--) {
	719	skb = qdisc->dequeue(qdisc);
	720	root_qd->q.qlen--;
	721	/* packet will be classified again and */
	722	/* skb->packet_data->queue will be overridden if needed */
	723	if (skb)
	724	wme_qdiscop_enqueue(skb, root_qd);
	725	}
	726	}