Source code
Revision control
Copy as Markdown
Other Tools
/* 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
#include "nsUrlClassifierTestUtils.h"
#include "chromium/safebrowsing_v5.pb.h"
#include "mozilla/ClearOnShutdown.h"
#include "mozilla/StaticPtr.h"
#include "Entries.h"
#include "mozilla/EndianUtils.h"
#include "nsString.h"
NS_IMPL_ISUPPORTS(nsUrlClassifierTestUtils, nsIUrlClassifierTestUtils)
using namespace mozilla;
using namespace mozilla::safebrowsing;
static mozilla::StaticRefPtr<nsUrlClassifierTestUtils> gUrlClassifierTestUtils;
// static
already_AddRefed<nsUrlClassifierTestUtils>
nsUrlClassifierTestUtils::GetXPCOMSingleton() {
if (gUrlClassifierTestUtils) {
return do_AddRef(gUrlClassifierTestUtils);
}
RefPtr<nsUrlClassifierTestUtils> utils = new nsUrlClassifierTestUtils();
// Note: This is cleared in the nsUrlClassifierUtils destructor.
gUrlClassifierTestUtils = utils.get();
ClearOnShutdown(&gUrlClassifierTestUtils);
return utils.forget();
}
NS_IMETHODIMP
nsUrlClassifierTestUtils::MakeUpdateResponseV5(const nsACString& aName,
uint32_t aSingleHash,
nsACString& aResponse) {
if (NS_WARN_IF(aName.IsEmpty())) {
return NS_ERROR_INVALID_ARG;
}
v5::BatchGetHashListsResponse response;
v5::HashList* hashList = response.add_hash_lists();
hashList->set_name("test-4b");
hashList->set_partial_update(false);
hashList->set_version("\x00\x00\x00\x01");
v5::RiceDeltaEncoded32Bit* riceDelta =
hashList->mutable_additions_four_bytes();
uint32_t prefix = BigEndian::readUint32(&aSingleHash);
riceDelta->set_first_value(prefix);
riceDelta->set_rice_parameter(30);
riceDelta->set_entries_count(0);
std::string s;
response.SerializeToString(&s);
nsCString out(s.c_str(), s.size());
aResponse = out;
return NS_OK;
}
NS_IMETHODIMP
nsUrlClassifierTestUtils::MakeFindFullHashResponseV5(
const nsACString& aFullHash, nsACString& aResponse) {
if (NS_WARN_IF(aFullHash.IsEmpty())) {
return NS_ERROR_INVALID_ARG;
}
v5::SearchHashesResponse response;
nsAutoCString fullHashData(aFullHash);
v5::FullHash* fullHash = response.add_full_hashes();
fullHash->set_full_hash(fullHashData.get(), fullHashData.Length());
v5::Duration* duration = response.mutable_cache_duration();
duration->set_seconds(1800);
duration->set_nanos(0);
std::string s;
response.SerializeToString(&s);
nsCString out(s.c_str(), s.size());
aResponse = out;
return NS_OK;
}
NS_IMETHODIMP
nsUrlClassifierTestUtils::GenerateLookupHash(const nsACString& aFragment,
uint32_t* aLookupHash) {
if (NS_WARN_IF(aFragment.IsEmpty())) {
return NS_ERROR_INVALID_ARG;
}
Completion lookupHash;
lookupHash.FromPlaintext(aFragment);
*aLookupHash = lookupHash.ToUint32();
return NS_OK;
}
NS_IMETHODIMP
nsUrlClassifierTestUtils::GenerateFullHash(const nsACString& aFragment,
nsACString& aFullHash) {
if (NS_WARN_IF(aFragment.IsEmpty())) {
return NS_ERROR_INVALID_ARG;
}
Completion lookupHash;
lookupHash.FromPlaintext(aFragment);
lookupHash.ToString(aFullHash);
return NS_OK;
}