forked from premiere/premiere-libtorrent
127 lines
4.0 KiB
C++
127 lines
4.0 KiB
C++
/*
|
|
|
|
Copyright (c) 2006-2014, Arvid Norberg
|
|
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.
|
|
|
|
*/
|
|
|
|
#ifndef TORRENT_FILE_POOL_HPP
|
|
#define TORRENT_FILE_POOL_HPP
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma warning(push, 1)
|
|
#endif
|
|
|
|
#include <boost/intrusive_ptr.hpp>
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma warning(pop)
|
|
#endif
|
|
|
|
#include <map>
|
|
#include "libtorrent/file.hpp"
|
|
#include "libtorrent/ptime.hpp"
|
|
#include "libtorrent/thread.hpp"
|
|
#include "libtorrent/file_storage.hpp"
|
|
|
|
namespace libtorrent
|
|
{
|
|
// this is an internal cache of open file handles. It's primarily used by
|
|
// storage_interface implementations. It provides semi weak guarantees of
|
|
// not opening more file handles than specified. Given multiple threads,
|
|
// each with the ability to lock a file handle (via smart pointer), there
|
|
// may be windows where more file handles are open.
|
|
struct TORRENT_EXPORT file_pool : boost::noncopyable
|
|
{
|
|
// ``size`` specifies the number of allowed files handles
|
|
// to hold open at any given time.
|
|
file_pool(int size = 40);
|
|
~file_pool();
|
|
|
|
// return an open file handle to file at ``file_index`` in the
|
|
// file_storage ``fs`` opened at save path ``p``. ``m`` is the
|
|
// file open mode (see file::open_mode_t).
|
|
boost::intrusive_ptr<file> open_file(void* st, std::string const& p
|
|
, int file_index, file_storage const& fs, int m, error_code& ec);
|
|
|
|
// release all files belonging to the specified storage_interface (``st``)
|
|
// the overload that takes ``file_index`` releases only the file with
|
|
// that index in storage ``st``.
|
|
void release(void* st);
|
|
void release(void* st, int file_index);
|
|
|
|
// update the allowed number of open file handles to ``size``.
|
|
void resize(int size);
|
|
|
|
// returns the current limit of number of allowed open file handles held
|
|
// by the file_pool.
|
|
int size_limit() const { return m_size; }
|
|
|
|
// internal
|
|
void set_low_prio_io(bool b) { m_low_prio_io = b; }
|
|
|
|
private:
|
|
|
|
void remove_oldest();
|
|
|
|
int m_size;
|
|
bool m_low_prio_io;
|
|
|
|
struct lru_file_entry
|
|
{
|
|
lru_file_entry(): key(0), last_use(time_now()), mode(0) {}
|
|
mutable boost::intrusive_ptr<file> file_ptr;
|
|
void* key;
|
|
ptime last_use;
|
|
int mode;
|
|
};
|
|
|
|
// maps storage pointer, file index pairs to the
|
|
// lru entry for the file
|
|
typedef std::map<std::pair<void*, int>, lru_file_entry> file_set;
|
|
|
|
file_set m_files;
|
|
mutex m_mutex;
|
|
#if TORRENT_USE_ASSERTS
|
|
int m_in_use;
|
|
#endif
|
|
|
|
#if TORRENT_CLOSE_MAY_BLOCK
|
|
void closer_thread_fun();
|
|
mutex m_closer_mutex;
|
|
std::vector<boost::intrusive_ptr<file> > m_queued_for_close;
|
|
bool m_stop_thread;
|
|
|
|
// used to close files
|
|
thread m_closer_thread;
|
|
#endif
|
|
};
|
|
}
|
|
|
|
#endif
|