mirror of
https://github.com/telegramdesktop/tdesktop
synced 2025-03-23 20:01:35 +00:00
MSVC instruction improved for OpenAL, merging autoupdate code for all OSs (not tested yet!)
This commit is contained in:
parent
c40758f30d
commit
ac2ae16f47
18
MSVC.md
18
MSVC.md
@ -66,13 +66,23 @@ or download in ZIP and extract to **D:\TBuild\Libraries\**, rename **libexif-0.6
|
|||||||
* Build Debug configuration
|
* Build Debug configuration
|
||||||
* Build Release configuration
|
* Build Release configuration
|
||||||
|
|
||||||
####OpenAL Soft
|
####OpenAL Soft, slightly patched
|
||||||
|
|
||||||
Get sources by git – in [Git Bash](http://git-scm.com/downloads) go to **/d/tbuild/libraries** and run
|
Get sources by git – in [Git Bash](http://git-scm.com/downloads) go to **/d/tbuild/libraries** and run
|
||||||
|
|
||||||
git clone git://repo.or.cz/openal-soft.git
|
git clone git://repo.or.cz/openal-soft.git
|
||||||
|
|
||||||
to have **D:\TBuild\Libraries\openal-soft\CMakeLists.txt**
|
to have **D:\TBuild\Libraries\openal-soft\CMakeLists.txt**, then in [Git Bash](http://git-scm.com/downloads) go to **/d/tbuild/libraries/openal-soft** and run
|
||||||
|
|
||||||
|
git checkout 9479ea656b
|
||||||
|
|
||||||
|
Apply patch
|
||||||
|
|
||||||
|
* OR copy (with overwrite!) everything from **D:\\TBuild\\tdesktop\\\_openal\_patch\\** to **D:\\TBuild\\Libraries\\openal-soft\\**
|
||||||
|
* OR in Git Bash go to **/d/tbuild/libraries/openal-soft/** and run
|
||||||
|
|
||||||
|
git apply ./../../tdesktop/Telegram/_openal_patch.diff
|
||||||
|
|
||||||
|
|
||||||
#####Building library
|
#####Building library
|
||||||
|
|
||||||
@ -138,9 +148,9 @@ Extract to **D:\TBuild\Libraries\**, rename **qt-everywhere-opensource-src-5.4.0
|
|||||||
Apply patch
|
Apply patch
|
||||||
|
|
||||||
* OR copy (with overwrite!) everything from **D:\TBuild\tdesktop\\\_qt\_5\_4\_0\_patch\** to **D:\TBuild\Libraries\QtStatic\**
|
* OR copy (with overwrite!) everything from **D:\TBuild\tdesktop\\\_qt\_5\_4\_0\_patch\** to **D:\TBuild\Libraries\QtStatic\**
|
||||||
* OR copy **D:\TBuild\tdesktop\\\_qt\_5\_4\_0\_patch.diff** to **D:\TBuild\Libraries\QtStatic\**, go there in Git Bash and run
|
* OR in Git Bash go to **/d/TBuild/Libraries/QtStatic/** and run
|
||||||
|
|
||||||
git apply _qt_5_4_0_patch.diff
|
git apply ./../../tdesktop/Telegram/_qt_5_4_0_patch.diff
|
||||||
|
|
||||||
#####Building library
|
#####Building library
|
||||||
|
|
||||||
|
@ -96,12 +96,11 @@ void writeLog(const wstring &msg) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void delFolder() {
|
void fullClearPath(const wstring &dir) {
|
||||||
wstring delPath = L"tupdates\\ready", delFolder = L"tupdates";
|
|
||||||
WCHAR path[4096];
|
WCHAR path[4096];
|
||||||
memcpy(path, delPath.c_str(), (delPath.size() + 1) * sizeof(WCHAR));
|
memcpy(path, dir.c_str(), (dir.size() + 1) * sizeof(WCHAR));
|
||||||
path[delPath.size() + 1] = 0;
|
path[dir.size() + 1] = 0;
|
||||||
writeLog(L"Fully clearing path '" + delPath + L"'..");
|
writeLog(L"Fully clearing path '" + dir + L"'..");
|
||||||
SHFILEOPSTRUCT file_op = {
|
SHFILEOPSTRUCT file_op = {
|
||||||
NULL,
|
NULL,
|
||||||
FO_DELETE,
|
FO_DELETE,
|
||||||
@ -116,13 +115,27 @@ void delFolder() {
|
|||||||
};
|
};
|
||||||
int res = SHFileOperation(&file_op);
|
int res = SHFileOperation(&file_op);
|
||||||
if (res) writeLog(L"Error: failed to clear path! :(");
|
if (res) writeLog(L"Error: failed to clear path! :(");
|
||||||
|
}
|
||||||
|
|
||||||
|
void delFolder() {
|
||||||
|
wstring delPathOld = L"tupdates\\ready", delPath = L"tupdates\\temp", delFolder = L"tupdates";
|
||||||
|
fullClearPath(delPathOld);
|
||||||
|
fullClearPath(delPath);
|
||||||
RemoveDirectory(delFolder.c_str());
|
RemoveDirectory(delFolder.c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
bool update() {
|
bool update() {
|
||||||
writeLog(L"Update started..");
|
writeLog(L"Update started..");
|
||||||
|
|
||||||
wstring updDir = L"tupdates\\ready";
|
wstring updDir = L"tupdates\\temp", readyFilePath = L"tupdates\\temp\\ready";
|
||||||
|
{
|
||||||
|
HANDLE readyFile = CreateFile(readyFilePath.c_str(), GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
|
||||||
|
if (readyFile != INVALID_HANDLE_VALUE) {
|
||||||
|
CloseHandle(readyFile);
|
||||||
|
} else {
|
||||||
|
updDir = L"tupdates\\ready"; // old
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
deque<wstring> dirs;
|
deque<wstring> dirs;
|
||||||
dirs.push_back(updDir);
|
dirs.push_back(updDir);
|
||||||
@ -166,10 +179,13 @@ bool update() {
|
|||||||
writeLog(L"Error: bad update, has Updater.exe! '" + tofname + L"' equal '" + exeName + L"'");
|
writeLog(L"Error: bad update, has Updater.exe! '" + tofname + L"' equal '" + exeName + L"'");
|
||||||
delFolder();
|
delFolder();
|
||||||
return false;
|
return false;
|
||||||
|
} else if (equal(fname, readyFilePath)) {
|
||||||
|
writeLog(L"Skipped ready file '" + fname + L"'");
|
||||||
|
} else {
|
||||||
|
from.push_back(fname);
|
||||||
|
to.push_back(tofname);
|
||||||
|
writeLog(L"Added file '" + fname + L"' to be copied to '" + tofname + L"'");
|
||||||
}
|
}
|
||||||
from.push_back(fname);
|
|
||||||
to.push_back(tofname);
|
|
||||||
writeLog(L"Added file '" + fname + L"' to be copied to '" + tofname + L"'");
|
|
||||||
}
|
}
|
||||||
} while (FindNextFile(findHandle, &findData));
|
} while (FindNextFile(findHandle, &findData));
|
||||||
DWORD errorCode = GetLastError();
|
DWORD errorCode = GetLastError();
|
||||||
@ -357,7 +373,7 @@ int APIENTRY WinMain(HINSTANCE instance, HINSTANCE prevInstance, LPSTR cmdParama
|
|||||||
updateRegistry();
|
updateRegistry();
|
||||||
}
|
}
|
||||||
if (writeprotected) { // if we can't clear all tupdates\ready (Updater.exe is there) - clear only version
|
if (writeprotected) { // if we can't clear all tupdates\ready (Updater.exe is there) - clear only version
|
||||||
if (DeleteFile(L"tupdates\\ready\\tdata\\version")) {
|
if (DeleteFile(L"tupdates\\temp\\tdata\\version") || DeleteFile(L"tupdates\\ready\\tdata\\version")) {
|
||||||
writeLog(L"Version file deleted!");
|
writeLog(L"Version file deleted!");
|
||||||
} else {
|
} else {
|
||||||
writeLog(L"Error: could not delete version file");
|
writeLog(L"Error: could not delete version file");
|
||||||
@ -386,7 +402,6 @@ int APIENTRY WinMain(HINSTANCE instance, HINSTANCE prevInstance, LPSTR cmdParama
|
|||||||
|
|
||||||
HRESULT hres = CoInitialize(0);
|
HRESULT hres = CoInitialize(0);
|
||||||
if (SUCCEEDED(hres)) {
|
if (SUCCEEDED(hres)) {
|
||||||
wstring lnk = L"tupdates\\ready\\temp.lnk";
|
|
||||||
IShellLink* psl;
|
IShellLink* psl;
|
||||||
HRESULT hres = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (LPVOID*)&psl);
|
HRESULT hres = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (LPVOID*)&psl);
|
||||||
if (SUCCEEDED(hres)) {
|
if (SUCCEEDED(hres)) {
|
||||||
@ -401,7 +416,12 @@ int APIENTRY WinMain(HINSTANCE instance, HINSTANCE prevInstance, LPSTR cmdParama
|
|||||||
hres = psl->QueryInterface(IID_IPersistFile, (LPVOID*)&ppf);
|
hres = psl->QueryInterface(IID_IPersistFile, (LPVOID*)&ppf);
|
||||||
|
|
||||||
if (SUCCEEDED(hres)) {
|
if (SUCCEEDED(hres)) {
|
||||||
|
wstring lnk = L"tupdates\\temp\\temp.lnk";
|
||||||
hres = ppf->Save(lnk.c_str(), TRUE);
|
hres = ppf->Save(lnk.c_str(), TRUE);
|
||||||
|
if (!SUCCEEDED(hres)) {
|
||||||
|
lnk = L"tupdates\\ready\\temp.lnk"; // old
|
||||||
|
hres = ppf->Save(lnk.c_str(), TRUE);
|
||||||
|
}
|
||||||
ppf->Release();
|
ppf->Release();
|
||||||
|
|
||||||
if (SUCCEEDED(hres)) {
|
if (SUCCEEDED(hres)) {
|
||||||
|
@ -136,7 +136,7 @@ bool remove_directory(const string &path) { // from http://stackoverflow.com/que
|
|||||||
|
|
||||||
if (!d) {
|
if (!d) {
|
||||||
writeLog("Could not open dir '%s'", path.c_str());
|
writeLog("Could not open dir '%s'", path.c_str());
|
||||||
return false;
|
return (errno == ENOENT);
|
||||||
}
|
}
|
||||||
|
|
||||||
while (struct dirent *p = readdir(d)) {
|
while (struct dirent *p = readdir(d)) {
|
||||||
@ -201,18 +201,30 @@ bool equal(string a, string b) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void delFolder() {
|
void delFolder() {
|
||||||
string delPath = workDir + "tupdates/ready", delFolder = workDir + "tupdates";
|
string delPathOld = workDir + "tupdates/ready", delPath = workDir + "tupdates/temp", delFolder = workDir + "tupdates";
|
||||||
writeLog("Fully clearing path '%s'..", delPath.c_str());
|
writeLog("Fully clearing old path '%s'..", delPathOld.c_str());
|
||||||
if (!remove_directory(delPath)) {
|
if (!remove_directory(delPathOld)) {
|
||||||
writeLog("Error: failed to clear path! :(");
|
writeLog("Error: failed to clear old path! :(");
|
||||||
}
|
}
|
||||||
rmdir(delFolder.c_str());
|
writeLog("Fully clearing path '%s'..", delPath.c_str());
|
||||||
|
if (!remove_directory(delPath)) {
|
||||||
|
writeLog("Error: failed to clear path! :(");
|
||||||
|
}
|
||||||
|
rmdir(delFolder.c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
bool update() {
|
bool update() {
|
||||||
writeLog("Update started..");
|
writeLog("Update started..");
|
||||||
|
|
||||||
string updDir = workDir + "tupdates/ready";
|
string updDir = workDir + "tupdates/temp", readyFilePath = workDir + "tupdates/temp/ready";
|
||||||
|
{
|
||||||
|
FILE *readyFile = fopen(readyFilePath.c_str(), "rb");
|
||||||
|
if (readyFile) {
|
||||||
|
fclose(readyFile);
|
||||||
|
} else {
|
||||||
|
updDir = workDir + "tupdates/ready"; // old
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
deque<string> dirs;
|
deque<string> dirs;
|
||||||
dirs.push_back(updDir);
|
dirs.push_back(updDir);
|
||||||
@ -253,9 +265,13 @@ bool update() {
|
|||||||
delFolder();
|
delFolder();
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
from.push_back(fname);
|
if (fname == readyFilePath) {
|
||||||
to.push_back(tofname);
|
writeLog("Skipped ready file '%s'", fname.c_str());
|
||||||
writeLog("Added file '%s' to be copied to '%s'", fname.c_str(), tofname.c_str());
|
} else {
|
||||||
|
from.push_back(fname);
|
||||||
|
to.push_back(tofname);
|
||||||
|
writeLog("Added file '%s' to be copied to '%s'", fname.c_str(), tofname.c_str());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
writeLog("Could not get stat() for file %s", fname.c_str());
|
writeLog("Could not get stat() for file %s", fname.c_str());
|
||||||
@ -336,7 +352,7 @@ int main(int argc, char *argv[]) {
|
|||||||
exeDir = exeName.substr(0, exeName.size() - 7);
|
exeDir = exeName.substr(0, exeName.size() - 7);
|
||||||
writeLog("Exe dir is: %s", exeDir.c_str());
|
writeLog("Exe dir is: %s", exeDir.c_str());
|
||||||
if (needupdate) {
|
if (needupdate) {
|
||||||
if (workDir.empty()) { // old app launched
|
if (workDir.empty()) { // old app launched, update prepared in tupdates/ready (not in tupdates/temp)
|
||||||
writeLog("No workdir, trying to figure it out");
|
writeLog("No workdir, trying to figure it out");
|
||||||
struct passwd *pw = getpwuid(getuid());
|
struct passwd *pw = getpwuid(getuid());
|
||||||
if (pw && pw->pw_dir && strlen(pw->pw_dir)) {
|
if (pw && pw->pw_dir && strlen(pw->pw_dir)) {
|
||||||
|
@ -55,7 +55,8 @@ void writeLog(NSString *msg) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void delFolder() {
|
void delFolder() {
|
||||||
[[NSFileManager defaultManager] removeItemAtPath:[workDir stringByAppendingString:@"tupdates/ready"] error:nil];
|
[[NSFileManager defaultManager] removeItemAtPath:[workDir stringByAppendingString:@"tupdates/ready"] error:nil]; // remove old
|
||||||
|
[[NSFileManager defaultManager] removeItemAtPath:[workDir stringByAppendingString:@"tupdates/temp"] error:nil];
|
||||||
rmdir([[workDir stringByAppendingString:@"tupdates"] fileSystemRepresentation]);
|
rmdir([[workDir stringByAppendingString:@"tupdates"] fileSystemRepresentation]);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -132,13 +133,19 @@ int main(int argc, const char * argv[]) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (update) {
|
if (update) {
|
||||||
writeLog([@"Starting update files iteration, path: " stringByAppendingString: [workDir stringByAppendingString:@"tupdates/ready"]]);
|
|
||||||
|
|
||||||
NSFileManager *fileManager = [NSFileManager defaultManager];
|
NSFileManager *fileManager = [NSFileManager defaultManager];
|
||||||
NSString *srcDir = [workDir stringByAppendingString:@"tupdates/ready/"];
|
NSString *readyFilePath = [workDir stringByAppendingString:@"tupdates/temp/ready"];
|
||||||
|
NSString *srcDir = [workDir stringByAppendingString:@"tupdates/temp/"], *srcEnum = [workDir stringByAppendingString:@"tupdates/temp"];
|
||||||
|
if (![fileManager fileExistsAtPath:readyFilePath]) {
|
||||||
|
srcDir = [workDir stringByAppendingString:@"tupdates/ready/"]; // old
|
||||||
|
srcEnum = [workDir stringByAppendingString:@"tupdates/ready"];
|
||||||
|
}
|
||||||
|
|
||||||
|
writeLog([@"Starting update files iteration, path: " stringByAppendingString: srcEnum]);
|
||||||
|
|
||||||
NSArray *keys = [NSArray arrayWithObject:NSURLIsDirectoryKey];
|
NSArray *keys = [NSArray arrayWithObject:NSURLIsDirectoryKey];
|
||||||
NSDirectoryEnumerator *enumerator = [fileManager
|
NSDirectoryEnumerator *enumerator = [fileManager
|
||||||
enumeratorAtURL:[NSURL fileURLWithPath:[workDir stringByAppendingString:@"tupdates/ready"]]
|
enumeratorAtURL:[NSURL fileURLWithPath:srcEnum]
|
||||||
includingPropertiesForKeys:keys
|
includingPropertiesForKeys:keys
|
||||||
options:0
|
options:0
|
||||||
errorHandler:^(NSURL *url, NSError *error) {
|
errorHandler:^(NSURL *url, NSError *error) {
|
||||||
@ -163,18 +170,20 @@ int main(int argc, const char * argv[]) {
|
|||||||
NSString *dstPath = [appDirFull stringByAppendingString:[pathPart substringFromIndex:r.length]];
|
NSString *dstPath = [appDirFull stringByAppendingString:[pathPart substringFromIndex:r.length]];
|
||||||
NSError *error;
|
NSError *error;
|
||||||
NSNumber *isDirectory = nil;
|
NSNumber *isDirectory = nil;
|
||||||
writeLog([[NSArray arrayWithObjects: @"Copying file ", srcPath, @" to ", dstPath, nil] componentsJoinedByString:@""]);
|
|
||||||
if (![url getResourceValue:&isDirectory forKey:NSURLIsDirectoryKey error:&error]) {
|
if (![url getResourceValue:&isDirectory forKey:NSURLIsDirectoryKey error:&error]) {
|
||||||
writeLog([@"Failed to get IsDirectory for file " stringByAppendingString:[url path]]);
|
writeLog([@"Failed to get IsDirectory for file " stringByAppendingString:[url path]]);
|
||||||
delFolder();
|
delFolder();
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
if ([isDirectory boolValue]) {
|
if ([isDirectory boolValue]) {
|
||||||
|
writeLog([[NSArray arrayWithObjects: @"Copying dir ", srcPath, @" to ", dstPath, nil] componentsJoinedByString:@""]);
|
||||||
if (![fileManager createDirectoryAtPath:dstPath withIntermediateDirectories:YES attributes:nil error:nil]) {
|
if (![fileManager createDirectoryAtPath:dstPath withIntermediateDirectories:YES attributes:nil error:nil]) {
|
||||||
writeLog([@"Failed to force path for directory " stringByAppendingString:dstPath]);
|
writeLog([@"Failed to force path for directory " stringByAppendingString:dstPath]);
|
||||||
delFolder();
|
delFolder();
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
} else if ([srcPath isEqualToString:readyFilePath]) {
|
||||||
|
writeLog([[NSArray arrayWithObjects: @"Skipping ready file ", srcPath, nil] componentsJoinedByString:@""]);
|
||||||
} else if ([fileManager fileExistsAtPath:dstPath]) {
|
} else if ([fileManager fileExistsAtPath:dstPath]) {
|
||||||
if (![[NSData dataWithContentsOfFile:srcPath] writeToFile:dstPath atomically:YES]) {
|
if (![[NSData dataWithContentsOfFile:srcPath] writeToFile:dstPath atomically:YES]) {
|
||||||
writeLog([@"Failed to edit file " stringByAppendingString:dstPath]);
|
writeLog([@"Failed to edit file " stringByAppendingString:dstPath]);
|
||||||
|
@ -29,6 +29,8 @@ Copyright (c) 2014 John Preston, https://desktop.telegram.org
|
|||||||
|
|
||||||
#include "localstorage.h"
|
#include "localstorage.h"
|
||||||
|
|
||||||
|
#include "autoupdater.h"
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
Application *mainApp = 0;
|
Application *mainApp = 0;
|
||||||
FileUploader *uploader = 0;
|
FileUploader *uploader = 0;
|
||||||
@ -180,29 +182,6 @@ Application::Application(int &argc, char **argv) : PsApplication(argc, argv),
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void Application::onAppUpdate(const MTPhelp_AppUpdate &response) {
|
|
||||||
updateRequestId = 0;
|
|
||||||
|
|
||||||
cSetLastUpdateCheck(unixtime());
|
|
||||||
Local::writeSettings();
|
|
||||||
if (response.type() == mtpc_help_noAppUpdate) {
|
|
||||||
startUpdateCheck();
|
|
||||||
} else {
|
|
||||||
updateThread = new QThread();
|
|
||||||
connect(updateThread, SIGNAL(finished()), updateThread, SLOT(deleteLater()));
|
|
||||||
updateDownloader = new PsUpdateDownloader(updateThread, response.c_help_appUpdate());
|
|
||||||
updateThread->start();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bool Application::onAppUpdateFail() {
|
|
||||||
updateRequestId = 0;
|
|
||||||
cSetLastUpdateCheck(unixtime());
|
|
||||||
Local::writeSettings();
|
|
||||||
startUpdateCheck();
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void Application::updateGotCurrent() {
|
void Application::updateGotCurrent() {
|
||||||
if (!updateReply || updateThread) return;
|
if (!updateReply || updateThread) return;
|
||||||
|
|
||||||
@ -213,7 +192,7 @@ void Application::updateGotCurrent() {
|
|||||||
if (currentVersion > AppVersion) {
|
if (currentVersion > AppVersion) {
|
||||||
updateThread = new QThread();
|
updateThread = new QThread();
|
||||||
connect(updateThread, SIGNAL(finished()), updateThread, SLOT(deleteLater()));
|
connect(updateThread, SIGNAL(finished()), updateThread, SLOT(deleteLater()));
|
||||||
updateDownloader = new PsUpdateDownloader(updateThread, m.captured(2));
|
updateDownloader = new UpdateDownloader(updateThread, m.captured(2));
|
||||||
updateThread->start();
|
updateThread->start();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -540,7 +519,6 @@ void Application::startUpdateCheck(bool forceWait) {
|
|||||||
updateReply = updateManager.get(checkVersion);
|
updateReply = updateManager.get(checkVersion);
|
||||||
connect(updateReply, SIGNAL(finished()), this, SLOT(updateGotCurrent()));
|
connect(updateReply, SIGNAL(finished()), this, SLOT(updateGotCurrent()));
|
||||||
connect(updateReply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(updateFailedCurrent(QNetworkReply::NetworkError)));
|
connect(updateReply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(updateFailedCurrent(QNetworkReply::NetworkError)));
|
||||||
// updateRequestId = MTP::send(MTPhelp_GetAppUpdate(MTP_string(cApiDeviceModel()), MTP_string(cApiSystemVersion()), MTP_string(cApiAppVersion()), MTP_string(cApiLang())), rpcDone(&Application::onAppUpdate), rpcFail(&Application::onAppUpdateFail);
|
|
||||||
emit updateChecking();
|
emit updateChecking();
|
||||||
} else {
|
} else {
|
||||||
updateCheckTimer.start((updateInSecs + 5) * 1000);
|
updateCheckTimer.start((updateInSecs + 5) * 1000);
|
||||||
@ -648,7 +626,7 @@ void Application::socketError(QLocalSocket::LocalSocketError e) {
|
|||||||
return App::quit();
|
return App::quit();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!cNoStartUpdate() && psCheckReadyUpdate()) {
|
if (!cNoStartUpdate() && checkReadyUpdate()) {
|
||||||
cSetRestartingUpdate(true);
|
cSetRestartingUpdate(true);
|
||||||
DEBUG_LOG(("Application Info: installing update instead of starting app.."));
|
DEBUG_LOG(("Application Info: installing update instead of starting app.."));
|
||||||
return App::quit();
|
return App::quit();
|
||||||
|
@ -27,6 +27,7 @@ Copyright (c) 2014 John Preston, https://desktop.telegram.org
|
|||||||
class MainWidget;
|
class MainWidget;
|
||||||
class FileUploader;
|
class FileUploader;
|
||||||
class Translator;
|
class Translator;
|
||||||
|
class UpdateDownloader;
|
||||||
|
|
||||||
class Application : public PsApplication, public RPCSender {
|
class Application : public PsApplication, public RPCSender {
|
||||||
Q_OBJECT
|
Q_OBJECT
|
||||||
@ -42,9 +43,6 @@ public:
|
|||||||
static int32 languageId();
|
static int32 languageId();
|
||||||
static MainWidget *main();
|
static MainWidget *main();
|
||||||
|
|
||||||
void onAppUpdate(const MTPhelp_AppUpdate &response);
|
|
||||||
bool onAppUpdateFail();
|
|
||||||
|
|
||||||
enum UpdatingState {
|
enum UpdatingState {
|
||||||
UpdatingNone,
|
UpdatingNone,
|
||||||
UpdatingDownload,
|
UpdatingDownload,
|
||||||
@ -80,6 +78,12 @@ public:
|
|||||||
|
|
||||||
signals:
|
signals:
|
||||||
|
|
||||||
|
void updateChecking();
|
||||||
|
void updateLatest();
|
||||||
|
void updateDownloading(qint64 ready, qint64 total);
|
||||||
|
void updateReady();
|
||||||
|
void updateFailed();
|
||||||
|
|
||||||
void peerPhotoDone(PeerId peer);
|
void peerPhotoDone(PeerId peer);
|
||||||
void peerPhotoFail(PeerId peer);
|
void peerPhotoFail(PeerId peer);
|
||||||
|
|
||||||
@ -143,7 +147,7 @@ private:
|
|||||||
QNetworkReply *updateReply;
|
QNetworkReply *updateReply;
|
||||||
SingleTimer updateCheckTimer;
|
SingleTimer updateCheckTimer;
|
||||||
QThread *updateThread;
|
QThread *updateThread;
|
||||||
PsUpdateDownloader *updateDownloader;
|
UpdateDownloader *updateDownloader;
|
||||||
|
|
||||||
QTimer writeUserConfigTimer;
|
QTimer writeUserConfigTimer;
|
||||||
|
|
||||||
|
520
Telegram/SourceFiles/autoupdater.cpp
Normal file
520
Telegram/SourceFiles/autoupdater.cpp
Normal file
@ -0,0 +1,520 @@
|
|||||||
|
/*
|
||||||
|
This file is part of Telegram Desktop,
|
||||||
|
the official desktop version of Telegram messaging app, see https://telegram.org
|
||||||
|
|
||||||
|
Telegram Desktop 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.
|
||||||
|
|
||||||
|
It 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.
|
||||||
|
|
||||||
|
Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
|
||||||
|
Copyright (c) 2014 John Preston, https://desktop.telegram.org
|
||||||
|
*/
|
||||||
|
#include "stdafx.h"
|
||||||
|
#include "application.h"
|
||||||
|
#include "pspecific.h"
|
||||||
|
#include "autoupdater.h"
|
||||||
|
|
||||||
|
#ifdef Q_OS_WIN
|
||||||
|
typedef DWORD VerInt;
|
||||||
|
typedef WCHAR VerChar;
|
||||||
|
#else
|
||||||
|
typedef int VerInt;
|
||||||
|
typedef wchar_t VerChar;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
UpdateDownloader::UpdateDownloader(QThread *thread, const QString &url) : already(0), reply(0), full(0) {
|
||||||
|
updateUrl = url;
|
||||||
|
moveToThread(thread);
|
||||||
|
manager.moveToThread(thread);
|
||||||
|
App::setProxySettings(manager);
|
||||||
|
|
||||||
|
connect(thread, SIGNAL(started()), this, SLOT(start()));
|
||||||
|
initOutput();
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::initOutput() {
|
||||||
|
QString fileName;
|
||||||
|
QRegularExpressionMatch m = QRegularExpression(qsl("/([^/\\?]+)(\\?|$)")).match(updateUrl);
|
||||||
|
if (m.hasMatch()) {
|
||||||
|
fileName = m.captured(1).replace(QRegularExpression(qsl("[^a-zA-Z0-9_\\-]")), QString());
|
||||||
|
}
|
||||||
|
if (fileName.isEmpty()) {
|
||||||
|
fileName = qsl("tupdate-%1").arg(rand());
|
||||||
|
}
|
||||||
|
QString dirStr = cWorkingDir() + qsl("tupdates/");
|
||||||
|
fileName = dirStr + fileName;
|
||||||
|
QFileInfo file(fileName);
|
||||||
|
|
||||||
|
QDir dir(dirStr);
|
||||||
|
if (dir.exists()) {
|
||||||
|
QFileInfoList all = dir.entryInfoList(QDir::Files);
|
||||||
|
for (QFileInfoList::iterator i = all.begin(), e = all.end(); i != e; ++i) {
|
||||||
|
if (i->absoluteFilePath() != file.absoluteFilePath()) {
|
||||||
|
QFile::remove(i->absoluteFilePath());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
dir.mkdir(dir.absolutePath());
|
||||||
|
}
|
||||||
|
outputFile.setFileName(fileName);
|
||||||
|
if (file.exists()) {
|
||||||
|
uint64 fullSize = file.size();
|
||||||
|
if (fullSize < INT_MAX) {
|
||||||
|
int32 goodSize = (int32)fullSize;
|
||||||
|
if (goodSize % UpdateChunk) {
|
||||||
|
goodSize = goodSize - (goodSize % UpdateChunk);
|
||||||
|
if (goodSize) {
|
||||||
|
if (outputFile.open(QIODevice::ReadOnly)) {
|
||||||
|
QByteArray goodData = outputFile.readAll().mid(0, goodSize);
|
||||||
|
outputFile.close();
|
||||||
|
if (outputFile.open(QIODevice::WriteOnly)) {
|
||||||
|
outputFile.write(goodData);
|
||||||
|
outputFile.close();
|
||||||
|
|
||||||
|
QMutexLocker lock(&mutex);
|
||||||
|
already = goodSize;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
QMutexLocker lock(&mutex);
|
||||||
|
already = goodSize;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!already) {
|
||||||
|
QFile::remove(fileName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::start() {
|
||||||
|
sendRequest();
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::sendRequest() {
|
||||||
|
QNetworkRequest req(updateUrl);
|
||||||
|
QByteArray rangeHeaderValue = "bytes=" + QByteArray::number(already) + "-";
|
||||||
|
req.setRawHeader("Range", rangeHeaderValue);
|
||||||
|
req.setAttribute(QNetworkRequest::HttpPipeliningAllowedAttribute, true);
|
||||||
|
if (reply) reply->deleteLater();
|
||||||
|
reply = manager.get(req);
|
||||||
|
connect(reply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(partFinished(qint64,qint64)));
|
||||||
|
connect(reply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(partFailed(QNetworkReply::NetworkError)));
|
||||||
|
connect(reply, SIGNAL(metaDataChanged()), this, SLOT(partMetaGot()));
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::partMetaGot() {
|
||||||
|
typedef QList<QNetworkReply::RawHeaderPair> Pairs;
|
||||||
|
Pairs pairs = reply->rawHeaderPairs();
|
||||||
|
for (Pairs::iterator i = pairs.begin(), e = pairs.end(); i != e; ++i) {
|
||||||
|
if (QString::fromUtf8(i->first).toLower() == "content-range") {
|
||||||
|
QRegularExpressionMatch m = QRegularExpression(qsl("/(\\d+)([^\\d]|$)")).match(QString::fromUtf8(i->second));
|
||||||
|
if (m.hasMatch()) {
|
||||||
|
{
|
||||||
|
QMutexLocker lock(&mutex);
|
||||||
|
full = m.captured(1).toInt();
|
||||||
|
}
|
||||||
|
emit App::app()->updateDownloading(already, full);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int32 UpdateDownloader::ready() {
|
||||||
|
QMutexLocker lock(&mutex);
|
||||||
|
return already;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32 UpdateDownloader::size() {
|
||||||
|
QMutexLocker lock(&mutex);
|
||||||
|
return full;
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::partFinished(qint64 got, qint64 total) {
|
||||||
|
if (!reply) return;
|
||||||
|
|
||||||
|
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
||||||
|
if (statusCode.isValid()) {
|
||||||
|
int status = statusCode.toInt();
|
||||||
|
if (status != 200 && status != 206 && status != 416) {
|
||||||
|
LOG(("Update Error: Bad HTTP status received in partFinished(): %1").arg(status));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!already && !full) {
|
||||||
|
QMutexLocker lock(&mutex);
|
||||||
|
full = total;
|
||||||
|
}
|
||||||
|
DEBUG_LOG(("Update Info: part %1 of %2").arg(got).arg(total));
|
||||||
|
|
||||||
|
if (!outputFile.isOpen()) {
|
||||||
|
if (!outputFile.open(QIODevice::Append)) {
|
||||||
|
LOG(("Update Error: Could not open output file '%1' for appending").arg(outputFile.fileName()));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
QByteArray r = reply->readAll();
|
||||||
|
if (!r.isEmpty()) {
|
||||||
|
outputFile.write(r);
|
||||||
|
|
||||||
|
QMutexLocker lock(&mutex);
|
||||||
|
already += r.size();
|
||||||
|
}
|
||||||
|
if (got >= total) {
|
||||||
|
reply->deleteLater();
|
||||||
|
reply = 0;
|
||||||
|
outputFile.close();
|
||||||
|
unpackUpdate();
|
||||||
|
} else {
|
||||||
|
emit App::app()->updateDownloading(already, full);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::partFailed(QNetworkReply::NetworkError e) {
|
||||||
|
if (!reply) return;
|
||||||
|
|
||||||
|
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
||||||
|
reply->deleteLater();
|
||||||
|
reply = 0;
|
||||||
|
if (statusCode.isValid()) {
|
||||||
|
int status = statusCode.toInt();
|
||||||
|
if (status == 416) { // Requested range not satisfiable
|
||||||
|
outputFile.close();
|
||||||
|
unpackUpdate();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
LOG(("Update Error: failed to download part starting from %1, error %2").arg(already).arg(e));
|
||||||
|
emit App::app()->updateFailed();
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::fatalFail() {
|
||||||
|
clearAll();
|
||||||
|
emit App::app()->updateFailed();
|
||||||
|
}
|
||||||
|
|
||||||
|
void UpdateDownloader::clearAll() {
|
||||||
|
psDeleteDir(cWorkingDir() + qsl("tupdates"));
|
||||||
|
}
|
||||||
|
|
||||||
|
//QString winapiErrorWrap() {
|
||||||
|
// WCHAR errMsg[2048];
|
||||||
|
// DWORD errorCode = GetLastError();
|
||||||
|
// LPTSTR errorText = NULL, errorTextDefault = L"(Unknown error)";
|
||||||
|
// FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, errorCode, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&errorText, 0, 0);
|
||||||
|
// if (!errorText) {
|
||||||
|
// errorText = errorTextDefault;
|
||||||
|
// }
|
||||||
|
// StringCbPrintf(errMsg, sizeof(errMsg), L"Error code: %d, error message: %s", errorCode, errorText);
|
||||||
|
// if (errorText != errorTextDefault) {
|
||||||
|
// LocalFree(errorText);
|
||||||
|
// }
|
||||||
|
// return QString::fromWCharArray(errMsg);
|
||||||
|
//}
|
||||||
|
|
||||||
|
void UpdateDownloader::unpackUpdate() {
|
||||||
|
QByteArray packed;
|
||||||
|
if (!outputFile.open(QIODevice::ReadOnly)) {
|
||||||
|
LOG(("Update Error: cant read updates file!"));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef Q_OS_WIN // use Lzma SDK for win
|
||||||
|
const int32 hSigLen = 128, hShaLen = 20, hPropsLen = LZMA_PROPS_SIZE, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hPropsLen + hOriginalSizeLen; // header
|
||||||
|
#else
|
||||||
|
const int32 hSigLen = 128, hShaLen = 20, hPropsLen = 0, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hOriginalSizeLen; // header
|
||||||
|
#endif
|
||||||
|
|
||||||
|
QByteArray compressed = outputFile.readAll();
|
||||||
|
int32 compressedLen = compressed.size() - hSize;
|
||||||
|
if (compressedLen <= 0) {
|
||||||
|
LOG(("Update Error: bad compressed size: %1").arg(compressed.size()));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
outputFile.close();
|
||||||
|
|
||||||
|
QString tempDirPath = cWorkingDir() + qsl("tupdates/temp"), readyFilePath = cWorkingDir() + qsl("tupdates/temp/ready");
|
||||||
|
psDeleteDir(tempDirPath);
|
||||||
|
|
||||||
|
QDir tempDir(tempDirPath);
|
||||||
|
if (tempDir.exists() || QFile(readyFilePath).exists()) {
|
||||||
|
LOG(("Update Error: cant clear tupdates/temp dir!"));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
|
||||||
|
uchar sha1Buffer[20];
|
||||||
|
bool goodSha1 = !memcmp(compressed.constData() + hSigLen, hashSha1(compressed.constData() + hSigLen + hShaLen, compressedLen + hPropsLen + hOriginalSizeLen, sha1Buffer), hShaLen);
|
||||||
|
if (!goodSha1) {
|
||||||
|
LOG(("Update Error: bad SHA1 hash of update file!"));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
|
||||||
|
RSA *pbKey = PEM_read_bio_RSAPublicKey(BIO_new_mem_buf(const_cast<char*>(DevChannel ? UpdatesPublicDevKey : UpdatesPublicKey), -1), 0, 0, 0);
|
||||||
|
if (!pbKey) {
|
||||||
|
LOG(("Update Error: cant read public rsa key!"));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
if (RSA_verify(NID_sha1, (const uchar*)(compressed.constData() + hSigLen), hShaLen, (const uchar*)(compressed.constData()), hSigLen, pbKey) != 1) { // verify signature
|
||||||
|
RSA_free(pbKey);
|
||||||
|
LOG(("Update Error: bad RSA signature of update file!"));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
RSA_free(pbKey);
|
||||||
|
|
||||||
|
QByteArray uncompressed;
|
||||||
|
|
||||||
|
int32 uncompressedLen;
|
||||||
|
memcpy(&uncompressedLen, compressed.constData() + hSigLen + hShaLen + hPropsLen, hOriginalSizeLen);
|
||||||
|
uncompressed.resize(uncompressedLen);
|
||||||
|
|
||||||
|
size_t resultLen = uncompressed.size();
|
||||||
|
#ifdef Q_OS_WIN // use Lzma SDK for win
|
||||||
|
SizeT srcLen = compressedLen;
|
||||||
|
int uncompressRes = LzmaUncompress((uchar*)uncompressed.data(), &resultLen, (const uchar*)(compressed.constData() + hSize), &srcLen, (const uchar*)(compressed.constData() + hSigLen + hShaLen), LZMA_PROPS_SIZE);
|
||||||
|
if (uncompressRes != SZ_OK) {
|
||||||
|
LOG(("Update Error: could not uncompress lzma, code: %1").arg(uncompressRes));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
lzma_stream stream = LZMA_STREAM_INIT;
|
||||||
|
|
||||||
|
lzma_ret ret = lzma_stream_decoder(&stream, UINT64_MAX, LZMA_CONCATENATED);
|
||||||
|
if (ret != LZMA_OK) {
|
||||||
|
const char *msg;
|
||||||
|
switch (ret) {
|
||||||
|
case LZMA_MEM_ERROR: msg = "Memory allocation failed"; break;
|
||||||
|
case LZMA_OPTIONS_ERROR: msg = "Specified preset is not supported"; break;
|
||||||
|
case LZMA_UNSUPPORTED_CHECK: msg = "Specified integrity check is not supported"; break;
|
||||||
|
default: msg = "Unknown error, possibly a bug"; break;
|
||||||
|
}
|
||||||
|
LOG(("Error initializing the decoder: %1 (error code %2)").arg(msg).arg(ret));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
|
||||||
|
stream.avail_in = compressedLen;
|
||||||
|
stream.next_in = (uint8_t*)(compressed.constData() + hSize);
|
||||||
|
stream.avail_out = resultLen;
|
||||||
|
stream.next_out = (uint8_t*)uncompressed.data();
|
||||||
|
|
||||||
|
lzma_ret res = lzma_code(&stream, LZMA_FINISH);
|
||||||
|
if (stream.avail_in) {
|
||||||
|
LOG(("Error in decompression, %1 bytes left in _in of %2 whole.").arg(stream.avail_in).arg(compressedLen));
|
||||||
|
return fatalFail();
|
||||||
|
} else if (stream.avail_out) {
|
||||||
|
LOG(("Error in decompression, %1 bytes free left in _out of %2 whole.").arg(stream.avail_out).arg(resultLen));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
lzma_end(&stream);
|
||||||
|
if (res != LZMA_OK && res != LZMA_STREAM_END) {
|
||||||
|
const char *msg;
|
||||||
|
switch (res) {
|
||||||
|
case LZMA_MEM_ERROR: msg = "Memory allocation failed"; break;
|
||||||
|
case LZMA_FORMAT_ERROR: msg = "The input data is not in the .xz format"; break;
|
||||||
|
case LZMA_OPTIONS_ERROR: msg = "Unsupported compression options"; break;
|
||||||
|
case LZMA_DATA_ERROR: msg = "Compressed file is corrupt"; break;
|
||||||
|
case LZMA_BUF_ERROR: msg = "Compressed data is truncated or otherwise corrupt"; break;
|
||||||
|
default: msg = "Unknown error, possibly a bug"; break;
|
||||||
|
}
|
||||||
|
LOG(("Error in decompression: %1 (error code %2)").arg(msg).arg(res));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
tempDir.mkdir(tempDir.absolutePath());
|
||||||
|
|
||||||
|
quint32 version;
|
||||||
|
{
|
||||||
|
QBuffer buffer(&uncompressed);
|
||||||
|
buffer.open(QIODevice::ReadOnly);
|
||||||
|
QDataStream stream(&buffer);
|
||||||
|
stream.setVersion(QDataStream::Qt_5_1);
|
||||||
|
|
||||||
|
stream >> version;
|
||||||
|
if (stream.status() != QDataStream::Ok) {
|
||||||
|
LOG(("Update Error: cant read version from downloaded stream, status: %1").arg(stream.status()));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
if (int32(version) <= AppVersion) {
|
||||||
|
LOG(("Update Error: downloaded version %1 is not greater, than mine %2").arg(version).arg(AppVersion));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
|
||||||
|
quint32 filesCount;
|
||||||
|
stream >> filesCount;
|
||||||
|
if (stream.status() != QDataStream::Ok) {
|
||||||
|
LOG(("Update Error: cant read files count from downloaded stream, status: %1").arg(stream.status()));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
if (!filesCount) {
|
||||||
|
LOG(("Update Error: update is empty!"));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
for (uint32 i = 0; i < filesCount; ++i) {
|
||||||
|
QString relativeName;
|
||||||
|
quint32 fileSize;
|
||||||
|
QByteArray fileInnerData;
|
||||||
|
bool executable = false;
|
||||||
|
|
||||||
|
stream >> relativeName >> fileSize >> fileInnerData;
|
||||||
|
#if defined Q_OS_MAC || defined Q_OS_LINUX
|
||||||
|
stream >> executable;
|
||||||
|
#endif
|
||||||
|
if (stream.status() != QDataStream::Ok) {
|
||||||
|
LOG(("Update Error: cant read file from downloaded stream, status: %1").arg(stream.status()));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
if (fileSize != quint32(fileInnerData.size())) {
|
||||||
|
LOG(("Update Error: bad file size %1 not matching data size %2").arg(fileSize).arg(fileInnerData.size()));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
|
||||||
|
QFile f(tempDirPath + '/' + relativeName);
|
||||||
|
if (!QDir().mkpath(QFileInfo(f).absolutePath())) {
|
||||||
|
LOG(("Update Error: cant mkpath for file '%1'").arg(tempDirPath + '/' + relativeName));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
if (!f.open(QIODevice::WriteOnly)) {
|
||||||
|
LOG(("Update Error: cant open file '%1' for writing").arg(tempDirPath + '/' + relativeName));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
if (f.write(fileInnerData) != fileSize) {
|
||||||
|
f.close();
|
||||||
|
LOG(("Update Error: cant write file '%1'").arg(tempDirPath + '/' + relativeName));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
f.close();
|
||||||
|
if (executable) {
|
||||||
|
QFileDevice::Permissions p = f.permissions();
|
||||||
|
p |= QFileDevice::ExeOwner | QFileDevice::ExeUser | QFileDevice::ExeGroup | QFileDevice::ExeOther;
|
||||||
|
f.setPermissions(p);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// create tdata/version file
|
||||||
|
tempDir.mkdir(QDir(tempDirPath + qsl("/tdata")).absolutePath());
|
||||||
|
std::wstring versionString = ((version % 1000) ? QString("%1.%2.%3").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000)).arg(int(version % 1000)) : QString("%1.%2").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000))).toStdWString();
|
||||||
|
|
||||||
|
VerInt versionNum = VerInt(version), versionLen = VerInt(versionString.size() * sizeof(VerChar));
|
||||||
|
VerChar versionStr[32];
|
||||||
|
memcpy(versionStr, versionString.c_str(), versionLen);
|
||||||
|
|
||||||
|
QFile fVersion(tempDirPath + qsl("/tdata/version"));
|
||||||
|
if (!fVersion.open(QIODevice::WriteOnly)) {
|
||||||
|
LOG(("Update Error: cant write version file '%1'").arg(tempDirPath + qsl("/version")));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
fVersion.write((const char*)&versionNum, sizeof(VerInt));
|
||||||
|
fVersion.write((const char*)&versionLen, sizeof(VerInt));
|
||||||
|
fVersion.write((const char*)&versionStr[0], versionLen);
|
||||||
|
fVersion.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
QFile readyFile(readyFilePath);
|
||||||
|
if (readyFile.open(QIODevice::WriteOnly)) {
|
||||||
|
if (readyFile.write("1", 1)) {
|
||||||
|
readyFile.close();
|
||||||
|
} else {
|
||||||
|
LOG(("Update Error: cant write ready file '%1'").arg(readyFilePath));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
LOG(("Update Error: cant create ready file '%1'").arg(readyFilePath));
|
||||||
|
return fatalFail();
|
||||||
|
}
|
||||||
|
outputFile.remove();
|
||||||
|
|
||||||
|
emit App::app()->updateReady();
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateDownloader::~UpdateDownloader() {
|
||||||
|
delete reply;
|
||||||
|
reply = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool checkReadyUpdate() {
|
||||||
|
QString readyFilePath = cWorkingDir() + qsl("tupdates/temp/ready"), readyPath = cWorkingDir() + qsl("tupdates/temp");
|
||||||
|
if (!QFile(readyFilePath).exists()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// check ready version
|
||||||
|
QString versionPath = readyPath + qsl("/tdata/version");
|
||||||
|
{
|
||||||
|
QFile fVersion(versionPath);
|
||||||
|
if (!fVersion.open(QIODevice::ReadOnly)) {
|
||||||
|
LOG(("Update Error: cant read version file '%1'").arg(versionPath));
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
VerInt versionNum;
|
||||||
|
if (fVersion.read((char*)&versionNum, sizeof(VerInt)) != sizeof(VerInt)) {
|
||||||
|
LOG(("Update Error: cant read version from file '%1'").arg(versionPath));
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
fVersion.close();
|
||||||
|
if (versionNum <= AppVersion) {
|
||||||
|
LOG(("Update Error: cant install version %1 having version %2").arg(versionNum).arg(AppVersion));
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef Q_OS_WIN
|
||||||
|
QString curUpdater = (cExeDir() + qsl("Updater.exe"));
|
||||||
|
QFileInfo updater(cWorkingDir() + qsl("tupdates/temp/Updater.exe"));
|
||||||
|
#elif defined Q_OS_MAC
|
||||||
|
QString curUpdater = (cExeDir() + cExeName() + qsl("/Contents/Frameworks/Updater"));
|
||||||
|
QFileInfo updater(cWorkingDir() + qsl("tupdates/temp/Telegram.app/Contents/Frameworks/Updater"));
|
||||||
|
#elif defined Q_OS_LINUX
|
||||||
|
QString curUpdater = (cExeDir() + qsl("Updater"));
|
||||||
|
QFileInfo updater(cWorkingDir() + qsl("tupdates/temp/Updater"));
|
||||||
|
#endif
|
||||||
|
if (!updater.exists()) {
|
||||||
|
QFileInfo current(curUpdater);
|
||||||
|
if (!current.exists()) {
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (!QFile(current.absoluteFilePath()).copy(updater.absoluteFilePath())) {
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#ifdef Q_OS_WIN
|
||||||
|
if (CopyFile(updater.absoluteFilePath().toStdWString().c_str(), curUpdater.toStdWString().c_str(), FALSE) == FALSE) {
|
||||||
|
DWORD errorCode = GetLastError();
|
||||||
|
if (errorCode == ERROR_ACCESS_DENIED) { // we are in write-protected dir, like Program Files
|
||||||
|
cSetWriteProtected(true);
|
||||||
|
return true;
|
||||||
|
} else {
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (DeleteFile(updater.absoluteFilePath().toStdWString().c_str()) == FALSE) {
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
#elif defined Q_OS_MAC
|
||||||
|
QDir().mkpath(QFileInfo(curUpdater).absolutePath());
|
||||||
|
DEBUG_LOG(("Update Info: moving %1 to %2..").arg(updater.absoluteFilePath()).arg(curUpdater));
|
||||||
|
if (!objc_moveFile(updater.absoluteFilePath(), curUpdater)) {
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
#elif defined Q_OS_LINUX
|
||||||
|
if (!moveFile(QFile::encodeName(updater.absoluteFilePath()).constData(), QFile::encodeName(curUpdater).constData())) {
|
||||||
|
UpdateDownloader::clearAll();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
return true;
|
||||||
|
}
|
62
Telegram/SourceFiles/autoupdater.h
Normal file
62
Telegram/SourceFiles/autoupdater.h
Normal file
@ -0,0 +1,62 @@
|
|||||||
|
/*
|
||||||
|
This file is part of Telegram Desktop,
|
||||||
|
the official desktop version of Telegram messaging app, see https://telegram.org
|
||||||
|
|
||||||
|
Telegram Desktop 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.
|
||||||
|
|
||||||
|
It 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.
|
||||||
|
|
||||||
|
Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
|
||||||
|
Copyright (c) 2014 John Preston, https://desktop.telegram.org
|
||||||
|
*/
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <QtNetwork/QLocalSocket>
|
||||||
|
#include <QtNetwork/QLocalServer>
|
||||||
|
#include <QtNetwork/QNetworkReply>
|
||||||
|
|
||||||
|
class UpdateDownloader : public QObject {
|
||||||
|
Q_OBJECT
|
||||||
|
|
||||||
|
public:
|
||||||
|
UpdateDownloader(QThread *thread, const QString &url);
|
||||||
|
|
||||||
|
void unpackUpdate();
|
||||||
|
|
||||||
|
int32 ready();
|
||||||
|
int32 size();
|
||||||
|
|
||||||
|
static void clearAll();
|
||||||
|
|
||||||
|
~UpdateDownloader();
|
||||||
|
|
||||||
|
public slots:
|
||||||
|
|
||||||
|
void start();
|
||||||
|
void partMetaGot();
|
||||||
|
void partFinished(qint64 got, qint64 total);
|
||||||
|
void partFailed(QNetworkReply::NetworkError e);
|
||||||
|
void sendRequest();
|
||||||
|
|
||||||
|
private:
|
||||||
|
void initOutput();
|
||||||
|
|
||||||
|
void fatalFail();
|
||||||
|
|
||||||
|
QString updateUrl;
|
||||||
|
QNetworkAccessManager manager;
|
||||||
|
QNetworkReply *reply;
|
||||||
|
int32 already, full;
|
||||||
|
QFile outputFile;
|
||||||
|
|
||||||
|
QMutex mutex;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
bool checkReadyUpdate();
|
@ -946,183 +946,6 @@ PsApplication::~PsApplication() {
|
|||||||
_psEventFilter = 0;
|
_psEventFilter = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const MTPDhelp_appUpdate &update) : reply(0), already(0), full(0) {
|
|
||||||
updateUrl = qs(update.vurl);
|
|
||||||
moveToThread(thread);
|
|
||||||
manager.moveToThread(thread);
|
|
||||||
App::setProxySettings(manager);
|
|
||||||
|
|
||||||
connect(thread, SIGNAL(started()), this, SLOT(start()));
|
|
||||||
initOutput();
|
|
||||||
}
|
|
||||||
|
|
||||||
PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const QString &url) : reply(0), already(0), full(0) {
|
|
||||||
updateUrl = url;
|
|
||||||
moveToThread(thread);
|
|
||||||
manager.moveToThread(thread);
|
|
||||||
App::setProxySettings(manager);
|
|
||||||
|
|
||||||
connect(thread, SIGNAL(started()), this, SLOT(start()));
|
|
||||||
initOutput();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::initOutput() {
|
|
||||||
QString fileName;
|
|
||||||
QRegularExpressionMatch m = QRegularExpression(qsl("/([^/\\?]+)(\\?|$)")).match(updateUrl);
|
|
||||||
if (m.hasMatch()) {
|
|
||||||
fileName = m.captured(1).replace(QRegularExpression(qsl("[^a-zA-Z0-9_\\-]")), QString());
|
|
||||||
}
|
|
||||||
if (fileName.isEmpty()) {
|
|
||||||
fileName = qsl("tupdate-%1").arg(rand());
|
|
||||||
}
|
|
||||||
QString dirStr = cWorkingDir() + qsl("tupdates/");
|
|
||||||
fileName = dirStr + fileName;
|
|
||||||
QFileInfo file(fileName);
|
|
||||||
|
|
||||||
QDir dir(dirStr);
|
|
||||||
if (dir.exists()) {
|
|
||||||
QFileInfoList all = dir.entryInfoList(QDir::Files);
|
|
||||||
for (QFileInfoList::iterator i = all.begin(), e = all.end(); i != e; ++i) {
|
|
||||||
if (i->absoluteFilePath() != file.absoluteFilePath()) {
|
|
||||||
QFile::remove(i->absoluteFilePath());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
dir.mkdir(dir.absolutePath());
|
|
||||||
}
|
|
||||||
outputFile.setFileName(fileName);
|
|
||||||
if (file.exists()) {
|
|
||||||
uint64 fullSize = file.size();
|
|
||||||
if (fullSize < INT_MAX) {
|
|
||||||
int32 goodSize = (int32)fullSize;
|
|
||||||
if (goodSize % UpdateChunk) {
|
|
||||||
goodSize = goodSize - (goodSize % UpdateChunk);
|
|
||||||
if (goodSize) {
|
|
||||||
if (outputFile.open(QIODevice::ReadOnly)) {
|
|
||||||
QByteArray goodData = outputFile.readAll().mid(0, goodSize);
|
|
||||||
outputFile.close();
|
|
||||||
if (outputFile.open(QIODevice::WriteOnly)) {
|
|
||||||
outputFile.write(goodData);
|
|
||||||
outputFile.close();
|
|
||||||
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already = goodSize;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already = goodSize;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (!already) {
|
|
||||||
QFile::remove(fileName);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::start() {
|
|
||||||
sendRequest();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::sendRequest() {
|
|
||||||
QNetworkRequest req(updateUrl);
|
|
||||||
QByteArray rangeHeaderValue = "bytes=" + QByteArray::number(already) + "-";// + QByteArray::number(already + cUpdateChunk() - 1);
|
|
||||||
req.setRawHeader("Range", rangeHeaderValue);
|
|
||||||
req.setAttribute(QNetworkRequest::HttpPipeliningAllowedAttribute, true);
|
|
||||||
if (reply) reply->deleteLater();
|
|
||||||
reply = manager.get(req);
|
|
||||||
connect(reply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(partFinished(qint64,qint64)));
|
|
||||||
connect(reply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(partFailed(QNetworkReply::NetworkError)));
|
|
||||||
connect(reply, SIGNAL(metaDataChanged()), this, SLOT(partMetaGot()));
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partMetaGot() {
|
|
||||||
typedef QList<QNetworkReply::RawHeaderPair> Pairs;
|
|
||||||
Pairs pairs = reply->rawHeaderPairs();
|
|
||||||
for (Pairs::iterator i = pairs.begin(), e = pairs.end(); i != e; ++i) {
|
|
||||||
if (QString::fromUtf8(i->first).toLower() == "content-range") {
|
|
||||||
QRegularExpressionMatch m = QRegularExpression(qsl("/(\\d+)([^\\d]|$)")).match(QString::fromUtf8(i->second));
|
|
||||||
if (m.hasMatch()) {
|
|
||||||
{
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
full = m.captured(1).toInt();
|
|
||||||
}
|
|
||||||
emit App::app()->updateDownloading(already, full);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int32 PsUpdateDownloader::ready() {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
return already;
|
|
||||||
}
|
|
||||||
|
|
||||||
int32 PsUpdateDownloader::size() {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
return full;
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partFinished(qint64 got, qint64 total) {
|
|
||||||
if (!reply) return;
|
|
||||||
|
|
||||||
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
|
||||||
if (statusCode.isValid()) {
|
|
||||||
int status = statusCode.toInt();
|
|
||||||
if (status != 200 && status != 206 && status != 416) {
|
|
||||||
LOG(("Update Error: Bad HTTP status received in partFinished(): %1").arg(status));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!already && !full) {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
full = total;
|
|
||||||
}
|
|
||||||
DEBUG_LOG(("Update Info: part %1 of %2").arg(got).arg(total));
|
|
||||||
|
|
||||||
if (!outputFile.isOpen()) {
|
|
||||||
if (!outputFile.open(QIODevice::Append)) {
|
|
||||||
LOG(("Update Error: Could not open output file '%1' for appending").arg(outputFile.fileName()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
QByteArray r = reply->readAll();
|
|
||||||
if (!r.isEmpty()) {
|
|
||||||
outputFile.write(r);
|
|
||||||
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already += r.size();
|
|
||||||
}
|
|
||||||
if (got >= total) {
|
|
||||||
reply->deleteLater();
|
|
||||||
reply = 0;
|
|
||||||
outputFile.close();
|
|
||||||
unpackUpdate();
|
|
||||||
} else {
|
|
||||||
emit App::app()->updateDownloading(already, full);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partFailed(QNetworkReply::NetworkError e) {
|
|
||||||
if (!reply) return;
|
|
||||||
|
|
||||||
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
|
||||||
reply->deleteLater();
|
|
||||||
reply = 0;
|
|
||||||
if (statusCode.isValid()) {
|
|
||||||
int status = statusCode.toInt();
|
|
||||||
if (status == 416) { // Requested range not satisfiable
|
|
||||||
outputFile.close();
|
|
||||||
unpackUpdate();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
LOG(("Update Error: failed to download part starting from %1, error %2").arg(already).arg(e));
|
|
||||||
emit App::app()->updateFailed();
|
|
||||||
}
|
|
||||||
|
|
||||||
bool _removeDirectory(const QString &path) { // from http://stackoverflow.com/questions/2256945/removing-a-non-empty-directory-programmatically-in-c-or-c
|
bool _removeDirectory(const QString &path) { // from http://stackoverflow.com/questions/2256945/removing-a-non-empty-directory-programmatically-in-c-or-c
|
||||||
QByteArray pathRaw = QFile::encodeName(path);
|
QByteArray pathRaw = QFile::encodeName(path);
|
||||||
DIR *d = opendir(pathRaw.constData());
|
DIR *d = opendir(pathRaw.constData());
|
||||||
@ -1154,236 +977,8 @@ bool _removeDirectory(const QString &path) { // from http://stackoverflow.com/qu
|
|||||||
return !rmdir(pathRaw.constData());
|
return !rmdir(pathRaw.constData());
|
||||||
}
|
}
|
||||||
|
|
||||||
void PsUpdateDownloader::deleteDir(const QString &dir) {
|
void psDeleteDir(const QString &dir) {
|
||||||
_removeDirectory(dir);
|
_removeDirectory(dir);
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::fatalFail() {
|
|
||||||
clearAll();
|
|
||||||
emit App::app()->updateFailed();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::clearAll() {
|
|
||||||
deleteDir(cWorkingDir() + qsl("tupdates"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef Q_OS_WIN
|
|
||||||
typedef DWORD VerInt;
|
|
||||||
typedef WCHAR VerChar;
|
|
||||||
#else
|
|
||||||
typedef int VerInt;
|
|
||||||
typedef wchar_t VerChar;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
void PsUpdateDownloader::unpackUpdate() {
|
|
||||||
QByteArray packed;
|
|
||||||
if (!outputFile.open(QIODevice::ReadOnly)) {
|
|
||||||
LOG(("Update Error: cant read updates file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
#ifdef Q_OS_WIN // use Lzma SDK for win
|
|
||||||
const int32 hSigLen = 128, hShaLen = 20, hPropsLen = LZMA_PROPS_SIZE, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hPropsLen + hOriginalSizeLen; // header
|
|
||||||
#else
|
|
||||||
const int32 hSigLen = 128, hShaLen = 20, hPropsLen = 0, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hOriginalSizeLen; // header
|
|
||||||
#endif
|
|
||||||
QByteArray compressed = outputFile.readAll();
|
|
||||||
int32 compressedLen = compressed.size() - hSize;
|
|
||||||
if (compressedLen <= 0) {
|
|
||||||
LOG(("Update Error: bad compressed size: %1").arg(compressed.size()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
outputFile.close();
|
|
||||||
|
|
||||||
QString tempDirPath = cWorkingDir() + qsl("tupdates/temp"), readyDirPath = cWorkingDir() + qsl("tupdates/ready");
|
|
||||||
deleteDir(tempDirPath);
|
|
||||||
deleteDir(readyDirPath);
|
|
||||||
|
|
||||||
QDir tempDir(tempDirPath), readyDir(readyDirPath);
|
|
||||||
if (tempDir.exists() || readyDir.exists()) {
|
|
||||||
LOG(("Update Error: cant clear tupdates/temp or tupdates/ready dir!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
uchar sha1Buffer[20];
|
|
||||||
bool goodSha1 = !memcmp(compressed.constData() + hSigLen, hashSha1(compressed.constData() + hSigLen + hShaLen, compressedLen + hPropsLen + hOriginalSizeLen, sha1Buffer), hShaLen);
|
|
||||||
if (!goodSha1) {
|
|
||||||
LOG(("Update Error: bad SHA1 hash of update file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
RSA *pbKey = PEM_read_bio_RSAPublicKey(BIO_new_mem_buf(const_cast<char*>(DevChannel ? UpdatesPublicDevKey : UpdatesPublicKey), -1), 0, 0, 0);
|
|
||||||
if (!pbKey) {
|
|
||||||
LOG(("Update Error: cant read public rsa key!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (RSA_verify(NID_sha1, (const uchar*)(compressed.constData() + hSigLen), hShaLen, (const uchar*)(compressed.constData()), hSigLen, pbKey) != 1) { // verify signature
|
|
||||||
RSA_free(pbKey);
|
|
||||||
LOG(("Update Error: bad RSA signature of update file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
RSA_free(pbKey);
|
|
||||||
|
|
||||||
QByteArray uncompressed;
|
|
||||||
|
|
||||||
int32 uncompressedLen;
|
|
||||||
memcpy(&uncompressedLen, compressed.constData() + hSigLen + hShaLen + hPropsLen, hOriginalSizeLen);
|
|
||||||
uncompressed.resize(uncompressedLen);
|
|
||||||
|
|
||||||
size_t resultLen = uncompressed.size();
|
|
||||||
#ifdef Q_OS_WIN // use Lzma SDK for win
|
|
||||||
SizeT srcLen = compressedLen;
|
|
||||||
int uncompressRes = LzmaUncompress((uchar*)uncompressed.data(), &resultLen, (const uchar*)(compressed.constData() + hSize), &srcLen, (const uchar*)(compressed.constData() + hSigLen + hShaLen), LZMA_PROPS_SIZE);
|
|
||||||
if (uncompressRes != SZ_OK) {
|
|
||||||
LOG(("Update Error: could not uncompress lzma, code: %1").arg(uncompressRes));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
lzma_stream stream = LZMA_STREAM_INIT;
|
|
||||||
|
|
||||||
lzma_ret ret = lzma_stream_decoder(&stream, UINT64_MAX, LZMA_CONCATENATED);
|
|
||||||
if (ret != LZMA_OK) {
|
|
||||||
const char *msg;
|
|
||||||
switch (ret) {
|
|
||||||
case LZMA_MEM_ERROR: msg = "Memory allocation failed"; break;
|
|
||||||
case LZMA_OPTIONS_ERROR: msg = "Specified preset is not supported"; break;
|
|
||||||
case LZMA_UNSUPPORTED_CHECK: msg = "Specified integrity check is not supported"; break;
|
|
||||||
default: msg = "Unknown error, possibly a bug"; break;
|
|
||||||
}
|
|
||||||
LOG(("Error initializing the decoder: %1 (error code %2)").arg(msg).arg(ret));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
stream.avail_in = compressedLen;
|
|
||||||
stream.next_in = (uint8_t*)(compressed.constData() + hSize);
|
|
||||||
stream.avail_out = resultLen;
|
|
||||||
stream.next_out = (uint8_t*)uncompressed.data();
|
|
||||||
|
|
||||||
lzma_ret res = lzma_code(&stream, LZMA_FINISH);
|
|
||||||
if (stream.avail_in) {
|
|
||||||
LOG(("Error in decompression, %1 bytes left in _in of %2 whole.").arg(stream.avail_in).arg(compressedLen));
|
|
||||||
return fatalFail();
|
|
||||||
} else if (stream.avail_out) {
|
|
||||||
LOG(("Error in decompression, %1 bytes free left in _out of %2 whole.").arg(stream.avail_out).arg(resultLen));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
lzma_end(&stream);
|
|
||||||
if (res != LZMA_OK && res != LZMA_STREAM_END) {
|
|
||||||
const char *msg;
|
|
||||||
switch (res) {
|
|
||||||
case LZMA_MEM_ERROR: msg = "Memory allocation failed"; break;
|
|
||||||
case LZMA_FORMAT_ERROR: msg = "The input data is not in the .xz format"; break;
|
|
||||||
case LZMA_OPTIONS_ERROR: msg = "Unsupported compression options"; break;
|
|
||||||
case LZMA_DATA_ERROR: msg = "Compressed file is corrupt"; break;
|
|
||||||
case LZMA_BUF_ERROR: msg = "Compressed data is truncated or otherwise corrupt"; break;
|
|
||||||
default: msg = "Unknown error, possibly a bug"; break;
|
|
||||||
}
|
|
||||||
LOG(("Error in decompression: %1 (error code %2)").arg(msg).arg(res));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
tempDir.mkdir(tempDir.absolutePath());
|
|
||||||
|
|
||||||
quint32 version;
|
|
||||||
{
|
|
||||||
QBuffer buffer(&uncompressed);
|
|
||||||
buffer.open(QIODevice::ReadOnly);
|
|
||||||
QDataStream stream(&buffer);
|
|
||||||
stream.setVersion(QDataStream::Qt_5_1);
|
|
||||||
|
|
||||||
stream >> version;
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read version from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (int32(version) <= AppVersion) {
|
|
||||||
LOG(("Update Error: downloaded version %1 is not greater, than mine %2").arg(version).arg(AppVersion));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
quint32 filesCount;
|
|
||||||
stream >> filesCount;
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read files count from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (!filesCount) {
|
|
||||||
LOG(("Update Error: update is empty!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
for (uint32 i = 0; i < filesCount; ++i) {
|
|
||||||
QString relativeName;
|
|
||||||
quint32 fileSize;
|
|
||||||
QByteArray fileInnerData;
|
|
||||||
bool executable = false;
|
|
||||||
|
|
||||||
stream >> relativeName >> fileSize >> fileInnerData;
|
|
||||||
#if defined Q_OS_MAC || defined Q_OS_LINUX
|
|
||||||
stream >> executable;
|
|
||||||
#endif
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read file from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (fileSize != quint32(fileInnerData.size())) {
|
|
||||||
LOG(("Update Error: bad file size %1 not matching data size %2").arg(fileSize).arg(fileInnerData.size()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
QFile f(tempDirPath + '/' + relativeName);
|
|
||||||
if (!QDir().mkpath(QFileInfo(f).absolutePath())) {
|
|
||||||
LOG(("Update Error: cant mkpath for file '%1'").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (!f.open(QIODevice::WriteOnly)) {
|
|
||||||
LOG(("Update Error: cant open file '%1' for writing").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (f.write(fileInnerData) != fileSize) {
|
|
||||||
f.close();
|
|
||||||
LOG(("Update Error: cant write file '%1'").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
f.close();
|
|
||||||
if (executable) {
|
|
||||||
QFileDevice::Permissions p = f.permissions();
|
|
||||||
p |= QFileDevice::ExeOwner | QFileDevice::ExeUser | QFileDevice::ExeGroup | QFileDevice::ExeOther;
|
|
||||||
f.setPermissions(p);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// create tdata/version file
|
|
||||||
tempDir.mkdir(QDir(tempDirPath + qsl("/tdata")).absolutePath());
|
|
||||||
std::wstring versionString = ((version % 1000) ? QString("%1.%2.%3").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000)).arg(int(version % 1000)) : QString("%1.%2").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000))).toStdWString();
|
|
||||||
|
|
||||||
VerInt versionNum = VerInt(version), versionLen = VerInt(versionString.size() * sizeof(VerChar));
|
|
||||||
VerChar versionStr[32];
|
|
||||||
memcpy(versionStr, versionString.c_str(), versionLen);
|
|
||||||
|
|
||||||
QFile fVersion(tempDirPath + qsl("/tdata/version"));
|
|
||||||
if (!fVersion.open(QIODevice::WriteOnly)) {
|
|
||||||
LOG(("Update Error: cant write version file '%1'").arg(tempDirPath + qsl("/version")));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
fVersion.write((const char*)&versionNum, sizeof(VerInt));
|
|
||||||
fVersion.write((const char*)&versionLen, sizeof(VerInt));
|
|
||||||
fVersion.write((const char*)&versionStr[0], versionLen);
|
|
||||||
fVersion.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!tempDir.rename(tempDir.absolutePath(), readyDir.absolutePath())) {
|
|
||||||
LOG(("Update Error: cant rename temp dir '%1' to ready dir '%2'").arg(tempDir.absolutePath()).arg(readyDir.absolutePath()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
deleteDir(tempDirPath);
|
|
||||||
outputFile.remove();
|
|
||||||
|
|
||||||
emit App::app()->updateReady();
|
|
||||||
}
|
|
||||||
|
|
||||||
PsUpdateDownloader::~PsUpdateDownloader() {
|
|
||||||
delete reply;
|
|
||||||
reply = 0;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
@ -1497,129 +1092,6 @@ int psFixPrevious() {
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef Q_OS_LINUX
|
|
||||||
bool moveFile(const char *from, const char *to) {
|
|
||||||
FILE *ffrom = fopen(from, "rb"), *fto = fopen(to, "wb");
|
|
||||||
if (!ffrom) {
|
|
||||||
if (fto) fclose(fto);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (!fto) {
|
|
||||||
fclose(ffrom);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
static const int BufSize = 65536;
|
|
||||||
char buf[BufSize];
|
|
||||||
while (size_t size = fread(buf, 1, BufSize, ffrom)) {
|
|
||||||
fwrite(buf, 1, size, fto);
|
|
||||||
}
|
|
||||||
|
|
||||||
struct stat fst; // from http://stackoverflow.com/questions/5486774/keeping-fileowner-and-permissions-after-copying-file-in-c
|
|
||||||
//let's say this wont fail since you already worked OK on that fp
|
|
||||||
if (fstat(fileno(ffrom), &fst) != 0) {
|
|
||||||
fclose(ffrom);
|
|
||||||
fclose(fto);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
//update to the same uid/gid
|
|
||||||
if (fchown(fileno(fto), fst.st_uid, fst.st_gid) != 0) {
|
|
||||||
fclose(ffrom);
|
|
||||||
fclose(fto);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
//update the permissions
|
|
||||||
if (fchmod(fileno(fto), fst.st_mode) != 0) {
|
|
||||||
fclose(ffrom);
|
|
||||||
fclose(fto);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
fclose(ffrom);
|
|
||||||
fclose(fto);
|
|
||||||
|
|
||||||
if (unlink(from)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
bool psCheckReadyUpdate() {
|
|
||||||
QString readyPath = cWorkingDir() + qsl("tupdates/ready");
|
|
||||||
if (!QDir(readyPath).exists()) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// check ready version
|
|
||||||
QString versionPath = readyPath + qsl("/tdata/version");
|
|
||||||
{
|
|
||||||
QFile fVersion(versionPath);
|
|
||||||
if (!fVersion.open(QIODevice::ReadOnly)) {
|
|
||||||
LOG(("Update Error: cant read version file '%1'").arg(versionPath));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
VerInt versionNum;
|
|
||||||
if (fVersion.read((char*)&versionNum, sizeof(VerInt)) != sizeof(VerInt)) {
|
|
||||||
LOG(("Update Error: cant read version from file '%1'").arg(versionPath));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
fVersion.close();
|
|
||||||
if (versionNum <= AppVersion) {
|
|
||||||
LOG(("Update Error: cant install version %1 having version %2").arg(versionNum).arg(AppVersion));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef Q_OS_WIN
|
|
||||||
QString curUpdater = (cExeDir() + "Updater.exe");
|
|
||||||
QFileInfo updater(cWorkingDir() + "tupdates/ready/Updater.exe");
|
|
||||||
#elif defined Q_OS_MAC
|
|
||||||
QString curUpdater = (cExeDir() + "Telegram.app/Contents/Frameworks/Updater");
|
|
||||||
QFileInfo updater(cWorkingDir() + "tupdates/ready/Telegram.app/Contents/Frameworks/Updater");
|
|
||||||
#elif defined Q_OS_LINUX
|
|
||||||
QString curUpdater = (cExeDir() + "Updater");
|
|
||||||
QFileInfo updater(cWorkingDir() + "tupdates/ready/Updater");
|
|
||||||
#endif
|
|
||||||
if (!updater.exists()) {
|
|
||||||
QFileInfo current(curUpdater);
|
|
||||||
if (!current.exists()) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (!QFile(current.absoluteFilePath()).copy(updater.absoluteFilePath())) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#ifdef Q_OS_WIN
|
|
||||||
if (CopyFile(updater.absoluteFilePath().toStdWString().c_str(), curUpdater.toStdWString().c_str(), FALSE) == FALSE) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (DeleteFile(updater.absoluteFilePath().toStdWString().c_str()) == FALSE) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
#elif defined Q_OS_MAC
|
|
||||||
QFileInfo to(curUpdater);
|
|
||||||
QDir().mkpath(to.absolutePath());
|
|
||||||
if (!objc_moveFile(updater.absoluteFilePath(), curUpdater)) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
#elif defined Q_OS_LINUX
|
|
||||||
if (!moveFile(QFile::encodeName(updater.absoluteFilePath()).constData(), QFile::encodeName(curUpdater).constData())) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void psPostprocessFile(const QString &name) {
|
void psPostprocessFile(const QString &name) {
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1781,8 +1253,7 @@ bool _execUpdater(bool update = true) {
|
|||||||
|
|
||||||
void psExecUpdater() {
|
void psExecUpdater() {
|
||||||
if (!_execUpdater()) {
|
if (!_execUpdater()) {
|
||||||
QString readyPath = cWorkingDir() + qsl("tupdates/ready");
|
psDeleteDir(cWorkingDir() + qsl("tupdates/temp"));
|
||||||
PsUpdateDownloader::deleteDir(readyPath);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1802,3 +1273,49 @@ void psSendToMenu(bool send, bool silent) {
|
|||||||
|
|
||||||
void psUpdateOverlayed(QWidget *widget) {
|
void psUpdateOverlayed(QWidget *widget) {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool linuxMoveFile(const char *from, const char *to) {
|
||||||
|
FILE *ffrom = fopen(from, "rb"), *fto = fopen(to, "wb");
|
||||||
|
if (!ffrom) {
|
||||||
|
if (fto) fclose(fto);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (!fto) {
|
||||||
|
fclose(ffrom);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
static const int BufSize = 65536;
|
||||||
|
char buf[BufSize];
|
||||||
|
while (size_t size = fread(buf, 1, BufSize, ffrom)) {
|
||||||
|
fwrite(buf, 1, size, fto);
|
||||||
|
}
|
||||||
|
|
||||||
|
struct stat fst; // from http://stackoverflow.com/questions/5486774/keeping-fileowner-and-permissions-after-copying-file-in-c
|
||||||
|
//let's say this wont fail since you already worked OK on that fp
|
||||||
|
if (fstat(fileno(ffrom), &fst) != 0) {
|
||||||
|
fclose(ffrom);
|
||||||
|
fclose(fto);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
//update to the same uid/gid
|
||||||
|
if (fchown(fileno(fto), fst.st_uid, fst.st_gid) != 0) {
|
||||||
|
fclose(ffrom);
|
||||||
|
fclose(fto);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
//update the permissions
|
||||||
|
if (fchmod(fileno(fto), fst.st_mode) != 0) {
|
||||||
|
fclose(ffrom);
|
||||||
|
fclose(fto);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
fclose(ffrom);
|
||||||
|
fclose(fto);
|
||||||
|
|
||||||
|
if (unlink(from)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
@ -116,55 +116,9 @@ public:
|
|||||||
void psInstallEventFilter();
|
void psInstallEventFilter();
|
||||||
~PsApplication();
|
~PsApplication();
|
||||||
|
|
||||||
signals:
|
|
||||||
|
|
||||||
void updateChecking();
|
|
||||||
void updateLatest();
|
|
||||||
void updateDownloading(qint64 ready, qint64 total);
|
|
||||||
void updateReady();
|
|
||||||
void updateFailed();
|
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class PsUpdateDownloader : public QObject {
|
void psDeleteDir(const QString &dir);
|
||||||
Q_OBJECT
|
|
||||||
|
|
||||||
public:
|
|
||||||
PsUpdateDownloader(QThread *thread, const MTPDhelp_appUpdate &update);
|
|
||||||
PsUpdateDownloader(QThread *thread, const QString &url);
|
|
||||||
|
|
||||||
void unpackUpdate();
|
|
||||||
|
|
||||||
int32 ready();
|
|
||||||
int32 size();
|
|
||||||
|
|
||||||
static void deleteDir(const QString &dir);
|
|
||||||
static void clearAll();
|
|
||||||
|
|
||||||
~PsUpdateDownloader();
|
|
||||||
|
|
||||||
public slots:
|
|
||||||
|
|
||||||
void start();
|
|
||||||
void partMetaGot();
|
|
||||||
void partFinished(qint64 got, qint64 total);
|
|
||||||
void partFailed(QNetworkReply::NetworkError e);
|
|
||||||
void sendRequest();
|
|
||||||
|
|
||||||
private:
|
|
||||||
void initOutput();
|
|
||||||
|
|
||||||
void fatalFail();
|
|
||||||
|
|
||||||
QString updateUrl;
|
|
||||||
QNetworkAccessManager manager;
|
|
||||||
QNetworkReply *reply;
|
|
||||||
int32 already, full;
|
|
||||||
QFile outputFile;
|
|
||||||
|
|
||||||
QMutex mutex;
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
void psUserActionDone();
|
void psUserActionDone();
|
||||||
bool psIdleSupported();
|
bool psIdleSupported();
|
||||||
@ -194,7 +148,6 @@ void psBringToBack(QWidget *w);
|
|||||||
int psCleanup();
|
int psCleanup();
|
||||||
int psFixPrevious();
|
int psFixPrevious();
|
||||||
|
|
||||||
bool psCheckReadyUpdate();
|
|
||||||
void psExecUpdater();
|
void psExecUpdater();
|
||||||
void psExecTelegram();
|
void psExecTelegram();
|
||||||
|
|
||||||
@ -212,3 +165,5 @@ void psUpdateOverlayed(QWidget *widget);
|
|||||||
inline QString psConvertFileUrl(const QString &url) {
|
inline QString psConvertFileUrl(const QString &url) {
|
||||||
return url;
|
return url;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool linuxMoveFile(const char *from, const char *to);
|
||||||
|
@ -522,415 +522,10 @@ PsApplication::~PsApplication() {
|
|||||||
_psEventFilter = 0;
|
_psEventFilter = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const MTPDhelp_appUpdate &update) : reply(0), already(0), full(0) {
|
void psDeleteDir(const QString &dir) {
|
||||||
updateUrl = qs(update.vurl);
|
|
||||||
moveToThread(thread);
|
|
||||||
manager.moveToThread(thread);
|
|
||||||
App::setProxySettings(manager);
|
|
||||||
|
|
||||||
connect(thread, SIGNAL(started()), this, SLOT(start()));
|
|
||||||
initOutput();
|
|
||||||
}
|
|
||||||
|
|
||||||
PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const QString &url) : reply(0), already(0), full(0) {
|
|
||||||
updateUrl = url;
|
|
||||||
moveToThread(thread);
|
|
||||||
manager.moveToThread(thread);
|
|
||||||
App::setProxySettings(manager);
|
|
||||||
|
|
||||||
connect(thread, SIGNAL(started()), this, SLOT(start()));
|
|
||||||
initOutput();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::initOutput() {
|
|
||||||
QString fileName;
|
|
||||||
QRegularExpressionMatch m = QRegularExpression(qsl("/([^/\\?]+)(\\?|$)")).match(updateUrl);
|
|
||||||
if (m.hasMatch()) {
|
|
||||||
fileName = m.captured(1).replace(QRegularExpression(qsl("[^a-zA-Z0-9_\\-]")), QString());
|
|
||||||
}
|
|
||||||
if (fileName.isEmpty()) {
|
|
||||||
fileName = qsl("tupdate-%1").arg(rand());
|
|
||||||
}
|
|
||||||
QString dirStr = cWorkingDir() + qsl("tupdates/");
|
|
||||||
fileName = dirStr + fileName;
|
|
||||||
QFileInfo file(fileName);
|
|
||||||
|
|
||||||
QDir dir(dirStr);
|
|
||||||
if (dir.exists()) {
|
|
||||||
QFileInfoList all = dir.entryInfoList(QDir::Files);
|
|
||||||
for (QFileInfoList::iterator i = all.begin(), e = all.end(); i != e; ++i) {
|
|
||||||
if (i->absoluteFilePath() != file.absoluteFilePath()) {
|
|
||||||
QFile::remove(i->absoluteFilePath());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
dir.mkdir(dir.absolutePath());
|
|
||||||
}
|
|
||||||
outputFile.setFileName(fileName);
|
|
||||||
if (file.exists()) {
|
|
||||||
uint64 fullSize = file.size();
|
|
||||||
if (fullSize < INT_MAX) {
|
|
||||||
int32 goodSize = (int32)fullSize;
|
|
||||||
if (goodSize % UpdateChunk) {
|
|
||||||
goodSize = goodSize - (goodSize % UpdateChunk);
|
|
||||||
if (goodSize) {
|
|
||||||
if (outputFile.open(QIODevice::ReadOnly)) {
|
|
||||||
QByteArray goodData = outputFile.readAll().mid(0, goodSize);
|
|
||||||
outputFile.close();
|
|
||||||
if (outputFile.open(QIODevice::WriteOnly)) {
|
|
||||||
outputFile.write(goodData);
|
|
||||||
outputFile.close();
|
|
||||||
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already = goodSize;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already = goodSize;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (!already) {
|
|
||||||
QFile::remove(fileName);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::start() {
|
|
||||||
sendRequest();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::sendRequest() {
|
|
||||||
QNetworkRequest req(updateUrl);
|
|
||||||
QByteArray rangeHeaderValue = "bytes=" + QByteArray::number(already) + "-";// + QByteArray::number(already + cUpdateChunk() - 1);
|
|
||||||
req.setRawHeader("Range", rangeHeaderValue);
|
|
||||||
req.setAttribute(QNetworkRequest::HttpPipeliningAllowedAttribute, true);
|
|
||||||
if (reply) reply->deleteLater();
|
|
||||||
reply = manager.get(req);
|
|
||||||
connect(reply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(partFinished(qint64,qint64)));
|
|
||||||
connect(reply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(partFailed(QNetworkReply::NetworkError)));
|
|
||||||
connect(reply, SIGNAL(metaDataChanged()), this, SLOT(partMetaGot()));
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partMetaGot() {
|
|
||||||
typedef QList<QNetworkReply::RawHeaderPair> Pairs;
|
|
||||||
Pairs pairs = reply->rawHeaderPairs();
|
|
||||||
for (Pairs::iterator i = pairs.begin(), e = pairs.end(); i != e; ++i) {
|
|
||||||
if (QString::fromUtf8(i->first).toLower() == "content-range") {
|
|
||||||
QRegularExpressionMatch m = QRegularExpression(qsl("/(\\d+)([^\\d]|$)")).match(QString::fromUtf8(i->second));
|
|
||||||
if (m.hasMatch()) {
|
|
||||||
{
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
full = m.captured(1).toInt();
|
|
||||||
}
|
|
||||||
emit App::app()->updateDownloading(already, full);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int32 PsUpdateDownloader::ready() {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
return already;
|
|
||||||
}
|
|
||||||
|
|
||||||
int32 PsUpdateDownloader::size() {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
return full;
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partFinished(qint64 got, qint64 total) {
|
|
||||||
if (!reply) return;
|
|
||||||
|
|
||||||
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
|
||||||
if (statusCode.isValid()) {
|
|
||||||
int status = statusCode.toInt();
|
|
||||||
if (status != 200 && status != 206 && status != 416) {
|
|
||||||
LOG(("Update Error: Bad HTTP status received in partFinished(): %1").arg(status));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!already && !full) {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
full = total;
|
|
||||||
}
|
|
||||||
DEBUG_LOG(("Update Info: part %1 of %2").arg(got).arg(total));
|
|
||||||
|
|
||||||
if (!outputFile.isOpen()) {
|
|
||||||
if (!outputFile.open(QIODevice::Append)) {
|
|
||||||
LOG(("Update Error: Could not open output file '%1' for appending").arg(outputFile.fileName()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
QByteArray r = reply->readAll();
|
|
||||||
if (!r.isEmpty()) {
|
|
||||||
outputFile.write(r);
|
|
||||||
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already += r.size();
|
|
||||||
}
|
|
||||||
if (got >= total) {
|
|
||||||
reply->deleteLater();
|
|
||||||
reply = 0;
|
|
||||||
outputFile.close();
|
|
||||||
unpackUpdate();
|
|
||||||
} else {
|
|
||||||
emit App::app()->updateDownloading(already, full);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partFailed(QNetworkReply::NetworkError e) {
|
|
||||||
if (!reply) return;
|
|
||||||
|
|
||||||
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
|
||||||
reply->deleteLater();
|
|
||||||
reply = 0;
|
|
||||||
if (statusCode.isValid()) {
|
|
||||||
int status = statusCode.toInt();
|
|
||||||
if (status == 416) { // Requested range not satisfiable
|
|
||||||
outputFile.close();
|
|
||||||
unpackUpdate();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
LOG(("Update Error: failed to download part starting from %1, error %2").arg(already).arg(e));
|
|
||||||
emit App::app()->updateFailed();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::deleteDir(const QString &dir) {
|
|
||||||
objc_deleteDir(dir);
|
objc_deleteDir(dir);
|
||||||
}
|
}
|
||||||
|
|
||||||
void PsUpdateDownloader::fatalFail() {
|
|
||||||
clearAll();
|
|
||||||
emit App::app()->updateFailed();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::clearAll() {
|
|
||||||
deleteDir(cWorkingDir() + qsl("tupdates"));
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef Q_OS_WIN
|
|
||||||
typedef DWORD VerInt;
|
|
||||||
typedef WCHAR VerChar;
|
|
||||||
#else
|
|
||||||
typedef int VerInt;
|
|
||||||
typedef wchar_t VerChar;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
void PsUpdateDownloader::unpackUpdate() {
|
|
||||||
QByteArray packed;
|
|
||||||
if (!outputFile.open(QIODevice::ReadOnly)) {
|
|
||||||
LOG(("Update Error: cant read updates file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
#ifdef Q_OS_WIN // use Lzma SDK for win
|
|
||||||
const int32 hSigLen = 128, hShaLen = 20, hPropsLen = LZMA_PROPS_SIZE, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hPropsLen + hOriginalSizeLen; // header
|
|
||||||
#else
|
|
||||||
const int32 hSigLen = 128, hShaLen = 20, hPropsLen = 0, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hOriginalSizeLen; // header
|
|
||||||
#endif
|
|
||||||
QByteArray compressed = outputFile.readAll();
|
|
||||||
int32 compressedLen = compressed.size() - hSize;
|
|
||||||
if (compressedLen <= 0) {
|
|
||||||
LOG(("Update Error: bad compressed size: %1").arg(compressed.size()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
outputFile.close();
|
|
||||||
|
|
||||||
QString tempDirPath = cWorkingDir() + qsl("tupdates/temp"), readyDirPath = cWorkingDir() + qsl("tupdates/ready");
|
|
||||||
deleteDir(tempDirPath);
|
|
||||||
deleteDir(readyDirPath);
|
|
||||||
|
|
||||||
QDir tempDir(tempDirPath), readyDir(readyDirPath);
|
|
||||||
if (tempDir.exists() || readyDir.exists()) {
|
|
||||||
LOG(("Update Error: cant clear tupdates/temp or tupdates/ready dir!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
uchar sha1Buffer[20];
|
|
||||||
bool goodSha1 = !memcmp(compressed.constData() + hSigLen, hashSha1(compressed.constData() + hSigLen + hShaLen, compressedLen + hPropsLen + hOriginalSizeLen, sha1Buffer), hShaLen);
|
|
||||||
if (!goodSha1) {
|
|
||||||
LOG(("Update Error: bad SHA1 hash of update file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
RSA *pbKey = PEM_read_bio_RSAPublicKey(BIO_new_mem_buf(const_cast<char*>(DevChannel ? UpdatesPublicDevKey : UpdatesPublicKey), -1), 0, 0, 0);
|
|
||||||
if (!pbKey) {
|
|
||||||
LOG(("Update Error: cant read public rsa key!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (RSA_verify(NID_sha1, (const uchar*)(compressed.constData() + hSigLen), hShaLen, (const uchar*)(compressed.constData()), hSigLen, pbKey) != 1) { // verify signature
|
|
||||||
RSA_free(pbKey);
|
|
||||||
LOG(("Update Error: bad RSA signature of update file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
RSA_free(pbKey);
|
|
||||||
|
|
||||||
QByteArray uncompressed;
|
|
||||||
|
|
||||||
int32 uncompressedLen;
|
|
||||||
memcpy(&uncompressedLen, compressed.constData() + hSigLen + hShaLen + hPropsLen, hOriginalSizeLen);
|
|
||||||
uncompressed.resize(uncompressedLen);
|
|
||||||
|
|
||||||
size_t resultLen = uncompressed.size();
|
|
||||||
#ifdef Q_OS_WIN // use Lzma SDK for win
|
|
||||||
SizeT srcLen = compressedLen;
|
|
||||||
int uncompressRes = LzmaUncompress((uchar*)uncompressed.data(), &resultLen, (const uchar*)(compressed.constData() + hSize), &srcLen, (const uchar*)(compressed.constData() + hSigLen + hShaLen), LZMA_PROPS_SIZE);
|
|
||||||
if (uncompressRes != SZ_OK) {
|
|
||||||
LOG(("Update Error: could not uncompress lzma, code: %1").arg(uncompressRes));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
lzma_stream stream = LZMA_STREAM_INIT;
|
|
||||||
|
|
||||||
lzma_ret ret = lzma_stream_decoder(&stream, UINT64_MAX, LZMA_CONCATENATED);
|
|
||||||
if (ret != LZMA_OK) {
|
|
||||||
const char *msg;
|
|
||||||
switch (ret) {
|
|
||||||
case LZMA_MEM_ERROR: msg = "Memory allocation failed"; break;
|
|
||||||
case LZMA_OPTIONS_ERROR: msg = "Specified preset is not supported"; break;
|
|
||||||
case LZMA_UNSUPPORTED_CHECK: msg = "Specified integrity check is not supported"; break;
|
|
||||||
default: msg = "Unknown error, possibly a bug"; break;
|
|
||||||
}
|
|
||||||
LOG(("Error initializing the decoder: %1 (error code %2)").arg(msg).arg(ret));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
stream.avail_in = compressedLen;
|
|
||||||
stream.next_in = (uint8_t*)(compressed.constData() + hSize);
|
|
||||||
stream.avail_out = resultLen;
|
|
||||||
stream.next_out = (uint8_t*)uncompressed.data();
|
|
||||||
|
|
||||||
lzma_ret res = lzma_code(&stream, LZMA_FINISH);
|
|
||||||
if (stream.avail_in) {
|
|
||||||
LOG(("Error in decompression, %1 bytes left in _in of %2 whole.").arg(stream.avail_in).arg(compressedLen));
|
|
||||||
return fatalFail();
|
|
||||||
} else if (stream.avail_out) {
|
|
||||||
LOG(("Error in decompression, %1 bytes free left in _out of %2 whole.").arg(stream.avail_out).arg(resultLen));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
lzma_end(&stream);
|
|
||||||
if (res != LZMA_OK && res != LZMA_STREAM_END) {
|
|
||||||
const char *msg;
|
|
||||||
switch (res) {
|
|
||||||
case LZMA_MEM_ERROR: msg = "Memory allocation failed"; break;
|
|
||||||
case LZMA_FORMAT_ERROR: msg = "The input data is not in the .xz format"; break;
|
|
||||||
case LZMA_OPTIONS_ERROR: msg = "Unsupported compression options"; break;
|
|
||||||
case LZMA_DATA_ERROR: msg = "Compressed file is corrupt"; break;
|
|
||||||
case LZMA_BUF_ERROR: msg = "Compressed data is truncated or otherwise corrupt"; break;
|
|
||||||
default: msg = "Unknown error, possibly a bug"; break;
|
|
||||||
}
|
|
||||||
LOG(("Error in decompression: %1 (error code %2)").arg(msg).arg(res));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
tempDir.mkdir(tempDir.absolutePath());
|
|
||||||
|
|
||||||
quint32 version;
|
|
||||||
{
|
|
||||||
QBuffer buffer(&uncompressed);
|
|
||||||
buffer.open(QIODevice::ReadOnly);
|
|
||||||
QDataStream stream(&buffer);
|
|
||||||
stream.setVersion(QDataStream::Qt_5_1);
|
|
||||||
|
|
||||||
stream >> version;
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read version from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (version <= AppVersion) {
|
|
||||||
LOG(("Update Error: downloaded version %1 is not greater, than mine %2").arg(version).arg(AppVersion));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
quint32 filesCount;
|
|
||||||
stream >> filesCount;
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read files count from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (!filesCount) {
|
|
||||||
LOG(("Update Error: update is empty!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
for (uint32 i = 0; i < filesCount; ++i) {
|
|
||||||
QString relativeName;
|
|
||||||
quint32 fileSize;
|
|
||||||
QByteArray fileInnerData;
|
|
||||||
bool executable = false;
|
|
||||||
|
|
||||||
stream >> relativeName >> fileSize >> fileInnerData;
|
|
||||||
#if defined Q_OS_MAC || defined Q_OS_LINUX
|
|
||||||
stream >> executable;
|
|
||||||
#endif
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read file from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (fileSize != quint32(fileInnerData.size())) {
|
|
||||||
LOG(("Update Error: bad file size %1 not matching data size %2").arg(fileSize).arg(fileInnerData.size()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
QFile f(tempDirPath + '/' + relativeName);
|
|
||||||
if (!QDir().mkpath(QFileInfo(f).absolutePath())) {
|
|
||||||
LOG(("Update Error: cant mkpath for file '%1'").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (!f.open(QIODevice::WriteOnly)) {
|
|
||||||
LOG(("Update Error: cant open file '%1' for writing").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (f.write(fileInnerData) != fileSize) {
|
|
||||||
f.close();
|
|
||||||
LOG(("Update Error: cant write file '%1'").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
f.close();
|
|
||||||
if (executable) {
|
|
||||||
QFileDevice::Permissions p = f.permissions();
|
|
||||||
p |= QFileDevice::ExeOwner | QFileDevice::ExeUser | QFileDevice::ExeGroup | QFileDevice::ExeOther;
|
|
||||||
f.setPermissions(p);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// create tdata/version file
|
|
||||||
tempDir.mkdir(QDir(tempDirPath + qsl("/tdata")).absolutePath());
|
|
||||||
std::wstring versionString = ((version % 1000) ? QString("%1.%2.%3").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000)).arg(int(version % 1000)) : QString("%1.%2").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000))).toStdWString();
|
|
||||||
|
|
||||||
VerInt versionNum = VerInt(version), versionLen = VerInt(versionString.size() * sizeof(VerChar));
|
|
||||||
VerChar versionStr[32];
|
|
||||||
memcpy(versionStr, versionString.c_str(), versionLen);
|
|
||||||
|
|
||||||
QFile fVersion(tempDirPath + qsl("/tdata/version"));
|
|
||||||
if (!fVersion.open(QIODevice::WriteOnly)) {
|
|
||||||
LOG(("Update Error: cant write version file '%1'").arg(tempDirPath + qsl("/version")));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
fVersion.write((const char*)&versionNum, sizeof(VerInt));
|
|
||||||
fVersion.write((const char*)&versionLen, sizeof(VerInt));
|
|
||||||
fVersion.write((const char*)&versionStr[0], versionLen);
|
|
||||||
fVersion.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!tempDir.rename(tempDir.absolutePath(), readyDir.absolutePath())) {
|
|
||||||
LOG(("Update Error: cant rename temp dir '%1' to ready dir '%2'").arg(tempDir.absolutePath()).arg(readyDir.absolutePath()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
deleteDir(tempDirPath);
|
|
||||||
outputFile.remove();
|
|
||||||
|
|
||||||
emit App::app()->updateReady();
|
|
||||||
}
|
|
||||||
|
|
||||||
PsUpdateDownloader::~PsUpdateDownloader() {
|
|
||||||
delete reply;
|
|
||||||
reply = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
uint64 _lastUserAction = 0;
|
uint64 _lastUserAction = 0;
|
||||||
}
|
}
|
||||||
@ -1029,73 +624,6 @@ int psFixPrevious() {
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool psCheckReadyUpdate() {
|
|
||||||
QString readyPath = cWorkingDir() + qsl("tupdates/ready");
|
|
||||||
if (!QDir(readyPath).exists()) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// check ready version
|
|
||||||
QString versionPath = readyPath + qsl("/tdata/version");
|
|
||||||
{
|
|
||||||
QFile fVersion(versionPath);
|
|
||||||
if (!fVersion.open(QIODevice::ReadOnly)) {
|
|
||||||
LOG(("Update Error: cant read version file '%1'").arg(versionPath));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
VerInt versionNum;
|
|
||||||
if (fVersion.read((char*)&versionNum, sizeof(VerInt)) != sizeof(VerInt)) {
|
|
||||||
LOG(("Update Error: cant read version from file '%1'").arg(versionPath));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
fVersion.close();
|
|
||||||
if (versionNum <= AppVersion) {
|
|
||||||
LOG(("Update Error: cant install version %1 having version %2").arg(versionNum).arg(AppVersion));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef Q_OS_WIN
|
|
||||||
QString curUpdater = (cExeDir() + qsl("Updater.exe"));
|
|
||||||
QFileInfo updater(cWorkingDir() + qsl("tupdates/ready/Updater.exe"));
|
|
||||||
#elif defined Q_OS_MAC
|
|
||||||
QString curUpdater = (cExeDir() + cExeName() + qsl("/Contents/Frameworks/Updater"));
|
|
||||||
QFileInfo updater(cWorkingDir() + qsl("tupdates/ready/Telegram.app/Contents/Frameworks/Updater"));
|
|
||||||
#endif
|
|
||||||
if (!updater.exists()) {
|
|
||||||
QFileInfo current(curUpdater);
|
|
||||||
if (!current.exists()) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (!QFile(current.absoluteFilePath()).copy(updater.absoluteFilePath())) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#ifdef Q_OS_WIN
|
|
||||||
if (CopyFile(updater.absoluteFilePath().toStdWString().c_str(), curUpdater.toStdWString().c_str(), FALSE) == FALSE) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (DeleteFile(updater.absoluteFilePath().toStdWString().c_str()) == FALSE) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
#elif defined Q_OS_MAC
|
|
||||||
QDir().mkpath(QFileInfo(curUpdater).absolutePath());
|
|
||||||
DEBUG_LOG(("Update Info: moving %1 to %2..").arg(updater.absoluteFilePath()).arg(curUpdater));
|
|
||||||
if (!objc_moveFile(updater.absoluteFilePath(), curUpdater)) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool psShowOpenWithMenu(int x, int y, const QString &file) {
|
bool psShowOpenWithMenu(int x, int y, const QString &file) {
|
||||||
return objc_showOpenWithMenu(x, y, file);
|
return objc_showOpenWithMenu(x, y, file);
|
||||||
}
|
}
|
||||||
@ -1125,8 +653,7 @@ void psRegisterCustomScheme() {
|
|||||||
|
|
||||||
void psExecUpdater() {
|
void psExecUpdater() {
|
||||||
if (!objc_execUpdater()) {
|
if (!objc_execUpdater()) {
|
||||||
QString readyPath = cWorkingDir() + qsl("tupdates/ready");
|
psDeleteDir(cWorkingDir() + qsl("tupdates/temp"));
|
||||||
PsUpdateDownloader::deleteDir(readyPath);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -144,55 +144,9 @@ public:
|
|||||||
void psInstallEventFilter();
|
void psInstallEventFilter();
|
||||||
~PsApplication();
|
~PsApplication();
|
||||||
|
|
||||||
signals:
|
|
||||||
|
|
||||||
void updateChecking();
|
|
||||||
void updateLatest();
|
|
||||||
void updateDownloading(qint64 ready, qint64 total);
|
|
||||||
void updateReady();
|
|
||||||
void updateFailed();
|
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class PsUpdateDownloader : public QObject {
|
void psDeleteDir(const QString &dir);
|
||||||
Q_OBJECT
|
|
||||||
|
|
||||||
public:
|
|
||||||
PsUpdateDownloader(QThread *thread, const MTPDhelp_appUpdate &update);
|
|
||||||
PsUpdateDownloader(QThread *thread, const QString &url);
|
|
||||||
|
|
||||||
void unpackUpdate();
|
|
||||||
|
|
||||||
int32 ready();
|
|
||||||
int32 size();
|
|
||||||
|
|
||||||
static void deleteDir(const QString &dir);
|
|
||||||
static void clearAll();
|
|
||||||
|
|
||||||
~PsUpdateDownloader();
|
|
||||||
|
|
||||||
public slots:
|
|
||||||
|
|
||||||
void start();
|
|
||||||
void partMetaGot();
|
|
||||||
void partFinished(qint64 got, qint64 total);
|
|
||||||
void partFailed(QNetworkReply::NetworkError e);
|
|
||||||
void sendRequest();
|
|
||||||
|
|
||||||
private:
|
|
||||||
void initOutput();
|
|
||||||
|
|
||||||
void fatalFail();
|
|
||||||
|
|
||||||
QString updateUrl;
|
|
||||||
QNetworkAccessManager manager;
|
|
||||||
QNetworkReply *reply;
|
|
||||||
int32 already, full;
|
|
||||||
QFile outputFile;
|
|
||||||
|
|
||||||
QMutex mutex;
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
void psUserActionDone();
|
void psUserActionDone();
|
||||||
bool psIdleSupported();
|
bool psIdleSupported();
|
||||||
@ -222,7 +176,6 @@ void psBringToBack(QWidget *w);
|
|||||||
int psCleanup();
|
int psCleanup();
|
||||||
int psFixPrevious();
|
int psFixPrevious();
|
||||||
|
|
||||||
bool psCheckReadyUpdate();
|
|
||||||
void psExecUpdater();
|
void psExecUpdater();
|
||||||
void psExecTelegram();
|
void psExecTelegram();
|
||||||
|
|
||||||
|
@ -1385,184 +1385,7 @@ PsApplication::~PsApplication() {
|
|||||||
_psEventFilter = 0;
|
_psEventFilter = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const MTPDhelp_appUpdate &update) : already(0), reply(0), full(0) {
|
void psDeleteDir(const QString &dir) {
|
||||||
updateUrl = qs(update.vurl);
|
|
||||||
moveToThread(thread);
|
|
||||||
manager.moveToThread(thread);
|
|
||||||
App::setProxySettings(manager);
|
|
||||||
|
|
||||||
connect(thread, SIGNAL(started()), this, SLOT(start()));
|
|
||||||
initOutput();
|
|
||||||
}
|
|
||||||
|
|
||||||
PsUpdateDownloader::PsUpdateDownloader(QThread *thread, const QString &url) : already(0), reply(0), full(0) {
|
|
||||||
updateUrl = url;
|
|
||||||
moveToThread(thread);
|
|
||||||
manager.moveToThread(thread);
|
|
||||||
App::setProxySettings(manager);
|
|
||||||
|
|
||||||
connect(thread, SIGNAL(started()), this, SLOT(start()));
|
|
||||||
initOutput();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::initOutput() {
|
|
||||||
QString fileName;
|
|
||||||
QRegularExpressionMatch m = QRegularExpression(qsl("/([^/\\?]+)(\\?|$)")).match(updateUrl);
|
|
||||||
if (m.hasMatch()) {
|
|
||||||
fileName = m.captured(1).replace(QRegularExpression(qsl("[^a-zA-Z0-9_\\-]")), QString());
|
|
||||||
}
|
|
||||||
if (fileName.isEmpty()) {
|
|
||||||
fileName = qsl("tupdate-%1").arg(rand());
|
|
||||||
}
|
|
||||||
QString dirStr = cWorkingDir() + qsl("tupdates/");
|
|
||||||
fileName = dirStr + fileName;
|
|
||||||
QFileInfo file(fileName);
|
|
||||||
|
|
||||||
QDir dir(dirStr);
|
|
||||||
if (dir.exists()) {
|
|
||||||
QFileInfoList all = dir.entryInfoList(QDir::Files);
|
|
||||||
for (QFileInfoList::iterator i = all.begin(), e = all.end(); i != e; ++i) {
|
|
||||||
if (i->absoluteFilePath() != file.absoluteFilePath()) {
|
|
||||||
QFile::remove(i->absoluteFilePath());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
dir.mkdir(dir.absolutePath());
|
|
||||||
}
|
|
||||||
outputFile.setFileName(fileName);
|
|
||||||
if (file.exists()) {
|
|
||||||
uint64 fullSize = file.size();
|
|
||||||
if (fullSize < INT_MAX) {
|
|
||||||
int32 goodSize = (int32)fullSize;
|
|
||||||
if (goodSize % UpdateChunk) {
|
|
||||||
goodSize = goodSize - (goodSize % UpdateChunk);
|
|
||||||
if (goodSize) {
|
|
||||||
if (outputFile.open(QIODevice::ReadOnly)) {
|
|
||||||
QByteArray goodData = outputFile.readAll().mid(0, goodSize);
|
|
||||||
outputFile.close();
|
|
||||||
if (outputFile.open(QIODevice::WriteOnly)) {
|
|
||||||
outputFile.write(goodData);
|
|
||||||
outputFile.close();
|
|
||||||
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already = goodSize;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already = goodSize;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (!already) {
|
|
||||||
QFile::remove(fileName);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::start() {
|
|
||||||
sendRequest();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::sendRequest() {
|
|
||||||
QNetworkRequest req(updateUrl);
|
|
||||||
QByteArray rangeHeaderValue = "bytes=" + QByteArray::number(already) + "-";// + QByteArray::number(already + cUpdateChunk() - 1);
|
|
||||||
req.setRawHeader("Range", rangeHeaderValue);
|
|
||||||
req.setAttribute(QNetworkRequest::HttpPipeliningAllowedAttribute, true);
|
|
||||||
if (reply) reply->deleteLater();
|
|
||||||
reply = manager.get(req);
|
|
||||||
connect(reply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(partFinished(qint64,qint64)));
|
|
||||||
connect(reply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(partFailed(QNetworkReply::NetworkError)));
|
|
||||||
connect(reply, SIGNAL(metaDataChanged()), this, SLOT(partMetaGot()));
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partMetaGot() {
|
|
||||||
typedef QList<QNetworkReply::RawHeaderPair> Pairs;
|
|
||||||
Pairs pairs = reply->rawHeaderPairs();
|
|
||||||
for (Pairs::iterator i = pairs.begin(), e = pairs.end(); i != e; ++i) {
|
|
||||||
if (QString::fromUtf8(i->first).toLower() == "content-range") {
|
|
||||||
QRegularExpressionMatch m = QRegularExpression(qsl("/(\\d+)([^\\d]|$)")).match(QString::fromUtf8(i->second));
|
|
||||||
if (m.hasMatch()) {
|
|
||||||
{
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
full = m.captured(1).toInt();
|
|
||||||
}
|
|
||||||
emit App::app()->updateDownloading(already, full);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int32 PsUpdateDownloader::ready() {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
return already;
|
|
||||||
}
|
|
||||||
|
|
||||||
int32 PsUpdateDownloader::size() {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
return full;
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partFinished(qint64 got, qint64 total) {
|
|
||||||
if (!reply) return;
|
|
||||||
|
|
||||||
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
|
||||||
if (statusCode.isValid()) {
|
|
||||||
int status = statusCode.toInt();
|
|
||||||
if (status != 200 && status != 206 && status != 416) {
|
|
||||||
LOG(("Update Error: Bad HTTP status received in partFinished(): %1").arg(status));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!already && !full) {
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
full = total;
|
|
||||||
}
|
|
||||||
DEBUG_LOG(("Update Info: part %1 of %2").arg(got).arg(total));
|
|
||||||
|
|
||||||
if (!outputFile.isOpen()) {
|
|
||||||
if (!outputFile.open(QIODevice::Append)) {
|
|
||||||
LOG(("Update Error: Could not open output file '%1' for appending").arg(outputFile.fileName()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
QByteArray r = reply->readAll();
|
|
||||||
if (!r.isEmpty()) {
|
|
||||||
outputFile.write(r);
|
|
||||||
|
|
||||||
QMutexLocker lock(&mutex);
|
|
||||||
already += r.size();
|
|
||||||
}
|
|
||||||
if (got >= total) {
|
|
||||||
reply->deleteLater();
|
|
||||||
reply = 0;
|
|
||||||
outputFile.close();
|
|
||||||
unpackUpdate();
|
|
||||||
} else {
|
|
||||||
emit App::app()->updateDownloading(already, full);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::partFailed(QNetworkReply::NetworkError e) {
|
|
||||||
if (!reply) return;
|
|
||||||
|
|
||||||
QVariant statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
|
||||||
reply->deleteLater();
|
|
||||||
reply = 0;
|
|
||||||
if (statusCode.isValid()) {
|
|
||||||
int status = statusCode.toInt();
|
|
||||||
if (status == 416) { // Requested range not satisfiable
|
|
||||||
outputFile.close();
|
|
||||||
unpackUpdate();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
LOG(("Update Error: failed to download part starting from %1, error %2").arg(already).arg(e));
|
|
||||||
emit App::app()->updateFailed();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::deleteDir(const QString &dir) {
|
|
||||||
std::wstring wDir = QDir::toNativeSeparators(dir).toStdWString();
|
std::wstring wDir = QDir::toNativeSeparators(dir).toStdWString();
|
||||||
WCHAR path[4096];
|
WCHAR path[4096];
|
||||||
memcpy(path, wDir.c_str(), (wDir.size() + 1) * sizeof(WCHAR));
|
memcpy(path, wDir.c_str(), (wDir.size() + 1) * sizeof(WCHAR));
|
||||||
@ -1582,185 +1405,6 @@ void PsUpdateDownloader::deleteDir(const QString &dir) {
|
|||||||
int res = SHFileOperation(&file_op);
|
int res = SHFileOperation(&file_op);
|
||||||
}
|
}
|
||||||
|
|
||||||
void PsUpdateDownloader::fatalFail() {
|
|
||||||
clearAll();
|
|
||||||
emit App::app()->updateFailed();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::clearAll() {
|
|
||||||
deleteDir(cWorkingDir() + qsl("tupdates"));
|
|
||||||
}
|
|
||||||
|
|
||||||
QString winapiErrorWrap() {
|
|
||||||
WCHAR errMsg[2048];
|
|
||||||
DWORD errorCode = GetLastError();
|
|
||||||
LPTSTR errorText = NULL, errorTextDefault = L"(Unknown error)";
|
|
||||||
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, errorCode, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&errorText, 0, 0);
|
|
||||||
if (!errorText) {
|
|
||||||
errorText = errorTextDefault;
|
|
||||||
}
|
|
||||||
StringCbPrintf(errMsg, sizeof(errMsg), L"Error code: %d, error message: %s", errorCode, errorText);
|
|
||||||
if (errorText != errorTextDefault) {
|
|
||||||
LocalFree(errorText);
|
|
||||||
}
|
|
||||||
return QString::fromWCharArray(errMsg);
|
|
||||||
}
|
|
||||||
|
|
||||||
void PsUpdateDownloader::unpackUpdate() {
|
|
||||||
QByteArray packed;
|
|
||||||
if (!outputFile.open(QIODevice::ReadOnly)) {
|
|
||||||
LOG(("Update Error: cant read updates file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
const int32 hSigLen = 128, hShaLen = 20, hPropsLen = LZMA_PROPS_SIZE, hOriginalSizeLen = sizeof(int32), hSize = hSigLen + hShaLen + hPropsLen + hOriginalSizeLen; // header
|
|
||||||
|
|
||||||
QByteArray compressed = outputFile.readAll();
|
|
||||||
int32 compressedLen = compressed.size() - hSize;
|
|
||||||
if (compressedLen <= 0) {
|
|
||||||
LOG(("Update Error: bad compressed size: %1").arg(compressed.size()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
outputFile.close();
|
|
||||||
|
|
||||||
QString tempDirPath = cWorkingDir() + qsl("tupdates/temp"), readyDirPath = cWorkingDir() + qsl("tupdates/ready");
|
|
||||||
deleteDir(tempDirPath);
|
|
||||||
deleteDir(readyDirPath);
|
|
||||||
|
|
||||||
{
|
|
||||||
QDir tempDir(tempDirPath), readyDir(readyDirPath);
|
|
||||||
if (tempDir.exists() || readyDir.exists()) {
|
|
||||||
LOG(("Update Error: cant clear tupdates/temp or tupdates/ready dir!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
tempDirPath = tempDir.absolutePath();
|
|
||||||
readyDirPath = readyDir.absolutePath();
|
|
||||||
|
|
||||||
uchar sha1Buffer[20];
|
|
||||||
bool goodSha1 = !memcmp(compressed.constData() + hSigLen, hashSha1(compressed.constData() + hSigLen + hShaLen, compressedLen + hPropsLen + hOriginalSizeLen, sha1Buffer), hShaLen);
|
|
||||||
if (!goodSha1) {
|
|
||||||
LOG(("Update Error: bad SHA1 hash of update file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
RSA *pbKey = PEM_read_bio_RSAPublicKey(BIO_new_mem_buf(const_cast<char*>(DevChannel ? UpdatesPublicDevKey : UpdatesPublicKey), -1), 0, 0, 0);
|
|
||||||
if (!pbKey) {
|
|
||||||
LOG(("Update Error: cant read public rsa key!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (RSA_verify(NID_sha1, (const uchar*)(compressed.constData() + hSigLen), hShaLen, (const uchar*)(compressed.constData()), hSigLen, pbKey) != 1) { // verify signature
|
|
||||||
RSA_free(pbKey);
|
|
||||||
LOG(("Update Error: bad RSA signature of update file!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
RSA_free(pbKey);
|
|
||||||
|
|
||||||
QByteArray uncompressed;
|
|
||||||
|
|
||||||
int32 uncompressedLen;
|
|
||||||
memcpy(&uncompressedLen, compressed.constData() + hSigLen + hShaLen + hPropsLen, hOriginalSizeLen);
|
|
||||||
uncompressed.resize(uncompressedLen);
|
|
||||||
|
|
||||||
size_t resultLen = uncompressed.size();
|
|
||||||
SizeT srcLen = compressedLen;
|
|
||||||
int uncompressRes = LzmaUncompress((uchar*)uncompressed.data(), &resultLen, (const uchar*)(compressed.constData() + hSize), &srcLen, (const uchar*)(compressed.constData() + hSigLen + hShaLen), LZMA_PROPS_SIZE);
|
|
||||||
if (uncompressRes != SZ_OK) {
|
|
||||||
LOG(("Update Error: could not uncompress lzma, code: %1").arg(uncompressRes));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
QDir().mkdir(tempDirPath);
|
|
||||||
|
|
||||||
quint32 version;
|
|
||||||
|
|
||||||
QBuffer buffer(&uncompressed);
|
|
||||||
buffer.open(QIODevice::ReadOnly);
|
|
||||||
QDataStream stream(&buffer);
|
|
||||||
stream.setVersion(QDataStream::Qt_5_1);
|
|
||||||
|
|
||||||
stream >> version;
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read version from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (version <= AppVersion) {
|
|
||||||
LOG(("Update Error: downloaded version %1 is not greater, than mine %2").arg(version).arg(AppVersion));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
quint32 filesCount;
|
|
||||||
stream >> filesCount;
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read files count from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (!filesCount) {
|
|
||||||
LOG(("Update Error: update is empty!"));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
for (int32 i = 0; i < filesCount; ++i) {
|
|
||||||
QString relativeName;
|
|
||||||
quint32 fileSize;
|
|
||||||
QByteArray fileInnerData;
|
|
||||||
|
|
||||||
stream >> relativeName >> fileSize >> fileInnerData;
|
|
||||||
if (stream.status() != QDataStream::Ok) {
|
|
||||||
LOG(("Update Error: cant read file from downloaded stream, status: %1").arg(stream.status()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (fileSize != fileInnerData.size()) {
|
|
||||||
LOG(("Update Error: bad file size %1 not matching data size %2").arg(fileSize).arg(fileInnerData.size()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
|
|
||||||
QFile f(tempDirPath + '/' + relativeName);
|
|
||||||
if (!f.open(QIODevice::WriteOnly)) {
|
|
||||||
LOG(("Update Error: cant open file '%1' for writing").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
if (f.write(fileInnerData) != fileSize) {
|
|
||||||
f.close();
|
|
||||||
LOG(("Update Error: cant write file '%1'").arg(tempDirPath + '/' + relativeName));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
f.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
// create tdata/version file
|
|
||||||
QDir().mkdir(tempDirPath + qsl("/tdata"));
|
|
||||||
std::wstring versionString = ((version % 1000) ? QString("%1.%2.%3").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000)).arg(int(version % 1000)) : QString("%1.%2").arg(int(version / 1000000)).arg(int((version % 1000000) / 1000))).toStdWString();
|
|
||||||
DWORD versionNum = DWORD(version), versionLen = DWORD(versionString.size() * sizeof(WCHAR));
|
|
||||||
WCHAR versionStr[32];
|
|
||||||
memcpy(versionStr, versionString.c_str(), versionLen);
|
|
||||||
|
|
||||||
QFile fVersion(tempDirPath + qsl("/tdata/version"));
|
|
||||||
if (!fVersion.open(QIODevice::WriteOnly)) {
|
|
||||||
LOG(("Update Error: cant write version file '%1'").arg(tempDirPath + qsl("/tdata/version")));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
fVersion.write((const char*)&versionNum, sizeof(DWORD));
|
|
||||||
fVersion.write((const char*)&versionLen, sizeof(DWORD));
|
|
||||||
fVersion.write((const char*)&versionStr[0], versionLen);
|
|
||||||
fVersion.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
std::wstring tempDirNative = QDir::toNativeSeparators(tempDirPath).toStdWString(), readyDirNative = QDir::toNativeSeparators(readyDirPath).toStdWString();
|
|
||||||
if (!MoveFile(tempDirNative.c_str(), readyDirNative.c_str())) {
|
|
||||||
LOG(("Update Error: cant rename temp dir '%1' to ready dir '%2'. %3").arg(QString::fromStdWString(tempDirNative)).arg(QString::fromStdWString(readyDirNative)).arg(winapiErrorWrap()));
|
|
||||||
return fatalFail();
|
|
||||||
}
|
|
||||||
deleteDir(tempDirPath);
|
|
||||||
outputFile.remove();
|
|
||||||
|
|
||||||
emit App::app()->updateReady();
|
|
||||||
}
|
|
||||||
|
|
||||||
PsUpdateDownloader::~PsUpdateDownloader() {
|
|
||||||
delete reply;
|
|
||||||
reply = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
BOOL CALLBACK _ActivateProcess(HWND hWnd, LPARAM lParam) {
|
BOOL CALLBACK _ActivateProcess(HWND hWnd, LPARAM lParam) {
|
||||||
uint64 &processId(*(uint64*)lParam);
|
uint64 &processId(*(uint64*)lParam);
|
||||||
@ -2134,65 +1778,6 @@ int psFixPrevious() {
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool psCheckReadyUpdate() {
|
|
||||||
QString readyPath = cWorkingDir() + qsl("tupdates/ready");
|
|
||||||
if (!QDir(readyPath).exists()) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
// check ready version
|
|
||||||
QString versionPath = readyPath + qsl("/tdata/version");
|
|
||||||
{
|
|
||||||
QFile fVersion(versionPath);
|
|
||||||
if (!fVersion.open(QIODevice::ReadOnly)) {
|
|
||||||
LOG(("Update Error: cant read version file '%1'").arg(versionPath));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
DWORD versionNum;
|
|
||||||
if (fVersion.read((char*)&versionNum, sizeof(DWORD)) != sizeof(DWORD)) {
|
|
||||||
LOG(("Update Error: cant read version from file '%1'").arg(versionPath));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
fVersion.close();
|
|
||||||
if (versionNum <= AppVersion) {
|
|
||||||
LOG(("Update Error: cant install version %1 having version %2").arg(versionNum).arg(AppVersion));
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
QString curUpdater = (cExeDir() + qsl("Updater.exe"));
|
|
||||||
QFileInfo updater(cWorkingDir() + qsl("tupdates/ready/Updater.exe"));
|
|
||||||
if (!updater.exists()) {
|
|
||||||
QFileInfo current(curUpdater);
|
|
||||||
if (!current.exists()) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (CopyFile(current.absoluteFilePath().toStdWString().c_str(), updater.absoluteFilePath().toStdWString().c_str(), TRUE) == FALSE) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (CopyFile(updater.absoluteFilePath().toStdWString().c_str(), curUpdater.toStdWString().c_str(), FALSE) == FALSE) {
|
|
||||||
DWORD errorCode = GetLastError();
|
|
||||||
if (errorCode == ERROR_ACCESS_DENIED) { // we are in write-protected dir, like Program Files
|
|
||||||
cSetWriteProtected(true);
|
|
||||||
return true;
|
|
||||||
} else {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (DeleteFile(updater.absoluteFilePath().toStdWString().c_str()) == FALSE) {
|
|
||||||
PsUpdateDownloader::clearAll();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void psPostprocessFile(const QString &name) {
|
void psPostprocessFile(const QString &name) {
|
||||||
std::wstring zoneFile = QDir::toNativeSeparators(name).toStdWString() + L":Zone.Identifier";
|
std::wstring zoneFile = QDir::toNativeSeparators(name).toStdWString() + L":Zone.Identifier";
|
||||||
HANDLE f = CreateFile(zoneFile.c_str(), GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, 0, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
|
HANDLE f = CreateFile(zoneFile.c_str(), GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, 0, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
|
||||||
@ -2474,7 +2059,7 @@ void psExecUpdater() {
|
|||||||
if (cStartInTray()) targs += qsl(" -startintray");
|
if (cStartInTray()) targs += qsl(" -startintray");
|
||||||
if (cWriteProtected()) targs += qsl(" -writeprotected \"") + cExeDir() + '"';
|
if (cWriteProtected()) targs += qsl(" -writeprotected \"") + cExeDir() + '"';
|
||||||
|
|
||||||
QString updaterPath = cWriteProtected() ? (cWorkingDir() + qsl("tupdates/ready/Updater.exe")) : (cExeDir() + qsl("Updater.exe"));
|
QString updaterPath = cWriteProtected() ? (cWorkingDir() + qsl("tupdates/temp/Updater.exe")) : (cExeDir() + qsl("Updater.exe"));
|
||||||
|
|
||||||
QString updater(QDir::toNativeSeparators(updaterPath)), wdir(QDir::toNativeSeparators(cWorkingDir()));
|
QString updater(QDir::toNativeSeparators(updaterPath)), wdir(QDir::toNativeSeparators(cWorkingDir()));
|
||||||
|
|
||||||
@ -2482,8 +2067,7 @@ void psExecUpdater() {
|
|||||||
HINSTANCE r = ShellExecute(0, cWriteProtected() ? L"runas" : 0, updater.toStdWString().c_str(), targs.toStdWString().c_str(), wdir.isEmpty() ? 0 : wdir.toStdWString().c_str(), SW_SHOWNORMAL);
|
HINSTANCE r = ShellExecute(0, cWriteProtected() ? L"runas" : 0, updater.toStdWString().c_str(), targs.toStdWString().c_str(), wdir.isEmpty() ? 0 : wdir.toStdWString().c_str(), SW_SHOWNORMAL);
|
||||||
if (long(r) < 32) {
|
if (long(r) < 32) {
|
||||||
DEBUG_LOG(("Application Error: failed to execute %1, working directory: '%2', result: %3").arg(updater).arg(wdir).arg(long(r)));
|
DEBUG_LOG(("Application Error: failed to execute %1, working directory: '%2', result: %3").arg(updater).arg(wdir).arg(long(r)));
|
||||||
QString readyPath = cWorkingDir() + qsl("tupdates/ready");
|
psDeleteDir(cWorkingDir() + qsl("tupdates/temp"));
|
||||||
PsUpdateDownloader::deleteDir(readyPath);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -117,55 +117,9 @@ public:
|
|||||||
void psInstallEventFilter();
|
void psInstallEventFilter();
|
||||||
~PsApplication();
|
~PsApplication();
|
||||||
|
|
||||||
signals:
|
|
||||||
|
|
||||||
void updateChecking();
|
|
||||||
void updateLatest();
|
|
||||||
void updateDownloading(qint64 ready, qint64 total);
|
|
||||||
void updateReady();
|
|
||||||
void updateFailed();
|
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class PsUpdateDownloader : public QObject {
|
void psDeleteDir(const QString &dir);
|
||||||
Q_OBJECT
|
|
||||||
|
|
||||||
public:
|
|
||||||
PsUpdateDownloader(QThread *thread, const MTPDhelp_appUpdate &update);
|
|
||||||
PsUpdateDownloader(QThread *thread, const QString &url);
|
|
||||||
|
|
||||||
void unpackUpdate();
|
|
||||||
|
|
||||||
int32 ready();
|
|
||||||
int32 size();
|
|
||||||
|
|
||||||
static void deleteDir(const QString &dir);
|
|
||||||
static void clearAll();
|
|
||||||
|
|
||||||
~PsUpdateDownloader();
|
|
||||||
|
|
||||||
public slots:
|
|
||||||
|
|
||||||
void start();
|
|
||||||
void partMetaGot();
|
|
||||||
void partFinished(qint64 got, qint64 total);
|
|
||||||
void partFailed(QNetworkReply::NetworkError e);
|
|
||||||
void sendRequest();
|
|
||||||
|
|
||||||
private:
|
|
||||||
void initOutput();
|
|
||||||
|
|
||||||
void fatalFail();
|
|
||||||
|
|
||||||
QString updateUrl;
|
|
||||||
QNetworkAccessManager manager;
|
|
||||||
QNetworkReply *reply;
|
|
||||||
int32 already, full;
|
|
||||||
QFile outputFile;
|
|
||||||
|
|
||||||
QMutex mutex;
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
void psUserActionDone();
|
void psUserActionDone();
|
||||||
bool psIdleSupported();
|
bool psIdleSupported();
|
||||||
@ -196,7 +150,6 @@ void psBringToBack(QWidget *w);
|
|||||||
int psCleanup();
|
int psCleanup();
|
||||||
int psFixPrevious();
|
int psFixPrevious();
|
||||||
|
|
||||||
bool psCheckReadyUpdate();
|
|
||||||
void psExecUpdater();
|
void psExecUpdater();
|
||||||
void psExecTelegram();
|
void psExecTelegram();
|
||||||
|
|
||||||
|
@ -37,6 +37,8 @@ Copyright (c) 2014 John Preston, https://desktop.telegram.org
|
|||||||
#include "langloaderplain.h"
|
#include "langloaderplain.h"
|
||||||
#include "gui/filedialog.h"
|
#include "gui/filedialog.h"
|
||||||
|
|
||||||
|
#include "autoupdater.h"
|
||||||
|
|
||||||
#include "localstorage.h"
|
#include "localstorage.h"
|
||||||
|
|
||||||
Slider::Slider(QWidget *parent, const style::slider &st, int32 count, int32 sel) : QWidget(parent),
|
Slider::Slider(QWidget *parent, const style::slider &st, int32 count, int32 sel) : QWidget(parent),
|
||||||
@ -1172,7 +1174,7 @@ void SettingsInner::onCheckNow() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void SettingsInner::onRestartNow() {
|
void SettingsInner::onRestartNow() {
|
||||||
psCheckReadyUpdate();
|
checkReadyUpdate();
|
||||||
if (_updatingState == UpdatingReady) {
|
if (_updatingState == UpdatingReady) {
|
||||||
cSetRestartingUpdate(true);
|
cSetRestartingUpdate(true);
|
||||||
} else {
|
} else {
|
||||||
|
@ -23,6 +23,7 @@ Copyright (c) 2014 John Preston, https://desktop.telegram.org
|
|||||||
#include "passcodewidget.h"
|
#include "passcodewidget.h"
|
||||||
#include "window.h"
|
#include "window.h"
|
||||||
#include "application.h"
|
#include "application.h"
|
||||||
|
#include "autoupdater.h"
|
||||||
|
|
||||||
SysBtn::SysBtn(QWidget *parent, const style::sysButton &st, const QString &text) : Button(parent),
|
SysBtn::SysBtn(QWidget *parent, const style::sysButton &st, const QString &text) : Button(parent),
|
||||||
_st(st), a_color(_st.color->c), _overLevel(0), _text(text) {
|
_st(st), a_color(_st.color->c), _overLevel(0), _text(text) {
|
||||||
@ -142,7 +143,7 @@ UpdateBtn::UpdateBtn(QWidget *parent, Window *window, const QString &text) : Sys
|
|||||||
}
|
}
|
||||||
|
|
||||||
void UpdateBtn::onClick() {
|
void UpdateBtn::onClick() {
|
||||||
psCheckReadyUpdate();
|
checkReadyUpdate();
|
||||||
if (App::app()->updatingState() == Application::UpdatingReady) {
|
if (App::app()->updatingState() == Application::UpdatingReady) {
|
||||||
cSetRestartingUpdate(true);
|
cSetRestartingUpdate(true);
|
||||||
} else {
|
} else {
|
||||||
|
@ -180,6 +180,10 @@
|
|||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
||||||
</ClCompile>
|
</ClCompile>
|
||||||
|
<ClCompile Include="GeneratedFiles\Debug\moc_autoupdater.cpp">
|
||||||
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
||||||
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
||||||
|
</ClCompile>
|
||||||
<ClCompile Include="GeneratedFiles\Debug\moc_backgroundbox.cpp">
|
<ClCompile Include="GeneratedFiles\Debug\moc_backgroundbox.cpp">
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
||||||
@ -442,6 +446,10 @@
|
|||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
||||||
</ClCompile>
|
</ClCompile>
|
||||||
|
<ClCompile Include="GeneratedFiles\Deploy\moc_autoupdater.cpp">
|
||||||
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
||||||
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
||||||
|
</ClCompile>
|
||||||
<ClCompile Include="GeneratedFiles\Deploy\moc_backgroundbox.cpp">
|
<ClCompile Include="GeneratedFiles\Deploy\moc_backgroundbox.cpp">
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">true</ExcludedFromBuild>
|
||||||
@ -729,6 +737,10 @@
|
|||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
||||||
</ClCompile>
|
</ClCompile>
|
||||||
|
<ClCompile Include="GeneratedFiles\Release\moc_autoupdater.cpp">
|
||||||
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
||||||
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
||||||
|
</ClCompile>
|
||||||
<ClCompile Include="GeneratedFiles\Release\moc_backgroundbox.cpp">
|
<ClCompile Include="GeneratedFiles\Release\moc_backgroundbox.cpp">
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">true</ExcludedFromBuild>
|
||||||
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
<ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">true</ExcludedFromBuild>
|
||||||
@ -964,6 +976,7 @@
|
|||||||
<ClCompile Include="SourceFiles\app.cpp" />
|
<ClCompile Include="SourceFiles\app.cpp" />
|
||||||
<ClCompile Include="SourceFiles\application.cpp" />
|
<ClCompile Include="SourceFiles\application.cpp" />
|
||||||
<ClCompile Include="SourceFiles\audio.cpp" />
|
<ClCompile Include="SourceFiles\audio.cpp" />
|
||||||
|
<ClCompile Include="SourceFiles\autoupdater.cpp" />
|
||||||
<ClCompile Include="SourceFiles\boxes\aboutbox.cpp" />
|
<ClCompile Include="SourceFiles\boxes\aboutbox.cpp" />
|
||||||
<ClCompile Include="SourceFiles\boxes\abstractbox.cpp" />
|
<ClCompile Include="SourceFiles\boxes\abstractbox.cpp" />
|
||||||
<ClCompile Include="SourceFiles\boxes\addcontactbox.cpp" />
|
<ClCompile Include="SourceFiles\boxes\addcontactbox.cpp" />
|
||||||
@ -1383,6 +1396,20 @@
|
|||||||
<Outputs Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">.\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp</Outputs>
|
<Outputs Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">.\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp</Outputs>
|
||||||
<Command Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">"$(QTDIR)\bin\moc.exe" "%(FullPath)" -o ".\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp" -DAL_LIBTYPE_STATIC -DUNICODE -D_WITH_DEBUG -DWIN32 -DWIN64 -DHAVE_STDINT_H -DZLIB_WINAPI -DQT_NO_DEBUG -DNDEBUG "-I.\..\..\Libraries\lzma\C" "-I.\..\..\Libraries\libexif-0.6.20" "-I.\..\..\Libraries\zlib-1.2.8" "-I.\..\..\Libraries\OpenSSL-Win32\include" "-I.\..\..\Libraries\libogg-1.3.2\include" "-I.\..\..\Libraries\opus\include" "-I.\..\..\Libraries\opusfile\include" "-I.\..\..\Libraries\mpg123-1.22.1\ports\MSVC++" "-I.\..\..\Libraries\mpg123-1.22.1\src\libmpg123" "-I.\..\..\Libraries\faad2-2.7\include" "-I.\..\..\Libraries\faad2-2.7\common\mp4ff" "-I.\..\..\Libraries\ffmpeg-2.6.3" "-I.\..\..\Libraries\openal-soft\include" "-I.\SourceFiles" "-I.\GeneratedFiles" "-I." "-I$(QTDIR)\include" "-I.\GeneratedFiles\$(ConfigurationName)\." "-I.\..\..\Libraries\QtStatic\qtbase\include\QtCore\5.4.0\QtCore" "-I.\..\..\Libraries\QtStatic\qtbase\include\QtGui\5.4.0\QtGui" "-fstdafx.h" "-f../../SourceFiles/boxes/stickersetbox.h"</Command>
|
<Command Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">"$(QTDIR)\bin\moc.exe" "%(FullPath)" -o ".\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp" -DAL_LIBTYPE_STATIC -DUNICODE -D_WITH_DEBUG -DWIN32 -DWIN64 -DHAVE_STDINT_H -DZLIB_WINAPI -DQT_NO_DEBUG -DNDEBUG "-I.\..\..\Libraries\lzma\C" "-I.\..\..\Libraries\libexif-0.6.20" "-I.\..\..\Libraries\zlib-1.2.8" "-I.\..\..\Libraries\OpenSSL-Win32\include" "-I.\..\..\Libraries\libogg-1.3.2\include" "-I.\..\..\Libraries\opus\include" "-I.\..\..\Libraries\opusfile\include" "-I.\..\..\Libraries\mpg123-1.22.1\ports\MSVC++" "-I.\..\..\Libraries\mpg123-1.22.1\src\libmpg123" "-I.\..\..\Libraries\faad2-2.7\include" "-I.\..\..\Libraries\faad2-2.7\common\mp4ff" "-I.\..\..\Libraries\ffmpeg-2.6.3" "-I.\..\..\Libraries\openal-soft\include" "-I.\SourceFiles" "-I.\GeneratedFiles" "-I." "-I$(QTDIR)\include" "-I.\GeneratedFiles\$(ConfigurationName)\." "-I.\..\..\Libraries\QtStatic\qtbase\include\QtCore\5.4.0\QtCore" "-I.\..\..\Libraries\QtStatic\qtbase\include\QtGui\5.4.0\QtGui" "-fstdafx.h" "-f../../SourceFiles/boxes/stickersetbox.h"</Command>
|
||||||
</CustomBuild>
|
</CustomBuild>
|
||||||
|
<CustomBuild Include="SourceFiles\autoupdater.h">
|
||||||
|
<AdditionalInputs Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">$(QTDIR)\bin\moc.exe;%(FullPath)</AdditionalInputs>
|
||||||
|
<Message Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">Moc%27ing autoupdater.h...</Message>
|
||||||
|
<Outputs Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">.\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp</Outputs>
|
||||||
|
<Command Condition="'$(Configuration)|$(Platform)'=='Deploy|Win32'">"$(QTDIR)\bin\moc.exe" "%(FullPath)" -o ".\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp" "-fstdafx.h" "-f../../SourceFiles/autoupdater.h" -DAL_LIBTYPE_STATIC -DCUSTOM_API_ID -DUNICODE -D_WITH_DEBUG -DWIN32 -DWIN64 -DHAVE_STDINT_H -DZLIB_WINAPI -DQT_NO_DEBUG -DNDEBUG "-I.\..\..\Libraries\lzma\C" "-I.\..\..\Libraries\libexif-0.6.20" "-I.\..\..\Libraries\zlib-1.2.8" "-I.\..\..\Libraries\OpenSSL-Win32\include" "-I.\..\..\Libraries\ffmpeg-2.6.3" "-I.\..\..\Libraries\openal-soft\include" "-I.\SourceFiles" "-I.\GeneratedFiles" "-I." "-I$(QTDIR)\include" "-I.\GeneratedFiles\$(ConfigurationName)\." "-I.\..\..\Libraries\QtStatic\qtbase\include\QtCore\5.4.0\QtCore" "-I.\..\..\Libraries\QtStatic\qtbase\include\QtGui\5.4.0\QtGui"</Command>
|
||||||
|
<AdditionalInputs Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">$(QTDIR)\bin\moc.exe;%(FullPath)</AdditionalInputs>
|
||||||
|
<Message Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Moc%27ing autoupdater.h...</Message>
|
||||||
|
<Outputs Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">.\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp</Outputs>
|
||||||
|
<Command Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">"$(QTDIR)\bin\moc.exe" "%(FullPath)" -o ".\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp" "-fstdafx.h" "-f../../SourceFiles/autoupdater.h" -DAL_LIBTYPE_STATIC -DUNICODE -DWIN32 -DWIN64 -DHAVE_STDINT_H -DZLIB_WINAPI "-I.\..\..\Libraries\lzma\C" "-I.\..\..\Libraries\libexif-0.6.20" "-I.\..\..\Libraries\zlib-1.2.8" "-I.\..\..\Libraries\OpenSSL-Win32\include" "-I.\..\..\Libraries\ffmpeg-2.6.3" "-I.\..\..\Libraries\openal-soft\include" "-I.\SourceFiles" "-I.\GeneratedFiles" "-I." "-I$(QTDIR)\include" "-I.\GeneratedFiles\$(ConfigurationName)\." "-I.\..\..\Libraries\QtStatic\qtbase\include\QtCore\5.4.0\QtCore" "-I.\..\..\Libraries\QtStatic\qtbase\include\QtGui\5.4.0\QtGui"</Command>
|
||||||
|
<AdditionalInputs Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">$(QTDIR)\bin\moc.exe;%(FullPath)</AdditionalInputs>
|
||||||
|
<Message Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Moc%27ing autoupdater.h...</Message>
|
||||||
|
<Outputs Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">.\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp</Outputs>
|
||||||
|
<Command Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">"$(QTDIR)\bin\moc.exe" "%(FullPath)" -o ".\GeneratedFiles\$(ConfigurationName)\moc_%(Filename).cpp" "-fstdafx.h" "-f../../SourceFiles/autoupdater.h" -DAL_LIBTYPE_STATIC -DUNICODE -D_WITH_DEBUG -DWIN32 -DWIN64 -DHAVE_STDINT_H -DZLIB_WINAPI -DQT_NO_DEBUG -DNDEBUG "-I.\..\..\Libraries\lzma\C" "-I.\..\..\Libraries\libexif-0.6.20" "-I.\..\..\Libraries\zlib-1.2.8" "-I.\..\..\Libraries\OpenSSL-Win32\include" "-I.\..\..\Libraries\libogg-1.3.2\include" "-I.\..\..\Libraries\opus\include" "-I.\..\..\Libraries\opusfile\include" "-I.\..\..\Libraries\mpg123-1.22.1\ports\MSVC++" "-I.\..\..\Libraries\mpg123-1.22.1\src\libmpg123" "-I.\..\..\Libraries\faad2-2.7\include" "-I.\..\..\Libraries\faad2-2.7\common\mp4ff" "-I.\..\..\Libraries\ffmpeg-2.6.3" "-I.\..\..\Libraries\openal-soft\include" "-I.\SourceFiles" "-I.\GeneratedFiles" "-I." "-I$(QTDIR)\include" "-I.\GeneratedFiles\$(ConfigurationName)\." "-I.\..\..\Libraries\QtStatic\qtbase\include\QtCore\5.4.0\QtCore" "-I.\..\..\Libraries\QtStatic\qtbase\include\QtGui\5.4.0\QtGui"</Command>
|
||||||
|
</CustomBuild>
|
||||||
<ClInclude Include="SourceFiles\config.h" />
|
<ClInclude Include="SourceFiles\config.h" />
|
||||||
<CustomBuild Include="SourceFiles\gui\animation.h">
|
<CustomBuild Include="SourceFiles\gui\animation.h">
|
||||||
<Message Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Moc%27ing animation.h...</Message>
|
<Message Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Moc%27ing animation.h...</Message>
|
||||||
|
@ -891,6 +891,18 @@
|
|||||||
<ClCompile Include="SourceFiles\boxes\stickersetbox.cpp">
|
<ClCompile Include="SourceFiles\boxes\stickersetbox.cpp">
|
||||||
<Filter>boxes</Filter>
|
<Filter>boxes</Filter>
|
||||||
</ClCompile>
|
</ClCompile>
|
||||||
|
<ClCompile Include="SourceFiles\autoupdater.cpp">
|
||||||
|
<Filter>Source Files</Filter>
|
||||||
|
</ClCompile>
|
||||||
|
<ClCompile Include="GeneratedFiles\Deploy\moc_autoupdater.cpp">
|
||||||
|
<Filter>Generated Files\Deploy</Filter>
|
||||||
|
</ClCompile>
|
||||||
|
<ClCompile Include="GeneratedFiles\Debug\moc_autoupdater.cpp">
|
||||||
|
<Filter>Generated Files\Debug</Filter>
|
||||||
|
</ClCompile>
|
||||||
|
<ClCompile Include="GeneratedFiles\Release\moc_autoupdater.cpp">
|
||||||
|
<Filter>Generated Files\Release</Filter>
|
||||||
|
</ClCompile>
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<ClInclude Include="SourceFiles\stdafx.h">
|
<ClInclude Include="SourceFiles\stdafx.h">
|
||||||
@ -1183,6 +1195,9 @@
|
|||||||
<CustomBuild Include="SourceFiles\boxes\stickersetbox.h">
|
<CustomBuild Include="SourceFiles\boxes\stickersetbox.h">
|
||||||
<Filter>boxes</Filter>
|
<Filter>boxes</Filter>
|
||||||
</CustomBuild>
|
</CustomBuild>
|
||||||
|
<CustomBuild Include="SourceFiles\autoupdater.h">
|
||||||
|
<Filter>Source Files</Filter>
|
||||||
|
</CustomBuild>
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<Image Include="SourceFiles\art\icon256.ico" />
|
<Image Include="SourceFiles\art\icon256.ico" />
|
||||||
|
35
Telegram/_openal_patch.diff
Normal file
35
Telegram/_openal_patch.diff
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
diff --git a/Alc/backends/mmdevapi.c b/Alc/backends/mmdevapi.c
|
||||||
|
index cfd12d8..8a6f9fb 100644
|
||||||
|
--- a/Alc/backends/mmdevapi.c
|
||||||
|
+++ b/Alc/backends/mmdevapi.c
|
||||||
|
@@ -1719,7 +1719,7 @@ static void ALCmmdevBackendFactory_deinit(ALCmmdevBackendFactory* UNUSED(self))
|
||||||
|
|
||||||
|
static ALCboolean ALCmmdevBackendFactory_querySupport(ALCmmdevBackendFactory* UNUSED(self), ALCbackend_Type type)
|
||||||
|
{
|
||||||
|
- if(type == ALCbackend_Playback || type == ALCbackend_Capture)
|
||||||
|
+ if(type == ALCbackend_Playback/* || type == ALCbackend_Capture*/)
|
||||||
|
return ALC_TRUE;
|
||||||
|
return ALC_FALSE;
|
||||||
|
}
|
||||||
|
diff --git a/Alc/backends/winmm.c b/Alc/backends/winmm.c
|
||||||
|
index 03805ab..77212c2 100644
|
||||||
|
--- a/Alc/backends/winmm.c
|
||||||
|
+++ b/Alc/backends/winmm.c
|
||||||
|
@@ -220,7 +220,7 @@ FORCE_ALIGN static int ALCwinmmPlayback_mixerProc(void *arg)
|
||||||
|
SetRTPriority();
|
||||||
|
althrd_setname(althrd_current(), MIXER_THREAD_NAME);
|
||||||
|
|
||||||
|
- while(GetMessage(&msg, NULL, 0, 0))
|
||||||
|
+ if (!self->killNow) while (GetMessage(&msg, NULL, 0, 0))
|
||||||
|
{
|
||||||
|
if(msg.message != WOM_DONE)
|
||||||
|
continue;
|
||||||
|
@@ -505,7 +505,7 @@ static int ALCwinmmCapture_captureProc(void *arg)
|
||||||
|
|
||||||
|
althrd_setname(althrd_current(), RECORD_THREAD_NAME);
|
||||||
|
|
||||||
|
- while(GetMessage(&msg, NULL, 0, 0))
|
||||||
|
+ if (!self->killNow) while(GetMessage(&msg, NULL, 0, 0))
|
||||||
|
{
|
||||||
|
if(msg.message != WIM_DATA)
|
||||||
|
continue;
|
File diff suppressed because it is too large
Load Diff
1789
Telegram/_openal_patch/Alc/backends/mmdevapi.c
Normal file
1789
Telegram/_openal_patch/Alc/backends/mmdevapi.c
Normal file
File diff suppressed because it is too large
Load Diff
@ -31,30 +31,13 @@
|
|||||||
#include "alu.h"
|
#include "alu.h"
|
||||||
#include "threads.h"
|
#include "threads.h"
|
||||||
|
|
||||||
|
#include "backends/base.h"
|
||||||
|
|
||||||
#ifndef WAVE_FORMAT_IEEE_FLOAT
|
#ifndef WAVE_FORMAT_IEEE_FLOAT
|
||||||
#define WAVE_FORMAT_IEEE_FLOAT 0x0003
|
#define WAVE_FORMAT_IEEE_FLOAT 0x0003
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
// MMSYSTEM Device
|
|
||||||
volatile ALboolean killNow;
|
|
||||||
althrd_t thread;
|
|
||||||
|
|
||||||
RefCount WaveBuffersCommitted;
|
|
||||||
WAVEHDR WaveBuffer[4];
|
|
||||||
|
|
||||||
union {
|
|
||||||
HWAVEIN In;
|
|
||||||
HWAVEOUT Out;
|
|
||||||
} WaveHandle;
|
|
||||||
|
|
||||||
WAVEFORMATEX Format;
|
|
||||||
|
|
||||||
RingBuffer *Ring;
|
|
||||||
} WinMMData;
|
|
||||||
|
|
||||||
|
|
||||||
TYPEDEF_VECTOR(al_string, vector_al_string)
|
TYPEDEF_VECTOR(al_string, vector_al_string)
|
||||||
static vector_al_string PlaybackDevices;
|
static vector_al_string PlaybackDevices;
|
||||||
static vector_al_string CaptureDevices;
|
static vector_al_string CaptureDevices;
|
||||||
@ -155,111 +138,117 @@ static void ProbeCaptureDevices(void)
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/*
|
typedef struct ALCwinmmPlayback {
|
||||||
WaveOutProc
|
DERIVE_FROM_TYPE(ALCbackend);
|
||||||
|
|
||||||
Posts a message to 'PlaybackThreadProc' everytime a WaveOut Buffer is completed and
|
RefCount WaveBuffersCommitted;
|
||||||
returns to the application (for more data)
|
WAVEHDR WaveBuffer[4];
|
||||||
*/
|
|
||||||
static void CALLBACK WaveOutProc(HWAVEOUT UNUSED(device), UINT msg, DWORD_PTR instance, DWORD_PTR param1, DWORD_PTR UNUSED(param2))
|
HWAVEOUT OutHdl;
|
||||||
|
|
||||||
|
WAVEFORMATEX Format;
|
||||||
|
|
||||||
|
volatile ALboolean killNow;
|
||||||
|
althrd_t thread;
|
||||||
|
} ALCwinmmPlayback;
|
||||||
|
|
||||||
|
static void ALCwinmmPlayback_Construct(ALCwinmmPlayback *self, ALCdevice *device);
|
||||||
|
static void ALCwinmmPlayback_Destruct(ALCwinmmPlayback *self);
|
||||||
|
|
||||||
|
static void CALLBACK ALCwinmmPlayback_waveOutProc(HWAVEOUT device, UINT msg, DWORD_PTR instance, DWORD_PTR param1, DWORD_PTR param2);
|
||||||
|
static int ALCwinmmPlayback_mixerProc(void *arg);
|
||||||
|
|
||||||
|
static ALCenum ALCwinmmPlayback_open(ALCwinmmPlayback *self, const ALCchar *name);
|
||||||
|
static void ALCwinmmPlayback_close(ALCwinmmPlayback *self);
|
||||||
|
static ALCboolean ALCwinmmPlayback_reset(ALCwinmmPlayback *self);
|
||||||
|
static ALCboolean ALCwinmmPlayback_start(ALCwinmmPlayback *self);
|
||||||
|
static void ALCwinmmPlayback_stop(ALCwinmmPlayback *self);
|
||||||
|
static DECLARE_FORWARD2(ALCwinmmPlayback, ALCbackend, ALCenum, captureSamples, ALCvoid*, ALCuint)
|
||||||
|
static DECLARE_FORWARD(ALCwinmmPlayback, ALCbackend, ALCuint, availableSamples)
|
||||||
|
static DECLARE_FORWARD(ALCwinmmPlayback, ALCbackend, ALint64, getLatency)
|
||||||
|
static DECLARE_FORWARD(ALCwinmmPlayback, ALCbackend, void, lock)
|
||||||
|
static DECLARE_FORWARD(ALCwinmmPlayback, ALCbackend, void, unlock)
|
||||||
|
DECLARE_DEFAULT_ALLOCATORS(ALCwinmmPlayback)
|
||||||
|
|
||||||
|
DEFINE_ALCBACKEND_VTABLE(ALCwinmmPlayback);
|
||||||
|
|
||||||
|
|
||||||
|
static void ALCwinmmPlayback_Construct(ALCwinmmPlayback *self, ALCdevice *device)
|
||||||
{
|
{
|
||||||
ALCdevice *Device = (ALCdevice*)instance;
|
ALCbackend_Construct(STATIC_CAST(ALCbackend, self), device);
|
||||||
WinMMData *data = Device->ExtraData;
|
SET_VTABLE2(ALCwinmmPlayback, ALCbackend, self);
|
||||||
|
|
||||||
|
InitRef(&self->WaveBuffersCommitted, 0);
|
||||||
|
self->OutHdl = NULL;
|
||||||
|
|
||||||
|
self->killNow = AL_TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void ALCwinmmPlayback_Destruct(ALCwinmmPlayback *self)
|
||||||
|
{
|
||||||
|
if(self->OutHdl)
|
||||||
|
waveOutClose(self->OutHdl);
|
||||||
|
self->OutHdl = 0;
|
||||||
|
|
||||||
|
ALCbackend_Destruct(STATIC_CAST(ALCbackend, self));
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* ALCwinmmPlayback_waveOutProc
|
||||||
|
*
|
||||||
|
* Posts a message to 'ALCwinmmPlayback_mixerProc' everytime a WaveOut Buffer
|
||||||
|
* is completed and returns to the application (for more data)
|
||||||
|
*/
|
||||||
|
static void CALLBACK ALCwinmmPlayback_waveOutProc(HWAVEOUT UNUSED(device), UINT msg, DWORD_PTR instance, DWORD_PTR param1, DWORD_PTR UNUSED(param2))
|
||||||
|
{
|
||||||
|
ALCwinmmPlayback *self = (ALCwinmmPlayback*)instance;
|
||||||
|
|
||||||
if(msg != WOM_DONE)
|
if(msg != WOM_DONE)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
DecrementRef(&data->WaveBuffersCommitted);
|
DecrementRef(&self->WaveBuffersCommitted);
|
||||||
PostThreadMessage(data->thread, msg, 0, param1);
|
PostThreadMessage(self->thread, msg, 0, param1);
|
||||||
}
|
}
|
||||||
|
|
||||||
FORCE_ALIGN static int PlaybackThreadProc(void *arg)
|
FORCE_ALIGN static int ALCwinmmPlayback_mixerProc(void *arg)
|
||||||
{
|
{
|
||||||
ALCdevice *Device = (ALCdevice*)arg;
|
ALCwinmmPlayback *self = arg;
|
||||||
WinMMData *data = Device->ExtraData;
|
ALCdevice *device = STATIC_CAST(ALCbackend, self)->mDevice;
|
||||||
WAVEHDR *WaveHdr;
|
WAVEHDR *WaveHdr;
|
||||||
MSG msg;
|
MSG msg;
|
||||||
|
|
||||||
SetRTPriority();
|
SetRTPriority();
|
||||||
althrd_setname(althrd_current(), MIXER_THREAD_NAME);
|
althrd_setname(althrd_current(), MIXER_THREAD_NAME);
|
||||||
|
|
||||||
while(GetMessage(&msg, NULL, 0, 0))
|
if (!self->killNow) while (GetMessage(&msg, NULL, 0, 0))
|
||||||
{
|
{
|
||||||
if(msg.message != WOM_DONE)
|
if(msg.message != WOM_DONE)
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
if(data->killNow)
|
if(self->killNow)
|
||||||
{
|
{
|
||||||
if(ReadRef(&data->WaveBuffersCommitted) == 0)
|
if(ReadRef(&self->WaveBuffersCommitted) == 0)
|
||||||
break;
|
break;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
WaveHdr = ((WAVEHDR*)msg.lParam);
|
WaveHdr = ((WAVEHDR*)msg.lParam);
|
||||||
aluMixData(Device, WaveHdr->lpData, WaveHdr->dwBufferLength /
|
aluMixData(device, WaveHdr->lpData, WaveHdr->dwBufferLength /
|
||||||
data->Format.nBlockAlign);
|
self->Format.nBlockAlign);
|
||||||
|
|
||||||
// Send buffer back to play more data
|
// Send buffer back to play more data
|
||||||
waveOutWrite(data->WaveHandle.Out, WaveHdr, sizeof(WAVEHDR));
|
waveOutWrite(self->OutHdl, WaveHdr, sizeof(WAVEHDR));
|
||||||
IncrementRef(&data->WaveBuffersCommitted);
|
IncrementRef(&self->WaveBuffersCommitted);
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
WaveInProc
|
|
||||||
|
|
||||||
Posts a message to 'CaptureThreadProc' everytime a WaveIn Buffer is completed and
|
|
||||||
returns to the application (with more data)
|
|
||||||
*/
|
|
||||||
static void CALLBACK WaveInProc(HWAVEIN UNUSED(device), UINT msg, DWORD_PTR instance, DWORD_PTR param1, DWORD_PTR UNUSED(param2))
|
|
||||||
{
|
|
||||||
ALCdevice *Device = (ALCdevice*)instance;
|
|
||||||
WinMMData *data = Device->ExtraData;
|
|
||||||
|
|
||||||
if(msg != WIM_DATA)
|
|
||||||
return;
|
|
||||||
|
|
||||||
DecrementRef(&data->WaveBuffersCommitted);
|
|
||||||
PostThreadMessage(data->thread, msg, 0, param1);
|
|
||||||
}
|
|
||||||
|
|
||||||
static int CaptureThreadProc(void *arg)
|
|
||||||
{
|
|
||||||
ALCdevice *Device = (ALCdevice*)arg;
|
|
||||||
WinMMData *data = Device->ExtraData;
|
|
||||||
WAVEHDR *WaveHdr;
|
|
||||||
MSG msg;
|
|
||||||
|
|
||||||
althrd_setname(althrd_current(), "alsoft-record");
|
|
||||||
|
|
||||||
if (!data->killNow) while(GetMessage(&msg, NULL, 0, 0))
|
|
||||||
{
|
|
||||||
if(msg.message != WIM_DATA)
|
|
||||||
continue;
|
|
||||||
/* Don't wait for other buffers to finish before quitting. We're
|
|
||||||
* closing so we don't need them. */
|
|
||||||
if(data->killNow)
|
|
||||||
break;
|
|
||||||
|
|
||||||
WaveHdr = ((WAVEHDR*)msg.lParam);
|
|
||||||
WriteRingBuffer(data->Ring, (ALubyte*)WaveHdr->lpData,
|
|
||||||
WaveHdr->dwBytesRecorded/data->Format.nBlockAlign);
|
|
||||||
|
|
||||||
// Send buffer back to capture more data
|
|
||||||
waveInAddBuffer(data->WaveHandle.In, WaveHdr, sizeof(WAVEHDR));
|
|
||||||
IncrementRef(&data->WaveBuffersCommitted);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
static ALCenum WinMMOpenPlayback(ALCdevice *Device, const ALCchar *deviceName)
|
static ALCenum ALCwinmmPlayback_open(ALCwinmmPlayback *self, const ALCchar *deviceName)
|
||||||
{
|
{
|
||||||
WinMMData *data = NULL;
|
ALCdevice *device = STATIC_CAST(ALCbackend, self)->mDevice;
|
||||||
const al_string *iter, *end;
|
const al_string *iter;
|
||||||
UINT DeviceID;
|
UINT DeviceID;
|
||||||
MMRESULT res;
|
MMRESULT res;
|
||||||
|
|
||||||
@ -267,129 +256,109 @@ static ALCenum WinMMOpenPlayback(ALCdevice *Device, const ALCchar *deviceName)
|
|||||||
ProbePlaybackDevices();
|
ProbePlaybackDevices();
|
||||||
|
|
||||||
// Find the Device ID matching the deviceName if valid
|
// Find the Device ID matching the deviceName if valid
|
||||||
iter = VECTOR_ITER_BEGIN(PlaybackDevices);
|
#define MATCH_DEVNAME(iter) (!al_string_empty(*(iter)) && \
|
||||||
end = VECTOR_ITER_END(PlaybackDevices);
|
(!deviceName || al_string_cmp_cstr(*(iter), deviceName) == 0))
|
||||||
for(;iter != end;iter++)
|
VECTOR_FIND_IF(iter, const al_string, PlaybackDevices, MATCH_DEVNAME);
|
||||||
{
|
if(iter == VECTOR_ITER_END(PlaybackDevices))
|
||||||
if(!al_string_empty(*iter) &&
|
|
||||||
(!deviceName || al_string_cmp_cstr(*iter, deviceName) == 0))
|
|
||||||
{
|
|
||||||
DeviceID = (UINT)(iter - VECTOR_ITER_BEGIN(PlaybackDevices));
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if(iter == end)
|
|
||||||
return ALC_INVALID_VALUE;
|
return ALC_INVALID_VALUE;
|
||||||
|
#undef MATCH_DEVNAME
|
||||||
|
|
||||||
data = calloc(1, sizeof(*data));
|
DeviceID = (UINT)(iter - VECTOR_ITER_BEGIN(PlaybackDevices));
|
||||||
if(!data)
|
|
||||||
return ALC_OUT_OF_MEMORY;
|
|
||||||
Device->ExtraData = data;
|
|
||||||
|
|
||||||
retry_open:
|
retry_open:
|
||||||
memset(&data->Format, 0, sizeof(WAVEFORMATEX));
|
memset(&self->Format, 0, sizeof(WAVEFORMATEX));
|
||||||
if(Device->FmtType == DevFmtFloat)
|
if(device->FmtType == DevFmtFloat)
|
||||||
{
|
{
|
||||||
data->Format.wFormatTag = WAVE_FORMAT_IEEE_FLOAT;
|
self->Format.wFormatTag = WAVE_FORMAT_IEEE_FLOAT;
|
||||||
data->Format.wBitsPerSample = 32;
|
self->Format.wBitsPerSample = 32;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
data->Format.wFormatTag = WAVE_FORMAT_PCM;
|
self->Format.wFormatTag = WAVE_FORMAT_PCM;
|
||||||
if(Device->FmtType == DevFmtUByte || Device->FmtType == DevFmtByte)
|
if(device->FmtType == DevFmtUByte || device->FmtType == DevFmtByte)
|
||||||
data->Format.wBitsPerSample = 8;
|
self->Format.wBitsPerSample = 8;
|
||||||
else
|
else
|
||||||
data->Format.wBitsPerSample = 16;
|
self->Format.wBitsPerSample = 16;
|
||||||
}
|
}
|
||||||
data->Format.nChannels = ((Device->FmtChans == DevFmtMono) ? 1 : 2);
|
self->Format.nChannels = ((device->FmtChans == DevFmtMono) ? 1 : 2);
|
||||||
data->Format.nBlockAlign = data->Format.wBitsPerSample *
|
self->Format.nBlockAlign = self->Format.wBitsPerSample *
|
||||||
data->Format.nChannels / 8;
|
self->Format.nChannels / 8;
|
||||||
data->Format.nSamplesPerSec = Device->Frequency;
|
self->Format.nSamplesPerSec = device->Frequency;
|
||||||
data->Format.nAvgBytesPerSec = data->Format.nSamplesPerSec *
|
self->Format.nAvgBytesPerSec = self->Format.nSamplesPerSec *
|
||||||
data->Format.nBlockAlign;
|
self->Format.nBlockAlign;
|
||||||
data->Format.cbSize = 0;
|
self->Format.cbSize = 0;
|
||||||
|
|
||||||
if((res=waveOutOpen(&data->WaveHandle.Out, DeviceID, &data->Format, (DWORD_PTR)&WaveOutProc, (DWORD_PTR)Device, CALLBACK_FUNCTION)) != MMSYSERR_NOERROR)
|
if((res=waveOutOpen(&self->OutHdl, DeviceID, &self->Format, (DWORD_PTR)&ALCwinmmPlayback_waveOutProc, (DWORD_PTR)self, CALLBACK_FUNCTION)) != MMSYSERR_NOERROR)
|
||||||
{
|
{
|
||||||
if(Device->FmtType == DevFmtFloat)
|
if(device->FmtType == DevFmtFloat)
|
||||||
{
|
{
|
||||||
Device->FmtType = DevFmtShort;
|
device->FmtType = DevFmtShort;
|
||||||
goto retry_open;
|
goto retry_open;
|
||||||
}
|
}
|
||||||
ERR("waveOutOpen failed: %u\n", res);
|
ERR("waveOutOpen failed: %u\n", res);
|
||||||
goto failure;
|
goto failure;
|
||||||
}
|
}
|
||||||
|
|
||||||
al_string_copy(&Device->DeviceName, VECTOR_ELEM(PlaybackDevices, DeviceID));
|
al_string_copy(&device->DeviceName, VECTOR_ELEM(PlaybackDevices, DeviceID));
|
||||||
return ALC_NO_ERROR;
|
return ALC_NO_ERROR;
|
||||||
|
|
||||||
failure:
|
failure:
|
||||||
if(data->WaveHandle.Out)
|
if(self->OutHdl)
|
||||||
waveOutClose(data->WaveHandle.Out);
|
waveOutClose(self->OutHdl);
|
||||||
|
self->OutHdl = NULL;
|
||||||
|
|
||||||
free(data);
|
|
||||||
Device->ExtraData = NULL;
|
|
||||||
return ALC_INVALID_VALUE;
|
return ALC_INVALID_VALUE;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void WinMMClosePlayback(ALCdevice *device)
|
static void ALCwinmmPlayback_close(ALCwinmmPlayback* UNUSED(self))
|
||||||
|
{ }
|
||||||
|
|
||||||
|
static ALCboolean ALCwinmmPlayback_reset(ALCwinmmPlayback *self)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)device->ExtraData;
|
ALCdevice *device = STATIC_CAST(ALCbackend, self)->mDevice;
|
||||||
|
|
||||||
// Close the Wave device
|
|
||||||
waveOutClose(data->WaveHandle.Out);
|
|
||||||
data->WaveHandle.Out = 0;
|
|
||||||
|
|
||||||
free(data);
|
|
||||||
device->ExtraData = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
static ALCboolean WinMMResetPlayback(ALCdevice *device)
|
|
||||||
{
|
|
||||||
WinMMData *data = (WinMMData*)device->ExtraData;
|
|
||||||
|
|
||||||
device->UpdateSize = (ALuint)((ALuint64)device->UpdateSize *
|
device->UpdateSize = (ALuint)((ALuint64)device->UpdateSize *
|
||||||
data->Format.nSamplesPerSec /
|
self->Format.nSamplesPerSec /
|
||||||
device->Frequency);
|
device->Frequency);
|
||||||
device->UpdateSize = (device->UpdateSize*device->NumUpdates + 3) / 4;
|
device->UpdateSize = (device->UpdateSize*device->NumUpdates + 3) / 4;
|
||||||
device->NumUpdates = 4;
|
device->NumUpdates = 4;
|
||||||
device->Frequency = data->Format.nSamplesPerSec;
|
device->Frequency = self->Format.nSamplesPerSec;
|
||||||
|
|
||||||
if(data->Format.wFormatTag == WAVE_FORMAT_IEEE_FLOAT)
|
if(self->Format.wFormatTag == WAVE_FORMAT_IEEE_FLOAT)
|
||||||
{
|
{
|
||||||
if(data->Format.wBitsPerSample == 32)
|
if(self->Format.wBitsPerSample == 32)
|
||||||
device->FmtType = DevFmtFloat;
|
device->FmtType = DevFmtFloat;
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
ERR("Unhandled IEEE float sample depth: %d\n", data->Format.wBitsPerSample);
|
ERR("Unhandled IEEE float sample depth: %d\n", self->Format.wBitsPerSample);
|
||||||
return ALC_FALSE;
|
return ALC_FALSE;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else if(data->Format.wFormatTag == WAVE_FORMAT_PCM)
|
else if(self->Format.wFormatTag == WAVE_FORMAT_PCM)
|
||||||
{
|
{
|
||||||
if(data->Format.wBitsPerSample == 16)
|
if(self->Format.wBitsPerSample == 16)
|
||||||
device->FmtType = DevFmtShort;
|
device->FmtType = DevFmtShort;
|
||||||
else if(data->Format.wBitsPerSample == 8)
|
else if(self->Format.wBitsPerSample == 8)
|
||||||
device->FmtType = DevFmtUByte;
|
device->FmtType = DevFmtUByte;
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
ERR("Unhandled PCM sample depth: %d\n", data->Format.wBitsPerSample);
|
ERR("Unhandled PCM sample depth: %d\n", self->Format.wBitsPerSample);
|
||||||
return ALC_FALSE;
|
return ALC_FALSE;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
ERR("Unhandled format tag: 0x%04x\n", data->Format.wFormatTag);
|
ERR("Unhandled format tag: 0x%04x\n", self->Format.wFormatTag);
|
||||||
return ALC_FALSE;
|
return ALC_FALSE;
|
||||||
}
|
}
|
||||||
|
|
||||||
if(data->Format.nChannels == 2)
|
if(self->Format.nChannels == 2)
|
||||||
device->FmtChans = DevFmtStereo;
|
device->FmtChans = DevFmtStereo;
|
||||||
else if(data->Format.nChannels == 1)
|
else if(self->Format.nChannels == 1)
|
||||||
device->FmtChans = DevFmtMono;
|
device->FmtChans = DevFmtMono;
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
ERR("Unhandled channel count: %d\n", data->Format.nChannels);
|
ERR("Unhandled channel count: %d\n", self->Format.nChannels);
|
||||||
return ALC_FALSE;
|
return ALC_FALSE;
|
||||||
}
|
}
|
||||||
SetDefaultWFXChannelOrder(device);
|
SetDefaultWFXChannelOrder(device);
|
||||||
@ -397,18 +366,18 @@ static ALCboolean WinMMResetPlayback(ALCdevice *device)
|
|||||||
return ALC_TRUE;
|
return ALC_TRUE;
|
||||||
}
|
}
|
||||||
|
|
||||||
static ALCboolean WinMMStartPlayback(ALCdevice *device)
|
static ALCboolean ALCwinmmPlayback_start(ALCwinmmPlayback *self)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)device->ExtraData;
|
ALCdevice *device = STATIC_CAST(ALCbackend, self)->mDevice;
|
||||||
ALbyte *BufferData;
|
ALbyte *BufferData;
|
||||||
ALint BufferSize;
|
ALint BufferSize;
|
||||||
ALuint i;
|
ALuint i;
|
||||||
|
|
||||||
data->killNow = AL_FALSE;
|
self->killNow = AL_FALSE;
|
||||||
if(althrd_create(&data->thread, PlaybackThreadProc, device) != althrd_success)
|
if(althrd_create(&self->thread, ALCwinmmPlayback_mixerProc, self) != althrd_success)
|
||||||
return ALC_FALSE;
|
return ALC_FALSE;
|
||||||
|
|
||||||
InitRef(&data->WaveBuffersCommitted, 0);
|
InitRef(&self->WaveBuffersCommitted, 0);
|
||||||
|
|
||||||
// Create 4 Buffers
|
// Create 4 Buffers
|
||||||
BufferSize = device->UpdateSize*device->NumUpdates / 4;
|
BufferSize = device->UpdateSize*device->NumUpdates / 4;
|
||||||
@ -417,49 +386,153 @@ static ALCboolean WinMMStartPlayback(ALCdevice *device)
|
|||||||
BufferData = calloc(4, BufferSize);
|
BufferData = calloc(4, BufferSize);
|
||||||
for(i = 0;i < 4;i++)
|
for(i = 0;i < 4;i++)
|
||||||
{
|
{
|
||||||
memset(&data->WaveBuffer[i], 0, sizeof(WAVEHDR));
|
memset(&self->WaveBuffer[i], 0, sizeof(WAVEHDR));
|
||||||
data->WaveBuffer[i].dwBufferLength = BufferSize;
|
self->WaveBuffer[i].dwBufferLength = BufferSize;
|
||||||
data->WaveBuffer[i].lpData = ((i==0) ? (CHAR*)BufferData :
|
self->WaveBuffer[i].lpData = ((i==0) ? (CHAR*)BufferData :
|
||||||
(data->WaveBuffer[i-1].lpData +
|
(self->WaveBuffer[i-1].lpData +
|
||||||
data->WaveBuffer[i-1].dwBufferLength));
|
self->WaveBuffer[i-1].dwBufferLength));
|
||||||
waveOutPrepareHeader(data->WaveHandle.Out, &data->WaveBuffer[i], sizeof(WAVEHDR));
|
waveOutPrepareHeader(self->OutHdl, &self->WaveBuffer[i], sizeof(WAVEHDR));
|
||||||
waveOutWrite(data->WaveHandle.Out, &data->WaveBuffer[i], sizeof(WAVEHDR));
|
waveOutWrite(self->OutHdl, &self->WaveBuffer[i], sizeof(WAVEHDR));
|
||||||
IncrementRef(&data->WaveBuffersCommitted);
|
IncrementRef(&self->WaveBuffersCommitted);
|
||||||
}
|
}
|
||||||
|
|
||||||
return ALC_TRUE;
|
return ALC_TRUE;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void WinMMStopPlayback(ALCdevice *device)
|
static void ALCwinmmPlayback_stop(ALCwinmmPlayback *self)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)device->ExtraData;
|
|
||||||
void *buffer = NULL;
|
void *buffer = NULL;
|
||||||
int i;
|
int i;
|
||||||
|
|
||||||
if(data->killNow)
|
if(self->killNow)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
// Set flag to stop processing headers
|
// Set flag to stop processing headers
|
||||||
data->killNow = AL_TRUE;
|
self->killNow = AL_TRUE;
|
||||||
althrd_join(data->thread, &i);
|
althrd_join(self->thread, &i);
|
||||||
|
|
||||||
// Release the wave buffers
|
// Release the wave buffers
|
||||||
for(i = 0;i < 4;i++)
|
for(i = 0;i < 4;i++)
|
||||||
{
|
{
|
||||||
waveOutUnprepareHeader(data->WaveHandle.Out, &data->WaveBuffer[i], sizeof(WAVEHDR));
|
waveOutUnprepareHeader(self->OutHdl, &self->WaveBuffer[i], sizeof(WAVEHDR));
|
||||||
if(i == 0) buffer = data->WaveBuffer[i].lpData;
|
if(i == 0) buffer = self->WaveBuffer[i].lpData;
|
||||||
data->WaveBuffer[i].lpData = NULL;
|
self->WaveBuffer[i].lpData = NULL;
|
||||||
}
|
}
|
||||||
free(buffer);
|
free(buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
static ALCenum WinMMOpenCapture(ALCdevice *Device, const ALCchar *deviceName)
|
|
||||||
|
typedef struct ALCwinmmCapture {
|
||||||
|
DERIVE_FROM_TYPE(ALCbackend);
|
||||||
|
|
||||||
|
RefCount WaveBuffersCommitted;
|
||||||
|
WAVEHDR WaveBuffer[4];
|
||||||
|
|
||||||
|
HWAVEIN InHdl;
|
||||||
|
|
||||||
|
RingBuffer *Ring;
|
||||||
|
|
||||||
|
WAVEFORMATEX Format;
|
||||||
|
|
||||||
|
volatile ALboolean killNow;
|
||||||
|
althrd_t thread;
|
||||||
|
} ALCwinmmCapture;
|
||||||
|
|
||||||
|
static void ALCwinmmCapture_Construct(ALCwinmmCapture *self, ALCdevice *device);
|
||||||
|
static void ALCwinmmCapture_Destruct(ALCwinmmCapture *self);
|
||||||
|
|
||||||
|
static void CALLBACK ALCwinmmCapture_waveInProc(HWAVEIN device, UINT msg, DWORD_PTR instance, DWORD_PTR param1, DWORD_PTR param2);
|
||||||
|
static int ALCwinmmCapture_captureProc(void *arg);
|
||||||
|
|
||||||
|
static ALCenum ALCwinmmCapture_open(ALCwinmmCapture *self, const ALCchar *name);
|
||||||
|
static void ALCwinmmCapture_close(ALCwinmmCapture *self);
|
||||||
|
static DECLARE_FORWARD(ALCwinmmCapture, ALCbackend, ALCboolean, reset)
|
||||||
|
static ALCboolean ALCwinmmCapture_start(ALCwinmmCapture *self);
|
||||||
|
static void ALCwinmmCapture_stop(ALCwinmmCapture *self);
|
||||||
|
static ALCenum ALCwinmmCapture_captureSamples(ALCwinmmCapture *self, ALCvoid *buffer, ALCuint samples);
|
||||||
|
static ALCuint ALCwinmmCapture_availableSamples(ALCwinmmCapture *self);
|
||||||
|
static DECLARE_FORWARD(ALCwinmmCapture, ALCbackend, ALint64, getLatency)
|
||||||
|
static DECLARE_FORWARD(ALCwinmmCapture, ALCbackend, void, lock)
|
||||||
|
static DECLARE_FORWARD(ALCwinmmCapture, ALCbackend, void, unlock)
|
||||||
|
DECLARE_DEFAULT_ALLOCATORS(ALCwinmmCapture)
|
||||||
|
|
||||||
|
DEFINE_ALCBACKEND_VTABLE(ALCwinmmCapture);
|
||||||
|
|
||||||
|
|
||||||
|
static void ALCwinmmCapture_Construct(ALCwinmmCapture *self, ALCdevice *device)
|
||||||
{
|
{
|
||||||
const al_string *iter, *end;
|
ALCbackend_Construct(STATIC_CAST(ALCbackend, self), device);
|
||||||
|
SET_VTABLE2(ALCwinmmCapture, ALCbackend, self);
|
||||||
|
|
||||||
|
InitRef(&self->WaveBuffersCommitted, 0);
|
||||||
|
self->InHdl = NULL;
|
||||||
|
|
||||||
|
self->killNow = AL_TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void ALCwinmmCapture_Destruct(ALCwinmmCapture *self)
|
||||||
|
{
|
||||||
|
if(self->InHdl)
|
||||||
|
waveInClose(self->InHdl);
|
||||||
|
self->InHdl = 0;
|
||||||
|
|
||||||
|
ALCbackend_Destruct(STATIC_CAST(ALCbackend, self));
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* ALCwinmmCapture_waveInProc
|
||||||
|
*
|
||||||
|
* Posts a message to 'ALCwinmmCapture_captureProc' everytime a WaveIn Buffer
|
||||||
|
* is completed and returns to the application (with more data).
|
||||||
|
*/
|
||||||
|
static void CALLBACK ALCwinmmCapture_waveInProc(HWAVEIN UNUSED(device), UINT msg, DWORD_PTR instance, DWORD_PTR param1, DWORD_PTR UNUSED(param2))
|
||||||
|
{
|
||||||
|
ALCwinmmCapture *self = (ALCwinmmCapture*)instance;
|
||||||
|
|
||||||
|
if(msg != WIM_DATA)
|
||||||
|
return;
|
||||||
|
|
||||||
|
DecrementRef(&self->WaveBuffersCommitted);
|
||||||
|
PostThreadMessage(self->thread, msg, 0, param1);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int ALCwinmmCapture_captureProc(void *arg)
|
||||||
|
{
|
||||||
|
ALCwinmmCapture *self = arg;
|
||||||
|
WAVEHDR *WaveHdr;
|
||||||
|
MSG msg;
|
||||||
|
|
||||||
|
althrd_setname(althrd_current(), RECORD_THREAD_NAME);
|
||||||
|
|
||||||
|
if (!self->killNow) while(GetMessage(&msg, NULL, 0, 0))
|
||||||
|
{
|
||||||
|
if(msg.message != WIM_DATA)
|
||||||
|
continue;
|
||||||
|
/* Don't wait for other buffers to finish before quitting. We're
|
||||||
|
* closing so we don't need them. */
|
||||||
|
if(self->killNow)
|
||||||
|
break;
|
||||||
|
|
||||||
|
WaveHdr = ((WAVEHDR*)msg.lParam);
|
||||||
|
WriteRingBuffer(self->Ring, (ALubyte*)WaveHdr->lpData,
|
||||||
|
WaveHdr->dwBytesRecorded/self->Format.nBlockAlign);
|
||||||
|
|
||||||
|
// Send buffer back to capture more data
|
||||||
|
waveInAddBuffer(self->InHdl, WaveHdr, sizeof(WAVEHDR));
|
||||||
|
IncrementRef(&self->WaveBuffersCommitted);
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
static ALCenum ALCwinmmCapture_open(ALCwinmmCapture *self, const ALCchar *name)
|
||||||
|
{
|
||||||
|
ALCdevice *device = STATIC_CAST(ALCbackend, self)->mDevice;
|
||||||
|
const al_string *iter;
|
||||||
ALbyte *BufferData = NULL;
|
ALbyte *BufferData = NULL;
|
||||||
DWORD CapturedDataSize;
|
DWORD CapturedDataSize;
|
||||||
WinMMData *data = NULL;
|
|
||||||
ALint BufferSize;
|
ALint BufferSize;
|
||||||
UINT DeviceID;
|
UINT DeviceID;
|
||||||
MMRESULT res;
|
MMRESULT res;
|
||||||
@ -469,21 +542,15 @@ static ALCenum WinMMOpenCapture(ALCdevice *Device, const ALCchar *deviceName)
|
|||||||
ProbeCaptureDevices();
|
ProbeCaptureDevices();
|
||||||
|
|
||||||
// Find the Device ID matching the deviceName if valid
|
// Find the Device ID matching the deviceName if valid
|
||||||
iter = VECTOR_ITER_BEGIN(CaptureDevices);
|
#define MATCH_DEVNAME(iter) (!al_string_empty(*(iter)) && (!name || al_string_cmp_cstr(*iter, name) == 0))
|
||||||
end = VECTOR_ITER_END(CaptureDevices);
|
VECTOR_FIND_IF(iter, const al_string, CaptureDevices, MATCH_DEVNAME);
|
||||||
for(;iter != end;iter++)
|
if(iter == VECTOR_ITER_END(CaptureDevices))
|
||||||
{
|
|
||||||
if(!al_string_empty(*iter) &&
|
|
||||||
(!deviceName || al_string_cmp_cstr(*iter, deviceName) == 0))
|
|
||||||
{
|
|
||||||
DeviceID = (UINT)(iter - VECTOR_ITER_BEGIN(CaptureDevices));
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if(iter == end)
|
|
||||||
return ALC_INVALID_VALUE;
|
return ALC_INVALID_VALUE;
|
||||||
|
#undef MATCH_DEVNAME
|
||||||
|
|
||||||
switch(Device->FmtChans)
|
DeviceID = (UINT)(iter - VECTOR_ITER_BEGIN(CaptureDevices));
|
||||||
|
|
||||||
|
switch(device->FmtChans)
|
||||||
{
|
{
|
||||||
case DevFmtMono:
|
case DevFmtMono:
|
||||||
case DevFmtStereo:
|
case DevFmtStereo:
|
||||||
@ -491,13 +558,14 @@ static ALCenum WinMMOpenCapture(ALCdevice *Device, const ALCchar *deviceName)
|
|||||||
|
|
||||||
case DevFmtQuad:
|
case DevFmtQuad:
|
||||||
case DevFmtX51:
|
case DevFmtX51:
|
||||||
case DevFmtX51Side:
|
case DevFmtX51Rear:
|
||||||
case DevFmtX61:
|
case DevFmtX61:
|
||||||
case DevFmtX71:
|
case DevFmtX71:
|
||||||
|
case DevFmtBFormat3D:
|
||||||
return ALC_INVALID_ENUM;
|
return ALC_INVALID_ENUM;
|
||||||
}
|
}
|
||||||
|
|
||||||
switch(Device->FmtType)
|
switch(device->FmtType)
|
||||||
{
|
{
|
||||||
case DevFmtUByte:
|
case DevFmtUByte:
|
||||||
case DevFmtShort:
|
case DevFmtShort:
|
||||||
@ -511,147 +579,134 @@ static ALCenum WinMMOpenCapture(ALCdevice *Device, const ALCchar *deviceName)
|
|||||||
return ALC_INVALID_ENUM;
|
return ALC_INVALID_ENUM;
|
||||||
}
|
}
|
||||||
|
|
||||||
data = calloc(1, sizeof(*data));
|
memset(&self->Format, 0, sizeof(WAVEFORMATEX));
|
||||||
if(!data)
|
self->Format.wFormatTag = ((device->FmtType == DevFmtFloat) ?
|
||||||
return ALC_OUT_OF_MEMORY;
|
|
||||||
Device->ExtraData = data;
|
|
||||||
|
|
||||||
memset(&data->Format, 0, sizeof(WAVEFORMATEX));
|
|
||||||
data->Format.wFormatTag = ((Device->FmtType == DevFmtFloat) ?
|
|
||||||
WAVE_FORMAT_IEEE_FLOAT : WAVE_FORMAT_PCM);
|
WAVE_FORMAT_IEEE_FLOAT : WAVE_FORMAT_PCM);
|
||||||
data->Format.nChannels = ChannelsFromDevFmt(Device->FmtChans);
|
self->Format.nChannels = ChannelsFromDevFmt(device->FmtChans);
|
||||||
data->Format.wBitsPerSample = BytesFromDevFmt(Device->FmtType) * 8;
|
self->Format.wBitsPerSample = BytesFromDevFmt(device->FmtType) * 8;
|
||||||
data->Format.nBlockAlign = data->Format.wBitsPerSample *
|
self->Format.nBlockAlign = self->Format.wBitsPerSample *
|
||||||
data->Format.nChannels / 8;
|
self->Format.nChannels / 8;
|
||||||
data->Format.nSamplesPerSec = Device->Frequency;
|
self->Format.nSamplesPerSec = device->Frequency;
|
||||||
data->Format.nAvgBytesPerSec = data->Format.nSamplesPerSec *
|
self->Format.nAvgBytesPerSec = self->Format.nSamplesPerSec *
|
||||||
data->Format.nBlockAlign;
|
self->Format.nBlockAlign;
|
||||||
data->Format.cbSize = 0;
|
self->Format.cbSize = 0;
|
||||||
|
|
||||||
if((res=waveInOpen(&data->WaveHandle.In, DeviceID, &data->Format, (DWORD_PTR)&WaveInProc, (DWORD_PTR)Device, CALLBACK_FUNCTION)) != MMSYSERR_NOERROR)
|
if((res=waveInOpen(&self->InHdl, DeviceID, &self->Format, (DWORD_PTR)&ALCwinmmCapture_waveInProc, (DWORD_PTR)self, CALLBACK_FUNCTION)) != MMSYSERR_NOERROR)
|
||||||
{
|
{
|
||||||
ERR("waveInOpen failed: %u\n", res);
|
ERR("waveInOpen failed: %u\n", res);
|
||||||
goto failure;
|
goto failure;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Allocate circular memory buffer for the captured audio
|
// Allocate circular memory buffer for the captured audio
|
||||||
CapturedDataSize = Device->UpdateSize*Device->NumUpdates;
|
CapturedDataSize = device->UpdateSize*device->NumUpdates;
|
||||||
|
|
||||||
// Make sure circular buffer is at least 100ms in size
|
// Make sure circular buffer is at least 100ms in size
|
||||||
if(CapturedDataSize < (data->Format.nSamplesPerSec / 10))
|
if(CapturedDataSize < (self->Format.nSamplesPerSec / 10))
|
||||||
CapturedDataSize = data->Format.nSamplesPerSec / 10;
|
CapturedDataSize = self->Format.nSamplesPerSec / 10;
|
||||||
|
|
||||||
data->Ring = CreateRingBuffer(data->Format.nBlockAlign, CapturedDataSize);
|
self->Ring = CreateRingBuffer(self->Format.nBlockAlign, CapturedDataSize);
|
||||||
if(!data->Ring)
|
if(!self->Ring) goto failure;
|
||||||
goto failure;
|
|
||||||
|
|
||||||
InitRef(&data->WaveBuffersCommitted, 0);
|
InitRef(&self->WaveBuffersCommitted, 0);
|
||||||
|
|
||||||
// Create 4 Buffers of 50ms each
|
// Create 4 Buffers of 50ms each
|
||||||
BufferSize = data->Format.nAvgBytesPerSec / 20;
|
BufferSize = self->Format.nAvgBytesPerSec / 20;
|
||||||
BufferSize -= (BufferSize % data->Format.nBlockAlign);
|
BufferSize -= (BufferSize % self->Format.nBlockAlign);
|
||||||
|
|
||||||
BufferData = calloc(4, BufferSize);
|
BufferData = calloc(4, BufferSize);
|
||||||
if(!BufferData)
|
if(!BufferData) goto failure;
|
||||||
goto failure;
|
|
||||||
|
|
||||||
for(i = 0;i < 4;i++)
|
for(i = 0;i < 4;i++)
|
||||||
{
|
{
|
||||||
memset(&data->WaveBuffer[i], 0, sizeof(WAVEHDR));
|
memset(&self->WaveBuffer[i], 0, sizeof(WAVEHDR));
|
||||||
data->WaveBuffer[i].dwBufferLength = BufferSize;
|
self->WaveBuffer[i].dwBufferLength = BufferSize;
|
||||||
data->WaveBuffer[i].lpData = ((i==0) ? (CHAR*)BufferData :
|
self->WaveBuffer[i].lpData = ((i==0) ? (CHAR*)BufferData :
|
||||||
(data->WaveBuffer[i-1].lpData +
|
(self->WaveBuffer[i-1].lpData +
|
||||||
data->WaveBuffer[i-1].dwBufferLength));
|
self->WaveBuffer[i-1].dwBufferLength));
|
||||||
data->WaveBuffer[i].dwFlags = 0;
|
self->WaveBuffer[i].dwFlags = 0;
|
||||||
data->WaveBuffer[i].dwLoops = 0;
|
self->WaveBuffer[i].dwLoops = 0;
|
||||||
waveInPrepareHeader(data->WaveHandle.In, &data->WaveBuffer[i], sizeof(WAVEHDR));
|
waveInPrepareHeader(self->InHdl, &self->WaveBuffer[i], sizeof(WAVEHDR));
|
||||||
waveInAddBuffer(data->WaveHandle.In, &data->WaveBuffer[i], sizeof(WAVEHDR));
|
waveInAddBuffer(self->InHdl, &self->WaveBuffer[i], sizeof(WAVEHDR));
|
||||||
IncrementRef(&data->WaveBuffersCommitted);
|
IncrementRef(&self->WaveBuffersCommitted);
|
||||||
}
|
}
|
||||||
|
|
||||||
if(althrd_create(&data->thread, CaptureThreadProc, Device) != althrd_success)
|
self->killNow = AL_FALSE;
|
||||||
|
if(althrd_create(&self->thread, ALCwinmmCapture_captureProc, self) != althrd_success)
|
||||||
goto failure;
|
goto failure;
|
||||||
|
|
||||||
al_string_copy(&Device->DeviceName, VECTOR_ELEM(CaptureDevices, DeviceID));
|
al_string_copy(&device->DeviceName, VECTOR_ELEM(CaptureDevices, DeviceID));
|
||||||
return ALC_NO_ERROR;
|
return ALC_NO_ERROR;
|
||||||
|
|
||||||
failure:
|
failure:
|
||||||
if(BufferData)
|
if(BufferData)
|
||||||
{
|
{
|
||||||
for(i = 0;i < 4;i++)
|
for(i = 0;i < 4;i++)
|
||||||
waveInUnprepareHeader(data->WaveHandle.In, &data->WaveBuffer[i], sizeof(WAVEHDR));
|
waveInUnprepareHeader(self->InHdl, &self->WaveBuffer[i], sizeof(WAVEHDR));
|
||||||
free(BufferData);
|
free(BufferData);
|
||||||
}
|
}
|
||||||
|
|
||||||
if(data->Ring)
|
if(self->Ring)
|
||||||
DestroyRingBuffer(data->Ring);
|
DestroyRingBuffer(self->Ring);
|
||||||
|
self->Ring = NULL;
|
||||||
|
|
||||||
if(data->WaveHandle.In)
|
if(self->InHdl)
|
||||||
waveInClose(data->WaveHandle.In);
|
waveInClose(self->InHdl);
|
||||||
|
self->InHdl = NULL;
|
||||||
|
|
||||||
free(data);
|
|
||||||
Device->ExtraData = NULL;
|
|
||||||
return ALC_INVALID_VALUE;
|
return ALC_INVALID_VALUE;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void WinMMCloseCapture(ALCdevice *Device)
|
static void ALCwinmmCapture_close(ALCwinmmCapture *self)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)Device->ExtraData;
|
|
||||||
void *buffer = NULL;
|
void *buffer = NULL;
|
||||||
int i;
|
int i;
|
||||||
|
|
||||||
/* Tell the processing thread to quit and wait for it to do so. */
|
/* Tell the processing thread to quit and wait for it to do so. */
|
||||||
data->killNow = AL_TRUE;
|
self->killNow = AL_TRUE;
|
||||||
PostThreadMessage(data->thread, WM_QUIT, 0, 0);
|
PostThreadMessage(self->thread, WM_QUIT, 0, 0);
|
||||||
|
|
||||||
althrd_join(data->thread, &i);
|
althrd_join(self->thread, &i);
|
||||||
|
|
||||||
/* Make sure capture is stopped and all pending buffers are flushed. */
|
/* Make sure capture is stopped and all pending buffers are flushed. */
|
||||||
waveInReset(data->WaveHandle.In);
|
waveInReset(self->InHdl);
|
||||||
|
|
||||||
// Release the wave buffers
|
// Release the wave buffers
|
||||||
for(i = 0;i < 4;i++)
|
for(i = 0;i < 4;i++)
|
||||||
{
|
{
|
||||||
waveInUnprepareHeader(data->WaveHandle.In, &data->WaveBuffer[i], sizeof(WAVEHDR));
|
waveInUnprepareHeader(self->InHdl, &self->WaveBuffer[i], sizeof(WAVEHDR));
|
||||||
if(i == 0) buffer = data->WaveBuffer[i].lpData;
|
if(i == 0) buffer = self->WaveBuffer[i].lpData;
|
||||||
data->WaveBuffer[i].lpData = NULL;
|
self->WaveBuffer[i].lpData = NULL;
|
||||||
}
|
}
|
||||||
free(buffer);
|
free(buffer);
|
||||||
|
|
||||||
DestroyRingBuffer(data->Ring);
|
DestroyRingBuffer(self->Ring);
|
||||||
data->Ring = NULL;
|
self->Ring = NULL;
|
||||||
|
|
||||||
// Close the Wave device
|
// Close the Wave device
|
||||||
waveInClose(data->WaveHandle.In);
|
waveInClose(self->InHdl);
|
||||||
data->WaveHandle.In = 0;
|
self->InHdl = NULL;
|
||||||
|
|
||||||
free(data);
|
|
||||||
Device->ExtraData = NULL;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void WinMMStartCapture(ALCdevice *Device)
|
static ALCboolean ALCwinmmCapture_start(ALCwinmmCapture *self)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)Device->ExtraData;
|
waveInStart(self->InHdl);
|
||||||
waveInStart(data->WaveHandle.In);
|
return ALC_TRUE;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void WinMMStopCapture(ALCdevice *Device)
|
static void ALCwinmmCapture_stop(ALCwinmmCapture *self)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)Device->ExtraData;
|
waveInStop(self->InHdl);
|
||||||
waveInStop(data->WaveHandle.In);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static ALCenum WinMMCaptureSamples(ALCdevice *Device, ALCvoid *Buffer, ALCuint Samples)
|
static ALCenum ALCwinmmCapture_captureSamples(ALCwinmmCapture *self, ALCvoid *buffer, ALCuint samples)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)Device->ExtraData;
|
ReadRingBuffer(self->Ring, buffer, samples);
|
||||||
ReadRingBuffer(data->Ring, Buffer, Samples);
|
|
||||||
return ALC_NO_ERROR;
|
return ALC_NO_ERROR;
|
||||||
}
|
}
|
||||||
|
|
||||||
static ALCuint WinMMAvailableSamples(ALCdevice *Device)
|
static ALCuint ALCwinmmCapture_availableSamples(ALCwinmmCapture *self)
|
||||||
{
|
{
|
||||||
WinMMData *data = (WinMMData*)Device->ExtraData;
|
return RingBufferSize(self->Ring);
|
||||||
return RingBufferSize(data->Ring);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -666,31 +721,29 @@ static inline void AppendCaptureDeviceList2(const al_string *name)
|
|||||||
AppendCaptureDeviceList(al_string_get_cstr(*name));
|
AppendCaptureDeviceList(al_string_get_cstr(*name));
|
||||||
}
|
}
|
||||||
|
|
||||||
static const BackendFuncs WinMMFuncs = {
|
typedef struct ALCwinmmBackendFactory {
|
||||||
WinMMOpenPlayback,
|
DERIVE_FROM_TYPE(ALCbackendFactory);
|
||||||
WinMMClosePlayback,
|
} ALCwinmmBackendFactory;
|
||||||
WinMMResetPlayback,
|
#define ALCWINMMBACKENDFACTORY_INITIALIZER { { GET_VTABLE2(ALCwinmmBackendFactory, ALCbackendFactory) } }
|
||||||
WinMMStartPlayback,
|
|
||||||
WinMMStopPlayback,
|
|
||||||
WinMMOpenCapture,
|
|
||||||
WinMMCloseCapture,
|
|
||||||
WinMMStartCapture,
|
|
||||||
WinMMStopCapture,
|
|
||||||
WinMMCaptureSamples,
|
|
||||||
WinMMAvailableSamples,
|
|
||||||
ALCdevice_GetLatencyDefault
|
|
||||||
};
|
|
||||||
|
|
||||||
ALCboolean alcWinMMInit(BackendFuncs *FuncList)
|
static ALCboolean ALCwinmmBackendFactory_init(ALCwinmmBackendFactory *self);
|
||||||
|
static void ALCwinmmBackendFactory_deinit(ALCwinmmBackendFactory *self);
|
||||||
|
static ALCboolean ALCwinmmBackendFactory_querySupport(ALCwinmmBackendFactory *self, ALCbackend_Type type);
|
||||||
|
static void ALCwinmmBackendFactory_probe(ALCwinmmBackendFactory *self, enum DevProbe type);
|
||||||
|
static ALCbackend* ALCwinmmBackendFactory_createBackend(ALCwinmmBackendFactory *self, ALCdevice *device, ALCbackend_Type type);
|
||||||
|
|
||||||
|
DEFINE_ALCBACKENDFACTORY_VTABLE(ALCwinmmBackendFactory);
|
||||||
|
|
||||||
|
|
||||||
|
static ALCboolean ALCwinmmBackendFactory_init(ALCwinmmBackendFactory* UNUSED(self))
|
||||||
{
|
{
|
||||||
VECTOR_INIT(PlaybackDevices);
|
VECTOR_INIT(PlaybackDevices);
|
||||||
VECTOR_INIT(CaptureDevices);
|
VECTOR_INIT(CaptureDevices);
|
||||||
|
|
||||||
*FuncList = WinMMFuncs;
|
|
||||||
return ALC_TRUE;
|
return ALC_TRUE;
|
||||||
}
|
}
|
||||||
|
|
||||||
void alcWinMMDeinit()
|
static void ALCwinmmBackendFactory_deinit(ALCwinmmBackendFactory* UNUSED(self))
|
||||||
{
|
{
|
||||||
clear_devlist(&PlaybackDevices);
|
clear_devlist(&PlaybackDevices);
|
||||||
VECTOR_DEINIT(PlaybackDevices);
|
VECTOR_DEINIT(PlaybackDevices);
|
||||||
@ -699,7 +752,14 @@ void alcWinMMDeinit()
|
|||||||
VECTOR_DEINIT(CaptureDevices);
|
VECTOR_DEINIT(CaptureDevices);
|
||||||
}
|
}
|
||||||
|
|
||||||
void alcWinMMProbe(enum DevProbe type)
|
static ALCboolean ALCwinmmBackendFactory_querySupport(ALCwinmmBackendFactory* UNUSED(self), ALCbackend_Type type)
|
||||||
|
{
|
||||||
|
if(type == ALCbackend_Playback || type == ALCbackend_Capture)
|
||||||
|
return ALC_TRUE;
|
||||||
|
return ALC_FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void ALCwinmmBackendFactory_probe(ALCwinmmBackendFactory* UNUSED(self), enum DevProbe type)
|
||||||
{
|
{
|
||||||
switch(type)
|
switch(type)
|
||||||
{
|
{
|
||||||
@ -714,3 +774,39 @@ void alcWinMMProbe(enum DevProbe type)
|
|||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static ALCbackend* ALCwinmmBackendFactory_createBackend(ALCwinmmBackendFactory* UNUSED(self), ALCdevice *device, ALCbackend_Type type)
|
||||||
|
{
|
||||||
|
if(type == ALCbackend_Playback)
|
||||||
|
{
|
||||||
|
ALCwinmmPlayback *backend;
|
||||||
|
|
||||||
|
backend = ALCwinmmPlayback_New(sizeof(*backend));
|
||||||
|
if(!backend) return NULL;
|
||||||
|
memset(backend, 0, sizeof(*backend));
|
||||||
|
|
||||||
|
ALCwinmmPlayback_Construct(backend, device);
|
||||||
|
|
||||||
|
return STATIC_CAST(ALCbackend, backend);
|
||||||
|
}
|
||||||
|
if(type == ALCbackend_Capture)
|
||||||
|
{
|
||||||
|
ALCwinmmCapture *backend;
|
||||||
|
|
||||||
|
backend = ALCwinmmCapture_New(sizeof(*backend));
|
||||||
|
if(!backend) return NULL;
|
||||||
|
memset(backend, 0, sizeof(*backend));
|
||||||
|
|
||||||
|
ALCwinmmCapture_Construct(backend, device);
|
||||||
|
|
||||||
|
return STATIC_CAST(ALCbackend, backend);
|
||||||
|
}
|
||||||
|
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
ALCbackendFactory *ALCwinmmBackendFactory_getFactory(void)
|
||||||
|
{
|
||||||
|
static ALCwinmmBackendFactory factory = ALCWINMMBACKENDFACTORY_INITIALIZER;
|
||||||
|
return STATIC_CAST(ALCbackendFactory, &factory);
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user