1/*
2 * Wi-Fi Protected Setup - internal definitions
3 * Copyright (c) 2008-2009, Jouni Malinen <j@w1.fi>
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation.
8 *
9 * Alternatively, this software may be distributed under the terms of BSD
10 * license.
11 *
12 * See README and COPYING for more details.
13 */
14
15#ifndef WPS_I_H
16#define WPS_I_H
17
18#include "wps.h"
19
20/**
21 * struct wps_data - WPS registration protocol data
22 *
23 * This data is stored at the EAP-WSC server/peer method and it is kept for a
24 * single registration protocol run.
25 */
26struct wps_data {
27	/**
28	 * wps - Pointer to long term WPS context
29	 */
30	struct wps_context *wps;
31
32	/**
33	 * registrar - Whether this end is a Registrar
34	 */
35	int registrar;
36
37	/**
38	 * er - Whether the local end is an external registrar
39	 */
40	int er;
41
42	enum {
43		/* Enrollee states */
44		SEND_M1, RECV_M2, SEND_M3, RECV_M4, SEND_M5, RECV_M6, SEND_M7,
45		RECV_M8, RECEIVED_M2D, WPS_MSG_DONE, RECV_ACK, WPS_FINISHED,
46		SEND_WSC_NACK,
47
48		/* Registrar states */
49		RECV_M1, SEND_M2, RECV_M3, SEND_M4, RECV_M5, SEND_M6,
50		RECV_M7, SEND_M8, RECV_DONE, SEND_M2D, RECV_M2D_ACK
51	} state;
52
53	u8 uuid_e[WPS_UUID_LEN];
54	u8 uuid_r[WPS_UUID_LEN];
55	u8 mac_addr_e[ETH_ALEN];
56	u8 nonce_e[WPS_NONCE_LEN];
57	u8 nonce_r[WPS_NONCE_LEN];
58	u8 psk1[WPS_PSK_LEN];
59	u8 psk2[WPS_PSK_LEN];
60	u8 snonce[2 * WPS_SECRET_NONCE_LEN];
61	u8 peer_hash1[WPS_HASH_LEN];
62	u8 peer_hash2[WPS_HASH_LEN];
63
64	struct wpabuf *dh_privkey;
65	struct wpabuf *dh_pubkey_e;
66	struct wpabuf *dh_pubkey_r;
67	u8 authkey[WPS_AUTHKEY_LEN];
68	u8 keywrapkey[WPS_KEYWRAPKEY_LEN];
69	u8 emsk[WPS_EMSK_LEN];
70
71	struct wpabuf *last_msg;
72
73	u8 *dev_password;
74	size_t dev_password_len;
75	u16 dev_pw_id;
76	int pbc;
77
78	/**
79	 * request_type - Request Type attribute from (Re)AssocReq
80	 */
81	u8 request_type;
82
83	/**
84	 * encr_type - Available encryption types
85	 */
86	u16 encr_type;
87
88	/**
89	 * auth_type - Available authentication types
90	 */
91	u16 auth_type;
92
93	u8 *new_psk;
94	size_t new_psk_len;
95
96	int wps_pin_revealed;
97	struct wps_credential cred;
98
99	struct wps_device_data peer_dev;
100
101	/**
102	 * config_error - Configuration Error value to be used in NACK
103	 */
104	u16 config_error;
105
106	int ext_reg;
107	int int_reg;
108
109	struct wps_credential *new_ap_settings;
110
111	void *dh_ctx;
112
113	void (*ap_settings_cb)(void *ctx, const struct wps_credential *cred);
114	void *ap_settings_cb_ctx;
115
116	struct wps_credential *use_cred;
117
118	int use_psk_key;
119};
120
121
122struct wps_parse_attr {
123	/* fixed length fields */
124	const u8 *version; /* 1 octet */
125	const u8 *msg_type; /* 1 octet */
126	const u8 *enrollee_nonce; /* WPS_NONCE_LEN (16) octets */
127	const u8 *registrar_nonce; /* WPS_NONCE_LEN (16) octets */
128	const u8 *uuid_r; /* WPS_UUID_LEN (16) octets */
129	const u8 *uuid_e; /* WPS_UUID_LEN (16) octets */
130	const u8 *auth_type_flags; /* 2 octets */
131	const u8 *encr_type_flags; /* 2 octets */
132	const u8 *conn_type_flags; /* 1 octet */
133	const u8 *config_methods; /* 2 octets */
134	const u8 *sel_reg_config_methods; /* 2 octets */
135	const u8 *primary_dev_type; /* 8 octets */
136	const u8 *rf_bands; /* 1 octet */
137	const u8 *assoc_state; /* 2 octets */
138	const u8 *config_error; /* 2 octets */
139	const u8 *dev_password_id; /* 2 octets */
140	const u8 *oob_dev_password; /* WPS_OOB_DEVICE_PASSWORD_ATTR_LEN (54)
141				     * octets */
142	const u8 *os_version; /* 4 octets */
143	const u8 *wps_state; /* 1 octet */
144	const u8 *authenticator; /* WPS_AUTHENTICATOR_LEN (8) octets */
145	const u8 *r_hash1; /* WPS_HASH_LEN (32) octets */
146	const u8 *r_hash2; /* WPS_HASH_LEN (32) octets */
147	const u8 *e_hash1; /* WPS_HASH_LEN (32) octets */
148	const u8 *e_hash2; /* WPS_HASH_LEN (32) octets */
149	const u8 *r_snonce1; /* WPS_SECRET_NONCE_LEN (16) octets */
150	const u8 *r_snonce2; /* WPS_SECRET_NONCE_LEN (16) octets */
151	const u8 *e_snonce1; /* WPS_SECRET_NONCE_LEN (16) octets */
152	const u8 *e_snonce2; /* WPS_SECRET_NONCE_LEN (16) octets */
153	const u8 *key_wrap_auth; /* WPS_KWA_LEN (8) octets */
154	const u8 *auth_type; /* 2 octets */
155	const u8 *encr_type; /* 2 octets */
156	const u8 *network_idx; /* 1 octet */
157	const u8 *network_key_idx; /* 1 octet */
158	const u8 *mac_addr; /* ETH_ALEN (6) octets */
159	const u8 *key_prov_auto; /* 1 octet (Bool) */
160	const u8 *dot1x_enabled; /* 1 octet (Bool) */
161	const u8 *selected_registrar; /* 1 octet (Bool) */
162	const u8 *request_type; /* 1 octet */
163	const u8 *response_type; /* 1 octet */
164	const u8 *ap_setup_locked; /* 1 octet */
165
166	/* variable length fields */
167	const u8 *manufacturer;
168	size_t manufacturer_len;
169	const u8 *model_name;
170	size_t model_name_len;
171	const u8 *model_number;
172	size_t model_number_len;
173	const u8 *serial_number;
174	size_t serial_number_len;
175	const u8 *dev_name;
176	size_t dev_name_len;
177	const u8 *public_key;
178	size_t public_key_len;
179	const u8 *encr_settings;
180	size_t encr_settings_len;
181	const u8 *ssid; /* <= 32 octets */
182	size_t ssid_len;
183	const u8 *network_key; /* <= 64 octets */
184	size_t network_key_len;
185	const u8 *eap_type; /* <= 8 octets */
186	size_t eap_type_len;
187	const u8 *eap_identity; /* <= 64 octets */
188	size_t eap_identity_len;
189
190	/* attributes that can occur multiple times */
191#define MAX_CRED_COUNT 10
192	const u8 *cred[MAX_CRED_COUNT];
193	size_t cred_len[MAX_CRED_COUNT];
194	size_t num_cred;
195};
196
197/* wps_common.c */
198void wps_kdf(const u8 *key, const u8 *label_prefix, size_t label_prefix_len,
199	     const char *label, u8 *res, size_t res_len);
200int wps_derive_keys(struct wps_data *wps);
201void wps_derive_psk(struct wps_data *wps, const u8 *dev_passwd,
202		    size_t dev_passwd_len);
203struct wpabuf * wps_decrypt_encr_settings(struct wps_data *wps, const u8 *encr,
204					  size_t encr_len);
205void wps_fail_event(struct wps_context *wps, enum wps_msg_type msg);
206void wps_success_event(struct wps_context *wps);
207void wps_pwd_auth_fail_event(struct wps_context *wps, int enrollee, int part);
208void wps_pbc_overlap_event(struct wps_context *wps);
209void wps_pbc_timeout_event(struct wps_context *wps);
210
211extern struct oob_device_data oob_ufd_device_data;
212extern struct oob_device_data oob_nfc_device_data;
213extern struct oob_nfc_device_data oob_nfc_pn531_device_data;
214
215/* wps_attr_parse.c */
216int wps_parse_msg(const struct wpabuf *msg, struct wps_parse_attr *attr);
217
218/* wps_attr_build.c */
219int wps_build_public_key(struct wps_data *wps, struct wpabuf *msg);
220int wps_build_req_type(struct wpabuf *msg, enum wps_request_type type);
221int wps_build_resp_type(struct wpabuf *msg, enum wps_response_type type);
222int wps_build_config_methods(struct wpabuf *msg, u16 methods);
223int wps_build_uuid_e(struct wpabuf *msg, const u8 *uuid);
224int wps_build_dev_password_id(struct wpabuf *msg, u16 id);
225int wps_build_config_error(struct wpabuf *msg, u16 err);
226int wps_build_authenticator(struct wps_data *wps, struct wpabuf *msg);
227int wps_build_key_wrap_auth(struct wps_data *wps, struct wpabuf *msg);
228int wps_build_encr_settings(struct wps_data *wps, struct wpabuf *msg,
229			    struct wpabuf *plain);
230int wps_build_version(struct wpabuf *msg);
231int wps_build_msg_type(struct wpabuf *msg, enum wps_msg_type msg_type);
232int wps_build_enrollee_nonce(struct wps_data *wps, struct wpabuf *msg);
233int wps_build_registrar_nonce(struct wps_data *wps, struct wpabuf *msg);
234int wps_build_auth_type_flags(struct wps_data *wps, struct wpabuf *msg);
235int wps_build_encr_type_flags(struct wps_data *wps, struct wpabuf *msg);
236int wps_build_conn_type_flags(struct wps_data *wps, struct wpabuf *msg);
237int wps_build_assoc_state(struct wps_data *wps, struct wpabuf *msg);
238int wps_build_oob_dev_password(struct wpabuf *msg, struct wps_context *wps);
239
240/* wps_attr_process.c */
241int wps_process_authenticator(struct wps_data *wps, const u8 *authenticator,
242			      const struct wpabuf *msg);
243int wps_process_key_wrap_auth(struct wps_data *wps, struct wpabuf *msg,
244			      const u8 *key_wrap_auth);
245int wps_process_cred(struct wps_parse_attr *attr,
246		     struct wps_credential *cred);
247int wps_process_ap_settings(struct wps_parse_attr *attr,
248			    struct wps_credential *cred);
249
250/* wps_enrollee.c */
251struct wpabuf * wps_enrollee_get_msg(struct wps_data *wps,
252				     enum wsc_op_code *op_code);
253enum wps_process_res wps_enrollee_process_msg(struct wps_data *wps,
254					      enum wsc_op_code op_code,
255					      const struct wpabuf *msg);
256
257/* wps_registrar.c */
258struct wpabuf * wps_registrar_get_msg(struct wps_data *wps,
259				      enum wsc_op_code *op_code);
260enum wps_process_res wps_registrar_process_msg(struct wps_data *wps,
261					       enum wsc_op_code op_code,
262					       const struct wpabuf *msg);
263int wps_build_cred(struct wps_data *wps, struct wpabuf *msg);
264int wps_device_store(struct wps_registrar *reg,
265		     struct wps_device_data *dev, const u8 *uuid);
266void wps_registrar_selected_registrar_changed(struct wps_registrar *reg);
267
268/* ndef.c */
269struct wpabuf * ndef_parse_wifi(struct wpabuf *buf);
270struct wpabuf * ndef_build_wifi(struct wpabuf *buf);
271
272static inline int wps_version_supported(const u8 *version)
273{
274	/* Require major version match, but allow minor version differences */
275	return version && (*version & 0xf0) == (WPS_VERSION & 0xf0);
276}
277
278#endif /* WPS_I_H */
279