首页 > 代码库 > Linux网络编程7——使用TCP实现双方聊天
Linux网络编程7——使用TCP实现双方聊天
思路
主线程负责发送消息,另一线程负责接收消息。服务端和客户端均是如此。
注意
当A方close掉用于通信的socket端口后,该端口是不会立即关闭的。因为此时可能B方的信息还没send完。因此,此时A方的recv仍旧处于阻塞状态,会最后再等待收一次信息。此时,当B方send一个信息给A后,A方recv到后,A的socket端口就正式关闭了,A的recv返回-1。
此时由于A的socket端口已关闭,因此B得recv返回0。
代码
本代码服务器和客户端程序写在一起了。具体请看注释。
/************************************************************************* > File Name: my_chat.c > Author: KrisChou > Mail:zhoujx0219@163.com > Created Time: Thu 28 Aug 2014 11:06:04 PM CST ************************************************************************/#include <stdio.h>#include <stdlib.h>#include <string.h>#include <sys/types.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#include <unistd.h>#include <pthread.h>#include <netdb.h>#define ACTIVE 1 #define PASSIVE 0#define SERVER_PORT 1314#define CLIENT_PORT 1350/* 接收消息通过线程实现 */void* chat_handler(void* arg){ int fd_client = (int)arg; char buf[1024]; int test; while(memset(buf, 0, 1024), (test=recv(fd_client, buf, 1024, 0)) > 0) { write(1, buf, strlen(buf)); } //用于测试recv返回值 printf("break child while!\n"); printf("test = %d \n",test); return (void*)0; //close(fd_client);}int main(int argc,char* argv[])// exe peer_ip{ int flag; int fd_peer; /* 对方socket描述符,由acccept返回 */ if(argc == 1) { flag = PASSIVE ; }else if(argc == 2) { flag = ACTIVE ; } /*---------------------------------------socket----------------------------------------*/ int sfd ; /* sfd为本地socket描述符 */ sfd = socket(AF_INET, SOCK_STREAM, 0) ; if(sfd == -1) { perror("socket"); exit(1); } /*------------ ---------------------------bind -----------------------------------------*/ char my_hostname[1024]=""; struct hostent * p ; gethostname(my_hostname, 1024); /* 获取本地hostname,以便根据hostname查出ip */ p = gethostbyname(my_hostname) ;/* 根据hostname,获取指向struct hostent结构体的指针 */ struct sockaddr_in local_addr ; /* 绑定socket端口以及IP的结构体 */ memset(&local_addr, 0, sizeof(local_addr)); local_addr.sin_family = AF_INET ; if(flag == ACTIVE) { local_addr.sin_port = htons(CLIENT_PORT); /* 作为主动方,需要绑定的端口是client_port */ }else { local_addr.sin_port = htons(SERVER_PORT); /* 作为服务器,需要绑定的端口server_port */ /*两者均为本机端口*/ } local_addr.sin_addr = *(struct in_addr *)(p ->h_addr_list)[0]; /* 绑定IP地址 */ /* 本地IP */ if(-1 == bind(sfd, (struct sockaddr *)&local_addr, sizeof(local_addr))) { perror("bind"); close(sfd); exit(1); } /*----------------------------------如果是服务器,listen+accept---------------------------------------------- */ if(flag == PASSIVE) { if(-1 == listen(sfd, 10)) { perror("listen"); close(sfd); exit(1); } struct sockaddr_in peer_addr ; /* 存放返回socket描述符方的联系方式的结构体,是传出参数 */ int len ; memset(&peer_addr, 0, sizeof(peer_addr)); len = sizeof(peer_addr); //accept返回对方socket描述符,并且peer_addr与len均为传出参数 fd_peer = accept(sfd,(struct sockaddr*)&peer_addr,&len); printf("%s:%d begin to talk!\n",inet_ntoa(peer_addr.sin_addr),ntohs(peer_addr.sin_port)); close(sfd);//kris }else if(flag == ACTIVE) /*-------------如果是主动方,connect -------------------------------------------------*/ { fd_peer = sfd ; //如果是主动方实际上本方就是从sfd通信的。 /* 主动方要去连对方,需要知道对方IP,通过命令行参数传;也需要知道服务器端口号,程序已经写死 */ struct sockaddr_in server_addr ; memset(&server_addr, 0, sizeof(server_addr)); server_addr.sin_family = AF_INET ; server_addr.sin_port = htons(SERVER_PORT);/*加了htons,如果右边是12345678,传到左边还是12345678;如果不加htons如果右边是12345678,传到左边就变成78563412了 */ server_addr.sin_addr.s_addr = inet_addr(argv[1]); /* connect连不上对方就返回-1,睡一秒后继续连 */ while( connect(sfd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) { sleep(1); printf("connecting....\n"); } printf("success ! \n"); } pthread_t thd ; pthread_create(&thd, NULL,chat_handler,(void*)fd_peer); char msg[1024]; while(memset(msg, 0, 1024), fgets(msg, 1024, stdin) != NULL) { send(fd_peer, msg, strlen(msg), 0); } close(fd_peer); //用于测试 printf("close close close \n"); pthread_join(thd, NULL); }/* 本代码通信是在fd_peer这个socket端口上 */
程序运行退出结果讨论
1. 如果A按ctrl+D(主线程退出while循环,close端口,阻塞在pthread_join上),那么B此时再发送一个消息,在线程中A recv 到并将其打印,此时A的socket端口真正关闭,recv返回-1,退出循环,退出线程,A结束。B由于A的socket端口关闭,线程中的recv返回0,退出循环,退出线程。此时B如果按下ctrl+D,也会退出程序。
2. 一方按下ctrl+D后,另一方马上也按下ctrl+D。则双方陷入僵局。如果一方按下ctrl+c,强行退出程序后,另一方也会跟着退。原因:一方强退,导致socket端口真正关闭,另一方子线程中的recv返回0,退出循环,退出线程。
3. 一方上来就ctrl+c强退。则对方的子线程会退掉,原因同上。此时另一方按ctrl+D就退出程序了。
小结
recv返回值分以下3种情况
1. 大于0。接收到的数据大小。
2. 等于0。对方的socket端口真正关闭。
3. 小于0。出错。其中一种情况是,本方的socket端口已经真正关闭后,还试图从该端口中获取数据。
recv返回值实际上和read是类似的。
发现
send一个空的消息,如下:
send(fd_peer,"",0,0);
对方的recv不会返回0,仍旧阻塞着。
这点和UDP有所不同,在UDP中,一方sendto一个空消息给另一方,对方的recvfrom是返回0的。
Linux网络编程7——使用TCP实现双方聊天