chiark / gitweb /
dhcp-server: simplify dhcp server unref call
[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 <net/if_arp.h>
27 #include <stdio.h>
28 #include <unistd.h>
29 #include <linux/filter.h>
30
31 #include "socket-util.h"
32
33 #include "dhcp-internal.h"
34
35 int dhcp_network_bind_raw_socket(int index, union sockaddr_union *link,
36                                  uint32_t xid) {
37         struct sock_filter filter[] = {
38             BPF_STMT(BPF_LD + BPF_W + BPF_LEN, 0),                                 /* A <- packet length */
39             BPF_JUMP(BPF_JMP + BPF_JGE + BPF_K, sizeof(DHCPPacket), 1, 0),         /* packet >= DHCPPacket ? */
40             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
41             BPF_STMT(BPF_LD + BPF_B + BPF_ABS, offsetof(DHCPPacket, ip.protocol)), /* A <- IP protocol */
42             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, IPPROTO_UDP, 1, 0),                /* IP protocol == UDP ? */
43             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
44             BPF_STMT(BPF_LD + BPF_H + BPF_ABS, offsetof(DHCPPacket, udp.dest)),    /* A <- UDP destination port */
45             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, DHCP_PORT_CLIENT, 1, 0),           /* UDP destination port == DHCP client port ? */
46             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
47             BPF_STMT(BPF_LD + BPF_B + BPF_ABS, offsetof(DHCPPacket, dhcp.op)),     /* A <- DHCP op */
48             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, BOOTREPLY, 1, 0),                  /* op == BOOTREPLY ? */
49             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
50             BPF_STMT(BPF_LD + BPF_B + BPF_ABS, offsetof(DHCPPacket, dhcp.htype)),  /* A <- DHCP header type */
51             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, ARPHRD_ETHER, 1, 0),               /* header type == ARPHRD_ETHER ? */
52             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
53             BPF_STMT(BPF_LD + BPF_B + BPF_ABS, offsetof(DHCPPacket, dhcp.hlen)),   /* A <- mac address length */
54             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, ETHER_ADDR_LEN, 1, 0),             /* address length == ETHER_ADDR_LEN ? */
55             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
56             BPF_STMT(BPF_LD + BPF_W + BPF_ABS, offsetof(DHCPPacket, dhcp.xid)),    /* A <- client identifier */
57             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, xid, 1, 0),                        /* client identifier == xid ? */
58             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
59             /* TODO: match chaddr */
60             BPF_STMT(BPF_LD + BPF_W + BPF_ABS, offsetof(DHCPPacket, dhcp.magic)),  /* A <- DHCP magic cookie */
61             BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, DHCP_MAGIC_COOKIE, 1, 0),          /* cookie == DHCP magic cookie ? */
62             BPF_STMT(BPF_RET + BPF_K, 0),                                          /* ignore */
63             BPF_STMT(BPF_RET + BPF_K, 65535),                                      /* return all */
64         };
65         struct sock_fprog fprog = {
66             .len = ELEMENTSOF(filter),
67             .filter = filter
68         };
69         _cleanup_close_ int s = -1;
70         int r, on = 1;
71
72         assert(index > 0);
73         assert(link);
74
75         s = socket(AF_PACKET, SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0);
76         if (s < 0)
77                 return -errno;
78
79         r = setsockopt (s, SOL_PACKET, PACKET_AUXDATA, &on, sizeof(on));
80         if (r < 0)
81                 return -errno;
82
83         r = setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER, &fprog, sizeof(fprog));
84         if (r < 0)
85                 return -errno;
86
87         link->ll.sll_family = AF_PACKET;
88         link->ll.sll_protocol = htons(ETH_P_IP);
89         link->ll.sll_ifindex =  index;
90         link->ll.sll_halen = ETH_ALEN;
91         memset(link->ll.sll_addr, 0xff, ETH_ALEN);
92
93         r = bind(s, &link->sa, sizeof(link->ll));
94         if (r < 0)
95                 return -errno;
96
97         r = s;
98         s = -1;
99
100         return r;
101 }
102
103 int dhcp_network_bind_udp_socket(be32_t address, uint16_t port) {
104         union sockaddr_union src = {
105                 .in.sin_family = AF_INET,
106                 .in.sin_port = htobe16(port),
107                 .in.sin_addr.s_addr = address,
108         };
109         _cleanup_close_ int s = -1;
110         int r, tos = IPTOS_CLASS_CS6;
111
112         s = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0);
113         if (s < 0)
114                 return -errno;
115
116         r = setsockopt(s, IPPROTO_IP, IP_TOS, &tos, sizeof(tos));
117         if (r < 0)
118                 return -errno;
119
120         if (address == INADDR_ANY) {
121                 int on = 1;
122
123                 r = setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
124                 if (r < 0)
125                         return -errno;
126
127                 r = setsockopt(s, IPPROTO_IP, IP_PKTINFO, &on, sizeof(on));
128                 if (r < 0)
129                         return -errno;
130         }
131
132         r = bind(s, &src.sa, sizeof(src.in));
133         if (r < 0)
134                 return -errno;
135
136         r = s;
137         s = -1;
138
139         return r;
140 }
141
142 int dhcp_network_send_raw_socket(int s, const union sockaddr_union *link,
143                                  const void *packet, size_t len) {
144         int r;
145
146         assert(link);
147         assert(packet);
148         assert(len);
149
150         r = sendto(s, packet, len, 0, &link->sa, sizeof(link->ll));
151         if (r < 0)
152                 return -errno;
153
154         return 0;
155 }
156
157 int dhcp_network_send_udp_socket(int s, be32_t address, uint16_t port,
158                                  const void *packet, size_t len) {
159         union sockaddr_union dest = {
160                 .in.sin_family = AF_INET,
161                 .in.sin_port = htobe16(port),
162                 .in.sin_addr.s_addr = address,
163         };
164         int r;
165
166         assert(s >= 0);
167         assert(packet);
168         assert(len);
169
170         r = sendto(s, packet, len, 0, &dest.sa, sizeof(dest.in));
171         if (r < 0)
172                 return -errno;
173
174         return 0;
175 }