一句话总结:防止阻塞在那里,任务该往下走的时候还要往下走。
1、接收超时,等待N秒,还是没有消息,那就接着往下走
服务端
#include <iostream>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdio.h>
using namespace std;
int main()
{
struct sockaddr_in serAddr;
int listenSock, cnntSock;
listenSock = socket(AF_INET, SOCK_STREAM, 0);
serAddr.sin_family = AF_INET;
serAddr.sin_addr.s_addr = INADDR_ANY;
serAddr.sin_port = htons(1270);
bind(listenSock, (const struct sockaddr *)&serAddr, sizeof(sockaddr_in));
listen(listenSock, 10);
struct timeval tv = {3,0};//3.0s 接收超时设置
setsockopt(listenSock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(struct timeval));
struct sockaddr_in cliAddr;
socklen_t len = sizeof(sockaddr_in);
cnntSock = accept(listenSock, (struct sockaddr*)&cliAddr, &len);
cout<< "IP:" <<inet_ntoa(cliAddr.sin_addr) <<", port:"<<cliAddr.sin_port<<endl;
unsigned int total = 0;
int cnt = 0;
while(1)
{
getchar();
cout <<"read msg..."<<endl;
char buf[6000] = {0};
int ret = recv(cnntSock, buf, sizeof(buf), 0); //wait 3 sec,超过3秒返回-1,继续往下走
total += ret;
cout << "msg:" <<endl;
cout << buf;
cout << "recv msg size:" << ret << ",total size:" << total <<endl;
cnt++;
if (5==cnt)
break;
}
close(cnntSock);
close(listenSock);
return 0;
}
客户端
#include <iostream>
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
using namespace std;
int main()
{
int sockClient = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in addrSrv;
addrSrv.sin_addr.s_addr = inet_addr("127.0.0.1");
addrSrv.sin_family = AF_INET;
addrSrv.sin_port = htons(1270);
struct timeval tv = {3,0};//3s 超时
setsockopt(sockClient, SOL_SOCKET, SO_SNDTIMEO, (char *)&tv, sizeof(struct timeval));
connect(sockClient, ( const struct sockaddr *)&addrSrv, sizeof(struct sockaddr_in));
char buf[10] = {0};
for(unsigned int i = 0; i < 10; i++)
{
buf[i] = 'd';
}
int ret = send(sockClient, buf, sizeof(buf) , 0);
cout << "buf size:" << sizeof(buf) << "ret:" << ret <<endl;
getchar();
for(unsigned int i = 0; i < 10; i++)
{
buf[i] = 'p';
}
ret = send(sockClient, buf, sizeof(buf) , 0);
cout << "buf size:" << sizeof(buf) << "ret:" << ret <<endl;
getchar();
close(sockClient);
return 0;
}
运行结果: