/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ // vim:set ts=2 sts=2 sw=2 et cin: /* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ #include "CocoaFileUtils.h" #include "nsCocoaFeatures.h" #include "nsCocoaUtils.h" #include <Cocoa/Cocoa.h> #include "nsObjCExceptions.h" #include "nsDebug.h" // Need to cope with us using old versions of the SDK and needing this on 10.10+ #if !defined(MAC_OS_X_VERSION_10_10) || (MAC_OS_X_VERSION_MAX_ALLOWED < MAC_OS_X_VERSION_10_10) const CFStringRef kCFURLQuarantinePropertiesKey = CFSTR("NSURLQuarantinePropertiesKey"); #endif namespace CocoaFileUtils { nsresult RevealFileInFinder(CFURLRef url) { NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT; if (NS_WARN_IF(!url)) return NS_ERROR_INVALID_ARG; NSAutoreleasePool* ap = [[NSAutoreleasePool alloc] init]; BOOL success = [[NSWorkspace sharedWorkspace] selectFile:[(NSURL*)url path] inFileViewerRootedAtPath:@""]; [ap release]; return (success ? NS_OK : NS_ERROR_FAILURE); NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT; } nsresult OpenURL(CFURLRef url) { NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT; if (NS_WARN_IF(!url)) return NS_ERROR_INVALID_ARG; NSAutoreleasePool* ap = [[NSAutoreleasePool alloc] init]; BOOL success = [[NSWorkspace sharedWorkspace] openURL:(NSURL*)url]; [ap release]; return (success ? NS_OK : NS_ERROR_FAILURE); NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT; } nsresult GetFileCreatorCode(CFURLRef url, OSType *creatorCode) { NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT; if (NS_WARN_IF(!url) || NS_WARN_IF(!creatorCode)) return NS_ERROR_INVALID_ARG; nsAutoreleasePool localPool; NSString *resolvedPath = [[(NSURL*)url path] stringByResolvingSymlinksInPath]; if (!resolvedPath) { return NS_ERROR_FAILURE; } NSDictionary* dict = [[NSFileManager defaultManager] attributesOfItemAtPath:resolvedPath error:nil]; if (!dict) { return NS_ERROR_FAILURE; } NSNumber* creatorNum = (NSNumber*)[dict objectForKey:NSFileHFSCreatorCode]; if (!creatorNum) { return NS_ERROR_FAILURE; } *creatorCode = [creatorNum unsignedLongValue]; return NS_OK; NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT; } nsresult SetFileCreatorCode(CFURLRef url, OSType creatorCode) { NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT; if (NS_WARN_IF(!url)) return NS_ERROR_INVALID_ARG; NSAutoreleasePool* ap = [[NSAutoreleasePool alloc] init]; NSDictionary* dict = [NSDictionary dictionaryWithObject:[NSNumber numberWithUnsignedLong:creatorCode] forKey:NSFileHFSCreatorCode]; BOOL success = [[NSFileManager defaultManager] setAttributes:dict ofItemAtPath:[(NSURL*)url path] error:nil]; [ap release]; return (success ? NS_OK : NS_ERROR_FAILURE); NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT; } nsresult GetFileTypeCode(CFURLRef url, OSType *typeCode) { NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT; if (NS_WARN_IF(!url) || NS_WARN_IF(!typeCode)) return NS_ERROR_INVALID_ARG; nsAutoreleasePool localPool; NSString *resolvedPath = [[(NSURL*)url path] stringByResolvingSymlinksInPath]; if (!resolvedPath) { return NS_ERROR_FAILURE; } NSDictionary* dict = [[NSFileManager defaultManager] attributesOfItemAtPath:resolvedPath error:nil]; if (!dict) { return NS_ERROR_FAILURE; } NSNumber* typeNum = (NSNumber*)[dict objectForKey:NSFileHFSTypeCode]; if (!typeNum) { return NS_ERROR_FAILURE; } *typeCode = [typeNum unsignedLongValue]; return NS_OK; NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT; } nsresult SetFileTypeCode(CFURLRef url, OSType typeCode) { NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT; if (NS_WARN_IF(!url)) return NS_ERROR_INVALID_ARG; NSAutoreleasePool* ap = [[NSAutoreleasePool alloc] init]; NSDictionary* dict = [NSDictionary dictionaryWithObject:[NSNumber numberWithUnsignedLong:typeCode] forKey:NSFileHFSTypeCode]; BOOL success = [[NSFileManager defaultManager] setAttributes:dict ofItemAtPath:[(NSURL*)url path] error:nil]; [ap release]; return (success ? NS_OK : NS_ERROR_FAILURE); NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT; } void AddOriginMetadataToFile(const CFStringRef filePath, const CFURLRef sourceURL, const CFURLRef referrerURL) { typedef OSStatus (*MDItemSetAttribute_type)(MDItemRef, CFStringRef, CFTypeRef); static MDItemSetAttribute_type mdItemSetAttributeFunc = NULL; static bool did_symbol_lookup = false; if (!did_symbol_lookup) { did_symbol_lookup = true; CFBundleRef metadata_bundle = ::CFBundleGetBundleWithIdentifier(CFSTR("com.apple.Metadata")); if (!metadata_bundle) { return; } mdItemSetAttributeFunc = (MDItemSetAttribute_type) ::CFBundleGetFunctionPointerForName(metadata_bundle, CFSTR("MDItemSetAttribute")); } if (!mdItemSetAttributeFunc) { return; } MDItemRef mdItem = ::MDItemCreate(NULL, filePath); if (!mdItem) { return; } CFMutableArrayRef list = ::CFArrayCreateMutable(kCFAllocatorDefault, 2, NULL); if (!list) { ::CFRelease(mdItem); return; } // The first item in the list is the source URL of the downloaded file. if (sourceURL) { ::CFArrayAppendValue(list, ::CFURLGetString(sourceURL)); } // If the referrer is known, store that in the second position. if (referrerURL) { ::CFArrayAppendValue(list, ::CFURLGetString(referrerURL)); } mdItemSetAttributeFunc(mdItem, kMDItemWhereFroms, list); ::CFRelease(list); ::CFRelease(mdItem); } void AddQuarantineMetadataToFile(const CFStringRef filePath, const CFURLRef sourceURL, const CFURLRef referrerURL, const bool isFromWeb) { CFURLRef fileURL = ::CFURLCreateWithFileSystemPath(kCFAllocatorDefault, filePath, kCFURLPOSIXPathStyle, false); // The properties key changed in 10.10: CFStringRef quarantinePropKey; if (nsCocoaFeatures::OnYosemiteOrLater()) { quarantinePropKey = kCFURLQuarantinePropertiesKey; } else { quarantinePropKey = kLSItemQuarantineProperties; } CFDictionaryRef quarantineProps = NULL; Boolean success = ::CFURLCopyResourcePropertyForKey(fileURL, quarantinePropKey, &quarantineProps, NULL); // If there aren't any quarantine properties then the user probably // set up an exclusion and we don't need to add metadata. if (!success || !quarantineProps) { ::CFRelease(fileURL); return; } // We don't know what to do if the props aren't a dictionary. if (::CFGetTypeID(quarantineProps) != ::CFDictionaryGetTypeID()) { ::CFRelease(fileURL); ::CFRelease(quarantineProps); return; } // Make a mutable copy of the properties. CFMutableDictionaryRef mutQuarantineProps = ::CFDictionaryCreateMutableCopy(kCFAllocatorDefault, 0, (CFDictionaryRef)quarantineProps); ::CFRelease(quarantineProps); // Add metadata that the OS couldn't infer. if (!::CFDictionaryGetValue(mutQuarantineProps, kLSQuarantineTypeKey)) { CFStringRef type = isFromWeb ? kLSQuarantineTypeWebDownload : kLSQuarantineTypeOtherDownload; ::CFDictionarySetValue(mutQuarantineProps, kLSQuarantineTypeKey, type); } if (!::CFDictionaryGetValue(mutQuarantineProps, kLSQuarantineOriginURLKey) && referrerURL) { ::CFDictionarySetValue(mutQuarantineProps, kLSQuarantineOriginURLKey, referrerURL); } if (!::CFDictionaryGetValue(mutQuarantineProps, kLSQuarantineDataURLKey) && sourceURL) { ::CFDictionarySetValue(mutQuarantineProps, kLSQuarantineDataURLKey, sourceURL); } // Set quarantine properties on file. ::CFURLSetResourcePropertyForKey(fileURL, quarantinePropKey, mutQuarantineProps, NULL); ::CFRelease(fileURL); ::CFRelease(mutQuarantineProps); } CFURLRef GetTemporaryFolderCFURLRef() { NSString* tempDir = ::NSTemporaryDirectory(); return tempDir == nil ? NULL : (CFURLRef)[NSURL fileURLWithPath:tempDir isDirectory:YES]; } } // namespace CocoaFileUtils