1/* Set IPv4 source filter. Linux version.
2 Copyright (C) 2004-2020 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, see
18 <https://www.gnu.org/licenses/>. */
19
20#include <alloca.h>
21#include <errno.h>
22#include <stdlib.h>
23#include <string.h>
24#include <stdint.h>
25#include <netinet/in.h>
26#include <sys/socket.h>
27
28
29int
30setipv4sourcefilter (int s, struct in_addr interface, struct in_addr group,
31 uint32_t fmode, uint32_t numsrc,
32 const struct in_addr *slist)
33{
34 /* We have to create an struct ip_msfilter object which we can pass
35 to the kernel. */
36 size_t needed = IP_MSFILTER_SIZE (numsrc);
37 int use_alloca = __libc_use_alloca (needed);
38
39 struct ip_msfilter *imsf;
40 if (use_alloca)
41 imsf = (struct ip_msfilter *) alloca (needed);
42 else
43 {
44 imsf = (struct ip_msfilter *) malloc (needed);
45 if (imsf == NULL)
46 return -1;
47 }
48
49 imsf->imsf_multiaddr = group;
50 imsf->imsf_interface = interface;
51 imsf->imsf_fmode = fmode;
52 imsf->imsf_numsrc = numsrc;
53 memcpy (imsf->imsf_slist, slist, numsrc * sizeof (struct in_addr));
54
55 int result = __setsockopt (s, SOL_IP, IP_MSFILTER, imsf, needed);
56
57 if (! use_alloca)
58 {
59 int save_errno = errno;
60 free (imsf);
61 __set_errno (save_errno);
62 }
63
64 return result;
65}
66