一步一步創建聊天程式2-利用epoll來創建簡單的聊天室

来源:https://www.cnblogs.com/JsonZhangAA/archive/2019/12/01/11965404.html
-Advertisement-
Play Games

如圖，這個是看視頻時，最後的作業，除了客戶端未使用select實現外，其它的要求都有簡單實現。 服務端代碼如下： 客戶端代碼還是之前的版本，沒有使用select實現： ...


 

 如圖，這個是看視頻時，最後的作業，除了客戶端未使用select實現外，其它的要求都有簡單實現。

服務端代碼如下：

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <unistd.h>
#include<netinet/tcp.h>

#define MAX_LISTEN 10
#define EPOLL_SIZE 100

struct message
{
    int target_id;
    char buf[100];
};

struct user_password
{
    char username[100];
    char password[100];
};

//設置保活參數 keepalive_time保活時間，keepalive_intvl保活間隔，keepalive_probes保活探測數。
int set_keepalive(int sockfd,int keepalive_time, int keepalive_intvl, int keepalive_probes)
{
    int optval;
    socklen_t optlen= sizeof(optval);

    optval=1;
    if(-1==setsockopt(sockfd,SOL_SOCKET, SO_KEEPALIVE, &optval, optlen))
    {
        perror("setsockopt failture.\n");
        return -1;
    }

    optval=keepalive_probes;
    if(-1==setsockopt(sockfd,SOL_TCP,TCP_KEEPCNT, &optval,optlen))
    {
        perror("setsockopt failture.\n");
        return -1;
    }

    optval=keepalive_intvl;
    if(-1==setsockopt(sockfd,SOL_TCP,TCP_KEEPINTVL, &optval,optlen))
    {
        perror("setsockopt failture.\n");
        return -1;
    }

    optval=keepalive_time;
    if(-1==setsockopt(sockfd,SOL_TCP,TCP_KEEPIDLE, &optval, optlen))
    {
        perror("setsockopt failture\n");
        return -1;
    }
    return 0;
}

void set_user_password(struct user_password users[100])
{
    int fd=fopen("./password.txt","r");
    if(fd==NULL)
    {
        perror("open password failed\n");
        return;
    }

    char buf[100];
    int count=0;
    int i=0;
    while(feof(fd)==0)
    {
        fscanf(fd,"%s",buf);
        if(count%2==0)
        {
            strcpy(users[i].username,buf);
            count++;
        }
        else
        {
            strcpy(users[i++].password,buf);
            count++;
        }
    }

    fclose(fd);
}

