UDP聊天室

pub.h   定義了一些消息的結構和宏變量

#ifndef _PUB_H_

#define _PUB_H_

#include <list>

#include <algorithm>

using namespace std;

// C2S

#define C2S_LOGIN            0x01

#define C2S_LOGOUT            0x02

#define C2S_ONLINE_USER        0x03

#define MSG_LEN                512

// S2C

#define S2C_LOGIN_OK        0x01

#define S2C_ALREADY_LOGINED    0x02

#define S2C_SOMEONE_LOGIN    0x03

#define S2C_SOMEONE_LOGOUT    0x04

#define S2C_ONLINE_USER        0x05

// C2C

#define C2C_CHAT            0x06

typedef struct message

{

    int cmd;

    char body[MSG_LEN];

} MESSAGE;

typedef struct user_info

    char username[16];

    unsigned int ip;

    unsigned short port;

} USER_INFO;

typedef struct chat_msg

    char msg[100];

}CHAT_MSG;

typedef list<USER_INFO> USER_LIST;

#endif /* _PUB_H_ */

chat.c

#include <unistd.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <arpa/inet.h>

#include <signal.h>

#include <stdlib.h>

#include <stdio.h>

#include <errno.h>

#include <string.h>

#include "pub.h"

#define ERR_EXIT(m) \

        do \

        { \

                perror(m); \

                exit(EXIT_FAILURE); \

        } while(0)

// 目前使用者名

char username[16];

// 聊天室成員清單

USER_LIST client_list;

void do_someone_login(MESSAGE& msg);

void do_someone_logout(MESSAGE& msg);

void do_getlist();

void do_chat();

void parse_cmd(char* cmdline, int sock, struct sockaddr_in *servaddr);

bool sendmsgto(int sock, char* username, char* msg);

void parse_cmd(char* cmdline, int sock, struct sockaddr_in *servaddr)

    char cmd[10]={0};

    char *p;

    p = strchr(cmdline, ‘ ‘);

    if (p != NULL)

        *p = ‘\0‘;

    strcpy(cmd, cmdline);

    if (strcmp(cmd, "exit") == 0)

    {

        MESSAGE msg;

        memset(&msg,0,sizeof(msg));

        msg.cmd = htonl(C2S_LOGOUT);

        strcpy(msg.body, username);

        if (sendto(sock, &msg, sizeof(msg), 0, (struct sockaddr *)servaddr, sizeof(struct sockaddr_in)) < 0)

            ERR_EXIT("sendto");

        printf("user %s has logout server\n", username);

        exit(EXIT_SUCCESS);

    }

    else if (strcmp(cmd, "send") == 0)

        char peername[16]={0};

        char msg[MSG_LEN]={0};

        /* send  user  msg  */

        /*       p     p2   */

        while (*p++ == ‘ ‘) ;

        char *p2;

        p2 = strchr(p, ‘ ‘);

        if (p2 == NULL)

        {

            printf("bad command\n");

            printf("\nCommands are:\n");

            printf("send username msg\n");

            printf("list\n");

            printf("exit\n");

            printf("\n");

            return;

        }

        *p2 = ‘\0‘;

        strcpy(peername, p);

        while (*p2++ == ‘ ‘) ;

        strcpy(msg, p2);

        sendmsgto(sock, peername, msg);

    else if (strcmp(cmd, "list") == 0)

        memset(&msg, 0, sizeof(msg));

        msg.cmd = htonl(C2S_ONLINE_USER);

    else

        printf("bad command\n");

        printf("\nCommands are:\n");

        printf("send username msg\n");

        printf("list\n");

        printf("exit\n");

        printf("\n");

}

bool sendmsgto(int sock, char* name, char* msg)

    if (strcmp(name, username) == 0)

        printf("can‘t send message to self\n");

        return false;

    USER_LIST::iterator it;

    for (it=client_list.begin(); it != client_list.end(); ++it)

        if (strcmp(it->username,name) == 0)

            break;

    if (it == client_list.end())

        printf("user %s has not logined server\n", name);

    MESSAGE m;

    memset(&m,0,sizeof(m));

    m.cmd = htonl(C2C_CHAT);

    CHAT_MSG cm;

    strcpy(cm.username, username);

    strcpy(cm.msg, msg);

    memcpy(m.body, &cm, sizeof(cm));

    //strcpy(m.body,msg);

    struct sockaddr_in    peeraddr;

    memset(&peeraddr,0,sizeof(peeraddr));

    peeraddr.sin_family      = AF_INET;

    peeraddr.sin_addr.s_addr = it->ip;

    peeraddr.sin_port        = it->port;

    in_addr tmp;

    tmp.s_addr = it->ip;

    printf("sending message [%s] to user [%s] <-> %s:%d\n",  msg, name, inet_ntoa(tmp), ntohs(it->port));

    sendto(sock, (const char*)&m, sizeof(m), 0, (struct sockaddr *)&peeraddr, sizeof(peeraddr));

    return true;

