blob: 6e30238305e83c4c4286b90ead6c110258c9a744 (
plain)
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
|
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/bind.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/prefs/pref_service.h"
#include "base/strings/string_util.h"
#include "base/task_runner_util.h"
#include "base/threading/worker_pool.h"
#include "chrome/browser/browser_process.h"
#include "chrome/common/pref_names.h"
#include "chromeos/system/statistics_provider.h"
#include "content/public/browser/browser_thread.h"
namespace google_brand {
namespace chromeos {
namespace {
// Path to file that stores the RLZ brand code on ChromeOS.
const base::FilePath::CharType kRLZBrandFilePath[] =
FILE_PATH_LITERAL("/opt/oem/etc/BRAND_CODE");
// Reads the brand code from file |kRLZBrandFilePath|.
std::string ReadBrandFromFile() {
std::string brand;
base::FilePath brand_file_path(kRLZBrandFilePath);
if (!base::ReadFileToString(brand_file_path, &brand))
LOG(WARNING) << "Brand code file missing: " << brand_file_path.value();
base::TrimWhitespaceASCII(brand, base::TRIM_ALL, &brand);
return brand;
}
// Sets the brand code to |brand|.
void SetBrand(const base::Closure& callback, const std::string& brand) {
g_browser_process->local_state()->SetString(prefs::kRLZBrand, brand);
callback.Run();
}
// True if brand code has been cleared for the current session.
bool g_brand_empty = false;
} // namespace
void ClearBrandForCurrentSession() {
DCHECK(!content::BrowserThread::IsThreadInitialized(
content::BrowserThread::UI) ||
content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
g_brand_empty = true;
}
std::string GetBrand() {
DCHECK(!content::BrowserThread::IsThreadInitialized(
content::BrowserThread::UI) ||
content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
if (g_brand_empty)
return std::string();
// Unit tests do not have prefs.
if (!g_browser_process->local_state())
return std::string();
return g_browser_process->local_state()->GetString(prefs::kRLZBrand);
}
void InitBrand(const base::Closure& callback) {
::chromeos::system::StatisticsProvider* provider =
::chromeos::system::StatisticsProvider::GetInstance();
std::string brand;
const bool found = provider->GetMachineStatistic(
::chromeos::system::kRlzBrandCodeKey, &brand);
if (found && !brand.empty()) {
SetBrand(callback, brand);
return;
}
base::PostTaskAndReplyWithResult(
base::WorkerPool::GetTaskRunner(false /* task_is_slow */).get(),
FROM_HERE,
base::Bind(&ReadBrandFromFile),
base::Bind(&SetBrand, callback));
}
} // namespace chromeos
} // namespace google_brand
|
