[mirror_ubuntu-jammy-kernel.git] / drivers / net / wireless / ath / ath9k / wmi.c

/*
 * Copyright (c) 2010-2011 Atheros Communications Inc.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include "htc.h"

static const char *wmi_cmd_to_name(enum wmi_cmd_id wmi_cmd)
{
	switch (wmi_cmd) {
	case WMI_ECHO_CMDID:
		return "WMI_ECHO_CMDID";
	case WMI_ACCESS_MEMORY_CMDID:
		return "WMI_ACCESS_MEMORY_CMDID";
	case WMI_GET_FW_VERSION:
		return "WMI_GET_FW_VERSION";
	case WMI_DISABLE_INTR_CMDID:
		return "WMI_DISABLE_INTR_CMDID";
	case WMI_ENABLE_INTR_CMDID:
		return "WMI_ENABLE_INTR_CMDID";
	case WMI_ATH_INIT_CMDID:
		return "WMI_ATH_INIT_CMDID";
	case WMI_ABORT_TXQ_CMDID:
		return "WMI_ABORT_TXQ_CMDID";
	case WMI_STOP_TX_DMA_CMDID:
		return "WMI_STOP_TX_DMA_CMDID";
	case WMI_ABORT_TX_DMA_CMDID:
		return "WMI_ABORT_TX_DMA_CMDID";
	case WMI_DRAIN_TXQ_CMDID:
		return "WMI_DRAIN_TXQ_CMDID";
	case WMI_DRAIN_TXQ_ALL_CMDID:
		return "WMI_DRAIN_TXQ_ALL_CMDID";
	case WMI_START_RECV_CMDID:
		return "WMI_START_RECV_CMDID";
	case WMI_STOP_RECV_CMDID:
		return "WMI_STOP_RECV_CMDID";
	case WMI_FLUSH_RECV_CMDID:
		return "WMI_FLUSH_RECV_CMDID";
	case WMI_SET_MODE_CMDID:
		return "WMI_SET_MODE_CMDID";
	case WMI_NODE_CREATE_CMDID:
		return "WMI_NODE_CREATE_CMDID";
	case WMI_NODE_REMOVE_CMDID:
		return "WMI_NODE_REMOVE_CMDID";
	case WMI_VAP_REMOVE_CMDID:
		return "WMI_VAP_REMOVE_CMDID";
	case WMI_VAP_CREATE_CMDID:
		return "WMI_VAP_CREATE_CMDID";
	case WMI_REG_READ_CMDID:
		return "WMI_REG_READ_CMDID";
	case WMI_REG_WRITE_CMDID:
		return "WMI_REG_WRITE_CMDID";
	case WMI_REG_RMW_CMDID:
		return "WMI_REG_RMW_CMDID";
	case WMI_RC_STATE_CHANGE_CMDID:
		return "WMI_RC_STATE_CHANGE_CMDID";
	case WMI_RC_RATE_UPDATE_CMDID:
		return "WMI_RC_RATE_UPDATE_CMDID";
	case WMI_TARGET_IC_UPDATE_CMDID:
		return "WMI_TARGET_IC_UPDATE_CMDID";
	case WMI_TX_AGGR_ENABLE_CMDID:
		return "WMI_TX_AGGR_ENABLE_CMDID";
	case WMI_TGT_DETACH_CMDID:
		return "WMI_TGT_DETACH_CMDID";
	case WMI_NODE_UPDATE_CMDID:
		return "WMI_NODE_UPDATE_CMDID";
	case WMI_INT_STATS_CMDID:
		return "WMI_INT_STATS_CMDID";
	case WMI_TX_STATS_CMDID:
		return "WMI_TX_STATS_CMDID";
	case WMI_RX_STATS_CMDID:
		return "WMI_RX_STATS_CMDID";
	case WMI_BITRATE_MASK_CMDID:
		return "WMI_BITRATE_MASK_CMDID";
	}

	return "Bogus";
}

struct wmi *ath9k_init_wmi(struct ath9k_htc_priv *priv)
{
	struct wmi *wmi;

	wmi = kzalloc(sizeof(struct wmi), GFP_KERNEL);
	if (!wmi)
		return NULL;

	wmi->drv_priv = priv;
	wmi->stopped = false;
	skb_queue_head_init(&wmi->wmi_event_queue);
	spin_lock_init(&wmi->wmi_lock);
	spin_lock_init(&wmi->event_lock);
	mutex_init(&wmi->op_mutex);
	mutex_init(&wmi->multi_write_mutex);
	mutex_init(&wmi->multi_rmw_mutex);
	init_completion(&wmi->cmd_wait);
	INIT_LIST_HEAD(&wmi->pending_tx_events);
	tasklet_setup(&wmi->wmi_event_tasklet, ath9k_wmi_event_tasklet);

	return wmi;
}

void ath9k_stop_wmi(struct ath9k_htc_priv *priv)
{
	struct wmi *wmi = priv->wmi;

	mutex_lock(&wmi->op_mutex);
	wmi->stopped = true;
	mutex_unlock(&wmi->op_mutex);
}

void ath9k_destroy_wmi(struct ath9k_htc_priv *priv)
{
	kfree(priv->wmi);
}

void ath9k_wmi_event_drain(struct ath9k_htc_priv *priv)
{
	unsigned long flags;

	tasklet_kill(&priv->wmi->wmi_event_tasklet);
	spin_lock_irqsave(&priv->wmi->wmi_lock, flags);
	__skb_queue_purge(&priv->wmi->wmi_event_queue);
	spin_unlock_irqrestore(&priv->wmi->wmi_lock, flags);
}

void ath9k_wmi_event_tasklet(struct tasklet_struct *t)
{
	struct wmi *wmi = from_tasklet(wmi, t, wmi_event_tasklet);
	struct ath9k_htc_priv *priv = wmi->drv_priv;
	struct wmi_cmd_hdr *hdr;
	void *wmi_event;
	struct wmi_event_swba *swba;
	struct sk_buff *skb = NULL;
	unsigned long flags;
	u16 cmd_id;

	do {
		spin_lock_irqsave(&wmi->wmi_lock, flags);
		skb = __skb_dequeue(&wmi->wmi_event_queue);
		if (!skb) {
			spin_unlock_irqrestore(&wmi->wmi_lock, flags);
			return;
		}
		spin_unlock_irqrestore(&wmi->wmi_lock, flags);

		hdr = (struct wmi_cmd_hdr *) skb->data;
		cmd_id = be16_to_cpu(hdr->command_id);
		wmi_event = skb_pull(skb, sizeof(struct wmi_cmd_hdr));

		switch (cmd_id) {
		case WMI_SWBA_EVENTID:
			swba = wmi_event;
			ath9k_htc_swba(priv, swba);
			break;
		case WMI_FATAL_EVENTID:
			ieee80211_queue_work(wmi->drv_priv->hw,
					     &wmi->drv_priv->fatal_work);
			break;
		case WMI_TXSTATUS_EVENTID:
			/* Check if ath9k_tx_init() completed. */
			if (!data_race(priv->tx.initialized))
				break;

			spin_lock_bh(&priv->tx.tx_lock);
			if (priv->tx.flags & ATH9K_HTC_OP_TX_DRAIN) {
				spin_unlock_bh(&priv->tx.tx_lock);
				break;
			}
			spin_unlock_bh(&priv->tx.tx_lock);

			ath9k_htc_txstatus(priv, wmi_event);
			break;
		default:
			break;
		}

		kfree_skb(skb);
	} while (1);
}

