1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/* -*- Mode: C++; tab-width: 8; 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/. */
#include "HashStore.h"
#include "LookupCache.h"
#include "LookupCacheV4.h"
#include "nsAppDirectoryServiceDefs.h"
#include "Common.h"
namespace mozilla {
namespace safebrowsing {
class PerProviderDirectoryTestUtils {
public:
template <typename T>
static nsIFile* InspectStoreDirectory(const T& aT) {
return aT.mStoreDirectory;
}
};
} // end of namespace safebrowsing
} // end of namespace mozilla
template <typename T>
static void VerifyPrivateStorePath(T* target, const nsCString& aTableName,
const nsCString& aProvider,
const nsCOMPtr<nsIFile>& aRootDir,
bool aUsePerProviderStore) {
nsString rootStorePath;
nsresult rv = aRootDir->GetPath(rootStorePath);
EXPECT_EQ(rv, NS_OK);
nsIFile* privateStoreDirectory =
PerProviderDirectoryTestUtils::InspectStoreDirectory(*target);
nsString privateStorePath;
rv = privateStoreDirectory->GetPath(privateStorePath);
ASSERT_EQ(rv, NS_OK);
nsString expectedPrivateStorePath = rootStorePath;
if (aUsePerProviderStore) {
// Use API to append "provider" to the root directoy path
nsCOMPtr<nsIFile> expectedPrivateStoreDir;
rv = aRootDir->Clone(getter_AddRefs(expectedPrivateStoreDir));
ASSERT_EQ(rv, NS_OK);
expectedPrivateStoreDir->AppendNative(aProvider);
rv = expectedPrivateStoreDir->GetPath(expectedPrivateStorePath);
ASSERT_EQ(rv, NS_OK);
}
printf("table: %s\nprovider: %s\nroot path: %s\nprivate path: %s\n\n",
aTableName.get(), aProvider.get(),
NS_ConvertUTF16toUTF8(rootStorePath).get(),
NS_ConvertUTF16toUTF8(privateStorePath).get());
ASSERT_TRUE(privateStorePath == expectedPrivateStorePath);
}
TEST(UrlClassifierPerProviderDirectory, LookupCache)
{
nsCOMPtr<nsIFile> rootDir;
NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(rootDir));
RunTestInNewThread([&]() -> void {
// For V2 tables (NOT ending with '-proto'), root directory should be
// used as the private store.
{
nsAutoCString table("goog-phish-shavar");
nsAutoCString provider("google");
RefPtr<LookupCacheV2> lc = new LookupCacheV2(table, provider, rootDir);
VerifyPrivateStorePath<LookupCacheV2>(lc, table, provider, rootDir,
false);
}
// For V4 tables, if provider is found, use per-provider subdirectory;
// If not found, use root directory.
{
nsAutoCString table("goog-noprovider-proto");
nsAutoCString provider("");
RefPtr<LookupCacheV4> lc = new LookupCacheV4(table, provider, rootDir);
VerifyPrivateStorePath<LookupCacheV4>(lc, table, provider, rootDir,
false);
}
{
nsAutoCString table("goog-phish-proto");
nsAutoCString provider("google4");
RefPtr<LookupCacheV4> lc = new LookupCacheV4(table, provider, rootDir);
VerifyPrivateStorePath<LookupCacheV4>(lc, table, provider, rootDir, true);
}
});
}
TEST(UrlClassifierPerProviderDirectory, HashStore)
{
nsCOMPtr<nsIFile> rootDir;
NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(rootDir));
RunTestInNewThread([&]() -> void {
// For V2 tables (NOT ending with '-proto'), root directory should be
// used as the private store.
{
nsAutoCString table("goog-phish-shavar");
nsAutoCString provider("google");
HashStore hs(table, provider, rootDir);
VerifyPrivateStorePath(&hs, table, provider, rootDir, false);
}
// For V4 tables, if provider is found, use per-provider subdirectory;
// If not found, use root directory.
{
nsAutoCString table("goog-noprovider-proto");
nsAutoCString provider("");
HashStore hs(table, provider, rootDir);
VerifyPrivateStorePath(&hs, table, provider, rootDir, false);
}
{
nsAutoCString table("goog-phish-proto");
nsAutoCString provider("google4");
HashStore hs(table, provider, rootDir);
VerifyPrivateStorePath(&hs, table, provider, rootDir, true);
}
});
}
|
