| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "components/metrics/chromeos/serialization_utils.h" | |
| 6 | |
| 7 #include <sys/file.h> | |
| 8 | |
| 9 #include <string> | |
| 10 #include <vector> | |
| 11 | |
| 12 #include "base/file_util.h" | |
| 13 #include "base/files/file_path.h" | |
| 14 #include "base/files/scoped_file.h" | |
| 15 #include "base/logging.h" | |
| 16 #include "base/memory/scoped_ptr.h" | |
| 17 #include "base/memory/scoped_vector.h" | |
| 18 #include "base/strings/string_split.h" | |
| 19 #include "base/strings/string_util.h" | |
| 20 #include "components/metrics/chromeos/metric_sample.h" | |
| 21 | |
| 22 #define READ_WRITE_ALL_FILE_FLAGS \ | |
| 23 (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH) | |
| 24 | |
| 25 namespace metrics { | |
| 26 namespace { | |
| 27 | |
| 28 // Reads the next message from |file_descriptor| into |message|. | |
| 29 // | |
| 30 // |message| will be set to the empty string if no message could be read (EOF) | |
| 31 // or the message was badly constructed. | |
| 32 // | |
| 33 // Returns false if no message can be read from this file anymore (EOF or | |
| 34 // unrecoverable error). | |
| 35 bool ReadMessage(int fd, std::string* message) { | |
| 36 CHECK(message); | |
| 37 | |
| 38 int result; | |
| 39 int32 message_size; | |
| 40 // The file containing the metrics do not leave the device so the writer and | |
| 41 // the reader will always have the same endianness. | |
| 42 result = HANDLE_EINTR(read(fd, &message_size, sizeof(message_size))); | |
| 43 if (result < 0) { | |
| 44 DPLOG(ERROR) << "reading metrics message header"; | |
| 45 return false; | |
| 46 } | |
| 47 if (result == 0) { | |
| 48 // This indicates a normal EOF. | |
| 49 return false; | |
| 50 } | |
| 51 if (result < static_cast<int>(sizeof(message_size))) { | |
| 52 DLOG(ERROR) << "bad read size " << result << ", expecting " | |
| 53 << sizeof(message_size); | |
| 54 return false; | |
| 55 } | |
| 56 | |
| 57 // kMessageMaxLength applies to the entire message: the 4-byte | |
| 58 // length field and the content. | |
| 59 if (message_size > metrics::SerializationUtils::kMessageMaxLength) { | |
| 60 DLOG(ERROR) << "message too long : " << message_size; | |
| 61 if (HANDLE_EINTR(lseek(fd, message_size - 4, SEEK_CUR)) == -1) { | |
| 62 DLOG(ERROR) << "error while skipping message. abort"; | |
| 63 return false; | |
| 64 } | |
| 65 // Badly formatted message was skipped. Treat the badly formatted sample as | |
| 66 // an empty sample. | |
| 67 message->clear(); | |
| 68 return true; | |
| 69 } | |
| 70 | |
| 71 message_size -= sizeof(message_size); // The message size includes itself. | |
| 72 char buffer[metrics::SerializationUtils::kMessageMaxLength]; | |
| 73 if (!base::ReadFromFD(fd, buffer, message_size)) { | |
| 74 DPLOG(ERROR) << "reading metrics message body"; | |
| 75 return false; | |
| 76 } | |
| 77 *message = std::string(buffer, message_size); | |
| 78 return true; | |
| 79 } | |
| 80 | |
| 81 } // namespace | |
| 82 | |
| 83 scoped_ptr<MetricSample> SerializationUtils::ParseSample( | |
| 84 const std::string& sample) { | |
| 85 if (sample.empty()) | |
| 86 return scoped_ptr<MetricSample>(); | |
| 87 | |
| 88 std::vector<std::string> parts; | |
| 89 base::SplitString(sample, '\0', &parts); | |
| 90 // We should have two null terminated strings so split should produce | |
| 91 // three chunks. | |
| 92 if (parts.size() != 3) { | |
| 93 DLOG(ERROR) << "splitting message on \\0 produced " << parts.size() | |
| 94 << " parts (expected 3)"; | |
| 95 return scoped_ptr<MetricSample>(); | |
| 96 } | |
| 97 const std::string& name = parts[0]; | |
| 98 const std::string& value = parts[1]; | |
| 99 | |
| 100 if (LowerCaseEqualsASCII(name, "crash")) { | |
| 101 return MetricSample::CrashSample(value); | |
| 102 } else if (LowerCaseEqualsASCII(name, "histogram")) { | |
| 103 return MetricSample::ParseHistogram(value); | |
| 104 } else if (LowerCaseEqualsASCII(name, "linearhistogram")) { | |
| 105 return MetricSample::ParseLinearHistogram(value); | |
| 106 } else if (LowerCaseEqualsASCII(name, "sparsehistogram")) { | |
| 107 return MetricSample::ParseSparseHistogram(value); | |
| 108 } else if (LowerCaseEqualsASCII(name, "useraction")) { | |
| 109 return MetricSample::UserActionSample(value); | |
| 110 } else { | |
| 111 DLOG(ERROR) << "invalid event type: " << name << ", value: " << value; | |
| 112 } | |
| 113 return scoped_ptr<MetricSample>(); | |
| 114 } | |
| 115 | |
| 116 void SerializationUtils::ReadAndTruncateMetricsFromFile( | |
| 117 const std::string& filename, | |
| 118 ScopedVector<MetricSample>* metrics) { | |
| 119 struct stat stat_buf; | |
| 120 int result; | |
| 121 | |
| 122 result = stat(filename.c_str(), &stat_buf); | |
| 123 if (result < 0) { | |
| 124 if (errno != ENOENT) | |
| 125 DPLOG(ERROR) << filename << ": bad metrics file stat"; | |
| 126 | |
| 127 // Nothing to collect---try later. | |
| 128 return; | |
| 129 } | |
| 130 if (stat_buf.st_size == 0) { | |
| 131 // Also nothing to collect. | |
| 132 return; | |
| 133 } | |
| 134 base::ScopedFD fd(open(filename.c_str(), O_RDWR)); | |
| 135 if (fd.get() < 0) { | |
| 136 DPLOG(ERROR) << filename << ": cannot open"; | |
| 137 return; | |
| 138 } | |
| 139 result = flock(fd.get(), LOCK_EX); | |
| 140 if (result < 0) { | |
| 141 DPLOG(ERROR) << filename << ": cannot lock"; | |
| 142 return; | |
| 143 } | |
| 144 | |
| 145 // This processes all messages in the log. When all messages are | |
| 146 // read and processed, or an error occurs, truncate the file to zero size. | |
| 147 for (;;) { | |
| 148 std::string message; | |
| 149 | |
| 150 if (!ReadMessage(fd.get(), &message)) | |
| 151 break; | |
| 152 | |
| 153 scoped_ptr<MetricSample> sample = ParseSample(message); | |
| 154 if (sample) | |
| 155 metrics->push_back(sample.release()); | |
| 156 } | |
| 157 | |
| 158 result = ftruncate(fd.get(), 0); | |
| 159 if (result < 0) | |
| 160 DPLOG(ERROR) << "truncate metrics log"; | |
| 161 | |
| 162 result = flock(fd.get(), LOCK_UN); | |
| 163 if (result < 0) | |
| 164 DPLOG(ERROR) << "unlock metrics log"; | |
| 165 } | |
| 166 | |
| 167 bool SerializationUtils::WriteMetricToFile(const MetricSample& sample, | |
| 168 const std::string& filename) { | |
| 169 if (!sample.IsValid()) | |
| 170 return false; | |
| 171 | |
| 172 base::ScopedFD file_descriptor(open(filename.c_str(), | |
| 173 O_WRONLY | O_APPEND | O_CREAT, | |
| 174 READ_WRITE_ALL_FILE_FLAGS)); | |
| 175 | |
| 176 if (file_descriptor.get() < 0) { | |
| 177 DLOG(ERROR) << "error openning the file"; | |
| 178 return false; | |
| 179 } | |
| 180 | |
| 181 fchmod(file_descriptor.get(), READ_WRITE_ALL_FILE_FLAGS); | |
| 182 // Grab a lock to avoid chrome truncating the file | |
| 183 // underneath us. Keep the file locked as briefly as possible. | |
| 184 // Freeing file_descriptor will close the file and and remove the lock. | |
| 185 if (HANDLE_EINTR(flock(file_descriptor.get(), LOCK_EX)) < 0) { | |
| 186 DLOG(ERROR) << "error locking" << filename << " : " << errno; | |
| 187 return false; | |
| 188 } | |
| 189 | |
| 190 std::string msg = sample.ToString(); | |
| 191 int32 size = msg.length() + sizeof(int32); | |
| 192 if (size > kMessageMaxLength) { | |
| 193 DLOG(ERROR) << "cannot write message: too long"; | |
| 194 return false; | |
| 195 } | |
| 196 | |
| 197 // The file containing the metrics samples will only be read by programs on | |
| 198 // the same device so we do not check endianness. | |
| 199 if (base::WriteFileDescriptor(file_descriptor.get(), | |
| 200 reinterpret_cast<char*>(&size), | |
| 201 sizeof(size)) != sizeof(size)) { | |
| 202 DLOG(ERROR) << "error writing message length " << errno; | |
| 203 return false; | |
| 204 } | |
| 205 | |
| 206 if (base::WriteFileDescriptor( | |
| 207 file_descriptor.get(), msg.c_str(), msg.length()) != | |
| 208 static_cast<int>(msg.length())) { | |
| 209 DLOG(ERROR) << "error writing message" << errno; | |
| 210 return false; | |
| 211 } | |
| 212 | |
| 213 return true; | |
| 214 } | |
| 215 | |
| 216 } // namespace metrics | |
| OLD | NEW |