1. 程式人生 > >Linux-根據域名建立Socket

Linux-根據域名建立Socket

【根據域名建立socket】:

#include <sys/socket.h>
#include <netinet/tcp.h>
#include <stdio.h>
#include <arpa/inet.h>
#include <sys/select.h>
#include <sys/time.h>
#include <errno.h>
#include <netdb.h>
#define LOGD printf
#define LOGI printf
#define LOGE printf
int just_sock_connect_url(char *url, int port)
{
    if(url == NULL) return -EINVAL;
    
    LOGI("just_sock_connect_url(%s:%d)", url, port);
    
    struct hostent *hptr = gethostbyname(url);
    if(hptr == NULL) return -EFAULT;
    
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd <= 0){
        LOGE("socket error");
        return -EFAULT;
    }

    struct sockaddr_in servaddr;
    memset((void*)&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(port);
    //inet_pton(AF_INET, ipBuf, &servaddr.sin_addr.s_addr);
    servaddr.sin_addr.s_addr = *((unsigned long*)hptr->h_addr_list[0]);
    
    int reval = connect(sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr));
    if (reval == 0) return sockfd;
    
    switch(errno)
    {
        case EISCONN:
            reval = 0;
            break;
        case EINPROGRESS:
        case EALREADY:
            reval = socket_wait_writable(sockfd, 30000)
; break; default: reval = -EFAULT; break; } if (reval < 0) { LOGE("connect [%s:%d] failed, reval=%d", url, port, reval); close(sockfd); return reval; } LOGI("connect [%s:%d] Success!", url, port); return sockfd; }

當connect返回負值但errno=EINPROGRESS/EALREADY時表示“正在處理中”,可用select機制來監控可寫性,socket一旦可寫就說明連線成功了:

static int socket_wait_writable(int sockfd, int timeout_ms)
{
    struct timeval tpstart,tpend,timeout_v;
    fd_set wset, eset;
    long time_used = 0;
    int reval = 0;

    LOGI("socket_wait(sockfd=%d,timeout_ms=%d)", sockfd, timeout_ms);

    gettimeofday(&tpstart,NULL);

    while(1)
    {
        gettimeofday(&tpend,NULL);
        time_used=1000*(tpend.tv_sec-tpstart.tv_sec);
        time_used += (tpend.tv_usec-tpstart.tv_usec)/1000;
        if (time_used > timeout_ms)
        {
            LOGI("write timeout1 [%d ms]", timeout_ms);
            return -ETIME;
        }
        long time_seft = timeout_ms - time_used;
        timeout_v.tv_sec = time_seft/1000;
        timeout_v.tv_usec = time_seft%1000 * 1000;

        FD_ZERO(&wset);
        FD_ZERO(&eset);
        FD_SET(sockfd, &wset);
        FD_SET(sockfd, &eset);
        reval = select(sockfd + 1, NULL, &wset, &eset , &timeout_v);
        if(reval == -1 && errno != EINTR)
        {
            LOGE("error in socket_wait 1:%s", strerror(errno));
            return -EFAULT;
        }
        else if( reval == 0)
        {
            LOGI("write timeout2 [%d ms]", JUST_SOCK_DEFAULT_TIMEOUT);
            return -ETIME;
        }
        else if(reval > 0 && FD_ISSET(sockfd, &eset))
        {
            LOGE("error in socket_wait 2:%s", strerror(errno));
            return -EFAULT;
        }
        else if ( reval > 0 && FD_ISSET(sockfd, &wset))
        {
            //write event:
            LOGI("socket[%d] writable ...", sockfd);
            break;
        }
        else
        {
            LOGE("unexpected err in socket_wait");
        }
    }
    return 0;
}

【測試用例】

void just_sock_connect_test(void)
{
    int reval = 0;
    int fd = -1;

    char *this_url = "www.baidu.com";
    fd = just_sock_connect_url(this_url, 80);
    if (fd < 0)
    {
        LOGE("just_sock_connect_url failed");
        return;
    }
    LOGE("just_sock_connect_url[%s] success", this_url);
}