import("http://build/lite/config/component/lite_component.gni")

lite_component("app") {
    features = [
        "startup",
        "tcpclient:net_demo",
    ]
}

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include "iot_gpio.h"

#include "ohos_init.h"
#include "cmsis_os2.h"

#include "net_demo.h"
#include "net_params.h"
#include "wifi_connecter.h"

#define LED_TASK_GPIO 9
static void NetDemoTask(void *arg) //一開始線程入口函數(shù)
{
    (void)arg;
    WifiDeviceConfig config = {0};                  //表示用于連接到指定 Wi-Fi 設(shè)備的 Wi-Fi 站配置。
    IoTGpioInit(LED_TASK_GPIO);                     //初始化IO口，為后文點燈做準(zhǔn)備
    IoTGpioSetDir(LED_TASK_GPIO, IOT_GPIO_DIR_OUT); //設(shè)置GPIO為輸出模式
    // 準(zhǔn)備AP的配置參數(shù)
    strcpy(config.ssid, PARAM_HOTSPOT_SSID); //從net_params.h拷貝WiFi的參數(shù)
    strcpy(config.preSharedKey, PARAM_HOTSPOT_PSK);
    config.securityType = PARAM_HOTSPOT_TYPE; //配置WiFi的安全模式
    osDelay(10);
    int netId = ConnectToHotspot(&config); //連接熱點
    int timeout = 10;
    while (timeout--) //等待10秒后開始執(zhí)行NetDemoTest
    {
        printf("After %d seconds, I will start %s test!\r\n", timeout, GetNetDemoName());
        osDelay(100);
    }
    while (1)
    {
        NetDemoTest(PARAM_SERVER_PORT, PARAM_SERVER_ADDR); //開始TCP連接，輸入端口號，ip地址
    }
    printf("disconnect to AP ...\r\n");
    // DisconnectWithHotspot(netId);
    printf("disconnect to AP done!\r\n");
}
static void NetDemoEntry(void)
{
    osThreadAttr_t attr;
    attr.name = "NetDemoTask";
    attr.attr_bits = 0U;
    attr.cb_mem = NULL;
    attr.cb_size = 0U;
    attr.stack_mem = NULL;
    attr.stack_size = 10240;
    attr.priority = osPriorityNormal;

    if (osThreadNew(NetDemoTask, NULL, &attr) == NULL)
    {
        printf("[NetDemoEntry] Falied to create NetDemoTask!\n");
    }
}
SYS_RUN(NetDemoEntry);

int sockfd = socket(AF_INET, SOCK_STREAM, 0); // AF_INET:IP 協(xié)議系列。SOCK_STREAM=1:TCP協(xié)議

struct sockaddr_in serverAddr = {0}; //描述互聯(lián)網(wǎng)套接字地址的結(jié)構(gòu)體
    serverAddr.sin_family = AF_INET;     // AF_INET表示IPv4協(xié)議
    serverAddr.sin_port = htons(port);   // 端口號，從主機字節(jié)序轉(zhuǎn)為網(wǎng)絡(luò)字節(jié)序
    if (inet_pton(AF_INET, host, &serverAddr.sin_addr) <= 0)
    { // 將主機IP地址從“點分十進制”字符串 轉(zhuǎn)化為 標(biāo)準(zhǔn)格式（32位整數(shù)）
        printf("inet_pton failed!\r\n");
        goto do_cleanup;
    }

// 嘗試和目標(biāo)主機建立連接，連接成功會返回0 ，失敗返回 -1
    if (connect(sockfd, (struct sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
    {
        printf("connect failed!\r\n");
        goto do_cleanup;
    }
    printf("connect to server %s success!\r\n", host);

// 建立連接成功之后，這個TCP socket描述符 —— sockfd 就具有了 “連接狀態(tài)”，發(fā)送、接收 對端都是 connect 參數(shù)指定的目標(biāo)主機和端口
    retval = send(sockfd, request, sizeof(request), 0); //發(fā)送request給目標(biāo)主機，成功會返回字符串長度 ，失敗返回 -1
    if (retval < 0)
    {
        printf("send request failed!\r\n");
        goto do_cleanup;
    }
    printf("send request{%s} %ld to server done!\r\n", request, retval);

retval = recv(sockfd, &response, sizeof(response), 0);//接收目標(biāo)主機的消息存入response，成功會返回字符串長度 ，失敗返回 -1
    if (retval <= 0) {
        printf("send response from server failed or done, %ld!\r\n", retval);
        goto do_cleanup;
    }
    response[retval] = '\0';
    printf("recv response{%s} %ld from server done!\r\n", response, retval);

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include "net_demo.h"
#include "net_common.h"
#define LED_TASK_GPIO 9
static char request[] = "Hello";
static char response[128] = "";
void TcpClientTest(const char *host, unsigned short port)
{
    ssize_t retval = 0;
    int sockfd = socket(AF_INET, SOCK_STREAM, 0); // AF_INET:IP 協(xié)議系列。SOCK_STREAM=1:TCP協(xié)議

    struct sockaddr_in serverAddr = {0}; //描述互聯(lián)網(wǎng)套接字地址的結(jié)構(gòu)體
    serverAddr.sin_family = AF_INET;     // AF_INET表示IPv4協(xié)議
    serverAddr.sin_port = htons(port);   // 端口號，從主機字節(jié)序轉(zhuǎn)為網(wǎng)絡(luò)字節(jié)序
    if (inet_pton(AF_INET, host, &serverAddr.sin_addr) <= 0)
    { // 將主機IP地址從“點分十進制”字符串 轉(zhuǎn)化為 標(biāo)準(zhǔn)格式（32位整數(shù)）
        printf("inet_pton failed!\r\n");
        goto do_cleanup;
    }
    // 嘗試和目標(biāo)主機建立連接，連接成功會返回0 ，失敗返回 -1
    if (connect(sockfd, (struct sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
    {
        printf("connect failed!\r\n");
        goto do_cleanup;
    }
    printf("connect to server %s success!\r\n", host);

    // 建立連接成功之后，這個TCP socket描述符 —— sockfd 就具有了 “連接狀態(tài)”，發(fā)送、接收 對端都是 connect 參數(shù)指定的目標(biāo)主機和端口
    retval = send(sockfd, request, sizeof(request), 0); //發(fā)送request給目標(biāo)主機，成功會返回字符串長度 ，失敗返回 -1
    if (retval < 0)
    {
        printf("send request failed!\r\n");
        goto do_cleanup;
    }
    printf("send request{%s} %ld to server done!\r\n", request, retval);
    while (1)
    {
        retval = recv(sockfd, &response, sizeof(response), 0); //接收目標(biāo)主機的消息存入response，成功會返回字符串長度 ，失敗返回 -1
        if (retval <= 0)
        {
            printf("send response from server failed or done, %ld!\r\n", retval);
            goto do_cleanup;
        }
        response[retval] = '\0';
        printf("recv response{%s} %ld from server done!\r\n", response, retval);

        if (response[0] == 'o' && response[1] == 'n')
        {
            IoTGpioSetOutputVal(LED_TASK_GPIO, 0); //開燈
            printf("The led is on\n");
        }
        if (response[0] == 'o' && response[1] == 'f' && response[2] == 'f')
        {
            IoTGpioSetOutputVal(LED_TASK_GPIO, 1); //關(guān)燈
            printf("The led is off\n");
        }
    }

do_cleanup:
    printf("do_cleanup...\r\n");
    //     close(sockfd);//關(guān)閉套接字
}
CLIENT_TEST_DEMO(TcpClientTest);