1a1157835SDaniel Fojt /*
2a1157835SDaniel Fojt  * Operating classes
3a1157835SDaniel Fojt  * Copyright(c) 2015 Intel Deutschland GmbH
4a1157835SDaniel Fojt  * Contact Information:
5a1157835SDaniel Fojt  * Intel Linux Wireless <ilw@linux.intel.com>
6a1157835SDaniel Fojt  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
7a1157835SDaniel Fojt  *
8a1157835SDaniel Fojt  * This software may be distributed under the terms of the BSD license.
9a1157835SDaniel Fojt  * See README for more details.
10a1157835SDaniel Fojt  */
11a1157835SDaniel Fojt 
12a1157835SDaniel Fojt #include "utils/includes.h"
13a1157835SDaniel Fojt 
14a1157835SDaniel Fojt #include "utils/common.h"
15a1157835SDaniel Fojt #include "common/ieee802_11_common.h"
16a1157835SDaniel Fojt #include "wpa_supplicant_i.h"
17a1157835SDaniel Fojt 
18a1157835SDaniel Fojt 
allow_channel(struct hostapd_hw_modes * mode,u8 chan,unsigned int * flags)19a1157835SDaniel Fojt static enum chan_allowed allow_channel(struct hostapd_hw_modes *mode, u8 chan,
20a1157835SDaniel Fojt 				       unsigned int *flags)
21a1157835SDaniel Fojt {
22a1157835SDaniel Fojt 	int i;
23a1157835SDaniel Fojt 
24a1157835SDaniel Fojt 	for (i = 0; i < mode->num_channels; i++) {
25a1157835SDaniel Fojt 		if (mode->channels[i].chan == chan)
26a1157835SDaniel Fojt 			break;
27a1157835SDaniel Fojt 	}
28a1157835SDaniel Fojt 
29a1157835SDaniel Fojt 	if (i == mode->num_channels ||
30a1157835SDaniel Fojt 	    (mode->channels[i].flag & HOSTAPD_CHAN_DISABLED))
31a1157835SDaniel Fojt 		return NOT_ALLOWED;
32a1157835SDaniel Fojt 
33a1157835SDaniel Fojt 	if (flags)
34a1157835SDaniel Fojt 		*flags = mode->channels[i].flag;
35a1157835SDaniel Fojt 
36a1157835SDaniel Fojt 	if (mode->channels[i].flag & HOSTAPD_CHAN_NO_IR)
37a1157835SDaniel Fojt 		return NO_IR;
38a1157835SDaniel Fojt 
39a1157835SDaniel Fojt 	return ALLOWED;
40a1157835SDaniel Fojt }
41a1157835SDaniel Fojt 
42a1157835SDaniel Fojt 
get_center_80mhz(struct hostapd_hw_modes * mode,u8 channel)43a1157835SDaniel Fojt static int get_center_80mhz(struct hostapd_hw_modes *mode, u8 channel)
44a1157835SDaniel Fojt {
45a1157835SDaniel Fojt 	u8 center_channels[] = { 42, 58, 106, 122, 138, 155 };
46a1157835SDaniel Fojt 	size_t i;
47a1157835SDaniel Fojt 
48a1157835SDaniel Fojt 	if (mode->mode != HOSTAPD_MODE_IEEE80211A)
49a1157835SDaniel Fojt 		return 0;
50a1157835SDaniel Fojt 
51a1157835SDaniel Fojt 	for (i = 0; i < ARRAY_SIZE(center_channels); i++) {
52a1157835SDaniel Fojt 		/*
53a1157835SDaniel Fojt 		 * In 80 MHz, the bandwidth "spans" 12 channels (e.g., 36-48),
54a1157835SDaniel Fojt 		 * so the center channel is 6 channels away from the start/end.
55a1157835SDaniel Fojt 		 */
56a1157835SDaniel Fojt 		if (channel >= center_channels[i] - 6 &&
57a1157835SDaniel Fojt 		    channel <= center_channels[i] + 6)
58a1157835SDaniel Fojt 			return center_channels[i];
59a1157835SDaniel Fojt 	}
60a1157835SDaniel Fojt 
61a1157835SDaniel Fojt 	return 0;
62a1157835SDaniel Fojt }
63a1157835SDaniel Fojt 
64a1157835SDaniel Fojt 
verify_80mhz(struct hostapd_hw_modes * mode,u8 channel)65a1157835SDaniel Fojt static enum chan_allowed verify_80mhz(struct hostapd_hw_modes *mode, u8 channel)
66a1157835SDaniel Fojt {
67a1157835SDaniel Fojt 	u8 center_chan;
68a1157835SDaniel Fojt 	unsigned int i;
69a1157835SDaniel Fojt 	unsigned int no_ir = 0;
70a1157835SDaniel Fojt 
71a1157835SDaniel Fojt 	center_chan = get_center_80mhz(mode, channel);
72a1157835SDaniel Fojt 	if (!center_chan)
73a1157835SDaniel Fojt 		return NOT_ALLOWED;
74a1157835SDaniel Fojt 
75a1157835SDaniel Fojt 	/* check all the channels are available */
76a1157835SDaniel Fojt 	for (i = 0; i < 4; i++) {
77a1157835SDaniel Fojt 		unsigned int flags;
78a1157835SDaniel Fojt 		u8 adj_chan = center_chan - 6 + i * 4;
79a1157835SDaniel Fojt 
80a1157835SDaniel Fojt 		if (allow_channel(mode, adj_chan, &flags) == NOT_ALLOWED)
81a1157835SDaniel Fojt 			return NOT_ALLOWED;
82a1157835SDaniel Fojt 
83a1157835SDaniel Fojt 		if ((i == 0 && !(flags & HOSTAPD_CHAN_VHT_10_70)) ||
84a1157835SDaniel Fojt 		    (i == 1 && !(flags & HOSTAPD_CHAN_VHT_30_50)) ||
85a1157835SDaniel Fojt 		    (i == 2 && !(flags & HOSTAPD_CHAN_VHT_50_30)) ||
86a1157835SDaniel Fojt 		    (i == 3 && !(flags & HOSTAPD_CHAN_VHT_70_10)))
87a1157835SDaniel Fojt 			return NOT_ALLOWED;
88a1157835SDaniel Fojt 
89a1157835SDaniel Fojt 		if (flags & HOSTAPD_CHAN_NO_IR)
90a1157835SDaniel Fojt 			no_ir = 1;
91a1157835SDaniel Fojt 	}
92a1157835SDaniel Fojt 
93a1157835SDaniel Fojt 	if (no_ir)
94a1157835SDaniel Fojt 		return NO_IR;
95a1157835SDaniel Fojt 
96a1157835SDaniel Fojt 	return ALLOWED;
97a1157835SDaniel Fojt }
98a1157835SDaniel Fojt 
99a1157835SDaniel Fojt 
get_center_160mhz(struct hostapd_hw_modes * mode,u8 channel)100a1157835SDaniel Fojt static int get_center_160mhz(struct hostapd_hw_modes *mode, u8 channel)
101a1157835SDaniel Fojt {
102a1157835SDaniel Fojt 	u8 center_channels[] = { 50, 114 };
103a1157835SDaniel Fojt 	unsigned int i;
104a1157835SDaniel Fojt 
105a1157835SDaniel Fojt 	if (mode->mode != HOSTAPD_MODE_IEEE80211A)
106a1157835SDaniel Fojt 		return 0;
107a1157835SDaniel Fojt 
108a1157835SDaniel Fojt 	for (i = 0; i < ARRAY_SIZE(center_channels); i++) {
109a1157835SDaniel Fojt 		/*
110a1157835SDaniel Fojt 		 * In 160 MHz, the bandwidth "spans" 28 channels (e.g., 36-64),
111a1157835SDaniel Fojt 		 * so the center channel is 14 channels away from the start/end.
112a1157835SDaniel Fojt 		 */
113a1157835SDaniel Fojt 		if (channel >= center_channels[i] - 14 &&
114a1157835SDaniel Fojt 		    channel <= center_channels[i] + 14)
115a1157835SDaniel Fojt 			return center_channels[i];
116a1157835SDaniel Fojt 	}
117a1157835SDaniel Fojt 
118a1157835SDaniel Fojt 	return 0;
119a1157835SDaniel Fojt }
120a1157835SDaniel Fojt 
121a1157835SDaniel Fojt 
verify_160mhz(struct hostapd_hw_modes * mode,u8 channel)122a1157835SDaniel Fojt static enum chan_allowed verify_160mhz(struct hostapd_hw_modes *mode,
123a1157835SDaniel Fojt 				       u8 channel)
124a1157835SDaniel Fojt {
125a1157835SDaniel Fojt 	u8 center_chan;
126a1157835SDaniel Fojt 	unsigned int i;
127a1157835SDaniel Fojt 	unsigned int no_ir = 0;
128a1157835SDaniel Fojt 
129a1157835SDaniel Fojt 	center_chan = get_center_160mhz(mode, channel);
130a1157835SDaniel Fojt 	if (!center_chan)
131a1157835SDaniel Fojt 		return NOT_ALLOWED;
132a1157835SDaniel Fojt 
133a1157835SDaniel Fojt 	/* Check all the channels are available */
134a1157835SDaniel Fojt 	for (i = 0; i < 8; i++) {
135a1157835SDaniel Fojt 		unsigned int flags;
136a1157835SDaniel Fojt 		u8 adj_chan = center_chan - 14 + i * 4;
137a1157835SDaniel Fojt 
138a1157835SDaniel Fojt 		if (allow_channel(mode, adj_chan, &flags) == NOT_ALLOWED)
139a1157835SDaniel Fojt 			return NOT_ALLOWED;
140a1157835SDaniel Fojt 
141a1157835SDaniel Fojt 		if ((i == 0 && !(flags & HOSTAPD_CHAN_VHT_10_150)) ||
142a1157835SDaniel Fojt 		    (i == 1 && !(flags & HOSTAPD_CHAN_VHT_30_130)) ||
143a1157835SDaniel Fojt 		    (i == 2 && !(flags & HOSTAPD_CHAN_VHT_50_110)) ||
144a1157835SDaniel Fojt 		    (i == 3 && !(flags & HOSTAPD_CHAN_VHT_70_90)) ||
145a1157835SDaniel Fojt 		    (i == 4 && !(flags & HOSTAPD_CHAN_VHT_90_70)) ||
146a1157835SDaniel Fojt 		    (i == 5 && !(flags & HOSTAPD_CHAN_VHT_110_50)) ||
147a1157835SDaniel Fojt 		    (i == 6 && !(flags & HOSTAPD_CHAN_VHT_130_30)) ||
148a1157835SDaniel Fojt 		    (i == 7 && !(flags & HOSTAPD_CHAN_VHT_150_10)))
149a1157835SDaniel Fojt 			return NOT_ALLOWED;
150a1157835SDaniel Fojt 
151a1157835SDaniel Fojt 		if (flags & HOSTAPD_CHAN_NO_IR)
152a1157835SDaniel Fojt 			no_ir = 1;
153a1157835SDaniel Fojt 	}
154a1157835SDaniel Fojt 
155a1157835SDaniel Fojt 	if (no_ir)
156a1157835SDaniel Fojt 		return NO_IR;
157a1157835SDaniel Fojt 
158a1157835SDaniel Fojt 	return ALLOWED;
159a1157835SDaniel Fojt }
160a1157835SDaniel Fojt 
161a1157835SDaniel Fojt 
verify_channel(struct hostapd_hw_modes * mode,u8 channel,u8 bw)162a1157835SDaniel Fojt enum chan_allowed verify_channel(struct hostapd_hw_modes *mode, u8 channel,
163a1157835SDaniel Fojt 				 u8 bw)
164a1157835SDaniel Fojt {
165a1157835SDaniel Fojt 	unsigned int flag = 0;
166a1157835SDaniel Fojt 	enum chan_allowed res, res2;
167a1157835SDaniel Fojt 
168a1157835SDaniel Fojt 	res2 = res = allow_channel(mode, channel, &flag);
169a1157835SDaniel Fojt 	if (bw == BW40MINUS) {
170a1157835SDaniel Fojt 		if (!(flag & HOSTAPD_CHAN_HT40MINUS))
171a1157835SDaniel Fojt 			return NOT_ALLOWED;
172a1157835SDaniel Fojt 		res2 = allow_channel(mode, channel - 4, NULL);
173a1157835SDaniel Fojt 	} else if (bw == BW40PLUS) {
174a1157835SDaniel Fojt 		if (!(flag & HOSTAPD_CHAN_HT40PLUS))
175a1157835SDaniel Fojt 			return NOT_ALLOWED;
176a1157835SDaniel Fojt 		res2 = allow_channel(mode, channel + 4, NULL);
177a1157835SDaniel Fojt 	} else if (bw == BW80) {
178a1157835SDaniel Fojt 		/*
179a1157835SDaniel Fojt 		 * channel is a center channel and as such, not necessarily a
180a1157835SDaniel Fojt 		 * valid 20 MHz channels. Override earlier allow_channel()
181a1157835SDaniel Fojt 		 * result and use only the 80 MHz specific version.
182a1157835SDaniel Fojt 		 */
183a1157835SDaniel Fojt 		res2 = res = verify_80mhz(mode, channel);
184a1157835SDaniel Fojt 	} else if (bw == BW160) {
185a1157835SDaniel Fojt 		/*
186a1157835SDaniel Fojt 		 * channel is a center channel and as such, not necessarily a
187a1157835SDaniel Fojt 		 * valid 20 MHz channels. Override earlier allow_channel()
188a1157835SDaniel Fojt 		 * result and use only the 160 MHz specific version.
189a1157835SDaniel Fojt 		 */
190a1157835SDaniel Fojt 		res2 = res = verify_160mhz(mode, channel);
191a1157835SDaniel Fojt 	} else if (bw == BW80P80) {
192a1157835SDaniel Fojt 		/*
193a1157835SDaniel Fojt 		 * channel is a center channel and as such, not necessarily a
194a1157835SDaniel Fojt 		 * valid 20 MHz channels. Override earlier allow_channel()
195a1157835SDaniel Fojt 		 * result and use only the 80 MHz specific version.
196a1157835SDaniel Fojt 		 */
197a1157835SDaniel Fojt 		res2 = res = verify_80mhz(mode, channel);
198a1157835SDaniel Fojt 	}
199a1157835SDaniel Fojt 
200a1157835SDaniel Fojt 	if (res == NOT_ALLOWED || res2 == NOT_ALLOWED)
201a1157835SDaniel Fojt 		return NOT_ALLOWED;
202a1157835SDaniel Fojt 
203a1157835SDaniel Fojt 	if (res == NO_IR || res2 == NO_IR)
204a1157835SDaniel Fojt 		return NO_IR;
205a1157835SDaniel Fojt 
206a1157835SDaniel Fojt 	return ALLOWED;
207a1157835SDaniel Fojt }
208a1157835SDaniel Fojt 
209a1157835SDaniel Fojt 
wpas_op_class_supported(struct wpa_supplicant * wpa_s,struct wpa_ssid * ssid,const struct oper_class_map * op_class)210a1157835SDaniel Fojt static int wpas_op_class_supported(struct wpa_supplicant *wpa_s,
211a1157835SDaniel Fojt 				   struct wpa_ssid *ssid,
212a1157835SDaniel Fojt 				   const struct oper_class_map *op_class)
213a1157835SDaniel Fojt {
214a1157835SDaniel Fojt 	int chan;
215a1157835SDaniel Fojt 	size_t i;
216a1157835SDaniel Fojt 	struct hostapd_hw_modes *mode;
217a1157835SDaniel Fojt 	int found;
218a1157835SDaniel Fojt 	int z;
219a1157835SDaniel Fojt 	int freq2 = 0;
220a1157835SDaniel Fojt 	int freq5 = 0;
221a1157835SDaniel Fojt 
222a1157835SDaniel Fojt 	mode = get_mode(wpa_s->hw.modes, wpa_s->hw.num_modes, op_class->mode);
223a1157835SDaniel Fojt 	if (!mode)
224a1157835SDaniel Fojt 		return 0;
225a1157835SDaniel Fojt 
226a1157835SDaniel Fojt 	/* If we are configured to disable certain things, take that into
227a1157835SDaniel Fojt 	 * account here. */
228a1157835SDaniel Fojt 	if (ssid->freq_list && ssid->freq_list[0]) {
229a1157835SDaniel Fojt 		for (z = 0; ; z++) {
230a1157835SDaniel Fojt 			int f = ssid->freq_list[z];
231a1157835SDaniel Fojt 
232a1157835SDaniel Fojt 			if (f == 0)
233a1157835SDaniel Fojt 				break; /* end of list */
234a1157835SDaniel Fojt 			if (f > 4000 && f < 6000)
235a1157835SDaniel Fojt 				freq5 = 1;
236a1157835SDaniel Fojt 			else if (f > 2400 && f < 2500)
237a1157835SDaniel Fojt 				freq2 = 1;
238a1157835SDaniel Fojt 		}
239a1157835SDaniel Fojt 	} else {
240a1157835SDaniel Fojt 		/* No frequencies specified, can use anything hardware supports.
241a1157835SDaniel Fojt 		 */
242a1157835SDaniel Fojt 		freq2 = freq5 = 1;
243a1157835SDaniel Fojt 	}
244a1157835SDaniel Fojt 
245a1157835SDaniel Fojt 	if (op_class->op_class >= 115 && op_class->op_class <= 130 && !freq5)
246a1157835SDaniel Fojt 		return 0;
247a1157835SDaniel Fojt 	if (op_class->op_class >= 81 && op_class->op_class <= 84 && !freq2)
248a1157835SDaniel Fojt 		return 0;
249a1157835SDaniel Fojt 
250a1157835SDaniel Fojt #ifdef CONFIG_HT_OVERRIDES
251a1157835SDaniel Fojt 	if (ssid->disable_ht) {
252a1157835SDaniel Fojt 		switch (op_class->op_class) {
253a1157835SDaniel Fojt 		case 83:
254a1157835SDaniel Fojt 		case 84:
255a1157835SDaniel Fojt 		case 104:
256a1157835SDaniel Fojt 		case 105:
257a1157835SDaniel Fojt 		case 116:
258a1157835SDaniel Fojt 		case 117:
259a1157835SDaniel Fojt 		case 119:
260a1157835SDaniel Fojt 		case 120:
261a1157835SDaniel Fojt 		case 122:
262a1157835SDaniel Fojt 		case 123:
263a1157835SDaniel Fojt 		case 126:
264a1157835SDaniel Fojt 		case 127:
265a1157835SDaniel Fojt 		case 128:
266a1157835SDaniel Fojt 		case 129:
267a1157835SDaniel Fojt 		case 130:
268a1157835SDaniel Fojt 			/* Disable >= 40 MHz channels if HT is disabled */
269a1157835SDaniel Fojt 			return 0;
270a1157835SDaniel Fojt 		}
271a1157835SDaniel Fojt 	}
272a1157835SDaniel Fojt #endif /* CONFIG_HT_OVERRIDES */
273a1157835SDaniel Fojt 
274a1157835SDaniel Fojt #ifdef CONFIG_VHT_OVERRIDES
275a1157835SDaniel Fojt 	if (ssid->disable_vht) {
276a1157835SDaniel Fojt 		if (op_class->op_class >= 128 && op_class->op_class <= 130) {
277a1157835SDaniel Fojt 			/* Disable >= 80 MHz channels if VHT is disabled */
278a1157835SDaniel Fojt 			return 0;
279a1157835SDaniel Fojt 		}
280a1157835SDaniel Fojt 	}
281a1157835SDaniel Fojt #endif /* CONFIG_VHT_OVERRIDES */
282a1157835SDaniel Fojt 
283a1157835SDaniel Fojt 	if (op_class->op_class == 128) {
284a1157835SDaniel Fojt 		u8 channels[] = { 42, 58, 106, 122, 138, 155 };
285a1157835SDaniel Fojt 
286a1157835SDaniel Fojt 		for (i = 0; i < ARRAY_SIZE(channels); i++) {
287a1157835SDaniel Fojt 			if (verify_channel(mode, channels[i], op_class->bw) !=
288a1157835SDaniel Fojt 			    NOT_ALLOWED)
289a1157835SDaniel Fojt 				return 1;
290a1157835SDaniel Fojt 		}
291a1157835SDaniel Fojt 
292a1157835SDaniel Fojt 		return 0;
293a1157835SDaniel Fojt 	}
294a1157835SDaniel Fojt 
295a1157835SDaniel Fojt 	if (op_class->op_class == 129) {
296a1157835SDaniel Fojt 		/* Check if either 160 MHz channels is allowed */
297a1157835SDaniel Fojt 		return verify_channel(mode, 50, op_class->bw) != NOT_ALLOWED ||
298a1157835SDaniel Fojt 			verify_channel(mode, 114, op_class->bw) != NOT_ALLOWED;
299a1157835SDaniel Fojt 	}
300a1157835SDaniel Fojt 
301a1157835SDaniel Fojt 	if (op_class->op_class == 130) {
302a1157835SDaniel Fojt 		/* Need at least two non-contiguous 80 MHz segments */
303a1157835SDaniel Fojt 		found = 0;
304a1157835SDaniel Fojt 
305a1157835SDaniel Fojt 		if (verify_channel(mode, 42, op_class->bw) != NOT_ALLOWED ||
306a1157835SDaniel Fojt 		    verify_channel(mode, 58, op_class->bw) != NOT_ALLOWED)
307a1157835SDaniel Fojt 			found++;
308a1157835SDaniel Fojt 		if (verify_channel(mode, 106, op_class->bw) != NOT_ALLOWED ||
309a1157835SDaniel Fojt 		    verify_channel(mode, 122, op_class->bw) != NOT_ALLOWED ||
310a1157835SDaniel Fojt 		    verify_channel(mode, 138, op_class->bw) != NOT_ALLOWED)
311a1157835SDaniel Fojt 			found++;
312a1157835SDaniel Fojt 		if (verify_channel(mode, 106, op_class->bw) != NOT_ALLOWED &&
313a1157835SDaniel Fojt 		    verify_channel(mode, 138, op_class->bw) != NOT_ALLOWED)
314a1157835SDaniel Fojt 			found++;
315a1157835SDaniel Fojt 		if (verify_channel(mode, 155, op_class->bw) != NOT_ALLOWED)
316a1157835SDaniel Fojt 			found++;
317a1157835SDaniel Fojt 
318a1157835SDaniel Fojt 		if (found >= 2)
319a1157835SDaniel Fojt 			return 1;
320a1157835SDaniel Fojt 
321a1157835SDaniel Fojt 		return 0;
322a1157835SDaniel Fojt 	}
323a1157835SDaniel Fojt 
324a1157835SDaniel Fojt 	found = 0;
325a1157835SDaniel Fojt 	for (chan = op_class->min_chan; chan <= op_class->max_chan;
326a1157835SDaniel Fojt 	     chan += op_class->inc) {
327a1157835SDaniel Fojt 		if (verify_channel(mode, chan, op_class->bw) != NOT_ALLOWED) {
328a1157835SDaniel Fojt 			found = 1;
329a1157835SDaniel Fojt 			break;
330a1157835SDaniel Fojt 		}
331a1157835SDaniel Fojt 	}
332a1157835SDaniel Fojt 
333a1157835SDaniel Fojt 	return found;
334a1157835SDaniel Fojt }
335a1157835SDaniel Fojt 
336a1157835SDaniel Fojt 
wpas_supp_op_class_ie(struct wpa_supplicant * wpa_s,struct wpa_ssid * ssid,int freq,u8 * pos,size_t len)337a1157835SDaniel Fojt size_t wpas_supp_op_class_ie(struct wpa_supplicant *wpa_s,
338a1157835SDaniel Fojt 			     struct wpa_ssid *ssid,
339a1157835SDaniel Fojt 			     int freq, u8 *pos, size_t len)
340a1157835SDaniel Fojt {
341a1157835SDaniel Fojt 	struct wpabuf *buf;
342a1157835SDaniel Fojt 	u8 op, current, chan;
343a1157835SDaniel Fojt 	u8 *ie_len;
344a1157835SDaniel Fojt 	size_t res;
345a1157835SDaniel Fojt 
346a1157835SDaniel Fojt 	/*
347a1157835SDaniel Fojt 	 * Assume 20 MHz channel for now.
348a1157835SDaniel Fojt 	 * TODO: Use the secondary channel and VHT channel width that will be
349a1157835SDaniel Fojt 	 * used after association.
350a1157835SDaniel Fojt 	 */
351a1157835SDaniel Fojt 	if (ieee80211_freq_to_channel_ext(freq, 0, CHANWIDTH_USE_HT,
352a1157835SDaniel Fojt 					  &current, &chan) == NUM_HOSTAPD_MODES)
353a1157835SDaniel Fojt 		return 0;
354a1157835SDaniel Fojt 
355a1157835SDaniel Fojt 	/*
356a1157835SDaniel Fojt 	 * Need 3 bytes for EID, length, and current operating class, plus
357a1157835SDaniel Fojt 	 * 1 byte for every other supported operating class.
358a1157835SDaniel Fojt 	 */
359a1157835SDaniel Fojt 	buf = wpabuf_alloc(global_op_class_size + 3);
360a1157835SDaniel Fojt 	if (!buf)
361a1157835SDaniel Fojt 		return 0;
362a1157835SDaniel Fojt 
363a1157835SDaniel Fojt 	wpabuf_put_u8(buf, WLAN_EID_SUPPORTED_OPERATING_CLASSES);
364a1157835SDaniel Fojt 	/* Will set the length later, putting a placeholder */
365a1157835SDaniel Fojt 	ie_len = wpabuf_put(buf, 1);
366a1157835SDaniel Fojt 	wpabuf_put_u8(buf, current);
367a1157835SDaniel Fojt 
368a1157835SDaniel Fojt 	for (op = 0; global_op_class[op].op_class; op++) {
369a1157835SDaniel Fojt 		if (wpas_op_class_supported(wpa_s, ssid, &global_op_class[op]))
370a1157835SDaniel Fojt 			wpabuf_put_u8(buf, global_op_class[op].op_class);
371a1157835SDaniel Fojt 	}
372a1157835SDaniel Fojt 
373a1157835SDaniel Fojt 	*ie_len = wpabuf_len(buf) - 2;
374a1157835SDaniel Fojt 	if (*ie_len < 2 || wpabuf_len(buf) > len) {
375*8716355dSDaniel Fojt 		wpa_printf(MSG_WARNING,
376a1157835SDaniel Fojt 			   "Failed to add supported operating classes IE");
377a1157835SDaniel Fojt 		res = 0;
378a1157835SDaniel Fojt 	} else {
379a1157835SDaniel Fojt 		os_memcpy(pos, wpabuf_head(buf), wpabuf_len(buf));
380a1157835SDaniel Fojt 		res = wpabuf_len(buf);
381a1157835SDaniel Fojt 		wpa_hexdump_buf(MSG_DEBUG,
382a1157835SDaniel Fojt 				"Added supported operating classes IE", buf);
383a1157835SDaniel Fojt 	}
384a1157835SDaniel Fojt 
385a1157835SDaniel Fojt 	wpabuf_free(buf);
386a1157835SDaniel Fojt 	return res;
387a1157835SDaniel Fojt }
388