PersistencyUtils.cpp [plain text]
#include "config.h"
#include "PersistencyUtils.h"
#include "Logging.h"
#include <WebCore/SharedBuffer.h>
#include <wtf/FileSystem.h>
#include <wtf/RunLoop.h>
namespace WebKit {
using namespace WebCore;
std::unique_ptr<KeyedDecoder> createForFile(const String& path)
{
ASSERT(!RunLoop::isMain());
auto handle = FileSystem::openAndLockFile(path, FileSystem::FileOpenMode::Read);
if (handle == FileSystem::invalidPlatformFileHandle)
return nullptr;
long long fileSize = 0;
if (!FileSystem::getFileSize(handle, fileSize) || !fileSize) {
FileSystem::unlockAndCloseFile(handle);
return nullptr;
}
size_t bytesToRead;
if (!WTF::convertSafely(fileSize, bytesToRead)) {
FileSystem::unlockAndCloseFile(handle);
return nullptr;
}
Vector<char> buffer(bytesToRead);
size_t totalBytesRead = FileSystem::readFromFile(handle, buffer.data(), buffer.size());
FileSystem::unlockAndCloseFile(handle);
if (totalBytesRead != bytesToRead)
return nullptr;
return KeyedDecoder::decoder(reinterpret_cast<const uint8_t*>(buffer.data()), buffer.size());
}
void writeToDisk(std::unique_ptr<KeyedEncoder>&& encoder, String&& path)
{
ASSERT(!RunLoop::isMain());
auto rawData = encoder->finishEncoding();
if (!rawData)
return;
FileSystem::PlatformFileHandle handle = FileSystem::openAndLockFile(path, FileSystem::FileOpenMode::Write);
if (handle == FileSystem::invalidPlatformFileHandle)
return;
auto writtenBytes = FileSystem::writeToFile(handle, rawData->data(), rawData->size());
FileSystem::unlockAndCloseFile(handle);
if (writtenBytes != static_cast<int64_t>(rawData->size()))
RELEASE_LOG_ERROR(DiskPersistency, "Disk persistency: We only wrote %d out of %zu bytes to disk", static_cast<unsigned>(writtenBytes), rawData->size());
}
}