forked from mirrors/gecko-dev
7962eacafd
MozReview-Commit-ID: NgjE1HZS7M --HG-- rename : browser/modules/AttributionCode.jsm => browser/components/attribution/AttributionCode.jsm rename : browser/modules/test/unit/test_AttributionCode.js => browser/components/attribution/test/xpcshell/test_AttributionCode.js extra : rebase_source : 197f427e0d2fcbdbe85d5ebe65067dcc9eee6407
70 lines
2.5 KiB
C++
70 lines
2.5 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* 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 "nsMacAttribution.h"
|
|
|
|
#include <CoreFoundation/CoreFoundation.h>
|
|
#include <ApplicationServices/ApplicationServices.h>
|
|
|
|
#include "../../../xpcom/io/CocoaFileUtils.h"
|
|
#include "nsCocoaFeatures.h"
|
|
#include "nsString.h"
|
|
|
|
using namespace mozilla;
|
|
|
|
NS_IMPL_ISUPPORTS(nsMacAttributionService, nsIMacAttributionService)
|
|
|
|
NS_IMETHODIMP
|
|
nsMacAttributionService::GetReferrerUrl(const nsACString& aFilePath,
|
|
nsAString& aReferrer)
|
|
{
|
|
const nsCString& flat = PromiseFlatCString(aFilePath);
|
|
CFStringRef filePath = ::CFStringCreateWithCString(kCFAllocatorDefault,
|
|
flat.get(),
|
|
kCFStringEncodingUTF8);
|
|
|
|
CocoaFileUtils::CopyQuarantineReferrerUrl(filePath, aReferrer);
|
|
|
|
::CFRelease(filePath);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsMacAttributionService::SetReferrerUrl(const nsACString& aFilePath,
|
|
const nsACString& aReferrerUrl,
|
|
const bool aCreate)
|
|
{
|
|
const nsCString& flat = PromiseFlatCString(aFilePath);
|
|
CFStringRef filePath = ::CFStringCreateWithCString(kCFAllocatorDefault,
|
|
flat.get(),
|
|
kCFStringEncodingUTF8);
|
|
|
|
if (!filePath) {
|
|
return NS_ERROR_UNEXPECTED;
|
|
}
|
|
|
|
const nsCString& flatReferrer = PromiseFlatCString(aReferrerUrl);
|
|
CFStringRef referrer = ::CFStringCreateWithCString(kCFAllocatorDefault,
|
|
flatReferrer.get(),
|
|
kCFStringEncodingUTF8);
|
|
if (!referrer) {
|
|
::CFRelease(filePath);
|
|
return NS_ERROR_UNEXPECTED;
|
|
}
|
|
CFURLRef referrerURL = ::CFURLCreateWithString(kCFAllocatorDefault,
|
|
referrer, nullptr);
|
|
|
|
CocoaFileUtils::AddQuarantineMetadataToFile(filePath,
|
|
NULL,
|
|
referrerURL,
|
|
true,
|
|
aCreate);
|
|
|
|
::CFRelease(filePath);
|
|
::CFRelease(referrer);
|
|
::CFRelease(referrerURL);
|
|
|
|
return NS_OK;
|
|
}
|