int main(int argc,char * argv[])
{

    //讀取設定的用戶名和密碼
    struct user_password users[100];
    set_user_password(users);
    for(int i=0;i<2;i++)
    {
        printf("user: %s,\t password: %s\n",users[i].username,users[i].password);
    }


    struct sockaddr_in server_ip,customer_ip;
    int err,sd;

    sd=socket(AF_INET,SOCK_STREAM,0);
    if(sd==-1)
    {
        perror("socket failed\n");
        close(sd);
        return -1;
    }

    server_ip.sin_family=AF_INET;
    server_ip.sin_port=htons(5678);
    server_ip.sin_addr.s_addr=htonl(INADDR_ANY);
    memset(server_ip.sin_zero,0,8);

    err=bind(sd,(struct sockaddr *)(&server_ip),sizeof(struct sockaddr));
    if(err==-1)
    {
        perror("bind failed\n");
        close(sd);
        return -1;
    }

    err=listen(sd,MAX_LISTEN);
    if(err==-1)
    {
        perror("listen failed\n");
        close(sd);
        return -1;
    }

    int epfd=epoll_create(EPOLL_SIZE);
    if(epfd<0)
    {
        perror("epoll create failed\n");
        return -1;
    }
    printf("epoll created, epollfd=%d\n",epfd);

    struct  epoll_event events[EPOLL_SIZE];
    struct  epoll_event  event;
    event.data.fd=sd;
    event.events=EPOLLIN;
    err=epoll_ctl(epfd,EPOLL_CTL_ADD,sd,&event);
    if(err==-1)
    {
        perror("epoll add failed\n");
        return -1;
    }

    while(1)
    {
        int epoll_events_count=epoll_wait(epfd,events,EPOLL_SIZE,-1);
        if(epoll_events_count<0)
        {
            perror("epoll wait failed\n");
            break;
        }

        printf("epoll_events_count= %d\n",epoll_events_count);

        for(int i=0;i<epoll_events_count;i++)
        {
            int sockfd=events[i].data.fd;
            if(sockfd==sd)
            {
                int length=sizeof(struct sockaddr);
                int client_fd=accept(sockfd,(struct sockaddr *)(&customer_ip),&length);

                //用戶名和密碼檢驗
                int flag=0;
                char buf[100]="please input username and password ";
                struct user_password login_user;
                
                send(client_fd,buf,100,0);
                recv(client_fd,&login_user,sizeof(struct user_password),0);
                memset(buf,0,100);
                for(int i=0;i<100;i++)
                {
                    if((strcmp(users[i].username,login_user.username)==0)&&(strcmp(users[i].password,login_user.password)==0))
                    {
                        flag=1;
                        break;
                    }
                }

                if(flag)
                {
                    err=set_keepalive(sockfd,120,20,3);
                    if(err!=0)
                    {
                        perror("set keep alive failed.\n");
                        continue;
                    }

                    printf("USER %d online\n",client_fd);
                    event.data.fd=client_fd;
                    event.events=EPOLLIN;
                    err=epoll_ctl(epfd,EPOLL_CTL_ADD,client_fd,&event);
                    if(err==-1)
                    {
                        perror("epoll add failed\n");
                        return -1;
                    }
                }
                else
                {
                    memset(buf,0,100);
                    strcpy(buf,"login failed\n");
                    send(client_fd,buf,100,0);
                }
                
            }
            else
            {
                struct message recv_message;
                int bytes=recv(sockfd,&recv_message,sizeof(struct message),0);
                if(bytes<0)
                {
                    perror("recv failed\n");
                    return -1;
                }
                else if(bytes==0)
                {
                    err=epoll_ctl(epfd,EPOLL_CTL_DEL,sockfd,events);
                    if(err==-1)
                    {
                        perror("epoll del failed\n");
                        return -1;
                    }
                    printf("USER %d offline\n",sockfd);
                    close(sockfd);
                }
                else
                {
                    printf("recv message from cliend %d, buf: %s\n",sockfd,recv_message.buf);
                    if(recv_message.target_id!=-1)//請求向客戶端發送信息
                    {
                        err=send(recv_message.target_id,recv_message.buf,100,0);
                        if(err==-1)
                        {
                            perror("send failed\n");
                        }
                    }
                    else if(strcmp(recv_message.buf,"quit")==0)//用戶輸入quit，客戶端推出。
                    {
                        printf("USER %d offline\n",sockfd);
                        send(sockfd,recv_message.buf,100,0);
                        close(sockfd);
                    }
                    else
                    {
                        char cmd[100];
                        strcat(cmd,recv_message.buf);
                        strcat(cmd," >> test.txt");

                        int fd=fopen("./test.txt","w");
                        if(fd==NULL)
                        {
                            perror("clear file failed\n");
                            continue;
                        }
                        close(fd);

                        printf("exec cmd: %s\n",cmd);

                        system(cmd);
                        memset(cmd,0,100);

                        //將結果回送客戶端
                        fd=fopen("./test.txt","r");
                        if(fd==NULL)
                        {
                            perror("open file failed\n");
                            continue;
                        }
                        char buf[100];
                        while(feof(fd)==0)
                        {
                            int fread_ret=fread(buf,sizeof(char),sizeof(buf),fd);
                            send(sockfd,buf,100,0);
                            memset(buf,0,128);
                        }
                        fclose(fd);
                    } 
                }
            }
        }
    }
    return 0;
}

客戶端代碼還是之前的版本，沒有使用select實現：

#include <stdio.h>
#include <pthread.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>

struct message
{
    int target_id;
    char buf[100];
};

struct user_password
{
    char username[100];
    char password[100];
};

int sd;
struct message send_message;

void * read_message(void * argv)
{
    while(1)
    {
        //讀伺服器發來的消息
        char revBuf[100];
        read(sd,revBuf,100);
        printf("recevice from server: %s",revBuf);
    }
}

void * write_message(void * argv)
{
    while(1)
    {
        printf("input message: \n");
        memset(send_message.buf,0,128);
        send_message.target_id=-1;
        scanf("%d %s",&send_message.target_id,send_message.buf);

        write(sd,&send_message,sizeof(send_message));
        sleep(3);
    }
}

