forked from mirrors/gecko-dev
5aef415cc6
In this patch, we will make Safebrowsing V2 caching use the same algorithm as V4. So we remove "mMissCache" for negative caching and TableFresness check for positive caching. But Safebrowsing V2 doesn't contain negative/positive cache duration information in gethash response. So we hard-code a fixed value, 15 minutes, as cache duration. In this way, we can sync the mechanism we handle caching for V2 and V4. An extra effort for V2 here is that we need to manually record prefixes misses because we won't get any response for those prefixes(implemented in nsUrlClassifierLookupCallback::CacheMisses). --HG-- extra : rebase_source : 0fb7938464ad24a01a51493ecb1b5c0197c16123
49 lines
1.6 KiB
C++
49 lines
1.6 KiB
C++
#include "HashStore.h"
|
|
#include "LookupCacheV4.h"
|
|
#include "nsIFile.h"
|
|
#include "nsTArray.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::safebrowsing;
|
|
|
|
namespace mozilla {
|
|
namespace safebrowsing {
|
|
class Classifier;
|
|
}
|
|
}
|
|
|
|
typedef nsCString _Fragment;
|
|
typedef nsTArray<nsCString> _PrefixArray;
|
|
|
|
template<typename Function>
|
|
void RunTestInNewThread(Function&& aFunction);
|
|
|
|
// Synchronously apply updates by calling Classifier::AsyncApplyUpdates.
|
|
nsresult SyncApplyUpdates(Classifier* aClassifier,
|
|
nsTArray<TableUpdate*>* aUpdates);
|
|
|
|
// Return nsIFile with root directory - NS_APP_USER_PROFILE_50_DIR
|
|
// Sub-directories are passed in path argument.
|
|
already_AddRefed<nsIFile>
|
|
GetFile(const nsTArray<nsString>& path);
|
|
|
|
// ApplyUpdate will call |ApplyUpdates| of Classifier within a new thread
|
|
void ApplyUpdate(nsTArray<TableUpdate*>& updates);
|
|
|
|
void ApplyUpdate(TableUpdate* update);
|
|
|
|
// This function converts lexigraphic-sorted prefixes to a hashtable
|
|
// which key is prefix size and value is concatenated prefix string.
|
|
void PrefixArrayToPrefixStringMap(const nsTArray<nsCString>& prefixArray,
|
|
PrefixStringMap& out);
|
|
|
|
nsresult PrefixArrayToAddPrefixArrayV2(const nsTArray<nsCString>& prefixArray,
|
|
AddPrefixArray& out);
|
|
|
|
// Generate a hash prefix from string
|
|
nsCString GeneratePrefix(const nsCString& aFragment, uint8_t aLength);
|
|
|
|
// Create a LookupCacheV4 object with sepecified prefix array.
|
|
template<typename T>
|
|
UniquePtr<T> SetupLookupCache(const _PrefixArray& prefixArray);
|