You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
2.3 KiB
118 lines
2.3 KiB
/*
|
|
This file is part of cpp-ethereum.
|
|
|
|
cpp-ethereum is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
cpp-ethereum is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
/** @file DownloadMan.h
|
|
* @author Gav Wood <i@gavwood.com>
|
|
* @date 2014
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
#include <set>
|
|
#include <libdevcore/Guards.h>
|
|
#include <libdevcore/Worker.h>
|
|
#include <libdevcore/RangeMask.h>
|
|
#include <libdevcore/FixedHash.h>
|
|
|
|
namespace dev
|
|
{
|
|
|
|
namespace eth
|
|
{
|
|
|
|
class DownloadMan;
|
|
|
|
class DownloadSub
|
|
{
|
|
friend class DownloadMan;
|
|
|
|
public:
|
|
DownloadSub(DownloadMan* _man);
|
|
~DownloadSub();
|
|
|
|
/// Finished last fetch - grab the next bunch of block hashes to download.
|
|
h256s nextFetch(unsigned _n);
|
|
|
|
/// Note that we've received a particular block.
|
|
void noteBlock(h256 _hash);
|
|
|
|
void doneFetch() { nextFetch(0); }
|
|
|
|
private:
|
|
void resetFetch() // Called by DownloadMan when we need to reset the download.
|
|
{
|
|
Guard l(m_fetch);
|
|
m_indices.clear();
|
|
m_asked.clear();
|
|
m_attempted.clear();
|
|
}
|
|
|
|
DownloadMan* m_man = nullptr;
|
|
|
|
Mutex m_fetch;
|
|
std::map<h256, unsigned> m_indices;
|
|
RangeMask<unsigned> m_asked;
|
|
RangeMask<unsigned> m_attempted;
|
|
};
|
|
|
|
class DownloadMan
|
|
{
|
|
friend class DownloadSub;
|
|
|
|
public:
|
|
~DownloadMan()
|
|
{
|
|
for (auto i: m_subs)
|
|
i->m_man = nullptr;
|
|
}
|
|
|
|
void resetToChain(h256s const& _chain)
|
|
{
|
|
{
|
|
ReadGuard l(x_subs);
|
|
for (auto i: m_subs)
|
|
i->resetFetch();
|
|
}
|
|
|
|
m_chain = _chain;
|
|
m_blocksGot = RangeMask<unsigned>(0, m_chain.size());
|
|
}
|
|
|
|
RangeMask<unsigned> taken(bool _desperate = false) const
|
|
{
|
|
auto ret = m_blocksGot;
|
|
if (!_desperate)
|
|
{
|
|
ReadGuard l(x_subs);
|
|
for (auto i: m_subs)
|
|
ret += i->m_asked;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
private:
|
|
h256s m_chain;
|
|
RangeMask<unsigned> m_blocksGot;
|
|
|
|
mutable SharedMutex x_subs;
|
|
std::set<DownloadSub*> m_subs;
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|