|  | // 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 "base/metrics/sparse_histogram.h" | 
|  |  | 
|  | #include <utility> | 
|  |  | 
|  | #include "base/memory/ptr_util.h" | 
|  | #include "base/metrics/dummy_histogram.h" | 
|  | #include "base/metrics/metrics_hashes.h" | 
|  | #include "base/metrics/persistent_histogram_allocator.h" | 
|  | #include "base/metrics/persistent_sample_map.h" | 
|  | #include "base/metrics/sample_map.h" | 
|  | #include "base/metrics/statistics_recorder.h" | 
|  | #include "base/pickle.h" | 
|  | #include "base/strings/stringprintf.h" | 
|  | #include "base/synchronization/lock.h" | 
|  |  | 
|  | namespace base { | 
|  |  | 
|  | typedef HistogramBase::Count Count; | 
|  | typedef HistogramBase::Sample Sample; | 
|  |  | 
|  | // static | 
|  | HistogramBase* SparseHistogram::FactoryGet(const std::string& name, | 
|  | int32_t flags) { | 
|  | HistogramBase* histogram = StatisticsRecorder::FindHistogram(name); | 
|  | if (!histogram) { | 
|  | // TODO(gayane): |HashMetricName| is called again in Histogram constructor. | 
|  | // Refactor code to avoid the additional call. | 
|  | bool should_record = | 
|  | StatisticsRecorder::ShouldRecordHistogram(HashMetricName(name)); | 
|  | if (!should_record) | 
|  | return DummyHistogram::GetInstance(); | 
|  | // Try to create the histogram using a "persistent" allocator. As of | 
|  | // 2016-02-25, the availability of such is controlled by a base::Feature | 
|  | // that is off by default. If the allocator doesn't exist or if | 
|  | // allocating from it fails, code below will allocate the histogram from | 
|  | // the process heap. | 
|  | PersistentMemoryAllocator::Reference histogram_ref = 0; | 
|  | std::unique_ptr<HistogramBase> tentative_histogram; | 
|  | PersistentHistogramAllocator* allocator = GlobalHistogramAllocator::Get(); | 
|  | if (allocator) { | 
|  | tentative_histogram = allocator->AllocateHistogram( | 
|  | SPARSE_HISTOGRAM, name, 0, 0, nullptr, flags, &histogram_ref); | 
|  | } | 
|  |  | 
|  | // Handle the case where no persistent allocator is present or the | 
|  | // persistent allocation fails (perhaps because it is full). | 
|  | if (!tentative_histogram) { | 
|  | DCHECK(!histogram_ref);  // Should never have been set. | 
|  | DCHECK(!allocator);      // Shouldn't have failed. | 
|  | flags &= ~HistogramBase::kIsPersistent; | 
|  | tentative_histogram.reset(new SparseHistogram(GetPermanentName(name))); | 
|  | tentative_histogram->SetFlags(flags); | 
|  | } | 
|  |  | 
|  | // Register this histogram with the StatisticsRecorder. Keep a copy of | 
|  | // the pointer value to tell later whether the locally created histogram | 
|  | // was registered or deleted. The type is "void" because it could point | 
|  | // to released memory after the following line. | 
|  | const void* tentative_histogram_ptr = tentative_histogram.get(); | 
|  | histogram = StatisticsRecorder::RegisterOrDeleteDuplicate( | 
|  | tentative_histogram.release()); | 
|  |  | 
|  | // Persistent histograms need some follow-up processing. | 
|  | if (histogram_ref) { | 
|  | allocator->FinalizeHistogram(histogram_ref, | 
|  | histogram == tentative_histogram_ptr); | 
|  | } | 
|  | } | 
|  |  | 
|  | CHECK_EQ(SPARSE_HISTOGRAM, histogram->GetHistogramType()); | 
|  | return histogram; | 
|  | } | 
|  |  | 
|  | // static | 
|  | std::unique_ptr<HistogramBase> SparseHistogram::PersistentCreate( | 
|  | PersistentHistogramAllocator* allocator, | 
|  | const char* name, | 
|  | HistogramSamples::Metadata* meta, | 
|  | HistogramSamples::Metadata* logged_meta) { | 
|  | return WrapUnique( | 
|  | new SparseHistogram(allocator, name, meta, logged_meta)); | 
|  | } | 
|  |  | 
|  | SparseHistogram::~SparseHistogram() = default; | 
|  |  | 
|  | uint64_t SparseHistogram::name_hash() const { | 
|  | return unlogged_samples_->id(); | 
|  | } | 
|  |  | 
|  | HistogramType SparseHistogram::GetHistogramType() const { | 
|  | return SPARSE_HISTOGRAM; | 
|  | } | 
|  |  | 
|  | bool SparseHistogram::HasConstructionArguments( | 
|  | Sample expected_minimum, | 
|  | Sample expected_maximum, | 
|  | uint32_t expected_bucket_count) const { | 
|  | // SparseHistogram never has min/max/bucket_count limit. | 
|  | return false; | 
|  | } | 
|  |  | 
|  | void SparseHistogram::Add(Sample value) { | 
|  | AddCount(value, 1); | 
|  | } | 
|  |  | 
|  | void SparseHistogram::AddCount(Sample value, int count) { | 
|  | if (count <= 0) { | 
|  | NOTREACHED(); | 
|  | return; | 
|  | } | 
|  | { | 
|  | base::AutoLock auto_lock(lock_); | 
|  | unlogged_samples_->Accumulate(value, count); | 
|  | } | 
|  |  | 
|  | FindAndRunCallback(value); | 
|  | } | 
|  |  | 
|  | std::unique_ptr<HistogramSamples> SparseHistogram::SnapshotSamples() const { | 
|  | std::unique_ptr<SampleMap> snapshot(new SampleMap(name_hash())); | 
|  |  | 
|  | base::AutoLock auto_lock(lock_); | 
|  | snapshot->Add(*unlogged_samples_); | 
|  | snapshot->Add(*logged_samples_); | 
|  | return std::move(snapshot); | 
|  | } | 
|  |  | 
|  | std::unique_ptr<HistogramSamples> SparseHistogram::SnapshotDelta() { | 
|  | DCHECK(!final_delta_created_); | 
|  |  | 
|  | std::unique_ptr<SampleMap> snapshot(new SampleMap(name_hash())); | 
|  | base::AutoLock auto_lock(lock_); | 
|  | snapshot->Add(*unlogged_samples_); | 
|  |  | 
|  | unlogged_samples_->Subtract(*snapshot); | 
|  | logged_samples_->Add(*snapshot); | 
|  | return std::move(snapshot); | 
|  | } | 
|  |  | 
|  | std::unique_ptr<HistogramSamples> SparseHistogram::SnapshotFinalDelta() const { | 
|  | DCHECK(!final_delta_created_); | 
|  | final_delta_created_ = true; | 
|  |  | 
|  | std::unique_ptr<SampleMap> snapshot(new SampleMap(name_hash())); | 
|  | base::AutoLock auto_lock(lock_); | 
|  | snapshot->Add(*unlogged_samples_); | 
|  |  | 
|  | return std::move(snapshot); | 
|  | } | 
|  |  | 
|  | void SparseHistogram::AddSamples(const HistogramSamples& samples) { | 
|  | base::AutoLock auto_lock(lock_); | 
|  | unlogged_samples_->Add(samples); | 
|  | } | 
|  |  | 
|  | bool SparseHistogram::AddSamplesFromPickle(PickleIterator* iter) { | 
|  | base::AutoLock auto_lock(lock_); | 
|  | return unlogged_samples_->AddFromPickle(iter); | 
|  | } | 
|  |  | 
|  | void SparseHistogram::WriteHTMLGraph(std::string* output) const { | 
|  | output->append("<PRE>"); | 
|  | WriteAsciiImpl(true, "<br>", output); | 
|  | output->append("</PRE>"); | 
|  | } | 
|  |  | 
|  | void SparseHistogram::WriteAscii(std::string* output) const { | 
|  | WriteAsciiImpl(true, "\n", output); | 
|  | } | 
|  |  | 
|  | void SparseHistogram::SerializeInfoImpl(Pickle* pickle) const { | 
|  | pickle->WriteString(histogram_name()); | 
|  | pickle->WriteInt(flags()); | 
|  | } | 
|  |  | 
|  | SparseHistogram::SparseHistogram(const char* name) | 
|  | : HistogramBase(name), | 
|  | unlogged_samples_(new SampleMap(HashMetricName(name))), | 
|  | logged_samples_(new SampleMap(unlogged_samples_->id())) {} | 
|  |  | 
|  | SparseHistogram::SparseHistogram(PersistentHistogramAllocator* allocator, | 
|  | const char* name, | 
|  | HistogramSamples::Metadata* meta, | 
|  | HistogramSamples::Metadata* logged_meta) | 
|  | : HistogramBase(name), | 
|  | // While other histogram types maintain a static vector of values with | 
|  | // sufficient space for both "active" and "logged" samples, with each | 
|  | // SampleVector being given the appropriate half, sparse histograms | 
|  | // have no such initial allocation. Each sample has its own record | 
|  | // attached to a single PersistentSampleMap by a common 64-bit identifier. | 
|  | // Since a sparse histogram has two sample maps (active and logged), | 
|  | // there must be two sets of sample records with diffent IDs. The | 
|  | // "active" samples use, for convenience purposes, an ID matching | 
|  | // that of the histogram while the "logged" samples use that number | 
|  | // plus 1. | 
|  | unlogged_samples_( | 
|  | new PersistentSampleMap(HashMetricName(name), allocator, meta)), | 
|  | logged_samples_(new PersistentSampleMap(unlogged_samples_->id() + 1, | 
|  | allocator, | 
|  | logged_meta)) {} | 
|  |  | 
|  | HistogramBase* SparseHistogram::DeserializeInfoImpl(PickleIterator* iter) { | 
|  | std::string histogram_name; | 
|  | int flags; | 
|  | if (!iter->ReadString(&histogram_name) || !iter->ReadInt(&flags)) { | 
|  | DLOG(ERROR) << "Pickle error decoding Histogram: " << histogram_name; | 
|  | return nullptr; | 
|  | } | 
|  |  | 
|  | flags &= ~HistogramBase::kIPCSerializationSourceFlag; | 
|  |  | 
|  | return SparseHistogram::FactoryGet(histogram_name, flags); | 
|  | } | 
|  |  | 
|  | void SparseHistogram::GetParameters(DictionaryValue* params) const { | 
|  | // TODO(kaiwang): Implement. (See HistogramBase::WriteJSON.) | 
|  | } | 
|  |  | 
|  | void SparseHistogram::GetCountAndBucketData(Count* count, | 
|  | int64_t* sum, | 
|  | ListValue* buckets) const { | 
|  | // TODO(kaiwang): Implement. (See HistogramBase::WriteJSON.) | 
|  | } | 
|  |  | 
|  | void SparseHistogram::WriteAsciiImpl(bool graph_it, | 
|  | const std::string& newline, | 
|  | std::string* output) const { | 
|  | // Get a local copy of the data so we are consistent. | 
|  | std::unique_ptr<HistogramSamples> snapshot = SnapshotSamples(); | 
|  | Count total_count = snapshot->TotalCount(); | 
|  | double scaled_total_count = total_count / 100.0; | 
|  |  | 
|  | WriteAsciiHeader(total_count, output); | 
|  | output->append(newline); | 
|  |  | 
|  | // Determine how wide the largest bucket range is (how many digits to print), | 
|  | // so that we'll be able to right-align starts for the graphical bars. | 
|  | // Determine which bucket has the largest sample count so that we can | 
|  | // normalize the graphical bar-width relative to that sample count. | 
|  | Count largest_count = 0; | 
|  | Sample largest_sample = 0; | 
|  | std::unique_ptr<SampleCountIterator> it = snapshot->Iterator(); | 
|  | while (!it->Done()) { | 
|  | Sample min; | 
|  | int64_t max; | 
|  | Count count; | 
|  | it->Get(&min, &max, &count); | 
|  | if (min > largest_sample) | 
|  | largest_sample = min; | 
|  | if (count > largest_count) | 
|  | largest_count = count; | 
|  | it->Next(); | 
|  | } | 
|  | size_t print_width = GetSimpleAsciiBucketRange(largest_sample).size() + 1; | 
|  |  | 
|  | // iterate over each item and display them | 
|  | it = snapshot->Iterator(); | 
|  | while (!it->Done()) { | 
|  | Sample min; | 
|  | int64_t max; | 
|  | Count count; | 
|  | it->Get(&min, &max, &count); | 
|  |  | 
|  | // value is min, so display it | 
|  | std::string range = GetSimpleAsciiBucketRange(min); | 
|  | output->append(range); | 
|  | for (size_t j = 0; range.size() + j < print_width + 1; ++j) | 
|  | output->push_back(' '); | 
|  |  | 
|  | if (graph_it) | 
|  | WriteAsciiBucketGraph(count, largest_count, output); | 
|  | WriteAsciiBucketValue(count, scaled_total_count, output); | 
|  | output->append(newline); | 
|  | it->Next(); | 
|  | } | 
|  | } | 
|  |  | 
|  | void SparseHistogram::WriteAsciiHeader(const Count total_count, | 
|  | std::string* output) const { | 
|  | StringAppendF(output, "Histogram: %s recorded %d samples", histogram_name(), | 
|  | total_count); | 
|  | if (flags()) | 
|  | StringAppendF(output, " (flags = 0x%x)", flags()); | 
|  | } | 
|  |  | 
|  | }  // namespace base |