client.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <signal.h>
#include <string.h>
#include <pthread.h>
#include <ncurses.h>
#include <locale.h>
#include <assert.h>

#define SERVER_ADDR "127.0.0.1"
#define SERVER_PORT 6666

#define SCREEN_BEGIN_LINE   1

#define SERVER_INS "SERVER-INS"
#define SERVER_INS_MESSAGE 0
#define SERVER_INS_USER_COME 1
#define SERVER_INS_USER_EXIT 2

#define USER_COME_TIPS                  " come in chatroom!"
#define USER_EXIT_TIPS                  " exit chatroom!"
#define ONLINE_GAY_TIPS                 "ChatRoom-online-gay: "
#define ONLINE_GAY_TIPS_LENGTH          (strlen(ONLINE_GAY_TIPS))
#define USER_COME_TIPS_LENGTH           (strlen(USER_COME_TIPS))
#define USER_EXIT_TIPS_LENGTH           (strlen(USER_EXIT_TIPS))

int sockfd;
int next_writed_row = SCREEN_BEGIN_LINE;
WINDOW *win;
char username[12];
int name_length;

void *read_user_input(void *sockfd);
void *recv_from_server(void *sockfd);
void sigint_handler(int num);
void refresh_input_box();
void init_client(int sockfd);
void write_display_win(char buffer[]);

int main(int argc, char *argv[]){
    // chech addr and port
    char *server_addr = SERVER_ADDR;
    int server_port = SERVER_PORT;
    if(argc == 3){
    	server_addr = argv[1];
	server_port = atoi(argv[2]);
    }
	
    printf("try to connect %s:%d......\n", server_addr, server_port);

    printf("please input your name(length between 1~12): ");
    scanf("%s", username);
    name_length = strlen(username);
    if(name_length > 12){
        printf("username length error!\n");
        exit(0);
    }

    // set sigint handler
    signal(SIGINT, sigint_handler);

    setlocale(LC_ALL, "");
    win = initscr(); /* 初始化屏幕 */

    curs_set(TRUE);
    clear();
    refresh_input_box();

    // create a socket  
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    assert(sockfd != -1);

    struct sockaddr_in addr;
    socklen_t addrlen = sizeof(addr);
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = inet_addr(server_addr);
    addr.sin_port = htons(server_port);

    int res = connect(sockfd, (struct sockaddr*)&addr, addrlen);
    assert(res != -1);

    init_client(sockfd);

    pthread_t p1, p2;
    assert(pthread_create(&p1, NULL, read_user_input, &sockfd) == 0);
    assert(pthread_create(&p2, NULL, recv_from_server, &sockfd) == 0);

    pthread_join(p1, NULL);
    pthread_join(p2, NULL);

    return 0;
}

void init_client(int sockfd){
    if(send(sockfd, username, name_length, 0) == -1){
        printf("init your name error!\n");
        exit(0);
    }
}

void *read_user_input(void *sockfd){
    int fd = *((int*)sockfd);
    char buffer[1024];
    while(1){
        getstr(buffer);
        send(fd, buffer, strlen(buffer), 0);
        refresh_input_box();
    }
}

void *recv_from_server(void *sockfd){
    int fd = *((int *)sockfd);
    char buffer[1024];
    int n = -1;
    while((n = recv(fd, buffer, 1024, 0)) > 0){
        buffer[n] = '\0';
        write_display_win(buffer);
        
    }
}

void sigint_handler(int num){
    close(sockfd);
    endwin();
    exit(0);
}

void refresh_input_box(){
    int row, clo;
    getmaxyx(stdscr, row, clo);
    int line_row = row * 4 / 5;
    for (int i = line_row; i < row; i++){
        move(i, 0);
        clrtoeol();
    }
    move(line_row, 0);
    for (int i = 0; i < clo; i++){
        printw("*");
    }
    move(line_row+1, 0);
    refresh();
}

void write_display_win(char buffer[]){
    // save current position
    int old_row, old_clo;
    getyx(win, old_row, old_clo);

    int cur_row, cur_clo;

    int row, clo;
    getmaxyx(stdscr, row, clo);
    int line_row = row * 4 / 5;
    int length = strlen(buffer);
    int last_row = next_writed_row + length / clo + (length % clo == 0 ? 0 : 1);
    if (last_row >= line_row){
        for (int i = SCREEN_BEGIN_LINE; i < line_row; i++){
            move(i, 0);
            clrtoeol();
        }
        next_writed_row = SCREEN_BEGIN_LINE;
    }

    char *begin = buffer;
    char *ins_header = strsep(&begin, "&");
    char *ins_num = strsep(&begin, "&");
    char *online_gay = strsep(&begin, "&");
    char *time = strsep(&begin, "&");
    char *message = begin;

    // update online gay
    move(0, 0);
    int online_gay_begin_clo = (clo - strlen(online_gay) - ONLINE_GAY_TIPS_LENGTH) / 2;
    for (int i = 0; i < online_gay_begin_clo; i++){
        printw("*");
    }
    move(0, online_gay_begin_clo);
    printw("%s%s", ONLINE_GAY_TIPS, online_gay);
    getyx(win, cur_row, cur_clo);
    for (int i = cur_clo; i < clo; i++){
        printw("*");
    }
    refresh();
    
    // write data
    move(next_writed_row, 0);
    switch (atoi(ins_num))
    {
        case SERVER_INS_MESSAGE:
            printw("[%s] [%s] say:\n%s", ins_header, time, message);
            break;
        case SERVER_INS_USER_COME:
            move(next_writed_row, (clo-strlen(message)-USER_COME_TIPS_LENGTH)/2);
            printw("%s%s", message, USER_COME_TIPS);
            break;
        case SERVER_INS_USER_EXIT:
            move(next_writed_row, (clo-strlen(message)-USER_EXIT_TIPS_LENGTH)/2);
            printw("%s%s", message, USER_EXIT_TIPS);
            break;
        default:
            printw("[%s] [%s] say:\n%s", "no-name gay", "just now", "fuck! it means occured bugs if you see this line!");
            break;
    }

    // mvprintw(next_writed_row, 0, buffer);
    refresh();
    // update next_writed_row
    getyx(win, row, clo);
    next_writed_row = (clo == 0 ? row : (row + 1));

    // back to old position
    move(old_row, old_clo);
    refresh();
}