/* Обход дерева каталогов в MS DOS при помощи смены текущего каталога.
* Аналог ls -R в UNIX. По аналогичному алгоритму работает программа
* find . -print (напишите команду find, используя match())
*/
#define STYLE2
#include <stdio.h>
#include <stdlib.h>
#include <dir.h>
#include <dos.h>
#include <alloc.h> /* для malloc() */
#include <string.h> /* strchr(), strrchr(), strcpy(), ... */
/* прототипы */
char *strend(char *s); char *strdup(const char *s);
void action(int, char **); void main(int, char **);
int listdir(char *); void printdir(int n);
#ifdef STYLE2
void lookdir(char *s, int ac, char **av, register int level);
#else
void lookdir(char *s, int ac, char **av);
#endif
char root[256]; /* имя стартового каталога */
char cwd[256]; /* полное имя текущего каталога */
char *strend(register char *s){ while(*s)s++; return s; }
char *strdup(const char *s){ /* прототип malloc в <stdlib.h> */
char *p = (char *) malloc(strlen(s) + 1);
if(p) strcpy(p, s); return p;
}
stop(){ /* Реакция на control/break */
chdir( root );
/* Это необходимо потому, что MS DOS имеет (в отличие от UNIX)
понятие "текущий каталог" как глобальное для всей системы.
Если мы прервем программу, то окажемся не в том каталоге,
откуда начинали. */
printf( "\nInterrupted by ctrl-break\n");
return 0; /* exit */
}
void main(int argc, char **argv){
/* получить имя текущего каталога */
(void) getcwd(root, sizeof root);
ctrlbrk( stop ); /* установить реакцию на ctrl/break */
#ifndef STYLE2
lookdir( "." /* корень дерева */, argc, argv );
#else
/* для примера: дерево от "\\" а не от "." */
lookdir( "\\", argc, argv, 0 /* начальный уровень */ );
#endif /*STYLE2*/
chdir(root); /* вернуться в исх. каталог */
}
# ifndef STYLE2
void lookdir(char *s, int ac, char **av){
static int level = 0; /* уровень рекурсии */
# else
void lookdir(char *s, int ac, char **av, register int level){
# endif /*STYLE2*/
struct ffblk dblk, *psd = &dblk;
register done;
if( chdir(s) < 0 ){ /* войти в каталог */
printf( "Cannot cd %s\n", s ); return;
} else if (level == 0){ /* верхний уровень */
(void) getcwd(cwd, sizeof cwd);
/* получить полное имя корня поддерева */
}
action(ac, av);
/* искать имена каталогов, удовлетворяющие шаблону "*" */
/* (не в алфавитном порядке !) */
done = findfirst("*.", psd, FA_DIREC);
while( !done ){
if((psd->ff_attrib & FA_DIREC) && psd->ff_name[0] != '.' ){
/* Видим каталог: войти в него! */
char *tail = strend(cwd); char *addplace;
if( tail[-1] == '\\' ){
addplace = tail;
}else{
*tail = '\\'; addplace = tail+1;
}
strcpy(addplace, psd->ff_name);
#ifndef STYLE2
level++; lookdir( psd->ff_name, ac, av ); level--;
#else
lookdir( psd->ff_name, ac, av, level+1 );
#endif
*tail = '\0';
}
/* Искать следующее имя. Информация о точке, где был
* прерван поиск, хранится в dblk */
done = findnext(psd);
}
if( level ) chdir( ".." ); /* выйти вверх */
}
/* Выполнить действия в каталоге */
void action(int ac, char **av){
extern int busy;
busy = 0;
if( ac == 1 ) listdir( "*.*" );
else{
av++;
while( *av ) listdir( *av++ );
}
printdir( busy );
}
#define MAXF 400
struct fst{
char *name; long size; short attr;
} files[MAXF];
int busy; /* сколько имен собрано */
/* Собрать имена, удовлетворяющие шаблону. */
int listdir( char *picture ){
int done, n; struct ffblk dentry;
for(n=0, done=findfirst(picture, &dentry,0xFF /* все типы */);
busy < MAXF && !done ;
done = findnext( &dentry )){
files[busy].name = strdup(dentry.ff_name);
files[busy].size = dentry.ff_fsize;
files[busy].attr = dentry.ff_attrib;
n++; busy++;
}
return n;
}
/* int cmp(struct fst *a, struct fst *b) */
/* новые веяния в Си требуют такого прототипа: */
int cmp(const void *a, const void *b){
return strcmp(((struct fst *) a) -> name,
((struct fst *) b) -> name );
}
/* отсортировать и напечатать */
void printdir(int n){
register i;
struct fst *f;
qsort( files, n, sizeof files[0], cmp );
printf( "Directory %s\n", cwd );
for( i=0, f = files; i < n; i++, f++ )
printf("\t%-16s\t%10ld\t%c%c%c%c%c%c\n",
f->name, f->size,
f->attr & FA_DIREC ? 'd':'-', /* directory */
f->attr & FA_RDONLY ? 'r':'-', /* read only */
f->attr & FA_HIDDEN ? 'h':'-', /* hidden */
f->attr & FA_SYSTEM ? 's':'-', /* system */
f->attr & FA_LABEL ? 'l':'-', /* volume label */
f->attr & FA_ARCH ? 'a':'-' /* archive */
), free(f->name);
putchar('\n');
}
© Copyright А. Богатырев, 1992-95
Си в UNIX
Назад | Содержание | Вперед