三种网络超时检测

三种网络超时检测
1:setsockopt

服务器：

#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netinet/ip.h>
#include<strings.h>
#include<string.h>
#include<arpa/inet.h>
#include<sys/time.h>

static int connfd;

int main(int argc, char **argv)
{
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);

int on = 1;
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof on);

struct sockaddr_in my_addr, cli_addr;
bzero(&my_addr, sizeof my_addr);
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(atoi(argv[1]));
my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr));

listen(sockfd, 4);

socklen_t len;
len = sizeof(cli_addr);
bzero(&cli_addr, len);
connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &len);

printf("new connection: %s:%hu\n",inet_ntoa(cli_addr.sin_addr),ntohs(cli_addr.sin_port));

struct timeval tv;
tv.tv_sec = 5;
tv.tv_usec = 0;
setsockopt(connfd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
char buf[1024];
int n;
while(1){
bzero(buf, 1024);
if((n = read(connfd, buf, 1023)) <= 0)
break;
write(connfd, buf, strlen(buf));
printf("srv: %s\n",buf);
}
return 0;
}

客服端：

#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netinet/ip.h>
#include<strings.h>
#include<string.h>

int main(int argc, char **argv)
{
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);

struct sockaddr_in src_addr;
bzero(&src_addr, sizeof src_addr);
src_addr.sin_family = AF_INET;
inet_pton(AF_INET, argv[1],(struct sockaddr *)&src_addr.sin_addr);
src_addr.sin_port = htons(atoi(argv[2]));

connect(sockfd, (struct sockaddr *)&src_addr, sizeof src_addr);

char buf[1024];
char b[1024];
while(1){
bzero(b, 1024);
fgets(buf, 1024, stdin);
write(sockfd, buf, strlen(buf));
read(sockfd, b ,1023);
printf("b: %s\n",b);
}
return 0;
}

2:select

服务器：

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

static int connfd;

int main(int argc, char **argv)
{
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);

int on = 1;
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof on);

struct sockaddr_in my_addr, cli_addr;
bzero(&my_addr, sizeof my_addr);
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(atoi(argv[1]));
my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr));

listen(sockfd, 4);

socklen_t len;
len = sizeof(cli_addr);
bzero(&cli_addr, len);
connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &len);

printf("new connection: %s:%hu\n",inet_ntoa(cli_addr.sin_addr),ntohs(cli_addr.sin_port));

char buf[1024];
int n;
fd_set ret;
struct timeval tv = {5, 0};
while(1){
FD_ZERO(&ret);
FD_SET(connfd, &ret);

int re;
re = select(connfd+1, &ret, NULL, NULL, &tv);

if(re == 0)
{
printf("time out.");
exit(0);

}
if(FD_ISSET(connfd, &ret))
{

bzero(buf, 1024);
if((n = read(connfd, buf, 1023)) <= 0)
break;
write(connfd, buf, strlen(buf));
printf("srv: %s\n",buf);

}
}
return 0;
}

客服端：

#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netinet/ip.h>
#include<strings.h>
#include<string.h>

int main(int argc, char **argv)
{
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);

struct sockaddr_in src_addr;
bzero(&src_addr, sizeof src_addr);
src_addr.sin_family = AF_INET;
inet_pton(AF_INET, argv[1],(struct sockaddr *)&src_addr.sin_addr);
src_addr.sin_port = htons(atoi(argv[2]));

connect(sockfd, (struct sockaddr *)&src_addr, sizeof src_addr);

char buf[1024];
char b[1024];
while(1){
bzero(b, 1024);
fgets(buf, 1024, stdin);
write(sockfd, buf, strlen(buf));
read(sockfd, b ,1023);
printf("b: %s\n",b);
}
return 0;
}

3：设置定时器

服务器：

#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netinet/ip.h>
#include<strings.h>
#include<string.h>
#include<arpa/inet.h>
#include<signal.h>

static int connfd;

void f(int reg)
{
printf("time out.");
exit(0);

}

int main(int argc, char **argv)
{
signal(SIGALRM,f);
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);

int on = 1;
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof on);

struct sockaddr_in my_addr, cli_addr;
bzero(&my_addr, sizeof my_addr);
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(atoi(argv[1]));
my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr));

listen(sockfd, 4);

socklen_t len;
len = sizeof(cli_addr);
bzero(&cli_addr, len);
connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &len);

printf("new connection: %s:%hu\n",inet_ntoa(cli_addr.sin_addr),ntohs(cli_addr.sin_port));

char buf[1024];
int n;
while(1){
alarm(5);
bzero(buf, 1024);
if((n = read(connfd, buf, 1023)) <= 0)
break;
write(connfd, buf, strlen(buf));
printf("srv: %s\n",buf);
}
return 0;
}

客服端：

#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netinet/ip.h>
#include<strings.h>
#include<string.h>

int main(int argc, char **argv)
{
int sockfd;
sockfd = socket(AF_INET, SOCK_STREAM, 0);

struct sockaddr_in src_addr;
bzero(&src_addr, sizeof src_addr);
src_addr.sin_family = AF_INET;
inet_pton(AF_INET, argv[1],(struct sockaddr *)&src_addr.sin_addr);
src_addr.sin_port = htons(atoi(argv[2]));

connect(sockfd, (struct sockaddr *)&src_addr, sizeof src_addr);

char buf[1024];
char b[1024];
while(1){
bzero(b, 1024);
fgets(buf, 1024, stdin);
write(sockfd, buf, strlen(buf));
read(sockfd, b ,1023);
printf("b: %s\n",b);
}
return 0;
}