[C++] Linux TCP Socket 实例- 阻塞

Linux平台 TCP Socket通信实例，发现用代码注释记笔记也不错

TCP server 阻塞

// 两个进程通过socket进行通信，client需要知道server的，server却不需要实现知道client的。
// socket需要知道type和address domain；
// address domain：unix domain用于同主机进程间，通过文件系统实现；Internet domain用于两台机器间，需要IP和Port信息。一般2000以下的port保留。
// socket type：stream（意味着持续读取，TCP是一种协议）、datagram（一次需要读完整个信息，UDP是一种协议）。

// port复用机制的理解

/**
stream server
1.socket
2.bind
3.listen
4.accept
*/
#include <cstdlib>              /* exit() */
#include <cstdio>               /* perror(): 打印信息+发生错误的原因，可用于定位。 */
#include <cstring>              /* memset memcpy*/
#include <iostream>             /* cin cout */
#include <sys/types.h>          /* 为了满足一些 BSD系统添加头文件*/
#include <sys/socket.h>         /* socket(); listen(); baccept(); socklen_t */
#include <netinet/in.h>         /* struct sockaddr_in: 保存socket信息; ntohl(), ntohs(), htonl() and htons()*/
#include <unistd.h>             /* read() write(), 不是C语言范畴，所以没有cxxxx的实现 */
    // http://man7.org/linux/man-pages/man2/read.2.html
    // http://man7.org/linux/man-pages/man2/write.2.html

/**
\brief  错误处理函数
*/
void tzError(const char *msg)
{
    perror(msg);
    exit(1);    // 一般不同原因不同的exit code更为规范
}

/**
\brief  主函数
*/
int main(int argc, char *argv[]){
    // 参数check
    if (argc < 2) {
        fprintf(stderr,"ERROR, no [port] arg provided\n");
        exit(1);
    }
    // 生成socke + check
    int sockfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
        // int socket(int domain, int type, int protocol);
        // http://man7.org/linux/man-pages/man2/socket.2.html
        // domain:
        //      AF_UNIX     用于本地通信
        //      AF_INET     IPv4
        //      AF_INET6    IPv6
        // type:
        //      SOCK_STREAM     面向连接通信
        //      SOCK_DGRAM      面向无连接通信
        // protocol:
        //      SOCK_NONBLOCK   非阻塞
        //      SOCK_CLOEXEC    子类不继承父类的socket fd
        // Return:
        //      建立的socket：对于server用于监听，对于client用于向指定位置发送
    if (sockfd < 0){
        tzError("socket()");
    }
    // bind socket + check
    int port_no = atoi(argv[1]);
    struct sockaddr_in serv_addr;
    serv_addr.sin_family = AF_INET;             // IPv4
    serv_addr.sin_addr.s_addr = INADDR_ANY;     // 相当于inet_addr("0.0.0.0")，所有网卡。
    serv_addr.sin_port = htons(port_no);         // 需要网络序转换
    int bind_ret = bind(sockfd, 
                    (struct sockaddr*)&serv_addr, 
                    sizeof(serv_addr));
        // int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
        // http://man7.org/linux/man-pages/man2/bind.2.html
        // socket()产生sockdf，有了name space(address family)但是没有具体地址，bind完成此任务
    if( bind_ret<0 ){
        tzError("bind()");
    }
    // listen
    listen(sockfd,5);
        // int listen(int sockfd, int backlog);
        // http://man7.org/linux/man-pages/man2/listen.2.html
        // listen让sockfd对应的socket成为被动socket，用于accetp
        // backlog: 最大可以入队等待处理的连接数量。如果大于128(somaxconn)可能被内核截断，cat /proc/sys/net/core/somaxconn
        // 如果尝试connect的链接超过backlog，client收到ECONNREFUSED错误，如果有重传机制可能忽略此错误。
    // accept + check
    struct sockaddr_in client_addr;
    socklen_t clilen_len = sizeof(client_addr);
    int new_sockfd = accept(sockfd, 
                 (struct sockaddr *) &client_addr, 
                 &clilen_len);
        // int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen)
        // http://man7.org/linux/man-pages/man2/accept.2.html
        // 用于面向连接socket(SOCK_STREAM, SOCK_SEQPACKET)，提取队列中第一个请求，创建新的connected socket
        // 返回新创建的socket的fd，原始的被动sockfd不受影响。
        // addr: socketaddr指针，记录peer的信息，信息格式取决于socket的定义。
            // 如果等待队列为空，且socket没有设置nonblocking-->阻塞等待；
            // 如果等待队列为空，且socket设置nonblocking，返回错误：EAGAIN or EWOULDBLOCK
        
            // 为了处理对socket的新的connect，可使用：select(2), poll(2), or epoll(7)-->有尝试连接的请求会出触发"readable event"
            // 也可以通过socket()设置发送SIGIO中断。
        // addrlen：传入传输参数，实际的大小会重新保存到addrlen中
    if (new_sockfd < 0) 
          tzError("accept()");
    // 接收-发送信息
#define RECV_BUF_SIZE   256
    char buffer[RECV_BUF_SIZE];
    std::memset(buffer, 0, RECV_BUF_SIZE);  // 用0进行set
    int valread = read( new_sockfd , buffer, RECV_BUF_SIZE);
    std::cout << "read:" << valread << "char, content: " << buffer << std::endl;
    write(new_sockfd , "get msg", 7);
    // close sockets
    close(new_sockfd); 
    close(sockfd);

    return 0;
}

