1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "EventLog.h"
#include <stdio.h>
#include "mozilla/UniquePtr.h"
// This is an easy way to expose `MOZ_APP_DISPLAYNAME` to Rust code.
const wchar_t* gWinEventLogSourceName =
L"" MOZ_APP_DISPLAYNAME " Default Browser Agent";
static void WriteEventLogErrorBuffer(const wchar_t* buffer, DWORD eventId) {
HANDLE source = RegisterEventSourceW(nullptr, gWinEventLogSourceName);
if (!source) {
// Not much we can do about this.
return;
}
const wchar_t* stringsArray[] = {buffer};
ReportEventW(source, EVENTLOG_ERROR_TYPE, 0, eventId, nullptr, 1, 0,
stringsArray, nullptr);
DeregisterEventSource(source);
}
void WriteEventLogHresult(HRESULT hr, const char* sourceFile, int sourceLine) {
const wchar_t* format = L"0x%X in %S:%d";
int bufferSize = _scwprintf(format, hr, sourceFile, sourceLine);
++bufferSize; // Extra character for terminating null
mozilla::UniquePtr<wchar_t[]> errorStr =
mozilla::MakeUnique<wchar_t[]>(bufferSize);
_snwprintf_s(errorStr.get(), bufferSize, _TRUNCATE, format, hr, sourceFile,
sourceLine);
WriteEventLogErrorBuffer(errorStr.get(), hr);
}
void WriteEventLogErrorMessage(const wchar_t* messageFormat,
const char* sourceFile, int sourceLine, ...) {
// First assemble the passed message
va_list ap;
va_start(ap, sourceLine);
int bufferSize = _vscwprintf(messageFormat, ap);
++bufferSize; // Extra character for terminating null
va_end(ap);
mozilla::UniquePtr<wchar_t[]> message =
mozilla::MakeUnique<wchar_t[]>(bufferSize);
va_start(ap, sourceLine);
vswprintf(message.get(), bufferSize, messageFormat, ap);
va_end(ap);
// Next, assemble the complete error message to print
const wchar_t* errorFormat = L"Error: %s (%S:%d)";
bufferSize = _scwprintf(errorFormat, message.get(), sourceFile, sourceLine);
++bufferSize; // Extra character for terminating null
mozilla::UniquePtr<wchar_t[]> errorStr =
mozilla::MakeUnique<wchar_t[]>(bufferSize);
_snwprintf_s(errorStr.get(), bufferSize, _TRUNCATE, errorFormat,
message.get(), sourceFile, sourceLine);
WriteEventLogErrorBuffer(errorStr.get(), 0);
}
|
