总结下近期写的小项目,在学习中同时积累解决问题的经验,以及真正的项目中解决问题的思路,如有不合理地方,请多指教!
一、项目名称
基于UDP的网络聊天室
二、功能
1.当有新用户登录时,其他在线用户可以收到该用户登录的消息。
2.当用户发送聊天信息时,其他所有在线用户均能收到群聊信息。
3.当有用户退出时,其他所有在线用户均能收到退出消息。
4.服务器可以发送系统消息。
三、初步构思
1.服务器需要给多个用户发送信息,需要创建链表保存各用户的网络信息。
2.消息类型有四种,上线、群聊、退出、系统消息,需要定义消息结构体,结构体中需要包含信息操作码、用户名、消息数据。
四、流程图
客户端
服务器端:
五、代码实现
服务器端:
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
//宏定义高频率使用的判断出错
#define PRINERR(msg) \
do { \
printf("%s:%s:%d\n", __FILE__, __func__, __LINE__); \
perror(msg); \
exit(-1); \
} while (0)
typedef struct _MSG {
char code; //操作码:'L'表示登录 'C'表示群聊 'Q'表示退出 'S'表示系统消息
//定义操作码为1个字符,免去考虑网络字节序问题
char name[45];
char txt[256];
} msg_t; //定义消息结构体类型
typedef struct _ADDR {
struct sockaddr_in clientaddr;
struct _ADDR* rear;
} addrlist_t; //定义保存客户端网络信息结构体的链表类型
//登录操作的函数
void do_login(int sockfd,msg_t msg,addrlist_t*addr,struct sockaddr_in clientaddr){
//先遍历链表 将新用户加入群聊的消息发给所有人
addrlist_t* tmp = addr;//记录链表头节点
while(tmp->rear != NULL){
tmp = tmp->rear;
if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(tmp->clientaddr), sizeof(tmp->clientaddr))){
PRINERR("sendto error");
}
}
//将新用户的网络信息结构体 头插入链表
addrlist_t* pnew = NULL;
if (NULL == (pnew = (addrlist_t*)malloc(sizeof(addrlist_t)))) {
printf("malloc error\n");
return;
}
pnew->clientaddr = clientaddr;
pnew->rear = addr->rear;
addr->rear = pnew;
return;
}
//群聊操作的函数
void do_chat(int sockfd,msg_t msg,addrlist_t*addr,struct sockaddr_in clientaddr){
//遍历链表,将群聊消息发给除了自己之外的所有人
addrlist_t* ptmp = addr;
while(ptmp->rear != NULL){
ptmp = ptmp->rear;
if(memcmp(&clientaddr, &(ptmp->clientaddr), sizeof(clientaddr))){
//说明不是自己 就发送数据
if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(ptmp->clientaddr), sizeof(ptmp->clientaddr))){
PRINERR("sendto error");
}
}
}
return;
}
//退出操作的函数
void do_quit(int sockfd,msg_t msg,addrlist_t*addr,struct sockaddr_in clientaddr){
//遍历链表 是自己就将自己在链表中删除
//不是自己 就发送 退出群聊的数据
addrlist_t* ptmp = addr;
addrlist_t* del = NULL;
while(ptmp->rear != NULL){
if(memcmp(&(ptmp->rear->clientaddr), &clientaddr, sizeof(clientaddr))){
//不是自己
ptmp = ptmp->rear;
if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(ptmp->clientaddr), sizeof(ptmp->clientaddr))){
PRINERR("sendto error");
}
}else{
//是自己
del = ptmp->rear;
ptmp->rear = del->rear;
free(del);
del = NULL;
}
}
return;
}
int main(int argc, const char* argv[])
{
if (3 != argc) { //考虑用命令行传参方式输入ip地址及端口号,先进行参数判断
printf("input error!\n");
printf("usage: %s <IP> <port>\n",argv[0]);
return -1;
}
int sockfd;
if (-1 == (sockfd = socket(AF_INET, SOCK_DGRAM, 0))) {
PRINERR("socket error");
}
//填充服务器网络信息结构体
struct sockaddr_in serveraddr;
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(atoi(argv[2]));
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
socklen_t serveraddr_len = sizeof(serveraddr);
//将套接字绑定网络信息
if (-1 == bind(sockfd, (struct sockaddr*)&serveraddr, serveraddr_len)) {
PRINERR("bind error");
}
//定义客户端网络信息结构体
struct sockaddr_in clientaddr;
socklen_t clientaddr_len = sizeof(clientaddr);
msg_t msg; //定义接收信息的变量msg
pid_t pid; //进程号
if (-1 == (pid = fork())) { //创建多进程
PRINERR("fock error");
}
if (0 == pid) { //子进程,用来收发数据
//创建保存客户端信息的链表头节点
addrlist_t* addr;
if (NULL == (addr = (addrlist_t*)malloc(sizeof(addrlist_t)))) {
printf("malloc error\n");
return -1;
}
memset(addr, 0, sizeof(addr));
addr->rear = NULL;
while (1) { //循环收发数据
//每次接收新数据及新用户信息前清空,确保信息准确
memset(&msg, 0, sizeof(msg));
memset(&clientaddr, 0, sizeof(clientaddr));
//接收客户端发送的消息,存放在msg中
if (-1 == recvfrom(sockfd, &msg, sizeof(msg), 0, (struct sockaddr*)&clientaddr, &clientaddr_len)) {
PRINERR("recvfrom error");
}
switch (msg.code) { //判断消息中的操作码,根据操作码执行对应操作
case 'L': //登录操作
do_login(sockfd,msg,addr,clientaddr);
break;
case 'C': //群聊操作
do_chat(sockfd,msg,addr,clientaddr);
break;
case 'Q': //退出操作
do_quit(sockfd,msg,addr,clientaddr);
break;
}
}
} else {
//父进程,用来发送系统消息
//将父进程当做一个客户端,向子进程发送群聊消息
//msg在fock 前定义的,所以父进程中同样有msg
strcpy(msg.name, "系统消息");
msg.code = 'C';
while(1){
memset(msg.txt,0,sizeof(msg.txt));
fgets(msg.txt, 256, stdin);//循环在终端接收消息
msg.txt[strlen(msg.txt)-1] = '\0';
if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, serveraddr_len)){
PRINERR("sendto error");
}
}
}
close(sockfd);
return 0;
}
客户端:
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <signal.h>
#include <arpa/inet.h>
#include <sys/wait.h>
//宏定义高频率使用的判断出错
#define PRINERR(msg) \
do { \
printf("%s:%s:%d\n", __FILE__, __func__, __LINE__); \
perror(msg); \
exit(-1); \
} while (0)
typedef struct _MSG {
char code; //操作码:'L'表示登录 'C'表示群聊 'Q'表示退出 'S'表示系统消息
//定义操作码为1个字符,免去考虑网络字节序问题
char name[45];
char txt[256];
} msg_t; //定义消息结构体类型
int main(int argc, const char *argv[])
{
if (3 != argc) { //考虑用命令行传参方式输入ip地址及端口号,先进行参数判断
printf("input error!\n");
printf("usage: %s <IP> <port>\n",argv[0]);
return -1;
}
int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
if(-1 == sockfd){
PRINERR("socket error");
}
struct sockaddr_in serveraddr;
memset(&serveraddr, 0, sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(atoi(argv[2]));
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
socklen_t serveraddr_len = sizeof(serveraddr);
msg_t msg;
memset(&msg, 0, sizeof(msg));
//输入用户名
printf("请输入用户名:>>");
fgets(msg.name, 45, stdin);
msg.name[strlen(msg.name)-1] = '\0';
msg.code = 'L';
strcpy(msg.txt, "加入群聊");
//给服务器发送登录的信息
if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, serveraddr_len)){
PRINERR("sendto error");
}
pid_t pid = 0;
if(-1 == (pid = fork())){
PRINERR("fork error");
}
if(0 == pid){//子进程,循环接收并打印收到的数据
while(1){
if(-1 == recvfrom(sockfd, &msg, sizeof(msg), 0, NULL, NULL)){
PRINERR("recvfrom error");
}
//打印收到的数据
printf("[%s]: %s\n", msg.name, msg.txt);
}
}else if(0 < pid){//父进程,循环在终端接收数据并发送给客户端
while(1){
memset(msg.txt,0,sizeof(msg.txt));
fgets(msg.txt, 128, stdin);//在终端获取聊天信息
msg.txt[strlen(msg.txt)-1] = '\0';
if(!strcmp(msg.txt, "quit")){
msg.code = 'Q';
strcpy(msg.txt, "退出群聊");
}else{
msg.code = 'C';
}
if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, serveraddr_len)){
PRINERR("sendto error");
}
if(!strcmp(msg.txt, "退出群聊")){
break;
}
}
//杀死子进程
kill(pid, SIGKILL);
wait(NULL);//等待回收子进程资源
}
close(sockfd);
return 0;
}
六、最终效果