/*
 *  wlan_test.c  --  wlan test application
 *
 *  Copyright (c) 2017 Rockchip Electronics Co. Ltd.
 *  Author: Panzhenzhuan Wang <randy.wang@rock-chips.com>
 *
 * 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.
 */
//WLAN test program
#include <stdio.h>

//open()相关头文件
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

//error相关头文件
#include <errno.h>
#include <string.h>
#include "wlan_test.h"

#define LOG_TAG "wlan_test"
#include "common.h"
#define WLAN_PROC_ERR -45

#define WLAN_START_UP_COMMAND "ifconfig wlan0 up"
#define WLAN_MANAGE_START "wpa_supplicant -i wlan0 -c /etc/wpa_supplicant.conf &"
#define WLAN_SCAN_COMMAND "wpa_cli -i wlan0 scan 'wlan0'"
#define WLAN_SCAN_RESULT  "wpa_cli -i wlan0 scan_r | busybox awk 'NR>=3{print $5,$3}'"
#define SCAN_RESULT_FILE "/tmp/wlan_scan_result.txt"

#define AP_SUPPORT_CMD "iw list | busybox grep AP > %s"
#define AP_SUPPORT_FILE "/tmp/wlan_ap_result.txt"
#define SOFTAP_MSG "dmesg | grep SSID | tail -n 6 > %s"
#define MSG_FILE "/tmp/softap_start_msg.txt"

#ifdef PCBA_PX3SE
#define CONNECT_AP_CMD "wpa_supplicant -i wlan0 -c /data/wpa_conf &"
#define CONNECT_AP_STATUS "wpa_cli status > %s"
#define CONNECT_AP_RESULT_FILE "/tmp/wlan_connect_result.txt"

#define REPORT_SSID     "Rockchip" //指定扫描的的wifi ssid名称
#endif

//* 1、关闭后台wpa_supplicant进程
static int close_wpa_supplicant(void)
{
    int test_flag = -1;
    char pid_buf[64];
    char cmd[64];
    FILE* pp;
    printf("====================function : %s start =================\n",__func__);
    pp = popen("ps |grep wpa_supplicant|awk 'NR==1 {print $1}'","r");
    //pp = popen("ps | grep wpa_supplicant | grep -v grep","r");
    //如果文件打开失败，则输出错误信息
    if (!pp)
    {
        printf("%s popen err%s\n",__func__,strerror(errno));
        return -1;
    }
    memset(pid_buf,0,sizeof(pid_buf));
    fgets(pid_buf,sizeof(pid_buf),pp);
    pclose(pp);
    printf("Get pid_buf is: \t %s\n",pid_buf);
    sprintf(cmd,"kill -9 %d",atoi(pid_buf));
    printf("cmd is: %s\n",cmd);
    system(cmd);
    printf("====================function : %s finish =================\n",__func__);
    return 0;
}

/*
 * RSSI Levels as used by notification icon
 *
 * Level 4  -55 <= RSSI
 * Level 3  -66 <= RSSI < -55
 * Level 2  -77 <= RSSI < -67
 * Level 1  -88 <= RSSI < -78
 * Level 0         RSSI < -88
 */
static int calc_rssi_lvl(int rssi)
{
	if (rssi >= -55)
		return 4;
	else if (rssi >= -66)
		return 3;
	else if (rssi >= -77)
		return 2;
	else if (rssi >= -88)
		return 1;
	else
		return 0;
}


/* 将扫描到的WiFi信息输出到path对应文件
 * 并显示第一个WiFi名字和信号强度
*/
int wlan_exec(const char *cmd, char *path)
{
    char result_buf[128];
    char ssid[128];
    int signal_level;
    int ch=0;
    int fd;

     printf("=================== function :%s start======================\n\n",__func__);

    memset(result_buf, 0, sizeof(result_buf));
    memset(ssid, 0, sizeof(ssid));
    //显示第一个WiFi信息
    FILE *pp = popen(cmd, "r");
    //如果文件打开失败，则输出错误信息
    if (!pp)
    {
        printf("%s popen err%s\n",__func__,strerror(errno));
        return -1;
    }

#ifdef PCBA_PX3SE
    /*chad.ma add below*/
    if (path != NULL) {
        fd = open(path, O_CREAT | O_WRONLY	| O_TRUNC);
        if (fd <0) {
            log_err("open %s fail, errno = %d\n", path, errno);
            return errno;
        }
    }
    /*chad.ma add up*/
#endif

	while(!feof(pp))
	{
	    fscanf(pp,"%s",ssid);
	    fscanf(pp,"%d",&signal_level);
	    //根据信号强度判断WiFi是否正常工作
	    if(signal_level< -200||signal_level >=0)
        {
            printf("wlan scan result is null\n");
            return -1;
        }
#ifdef PCBA_PX3SE
        snprintf(result_buf,sizeof(result_buf),
                    "SSID:%s Signal Level:%d RSSI:%d\n",
                    ssid, signal_level, calc_rssi_lvl(signal_level));
        int w_len = write(fd, result_buf, strlen(result_buf));
		if (w_len <= 0) {
			log_err("Write %s fail, errno = %d\n", path, errno);
			return errno;
		}
#endif

	    printf("SSID is: %s,signal level is: %d\n",ssid,signal_level);
	    printf("SSID is: %s,rssi is: %d\n",ssid,calc_rssi_lvl(signal_level));
	}
	pclose(pp);

#ifdef PCBA_PX3SE
    close(fd);
#endif
	printf("\n=================== function :%s finish======================\n",__func__);
	return 0;
}

