src/wps/wps.h

   1 /*
   2  * Wi-Fi Protected Setup
   3  * Copyright (c) 2007-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_H
  16 #define WPS_H
  17
  18 #include "wps_defs.h"
  19
  20 /**
  21  * enum wsc_op_code - EAP-WSC OP-Code values
  22  */
  23 enum wsc_op_code {
  24         WSC_UPnP = 0 /* No OP Code in UPnP transport */,
  25         WSC_Start = 0x01,
  26         WSC_ACK = 0x02,
  27         WSC_NACK = 0x03,
  28         WSC_MSG = 0x04,
  29         WSC_Done = 0x05,
  30         WSC_FRAG_ACK = 0x06
  31 };
  32
  33 struct wps_registrar;
  34 struct upnp_wps_device_sm;
  35
  36 /**
  37  * struct wps_credential - WPS Credential
  38  * @ssid: SSID
  39  * @ssid_len: Length of SSID
  40  * @auth_type: Authentication Type (WPS_AUTH_OPEN, .. flags)
  41  * @encr_type: Encryption Type (WPS_ENCR_NONE, .. flags)
  42  * @key_idx: Key index
  43  * @key: Key
  44  * @key_len: Key length in octets
  45  * @mac_addr: MAC address of the peer
  46  * @cred_attr: Unparsed Credential attribute data (used only in cred_cb());
  47  *      this may be %NULL, if not used
  48  * @cred_attr_len: Length of cred_attr in octets
  49  */
  50 struct wps_credential {
  51         u8 ssid[32];
  52         size_t ssid_len;
  53         u16 auth_type;
  54         u16 encr_type;
  55         u8 key_idx;
  56         u8 key[64];
  57         size_t key_len;
  58         u8 mac_addr[ETH_ALEN];
  59         const u8 *cred_attr;
  60         size_t cred_attr_len;
  61 };
  62
  63 /**
  64  * struct wps_device_data - WPS Device Data
  65  * @mac_addr: Device MAC address
  66  * @device_name: Device Name (0..32 octets encoded in UTF-8)
  67  * @manufacturer: Manufacturer (0..64 octets encoded in UTF-8)
  68  * @model_name: Model Name (0..32 octets encoded in UTF-8)
  69  * @model_number: Model Number (0..32 octets encoded in UTF-8)
  70  * @serial_number: Serial Number (0..32 octets encoded in UTF-8)
  71  * @categ: Primary Device Category
  72  * @oui: Primary Device OUI
  73  * @sub_categ: Primary Device Sub-Category
  74  * @os_version: OS Version
  75  * @rf_bands: RF bands (WPS_RF_24GHZ, WPS_RF_50GHZ flags)
  76  */
  77 struct wps_device_data {
  78         u8 mac_addr[ETH_ALEN];
  79         char *device_name;
  80         char *manufacturer;
  81         char *model_name;
  82         char *model_number;
  83         char *serial_number;
  84         u16 categ;
  85         u32 oui;
  86         u16 sub_categ;
  87         u32 os_version;
  88         u8 rf_bands;
  89 };
  90
  91 struct oob_conf_data {
  92         enum {
  93                 OOB_METHOD_UNKNOWN = 0,
  94                 OOB_METHOD_DEV_PWD_E,
  95                 OOB_METHOD_DEV_PWD_R,
  96                 OOB_METHOD_CRED,
  97         } oob_method;
  98         struct wpabuf *dev_password;
  99         struct wpabuf *pubkey_hash;
 100 };
 101
 102 /**
 103  * struct wps_config - WPS configuration for a single registration protocol run
 104  */
 105 struct wps_config {
 106         /**
 107          * wps - Pointer to long term WPS context
 108          */
 109         struct wps_context *wps;
 110
 111         /**
 112          * registrar - Whether this end is a Registrar
 113          */
 114         int registrar;
 115
 116         /**
 117          * pin - Enrollee Device Password (%NULL for Registrar or PBC)
 118          */
 119         const u8 *pin;
 120
 121         /**
 122          * pin_len - Length on pin in octets
 123          */
 124         size_t pin_len;
 125
 126         /**
 127          * pbc - Whether this is protocol run uses PBC
 128          */
 129         int pbc;
 130
 131         /**
 132          * assoc_wps_ie: (Re)AssocReq WPS IE (in AP; %NULL if not AP)
 133          */
 134         const struct wpabuf *assoc_wps_ie;
 135
 136         /**
 137          * new_ap_settings - New AP settings (%NULL if not used)
 138          *
 139          * This parameter provides new AP settings when using a wireless
 140          * stations as a Registrar to configure the AP. %NULL means that AP
 141          * will not be reconfigured, i.e., the station will only learn the
 142          * current AP settings by using AP PIN.
 143          */
 144         const struct wps_credential *new_ap_settings;
 145 };
 146
 147 struct wps_data * wps_init(const struct wps_config *cfg);
 148
 149 void wps_deinit(struct wps_data *data);
 150
 151 /**
 152  * enum wps_process_res - WPS message processing result
 153  */
 154 enum wps_process_res {
 155         /**
 156          * WPS_DONE - Processing done
 157          */
 158         WPS_DONE,
 159
 160         /**
 161          * WPS_CONTINUE - Processing continues
 162          */
 163         WPS_CONTINUE,
 164
 165         /**
 166          * WPS_FAILURE - Processing failed
 167          */
 168         WPS_FAILURE,
 169
 170         /**
 171          * WPS_PENDING - Processing continues, but waiting for an external
 172          *      event (e.g., UPnP message from an external Registrar)
 173          */
 174         WPS_PENDING
 175 };
 176 enum wps_process_res wps_process_msg(struct wps_data *wps,
 177                                      enum wsc_op_code op_code,
 178                                      const struct wpabuf *msg);
 179
 180 struct wpabuf * wps_get_msg(struct wps_data *wps, enum wsc_op_code *op_code);
 181
 182 int wps_is_selected_pbc_registrar(const struct wpabuf *msg);
 183 int wps_is_selected_pin_registrar(const struct wpabuf *msg);
 184 const u8 * wps_get_uuid_e(const struct wpabuf *msg);
 185
 186 struct wpabuf * wps_build_assoc_req_ie(enum wps_request_type req_type);
 187 struct wpabuf * wps_build_probe_req_ie(int pbc, struct wps_device_data *dev,
 188                                        const u8 *uuid,
 189                                        enum wps_request_type req_type);
 190
 191
 192 /**
 193  * struct wps_registrar_config - WPS Registrar configuration
 194  */
 195 struct wps_registrar_config {
 196         /**
 197          * new_psk_cb - Callback for new PSK
 198          * @ctx: Higher layer context data (cb_ctx)
 199          * @mac_addr: MAC address of the Enrollee
 200          * @psk: The new PSK
 201          * @psk_len: The length of psk in octets
 202          * Returns: 0 on success, -1 on failure
 203          *
 204          * This callback is called when a new per-device PSK is provisioned.
 205          */
 206         int (*new_psk_cb)(void *ctx, const u8 *mac_addr, const u8 *psk,
 207                           size_t psk_len);
 208
 209         /**
 210          * set_ie_cb - Callback for WPS IE changes
 211          * @ctx: Higher layer context data (cb_ctx)
 212          * @beacon_ie: WPS IE for Beacon
 213          * @beacon_ie_len: WPS IE length for Beacon
 214          * @probe_resp_ie: WPS IE for Probe Response
 215          * @probe_resp_ie_len: WPS IE length for Probe Response
 216          * Returns: 0 on success, -1 on failure
 217          *
 218          * This callback is called whenever the WPS IE in Beacon or Probe
 219          * Response frames needs to be changed (AP only).
 220          */
 221         int (*set_ie_cb)(void *ctx, const u8 *beacon_ie, size_t beacon_ie_len,
 222                          const u8 *probe_resp_ie, size_t probe_resp_ie_len);
 223
 224         /**
 225          * pin_needed_cb - Callback for requesting a PIN
 226          * @ctx: Higher layer context data (cb_ctx)
 227          * @uuid_e: UUID-E of the unknown Enrollee
 228          * @dev: Device Data from the unknown Enrollee
 229          *
 230          * This callback is called whenever an unknown Enrollee requests to use
 231          * PIN method and a matching PIN (Device Password) is not found in
 232          * Registrar data.
 233          */
 234         void (*pin_needed_cb)(void *ctx, const u8 *uuid_e,
 235                               const struct wps_device_data *dev);
 236
 237         /**
 238          * reg_success_cb - Callback for reporting successful registration
 239          * @ctx: Higher layer context data (cb_ctx)
 240          * @mac_addr: MAC address of the Enrollee
 241          * @uuid_e: UUID-E of the Enrollee
 242          *
 243          * This callback is called whenever an Enrollee completes registration
 244          * successfully.
 245          */
 246         void (*reg_success_cb)(void *ctx, const u8 *mac_addr,
 247                                const u8 *uuid_e);
 248
 249         /**
 250          * cb_ctx: Higher layer context data for Registrar callbacks
 251          */
 252         void *cb_ctx;
 253
 254         /**
 255          * skip_cred_build: Do not build credential
 256          *
 257          * This option can be used to disable internal code that builds
 258          * Credential attribute into M8 based on the current network
 259          * configuration and Enrollee capabilities. The extra_cred data will
 260          * then be used as the Credential(s).
 261          */
 262         int skip_cred_build;
 263
 264         /**
 265          * extra_cred: Additional Credential attribute(s)
 266          *
 267          * This optional data (set to %NULL to disable) can be used to add
 268          * Credential attribute(s) for other networks into M8. If
 269          * skip_cred_build is set, this will also override the automatically
 270          * generated Credential attribute.
 271          */
 272         const u8 *extra_cred;
 273
 274         /**
 275          * extra_cred_len: Length of extra_cred in octets
 276          */
 277         size_t extra_cred_len;
 278
 279         /**
 280          * disable_auto_conf - Disable auto-configuration on first registration
 281          *
 282          * By default, the AP that is started in not configured state will
 283          * generate a random PSK and move to configured state when the first
 284          * registration protocol run is completed successfully. This option can
 285          * be used to disable this functionality and leave it up to an external
 286          * program to take care of configuration. This requires the extra_cred
 287          * to be set with a suitable Credential and skip_cred_build being used.
 288          */
 289         int disable_auto_conf;
 290
 291         /**
 292          * static_wep_only - Whether the BSS supports only static WEP
 293          */
 294         int static_wep_only;
 295 };
 296
 297
 298 /**
 299  * enum wps_event - WPS event types
 300  */
 301 enum wps_event {
 302         /**
 303          * WPS_EV_M2D - M2D received (Registrar did not know us)
 304          */
 305         WPS_EV_M2D,
 306
 307         /**
 308          * WPS_EV_FAIL - Registration failed
 309          */
 310         WPS_EV_FAIL,
 311
 312         /**
 313          * WPS_EV_SUCCESS - Registration succeeded
 314          */
 315         WPS_EV_SUCCESS,
 316
 317         /**
 318          * WPS_EV_PWD_AUTH_FAIL - Password authentication failed
 319          */
 320         WPS_EV_PWD_AUTH_FAIL
 321 };
 322
 323 /**
 324  * union wps_event_data - WPS event data
 325  */
 326 union wps_event_data {
 327         /**
 328          * struct wps_event_m2d - M2D event data
 329          */
 330         struct wps_event_m2d {
 331                 u16 config_methods;
 332                 const u8 *manufacturer;
 333                 size_t manufacturer_len;
 334                 const u8 *model_name;
 335                 size_t model_name_len;
 336                 const u8 *model_number;
 337                 size_t model_number_len;
 338                 const u8 *serial_number;
 339                 size_t serial_number_len;
 340                 const u8 *dev_name;
 341                 size_t dev_name_len;
 342                 const u8 *primary_dev_type; /* 8 octets */
 343                 u16 config_error;
 344                 u16 dev_password_id;
 345         } m2d;
 346
 347         /**
 348          * struct wps_event_fail - Registration failure information
 349          * @msg: enum wps_msg_type
 350          */
 351         struct wps_event_fail {
 352                 int msg;
 353         } fail;
 354
 355         struct wps_event_pwd_auth_fail {
 356                 int enrollee;
 357                 int part;
 358         } pwd_auth_fail;
 359 };
 360
 361 /**
 362  * struct upnp_pending_message - Pending PutWLANResponse messages
 363  * @next: Pointer to next pending message or %NULL
 364  * @addr: NewWLANEventMAC
 365  * @msg: NewMessage
 366  * @type: Message Type
 367  */
 368 struct upnp_pending_message {
 369         struct upnp_pending_message *next;
 370         u8 addr[ETH_ALEN];
 371         struct wpabuf *msg;
 372         enum wps_msg_type type;
 373 };
 374
 375 /**
 376  * struct wps_context - Long term WPS context data
 377  *
 378  * This data is stored at the higher layer Authenticator or Supplicant data
 379  * structures and it is maintained over multiple registration protocol runs.
 380  */
 381 struct wps_context {
 382         /**
 383          * ap - Whether the local end is an access point
 384          */
 385         int ap;
 386
 387         /**
 388          * registrar - Pointer to WPS registrar data from wps_registrar_init()
 389          */
 390         struct wps_registrar *registrar;
 391
 392         /**
 393          * wps_state - Current WPS state
 394          */
 395         enum wps_state wps_state;
 396
 397         /**
 398          * ap_setup_locked - Whether AP setup is locked (only used at AP)
 399          */
 400         int ap_setup_locked;
 401
 402         /**
 403          * uuid - Own UUID
 404          */
 405         u8 uuid[16];
 406
 407         /**
 408          * ssid - SSID
 409          *
 410          * This SSID is used by the Registrar to fill in information for
 411          * Credentials. In addition, AP uses it when acting as an Enrollee to
 412          * notify Registrar of the current configuration.
 413          */
 414         u8 ssid[32];
 415
 416         /**
 417          * ssid_len - Length of ssid in octets
 418          */
 419         size_t ssid_len;
 420
 421         /**
 422          * dev - Own WPS device data
 423          */
 424         struct wps_device_data dev;
 425
 426         /**
 427          * oob_conf - OOB Config data
 428          */
 429         struct oob_conf_data oob_conf;
 430
 431         /**
 432          * oob_dev_pw_id - OOB Device password id
 433          */
 434         u16 oob_dev_pw_id;
 435
 436         /**
 437          * dh_privkey - Diffie-Hellman private key
 438          */
 439         struct wpabuf *dh_privkey;
 440
 441         /**
 442          * dh_pubkey_oob - Diffie-Hellman public key
 443          */
 444         struct wpabuf *dh_pubkey;
 445
 446         /**
 447          * config_methods - Enabled configuration methods
 448          *
 449          * Bit field of WPS_CONFIG_*
 450          */
 451         u16 config_methods;
 452
 453         /**
 454          * encr_types - Enabled encryption types (bit field of WPS_ENCR_*)
 455          */
 456         u16 encr_types;
 457
 458         /**
 459          * auth_types - Authentication types (bit field of WPS_AUTH_*)
 460          */
 461         u16 auth_types;
 462
 463         /**
 464          * network_key - The current Network Key (PSK) or %NULL to generate new
 465          *
 466          * If %NULL, Registrar will generate per-device PSK. In addition, AP
 467          * uses this when acting as an Enrollee to notify Registrar of the
 468          * current configuration.
 469          */
 470         u8 *network_key;
 471
 472         /**
 473          * network_key_len - Length of network_key in octets
 474          */
 475         size_t network_key_len;
 476
 477         /**
 478          * ap_settings - AP Settings override for M7 (only used at AP)
 479          *
 480          * If %NULL, AP Settings attributes will be generated based on the
 481          * current network configuration.
 482          */
 483         u8 *ap_settings;
 484
 485         /**
 486          * ap_settings_len - Length of ap_settings in octets
 487          */
 488         size_t ap_settings_len;
 489
 490         /**
 491          * friendly_name - Friendly Name (required for UPnP)
 492          */
 493         char *friendly_name;
 494
 495         /**
 496          * manufacturer_url - Manufacturer URL (optional for UPnP)
 497          */
 498         char *manufacturer_url;
 499
 500         /**
 501          * model_description - Model Description (recommended for UPnP)
 502          */
 503         char *model_description;
 504
 505         /**
 506          * model_url - Model URL (optional for UPnP)
 507          */
 508         char *model_url;
 509
 510         /**
 511          * upc - Universal Product Code (optional for UPnP)
 512          */
 513         char *upc;
 514
 515         /**
 516          * cred_cb - Callback to notify that new Credentials were received
 517          * @ctx: Higher layer context data (cb_ctx)
 518          * @cred: The received Credential
 519          * Return: 0 on success, -1 on failure
 520          */
 521         int (*cred_cb)(void *ctx, const struct wps_credential *cred);
 522
 523         /**
 524          * event_cb - Event callback (state information about progress)
 525          * @ctx: Higher layer context data (cb_ctx)
 526          * @event: Event type
 527          * @data: Event data
 528          */
 529         void (*event_cb)(void *ctx, enum wps_event event,
 530                          union wps_event_data *data);
 531
 532         /**
 533          * cb_ctx: Higher layer context data for callbacks
 534          */
 535         void *cb_ctx;
 536
 537         struct upnp_wps_device_sm *wps_upnp;
 538
 539         /* Pending messages from UPnP PutWLANResponse */
 540         struct upnp_pending_message *upnp_msgs;
 541 };
 542
 543 struct oob_device_data {
 544         char *device_name;
 545         char *device_path;
 546         void * (*init_func)(struct wps_context *, struct oob_device_data *,
 547                             int);
 548         struct wpabuf * (*read_func)(void *);
 549         int (*write_func)(void *, struct wpabuf *);
 550         void (*deinit_func)(void *);
 551 };
 552
 553 struct oob_nfc_device_data {
 554         int (*init_func)(char *);
 555         void * (*read_func)(size_t *);
 556         int (*write_func)(void *, size_t);
 557         void (*deinit_func)(void);
 558 };
 559
 560 struct wps_registrar *
 561 wps_registrar_init(struct wps_context *wps,
 562                    const struct wps_registrar_config *cfg);
 563 void wps_registrar_deinit(struct wps_registrar *reg);
 564 int wps_registrar_add_pin(struct wps_registrar *reg, const u8 *uuid,
 565                           const u8 *pin, size_t pin_len, int timeout);
 566 int wps_registrar_invalidate_pin(struct wps_registrar *reg, const u8 *uuid);
 567 int wps_registrar_unlock_pin(struct wps_registrar *reg, const u8 *uuid);
 568 int wps_registrar_button_pushed(struct wps_registrar *reg);
 569 void wps_registrar_probe_req_rx(struct wps_registrar *reg, const u8 *addr,
 570                                 const struct wpabuf *wps_data);
 571 int wps_registrar_update_ie(struct wps_registrar *reg);
 572 int wps_registrar_set_selected_registrar(struct wps_registrar *reg,
 573                                          const struct wpabuf *msg);
 574
 575 unsigned int wps_pin_checksum(unsigned int pin);
 576 unsigned int wps_pin_valid(unsigned int pin);
 577 unsigned int wps_generate_pin(void);
 578 void wps_free_pending_msgs(struct upnp_pending_message *msgs);
 579
 580 struct oob_device_data * wps_get_oob_device(char *device_type);
 581 struct oob_nfc_device_data * wps_get_oob_nfc_device(char *device_name);
 582 int wps_get_oob_method(char *method);
 583 int wps_process_oob(struct wps_context *wps, struct oob_device_data *oob_dev,
 584                     int registrar);
 585
 586 #endif /* WPS_H */