13
0
livetrax/libs/pbd/file_utils.cc
Robin Gareus d257c8c0ce
Prevent endless recursive directory scans
This can happen when a folder symlinks to itself or subfolders
in some similar recursive pattern.
2024-09-08 01:33:57 +02:00

625 lines
16 KiB
C++

/*
* Copyright (C) 1998-2016 Paul Davis <paul@linuxaudiosystems.com>
* Copyright (C) 2007-2016 Tim Mayberry <mojofunk@gmail.com>
* Copyright (C) 2008-2009 David Robillard <d@drobilla.net>
* Copyright (C) 2013-2015 John Emmas <john@creativepost.co.uk>
* Copyright (C) 2014-2018 Robin Gareus <robin@gareus.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 <algorithm>
#include <climits>
#include <set>
#include <vector>
#include <glib.h>
#include "pbd/gstdio_compat.h"
#if defined COMPILER_MINGW || defined COMPILER_MSVC
#if defined COMPILER_MINGW
#include <io.h> // For W_OK
#include <windows.h>
#endif
#include <sys/utime.h>
#else
#include <utime.h>
#endif
#include <glibmm/convert.h>
#include <glibmm/fileutils.h>
#include <glibmm/miscutils.h>
#include <glibmm/pattern.h>
#include <errno.h>
#include <string.h> /* strerror */
/* open() */
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
/* close(), read(), write() */
#ifdef COMPILER_MSVC
#include <io.h> // Microsoft's nearest equivalent to <unistd.h>
#include <ardourext/misc.h>
#else
#include <unistd.h>
#include <regex.h>
#endif
#include "pbd/compose.h"
#include "pbd/file_utils.h"
#include "pbd/debug.h"
#include "pbd/error.h"
#include "pbd/pathexpand.h"
#include "pbd/scoped_file_descriptor.h"
#include "pbd/stl_delete.h"
#include "pbd/i18n.h"
using namespace std;
namespace PBD {
static void
run_functor_for_paths (vector<string>& result,
const Searchpath& paths,
bool (*functor)(const string &, void *),
void *arg,
bool pass_files_only,
bool pass_fullpath, bool return_fullpath,
bool recurse,
set<string>& scanned_paths)
{
for (vector<string>::const_iterator i = paths.begin(); i != paths.end(); ++i) {
try
{
string expanded_path = path_expand (*i);
DEBUG_TRACE (DEBUG::FileUtils,
string_compose("Find files in expanded path: %1\n", expanded_path));
if (!Glib::file_test (expanded_path, Glib::FILE_TEST_IS_DIR)) continue;
Glib::Dir dir(expanded_path);
for (Glib::DirIterator di = dir.begin(); di != dir.end(); di++) {
string fullpath = Glib::build_filename (expanded_path, *di);
string basename = *di;
bool is_dir = Glib::file_test (fullpath, Glib::FILE_TEST_IS_DIR);
if (is_dir && recurse) {
if (scanned_paths.find (fullpath) == scanned_paths.end ()) {
scanned_paths.insert (fullpath);
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, scanned_paths);
}
}
if (is_dir && pass_files_only) {
continue;
}
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 const& err) {
char errstr[PATH_MAX*2];
snprintf (errstr, sizeof (errstr), "Cannot access file: %s", err.what().c_str());
warning << errstr << endmsg;
}
catch (Glib::ConvertError const& err) {
char errstr[PATH_MAX*2];
snprintf (errstr, sizeof (errstr), "Cannot convert filename: %s", err.what().c_str());
warning << errstr << endmsg;
}
}
}
static
bool accept_all_files (string const &, void *)
{
return true;
}
void
get_paths (vector<string>& result,
const Searchpath& paths,
bool files_only,
bool recurse)
{
set<string> scanned_path;
run_functor_for_paths (result, paths, accept_all_files, 0,
files_only, true, true, recurse, scanned_path);
}
void
get_files (vector<string>& result, const Searchpath& paths)
{
return get_paths (result, paths, true, false);
}
static
bool
pattern_filter (const string& str, void *arg)
{
Glib::PatternSpec* pattern = (Glib::PatternSpec*)arg;
return pattern->match(str);
}
void
find_files_matching_pattern (vector<string>& result,
const Searchpath& paths,
const Glib::PatternSpec& pattern)
{
set<string> unused;
run_functor_for_paths (result, paths, pattern_filter,
const_cast<Glib::PatternSpec*>(&pattern),
true, false, true, false, unused);
}
void
find_files_matching_pattern (vector<string>& 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<std::string> 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()));
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()));
}
result = tmp.front();
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<string>& result,
const Searchpath& paths,
const std::string& regexp,
bool recurse)
{
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, recurse);
regfree (&compiled_pattern);
}
void
find_paths_matching_filter (vector<string>& result,
const Searchpath& paths,
bool (*filter)(const string &, void *),
void *arg,
bool pass_fullpath, bool return_fullpath,
bool recurse)
{
set<string> scanned_path;
run_functor_for_paths (result, paths, filter, arg, false, pass_fullpath, return_fullpath, recurse, scanned_path);
}
void
find_files_matching_filter (vector<string>& result,
const Searchpath& paths,
bool (*filter)(const string &, void *),
void *arg,
bool pass_fullpath, bool return_fullpath,
bool recurse)
{
set<string> scanned_path;
run_functor_for_paths (result, paths, filter, arg, true, pass_fullpath, return_fullpath, recurse, scanned_path);
}
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;
PBD::ScopedFileDescriptor fd_from (g_open (from_path.c_str(), O_RDONLY, 0444));
PBD::ScopedFileDescriptor fd_to (g_open (to_path.c_str(), O_CREAT|O_TRUNC|O_RDWR, 0666));
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;
}
void
copy_files(const std::string & from_path, const std::string & to_dir)
{
vector<string> files;
find_files_matching_filter (files, from_path, accept_all_files, 0, true, false);
for (vector<string>::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_recurse(const std::string & from_path, const std::string & to_dir, bool preseve_timestamps)
{
vector<string> 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<string>::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);
if (copy_file (from, to) && preseve_timestamps) {
GStatBuf sb;
if (g_stat (from.c_str(), &sb) != 0) {
error << string_compose (_("Unable to query file timestamp from %1 to %2"), from) << endmsg;
continue;
}
struct utimbuf utb;
utb.actime = sb.st_atime;
utb.modtime = sb.st_mtime;
if (0 != g_utime (to.c_str (), &utb)) {
error << string_compose (_("Unable to preseve file timestamp from %1 to %2"), from, to) << endmsg;
}
}
}
}
bool
touch_file (const std::string& path)
{
int fd = g_open (path.c_str(), O_RDWR|O_CREAT, 0660);
if (fd >= 0) {
close (fd);
return true;
}
return false;
}
bool
hard_link (const std::string& existing_file, const std::string& new_path)
{
#ifdef PLATFORM_WINDOWS
# if defined (COMPILER_MINGW) && defined(__GNUC__) && __GNUC__ == 8
/* For some reason mingx 8.3.0 does not support CreateHardLinkA()
* (mingw/gcc-4.9 does) */
return false;
# else
/* see also ntfs_link -- msvc only pbd extension */
return CreateHardLinkA (new_path.c_str(), existing_file.c_str(), NULL);
# endif
#else
return 0 == link (existing_file.c_str(), new_path.c_str());
#endif
}
std::string
get_absolute_path (const std::string & p)
{
if (Glib::path_is_absolute(p)) return p;
return Glib::build_filename (Glib::get_current_dir(), p);
}
string
canonical_path (const std::string& path)
{
#ifdef PLATFORM_WINDOWS
wchar_t resolved_wpath[_MAX_PATH];
// sizeof(wchar_t) is 2 bytes using gcc/mingw and VC++ but 4 bytes using gcc/linux
assert (sizeof(wchar_t) == 2);
wchar_t* wfilepath = (wchar_t*)g_utf8_to_utf16 (path.c_str(), -1, NULL, NULL, NULL);
if (wfilepath == NULL) {
DEBUG_TRACE (
DEBUG::FileUtils,
string_compose ("PBD::canonical_path: Unable to convert path from utf8 to utf16 : %1\n",
path));
return path;
}
if (_wfullpath (resolved_wpath, wfilepath, _MAX_PATH) == NULL) {
DEBUG_TRACE (DEBUG::FileUtils,
string_compose ("PBD::canonical_path: Unable to resolve %1\n", wfilepath));
return path;
}
gchar* resolved_utf8_path =
g_utf16_to_utf8 (reinterpret_cast<const gunichar2*>(resolved_wpath), -1, NULL, NULL, NULL);
if (resolved_utf8_path == NULL) {
DEBUG_TRACE (
DEBUG::FileUtils,
string_compose ("PBD::canonical_path: Unable to convert path from utf16 to utf8 : %1\n",
resolved_wpath));
return path;
}
const string retval(resolved_utf8_path);
g_free (wfilepath);
g_free (resolved_utf8_path);
return retval;
#else
char buf[PATH_MAX+1];
if (realpath (path.c_str(), buf) == NULL) {
DEBUG_TRACE (DEBUG::FileUtils,
string_compose ("PBD::canonical_path: Unable to resolve %1: %2\n", path,
g_strerror (errno)));
return path;
}
DEBUG_TRACE (DEBUG::FileUtils,
string_compose ("PBD::canonical_path %1 resolved to: %2\n", path, string (buf)));
return string (buf);
#endif
}
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
equivalent_paths (const std::string& a, const std::string& b)
{
GStatBuf bA;
int const rA = g_stat (a.c_str(), &bA);
GStatBuf bB;
int const rB = g_stat (b.c_str(), &bB);
return (rA == 0 && rB == 0 && bA.st_dev == bB.st_dev && bA.st_ino == bB.st_ino);
}
bool
path_is_within (std::string const & haystack, std::string needle)
{
while (1) {
if (equivalent_paths (haystack, needle)) {
return true;
}
needle = Glib::path_get_dirname (needle);
if (needle == "." || needle == "/" || Glib::path_skip_root(needle).empty()) {
break;
}
}
return false;
}
bool
exists_and_writable (const std::string & p)
{
/* writable() really reflects the whole folder, but if for any
reason the session state file can't be written to, still
make us unwritable.
*/
GStatBuf statbuf;
if (g_stat (p.c_str(), &statbuf) != 0) {
/* doesn't exist - not writable */
return false;
} else {
if (!(statbuf.st_mode & S_IWUSR)) {
/* exists and is not writable */
return false;
}
/* filesystem may be mounted read-only, so even though file
* permissions permit access, the mount status does not.
* access(2) seems like the best test for this.
*/
if (g_access (p.c_str(), W_OK) != 0) {
return false;
}
}
return true;
}
int
remove_directory_internal (const string& dir, size_t* size, vector<string>* paths,
bool just_remove_files)
{
vector<string> tmp_paths;
GStatBuf statbuf;
int ret = 0;
get_paths (tmp_paths, dir, just_remove_files, true);
for (vector<string>::const_iterator i = tmp_paths.begin();
i != tmp_paths.end(); ++i) {
if (g_stat (i->c_str(), &statbuf) && g_lstat (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;
continue;
}
if (paths) {
paths->push_back (Glib::path_get_basename(*i));
}
// statbuf.st_size is off_t
if (size && statbuf.st_size > 0) {
*size += statbuf.st_size;
}
}
return ret;
}
int
clear_directory (const string& dir, size_t* size, vector<string>* paths)
{
return remove_directory_internal (dir, size, paths, true);
}
// rm -rf <dir> -- used to remove saved plugin state
void
remove_directory (const std::string& dir)
{
remove_directory_internal (dir, 0, 0, false);
g_rmdir (dir.c_str());
}
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;
}
int
toggle_file_existence (string const & path)
{
if (Glib::file_test (path, Glib::FILE_TEST_IS_REGULAR)) {
return g_unlink (path.c_str());
}
PBD::ScopedFileDescriptor fd = g_open (path.c_str(), O_CREAT|O_TRUNC|O_RDWR, 0666);
return !((int) fd >= 0);
}
} // namespace PBD