chat客户端-双线程

chat_client.c

#include <stdio.h> //printf、fput
#include <stdlib.h> //atoi
#include <string.h> //memset
#include <unistd.h> //read、close
#include <arpa/inet.h> //sockaddr_in结构、inet_addr
#include <sys/socket.h>
#include <pthread.h>
#define BUF_SIZE 100
#define NAME_SIZE 20

void* send_msg(void* arg);
void* recv_msg(void* arg);

char name[NAME_SIZE] = "";
char buf[BUF_SIZE];

int main(int argc, char *argv[]) {
    if (argc != 4) {
        printf("Usage: %s <IP> <port> <name>\n", argv[0]);
        exit(1);
    }
    sprintf(name, "[%s]", argv[3]); //把输入的名字加上方括号，放到name指向的区域
    
    int sock = socket(PF_INET, SOCK_STREAM, 0);
    if (sock == -1) {
        fputs("socket() error", stderr);
        exit(1);
    }

    struct sockaddr_in serv_addr;
    memset(&serv_addr, 0, sizeof(serv_addr)); //全部位置0
    serv_addr.sin_family = AF_INET; //
    serv_addr.sin_addr.s_addr = inet_addr(argv[1]); //取1号参数作为目标IP地址
    serv_addr.sin_port = htons(atoi(argv[2])); //取2号参数作为目标端口

    if (connect(sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1) {
        fputs("connect() error", stderr);
        exit(1);
    }

    pthread_t send_thread, recv_thread;
    pthread_create(&send_thread, NULL, send_msg, (void*)&sock);
    pthread_create(&recv_thread, NULL, recv_msg, (void*)&sock);
    void* thread_return;
    pthread_join(send_thread, &thread_return);
    pthread_join(recv_thread, &thread_return);
    
    close(sock);
    return 0;
}

void* send_msg(void* arg) {
    int sock = *((int*)arg);
    char name_msg[NAME_SIZE + BUF_SIZE];
    while(1) {
        fgets(buf, BUF_SIZE, stdin);
        if (!strcmp(buf, "q\n")) {
            shutdown(sock, SHUT_WR);
            exit(0);
        }
        sprintf(name_msg, "%s %s", name, buf); //把name和buf指向的区域内容合并
        write(sock, name_msg, strlen(name_msg));
    }
    return NULL;
}

void* recv_msg(void* arg) {
    int sock = *((int*)arg);
    char name_msg[NAME_SIZE + BUF_SIZE];
    while(1) {
        int str_len = read(sock, name_msg, NAME_SIZE + BUF_SIZE - 1);
        if (str_len == -1) {
            return (void*)-1;
        }
        name_msg[str_len] = 0;
        printf("Message from server: %s \n", name_msg);
    }
}

• 编译

  gcc chat_client.c -o cclient
  

• 执行-两个终端

  ./cclient 127.0.0.1 8100 Jack
  

  ./cclient 127.0.0.1 8100 Mike