15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/crash_upload_list_win.h"
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h"
8868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h"
9c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/strings/sys_string_conversions.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
11eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen MurdochCrashUploadListWin::CrashUploadListWin(Delegate* delegate,
12eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch                                       const base::FilePath& upload_log_path)
13eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch    : CrashUploadList(delegate, upload_log_path) {}
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
15eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdochvoid CrashUploadListWin::LoadUploadList() {
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::vector<uint8> buffer(1024);
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  HANDLE event_log = OpenEventLog(NULL, L"Application");
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (event_log) {
191e9bf3e0803691d0a228da41fc608347b6db4340Torne (Richard Coles)    ClearUploads();
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    while (true) {
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DWORD bytes_read;
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DWORD bytes_needed;
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      BOOL success =
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          ReadEventLog(event_log,
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ,
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       0,
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       &buffer[0],
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       buffer.size(),
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       &bytes_read,
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       &bytes_needed);
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (success) {
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        DWORD record_offset = 0;
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // The ReadEventLog() API docs imply, but do not explicitly state that
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // partial records will not be returned. Use DCHECK() to affirm this.
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        while (record_offset < bytes_read) {
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          DCHECK(record_offset + sizeof(EVENTLOGRECORD) <= bytes_read);
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          EVENTLOGRECORD* record = (EVENTLOGRECORD*)&buffer[record_offset];
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          DCHECK(record_offset + record->Length <= bytes_read);
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          if (IsPossibleCrashLogRecord(record))
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)            ProcessPossibleCrashLogRecord(record);
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          record_offset += record->Length;
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        }
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      } else if (GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Resize buffer to the required minimum size.
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        buffer.resize(bytes_needed);
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      } else {
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Stop on any other error, including the expected case
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // of ERROR_HANDLE_EOF.
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        DCHECK(GetLastError() == ERROR_HANDLE_EOF);
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        break;
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    CloseEventLog(event_log);
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool CrashUploadListWin::IsPossibleCrashLogRecord(
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    EVENTLOGRECORD* record) const {
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LPWSTR provider_name = (LPWSTR)((uint8*)record + sizeof(EVENTLOGRECORD));
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return !wcscmp(L"Chrome", provider_name) &&
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      record->EventType == EVENTLOG_INFORMATION_TYPE &&
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      record->NumStrings >= 1;
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void CrashUploadListWin::ProcessPossibleCrashLogRecord(EVENTLOGRECORD* record) {
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Add the crash if the message matches the expected pattern.
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const std::wstring pattern_prefix(L"Id=");
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const std::wstring pattern_suffix(L".");
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring message((LPWSTR)((uint8*)record + record->StringOffset));
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  size_t start_index = message.find(pattern_prefix);
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (start_index != std::wstring::npos) {
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    start_index += pattern_prefix.size();
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    size_t end_index = message.find(pattern_suffix, start_index);
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (end_index != std::wstring::npos) {
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      std::wstring crash_id =
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          message.substr(start_index, end_index - start_index);
77eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch      AppendUploadInfo(
78eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch          UploadInfo(base::SysWideToUTF8(crash_id),
79eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch                     base::Time::FromDoubleT(record->TimeGenerated)));
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
83