Linux下使用多线程模拟异步网络通信

/* socket server
 * 2014-12-15 CopyRight (c) arbboter
 */
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netdb.h>
#include <string.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <vector>
using namespace std;

vector<int> vecClient;
// 线程同步锁
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;  

void* recv_thr(void*p);
void* send_thr(void*p);
int main()
{
    int sockfd_server;
    int sockfd_client;
    struct sockaddr_in addr_server;
    struct sockaddr_in addr_client;
    socklen_t addr_len = 0;
    int client_max = 10;
    int server_port = 33891;

    srand(time(NULL));
    
    // create socket
    sockfd_server = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd_server == -1)
    {
        printf("init socket failed\n");
        return -1;
    }

    // set address
    memset(&addr_server, 0, sizeof(addr_server));
    addr_server.sin_family = AF_INET;
    addr_server.sin_addr.s_addr = htonl(INADDR_ANY);
    addr_server.sin_port = htons(server_port);
    
    // socket bind with address
    if(bind(sockfd_server, (struct sockaddr*)&addr_server, sizeof(struct sockaddr)) == -1)
    {
        printf("bind socket failed\n");
        return -1;
    }

    // server socket start list, waitting client to connect
    // 这个client_max是指同时连接数
    if(listen(sockfd_server, client_max) == -1)
    {
        printf("start listen socket failed\n");
        return -1;
    }

    // 数据线程
    pthread_t tid = 0;
    
    pthread_create(&tid, NULL, send_thr, NULL);
    while(1)
    {
        addr_len = sizeof(struct sockaddr_in);
        printf("waitting for connected...\n");
        
        // waitting for connected
        sockfd_client = accept(sockfd_server, (struct sockaddr*)&addr_client, &addr_len);
        if(sockfd_client == -1)
        {
            printf("accept socket failed\n");
            return -1;
        }
        
        // create a thread to talk with client
        printf("recived connection from : %s\n", inet_ntoa(addr_client.sin_addr));
        pthread_create(&tid, NULL, recv_thr, (void*)&sockfd_client);
        
        // 加到发数据对象集合中
        pthread_mutex_lock(&mutex);  
        vecClient.push_back(sockfd_client);
        pthread_mutex_unlock(&mutex); 
    }
    close(sockfd_server);
    pthread_mutex_destroy(&mutex);
    return 0;
}

// 收数据
void* recv_thr(void*p)
{
    int fd = *((int*)p);
    int nRead = 0;
    char szBuf[512] = {0};
    
    while(1)
    {
        memset(szBuf, 0, sizeof(szBuf));
        nRead = read(fd, szBuf, sizeof(szBuf));
        printf("client : %s\n", szBuf);
        
        if(strcmp(szBuf, "quit") == 0)
        {
            break;
        }
        
        sprintf(szBuf, "Your magic number is : %d", rand()%100);
        printf("server : %s\n", szBuf);
        write(fd, szBuf, strlen(szBuf));
    }
    return NULL;
}

// 发数据
void* send_thr(void*p)
{
    size_t i = 0;
    char szBuf[512] = {0};
    
    while(1)
    {
        sleep(rand()%10);
        pthread_mutex_lock(&mutex);  
        for(int i=0; i<vecClient.size(); i++)
        {
            sprintf(szBuf, "Your magic number is : %d", rand()%100);
            printf("[%02d] server : %s\n", i, szBuf);
            write(vecClient[i], szBuf, strlen(szBuf));
        }
        pthread_mutex_unlock(&mutex); 
    }
    return NULL;
}

/* socket client
 * 2014-12-15 CopyRight (c) arbboter
 */
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netdb.h>
#include <string.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <signal.h>
#include <errno.h>
#include <string.h>

void* recv_thr(void*p);
int main()
{
    int sockfd_server;
    struct sockaddr_in addr_server;
    socklen_t addr_len = 0;
    int server_port = 33891;
    char* server_addr = "192.168.2.200";


    // create socket
    sockfd_server = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd_server == -1)
    {
        printf("init socket failed\n");
        return -1;
    }
    
    // set server address
    memset(&addr_server, 0, sizeof(addr_server));
    addr_server.sin_family = AF_INET;
    addr_server.sin_addr.s_addr = inet_addr(server_addr);;
    addr_server.sin_port = htons(server_port);
    
    // connect server
    if(connect(sockfd_server,(struct sockaddr *)&addr_server,sizeof(struct sockaddr))==-1)
    {
        printf("connect server failed\n");
        return -1;
    }
    
    // 启动收数据线程
    pthread_t tid = 0;
    pthread_create(&tid, NULL, recv_thr, (void*)&sockfd_server);
    
    // 发送数据
    char szBuf[512] = {0};
    int nLen = 0;
    while(1)
    {
        // 收线程已退出
        if( ESRCH == pthread_kill(tid,0) )
        {
            break;
        }
        // 等待输入数据
        gets(szBuf);
        nLen = strlen(szBuf);
        if(nLen > 0)
        {
            szBuf[nLen] = ‘\0‘;
            write(sockfd_server, szBuf, nLen);
        }
    }
    close(sockfd_server);
    return 0;
}

// 收数据
void* recv_thr(void*p)
{
    int fd = *((int*)p);
    int nRead = 0;
    char szBuf[512] = {0};
    
    while(1)
    {
        memset(szBuf, 0, sizeof(szBuf));
        nRead = read(fd, szBuf, sizeof(szBuf));
        printf("server : %s\n", szBuf);
        
        if(strcmp(szBuf, "quit") == 0)
        {
            break;
        }
    }
    return NULL;
}