OpenWrt：libubox之usock

usock便是對網絡套接字鏈接的簡單封裝。

類型標誌：

#define USOCK_TCP 0
#define USOCK_UDP 1
#define USOCK_SERVER          0x0100
#define USOCK_NOCLOEXEC       0x0200
#define USOCK_NONBLOCK        0x0400
#define USOCK_NUMERIC         0x0800
#define USOCK_IPV6ONLY        0x2000
#define USOCK_IPV4ONLY        0x4000
#define USOCK_UNIX            0x8000

接口函數：

/**
 * 建立一個sock
 * 
 * @param type - 類型標誌
 * @param host - 做爲server表示綁定本地地址；做爲client表示需鏈接的地址
 * @param service - 端口
 * @return - sock fd > 0； 錯誤 < 0
 */

int usock(int type, const char *host, const char *service);

客戶端代碼：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>

#include "libubox/usock.h"


int main(int argc, char** argv)
{
    struct sockaddr cli_addr;
    socklen_t len = sizeof(struct sockaddr);
    int type = USOCK_TCP | USOCK_NOCLOEXEC | USOCK_IPV4ONLY;
    const char *host = "127.0.0.1";
    const char *service = "8000";
    char recv_buf[1024] = {0};

    // printf("port: %s\n", usock_port(8888));

    int c_fd = usock(type, host, service);
    if(c_fd < 0) {
        perror("usock");
        return -1;
    }

    send(c_fd, "helloworld", 10, 0);
    sleep(2);
    close(c_fd);

    return 0;
}

服務端代碼：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include "libubox/usock.h"


int main(int argc, char** argv)
{
    struct sockaddr_in cli_addr;
    socklen_t len = sizeof(struct sockaddr);
    int type = USOCK_TCP | USOCK_SERVER | USOCK_NOCLOEXEC | USOCK_IPV4ONLY;
    const char *host = "127.0.0.1";
    const char *service = "8000";
    char recv_buf[1024] = {0};
    int connect_fd, u_fd;

    u_fd = usock(type, host, service);    
    if(u_fd < 0) {
        perror("usock");
        return -1;
    }

    while (1) {
        connect_fd = accept(u_fd, (struct sockaddr *)(&cli_addr), &len);
        if (connect_fd < 0) {
            perror("accept");
            return -1;
        }

        printf("client_addr: %s\n", inet_ntoa(cli_addr.sin_addr));
        recv(connect_fd, recv_buf, 1024, 0);
        printf("recv %s\n", recv_buf);
        close(connect_fd);
    }

    return 0;
}