Linux 系统应用编程——网络编程（高级篇）

最新推荐文章于 2024-10-09 19:49:21 发布

玛丽奥ZJY

最新推荐文章于 2024-10-09 19:49:21 发布

阅读量790

点赞数

分类专栏：【linux 网络编程】

【linux 网络编程】专栏收录该内容

50 篇文章 4 订阅

订阅专栏

一、网络超时检测

在网络通信过程中，经常会出现不可预知的各种情况。例如网络线路突发故障、通信一方异常结束等。一旦出现上述情况，很可能长时间都不会收到数据，而且无法判断是没有数据还是数据无法到达。如果使用的是TCP协议，可以检测出来；但如果使用UDP协议的话，需要在程序中进行相关检测。所以，为避免进程在没有数据时无限制的阻塞，使用网络超时检测很有必要。

1、套接字接收超时检测

这里先介绍设置套接字选项的函数 setsockopt() 函数：

所需头文件	#include <sys/types.h> #include <sys/socket.h>
函数原型	int setsockopt (int sockfd, int level, int optname, const void *optval, socklen_t optlen );
函数参数	sockfd：套接字描述符 level：选项所属协议层 optval：保存选项值的缓冲区 optlen：选项值的长度
函数返回值	成功：0 出错：-1，并设置 errno

下面是套接字常用选项及其说明：

LEVEL：SOL_SOCKET

选项名称	说明	数据类型
SO_BROADCAST	允许发送广播数据	int
SO_DEBUG	允许调试	int
SO_DONTRUOTE	不查找路由	int
SO_ERROR	获得套接字错误	int
SO_KEEPALIVE	保持连接	int
SO_LINGER	延迟关闭连接	struct linger
SO_OOBINLINE	带外数据放入正常数据流	int
SO_RCVBUF	接收缓冲区大小	int
SO_SNDBUF	发送缓冲区大小	int
SO_RCVTIMEO	接收超时	struct timeval
SO_SNDTIMEO	发送超时	struct timeval
SO_REUSERADDR	允许重用本地地址和端口	int
SO_TYPE	获得套接字类型	int

下面利用SO_RCVTIMEO的选项实现套接字的接收超时检测：

[cpp] view plain copy

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define N 64
#define PORT 8888
int main()
{
int sockfd;
char buf[N];
struct sockaddr_in seraddr;
struct timeval t = {6, 0};
if((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
perror("socket error");
exit(-1);
}
else
{
printf("socket successfully!\n");
printf("sockfd:%d\n",sockfd);
}
memset(&seraddr, 0, sizeof(seraddr));
seraddr.sin_family = AF_INET;
seraddr.sin_port = htons(PORT);
seraddr.sin_addr.s_addr = htonl(INADDR_ANY);
if(bind(sockfd, (struct sockaddr *)&seraddr, sizeof(seraddr)) == -1)
{
perror("bind error");
exit(-1);
}
else
{
printf("bind successfully!\n");
printf("PORT:%d\n",PORT);
}
if(setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &t, sizeof(t)) < 0)
{
perror("setsockopt error");
exit(-1);
}
if(recvfrom(sockfd, buf, N, 0, NULL, NULL) < 0)
{
perror("fail to recvfrom");
exit(-1);
}
else
{
printf("recv data: %s\n",buf);
}
return 0;
}

执行结果如下：

[cpp] view plain copy

fs@ubuntu:~/qiang/socket/time$ ./setsockopt
socket successfully!
sockfd:3
bind successfully!
PORT:8888
fail to recvfrom: Resource temporarily unavailable
fs@ubuntu:~/qiang/socket/time$

可以看到，6s之内没有数据包到来，程序会从 recvfrom 函数返回，进行相应的错误处理。

注意：套接字一旦设置了超时之后，每一次发送或接收时都会检测，如果要取消超时检测，重新用setsockopt函数设置即可（把时间值指定为 0）。

2、定时器超时检测

这里利用定时器信号SIGALARM，可以在程序中创建一个闹钟。当到达目标时间后，指定的信号处理函数被执行。这样同样可以利用SIGALARM信号实现检测，下面分别介绍相关数据类型和函数。

struct sigaction 是 Linux 中用来描述信号行为的结构体类型，其定义如下：

[cpp] view plain copy

struct sigaction
{
void (*sa_handler) (int);
void (*sa_sigaction)(int, siginfo_t *, void *);
sigset_t sa_mask;
int sa_flags;
void (*sa_restorer) (void);
}

