• 网络编程day6作业

    完成网络聊天室编写

    ser

    1. #define ERR_MSG(msg) do{\
    2. 	fprintf(stderr,"__%d__",__LINE__);\
    3. 	perror(msg);\
    4. }while(0)
    5. #define IP "127.0.0.1"
    6. #define PORT 6666
    7. //创建链表
    8. Linklistptr list_create();
    9. Linklistptr node_buy(datatype e);
    10. int list_insert_head(Linklistptr L,datatype e);
    11. int chat_login(Linklistptr L,int sfd,msg_t msg,datatype e);
    12. int chat_enter(Linklistptr l,int sfd,msg_t msg,datatype e);
    13. int chat_exit(Linklistptr L,int sfd,msg_t msg,datatype e);
    14. typedef struct sockaddr_in datatype;//类型重命名
    15. //创建信息结构体
    16. typedef struct msg
    17. {
    18. 	char type;//操作码 'L'登录 'C'群聊 'Q'退出
    19. 	char name[20];
    20. 	char text[128];
    21. }msg_t;
    22. //创建链表保存地址信息
    23. typedef struct Node
    24. {
    25. 	union
    26. 	{
    27. 		datatype res_addr;//数据域
    28. 		int len;//头结点数据域
    29. 	};
    30. 	struct Node *next;//指针域
    31. }Node, *Linklistptr;
    32. int main(int argc, const char *argv[])
    33. {
    34. 	//创建报式套接字
    35. 	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
    36. 	if(sfd < 0)
    37. 	{
    38. 		ERR_MSG("socket");
    39. 		return -1;
    40. 	}
    41. 	printf("socket create success sfd=%d\n",sfd);
    42. 	//填充接收方的地址信息结构体，给bind函数使用
    43. 	datatype sin;
    44. 	sin.sin_family = AF_INET;
    45. 	sin.sin_port = htons(PORT);
    46. 	sin.sin_addr.s_addr = inet_addr(IP);
    47. 	//绑定地址信息结构体
    48. 	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    49. 	{
    50. 		ERR_MSG("bind");
    51. 		return -1;
    52. 	}
    53. 	printf("bind success");
    54. 	//创建接收地址信息结构体
    55. 	datatype cin;
    56. 	socklen_t addrlen = sizeof(cin);
    57. 	pthread_t tid;
    58. 	msg_t msg;
    59. 	//创建一个链表
    60. 	Linklistptr L = list_create();
    61. 	if(NULL == L)
    62. 	{
    63. 		return -1;
    64. 	}
    65. 	while(1)
    66. 	{
    67. 		//主线程负责接收并处理
    68. 		if(recvfrom(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&cin, &addrlen) < 0)
    69. 		{
    70. 			ERR_MSG("recvfrom");
    71. 			return -1;
    72. 		}
    73. 		printf("[%s:%d] : %s\n", msg.name, ntohs(cin.sin_port), msg.text);
    74. 		switch(msg.type)
    75. 		{
    76. 		case 'L'://登录
    77. 			chat_login(L,sfd,msg,e);
    78. 			break;
    79. 		case 'C'://群聊
    80. 			chat_enter(L,sfd,msg,e);
    81. 			break;
    82. 		case 'Q'://退出
    83. 			chat_exit(L,sfd,msg,e);
    84. 			break;
    85. 		default :
    86. 			printf("输入错误\n");
    87. 			break;
    88. 		}
    89. 		info.sfd=sfd;
    90. 		info.sin=sin;
    91. 		//分支线程只负责发送系统信息
    92. 		if(pthread_create(&tid, NULL, task,(void*)&info) != 0)
    93. 		{
    94. 			fprintf(stderr, "pthread_create  failed__%d__\n",__LINE__);
    95. 			return -1;
    96. 		}
    97. 		pthread_detach(tid);
    98. 	}
    99. 	//关闭文件描述符
    100. 	if(close(sfd) < 0)
    101. 	{
    102. 		ERR_MSG("close");
    103. 		return -1;
    104. 	}
    105. 	return 0;
    106. }
    107. void *task(void *arg)
    108. {
    109. 	int sfd = ((struct Climsg*)arg)->sfd;
    110. 	datatype sin = ((struct Climsg*)arg)->sin;
    111. 	msg_t msg;
    112. 	msg.type = 'C';
    113.  
    114. 	while(1)
    115. 	{
    116.  
    117. 		//从终端获取消息文本
    118. 		fgets(msg.text, sizeof(msg.text), stdin);
    119. 		msg.text[strlen(msg.text)-1] = '\0';
    120. 		//将信息包名定为服务器
    121. 		strcpy(msg.name,"servce");
    122. 		if(sendto(sfd , &msg, sizeof(msg), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    123. 		{
    124. 			ERR_MSG("sendto");
    125. 		}
    126. 	}
    127. 	close(sfd);
    128. 	pthread_exit(NULL);
    129. }
    130. //创建链表
    131. Linklistptr list_create()
    132. {
    133. 	//从堆区申请一个头结点类型
    134. 	Linklistptr L = (Linklistptr)malloc(sizeof(Node));
    135. 	if(NULL == L)
    136. 	{
    137. 		printf("创建失败\n");
    138. 		return NULL;
    139. 	}
    140. 	//创建成功后，对节点进行初始化工作
    141. 	L->len = 0;
    142. 	L->next = NULL;
    143. 	return L;
    144. }
    145. //申请节点封装地址信息
    146. Linklistptr node_buy(datatype e)
    147. {
    148. 	//在堆区申请节点
    149. 	Linklistptr p = (Linklistptr)malloc(sizeof(Node));
    150. 	if(NULL == p)
    151. 	{
    152. 		printf("申请失败]n");
    153. 		return NULL;
    154. 	}
    155. 	//节点申请成功,封装数据
    156. 	p->res_addr = e;
    157. 	p->next = NULL;
    158. 	return p;
    159. }
    160. //头插
    161. int list_insert_head(Linklistptr L,datatype e)
    162. {
    163. 	//判断逻辑
    164. 	if(NULL == L)
    165. 	{
    166. 		printf("所给链表不合法\n");
    167. 		return -1;
    168. 	}
    169. 	//调用节点封装数据
    170. 	
    171. 	Linklistptr p = node_buy(e);
    172. 	if(NULL == p)
    173. 	{
    174. 		return -1;
    175. 	}
    176. 	//头插
    177. 	p->next = L->next;
    178. 	L->next = p;
    179. 	//表长变化
    180. 	L->len++;
    181. 	return 0;
    182. }
    183. int chat_login(Linklistptr L,int sfd,msg_t msg,datatype e)
    184. {
    185. 	Linklistptr p=L;//定义一个遍历指针
    186. 	while(p->next!=NULL)
    187. 	{
    188. 		p=p->next;
    189. 		if(sendto(sfd,&msg,sizeof(msg_t),0,(struct sockaddr *)&(p->res_addr),sizeof(p->res_addr))<0)
    190. 		{
    191. 			ERR_MSG("sendto");
    192. 			return -1;
    193. 		}
    194. 	}
    195. 	//头插，将自己的地址存入链表中
    196. 	list_insert_head(L,e);
    197. 	return 0;
    198. }
    199. //群聊
    200. int chat_enter(Linklistptr l,int sfd,msg_t msg,datatype e)
    201. {
    202. 	//定义一个遍历指针
    203. 	Linklistptr p=L;
    204. 	while(p->next!=NULL)
    205. 	{
    206. 		p=p->next;
    207. 		//判断链表客户端信息
    208. 		if(memcmp(&(p->res_addr),&e,sizeof(e))==0)
    209. 		{
    210. 			if(sendto(sfd,&msg,sizeof(msg_t),0,(struct sockaddr*)&(p->res_addr),sizeof(p->res_addr))<0)
    211. 			{
    212. 				ERR_MSG("sendto");
    213. 				return -1;
    214. 			}
    215. 		}
    216. 	}
    217. 	return 0;
    218. }
    219. //退出
    220. int chat_exit(Linklistptr L,int sfd,msg_t msg,datatype e)
    221. {
    222. 	Linklistptr p=L;
    223. 	while(p->next!=NULL)
    224. 	{	
    225. 		p=p->next;
    226. 		//判断链表客户端信息
    227. 		if(memcmp(&(p->res_addr),&e,sizeof(e))==0)
    228. 		{
    229. 			if(sendto(sfd,&msg,sizeof(msg_t),0,(struct sockaddr*)&(p->res_addr),sizeof(p->res_addr))<0)
    230. 			{
    231. 				ERR_MSG("sendto");
    232. 				return -1;
    233. 			}
    234. 		}
    235. 		else//此时当前节点的下一个节点保存的就是要退出的成员的信息
    236. 		{
    237. 			Linklistptr q=p->next;
    238. 			if(sendto(sfd, &msg, sizeof(msg_t), 0, (struct sockaddr*)&(q->resin), sizeof(q->resin)) < 0)
    239. 			{
    240. 				ERR_MSG("sendto");
    241. 				return -1;
    242. 			}
    243. 			p->next=q->next;
    244. 			free(q);
    245. 			q=NULL;
    246. 		}
    247. 	}
    248. 	return 0;
    249. }

    cli

    1. #include 
    2. #define ERR_MSG(msg) do{\
    3. 	fprintf(stderr,"__%d__",__LINE__);\
    4. 	perror(msg);\
    5. }while(0)
    6. #define IP "127.0.0.1"
    7. #define PORT 4399
    8. typedef struct sockaddr_in datatype;//类型重命名
    9. //创建信息结构体
    10. typedef struct msg
    11. {
    12. 	char type;//操作码 'L'登录 'C'群聊 'Q'退出
    13. 	char name[20];
    14. 	char text[128];
    15. }msg_t;
    16.  
    17. struct Climsg
    18. {
    19. 	int cfd;
    20. 	datatype cin;
    21. 	msg_t msg;
    22. };
    23. void *task(void *arg);
    24. int main(int argc, const char *argv[])
    25. {
    26. 		//创建报式套接字
    27. 	int cfd = socket(AF_INET, SOCK_DGRAM, 0);
    28. 	if(cfd < 0)
    29. 	{
    30. 		ERR_MSG("socket");
    31. 		return -1;
    32. 	}
    33. 	printf("socket create success cfd=%d\n",cfd);
    34. 	//填充接收方的地址信息结构体，给sendto函数使用
    35. 	datatype sin;
    36. 	sin.sin_family = AF_INET;
    37. 	sin.sin_port = htons(PORT);
    38. 	sin.sin_addr.s_addr = inet_addr(IP);
    39. 	//创建接收地址信息结构体
    40. 	datatype cin;
    41. 	socklen_t c_addrlen = sizeof(cin);	
    42. 	pthread_t tid;
    43. 	msg_t msg;
    44. 	memset(&msg, 0, sizeof(msg_t));
    45. 	struct Climsg info;
    46. 	printf("请输入登录名：");
    47. 	fgets(msg.name, sizeof(msg.name), stdin);
    48. 	msg.name[strlen(msg.name)-1] = '\0';	
    49. 	msg_t msg1=msg;
    50. 	msg.type = 'L';
    51. 	strcpy(msg.text,"已进入群聊");
    52. 	//发送登录请求包
    53. 	if(sendto(cfd, &msg, sizeof(msg_t), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    54. 	{
    55. 		ERR_MSG("sendto");
    56. 		return -1;
    57. 	}
    58. 	printf("登陆成功\n");
    59. 	info.cfd = cfd;
    60. 	info.cin = sin;
    61. 	info.msg = msg;	
    62. 	while(1)
    63. 	{
    64. 		
    65. 		memset(&msg, 0, sizeof(msg));
    66. 		if(recvfrom(cfd, &msg, sizeof(msg_t), 0, (struct sockaddr*)&cin, &c_addrlen) < 0)
    67. 		{
    68. 			ERR_MSG("recvfrom");
    69. 			return -1;
    70. 		}
    71. 		if(strcmp(msg.name, msg1.name) == 0)
    72. 		{
    73. 			break;
    74. 		}
    75.  
    76. 		printf("[%s] : %s\n", msg.name, msg.text);
    77.  
    78.  
    79. 		if(pthread_create(&tid, NULL, task, (void *)&info) != 0)
    80. 		{
    81. 			ERR_MSG("pthread_create");
    82. 			return -1;
    83. 		}
    84. 	
    85. 		//将分支线程设置为分离态
    86. 		pthread_detach(tid);
    87. 		}
    88.  
    89. 	//关闭套接字描述符
    90. 	close(cfd);
    91. 	return 0;
    92. }
    93. //线程体函数
    94. void *task(void *arg)
    95. {
    96.  
    97. 	int cfd = ((struct Climsg*)arg)->cfd;
    98. 	datatype sin = ((struct Climsg*)arg)->cin;
    99. 	msg_t msg = ((struct Climsg*)arg)->msg;
    100.  
    101. 	while(1)
    102. 	{
    103. 		//清空文本内容
    104. 		bzero(msg.text, sizeof(msg.text));
    105. 		//从终端获取数据
    106. 		fgets(msg.text, sizeof(msg.text), stdin);
    107. 		msg.text[strlen(msg.text)-1] = '\0';
    108.  
    109. 		msg.type = 'C';
    110. 		if(strcmp(msg.text,"quit") == 0)
    111. 		{
    112. 			msg.type = 'Q';
    113. 			strcpy(msg.text, "已下线");
    114. 		}
    115. 		if(sendto(cfd, &msg, sizeof(msg_t), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    116. 		{
    117. 			ERR_MSG("sendto");
    118. 		}
    119. 		if(strcmp(msg.text, "已下线") == 0)
    120. 		{
    121. 			break;
    122. 		}
    123. 	}
    124. 	close(cfd);
    125. 	pthread_exit(NULL);
    126. }

    思维导图：https://mubu.com/app/edit/home/5fnWgXpb5GT#m

  • 相关阅读:
    【疯狂Java】数组
    SpringBoot+Kafka
    CSS3自定义滚动条的颜色等样式 - hover显示切换 - 伪类::-webkit-scrollbar的样式控制
    vulhub中Apache Log4j2 lookup JNDI 注入漏洞（CVE-2021-44228）
    短信的信令过程
    论文《Unsupervised Dialog Structure Learning》笔记：详解DD-VRNN
    uniapp navigateTo url路由传参(传递对象)
    PHP：命名空间必知必会
    【HMS Core】华为分析服务通过REST方式上报用户行为，控制台为何无法查询到相关数据？
    ElasticSearch三种分页对比
  • 原文地址：https://blog.csdn.net/m0_59031281/article/details/132656609