Remove direct driver calls from sta_info.c
[libeap.git] / hostapd / wps_hostapd.c
index 1ca0ff9..3004ba1 100644 (file)
 
 #include "includes.h"
 
-#include "hostapd.h"
-#include "driver_i.h"
+#include "common.h"
 #include "eloop.h"
 #include "uuid.h"
+#include "crypto/dh_groups.h"
 #include "common/wpa_ctrl.h"
 #include "common/ieee802_11_defs.h"
 #include "common/ieee802_11_common.h"
-#include "sta_info.h"
-#include "eapol_sm.h"
+#include "eapol_auth/eapol_auth_sm.h"
+#include "eapol_auth/eapol_auth_sm_i.h"
 #include "wps/wps.h"
 #include "wps/wps_defs.h"
 #include "wps/wps_dev_attr.h"
+#include "hostapd.h"
+#include "config.h"
+#include "sta_info.h"
 #include "wps_hostapd.h"
-#include "dh_groups.h"
 
 
 #ifdef CONFIG_WPS_UPNP
@@ -88,46 +90,16 @@ static int hostapd_wps_new_psk_cb(void *ctx, const u8 *mac_addr, const u8 *psk,
 }
 
 
-static int hostapd_wps_set_ie_cb(void *ctx, const u8 *beacon_ie,
-                                size_t beacon_ie_len, const u8 *probe_resp_ie,
-                                size_t probe_resp_ie_len)
+static int hostapd_wps_set_ie_cb(void *ctx, struct wpabuf *beacon_ie,
+                                struct wpabuf *probe_resp_ie)
 {
        struct hostapd_data *hapd = ctx;
-
-       os_free(hapd->wps_beacon_ie);
-       if (beacon_ie_len == 0) {
-               hapd->wps_beacon_ie = NULL;
-               hapd->wps_beacon_ie_len = 0;
-       } else {
-               hapd->wps_beacon_ie = os_malloc(beacon_ie_len);
-               if (hapd->wps_beacon_ie == NULL) {
-                       hapd->wps_beacon_ie_len = 0;
-                       return -1;
-               }
-               os_memcpy(hapd->wps_beacon_ie, beacon_ie, beacon_ie_len);
-               hapd->wps_beacon_ie_len = beacon_ie_len;
-       }
-       hostapd_set_wps_beacon_ie(hapd, hapd->wps_beacon_ie,
-                                 hapd->wps_beacon_ie_len);
-
-       os_free(hapd->wps_probe_resp_ie);
-       if (probe_resp_ie_len == 0) {
-               hapd->wps_probe_resp_ie = NULL;
-               hapd->wps_probe_resp_ie_len = 0;
-       } else {
-               hapd->wps_probe_resp_ie = os_malloc(probe_resp_ie_len);
-               if (hapd->wps_probe_resp_ie == NULL) {
-                       hapd->wps_probe_resp_ie_len = 0;
-                       return -1;
-               }
-               os_memcpy(hapd->wps_probe_resp_ie, probe_resp_ie,
-                         probe_resp_ie_len);
-               hapd->wps_probe_resp_ie_len = probe_resp_ie_len;
-       }
-       hostapd_set_wps_probe_resp_ie(hapd, hapd->wps_probe_resp_ie,
-                                     hapd->wps_probe_resp_ie_len);
-
-       return 0;
+       wpabuf_free(hapd->wps_beacon_ie);
+       hapd->wps_beacon_ie = beacon_ie;
+       wpabuf_free(hapd->wps_probe_resp_ie);
+       hapd->wps_probe_resp_ie = probe_resp_ie;
+       return hapd->drv.set_ap_wps_ie(hapd, hapd->wps_beacon_ie,
+                                      hapd->wps_probe_resp_ie);
 }
 
 
