《APUE》:将文件复制到分页程序


《Unix环境高级编程》这本书附带了许多短小精美的小程序,我在阅读此书的时候,将书上的代码按照自己的理解重写了一遍(大部分是抄书上的),加深一下自己的理解(纯看书太困了,呵呵)。此例子在Ubuntu10.04上测试通过。

程序简介:这个程序的功能的是每次显示每次一页产生已产生的输出,已经有很多UNIX系统的应用程序具有分页的功能,因此无需再构造一个新的分页程序,只需要调用用户最喜欢的分页程序就可以了(为了演示底层原理,本程序故意不使用popen和pclose函数)。

相关链接

  • 《UNIX环境高级编程》(第二版)apue.h的错误
  • Unix环境高级编程 源代码地址
  1. //《APUE》程序15-2:将文件复制到分页程序  
  2. #include <stdio.h>  
  3. #include <stdlib.h>  
  4. #include <unistd.h>  
  5. #include <fcntl.h>  
  6. #include <errno.h>  
  7. #include <signal.h>  
  8. #include <string.h>  
  9. #include <sys/stat.h>   
  10. #include <stropts.h>  
  11.  
  12. //默认的more路径  
  13. #define DEF_PAGER   "/bin/more"  
  14. //每次读取文件字符的长度  
  15. #define MAXLINE 4096  
  16.  
  17. //输出错误信息并退出     
  18. void error_quit(const char *str)     
  19. {     
  20.     fprintf(stderr, "%s\n", str);     
  21.     exit(1);     
  22.  
  23. int main(int argc, char **argv) 
  24.     int n, terrno;; 
  25.     int fd[2]; 
  26.     pid_t pid; 
  27.     char *paper, *argv0; 
  28.     char line[MAXLINE]; 
  29.     FILE *fp; 
  30.  
  31.     if( argc != 2 ) 
  32.         error_quit("usage: temp <pathname>"); 
  33.  
  34.     //首先打开目标文件  
  35.     fp = fopen(argv[1], "r"); 
  36.     if( NULL == fp ) 
  37.         error_quit("can't open the file"); 
  38.     //创建一个管理  
  39.     if( pipe(fd) < 0 ) 
  40.         error_quit("pipe error"); 
  41.  
  42.     //创建子进程  
  43.     pid = fork(); 
  44.     if( pid < 0 ) 
  45.         error_quit("fork error"); 
  46.     else if( pid > 0 ) 
  47.     { 
  48.         //父进程关闭管道的读端  
  49.         close(fd[0]); 
  50.          
  51.         while( fgets(line, MAXLINE, fp) != NULL ) 
  52.         { 
  53.             n = strlen(line); 
  54.             //将目标文件的内容写入到管道中  
  55.             if( write(fd[1], line, n) != n ) 
  56.                 error_quit("write error to pipe"); 
  57.         } 
  58.  
  59.         if( ferror(fp) ) 
  60.             error_quit("fgets error"); 
  61.         //读取结束,关闭文件  
  62.         close(fd[1]); 
  63.  
  64.         //等待子进程结束后,父进程才能结束  
  65.         if( waitpid(pid, NULL, 0) < 0 ) 
  66.             error_quit("waitpid error"); 
  67.         exit(0); 
  68.     } 
  69.     else 
  70.     { 
  71.         //子进程关闭管道的写端  
  72.         close(fd[1]); 
  73.         //将管道的读端设置为标准输入  
  74.         if( fd[0] != STDIN_FILENO ) 
  75.         { 
  76.             if( dup2(fd[0], STDIN_FILENO) != STDIN_FILENO ) 
  77.                 error_quit("dup2 error to stdin"); 
  78.             close(fd[0]); 
  79.         } 
  80.  
  81.         //得到分页程序的名字,一般是more  
  82.         paper = getenv("PAGER"); 
  83.         if( NULL == paper ) 
  84.             paper = DEF_PAGER; 
  85.         argv0 = strrchr(paper, '/'); 
  86.         if( argv0 != NULL ) 
  87.             argv0++; 
  88.         else 
  89.             argv0 = paper; 
  90.  
  91.         //将进程替换为分页进程  
  92.         if( execl(paper, argv0, (char*)0 ) < 0 ) 
  93.             error_quit("execl error"); 
  94.     } 
  95.     return 0; 

运行示例(红色字体的为输入):

www.bkjia.com @ubuntu:~/code$ gcc temp.c -o temp

#将test.txt的内容设置成man gcc的输出
www.bkjia.com @ubuntu:~/code$ man gcc > test.txt
www.bkjia.com @ubuntu:~/code$ ./temp test.txt

#接下来的输出就和man gcc一模一样了

相关内容