premiere-libtorrent/src/file.cpp

1344 lines
33 KiB
C++
Raw Normal View History

2004-01-16 03:57:45 +01:00
/*
Copyright (c) 2003-2016, Arvid Norberg
2004-01-16 03:57:45 +01:00
All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions
are met:
* Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in
the documentation and/or other materials provided with the distribution.
* Neither the name of the author nor the names of its
contributors may be used to endorse or promote products derived
from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
POSSIBILITY OF SUCH DAMAGE.
*/
#include "libtorrent/config.hpp"
#include "libtorrent/aux_/disable_warnings_push.hpp"
2015-08-16 18:17:23 +02:00
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-macros"
#endif
#ifdef __clang__
#pragma clang diagnostic push
2016-11-19 20:19:35 +01:00
#pragma clang diagnostic ignored "-Wunknown-pragmas"
2015-08-16 18:17:23 +02:00
#pragma clang diagnostic ignored "-Wunused-macros"
2016-11-19 20:19:35 +01:00
#pragma clang diagnostic ignored "-Wreserved-id-macro"
2015-08-16 18:17:23 +02:00
#endif
// these defines are just in case the system we're on needs them for 64 bit file
// support
2013-11-08 09:10:22 +01:00
#define _FILE_OFFSET_BITS 64
#define _LARGE_FILES 1
#ifndef TORRENT_WINDOWS
#include <sys/uio.h> // for iovec
#else
namespace {
struct iovec
{
void* iov_base;
std::size_t iov_len;
};
} // anonymous namespace
#endif
2014-02-15 03:45:04 +01:00
// on mingw this is necessary to enable 64-bit time_t, specifically used for
// the stat struct. Without this, modification times returned by stat may be
// incorrect and consistently fail resume data
#ifndef __MINGW_USE_VC2005_COMPAT
# define __MINGW_USE_VC2005_COMPAT
#endif
2015-08-16 18:17:23 +02:00
#ifdef __clang__
#pragma clang diagnostic pop
#endif
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
#include "libtorrent/aux_/disable_warnings_pop.hpp"
#include "libtorrent/aux_/alloca.hpp"
2014-07-06 21:18:00 +02:00
#include "libtorrent/file.hpp"
#include "libtorrent/aux_/path.hpp"
#include "libtorrent/string_util.hpp"
#include "libtorrent/aux_/max_path.hpp" // for TORRENT_MAX_PATH
2014-07-06 21:18:00 +02:00
#include <cstring>
2014-07-06 21:18:00 +02:00
// for convert_to_wstring and convert_to_native
#include "libtorrent/aux_/escape_string.hpp"
2014-07-06 21:18:00 +02:00
#include "libtorrent/assert.hpp"
#include "libtorrent/aux_/throw.hpp"
#include "libtorrent/aux_/disable_warnings_push.hpp"
2012-08-21 23:54:07 +02:00
#include <sys/stat.h>
#include <climits> // for IOV_MAX
2012-08-21 23:54:07 +02:00
#ifdef TORRENT_WINDOWS
2004-09-24 12:50:03 +02:00
// windows part
2005-08-18 13:20:17 +02:00
2011-02-02 08:41:32 +01:00
#ifndef PtrToPtr64
#define PtrToPtr64(x) (x)
#endif
#include "libtorrent/utf8.hpp"
#include "libtorrent/aux_/win_util.hpp"
2009-11-23 00:55:54 +01:00
#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif
#include <windows.h>
#include <winioctl.h>
2012-08-21 23:54:07 +02:00
#ifndef TORRENT_MINGW
#include <direct.h> // for _getcwd, _mkdir
2012-08-21 23:54:07 +02:00
#else
#include <dirent.h>
#endif
#include <sys/types.h>
2004-03-31 01:55:52 +02:00
#else
// posix part
2004-03-31 01:55:52 +02:00
#include <unistd.h>
2004-04-02 12:43:37 +02:00
#include <sys/types.h>
#include <cerrno>
#include <dirent.h>
2009-09-05 17:02:49 +02:00
#ifdef TORRENT_LINUX
// linux specifics
2009-09-05 17:02:49 +02:00
#include <sys/ioctl.h>
2013-11-27 17:58:02 +01:00
#ifdef TORRENT_ANDROID
#include <sys/syscall.h>
#define lseek lseek64
#define pread pread64
#define pwrite pwrite64
#define ftruncate ftruncate64
2013-11-27 17:58:02 +01:00
#endif
2011-02-14 05:12:26 +01:00
#elif defined __APPLE__ && defined __MACH__ && MAC_OS_X_VERSION_MIN_REQUIRED >= 1050
// mac specifics
#include <copyfile.h>
#endif
2004-03-31 01:55:52 +02:00
2005-08-23 11:59:56 +02:00
// make sure the _FILE_OFFSET_BITS define worked
// on this platform. It's supposed to make file
// related functions support 64-bit offsets.
2009-09-05 17:02:49 +02:00
// this test makes sure lseek() returns a type
// at least 64 bits wide
static_assert(sizeof(lseek(0, 0, 0)) >= 8, "64 bit file operations are required");
2005-08-23 11:59:56 +02:00
#endif // posix part
2004-03-31 01:55:52 +02:00
#include "libtorrent/aux_/disable_warnings_pop.hpp"
2014-07-06 21:18:00 +02:00
#if TORRENT_USE_PREADV
# if defined TORRENT_WINDOWS
namespace {
2014-07-06 21:18:00 +02:00
// wrap the windows function in something that looks
// like preadv() and pwritev()
2005-08-23 11:59:56 +02:00
// windows only lets us wait for 64 handles at a time, so this function makes
// sure we wait for all of them, partially in sequence
DWORD wait_for_multiple_objects(int num_handles, HANDLE* h)
{
2016-01-20 03:21:00 +01:00
int batch_size = (std::min)(num_handles, MAXIMUM_WAIT_OBJECTS);
while (WaitForMultipleObjects(batch_size, h, TRUE, INFINITE) != WAIT_FAILED)
{
h += batch_size;
num_handles -= batch_size;
2016-01-20 03:21:00 +01:00
batch_size = (std::min)(num_handles, MAXIMUM_WAIT_OBJECTS);
if (batch_size <= 0) return WAIT_OBJECT_0;
}
return WAIT_FAILED;
}
int preadv(HANDLE fd, ::iovec const* bufs, int num_bufs, std::int64_t file_offset)
2014-07-06 21:18:00 +02:00
{
TORRENT_ALLOCA(ol, OVERLAPPED, num_bufs);
std::memset(ol.data(), 0, sizeof(OVERLAPPED) * num_bufs);
2014-07-06 21:18:00 +02:00
TORRENT_ALLOCA(h, HANDLE, num_bufs);
2014-07-06 21:18:00 +02:00
for (int i = 0; i < num_bufs; ++i)
{
ol[i].OffsetHigh = file_offset >> 32;
ol[i].Offset = file_offset & 0xffffffff;
ol[i].hEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr);
2014-07-06 21:18:00 +02:00
h[i] = ol[i].hEvent;
if (h[i] == nullptr)
2014-07-06 21:18:00 +02:00
{
// we failed to create the event, roll-back and return an error
for (int j = 0; j < i; ++j) CloseHandle(h[i]);
return -1;
}
file_offset += bufs[i].iov_len;
}
int ret = 0;
for (int i = 0; i < num_bufs; ++i)
{
DWORD num_read;
if (ReadFile(fd, bufs[i].iov_base, DWORD(bufs[i].iov_len), &num_read, &ol[i]) == FALSE
2014-07-06 21:18:00 +02:00
&& GetLastError() != ERROR_IO_PENDING
#ifdef ERROR_CANT_WAIT
&& GetLastError() != ERROR_CANT_WAIT
#endif
)
{
ret = -1;
goto done;
}
}
if (wait_for_multiple_objects(int(h.size()), h.data()) == WAIT_FAILED)
{
ret = -1;
2016-01-13 07:07:27 +01:00
goto done;
}
2014-07-06 21:18:00 +02:00
for (auto& o : ol)
2014-07-06 21:18:00 +02:00
{
if (WaitForSingleObject(o.hEvent, INFINITE) == WAIT_FAILED)
{
ret = -1;
break;
}
2014-07-06 21:18:00 +02:00
DWORD num_read;
if (GetOverlappedResult(fd, &o, &num_read, FALSE) == FALSE)
2014-07-06 21:18:00 +02:00
{
#ifdef ERROR_CANT_WAIT
TORRENT_ASSERT(GetLastError() != ERROR_CANT_WAIT);
#endif
ret = -1;
break;
}
ret += num_read;
}
done:
for (auto hnd : h)
CloseHandle(hnd);
2014-07-06 21:18:00 +02:00
return ret;
}
int pwritev(HANDLE fd, ::iovec const* bufs, int num_bufs, std::int64_t file_offset)
2014-07-06 21:18:00 +02:00
{
TORRENT_ALLOCA(ol, OVERLAPPED, num_bufs);
std::memset(ol.data(), 0, sizeof(OVERLAPPED) * num_bufs);
2014-07-06 21:18:00 +02:00
TORRENT_ALLOCA(h, HANDLE, num_bufs);
2014-07-06 21:18:00 +02:00
for (int i = 0; i < num_bufs; ++i)
{
ol[i].OffsetHigh = file_offset >> 32;
ol[i].Offset = file_offset & 0xffffffff;
ol[i].hEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr);
2014-07-06 21:18:00 +02:00
h[i] = ol[i].hEvent;
if (h[i] == nullptr)
2014-07-06 21:18:00 +02:00
{
// we failed to create the event, roll-back and return an error
for (int j = 0; j < i; ++j) CloseHandle(h[i]);
return -1;
}
file_offset += bufs[i].iov_len;
}
int ret = 0;
for (int i = 0; i < num_bufs; ++i)
{
DWORD num_written;
if (WriteFile(fd, bufs[i].iov_base, DWORD(bufs[i].iov_len), &num_written, &ol[i]) == FALSE
2014-07-06 21:18:00 +02:00
&& GetLastError() != ERROR_IO_PENDING
#ifdef ERROR_CANT_WAIT
&& GetLastError() != ERROR_CANT_WAIT
#endif
)
{
ret = -1;
goto done;
}
}
if (wait_for_multiple_objects(int(h.size()), h.data()) == WAIT_FAILED)
{
ret = -1;
2016-01-13 07:07:27 +01:00
goto done;
}
2014-07-06 21:18:00 +02:00
for (auto& o : ol)
2014-07-06 21:18:00 +02:00
{
if (WaitForSingleObject(o.hEvent, INFINITE) == WAIT_FAILED)
{
ret = -1;
break;
}
2014-07-06 21:18:00 +02:00
DWORD num_written;
if (GetOverlappedResult(fd, &o, &num_written, FALSE) == FALSE)
2014-07-06 21:18:00 +02:00
{
#ifdef ERROR_CANT_WAIT
TORRENT_ASSERT(GetLastError() != ERROR_CANT_WAIT);
#endif
ret = -1;
break;
}
ret += num_written;
}
done:
for (auto hnd : h)
CloseHandle(hnd);
2014-07-06 21:18:00 +02:00
return ret;
}
}
# else
# undef _BSD_SOURCE
# define _BSD_SOURCE // deprecated since glibc 2.20
# undef _DEFAULT_SOURCE
# define _DEFAULT_SOURCE
2014-07-06 21:18:00 +02:00
# include <sys/uio.h>
# endif
#endif
2004-04-05 00:15:31 +02:00
namespace libtorrent {
static_assert(!(open_mode::rw_mask & open_mode::sparse), "internal flags error");
static_assert(!(open_mode::rw_mask & open_mode::attribute_mask), "internal flags error");
static_assert(!(open_mode::sparse & open_mode::attribute_mask), "internal flags error");
directory::directory(std::string const& path, error_code& ec)
: m_done(false)
{
ec.clear();
std::string p{ path };
#ifdef TORRENT_WINDOWS
// the path passed to FindFirstFile() must be
// a pattern
p.append((!p.empty() && p.back() != '\\') ? "\\*" : "*");
#else
// the path passed to opendir() may not
// end with a /
if (!p.empty() && p.back() == '/')
p.pop_back();
#endif
native_path_string f = convert_to_native_path_string(p);
#ifdef TORRENT_WINDOWS
m_inode = 0;
m_handle = FindFirstFileW(f.c_str(), &m_fd);
if (m_handle == INVALID_HANDLE_VALUE)
{
ec.assign(GetLastError(), system_category());
m_done = true;
return;
}
#else
m_handle = ::opendir(f.c_str());
2016-07-09 22:26:26 +02:00
if (m_handle == nullptr)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
m_done = true;
return;
}
// read the first entry
next(ec);
#endif
}
directory::~directory()
{
#ifdef TORRENT_WINDOWS
if (m_handle != INVALID_HANDLE_VALUE)
FindClose(m_handle);
#else
2017-09-12 23:10:11 +02:00
if (m_handle) ::closedir(m_handle);
#endif
}
std::uint64_t directory::inode() const
{
return m_inode;
}
std::string directory::file() const
{
#ifdef TORRENT_WINDOWS
return convert_from_native_path(m_fd.cFileName);
#else
return convert_from_native(m_name);
#endif
}
void directory::next(error_code& ec)
{
ec.clear();
#ifdef TORRENT_WINDOWS
if (FindNextFileW(m_handle, &m_fd) == 0)
{
m_done = true;
int err = GetLastError();
if (err != ERROR_NO_MORE_FILES)
ec.assign(err, system_category());
}
++m_inode;
#else
struct dirent* de;
errno = 0;
if ((de = ::readdir(m_handle)) != nullptr)
{
m_inode = de->d_ino;
m_name = de->d_name;
}
else
{
if (errno) ec.assign(errno, system_category());
m_done = true;
}
#endif
}
2007-06-10 22:46:09 +02:00
2014-07-06 21:18:00 +02:00
#ifndef INVALID_HANDLE_VALUE
2016-07-10 13:34:45 +02:00
#define INVALID_HANDLE_VALUE (-1)
2014-07-06 21:18:00 +02:00
#endif
2013-02-16 09:26:55 +01:00
#ifdef TORRENT_WINDOWS
struct overlapped_t
{
overlapped_t()
{
2016-08-13 01:24:03 +02:00
std::memset(&ol, 0, sizeof(ol));
2013-02-16 09:26:55 +01:00
ol.hEvent = CreateEvent(0, true, false, 0);
}
~overlapped_t()
{
if (ol.hEvent != INVALID_HANDLE_VALUE)
CloseHandle(ol.hEvent);
}
int wait(HANDLE file, error_code& ec)
{
2013-04-11 01:37:22 +02:00
if (ol.hEvent != INVALID_HANDLE_VALUE
&& WaitForSingleObject(ol.hEvent, INFINITE) == WAIT_FAILED)
2013-02-16 09:26:55 +01:00
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
2013-02-16 09:26:55 +01:00
return -1;
}
2016-12-13 16:30:36 +01:00
DWORD ret;
2013-02-16 09:26:55 +01:00
if (GetOverlappedResult(file, &ol, &ret, false) == 0)
{
DWORD last_error = GetLastError();
if (last_error != ERROR_HANDLE_EOF)
{
2014-03-24 08:42:55 +01:00
#ifdef ERROR_CANT_WAIT
2013-02-16 09:26:55 +01:00
TORRENT_ASSERT(last_error != ERROR_CANT_WAIT);
#endif
2014-07-06 21:18:00 +02:00
ec.assign(last_error, system_category());
2013-02-16 09:26:55 +01:00
return -1;
}
}
return ret;
}
OVERLAPPED ol;
};
#endif // TORRENT_WINDOWS
#ifdef TORRENT_WINDOWS
bool get_manage_volume_privs();
// this needs to be run before CreateFile
bool file::has_manage_volume_privs = get_manage_volume_privs();
#endif
file::file() : m_file_handle(INVALID_HANDLE_VALUE)
{}
file::file(std::string const& path, open_mode_t const mode, error_code& ec)
: m_file_handle(INVALID_HANDLE_VALUE)
{
// the return value is not important, since the
// error code contains the same information
open(path, mode, ec);
}
file::~file()
{
close();
}
2004-01-16 03:57:45 +01:00
bool file::open(std::string const& path, open_mode_t mode, error_code& ec)
{
close();
native_path_string file_path = convert_to_native_path_string(path);
2014-07-06 21:18:00 +02:00
#ifdef TORRENT_WINDOWS
2004-04-03 00:21:20 +02:00
struct win_open_mode_t
{
DWORD rw_mode;
DWORD create_mode;
};
static std::array<win_open_mode_t, 3> const mode_array{
{
// read_only
2011-06-09 08:08:24 +02:00
{GENERIC_READ, OPEN_EXISTING},
// write_only
2011-06-09 08:08:24 +02:00
{GENERIC_WRITE, OPEN_ALWAYS},
// read_write
2011-06-09 08:08:24 +02:00
{GENERIC_WRITE | GENERIC_READ, OPEN_ALWAYS},
}};
static std::array<DWORD, 4> const attrib_array{
{
FILE_ATTRIBUTE_NORMAL, // no attrib
FILE_ATTRIBUTE_HIDDEN, // hidden
FILE_ATTRIBUTE_NORMAL, // executable
FILE_ATTRIBUTE_HIDDEN, // hidden + executable
}};
TORRENT_ASSERT(static_cast<std::uint32_t>(mode & open_mode::rw_mask) < mode_array.size());
win_open_mode_t const& m = mode_array[static_cast<std::uint32_t>(mode & open_mode::rw_mask)];
DWORD a = attrib_array[static_cast<std::uint32_t>(mode & open_mode::attribute_mask) >> 12];
2012-05-27 20:17:51 +02:00
// one might think it's a good idea to pass in FILE_FLAG_RANDOM_ACCESS. It
// turns out that it isn't. That flag will break your operating system:
// http://support.microsoft.com/kb/2549369
DWORD const flags = ((mode & open_mode::random_access) ? 0 : FILE_FLAG_SEQUENTIAL_SCAN)
2011-06-09 08:08:24 +02:00
| (a ? a : FILE_ATTRIBUTE_NORMAL)
2014-07-06 21:18:00 +02:00
| FILE_FLAG_OVERLAPPED
| ((mode & open_mode::no_cache) ? FILE_FLAG_WRITE_THROUGH : 0);
handle_type handle = CreateFileW(file_path.c_str(), m.rw_mode
, FILE_SHARE_READ | FILE_SHARE_WRITE
2011-06-09 08:08:24 +02:00
, 0, m.create_mode, flags, 0);
2014-07-06 21:18:00 +02:00
if (handle == INVALID_HANDLE_VALUE)
2004-01-16 03:57:45 +01:00
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
TORRENT_ASSERT(ec);
return false;
}
2004-03-31 01:55:52 +02:00
2014-07-06 21:18:00 +02:00
m_file_handle = handle;
// try to make the file sparse if supported
// only set this flag if the file is opened for writing
if ((mode & open_mode::sparse)
&& (mode & open_mode::rw_mask) != open_mode::read_only)
{
DWORD temp;
2013-02-16 09:26:55 +01:00
overlapped_t ol;
2014-07-06 21:18:00 +02:00
BOOL ret = ::DeviceIoControl(native_handle(), FSCTL_SET_SPARSE, 0, 0
, 0, 0, &temp, &ol.ol);
2013-02-16 09:26:55 +01:00
error_code error;
2014-07-06 21:18:00 +02:00
if (ret == FALSE && GetLastError() == ERROR_IO_PENDING)
ol.wait(native_handle(), error);
2004-01-16 03:57:45 +01:00
}
#else // TORRENT_WINDOWS
// rely on default umask to filter x and w permissions
// for group and others
int permissions = S_IRUSR | S_IWUSR
| S_IRGRP | S_IWGRP
| S_IROTH | S_IWOTH;
if ((mode & open_mode::attribute_executable))
permissions |= S_IXGRP | S_IXOTH | S_IXUSR;
2012-09-24 18:13:57 +02:00
#ifdef O_BINARY
static const int mode_array[] = {O_RDONLY | O_BINARY, O_WRONLY | O_CREAT | O_BINARY, O_RDWR | O_CREAT | O_BINARY};
#else
static const int mode_array[] = {O_RDONLY, O_WRONLY | O_CREAT, O_RDWR | O_CREAT};
2012-09-24 18:13:57 +02:00
#endif
int open_mode = 0
#ifdef O_NOATIME
| ((mode & open_mode::no_atime) ? O_NOATIME : 0)
#endif
2015-01-08 00:21:54 +01:00
#ifdef O_SYNC
| ((mode & open_mode::no_cache) ? O_SYNC : 0)
#endif
;
handle_type handle = ::open(file_path.c_str()
, mode_array[static_cast<std::uint32_t>(mode & open_mode::rw_mask)] | open_mode
2014-07-06 21:18:00 +02:00
, permissions);
2009-01-17 10:37:40 +01:00
#ifdef O_NOATIME
// O_NOATIME is not allowed for files we don't own
// so, if we get EPERM when we try to open with it
// try again without O_NOATIME
if (handle == -1 && (mode & open_mode::no_atime) && errno == EPERM)
{
mode &= ~open_mode::no_atime;
open_mode &= ~O_NOATIME;
handle = ::open(file_path.c_str()
, mode_array[static_cast<std::uint32_t>(mode & open_mode::rw_mask)] | open_mode
, permissions);
}
2009-01-17 10:37:40 +01:00
#endif
2014-07-06 21:18:00 +02:00
if (handle == -1)
2004-01-16 03:57:45 +01:00
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
TORRENT_ASSERT(ec);
return false;
}
2014-07-06 21:18:00 +02:00
m_file_handle = handle;
2010-03-21 18:26:21 +01:00
#ifdef DIRECTIO_ON
// for solaris
if ((mode & open_mode::no_cache))
2010-03-21 18:26:21 +01:00
{
int yes = 1;
2014-07-06 21:18:00 +02:00
directio(native_handle(), DIRECTIO_ON);
2010-03-21 18:26:21 +01:00
}
#endif
#ifdef F_NOCACHE
2010-03-21 18:26:21 +01:00
// for BSD/Mac
if ((mode & open_mode::no_cache))
{
int yes = 1;
2017-09-12 23:10:11 +02:00
::fcntl(native_handle(), F_NOCACHE, &yes);
2014-07-06 21:18:00 +02:00
#ifdef F_NODIRECT
// it's OK to temporarily cache written pages
2017-09-12 23:10:11 +02:00
::fcntl(native_handle(), F_NODIRECT, &yes);
2014-07-06 21:18:00 +02:00
#endif
}
#endif
#ifdef POSIX_FADV_RANDOM
if ((mode & open_mode::random_access))
{
// disable read-ahead
// NOTE: in android this function was introduced in API 21,
// but the constant POSIX_FADV_RANDOM is there for lower
// API levels, just don't add :: to allow a macro workaround
posix_fadvise(native_handle(), 0, 0, POSIX_FADV_RANDOM);
}
#endif
#endif
m_open_mode = mode;
2008-07-20 18:34:01 +02:00
TORRENT_ASSERT(is_open());
return true;
}
2004-03-31 01:55:52 +02:00
2014-07-06 21:18:00 +02:00
bool file::is_open() const
{
2014-07-06 21:18:00 +02:00
return m_file_handle != INVALID_HANDLE_VALUE;
}
2013-02-16 09:26:55 +01:00
#ifdef TORRENT_WINDOWS
2014-07-06 21:18:00 +02:00
// returns true if the given file has any regions that are
// sparse, i.e. not allocated.
bool is_sparse(HANDLE file)
2013-02-16 09:26:55 +01:00
{
LARGE_INTEGER file_size;
if (!GetFileSizeEx(file, &file_size))
2014-05-10 05:23:05 +02:00
return false;
2013-02-16 09:26:55 +01:00
overlapped_t ol;
if (ol.ol.hEvent == nullptr) return false;
2013-02-16 09:26:55 +01:00
#ifndef FSCTL_QUERY_ALLOCATED_RANGES
2013-02-16 09:26:55 +01:00
typedef struct _FILE_ALLOCATED_RANGE_BUFFER {
LARGE_INTEGER FileOffset;
LARGE_INTEGER Length;
} FILE_ALLOCATED_RANGE_BUFFER;
2013-02-16 09:26:55 +01:00
#define FSCTL_QUERY_ALLOCATED_RANGES ((0x9 << 16) | (1 << 14) | (51 << 2) | 3)
#endif
FILE_ALLOCATED_RANGE_BUFFER in;
in.FileOffset.QuadPart = 0;
in.Length.QuadPart = file_size.QuadPart;
FILE_ALLOCATED_RANGE_BUFFER out[2];
DWORD returned_bytes = 0;
BOOL ret = DeviceIoControl(file, FSCTL_QUERY_ALLOCATED_RANGES, (void*)&in, sizeof(in)
2014-07-06 21:18:00 +02:00
, out, sizeof(out), &returned_bytes, &ol.ol);
2013-02-16 09:26:55 +01:00
2014-07-06 21:18:00 +02:00
if (ret == FALSE && GetLastError() == ERROR_IO_PENDING)
2013-02-16 09:26:55 +01:00
{
error_code ec;
returned_bytes = ol.wait(file, ec);
if (ec) return true;
}
else if (ret == FALSE)
{
return true;
}
2014-05-11 01:38:54 +02:00
// if we have more than one range in the file, we're sparse
if (returned_bytes != sizeof(FILE_ALLOCATED_RANGE_BUFFER)) {
return true;
}
return (in.Length.QuadPart != out[0].Length.QuadPart);
2013-02-16 09:26:55 +01:00
}
#endif
void file::close()
{
2014-07-06 21:18:00 +02:00
if (!is_open()) return;
#ifdef TORRENT_WINDOWS
2013-02-16 09:26:55 +01:00
// if this file is open for writing, has the sparse
// flag set, but there are no sparse regions, unset
// the flag
open_mode_t const rw_mode = m_open_mode & open_mode::rw_mask;
if ((rw_mode != open_mode::read_only)
&& (m_open_mode & open_mode::sparse)
2014-07-06 21:18:00 +02:00
&& !is_sparse(native_handle()))
2013-02-16 09:26:55 +01:00
{
overlapped_t ol;
// according to MSDN, clearing the sparse flag of a file only
// works on windows vista and later
#ifdef TORRENT_MINGW
2014-07-06 21:18:00 +02:00
typedef struct _FILE_SET_SPARSE_BUFFER {
BOOLEAN SetSparse;
} FILE_SET_SPARSE_BUFFER;
2013-02-16 09:26:55 +01:00
#endif
DWORD temp;
FILE_SET_SPARSE_BUFFER b;
b.SetSparse = FALSE;
2014-07-06 21:18:00 +02:00
BOOL ret = ::DeviceIoControl(native_handle(), FSCTL_SET_SPARSE, &b, sizeof(b)
, 0, 0, &temp, &ol.ol);
2013-02-16 09:26:55 +01:00
error_code ec;
2014-07-06 21:18:00 +02:00
if (ret == FALSE && GetLastError() == ERROR_IO_PENDING)
2013-02-16 09:26:55 +01:00
{
2014-07-06 21:18:00 +02:00
ol.wait(native_handle(), ec);
2013-02-16 09:26:55 +01:00
}
}
2014-07-06 21:18:00 +02:00
CloseHandle(native_handle());
#else
2014-07-06 21:18:00 +02:00
if (m_file_handle != INVALID_HANDLE_VALUE)
::close(m_file_handle);
#endif
2014-07-06 21:18:00 +02:00
m_file_handle = INVALID_HANDLE_VALUE;
m_open_mode = open_mode_t{};
}
2015-04-19 00:00:27 +02:00
namespace {
#if !TORRENT_USE_PREADV
void gather_copy(span<iovec_t const> bufs, char* dst)
{
2015-11-19 01:51:17 +01:00
std::size_t offset = 0;
for (auto buf : bufs)
2014-07-06 21:18:00 +02:00
{
std::memcpy(dst + offset, buf.data(), buf.size());
offset += buf.size();
2014-07-06 21:18:00 +02:00
}
}
void scatter_copy(span<iovec_t const> bufs, char const* src)
{
2015-11-19 01:51:17 +01:00
std::size_t offset = 0;
for (auto buf : bufs)
2013-01-06 19:38:33 +01:00
{
std::memcpy(buf.data(), src + offset, buf.size());
offset += buf.size();
2013-01-06 19:38:33 +01:00
}
2014-07-06 21:18:00 +02:00
}
bool coalesce_read_buffers(span<iovec_t const>& bufs
, iovec_t& tmp)
2014-07-06 21:18:00 +02:00
{
std::size_t const buf_size = aux::numeric_cast<std::size_t>(bufs_size(bufs));
char* buf = new char[buf_size];
tmp = { buf, buf_size };
bufs = span<iovec_t const>(tmp);
2014-07-06 21:18:00 +02:00
return true;
}
void coalesce_read_buffers_end(span<iovec_t const> bufs
, char* const buf, bool const copy)
2014-07-06 21:18:00 +02:00
{
if (copy) scatter_copy(bufs, buf);
delete[] buf;
2014-07-06 21:18:00 +02:00
}
bool coalesce_write_buffers(span<iovec_t const>& bufs
, iovec_t& tmp)
2014-07-06 21:18:00 +02:00
{
std::size_t const buf_size = aux::numeric_cast<std::size_t>(bufs_size(bufs));
char* buf = new char[buf_size];
gather_copy(bufs, buf);
tmp = { buf, buf_size };
bufs = span<iovec_t const>(tmp);
2014-07-06 21:18:00 +02:00
return true;
}
#else
namespace {
int bufs_size(span<::iovec> bufs)
{
std::size_t size = 0;
for (auto buf : bufs) size += buf.iov_len;
return int(size);
}
}
2015-12-12 06:15:24 +01:00
#endif // TORRENT_USE_PREADV
2014-07-06 21:18:00 +02:00
template <class Fun>
std::int64_t iov(Fun f, handle_type fd, std::int64_t file_offset
, span<iovec_t const> bufs, error_code& ec)
2014-07-06 21:18:00 +02:00
{
#if TORRENT_USE_PREADV
TORRENT_ALLOCA(vec, ::iovec, bufs.size());
auto it = vec.begin();
for (auto const& b : bufs)
{
it->iov_base = b.data();
it->iov_len = b.size();
++it;
}
2014-07-06 21:18:00 +02:00
int ret = 0;
while (!vec.empty())
{
#ifdef IOV_MAX
auto const nbufs = vec.first(std::min(int(vec.size()), IOV_MAX));
#else
auto const nbufs = vec;
#endif
2014-07-06 21:18:00 +02:00
int tmp_ret = 0;
tmp_ret = f(fd, nbufs.data(), int(nbufs.size()), file_offset);
2014-07-06 21:18:00 +02:00
if (tmp_ret < 0)
{
2014-07-06 21:18:00 +02:00
#ifdef TORRENT_WINDOWS
ec.assign(GetLastError(), system_category());
#else
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
2014-07-06 21:18:00 +02:00
#endif
return -1;
}
2014-07-06 21:18:00 +02:00
file_offset += tmp_ret;
ret += tmp_ret;
// we got a short read/write. It's either 0, and we're at EOF, or we
// just need to issue the read/write operation again. In either case,
// punt that to the upper layer, as reissuing the operations is
// complicated here
const int expected_len = bufs_size(nbufs);
if (tmp_ret < expected_len) break;
vec = vec.subspan(nbufs.size());
}
2014-07-06 21:18:00 +02:00
return ret;
2004-04-05 00:15:31 +02:00
2014-07-06 21:18:00 +02:00
#elif TORRENT_USE_PREAD
std::int64_t ret = 0;
for (auto i : bufs)
{
std::int64_t const tmp_ret = f(fd, i.data(), i.size(), file_offset);
2014-07-06 21:18:00 +02:00
if (tmp_ret < 0)
{
2014-07-06 21:18:00 +02:00
#ifdef TORRENT_WINDOWS
2014-07-20 10:59:02 +02:00
ec.assign(GetLastError(), system_category());
2014-07-06 21:18:00 +02:00
#else
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
2013-01-11 05:49:08 +01:00
#endif
return -1;
}
2014-07-06 21:18:00 +02:00
file_offset += tmp_ret;
ret += tmp_ret;
if (tmp_ret < int(i.size())) break;
}
2014-07-06 21:18:00 +02:00
return ret;
2014-07-20 10:59:02 +02:00
#else // not PREADV nor PREAD
2014-07-06 21:18:00 +02:00
int ret = 0;
2014-07-06 21:18:00 +02:00
#ifdef TORRENT_WINDOWS
if (SetFilePointerEx(fd, offs, &offs, FILE_BEGIN) == FALSE)
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
return -1;
}
2014-07-06 21:18:00 +02:00
#else
if (lseek(fd, file_offset, SEEK_SET) < 0)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
2014-07-06 21:18:00 +02:00
return -1;
}
2014-07-06 21:18:00 +02:00
#endif
for (auto i : bufs)
{
int tmp_ret = f(fd, i.data(), i.size());
2014-07-06 21:18:00 +02:00
if (tmp_ret < 0)
{
2014-07-06 21:18:00 +02:00
#ifdef TORRENT_WINDOWS
ec.assign(GetLastError(), system_category());
#else
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
2014-07-06 21:18:00 +02:00
#endif
return -1;
}
2014-07-06 21:18:00 +02:00
file_offset += tmp_ret;
ret += tmp_ret;
if (tmp_ret < int(i.size())) break;
}
2014-07-06 21:18:00 +02:00
return ret;
#endif // USE_PREADV
}
2015-04-19 00:00:27 +02:00
} // anonymous namespace
2014-07-06 21:18:00 +02:00
// this has to be thread safe and atomic. i.e. on posix systems it has to be
// turned into a series of pread() calls
std::int64_t file::readv(std::int64_t file_offset, span<iovec_t const> bufs
, error_code& ec, open_mode_t flags)
{
2013-01-06 19:38:33 +01:00
if (m_file_handle == INVALID_HANDLE_VALUE)
{
2014-07-06 21:18:00 +02:00
#ifdef TORRENT_WINDOWS
ec = error_code(ERROR_INVALID_HANDLE, system_category());
2013-01-06 19:38:33 +01:00
#else
2015-11-24 06:50:51 +01:00
ec = error_code(boost::system::errc::bad_file_descriptor, generic_category());
2014-07-06 21:18:00 +02:00
#endif
2013-01-06 19:38:33 +01:00
return -1;
}
TORRENT_ASSERT((m_open_mode & open_mode::rw_mask) == open_mode::read_only
|| (m_open_mode & open_mode::rw_mask) == open_mode::read_write);
TORRENT_ASSERT(!bufs.empty());
TORRENT_ASSERT(is_open());
2014-07-06 21:18:00 +02:00
#if TORRENT_USE_PREADV
2015-08-21 10:05:51 +02:00
TORRENT_UNUSED(flags);
2016-12-13 16:30:36 +01:00
std::int64_t ret = iov(&::preadv, native_handle(), file_offset, bufs, ec);
2014-07-06 21:18:00 +02:00
#else
// there's no point in coalescing single buffer writes
if (bufs.size() == 1)
{
flags &= ~open_mode::coalesce_buffers;
}
iovec_t tmp;
span<iovec_t const> tmp_bufs = bufs;
if (flags & open_mode::coalesce_buffers)
{
if (!coalesce_read_buffers(tmp_bufs, tmp))
2014-07-06 21:18:00 +02:00
// ok, that failed, don't coalesce this read
flags &= ~open_mode::coalesce_buffers;
}
2014-07-06 21:18:00 +02:00
#if TORRENT_USE_PREAD
std::int64_t ret = iov(&::pread, native_handle(), file_offset, tmp_bufs, ec);
2014-07-06 21:18:00 +02:00
#else
std::int64_t ret = iov(&::read, native_handle(), file_offset, tmp_bufs, ec);
#endif
if (flags & open_mode::coalesce_buffers)
coalesce_read_buffers_end(bufs
, tmp.data(), !ec);
2013-01-11 05:49:08 +01:00
#endif
return ret;
2014-07-06 21:18:00 +02:00
}
// This has to be thread safe, i.e. atomic.
// that means, on posix this has to be turned into a series of
// pwrite() calls
std::int64_t file::writev(std::int64_t file_offset, span<iovec_t const> bufs
, error_code& ec, open_mode_t flags)
2014-07-06 21:18:00 +02:00
{
if (m_file_handle == INVALID_HANDLE_VALUE)
{
2014-07-06 21:18:00 +02:00
#ifdef TORRENT_WINDOWS
ec = error_code(ERROR_INVALID_HANDLE, system_category());
#else
2015-11-24 06:50:51 +01:00
ec = error_code(boost::system::errc::bad_file_descriptor, generic_category());
2014-07-06 21:18:00 +02:00
#endif
return -1;
}
TORRENT_ASSERT((m_open_mode & open_mode::rw_mask) == open_mode::write_only
|| (m_open_mode & open_mode::rw_mask) == open_mode::read_write);
TORRENT_ASSERT(!bufs.empty());
2014-07-06 21:18:00 +02:00
TORRENT_ASSERT(is_open());
2014-07-06 21:18:00 +02:00
ec.clear();
2014-07-06 21:18:00 +02:00
#if TORRENT_USE_PREADV
2015-08-21 10:05:51 +02:00
TORRENT_UNUSED(flags);
2016-12-13 16:30:36 +01:00
std::int64_t ret = iov(&::pwritev, native_handle(), file_offset, bufs, ec);
2014-07-06 21:18:00 +02:00
#else
// there's no point in coalescing single buffer writes
if (bufs.size() == 1)
{
flags &= ~open_mode::coalesce_buffers;
}
iovec_t tmp;
if (flags & open_mode::coalesce_buffers)
2014-07-06 21:18:00 +02:00
{
if (!coalesce_write_buffers(bufs, tmp))
2014-07-06 21:18:00 +02:00
// ok, that failed, don't coalesce writes
flags &= ~open_mode::coalesce_buffers;
}
2014-07-06 21:18:00 +02:00
#if TORRENT_USE_PREAD
std::int64_t ret = iov(&::pwrite, native_handle(), file_offset, bufs, ec);
2014-07-06 21:18:00 +02:00
#else
std::int64_t ret = iov(&::write, native_handle(), file_offset, bufs, ec);
2014-07-06 21:18:00 +02:00
#endif
if (flags & open_mode::coalesce_buffers)
delete[] tmp.data();
2014-07-06 21:18:00 +02:00
#endif
2015-08-09 04:53:11 +02:00
#if TORRENT_USE_FDATASYNC \
2014-07-06 21:18:00 +02:00
&& !defined F_NOCACHE && \
!defined DIRECTIO_ON
if (m_open_mode & open_mode::no_cache)
{
2017-09-12 23:10:11 +02:00
if (::fdatasync(native_handle()) != 0
2014-07-06 21:18:00 +02:00
&& errno != EINVAL
&& errno != ENOSYS)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
}
}
2014-07-06 21:18:00 +02:00
#endif
return ret;
}
#ifdef TORRENT_WINDOWS
bool get_manage_volume_privs()
{
typedef BOOL (WINAPI *OpenProcessToken_t)(
HANDLE ProcessHandle,
DWORD DesiredAccess,
PHANDLE TokenHandle);
typedef BOOL (WINAPI *LookupPrivilegeValue_t)(
LPCSTR lpSystemName,
LPCSTR lpName,
PLUID lpLuid);
typedef BOOL (WINAPI *AdjustTokenPrivileges_t)(
HANDLE TokenHandle,
BOOL DisableAllPrivileges,
PTOKEN_PRIVILEGES NewState,
DWORD BufferLength,
PTOKEN_PRIVILEGES PreviousState,
PDWORD ReturnLength);
auto OpenProcessToken =
aux::get_library_procedure<aux::advapi32, OpenProcessToken_t>("OpenProcessToken");
auto LookupPrivilegeValue =
aux::get_library_procedure<aux::advapi32, LookupPrivilegeValue_t>("LookupPrivilegeValueA");
auto AdjustTokenPrivileges =
aux::get_library_procedure<aux::advapi32, AdjustTokenPrivileges_t>("AdjustTokenPrivileges");
if (OpenProcessToken == nullptr || LookupPrivilegeValue == nullptr || AdjustTokenPrivileges == nullptr) return false;
HANDLE token;
if (!OpenProcessToken(GetCurrentProcess()
, TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token))
return false;
TOKEN_PRIVILEGES privs;
if (!LookupPrivilegeValue(nullptr, "SeManageVolumePrivilege"
, &privs.Privileges[0].Luid))
{
CloseHandle(token);
return false;
}
privs.PrivilegeCount = 1;
privs.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
bool ret = AdjustTokenPrivileges(token, FALSE, &privs, 0, nullptr, nullptr)
&& GetLastError() == ERROR_SUCCESS;
CloseHandle(token);
return ret;
}
void set_file_valid_data(HANDLE f, std::int64_t size)
{
typedef BOOL (WINAPI *SetFileValidData_t)(HANDLE, LONGLONG);
auto SetFileValidData =
aux::get_library_procedure<aux::kernel32, SetFileValidData_t>("SetFileValidData");
if (SetFileValidData == nullptr) return;
// we don't necessarily expect to have enough
// privilege to do this, so ignore errors.
SetFileValidData(f, size);
}
#endif
bool file::set_size(std::int64_t s, error_code& ec)
{
TORRENT_ASSERT(is_open());
TORRENT_ASSERT(s >= 0);
2004-01-16 03:57:45 +01:00
#ifdef TORRENT_WINDOWS
2011-11-13 05:12:56 +01:00
LARGE_INTEGER offs;
LARGE_INTEGER cur_size;
2014-07-06 21:18:00 +02:00
if (GetFileSizeEx(native_handle(), &cur_size) == FALSE)
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
return false;
}
offs.QuadPart = s;
// only set the file size if it's not already at
// the right size. We don't want to update the
// modification time if we don't have to
if (cur_size.QuadPart != s)
{
2014-07-06 21:18:00 +02:00
if (SetFilePointerEx(native_handle(), offs, &offs, FILE_BEGIN) == FALSE)
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
return false;
}
2014-07-06 21:18:00 +02:00
if (::SetEndOfFile(native_handle()) == FALSE)
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
return false;
}
}
#if _WIN32_WINNT >= 0x0600 // only if Windows Vista or newer
if (!(m_open_mode & open_mode::sparse))
{
typedef DWORD (WINAPI *GetFileInformationByHandleEx_t)(HANDLE hFile
, FILE_INFO_BY_HANDLE_CLASS FileInformationClass
, LPVOID lpFileInformation
, DWORD dwBufferSize);
auto GetFileInformationByHandleEx =
aux::get_library_procedure<aux::kernel32, GetFileInformationByHandleEx_t>("GetFileInformationByHandleEx");
offs.QuadPart = 0;
if (GetFileInformationByHandleEx != nullptr)
{
// only allocate the space if the file
// is not fully allocated
FILE_STANDARD_INFO inf;
if (GetFileInformationByHandleEx(native_handle()
, FileStandardInfo, &inf, sizeof(inf)) == FALSE)
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
if (ec) return false;
}
offs = inf.AllocationSize;
}
if (offs.QuadPart < s)
{
// if the user has permissions, avoid filling
// the file with zeroes, but just fill it with
// garbage instead
set_file_valid_data(m_file_handle, s);
}
}
#endif // if Windows Vista
#else // NON-WINDOWS
struct stat st;
2017-09-12 23:10:11 +02:00
if (::fstat(native_handle(), &st) != 0)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
return false;
}
// only truncate the file if it doesn't already
// have the right size. We don't want to update
2017-09-12 23:10:11 +02:00
if (st.st_size != s && ::ftruncate(native_handle(), s) < 0)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
return false;
}
// if we're not in sparse mode, allocate the storage
// but only if the number of allocated blocks for the file
// is less than the file size. Otherwise we would just
// update the modification time of the file for no good
// reason.
if (!(m_open_mode & open_mode::sparse)
2016-12-12 03:37:07 +01:00
&& std::int64_t(st.st_blocks) < (s + st.st_blksize - 1) / st.st_blksize)
{
// How do we know that the file is already allocated?
// if we always try to allocate the space, we'll update
// the modification time without actually changing the file
// but if we don't do anything if the file size is
#ifdef F_PREALLOCATE
fstore_t f = {F_ALLOCATECONTIG, F_PEOFPOSMODE, 0, s, 0};
2014-07-06 21:18:00 +02:00
if (fcntl(native_handle(), F_PREALLOCATE, &f) < 0)
{
2011-04-13 08:54:56 +02:00
if (errno != ENOSPC)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
2011-04-13 08:54:56 +02:00
return false;
}
// ok, let's try to allocate non contiguous space then
2015-08-09 04:53:11 +02:00
f.fst_flags = F_ALLOCATEALL;
2014-07-06 21:18:00 +02:00
if (fcntl(native_handle(), F_PREALLOCATE, &f) < 0)
2011-04-13 08:54:56 +02:00
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
2011-04-13 08:54:56 +02:00
return false;
}
}
2010-04-08 07:09:11 +02:00
#endif // F_PREALLOCATE
2011-09-09 01:08:37 +02:00
#ifdef F_ALLOCSP64
flock64 fl64;
fl64.l_whence = SEEK_SET;
fl64.l_start = 0;
fl64.l_len = s;
if (fcntl(native_handle(), F_ALLOCSP64, &fl64) < 0)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
2011-09-09 01:08:37 +02:00
return false;
}
#endif // F_ALLOCSP64
2010-04-08 07:09:11 +02:00
#if TORRENT_HAS_FALLOCATE
// if fallocate failed, we have to use posix_fallocate
// which can be painfully slow
2010-04-08 07:31:00 +02:00
// if you get a compile error here, you might want to
// define TORRENT_HAS_FALLOCATE to 0.
int const ret = posix_fallocate(native_handle(), 0, s);
// posix_allocate fails with EINVAL in case the underlying
2015-09-21 23:58:20 +02:00
// filesystem does not support this operation
if (ret != 0 && ret != EINVAL)
2010-01-18 01:51:40 +01:00
{
2015-11-24 06:50:51 +01:00
ec.assign(ret, system_category());
2010-01-18 01:51:40 +01:00
return false;
}
2010-04-08 07:09:11 +02:00
#endif // TORRENT_HAS_FALLOCATE
}
2010-04-08 07:09:11 +02:00
#endif // TORRENT_WINDOWS
return true;
2004-01-16 03:57:45 +01:00
}
std::int64_t file::get_size(error_code& ec) const
2004-01-16 03:57:45 +01:00
{
#ifdef TORRENT_WINDOWS
LARGE_INTEGER file_size;
2014-07-06 21:18:00 +02:00
if (!GetFileSizeEx(native_handle(), &file_size))
{
2014-07-06 21:18:00 +02:00
ec.assign(GetLastError(), system_category());
return -1;
}
return file_size.QuadPart;
#else
struct stat fs;
2017-09-12 23:10:11 +02:00
if (::fstat(native_handle(), &fs) != 0)
{
2015-11-24 06:50:51 +01:00
ec.assign(errno, system_category());
return -1;
}
return fs.st_size;
#endif
2004-01-16 03:57:45 +01:00
}
std::int64_t file::sparse_end(std::int64_t start) const
{
#ifdef TORRENT_WINDOWS
2014-07-06 21:18:00 +02:00
#ifndef FSCTL_QUERY_ALLOCATED_RANGES
2010-02-14 05:05:18 +01:00
typedef struct _FILE_ALLOCATED_RANGE_BUFFER {
LARGE_INTEGER FileOffset;
LARGE_INTEGER Length;
} FILE_ALLOCATED_RANGE_BUFFER;
2010-02-14 05:05:18 +01:00
#define FSCTL_QUERY_ALLOCATED_RANGES ((0x9 << 16) | (1 << 14) | (51 << 2) | 3)
2014-07-06 21:18:00 +02:00
#endif // TORRENT_MINGW
FILE_ALLOCATED_RANGE_BUFFER buffer;
DWORD bytes_returned = 0;
FILE_ALLOCATED_RANGE_BUFFER in;
error_code ec;
std::int64_t file_size = get_size(ec);
if (ec) return start;
2014-07-06 21:18:00 +02:00
in.FileOffset.QuadPart = start;
in.Length.QuadPart = file_size - start;
2014-07-06 21:18:00 +02:00
if (!DeviceIoControl(native_handle(), FSCTL_QUERY_ALLOCATED_RANGES
, &in, sizeof(FILE_ALLOCATED_RANGE_BUFFER)
, &buffer, sizeof(FILE_ALLOCATED_RANGE_BUFFER), &bytes_returned, 0))
{
if (GetLastError() != ERROR_INSUFFICIENT_BUFFER) return start;
}
// if there are no allocated regions within the rest
// of the file, return the end of the file
if (bytes_returned == 0) return file_size;
// assume that this range overlaps the start of the
// region we were interested in, and that start actually
// resides in an allocated region.
if (buffer.FileOffset.QuadPart < start) return start;
// return the offset to the next allocated region
return buffer.FileOffset.QuadPart;
#elif defined SEEK_DATA
// this is supported on solaris
2017-09-12 23:10:11 +02:00
std::int64_t ret = ::lseek(native_handle(), start, SEEK_DATA);
if (ret < 0) return start;
2011-09-05 01:29:47 +02:00
return start;
#else
return start;
#endif
}
2004-01-16 03:57:45 +01:00
}