| // Copyright (c) 2012 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 "chrome/browser/chromeos/policy/user_policy_disk_cache.h" |
| |
| #include "base/bind.h" |
| #include "base/file_util.h" |
| #include "base/logging.h" |
| #include "base/metrics/histogram.h" |
| #include "chrome/browser/policy/cloud/enterprise_metrics.h" |
| #include "chrome/browser/policy/proto/cloud/device_management_local.pb.h" |
| #include "content/public/browser/browser_thread.h" |
| |
| using content::BrowserThread; |
| |
| namespace em = enterprise_management; |
| |
| namespace { |
| |
| // Other places can sample on the same UMA counter, so make sure they all do |
| // it on the same thread (UI). |
| void SampleUMAOnUIThread(policy::MetricPolicy sample) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| UMA_HISTOGRAM_ENUMERATION(policy::kMetricPolicy, sample, |
| policy::kMetricPolicySize); |
| } |
| |
| void SampleUMA(policy::MetricPolicy sample) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
| base::Bind(&SampleUMAOnUIThread, sample)); |
| } |
| |
| } // namespace |
| |
| namespace policy { |
| |
| UserPolicyDiskCache::Delegate::~Delegate() {} |
| |
| UserPolicyDiskCache::UserPolicyDiskCache( |
| const base::WeakPtr<Delegate>& delegate, |
| const base::FilePath& backing_file_path) |
| : delegate_(delegate), |
| backing_file_path_(backing_file_path) {} |
| |
| void UserPolicyDiskCache::Load() { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| bool ret = BrowserThread::PostTask( |
| BrowserThread::FILE, FROM_HERE, |
| base::Bind(&UserPolicyDiskCache::LoadOnFileThread, this)); |
| DCHECK(ret); |
| } |
| |
| void UserPolicyDiskCache::Store( |
| const em::CachedCloudPolicyResponse& policy) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| BrowserThread::PostTask( |
| BrowserThread::FILE, FROM_HERE, |
| base::Bind(&UserPolicyDiskCache::StoreOnFileThread, this, policy)); |
| } |
| |
| UserPolicyDiskCache::~UserPolicyDiskCache() {} |
| |
| void UserPolicyDiskCache::LoadOnFileThread() { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| |
| em::CachedCloudPolicyResponse cached_response; |
| if (!base::PathExists(backing_file_path_)) { |
| LoadDone(LOAD_RESULT_NOT_FOUND, cached_response); |
| return; |
| } |
| |
| // Read the protobuf from the file. |
| std::string data; |
| if (!base::ReadFileToString(backing_file_path_, &data)) { |
| LOG(WARNING) << "Failed to read policy data from " |
| << backing_file_path_.value(); |
| LoadDone(LOAD_RESULT_READ_ERROR, cached_response); |
| return; |
| } |
| |
| // Decode it. |
| if (!cached_response.ParseFromArray(data.c_str(), data.size())) { |
| LOG(WARNING) << "Failed to parse policy data read from " |
| << backing_file_path_.value(); |
| LoadDone(LOAD_RESULT_PARSE_ERROR, cached_response); |
| return; |
| } |
| |
| LoadDone(LOAD_RESULT_SUCCESS, cached_response); |
| } |
| |
| void UserPolicyDiskCache::LoadDone( |
| LoadResult result, |
| const em::CachedCloudPolicyResponse& policy) { |
| BrowserThread::PostTask( |
| BrowserThread::UI, FROM_HERE, |
| base::Bind(&UserPolicyDiskCache::ReportResultOnUIThread, this, |
| result, policy)); |
| } |
| |
| void UserPolicyDiskCache::ReportResultOnUIThread( |
| LoadResult result, |
| const em::CachedCloudPolicyResponse& policy) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| |
| switch (result) { |
| case LOAD_RESULT_NOT_FOUND: |
| break; |
| case LOAD_RESULT_READ_ERROR: |
| case LOAD_RESULT_PARSE_ERROR: |
| SampleUMAOnUIThread(kMetricPolicyLoadFailed); |
| break; |
| case LOAD_RESULT_SUCCESS: |
| SampleUMAOnUIThread(kMetricPolicyLoadSucceeded); |
| } |
| |
| if (delegate_.get()) |
| delegate_->OnDiskCacheLoaded(result, policy); |
| } |
| |
| void UserPolicyDiskCache::StoreOnFileThread( |
| const em::CachedCloudPolicyResponse& policy) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| std::string data; |
| if (!policy.SerializeToString(&data)) { |
| LOG(WARNING) << "Failed to serialize policy data"; |
| SampleUMA(kMetricPolicyStoreFailed); |
| return; |
| } |
| |
| if (!file_util::CreateDirectory(backing_file_path_.DirName())) { |
| LOG(WARNING) << "Failed to create directory " |
| << backing_file_path_.DirName().value(); |
| SampleUMA(kMetricPolicyStoreFailed); |
| return; |
| } |
| |
| int size = data.size(); |
| if (file_util::WriteFile(backing_file_path_, data.c_str(), size) != size) { |
| LOG(WARNING) << "Failed to write " << backing_file_path_.value(); |
| SampleUMA(kMetricPolicyStoreFailed); |
| return; |
| } |
| SampleUMA(kMetricPolicyStoreSucceeded); |
| } |
| |
| } // namespace policy |