void ath9k_fatal_work(struct work_struct *work)
{
	struct ath9k_htc_priv *priv = container_of(work, struct ath9k_htc_priv,
						   fatal_work);
	struct ath_common *common = ath9k_hw_common(priv->ah);

	ath_dbg(common, FATAL, "FATAL Event received, resetting device\n");
	ath9k_htc_reset(priv);
}

static void ath9k_wmi_rsp_callback(struct wmi *wmi, struct sk_buff *skb)
{
	skb_pull(skb, sizeof(struct wmi_cmd_hdr));

	if (wmi->cmd_rsp_buf != NULL && wmi->cmd_rsp_len != 0)
		memcpy(wmi->cmd_rsp_buf, skb->data, wmi->cmd_rsp_len);

	complete(&wmi->cmd_wait);
}

static void ath9k_wmi_ctrl_rx(void *priv, struct sk_buff *skb,
			      enum htc_endpoint_id epid)
{
	struct wmi *wmi = priv;
	struct wmi_cmd_hdr *hdr;
	unsigned long flags;
	u16 cmd_id;

	if (unlikely(wmi->stopped))
		goto free_skb;

	hdr = (struct wmi_cmd_hdr *) skb->data;
	cmd_id = be16_to_cpu(hdr->command_id);

	if (cmd_id & 0x1000) {
		spin_lock_irqsave(&wmi->wmi_lock, flags);
		__skb_queue_tail(&wmi->wmi_event_queue, skb);
		spin_unlock_irqrestore(&wmi->wmi_lock, flags);
		tasklet_schedule(&wmi->wmi_event_tasklet);
		return;
	}

