Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(57)

Side by Side Diff: chrome/installer/setup/setup_util.cc

Issue 2507753002: Install the chrome event log provider together with the browser. (Closed)
Patch Set: Address comments. Created 4 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « chrome/installer/setup/setup_util.h ('k') | chrome/installer/setup/setup_util_unittest.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2012 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 // This file declares util functions for setup project. 5 // This file declares util functions for setup project.
6 6
7 #include "chrome/installer/setup/setup_util.h" 7 #include "chrome/installer/setup/setup_util.h"
8 8
9 #include <windows.h> 9 #include <windows.h>
10 #include <stddef.h> 10 #include <stddef.h>
11 11
12 #include <algorithm> 12 #include <algorithm>
13 #include <iterator> 13 #include <iterator>
14 #include <limits>
15 #include <memory>
14 #include <set> 16 #include <set>
15 #include <string> 17 #include <string>
16 18
17 #include "base/bind.h" 19 #include "base/bind.h"
18 #include "base/command_line.h" 20 #include "base/command_line.h"
19 #include "base/cpu.h" 21 #include "base/cpu.h"
20 #include "base/files/file_enumerator.h" 22 #include "base/files/file_enumerator.h"
21 #include "base/files/file_path.h" 23 #include "base/files/file_path.h"
22 #include "base/files/file_util.h" 24 #include "base/files/file_util.h"
23 #include "base/logging.h" 25 #include "base/logging.h"
24 #include "base/macros.h" 26 #include "base/macros.h"
25 #include "base/metrics/histogram.h" 27 #include "base/metrics/histogram.h"
26 #include "base/numerics/safe_conversions.h" 28 #include "base/numerics/safe_conversions.h"
27 #include "base/strings/string_number_conversions.h" 29 #include "base/strings/string_number_conversions.h"
28 #include "base/strings/string_util.h" 30 #include "base/strings/string_util.h"
29 #include "base/strings/utf_string_conversions.h" 31 #include "base/strings/utf_string_conversions.h"
30 #include "base/time/time.h" 32 #include "base/time/time.h"
31 #include "base/version.h" 33 #include "base/version.h"
32 #include "base/win/registry.h" 34 #include "base/win/registry.h"
33 #include "base/win/windows_version.h" 35 #include "base/win/windows_version.h"
34 #include "chrome/installer/setup/installer_state.h" 36 #include "chrome/installer/setup/installer_state.h"
35 #include "chrome/installer/setup/setup_constants.h" 37 #include "chrome/installer/setup/setup_constants.h"
36 #include "chrome/installer/setup/user_hive_visitor.h" 38 #include "chrome/installer/setup/user_hive_visitor.h"
37 #include "chrome/installer/util/app_registration_data.h" 39 #include "chrome/installer/util/app_registration_data.h"
38 #include "chrome/installer/util/google_update_constants.h" 40 #include "chrome/installer/util/google_update_constants.h"
41 #include "chrome/installer/util/install_util.h"
39 #include "chrome/installer/util/installation_state.h" 42 #include "chrome/installer/util/installation_state.h"
40 #include "chrome/installer/util/master_preferences.h" 43 #include "chrome/installer/util/master_preferences.h"
41 #include "chrome/installer/util/master_preferences_constants.h" 44 #include "chrome/installer/util/master_preferences_constants.h"
42 #include "chrome/installer/util/util_constants.h" 45 #include "chrome/installer/util/util_constants.h"
43 #include "courgette/courgette.h" 46 #include "courgette/courgette.h"
44 #include "courgette/third_party/bsdiff/bsdiff.h" 47 #include "courgette/third_party/bsdiff/bsdiff.h"
45 48
46 namespace installer { 49 namespace installer {
47 50
48 namespace { 51 namespace {
49 52
53 // Event log providers registry location.
54 constexpr wchar_t kEventLogProvidersRegPath[] =
55 L"SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\";
56
57 // TODO(grt): use install_static::InstallDetails::Get().install_full_name() when
58 // InstallDetails is initialized in the installer.
59 base::string16 InstallFullName() {
60 base::string16 reg_path(kEventLogProvidersRegPath);
grt (UTC plus 2) 2016/12/16 13:03:58 remove
pastarmovj 2016/12/16 14:58:12 Done.
61 #if defined(GOOGLE_CHROME_BUILD)
62 reg_path = L"Chrome";
63 if (InstallUtil::IsChromeSxSProcess())
64 reg_path.append(L" SxS");
65 #else
66 reg_path = L"Chromium";
67 #endif
68 return reg_path;
69 }
70
50 // Returns true if product |type| cam be meaningfully installed without the 71 // Returns true if product |type| cam be meaningfully installed without the
51 // --multi-install flag. 72 // --multi-install flag.
52 bool SupportsSingleInstall(BrowserDistribution::Type type) { 73 bool SupportsSingleInstall(BrowserDistribution::Type type) {
53 return (type == BrowserDistribution::CHROME_BROWSER || 74 return (type == BrowserDistribution::CHROME_BROWSER ||
54 type == BrowserDistribution::CHROME_FRAME); 75 type == BrowserDistribution::CHROME_FRAME);
55 } 76 }
56 77
57 // Returns true if the "lastrun" value in |root|\|key_path| (a path to Chrome's 78 // Returns true if the "lastrun" value in |root|\|key_path| (a path to Chrome's
58 // ClientState key for a user) indicates that Chrome has been used within the 79 // ClientState key for a user) indicates that Chrome has been used within the
59 // last 28 days. 80 // last 28 days.
(...skipping 593 matching lines...) Expand 10 before | Expand all | Expand 10 after
653 UNPACK_STATUS_COUNT, UNPACK_STATUS_COUNT + 1, 674 UNPACK_STATUS_COUNT, UNPACK_STATUS_COUNT + 1,
654 base::HistogramBase::kUmaTargetedHistogramFlag) 675 base::HistogramBase::kUmaTargetedHistogramFlag)
655 ->Add(unpack_status); 676 ->Add(unpack_status);
656 677
657 base::SparseHistogram::FactoryGet( 678 base::SparseHistogram::FactoryGet(
658 std::string(kUnPackNTSTATUSMetricsName) + "_" + consumer_name, 679 std::string(kUnPackNTSTATUSMetricsName) + "_" + consumer_name,
659 base::HistogramBase::kUmaTargetedHistogramFlag) 680 base::HistogramBase::kUmaTargetedHistogramFlag)
660 ->Add(status); 681 ->Add(status);
661 } 682 }
662 683
684 void RegisterEventLogProvider(const base::FilePath& install_directory,
685 const base::Version& version) {
686 base::string16 reg_path(kEventLogProvidersRegPath);
687 reg_path.append(InstallFullName());
688 VLOG(1) << "Registering Chrome's event log provider at " << reg_path;
689
690 std::unique_ptr<WorkItemList> work_item_list(WorkItem::CreateWorkItemList());
691 work_item_list->set_log_message("Register event log provider");
692
693 work_item_list->AddCreateRegKeyWorkItem(HKEY_LOCAL_MACHINE, reg_path,
694 WorkItem::kWow64Default);
695 // Speicifes the number of event categories defined in the dll.
696 work_item_list->AddSetRegValueWorkItem(
697 HKEY_LOCAL_MACHINE, reg_path, WorkItem::kWow64Default, L"CategoryCount",
698 static_cast<DWORD>(1), true);
699 // Specifies the event type emitted by this event source.
700 work_item_list->AddSetRegValueWorkItem(
701 HKEY_LOCAL_MACHINE, reg_path, WorkItem::kWow64Default, L"TypesSupported",
702 static_cast<DWORD>(EVENTLOG_ERROR_TYPE | EVENTLOG_INFORMATION_TYPE |
703 EVENTLOG_WARNING_TYPE),
704 true);
705
706 const base::FilePath provider(
707 install_directory.AppendASCII(version.GetString())
708 .Append(FILE_PATH_LITERAL("eventlog_provider.dll")));
709
710 static constexpr const wchar_t* kFileKeys[] = {
711 L"CategoryMessageFile", L"EventMessageFile", L"ParameterMessageFile",
712 };
713 for (const wchar_t* file_key : kFileKeys) {
714 work_item_list->AddSetRegValueWorkItem(HKEY_LOCAL_MACHINE, reg_path,
715 WorkItem::kWow64Default, file_key,
716 provider.value(), true);
717 }
718
719 // if the operation fails we log the error but still continue because none of
720 // these are critical for the proper operation of the browser.
721 if (!work_item_list->Do())
722 work_item_list->Rollback();
723 }
724
725 void DeRegisterEventLogProvider() {
726 base::string16 reg_path(kEventLogProvidersRegPath);
727 reg_path.append(InstallFullName());
728
729 // TODO(http://crbug.com/668120): If the Event Viewer is open the provider dll
730 // will fail to get deleted. This doesn't fail the uninstallation altogether
731 // but leaves files behind.
732 InstallUtil::DeleteRegistryKey(HKEY_LOCAL_MACHINE, reg_path,
733 WorkItem::kWow64Default);
734 }
735
663 ScopedTokenPrivilege::ScopedTokenPrivilege(const wchar_t* privilege_name) 736 ScopedTokenPrivilege::ScopedTokenPrivilege(const wchar_t* privilege_name)
664 : is_enabled_(false) { 737 : is_enabled_(false) {
665 HANDLE temp_handle; 738 HANDLE temp_handle;
666 if (!::OpenProcessToken(::GetCurrentProcess(), 739 if (!::OpenProcessToken(::GetCurrentProcess(),
667 TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, 740 TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
668 &temp_handle)) { 741 &temp_handle)) {
669 return; 742 return;
670 } 743 }
671 token_.Set(temp_handle); 744 token_.Set(temp_handle);
672 745
(...skipping 22 matching lines...) Expand all
695 } 768 }
696 769
697 ScopedTokenPrivilege::~ScopedTokenPrivilege() { 770 ScopedTokenPrivilege::~ScopedTokenPrivilege() {
698 if (is_enabled_ && previous_privileges_.PrivilegeCount != 0) { 771 if (is_enabled_ && previous_privileges_.PrivilegeCount != 0) {
699 ::AdjustTokenPrivileges(token_.Get(), FALSE, &previous_privileges_, 772 ::AdjustTokenPrivileges(token_.Get(), FALSE, &previous_privileges_,
700 sizeof(TOKEN_PRIVILEGES), NULL, NULL); 773 sizeof(TOKEN_PRIVILEGES), NULL, NULL);
701 } 774 }
702 } 775 }
703 776
704 } // namespace installer 777 } // namespace installer
OLDNEW
« no previous file with comments | « chrome/installer/setup/setup_util.h ('k') | chrome/installer/setup/setup_util_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698