基于linux原生異步io創(chuàng)建的簡易聊天程序

xcold 發(fā)布于2019-06-28 10:27 / 528人閱讀

摘要：分為兩個(gè)獨(dú)立的程序編譯前先確定自己服務(wù)器的地址，比如想要在自己的下運(yùn)行，先獲取自己的地址，修改，里面的地址為自己的地址服務(wù)端編譯運(yùn)行客戶端編譯運(yùn)行服務(wù)端先運(yùn)行起來后，再啟動(dòng)客戶端，可以在多態(tài)機(jī)器上啟多個(gè)客戶端，互相聊天

分為兩個(gè)獨(dú)立的程序
編譯前先確定自己服務(wù)器的地址，比如想要在自己的ubuntu下運(yùn)行，先ip addr獲取自己的ip地址，修改chat_server.c，chat_client.c里面的ip地址為自己的ip地址

服務(wù)端chat_server.c
編譯: gcc chat_server.c -o server
運(yùn)行: ./server

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define    MAX_USER        500
#define    MAX_MSG_LEN        1024
#define MAX_IO_RETRY_TIMES    5

int server_fd;

int user_cnt = 0;
int user_fds[MAX_USER];

char msg_buffer[MAX_MSG_LEN];

void down()
{
    int i;
    for(i = 0; i < user_cnt; i++)
    {
        shutdown(user_fds[i], SHUT_RDWR);
        close(user_fds[i]);
        user_fds[i] = 0;
    }
    shutdown(server_fd, SHUT_RDWR);
    close(server_fd);
    server_fd = 0;
}

int go_on = 1;
void sig_handler(int signal)
{
    go_on = 0;
}

