1600 lines
50 KiB
C++
1600 lines
50 KiB
C++
#include "foobar2000.h"
|
|
|
|
// For reasons unknown, MS linker will not throw these global constants out if the code using them is not referenced
|
|
// To verify, flip the #if and search your DLL for "unpack://"
|
|
// Using #defines instead fixes it
|
|
#if 1
|
|
#define unpack_prefix "unpack://"
|
|
#define unpack_prefix_len 9
|
|
#else
|
|
static const char unpack_prefix[] = "unpack://";
|
|
static const unsigned unpack_prefix_len = 9;
|
|
#endif
|
|
|
|
void unpacker::g_open(service_ptr_t<file> & p_out,const service_ptr_t<file> & p,abort_callback & p_abort)
|
|
{
|
|
service_enum_t<unpacker> e;
|
|
service_ptr_t<unpacker> ptr;
|
|
if (e.first(ptr)) do {
|
|
p->reopen(p_abort);
|
|
try {
|
|
ptr->open(p_out,p,p_abort);
|
|
return;
|
|
} catch(exception_io_data const &) {}
|
|
} while(e.next(ptr));
|
|
throw exception_io_data();
|
|
}
|
|
|
|
void file::seek_probe(t_filesize p_position, abort_callback & p_abort) {
|
|
try { seek(p_position, p_abort); } catch(exception_io_seek_out_of_range) {throw exception_io_data();}
|
|
}
|
|
|
|
void file::seek_ex(t_sfilesize p_position, file::t_seek_mode p_mode, abort_callback &p_abort) {
|
|
switch(p_mode) {
|
|
case seek_from_beginning:
|
|
seek(p_position,p_abort);
|
|
break;
|
|
case seek_from_current:
|
|
seek(p_position + get_position(p_abort),p_abort);
|
|
break;
|
|
case seek_from_eof:
|
|
seek(p_position + get_size_ex(p_abort),p_abort);
|
|
break;
|
|
default:
|
|
throw exception_io_data();
|
|
}
|
|
}
|
|
|
|
static void makeBuffer(pfc::array_t<uint8_t> & buffer, size_t size) {
|
|
for(;;) {// Tolerant malloc - allocate a smaller buffer if we're unable to acquire the requested size.
|
|
try {
|
|
buffer.set_size_discard( size );
|
|
return;
|
|
} catch(std::bad_alloc) {
|
|
if (size < 256) throw;
|
|
size >>= 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
t_filesize file::g_transfer(stream_reader * p_src,stream_writer * p_dst,t_filesize p_bytes,abort_callback & p_abort) {
|
|
pfc::array_t<t_uint8> temp;
|
|
makeBuffer(temp, (t_size)pfc::min_t<t_filesize>(1024*1024*8,p_bytes));
|
|
void* ptr = temp.get_ptr();
|
|
t_filesize done = 0;
|
|
while(done<p_bytes) {
|
|
p_abort.check_e();
|
|
t_size delta = (t_size)pfc::min_t<t_filesize>(temp.get_size(),p_bytes-done);
|
|
delta = p_src->read(ptr,delta,p_abort);
|
|
if (delta<=0) break;
|
|
p_dst->write(ptr,delta,p_abort);
|
|
done += delta;
|
|
}
|
|
return done;
|
|
}
|
|
|
|
void file::g_transfer_object(stream_reader * p_src,stream_writer * p_dst,t_filesize p_bytes,abort_callback & p_abort) {
|
|
if (g_transfer(p_src,p_dst,p_bytes,p_abort) != p_bytes)
|
|
throw exception_io_data_truncation();
|
|
}
|
|
|
|
|
|
void filesystem::g_get_canonical_path(const char * path,pfc::string_base & out)
|
|
{
|
|
TRACK_CALL_TEXT("filesystem::g_get_canonical_path");
|
|
|
|
service_enum_t<filesystem> e;
|
|
service_ptr_t<filesystem> ptr;
|
|
if (e.first(ptr)) do {
|
|
if (ptr->get_canonical_path(path,out)) return;
|
|
} while(e.next(ptr));
|
|
//no one wants to process this, let's copy over
|
|
out = path;
|
|
}
|
|
|
|
void filesystem::g_get_display_path(const char * path,pfc::string_base & out)
|
|
{
|
|
TRACK_CALL_TEXT("filesystem::g_get_display_path");
|
|
service_ptr_t<filesystem> ptr;
|
|
if (!g_get_interface(ptr,path))
|
|
{
|
|
//no one wants to process this, let's copy over
|
|
out = path;
|
|
}
|
|
else
|
|
{
|
|
if (!ptr->get_display_path(path,out))
|
|
out = path;
|
|
}
|
|
}
|
|
|
|
bool filesystem::g_get_native_path( const char * path, pfc::string_base & out) {
|
|
// Is proper file:// path?
|
|
if (foobar2000_io::extract_native_path( path, out ) ) return true;
|
|
|
|
// Set anyway
|
|
out = path;
|
|
|
|
// Maybe just a file:// less local path? Check for other protocol markers
|
|
// If no :// present, return true anyway
|
|
return strstr( path, "://" ) == NULL;
|
|
}
|
|
|
|
filesystem::ptr filesystem::g_get_interface(const char * path) {
|
|
filesystem::ptr rv;
|
|
if (!g_get_interface(rv, path)) throw exception_io_no_handler_for_path();
|
|
return rv;
|
|
|
|
}
|
|
bool filesystem::g_get_interface(service_ptr_t<filesystem> & p_out,const char * path)
|
|
{
|
|
PFC_ASSERT( path != nullptr );
|
|
PFC_ASSERT( path[0] != 0 );
|
|
|
|
service_enum_t<filesystem> e;
|
|
service_ptr_t<filesystem> ptr;
|
|
if (e.first(ptr)) do {
|
|
if (ptr->is_our_path(path))
|
|
{
|
|
p_out = ptr;
|
|
return true;
|
|
}
|
|
} while(e.next(ptr));
|
|
return false;
|
|
}
|
|
|
|
|
|
void filesystem::g_open(service_ptr_t<file> & p_out,const char * path,t_open_mode mode,abort_callback & p_abort)
|
|
{
|
|
TRACK_CALL_TEXT("filesystem::g_open");
|
|
g_get_interface(path)->open(p_out,path,mode,p_abort);
|
|
}
|
|
|
|
|
|
void filesystem::g_open_timeout(service_ptr_t<file> & p_out,const char * p_path,t_open_mode p_mode,double p_timeout,abort_callback & p_abort) {
|
|
FB2K_RETRY_ON_SHARING_VIOLATION( g_open(p_out, p_path, p_mode, p_abort), p_abort, p_timeout);
|
|
}
|
|
|
|
bool filesystem::g_exists(const char * p_path,abort_callback & p_abort)
|
|
{
|
|
t_filestats stats;
|
|
bool dummy;
|
|
try {
|
|
g_get_stats(p_path,stats,dummy,p_abort);
|
|
} catch(exception_io_not_found) {return false;}
|
|
return true;
|
|
}
|
|
|
|
bool filesystem::g_exists_writeable(const char * p_path,abort_callback & p_abort)
|
|
{
|
|
t_filestats stats;
|
|
bool writeable;
|
|
try {
|
|
g_get_stats(p_path,stats,writeable,p_abort);
|
|
} catch(exception_io_not_found) {return false;}
|
|
return writeable;
|
|
}
|
|
|
|
void filesystem::g_remove(const char * p_path,abort_callback & p_abort) {
|
|
g_get_interface(p_path)->remove(p_path,p_abort);
|
|
}
|
|
|
|
void filesystem::g_remove_timeout(const char * p_path,double p_timeout,abort_callback & p_abort) {
|
|
FB2K_RETRY_FILE_MOVE( g_remove(p_path, p_abort), p_abort, p_timeout );
|
|
}
|
|
|
|
void filesystem::g_move_timeout(const char * p_src,const char * p_dst,double p_timeout,abort_callback & p_abort) {
|
|
FB2K_RETRY_FILE_MOVE( g_move(p_src, p_dst, p_abort), p_abort, p_timeout );
|
|
}
|
|
|
|
void filesystem::g_copy_timeout(const char * p_src,const char * p_dst,double p_timeout,abort_callback & p_abort) {
|
|
FB2K_RETRY_FILE_MOVE( g_copy(p_src, p_dst, p_abort), p_abort, p_timeout );
|
|
}
|
|
|
|
void filesystem::g_create_directory(const char * p_path,abort_callback & p_abort)
|
|
{
|
|
g_get_interface(p_path)->create_directory(p_path,p_abort);
|
|
}
|
|
|
|
void filesystem::g_move(const char * src,const char * dst,abort_callback & p_abort) {
|
|
service_enum_t<filesystem> e;
|
|
service_ptr_t<filesystem> ptr;
|
|
if (e.first(ptr)) do {
|
|
if (ptr->is_our_path(src) && ptr->is_our_path(dst)) {
|
|
ptr->move(src,dst,p_abort);
|
|
return;
|
|
}
|
|
} while(e.next(ptr));
|
|
throw exception_io_no_handler_for_path();
|
|
}
|
|
|
|
void filesystem::g_link(const char * p_src,const char * p_dst,abort_callback & p_abort) {
|
|
if (!foobar2000_io::_extract_native_path_ptr(p_src) || !foobar2000_io::_extract_native_path_ptr(p_dst)) throw exception_io_no_handler_for_path();
|
|
WIN32_IO_OP( CreateHardLink( pfc::stringcvt::string_os_from_utf8( pfc::winPrefixPath( p_dst ) ), pfc::stringcvt::string_os_from_utf8( pfc::winPrefixPath( p_src ) ), NULL) );
|
|
}
|
|
|
|
void filesystem::g_link_timeout(const char * p_src,const char * p_dst,double p_timeout,abort_callback & p_abort) {
|
|
FB2K_RETRY_FILE_MOVE( g_link(p_src, p_dst, p_abort), p_abort, p_timeout );
|
|
}
|
|
|
|
|
|
void filesystem::g_list_directory(const char * p_path,directory_callback & p_out,abort_callback & p_abort)
|
|
{
|
|
TRACK_CALL_TEXT("filesystem::g_list_directory");
|
|
g_get_interface(p_path)->list_directory(p_path,p_out,p_abort);
|
|
}
|
|
|
|
|
|
static void path_pack_string(pfc::string_base & out,const char * src)
|
|
{
|
|
out.add_char('|');
|
|
out << (unsigned) strlen(src);
|
|
out.add_char('|');
|
|
out << src;
|
|
out.add_char('|');
|
|
}
|
|
|
|
static int path_unpack_string(pfc::string_base & out,const char * src)
|
|
{
|
|
int ptr=0;
|
|
if (src[ptr++]!='|') return -1;
|
|
int len = atoi(src+ptr);
|
|
if (len<=0) return -1;
|
|
while(src[ptr]!=0 && src[ptr]!='|') ptr++;
|
|
if (src[ptr]!='|') return -1;
|
|
ptr++;
|
|
int start = ptr;
|
|
while(ptr-start<len)
|
|
{
|
|
if (src[ptr]==0) return -1;
|
|
ptr++;
|
|
}
|
|
if (src[ptr]!='|') return -1;
|
|
out.set_string(&src[start],len);
|
|
ptr++;
|
|
return ptr;
|
|
}
|
|
|
|
|
|
void filesystem::g_open_precache(service_ptr_t<file> & p_out,const char * p_path,abort_callback & p_abort) {
|
|
service_ptr_t<filesystem> fs = g_get_interface(p_path);
|
|
if (fs->is_remote(p_path)) throw exception_io_object_is_remote();
|
|
fs->open(p_out,p_path,open_mode_read,p_abort);
|
|
}
|
|
|
|
bool filesystem::g_is_remote(const char * p_path) {
|
|
return g_get_interface(p_path)->is_remote(p_path);
|
|
}
|
|
|
|
bool filesystem::g_is_recognized_and_remote(const char * p_path) {
|
|
service_ptr_t<filesystem> fs;
|
|
if (g_get_interface(fs,p_path)) return fs->is_remote(p_path);
|
|
else return false;
|
|
}
|
|
|
|
bool filesystem::g_is_remote_or_unrecognized(const char * p_path) {
|
|
service_ptr_t<filesystem> fs;
|
|
if (g_get_interface(fs,p_path)) return fs->is_remote(p_path);
|
|
else return true;
|
|
}
|
|
|
|
bool filesystem::g_relative_path_create(const char * file_path,const char * playlist_path,pfc::string_base & out)
|
|
{
|
|
|
|
bool rv = false;
|
|
service_ptr_t<filesystem> fs;
|
|
|
|
if (g_get_interface(fs,file_path))
|
|
rv = fs->relative_path_create(file_path,playlist_path,out);
|
|
|
|
return rv;
|
|
}
|
|
|
|
bool filesystem::g_relative_path_parse(const char * relative_path,const char * playlist_path,pfc::string_base & out)
|
|
{
|
|
service_enum_t<filesystem> e;
|
|
service_ptr_t<filesystem> ptr;
|
|
if (e.first(ptr)) do {
|
|
if (ptr->relative_path_parse(relative_path,playlist_path,out)) return true;
|
|
} while(e.next(ptr));
|
|
return false;
|
|
}
|
|
|
|
bool archive::is_our_archive( const char * path ) {
|
|
archive_v2::ptr v2;
|
|
if ( v2 &= this ) return v2->is_our_archive( path );
|
|
return true; // accept all files
|
|
}
|
|
|
|
bool archive_impl::get_canonical_path(const char * path,pfc::string_base & out)
|
|
{
|
|
if (is_our_path(path))
|
|
{
|
|
pfc::string8 archive,file,archive_canonical;
|
|
if (g_parse_unpack_path(path,archive,file))
|
|
{
|
|
g_get_canonical_path(archive,archive_canonical);
|
|
make_unpack_path(out,archive_canonical,file);
|
|
|
|
return true;
|
|
}
|
|
else return false;
|
|
}
|
|
else return false;
|
|
}
|
|
|
|
bool archive_impl::is_our_path(const char * path)
|
|
{
|
|
if (!g_is_unpack_path(path)) return false;
|
|
const char * type = get_archive_type();
|
|
path += 9;
|
|
while(*type)
|
|
{
|
|
if (*type!=*path) return false;
|
|
type++;
|
|
path++;
|
|
}
|
|
if (*path!='|') return false;
|
|
return true;
|
|
}
|
|
|
|
bool archive_impl::get_display_path(const char * path,pfc::string_base & out)
|
|
{
|
|
pfc::string8 archive,file;
|
|
if (g_parse_unpack_path(path,archive,file))
|
|
{
|
|
g_get_display_path(archive,out);
|
|
out.add_string("|");
|
|
out.add_string(file);
|
|
return true;
|
|
}
|
|
else return false;
|
|
}
|
|
|
|
void archive_impl::open(service_ptr_t<file> & p_out,const char * path,t_open_mode mode, abort_callback & p_abort)
|
|
{
|
|
if (mode != open_mode_read) throw exception_io_denied();
|
|
pfc::string8 archive,file;
|
|
if (!g_parse_unpack_path(path,archive,file)) throw exception_io_not_found();
|
|
open_archive(p_out,archive,file,p_abort);
|
|
}
|
|
|
|
|
|
void archive_impl::remove(const char * path,abort_callback & p_abort) {
|
|
throw exception_io_denied();
|
|
}
|
|
|
|
void archive_impl::move(const char * src,const char * dst,abort_callback & p_abort) {
|
|
throw exception_io_denied();
|
|
}
|
|
|
|
bool archive_impl::is_remote(const char * src) {
|
|
pfc::string8 archive,file;
|
|
if (g_parse_unpack_path(src,archive,file)) return g_is_remote(archive);
|
|
else throw exception_io_not_found();
|
|
}
|
|
|
|
bool archive_impl::relative_path_create(const char * file_path,const char * playlist_path,pfc::string_base & out) {
|
|
pfc::string8 archive,file;
|
|
if (g_parse_unpack_path(file_path,archive,file))
|
|
{
|
|
pfc::string8 archive_rel;
|
|
if (g_relative_path_create(archive,playlist_path,archive_rel))
|
|
{
|
|
pfc::string8 out_path;
|
|
make_unpack_path(out_path,archive_rel,file);
|
|
out.set_string(out_path);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool archive_impl::relative_path_parse(const char * relative_path,const char * playlist_path,pfc::string_base & out)
|
|
{
|
|
if (!is_our_path(relative_path)) return false;
|
|
pfc::string8 archive_rel,file;
|
|
if (g_parse_unpack_path(relative_path,archive_rel,file))
|
|
{
|
|
pfc::string8 archive;
|
|
if (g_relative_path_parse(archive_rel,playlist_path,archive))
|
|
{
|
|
pfc::string8 out_path;
|
|
make_unpack_path(out_path,archive,file);
|
|
out.set_string(out_path);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool archive_impl::g_parse_unpack_path_ex(const char * path,pfc::string_base & archive,pfc::string_base & file, pfc::string_base & type) {
|
|
PFC_ASSERT( g_is_unpack_path(path) );
|
|
const char * base = path + unpack_prefix_len; // strstr(path, "//");
|
|
const char * split = strchr(path,'|');
|
|
if (base == NULL || split == NULL || base > split) return false;
|
|
// base += 2;
|
|
type.set_string( base, split - base );
|
|
int delta = path_unpack_string(archive,split);
|
|
if (delta<0) return false;
|
|
split += delta;
|
|
file = split;
|
|
return true;
|
|
}
|
|
bool archive_impl::g_parse_unpack_path(const char * path,pfc::string_base & archive,pfc::string_base & file) {
|
|
PFC_ASSERT( g_is_unpack_path(path) );
|
|
path = strchr(path,'|');
|
|
if (!path) return false;
|
|
int delta = path_unpack_string(archive,path);
|
|
if (delta<0) return false;
|
|
path += delta;
|
|
file = path;
|
|
return true;
|
|
}
|
|
|
|
bool archive_impl::g_is_unpack_path(const char * path) {
|
|
return strncmp(path,unpack_prefix,unpack_prefix_len) == 0;
|
|
}
|
|
|
|
void archive_impl::g_make_unpack_path(pfc::string_base & path,const char * archive,const char * file,const char * name)
|
|
{
|
|
path = unpack_prefix;
|
|
path += name;
|
|
path_pack_string(path,archive);
|
|
path += file;
|
|
}
|
|
|
|
void archive_impl::make_unpack_path(pfc::string_base & path,const char * archive,const char * file) {g_make_unpack_path(path,archive,file,get_archive_type());}
|
|
|
|
|
|
FILE * filesystem::streamio_open(const char * path,const char * flags)
|
|
{
|
|
FILE * ret = 0;
|
|
pfc::string8 temp;
|
|
g_get_canonical_path(path,temp);
|
|
if (!strncmp(temp,"file://",7))
|
|
{
|
|
ret = _wfopen(pfc::stringcvt::string_wide_from_utf8(path+7),pfc::stringcvt::string_wide_from_utf8(flags));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
|
|
namespace {
|
|
|
|
class directory_callback_isempty : public directory_callback
|
|
{
|
|
bool m_isempty;
|
|
public:
|
|
directory_callback_isempty() : m_isempty(true) {}
|
|
bool on_entry(filesystem * owner,abort_callback & p_abort,const char * url,bool is_subdirectory,const t_filestats & p_stats)
|
|
{
|
|
m_isempty = false;
|
|
return false;
|
|
}
|
|
bool isempty() {return m_isempty;}
|
|
};
|
|
|
|
class directory_callback_dummy : public directory_callback
|
|
{
|
|
public:
|
|
bool on_entry(filesystem * owner,abort_callback & p_abort,const char * url,bool is_subdirectory,const t_filestats & p_stats) {return false;}
|
|
};
|
|
|
|
}
|
|
|
|
bool filesystem::g_is_empty_directory(const char * path,abort_callback & p_abort)
|
|
{
|
|
directory_callback_isempty callback;
|
|
try {
|
|
g_list_directory(path,callback,p_abort);
|
|
} catch(exception_io const &) {return false;}
|
|
return callback.isempty();
|
|
}
|
|
|
|
bool filesystem::g_is_valid_directory(const char * path,abort_callback & p_abort) {
|
|
if ( path == NULL || path[0] == 0 ) return false;
|
|
|
|
return get(path)->directory_exists( path, p_abort );
|
|
}
|
|
|
|
bool directory_callback_impl::on_entry(filesystem * owner,abort_callback & p_abort,const char * url,bool is_subdirectory,const t_filestats & p_stats) {
|
|
p_abort.check_e();
|
|
if (is_subdirectory) {
|
|
if (m_recur) {
|
|
try {
|
|
owner->list_directory(url,*this,p_abort);
|
|
} catch(exception_io const &) {}
|
|
}
|
|
} else {
|
|
m_data.add_item(pfc::rcnew_t<t_entry>(url,p_stats));
|
|
}
|
|
return true;
|
|
}
|
|
|
|
namespace {
|
|
class directory_callback_impl_copy : public directory_callback
|
|
{
|
|
public:
|
|
directory_callback_impl_copy(const char * p_target, filesystem::ptr fs) : m_fs(fs)
|
|
{
|
|
m_target = p_target;
|
|
m_target.fix_dir_separator('\\');
|
|
}
|
|
|
|
bool on_entry(filesystem * owner,abort_callback & p_abort,const char * url,bool is_subdirectory,const t_filestats & p_stats) {
|
|
const char * fn = url + pfc::scan_filename(url);
|
|
t_size truncat = m_target.length();
|
|
m_target += fn;
|
|
if (is_subdirectory) {
|
|
try {
|
|
m_fs->create_directory(m_target,p_abort);
|
|
} catch(exception_io_already_exists) {}
|
|
m_target += "\\";
|
|
owner->list_directory(url,*this,p_abort);
|
|
} else {
|
|
_copy(url, m_target, owner, p_abort);
|
|
}
|
|
m_target.truncate(truncat);
|
|
return true;
|
|
}
|
|
void _copy(const char * src, const char * dst, filesystem * srcFS, abort_callback & p_abort) {
|
|
service_ptr_t<file> r_src,r_dst;
|
|
t_filesize size;
|
|
|
|
srcFS->open(r_src,src,filesystem::open_mode_read,p_abort);
|
|
size = r_src->get_size_ex(p_abort);
|
|
m_fs->open(r_dst,dst,filesystem::open_mode_write_new,p_abort);
|
|
|
|
if (size > 0) {
|
|
try {
|
|
file::g_transfer_object(r_src,r_dst,size,p_abort);
|
|
} catch(...) {
|
|
r_dst.release();
|
|
try {m_fs->remove(dst,fb2k::noAbort);} catch(...) {}
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
private:
|
|
pfc::string8_fastalloc m_target;
|
|
filesystem::ptr m_fs;
|
|
};
|
|
}
|
|
|
|
file::ptr filesystem::openEx(const char * path, filesystem::t_open_mode mode, abort_callback & abort, double timeout) {
|
|
file::ptr f;
|
|
retryOnSharingViolation([&] {
|
|
this->open(f, path, mode, abort);
|
|
}, timeout, abort);
|
|
return f;
|
|
}
|
|
|
|
file::ptr filesystem::openRead(const char * path, abort_callback & abort, double timeout) {
|
|
return this->openEx(path, open_mode_read, abort, timeout);
|
|
}
|
|
|
|
file::ptr filesystem::openWriteExisting(const char * path, abort_callback & abort, double timeout) {
|
|
return this->openEx(path, open_mode_write_existing, abort, timeout);
|
|
}
|
|
|
|
file::ptr filesystem::openWriteNew(const char * path, abort_callback & abort, double timeout) {
|
|
return this->openEx( path, open_mode_write_new, abort, timeout );
|
|
}
|
|
|
|
void filesystem::copy_directory(const char * src, const char * dst, abort_callback & p_abort) {
|
|
try {
|
|
this->create_directory( dst, p_abort );
|
|
} catch(exception_io_already_exists) {}
|
|
directory_callback_impl_copy cb(dst, this);
|
|
list_directory(src, cb, p_abort);
|
|
}
|
|
|
|
void filesystem::g_copy_directory(const char * src,const char * dst,abort_callback & p_abort) {
|
|
filesystem::ptr dstFS = filesystem::g_get_interface(dst);
|
|
try {
|
|
dstFS->create_directory( dst, p_abort );
|
|
} catch(exception_io_already_exists) {}
|
|
directory_callback_impl_copy cb(dst, dstFS);
|
|
g_list_directory(src,cb,p_abort);
|
|
}
|
|
|
|
void filesystem::g_copy(const char * src,const char * dst,abort_callback & p_abort) {
|
|
service_ptr_t<file> r_src,r_dst;
|
|
t_filesize size;
|
|
|
|
g_open(r_src,src,open_mode_read,p_abort);
|
|
size = r_src->get_size_ex(p_abort);
|
|
g_open(r_dst,dst,open_mode_write_new,p_abort);
|
|
|
|
if (size > 0) {
|
|
try {
|
|
file::g_transfer_object(r_src,r_dst,size,p_abort);
|
|
} catch(...) {
|
|
r_dst.release();
|
|
try {g_remove(dst,fb2k::noAbort);} catch(...) {}
|
|
throw;
|
|
}
|
|
}
|
|
|
|
try {
|
|
file::g_copy_timestamps(r_src, r_dst, p_abort);
|
|
} catch (exception_io) {}
|
|
}
|
|
|
|
void stream_reader::read_object(void * p_buffer,t_size p_bytes,abort_callback & p_abort) {
|
|
if (read(p_buffer,p_bytes,p_abort) != p_bytes) throw exception_io_data_truncation();
|
|
}
|
|
|
|
t_filestats file::get_stats(abort_callback & p_abort)
|
|
{
|
|
t_filestats temp;
|
|
temp.m_size = get_size(p_abort);
|
|
temp.m_timestamp = get_timestamp(p_abort);
|
|
return temp;
|
|
}
|
|
|
|
t_filesize stream_reader::skip(t_filesize p_bytes,abort_callback & p_abort)
|
|
{
|
|
t_uint8 temp[256];
|
|
t_filesize todo = p_bytes, done = 0;
|
|
while(todo > 0) {
|
|
t_size delta,deltadone;
|
|
delta = sizeof(temp);
|
|
if (delta > todo) delta = (t_size) todo;
|
|
deltadone = read(temp,delta,p_abort);
|
|
done += deltadone;
|
|
todo -= deltadone;
|
|
if (deltadone < delta) break;
|
|
}
|
|
return done;
|
|
}
|
|
|
|
void stream_reader::skip_object(t_filesize p_bytes,abort_callback & p_abort) {
|
|
if (skip(p_bytes,p_abort) != p_bytes) throw exception_io_data_truncation();
|
|
}
|
|
|
|
void filesystem::g_open_write_new(service_ptr_t<file> & p_out,const char * p_path,abort_callback & p_abort) {
|
|
g_open(p_out,p_path,open_mode_write_new,p_abort);
|
|
}
|
|
void file::g_transfer_file(const service_ptr_t<file> & p_from,const service_ptr_t<file> & p_to,abort_callback & p_abort) {
|
|
t_filesize length = p_from->get_size(p_abort);
|
|
p_from->reopen( p_abort );
|
|
// p_from->seek(0,p_abort);
|
|
p_to->seek(0,p_abort);
|
|
p_to->set_eof(p_abort);
|
|
if (length == filesize_invalid) {
|
|
g_transfer(p_from, p_to, ~0, p_abort);
|
|
} else if (length > 0) {
|
|
g_transfer_object(p_from,p_to,length,p_abort);
|
|
}
|
|
}
|
|
|
|
void filesystem::g_open_temp(service_ptr_t<file> & p_out,abort_callback & p_abort) {
|
|
g_open(p_out,"tempfile://",open_mode_write_new,p_abort);
|
|
}
|
|
|
|
void filesystem::g_open_tempmem(service_ptr_t<file> & p_out,abort_callback & p_abort) {
|
|
g_open(p_out,"tempmem://",open_mode_write_new,p_abort);
|
|
}
|
|
|
|
file::ptr filesystem::g_open_tempmem() {
|
|
file::ptr f; g_open_tempmem(f, fb2k::noAbort); return f;
|
|
}
|
|
|
|
void archive_impl::list_directory(const char * p_path,directory_callback & p_out,abort_callback & p_abort) {
|
|
throw exception_io_not_found();
|
|
}
|
|
|
|
void archive_impl::create_directory(const char * path,abort_callback &) {
|
|
throw exception_io_denied();
|
|
}
|
|
|
|
void filesystem::g_get_stats(const char * p_path,t_filestats & p_stats,bool & p_is_writeable,abort_callback & p_abort) {
|
|
TRACK_CALL_TEXT("filesystem::g_get_stats");
|
|
return g_get_interface(p_path)->get_stats(p_path,p_stats,p_is_writeable,p_abort);
|
|
}
|
|
|
|
void archive_impl::get_stats(const char * p_path,t_filestats & p_stats,bool & p_is_writeable,abort_callback & p_abort) {
|
|
pfc::string8 archive,file;
|
|
if (g_parse_unpack_path(p_path,archive,file)) {
|
|
if (g_is_remote(archive)) throw exception_io_object_is_remote();
|
|
p_is_writeable = false;
|
|
p_stats = get_stats_in_archive(archive,file,p_abort);
|
|
}
|
|
else throw exception_io_not_found();
|
|
}
|
|
|
|
|
|
bool file::is_eof(abort_callback & p_abort) {
|
|
t_filesize position,size;
|
|
position = get_position(p_abort);
|
|
size = get_size(p_abort);
|
|
if (size == filesize_invalid) return false;
|
|
return position >= size;
|
|
}
|
|
|
|
|
|
t_filetimestamp foobar2000_io::import_DOS_time(uint32_t v) {
|
|
#ifdef _WIN32
|
|
FILETIME ft = {};
|
|
if (DosDateTimeToFileTime(HIWORD(v), LOWORD(v), &ft)) {
|
|
FILETIME ft2 = {};
|
|
if (LocalFileTimeToFileTime(&ft, &ft2)) {
|
|
return ((uint64_t)ft2.dwHighDateTime << 32) | (uint64_t)ft2.dwLowDateTime;
|
|
}
|
|
|
|
}
|
|
#endif
|
|
// FIX ME
|
|
return filetimestamp_invalid;
|
|
}
|
|
|
|
t_filetimestamp foobar2000_io::filetimestamp_from_system_timer()
|
|
{
|
|
return pfc::fileTimeNow();
|
|
}
|
|
|
|
void stream_reader::read_string_ex(pfc::string_base & p_out,t_size p_bytes,abort_callback & p_abort) {
|
|
const t_size expBase = 64*1024;
|
|
if (p_bytes > expBase) {
|
|
pfc::array_t<char> temp;
|
|
t_size allocWalk = expBase;
|
|
t_size done = 0;
|
|
for(;;) {
|
|
const t_size target = pfc::min_t(allocWalk, p_bytes);
|
|
temp.set_size(target);
|
|
read_object(temp.get_ptr() + done, target - done, p_abort);
|
|
if (target == p_bytes) break;
|
|
done = target;
|
|
allocWalk <<= 1;
|
|
}
|
|
p_out.set_string(temp.get_ptr(), p_bytes);
|
|
} else {
|
|
pfc::string_buffer buf(p_out, p_bytes);
|
|
read_object(buf.get_ptr(),p_bytes,p_abort);
|
|
}
|
|
}
|
|
void stream_reader::read_string(pfc::string_base & p_out,abort_callback & p_abort)
|
|
{
|
|
t_uint32 length;
|
|
read_lendian_t(length,p_abort);
|
|
read_string_ex(p_out,length,p_abort);
|
|
}
|
|
|
|
void stream_reader::read_string_raw(pfc::string_base & p_out,abort_callback & p_abort) {
|
|
enum {delta = 256};
|
|
char buffer[delta];
|
|
p_out.reset();
|
|
for(;;) {
|
|
t_size delta_done;
|
|
delta_done = read(buffer,delta,p_abort);
|
|
p_out.add_string(buffer,delta_done);
|
|
if (delta_done < delta) break;
|
|
}
|
|
}
|
|
void stream_writer::write_string(const char * p_string,t_size p_len,abort_callback & p_abort) {
|
|
t_uint32 len = pfc::downcast_guarded<t_uint32>(pfc::strlen_max(p_string,p_len));
|
|
write_lendian_t(len,p_abort);
|
|
write_object(p_string,len,p_abort);
|
|
}
|
|
|
|
void stream_writer::write_string(const char * p_string,abort_callback & p_abort) {
|
|
write_string(p_string,~0,p_abort);
|
|
}
|
|
|
|
void stream_writer::write_string_raw(const char * p_string,abort_callback & p_abort) {
|
|
write_object(p_string,strlen(p_string),p_abort);
|
|
}
|
|
|
|
void file::truncate(t_uint64 p_position,abort_callback & p_abort) {
|
|
if (p_position < get_size(p_abort)) resize(p_position,p_abort);
|
|
}
|
|
|
|
|
|
#ifdef _WIN32
|
|
namespace {
|
|
//rare/weird win32 errors that didn't make it to the main API
|
|
PFC_DECLARE_EXCEPTION(exception_io_device_not_ready, exception_io,"Device not ready");
|
|
PFC_DECLARE_EXCEPTION(exception_io_invalid_drive, exception_io_not_found,"Drive not found");
|
|
PFC_DECLARE_EXCEPTION(exception_io_win32, exception_io,"Generic win32 I/O error");
|
|
PFC_DECLARE_EXCEPTION(exception_io_buffer_overflow, exception_io,"The file name is too long");
|
|
PFC_DECLARE_EXCEPTION(exception_io_invalid_path_syntax, exception_io,"Invalid path syntax");
|
|
|
|
class exception_io_win32_ex : public exception_io_win32 {
|
|
public:
|
|
static pfc::string8 format(DWORD code) {
|
|
pfc::string8 ret;
|
|
ret << "I/O error (win32 ";
|
|
if (code & 0x80000000) {
|
|
ret << "0x" << pfc::format_hex(code, 8);
|
|
} else {
|
|
ret << "#" << (uint32_t)code;
|
|
}
|
|
ret << ")";
|
|
return ret;
|
|
}
|
|
exception_io_win32_ex(DWORD p_code) : m_msg(format(p_code)) {}
|
|
exception_io_win32_ex(const exception_io_win32_ex & p_other) {*this = p_other;}
|
|
const char * what() const throw() {return m_msg;}
|
|
private:
|
|
pfc::string8 m_msg;
|
|
};
|
|
}
|
|
|
|
PFC_NORETURN void foobar2000_io::win32_file_write_failure(DWORD p_code, const char * path) {
|
|
if (p_code == ERROR_ACCESS_DENIED) {
|
|
const DWORD attr = uGetFileAttributes(path);
|
|
if (attr != ~0 && (attr & FILE_ATTRIBUTE_READONLY) != 0) throw exception_io_denied_readonly();
|
|
}
|
|
exception_io_from_win32(p_code);
|
|
}
|
|
|
|
PFC_NORETURN void foobar2000_io::exception_io_from_win32(DWORD p_code) {
|
|
#if PFC_DEBUG
|
|
PFC_DEBUGLOG << "exception_io_from_win32: " << p_code;
|
|
#endif
|
|
//pfc::string_fixed_t<32> debugMsg; debugMsg << "Win32 I/O error #" << (t_uint32)p_code;
|
|
//TRACK_CALL_TEXT(debugMsg);
|
|
switch(p_code) {
|
|
case ERROR_ALREADY_EXISTS:
|
|
case ERROR_FILE_EXISTS:
|
|
throw exception_io_already_exists();
|
|
case ERROR_NETWORK_ACCESS_DENIED:
|
|
case ERROR_ACCESS_DENIED:
|
|
throw exception_io_denied();
|
|
case ERROR_WRITE_PROTECT:
|
|
throw exception_io_write_protected();
|
|
case ERROR_BUSY:
|
|
case ERROR_PATH_BUSY:
|
|
case ERROR_SHARING_VIOLATION:
|
|
case ERROR_LOCK_VIOLATION:
|
|
throw exception_io_sharing_violation();
|
|
case ERROR_HANDLE_DISK_FULL:
|
|
case ERROR_DISK_FULL:
|
|
throw exception_io_device_full();
|
|
case ERROR_FILE_NOT_FOUND:
|
|
case ERROR_PATH_NOT_FOUND:
|
|
throw exception_io_not_found();
|
|
case ERROR_BROKEN_PIPE:
|
|
case ERROR_NO_DATA:
|
|
throw exception_io_no_data();
|
|
case ERROR_NETWORK_UNREACHABLE:
|
|
case ERROR_NETNAME_DELETED:
|
|
throw exception_io_network_not_reachable();
|
|
case ERROR_NOT_READY:
|
|
throw exception_io_device_not_ready();
|
|
case ERROR_INVALID_DRIVE:
|
|
throw exception_io_invalid_drive();
|
|
case ERROR_CRC:
|
|
case ERROR_FILE_CORRUPT:
|
|
case ERROR_DISK_CORRUPT:
|
|
throw exception_io_file_corrupted();
|
|
case ERROR_BUFFER_OVERFLOW:
|
|
throw exception_io_buffer_overflow();
|
|
case ERROR_DISK_CHANGE:
|
|
throw exception_io_disk_change();
|
|
case ERROR_DIR_NOT_EMPTY:
|
|
throw exception_io_directory_not_empty();
|
|
case ERROR_INVALID_NAME:
|
|
throw exception_io_invalid_path_syntax();
|
|
case ERROR_NO_SYSTEM_RESOURCES:
|
|
case ERROR_NONPAGED_SYSTEM_RESOURCES:
|
|
case ERROR_PAGED_SYSTEM_RESOURCES:
|
|
case ERROR_WORKING_SET_QUOTA:
|
|
case ERROR_PAGEFILE_QUOTA:
|
|
case ERROR_COMMITMENT_LIMIT:
|
|
throw exception_io("Insufficient system resources");
|
|
case ERROR_IO_DEVICE:
|
|
throw exception_io("Device error");
|
|
case ERROR_BAD_NETPATH:
|
|
// known to be inflicted by momentary net connectivity issues - NOT the same as exception_io_not_found
|
|
throw exception_io("Network path not found");
|
|
#if FB2K_SUPPORT_TRANSACTED_FILESYSTEM
|
|
case ERROR_TRANSACTIONAL_OPEN_NOT_ALLOWED:
|
|
case ERROR_TRANSACTIONS_UNSUPPORTED_REMOTE:
|
|
case ERROR_RM_NOT_ACTIVE:
|
|
case ERROR_RM_METADATA_CORRUPT:
|
|
case ERROR_DIRECTORY_NOT_RM:
|
|
throw exception_io_transactions_unsupported();
|
|
case ERROR_TRANSACTIONAL_CONFLICT:
|
|
throw exception_io_transactional_conflict();
|
|
case ERROR_TRANSACTION_ALREADY_ABORTED:
|
|
throw exception_io_transaction_aborted();
|
|
case ERROR_EFS_NOT_ALLOWED_IN_TRANSACTION:
|
|
throw exception_io("Transacted updates of encrypted content are not supported");
|
|
#endif // FB2K_SUPPORT_TRANSACTED_FILESYSTEM
|
|
case ERROR_UNEXP_NET_ERR:
|
|
// QNAP threw this when messing with very long file paths and concurrent conversion, probably SMB daemon crashed
|
|
throw exception_io("Unexpected network error");
|
|
case ERROR_NOT_SAME_DEVICE:
|
|
throw exception_io("Source and destination must be on the same device");
|
|
case 0x80310000:
|
|
throw exception_io("Drive locked by BitLocker");
|
|
default:
|
|
throw exception_io_win32_ex(p_code);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
t_filesize file::get_size_ex(abort_callback & p_abort) {
|
|
t_filesize temp = get_size(p_abort);
|
|
if (temp == filesize_invalid) throw exception_io_no_length();
|
|
return temp;
|
|
}
|
|
|
|
void file::ensure_local() {
|
|
if (is_remote()) throw exception_io_object_is_remote();
|
|
}
|
|
|
|
void file::ensure_seekable() {
|
|
if (!can_seek()) throw exception_io_object_not_seekable();
|
|
}
|
|
|
|
bool filesystem::g_is_recognized_path(const char * p_path) {
|
|
filesystem::ptr obj;
|
|
return g_get_interface(obj,p_path);
|
|
}
|
|
|
|
t_filesize file::get_remaining(abort_callback & p_abort) {
|
|
t_filesize length = get_size_ex(p_abort);
|
|
t_filesize position = get_position(p_abort);
|
|
pfc::dynamic_assert(position <= length);
|
|
return length - position;
|
|
}
|
|
|
|
void file::probe_remaining(t_filesize bytes, abort_callback & p_abort) {
|
|
t_filesize length = get_size(p_abort);
|
|
if (length != ~0) {
|
|
t_filesize remaining = length - get_position(p_abort);
|
|
if (remaining < bytes) throw exception_io_data_truncation();
|
|
}
|
|
}
|
|
|
|
|
|
t_filesize file::g_transfer(service_ptr_t<file> p_src,service_ptr_t<file> p_dst,t_filesize p_bytes,abort_callback & p_abort) {
|
|
return g_transfer(pfc::implicit_cast<stream_reader*>(p_src.get_ptr()),pfc::implicit_cast<stream_writer*>(p_dst.get_ptr()),p_bytes,p_abort);
|
|
}
|
|
|
|
void file::g_transfer_object(service_ptr_t<file> p_src,service_ptr_t<file> p_dst,t_filesize p_bytes,abort_callback & p_abort) {
|
|
if (p_bytes > 1024) /* don't bother on small objects */
|
|
{
|
|
t_filesize srcFileSize = p_src->get_size(p_abort); // detect truncation
|
|
if (srcFileSize != ~0) {
|
|
t_filesize remaining = srcFileSize - p_src->get_position(p_abort);
|
|
if (p_bytes > remaining) throw exception_io_data_truncation();
|
|
}
|
|
|
|
t_filesize oldsize = p_dst->get_size(p_abort); // pre-resize the target file
|
|
if (oldsize != filesize_invalid) {
|
|
t_filesize newpos = p_dst->get_position(p_abort) + p_bytes;
|
|
if (newpos > oldsize) p_dst->resize(newpos ,p_abort);
|
|
}
|
|
|
|
}
|
|
g_transfer_object(pfc::implicit_cast<stream_reader*>(p_src.get_ptr()),pfc::implicit_cast<stream_writer*>(p_dst.get_ptr()),p_bytes,p_abort);
|
|
}
|
|
|
|
|
|
void foobar2000_io::generate_temp_location_for_file(pfc::string_base & p_out, const char * p_origpath,const char * p_extension,const char * p_magic) {
|
|
hasher_md5_result hash;
|
|
{
|
|
auto hasher = hasher_md5::get();
|
|
hasher_md5_state state;
|
|
hasher->initialize(state);
|
|
hasher->process(state,p_origpath,strlen(p_origpath));
|
|
hasher->process(state,p_extension,strlen(p_extension));
|
|
hasher->process(state,p_magic,strlen(p_magic));
|
|
hash = hasher->get_result(state);
|
|
}
|
|
|
|
p_out = p_origpath;
|
|
p_out.truncate(p_out.scan_filename());
|
|
p_out += "temp-";
|
|
p_out += pfc::format_hexdump(hash.m_data,sizeof(hash.m_data),"");
|
|
p_out += ".";
|
|
p_out += p_extension;
|
|
}
|
|
|
|
t_filesize file::skip_seek(t_filesize p_bytes,abort_callback & p_abort) {
|
|
const t_filesize size = get_size(p_abort);
|
|
if (size != filesize_invalid) {
|
|
const t_filesize position = get_position(p_abort);
|
|
const t_filesize toskip = pfc::min_t( p_bytes, size - position );
|
|
seek(position + toskip,p_abort);
|
|
return toskip;
|
|
} else {
|
|
this->seek_ex( p_bytes, seek_from_current, p_abort );
|
|
return p_bytes;
|
|
}
|
|
}
|
|
|
|
t_filesize file::skip(t_filesize p_bytes,abort_callback & p_abort) {
|
|
if (p_bytes > 1024 && can_seek()) {
|
|
const t_filesize size = get_size(p_abort);
|
|
if (size != filesize_invalid) {
|
|
const t_filesize position = get_position(p_abort);
|
|
const t_filesize toskip = pfc::min_t( p_bytes, size - position );
|
|
seek(position + toskip,p_abort);
|
|
return toskip;
|
|
}
|
|
}
|
|
return stream_reader::skip(p_bytes,p_abort);
|
|
}
|
|
|
|
bool foobar2000_io::is_native_filesystem( const char * p_fspath ) {
|
|
return _extract_native_path_ptr( p_fspath );
|
|
}
|
|
|
|
bool foobar2000_io::_extract_native_path_ptr(const char * & p_fspath) {
|
|
static const char header[] = "file://"; static const t_size headerLen = 7;
|
|
if (strncmp(p_fspath,header,headerLen) != 0) return false;
|
|
p_fspath += headerLen;
|
|
return true;
|
|
}
|
|
bool foobar2000_io::extract_native_path(const char * p_fspath,pfc::string_base & p_native) {
|
|
if (!_extract_native_path_ptr(p_fspath)) return false;
|
|
p_native = p_fspath;
|
|
return true;
|
|
}
|
|
|
|
bool foobar2000_io::extract_native_path_ex(const char * p_fspath, pfc::string_base & p_native) {
|
|
if (!_extract_native_path_ptr(p_fspath)) return false;
|
|
if (p_fspath[0] != '\\' || p_fspath[1] != '\\') {
|
|
p_native = "\\\\?\\";
|
|
p_native += p_fspath;
|
|
} else {
|
|
p_native = p_fspath;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool foobar2000_io::extract_native_path_archive_aware(const char * in, pfc::string_base & out) {
|
|
if (foobar2000_io::extract_native_path(in, out)) return true;
|
|
if (archive_impl::g_is_unpack_path(in)) {
|
|
pfc::string8 arc, dummy;
|
|
if (archive_impl::g_parse_unpack_path(in, arc, dummy)) {
|
|
return foobar2000_io::extract_native_path(arc, out);
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
pfc::string stream_reader::read_string(abort_callback & p_abort) {
|
|
t_uint32 len;
|
|
read_lendian_t(len,p_abort);
|
|
return read_string_ex(len,p_abort);
|
|
}
|
|
pfc::string stream_reader::read_string_ex(t_size p_len,abort_callback & p_abort) {
|
|
pfc::rcptr_t<pfc::string8> temp; temp.new_t();
|
|
read_object(temp->lock_buffer(p_len),p_len,p_abort);
|
|
temp->unlock_buffer();
|
|
return pfc::string::t_data(temp);
|
|
}
|
|
|
|
|
|
void filesystem::remove_directory_content(const char * path, abort_callback & abort) {
|
|
class myCallback : public directory_callback {
|
|
public:
|
|
bool on_entry(filesystem * p_owner,abort_callback & p_abort,const char * p_url,bool p_is_subdirectory,const t_filestats & p_stats) {
|
|
if (p_is_subdirectory) p_owner->list_directory(p_url, *this, p_abort);
|
|
try {
|
|
p_owner->remove(p_url, p_abort);
|
|
} catch(exception_io_not_found) {}
|
|
return true;
|
|
}
|
|
};
|
|
myCallback cb;
|
|
list_directory(path, cb, abort);
|
|
}
|
|
void filesystem::remove_object_recur(const char * path, abort_callback & abort) {
|
|
try {
|
|
remove_directory_content(path, abort);
|
|
} catch(exception_io_not_found) {}
|
|
remove(path, abort);
|
|
}
|
|
|
|
void filesystem::g_remove_object_recur_timeout(const char * path, double timeout, abort_callback & abort) {
|
|
FB2K_RETRY_FILE_MOVE( g_remove_object_recur(path, abort), abort, timeout );
|
|
}
|
|
|
|
void filesystem::g_remove_object_recur(const char * path, abort_callback & abort) {
|
|
g_get_interface(path)->remove_object_recur(path, abort);
|
|
}
|
|
|
|
void foobar2000_io::purgeOldFiles(const char * directory, t_filetimestamp period, abort_callback & abort) {
|
|
|
|
class myCallback : public directory_callback {
|
|
public:
|
|
myCallback(t_filetimestamp period) : m_base(filetimestamp_from_system_timer() - period) {}
|
|
bool on_entry(filesystem * p_owner,abort_callback & p_abort,const char * p_url,bool p_is_subdirectory,const t_filestats & p_stats) {
|
|
if (!p_is_subdirectory && p_stats.m_timestamp < m_base) {
|
|
try {
|
|
filesystem::g_remove_timeout(p_url, 1, p_abort);
|
|
} catch(exception_io_not_found) {}
|
|
}
|
|
return true;
|
|
}
|
|
private:
|
|
const t_filetimestamp m_base;
|
|
};
|
|
|
|
myCallback cb(period);
|
|
filesystem::g_list_directory(directory, cb, abort);
|
|
}
|
|
|
|
void stream_reader::read_string_nullterm( pfc::string_base & out, abort_callback & abort ) {
|
|
enum { bufCount = 256 };
|
|
char buffer[bufCount];
|
|
out.reset();
|
|
size_t w = 0;
|
|
for(;;) {
|
|
char & c = buffer[w];
|
|
this->read_object( &c, 1, abort );
|
|
if (c == 0) {
|
|
out.add_string( buffer, w ); break;
|
|
}
|
|
if (++w == bufCount ) {
|
|
out.add_string( buffer, bufCount ); w = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
t_filesize stream_reader::skip_till_eof(abort_callback & abort) {
|
|
t_filesize atOnce = 1024 * 1024;
|
|
t_filesize done = 0;
|
|
for (;; ) {
|
|
abort.check();
|
|
t_filesize did = this->skip(atOnce, abort);
|
|
done += did;
|
|
if (did != atOnce) break;
|
|
}
|
|
return done;
|
|
}
|
|
|
|
uint8_t stream_reader::read_byte( abort_callback & abort ) {
|
|
uint8_t b;
|
|
read_object(&b, 1, abort );
|
|
return b;
|
|
}
|
|
|
|
bool foobar2000_io::matchContentType(const char * fullString, const char * ourType) {
|
|
t_size lim = pfc::string_find_first(fullString, ';');
|
|
if (lim != ~0) {
|
|
while(lim > 0 && fullString[lim-1] == ' ') --lim;
|
|
}
|
|
return pfc::stricmp_ascii_ex(fullString,lim, ourType, ~0) == 0;
|
|
}
|
|
|
|
const char * foobar2000_io::contentTypeFromExtension( const char * ext ) {
|
|
if ( pfc::stringEqualsI_ascii( ext, "mp3" ) ) return "audio/mpeg";
|
|
if ( pfc::stringEqualsI_ascii( ext, "flac" ) ) return "audio/flac";
|
|
if ( pfc::stringEqualsI_ascii( ext, "mp4" ) ) return "application/mp4"; // We don't know if it's audio-only or other.
|
|
if ( pfc::stringEqualsI_ascii( ext, "m4a" ) ) return "audio/mp4";
|
|
if ( pfc::stringEqualsI_ascii( ext, "mpc" ) ) return "audio/musepack";
|
|
if ( pfc::stringEqualsI_ascii( ext, "ogg" ) ) return "audio/ogg";
|
|
if ( pfc::stringEqualsI_ascii( ext, "opus" ) ) return "audio/opus";
|
|
if ( pfc::stringEqualsI_ascii( ext, "wav" ) ) return "audio/vnd.wave";
|
|
if ( pfc::stringEqualsI_ascii( ext, "wv" ) ) return "audio/wavpack";
|
|
if ( pfc::stringEqualsI_ascii( ext, "txt" ) || pfc::stringEqualsI_ascii( ext, "cue" ) || pfc::stringEqualsI_ascii( ext, "log" ) ) return "text/plain";
|
|
return "application/binary";
|
|
}
|
|
|
|
const char * foobar2000_io::extensionFromContentType( const char * contentType ) {
|
|
if (matchContentType_MP3( contentType )) return "mp3";
|
|
if (matchContentType_FLAC( contentType )) return "flac";
|
|
if (matchContentType_MP4audio( contentType)) return "m4a";
|
|
if (matchContentType_MP4( contentType)) return "mp4";
|
|
if (matchContentType_Musepack( contentType )) return "mpc";
|
|
if (matchContentType_Ogg( contentType )) return "ogg";
|
|
if (matchContentType_Opus( contentType )) return "opus";
|
|
if (matchContentType_WAV( contentType )) return "wav";
|
|
if (matchContentType_WavPack( contentType )) return "wv";
|
|
if (matchContentType(contentType, "image/jpeg")) return "jpg";
|
|
if (matchContentType(contentType, "image/png")) return "png";
|
|
return "";
|
|
}
|
|
|
|
bool foobar2000_io::matchContentType_MP3( const char * type) {
|
|
return matchContentType(type,"audio/mp3") || matchContentType(type,"audio/mpeg") || matchContentType(type,"audio/mpg") || matchContentType(type,"audio/x-mp3") || matchContentType(type,"audio/x-mpeg") || matchContentType(type,"audio/x-mpg");
|
|
}
|
|
bool foobar2000_io::matchContentType_MP4( const char * type ) {
|
|
return matchContentType(type, "audio/mp4") || matchContentType(type, "audio/x-mp4")
|
|
|| matchContentType(type, "video/mp4") || matchContentType(type, "video/x-mp4")
|
|
|| matchContentType(type, "application/mp4") || matchContentType(type, "application/x-mp4");
|
|
|
|
}
|
|
bool foobar2000_io::matchContentType_MP4audio( const char * type ) {
|
|
return matchContentType(type, "audio/mp4") || matchContentType(type, "audio/x-mp4");
|
|
}
|
|
bool foobar2000_io::matchContentType_Ogg( const char * type) {
|
|
return matchContentType(type, "application/ogg") || matchContentType(type, "application/x-ogg") || matchContentType(type, "audio/ogg") || matchContentType(type, "audio/x-ogg");
|
|
}
|
|
bool foobar2000_io::matchContentType_Opus( const char * type) {
|
|
return matchContentType(type, "audio/opus") || matchContentType(type, "audio/x-opus");
|
|
}
|
|
bool foobar2000_io::matchContentType_WAV( const char * type ) {
|
|
return matchContentType(type, "audio/vnd.wave" ) || matchContentType(type, "audio/wav") || matchContentType(type, "audio/wave") || matchContentType(type, "audio/x-wav") || matchContentType(type, "audio/x-wave");
|
|
}
|
|
bool foobar2000_io::matchContentType_FLAC( const char * type) {
|
|
return matchContentType(type, "audio/flac") || matchContentType(type, "audio/x-flac") || matchContentType(type, "application/flac") || matchContentType(type, "application/x-flac");
|
|
}
|
|
bool foobar2000_io::matchContentType_WavPack( const char * type) {
|
|
return matchContentType( type, "audio/wavpack" ) || matchContentType( type, "audio/x-wavpack");
|
|
}
|
|
bool foobar2000_io::matchContentType_Musepack( const char * type) {
|
|
return matchContentType(type,"audio/musepack") || matchContentType(type,"audio/x-musepack");
|
|
}
|
|
|
|
const char * foobar2000_io::afterProtocol( const char * fullString ) {
|
|
const char * s = strstr( fullString, "://" );
|
|
if ( s != nullptr ) return s + 3;
|
|
s = strchr(fullString, ':' );
|
|
if ( s != nullptr && s[1] != '\\' && s[1] != 0 ) return s + 1;
|
|
PFC_ASSERT(!"Should not get here");
|
|
return fullString;
|
|
}
|
|
|
|
bool foobar2000_io::matchProtocol(const char * fullString, const char * protocolName) {
|
|
const t_size len = strlen(protocolName);
|
|
if (pfc::stricmp_ascii_ex(fullString, len, protocolName, len) != 0) return false;
|
|
return fullString[len] == ':' && fullString[len+1] == '/' && fullString[len+2] == '/';
|
|
}
|
|
void foobar2000_io::substituteProtocol(pfc::string_base & out, const char * fullString, const char * protocolName) {
|
|
const char * base = strstr(fullString, "://");
|
|
if (base) {
|
|
out = protocolName; out << base;
|
|
} else {
|
|
PFC_ASSERT(!"Should not get here");
|
|
out = fullString;
|
|
}
|
|
}
|
|
|
|
void filesystem::move_overwrite(const char * src, const char * dst, abort_callback & abort) {
|
|
{
|
|
filesystem_v2::ptr v2;
|
|
if (v2 &= this) {
|
|
v2->move_overwrite(src, dst, abort); return;
|
|
}
|
|
}
|
|
try {
|
|
this->remove(dst, abort);
|
|
} catch (exception_io_not_found) {}
|
|
this->move(src, dst, abort);
|
|
}
|
|
|
|
void filesystem::replace_file(const char * src, const char * dst, abort_callback & abort) {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) {
|
|
v2->replace_file( src, dst, abort ); return;
|
|
}
|
|
move_overwrite( src, dst, abort );
|
|
}
|
|
|
|
void filesystem::make_directory(const char * path, abort_callback & abort, bool * didCreate) {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) {
|
|
v2->make_directory( path, abort, didCreate );
|
|
return;
|
|
}
|
|
bool rv = false;
|
|
try {
|
|
create_directory( path, abort );
|
|
rv = true;
|
|
} catch(exception_io_already_exists) {
|
|
}
|
|
if (didCreate != nullptr) * didCreate = rv;
|
|
}
|
|
|
|
bool filesystem::make_directory_check(const char * path, abort_callback & abort) {
|
|
bool rv = false;
|
|
make_directory(path, abort, &rv);
|
|
return rv;
|
|
}
|
|
|
|
bool filesystem::directory_exists(const char * path, abort_callback & abort) {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) {
|
|
return v2->directory_exists( path, abort );
|
|
}
|
|
try {
|
|
directory_callback_dummy cb;
|
|
list_directory(path, cb, abort);
|
|
return true;
|
|
} catch (exception_io const &) { return false; }
|
|
}
|
|
bool filesystem::file_exists(const char * path, abort_callback & abort) {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) {
|
|
return v2->file_exists( path, abort );
|
|
}
|
|
try {
|
|
t_filestats stats; bool writable;
|
|
get_stats(path, stats, writable, abort );
|
|
return true;
|
|
} catch(exception_io) { return false; }
|
|
}
|
|
|
|
char filesystem::pathSeparator() {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) return v2->pathSeparator();
|
|
return '/';
|
|
}
|
|
|
|
void filesystem::extract_filename_ext(const char * path, pfc::string_base & outFN) {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) {
|
|
v2->extract_filename_ext( path, outFN );
|
|
return;
|
|
}
|
|
outFN = pfc::filename_ext_v2( path );
|
|
}
|
|
|
|
bool filesystem::get_parent_helper( const char * path, char separator, pfc::string_base & out ) {
|
|
auto proto = path;
|
|
path = afterProtocol(path);
|
|
|
|
auto sep_ptr = strrchr( path, separator );
|
|
if ( sep_ptr == path ) return false;
|
|
if ( sep_ptr >= path + 1 && sep_ptr[-1] == separator ) return false;
|
|
|
|
out.set_string(proto, path - proto);
|
|
out.add_string(path, sep_ptr - path);
|
|
return true;
|
|
}
|
|
|
|
bool filesystem::get_parent_path(const char * path, pfc::string_base & out) {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) {
|
|
return v2->get_parent_path(path, out);
|
|
}
|
|
return get_parent_helper( path, '/', out );
|
|
}
|
|
|
|
void filesystem::read_whole_file(const char * path, mem_block_container & out, pfc::string_base & outContentType, size_t maxBytes, abort_callback & abort) {
|
|
filesystem_v2::ptr v2;
|
|
if ( v2 &= this ) {
|
|
v2->read_whole_file( path, out, outContentType, maxBytes, abort );
|
|
return;
|
|
}
|
|
read_whole_file_fallback(path, out, outContentType, maxBytes, abort);
|
|
}
|
|
|
|
void filesystem::read_whole_file_fallback(const char * path, mem_block_container & out, pfc::string_base & outContentType, size_t maxBytes, abort_callback & abort) {
|
|
auto f = this->openRead( path, abort, 0 );
|
|
if (!f->get_content_type(outContentType)) outContentType = "";
|
|
auto s64 = f->get_size( abort );
|
|
if ( s64 == filesize_invalid ) {
|
|
// unknown length, perform streamed read
|
|
size_t done = 0, alloc = 0;
|
|
|
|
while(alloc < maxBytes ) {
|
|
if ( alloc == 0 ) alloc = 4096;
|
|
else {
|
|
size_t next = alloc * 2;
|
|
if ( next <= alloc ) throw exception_io_data();
|
|
alloc = next;
|
|
}
|
|
if ( alloc > maxBytes ) alloc = maxBytes;
|
|
|
|
out.set_size( alloc );
|
|
size_t delta = alloc - done;
|
|
size_t deltaGot = f->read( (uint8_t*) out.get_ptr() + done, delta, abort );
|
|
PFC_ASSERT( deltaGot <= delta );
|
|
done += deltaGot;
|
|
if ( deltaGot != delta ) {
|
|
out.set_size( done ); return;
|
|
}
|
|
}
|
|
// maxbytes reached
|
|
PFC_ASSERT( done == maxBytes );
|
|
// corner case check
|
|
if ( f->skip(1, abort) != 0 ) throw exception_io_data();
|
|
} else if ( s64 > maxBytes ) {
|
|
throw exception_io_data();
|
|
} else {
|
|
size_t s = (size_t) s64;
|
|
out.set_size( s );
|
|
if (s > 0) f->read_object( out.get_ptr(), s, abort);
|
|
}
|
|
}
|
|
|
|
bool filesystem::is_transacted() {
|
|
filesystem_transacted::ptr p;
|
|
return ( p &= this );
|
|
}
|
|
|
|
void filesystem::rewrite_file(const char * path, abort_callback & abort, double opTimeout, std::function<void(file::ptr) > worker) {
|
|
if ( this->is_transacted() ) {
|
|
auto f = this->openWriteNew( path, abort, opTimeout );
|
|
worker(f);
|
|
} else {
|
|
pfc::string_formatter temp(path); temp << ".new.tmp";
|
|
try {
|
|
{
|
|
auto f = this->openWriteNew( temp, abort, opTimeout );
|
|
worker(f);
|
|
f->flushFileBuffers_( abort );
|
|
}
|
|
|
|
retryOnSharingViolation(opTimeout, abort, [&] {
|
|
this->replace_file(temp, path, abort);
|
|
});
|
|
|
|
} catch(...) {
|
|
try {
|
|
retryOnSharingViolation(opTimeout, abort, [&] { this->remove(temp, fb2k::noAbort); } );
|
|
} catch(...) {}
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
|
|
void filesystem::rewrite_directory(const char * path, abort_callback & abort, double opTimeout, std::function<void(const char *) > worker) {
|
|
if ( this->is_transacted() ) {
|
|
// so simple
|
|
if ( ! this->make_directory_check( path, abort) ) {
|
|
retryFileDelete(opTimeout, abort, [&] { this->remove_directory_content(path, abort); });
|
|
}
|
|
worker( path );
|
|
} else {
|
|
// so complex
|
|
pfc::string8 fnNew( path ); fnNew += ".new.tmp";
|
|
pfc::string8 fnOld( path ); fnOld += ".old.tmp";
|
|
|
|
if ( !this->make_directory_check( fnNew, abort ) ) {
|
|
// folder.new folder already existed? clear contents
|
|
try {
|
|
retryFileDelete(opTimeout, abort, [&] { this->remove_directory_content(fnNew, abort); });
|
|
} catch(exception_io_not_found) {}
|
|
}
|
|
|
|
// write to folder.new
|
|
worker( fnNew );
|
|
|
|
bool haveOld = false;
|
|
if ( directory_exists( path, abort ) ) {
|
|
// move folder to folder.old
|
|
if (this->directory_exists(fnOld, abort)) {
|
|
try {
|
|
retryFileDelete(opTimeout, abort, [&] { this->remove_object_recur(fnOld, abort); });
|
|
} catch (exception_io_not_found) {}
|
|
}
|
|
try {
|
|
retryFileMove(opTimeout, abort, [&] { this->move( path, fnOld, abort ); } ) ;
|
|
haveOld = true;
|
|
} catch(exception_io_not_found) {}
|
|
}
|
|
|
|
// move folder.new to folder
|
|
retryFileMove( opTimeout, abort, [&] {
|
|
this->move( fnNew, path, abort );
|
|
} );
|
|
|
|
if ( haveOld ) {
|
|
// delete folder.old if we made one
|
|
try {
|
|
retryFileDelete( opTimeout, abort, [&] { this->remove_object_recur( fnOld, abort); } );
|
|
} catch (exception_io_not_found) {}
|
|
}
|
|
}
|
|
}
|
|
|
|
void filesystem_v2::list_directory(const char * p_path, directory_callback & p_out, abort_callback & p_abort) {
|
|
list_directory_ex(p_path, p_out, listMode::filesAndFolders | listMode::hidden, p_abort);
|
|
}
|
|
|
|
void filesystem_v2::extract_filename_ext(const char * path, pfc::string_base & outFN) {
|
|
outFN = pfc::filename_ext_v2(path, this->pathSeparator() );
|
|
}
|
|
|
|
bool filesystem_v2::get_parent_path(const char * path, pfc::string_base & out) {
|
|
return get_parent_helper(path, pathSeparator(), out);
|
|
}
|
|
|
|
void filesystem_v2::replace_file(const char * src, const char * dst, abort_callback & abort) {
|
|
this->move_overwrite( src, dst, abort );
|
|
}
|
|
|
|
void filesystem_v2::read_whole_file(const char * path, mem_block_container & out, pfc::string_base & outContentType, size_t maxBytes, abort_callback & abort) {
|
|
read_whole_file_fallback( path, out, outContentType, maxBytes, abort );
|
|
}
|
|
|
|
bool filesystem_v2::make_directory_check(const char * path, abort_callback & abort) {
|
|
bool rv = false;
|
|
make_directory(path, abort, &rv);
|
|
return rv;
|
|
}
|
|
|
|
|
|
filesystem_transacted::ptr filesystem_transacted::create( const char * pathFor ) {
|
|
service_enum_t<filesystem_transacted_entry> e;
|
|
filesystem_transacted_entry::ptr p;
|
|
while(e.next(p)) {
|
|
if ( p->is_our_path( pathFor ) ) {
|
|
auto ret = p->create(pathFor);
|
|
if (ret.is_valid()) return ret;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
bool filesystem::commit_if_transacted(abort_callback &abort) {
|
|
bool rv = false;
|
|
filesystem_transacted::ptr t;
|
|
if ( t &= this ) {
|
|
t->commit( abort ); rv = true;
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
t_filestats filesystem::get_stats(const char * path, abort_callback & abort) {
|
|
t_filestats s; bool dummy;
|
|
this->get_stats(path, s, dummy, abort);
|
|
return s;
|
|
}
|
|
|
|
bool file_dynamicinfo_v2::get_dynamic_info(class file_info & p_out) {
|
|
t_filesize dummy = 0;
|
|
return this->get_dynamic_info_v2(p_out, dummy);
|
|
}
|
|
|
|
void file::flushFileBuffers_(abort_callback&a) {
|
|
file_lowLevelIO::ptr f;
|
|
if ( f &= this ) f->flushFileBuffers(a);
|
|
}
|
|
|
|
size_t file::lowLevelIO_(const GUID & guid, size_t arg1, void * arg2, size_t arg2size, abort_callback & abort) {
|
|
size_t retval = 0;
|
|
file_lowLevelIO::ptr f;
|
|
if (f &= this) retval = f->lowLevelIO(guid, arg1, arg2, arg2size, abort);
|
|
return retval;
|
|
}
|
|
|
|
bool file_lowLevelIO::flushFileBuffers(abort_callback & abort) {
|
|
return this->lowLevelIO( guid_flushFileBuffers, 0, nullptr, 0, abort) != 0;
|
|
}
|
|
|
|
bool file_lowLevelIO::getFileTimes(filetimes_t & out, abort_callback & a) {
|
|
return this->lowLevelIO(guid_getFileTimes, 0, &out, sizeof(out), a) != 0;
|
|
}
|
|
|
|
bool file_lowLevelIO::setFileTimes(filetimes_t const & in, abort_callback & a) {
|
|
return this->lowLevelIO(guid_setFileTimes, 0, (void*)&in, sizeof(in), a) != 0;
|
|
}
|
|
|
|
bool file::g_copy_creation_time(service_ptr_t<file> from, service_ptr_t<file> to, abort_callback& a) {
|
|
file_lowLevelIO::ptr llFrom, llTo;
|
|
bool rv = false;
|
|
if (llTo &= to) {
|
|
if (llFrom &= from) {
|
|
file_lowLevelIO::filetimes_t filetimes;
|
|
if (llFrom->getFileTimes(filetimes, a)) {
|
|
if (filetimes.creation != filetimestamp_invalid) {
|
|
file_lowLevelIO::filetimes_t ft2;
|
|
ft2.creation = filetimes.creation;
|
|
rv = llTo->setFileTimes(ft2, a);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return rv;
|
|
}
|
|
bool file::g_copy_timestamps(file::ptr from, file::ptr to, abort_callback& a) {
|
|
file_lowLevelIO::ptr llFrom, llTo;
|
|
if ( llTo &= to ) {
|
|
if (llFrom &= from) {
|
|
file_lowLevelIO::filetimes_t filetimes = {};
|
|
if (llFrom->getFileTimes(filetimes, a)) {
|
|
return llTo->setFileTimes(filetimes, a);
|
|
}
|
|
}
|
|
file_lowLevelIO::filetimes_t filetimes = {};
|
|
filetimes.lastWrite = from->get_timestamp(a);
|
|
if ( filetimes.lastWrite != filetimestamp_invalid ) {
|
|
return llTo->setFileTimes(filetimes, a);
|
|
}
|
|
}
|
|
return false;
|
|
}
|