基于UDP的网络聊天室

项目需求：

1. 如果有用户登录，其他用户可以收到这个人的登录信息
2. 如果有人发送信息，其他用户可以收到这个人的群聊信息
3. 如果有人下线，其他用户可以收到这个人的下线信息
4. 服务器可以发送系统信息

1.服务器

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:%d ",__LINE__);\
	perror(msg);\
}while(0);
#define IP "192.168.9.19"   //本机IP ifconfig
#define PORT 6665           //1024-49151

//用户信息结构体
typedef struct Node
{

	struct sockaddr_in cin;
	struct Node *next;
}Linklist;
struct _msg
{
	char type;  //登录L,群聊C,退出Q
	char name[32];
	char text[128];
};

//Linklist Create_head();                                 //申请头部堆区空间
//Linklist Create_node(struct sockaddr_in cin);           //申请其他结点堆区空间
void insert_head (Linklist L,Linklist p);                //头插

int do_recv(int sfd,struct _msg msg,Linklist *L);
int do_login(int sfd,struct _msg msg,Linklist *L,struct sockaddr_in cin);
int do_chat(int sfd,struct _msg msg,Linklist *L,struct sockaddr_in cin);
int do_quit(int sfd,struct _msg msg,Linklist *L,struct sockaddr_in cin);
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd = socket(AF_INET,SOCK_DGRAM,0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("sfd = %d\n",sfd);

	//填充地址信息结构体
	//真实的地址信息结构体根据地址族指定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;        //必须填AF_INEF
	sin.sin_port   =htons(PORT);     //端口号,1024-49151
	sin.sin_addr.s_addr =inet_addr(IP); //本机IP

	//将IP和端口绑定到套接字上
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n",__LINE__);
		

	ssize_t res = 0;
	struct _msg msg;   //定义接收信息的变量

//创建子进程
	int cpid = 0;
	cpid= fork();
	if(-1 == cpid)
	{
		perror("fork");
		return -1;
	}
	//父进程接受recvfrom
	if(cpid > 0)
	{
		Linklist* L = (Linklist*)malloc(sizeof(Linklist));
		L->next=NULL;
		do_recv(sfd,msg,L);

	}
	//子进程做sendto
	else if(0 == cpid)
	{
		//子进程,用来发送系统消息
		strcpy(msg.name,"系统信息");
      	msg.type = 'C';
		while(1)
		{
			bzero(msg.text,sizeof(msg.text));
			fgets(msg.text,sizeof(msg.text),stdin);
			msg.text[strlen(msg.text)-1]='\0';
			if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&sin,sizeof(sin))<0)
			{
				perror("sendto");
				return -1;
			}
		}
		printf("系统消息发送成功\n");
	}

	close(sfd);
	return 0;
}

//登录信息更新
int  do_login(int sfd,struct _msg msg,Linklist* L,struct sockaddr_in cin)
{
	printf("%s[%s:%d]登录成功\n",msg.name,(char *)inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
	sprintf(msg.text,"-----%s登录成功-----",msg.name);
	//先遍历链表,将新用户加入群聊的消息发送给所有人
	while(L->next!=NULL)
	{
		L = L->next;
		if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&(L->cin),sizeof(L->cin))<0)
		{
			perror("sendto");
			return -1 ;
		}
	}
	//将新用户的网络信息结构体,头插入链表

	Linklist* temp = (Linklist*)malloc(sizeof(Linklist));
	temp->cin=cin;
	temp->next = NULL;
	L->next = temp;
	return 0 ;
}
//接收并打印群聊
int do_chat(int sfd,struct _msg msg,Linklist* L,struct sockaddr_in cin)
{
	printf("%s[%s:%d]chat成功\n",msg.name,(char*)inet_ntoa(cin.sin_addr),\
			ntohs(cin.sin_port));

	//遍历链表,将群聊发给除自己以外的所有人
	while(L->next!=NULL)
	{
		L = L->next;
		if(memcmp(&cin,&L->cin,sizeof(cin))!=0)    //判断是是否为自己
		{
			if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&(L->cin),sizeof(L->cin))<0)
			{
				perror("sendto");
				return -1;
			}
		}
	}
	return 0;

}

