网络编程系列2

多线程处理tcp连接

接上篇，其它未动，创建线程来处理accept到socket描述符。
代码如下

#include <iostream>
#include <arpa/inet.h>
#include <err.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <thread>

#define PORT 12352
#define  MAXSIZE 10240

using namespace std;

/*
**多线程处理多clientd读写数据
*/
void recv_handler(int sockfd){
    long tid = getpid();
//    std::cout<<"this is thread :"<< tid;
    char *buffer = (char*)malloc( 1000 * sizeof(char));
    while(1){
        int n = read(sockfd, (void*)buffer, sizeof(buffer)/sizeof(char));
        if(n ==-1){
            perror("read error");
            strerror(errno);
            std::cout<<"socket fd =" << sockfd<<std::endl;
            free(buffer);
            exit(errno);
        }
        std::cout<< "thread: "<<tid << " recv data:" <<string(buffer,n)<<std::endl;
    }

     free(buffer);
}

int main(int args, char *argv[])
{
//    cout << "Hello world!" << endl;
    int sockfd, newsockfd;
    struct sockaddr_in server_addr;
    struct sockaddr_in client_addr;

    int sin_size, portnum;
    char buf[MAXSIZE];

    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
        fprintf(stderr, "create socket failed!");
        strerror(errno);
        close(sockfd);
        exit(1);
    }

    std::cout<<"socket fd is " << sockfd <<std::endl;

    bzero(&server_addr, sizeof(struct sockaddr_in));
    server_addr.sin_port = htons(PORT);
    server_addr.sin_family = AF_INET;
//    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_addr.s_addr = inet_addr("0.0.0.0");

    if( bind(sockfd, (struct sockaddr*)(&server_addr), sizeof(struct sockaddr))){
        fprintf(stderr, "bind failed");
        strerror(errno);
        close(sockfd);
        exit(2);
    }

    struct sockaddr_in local_addr;
    int len;
    if (getsockname(sockfd, (struct sockaddr *)&local_addr, (socklen_t* )&len) == 0) {
        struct sockaddr_in* sin = (struct sockaddr_in*)(&local_addr);
//        *port = sin->sin_port;
        char addr_buffer[INET_ADDRSTRLEN];
        void * tmp = &(sin->sin_addr);
        if (inet_ntop(AF_INET, tmp, addr_buffer, INET_ADDRSTRLEN) == NULL){
         cerr << "inet_ntop err";
         return false;
        }
        printf("local ip =%s\n", addr_buffer);
    }

    if(listen(sockfd, 10) < 0){
        fprintf(stderr,"listen failed");
        strerror(errno);
        close(sockfd);
        exit(3);
    }
    sin_size = sizeof(struct sockaddr_in);

    while(1){
        if( (newsockfd = accept(sockfd, (struct sockaddr *) (&client_addr),(socklen_t*) &sin_size))< 0){
            fprintf(stderr, "listen failed");
            strerror(errno);
            exit(3);
        }
        std::cout<<"accept connect success, client port=" << client_addr.sin_port<<
            "ip =" <<inet_ntoa(client_addr.sin_addr)<<std::endl;
        std::cout<< "newsockfd=" << newsockfd<<std::endl;
        std::thread task(recv_handler, newsockfd);
        task.detach();///detach()不会造成堵塞，join()会主线程造成堵
                        ///不加这条，程序会崩溃,为什么？？
//        strerror(errno);
    }

    close(sockfd);
    return 0;
}