这类要求最关键的地方在于:如果要使程序更加有用,最好不要对文件集合的大小加以限制。因此,类似于文件指针一类的变量,如果采取本例的如 FILE *p[MAX],尽管MAX的值是一万,也就是说允许在一万个文件内查找模式匹配——这在大多数情况下都能够应付,但仍然不尽完美。
因此,可以改良的地方在于,数组dir[MAX]存放输入的文件名(有可能是命令行参数也可能从stdin端输入)应该替换成一个不断申请储存空间的指针的指针,包括文件指针也应该如此,这样,可以查找的文件将只受制于机器。
这个版本是简单地将文件数控制在10000个以内:
#include<stdio.h>
#include<string.h>
#include<ctype.h>
#include<stdlib.h>
#define MAX 10000
int comfile(FILE *, char*);
/* 编写函数,从一个命名文件的集合中读取输入(有文件名参数时),否则从标准输入中读取输入,-x表示打印非匹配文件,--后是匹配模式 */
main(int argc, char *argv[]) {
FILE *p[MAX]; /* 指向文件的指针数组 */
int flag,except=0,nfound=0,i=0,k=0;
char point[MAX],*mode=NULL,*dir[MAX],*prog; /* dir字符数组用来保存文件集合中各个文件名 */
prog=argv[0]; /* 保留程序自身文件名 */
if(argc!=1) {
while(--argc>0) { /* 如果从命令行输入 */
if((*++argv)[0]=='-') {
flag=(*argv)[1];
switch(flag) {
case 'x':
except=1;
break;
case '-':
mode=(*argv)+2;
break;
default :
fprintf(stderr,"%s: usage: find -x or --mode pattern\n",prog);
exit(1);
}
} else if(strstr(*argv,":\\")) /* 是文件名 */
dir[i++]=*argv;
else {
fprintf(stderr,"%s: unknown command\n",prog);
exit(2);
}
}
} else {
while(gets(point)!=NULL) { /* 从stdin端输入 */
if(point[0]=='-'&&point[1]=='-')
mode=point+2;
else
dir[i++]=point;
}
}
if(mode==NULL) { /* 没有模式 */
fprintf(stderr,"%s: please enter the mode \n",prog);
exit(3);
}
if(i==0) { /* 没有文件 */
fprintf(stderr,"%s: no file to found\n",prog);
exit(4);
}
while(--i>=0) {
p[k]=fopen(dir[i],"r+"); /* 打开文件 */
if(p[k]==NULL)
fprintf(stderr,"%s: can't open file %s\n",prog,dir[i]);
else {
if((comfile(p[k],mode)!=0)!=except) /* 查找是否存在匹配的行 */
printf("%s\n",dir[i]);
}
++k;
}
return 0;
}
int comfile(FILE *p, char *s) {
char m[MAX];
while(fgets(m,MAX,p)!=NULL)
if (strstr(m,s))
return 1; /* 匹配返回1,不匹配返回0,出错返回EOF */
return ferror(p)?EOF: 0;
}