int main()
{
    struct sockaddr_in server_ip,customer_ip;
    int err;

    sd=socket(AF_INET,SOCK_STREAM,0);
    if(sd==-1)
    {
        printf("socket failed\n");
        close(sd);
        return -1;
    }

    //server_ip初始化
    server_ip.sin_family=AF_INET;
    server_ip.sin_port=htons(5678);
    server_ip.sin_addr.s_addr=htonl(INADDR_ANY);
    //err=inet_aton("115.157.201.179",&server_ip.sin_addr.s_addr);
    memset(server_ip.sin_zero,0,8);

    err=connect(sd,(struct sockaddr *)(&server_ip),sizeof(server_ip));
    if(err==-1)
    {
        printf("connect failed\n");
        close(sd);
        return -1;
    }

    pid_t pid=fork();
    if(pid==0)
    {
        while(1)
        {
            //讀伺服器發來的消息
            //printf("read message: \n");
            char revBuf[100];
            recv(sd,revBuf,100,0);
            //read(sd,revBuf,100);
            if(strcmp(revBuf,"quit")==0)
            {
                return 0;
            }
            printf("recevice from server: %s\n",revBuf);
        }
    }
    int flag=0;
    while(1)
    {
        printf("input message: \n");
        if(flag++==0)
        {
            struct user_password login_user;
            scanf("%s %s",login_user.username,login_user.password);
            err=send(sd,&login_user,sizeof(struct user_password),0);
            if(err==-1)
            {
                printf("send failed\n");
            }
            continue;
        }
        memset(send_message.buf,0,128);
        send_message.target_id=-1;
        scanf("%d %s",&send_message.target_id,send_message.buf);

        //if(send_message.target_id!=-1&&(strcmp(send_message.buf,"")!=0))
        //{
            //err=send(sd,send_message.buf,100,0);
            err=send(sd,&send_message,sizeof(send_message),0);
            if(err==-1)
            {
                printf("send failed\n");
            }
            //write(sd,&send_message,sizeof(send_message));

        //}
        if(strcmp(send_message.buf,"quit")==0)
        {
            printf("USER offline\n");
            close(sd);
            return 0;
        }
        send_message.target_id=-1;
        memset(send_message.buf,0,sizeof(send_message.buf));
        sleep(3);
    }

    close(sd);
    return 0;
}

 


您的分享是我們最大的動力!

-Advertisement-
Play Games
更多相關文章
  • 【Powershell 教程】學習資源彙總
    【Powershell 教程】學習資源彙總 一、書籍教程 1、《Windows PowerShell實戰指南》第3版 作者:[美] Don，Jones（道·瓊斯），Jeffery，Hicks（傑弗瑞·希克斯） 譯者：宋沄劍 譯 資源：微信讀書、iReader、網上有第 2 版的電子版 實體：淘寶、京 ...
  • Red Hat Enterprise Linux 官方正式版鏡像下載
    Red Hat Enterprise Linux是美國紅帽公司開發的商業市場導向的Linux發行版，為方便大家學習研究，整理分享歷代紅帽官方正式版鏡像給有需要的朋友們。 ...
  • 人生第一次研讀MFC截圖工具的筆記心得
    截圖工具： 其中用到了動態鏈接庫DLL技術（Dynamic Link Library）技術，鍵盤鉤子技術，游標捕獲技術，類橡皮類CRectTracker 頭文件：尾碼名為.cpp,主要是定義和聲明之類的，比如類的定義和常量定義 源文件：尾碼名為.cpp，主要是實現之類的，比如類方法的實現 資源文件： ...
  • tar文件歸檔
    tar是UNIX和類UNIX系統上的壓縮，備份工具， 名字來源於Tape archive--磁碟歸檔，最初的時候是用來將數據儲存，備份到磁帶上的。而今最簡單的備份方法是添加新的磁碟或者在雲端存儲，但即使是最簡單和便宜的磁碟設備和雲儲存，也應該壓縮備份數據，降低存儲空間需求和傳輸時間，linux系統默 ...
  • 【原創】（十二）Linux記憶體管理之vmap與vmalloc
    背景 By 魯迅 By 高爾基 說明： 1. Kernel版本：4.14 2. ARM64處理器，Contex A53，雙核 3. 使用工具：Source Insight 3.5， Visio 1. 概述 在之前的系列文章中，分析到了 的頁框分配， 的小塊記憶體對象分配，這些分配的地址都是物理記憶體連續 ...
  • alluxio 信息索引
    最近要使用到 alluxio，發現網上還是有一些文檔很是不錯，現在通過這篇文章進行索引一下，進行備忘： https://edgedef.com/2017/08/17/alluxio-%E5%B0%8F%E8%B0%83/ （以為網友關於 alluxio的調研） https://www2.eecs.b ...
  • 安裝完dlib以及face_recognition，但是在python IDE中無法導入的問題
    在dlib以及face_recognition通過pip安裝成功之後（如何通過pip進行安裝，在博主的上一篇隨筆中有提及），在pycharm中無法導入。 ...
  • 痞子衡嵌入式：恩智浦機器視覺模塊OpenMV-RT那些事（1）- 初體驗
    近些年機器視覺應用一直是個很火的方向，想象一下機器如果能長上“眼睛”，是不是就可以做一些人類才能做的事情，因此機器視覺是人工智慧實現的一個重要基礎。痞子衡當年碩士畢業論文課題就是工業相機圖像處理相關的，算是機器視覺系統的前端核心。遙想十年以前，想要從事機器視覺系統的開發一直是比較複雜的，需要開發者有... ...
