1 #include "test-tool.h" 2 #include "git-compat-util.h" 3 #include "strbuf.h" 4 #include "iterator.h" 5 #include "dir-iterator.h" 6 7 static const char *error_name(int error_number) 8 { 9 switch (error_number) { 10 case ENOENT: return "ENOENT"; 11 case ENOTDIR: return "ENOTDIR"; 12 default: return "ESOMETHINGELSE"; 13 } 14 } 15 16 /* 17 * usage: 18 * tool-test dir-iterator [--follow-symlinks] [--pedantic] directory_path 19 */ 20 int cmd__dir_iterator(int argc, const char **argv) 21 { 22 struct dir_iterator *diter; 23 unsigned int flags = 0; 24 int iter_status; 25 26 for (++argv, --argc; *argv && starts_with(*argv, "--"); ++argv, --argc) { 27 if (strcmp(*argv, "--follow-symlinks") == 0) 28 flags |= DIR_ITERATOR_FOLLOW_SYMLINKS; 29 else if (strcmp(*argv, "--pedantic") == 0) 30 flags |= DIR_ITERATOR_PEDANTIC; 31 else 32 die("invalid option '%s'", *argv); 33 } 34 35 if (!*argv || argc != 1) 36 die("dir-iterator needs exactly one non-option argument"); 37 38 diter = dir_iterator_begin(*argv, flags); 39 40 if (!diter) { 41 printf("dir_iterator_begin failure: %s\n", error_name(errno)); 42 exit(EXIT_FAILURE); 43 } 44 45 while ((iter_status = dir_iterator_advance(diter)) == ITER_OK) { 46 if (S_ISDIR(diter->st.st_mode)) 47 printf("[d] "); 48 else if (S_ISREG(diter->st.st_mode)) 49 printf("[f] "); 50 else if (S_ISLNK(diter->st.st_mode)) 51 printf("[s] "); socket_connect(const char * sockpath)52 else 53 printf("[?] "); 54 55 printf("(%s) [%s] %s\n", diter->relative_path, diter->basename, 56 diter->path.buf); 57 } 58 59 if (iter_status != ITER_DONE) { 60 printf("dir_iterator_advance failure\n"); 61 return 1; 62 } 63 64 return 0; 65 } 66