	/* Check if there has been a timeout. */
	spin_lock_irqsave(&wmi->wmi_lock, flags);
	if (be16_to_cpu(hdr->seq_no) != wmi->last_seq_id) {
		spin_unlock_irqrestore(&wmi->wmi_lock, flags);
		goto free_skb;
	}
	spin_unlock_irqrestore(&wmi->wmi_lock, flags);

	/* WMI command response */
	ath9k_wmi_rsp_callback(wmi, skb);

free_skb:
	kfree_skb(skb);
}

static void ath9k_wmi_ctrl_tx(void *priv, struct sk_buff *skb,
			      enum htc_endpoint_id epid, bool txok)
{
	kfree_skb(skb);
}

int ath9k_wmi_connect(struct htc_target *htc, struct wmi *wmi,
		      enum htc_endpoint_id *wmi_ctrl_epid)
{
	struct htc_service_connreq connect;
	int ret;

	wmi->htc = htc;

	memset(&connect, 0, sizeof(connect));

	connect.ep_callbacks.priv = wmi;
	connect.ep_callbacks.tx = ath9k_wmi_ctrl_tx;
	connect.ep_callbacks.rx = ath9k_wmi_ctrl_rx;
	connect.service_id = WMI_CONTROL_SVC;

	ret = htc_connect_service(htc, &connect, &wmi->ctrl_epid);
	if (ret)
		return ret;

	*wmi_ctrl_epid = wmi->ctrl_epid;

	return 0;
}

static int ath9k_wmi_cmd_issue(struct wmi *wmi,
			       struct sk_buff *skb,
			       enum wmi_cmd_id cmd, u16 len)
{
	struct wmi_cmd_hdr *hdr;
	unsigned long flags;

	hdr = skb_push(skb, sizeof(struct wmi_cmd_hdr));
	hdr->command_id = cpu_to_be16(cmd);
	hdr->seq_no = cpu_to_be16(++wmi->tx_seq_id);

	spin_lock_irqsave(&wmi->wmi_lock, flags);
	wmi->last_seq_id = wmi->tx_seq_id;
	spin_unlock_irqrestore(&wmi->wmi_lock, flags);

	return htc_send_epid(wmi->htc, skb, wmi->ctrl_epid);
}

