UDP聊天室

//client.c
struct message
{
     long type;
     char name[20];
     char peer_name[20];
     char mtext[512];
 };
  
 void recv_message(int );
 void send_message(int , struct sockaddr_in *, char *, pid_t);
 void login_msg(struct message *);
 void group_msg(struct message *);
 void quit_msg(struct message *);
 void server_msg(struct message *);
 void server_quit(void);
 void online_msg(struct message *);
 void private_msg(struct message *);
 void print_online(int , struct message *, struct sockaddr_in *);
 void group_chat(int , struct message *, struct sockaddr_in *);
 void private_chat(int ,struct message *, struct sockaddr_in *);
 void client_quit(int , struct message *, struct sockaddr_in *, pid_t );
int main(int argc, char *argv[])
{
    pid_t pid;
    int server_fd;
    struct sockaddr_in server_addr;

    if (argc < 4)
    {
        fprintf(stderr, "usages: %s ip port name\n", argv[0]);
        exit(-1);
    }

    if ((server_fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
    {
        perror("failed to create server_fd");
        exit(-1);
    }

    server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(atoi(argv[2]));
    server_addr.sin_addr.s_addr = inet_addr(argv[1]);
 if ((pid = fork()) < 0)
    {
        perror("failed to fork pid");
        exit(-1);
    }

    if (pid == 0)
        recv_message(server_fd);
    else
        send_message(server_fd, &server_addr, argv[3], pid);

    return 0;
}

void recv_message(int server_fd)
{
    struct message msg;
while (1)
    {
        memset(&msg, 0, sizeof(msg));

        if (recvfrom(server_fd, &msg, sizeof(msg), 0, NULL, NULL) < 0)
        {
            perror("failed to recv server message");
            exit(-1);
        }

        switch(msg.type)
        {
            case CLIENT_LOGIN:
                login_msg(&msg);
                break;

            case CLIENT_CHAT:
                group_msg(&msg);
                break;

            case CLIENT_QUIT:
                quit_msg(&msg);
                break;

            case SERVER_CHAT:
                server_msg(&msg);
                break;

            case SERVER_QUIT:
                server_quit();
                break;

            case PRINT_ONLINE:
                online_msg(&msg);
                break;

            case PRIVATE_CHAT:
                private_msg(&msg);
                break;
            default:
                break;
        }
    }
  return ;
}
void send_message(int server_fd, struct sockaddr_in *server_addr, char *name, pid_t pid)
{
    struct message msg;
    char buf[512];
    int c;

    msg.type = CLIENT_LOGIN;
    strcpy(msg.name, name);

    if (sendto(server_fd, &msg, sizeof(msg), 0,
                (struct sockaddr *)server_addr, sizeof(struct sockaddr)) < 0)
    {
        perror("failed to send login message");
        exit(-1);
    }
while(1)
    {
        usleep(1000);
        printf("***************************************\n");
        printf("the four choosen you can choose:\n");
        printf("input 1 to show who can chat with\n");
        printf("input 2 to chat with others in chatroom\n");
        printf("input 3 to chat with others privately\n");
        printf("input q to quit the chatroom\n");
        printf("***************************************\n");

        printf(">");
        c = getchar(); //输入回车符时停止输入到缓冲区
        while (getchar() != '\n'); //读出全部缓冲区的数据,包括'\n'

        switch(c)
        {
            case '1':
                print_online(server_fd, &msg, server_addr);
                break;
            case '2':
                group_chat(server_fd, &msg, server_addr);
                break;

            case '3':
                private_chat(server_fd, &msg, server_addr);
                break;

            case 'q':
                client_quit(server_fd, &msg, server_addr, pid);
                break;

            default:
                break;
        }
    }

    return ;
}
void login_msg(struct message *msg)
{
    printf("-------- Join  In --------\n");
    printf("%s is join chat\n", msg->name);
    printf("--------------------------\n");

    return ;
}
void quit_msg(struct message *msg)
{
    printf("--------   Quit   --------\n");
    printf("%s is Quit\n", msg->name);
    printf("--------------------------\n");

    return ;
}
void server_quit(void )
{
    kill(getppid(), SIGKILL);
    exit(0);
}
void online_msg(struct message *msg)
{
    printf("-------- Online Client --------\n");
    printf("%s\n", msg->mtext);
    printf("-------------------------------\n");

    return ;
}
void private_msg(struct message *msg)
{
    printf("-------- Private Msg ---------\n");
    printf("name: %s\n", msg->name);
    printf("msg: %s\n", msg->mtext);
    printf("------------------------------\n");

    return ;
}

void print_online(int server_fd, struct message *msg, struct sockaddr_in *server_addr)
{
    msg->type = PRINT_ONLINE;

    if (sendto(server_fd, msg, sizeof(struct message), 0,
                (struct sockaddr *)server_addr, sizeof(struct sockaddr)) < 0)
    {
        perror("failed to send online message");
        exit(-1);
    }
   return ;
}
void group_chat(int server_fd, struct message *msg, struct sockaddr_in *server_addr)
{
    char buf[512];

    memset(buf, 0, sizeof(buf));
    printf("****************************************\n");
    printf("****** Welcome to group chat room ******\n");
    printf("* if you want to quit, please input quit\n");
    printf("****************************************\n");

    while(1)
    {
        memset(buf, 0, sizeof(buf));
        usleep(500);
        printf(">");
        fgets(buf, sizeof(buf), stdin);
        buf[strlen(buf) - 1] = 0;
        msg->type = CLIENT_CHAT;
        strcpy(msg->mtext, buf);

        if (strncmp(buf, "quit", 4) == 0)
            break;

        if (sendto(server_fd, msg, sizeof(struct message), 0,
                    (struct sockaddr *)server_addr, sizeof(struct sockaddr)) < 0)
        {
            perror("failed to send group message");
            exit(-1);
        }
    }
   return ;
}
void private_chat(int server_fd, struct message *msg, struct sockaddr_in *server_addr)
{
    char name[20];
    char buf[512];

    memset(name, 0, sizeof(name));
    printf("please input the peer_name\n");
    printf(">");
    fgets(name, sizeof(name), stdin);
    name[strlen(name) - 1] = 0;

    strcpy(msg->peer_name, name);
    msg->type = PRIVATE_CHAT;

    printf("connect to %s OK!\n", msg->peer_name);
    while(1)
    {
        usleep(500);
        printf(">");
        fgets(buf, sizeof(buf), stdin);
        buf[strlen(buf) - 1] = 0;

        strcpy(msg->mtext, buf);

        if (strncmp(buf, "quit", 4) == 0)
            break;

        if (sendto(server_fd, msg, sizeof(struct message), 0,
                    (struct sockaddr *)server_addr, sizeof(struct sockaddr)) < 0)
        {
            perror("failed to send private message");
            exit(-1);
        }
    }
    return;
}
void client_quit(int server_fd, struct message *msg, struct sockaddr_in *server_addr, pid_t pid)
{
    msg->type = CLIENT_QUIT;

    if (sendto(server_fd, msg, sizeof(struct message), 0,
                (struct sockaddr *)server_addr, sizeof(struct sockaddr)) < 0)
    {
        perror("failed to send quit message");
        exit(-1);
    }

    kill(pid, SIGKILL);
    waitpid(pid, NULL, 0);
    exit(0);
}
//server.c
struct node
{
    char name[20];
    struct sockaddr_in client_addr;
    struct node *next;
};

struct message
{
    long type;
    char name[20];
    char peer_name[20];
    char mtext[512];
};
struct node *create_list(void);
void insert_list(struct node *, char *,struct sockaddr_in *);
void delete_list(struct node *, char *);
void recv_message(int , struct node *);
void send_message(int , struct sockaddr_in *, pid_t );
void client_join(int , struct node *, struct message *, struct sockaddr_in *);
void client_chat(int , struct node *, struct message *);
void client_quit(int , struct node *, struct message *);
void server_chat(int , struct node *, struct message *);
void server_quit(int , struct node *, struct message *);
void brocast_msg(int , struct node *, struct message *);
void print_online(int , struct node *, struct message *);
void private_chat(int , struct node *, struct message *);

void father_func(int sig_no)
{
    return ;
}
et_fd;
    pid_t pid;
    struct sockaddr_in server_addr;
    struct node *head;

    if (argc < 3)
    {
        fprintf(stderr, "usages : %s ip port\n", argv[0]);
        exit(-1);
    }

    if ((socket_fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
    {
        perror("failed to create socket");
        exit(-1);
    }

    head = create_list();//创建链表,存储用户信息
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(atoi(argv[2]));
    server_addr.sin_addr.s_addr = inet_addr(argv[1]);

    if (bind(socket_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0)//绑定套接字
    {
        perror("failed to bind");
        exit(-1);
    }

    if ((pid = fork()) < 0)
    {
        perror("failed to fork pid");
        exit(-1);
    }

    if (pid == 0)//创建父子进程,子进程接收信息,父进程发信息
        recv_message(socket_fd, head);
        else
        send_message(socket_fd, &server_addr, pid);
    return 0;
}
struct node *create_list(void)//创建链表
{
    struct node *head;

    head = (struct node *)malloc(sizeof(struct node));
    head->next = NULL;

    return head;
}
void insert_list(struct node *head, char *name, struct sockaddr_in *client_addr)//添加用户信息
{
    struct node *new;

    new = (struct node *)malloc(sizeof(struct node));
    strcpy(new->name, name);
    new->client_addr = *client_addr;

    new->next = head->next;
    head->next = new;

    return ;
}
void delete_list(struct node *head, char *name)//用户退出,删除用户信息
{
    struct node *p = head->next;
    struct node *q = head;

    while (p != NULL)
    {
        if (strcmp(p->name, name) == 0)
            break;

        p = p->next;
        q = q->next;
    }

    q->next = p->next;
    p->next = NULL;
    free(p);
    return ;
}
void recv_message(int socket_fd, struct node *head)//接收信息
{
    struct message msg;
    struct sockaddr_in client_addr;
    int client_addrlen = sizeof(struct sockaddr);

    while (1)
    {
        memset(&msg, 0, sizeof(msg));
        if (recvfrom(socket_fd, &msg, sizeof(msg), 0, (struct sockaddr *)&client_addr, &client_addrlen) < 0)
        {
            perror("failed to recvform client");
            exit(-1);
        }

        switch(msg.type)
        {
            case CLIENT_JOIN:
                client_join(socket_fd, head, &msg, &client_addr);
                break;
             case CLIENT_CHAT:
                client_chat(socket_fd, head, &msg);
                break;

            case CLIENT_QUIT:
                client_quit(socket_fd, head, &msg);
                break;

            case SERVER_CHAT:
                server_chat(socket_fd, head, &msg);
                break;

            case SERVER_QUIT:
                server_quit(socket_fd, head, &msg);
                break;

            case PRINT_ONLINE:
                print_online(socket_fd, head, &msg);
                break;
            case PRIVATE_CHAT:
                private_chat(socket_fd, head, &msg);
                break;

            default:
                break;
        }
    }
   return ;
}
void send_message(int socket_fd, struct sockaddr_in *server_addr, pid_t pid)//发
送信息
{
    struct message msg;
    char buf[512];

    signal(getppid(), father_func);

    while (1)
    {
        usleep(500);
        printf(">");
        fgets(buf, sizeof(buf), stdin);
        buf[strlen(buf) - 1] = 0;

        strcpy(msg.mtext, buf);
        strcpy(msg.name , "server");
        msg.type = SERVER_CHAT;

        if (strncmp(buf, "quit", 4) == 0)
        {
            msg.type = SERVER_QUIT;
            if (sendto(socket_fd, &msg, sizeof(msg), 0,
                        (struct sockaddr *)server_addr, sizeof(struct sockaddr)) < 0)
            {
                perror("failed to send server_quit message");
                exit(-1);
            }

            pause();

            kill(pid, SIGKILL);
            waitpid(pid, NULL, 0);
            close(socket_fd);
            exit(0);
        }

        if (sendto(socket_fd, &msg, sizeof(msg), 0, (struct sockaddr *)server_addr, sizeof(struct sockaddr)) < 0)
          {
            perror("failed to send server_chat message");
            exit(-1);
        }
    }
    return ;
}
void client_join(int socket_fd, struct node *head, struct message *msg, struct sockaddr_in *client_addr)
{
    printf("*********Join In********\n");
    printf("%s is Join In\n", msg->name);
    printf("************************\n");

    insert_list(head, msg->name, client_addr);
    brocast_msg(socket_fd, head, msg);
    return ;
}
void client_chat(int socket_fd, struct node *head, struct message *msg)
{
    printf("********Group Chat********\n");
    printf("name: %s\n", msg->name);
    printf("msg: %s\n", msg->mtext);
    printf("**************************\n");
    brocast_msg(socket_fd, head, msg);
    return ;
}
void client_quit(int socket_fd, struct node *head, struct message *msg)
{
    printf("*********Quit Msg********\n");
    printf("%s is Quit\n", msg->name);
    printf("*************************\n");
    delete_list(head, msg->name);
    brocast_msg(socket_fd, head, msg);
    return ;
}
void server_chat(int socket_fd, struct node *head, struct message *msg)
{
    printf("********Server Msg*********\n");
    printf("msg: %s\n", msg->mtext);
    printf("***************************\n");
    brocast_msg(socket_fd, head, msg);
    return ;
}
void server_quit(int socket_fd, struct node *head, struct message *msg)
{
    brocast_msg(socket_fd, head, msg);
    kill(getppid(), SIGUSR1);
    return ;
}
void print_online(int socket_fd, struct node *head, struct message *msg)
{
    struct node *p = head->next;
    struct sockaddr_in my_addr;
    char buf[512];

    printf("%s is request to print online client\n", msg->name);
    memset(buf, 0, sizeof(buf));

    while (p != NULL)
    {
        if (strcmp(p->name, msg->name) == 0)
            my_addr = p->client_addr;

        strcat(buf, p->name);
        strcat(buf, " ");

        p = p->next;
    }
     strcpy(msg->mtext, buf);
    msg->type = PRINT_ONLINE;

    if (sendto(socket_fd, msg, sizeof(struct message), 0,
                (struct sockaddr *)&my_addr, sizeof(my_addr)) < 0)
    {
        perror("failed to send print_online message");
        exit(-1);
    }

    return ;
}
void private_chat(int socket_fd, struct node *head, struct message *msg)
{
    struct node *p = head->next;
    struct sockaddr_in peer_addr;

    printf("******** Private Msg ********\n");
    printf("from %s\n", msg->name);
    printf("to %s\n", msg->peer_name);
    printf("msg: %s\n", msg->mtext);
    printf("*****************************\n");

    while (p != NULL)
    {
        if (strcmp(p->name, msg->peer_name) == 0)
        {
            peer_addr = p->client_addr;
            break;
        }

        p = p->next;
    }
if (sendto(socket_fd, msg, sizeof(struct message), 0,
                (struct sockaddr *)&peer_addr, sizeof(peer_addr)) < 0)
    {
        perror("failed to send private message");
        exit(-1);
    }
    return ;
}
void brocast_msg(int socket_fd, struct node *head, struct message *msg)
{
    struct node *p = head->next;

    while(p != NULL)
    {
        if (msg->type == CLIENT_JOIN)
        {
            if (strcmp(p->name, msg->name) == 0)
            {
                p = p->next;
                continue;
            }
        }

        sendto(socket_fd, msg, sizeof(struct message), 0, (struct sockaddr *)&(p->client_addr), sizeof(struct sockaddr));
        p = p->next;
    }
    return ;
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值