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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=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 nsToolkitProfileService_h
#define nsToolkitProfileService_h
#include "mozilla/Components.h"
#include "mozilla/LinkedList.h"
#include "nsIToolkitProfileService.h"
#include "nsIToolkitProfile.h"
#include "nsIFactory.h"
#include "nsSimpleEnumerator.h"
#include "nsProfileLock.h"
#include "nsINIParser.h"
#include "mozilla/MozPromise.h"
#include "nsProxyRelease.h"
class nsStartupLock;
struct CurrentProfileData {
nsCString mPath;
nsCString mStoreID;
bool mShowSelector;
};
struct IniData {
nsCString mProfiles;
nsCString mInstalls;
};
class nsToolkitProfile final
: public nsIToolkitProfile,
public mozilla::LinkedListElement<RefPtr<nsToolkitProfile>> {
public:
NS_DECL_ISUPPORTS
NS_DECL_NSITOOLKITPROFILE
friend class nsToolkitProfileService;
private:
~nsToolkitProfile() = default;
nsToolkitProfile(const nsACString& aName, nsIFile* aRootDir,
nsIFile* aLocalDir, bool aFromDB, const nsACString& aStoreID,
bool aShowProfileSelector);
nsresult RemoveInternal(bool aRemoveFiles, bool aInBackground);
friend class nsToolkitProfileLock;
nsCString mName;
nsCOMPtr<nsIFile> mRootDir;
nsCOMPtr<nsIFile> mLocalDir;
nsCString mStoreID;
bool mShowProfileSelector;
nsIProfileLock* mLock;
uint32_t mIndex;
nsCString mSection;
};
class nsToolkitProfileLock final : public nsIProfileLock {
public:
NS_DECL_ISUPPORTS
NS_DECL_NSIPROFILELOCK
nsresult Init(nsToolkitProfile* aProfile, nsIProfileUnlocker** aUnlocker);
nsresult Init(nsIFile* aDirectory, nsIFile* aLocalDirectory,
nsIProfileUnlocker** aUnlocker);
nsToolkitProfileLock() = default;
private:
~nsToolkitProfileLock();
RefPtr<nsToolkitProfile> mProfile;
nsCOMPtr<nsIFile> mDirectory;
nsCOMPtr<nsIFile> mLocalDirectory;
nsProfileLock mLock;
};
class nsToolkitProfileService final : public nsIToolkitProfileService {
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSITOOLKITPROFILESERVICE
nsresult SelectStartupProfile(int* aArgc, char* aArgv[], bool aIsResetting,
nsIFile** aRootDir, nsIFile** aLocalDir,
nsIToolkitProfile** aProfile, bool* aDidCreate,
bool* aWasDefaultSelection);
nsresult CreateResetProfile(nsIToolkitProfile** aNewProfile);
nsresult ApplyResetProfile(nsIToolkitProfile* aOldProfile);
bool HasShowProfileSelector();
void UpdateCurrentProfile();
void CompleteStartup();
using AsyncFlushPromise =
mozilla::MozPromise<bool /* ignored */, nsresult, false>;
private:
friend class nsToolkitProfile;
friend already_AddRefed<nsToolkitProfileService>
NS_GetToolkitProfileService();
nsToolkitProfileService();
~nsToolkitProfileService();
nsresult Init();
nsresult CreateTimesInternal(nsIFile* profileDir);
void GetProfileByDir(nsIFile* aRootDir, nsIFile* aLocalDir,
nsToolkitProfile** aResult);
already_AddRefed<nsToolkitProfile> GetProfileByStoreID(
const nsACString& aStoreID);
nsresult GetProfileDescriptor(nsIFile* aRootDir, nsACString& aDescriptor,
bool* aIsRelative);
nsresult GetProfileDescriptor(nsToolkitProfile* aProfile,
nsACString& aDescriptor, bool* aIsRelative);
bool IsProfileForCurrentInstall(nsToolkitProfile* aProfile);
void ClearProfileFromOtherInstalls(nsToolkitProfile* aProfile);
nsresult MaybeMakeDefaultDedicatedProfile(nsToolkitProfile* aProfile,
bool* aResult);
bool IsSnapEnvironment();
bool UseLegacyProfiles();
nsresult CreateDefaultProfile(nsToolkitProfile** aResult);
nsresult CreateUniqueProfile(nsIFile* aRootDir, const nsACString& aNamePrefix,
nsToolkitProfile** aResult);
nsresult CreateProfile(nsIFile* aRootDir, const nsACString& aName,
nsToolkitProfile** aResult);
already_AddRefed<nsToolkitProfile> GetProfileByName(const nsACString& aName);
void SetNormalDefault(nsToolkitProfile* aProfile);
already_AddRefed<nsToolkitProfile> GetDefaultProfile();
nsresult GetLocalDirFromRootDir(nsIFile* aRootDir, nsIFile** aResult);
void FlushProfileData(
const nsMainThreadPtrHandle<nsStartupLock>& aStartupLock,
const CurrentProfileData* aProfileInfo);
void BuildIniData(nsCString& aProfilesIniData, nsCString& aInstallsIniData);
nsresult FlushData(const nsCString& aProfilesIniData,
const nsCString& aInstallsIniData);
// Returns the known install hashes from the installs database. Modifying the
// installs database is safe while iterating the returned array.
nsTArray<nsCString> GetKnownInstalls();
// Tracks whether SelectStartupProfile has been called.
bool mStartupProfileSelected;
// The profiles loaded from profiles.ini.
mozilla::LinkedList<RefPtr<nsToolkitProfile>> mProfiles;
// The profile selected for use at startup, if it exists in profiles.ini.
RefPtr<nsToolkitProfile> mCurrent;
// The profile selected for this install in installs.ini.
RefPtr<nsToolkitProfile> mDedicatedProfile;
// The default profile used by non-dev-edition builds.
RefPtr<nsToolkitProfile> mNormalDefault;
// The profile used if mUseDevEditionProfile is true (the default on
// dev-edition builds).
RefPtr<nsToolkitProfile> mDevEditionDefault;
// The directory that holds profiles.ini and profile directories.
nsCOMPtr<nsIFile> mAppData;
// The directory that holds the cache files for profiles.
nsCOMPtr<nsIFile> mTempData;
// The location of profiles.ini.
nsCOMPtr<nsIFile> mProfileDBFile;
// The location of installs.ini.
nsCOMPtr<nsIFile> mInstallDBFile;
// The data loaded from profiles.ini.
nsINIParser mProfileDB;
// The section in the profiles db for the current install.
nsCString mInstallSection;
// A legacy install section which may have been generated against an
// installation directory with an incorrect case (see bug 1555319). It is only
// really held here so that it can be overridden by tests.
nsCString mLegacyInstallSection;
// Whether to start with the selected profile by default.
bool mStartWithLast;
// True if during startup it appeared that this is the first run.
bool mIsFirstRun;
// True if the default profile is the separate dev-edition-profile.
bool mUseDevEditionProfile;
// True if this install should use a dedicated default profile.
const bool mUseDedicatedProfile;
nsCString mStartupReason;
// Records the version of the profiles.ini file as it was when it was loaded
// during startup.
nsCString mStartupFileVersion;
bool mMaybeLockProfile;
// Holds the current application update channel. This is only really held
// so it can be overriden in tests.
nsCString mUpdateChannel;
// Keep track of some attributes of the databases so we can tell if another
// process has changed them.
bool mProfileDBExists;
int64_t mProfileDBFileSize;
PRTime mProfileDBModifiedTime;
// A background task queue for the async flushing operations.
nsCOMPtr<nsISerialEventTarget> mAsyncQueue;
nsISerialEventTarget* AsyncQueue();
static nsToolkitProfileService* gService;
class ProfileEnumerator final : public nsSimpleEnumerator {
public:
NS_DECL_NSISIMPLEENUMERATOR
const nsID& DefaultInterface() override {
return NS_GET_IID(nsIToolkitProfile);
}
explicit ProfileEnumerator(nsToolkitProfile* first) { mCurrent = first; }
private:
RefPtr<nsToolkitProfile> mCurrent;
};
};
already_AddRefed<nsToolkitProfileService> NS_GetToolkitProfileService();
#endif
|