int ath9k_wmi_cmd(struct wmi *wmi, enum wmi_cmd_id cmd_id,
		  u8 *cmd_buf, u32 cmd_len,
		  u8 *rsp_buf, u32 rsp_len,
		  u32 timeout)
{
	struct ath_hw *ah = wmi->drv_priv->ah;
	struct ath_common *common = ath9k_hw_common(ah);
	u16 headroom = sizeof(struct htc_frame_hdr) +
		       sizeof(struct wmi_cmd_hdr);
	struct sk_buff *skb;
	unsigned long time_left;
	int ret = 0;

	if (ah->ah_flags & AH_UNPLUGGED)
		return 0;

	skb = alloc_skb(headroom + cmd_len, GFP_ATOMIC);
	if (!skb)
		return -ENOMEM;

	skb_reserve(skb, headroom);

	if (cmd_len != 0 && cmd_buf != NULL) {
		skb_put_data(skb, cmd_buf, cmd_len);
	}

	mutex_lock(&wmi->op_mutex);

	/* check if wmi stopped flag is set */
	if (unlikely(wmi->stopped)) {
		ret = -EPROTO;
		goto out;
	}

	/* record the rsp buffer and length */
	wmi->cmd_rsp_buf = rsp_buf;
	wmi->cmd_rsp_len = rsp_len;

	ret = ath9k_wmi_cmd_issue(wmi, skb, cmd_id, cmd_len);
	if (ret)
		goto out;

	time_left = wait_for_completion_timeout(&wmi->cmd_wait, timeout);
	if (!time_left) {
		ath_dbg(common, WMI, "Timeout waiting for WMI command: %s\n",
			wmi_cmd_to_name(cmd_id));
		mutex_unlock(&wmi->op_mutex);
		return -ETIMEDOUT;
	}

	mutex_unlock(&wmi->op_mutex);

	return 0;

out:
	ath_dbg(common, WMI, "WMI failure for: %s\n", wmi_cmd_to_name(cmd_id));
	mutex_unlock(&wmi->op_mutex);
	kfree_skb(skb);

	return ret;
}
Commit	Line	Data
fb9987d0	1	/*
5b68138e	2	* Copyright (c) 2010-2011 Atheros Communications Inc.
fb9987d0 S	3	*
	4	* Permission to use, copy, modify, and/or distribute this software for any
	5	* purpose with or without fee is hereby granted, provided that the above
	6	* copyright notice and this permission notice appear in all copies.
	7	*
	8	* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
	9	* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
	10	* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
	11	* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
	12	* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
	13	* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
	14	* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
	15	*/
	16
	17	#include "htc.h"
	18
	19	static const char *wmi_cmd_to_name(enum wmi_cmd_id wmi_cmd)
	20	{
	21	switch (wmi_cmd) {
	22	case WMI_ECHO_CMDID:
	23	return "WMI_ECHO_CMDID";
	24	case WMI_ACCESS_MEMORY_CMDID:
	25	return "WMI_ACCESS_MEMORY_CMDID";
29bbfb24 SM	26	case WMI_GET_FW_VERSION:
29bbfb24 SM	27	return "WMI_GET_FW_VERSION";
fb9987d0 S	28	case WMI_DISABLE_INTR_CMDID:
	29	return "WMI_DISABLE_INTR_CMDID";
	30	case WMI_ENABLE_INTR_CMDID:
	31	return "WMI_ENABLE_INTR_CMDID";
fb9987d0 S	32	case WMI_ATH_INIT_CMDID:
	33	return "WMI_ATH_INIT_CMDID";
	34	case WMI_ABORT_TXQ_CMDID:
	35	return "WMI_ABORT_TXQ_CMDID";
	36	case WMI_STOP_TX_DMA_CMDID:
	37	return "WMI_STOP_TX_DMA_CMDID";
fb9987d0 S	38	case WMI_ABORT_TX_DMA_CMDID:
	39	return "WMI_ABORT_TX_DMA_CMDID";
	40	case WMI_DRAIN_TXQ_CMDID:
	41	return "WMI_DRAIN_TXQ_CMDID";
	42	case WMI_DRAIN_TXQ_ALL_CMDID:
	43	return "WMI_DRAIN_TXQ_ALL_CMDID";
	44	case WMI_START_RECV_CMDID:
	45	return "WMI_START_RECV_CMDID";
	46	case WMI_STOP_RECV_CMDID:
	47	return "WMI_STOP_RECV_CMDID";
	48	case WMI_FLUSH_RECV_CMDID:
	49	return "WMI_FLUSH_RECV_CMDID";
	50	case WMI_SET_MODE_CMDID:
	51	return "WMI_SET_MODE_CMDID";
fb9987d0 S	52	case WMI_NODE_CREATE_CMDID:
	53	return "WMI_NODE_CREATE_CMDID";
	54	case WMI_NODE_REMOVE_CMDID:
	55	return "WMI_NODE_REMOVE_CMDID";
	56	case WMI_VAP_REMOVE_CMDID:
	57	return "WMI_VAP_REMOVE_CMDID";
	58	case WMI_VAP_CREATE_CMDID:
	59	return "WMI_VAP_CREATE_CMDID";
fb9987d0 S	60	case WMI_REG_READ_CMDID:
	61	return "WMI_REG_READ_CMDID";
	62	case WMI_REG_WRITE_CMDID:
	63	return "WMI_REG_WRITE_CMDID";
8badb50c OR	64	case WMI_REG_RMW_CMDID:
8badb50c OR	65	return "WMI_REG_RMW_CMDID";
fb9987d0 S	66	case WMI_RC_STATE_CHANGE_CMDID:
	67	return "WMI_RC_STATE_CHANGE_CMDID";
	68	case WMI_RC_RATE_UPDATE_CMDID:
	69	return "WMI_RC_RATE_UPDATE_CMDID";
fb9987d0 S	70	case WMI_TARGET_IC_UPDATE_CMDID:
fb9987d0 S	71	return "WMI_TARGET_IC_UPDATE_CMDID";
fb9987d0 S	72	case WMI_TX_AGGR_ENABLE_CMDID:
	73	return "WMI_TX_AGGR_ENABLE_CMDID";
	74	case WMI_TGT_DETACH_CMDID:
	75	return "WMI_TGT_DETACH_CMDID";
fbc29d6c SM	76	case WMI_NODE_UPDATE_CMDID:
	77	return "WMI_NODE_UPDATE_CMDID";
	78	case WMI_INT_STATS_CMDID:
	79	return "WMI_INT_STATS_CMDID";
	80	case WMI_TX_STATS_CMDID:
	81	return "WMI_TX_STATS_CMDID";
	82	case WMI_RX_STATS_CMDID:
	83	return "WMI_RX_STATS_CMDID";
a55bb94a SM	84	case WMI_BITRATE_MASK_CMDID:
a55bb94a SM	85	return "WMI_BITRATE_MASK_CMDID";
fb9987d0 S	86	}
	87
	88	return "Bogus";
	89	}
	90
	91	struct wmi ath9k_init_wmi(struct ath9k_htc_priv priv)
	92	{
	93	struct wmi *wmi;
	94
	95	wmi = kzalloc(sizeof(struct wmi), GFP_KERNEL);
	96	if (!wmi)
	97	return NULL;
	98
	99	wmi->drv_priv = priv;
	100	wmi->stopped = false;
f4c88991	101	skb_queue_head_init(&wmi->wmi_event_queue);
859c3ca1 SM	102	spin_lock_init(&wmi->wmi_lock);
859c3ca1 SM	103	spin_lock_init(&wmi->event_lock);
fb9987d0	104	mutex_init(&wmi->op_mutex);
4a22fe10	105	mutex_init(&wmi->multi_write_mutex);
8badb50c	106	mutex_init(&wmi->multi_rmw_mutex);
fb9987d0	107	init_completion(&wmi->cmd_wait);
859c3ca1	108	INIT_LIST_HEAD(&wmi->pending_tx_events);
f91a35b4	109	tasklet_setup(&wmi->wmi_event_tasklet, ath9k_wmi_event_tasklet);
fb9987d0 S	110
	111	return wmi;
	112	}
	113
