crash_upload_list_win.cc revision 5821806d5e7f356e8fa4b058a389a808ea183019
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)
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stringprintf.h"
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/string_util.h"
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/sys_string_conversions.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)CrashUploadListWin::CrashUploadListWin(Delegate* delegate)
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    : CrashUploadList(delegate) {}
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void CrashUploadListWin::LoadCrashList() {
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::vector<uint8> buffer(1024);
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  HANDLE event_log = OpenEventLog(NULL, L"Application");
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (event_log) {
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    while (true) {
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DWORD bytes_read;
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DWORD bytes_needed;
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      BOOL success =
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          ReadEventLog(event_log,
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       EVENTLOG_SEQUENTIAL_READ | EVENTLOG_BACKWARDS_READ,
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       0,
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       &buffer[0],
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       buffer.size(),
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       &bytes_read,
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       &bytes_needed);
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (success) {
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        DWORD record_offset = 0;
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // The ReadEventLog() API docs imply, but do not explicitly state that
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // partial records will not be returned. Use DCHECK() to affirm this.
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        while (record_offset < bytes_read) {
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          DCHECK(record_offset + sizeof(EVENTLOGRECORD) <= bytes_read);
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          EVENTLOGRECORD* record = (EVENTLOGRECORD*)&buffer[record_offset];
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          DCHECK(record_offset + record->Length <= bytes_read);
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          if (IsPossibleCrashLogRecord(record))
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)            ProcessPossibleCrashLogRecord(record);
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          record_offset += record->Length;
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        }
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      } else if (GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Resize buffer to the required minimum size.
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        buffer.resize(bytes_needed);
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      } else {
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Stop on any other error, including the expected case
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // of ERROR_HANDLE_EOF.
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        DCHECK(GetLastError() == ERROR_HANDLE_EOF);
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        break;
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    CloseEventLog(event_log);
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool CrashUploadListWin::IsPossibleCrashLogRecord(
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    EVENTLOGRECORD* record) const {
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  LPWSTR provider_name = (LPWSTR)((uint8*)record + sizeof(EVENTLOGRECORD));
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return !wcscmp(L"Chrome", provider_name) &&
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      record->EventType == EVENTLOG_INFORMATION_TYPE &&
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      record->NumStrings >= 1;
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void CrashUploadListWin::ProcessPossibleCrashLogRecord(EVENTLOGRECORD* record) {
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Add the crash if the message matches the expected pattern.
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const std::wstring pattern_prefix(L"Id=");
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const std::wstring pattern_suffix(L".");
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::wstring message((LPWSTR)((uint8*)record + record->StringOffset));
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  size_t start_index = message.find(pattern_prefix);
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (start_index != std::wstring::npos) {
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    start_index += pattern_prefix.size();
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    size_t end_index = message.find(pattern_suffix, start_index);
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (end_index != std::wstring::npos) {
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      std::wstring crash_id =
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          message.substr(start_index, end_index - start_index);
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      crashes().push_back(
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          CrashInfo(base::SysWideToUTF8(crash_id),
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                    base::Time::FromDoubleT(record->TimeGenerated)));
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
81