287 lines
12 KiB
Objective-C
287 lines
12 KiB
Objective-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
|
|
*/
|
|
#import <Cocoa/Cocoa.h>
|
|
#include <sys/xattr.h>
|
|
|
|
NSString *appName = @"Telegram.app";
|
|
NSString *appDir = nil;
|
|
NSString *workDir = nil;
|
|
|
|
#ifdef _DEBUG
|
|
BOOL _debug = YES;
|
|
#else
|
|
BOOL _debug = NO;
|
|
#endif
|
|
|
|
NSFileHandle *_logFile = nil;
|
|
void openLog() {
|
|
if (!_debug || _logFile) return;
|
|
NSString *logDir = [workDir stringByAppendingString:@"DebugLogs"];
|
|
if (![[NSFileManager defaultManager] createDirectoryAtPath:logDir withIntermediateDirectories:YES attributes:nil error:nil]) {
|
|
return;
|
|
}
|
|
|
|
NSDateFormatter *fmt = [[NSDateFormatter alloc] initWithDateFormat:@"DebugLogs/%Y%m%d_%H%M%S_upd.txt" allowNaturalLanguage:NO];
|
|
NSString *logPath = [workDir stringByAppendingString:[fmt stringFromDate:[NSDate date]]];
|
|
[[NSFileManager defaultManager] createFileAtPath:logPath contents:nil attributes:nil];
|
|
_logFile = [NSFileHandle fileHandleForWritingAtPath:logPath];
|
|
}
|
|
|
|
void closeLog() {
|
|
if (!_logFile) return;
|
|
|
|
[_logFile closeFile];
|
|
}
|
|
|
|
void writeLog(NSString *msg) {
|
|
if (!_logFile) return;
|
|
|
|
[_logFile writeData:[[msg stringByAppendingString:@"\n"] dataUsingEncoding:NSUTF8StringEncoding]];
|
|
[_logFile synchronizeFile];
|
|
}
|
|
|
|
void RemoveQuarantineAttribute(NSString *path) {
|
|
const char *kQuarantineAttribute = "com.apple.quarantine";
|
|
|
|
writeLog([@"Removing quarantine: " stringByAppendingString:path]);
|
|
removexattr([path fileSystemRepresentation], kQuarantineAttribute, 0);
|
|
}
|
|
|
|
void RemoveQuarantineFromBundle(NSString *path) {
|
|
RemoveQuarantineAttribute(path);
|
|
RemoveQuarantineAttribute([path stringByAppendingString:@"/Contents/MacOS/Telegram"]);
|
|
RemoveQuarantineAttribute([path stringByAppendingString:@"/Contents/Helpers/crashpad_handler"]);
|
|
RemoveQuarantineAttribute([path stringByAppendingString:@"/Contents/Frameworks/Updater"]);
|
|
}
|
|
|
|
void delFolder() {
|
|
writeLog([@"Fully clearing old path: " stringByAppendingString:[workDir stringByAppendingString:@"tupdates/ready"]]);
|
|
if (![[NSFileManager defaultManager] removeItemAtPath:[workDir stringByAppendingString:@"tupdates/ready"] error:nil]) {
|
|
writeLog(@"Failed to clear old path! :( New path was used?..");
|
|
}
|
|
writeLog([@"Fully clearing new path: " stringByAppendingString:[workDir stringByAppendingString:@"tupdates/temp"]]);
|
|
if (![[NSFileManager defaultManager] removeItemAtPath:[workDir stringByAppendingString:@"tupdates/temp"] error:nil]) {
|
|
writeLog(@"Error: failed to clear new path! :(");
|
|
}
|
|
rmdir([[workDir stringByAppendingString:@"tupdates"] fileSystemRepresentation]);
|
|
}
|
|
|
|
int main(int argc, const char * argv[]) {
|
|
NSString *path = [[NSBundle mainBundle] bundlePath];
|
|
if (!path) {
|
|
return -1;
|
|
}
|
|
NSRange range = [path rangeOfString:@".app/" options:NSBackwardsSearch];
|
|
if (range.location == NSNotFound) {
|
|
return -1;
|
|
}
|
|
path = [path substringToIndex:range.location > 0 ? range.location : 0];
|
|
|
|
range = [path rangeOfString:@"/" options:NSBackwardsSearch];
|
|
NSString *appRealName = (range.location == NSNotFound) ? path : [path substringFromIndex:range.location + 1];
|
|
appRealName = [[NSArray arrayWithObjects:appRealName, @".app", nil] componentsJoinedByString:@""];
|
|
appDir = (range.location == NSNotFound) ? @"" : [path substringToIndex:range.location + 1];
|
|
NSString *appDirFull = [appDir stringByAppendingString:appRealName];
|
|
|
|
openLog();
|
|
pid_t procId = 0;
|
|
BOOL update = YES, toSettings = NO, autoStart = NO, startInTray = NO, freeType = NO;
|
|
BOOL customWorkingDir = NO;
|
|
NSString *key = nil;
|
|
for (int i = 0; i < argc; ++i) {
|
|
if ([@"-workpath" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
if (++i < argc) {
|
|
workDir = [NSString stringWithUTF8String:argv[i]];
|
|
}
|
|
} else if ([@"-procid" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
if (++i < argc) {
|
|
NSNumberFormatter *formatter = [[NSNumberFormatter alloc] init];
|
|
[formatter setNumberStyle:NSNumberFormatterDecimalStyle];
|
|
procId = [[formatter numberFromString:[NSString stringWithUTF8String:argv[i]]] intValue];
|
|
}
|
|
} else if ([@"-noupdate" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
update = NO;
|
|
} else if ([@"-tosettings" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
toSettings = YES;
|
|
} else if ([@"-autostart" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
autoStart = YES;
|
|
} else if ([@"-debug" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
_debug = YES;
|
|
} else if ([@"-startintray" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
startInTray = YES;
|
|
} else if ([@"-freetype" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
freeType = YES;
|
|
} else if ([@"-workdir_custom" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
customWorkingDir = YES;
|
|
} else if ([@"-key" isEqualToString:[NSString stringWithUTF8String:argv[i]]]) {
|
|
if (++i < argc) key = [NSString stringWithUTF8String:argv[i]];
|
|
}
|
|
}
|
|
if (!workDir) {
|
|
workDir = appDir;
|
|
customWorkingDir = NO;
|
|
}
|
|
openLog();
|
|
NSMutableArray *argsArr = [[NSMutableArray alloc] initWithCapacity:argc];
|
|
for (int i = 0; i < argc; ++i) {
|
|
[argsArr addObject:[NSString stringWithUTF8String:argv[i]]];
|
|
}
|
|
writeLog([[NSArray arrayWithObjects:@"Arguments: '", [argsArr componentsJoinedByString:@"' '"], @"'..", nil] componentsJoinedByString:@""]);
|
|
if (key) writeLog([@"Key: " stringByAppendingString:key]);
|
|
if (toSettings) writeLog(@"To Settings!");
|
|
|
|
if (procId) {
|
|
NSRunningApplication *app = [NSRunningApplication runningApplicationWithProcessIdentifier:procId];
|
|
for (int i = 0; i < 5 && app != nil && ![app isTerminated]; ++i) {
|
|
usleep(200000);
|
|
app = [NSRunningApplication runningApplicationWithProcessIdentifier:procId];
|
|
}
|
|
if (app) [app forceTerminate];
|
|
app = [NSRunningApplication runningApplicationWithProcessIdentifier:procId];
|
|
for (int i = 0; i < 5 && app != nil && ![app isTerminated]; ++i) {
|
|
usleep(200000);
|
|
app = [NSRunningApplication runningApplicationWithProcessIdentifier:procId];
|
|
}
|
|
}
|
|
|
|
if (update) {
|
|
NSFileManager *fileManager = [NSFileManager defaultManager];
|
|
NSString *readyFilePath = [workDir stringByAppendingString:@"tupdates/temp/ready"];
|
|
NSString *srcDir = [workDir stringByAppendingString:@"tupdates/temp/"], *srcEnum = [workDir stringByAppendingString:@"tupdates/temp"];
|
|
if ([fileManager fileExistsAtPath:readyFilePath]) {
|
|
writeLog([@"Ready file found! Using new path: " stringByAppendingString: srcEnum]);
|
|
} else {
|
|
srcDir = [workDir stringByAppendingString:@"tupdates/ready/"]; // old
|
|
srcEnum = [workDir stringByAppendingString:@"tupdates/ready"];
|
|
writeLog([@"Ready file not found! Using old path: " stringByAppendingString: srcEnum]);
|
|
}
|
|
|
|
writeLog([@"Starting update files iteration, path: " stringByAppendingString: srcEnum]);
|
|
|
|
// Take the Updater (this currently running binary) from the place where it was placed by Telegram
|
|
// and copy it to the folder with the new version of the app (ready),
|
|
// so it won't be deleted when we will clear the "Telegram.app/Contents" folder.
|
|
NSString *oldVersionUpdaterPath = [appDirFull stringByAppendingString: @"/Contents/Frameworks/Updater" ];
|
|
NSString *newVersionUpdaterPath = [srcEnum stringByAppendingString:[[NSArray arrayWithObjects:@"/", appName, @"/Contents/Frameworks/Updater", nil] componentsJoinedByString:@""]];
|
|
writeLog([[NSArray arrayWithObjects: @"Copying Updater from old path ", oldVersionUpdaterPath, @" to new path ", newVersionUpdaterPath, nil] componentsJoinedByString:@""]);
|
|
if (![fileManager fileExistsAtPath:newVersionUpdaterPath]) {
|
|
if (![fileManager copyItemAtPath:oldVersionUpdaterPath toPath:newVersionUpdaterPath error:nil]) {
|
|
writeLog([[NSArray arrayWithObjects: @"Failed to copy file from ", oldVersionUpdaterPath, @" to ", newVersionUpdaterPath, nil] componentsJoinedByString:@""]);
|
|
delFolder();
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
|
|
NSString *contentsPath = [appDirFull stringByAppendingString: @"/Contents"];
|
|
writeLog([[NSArray arrayWithObjects: @"Clearing dir ", contentsPath, nil] componentsJoinedByString:@""]);
|
|
if (![fileManager removeItemAtPath:contentsPath error:nil]) {
|
|
writeLog([@"Failed to clear path for directory " stringByAppendingString:contentsPath]);
|
|
delFolder();
|
|
return -1;
|
|
}
|
|
|
|
NSArray *keys = [NSArray arrayWithObject:NSURLIsDirectoryKey];
|
|
NSDirectoryEnumerator *enumerator = [fileManager
|
|
enumeratorAtURL:[NSURL fileURLWithPath:srcEnum]
|
|
includingPropertiesForKeys:keys
|
|
options:0
|
|
errorHandler:^(NSURL *url, NSError *error) {
|
|
writeLog([[[@"Error in enumerating " stringByAppendingString:[url absoluteString]] stringByAppendingString: @" error is: "] stringByAppendingString: [error description]]);
|
|
return NO;
|
|
}];
|
|
for (NSURL *url in enumerator) {
|
|
NSString *srcPath = [url path];
|
|
writeLog([@"Handling file " stringByAppendingString:srcPath]);
|
|
NSRange r = [srcPath rangeOfString:srcDir];
|
|
if (r.location != 0) {
|
|
writeLog([@"Bad file found, no base path " stringByAppendingString:srcPath]);
|
|
delFolder();
|
|
break;
|
|
}
|
|
NSString *pathPart = [srcPath substringFromIndex:r.length];
|
|
r = [pathPart rangeOfString:appName];
|
|
if (r.location != 0) {
|
|
writeLog([@"Skipping not app file " stringByAppendingString:srcPath]);
|
|
continue;
|
|
}
|
|
NSString *dstPath = [appDirFull stringByAppendingString:[pathPart substringFromIndex:r.length]];
|
|
NSError *error;
|
|
NSNumber *isDirectory = nil;
|
|
if (![url getResourceValue:&isDirectory forKey:NSURLIsDirectoryKey error:&error]) {
|
|
writeLog([@"Failed to get IsDirectory for file " stringByAppendingString:[url path]]);
|
|
delFolder();
|
|
break;
|
|
}
|
|
if ([isDirectory boolValue]) {
|
|
writeLog([[NSArray arrayWithObjects: @"Copying dir ", srcPath, @" to ", dstPath, nil] componentsJoinedByString:@""]);
|
|
if (![fileManager createDirectoryAtPath:dstPath withIntermediateDirectories:YES attributes:nil error:nil]) {
|
|
writeLog([@"Failed to force path for directory " stringByAppendingString:dstPath]);
|
|
delFolder();
|
|
break;
|
|
}
|
|
} else if ([srcPath isEqualToString:readyFilePath]) {
|
|
writeLog([[NSArray arrayWithObjects: @"Skipping ready file ", srcPath, nil] componentsJoinedByString:@""]);
|
|
} else if ([fileManager fileExistsAtPath:dstPath]) {
|
|
if (![[NSData dataWithContentsOfFile:srcPath] writeToFile:dstPath atomically:YES]) {
|
|
writeLog([@"Failed to edit file " stringByAppendingString:dstPath]);
|
|
delFolder();
|
|
break;
|
|
}
|
|
} else {
|
|
if (![fileManager copyItemAtPath:srcPath toPath:dstPath error:nil]) {
|
|
writeLog([@"Failed to copy file to " stringByAppendingString:dstPath]);
|
|
delFolder();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
delFolder();
|
|
}
|
|
|
|
NSString *appPath = [[NSArray arrayWithObjects:appDir, appRealName, nil] componentsJoinedByString:@""];
|
|
|
|
RemoveQuarantineFromBundle(appPath);
|
|
|
|
NSMutableArray *args = [[NSMutableArray alloc] initWithObjects: @"-noupdate", nil];
|
|
if (toSettings) [args addObject:@"-tosettings"];
|
|
if (_debug) [args addObject:@"-debug"];
|
|
if (startInTray) [args addObject:@"-startintray"];
|
|
if (freeType) [args addObject:@"-freetype"];
|
|
if (autoStart) [args addObject:@"-autostart"];
|
|
if (key) {
|
|
[args addObject:@"-key"];
|
|
[args addObject:key];
|
|
}
|
|
if (customWorkingDir) {
|
|
[args addObject:@"-workdir"];
|
|
[args addObject:workDir];
|
|
}
|
|
writeLog([[NSArray arrayWithObjects:@"Running application '", appPath, @"' with args '", [args componentsJoinedByString:@"' '"], @"'..", nil] componentsJoinedByString:@""]);
|
|
|
|
for (int i = 0; i < 5; ++i) {
|
|
NSError *error = nil;
|
|
NSRunningApplication *result = [[NSWorkspace sharedWorkspace]
|
|
launchApplicationAtURL:[NSURL fileURLWithPath:appPath]
|
|
options:NSWorkspaceLaunchDefault
|
|
configuration:[NSDictionary
|
|
dictionaryWithObject:args
|
|
forKey:NSWorkspaceLaunchConfigurationArguments]
|
|
error:&error];
|
|
if (result) {
|
|
closeLog();
|
|
return 0;
|
|
}
|
|
writeLog([[NSString stringWithFormat:@"Could not run application, error %ld: ", (long)[error code]] stringByAppendingString: error ? [error localizedDescription] : @"(nil)"]);
|
|
usleep(200000);
|
|
}
|
|
closeLog();
|
|
return -1;
|
|
}
|
|
|