用多线程实现并发

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/socket.h>
#include <arpa/inet.h>

#include <pthread.h>
#include <sys/syscall.h>  

#define INFO printf

#define LISTENPORT  7777    //监听端口
#define BACKLOG 5           //最大监听数
#define MAXLEN  2048        //recv缓冲区长度


int listen_socket(int listen_port);
void* client_routine(void* arg);

int main(int argc, char const *argv[])
{
    int connfd = 0; //客户端的fd
    int n = 0;  //recv的返回值
    char buff[MAXLEN];  //存储客户端发送过来的信息
    int listenfd = listen_socket(LISTENPORT);    
    if (listenfd == -1)
    {
        INFO("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
        return -1;          
    }

    // 用多线程来实现多个客户端同时连接
    while (1) 
    {
		struct sockaddr_in client;
	    socklen_t len = sizeof(client);
	    if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
	        INFO("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
	        return -1;
	    }

		pthread_t threadid;
		pthread_create(&threadid, NULL, client_routine, (void*)&connfd);

    }
    close(listenfd);

    return 0;
}


int listen_socket(int listen_port)
{
    int connfd = 0; //客户端的fd
    struct sockaddr_in serv_addr;
    int listenfd = socket(AF_INET, SOCK_STREAM, 0);    //创建套接字

    if (listenfd == -1)
    {
        INFO("create socket error: %s(errno: %d)\n", strerror(errno), errno);
        return -1;
    }

    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);  //监听服务器所有IP
    serv_addr.sin_port = htons(listen_port);

    if (bind(listenfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) == -1)  //绑定地址结构体和socket
    {
        INFO("bind socket error: %s(errno: %d)\n", strerror(errno), errno);
        close(listenfd);
        return -1;
    }

    if (listen(listenfd, BACKLOG) == -1)      //开启监听 ，第二个参数是最大监听数
    {
        INFO("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
        close(listenfd);
        return -1;        
    }

    return listenfd;
}

void* client_routine(void* arg)
{
    int connfd = *(int*)arg;
    char buff[MAXLEN];
    int n = 0;
    INFO(" TID: %ld, PID: %ld", syscall(__NR_gettid), syscall(__NR_getpid));  
    while (1)
    {
        n = recv(connfd, buff, MAXLEN, 0);
        if (n > 0)
        {
            buff[n] = '\0';
            INFO("recv msg from client: %s\n", buff);
            send(connfd, buff, n, 0);
        }
        else if (n == 0)
        {
            INFO("close--------\n");
            close(connfd);
            break;
        }        
    }
    return NULL;
}