//退出下线操作
int do_quit(int sfd,struct _msg msg,Linklist* L,struct sockaddr_in cin)
{
	//遍历链表,是自己就把自己在链表中删除,不是自己,就发生退出群聊的数据
	while(L->next!=NULL)
	{
		if(memcmp(&L->next->cin,&cin,sizeof(cin))!=0)
		{
			L = L->next;
			if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&L->cin,sizeof(L->cin))<0)
			{
				perror("sendto");
				return -1;
			}
		}
		else
		{
			Linklist * ptemp = L->next;
			L->next=ptemp->next;
			free(ptemp);
			ptemp = NULL;
		}
	}
	return 0 ;
}
int do_recv(int sfd,struct _msg msg,Linklist *L)
{
	struct sockaddr_in cin; //存储连接成功的客户端地址信息
	socklen_t addrlen=sizeof(cin);

	//		Linklist L= Create_head();   //申请头部堆区空间
	while(1)
	{
		//接收消息
		memset(&msg,0,sizeof(msg));
		memset(&cin,0,sizeof(cin));
		int res = 0;
		res = recvfrom(sfd,&msg,sizeof(msg),0,(struct sockaddr *)&cin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		//提取出协议
		switch(msg.type)
		{
		case 'L':
			do_login(sfd,msg,L,cin);
			break;
		case 'C':
			do_chat(sfd,msg,L,cin);
			break;
		case 'Q':
			do_quit(sfd,msg,L,cin);
			break;
		}

	}
}

2.客户端

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/wait.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:%d ",__LINE__);\
	perror(msg);\
}while(0);
#define SER_IP "192.168.9.19"   //本机IP ifconfig
#define SER_PORT 6665           //1024-49151

struct node
{
	char type;   //协议'L'登录,'C'发送群聊,'Q'下线
	char name[32];
	char text[128];
};
void handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);
	exit(0);
}
int main(int argc, const char *argv[])
{
	__sighandler_t s=signal(SIGCHLD,handler);
	if(s == SIG_ERR)
	{
		perror("signal");
		return -1;
	}

	//创建报式套接字
	int cfd = socket(AF_INET,SOCK_DGRAM,0);
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("cfd = %d\n",cfd);

	//填充地址信息结构体
	//真实的地址信息结构体根据地址族指定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;        //必须填AF_INEF
	sin.sin_port   =htons(SER_PORT);     //端口号,1024-49151
	sin.sin_addr.s_addr =inet_addr(SER_IP); //本机IP


//	struct sockaddr_in sinaddr; //存储连接成功的服务端地址信息
//	socklen_t addrlen=sizeof(sinaddr);
 
	//发送登录信息给服务器
	struct node argee;
	memset(&argee,0,sizeof(argee));
	printf("请输入登录用户>>>");
	fgets(argee.name,sizeof(argee.name),stdin);
	argee.name[strlen(argee.name)-1]='\0';
	argee.type = 'L';
	strcpy(argee.text,"加入群聊");
	if(sendto(cfd,&argee,sizeof(argee),0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	int cpid=0;
	cpid = fork();
	if(cpid <0)
	{
		perror("fork");
		return -1;
	}
	//父进程接收recvfrom
	if(cpid > 0)
	{
		while(1)
		{
			if(recvfrom(cfd,&argee,sizeof(argee),0,NULL,NULL)<0)
			{
				perror("recvfrom");
				return -1;
			}
			printf("[%s]:%s\n",argee.name,argee.text);
		}
		
	}
	//子进程发送sendto
	else if(cpid == 0)
	{
		while(1)
		{
			bzero(argee.text,sizeof(argee.text));
			//发送数据包
			fgets(argee.text,sizeof(argee.text),stdin);
			argee.text[strlen(argee.text)-1]='\0';
			if(!strcmp(argee.text,"quit"))
			{
				argee.type='Q';
				strcpy(argee.text,"退出群聊");
			}
			else
			{
				argee.type='C';
			}
			if(sendto(cfd,&argee,sizeof(argee),0,(struct sockaddr*)&sin,sizeof(sin))<0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			if(!strcmp(argee.text,"退出群聊"))
			{
				break;
			}

		}

	}
	close(cfd);
	return 0;
}

3.电子词典的上传

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sqlite3.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
int main(int argc, const char *argv[])
{
	//创建打开库文件
	sqlite3 *db = NULL;
	if(sqlite3_open("./dict.db",&db)!=SQLITE_OK)
	{
		fprintf(stderr,"sqlite3_open :%s  errcode : %d\n",sqlite3_errmsg(db),sqlite3_errcode(db));
		return -1;
	}

	//创建一个表
	char mu[128]="";
	char sql[128]="create table if not exists stu1 (word char ,chinese char);";
	char *errmsg = NULL;
	
	if(sqlite3_exec(db,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
	{
		fprintf(stderr,"line: %d sqlite3_exec :%s\n",__LINE__,errmsg);
		return -1;
	}
	printf("create table stu1 success\n");
	
	FILE *fp =fopen("./dict.txt","r");
	if(NULL == fp)
	{
		perror("fopen");
		return -1;
	}
	ssize_t res=0;
	char buff[128]="";
//	char *word;
//	char *chinese;
	//读出文件
	while(1)
	{
		bzero(buff,sizeof(buff));
		if(fgets(buff,sizeof(buff),fp)==NULL)
			break;
		int len =strlen(buff);
		buff[len-1]='\0';
		int i;
		for(i=0;i<len;i++)
		{
			if(buff[i]==' '&& buff[i+1]==' ')
			{
				buff[i+1]='\0';
				sprintf(mu,"insert into stu1 values (\"%s\",'%s')",buff,&(buff[i+3]));
				if(sqlite3_exec(db,mu,NULL,NULL,&errmsg)!=SQLITE_OK)
				{
					fprintf(stderr,"line: %d sqlite3_exec :%s\n",__LINE__,errmsg);
					return -1;
				}

				break;
			}
		}
	}

	fclose(fp);
	sqlite3_close(db);
	return 0;
}