| /* Set source filter. Linux version. |
| Copyright (C) 2004-2018 Free Software Foundation, Inc. |
| This file is part of the GNU C Library. |
| Contributed by Ulrich Drepper <drepper@redhat.com>, 2004. |
| |
| The GNU C Library is free software; you can redistribute it and/or |
| modify it under the terms of the GNU Lesser General Public |
| License as published by the Free Software Foundation; either |
| version 2.1 of the License, or (at your option) any later version. |
| |
| The GNU C Library is distributed in the hope that it will be useful, |
| but WITHOUT ANY WARRANTY; without even the implied warranty of |
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| Lesser General Public License for more details. |
| |
| You should have received a copy of the GNU Lesser General Public |
| License along with the GNU C Library; if not, see |
| <http://www.gnu.org/licenses/>. */ |
| |
| #include <alloca.h> |
| #include <errno.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <stdint.h> |
| #include <netinet/in.h> |
| #include <sys/socket.h> |
| #include "getsourcefilter.h" |
| |
| |
| int |
| setsourcefilter (int s, uint32_t interface, const struct sockaddr *group, |
| socklen_t grouplen, uint32_t fmode, uint32_t numsrc, |
| const struct sockaddr_storage *slist) |
| { |
| /* We have to create an struct ip_msfilter object which we can pass |
| to the kernel. */ |
| size_t needed = GROUP_FILTER_SIZE (numsrc); |
| int use_alloca = __libc_use_alloca (needed); |
| |
| struct group_filter *gf; |
| if (use_alloca) |
| gf = (struct group_filter *) alloca (needed); |
| else |
| { |
| gf = (struct group_filter *) malloc (needed); |
| if (gf == NULL) |
| return -1; |
| } |
| |
| gf->gf_interface = interface; |
| memcpy (&gf->gf_group, group, grouplen); |
| gf->gf_fmode = fmode; |
| gf->gf_numsrc = numsrc; |
| memcpy (gf->gf_slist, slist, numsrc * sizeof (struct sockaddr_storage)); |
| |
| /* We need to provide the appropriate socket level value. */ |
| int result; |
| int sol = __get_sol (group->sa_family, grouplen); |
| if (sol == -1) |
| { |
| __set_errno (EINVAL); |
| result = -1; |
| } |
| else |
| result = __setsockopt (s, sol, MCAST_MSFILTER, gf, needed); |
| |
| if (! use_alloca) |
| { |
| int save_errno = errno; |
| free (gf); |
| __set_errno (save_errno); |
| } |
| |
| return result; |
| } |