/////////////////////////////////////////////////////////////////////////////// // LameXP - Audio Encoder Front-End // Copyright (C) 2004-2013 LoRd_MuldeR // // This program 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 2 of the License, or // (at your option) any later version. // // This program 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 this program; if not, write to the Free Software Foundation, Inc., // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. // // http://www.gnu.org/licenses/gpl-2.0.txt /////////////////////////////////////////////////////////////////////////////// #include "LockedFile.h" #include "Global.h" #include #include #include #include #include #include #include #include #include #include //Windows includes #define NOMINMAX #define WIN32_LEAN_AND_MEAN #include /////////////////////////////////////////////////////////////////////////////// // WARNING: Passing file descriptors into Qt does NOT work with dynamically linked CRT! #ifdef QT_NODLL static const bool g_useFileDescr = 1; #else static const bool g_useFileDescr = 0; #endif /////////////////////////////////////////////////////////////////////////////// static const char *g_blnk = "deadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeefdeadbeef"; static const char *g_seed = "c375d83b4388329408dfcbb4d9a065b6e06d28272f25ef299c70b506e26600af79fd2f866ae24602daf38f25c9d4b7e1"; static const char *g_salt = "ee9f7bdabc170763d2200a7e3030045aafe380011aefc1730e547e9244c62308aac42a976feeca224ba553de0c4bb883"; QByteArray LockedFile::fileHash(QFile &file) { QByteArray hash = QByteArray::fromHex(g_blnk); if(file.isOpen() && file.reset()) { QKeccakHash keccak; const QByteArray data = file.readAll(); const QByteArray seed = QByteArray::fromHex(g_seed); const QByteArray salt = QByteArray::fromHex(g_salt); if(keccak.init(QKeccakHash::hb384)) { bool ok = true; ok = ok && keccak.addData(seed); ok = ok && keccak.addData(data); ok = ok && keccak.addData(salt); if(ok) { const QByteArray digest = keccak.finalize(); if(!digest.isEmpty()) hash = digest.toHex(); } } } return hash; } /////////////////////////////////////////////////////////////////////////////// LockedFile::LockedFile(QResource *const resource, const QString &outPath, const QByteArray &expectedHash) { m_fileHandle = NULL; //Make sure the resource is valid if(!(resource->isValid() && resource->data())) { THROW_FMT("The resource at %p is invalid!", resource); } QFile outFile(outPath); m_filePath = QFileInfo(outFile).absoluteFilePath(); //Open output file for(int i = 0; i < 64; i++) { if(outFile.open(QIODevice::WriteOnly)) break; if(!i) qWarning("Failed to open file on first attemp, retrying..."); Sleep(100); } //Write data to file if(outFile.isOpen() && outFile.isWritable()) { if(outFile.write(reinterpret_cast(resource->data()), resource->size()) != resource->size()) { QFile::remove(QFileInfo(outFile).canonicalFilePath()); THROW_FMT("File '%s' could not be written!", QUTF8(QFileInfo(outFile).fileName())); } outFile.close(); } else { THROW_FMT("File '%s' could not be created!", QUTF8(QFileInfo(outFile).fileName())); } //Now lock the file! for(int i = 0; i < 64; i++) { m_fileHandle = CreateFileW(QWCHAR(QDir::toNativeSeparators(m_filePath)), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, NULL, NULL); if((m_fileHandle != NULL) && (m_fileHandle != INVALID_HANDLE_VALUE)) break; if(!i) qWarning("Failed to lock file on first attemp, retrying..."); Sleep(100); } //Locked successfully? if((m_fileHandle == NULL) || (m_fileHandle == INVALID_HANDLE_VALUE)) { QFile::remove(QFileInfo(outFile).canonicalFilePath()); THROW_FMT("File '%s' could not be locked!", QUTF8(QFileInfo(outFile).fileName())); } //Open file for reading if(g_useFileDescr) { int fd = _open_osfhandle(reinterpret_cast(m_fileHandle), _O_RDONLY | _O_BINARY); if(fd >= 0) outFile.open(fd, QIODevice::ReadOnly); } else { for(int i = 0; i < 64; i++) { if(outFile.open(QIODevice::ReadOnly)) break; if(!i) qWarning("Failed to re-open file on first attemp, retrying..."); Sleep(100); } } //Verify file contents QByteArray hash; if(outFile.isOpen()) { hash = fileHash(outFile); outFile.close(); } else { QFile::remove(m_filePath); THROW_FMT("File '%s' could not be read!", QUTF8(QFileInfo(outFile).fileName())); } //Compare hashes if(hash.isNull() || _stricmp(hash.constData(), expectedHash.constData())) { qWarning("\nFile checksum error:\n A = %s\n B = %s\n", expectedHash.constData(), hash.constData()); LAMEXP_CLOSE(m_fileHandle); QFile::remove(m_filePath); THROW_FMT("File '%s' is corruputed, take care!", QUTF8(QFileInfo(outFile).fileName())); } } LockedFile::LockedFile(const QString &filePath) { m_fileHandle = NULL; QFileInfo existingFile(filePath); existingFile.setCaching(false); //Make sure the file exists, before we try to lock it if(!existingFile.exists()) { THROW_FMT("File '%s' does not exist!", QUTF8(existingFile.fileName())); } //Remember file path m_filePath = existingFile.canonicalFilePath(); //Now lock the file for(int i = 0; i < 64; i++) { m_fileHandle = CreateFileW(QWCHAR(QDir::toNativeSeparators(filePath)), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, NULL, NULL); if((m_fileHandle != NULL) && (m_fileHandle != INVALID_HANDLE_VALUE)) break; if(!i) qWarning("Failed to lock file on first attemp, retrying..."); Sleep(100); } //Locked successfully? if((m_fileHandle == NULL) || (m_fileHandle == INVALID_HANDLE_VALUE)) { THROW_FMT("File '%s' could not be locked!", QUTF8(existingFile.fileName())); } } LockedFile::~LockedFile(void) { LAMEXP_CLOSE(m_fileHandle); } const QString &LockedFile::filePath() { return m_filePath; } void LockedFile::selfTest() { if(!QKeccakHash::selfTest()) { THROW("QKeccakHash self-test has failed!"); } }