browser/components/attribution/nsMacAttribution.cpp
author L10n Bumper Bot <release+l10nbumper@mozilla.com>
Mon, 14 Jan 2019 05:00:14 -0800
changeset 506681 ac40678709f6f3af82e30819cea10260b6e497fc
parent 505383 6f3709b3878117466168c40affa7bca0b60cf75b
permissions -rw-r--r--
no bug - Bumping Fennec l10n changesets r=release a=l10n-bump DONTBUILD en-CA -> 1df3b5370c30 es-CL -> ff997f81eae6 es-MX -> 7d161fdf265c eu -> 506f20acd46d hi-IN -> 504d2fde3fc7 hu -> 6477fd6ddf64 mai -> d524e539eefb nn-NO -> 656adee29da6 sq -> f7c2ccfb7710 ta -> 047b3e205cca ur -> 03708a03f1b8 zh-CN -> b9c275f71f13 zh-TW -> f2b2ea4cd960

/* -*- 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;
}