5 files changed, 23 insertions, 54 deletions

diff --git a/drivers/net/wireless/libertas/assoc.c b/drivers/net/wireless/libertas/assoc.c
index e91cd6724de..c622e9b63cd 100644
--- a/drivers/net/wireless/libertas/assoc.c
+++ b/drivers/net/wireless/libertas/assoc.c
@@ -305,8 +305,8 @@ static int assoc_helper_secinfo(struct lbs_private *priv,
                                 struct assoc_request * assoc_req)
 {
         int ret = 0;
-        u32 do_wpa;
-        u32 rsn = 0;
+        uint16_t do_wpa;
+        uint16_t rsn = 0;
 
         lbs_deb_enter(LBS_DEB_ASSOC);
 
@@ -323,28 +323,19 @@ static int assoc_helper_secinfo(struct lbs_private *priv,
          */
 
         /* Get RSN enabled/disabled */
-        ret = lbs_prepare_and_send_command(priv,
-                                    CMD_802_11_ENABLE_RSN,
-                                    CMD_ACT_GET,
-                                    CMD_OPTION_WAITFORRSP,
-                                    0, &rsn);
+        ret = lbs_cmd_802_11_enable_rsn(priv, CMD_ACT_GET, &rsn);
         if (ret) {
                 lbs_deb_assoc("Failed to get RSN status: %d\n", ret);
                 goto out;
         }
 
         /* Don't re-enable RSN if it's already enabled */
-        do_wpa = (assoc_req->secinfo.WPAenabled || assoc_req->secinfo.WPA2enabled);
+        do_wpa = assoc_req->secinfo.WPAenabled || assoc_req->secinfo.WPA2enabled;
         if (do_wpa == rsn)
                 goto out;
 
         /* Set RSN enabled/disabled */
-        rsn = do_wpa;
-        ret = lbs_prepare_and_send_command(priv,
-                                    CMD_802_11_ENABLE_RSN,
-                                    CMD_ACT_SET,
-                                    CMD_OPTION_WAITFORRSP,
-                                    0, &rsn);
+        ret = lbs_cmd_802_11_enable_rsn(priv, CMD_ACT_SET, &do_wpa);
 
 out:
         lbs_deb_leave_args(LBS_DEB_ASSOC, "ret %d", ret);
diff --git a/drivers/net/wireless/libertas/cmd.c b/drivers/net/wireless/libertas/cmd.c
index 05e25cef5fe..5d2c928fdd3 100644
--- a/drivers/net/wireless/libertas/cmd.c
+++ b/drivers/net/wireless/libertas/cmd.c
@@ -300,30 +300,31 @@ done:
         return ret;
 }
 
-static int lbs_cmd_802_11_enable_rsn(struct lbs_private *priv,
-                                      struct cmd_ds_command *cmd,
-                                      u16 cmd_action,
-                                      void * pdata_buf)
+int lbs_cmd_802_11_enable_rsn(struct lbs_private *priv, uint16_t cmd_action,
+                              uint16_t *enable)
 {
-        struct cmd_ds_802_11_enable_rsn *penableRSN = &cmd->params.enbrsn;
-        u32 * enable = pdata_buf;
+        struct cmd_ds_802_11_enable_rsn cmd;
+        int ret;
 
         lbs_deb_enter(LBS_DEB_CMD);
 
-        cmd->command = cpu_to_le16(CMD_802_11_ENABLE_RSN);
-        cmd->size = cpu_to_le16(sizeof(*penableRSN) + S_DS_GEN);
-        penableRSN->action = cpu_to_le16(cmd_action);
+        cmd.hdr.size = cpu_to_le16(sizeof(cmd));
+        cmd.action = cpu_to_le16(cmd_action);
 
         if (cmd_action == CMD_ACT_SET) {
                 if (*enable)
-                        penableRSN->enable = cpu_to_le16(CMD_ENABLE_RSN);
+                        cmd.enable = cpu_to_le16(CMD_ENABLE_RSN);
                 else
-                        penableRSN->enable = cpu_to_le16(CMD_DISABLE_RSN);
+                        cmd.enable = cpu_to_le16(CMD_DISABLE_RSN);
                 lbs_deb_cmd("ENABLE_RSN: %d\n", *enable);
         }
 
-        lbs_deb_leave(LBS_DEB_CMD);
-        return 0;
+        ret = lbs_cmd_with_response(priv, CMD_802_11_ENABLE_RSN, &cmd);
+        if (!ret && cmd_action == CMD_ACT_GET)
+                *enable = le16_to_cpu(cmd.enable);
+
+        lbs_deb_leave_args(LBS_DEB_CMD, "ret %d", ret);
+        return ret;
 }
 
 
@@ -1473,11 +1474,6 @@ int lbs_prepare_and_send_command(struct lbs_private *priv,
                 ret = lbs_cmd_80211_ad_hoc_stop(priv, cmdptr);
                 break;
 
-        case CMD_802_11_ENABLE_RSN:
-                ret = lbs_cmd_802_11_enable_rsn(priv, cmdptr, cmd_action,
-                                pdata_buf);
-                break;
-
         case CMD_802_11_KEY_MATERIAL:
                 ret = lbs_cmd_802_11_key_material(priv, cmdptr, cmd_action,
                                 cmd_oid, pdata_buf);
diff --git a/drivers/net/wireless/libertas/cmd.h b/drivers/net/wireless/libertas/cmd.h
index d9947c3a507..6efb81ec597 100644
--- a/drivers/net/wireless/libertas/cmd.h
+++ b/drivers/net/wireless/libertas/cmd.h
@@ -53,5 +53,7 @@ int lbs_cmd_802_11_sleep_params(struct lbs_private *priv, uint16_t cmd_action,
                                 struct sleep_params *sp);
 int lbs_cmd_802_11_set_wep(struct lbs_private *priv, uint16_t cmd_action,
                            struct assoc_request *assoc);
+int lbs_cmd_802_11_enable_rsn(struct lbs_private *priv, uint16_t cmd_action,
+                              uint16_t *enable);
 
 #endif /* _LBS_CMD_H */
diff --git a/drivers/net/wireless/libertas/cmdresp.c b/drivers/net/wireless/libertas/cmdresp.c
index d670a811af0..8b71926d7a9 100644
--- a/drivers/net/wireless/libertas/cmdresp.c
+++ b/drivers/net/wireless/libertas/cmdresp.c
@@ -372,23 +372,6 @@ static int lbs_ret_get_log(struct lbs_private *priv,
         return 0;
 }
 
-static int lbs_ret_802_11_enable_rsn(struct lbs_private *priv,
-                                          struct cmd_ds_command *resp)
-{
-        struct cmd_ds_802_11_enable_rsn *enable_rsn = &resp->params.enbrsn;
-        uint32_t * pdata_buf = (uint32_t *)priv->cur_cmd->callback_arg;
-
-        lbs_deb_enter(LBS_DEB_CMD);
-
-        if (enable_rsn->action == cpu_to_le16(CMD_ACT_GET)) {
-                if (pdata_buf)
-                        *pdata_buf = (uint32_t) le16_to_cpu(enable_rsn->enable);
-        }
-
-        lbs_deb_leave(LBS_DEB_CMD);
-        return 0;
-}
-
 static int lbs_ret_802_11_bcn_ctrl(struct lbs_private * priv,
                                         struct cmd_ds_command *resp)
 {
@@ -495,10 +478,6 @@ static inline int handle_cmd_response(struct lbs_private *priv,
         case CMD_RET(CMD_802_11_BEACON_STOP):
                 break;
 
-        case CMD_RET(CMD_802_11_ENABLE_RSN):
-                ret = lbs_ret_802_11_enable_rsn(priv, resp);
-                break;
-
         case CMD_RET(CMD_802_11_RATE_ADAPT_RATESET):
                 ret = lbs_ret_802_11_rate_adapt_rateset(priv, resp);
                 break;
diff --git a/drivers/net/wireless/libertas/hostcmd.h b/drivers/net/wireless/libertas/hostcmd.h
index 6d76634f1e1..fffec11d065 100644
--- a/drivers/net/wireless/libertas/hostcmd.h
+++ b/drivers/net/wireless/libertas/hostcmd.h
@@ -536,6 +536,8 @@ struct cmd_ds_802_11_ad_hoc_join {
 } __attribute__ ((packed));
 
 struct cmd_ds_802_11_enable_rsn {
+        struct cmd_header hdr;
+
         __le16 action;
         __le16 enable;
 } __attribute__ ((packed));
@@ -708,7 +710,6 @@ struct cmd_ds_command {
                 struct cmd_ds_802_11_rssi_rsp rssirsp;
                 struct cmd_ds_802_11_disassociate dassociate;
                 struct cmd_ds_802_11_mac_address macadd;
-                struct cmd_ds_802_11_enable_rsn enbrsn;
                 struct cmd_ds_802_11_key_material keymaterial;
                 struct cmd_ds_mac_reg_access macreg;
                 struct cmd_ds_bbp_reg_access bbpreg;