Skip to content

Commit

Permalink
staging: rtl8712: check for alloc fail in _r8712_init_recv_priv()
Browse files Browse the repository at this point in the history
The function _r8712_init_recv_priv() and also r8712_init_recv_priv()
just returns silently if they fail to allocate memory. Change their
return type to int and add necessary checks and handling if they return
-ENOMEM

Tested-by: Philipp Hortmann <[email protected]>
Signed-off-by: Nam Cao <[email protected]>
Link: https://lore.kernel.org/r/506ac35a667e511db568b06b86834fd0ceeba453.1666688642.git.namcaov@gmail.com
Signed-off-by: Greg Kroah-Hartman <[email protected]>
  • Loading branch information
covanam authored and gregkh committed Oct 31, 2022
1 parent c5a7eec commit 63b5e50
Show file tree
Hide file tree
Showing 4 changed files with 20 additions and 12 deletions.
4 changes: 3 additions & 1 deletion drivers/staging/rtl8712/os_intfs.c
Original file line number Diff line number Diff line change
Expand Up @@ -309,7 +309,9 @@ int r8712_init_drv_sw(struct _adapter *padapter)
if (ret)
return ret;
_r8712_init_xmit_priv(&padapter->xmitpriv, padapter);
_r8712_init_recv_priv(&padapter->recvpriv, padapter);
ret = _r8712_init_recv_priv(&padapter->recvpriv, padapter);
if (ret)
return ret;
memset((unsigned char *)&padapter->securitypriv, 0,
sizeof(struct security_priv));
timer_setup(&padapter->securitypriv.tkip_timer,
Expand Down
8 changes: 4 additions & 4 deletions drivers/staging/rtl8712/recv_osdep.h
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,15 @@
#include "drv_types.h"
#include <linux/skbuff.h>

void _r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter);
int _r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter);
void _r8712_free_recv_priv(struct recv_priv *precvpriv);
void r8712_recv_entry(union recv_frame *precv_frame);
void r8712_recv_indicatepkt(struct _adapter *adapter,
union recv_frame *precv_frame);
void r8712_handle_tkip_mic_err(struct _adapter *padapter, u8 bgroup);
void r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter);
int r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter);
void r8712_free_recv_priv(struct recv_priv *precvpriv);
void r8712_os_recv_resource_alloc(struct _adapter *padapter,
union recv_frame *precvframe);
Expand Down
7 changes: 4 additions & 3 deletions drivers/staging/rtl8712/rtl8712_recv.c
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@

static void recv_tasklet(struct tasklet_struct *t);

void r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter)
int r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter)
{
int i;
struct recv_buf *precvbuf;
Expand All @@ -44,7 +44,7 @@ void r8712_init_recv_priv(struct recv_priv *precvpriv,
precvpriv->pallocated_recv_buf =
kzalloc(NR_RECVBUFF * sizeof(struct recv_buf) + 4, GFP_ATOMIC);
if (!precvpriv->pallocated_recv_buf)
return;
return -ENOMEM;
precvpriv->precv_buf = precvpriv->pallocated_recv_buf + 4 -
((addr_t)(precvpriv->pallocated_recv_buf) & 3);
precvbuf = (struct recv_buf *)precvpriv->precv_buf;
Expand Down Expand Up @@ -75,6 +75,7 @@ void r8712_init_recv_priv(struct recv_priv *precvpriv,
}
pskb = NULL;
}
return 0;
}

void r8712_free_recv_priv(struct recv_priv *precvpriv)
Expand Down
13 changes: 9 additions & 4 deletions drivers/staging/rtl8712/rtl871x_recv.c
Original file line number Diff line number Diff line change
Expand Up @@ -42,9 +42,10 @@ void _r8712_init_sta_recv_priv(struct sta_recv_priv *psta_recvpriv)
_init_queue(&psta_recvpriv->defrag_q);
}

void _r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter)
int _r8712_init_recv_priv(struct recv_priv *precvpriv,
struct _adapter *padapter)
{
int ret;
sint i;
union recv_frame *precvframe;

Expand All @@ -58,7 +59,7 @@ void _r8712_init_recv_priv(struct recv_priv *precvpriv,
sizeof(union recv_frame) + RXFRAME_ALIGN_SZ,
GFP_ATOMIC);
if (!precvpriv->pallocated_frame_buf)
return;
return -ENOMEM;
precvpriv->precv_frame_buf = precvpriv->pallocated_frame_buf +
RXFRAME_ALIGN_SZ -
((addr_t)(precvpriv->pallocated_frame_buf) &
Expand All @@ -73,7 +74,11 @@ void _r8712_init_recv_priv(struct recv_priv *precvpriv,
precvframe++;
}
precvpriv->rx_pending_cnt = 1;
r8712_init_recv_priv(precvpriv, padapter);
ret = r8712_init_recv_priv(precvpriv, padapter);
if (ret)
kfree(precvpriv->pallocated_frame_buf);

return ret;
}

void _r8712_free_recv_priv(struct recv_priv *precvpriv)
Expand Down

0 comments on commit 63b5e50

Please sign in to comment.