int my_safe_read(int fd)
{
    int ret;
    int i;
    
    for(i = 0; i < MAX_IO_RETRY_TIMES; i++)
    {
        errno = 0;
        int ret = read(fd, msg_buffer, MAX_MSG_LEN);
        if(ret < 0)
        {
            if(EAGAIN == errno || EWOULDBLOCK == errno)
            {
                printf("Safe read [%d] cnt
", i + 1);
                continue;
            }
            else
            {
                return ret;
            }
        }
        else if(0 == ret)
        {
            return 0;
        }
        else
        {
            return ret;
        }
    }
    return ret;
}

int my_safe_write(int fd)
{
    write(fd, msg_buffer, MAX_MSG_LEN);
}

int main(int argc, char *argv[])
{
    signal(SIGINT, sig_handler);
    server_fd = socket(AF_INET, SOCK_STREAM, 0);
    if(-1 == server_fd)
    {
        printf("socket err
");
        return -1;
    }
    printf("socket ok
");

    struct sockaddr_in server_addr;
    bzero(&server_addr, sizeof(struct sockaddr_in));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    unsigned short portnum = 0x8888;
    if(argc >= 2)
    {
        int port = atoi(argv[1]);
        if(port > 20000)
        {
            portnum = port;
        }
    }
    printf("port = %d
", portnum);
    
    server_addr.sin_port = portnum;

    if(-1 == bind(server_fd, (struct sockaddr *)(&server_addr), sizeof(struct sockaddr)))
    {
        printf("bind err
");
        return -1;
    }
    printf("bind ok
");

    if(-1 == listen(server_fd, MAX_USER))
    {
        printf("listen err
");
        return -1;
    }
    printf("listen ok
");

    int i;
    fd_set fds;
    struct timeval tv;
    while(go_on)
    {
        FD_ZERO(&fds);
        FD_SET(server_fd, &fds);
        for(i = 0; i < user_cnt; i++)
        {
            printf("-%d
", user_fds[i]);
            if(user_fds[i])
            {
                FD_SET(user_fds[i], &fds);
            }
        }

        int fd_cnt_to_selected = 0;
        for(i = 0; i < user_cnt; i++)
        {
            if(user_fds[i] > fd_cnt_to_selected)
            {
                fd_cnt_to_selected = user_fds[i];
            }
        }
        fd_cnt_to_selected = (0 == user_cnt)? (server_fd + 1) : (fd_cnt_to_selected + 1);

        tv.tv_sec = 2;
        tv.tv_usec = 0;
        int ret = select(fd_cnt_to_selected, &fds, 0, 0, &tv);
        if(ret < 0)
        {
            printf("select err
");
            break;
        }
        else if(0 == ret)
        {
            continue;
        }

        struct sockaddr_in client_addr;
        for(i = 0; i < user_cnt; i++)
        {
            if(FD_ISSET(user_fds[i], &fds))
            {
                bzero(msg_buffer, MAX_MSG_LEN);
                int ret = my_safe_read(user_fds[i]);
                if(ret <= 0)
                {
                    int connection_state = 1; 
                    int len_of_int = sizeof(int);
                    getsockopt(user_fds[i], IPPROTO_TCP, SO_KEEPALIVE, (void*)&connection_state, &len_of_int); 
                    printf("connection_state = %s
", (0 == connection_state)? "break" : "alive");
                    if(0 == connection_state)
                    {
                        close(user_fds[i]);
                        FD_CLR(user_fds[i], &fds);
                        if(user_cnt - 1 == i)
                        {
                            user_cnt = user_cnt - 1;
                            user_fds[i] = 0;
                        }
                        else
                        {
                            int j;
                            for(j = i; j < user_cnt; j++)
                            {
                                user_fds[j] = user_fds[j+1];
                            }
                            user_fds[j] = 0;
                            user_cnt = user_cnt -1;
                        }
                    }
                }
                else
                {
                    printf("receive[%s]
", msg_buffer);
                    int j;
                    for(j = 0; j < user_cnt; j++)
                    {
                        if(j == i)
                        {
                            continue;
                        }
                        my_safe_write(user_fds[j]);
                    }
                }
            }
        }

        if(FD_ISSET(server_fd, &fds))
        {
            int sin_size = sizeof(struct sockaddr);
        
            user_fds[user_cnt] = accept(server_fd, (struct sockaddr *)(&client_addr), &sin_size);
            if(-1 == user_fds[user_cnt])
            {
                printf("accept err
");
                return -1;
            }
            printf("accept ok!
");
            printf("user_fds[user_cnt] = %d
", user_fds[user_cnt]);
            user_cnt++;
        }
    }
    
    down();
    return 0;
}

客戶端chat_client.c
編譯: gcc chat_client -lpthread -o chat_client
運(yùn)行: 服務(wù)端先運(yùn)行起來后，再啟動(dòng)客戶端，可以在多態(tài)機(jī)器上啟多個(gè)客戶端，互相聊天
./chat_client

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

int cfd = 0;
struct passwd *pwd;

void sig_handler(int signal)
{
    char bye_msg[1024] = { 0 };
    sprintf(bye_msg, "WARNING------>User[%s] have go, he said all is "SB"! Oh, what the fuck!", pwd->pw_name);
    write(cfd, bye_msg, 1024);
    
    shutdown(cfd, SHUT_RDWR);
    close(cfd);
    cfd = 0;
    exit(-1);
}

void* f(void* t)
{
    struct timeval tv;
    fd_set fds;
    while(1)
    {
        tv.tv_sec = 0;
        tv.tv_usec = 100;
        FD_ZERO(&fds);
        FD_SET(cfd, &fds);
        
        int r = select(cfd + 1, &fds, 0, 0, &tv);
        if(r<0)
        {
            printf("select err
");
        }
        else if(r==0)
        {
        }
        
        char buf[1024*10] = { 0 };
        int n = read(cfd, buf, 1024*10);
        if(buf[0] != 0)
            printf("%s
", buf);
    }    
}

int main(int argc, char * argv[])
{
    signal(SIGINT, sig_handler);
    pwd = getpwuid(getuid());

    int recbytes;
    int sin_size;
    char buffer[1024] ={0}; 
    struct sockaddr_in s_add, c_add;
    unsigned short portnum =0x8888;
    if(argc >= 2)
    {
        int port = atoi(argv[1]);
        if(port > 20000)
        portnum = port;
    }
    printf("port = %d
", portnum);
    
    cfd = socket(AF_INET, SOCK_STREAM, 0);
    if(-1 == cfd)
    {
        printf("socket err
");
        return -1;
    }
    //printf("socket ok
");

    bzero(&s_add, sizeof(struct sockaddr_in));
    s_add.sin_family = AF_INET;
    s_add.sin_addr.s_addr = inet_addr("192.168.60.104");
    s_add.sin_port = htons(portnum);
    
    if(-1 == connect(cfd, (struct sockaddr *)(&s_add), sizeof(struct sockaddr)))
    {
        printf("connect err
");
        return -1;
    }
    char personal_info[128] = { 0 };
    sprintf(personal_info, "DouBi[%s] is comming!!!HaHaaa^-^", pwd->pw_name);
    write(cfd, personal_info, 128);
    //printf("connect ok
");

    pthread_t th;
    int rr = pthread_create((pthread_t *)&th, NULL, f, NULL);
    
    while(1)
    {    
        //printf("enter what you want to say:_
");
        char b[10240] = { 0 };
        sprintf(b, "[%s]: ", pwd->pw_name);
        gets(b + strlen(b));
        //printf("b = %s
", b);
        if(-1 == (recbytes = write(cfd, b, 1024*10)))
        {
            printf("send data err
");
            return -1;
        }
        //printf("send ok
");
    }
    

    //close(cfd);
//    sleep(100);
    
    return 0;
}

GPU云服務(wù)器云服務(wù)器原生js簡易日歷基于webrtc的多人視頻聊天服務(wù)實(shí)例簡易網(wǎng)頁聊天jsp js原生異步怎么寫

文章版權(quán)歸作者所有，未經(jīng)允許請(qǐng)勿轉(zhuǎn)載,若此文章存在違規(guī)行為，您可以聯(lián)系管理員刪除。

轉(zhuǎn)載請(qǐng)注明本文地址：http://systransis.cn/yun/25004.html

發(fā)表評(píng)論

登陸后可評(píng)論

0條評(píng)論

xcold

男|高級(jí)講師

我要關(guān)注我要私信

TA的文章

【LeetCode 二叉樹專項(xiàng)】把二叉搜索樹轉(zhuǎn)換為累加樹（538）

閱讀 3269·2021-11-18 10:02
UCloud金秋狂歡盛典-烏蘭察布上新首促，快杰共享型低至3元/1個(gè)月或37元/年-老劉博客

閱讀 1474·2021-10-12 10:08
Docker安裝InfluxDB_用戶名密碼和策略使用

閱讀 1274·2021-10-11 10:58
安裝鴻蒙開發(fā)工具-DevEco Studio

閱讀 1289·2021-10-11 10:57
golang實(shí)現(xiàn)儀表控制-visa32.dll方式

閱讀 1185·2021-10-08 10:04
【C++從0到1】新手都能看懂的C++入門（上篇），建議收藏

閱讀 2139·2021-09-29 09:35
彈性云主機(jī)是什么原因-電信云主機(jī)是什么？

閱讀 790·2021-09-22 15:44
微軟宣布將于 9 月 22 日舉行 Surface 和 Windows 11 活動(dòng)

閱讀 1287·2021-09-03 10:30

成人国产在线小视频_日韩寡妇人妻调教在线播放_色成人www永久在线观看_2018国产精品久久_亚洲欧美高清在线30p_亚洲少妇综合一区_黄色在线播放国产_亚洲另类技巧小说校园_国产主播xx日韩_a级毛片在线免费

資訊專欄INFORMATION COLUMN

上云采購季！| 2核2G4M爆款云服務(wù)器低至59元/年，更有多臺(tái)、長期優(yōu)惠，快來選購！

基于linux原生異步io創(chuàng)建的簡易聊天程序

相關(guān)文章

利用express+socket.io實(shí)現(xiàn)一個(gè)簡易版聊天室

基于express和socket.io的簡易版聊天室

PHP并發(fā)IO編程之路

微信小程序資源匯總

發(fā)表評(píng)論

0條評(píng)論

xcold

男|高級(jí)講師

TA的文章

【LeetCode 二叉樹專項(xiàng)】把二叉搜索樹轉(zhuǎn)換為累加樹（538）

UCloud金秋狂歡盛典-烏蘭察布上新首促，快杰共享型低至3元/1個(gè)月或37元/年-老劉博客

Docker安裝InfluxDB_用戶名密碼和策略使用

安裝鴻蒙開發(fā)工具-DevEco Studio

golang實(shí)現(xiàn)儀表控制-visa32.dll方式

【C++從0到1】新手都能看懂的C++入門（上篇），建議收藏

彈性云主機(jī)是什么原因-電信云主機(jī)是什么？

微軟宣布將于 9 月 22 日舉行 Surface 和 Windows 11 活動(dòng)

最新活動(dòng)

資訊專欄INFORMATION COLUMN

上云采購季！| 2核2G4M爆款云服務(wù)器低至59元/年，更有多臺(tái)、長期優(yōu)惠，快來選購！

基于linux原生異步io創(chuàng)建的簡易聊天程序

相關(guān)文章

發(fā)表評(píng)論

0條評(píng)論

男|高級(jí)講師

TA的文章

最新活動(dòng)

上云采購季！| 2核2G4M爆款云服務(wù)器低至59元/年，更有多臺(tái)、長期優(yōu)惠，快來選購！