chiark / gitweb /
sd-dhcp-client: use BPF on raw socket
[elogind.git] / src / libsystemd-network / dhcp-network.c
1 /***
2   This file is part of systemd.
3
4   Copyright (C) 2013 Intel Corporation. All rights reserved.
5
6   systemd is free software; you can redistribute it and/or modify it
7   under the terms of the GNU Lesser General Public License as published by
8   the Free Software Foundation; either version 2.1 of the License, or
9   (at your option) any later version.
10
11   systemd is distributed in the hope that it will be useful, but
12   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 License
17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
18 ***/
19
20 #include <errno.h>
21 #include <sys/types.h>
22 #include <sys/socket.h>
23 #include <string.h>
24 #include <linux/if_packet.h>
25 #include <net/ethernet.h>
26 #include <stdio.h>
27 #include <unistd.h>
28 #include <linux/filter.h>
29
30 #include "socket-util.h"
31
32 #include "dhcp-internal.h"
33
34 int dhcp_network_bind_raw_socket(int index, union sockaddr_union *link)
35 {
36         struct sock_filter filter[] = {
37             BPF_STMT(BPF_LD + BPF_B + BPF_ABS, offsetof(DHCPPacket, ip.protocol)), /* A <- IP protocol */
38             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, IPPROTO_UDP, 1, 0),                /* IP protocol = UDP? */
39             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
40             BPF_STMT(BPF_LD + BPF_H + BPF_ABS, offsetof(DHCPPacket, udp.dest)),    /* A <- UDP destination port */
41             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, DHCP_PORT_CLIENT, 1, 0),           /* UDP destination port = DHCP client? */
42             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
43             BPF_STMT(BPF_RET + BPF_K, 65535),                                      /* return all */
44         };
45         struct sock_fprog fprog = {
46             .len = ELEMENTSOF(filter),
47             .filter = filter
48         };
49         int s, one = 1;
50
51         assert(index > 0);
52         assert(link);
53
54         s = socket(AF_PACKET, SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0);
55         if (s < 0)
56                 return -errno;
57
58         link->ll.sll_family = AF_PACKET;
59         link->ll.sll_protocol = htons(ETH_P_IP);
60         link->ll.sll_ifindex =  index;
61         link->ll.sll_halen = ETH_ALEN;
62         memset(link->ll.sll_addr, 0xff, ETH_ALEN);
63
64         if (setsockopt (s, SOL_PACKET, PACKET_AUXDATA, &one, sizeof(one)) < 0)
65                 return -errno;
66
67         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER, &fprog, sizeof(fprog)) < 0) {
68                 return -errno;
69         }
70
71         if (bind(s, &link->sa, sizeof(link->ll)) < 0) {
72                 safe_close(s);
73                 return -errno;
74         }
75
76         return s;
77 }
78
79 int dhcp_network_bind_udp_socket(int index, be32_t address, uint16_t port)
80 {
81         int s;
82         union sockaddr_union src = {
83                 .in.sin_family = AF_INET,
84                 .in.sin_port = htobe16(port),
85                 .in.sin_addr.s_addr = address,
86         };
87
88         s = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0);
89         if (s < 0)
90                 return -errno;
91
92         if (bind(s, &src.sa, sizeof(src.in)) < 0) {
93                 safe_close(s);
94                 return -errno;
95         }
96
97         return s;
98 }
99
100 int dhcp_network_send_raw_socket(int s, const union sockaddr_union *link,
101                                  const void *packet, size_t len)
102 {
103         assert(link);
104         assert(packet);
105         assert(len);
106
107         if (sendto(s, packet, len, 0, &link->sa, sizeof(link->ll)) < 0)
108                 return -errno;
109
110         return 0;
111 }
112
113 int dhcp_network_send_udp_socket(int s, be32_t address, uint16_t port,
114                                  const void *packet, size_t len)
115 {
116         union sockaddr_union dest = {
117                 .in.sin_family = AF_INET,
118                 .in.sin_port = htobe16(port),
119                 .in.sin_addr.s_addr = address,
120         };
121
122         if (sendto(s, packet, len, 0, &dest.sa, sizeof(dest.in)) < 0)
123                 return -errno;
124
125         return 0;
126 }