void do_getlist(int sock)

    int count;

    recvfrom(sock, &count, sizeof(int), 0, NULL, NULL);

    printf("has %d users logined server\n", ntohl(count));

    client_list.clear();

    int n = ntohl(count);

    for (int i=0; i<n; i++)

        USER_INFO user;

        recvfrom(sock,&user, sizeof(USER_INFO), 0, NULL, NULL);

        client_list.push_back(user);

        in_addr tmp;

        tmp.s_addr = user.ip;

        printf("%s <-> %s:%d\n", user.username, inet_ntoa(tmp), ntohs(user.port));

void do_someone_login(MESSAGE& msg)

    USER_INFO *user = (USER_INFO*)msg.body;

    tmp.s_addr = user->ip;

    printf("%s <-> %s:%d has logined server\n", user->username, inet_ntoa(tmp), ntohs(user->port));

    client_list.push_back(*user);

void do_someone_logout(MESSAGE& msg)

        if (strcmp(it->username,msg.body) == 0)

    if (it != client_list.end())

        client_list.erase(it);

    printf("user %s has logout server\n", msg.body);

void do_chat(const MESSAGE& msg)

    CHAT_MSG *cm = (CHAT_MSG*)msg.body;

    printf("recv a msg [%s] from [%s]\n", cm->msg, cm->username);

    //recvfrom(sock, &count, sizeof(int), 0, NULL, NULL);

void chat_cli(int sock)

    struct sockaddr_in servaddr;

    memset(&servaddr, 0, sizeof(servaddr));

    servaddr.sin_family = AF_INET;

    servaddr.sin_port = htons(5188);

    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");

    struct sockaddr_in peeraddr;

    socklen_t peerlen;

    MESSAGE msg;

    while (1)

        memset(username,0,sizeof(username));

        printf("please inpt your name:");

        fflush(stdout);

        scanf("%s", username);

        msg.cmd = htonl(C2S_LOGIN);

        sendto(sock, &msg, sizeof(msg), 0, (struct sockaddr *)&servaddr, sizeof(servaddr));

        recvfrom(sock, &msg, sizeof(msg), 0, NULL, NULL);

        int cmd = ntohl(msg.cmd);

        if (cmd == S2C_ALREADY_LOGINED)

            printf("user %s already logined server, please use another username\n", username);

        else if (cmd == S2C_LOGIN_OK)

            printf("user %s has logined server\n", username);

    printf("has %d users logined server\n", n);

        recvfrom(sock, &user, sizeof(USER_INFO), 0, NULL, NULL);

        printf("%d %s <-> %s:%d\n", i, user.username, inet_ntoa(tmp), ntohs(user.port));

    printf("\nCommands are:\n");

    printf("send username msg\n");

    printf("list\n");

    printf("exit\n");

    printf("\n");

    fd_set rset;

    FD_ZERO(&rset);

    int nready;

        FD_SET(STDIN_FILENO, &rset);

        FD_SET(sock, &rset);

        nready = select(sock+1, &rset, NULL, NULL, NULL);

        if (nready == -1)

                ERR_EXIT("select");

        if (nready == 0)

                continue;

        if (FD_ISSET(sock, &rset))

            peerlen = sizeof(peeraddr);

            memset(&msg,0,sizeof(msg));

            recvfrom(sock, &msg, sizeof(msg), 0, (struct sockaddr *)&peeraddr, &peerlen);

            int cmd = ntohl(msg.cmd);

            switch (cmd)

            {

            case S2C_SOMEONE_LOGIN:

                do_someone_login(msg);

                break;

            case S2C_SOMEONE_LOGOUT:

                do_someone_logout(msg);

            case S2C_ONLINE_USER:

                do_getlist(sock);

            case C2C_CHAT:

                do_chat(msg);

        default:

            }

        if (FD_ISSET(STDIN_FILENO, &rset))

            char cmdline[100] = {0};

            if (fgets(cmdline, sizeof(cmdline), stdin) == NULL)

            if (cmdline[0] == ‘\n‘)

            cmdline[strlen(cmdline) - 1] = ‘\0‘;

            parse_cmd(cmdline, sock, &servaddr);

    memset(&msg,0,sizeof(msg));

    msg.cmd = htonl(C2S_LOGOUT);

    strcpy(msg.body, username);

    sendto(sock, (const char*)&msg, sizeof(msg), 0, (struct sockaddr *)&servaddr, sizeof(servaddr));

    close(sock);

int main(void)

    int sock;

    if ((sock = socket(PF_INET, SOCK_DGRAM, 0)) < 0)

        ERR_EXIT("socket");

    chat_cli(sock);

    return 0;

chatsrv.cpp

makefile:

.PHONY:clean all

CC=g++

CPPFLAGS=-Wall -g

BIN=chatsrv chatcli

all:$(BIN)

%.o:%.cpp

    $(CC) $(CPPFLAGS) -c $< -o $@

clean:

    rm -f *.o $(BIN)

實作的代碼，有看不懂的可以留言！