From a654f146d1b39a363cb10853a7670ef099570e54 Mon Sep 17 00:00:00 2001 From: Max Kellermann Date: Wed, 2 Jan 2013 19:22:15 +0100 Subject: update: convert to C++ --- src/UpdateWalk.cxx | 511 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 511 insertions(+) create mode 100644 src/UpdateWalk.cxx (limited to 'src/UpdateWalk.cxx') diff --git a/src/UpdateWalk.cxx b/src/UpdateWalk.cxx new file mode 100644 index 000000000..a8ca1e4ed --- /dev/null +++ b/src/UpdateWalk.cxx @@ -0,0 +1,511 @@ +/* + * Copyright (C) 2003-2013 The Music Player Daemon Project + * http://www.musicpd.org + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include "config.h" /* must be first for large file support */ +#include "UpdateWalk.hxx" +#include "UpdateIO.hxx" +#include "UpdateDatabase.hxx" +#include "UpdateSong.hxx" +#include "UpdateArchive.hxx" +#include "db_lock.h" +#include "directory.h" +#include "song.h" + +extern "C" { +#include "exclude.h" +#include "database.h" +#include "playlist_vector.h" +#include "uri.h" +#include "mapper.h" +#include "path.h" +#include "playlist_list.h" +#include "conf.h" +} + +#include + +#include +#include +#include +#include +#include +#include +#include +#include + +#undef G_LOG_DOMAIN +#define G_LOG_DOMAIN "update" + +bool walk_discard; +bool modified; + +#ifndef WIN32 + +enum { + DEFAULT_FOLLOW_INSIDE_SYMLINKS = true, + DEFAULT_FOLLOW_OUTSIDE_SYMLINKS = true, +}; + +static bool follow_inside_symlinks; +static bool follow_outside_symlinks; + +#endif + +void +update_walk_global_init(void) +{ +#ifndef WIN32 + follow_inside_symlinks = + config_get_bool(CONF_FOLLOW_INSIDE_SYMLINKS, + DEFAULT_FOLLOW_INSIDE_SYMLINKS); + + follow_outside_symlinks = + config_get_bool(CONF_FOLLOW_OUTSIDE_SYMLINKS, + DEFAULT_FOLLOW_OUTSIDE_SYMLINKS); +#endif +} + +void +update_walk_global_finish(void) +{ +} + +static void +directory_set_stat(struct directory *dir, const struct stat *st) +{ + dir->inode = st->st_ino; + dir->device = st->st_dev; + dir->have_stat = true; +} + +static void +remove_excluded_from_directory(struct directory *directory, + GSList *exclude_list) +{ + db_lock(); + + struct directory *child, *n; + directory_for_each_child_safe(child, n, directory) { + char *name_fs = utf8_to_fs_charset(directory_get_name(child)); + + if (exclude_list_check(exclude_list, name_fs)) { + delete_directory(child); + modified = true; + } + + g_free(name_fs); + } + + struct song *song, *ns; + directory_for_each_song_safe(song, ns, directory) { + assert(song->parent == directory); + + char *name_fs = utf8_to_fs_charset(song->uri); + if (exclude_list_check(exclude_list, name_fs)) { + delete_song(directory, song); + modified = true; + } + + g_free(name_fs); + } + + db_unlock(); +} + +static void +purge_deleted_from_directory(struct directory *directory) +{ + struct directory *child, *n; + directory_for_each_child_safe(child, n, directory) { + if (directory_exists(child)) + continue; + + db_lock(); + delete_directory(child); + db_unlock(); + + modified = true; + } + + struct song *song, *ns; + directory_for_each_song_safe(song, ns, directory) { + char *path; + struct stat st; + if ((path = map_song_fs(song)) == NULL || + stat(path, &st) < 0 || !S_ISREG(st.st_mode)) { + db_lock(); + delete_song(directory, song); + db_unlock(); + + modified = true; + } + + g_free(path); + } + + struct playlist_metadata *pm, *np; + directory_for_each_playlist_safe(pm, np, directory) { + if (!directory_child_is_regular(directory, pm->name)) { + db_lock(); + playlist_vector_remove(&directory->playlists, pm->name); + db_unlock(); + } + } +} + +#ifndef G_OS_WIN32 +static int +update_directory_stat(struct directory *directory) +{ + struct stat st; + if (stat_directory(directory, &st) < 0) + return -1; + + directory_set_stat(directory, &st); + return 0; +} +#endif + +static int +find_inode_ancestor(struct directory *parent, ino_t inode, dev_t device) +{ +#ifndef G_OS_WIN32 + while (parent) { + if (!parent->have_stat && update_directory_stat(parent) < 0) + return -1; + + if (parent->inode == inode && parent->device == device) { + g_debug("recursive directory found"); + return 1; + } + + parent = parent->parent; + } +#else + (void)parent; + (void)inode; + (void)device; +#endif + + return 0; +} + +static bool +update_playlist_file2(struct directory *directory, + const char *name, const char *suffix, + const struct stat *st) +{ + if (!playlist_suffix_supported(suffix)) + return false; + + db_lock(); + if (playlist_vector_update_or_add(&directory->playlists, name, + st->st_mtime)) + modified = true; + db_unlock(); + return true; +} + +static bool +update_regular_file(struct directory *directory, + const char *name, const struct stat *st) +{ + const char *suffix = uri_get_suffix(name); + if (suffix == NULL) + return false; + + return update_song_file(directory, name, suffix, st) || + update_archive_file(directory, name, suffix, st) || + update_playlist_file2(directory, name, suffix, st); +} + +static bool +update_directory(struct directory *directory, const struct stat *st); + +static void +update_directory_child(struct directory *directory, + const char *name, const struct stat *st) +{ + assert(strchr(name, '/') == NULL); + + if (S_ISREG(st->st_mode)) { + update_regular_file(directory, name, st); + } else if (S_ISDIR(st->st_mode)) { + if (find_inode_ancestor(directory, st->st_ino, st->st_dev)) + return; + + db_lock(); + struct directory *subdir = + directory_make_child(directory, name); + db_unlock(); + + assert(directory == subdir->parent); + + if (!update_directory(subdir, st)) { + db_lock(); + delete_directory(subdir); + db_unlock(); + } + } else { + g_debug("update: %s is not a directory, archive or music", name); + } +} + +/* we don't look at "." / ".." nor files with newlines in their name */ +G_GNUC_PURE +static bool skip_path(const char *path) +{ + return (path[0] == '.' && path[1] == 0) || + (path[0] == '.' && path[1] == '.' && path[2] == 0) || + strchr(path, '\n') != NULL; +} + +G_GNUC_PURE +static bool +skip_symlink(const struct directory *directory, const char *utf8_name) +{ +#ifndef WIN32 + char *path_fs = map_directory_child_fs(directory, utf8_name); + if (path_fs == NULL) + return true; + + char buffer[MPD_PATH_MAX]; + ssize_t length = readlink(path_fs, buffer, sizeof(buffer)); + g_free(path_fs); + if (length < 0) + /* don't skip if this is not a symlink */ + return errno != EINVAL; + + if ((size_t)length >= sizeof(buffer)) + /* skip symlinks when the buffer is too small for the + link target */ + return true; + + /* null-terminate the buffer, because readlink() will not */ + buffer[length] = 0; + + if (!follow_inside_symlinks && !follow_outside_symlinks) { + /* ignore all symlinks */ + return true; + } else if (follow_inside_symlinks && follow_outside_symlinks) { + /* consider all symlinks */ + return false; + } + + if (g_path_is_absolute(buffer)) { + /* if the symlink points to an absolute path, see if + that path is inside the music directory */ + const char *relative = map_to_relative_path(buffer); + return relative > buffer + ? !follow_inside_symlinks + : !follow_outside_symlinks; + } + + const char *p = buffer; + while (*p == '.') { + if (p[1] == '.' && G_IS_DIR_SEPARATOR(p[2])) { + /* "../" moves to parent directory */ + directory = directory->parent; + if (directory == NULL) { + /* we have moved outside the music + directory - skip this symlink + if such symlinks are not allowed */ + return !follow_outside_symlinks; + } + p += 3; + } else if (G_IS_DIR_SEPARATOR(p[1])) + /* eliminate "./" */ + p += 2; + else + break; + } + + /* we are still in the music directory, so this symlink points + to a song which is already in the database - skip according + to the follow_inside_symlinks param*/ + return !follow_inside_symlinks; +#else + /* no symlink checking on WIN32 */ + + (void)directory; + (void)utf8_name; + + return false; +#endif +} + +static bool +update_directory(struct directory *directory, const struct stat *st) +{ + assert(S_ISDIR(st->st_mode)); + + directory_set_stat(directory, st); + + char *path_fs = map_directory_fs(directory); + if (path_fs == NULL) + return false; + + DIR *dir = opendir(path_fs); + if (!dir) { + g_warning("Failed to open directory %s: %s", + path_fs, g_strerror(errno)); + g_free(path_fs); + return false; + } + + char *exclude_path_fs = g_build_filename(path_fs, ".mpdignore", NULL); + GSList *exclude_list = exclude_list_load(exclude_path_fs); + g_free(exclude_path_fs); + + g_free(path_fs); + + if (exclude_list != NULL) + remove_excluded_from_directory(directory, exclude_list); + + purge_deleted_from_directory(directory); + + struct dirent *ent; + while ((ent = readdir(dir))) { + char *utf8; + struct stat st2; + + if (skip_path(ent->d_name) || + exclude_list_check(exclude_list, ent->d_name)) + continue; + + utf8 = fs_charset_to_utf8(ent->d_name); + if (utf8 == NULL) + continue; + + if (skip_symlink(directory, utf8)) { + modified |= delete_name_in(directory, utf8); + g_free(utf8); + continue; + } + + if (stat_directory_child(directory, utf8, &st2) == 0) + update_directory_child(directory, utf8, &st2); + else + modified |= delete_name_in(directory, utf8); + + g_free(utf8); + } + + exclude_list_free(exclude_list); + + closedir(dir); + + directory->mtime = st->st_mtime; + + return true; +} + +static struct directory * +directory_make_child_checked(struct directory *parent, const char *name_utf8) +{ + db_lock(); + struct directory *directory = directory_get_child(parent, name_utf8); + db_unlock(); + + if (directory != NULL) + return directory; + + struct stat st; + if (stat_directory_child(parent, name_utf8, &st) < 0 || + find_inode_ancestor(parent, st.st_ino, st.st_dev)) + return NULL; + + if (skip_symlink(parent, name_utf8)) + return NULL; + + /* if we're adding directory paths, make sure to delete filenames + with potentially the same name */ + db_lock(); + struct song *conflicting = directory_get_song(parent, name_utf8); + if (conflicting) + delete_song(parent, conflicting); + + directory = directory_new_child(parent, name_utf8); + db_unlock(); + + directory_set_stat(directory, &st); + return directory; +} + +static struct directory * +directory_make_uri_parent_checked(const char *uri) +{ + struct directory *directory = db_get_root(); + char *duplicated = g_strdup(uri); + char *name_utf8 = duplicated, *slash; + + while ((slash = strchr(name_utf8, '/')) != NULL) { + *slash = 0; + + if (*name_utf8 == 0) + continue; + + directory = directory_make_child_checked(directory, name_utf8); + if (directory == NULL) + break; + + name_utf8 = slash + 1; + } + + g_free(duplicated); + return directory; +} + +static void +update_uri(const char *uri) +{ + struct directory *parent = directory_make_uri_parent_checked(uri); + if (parent == NULL) + return; + + char *name = g_path_get_basename(uri); + + struct stat st; + if (!skip_symlink(parent, name) && + stat_directory_child(parent, name, &st) == 0) + update_directory_child(parent, name, &st); + else + modified |= delete_name_in(parent, name); + + g_free(name); +} + +bool +update_walk(const char *path, bool discard) +{ + walk_discard = discard; + modified = false; + + if (path != NULL && !isRootDirectory(path)) { + update_uri(path); + } else { + struct directory *directory = db_get_root(); + struct stat st; + + if (stat_directory(directory, &st) == 0) + update_directory(directory, &st); + } + + return modified; +} -- cgit v1.2.3