gecko-dev/netwerk/base/nsURIHashKey.h
Valentin Gosu ac02cc4197 Bug 548685 - Avoid null pointer deref in nsURIHashKey r=mcmanus
MozReview-Commit-ID: 5wCZ0DTHEUS
2016-04-19 15:46:34 +02:00

59 lines
1.7 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* 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 nsURIHashKey_h__
#define nsURIHashKey_h__
#include "PLDHashTable.h"
#include "nsCOMPtr.h"
#include "nsIURI.h"
#include "nsHashKeys.h"
/**
* Hashtable key class to use with nsTHashtable/nsBaseHashtable
*/
class nsURIHashKey : public PLDHashEntryHdr
{
public:
typedef nsIURI* KeyType;
typedef const nsIURI* KeyTypePointer;
explicit nsURIHashKey(const nsIURI* aKey) :
mKey(const_cast<nsIURI*>(aKey)) { MOZ_COUNT_CTOR(nsURIHashKey); }
nsURIHashKey(const nsURIHashKey& toCopy) :
mKey(toCopy.mKey) { MOZ_COUNT_CTOR(nsURIHashKey); }
~nsURIHashKey() { MOZ_COUNT_DTOR(nsURIHashKey); }
nsIURI* GetKey() const { return mKey; }
bool KeyEquals(const nsIURI* aKey) const {
bool eq;
if (!mKey) {
return !aKey;
}
if (NS_SUCCEEDED(mKey->Equals(const_cast<nsIURI*>(aKey), &eq))) {
return eq;
}
return false;
}
static const nsIURI* KeyToPointer(nsIURI* aKey) { return aKey; }
static PLDHashNumber HashKey(const nsIURI* aKey) {
if (!aKey) {
// If the key is null, return hash for empty string.
return mozilla::HashString(EmptyCString());
}
nsAutoCString spec;
const_cast<nsIURI*>(aKey)->GetSpec(spec);
return mozilla::HashString(spec);
}
enum { ALLOW_MEMMOVE = true };
protected:
nsCOMPtr<nsIURI> mKey;
};
#endif // nsURIHashKey_h__