[C 语言] linux tcp socket 多线程实例

服务器
--
/* gcc -o server -g -lpthread server.c */

#include <netinet/in.h>   // for sockaddr_in
#include <sys/types.h>    // for socket
#include <sys/socket.h>   // for socket
#include <stdio.h>        // for printf
#include <stdlib.h>       // for exit
#include <string.h>       // for bzero
#include <pthread.h>
#include <sys/errno.h>    // for errno

#define HELLO_WORLD_SERVER_PORT 5555
#define LENGTH_OF_LISTEN_QUEUE  20
#define BUFFER_SIZE             1024
#define THREAD_MAX              5

void *talk_to_client(void *data)
{
    int new_server_socket = (int)data;
    char buffer[BUFFER_SIZE];
    bzero(buffer, BUFFER_SIZE);
    strcpy(buffer, "hello world\n");
    send(new_server_socket, buffer, BUFFER_SIZE, 0);
    bzero(buffer, BUFFER_SIZE);
    int length = recv(new_server_socket,buffer,BUFFER_SIZE,0);
    if (length < 0)
    {
        printf("server recieve data failed!\n");
        exit(1);
    }
    printf("socket num: %d %s\n", new_server_socket, buffer);

    close(new_server_socket);
    pthread_exit(NULL);
}

int main(int argc, char **argv)
{
    struct sockaddr_in server_addr;
    bzero(&server_addr, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htons(INADDR_ANY);
    server_addr.sin_port = htons(HELLO_WORLD_SERVER_PORT);

    int server_socket = socket(AF_INET, SOCK_STREAM, 0);
    if(server_socket < 0)
    {
        printf("create socket failed!\n");
        exit(1);
    }

    if(bind(server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr)))
    {
        printf("server bind port : %d failed!\n", HELLO_WORLD_SERVER_PORT);
        exit(1);
    }

    if (listen(server_socket, LENGTH_OF_LISTEN_QUEUE))
    {
        printf("server listen failed!\n");
        exit(1);
    }

    int i;
    while(1)
    {
        struct sockaddr_in client_addr;
        socklen_t length = sizeof(client_addr);

        int new_server_socket = accept(server_socket, (struct sockaddr*)&client_addr, &length);
        if ( new_server_socket < 0)
        {
            printf("server accept failed!\n");
            break;
        }

        pthread_t child_thread;
        pthread_attr_t child_thread_attr;
        pthread_attr_init(&child_thread_attr);
        pthread_attr_setdetachstate(&child_thread_attr, PTHREAD_CREATE_DETACHED);
        if(pthread_create(&child_thread, &child_thread_attr, talk_to_client, (void *)new_server_socket) < 0)
            printf("pthread_create failed : %s\n", strerror(errno));
    }

    close(server_socket);
    exit(0);
}
================================================================================
客户端
--
/* gcc -o client -g -lpthread client.c */

#include <netinet/in.h>    // for sockaddr_in
#include <sys/types.h>     // for socket
#include <sys/socket.h>    // for socket
#include <stdio.h>         // for printf
#include <stdlib.h>        // for exit
#include <string.h>        // for bzero
#include <sys/errno.h>     // for errno
#include <pthread.h>

#define HELLO_WORLD_SERVER_PORT 5555
#define BUFFER_SIZE             1024

char *server_IP = NULL;

void *talk_to_server(void * thread_num)
{
    struct sockaddr_in client_addr;
    bzero(&client_addr, sizeof(client_addr));
    client_addr.sin_family = AF_INET;
    client_addr.sin_addr.s_addr = htons(INADDR_ANY);
    client_addr.sin_port = htons(0);
    int client_socket = socket(AF_INET, SOCK_STREAM, 0);
    if( client_socket < 0)
    {
        printf("create socket failed!\n");
        exit(1);
    }

    if(bind(client_socket, (struct sockaddr*)&client_addr, sizeof(client_addr)))
    {
        printf("client bind port failed!\n");
        exit(1);
    }

    struct sockaddr_in server_addr;
    bzero(&server_addr, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    if(inet_aton(server_IP, &server_addr.sin_addr) == 0)
    {
        printf("server iP address error!\n");
        exit(1);
    }
    server_addr.sin_port = htons(HELLO_WORLD_SERVER_PORT);
    socklen_t server_addr_length = sizeof(server_addr);
    if(connect(client_socket, (struct sockaddr*)&server_addr, server_addr_length) < 0)
    {
        printf("can not connect to %s!\n", server_IP);
        exit(1);
    }

    char buffer[BUFFER_SIZE];
    bzero(buffer, BUFFER_SIZE);
    int length = recv(client_socket, buffer, BUFFER_SIZE, 0);
    if(length < 0)
    {
        printf("recieve data from server %s failed!\n", server_IP);
        exit(1);
    }
    printf("from server %s : %s", server_IP, buffer);

    bzero(buffer, BUFFER_SIZE);
    sprintf(buffer,"hello world! from client thread num : %d\n", (int)thread_num);
    send(client_socket, buffer, BUFFER_SIZE, 0);
    close(client_socket);
    pthread_exit(NULL);
}

int main(int argc, char **argv)
{
    if (argc != 2)
    {
        printf("usage: %s server ip address\n", argv[0]);
        exit(1);
    }
    server_IP = argv[1];

    pthread_t child_thread;
    pthread_attr_t child_thread_attr;
    pthread_attr_init(&child_thread_attr);
    pthread_attr_setdetachstate(&child_thread_attr, PTHREAD_CREATE_DETACHED);
    int i=0;
    for(i=0; i < 100; i++)
    {
        if(pthread_create(&child_thread, &child_thread_attr, talk_to_server, (void *)i) < 0)
            printf("pthread_create failed : %s\n", strerror(errno));
    }
    exit(0);
}