X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=libs%2Fpbd%2Ffile_utils.cc;h=89f6818704c9d8c05d0ad2dc7c668cadc8e2d05d;hb=294b99aabf3eb96323a3159b7a5e1b4bfc1ff04a;hp=44254989c1bfbd227930aa9da8e206f1143ab6aa;hpb=fb313fb1741a04f270fff3703967df572ac4fc45;p=ardour.git diff --git a/libs/pbd/file_utils.cc b/libs/pbd/file_utils.cc index 44254989c1..89f6818704 100644 --- a/libs/pbd/file_utils.cc +++ b/libs/pbd/file_utils.cc @@ -1,5 +1,6 @@ /* - Copyright (C) 2007 Tim Mayberry + Copyright (C) 2007-2014 Tim Mayberry + Copyright (C) 1998-2014 Paul Davis 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 @@ -31,13 +32,28 @@ #include #include -#include +#include +#include /* strerror */ + +/* open() */ +#include +#include +#include + +/* close(), read(), write() */ +#ifdef COMPILER_MSVC +#include // Microsoft's nearest equivalent to +#include +#else +#include +#include +#endif #include "pbd/compose.h" #include "pbd/file_utils.h" #include "pbd/debug.h" #include "pbd/error.h" -#include "pbd/pathscanner.h" +#include "pbd/pathexpand.h" #include "pbd/stl_delete.h" #include "i18n.h" @@ -47,156 +63,299 @@ using namespace std; namespace PBD { void -get_files_in_directory (const std::string& directory_path, vector& result) +run_functor_for_paths (vector& result, + const Searchpath& paths, + bool (*functor)(const string &, void *), + void *arg, + bool pass_files_only, + bool pass_fullpath, bool return_fullpath, + bool recurse) { - if (!Glib::file_test (directory_path, Glib::FILE_TEST_IS_DIR)) return; + for (vector::const_iterator i = paths.begin(); i != paths.end(); ++i) { + string expanded_path = path_expand (*i); + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("Find files in expanded path: %1\n", expanded_path)); - try - { - Glib::Dir dir(directory_path); - std::copy(dir.begin(), dir.end(), std::back_inserter(result)); - } - catch (Glib::FileError& err) - { - warning << err.what() << endmsg; - } -} + if (!Glib::file_test (expanded_path, Glib::FILE_TEST_IS_DIR)) continue; -void -find_matching_files_in_directory (const std::string& directory, - const Glib::PatternSpec& pattern, - vector& result) -{ - vector tmp_files; + try + { + Glib::Dir dir(expanded_path); + + for (Glib::DirIterator di = dir.begin(); di != dir.end(); di++) { - get_files_in_directory (directory, tmp_files); - result.reserve(tmp_files.size()); + string fullpath = Glib::build_filename (expanded_path, *di); + string basename = *di; - for (vector::iterator file_iter = tmp_files.begin(); - file_iter != tmp_files.end(); - ++file_iter) - { - if (!pattern.match(*file_iter)) continue; + bool is_dir = Glib::file_test (fullpath, Glib::FILE_TEST_IS_DIR); - std::string full_path(directory); - full_path = Glib::build_filename (full_path, *file_iter); + if (is_dir && recurse) { + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("Descending into directory: %1\n", + fullpath)); + run_functor_for_paths (result, fullpath, functor, arg, pass_files_only, + pass_fullpath, return_fullpath, recurse); + } - DEBUG_TRACE ( - DEBUG::FileUtils, - string_compose("Found file %1\n", full_path) - ); + if (is_dir && pass_files_only) { + continue; + } - result.push_back(full_path); + string functor_str; + + if (pass_fullpath) { + functor_str = fullpath; + } else { + functor_str = basename; + } + + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("Run Functor using string: %1\n", functor_str)); + + if (!functor(functor_str, arg)) { + continue; + } + + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("Found file %1 matching functor\n", functor_str)); + + if (return_fullpath) { + result.push_back(fullpath); + } else { + result.push_back(basename); + } + } + } + catch (Glib::FileError& err) + { + warning << err.what() << endmsg; + } } } +static +bool accept_all_files (string const &, void *) +{ + return true; +} + void -find_matching_files_in_directories (const vector& paths, - const Glib::PatternSpec& pattern, - vector& result) +get_paths (vector& result, + const Searchpath& paths, + bool files_only, + bool recurse) { - for (vector::const_iterator path_iter = paths.begin(); - path_iter != paths.end(); - ++path_iter) - { - find_matching_files_in_directory (*path_iter, pattern, result); - } + run_functor_for_paths (result, paths, accept_all_files, 0, + files_only, true, true, recurse); } void -find_matching_files_in_search_path (const Searchpath& search_path, - const Glib::PatternSpec& pattern, - vector& result) +get_files (vector& result, const Searchpath& paths) { - find_matching_files_in_directories (search_path, pattern, result); + return get_paths (result, paths, true, false); } +static bool -find_file_in_search_path(const Searchpath& search_path, - const string& filename, - std::string& result) +pattern_filter (const string& str, void *arg) +{ + Glib::PatternSpec* pattern = (Glib::PatternSpec*)arg; + return pattern->match(str); +} + +void +find_files_matching_pattern (vector& result, + const Searchpath& paths, + const Glib::PatternSpec& pattern) +{ + run_functor_for_paths (result, paths, pattern_filter, + const_cast(&pattern), + true, false, true, false); +} + +void +find_files_matching_pattern (vector& result, + const Searchpath& paths, + const string& pattern) +{ + Glib::PatternSpec tmp(pattern); + find_files_matching_pattern (result, paths, tmp); +} + +bool +find_file (const Searchpath& search_path, + const string& filename, + std::string& result) { vector tmp; - Glib::PatternSpec tmp_pattern(filename); - find_matching_files_in_search_path (search_path, tmp_pattern, tmp); + find_files_matching_pattern (tmp, search_path, filename); - if (tmp.size() == 0) - { - DEBUG_TRACE ( - DEBUG::FileUtils, - string_compose("No file matching %1 found in Path: %2\n", filename, search_path.to_string()) - ); + if (tmp.size() == 0) { + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("No file matching %1 found in Path: %2\n", + filename, search_path.to_string())); return false; } - if (tmp.size() != 1) - { - DEBUG_TRACE ( - DEBUG::FileUtils, - string_compose("Found more that one file matching %1 in Path: %2\n", filename, search_path.to_string()) - ); + if (tmp.size() != 1) { + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("Found more that one file matching %1 in Path: %2\n", + filename, search_path.to_string())); } result = tmp.front(); - DEBUG_TRACE ( - DEBUG::FileUtils, - string_compose("Found file %1 in Path: %2\n", filename, search_path.to_string()) - ); + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("Found file %1 in Path: %2\n", + filename, search_path.to_string())); return true; } +static +bool +regexp_filter (const string& str, void *arg) +{ + regex_t* pattern = (regex_t*)arg; + return regexec (pattern, str.c_str(), 0, 0, 0) == 0; +} + +void +find_files_matching_regex (vector& result, + const Searchpath& paths, + const std::string& regexp) +{ + int err; + char msg[256]; + regex_t compiled_pattern; + + if ((err = regcomp (&compiled_pattern, regexp.c_str(), + REG_EXTENDED|REG_NOSUB))) { + + regerror (err, &compiled_pattern, + msg, sizeof (msg)); + + error << "Cannot compile soundfile regexp for use (" + << msg + << ")" + << endmsg; + + return; + } + + DEBUG_TRACE (DEBUG::FileUtils, + string_compose("Matching files using regexp: %1\n", regexp)); + + find_files_matching_filter (result, paths, + regexp_filter, &compiled_pattern, + true, true, false); + + regfree (&compiled_pattern); +} + +void +find_paths_matching_filter (vector& result, + const Searchpath& paths, + bool (*filter)(const string &, void *), + void *arg, + bool pass_fullpath, bool return_fullpath, + bool recurse) +{ + run_functor_for_paths (result, paths, filter, arg, false, pass_fullpath, return_fullpath, recurse); +} + +void +find_files_matching_filter (vector& result, + const Searchpath& paths, + bool (*filter)(const string &, void *), + void *arg, + bool pass_fullpath, bool return_fullpath, + bool recurse) +{ + run_functor_for_paths (result, paths, filter, arg, true, pass_fullpath, return_fullpath, recurse); +} + bool copy_file(const std::string & from_path, const std::string & to_path) { if (!Glib::file_test (from_path, Glib::FILE_TEST_EXISTS)) return false; - Glib::RefPtr from_file = Gio::File::create_for_path(from_path); - Glib::RefPtr to_file = Gio::File::create_for_path(to_path); + int fd_from (::open (from_path.c_str(), O_RDONLY)); + int fd_to (::open (to_path.c_str(), O_CREAT|O_TRUNC|O_RDWR, 0666)); - try - { - from_file->copy (to_file, Gio::FILE_COPY_OVERWRITE); - } - catch(const Glib::Exception& ex) - { - error << string_compose (_("Unable to Copy file %1 to %2 (%3)"), - from_path, to_path, ex.what()) - << endmsg; + char buf[4096]; // BUFSIZ ?? + ssize_t nread; + + if ((fd_from < 0) || (fd_to < 0)) { + error << string_compose (_("Unable to Open files %1 to %2 for Copying(%3)"), + from_path, to_path, g_strerror(errno)) + << endmsg; return false; } + + while (nread = ::read(fd_from, buf, sizeof(buf)), nread > 0) { + char *out_ptr = buf; + do { + ssize_t nwritten = ::write(fd_to, out_ptr, nread); + if (nwritten >= 0) { + nread -= nwritten; + out_ptr += nwritten; + } else if (errno != EINTR) { + error << string_compose (_("Unable to Copy files %1 to %2(%3)"), + from_path, to_path, g_strerror(errno)) + << endmsg; + return false; + } + } while (nread > 0); + } + return true; } -static -bool accept_all_files (string const &, void *) +void +copy_files(const std::string & from_path, const std::string & to_dir) { - return true; + vector files; + find_files_matching_filter (files, from_path, accept_all_files, 0, true, false); + + for (vector::iterator i = files.begin(); i != files.end(); ++i) { + std::string from = Glib::build_filename (from_path, *i); + std::string to = Glib::build_filename (to_dir, *i); + copy_file (from, to); + } } void -copy_files(const std::string & from_path, const std::string & to_dir) +copy_recurse(const std::string & from_path, const std::string & to_dir) { - PathScanner scanner; - vector* files = scanner (from_path, accept_all_files, 0, true, false); - - if (files) { - for (vector::iterator i = files->begin(); i != files->end(); ++i) { - std::string from = Glib::build_filename (from_path, **i); - std::string to = Glib::build_filename (to_dir, **i); - copy_file (from, to); - } - vector_delete (files); + vector files; + find_files_matching_filter (files, from_path, accept_all_files, 0, false, true, true); + + const size_t prefix_len = from_path.size(); + for (vector::iterator i = files.begin(); i != files.end(); ++i) { + std::string from = *i; + std::string to = Glib::build_filename (to_dir, (*i).substr(prefix_len)); + g_mkdir_with_parents (Glib::path_get_dirname (to).c_str(), 0755); + copy_file (from, to); } } std::string get_absolute_path (const std::string & p) { - Glib::RefPtr f = Gio::File::create_for_path (p); - return f->get_path (); + if (Glib::path_is_absolute(p)) return p; + return Glib::build_filename (Glib::get_current_dir(), p); +} + +std::string +get_suffix (const std::string & p) +{ + string::size_type period = p.find_last_of ('.'); + if (period == string::npos || period == p.length() - 1) { + return string(); + } + return p.substr (period+1); } bool @@ -257,4 +416,70 @@ exists_and_writable (const std::string & p) return true; } +int +remove_directory_internal (const string& dir, size_t* size, vector* paths, + bool just_remove_files) +{ + vector tmp_paths; + GStatBuf statbuf; + int ret = 0; + + get_paths (tmp_paths, dir, just_remove_files, true); + + for (vector::const_iterator i = tmp_paths.begin(); + i != tmp_paths.end(); ++i) { + + if (g_stat (i->c_str(), &statbuf)) { + continue; + } + + if (::g_remove (i->c_str())) { + error << string_compose (_("cannot remove path %1 (%2)"), *i, strerror (errno)) + << endmsg; + ret = 1; + } + + if (paths) { + paths->push_back (Glib::path_get_basename(*i)); + } + + if (size) { + *size += statbuf.st_size; + } + + } + + return ret; +} + +int +clear_directory (const string& dir, size_t* size, vector* paths) +{ + return remove_directory_internal (dir, size, paths, true); +} + +// rm -rf -- used to remove saved plugin state +void +remove_directory (const std::string& dir) +{ + remove_directory_internal (dir, 0, 0, false); +} + +string +tmp_writable_directory (const char* domain, const string& prefix) +{ + std::string tmp_dir = Glib::build_filename (g_get_tmp_dir(), domain); + std::string dir_name; + std::string new_test_dir; + do { + ostringstream oss; + oss << prefix; + oss << g_random_int (); + dir_name = oss.str(); + new_test_dir = Glib::build_filename (tmp_dir, dir_name); + if (Glib::file_test (new_test_dir, Glib::FILE_TEST_EXISTS)) continue; + } while (g_mkdir_with_parents (new_test_dir.c_str(), 0755) != 0); + return new_test_dir; +} + } // namespace PBD