@@ -475,15 +447,13 @@ static void hostapd_wps_event_cb(void *ctx, enum wps_event event,
 
 static void hostapd_wps_clear_ies(struct hostapd_data *hapd)
 {
-       os_free(hapd->wps_beacon_ie);
+       wpabuf_free(hapd->wps_beacon_ie);
        hapd->wps_beacon_ie = NULL;
-       hapd->wps_beacon_ie_len = 0;
-       hostapd_set_wps_beacon_ie(hapd, NULL, 0);
 
-       os_free(hapd->wps_probe_resp_ie);
+       wpabuf_free(hapd->wps_probe_resp_ie);
        hapd->wps_probe_resp_ie = NULL;
-       hapd->wps_probe_resp_ie_len = 0;
-       hostapd_set_wps_probe_resp_ie(hapd, NULL, 0);
+
+       hapd->drv.set_ap_wps_ie(hapd, NULL, NULL);
 }
 
 
@@ -529,17 +499,8 @@ int hostapd_init_wps(struct hostapd_data *hapd,
                os_strdup(hapd->conf->model_number) : NULL;
        wps->dev.serial_number = hapd->conf->serial_number ?
                os_strdup(hapd->conf->serial_number) : NULL;
-       if (hapd->conf->config_methods) {
-               char *m = hapd->conf->config_methods;
-               if (os_strstr(m, "label"))
-                       wps->config_methods |= WPS_CONFIG_LABEL;
-               if (os_strstr(m, "display"))
-                       wps->config_methods |= WPS_CONFIG_DISPLAY;
-               if (os_strstr(m, "push_button"))
-                       wps->config_methods |= WPS_CONFIG_PUSHBUTTON;
-               if (os_strstr(m, "keypad"))
-                       wps->config_methods |= WPS_CONFIG_KEYPAD;
-       }
+       wps->config_methods =
+               wps_config_methods_str2bin(hapd->conf->config_methods);
        if (hapd->conf->device_type &&
            wps_dev_type_str2bin(hapd->conf->device_type,
                                 wps->dev.pri_dev_type) < 0) {
@@ -617,6 +578,11 @@ int hostapd_init_wps(struct hostapd_data *hapd,
                wps->network_key_len = conf->ssid.wep.len[0];
        }
 
+       if (conf->ssid.wpa_psk) {
+               os_memcpy(wps->psk, conf->ssid.wpa_psk->psk, PMK_LEN);
+               wps->psk_set = 1;
+       }
+
        if (conf->wps_state == WPS_STATE_NOT_CONFIGURED) {
                /* Override parameters to enable security by default */
                wps->auth_types = WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK;
@@ -802,61 +768,6 @@ static void hostapd_wps_probe_req_rx(void *ctx, const u8 *addr,
 
 #ifdef CONFIG_WPS_UPNP
 
-static struct wpabuf *
-hostapd_rx_req_get_device_info(void *priv, struct upnp_wps_peer *peer)
-{
-       struct hostapd_data *hapd = priv;
-       struct wps_config cfg;
-       struct wps_data *wps;
-       enum wsc_op_code op_code;
-       struct wpabuf *m1;
-
-       /*
-        * Request for DeviceInfo, i.e., M1 TLVs. This is a start of WPS
-        * registration over UPnP with the AP acting as an Enrollee. It should
-        * be noted that this is frequently used just to get the device data,
-        * i.e., there may not be any intent to actually complete the
-        * registration.
-        */
-
-       if (peer->wps)
-               wps_deinit(peer->wps);
-
-       os_memset(&cfg, 0, sizeof(cfg));
-       cfg.wps = hapd->wps;
-       cfg.pin = (u8 *) hapd->conf->ap_pin;
-       cfg.pin_len = os_strlen(hapd->conf->ap_pin);
-       wps = wps_init(&cfg);
-       if (wps == NULL)
-               return NULL;
-
-       m1 = wps_get_msg(wps, &op_code);
-       if (m1 == NULL) {
-               wps_deinit(wps);
-               return NULL;
-       }
-
-       peer->wps = wps;
-
-       return m1;
-}
-
-
-static struct wpabuf *
-hostapd_rx_req_put_message(void *priv, struct upnp_wps_peer *peer,
-                          const struct wpabuf *msg)
-{
-       enum wps_process_res res;
-       enum wsc_op_code op_code;
-
-       /* PutMessage: msg = InMessage, return OutMessage */
-       res = wps_process_msg(peer->wps, WSC_UPnP, msg);
-       if (res == WPS_FAILURE)
-               return NULL;
-       return wps_get_msg(peer->wps, &op_code);
-}
-
-
 static int hostapd_rx_req_put_wlan_response(
        void *priv, enum upnp_wps_wlanevent_type ev_type,
        const u8 *mac_addr, const struct wpabuf *msg,
@@ -914,14 +825,6 @@ static int hostapd_rx_req_put_wlan_response(
 }
 
 
-static int hostapd_rx_req_set_selected_registrar(void *priv,
-                                                const struct wpabuf *msg)
-{
-       struct hostapd_data *hapd = priv;
-       return wps_registrar_set_selected_registrar(hapd->wps->registrar, msg);
-}
-
-
 static int hostapd_wps_upnp_init(struct hostapd_data *hapd,
                                 struct wps_context *wps)
 {
@@ -933,11 +836,9 @@ static int hostapd_wps_upnp_init(struct hostapd_data *hapd,
        if (ctx == NULL)
                return -1;
 
-       ctx->rx_req_get_device_info = hostapd_rx_req_get_device_info;
-       ctx->rx_req_put_message = hostapd_rx_req_put_message;
        ctx->rx_req_put_wlan_response = hostapd_rx_req_put_wlan_response;
-       ctx->rx_req_set_selected_registrar =
-               hostapd_rx_req_set_selected_registrar;
+       if (hapd->conf->ap_pin)
+               ctx->ap_pin = os_strdup(hapd->conf->ap_pin);
 
        hapd->wps_upnp = upnp_wps_device_init(ctx, wps, hapd);
        if (hapd->wps_upnp == NULL) {