SnapshotStorage.cpp

Go to the documentation of this file.

/*
    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/>.
*/
 
#include "SnapshotStorage.h"
#include <libdevcore/CommonIO.h>
#include <libdevcore/SHA3.h>
 
#include <snappy.h>
 
namespace fs = boost::filesystem;
 
namespace dev
{
namespace eth
{
 
namespace
{
 
size_t const c_maxChunkUncomressedSize = 10 * 1024 * 1024;
 
 
std::string snappyUncompress(std::string const& _compressed)
{
    size_t uncompressedSize = 0;
    if (!snappy::GetUncompressedLength(_compressed.data(), _compressed.size(), &uncompressedSize))
        BOOST_THROW_EXCEPTION(FailedToGetUncompressedLength());
 
    if (uncompressedSize > c_maxChunkUncomressedSize)
        BOOST_THROW_EXCEPTION(ChunkIsTooBig());
 
    std::string uncompressed;
    if (!snappy::Uncompress(_compressed.data(), _compressed.size(), &uncompressed))
        BOOST_THROW_EXCEPTION(FailedToUncompressedSnapshotChunk());
 
    return uncompressed;
}
 
class SnapshotStorage: public SnapshotStorageFace
{
public:
    explicit SnapshotStorage(fs::path const& _snapshotDir) : m_snapshotDir(_snapshotDir) {}
 
    bytes readManifest() const override
    {
        bytes const manifestBytes = dev::contents((m_snapshotDir / "MANIFEST").string());
        if (manifestBytes.empty())
            BOOST_THROW_EXCEPTION(FailedToReadSnapshotManifestFile());
 
        return manifestBytes;
    }
 
    std::string readCompressedChunk(h256 const& _chunkHash) const override
    {
        std::string const chunkCompressed = dev::contentsString((m_snapshotDir / toHex(_chunkHash)).string());
        if (chunkCompressed.empty())
            BOOST_THROW_EXCEPTION(FailedToReadChunkFile() << errinfo_hash256(_chunkHash));
 
        return chunkCompressed;
    }
 
    std::string readChunk(h256 const& _chunkHash) const override
    {
        std::string const chunkCompressed = readCompressedChunk(_chunkHash);
 
        h256 const chunkHash = sha3(chunkCompressed);
        if (chunkHash != _chunkHash)
            BOOST_THROW_EXCEPTION(ChunkDataCorrupted() << errinfo_hash256(_chunkHash));
 
        std::string const chunkUncompressed = snappyUncompress(chunkCompressed);
        assert(!chunkUncompressed.empty());
 
        return chunkUncompressed;
    }
 
    void copyTo(fs::path const& _path) const override { copyDirectory(m_snapshotDir, _path); }
 
private:
    fs::path const m_snapshotDir;
};
 
}
 
std::unique_ptr<SnapshotStorageFace> createSnapshotStorage(fs::path const& _snapshotDirPath)
{
    return std::unique_ptr<SnapshotStorageFace>(new SnapshotStorage(_snapshotDirPath));
}
 
fs::path importedSnapshotPath(fs::path const& _dataDir, h256 const& _genesisHash)
{
    return _dataDir / toHex(_genesisHash.ref().cropped(0, 4)) / "snapshot";
}
}
}