ESP8266_RTOS_SDK/examples/wifi/smart_config/main/user_main.c

/* smartconfig example

   This example code is in the Public Domain (or CC0 licensed, at your option.)

   Unless required by applicable law or agreed to in writing, this
   software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
   CONDITIONS OF ANY KIND, either express or implied.
*/

#include "freertos/FreeRTOS.h"
#include "freertos/task.h"

#include "lwip/sockets.h"
#include "lwip/dns.h"
#include "lwip/netdb.h"

#include "airkiss.h"

#define server_ip "192.168.101.142"
#define server_port 9669


#define DEVICE_TYPE         "gh_9e2cff3dfa51"   // wechat public number
#define DEVICE_ID           "122475"            // model ID

#define DEFAULT_LAN_PORT    12476

static esp_udp ssdp_udp;
static struct espconn pssdpudpconn;
static os_timer_t ssdp_time_serv;

uint8_t  lan_buf[200];
uint16_t lan_buf_len;
uint8_t  udp_sent_cnt = 0;

const airkiss_config_t akconf = {
    (airkiss_memset_fn)& memset,
    (airkiss_memcpy_fn)& memcpy,
    (airkiss_memcmp_fn)& memcmp,
    0,
};

static void airkiss_wifilan_time_callback(void)
{
    uint16_t i;
    airkiss_lan_ret_t ret;

    if ((udp_sent_cnt++) > 30) {
        udp_sent_cnt = 0;
        os_timer_disarm(&ssdp_time_serv);//s
        //return;
    }

    ssdp_udp.remote_port = DEFAULT_LAN_PORT;
    ssdp_udp.remote_ip[0] = 255;
    ssdp_udp.remote_ip[1] = 255;
    ssdp_udp.remote_ip[2] = 255;
    ssdp_udp.remote_ip[3] = 255;
    lan_buf_len = sizeof(lan_buf);
    ret = airkiss_lan_pack(AIRKISS_LAN_SSDP_NOTIFY_CMD,
                           DEVICE_TYPE, DEVICE_ID, 0, 0, lan_buf, &lan_buf_len, &akconf);

    if (ret != AIRKISS_LAN_PAKE_READY) {
        printf("Pack lan packet error!");
        return;
    }

    ret = espconn_sendto(&pssdpudpconn, lan_buf, lan_buf_len);

    if (ret != 0) {
        printf("UDP send error!");
    }

    printf("Finish send notify!\n");
}

static void airkiss_wifilan_recv_callbk(void* arg, char* pdata, unsigned short len)
{
    uint16_t i;
    remot_info* pcon_info = NULL;

    airkiss_lan_ret_t ret = airkiss_lan_recv(pdata, len, &akconf);
    airkiss_lan_ret_t packret;

    switch (ret) {
        case AIRKISS_LAN_SSDP_REQ:
            espconn_get_connection_info(&pssdpudpconn, &pcon_info, 0);
            printf("remote ip: %d.%d.%d.%d \r\n", pcon_info->remote_ip[0], pcon_info->remote_ip[1],
                   pcon_info->remote_ip[2], pcon_info->remote_ip[3]);
            printf("remote port: %d \r\n", pcon_info->remote_port);

            pssdpudpconn.proto.udp->remote_port = pcon_info->remote_port;
            memcpy(pssdpudpconn.proto.udp->remote_ip, pcon_info->remote_ip, 4);
            ssdp_udp.remote_port = DEFAULT_LAN_PORT;

            lan_buf_len = sizeof(lan_buf);
            packret = airkiss_lan_pack(AIRKISS_LAN_SSDP_RESP_CMD,
                                       DEVICE_TYPE, DEVICE_ID, 0, 0, lan_buf, &lan_buf_len, &akconf);

            if (packret != AIRKISS_LAN_PAKE_READY) {
                printf("Pack lan packet error!");
                return;
            }

            printf("\r\n\r\n");

            for (i = 0; i < lan_buf_len; i++) {
                printf("%c", lan_buf[i]);
            }

            printf("\r\n\r\n");

            packret = espconn_sendto(&pssdpudpconn, lan_buf, lan_buf_len);

            if (packret != 0) {
                printf("LAN UDP Send err!");
            }

            break;

        default:
            printf("Pack is not ssdq req!%d\r\n", ret);
            break;
    }
}

void airkiss_start_discover(void)
{
    ssdp_udp.local_port = DEFAULT_LAN_PORT;
    pssdpudpconn.type = ESPCONN_UDP;
    pssdpudpconn.proto.udp = &(ssdp_udp);
    espconn_regist_recvcb(&pssdpudpconn, airkiss_wifilan_recv_callbk);
    espconn_create(&pssdpudpconn);

    os_timer_disarm(&ssdp_time_serv);
    os_timer_setfn(&ssdp_time_serv, (os_timer_func_t*)airkiss_wifilan_time_callback, NULL);
    os_timer_arm(&ssdp_time_serv, 1000, 1);//1s
}


void smartconfig_done(sc_status status, void* pdata)
{
    switch (status) {
        case SC_STATUS_WAIT:
            printf("SC_STATUS_WAIT\n");
            break;

        case SC_STATUS_FIND_CHANNEL:
            printf("SC_STATUS_FIND_CHANNEL\n");
            break;

        case SC_STATUS_GETTING_SSID_PSWD:
            printf("SC_STATUS_GETTING_SSID_PSWD\n");
            sc_type* type = pdata;

            if (*type == SC_TYPE_ESPTOUCH) {
                printf("SC_TYPE:SC_TYPE_ESPTOUCH\n");
            } else {
                printf("SC_TYPE:SC_TYPE_AIRKISS\n");
            }

            break;

        case SC_STATUS_LINK:
            printf("SC_STATUS_LINK\n");
            struct station_config* sta_conf = pdata;

            wifi_station_set_config(sta_conf);
            wifi_station_disconnect();
            wifi_station_connect();
            break;

        case SC_STATUS_LINK_OVER:
            printf("SC_STATUS_LINK_OVER\n");

            if (pdata != NULL) {
                //SC_TYPE_ESPTOUCH
                uint8 phone_ip[4] = {0};

                memcpy(phone_ip, (uint8*)pdata, 4);
                printf("Phone ip: %d.%d.%d.%d\n", phone_ip[0], phone_ip[1], phone_ip[2], phone_ip[3]);
            } else {
                //SC_TYPE_AIRKISS - support airkiss v2.0
                airkiss_start_discover();
            }

            smartconfig_stop();
            break;
    }

}

void smartconfig_task(void* pvParameters)
{
    smartconfig_start(smartconfig_done);

    vTaskDelete(NULL);
}

/******************************************************************************
 * FunctionName : user_init
 * Description  : entry of user application, init user function here
 * Parameters   : none
 * Returns      : none
*******************************************************************************/
void user_init(void)
{
    printf("SDK version:%s\n", esp_get_idf_version());

    /* need to set opmode before you set config */
    wifi_set_opmode(STATION_MODE);

    xTaskCreate(smartconfig_task, "smartconfig_task", 1024, NULL, 2, NULL);
}