// ref:
//      https://www.cnblogs.com/fuchongjundream/p/3914696.html
//      http://www.linuxhowtos.org/C_C++/socket.htm
//      

TCP Client - 单次连接发送接收测试

// 两个进程通过socket进行通信，client需要知道server的，server却不需要实现知道client的。
// socket需要知道type和address domain；
// address domain：unix domain用于同主机进程间，通过文件系统实现；Internet domain用于两台机器间，需要IP和Port信息。一般2000以下的port保留。
// socket type：stream（意味着持续读取，TCP是一种协议）、datagram（一次需要读完整个信息，UDP是一种协议）。

// port复用机制的理解

/**
stream client
1.socket
2.connect
*/
#include <cstdlib>              /* exit() */
#include <cstdio>               /* perror(): 打印信息+发生错误的原因，可用于定位。 */
#include <cstring>              /* memset memcpy strcpy string.c_str()*/
#include <iostream>             /* cin cout */
#include <sys/types.h>          /* 为了满足一些 BSD系统添加头文件*/
#include <sys/socket.h>         /* socket(); listen(); baccept(); socklen_t */
// #include <netinet/in.h>         /* struct sockaddr_in: 保存socket信息; ntohl(), ntohs(), htonl() and htons()*/
#include <netdb.h>              /* hostent,IP地址 */

#include <unistd.h>             /* read() write(), 不是C语言范畴，所以没有cxxxx的实现 */
    // http://man7.org/linux/man-pages/man2/read.2.html
    // http://man7.org/linux/man-pages/man2/write.2.html

/**
\brief  错误处理函数
*/
void tzError(const char *msg)
{
    perror(msg);
    exit(1);    // 一般不同原因不同的exit code更为规范
}

/**
\brief  主函数
*/
int main(int argc, char *argv[]){
    // 参数check
    if (argc < 3) {
        tzError("ERROR, no [IP] [port] arg provided");
    }
    // 生成socke + check
    int sockfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
        // int socket(int domain, int type, int protocol);
        // http://man7.org/linux/man-pages/man2/socket.2.html
        // domain:
        //      AF_UNIX     用于本地通信
        //      AF_INET     IPv4
        //      AF_INET6    IPv6
        // type:
        //      SOCK_STREAM     面向连接通信
        //      SOCK_DGRAM      面向无连接通信
        // protocol:
        //      SOCK_NONBLOCK   非阻塞
        //      SOCK_CLOEXEC    子类不继承父类的socket fd
        // Return:
        //      建立的socket：对于server用于监听，对于client用于向指定位置发送
    if (sockfd < 0){
        tzError("socket()");
    }
    // connect + check
    struct hostent *server;
        // struct hostent {
        //     char  *h_name;            /* official name of host */
        //     char **h_aliases;         /* alias list */
        //     int    h_addrtype;        /* host address type */
        //     int    h_length;          /* length of address */
        //     char **h_addr_list;       /* list of addresses */
        // }
        // #define h_addr h_addr_list[0] /* for backward compatibility，第一个地址 */
    server = gethostbyname(argv[1]);
    if (server == NULL) {
        tzError("[IP] error");
    }
    int port_no = atoi(argv[2]);
    struct sockaddr_in serv_addr;
    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;     // set type
    memcpy((char *)server->h_addr,      // set addr
        (char *)&serv_addr.sin_addr.s_addr,
        server->h_length);
    serv_addr.sin_port = htons(port_no); // set port
    int conntect_ret = connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr));
    if(conntect_ret<0){
        tzError("connect()");
    }

    // write and read
    std::cout << "msg>";
    std::string msg;
    std::cin >> msg;
#define BUF_SIEE    128
    char buf[BUF_SIEE];
    memset(buf, 0, BUF_SIEE);
    strcpy(buf, msg.c_str());
    int write_n = write(sockfd, buf, msg.size());
    std::cout << write_n << " chars send." << std::endl;
    memset(buf, 0, BUF_SIEE);   // 用于输出，依赖\0
    int read_n = read(sockfd, buf, BUF_SIEE);
    std::cout << buf << std::endl;
    std::cout << read_n << " chars read." << std::endl;
    
    // close sockets
    close(sockfd);

    return 0;
}

// ref:
//      http://www.linuxhowtos.org/C_C++/socket.htm