third-party-mirrors/sslh
1
0
Fork 0
mirror of https://github.com/yrutschle/sslh.git synced 2025-06-06 02:13:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix potential file descriptor leak if set_nonblock () fails

Browse Source
This commit is contained in:
Yves Rutschle 2023-05-13 22:36:21 +02:00
parent 842f6b0473
commit ee48dae8c5
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
udp-listener.c
View File

@ -221,6 +221,20 @@ static void mark_active(struct connection* cnx)
} }
/* Creates a new non-blocking socket */
static int nonblocking_socket(struct sslhcfg_protocols_item* proto)
{
int out = socket(proto->saddr->ai_family, SOCK_DGRAM, 0);
int res = set_nonblock(out);
if (res == -1) {
print_message(msg_system_error, "%s:%d:%s:%d:%s\n", __FILE__, __LINE__, "udp:socket:nonblock", errno, strerror(errno));
close(out);
return -1;
}
return out;
}
/* Process UDP coming from outside (client towards server) /* Process UDP coming from outside (client towards server)
* If it's a new source, probe; otherwise, forward to previous target * If it's a new source, probe; otherwise, forward to previous target
* Returns: newly allocate connections, for new connections * Returns: newly allocate connections, for new connections
@ -266,9 +280,8 @@ struct connection* udp_c2s_forward(int sockfd, struct loop_info* fd_info)
return NULL; return NULL;
} }
out = socket(proto->saddr->ai_family, SOCK_DGRAM, 0); out = nonblocking_socket(proto);
res = set_nonblock(out); if (out == -1) return NULL;
CHECK_RES_RETURN(res, "udp:socket:nonblock", NULL);
struct connection* cnx = collection_alloc_cnx_from_fd(collection, out); struct connection* cnx = collection_alloc_cnx_from_fd(collection, out);
if (!cnx) return NULL; if (!cnx) return NULL;
target = out; target = out;