Chromium Code Reviews| 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..1177da6902f13a6c44020224f62e987c54a15872 |
| --- /dev/null |
| +++ b/chrome/installer/mac/app/Unpacker.m |
| @@ -0,0 +1,122 @@ |
| +// 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 <copyfile.h> |
| +#include <DiskArbitration/DiskArbitration.h> |
| +#include <dispatch/dispatch.h> |
| + |
| +#import "Downloader.h" |
| + |
| +dispatch_semaphore_t mount_semaphore; |
| + |
| +@implementation Unpacker |
| + |
| +@synthesize delegate = delegate_; |
| + |
| +DASessionRef session; |
| +dispatch_queue_t unpack_dq; |
|
Sidney San Martín
2016/08/08 18:43:17
These should be instance variables, not globals (l
Anna Zeng
2016/08/12 22:56:18
Done.
|
| + |
| +static void unmount_callback(DADiskRef disk, |
| + DADissenterRef dissenter, |
| + void* context) { |
| + if (dissenter) { |
| + DAReturn status = DADissenterGetStatus(dissenter); |
| + if (unix_err(status)) { |
| + int code = err_get_code(status); |
| + NSLog(@"Unmount error code %d", code); |
| + // TODO: error handling |
| + } |
| + } else { |
| + DADiskEject(disk, kDADiskEjectOptionDefault, eject_callback, context); |
| + } |
| +} |
| + |
| +static void eject_callback(DADiskRef disk, |
| + DADissenterRef dissenter, |
| + void* context) { |
| + NSError* error; |
| + if (dissenter) { |
| + DAReturn status = DADissenterGetStatus(dissenter); |
| + if (unix_err(status)) { |
| + int code = err_get_code(status); |
| + NSLog(@"Eject error code %d", code); |
| + // TODO: error handling |
| + error = [NSError errorWithDomain:@"ChromeErrorDomain" |
| + code:code |
| + userInfo:@{ |
| + NSLocalizedDescriptionKey : |
| + DADissenterGetStatusString(dissenter) |
| + }]; |
| + } |
| + } |
| + Unpacker* unpacker = (__bridge Unpacker*)context; |
| + [unpacker didFinishEjectingDisk:disk WithError:error]; |
| +} |
| + |
| +- (void)mountDMG { |
| + // TODO: how to find this from the PATH env variable? |
|
Sidney San Martín
2016/08/08 18:43:17
TBH, since this is a system utility I think that h
Anna Zeng
2016/08/12 22:56:18
Done.
|
| + NSString* path = @"/usr/bin/hdiutil"; |
| + NSArray* args = @[ |
| + // TODO: break getDownloadsFilePath out of the downloader file |
| + @"mount", [Downloader getDownloadsFilePath], @"-nobrowse", @"-mountpoint", |
| + PATH_FROM_EXECUTABLE(@"tmp") |
| + ]; |
| + |
| + 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", |
| + DISPATCH_QUEUE_SERIAL); |
| + dispatch_async(unpack_dq, ^{ |
| + [self extractChrome]; |
| + }); |
| + }; |
| + [mountTask launch]; |
| + // TODO: how to error handling |
| +} |
| + |
| +- (void)extractChrome { |
| + NSFileManager* fileManager = [NSFileManager defaultManager]; |
| + if (![fileManager |
| + fileExistsAtPath:PATH_FROM_EXECUTABLE(@"tmp/Google Chrome.app")]) { |
| + NSLog(@"File in DMG doesn't exist"); |
| + // TODO: error handling |
| + } |
| + |
| + // TODO: add progress |
| + NSError* err; |
| + if (![fileManager |
| + copyItemAtPath:PATH_FROM_EXECUTABLE(@"tmp/Google Chrome.app") |
| + toPath:@"/Applications/Google Chromo.app" |
|
Sidney San Martín
2016/08/08 18:43:18
This path should come from outside (property, argu
Anna Zeng
2016/08/12 22:56:18
Done.
|
| + error:&err]) { |
| + NSLog(@"%@", err); |
| + } |
| + dispatch_semaphore_signal(mount_semaphore); |
| +} |
| + |
| +- (void)unmountDMG { |
| + session = DASessionCreate(nil); |
| + DASessionSetDispatchQueue(session, unpack_dq); |
| + DADiskRef child_disk = DADiskCreateFromVolumePath( |
| + nil, session, |
| + (CFURLRef)[NSURL URLWithString:PATH_FROM_EXECUTABLE(@"tmp")]); |
| + DADiskRef whole_disk = DADiskCopyWholeDisk(child_disk); |
| + |
| + DADiskUnmount(whole_disk, kDADiskUnmountOptionWhole, unmount_callback, |
| + (void*)self); |
| + |
| + CFRelease(whole_disk); |
| + CFRelease(child_disk); |
| +} |
| + |
| +- (void)didFinishEjectingDisk:(DADiskRef)disk WithError:(NSError*)error { |
| + DASessionSetDispatchQueue(session, NULL); |
| + dispatch_release(unpack_dq); |
| + [delegate_ onUnpackSuccess]; |
| +} |
| + |
| +@end |