/*
 * Copyright (C) 2017 XRADIO TECHNOLOGY CO., LTD. All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *    1. Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *    2. Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the
 *       distribution.
 *    3. Neither the name of XRADIO TECHNOLOGY CO., LTD. nor the names of
 *       its contributors may be used to endorse or promote products derived
 *       from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <stdio.h>
#include <string.h>

#include "net/wlan/wlan.h"
#include "net/wlan/wlan_defs.h"
#include "kernel/os/os.h"

#include "common/framework/platform_init.h"
#include "common/framework/net_ctrl.h"

#include "smartlink/sc_assistant.h"
#include "smartlink/smart_config/wlan_smart_config.h"

#define SC_TIME_OUT 120000 //120s

static char sc_key[17] = "1234567812345678"; // smart config aes key

static void SmartConfigDemo(void)
{
	wlan_smart_config_status_t sc_status;
	wlan_smart_config_result_t sc_result;
	sc_assistant_fun_t sca_fun;
	sc_assistant_time_config_t config;

	memset(&sc_result, 0, sizeof(wlan_smart_config_result_t));

	/*
	 * Initialize the distribution network assistant,
	 * which can improve the success rate of distribution network
	 */
	sc_assistant_get_fun(&sca_fun);
	config.time_total = SC_TIME_OUT;
	config.time_sw_ch_long = 400;
	config.time_sw_ch_short = 100;
	sc_assistant_init(wlan_netif_get(WLAN_MODE_NONE), &sca_fun, &config);

	/* start smart config */
	sc_status = wlan_smart_config_start(wlan_netif_get(WLAN_MODE_NONE), sc_key);
	if (sc_status != WLAN_SMART_CONFIG_SUCCESS) {
		printf("smartconfig start fiald!\n");
		goto out;
	}

	printf("%s getting ssid and psk...\n", __func__);

	/* Blocking waiting for distribution network to complete */
	if (wlan_smart_config_wait(SC_TIME_OUT) == WLAN_SMART_CONFIG_TIMEOUT) {
		printf("%s get ssid and psk timeout\n", __func__);
		goto out;
	}
	printf("%s get ssid and psk finished\n", __func__);

	/* if complete, then get the smart config result */
	if (wlan_smart_config_get_status() == SC_STATUS_COMPLETE && \
	    smartconfig_get_result(&sc_result) == WLAN_SMART_CONFIG_SUCCESS) {
		printf("ssid:%s psk:%s random:%d\n", (char *)sc_result.ssid,
		       (char *)sc_result.passphrase, sc_result.random_num);

		/* if get result, then connect ap and send the smart config connect ack */
		if (!wlan_smart_config_connect_ack(wlan_netif_get(WLAN_MODE_NONE),
		                                   SC_TIME_OUT, &sc_result)) {
			printf("connect and ack ok\n");
		}
	}

out:
	/* stop airkiss and deinit distribution network assistant */
	wlan_smart_config_stop();
	sc_assistant_deinit(wlan_netif_get(WLAN_MODE_NONE));
}

int main(void)
{
	platform_init();

	printf("SmartConfig demo started.\n\n");

	SmartConfigDemo();

	printf("\nSmartConfig demo over\n");
	return 0;
}