一周排行
    -Advertisement-
    Play Games
  • C#自增運算符詳解：++i與i++的區別及應用場景
    概述：在C#中，++i和i++都是自增運算符，其中++i先增加值再返回，而i++先返回值再增加。應用場景根據需求選擇，首碼適合先增後用，尾碼適合先用後增。詳細示例提供清晰的代碼演示這兩者的操作時機和實際應用。 在C#中，++i 和 i++ 都是自增運算符，但它們在操作上有細微的差異，主要體現在操作的 ...
  • Taurus.MVC 性能壓力測試（ap 壓測 和 linux 下wrk 壓測）：.NET 版本
    上次發佈了：Taurus.MVC 性能壓力測試（ap 壓測 和 linux 下wrk 壓測）：.NET Core 版本，今天計劃準備壓測一下 .NET 版本，來測試並記錄一下 Taurus.MVC 框架在 .NET 版本的性能，以便後續持續優化改進。 為了方便對比，本文章的電腦環境和測試思路，儘量和... ...
  • 一文讀懂.NET WebAPI中FromRoute、FromQuery、FromBody的關鍵角色與用法
    .NET WebAPI作為一種構建RESTful服務的強大工具，為開發者提供了便捷的方式來定義、處理HTTP請求並返迴響應。在設計API介面時，正確地接收和解析客戶端發送的數據至關重要。.NET WebAPI提供了一系列特性，如[FromRoute]、[FromQuery]和[FromBody]，用 ...
  • WPF/C#實現圖像濾鏡優化方案：打造炫目視覺體驗！
    原因：我之所以想做這個項目，是因為在之前查找關於C#/WPF相關資料時，我發現講解圖像濾鏡的資源非常稀缺。此外，我註意到許多現有的開源庫主要基於CPU進行圖像渲染。這種方式在處理大量圖像時，會導致CPU的渲染負擔過重。因此，我將在下文中介紹如何通過GPU渲染來有效實現圖像的各種濾鏡效果。 生成的效果 ...
  • Bogus 實戰：使用 Bogus 和 EFCore 生成模擬數據和種子數據【完整教程】
    引言 上一章我們介紹了在xUnit單元測試中用xUnit.DependencyInject來使用依賴註入，上一章我們的Sample.Repository倉儲層有一個批量註入的介面沒有做單元測試，今天用這個示例來演示一下如何用Bogus創建模擬數據 ，和 EFCore 的種子數據生成 Bogus 的優 ...
  • WPF隨筆收錄-實時繪製心率曲線
    一、前言 在自己的項目中，涉及到實時心率曲線的繪製，項目上的曲線繪製，一般很難找到能直接用的第三方庫，而且有些還是定製化的功能，所以還是自己繪製比較方便。很多人一聽到自己畫就害怕，感覺很難，今天就分享一個完整的實時心率數據繪製心率曲線圖的例子；之前的博客也分享給DrawingVisual繪製曲線的方 ...
  • wpf程式修改啟動入口註意事項
    如果你在自定義的 Main 方法中直接使用 App 類並啟動應用程式，但發現 App.xaml 中定義的資源沒有被正確載入，那麼問題可能在於如何正確配置 App.xaml 與你的 App 類的交互。 確保 App.xaml 文件中的 x:Class 屬性正確指向你的 App 類。這樣，當你創建 Ap ...
  • 記一次 .NET某爐膛鍋爐檢測系統 崩潰分析
    一：背景 1. 講故事 上個月有個朋友在微信上找到我，說他們的軟體在客戶那邊隔幾天就要崩潰一次，一直都沒有找到原因，讓我幫忙看下怎麼回事，確實工控類的軟體環境複雜難搞，朋友手上有一個崩潰的dump，剛好丟給我來分析一下。 二：WinDbg分析 1. 程式為什麼會崩潰 windbg 有一個厲害之處在於 ...
  • 一個.NET內置依賴註入的小型強化版
    前言 .NET生態中有許多依賴註入容器。在大多數情況下，微軟提供的內置容器在易用性和性能方面都非常優秀。外加ASP.NET Core預設使用內置容器，使用很方便。 但是筆者在使用中一直有一個頭疼的問題：服務工廠無法提供請求的服務類型相關的信息。這在一般情況下並沒有影響，但是內置容器支持註冊開放泛型服 ...
  • WPF隨筆收錄-DataGrid固定右側列
    一、前言 在項目開發過程中，DataGrid是經常使用到的一個數據展示控制項，而通常表格的最後一列是作為操作列存在，比如會有編輯、刪除等功能按鈕。但WPF的原始DataGrid中，預設只支持固定左側列，這跟大家習慣性操作列放最後不符，今天就來介紹一種簡單的方式實現固定右側列。（這裡的實現方式參考的大佬 ...
所有分類