staging: r8188eu: off by one bugs

[karo-tx-linux.git] / drivers / staging / rtl8188eu / core / rtw_cmd.c

diff --git a/drivers/staging/rtl8188eu/core/rtw_cmd.c b/drivers/staging/rtl8188eu/core/rtw_cmd.c
index 859fc1919ad64c3dae916dd1b696e8619582d88c..9632ef48fbc1234877a19d031ade4633d14483ad 100644 (file)
--- a/drivers/staging/rtl8188eu/core/rtw_cmd.c
+++ b/drivers/staging/rtl8188eu/core/rtw_cmd.c
@@ -367,7 +367,7 @@ _next:
 
                memcpy(pcmdbuf, pcmd->parmbuf, pcmd->cmdsz);
 
-               if (pcmd->cmdcode <= (sizeof(wlancmds) / sizeof(struct cmd_hdl))) {
+               if (pcmd->cmdcode < ARRAY_SIZE(wlancmds)) {
                        cmd_hdl = wlancmds[pcmd->cmdcode].h2cfuns;
 
                        if (cmd_hdl) {
@@ -385,7 +385,7 @@ _next:
 post_process:
 
                /* call callback function for post-processed */
-               if (pcmd->cmdcode <= (sizeof(rtw_cmd_callback) / sizeof(struct _cmd_callback))) {
+               if (pcmd->cmdcode < ARRAY_SIZE(rtw_cmd_callback)) {
                        pcmd_callback = rtw_cmd_callback[pcmd->cmdcode].callback;
                        if (pcmd_callback == NULL) {
                                RT_TRACE(_module_rtl871x_cmd_c_, _drv_info_, ("mlme_cmd_hdl(): pcmd_callback = 0x%p, cmdcode = 0x%x\n", pcmd_callback, pcmd->cmdcode));