[질문] ls를 흉내내고 싶은데..

재귀 함수를 사용하면 간단할 듯 ^^;

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <dirent.h>
#include <sys/types.h>
#include <sys/stat.h>

#define DIR_SEPARATOR	"/"

/* 
 * Callback prototype for dir_traverse.
 *
 * If the callback returns -1, dir_traverse stop traversing, 
 * return immediately.  If the callback returns 0, dir_traverse stop calling
 * itself (recursion) for that directory. If the callback received 
 * non-directory filename, return value 0 is the same as return value 1.
 * If the callback returns 1, dir_traverse works normally.
 *
 * Note that callbacks never receive "." and ".." as its argument.
 */
typedef int (*visit_t)(const char *pathname);

int dir_traverse(const char *pathname, visit_t func);
int is_dir(const char *pathname);

int print_name(const char *pathname);

int
main(int argc, char *argv[])
{
  if (argc != 2) {
    fprintf(stderr, "usage: %s DIR\n", argv[0]);
    return 1;
  }

  dir_traverse(argv[1], print_name);
  return 0;
}

int
print_name(const char *pathname)
{
  printf("%s\n", pathname);
  return 0;
}

int
dir_traverse(const char *pathname, visit_t func)
{
  char *p;
  int len, ret;
  struct dirent *ent;
  DIR *dir;

  p = malloc(pathconf(pathname, _PC_PATH_MAX) + 1);
  if (!p)
    return -1;

  strcpy(p, pathname);
  len = strlen(p);
  if (p[len - 1] != DIR_SEPARATOR[0]) {
    strcpy(p + len, DIR_SEPARATOR);
    len++;
  }

  dir = opendir(pathname);
  if (!dir) {
    free(p);
    return -1;
  }
  while ((ent = readdir(dir)) != NULL) {
    if (strcmp(ent->d_name, ".") == 0 ||
        strcmp(ent->d_name, "..") == 0)
      continue;
    strcpy(p + len, ent->d_name);

    ret = func(p);

    if (ret > 0 && is_dir(p))
      dir_traverse(p, func);
  }
  closedir(dir);
  free(p);
  return 0;
}

int
is_dir(const char *pathname)
{
  struct stat statbuf;
  
  if (stat(pathname, &statbuf) == -1)
    return 0;
  return S_ISDIR(statbuf.st_mode);
}

(주석을 달았습니다만) 버그는 아닙니다.

다만 print_name이 1을 리턴하도록 예제를 만드는게 좋을 거 같군요.

print_name이 -1을 리턴하면 dir_traverse 즉시 종료,
0을 리턴하면 그 때 인자로 들어왔던 directory에 대해 recursion 안함,
1을 리턴하면 계속 recursive하게 탐색.

하게 했던 것이죠. :wink:

속도가 걱정이 되시는 건지? 아니면 뭔가 다른 문제가 있나요?