examples/protocols/sockets/udp_client/main/xf_main.c

sockets udp 客户端 示例。

 
/* ==================== [Includes] ========================================== */
 
#include "lwip/err.h"
#include "lwip/sockets.h"
#include "lwip/sys.h"
#include "lwip/netdb.h"
#include "lwip/inet.h"
 
#include "xf_utils.h"
#include "xf_wifi.h"
#include "xf_netif.h"
#include "xf_osal.h"
 
#include "ex_easy_wifi.h"
 
/* ==================== [Defines] =========================================== */
 
#define PORT                CONFIG_EXAMPLE_PORT
#define EXAMPLE_DELAY_MS    1000
 
/* ==================== [Typedefs] ========================================== */
 
/* ==================== [Static Prototypes] ================================= */
 
static void _example_thread(void *argument);
 
/* ==================== [Static Variables] ================================== */
 
static const char *TAG = "sta";
static const char *payload = "This is a message from client.";
 
static xf_osal_thread_t s_thread_hdl = NULL;
#define EX_THREAD_NAME          "ex_thread"
#define EX_THREAD_PRIORITY      XF_OSAL_PRIORITY_NORMOL
#define EX_THREAD_STACK_SIZE    (1024 * 4)
static const xf_osal_thread_attr_t s_thread_attr = {
    .name       = EX_THREAD_NAME,
    .priority   = EX_THREAD_PRIORITY,
    .stack_size = EX_THREAD_STACK_SIZE,
};
 
/* ==================== [Macros] ============================================ */
 
/* ==================== [Global Functions] ================================== */
 
void xf_main(void)
{
    s_thread_hdl = xf_osal_thread_create(_example_thread, NULL, &s_thread_attr);
    if (s_thread_hdl == NULL) {
        XF_LOGE(TAG, "xf_osal_thread_create error");
        return;
    }
}
 
/* ==================== [Static Functions] ================================== */
 
static void _example_thread(void *argument)
{
    UNUSED(argument);
 
    ex_easy_wifi_sta();
 
    /* 默认网关为主机 */
    xf_ip4_addr_t host_ip4 = {0};
    host_ip4 = ex_easy_wifi_sta_get_gw_ip();
 
    char rx_buffer[128];
    int addr_family = 0;
    int ip_protocol = 0;
    int sockfd = 0;
 
    struct sockaddr_in dest_addr = {0};
    struct sockaddr_in source_addr = {0};
    struct timeval timeout = {0};
 
    dest_addr.sin_addr.s_addr   = host_ip4.addr;
    dest_addr.sin_family        = AF_INET;
    dest_addr.sin_port          = htons(PORT);
    addr_family                 = AF_INET;
    ip_protocol                 = IPPROTO_IP;
 
    sockfd = socket(addr_family, SOCK_DGRAM, ip_protocol);
    if (sockfd < 0) {
        XF_LOGE(TAG, "Unable to create socket: errno %d", errno);
        goto l_err;
    }
 
    /* 设置超时时间 */
    timeout.tv_sec = 10;
    timeout.tv_usec = 0;
    setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(timeout));
 
    XF_LOGI(TAG, "Socket created, sending to " XF_IPSTR ":%d",
            XF_IP2STR(&host_ip4), PORT);
 
    for (;;) {
        xf_osal_delay_ms(EXAMPLE_DELAY_MS);
 
        /* 发送 */
        int err = sendto(sockfd,
                         payload, strlen(payload), 0,
                         (struct sockaddr *)&dest_addr, sizeof(dest_addr));
        if (err < 0) {
            XF_LOGE(TAG, "Error occurred during sending: errno %d", errno);
            continue;
        }
        XF_LOGI(TAG, "Message sent: \"%s\".", payload);
 
        /* 接收 */
        socklen_t socklen = sizeof(source_addr);
        int len = recvfrom(sockfd,
                           rx_buffer, sizeof(rx_buffer) - 1, 0,
                           (struct sockaddr *)&source_addr, &socklen);
        if (len < 0) {
            XF_LOGE(TAG, "recvfrom failed: errno %d", errno);
            continue;
        }
 
        /* 收到数据 */
        rx_buffer[len] = 0;
        XF_LOGI(TAG, "Received %d bytes from " XF_IPSTR ":",
                len, XF_IP2STR(&host_ip4));
        XF_LOGI(TAG, "%s", rx_buffer);
    }
 
l_err:;
    if (sockfd != -1) {
        XF_LOGE(TAG, "Shutting down socket...");
        shutdown(sockfd, 0);
        closesocket(sockfd);
    }
    xf_osal_thread_delete(NULL);
}