Переглянути джерело

qcacmn: Add module_id check to avoid buffer overflow

Add module_id check to check to avoid the buffer overflow
of mod Array.

Change-Id: I73cfa1434d721e570df0414af8da35252ced5d7a
CRs-Fixed: 3589970
Azmath Mohammed 1 рік тому
батько
коміт
c0d6f0176e
1 змінених файлів з 8 додано та 0 видалено
  1. 8 0
      wbuff/src/wbuff.c

+ 8 - 0
wbuff/src/wbuff.c

@@ -373,6 +373,10 @@ QDF_STATUS wbuff_module_deregister(struct wbuff_mod_handle *hdl)
 		return QDF_STATUS_E_INVAL;
 
 	module_id = handle->id;
+
+	if (module_id > WBUFF_MAX_MODULES)
+		return -EINVAL;
+
 	mod = &wbuff.mod[module_id];
 
 	qdf_spin_lock_bh(&mod->lock);
@@ -417,6 +421,10 @@ wbuff_buff_get(struct wbuff_mod_handle *hdl, uint8_t pool_id, uint32_t len,
 		return NULL;
 
 	module_id = handle->id;
+
+	if (module_id > WBUFF_MAX_MODULES)
+		return NULL;
+
 	mod = &wbuff.mod[module_id];
 
 	if (pool_id == WBUFF_MAX_POOL_ID && len)