① sa_handler：此参数和signal()的参数handler相同，此参数主要用来对信号旧的安装函数signal()处理形式的支持；
② sa_sigaction：新的信号安装机制，处理函数被调用的时候，不但可以得到信号编号，而且可以获悉被调用的原因以及产生问题的上下文的相关信息。
③ sa_mask：用来设置在处理该信号时暂时将sa_mask指定的信号搁置；
④ sa_restorer：此参数没有使用；
⑤ sa_flags：用来设置信号处理的其他相关操作，下列的数值可用。可用OR 运算（|）组合：
ŸA_NOCLDSTOP:如果参数signum为SIGCHLD，则当子进程暂停时并不会通知父进程
SA_ONESHOT/SA_RESETHAND:当调用新的信号处理函数前，将此信号处理方式改为系统预设的方式
SA_RESTART:被信号中断的系统调用会自行重启
SA_NOMASK/SA_NODEFER:在处理此信号未结束前不理会此信号的再次到来
SA_SIGINFO：信号处理函数是带有三个参数的sa_sigaction。

所需头文件	#include <signal.h>
函数原型	int sigaction(int signum, const struct sigaction act , struct sigaction oldact );
函数传入值	signum：可以指定SIGKILL和SIGSTOP以外的所有信号 act ：act 是一个结构体，里面包含信号处理函数的地址、处理方式等信息； oldact ：参数oldact 是一个传出参数，sigaction 函数调用成功后， oldact 里面包含以前对 signum 信号的处理方式的信息；
函数返回值	成功：0 出错：-1

使用定时器信号检测超时的示例代码如下：

[cpp] view plain copy

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define N 64
#define PORT 8888
void handler(int signo)
{
printf("interrupted by SIGALRM\n");
}
int main()
{
int sockfd;
char buf[N];
struct sockaddr_in seraddr;
struct sigaction act;
if((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
perror("socket error");
exit(-1);
}
else
{
printf("socket successfully!\n");
printf("sockfd:%d\n",sockfd);
}
memset(&seraddr, 0, sizeof(seraddr));
seraddr.sin_family = AF_INET;
seraddr.sin_port = htons(PORT);
seraddr.sin_addr.s_addr = htonl(INADDR_ANY);
if(bind(sockfd, (struct sockaddr *)&seraddr, sizeof(seraddr)) == -1)
{
perror("bind error");
exit(-1);
}
else
{
printf("bind successfully!\n");
printf("PORT:%d\n",PORT);
}
sigaction(SIGALRM, NULL, &act);
act.sa_handler = handler;
act.sa_flags &= ~SA_RESTART;
sigaction(SIGALRM, &act, NULL);
alarm(6);
if(recvfrom(sockfd, buf, N, 0, NULL, NULL) < 0)
{
perror("fail to recvfrom");
exit(-1);
}
printf("recv data: %s\n",buf);
alarm(0);
return 0;
}

执行结果如下：

[cpp] view plain copy

fs@ubuntu:~/qiang/socket/time$ ./alarm
socket successfully!
sockfd:3
bind successfully!
PORT:8888
interrupted by SIGALRM
fail to recvfrom: Interrupted system call
fs@ubuntu:~/qiang/socket/time$

二、广播

前面的网络通信中，采用的都是单播（唯一的发送方和接收方）的方式。很多时候，需要把数据同时发送给局域网中的所有主机。例如，通过广播ARP包获取目标主机的MAC地址。

1、广播地址

IP地址用来标识网络中的一台主机。IPv4 协议用一个 32 位的无符号数表示网络地址，包括网络号和主机号。子网掩码表示 IP 地址中网络和占几个字节。对于一个Ｃ类地址来说，子网掩码为　255.255.255.0。

每个网段都有其对应的广播地址。以 C 类地址网段 192.168.1.x为例，其中最小的地址 192.168.1.0 代表该网段；而最大的地址192.168.1.255 则是该网段中的广播地址。当我们向这个地址发送数据包时，该网段中所以的主机都会接收并处理。

注意：发送广播包时，目标IP 为广播地址而目标 MAC 是 ff:ff:ff:ff:ff。

2、广播包的发送和接收

广播包的发送和接收通过UDP套接字实现。

1）广播包发送流程如下：

创建udp 套接字

指定目标地址和端口

设置套接字选项允许发送广播包

发送数据包

发送广播包的示例如下：

[cpp] view plain copy

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define N 64
#define PORT 8888
int main()
{
int sockfd;
int on = 1;
char buf[N] = "This is a broadcast package!";
struct sockaddr_in dstaddr;
if((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
perror("socket error");
exit(-1);
}
else
{
printf("socket successfully!\n");
printf("sockfd:%d\n",sockfd);
}
memset(&dstaddr, 0, sizeof(dstaddr));
dstaddr.sin_family = AF_INET;
dstaddr.sin_port = htons(PORT);
dstaddr.sin_addr.s_addr = inet_addr("192.168.1.255"); // 192.168.1.x 网段的广播地址
if(setsockopt(sockfd, SOL_SOCKET, SO_BROADCAST, &on, sizeof(on)) < 0) //套接字默认不允许发送广播包，通过修改 SO_BROADCAST 选项使能
{
perror("setsockopt error");
exit(-1);
}
while(1)
{
sendto(sockfd, buf, N, 0,(struct sockaddr *)&dstaddr, sizeof(dstaddr));
sleep(1);
}
return 0;
}