147 lines
4.8 KiB
C++
147 lines
4.8 KiB
C++
/*
|
|
* Copyright (C) 2009 Google Inc. 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 Google Inc. 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 "config.h"
|
|
#include "SQLiteFileSystem.h"
|
|
|
|
#include "SQLiteDatabase.h"
|
|
#include "SQLiteStatement.h"
|
|
#include <pal/crypto/CryptoDigest.h>
|
|
#include <sqlite3.h>
|
|
#include <wtf/FileSystem.h>
|
|
|
|
#if PLATFORM(IOS_FAMILY)
|
|
#include <pal/spi/ios/SQLite3SPI.h>
|
|
#endif
|
|
|
|
namespace WebCore {
|
|
|
|
SQLiteFileSystem::SQLiteFileSystem()
|
|
{
|
|
}
|
|
|
|
String SQLiteFileSystem::appendDatabaseFileNameToPath(const String& path, const String& fileName)
|
|
{
|
|
return FileSystem::pathByAppendingComponent(path, fileName);
|
|
}
|
|
|
|
bool SQLiteFileSystem::ensureDatabaseDirectoryExists(const String& path)
|
|
{
|
|
if (path.isEmpty())
|
|
return false;
|
|
return FileSystem::makeAllDirectories(path);
|
|
}
|
|
|
|
bool SQLiteFileSystem::ensureDatabaseFileExists(const String& fileName, bool checkPathOnly)
|
|
{
|
|
if (fileName.isEmpty())
|
|
return false;
|
|
|
|
if (checkPathOnly) {
|
|
String dir = FileSystem::parentPath(fileName);
|
|
return ensureDatabaseDirectoryExists(dir);
|
|
}
|
|
|
|
return FileSystem::fileExists(fileName);
|
|
}
|
|
|
|
bool SQLiteFileSystem::deleteEmptyDatabaseDirectory(const String& path)
|
|
{
|
|
return FileSystem::deleteEmptyDirectory(path);
|
|
}
|
|
|
|
bool SQLiteFileSystem::deleteDatabaseFile(const String& fileName)
|
|
{
|
|
auto walFileName = makeString(fileName, "-wal"_s);
|
|
auto shmFileName = makeString(fileName, "-shm"_s);
|
|
|
|
// Try to delete all three files whether or not they are there.
|
|
FileSystem::deleteFile(fileName);
|
|
FileSystem::deleteFile(walFileName);
|
|
FileSystem::deleteFile(shmFileName);
|
|
|
|
// If any of the wal or shm files remain after the delete attempt, the overall delete operation failed.
|
|
return !FileSystem::fileExists(fileName) && !FileSystem::fileExists(walFileName) && !FileSystem::fileExists(shmFileName);
|
|
}
|
|
|
|
#if PLATFORM(IOS_FAMILY)
|
|
bool SQLiteFileSystem::truncateDatabaseFile(sqlite3* database)
|
|
{
|
|
return sqlite3_file_control(database, 0, SQLITE_TRUNCATE_DATABASE, 0) == SQLITE_OK;
|
|
}
|
|
#endif
|
|
|
|
uint64_t SQLiteFileSystem::databaseFileSize(const String& fileName)
|
|
{
|
|
uint64_t totalSize = 0;
|
|
|
|
if (auto fileSize = FileSystem::fileSize(fileName))
|
|
totalSize += *fileSize;
|
|
|
|
if (auto fileSize = FileSystem::fileSize(makeString(fileName, "-wal"_s)))
|
|
totalSize += *fileSize;
|
|
|
|
if (auto fileSize = FileSystem::fileSize(makeString(fileName, "-shm"_s)))
|
|
totalSize += *fileSize;
|
|
|
|
return totalSize;
|
|
}
|
|
|
|
std::optional<WallTime> SQLiteFileSystem::databaseCreationTime(const String& fileName)
|
|
{
|
|
return FileSystem::fileCreationTime(fileName);
|
|
}
|
|
|
|
std::optional<WallTime> SQLiteFileSystem::databaseModificationTime(const String& fileName)
|
|
{
|
|
return FileSystem::fileModificationTime(fileName);
|
|
}
|
|
|
|
String SQLiteFileSystem::computeHashForFileName(const String& fileName)
|
|
{
|
|
auto cryptoDigest = PAL::CryptoDigest::create(PAL::CryptoDigest::Algorithm::SHA_256);
|
|
auto utf8FileName = fileName.utf8();
|
|
cryptoDigest->addBytes(utf8FileName.data(), utf8FileName.length());
|
|
auto digest = cryptoDigest->computeHash();
|
|
|
|
// Convert digest to hex.
|
|
char* start = 0;
|
|
unsigned digestLength = digest.size();
|
|
CString result = CString::newUninitialized(digestLength * 2, start);
|
|
char* buffer = start;
|
|
for (size_t i = 0; i < digestLength; ++i) {
|
|
snprintf(buffer, 3, "%02X", digest.at(i));
|
|
buffer += 2;
|
|
}
|
|
return String::fromUTF8(result);
|
|
}
|
|
|
|
} // namespace WebCore
|