1/*
2 * DHCP snooping for Proxy ARP
3 * Copyright (c) 2014, Qualcomm Atheros, Inc.
4 *
5 * This software may be distributed under the terms of the BSD license.
6 * See README for more details.
7 */
8
9#include "utils/includes.h"
10
11#include "utils/common.h"
12#include "common/dhcp.h"
13#include "l2_packet/l2_packet.h"
14#include "hostapd.h"
15#include "sta_info.h"
16#include "ap_drv_ops.h"
17#include "x_snoop.h"
18#include "dhcp_snoop.h"
19
20
21static const char * ipaddr_str(u32 addr)
22{
23	static char buf[17];
24
25	os_snprintf(buf, sizeof(buf), "%u.%u.%u.%u",
26		    (addr >> 24) & 0xff, (addr >> 16) & 0xff,
27		    (addr >> 8) & 0xff, addr & 0xff);
28	return buf;
29}
30
31
32static void handle_dhcp(void *ctx, const u8 *src_addr, const u8 *buf,
33			size_t len)
34{
35	struct hostapd_data *hapd = ctx;
36	const struct bootp_pkt *b;
37	struct sta_info *sta;
38	int exten_len;
39	const u8 *end, *pos;
40	int res, msgtype = 0, prefixlen = 32;
41	u32 subnet_mask = 0;
42	u16 tot_len;
43
44	exten_len = len - ETH_HLEN - (sizeof(*b) - sizeof(b->exten));
45	if (exten_len < 4)
46		return;
47
48	b = (const struct bootp_pkt *) &buf[ETH_HLEN];
49	tot_len = ntohs(b->iph.tot_len);
50	if (tot_len > (unsigned int) (len - ETH_HLEN))
51		return;
52
53	if (WPA_GET_BE32(b->exten) != DHCP_MAGIC)
54		return;
55
56	/* Parse DHCP options */
57	end = (const u8 *) b + tot_len;
58	pos = &b->exten[4];
59	while (pos < end && *pos != DHCP_OPT_END) {
60		const u8 *opt = pos++;
61
62		if (*opt == DHCP_OPT_PAD)
63			continue;
64
65		if (pos >= end || 1 + *pos > end - pos)
66			break;
67		pos += *pos + 1;
68		if (pos >= end)
69			break;
70
71		switch (*opt) {
72		case DHCP_OPT_SUBNET_MASK:
73			if (opt[1] == 4)
74				subnet_mask = WPA_GET_BE32(&opt[2]);
75			if (subnet_mask == 0)
76				return;
77			while (!(subnet_mask & 0x1)) {
78				subnet_mask >>= 1;
79				prefixlen--;
80			}
81			break;
82		case DHCP_OPT_MSG_TYPE:
83			if (opt[1])
84				msgtype = opt[2];
85			break;
86		default:
87			break;
88		}
89	}
90
91	if (msgtype == DHCPACK) {
92		if (b->your_ip == 0)
93			return;
94
95		/* DHCPACK for DHCPREQUEST */
96		sta = ap_get_sta(hapd, b->hw_addr);
97		if (!sta)
98			return;
99
100		wpa_printf(MSG_DEBUG, "dhcp_snoop: Found DHCPACK for " MACSTR
101			   " @ IPv4 address %s/%d",
102			   MAC2STR(sta->addr),
103			   ipaddr_str(be_to_host32(b->your_ip)),
104			   prefixlen);
105
106		if (sta->ipaddr == b->your_ip)
107			return;
108
109		if (sta->ipaddr != 0) {
110			wpa_printf(MSG_DEBUG,
111				   "dhcp_snoop: Removing IPv4 address %s from the ip neigh table",
112				   ipaddr_str(be_to_host32(sta->ipaddr)));
113			hostapd_drv_br_delete_ip_neigh(hapd, 4,
114						       (u8 *) &sta->ipaddr);
115		}
116
117		res = hostapd_drv_br_add_ip_neigh(hapd, 4, (u8 *) &b->your_ip,
118						  prefixlen, sta->addr);
119		if (res) {
120			wpa_printf(MSG_DEBUG,
121				   "dhcp_snoop: Adding ip neigh table failed: %d",
122				   res);
123			return;
124		}
125		sta->ipaddr = b->your_ip;
126	}
127
128	if (hapd->conf->disable_dgaf && is_broadcast_ether_addr(buf)) {
129		for (sta = hapd->sta_list; sta; sta = sta->next) {
130			if (!(sta->flags & WLAN_STA_AUTHORIZED))
131				continue;
132			x_snoop_mcast_to_ucast_convert_send(hapd, sta,
133							    (u8 *) buf, len);
134		}
135	}
136}
137
138
139int dhcp_snoop_init(struct hostapd_data *hapd)
140{
141	hapd->sock_dhcp = x_snoop_get_l2_packet(hapd, handle_dhcp,
142						L2_PACKET_FILTER_DHCP);
143	if (hapd->sock_dhcp == NULL) {
144		wpa_printf(MSG_DEBUG,
145			   "dhcp_snoop: Failed to initialize L2 packet processing for DHCP packet: %s",
146			   strerror(errno));
147		return -1;
148	}
149
150	return 0;
151}
152
153
154void dhcp_snoop_deinit(struct hostapd_data *hapd)
155{
156	l2_packet_deinit(hapd->sock_dhcp);
157}
158