OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_ELF_BLACKLIST_BLACKLIST_H_ | 5 #ifndef CHROME_ELF_BLACKLIST_BLACKLIST_H_ |
6 #define CHROME_ELF_BLACKLIST_BLACKLIST_H_ | 6 #define CHROME_ELF_BLACKLIST_BLACKLIST_H_ |
7 | 7 |
8 namespace blacklist { | 8 namespace blacklist { |
9 | 9 |
10 // Max size of the DLL blacklist. | 10 // Max size of the DLL blacklist. |
11 const int kTroublesomeDllsMaxCount = 64; | 11 const int kTroublesomeDllsMaxCount = 64; |
12 | 12 |
13 // The DLL blacklist. | 13 // The DLL blacklist. |
14 extern const wchar_t* g_troublesome_dlls[kTroublesomeDllsMaxCount]; | 14 extern const wchar_t* g_troublesome_dlls[kTroublesomeDllsMaxCount]; |
15 | 15 |
16 // Cursor to the current last element in the blacklist. | 16 // Cursor to the current last element in the blacklist. |
17 extern int g_troublesome_dlls_cur_index; | 17 extern int g_troublesome_dlls_cur_index; |
18 | 18 |
19 // The registry path of the blacklist beacon. | 19 // The registry path of the blacklist beacon. Exposed here for testing. |
20 extern const wchar_t kRegistryBeaconPath[]; | 20 extern const wchar_t kRegistryBeaconPath[]; |
21 | 21 |
22 // The properties for the blacklist beacon. | 22 // Attempts to create a beacon in the current user's registry hive. |
23 extern const wchar_t kBeaconVersion[]; | 23 // If the beacon already exists or any other error occurs when creating the |
24 extern const wchar_t kBeaconState[]; | 24 // beacon, returns false. Otherwise returns true. |
25 | |
26 // The states for the blacklist setup code. | |
27 enum BlacklistState { | |
28 BLACKLIST_DISABLED = 0, | |
29 BLACKLIST_ENABLED, | |
30 // The blacklist setup code is running. If this is still set at startup, | |
31 // it means the last setup crashed. | |
32 BLACKLIST_SETUP_RUNNING, | |
33 // Always keep this at the end. | |
34 BLACKLIST_STATE_MAX, | |
35 }; | |
36 | |
37 // Attempts to leave a beacon in the current user's registry hive. | |
38 // If the blacklist beacon doesn't say it is enabled or there are any other | |
39 // errors when creating the beacon, returns false. Otherwise returns true. | |
40 // The intent of the beacon is to act as an extra failure mode protection | 25 // The intent of the beacon is to act as an extra failure mode protection |
41 // whereby if Chrome for some reason fails to start during blacklist setup, | 26 // whereby if Chrome for some reason fails to start during blacklist setup, |
42 // it will skip blacklisting on the subsequent run. | 27 // it will skip blacklisting on the subsequent run. |
43 bool LeaveSetupBeacon(); | 28 bool CreateBeacon(); |
44 | 29 |
45 // Looks for the beacon that LeaveSetupBeacon() creates and resets it to | 30 // Looks for the beacon that CreateBeacon() creates and attempts to delete it. |
46 // to show the setup was successful. | 31 // Returns true if the beacon was found and deleted. |
47 // Returns true if the beacon was successfully set to BLACKLIST_ENABLED. | 32 bool ClearBeacon(); |
48 bool ResetBeacon(); | |
49 | 33 |
50 // Adds the given dll name to the blacklist. Returns true if the dll name is in | 34 // Adds the given dll name to the blacklist. Returns true if the dll name is in |
51 // the blacklist when this returns, false on error. Note that this will copy | 35 // the blacklist when this returns, false on error. Note that this will copy |
52 // |dll_name| and will leak it on exit if the string is not subsequently removed | 36 // |dll_name| and will leak it on exit if the string is not subsequently removed |
53 // using RemoveDllFromBlacklist. | 37 // using RemoveDllFromBlacklist. |
54 extern "C" bool AddDllToBlacklist(const wchar_t* dll_name); | 38 extern "C" bool AddDllToBlacklist(const wchar_t* dll_name); |
55 | 39 |
56 // Removes the given dll name from the blacklist. Returns true if it was | 40 // Removes the given dll name from the blacklist. Returns true if it was |
57 // removed, false on error. | 41 // removed, false on error. |
58 extern "C" bool RemoveDllFromBlacklist(const wchar_t* dll_name); | 42 extern "C" bool RemoveDllFromBlacklist(const wchar_t* dll_name); |
59 | 43 |
60 // Initializes the DLL blacklist in the current process. This should be called | 44 // Initializes the DLL blacklist in the current process. This should be called |
61 // before any undesirable DLLs might be loaded. If |force| is set to true, then | 45 // before any undesirable DLLs might be loaded. If |force| is set to true, then |
62 // initialization will take place even if a beacon is present. This is useful | 46 // initialization will take place even if a beacon is present. This is useful |
63 // for tests. | 47 // for tests. |
64 bool Initialize(bool force); | 48 bool Initialize(bool force); |
65 | 49 |
66 } // namespace blacklist | 50 } // namespace blacklist |
67 | 51 |
68 #endif // CHROME_ELF_BLACKLIST_BLACKLIST_H_ | 52 #endif // CHROME_ELF_BLACKLIST_BLACKLIST_H_ |
OLD | NEW |