void *wlan_test(void *argv)
{
    int test_flag=-1;
    char cmd[128];
    printf("===================Wlan test start======================\n");
    //sprintf(cmd,"aplay %s/wlan_test_start.wav",AUDIO_PATH);
    //system(cmd);
    //system("aplay /data/cfg/test/wlan_test_start.wav");

    //1、先关闭后台wpa_supplicant进程，然后先卸载WLAN驱动固件，然后重新加载驱动固件
    test_flag = close_wpa_supplicant();
    system("echo 1 > /sys/class/rkwifi/driver");

    //2、开启wlan0端口
    int status = system(WLAN_START_UP_COMMAND);

    /* add by chad ma 20180629 below judge*/
    if(status == -1) {
        printf("system run error ....\n");
        goto fail;
    } else {
        log_info("exit status value = [0x%x]\n", status);
        if (WIFEXITED(status))
        {
            if (0 == WEXITSTATUS(status))
            {
                log_info("run WLAN_START_UP_COMMAND successfully.\n");
                //ret = 0;
            }
            else
            {
                system("rfkill unblock wifi");
            }
        }
        else
        {
            log_info("exit status = [%d]\n", WEXITSTATUS(status));

            //ret = -1;
        }
    }
    sleep(2);

    //3、启动无线网卡管理程序wpa_supplicant
    system(WLAN_MANAGE_START);
    sleep(2);

    //4、扫描无线网络
    system(WLAN_SCAN_COMMAND);
    sleep(3);

    //5、显示出WiFi信息
    test_flag = wlan_exec(WLAN_SCAN_RESULT,SCAN_RESULT_FILE);
    if(test_flag<0)
    {
        //第一次scan失败，再执行一遍;
        system(WLAN_SCAN_COMMAND);
        sleep(2);
        test_flag = wlan_exec(WLAN_SCAN_RESULT,SCAN_RESULT_FILE);
        if(test_flag<0)
        goto fail;
        //return -1;
    }

    printf("===================Wlan test success ======================\n");
    system("ifconfig wlan0 down");
    system("busybox killall wpa_supplicant");      //关闭无线管理程序wpa_supplicant
    return (void*)test_flag;
fail:
    printf("===================Wlan test failed======================\n");
    system("ifconfig wlan0 down");
    system("busybox killall wpa_supplicant");      //关闭无线管理程序wpa_supplicant
    return (void*)test_flag;
}

int main(int argc, char *argv[])
{
    int test_flag = 0,err_code = 0;
    char read_buf[128];
    int find_flag = 0;
    FILE *fp;
    char buf[COMMAND_VALUESIZE] = "wlan_test";
    char result[COMMAND_VALUESIZE] = RESULT_PASS;
    test_flag = (int)wlan_test(argv[0]);
    if(test_flag < 0)
    {
        strcpy(result,RESULT_FAIL);
        err_code = WLAN_PROC_ERR;
    }

#ifdef PCBA_PX3SE
    //从扫描结果保存文件中，扫描指定ssid的signal
    fp = fopen(SCAN_RESULT_FILE, "r");
    memset(read_buf, 0 ,sizeof(read_buf));
     //如果文件打开失败，则输出错误信息
    if (!fp)
    {
        printf("%s fopen err:%s\n",__func__,strerror(errno));
        return -1;
    }

    while(!feof(fp))
    {
        fgets(read_buf,sizeof(read_buf),fp);
        if(strstr(read_buf,REPORT_SSID)!= NULL)
        {
            find_flag = 1;
            //注意：read_buf会将一行的回车符也读进去，这里要对read_buf进行改造
            int len = strlen(read_buf);
            if ( len < 128) {
                //assert(len < 128);
                read_buf[len - 1] = '\0';
                printf("\n ###Found!!! %s ###\n",read_buf);
                break;
            } else {
                printf("%s ERROR!!! read to much from one line\n",__func__);
            }
        }
    }
    fclose(fp);
    //将保存结果复制到buf中
    if (find_flag) {
        strcat(buf, ": ");
        strcat(buf, read_buf);
    }

    //删除扫描文件
    //remove(SCAN_RESULT_FILE);
    printf("\n $$ %s :not found %s $$\n",buf , REPORT_SSID);
#endif
    send_msg_to_server(buf, result, err_code);
}