Author: hbelusca
Date: Sun May 22 19:00:55 2016
New Revision: 71378
URL: http://svn.reactos.org/svn/reactos?rev=71378&view=rev
Log:
[EVENTVWR]
- Fix the flags used in LoadLibraryEx calls for loading DLLs as resource files.
- Use a nice flag (FORMAT_MESSAGE_MAX_WIDTH_MASK) in FormatMessage calls so
that no useless extra newlines are appended to the message strings that are
recovered.
This is especially useful when some strings as localized event log names are
retrieved to be displayed in the menu (amongst other things)...
This fixes CORE-8915 and fixes also the presentation of some event
descriptions.
- Fix BuildLogList.
Modified:
trunk/reactos/base/applications/mscutils/eventvwr/eventvwr.c
Modified: trunk/reactos/base/applications/mscutils/eventvwr/eventvwr.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/base/applications/mscutils/eventvwr/eventvwr.c?rev=71378&r1=71377&r2=71378&view=diff
==============================================================================
--- trunk/reactos/base/applications/mscutils/eventvwr/eventvwr.c
[iso-8859-1] (original)
+++ trunk/reactos/base/applications/mscutils/eventvwr/eventvwr.c
[iso-8859-1] Sun May 22 19:00:55 2016
@@ -83,8 +83,8 @@
LPWSTR lpSourceLogName = NULL;
LPWSTR lpComputerName = NULL;
-DWORD dwNumLogs = 0;
-LPWSTR* LogNames;
+DWORD dwNumLogs = 0;
+LPWSTR* LogNames = NULL;
/* Forward declarations of functions included in this code module: */
ATOM MyRegisterClass(HINSTANCE hInstance);
@@ -294,13 +294,13 @@
goto Quit;
hLibrary = LoadLibraryExW(szMessageDLL, NULL,
- DONT_RESOLVE_DLL_REFERENCES |
LOAD_LIBRARY_AS_DATAFILE);
+ LOAD_LIBRARY_AS_IMAGE_RESOURCE |
LOAD_LIBRARY_AS_DATAFILE);
if (hLibrary == NULL)
goto Quit;
- /* Retrieve the message string. */
+ /* Retrieve the message string without appending extra newlines */
if (FormatMessageW(FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_FROM_HMODULE |
- FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY,
+ FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY | FORMAT_MESSAGE_MAX_WIDTH_MASK,
hLibrary,
pevlr->EventCategory,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
@@ -395,7 +395,7 @@
while ((szDll != NULL) && (!Success))
{
hLibrary = LoadLibraryExW(szDll, NULL,
- DONT_RESOLVE_DLL_REFERENCES |
LOAD_LIBRARY_AS_DATAFILE);
+ LOAD_LIBRARY_AS_IMAGE_RESOURCE |
LOAD_LIBRARY_AS_DATAFILE);
if (hLibrary == NULL)
{
/* The DLL could not be loaded try the next one (if any) */
@@ -403,9 +403,9 @@
continue;
}
- /* Retrieve the message string. */
+ /* Retrieve the message string without appending extra newlines */
if (FormatMessageW(FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_FROM_HMODULE |
- FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY,
+ FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY | FORMAT_MESSAGE_MAX_WIDTH_MASK,
hLibrary,
pevlr->EventID,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
@@ -894,7 +894,7 @@
}
-VOID
+BOOL
GetDisplayNameFile(IN LPCWSTR lpLogName,
OUT PWCHAR lpModuleName)
{
@@ -908,7 +908,7 @@
KeyPath = HeapAlloc(GetProcessHeap(), 0, cbKeyPath);
if (!KeyPath)
{
- return;
+ return FALSE;
}
StringCbCopyW(KeyPath, cbKeyPath, EVENTLOG_BASE_KEY);
@@ -917,7 +917,7 @@
if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, KeyPath, 0, KEY_READ, &hKey) !=
ERROR_SUCCESS)
{
HeapFree(GetProcessHeap(), 0, KeyPath);
- return;
+ return FALSE;
}
cbData = sizeof(szModuleName);
@@ -928,6 +928,8 @@
RegCloseKey(hKey);
HeapFree(GetProcessHeap(), 0, KeyPath);
+
+ return TRUE;
}
@@ -970,86 +972,114 @@
BuildLogList(void)
{
HKEY hKey;
+ DWORD dwIndex;
+ DWORD dwMaxKeyLength;
+ LPWSTR LogName = NULL;
+ WCHAR szModuleName[MAX_PATH];
DWORD lpcName;
- DWORD dwIndex;
DWORD dwMessageID;
- DWORD dwMaxKeyLength;
- WCHAR szModuleName[MAX_PATH];
LPWSTR lpDisplayName;
HMODULE hLibrary = NULL;
+ /* Open the EventLog key */
if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, EVENTLOG_BASE_KEY, 0, KEY_READ,
&hKey) != ERROR_SUCCESS)
{
return;
}
+ /* Retrieve the number of event logs enumerated as registry keys */
if (RegQueryInfoKeyW(hKey, NULL, NULL, NULL, &dwNumLogs, &dwMaxKeyLength,
NULL, NULL, NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
{
RegCloseKey(hKey);
return;
}
-
if (!dwNumLogs)
{
RegCloseKey(hKey);
return;
}
- LogNames = HeapAlloc(GetProcessHeap(), 0, (dwNumLogs + 1) *
sizeof(LPWSTR));
-
- if (!LogNames)
+ /* Take the NULL terminator into account */
+ ++dwMaxKeyLength;
+
+ /* Allocate the temporary buffer */
+ LogName = HeapAlloc(GetProcessHeap(), 0, dwMaxKeyLength * sizeof(WCHAR));
+ if (!LogName)
{
RegCloseKey(hKey);
return;
}
+ /* Allocate the list of event logs (add 1 to dwNumLogs in order to have
one guard entry) */
+ LogNames = HeapAlloc(GetProcessHeap(), 0, (dwNumLogs + 1) *
sizeof(LPWSTR));
+ if (!LogNames)
+ {
+ HeapFree(GetProcessHeap(), 0, LogName);
+ RegCloseKey(hKey);
+ return;
+ }
+
+ /* Enumerate and retrieve each event log name */
for (dwIndex = 0; dwIndex < dwNumLogs; dwIndex++)
{
- LogNames[dwIndex] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
((dwMaxKeyLength + 1) * sizeof(WCHAR)));
-
- if (LogNames[dwIndex] != NULL)
- {
- lpcName = dwMaxKeyLength + 1;
-
- if (RegEnumKeyExW(hKey, dwIndex, LogNames[dwIndex], &lpcName,
NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
+ lpcName = dwMaxKeyLength;
+ if (RegEnumKeyExW(hKey, dwIndex, LogName, &lpcName, NULL, NULL, NULL,
NULL) != ERROR_SUCCESS)
+ {
+ LogNames[dwIndex] = NULL;
+ continue;
+ }
+
+ /* Take the NULL terminator into account */
+ ++lpcName;
+
+ /* Allocate the event log name string and copy it from the temporary
buffer */
+ LogNames[dwIndex] = HeapAlloc(GetProcessHeap(), 0, lpcName *
sizeof(WCHAR));
+ if (LogNames[dwIndex] == NULL)
+ continue;
+
+ StringCchCopyW(LogNames[dwIndex], lpcName, LogName);
+
+ /* Get the display name for the event log */
+ lpDisplayName = NULL;
+
+ ZeroMemory(szModuleName, sizeof(szModuleName));
+ if (GetDisplayNameFile(LogNames[dwIndex], szModuleName))
+ {
+ dwMessageID = GetDisplayNameID(LogNames[dwIndex]);
+
+ hLibrary = LoadLibraryExW(szModuleName, NULL,
LOAD_LIBRARY_AS_IMAGE_RESOURCE | LOAD_LIBRARY_AS_DATAFILE);
+ if (hLibrary != NULL)
{
- lpDisplayName = NULL;
-
- ZeroMemory(szModuleName, sizeof(szModuleName));
- GetDisplayNameFile(LogNames[dwIndex], szModuleName);
- dwMessageID = GetDisplayNameID(LogNames[dwIndex]);
-
- hLibrary = LoadLibraryExW(szModuleName, NULL,
DONT_RESOLVE_DLL_REFERENCES | LOAD_LIBRARY_AS_DATAFILE);
- if (hLibrary != NULL)
- {
- FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_HMODULE,
- hLibrary,
- dwMessageID,
- 0,
- (LPWSTR)&lpDisplayName,
- 0,
- NULL);
- FreeLibrary(hLibrary);
- }
-
- if (lpDisplayName)
- {
- InsertMenuW(hMainMenu, IDM_SAVE_EVENTLOG, MF_BYCOMMAND |
MF_STRING, ID_FIRST_LOG + dwIndex, lpDisplayName);
- }
- else
- {
- InsertMenuW(hMainMenu, IDM_SAVE_EVENTLOG, MF_BYCOMMAND |
MF_STRING, ID_FIRST_LOG + dwIndex, LogNames[dwIndex]);
- }
-
- /* Free the buffer allocated by FormatMessage */
- if (lpDisplayName)
- LocalFree(lpDisplayName);
+ /* Retrieve the message string without appending extra
newlines */
+ FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_HMODULE |
+ FORMAT_MESSAGE_IGNORE_INSERTS |
FORMAT_MESSAGE_MAX_WIDTH_MASK,
+ hLibrary,
+ dwMessageID,
+ MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+ (LPWSTR)&lpDisplayName,
+ 0,
+ NULL);
+ FreeLibrary(hLibrary);
}
}
+
+ if (lpDisplayName)
+ {
+ InsertMenuW(hMainMenu, IDM_SAVE_EVENTLOG, MF_BYCOMMAND |
MF_STRING, ID_FIRST_LOG + dwIndex, lpDisplayName);
+ }
+ else
+ {
+ InsertMenuW(hMainMenu, IDM_SAVE_EVENTLOG, MF_BYCOMMAND |
MF_STRING, ID_FIRST_LOG + dwIndex, LogNames[dwIndex]);
+ }
+
+ /* Free the buffer allocated by FormatMessage */
+ if (lpDisplayName)
+ LocalFree(lpDisplayName);
}
InsertMenuW(hMainMenu, IDM_SAVE_EVENTLOG, MF_BYCOMMAND | MF_SEPARATOR,
ID_FIRST_LOG + dwIndex + 1, NULL);
+ HeapFree(GetProcessHeap(), 0, LogName);
RegCloseKey(hKey);
return;
@@ -1079,7 +1109,7 @@
DeleteMenu(hMainMenu, ID_FIRST_LOG + dwIndex + 1, MF_BYCOMMAND);
HeapFree(GetProcessHeap(), 0, LogNames);
-
+ LogNames = NULL;
dwNumLogs = 0;
return;
