image/ImageURL.h
author Jean-Yves Avenard <jyavenard@mozilla.com>
Thu, 13 Aug 2015 08:25:23 +1000
changeset 289089 d710215ba3d5f56c1e12d9725be2304ee2901dbc
parent 278118 4f873b18b4bfa16f7678316719f82461440f8880
child 354664 d008348cd4f450faf68fdeba7ca2d8146ddce83a
permissions -rw-r--r--
Bug 1188871: P2. Call DrainComplete should an error occurs while draining. r=snorp a=ritu Part 1 prevented draining when an error occurred first. This handles the case where an error occurs during draining.

/* -*- 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/. */

#ifndef mozilla_image_ImageURL_h
#define mozilla_image_ImageURL_h

#include "nsIURI.h"
#include "MainThreadUtils.h"
#include "nsNetUtil.h"

namespace mozilla {
namespace image {

/** ImageURL
 *
 * nsStandardURL is not threadsafe, so this class is created to hold only the
 * necessary URL data required for image loading and decoding.
 *
 * Note: Although several APIs have the same or similar prototypes as those
 * found in nsIURI/nsStandardURL, the class does not implement nsIURI. This is
 * intentional; functionality is limited, and is only useful for imagelib code.
 * By not implementing nsIURI, external code cannot unintentionally be given an
 * nsIURI pointer with this limited class behind it; instead, conversion to a
 * fully implemented nsIURI is required (e.g. through NS_NewURI).
 */
class ImageURL
{
public:
  explicit ImageURL(nsIURI* aURI)
  {
    MOZ_ASSERT(NS_IsMainThread(), "Cannot use nsIURI off main thread!");
    aURI->GetSpec(mSpec);
    aURI->GetScheme(mScheme);
    aURI->GetRef(mRef);
  }

  NS_INLINE_DECL_THREADSAFE_REFCOUNTING(ImageURL)

  nsresult GetSpec(nsACString& result)
  {
    result = mSpec;
    return NS_OK;
  }

  /// A weak pointer to the URI spec for this ImageURL. For logging only.
  const char* Spec() const { return mSpec.get(); }

  enum TruncatedSpecStatus {
    FitsInto1k,
    TruncatedTo1k
  };
  TruncatedSpecStatus GetSpecTruncatedTo1k(nsACString& result)
  {
    static const size_t sMaxTruncatedLength = 1024;

    if (sMaxTruncatedLength >= mSpec.Length()) {
      result = mSpec;
      return FitsInto1k;
    }

    result = Substring(mSpec, 0, sMaxTruncatedLength);
    return TruncatedTo1k;
  }

  nsresult GetScheme(nsACString& result)
  {
    result = mScheme;
    return NS_OK;
  }

  nsresult SchemeIs(const char* scheme, bool* result)
  {
    NS_PRECONDITION(scheme, "scheme is null");
    NS_PRECONDITION(result, "result is null");

    *result = mScheme.Equals(scheme);
    return NS_OK;
  }

  nsresult GetRef(nsACString& result)
  {
    result = mRef;
    return NS_OK;
  }

  already_AddRefed<nsIURI> ToIURI()
  {
    MOZ_ASSERT(NS_IsMainThread(),
               "Convert to nsIURI on main thread only; it is not threadsafe.");
    nsCOMPtr<nsIURI> newURI;
    NS_NewURI(getter_AddRefs(newURI), mSpec);
    return newURI.forget();
  }

  bool operator==(const ImageURL& aOther) const
  {
    // Note that we don't need to consider mScheme and mRef, because they're
    // already represented in mSpec.
    return mSpec == aOther.mSpec;
  }

  bool HasSameRef(const ImageURL& aOther) const
  {
    return mRef == aOther.mRef;
  }

private:
  // Since this is a basic storage class, no duplication of spec parsing is
  // included in the functionality. Instead, the class depends upon the
  // parsing implementation in the nsIURI class used in object construction.
  // This means each field is stored separately, but since only a few are
  // required, this small memory tradeoff for threadsafe usage should be ok.
  nsAutoCString mSpec;
  nsAutoCString mScheme;
  nsAutoCString mRef;

  ~ImageURL() { }
};

} // namespace image
} // namespace mozilla

#endif // mozilla_image_ImageURL_h