基于UDP的网络聊天室

1、如果有用户登录，其他用户可以收到这个人的登录信息

2、如果有人发送信息，其他用户可以收到这个人的群聊信息

3、如果有人下线，其他用户可以收到这个人的下线信息

4、服务器可以发送系统信息
服务器

#include<myhead.h>

typedef struct Node
{
    char usrName[20];//用户名
    struct sockaddr_in cin;
    struct Node *next; //指针
}__link;

//收发信息结构体
typedef struct msgtyp
{
    int type;
    char name[20];
    char text[128];
}msg;
int do_lodin(int sfd,__link* head,msg recv_msg,struct sockaddr_in cin)
{
    printf("%s[%s:%d]登陆成功\n",recv_msg.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
    sprintf(recv_msg.text,"----%s登陆成功----",recv_msg.name);
    while(head->next!=NULL)
    {
        head=head->next;
        if(sendto(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&(head->cin),sizeof(head->cin))==-1)
        {
            perror("sendto");
            return -1;
        }
    }

    //添加客户到链表
    __link *new=(__link*)malloc(sizeof(__link));
    new->cin=cin;
    new->next=NULL;
    head->next=new;
    return 0;
}
int do_chat(int sfd,__link* head, msg recv_msg, struct sockaddr_in cin)
{
    printf("%s[%s:%d]chat成功\n",recv_msg.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
    char buf[258]="";
    sprintf(buf,"%s:%s",recv_msg.name,recv_msg.text);
    strcpy(recv_msg.text,buf);

    //给其他客户端发送消息
    while(head->next!=NULL)
    {
        head=head->next;
        head->cin=cin;
        if(memcmp(&cin, &head->cin, sizeof(cin)) != 0)
        {
            if(sendto(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&(head->cin),sizeof(head->cin))==-1)
            {
                perror("sendto");
                return -1;
            }
        }
    }
    return 0;
}
int do_quit(int sfd,__link* head, msg recv_msg, struct sockaddr_in cin)
{
    sprintf(recv_msg.text, "-----%s 已下线-----\n", recv_msg.name);
    while(head->next != NULL)
    {
        if(memcmp(&cin, &head->next->cin, sizeof(cin)) == 0)
        {
            //删除
            __link* temp = head->next;
            head->next = temp->next;
            free(temp);
        }
        else
        {
            head = head->next;
            if(sendto(sfd, &recv_msg, sizeof(recv_msg),0, (struct sockaddr*)&(head->cin), sizeof(head->cin))==-1)
            {
                perror("sendto");
                return -1;
            }
        }
    }
    return 0;
}

int main(int argc, char *argv[])
{
    if(argc<3)
    {
        printf("请输入ip和端口号\n");
        return -1;
    }
    int sfd=socket(AF_INET,SOCK_DGRAM,0);
    if(sfd==-1)
    {
        perror("socket");
        return -1;
    }
    int reuse=-1;
    if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
    {
        perror("setsockopt");
        return -1;
    }
    struct sockaddr_in sin;
    sin.sin_family=AF_INET;
    sin.sin_port=htons(atoi(argv[2]));
    sin.sin_addr.s_addr=inet_addr(argv[1]);
    
    if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
    {
        perror("bind");
        return -1;
    }
    //printf("bind success\n");
    
    int res=fork();
    //父进程接收消息
    if(res>0)
    {
        __link *head=(__link*)malloc(sizeof(__link));
        head->next=NULL;
        msg recv_msg;
        int recv_len=0;
        struct sockaddr_in cin;
        socklen_t len=sizeof(cin);
        while(1)
        {
            recv_len=recvfrom(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&cin,&len);
            if(recv_len<0)
            {
                perror("recvfrom");
                return -1;
            }
            int type=ntohl(recv_msg.type);
            switch(type)
            {
            case 1:
                {
                    do_lodin(sfd,head,recv_msg,cin);
                }
                break;
            case 2:
                {
                    do_chat(sfd,head,recv_msg,cin);
                }
                break;
            case 3:
                {
                    do_quit(sfd,head,recv_msg,cin);
                }
                break;
            }
        }
    }else if(res==0)
    {
        //子进程,发送系统消息
        msg sys_msg={htonl(1),"**system**"};
        while(1)
        {
            bzero(sys_msg.text,1024);
            fgets(sys_msg.text,1024,stdin);
            sys_msg.text[strlen(sys_msg.text)-1]=0;
            if(sendto(sfd,&sys_msg,sizeof(sys_msg),0,(struct sockaddr*)&sin,sizeof(sin))==-1)
            {
                perror("sendto");
                return -1;
            }
        }
        printf("系统消息发送成功\n");
    }
    close(sfd);
    return 0;
}

客户端代码

#include<myhead.h>
#define LOGIN     1
#define CHAT     2
#define QUIT     3
#define ERR_LOG(msg) do{\
    printf("%d %s %s \n", __LINE__, __func__, __FILE__);\
    perror(msg);\
}while(0)

//收发信息的结构体
typedef struct
{
    int type;
    char name[20];
    char text[128] ;

}MSG;

typedef void (*sighandler_t)(int);

int do_chat(int sfd, MSG msg, struct sockaddr_in sin)
{
    while(1)
    {
        //从终端获取聊天信息
        bzero(msg.text, 128);
        fgets(msg.text, 128,stdin);
        msg.text[strlen(msg.text)-1] = 0;

        if(strncasecmp(msg.text, "QUIT",4) == 0)
        {
            msg.type = htonl(QUIT);
        }
        else
        {
            msg.type = htonl(CHAT);
        }

        //发送 
        if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
        {
            ERR_LOG("sendto");
            return -1;
        }

        //如果是退出信息,则客户端要退出
        if(msg.type == htonl(QUIT))
        {
            exit(0);         //退出子进程;
        }
    }
}

int do_recv(int sfd)
{
    MSG rcv_msg ;
    while(1)
    {
        if(recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, NULL, NULL) < 0)
        {
            perror("recvfrom");
            return -1;
        }
        printf("%s\n",rcv_msg.text);
    }
}

void handler(int sig)
{
    //回收子进程资源并退出
    while(waitpid(-1,NULL, WNOHANG)>0);
    exit(0);
}

int main(int argc, char *argv[])
{
    if(argc < 3)
        {   
            printf("请输入 ip 和端口号\n");
            return -1; 
        } 

    sighandler_t s = signal(SIGCHLD, handler);
    if(s == SIG_ERR)
        {
            ERR_LOG("signal");
            return -1;
        }

    int sfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sfd < 0)
        {
            perror("socket");
            return -1;
        }

    //2.填充服务器ip和端口号    
    struct sockaddr_in sin;
    sin.sin_family=AF_INET;
    sin.sin_port=htons(atoi(argv[2]));
    sin.sin_addr.s_addr=inet_addr(argv[1]);
    socklen_t addrlen = sizeof(sin);

    //登录协议
    MSG msg ;
    msg.type = htonl(1);

    printf("请输入姓名>>>");
    fgets(msg.name, 20, stdin);
    msg.name[strlen(msg.name)-1] = 0;

    if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
    {
        ERR_LOG("sendto");    
        return -1;
    }

    pid_t pid = fork();

    if(pid > 0)
    {
        //父进程获取信息
        do_recv(sfd);
    }
    else if(0 == pid)
    {
        //子进程发送信息
        do_chat(sfd, msg, sin);
    }

    //4.关闭套接字
    close(sfd);
    return 0;
}

效果