socket信号驱动
为了使一个套接字能够使用信号驱动I/O，至少需要以下3步操作。
- 1.安装SIGIO信号
- 2.套接字的拥有者设定为当前进程。因为SIGIO信号只会送到socket拥有者进程. 通过fcntl的F_SETOWN
- 3.套接字必须被允许使用异步I/O。 通过fcntl的F_SETFL，设置为O_ASYNC
在UDP通信中，下面情况会产生SIGIO信号
在TCP通信中，下面情况会产生SIGIO信号
例子：
下面的代码好奇怪，说是UDP的，但是发送接收用的是send, recv 而且客户端还跟服务器连接了；说是TCP，但是socket建立的时候用的是SOCK_DGRAM.
而且代码是可以跑通的。
服务器：
#include<stdio.h> #include<stdlib.h> #include<errno.h> #include<string.h> #include<sys/types.h> #include<netinet/in.h> #include<netinet/tcp.h> #include<sys/socket.h> #include<sys/wait.h> #include<unistd.h> #include<arpa/inet.h> #include<sys/time.h> #include<netdb.h> #include<fcntl.h> #include<signal.h> #include<sys/ioctl.h> #define MAX_LENTH 1500 //针对SIGIO信号处理 static int nqueue = 0; void sigio_handler(int signum) { if(signum == SIGIO) nqueue++; printf("signum = %d, nqueue = %d ", signum, nqueue); //打印信号值 return; } static recv_buf[MAX_LENTH]; int main(int argc, char *argv[]) { int sockfd, on = 1; struct sigaction action; sigset_t newmask, oldmask; struct sockaddr_in ser_addr; if(argc != 3) { printf("use: %s ip_add port ", argv[0]); exit(EXIT_FAILURE); } memset(&ser_addr, 0, sizeof(ser_addr)); ser_addr.sin_family = AF_INET; //使用IPv4 ser_addr.sin_port = htons(atoi(argv[2])); if(inet_aton(argv[1], (struct in_addr *)&ser_addr.sin_addr.s_addr) == 0) { perror(argv[1]); exit(EXIT_FAILURE); } //创建socket if((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) { perror("socket"); exit(EXIT_FAILURE); } //绑定IP地址 if(bind(sockfd, (struct sockaddr *)&ser_addr, sizeof(ser_addr)) == -1) { perror("bind"); exit(EXIT_FAILURE); } memset(&action, 0, sizeof(action)); action.sa_handler = sigio_handler; action.sa_flags = 0; //安装信号 sigaction(SIGIO, &action, NULL); //设置socket拥有者 if(fcntl(sockfd, F_SETOWN, getpid()) == -1) { perror("fcntl F_SETOWN"); exit(EXIT_FAILURE); } //设置socket为信号驱动型 if(ioctl(sockfd, FIOASYNC, &on) == -1) { perror("ioctl FIOASYNC"); exit(EXIT_FAILURE); } sigemptyset(&oldmask); sigemptyset(&newmask); sigaddset(&newmask, SIGIO); printf("get ready "); while(1) { int len; //设置当前阻塞的信号 sigprocmask(SIG_BLOCK, &newmask, &oldmask); //等待信号 while(nqueue == 0) sigsuspend(&oldmask); memset(recv_buf, ' ', MAX_LENTH); //非阻塞接收数据 len = recv(sockfd, recv_buf, MAX_LENTH, MSG_DONTWAIT); if(len == -1 && errno == EAGAIN) nqueue = 0; //修改进程阻塞的信号 sigprocmask(SIG_SETMASK, &oldmask, NULL); if(len >= 0) printf("recv %d byte, msg is %s ", len, recv_buf); } }
客户端
#include<stdio.h> #include<stdlib.h> #include<errno.h> #include<string.h> #include<sys/types.h> #include<netinet/in.h> #include<netinet/tcp.h> #include<sys/socket.h> #include<sys/wait.h> #include<unistd.h> #include<arpa/inet.h> #include<sys/time.h> #include<netdb.h> #include<fcntl.h> #include<signal.h> #include<sys/ioctl.h> #define MAX_LENTH 1500 int main(int argc, char *argv[]) { struct sockaddr_in addr; int sock_fd, ret; char snd_buf[MAX_LENTH]; if(argc != 3) //参数需要服务器的IP和端口 { printf("use: %s ip_add port ", argv[0]); exit(EXIT_FAILURE); } memset(&addr, 0, sizeof(addr)); addr.sin_family = AF_INET; if(inet_aton(argv[1], (struct in_addr *)&addr.sin_addr.s_addr) == 0) { perror(argv[1]); exit(EXIT_FAILURE); } addr.sin_port = htons(atoi(argv[2])); //创建socket if((sock_fd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) { perror("socket"); exit(EXIT_FAILURE); } //向服务器发起连接 ？？这不是TCP的么 if(ret = connect(sock_fd, (struct sockaddr *)&addr, sizeof(addr)) == -1) { perror("connect"); exit(EXIT_FAILURE); } while(1) { printf("input msg to send:"); memset(snd_buf, ' ', MAX_LENTH); fgets(snd_buf, MAX_LENTH - 1, stdin); write(sock_fd, snd_buf, MAX_LENTH - 1); } }
