Handle *NUMSRC correctly.
[kopensolaris-gnu/glibc.git] / sysdeps / unix / sysv / linux / getsourcefilter.c
1 /* Get source filter.  Linux version.
2    Copyright (C) 2004 Free Software Foundation, Inc.
3    This file is part of the GNU C Library.
4    Contributed by Ulrich Drepper <drepper@redhat.com>, 2004.
5
6    The GNU C Library is free software; you can redistribute it and/or
7    modify it under the terms of the GNU Lesser General Public
8    License as published by the Free Software Foundation; either
9    version 2.1 of the License, or (at your option) any later version.
10
11    The GNU C Library is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14    Lesser General Public License for more details.
15
16    You should have received a copy of the GNU Lesser General Public
17    License along with the GNU C Library; if not, write to the Free
18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19    02111-1307 USA.  */
20
21 #include <alloca.h>
22 #include <errno.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <netinet/in.h>
26 #include <sys/param.h>
27 #include <sys/socket.h>
28
29
30 int
31 getsourcefilter (int s, uint32_t interface, struct sockaddr *group,
32                  socklen_t grouplen, uint32_t *fmode, uint32_t *numsrc,
33                  struct sockaddr_storage *slist)
34 {
35   /* We have to create an struct ip_msfilter object which we can pass
36      to the kernel.  */
37   socklen_t needed = GROUP_FILTER_SIZE (*numsrc);
38   int use_malloc = __libc_use_alloca (needed);
39
40   struct group_filter *gf;
41   if (use_malloc)
42     {
43       gf = (struct group_filter *) malloc (needed);
44       if (gf == NULL)
45         return -1;
46     }
47   else
48     gf = (struct group_filter *) alloca (needed);
49
50   gf->gf_interface = interface;
51   memcpy (&gf->gf_group, group, grouplen);
52   gf->gf_numsrc = *numsrc;
53
54   int result = __getsockopt (s, SOL_IP, MCAST_MSFILTER, gf, &needed);
55
56   /* If successful, copy the results to the places the caller wants
57      them in.  */
58   if (result == 0)
59     {
60       *fmode = gf->gf_fmode;
61       memcpy (slist, gf->gf_slist,
62               MIN (*numsrc, gf->gf_numsrc) * sizeof (struct sockaddr_storage));
63       *numsrc = gf->gf_numsrc;
64     }
65
66   if (use_malloc)
67     {
68       int save_errno = errno;
69       free (gf);
70       __set_errno (save_errno);
71     }
72
73   return result;
74 }