149 lines
		
	
	
	
		
			4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			149 lines
		
	
	
	
		
			4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
This file is part of Telegram Desktop,
 | 
						|
the official desktop application for the Telegram messaging service.
 | 
						|
 | 
						|
For license and copyright information please follow this link:
 | 
						|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
 | 
						|
*/
 | 
						|
#include "platform/linux/launcher_linux.h"
 | 
						|
 | 
						|
#include "kotato/kotato_settings.h"
 | 
						|
#include "core/crash_reports.h"
 | 
						|
#include "core/update_checker.h"
 | 
						|
#include "webview/platform/linux/webview_linux_webkitgtk.h"
 | 
						|
 | 
						|
#include <QtWidgets/QApplication>
 | 
						|
#include <glib/glib.hpp>
 | 
						|
 | 
						|
using namespace gi::repository;
 | 
						|
 | 
						|
namespace Platform {
 | 
						|
 | 
						|
Launcher::Launcher(int argc, char *argv[])
 | 
						|
: Core::Launcher(argc, argv) {
 | 
						|
}
 | 
						|
 | 
						|
int Launcher::exec() {
 | 
						|
	for (auto i = arguments().begin(), e = arguments().end(); i != e; ++i) {
 | 
						|
		if (*i == u"-webviewhelper"_q && std::distance(i, e) > 1) {
 | 
						|
			Webview::WebKitGTK::SetSocketPath((i + 1)->toStdString());
 | 
						|
			return Webview::WebKitGTK::Exec();
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return Core::Launcher::exec();
 | 
						|
}
 | 
						|
 | 
						|
bool Launcher::launchUpdater(UpdaterLaunch action) {
 | 
						|
	if (cExeName().isEmpty()) {
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
 | 
						|
	const auto justRelaunch = action == UpdaterLaunch::JustRelaunch;
 | 
						|
	if (action == UpdaterLaunch::PerformUpdate) {
 | 
						|
		_updating = true;
 | 
						|
	}
 | 
						|
 | 
						|
	std::vector<std::string> argumentsList;
 | 
						|
 | 
						|
	// What we are launching.
 | 
						|
	const auto launching = justRelaunch
 | 
						|
		? (cExeDir() + cExeName())
 | 
						|
		: cWriteProtected()
 | 
						|
		? u"pkexec"_q
 | 
						|
		: (cExeDir() + u"Updater"_q);
 | 
						|
	argumentsList.push_back(launching.toStdString());
 | 
						|
 | 
						|
	if (justRelaunch) {
 | 
						|
		// argv[0] that is passed to what we are launching.
 | 
						|
		// It should be added explicitly in case of FILE_AND_ARGV_ZERO_.
 | 
						|
		const auto argv0 = !arguments().isEmpty()
 | 
						|
			? arguments().first()
 | 
						|
			: launching;
 | 
						|
		argumentsList.push_back(argv0.toStdString());
 | 
						|
	} else if (cWriteProtected()) {
 | 
						|
		// Elevated process that pkexec should launch.
 | 
						|
		const auto elevated = cWorkingDir() + u"tupdates/temp/Updater"_q;
 | 
						|
		argumentsList.push_back(elevated.toStdString());
 | 
						|
	}
 | 
						|
 | 
						|
	if (Logs::DebugEnabled()) {
 | 
						|
		argumentsList.push_back("-debug");
 | 
						|
	}
 | 
						|
 | 
						|
	if (justRelaunch) {
 | 
						|
		if (cLaunchMode() == LaunchModeAutoStart) {
 | 
						|
			argumentsList.push_back("-autostart");
 | 
						|
		}
 | 
						|
		if (cStartInTray()) {
 | 
						|
			argumentsList.push_back("-startintray");
 | 
						|
		}
 | 
						|
		if (cDataFile() != u"data"_q) {
 | 
						|
			argumentsList.push_back("-key");
 | 
						|
			argumentsList.push_back(cDataFile().toStdString());
 | 
						|
		}
 | 
						|
		if (!_updating) {
 | 
						|
			argumentsList.push_back("-noupdate");
 | 
						|
			argumentsList.push_back("-tosettings");
 | 
						|
		}
 | 
						|
		if (customWorkingDir()) {
 | 
						|
			argumentsList.push_back("-workdir");
 | 
						|
			argumentsList.push_back(cWorkingDir().toStdString());
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		// Don't relaunch Telegram.
 | 
						|
		argumentsList.push_back("-justupdate");
 | 
						|
 | 
						|
		argumentsList.push_back("-workpath");
 | 
						|
		argumentsList.push_back(cWorkingDir().toStdString());
 | 
						|
		argumentsList.push_back("-exename");
 | 
						|
		argumentsList.push_back(cExeName().toStdString());
 | 
						|
		argumentsList.push_back("-exepath");
 | 
						|
		argumentsList.push_back(cExeDir().toStdString());
 | 
						|
		if (cWriteProtected()) {
 | 
						|
			argumentsList.push_back("-writeprotected");
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (!::Kotato::JsonSettings::GetBool("api_use_env")) {
 | 
						|
		argumentsList.push_back("-no-env-api");
 | 
						|
	}
 | 
						|
 | 
						|
	if (::Kotato::JsonSettings::GetBool("api_start_params")) {
 | 
						|
		argumentsList.push_back("-api-id");
 | 
						|
		argumentsList.push_back(std::to_string(::Kotato::JsonSettings::GetInt("api_id")));
 | 
						|
		argumentsList.push_back("-api-hash");
 | 
						|
		argumentsList.push_back(::Kotato::JsonSettings::GetString("api_hash").toStdString());
 | 
						|
	}
 | 
						|
 | 
						|
	Logs::closeMain();
 | 
						|
	CrashReports::Finish();
 | 
						|
 | 
						|
	if (justRelaunch) {
 | 
						|
		return GLib::spawn_async(
 | 
						|
			initialWorkingDir().toStdString(),
 | 
						|
			argumentsList,
 | 
						|
			{},
 | 
						|
			GLib::SpawnFlags::FILE_AND_ARGV_ZERO_,
 | 
						|
			nullptr,
 | 
						|
			nullptr,
 | 
						|
			nullptr);
 | 
						|
	} else if (!GLib::spawn_sync(
 | 
						|
			argumentsList,
 | 
						|
			{},
 | 
						|
			// if the spawn is sync, working directory is not set
 | 
						|
			// and GLib::SpawnFlags::LEAVE_DESCRIPTORS_OPEN_ is set,
 | 
						|
			// it goes through an optimized code path
 | 
						|
			GLib::SpawnFlags::SEARCH_PATH_
 | 
						|
				| GLib::SpawnFlags::LEAVE_DESCRIPTORS_OPEN_,
 | 
						|
			nullptr,
 | 
						|
			nullptr,
 | 
						|
			nullptr,
 | 
						|
			nullptr,
 | 
						|
			nullptr)) {
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
	return launchUpdater(UpdaterLaunch::JustRelaunch);
 | 
						|
}
 | 
						|
 | 
						|
} // namespace
 |