initial git commit
[paraslash.git] / db.c
1 /*
2  * Copyright (C) 2005-2006 Andre Noll <maan@systemlinux.org>
3  *
4  *     This program is free software; you can redistribute it and/or modify
5  *     it under the terms of the GNU General Public License as published by
6  *     the Free Software Foundation; either version 2 of the License, or
7  *     (at your option) any later version.
8  *
9  *     This program is distributed in the hope that it will be useful,
10  *     but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  *     GNU General Public License for more details.
13  *
14  *     You should have received a copy of the GNU General Public License
15  *     along with this program; if not, write to the Free Software
16  *     Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
17  */
18
19
20 /** \file db.c functions common to all database tools. */
21
22 #include "server.cmdline.h"
23 #include "server.h"
24 #include "afs.h"
25 #include <dirent.h> /* readdir() */
26 #include <sys/stat.h> /* stat */
27 #include <sys/types.h> /* mode_t */
28 #include "error.h"
29 #include "string.h"
30
31 /*
32  * return 1 if name matches any supported audio format
33  */
34 static int match_audio_file_name(char *name)
35 {
36         int i, len = strlen(name);
37         const char *pattern[] = {SUPPORTED_AUDIO_FORMATS_ARRAY};
38
39         for (i = 0; pattern[i]; i++) {
40                 const char *p = pattern[i];
41                 int plen = strlen(p);
42                 if (len < plen + 1)
43                         continue;
44                 if (name[len - plen - 1] != '.')
45                         continue;
46                 if (strcasecmp(name + len - plen, p))
47                         continue;
48                 return 1;
49         }
50         return 0;
51 }
52
53 /**
54  * traverse the given directory recursively
55  *
56  * @param dirname the directory to traverse
57  * @param f: the function to call for each entry.
58  *
59  * for each regular file whose filename ends in .yyy, where yyy is a supported
60  * audio format, the supplied function \a f is called.  The directory and
61  * filename component of the regular file are passed to \a f.
62  *
63  * \return On success, 1 is returned. Otherwise, this function returns a
64  * negative value which indicates the kind of the error.
65  */
66 int find_audio_files(const char *dirname, int (*f)(const char *, const char *))
67 {
68         DIR *dir = NULL;
69         struct dirent *entry;
70         /*
71          * Opening the current directory (".") and calling fchdir() to return
72          * is usually faster and more reliable than saving cwd in some buffer
73          * and calling chdir() afterwards (see man 3 getcwd).
74          */
75         char cwd_fd = open(".", O_RDONLY);
76         struct stat s;
77         int ret = -1;
78
79 //      PARA_DEBUG_LOG("dirname: %s\n", dirname);
80         if (cwd_fd < 0)
81                 return -E_GETCWD;
82         ret = -E_CHDIR;
83         if (chdir(dirname) < 0)
84                 goto out;
85         ret = -E_OPENDIR;
86         dir = opendir(".");
87         if (!dir)
88                 goto out;
89         /* scan cwd recursively */
90         while ((entry = readdir(dir))) {
91                 mode_t m;
92                 char *tmp;
93
94                 if (!strcmp(entry->d_name, "."))
95                         continue;
96                 if (!strcmp(entry->d_name, ".."))
97                         continue;
98                 ret = -E_LSTAT;
99                 if (lstat(entry->d_name, &s) == -1)
100                         goto out;
101                 m = s.st_mode;
102                 if (!S_ISREG(m) && !S_ISDIR(m)) /* skip links, sockets, ... */
103                         continue;
104                 if (S_ISREG(m)) { /* regular file */
105                         if (!match_audio_file_name(entry->d_name))
106                                 continue;
107                         if (f(dirname, entry->d_name) < 0)
108                                 goto out;
109                         continue;
110                 }
111                 /* directory */
112                 tmp = make_message("%s/%s", dirname, entry->d_name);
113                 ret = find_audio_files(tmp, f);
114                 free(tmp);
115                 if (ret < 0)
116                         goto out;
117         }
118         ret = 1;
119 out:
120         if (dir)
121                 closedir(dir);
122         if (fchdir(cwd_fd) < 0)
123                 ret = -E_CHDIR;
124         close(cwd_fd);
125         if (ret < 0)
126                 PARA_ERROR_LOG("ret = %x\n", -ret);
127         return ret;
128 }