src/utils/file_utils.c

   1 #define  _BSD_SOURCE
   2
   3 #include <stdio.h>
   4 #include <string.h>
   5 #include <unistd.h>
   6 #include <dirent.h>
   7 #include <errno.h>
   8 #include <fcntl.h>
   9
  10 #include <sys/types.h>
  11 #include <sys/stat.h>
  12
  13 /**
  14  * file_type
  15  *
  16  * @param[in] file      file name
  17  *
  18  * @retval 0 file is "regular", 1 otherwise, -1 on error
  19  */
  20 int file_type(const char *file)
  21 {
  22         int fd;
  23         struct stat fsb;
  24
  25         if (!file)
  26                 return -1;
  27
  28         fd = open(file, O_RDONLY);
  29         if (fd < 0) {
  30                 fprintf(stderr, "%s - fstat %s: %s\n",
  31                                  __func__, file, strerror(errno));
  32                 return -1;
  33         }
  34
  35         if (fstat(fd, &fsb)) {
  36                 fprintf(stderr, "%s - fstat %s: %s\n",
  37                                  __func__, file, strerror(errno));
  38                 close(fd);
  39                 return -1;
  40         }
  41
  42         if (!S_ISREG(fsb.st_mode)) {
  43                 close(fd);
  44                 return 1;
  45         }
  46
  47         close(fd);
  48         return 0;
  49 }
  50
  51 int scan_dir(const char *path, void (*file_callback)(const char *))
  52 {
  53         int rc = 0;
  54         DIR *dir;
  55         struct dirent *de;
  56
  57         if (!path)
  58                 return -1;
  59
  60         dir = opendir(path);
  61         if (!dir) {
  62                 fprintf(stderr, "%s - %s: %s\n", __func__, path, strerror(errno));
  63                 return -1;
  64         }
  65
  66         do {
  67                 de = readdir(dir);
  68                 if (de) {
  69                         file_callback(de->d_name);
  70                         continue;
  71                 }
  72
  73                 /* de is null on error or EOD */
  74                 if (errno) {
  75                         rc = -1;
  76                         fprintf(stderr, "%s - %s!\n", __func__, strerror(errno));
  77                         break;
  78                 }
  79         } while (de);
  80
  81         closedir(dir);
  82
  83         return rc;
  84 }
  85
  86 size_t dir_entries(const char *path)
  87 {
  88         DIR *dir;
  89         struct dirent *de;
  90         size_t count = 0;
  91
  92         if (!path)
  93                 return 0;
  94
  95         dir = opendir(path);
  96         if (!dir) {
  97                 fprintf(stderr, "%s - %s: %s\n", __func__, path, strerror(errno));
  98                 return 0;
  99         }
 100
 101         do {
 102                 de = readdir(dir);
 103                 if (de) {
 104                         if (file_type(de->d_name) == 0)
 105                                 count++;
 106                         continue;
 107                 }
 108
 109                 /* de is null on error or EOD */
 110                 if (errno) {
 111                         fprintf(stderr, "%s - %s!\n", __func__, strerror(errno));
 112                         break;
 113                 }
 114         } while (de);
 115
 116         closedir(dir);
 117
 118         return count;
 119 }
 120