During memory allocation failure in handle_key() the complete() was not
called for comp_test_key_block event. So now added the code to call
complete() for event during error scenario.

Signed-off-by: Ajay Singh <ajay.kat...@microchip.com>
---
 drivers/staging/wilc1000/host_interface.c | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/drivers/staging/wilc1000/host_interface.c 
b/drivers/staging/wilc1000/host_interface.c
index 28edd90..d903ae5 100644
--- a/drivers/staging/wilc1000/host_interface.c
+++ b/drivers/staging/wilc1000/host_interface.c
@@ -1525,8 +1525,10 @@ static int handle_key(struct wilc_vif *vif, struct 
key_attr *hif_key)
 
                        key_buf = kmalloc(hif_key->attr.wep.key_len + 2,
                                          GFP_KERNEL);
-                       if (!key_buf)
-                               return -ENOMEM;
+                       if (!key_buf) {
+                               result = -ENOMEM;
+                               goto out_wep;
+                       }
 
                        key_buf[0] = hif_key->attr.wep.index;
                        key_buf[1] = hif_key->attr.wep.key_len;
@@ -1547,8 +1549,10 @@ static int handle_key(struct wilc_vif *vif, struct 
key_attr *hif_key)
                        kfree(key_buf);
                } else if (hif_key->action & ADDKEY) {
                        key_buf = kmalloc(hif_key->attr.wep.key_len + 2, 
GFP_KERNEL);
-                       if (!key_buf)
-                               return -ENOMEM;
+                       if (!key_buf) {
+                               result = -ENOMEM;
+                               goto out_wep;
+                       }
                        key_buf[0] = hif_key->attr.wep.index;
                        memcpy(key_buf + 1, &hif_key->attr.wep.key_len, 1);
                        memcpy(key_buf + 2, hif_key->attr.wep.key,
@@ -1585,6 +1589,7 @@ static int handle_key(struct wilc_vif *vif, struct 
key_attr *hif_key)
                                                      &wid, 1,
                                                      wilc_get_vif_idx(vif));
                }
+out_wep:
                complete(&hif_drv->comp_test_key_block);
                break;
 
@@ -1619,7 +1624,6 @@ static int handle_key(struct wilc_vif *vif, struct 
key_attr *hif_key)
                                                      wilc_get_vif_idx(vif));
 
                        kfree(key_buf);
-                       complete(&hif_drv->comp_test_key_block);
                } else if (hif_key->action & ADDKEY) {
                        key_buf = kzalloc(RX_MIC_KEY_MSG_LEN, GFP_KERNEL);
                        if (!key_buf) {
@@ -1648,9 +1652,9 @@ static int handle_key(struct wilc_vif *vif, struct 
key_attr *hif_key)
                                                      wilc_get_vif_idx(vif));
 
                        kfree(key_buf);
-                       complete(&hif_drv->comp_test_key_block);
                }
 out_wpa_rx_gtk:
+               complete(&hif_drv->comp_test_key_block);
                kfree(hif_key->attr.wpa.key);
                kfree(hif_key->attr.wpa.seq);
                if (ret)
@@ -1686,7 +1690,6 @@ static int handle_key(struct wilc_vif *vif, struct 
key_attr *hif_key)
                                                      wid_list, 2,
                                                      wilc_get_vif_idx(vif));
                        kfree(key_buf);
-                       complete(&hif_drv->comp_test_key_block);
                } else if (hif_key->action & ADDKEY) {
                        key_buf = kmalloc(PTK_KEY_MSG_LEN, GFP_KERNEL);
                        if (!key_buf) {
@@ -1708,10 +1711,10 @@ static int handle_key(struct wilc_vif *vif, struct 
key_attr *hif_key)
                                                      &wid, 1,
                                                      wilc_get_vif_idx(vif));
                        kfree(key_buf);
-                       complete(&hif_drv->comp_test_key_block);
                }
 
 out_wpa_ptk:
+               complete(&hif_drv->comp_test_key_block);
                kfree(hif_key->attr.wpa.key);
                if (ret)
                        return ret;
-- 
2.7.4

Reply via email to