d/freebsd-dev
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

In ng_attach_cntl() first allocate things that may fail, and then

Browse Source 
do the rest of initialization. This simplifies code and fixes
a double free in failure scenario.

Reviewed by:	bz
This commit is contained in:
Gleb Smirnoff 2011-07-14 18:38:10 +00:00
parent b8ea56d7e4
commit 0a7d45d349
Notes: svn2git 2020-12-20 02:59:44 +00:00 
svn path=/head/; revision=224031
1 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
sys/netgraph/ng_socket.c
View File

@ -525,32 +525,31 @@ ng_attach_cntl(struct socket *so)
{
struct ngsock *priv;
struct ngpcb *pcbp;
node_p node;
int error;
/* Setup protocol control block */
if ((error = ng_attach_common(so, NG_CONTROL)) != 0)
return (error);
pcbp = sotongpcb(so);
/* Make the generic node components */
if ((error = ng_make_node_common(&typestruct, &node)) != 0) {
ng_detach_common(pcbp, NG_CONTROL);
return (error);
}
/* Allocate node private info */
priv = malloc(sizeof(*priv), M_NETGRAPH_SOCK, M_WAITOK | M_ZERO);
/* Setup protocol control block */
if ((error = ng_attach_common(so, NG_CONTROL)) != 0) {
free(priv, M_NETGRAPH_SOCK);
return (error);
}
pcbp = sotongpcb(so);
/* Initialize mutex. */
mtx_init(&priv->mtx, "ng_socket", NULL, MTX_DEF);
/* Link the pcb the private data. */
priv->ctlsock = pcbp;
pcbp->sockdata = priv;
priv->refs++;
/* Initialize mutex. */
mtx_init(&priv->mtx, "ng_socket", NULL, MTX_DEF);
/* Make the generic node components */
if ((error = ng_make_node_common(&typestruct, &priv->node)) != 0) {
free(priv, M_NETGRAPH_SOCK);
ng_detach_common(pcbp, NG_CONTROL);
return (error);
}
priv->node = node;
/* Store a hint for netstat(1). */
priv->node_id = priv->node->nd_ID;