目标:
如果有用户登录,其他用户可以收到这个人的登录信息
如果有人发送信息,其他用户可以收到这个人的群聊信息
如果有人下线,其他用户可以收到这个人的下线信息
服务器可以发送系统信息
分析:
服务端:
因为有终端输入和udp接收消息两个阻塞函数, 可以使用多线程或者多进程实现,一个处理输入消息发送,一个处理udp接收消息显示终端.
需要记录每个发送消息的客户端的地址信息,可用链表, 顺序表等等,这里使用单向链表.
需要处理三种消息,登录,登出,聊天,可以定义数据结构,通过字段code 定义不同的行为.
(这里定义L 登录, Q退出, C聊天)
发送系统消息功能,一种方法是可以认为服务端是一个用户,只需要给自己发送一条系统用户的聊天消息即可.
服务端作为中转, 收到客户端数据后, 直接遍历全部客户端地址列表并过滤掉发送方,然后转发即可. 不需要对数据包进行拆包组装等.
客户端:
因为需要终端输入和ddp接收消息两个阻塞函数, 可以使用多线程或者多进程实现,一个处理输入消息发送,一个处理udp接收消息显示终端.
通过用户行为组装定义好的数据结构体, 发送给客户端即可.包括聊天,登录,登出等.
head.h
#ifndef __HEAD_H__
#define __HEAD_H__
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/un.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/sem.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <fcntl.h>
#include <signal.h>
#include <pwd.h>
#include <grp.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <unistd.h>
#include <errno.h>
#include <semaphore.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <sqlite3.h>
#include <dirent.h>
#include <linux/input.h>
#define PRINT_ERR(msg) \
do { \
printf("%s:%s:%d\n", __FILE__, __func__, __LINE__); \
perror(msg); \
exit(-1); \
} while (0)
#endif
server.c
#include <head.h>
typedef struct {
char code;
char name[128];
char text[128];
}msg_t;
typedef struct Node {
struct sockaddr_in sockaddrIn;
struct Node* next;
}sockaddr_t;
sockaddr_t* sockaddr_link_create() {
sockaddr_t *s;
if ((s = (sockaddr_t *)malloc(sizeof(&s))) == NULL) {
PRINT_ERR("create_sockaddr_link malloc error");
}
s->next = NULL;
return s;
}
void sockaddr_link_insert(sockaddr_t *root, struct sockaddr_in client_addr) {
sockaddr_t *s;
if ((s = (sockaddr_t *)malloc(sizeof(&s))) == NULL) {
PRINT_ERR("create_sockaddr_link malloc error");
}
s->sockaddrIn = client_addr;
s->next = root->next;
root->next = s;
}
void sockaddr_link_delete(sockaddr_t *root, struct sockaddr_in client_addr) {
sockaddr_t *del;
while (root->next != NULL) {
if (!memcmp(&(root->next->sockaddrIn), &client_addr, sizeof(client_addr))) {
del = root->next;
root->next = root->next->next;
free(del);
del = NULL;
break;
}
root = root->next;
}
}
void showAddrs(sockaddr_t *root) {
while (root->next != NULL) {
printf("addr ---> %p\n",root->next);
root = root->next;
}
}
void doLogin(int sfd, struct sockaddr_in client_addr, msg_t msg, sockaddr_t* root) {
while (root->next != NULL) {
if (sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&(root->next->sockaddrIn), sizeof(root->next->sockaddrIn)) < 0) {
PRINT_ERR("send to error");
}
root = root->next;
}
sockaddr_link_insert(root, client_addr);
//showAddrs(root);
}
void doChat(int sfd, struct sockaddr_in client_addr, msg_t msg, sockaddr_t* root) {
while (root->next != NULL) {
if (memcmp(&client_addr, &(root->next->sockaddrIn),
sizeof(client_addr))) {
if (sendto(sfd, &msg, sizeof(msg), 0,
(struct sockaddr *)&(root->next->sockaddrIn),
sizeof(root->next->sockaddrIn)) < 0) {
PRINT_ERR("send to error");
}
}
root = root->next;
}
}
void doLogout(int sfd, struct sockaddr_in client_addr, msg_t msg, sockaddr_t* root) {
sockaddr_link_delete(root, client_addr);
//showAddrs(root);
while (root->next != NULL) {
if (sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&(root->next->sockaddrIn), sizeof(root->next->sockaddrIn)) < 0) {
PRINT_ERR("send to error");
}
root = root->next;
}
}
int main(int argc, const char *argv[]) {
if (argc < 3) {
printf("input ip port\n");
return -1;
}
int sfd;
if ((sfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
PRINT_ERR("sock error");
}
struct sockaddr_in sockaddrIn;
sockaddrIn.sin_family = AF_INET;
sockaddrIn.sin_addr.s_addr = inet_addr(argv[1]);
sockaddrIn.sin_port = htons(atoi(argv[2]));
socklen_t socklen = sizeof(sockaddrIn);
if (bind(sfd, (struct sockaddr*)&sockaddrIn, socklen) < 0) {
PRINT_ERR("bind error");
}
pid_t pid = fork();
if (pid < 0) {
PRINT_ERR("fork error");
} else if (pid == 0) {
sockaddr_t *root = sockaddr_link_create();
msg_t msg;
while (1) {
struct sockaddr_in client_sockaddr_in;
memset(&client_sockaddr_in, 0, sizeof(client_sockaddr_in));
socklen_t client_socklen = sizeof(client_sockaddr_in);
memset(&msg, 0, sizeof(msg));
if (recvfrom(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&client_sockaddr_in, &client_socklen) < 0) {
PRINT_ERR("recvfrom error");
}
printf("reveive msg, msg.code[%c], msg.name[%s], msg.text[%s]\n", msg.code, msg.name, msg.text);
switch (msg.code) {
case 'Q':
doLogout(sfd, client_sockaddr_in, msg, root);
break;
case 'C':
doChat(sfd, client_sockaddr_in, msg, root);
break;
case 'L':
doLogin(sfd, client_sockaddr_in, msg, root);
break;
}
}
} else {
while (1) {
msg_t msg;
fgets(msg.text, sizeof(msg.text), stdin);
msg.text[strlen(msg.text) - 1] = '\0';
strcpy(msg.name, "system");
msg.code = 'C';
if (sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&sockaddrIn, socklen) < 0) {
PRINT_ERR("send system message error");
}
}
}
return 0;
}
client.c
#include <head.h>
typedef struct {
char code;
char name[128];
char text[128];
}msg_t;
int main(int argc, const char *argv[]) {
int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
if (sockfd == -1) {
PRINT_ERR("socket error");
}
struct sockaddr_in serveraddr;
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
serveraddr.sin_port = htons(atoi(argv[2]));
socklen_t serveraddr_len = sizeof(serveraddr);
msg_t msg;
memset(&msg, 0, sizeof(msg));
fgets(msg.name, sizeof(msg.name), stdin);
msg.name[strlen(msg.name) - 1] = '\0';
msg.code = 'L';
strcpy(msg.text, "come in !!!");
if (sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, serveraddr_len) == -1) {
PRINT_ERR("sendto error");
}
pid_t pid = fork();
if (pid == 0) {
while (1) {
memset(&msg, 0, sizeof(msg));
if (recvfrom(sockfd, &msg, sizeof(msg), 0, NULL, NULL) < 0) {
printf("recvfrom error");
}
printf("%s: %s\n", msg.name, msg.text);
}
} else {
while (1) {
memset(msg.text, 0, sizeof(msg.text));
char buf[128] = {0};
fgets(buf, sizeof(buf), stdin);
buf[strlen(buf) - 1] = '\0';
if (!strncmp(buf, "quit", 4)) {
msg.code = 'Q';
strcpy(msg.text, "come out !!!");
} else {
msg.code = 'C';
strcpy(msg.text, buf);
}
if (sendto(sockfd, &msg, sizeof(msg), 0,
(struct sockaddr *)&serveraddr, serveraddr_len) < 0) {
PRINT_ERR("send to error");
}
if (msg.code == 'Q') {
break;
}
}
}
return 0;
}
执行示例
开启服务器和三个客户端
操作顺序:
client1 输入A, client2 输入B, client3 输入C, 此时A,B,C三个用户登录成功
client2 输入hello, client1 输入world, client3 输入nihao, 可以看到聊天内容
client2 输入quit, client1 输入 quit, client3 输入quit, 可以看到登出过程
server
ubuntu@ubuntu:~ $ ./server 127.0.0.1 8888
reveive msg, msg.code[L], msg.name[A], msg.text[come in !!!]
reveive msg, msg.code[L], msg.name[B], msg.text[come in !!!]
reveive msg, msg.code[L], msg.name[C], msg.text[come in !!!]
reveive msg, msg.code[C], msg.name[B], msg.text[hello]
reveive msg, msg.code[C], msg.name[A], msg.text[world]
reveive msg, msg.code[C], msg.name[C], msg.text[nihao]
reveive msg, msg.code[Q], msg.name[B], msg.text[come out !!!]
reveive msg, msg.code[Q], msg.name[A], msg.text[come out !!!]
reveive msg, msg.code[Q], msg.name[C], msg.text[come out !!!]
^C
ubuntu@ubuntu:~ $
client1
ubuntu@ubuntu:~ $ ./client 127.0.0.1 8888
A
B: come in !!!
C: come in !!!
B: hello
world
C: nihao
B: come out !!!
quit
ubuntu@ubuntu:~ $
client2
ubuntu@ubuntu:~ $ ./client 127.0.0.1 8888
B
C: come in !!!
hello
A: world
C: nihao
quit
ubuntu@ubuntu:~ $
client3
ubuntu@ubuntu:~ $ ./client 127.0.0.1 8888
C
B: hello
A: world
nihao
B: come out !!!
A: come out !!!
quit
ubuntu@ubuntu:~ $