abeaa850	114	void ath9k_stop_wmi(struct ath9k_htc_priv *priv)
fb9987d0 S	115	{
	116	struct wmi *wmi = priv->wmi;
	117
	118	mutex_lock(&wmi->op_mutex);
	119	wmi->stopped = true;
	120	mutex_unlock(&wmi->op_mutex);
abeaa850	121	}
fb9987d0	122
743adae9	123	void ath9k_destroy_wmi(struct ath9k_htc_priv *priv)
abeaa850	124	{
fb9987d0 S	125	kfree(priv->wmi);
	126	}
	127
f4c88991	128	void ath9k_wmi_event_drain(struct ath9k_htc_priv *priv)
fb9987d0	129	{
f4c88991	130	unsigned long flags;
fb9987d0	131
f4c88991 SM	132	tasklet_kill(&priv->wmi->wmi_event_tasklet);
	133	spin_lock_irqsave(&priv->wmi->wmi_lock, flags);
	134	__skb_queue_purge(&priv->wmi->wmi_event_queue);
	135	spin_unlock_irqrestore(&priv->wmi->wmi_lock, flags);
	136	}
	137
f91a35b4	138	void ath9k_wmi_event_tasklet(struct tasklet_struct *t)
f4c88991	139	{
f91a35b4	140	struct wmi *wmi = from_tasklet(wmi, t, wmi_event_tasklet);
f4c88991 SM	141	struct ath9k_htc_priv *priv = wmi->drv_priv;
	142	struct wmi_cmd_hdr *hdr;
	143	void *wmi_event;
	144	struct wmi_event_swba *swba;
	145	struct sk_buff *skb = NULL;
	146	unsigned long flags;
	147	u16 cmd_id;
f4c88991 SM	148
	149	do {
	150	spin_lock_irqsave(&wmi->wmi_lock, flags);
	151	skb = __skb_dequeue(&wmi->wmi_event_queue);
	152	if (!skb) {
	153	spin_unlock_irqrestore(&wmi->wmi_lock, flags);
	154	return;
	155	}
	156	spin_unlock_irqrestore(&wmi->wmi_lock, flags);
	157
	158	hdr = (struct wmi_cmd_hdr *) skb->data;
	159	cmd_id = be16_to_cpu(hdr->command_id);
	160	wmi_event = skb_pull(skb, sizeof(struct wmi_cmd_hdr));
	161
	162	switch (cmd_id) {
	163	case WMI_SWBA_EVENTID:
50c8cd44	164	swba = wmi_event;
f4c88991 SM	165	ath9k_htc_swba(priv, swba);
	166	break;
	167	case WMI_FATAL_EVENTID:
	168	ieee80211_queue_work(wmi->drv_priv->hw,
	169	&wmi->drv_priv->fatal_work);
	170	break;
27876a29	171	case WMI_TXSTATUS_EVENTID:
a429b03e TH	172	/* Check if ath9k_tx_init() completed. */
	173	if (!data_race(priv->tx.initialized))
	174	break;
	175
27876a29 SM	176	spin_lock_bh(&priv->tx.tx_lock);
	177	if (priv->tx.flags & ATH9K_HTC_OP_TX_DRAIN) {
	178	spin_unlock_bh(&priv->tx.tx_lock);
	179	break;
	180	}
	181	spin_unlock_bh(&priv->tx.tx_lock);
	182
	183	ath9k_htc_txstatus(priv, wmi_event);
	184	break;
f4c88991 SM	185	default:
	186	break;
	187	}
	188
	189	kfree_skb(skb);
	190	} while (1);
fb9987d0 S	191	}
fb9987d0 S	192
73908674 SM	193	void ath9k_fatal_work(struct work_struct *work)
	194	{
	195	struct ath9k_htc_priv *priv = container_of(work, struct ath9k_htc_priv,
	196	fatal_work);
	197	struct ath_common *common = ath9k_hw_common(priv->ah);
	198
d2182b69	199	ath_dbg(common, FATAL, "FATAL Event received, resetting device\n");
73908674 SM	200	ath9k_htc_reset(priv);
	201	}
	202
