123 lines
4.2 KiB
C#
123 lines
4.2 KiB
C#
using System;
|
|
using System.Collections;
|
|
using System.Diagnostics;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
|
|
|
|
namespace Microsoft.PowerShell.Commands.Diagnostics.Common
|
|
{
|
|
|
|
internal class CommonUtilities
|
|
{
|
|
//
|
|
// No-op private ctor to prevent the default ctor from getting generated.
|
|
// This class is intended to only have static members.
|
|
//
|
|
private CommonUtilities() {
|
|
}
|
|
|
|
//
|
|
// StringArrayToString helper converts a string array into a comma-separated string.
|
|
// Note this has only limited use, individual strings cannot have commas.
|
|
//
|
|
public static string StringArrayToString (IEnumerable input)
|
|
{
|
|
string ret = "";
|
|
foreach (string element in input)
|
|
{
|
|
ret += element + ", ";
|
|
}
|
|
|
|
ret = ret.TrimEnd();
|
|
ret = ret.TrimEnd(',');
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
const uint FORMAT_MESSAGE_ALLOCATE_BUFFER = 0x00000100;
|
|
const uint FORMAT_MESSAGE_IGNORE_INSERTS = 0x00000200;
|
|
const uint FORMAT_MESSAGE_FROM_SYSTEM = 0x00001000;
|
|
const uint LOAD_LIBRARY_AS_DATAFILE = 0x00000002;
|
|
const uint FORMAT_MESSAGE_FROM_HMODULE = 0x00000800;
|
|
|
|
[DllImport("kernel32.dll", SetLastError = true, CharSet = CharSet.Unicode)]
|
|
private static extern uint FormatMessage(uint dwFlags, IntPtr lpSource,
|
|
uint dwMessageId, uint dwLanguageId,
|
|
[MarshalAs(UnmanagedType.LPWStr)]
|
|
StringBuilder lpBuffer,
|
|
uint nSize, IntPtr Arguments);
|
|
|
|
[DllImport("kernel32.dll", SetLastError = true, CharSet = CharSet.Unicode)]
|
|
private static extern IntPtr LoadLibraryEx(
|
|
[MarshalAs(UnmanagedType.LPWStr)] string lpFileName,
|
|
IntPtr hFile,
|
|
uint dwFlags
|
|
);
|
|
|
|
[DllImport("kernel32.dll")]
|
|
private static extern bool FreeLibrary(IntPtr hModule);
|
|
|
|
|
|
[DllImport("kernel32.dll", EntryPoint = "GetUserDefaultLangID", CallingConvention = CallingConvention.Winapi, SetLastError = true)]
|
|
private static extern ushort GetUserDefaultLangID();
|
|
|
|
|
|
public static uint FormatMessageFromModule(uint lastError, string moduleName, out String msg)
|
|
{
|
|
Debug.Assert (!string.IsNullOrEmpty(moduleName));
|
|
|
|
uint formatError = 0;
|
|
msg = String.Empty;
|
|
IntPtr moduleHandle = IntPtr.Zero;
|
|
|
|
moduleHandle = LoadLibraryEx(moduleName, IntPtr.Zero, LOAD_LIBRARY_AS_DATAFILE);
|
|
if (moduleHandle == IntPtr.Zero)
|
|
{
|
|
return (uint)Marshal.GetLastWin32Error();
|
|
}
|
|
|
|
try
|
|
{
|
|
uint dwFormatFlags = FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_FROM_HMODULE;
|
|
uint LANGID = (uint)GetUserDefaultLangID();
|
|
uint langError = (uint)Marshal.GetLastWin32Error();
|
|
if (langError != 0)
|
|
{
|
|
LANGID = 0; // neutral
|
|
}
|
|
|
|
StringBuilder outStringBuilder = new StringBuilder(1024);
|
|
uint nChars = FormatMessage(dwFormatFlags,
|
|
moduleHandle,
|
|
lastError,
|
|
LANGID,
|
|
outStringBuilder,
|
|
(uint)outStringBuilder.Capacity,
|
|
IntPtr.Zero);
|
|
|
|
if (nChars == 0)
|
|
{
|
|
formatError = (uint)Marshal.GetLastWin32Error();
|
|
//Console.WriteLine("Win32FormatMessage", String.Format(null, "Error formatting message: {0}", formatError));
|
|
}
|
|
else
|
|
{
|
|
msg = outStringBuilder.ToString();
|
|
if (msg.EndsWith(Environment.NewLine, StringComparison.Ordinal))
|
|
{
|
|
msg = msg.Substring(0, msg.Length - 2);
|
|
}
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
FreeLibrary(moduleHandle);
|
|
}
|
|
return formatError;
|
|
}
|
|
|
|
}
|
|
}
|
|
|