From bdd7fb7a137c4b6e2bacc95976f6c901f590e9ed Mon Sep 17 00:00:00 2001
From: Xia Xiao Tian <xxt@201608-4.(none)>
Date: Mon, 7 Nov 2016 14:59:35 +0800
Subject: [PATCH] Add wps API

---
 components/esp32/component.mk      |   2 +-
 components/esp32/include/esp_wps.h | 131 +++++++++++++++++++++++++++++
 2 files changed, 132 insertions(+), 1 deletion(-)
 create mode 100644 components/esp32/include/esp_wps.h

diff --git a/components/esp32/component.mk b/components/esp32/component.mk
index c658787d8..e483b58ee 100644
--- a/components/esp32/component.mk
+++ b/components/esp32/component.mk
@@ -10,7 +10,7 @@
 
 COMPONENT_SRCDIRS := . hwcrypto
 
-LIBS := core net80211 phy rtc pp wpa smartconfig coexist
+LIBS := core net80211 phy rtc pp wpa smartconfig coexist wps
 
 LINKER_SCRIPTS += -T esp32_out.ld -T esp32.common.ld -T esp32.rom.ld -T esp32.peripherals.ld
 
diff --git a/components/esp32/include/esp_wps.h b/components/esp32/include/esp_wps.h
new file mode 100644
index 000000000..1588d3619
--- /dev/null
+++ b/components/esp32/include/esp_wps.h
@@ -0,0 +1,131 @@
+// Copyright 2015-2016 Espressif Systems (Shanghai) PTE LTD
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#ifndef __ESP_WPS_H__
+#define __ESP_WPS_H__
+
+#include <stdbool.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+/** \defgroup WiFi_APIs WiFi Related APIs
+  * @brief WiFi APIs
+  */
+
+/** @addtogroup WiFi_APIs
+  * @{
+  */
+
+/** \defgroup WPS_APIs  WPS APIs
+  * @brief ESP32 WPS APIs
+  *
+  * WPS can only be used when ESP32 station is enabled.
+  *
+  */
+
+/** @addtogroup WPS_APIs
+  * @{
+  */
+
+typedef enum wps_type {
+    WPS_TYPE_DISABLE = 0,
+    WPS_TYPE_PBC,
+    WPS_TYPE_PIN,
+    WPS_TYPE_DISPLAY,
+    WPS_TYPE_MAX,
+} WPS_TYPE_t;
+
+enum wps_cb_status {
+    WPS_CB_ST_SUCCESS = 0,     /**< WPS succeed */
+    WPS_CB_ST_FAILED,          /**< WPS fail */
+    WPS_CB_ST_TIMEOUT,         /**< WPS timeout, fail */
+    WPS_CB_ST_WEP,             /**< WPS failed because that WEP is not supported */
+    WPS_CB_ST_SCAN_ERR,        /**< can not find the target WPS AP */
+};
+
+/**
+  * @brief     Enable Wi-Fi WPS function.
+  *
+  * @attention WPS can only be used when ESP32 station is enabled.
+  *
+  * @param     WPS_TYPE_t wps_type : WPS type, so far only WPS_TYPE_PBC and WPS_TYPE_PIN is supported
+  *
+  * @return    true  : succeed
+  * @return    false : fail
+  */
+bool esp_wifi_wps_enable(WPS_TYPE_t wps_type);
+
+/**
+  * @brief  Disable Wi-Fi WPS function and release resource it taken.
+  *
+  * @param  null
+  *
+  * @return true  : succeed
+  * @return false : fail
+  */
+bool esp_wifi_wps_disable(void);
+
+/**
+  * @brief     WPS starts to work.
+  *
+  * @attention WPS can only be used when ESP32 station is enabled.
+  *
+  * @param     null
+  *
+  * @return    true  : WPS starts to work successfully, but does not mean WPS succeed.
+  * @return    false : fail
+  */
+bool esp_wifi_wps_start(void);
+
+/**
+  * @brief  WPS callback.
+  *
+  * @param  int status : status of WPS, enum wps_cb_status.
+  *    -  If parameter status == WPS_CB_ST_SUCCESS in WPS callback, it means WPS got AP's
+  *       information, user can call wifi_wps_disable to disable WPS and release resource,
+  *       then call wifi_station_connect to connect to target AP.
+  *    -  Otherwise, it means that WPS fail, user can create a timer to retry WPS by
+  *       wifi_wps_start after a while, or call wifi_wps_disable to disable WPS and release resource.
+  *
+  * @return null
+  */
+typedef void (*wps_st_cb_t)(int status);
+
+/**
+  * @brief     Set WPS callback.
+  *
+  * @attention WPS can only be used when ESP32 station is enabled.
+  *
+  * @param     wps_st_cb_t cb : callback.
+  *
+  * @return    true  : WPS starts to work successfully, but does not mean WPS succeed.
+  * @return    false : fail
+  */
+bool esp_wifi_set_wps_cb(wps_st_cb_t cb);
+
+/**
+  * @}
+  */
+
+/**
+  * @}
+  */
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif /* __ESP_WPS_H__ */