/*
 * 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.
 */

#ifndef WMI_H
#define WMI_H

struct wmi_event_txrate {
 __be32 txrate;
 struct {
  u8 rssi_thresh;
  u8 per;
 } rc_stats;
} __packed;

struct wmi_cmd_hdr {
 __be16 command_id;
 __be16 seq_no;
} __packed;

struct wmi_fw_version {
 __be16 major;
 __be16 minor;

} __packed;

struct wmi_event_swba {
 __be64 tsf;
 u8 beacon_pending;
} __packed;

/*
 * 64 - HTC header - WMI header - 1 / txstatus
 * And some other hdr. space is also accounted for.
 * 12 seems to be the magic number.
 */
#define HTC_MAX_TX_STATUS 12

#define ATH9K_HTC_TXSTAT_ACK        BIT(0)
#define ATH9K_HTC_TXSTAT_FILT       BIT(1)
#define ATH9K_HTC_TXSTAT_RTC_CTS    BIT(2)
#define ATH9K_HTC_TXSTAT_MCS        BIT(3)
#define ATH9K_HTC_TXSTAT_CW40       BIT(4)
#define ATH9K_HTC_TXSTAT_SGI        BIT(5)

/*
 * Legacy rates are indicated as indices.
 * HT rates are indicated as dot11 numbers.
 * This allows us to resrict the rate field
 * to 4 bits.
 */
#define ATH9K_HTC_TXSTAT_RATE       0x0f
#define ATH9K_HTC_TXSTAT_RATE_S     0

#define ATH9K_HTC_TXSTAT_EPID       0xf0
#define ATH9K_HTC_TXSTAT_EPID_S     4

struct __wmi_event_txstatus {
 u8 cookie;
 u8 ts_rate; /* Also holds EP ID */
 u8 ts_flags;
};

struct wmi_event_txstatus {
 u8 cnt;
 struct __wmi_event_txstatus txstatus[HTC_MAX_TX_STATUS];
} __packed;

enum wmi_cmd_id {
 WMI_ECHO_CMDID = 0x0001,
 WMI_ACCESS_MEMORY_CMDID,

 /* Commands to Target */
 WMI_GET_FW_VERSION,
 WMI_DISABLE_INTR_CMDID,
 WMI_ENABLE_INTR_CMDID,
 WMI_ATH_INIT_CMDID,
 WMI_ABORT_TXQ_CMDID,
 WMI_STOP_TX_DMA_CMDID,
 WMI_ABORT_TX_DMA_CMDID,
 WMI_DRAIN_TXQ_CMDID,
 WMI_DRAIN_TXQ_ALL_CMDID,
 WMI_START_RECV_CMDID,
 WMI_STOP_RECV_CMDID,
 WMI_FLUSH_RECV_CMDID,
 WMI_SET_MODE_CMDID,
 WMI_NODE_CREATE_CMDID,
 WMI_NODE_REMOVE_CMDID,
 WMI_VAP_REMOVE_CMDID,
 WMI_VAP_CREATE_CMDID,
 WMI_REG_READ_CMDID,
 WMI_REG_WRITE_CMDID,
 WMI_RC_STATE_CHANGE_CMDID,
 WMI_RC_RATE_UPDATE_CMDID,
 WMI_TARGET_IC_UPDATE_CMDID,
 WMI_TX_AGGR_ENABLE_CMDID,
 WMI_TGT_DETACH_CMDID,
 WMI_NODE_UPDATE_CMDID,
 WMI_INT_STATS_CMDID,
 WMI_TX_STATS_CMDID,
 WMI_RX_STATS_CMDID,
 WMI_BITRATE_MASK_CMDID,
 WMI_REG_RMW_CMDID,
};

enum wmi_event_id {
 WMI_TGT_RDY_EVENTID = 0x1001,
 WMI_SWBA_EVENTID,
 WMI_FATAL_EVENTID,
 WMI_TXTO_EVENTID,
 WMI_BMISS_EVENTID,
 WMI_DELBA_EVENTID,
 WMI_TXSTATUS_EVENTID,
};

#define MAX_CMD_NUMBER 62
#define MAX_RMW_CMD_NUMBER 15

struct register_write {
 __be32 reg;
 __be32 val;
};

struct register_rmw {
 __be32 reg;
 __be32 set;
 __be32 clr;
} __packed;

struct ath9k_htc_tx_event {
 int count;
 struct __wmi_event_txstatus txs;
 struct list_head list;
};

struct wmi {
 struct ath9k_htc_priv *drv_priv;
 struct htc_target *htc;
 enum htc_endpoint_id ctrl_epid;
 struct mutex op_mutex;
 struct completion cmd_wait;
 u16 last_seq_id;
 struct sk_buff_head wmi_event_queue;
 struct tasklet_struct wmi_event_tasklet;
 u16 tx_seq_id;
 u8 *cmd_rsp_buf;
 u32 cmd_rsp_len;
 bool stopped;

 struct list_head pending_tx_events;
 spinlock_t event_lock;

 spinlock_t wmi_lock;

 /* multi write section */
 atomic_t mwrite_cnt;
 struct register_write multi_write[MAX_CMD_NUMBER];
 u32 multi_write_idx;
 struct mutex multi_write_mutex;

 /* multi rmw section */
 atomic_t m_rmw_cnt;
 struct register_rmw multi_rmw[MAX_RMW_CMD_NUMBER];
 u32 multi_rmw_idx;
 struct mutex multi_rmw_mutex;

};

struct wmi *ath9k_init_wmi(struct ath9k_htc_priv *priv);
int ath9k_wmi_connect(struct htc_target *htc, struct wmi *wmi,
        enum htc_endpoint_id *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);
void ath9k_wmi_event_tasklet(struct tasklet_struct *t);
void ath9k_fatal_work(struct work_struct *work);
void ath9k_wmi_event_drain(struct ath9k_htc_priv *priv);
void ath9k_stop_wmi(struct ath9k_htc_priv *priv);
void ath9k_destroy_wmi(struct ath9k_htc_priv *priv);

#define WMI_CMD(_wmi_cmd)      \
 do {        \
  ret = ath9k_wmi_cmd(priv->wmi, _wmi_cmd, NULL, 0, \
        (u8 *) &cmd_rsp,   \
        sizeof(cmd_rsp), HZ*2);  \
 } while (0)

#define WMI_CMD_BUF(_wmi_cmd, _buf)     \
 do {        \
  ret = ath9k_wmi_cmd(priv->wmi, _wmi_cmd,  \
        (u8 *) _buf, sizeof(*_buf),  \
        &cmd_rsp, sizeof(cmd_rsp), HZ*2); \
 } while (0)

#endif /* WMI_H */