UNIX環境高級編程學習之第十章信號-用信號和非局部轉移函數寫非阻塞的IO函數

UNIX環境高級編程學習之第十章信號-用信號和非局部轉移函數寫非阻塞的IO函數

 

/* File: NonblockingIO.c Desc: 用信號和非局部轉移函數寫非阻塞的IO函數 */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include <errno.h> #include <signal.h> #include <sys/types.h> #include <unistd.h> #include <setjmp.h> #define MAXBUFSIZE 1024 #define READ_ERROR -1 #define READ_TIMEOUT -2 static sigjmp_buf jmpbuf; /* 處理計時器超時信號 */ static void sig_alrm(int signo) { siglongjmp(jmpbuf, 1); // 信號處理中專用的非局部轉移函數， 功能: 轉移到sigsetjmp()位置 } int my_read(char* buf, const int len, const int time) { signal(SIGALRM, sig_alrm); int ret, readLen; ret = sigsetjmp(jmpbuf, 1); // 信號處理中專用的非局部轉移函數， 功能: 設置轉換點 if (ret == 0) { alarm(time); // 設置計時器 if (( readLen = read(STDIN_FILENO, buf, len)) < 0) // 阻塞的IO { alarm(0); // 清楚計時器 return READ_ERROR; } alarm(0); return readLen; } return READ_TIMEOUT; } int main(int argc, char* argv[]) { char buf[MAXBUFSIZE]; int ret; while (1) { memset(buf, 0x00, sizeof(buf)); ret = my_read(buf, sizeof(buf), 5); if (ret == READ_ERROR) { printf("my_read Error ! /n"); exit(-1); }else if (ret == READ_TIMEOUT) { printf("my_read Timeout!/n"); }else{ printf("my_read data:%s", buf); } } return 0; }