read实现按行读取，计算一个文件的行数。

mike_chen

昨天答应了“gooderfeng ”兄晚上回家写一个，但是回去后就跟朋友扯了会蛋，就没来得及写，今天早上匆忙之间就写了一个，方法比较笨，希望各位多提意见，谢谢！

1. 
   
2. #include <unistd.h>
   
3. #include <fcntl.h>
   
4. #include <stdio.h>
   
5. #include <sys/types.h>
   
6. #include <sys/stat.h>
   
7. #include <errno.h>
   
8. #include <string.h>
   
9. 
   
10. #define BUFFER_SIZE 8192
    
11. 
    
12. int main(int argc, char **argv)
    
13. {
    
14.         int  fd;
    
15.         int  bytes_read, enter_end, count;
    
16.         char buffer[2*BUFFER_SIZE], tmp_buf[BUFFER_SIZE+1];
    
17.         char *token;
    
18.         char *delim = "\n";
    
19.         static int n=0;
    
20.        
    
21.         if (argc != 2) {
    
22.                 fprintf(stderr, "Usage:%s filename\n\a", argv[0]);
    
23.                 exit(1);
    
24.         }
    
25. 
    
26.         if ((fd = open(argv[1], O_RDONLY)) == -1) {
    
27.                 fprintf(stderr, "Open %s Error:%s\n", argv[1], strerror(errno));
    
28.                 exit(1);
    
29.         }
    
30. 
    
31.         memset(tmp_buf, 0x00, sizeof(tmp_buf));
    
32.         memset(buffer, 0x00, sizeof(buffer));
    
33.         while (bytes_read = read(fd, tmp_buf, BUFFER_SIZE)) {
    
34.                 enter_end = 0;
    
35.                 count     = 0;
    
36.                 bytes_read = strlen(strcat(buffer, tmp_buf));
    
37.                 if ((bytes_read == -1) && (errno != EINTR)) {
    
38.                         perror("read data");
    
39.                         break;
    
40.                 } else if (bytes_read > 0) {
    
41.                         if (buffer[bytes_read-1] == '\n')
    
42.                                 enter_end = 1;
    
43.                         token = strtok(buffer, delim);
    
44.                         printf("%s\n", token);
    
45.                         count += strlen(token) + 1;
    
46.                         n++;
    
47.                         for ( ; (token = strtok(NULL, delim)) != NULL; n++) {
    
48.                                 count += strlen(token) + 1;
    
49.                                 if (count >= bytes_read && enter_end != 1) {
    
50.                                         strcpy(buffer, token);
    
51.                                         n--;
    
52.                                 } else
    
53.                                         printf("%s\n", token);
    
54.                         }
    
55.                 }
    
56. 
    
57.                 if (enter_end == 1)
    
58.                         memset(buffer, 0x00, sizeof(buffer));
    
59.                 memset(tmp_buf, 0x00, sizeof(tmp_buf));
    
60.         }
    
61.        
    
62.         printf("Total lines: %d\n", n);
    
63.         close(fd);
    
64.        
    
65.         return 0;
    
66. }
    
67. 
    

复制代码

stiandao

对一个文件统计回车换行的个数不行吗

mike_chen

原帖由 stiandao 于 2006-4-19 10:49 发表
对一个文件统计回车换行的个数不行吗

其主要目的是使文件按行输出，有的时候从网络接收一个文件，在接收的同时可能还需要对文件进行分析、分解。

westgarden

必须是用 read吗？是为了在并发情况下的异步信号安全？

哦，现在才看到

其主要目的是使文件按行输出，有的时候从网络接收一个文件，在接收的同时可能还需要对文件进行分析、分解。

[ 本帖最后由 westgarden 于 2006-4-19 16:09 编辑 ]

gooderfeng

谢谢，好好看看

wolf0403

无非就是自己做缓冲而已。fgets 的实现就是将缓冲放在了 FILE * 里，由标准库函数维护。