forked from mirrors/gecko-dev
e73ed82b2f
1. VariableLengthPrefixSet supports getting/setting prefixes with AddPrefixArray and AddCompletesArray 2. VariableLengthPrefixSet supports passing prefix as an integer in Match API. This is because how V2 and V4 see prefixes as an integer works differently. Differential Revision: https://phabricator.services.mozilla.com/D34547 --HG-- extra : moz-landing-system : lando
70 lines
2.3 KiB
C++
70 lines
2.3 KiB
C++
//* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=2 et sw=2 tw=80: */
|
|
/* 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 VariableLengthPrefixSet_h
|
|
#define VariableLengthPrefixSet_h
|
|
|
|
#include "nsISupports.h"
|
|
#include "nsIMemoryReporter.h"
|
|
#include "Entries.h"
|
|
#include "nsTArray.h"
|
|
#include "mozilla/MemoryReporting.h"
|
|
#include "mozilla/Mutex.h"
|
|
|
|
class nsUrlClassifierPrefixSet;
|
|
|
|
namespace mozilla {
|
|
namespace safebrowsing {
|
|
|
|
class VariableLengthPrefixSet final : public nsIMemoryReporter {
|
|
public:
|
|
VariableLengthPrefixSet();
|
|
|
|
nsresult Init(const nsACString& aName);
|
|
nsresult SetPrefixes(mozilla::safebrowsing::PrefixStringMap& aPrefixMap);
|
|
nsresult SetPrefixes(AddPrefixArray& aAddPrefixes,
|
|
AddCompleteArray& aAddCompletes);
|
|
nsresult GetPrefixes(mozilla::safebrowsing::PrefixStringMap& aPrefixMap);
|
|
nsresult GetFixedLengthPrefixes(FallibleTArray<uint32_t>* aPrefixes,
|
|
FallibleTArray<nsCString>* aCompletes);
|
|
nsresult Matches(uint32_t aPrefix, const nsACString& aFullHash,
|
|
uint32_t* aLength) const;
|
|
nsresult IsEmpty(bool* aEmpty) const;
|
|
|
|
nsresult WritePrefixes(nsCOMPtr<nsIOutputStream>& out) const;
|
|
nsresult LoadPrefixes(nsCOMPtr<nsIInputStream>& in);
|
|
uint32_t CalculatePreallocateSize() const;
|
|
|
|
size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const;
|
|
|
|
NS_DECL_THREADSAFE_ISUPPORTS
|
|
NS_DECL_NSIMEMORYREPORTER
|
|
|
|
private:
|
|
virtual ~VariableLengthPrefixSet();
|
|
|
|
static const uint32_t PREFIXSET_VERSION_MAGIC = 1;
|
|
|
|
bool BinarySearch(const nsACString& aFullHash, const nsACString& aPrefixes,
|
|
uint32_t aPrefixSize) const;
|
|
|
|
// Lock to prevent races between the url-classifier thread (which does most
|
|
// of the operations) and the main thread (which does memory reporting).
|
|
// It should be held for all operations between Init() and destruction that
|
|
// touch this class's data members.
|
|
mutable mozilla::Mutex mLock;
|
|
|
|
const RefPtr<nsUrlClassifierPrefixSet> mFixedPrefixSet;
|
|
mozilla::safebrowsing::PrefixStringMap mVLPrefixSet;
|
|
|
|
nsCString mName;
|
|
nsCString mMemoryReportPath;
|
|
};
|
|
|
|
} // namespace safebrowsing
|
|
} // namespace mozilla
|
|
|
|
#endif
|