Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(6131)

Unified Diff: chrome/installer/mac/app/Unpacker.m

Issue 2203583002: Added unpacking step (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Added use of temporary folders, removed semaphore from main installer code, adjusted some files' APIs, resolved remaining comments Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: chrome/installer/mac/app/Unpacker.m
diff --git a/chrome/installer/mac/app/Unpacker.m b/chrome/installer/mac/app/Unpacker.m
new file mode 100644
index 0000000000000000000000000000000000000000..af7b2168232fe4f71cd2197ce9fef26e347506c6
--- /dev/null
+++ b/chrome/installer/mac/app/Unpacker.m
@@ -0,0 +1,174 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#import "Unpacker.h"
+
+#include <DiskArbitration/DiskArbitration.h>
+#include <dispatch/dispatch.h>
+
+#import "Downloader.h"
+
+@interface Unpacker () {
+ DASessionRef session;
+ dispatch_queue_t unpack_dq;
+
+ // TODO: narrow the scope for these variables
+ NSURL* temporaryDirectoryURL;
+ NSURL* temporaryDiskImageURL;
+ NSString* mountPath;
Sidney San Martín 2016/08/13 12:37:18 I believe instance variables should always have tr
Anna Zeng 2016/08/16 23:07:39 Done.
+}
+@end
+
+@implementation Unpacker
+
+@synthesize delegate = delegate_;
+@synthesize appPath = appPath_;
+
+// TODO: clean up error handling & pass up to appdelegate
+static void unmount_callback(DADiskRef disk,
Sidney San Martín 2016/08/13 12:37:19 These functions should not be in the @implementati
Anna Zeng 2016/08/16 23:07:39 Done.
+ DADissenterRef dissenter,
+ void* context) {
+ if (dissenter) {
Sidney San Martín 2016/08/13 12:37:19 There's a similar block of code in eject_callback.
Anna Zeng 2016/08/16 23:07:40 Done.
+ DAReturn status = DADissenterGetStatus(dissenter);
+ if (unix_err(status)) {
+ int code = err_get_code(status);
+ NSLog(@"Unmount error code %d", code);
+ }
+ } else {
+ DADiskEject(disk, kDADiskEjectOptionDefault, eject_callback, context);
+ }
+}
+
+static void eject_callback(DADiskRef disk,
+ DADissenterRef dissenter,
+ void* context) {
+ NSError* error = nil;
+ if (dissenter) {
+ DAReturn status = DADissenterGetStatus(dissenter);
+ if (unix_err(status)) {
+ int code = err_get_code(status);
+ NSLog(@"Eject error code %d", code);
+ error = [NSError
+ errorWithDomain:@"ChromeErrorDomain"
+ code:code
+ userInfo:@{
+ NSLocalizedDescriptionKey :
+ (__bridge NSString*)DADissenterGetStatusString(dissenter)
+ }];
+ }
+ } else {
+ Unpacker* unpacker = (__bridge Unpacker*)context;
+ [unpacker didFinishEjectingDisk:disk WithError:error];
+ }
+}
+
+// TODO: presumably we'd pass in the authenticated/unauthenticated path in here;
+// however we will change this to a different model that will allow for a
+// separate authenticated process to run
+- (id)initWithFinalAppPath:(NSString*)appPath {
+ if ((self = [super init])) {
+ appPath_ = appPath;
+ }
+ return self;
+}
+
+// TODO: the failure delegate methods need to be revised to be more meaningfully
+// deal with the errors (pipe in stderr / stdout)
+- (void)mountDMGFromURL:(NSURL*)fileURL {
+ NSError* error = nil;
+ temporaryDirectoryURL = [[NSFileManager defaultManager]
+ URLForDirectory:NSItemReplacementDirectory
+ inDomain:NSUserDomainMask
+ appropriateForURL:[NSURL fileURLWithPath:@"/" isDirectory:YES]
+ create:YES
+ error:&error];
+ if (error) {
+ [delegate_ unpacker:self onMountFailure:nil];
+ }
+
+ temporaryDiskImageURL =
+ [temporaryDirectoryURL URLByAppendingPathComponent:@"GoogleChrome.dmg"];
+ mountPath = [[temporaryDirectoryURL URLByAppendingPathComponent:@"mnt"
+ isDirectory:YES] path];
+ [[NSFileManager defaultManager] createDirectoryAtPath:mountPath
+ withIntermediateDirectories:YES
+ attributes:nil
+ error:&error];
+ if (error) {
+ [delegate_ unpacker:self onMountFailure:nil];
+ }
+
+ [[NSFileManager defaultManager] moveItemAtURL:fileURL
+ toURL:temporaryDiskImageURL
+ error:nil];
+
+ dispatch_semaphore_t mount_semaphore = dispatch_semaphore_create(0);
+ NSString* path = @"/usr/bin/hdiutil";
+ NSArray* args = @[
+ @"mount", temporaryDiskImageURL, @"-nobrowse", @"-mountpoint", mountPath
+ ];
+
+ NSTask* mountTask = [[NSTask alloc] init];
+ mountTask.launchPath = path;
+ mountTask.arguments = args;
+ mountTask.terminationHandler = ^void(NSTask* task) {
+ // unpack_dq = dispatch_queue_create("com.google.chrome.unpack",
Elly Fong-Jones 2016/08/16 15:26:18 does this commented code need to be here?
Anna Zeng 2016/08/16 23:07:40 This code was not completely deleted in case I nee
+ // DISPATCH_QUEUE_SERIAL);
+ // dispatch_async(unpack_dq, ^{
+ dispatch_semaphore_signal(mount_semaphore);
+ // });
+ };
+ [mountTask launch];
+
+ dispatch_semaphore_wait(mount_semaphore, DISPATCH_TIME_FOREVER);
Elly Fong-Jones 2016/08/16 15:26:18 is this running on a work queue or somesuch? other
Anna Zeng 2016/08/16 23:07:40 Currently, it is not. Would you recommend running
+ [delegate_ unpacker:self onMountSuccess:mountPath];
+}
+
+- (void)extractChrome {
+ NSString* diskAppPath = [[temporaryDirectoryURL
+ URLByAppendingPathComponent:@"mnt/Google Chrome.app"
+ isDirectory:YES] path];
+ if (![[NSFileManager defaultManager] fileExistsAtPath:diskAppPath]) {
+ NSLog(@"File in DMG doesn't exist");
+ [delegate_ unpacker:self onExtractFailure:appPath_];
+ }
+
+ // TODO: add progress
+ NSError* err;
+ if (![[NSFileManager defaultManager] copyItemAtPath:diskAppPath
+ toPath:appPath_
+ error:&err]) {
+ NSLog(@"%@", err);
+ [delegate_ unpacker:self onExtractFailure:appPath_];
+ }
+ [delegate_ unpacker:self onExtractSuccess:appPath_];
+}
+
+- (void)unmountDMG {
+ session = DASessionCreate(nil);
+ unpack_dq =
+ dispatch_queue_create("com.google.chrome.unpack", DISPATCH_QUEUE_SERIAL);
+ DASessionSetDispatchQueue(session, unpack_dq);
+ DADiskRef child_disk = DADiskCreateFromVolumePath(
+ nil, session, (CFURLRef)[NSURL URLWithString:mountPath]);
+ DADiskRef whole_disk = DADiskCopyWholeDisk(child_disk);
+
+ DADiskUnmount(whole_disk, kDADiskUnmountOptionWhole, unmount_callback,
Sidney San Martín 2016/08/13 12:37:18 I might also pass `kDADiskUnmountOptionForce`. If
Anna Zeng 2016/08/16 23:07:40 Done.
+ (void*)self);
+
+ CFRelease(whole_disk);
+ CFRelease(child_disk);
+}
+
+- (void)didFinishEjectingDisk:(DADiskRef)disk WithError:(NSError*)error {
+ DASessionSetDispatchQueue(session, NULL);
+ dispatch_release(unpack_dq);
+ if (error) {
+ [delegate_ unpacker:self onUnmountFailure:error];
+ } else {
+ [delegate_ unpacker:self onUnmountSuccess:nil];
+ }
+}
+
+@end

Powered by Google App Engine
This is Rietveld 408576698