fb9987d0 S	203	static void ath9k_wmi_rsp_callback(struct wmi wmi, struct sk_buff skb)
	204	{
	205	skb_pull(skb, sizeof(struct wmi_cmd_hdr));
	206
	207	if (wmi->cmd_rsp_buf != NULL && wmi->cmd_rsp_len != 0)
	208	memcpy(wmi->cmd_rsp_buf, skb->data, wmi->cmd_rsp_len);
	209
	210	complete(&wmi->cmd_wait);
	211	}
	212
	213	static void ath9k_wmi_ctrl_rx(void priv, struct sk_buff skb,
	214	enum htc_endpoint_id epid)
	215	{
50c8cd44	216	struct wmi *wmi = priv;
fb9987d0	217	struct wmi_cmd_hdr *hdr;
84a0d466	218	unsigned long flags;
fb9987d0 S	219	u16 cmd_id;
	220
	221	if (unlikely(wmi->stopped))
	222	goto free_skb;
	223
	224	hdr = (struct wmi_cmd_hdr *) skb->data;
	225	cmd_id = be16_to_cpu(hdr->command_id);
	226
	227	if (cmd_id & 0x1000) {
84a0d466	228	spin_lock_irqsave(&wmi->wmi_lock, flags);
f4c88991	229	__skb_queue_tail(&wmi->wmi_event_queue, skb);
84a0d466	230	spin_unlock_irqrestore(&wmi->wmi_lock, flags);
f4c88991	231	tasklet_schedule(&wmi->wmi_event_tasklet);
fb9987d0 S	232	return;
	233	}
	234
6ce34ec1	235	/* Check if there has been a timeout. */
84a0d466	236	spin_lock_irqsave(&wmi->wmi_lock, flags);
00ca1803	237	if (be16_to_cpu(hdr->seq_no) != wmi->last_seq_id) {
84a0d466	238	spin_unlock_irqrestore(&wmi->wmi_lock, flags);
6ce34ec1 S	239	goto free_skb;
6ce34ec1 S	240	}
84a0d466	241	spin_unlock_irqrestore(&wmi->wmi_lock, flags);
6ce34ec1	242
fb9987d0 S	243	/* WMI command response */
	244	ath9k_wmi_rsp_callback(wmi, skb);
	245
	246	free_skb:
e6c6d33c	247	kfree_skb(skb);
fb9987d0 S	248	}
	249
	250	static void ath9k_wmi_ctrl_tx(void priv, struct sk_buff skb,
	251	enum htc_endpoint_id epid, bool txok)
	252	{
0fa35a58	253	kfree_skb(skb);
fb9987d0 S	254	}
	255
	256	int ath9k_wmi_connect(struct htc_target htc, struct wmi wmi,
	257	enum htc_endpoint_id *wmi_ctrl_epid)
	258	{
	259	struct htc_service_connreq connect;
	260	int ret;
	261
	262	wmi->htc = htc;
	263
	264	memset(&connect, 0, sizeof(connect));
	265
	266	connect.ep_callbacks.priv = wmi;
	267	connect.ep_callbacks.tx = ath9k_wmi_ctrl_tx;
	268	connect.ep_callbacks.rx = ath9k_wmi_ctrl_rx;
	269	connect.service_id = WMI_CONTROL_SVC;
	270
	271	ret = htc_connect_service(htc, &connect, &wmi->ctrl_epid);
	272	if (ret)
	273	return ret;
	274
	275	*wmi_ctrl_epid = wmi->ctrl_epid;
	276
	277	return 0;
	278	}
	279
	280	static int ath9k_wmi_cmd_issue(struct wmi *wmi,
	281	struct sk_buff *skb,
	282	enum wmi_cmd_id cmd, u16 len)
	283	{
	284	struct wmi_cmd_hdr *hdr;
00ca1803	285	unsigned long flags;
fb9987d0	286
d58ff351	287	hdr = skb_push(skb, sizeof(struct wmi_cmd_hdr));
fb9987d0 S	288	hdr->command_id = cpu_to_be16(cmd);
	289	hdr->seq_no = cpu_to_be16(++wmi->tx_seq_id);
	290
00ca1803 FC	291	spin_lock_irqsave(&wmi->wmi_lock, flags);
	292	wmi->last_seq_id = wmi->tx_seq_id;
	293	spin_unlock_irqrestore(&wmi->wmi_lock, flags);
	294
d67ee533	295	return htc_send_epid(wmi->htc, skb, wmi->ctrl_epid);
fb9987d0 S	296	}
	297
	298	int ath9k_wmi_cmd(struct wmi *wmi, enum wmi_cmd_id cmd_id,
	299	u8 *cmd_buf, u32 cmd_len,
	300	u8 *rsp_buf, u32 rsp_len,
	301	u32 timeout)
	302	{
	303	struct ath_hw *ah = wmi->drv_priv->ah;
	304	struct ath_common *common = ath9k_hw_common(ah);
	305	u16 headroom = sizeof(struct htc_frame_hdr) +
	306	sizeof(struct wmi_cmd_hdr);
	307	struct sk_buff *skb;
61fc3920 NMG	308	unsigned long time_left;
61fc3920 NMG	309	int ret = 0;
fb9987d0	310
97dcec57	311	if (ah->ah_flags & AH_UNPLUGGED)
a3be14b7 S	312	return 0;
a3be14b7 S	313
0fa35a58	314	skb = alloc_skb(headroom + cmd_len, GFP_ATOMIC);
fb9987d0 S	315	if (!skb)
	316	return -ENOMEM;
	317
	318	skb_reserve(skb, headroom);
	319
	320	if (cmd_len != 0 && cmd_buf != NULL) {
b952f4df	321	skb_put_data(skb, cmd_buf, cmd_len);
fb9987d0 S	322	}
	323
	324	mutex_lock(&wmi->op_mutex);
	325
	326	/* check if wmi stopped flag is set */
	327	if (unlikely(wmi->stopped)) {
	328	ret = -EPROTO;
	329	goto out;
	330	}
	331
	332	/* record the rsp buffer and length */
	333	wmi->cmd_rsp_buf = rsp_buf;
	334	wmi->cmd_rsp_len = rsp_len;
	335
0d36d71d S	336	ret = ath9k_wmi_cmd_issue(wmi, skb, cmd_id, cmd_len);
	337	if (ret)
	338	goto out;
	339
fb9987d0 S	340	time_left = wait_for_completion_timeout(&wmi->cmd_wait, timeout);
fb9987d0 S	341	if (!time_left) {
d2182b69	342	ath_dbg(common, WMI, "Timeout waiting for WMI command: %s\n",
226afe68	343	wmi_cmd_to_name(cmd_id));
fb9987d0 S	344	mutex_unlock(&wmi->op_mutex);
	345	return -ETIMEDOUT;
	346	}
	347
	348	mutex_unlock(&wmi->op_mutex);
	349
	350	return 0;
	351
	352	out:
d2182b69	353	ath_dbg(common, WMI, "WMI failure for: %s\n", wmi_cmd_to_name(cmd_id));
fb9987d0	354	mutex_unlock(&wmi->op_mutex);
0fa35a58	355	kfree_skb(skb);
fb9987d0 S	356
	357	return ret;
	358	}