stv0g/tvheadend
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

udp: multipacket - add NULL checks

Browse source
This commit is contained in:
Jaroslav Kysela 2014-07-07 19:54:34 +02:00
parent 37d7156f6c
commit 069baa5ad9
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/udp.c
View file

@ -517,6 +517,7 @@ udp_multirecv_init( udp_multirecv_t *um, int packets, int psize )
{
int i;
assert(um);
um->um_psize = psize;
um->um_packets = packets;
um->um_data = malloc(packets * psize);
@ -535,6 +536,8 @@ udp_multirecv_init( udp_multirecv_t *um, int packets, int psize )
void
udp_multirecv_free( udp_multirecv_t *um )
{
if (um == NULL)
return;
free(um->um_msg); um->um_msg = NULL;
free(um->um_riovec); um->um_riovec = NULL;
free(um->um_iovec); um->um_iovec = NULL;
@ -548,6 +551,10 @@ udp_multirecv_read( udp_multirecv_t *um, int fd, int packets,
struct iovec **iovec )
{
int n, i;
if (um == NULL || iovec == NULL) {
errno = EINVAL;
return -1;
}
if (packets > um->um_packets)
packets = um->um_packets;
n = recvmmsg(fd, (struct mmsghdr *)um->um_msg, packets, MSG_DONTWAIT, NULL);