/// <summary>
/// Enumerates modules in the target process.
/// </summary>
/// <returns>
/// A list of the modules in the target process.
/// </returns>
public IList <ModuleInfo> EnumerateModules()
{
if (modulesCache == null)
{
lock (this)
{
DkmNativeRuntimeInstance runtimeInstance = Process.GetRuntimeInstances().OfType <DkmNativeRuntimeInstance>().FirstOrDefault();
List <ModuleInfo> modules;
if (runtimeInstance != null)
{
modules = runtimeInstance.GetModuleInstances().Select(m =>
{
return(new ModuleInfo(this)
{
FileName = m.FullName,
ImageBase = m.BaseAddress,
FileSize = m.Size,
TimeStamp = (uint)Math.Round((DateTime.FromFileTime((long)m.TimeDateStamp) - new DateTime(1970, 1, 1)).TotalSeconds),
Version = ExtractVersion(m.Version?.FileVersionString),
});
}).ToList();
}
else
{
System.Diagnostics.Process diagProcess = System.Diagnostics.Process.GetProcessById(Process.LivePart.Id);
modules = new List <ModuleInfo>();
foreach (System.Diagnostics.ProcessModule module in diagProcess.Modules)
{
modules.Add(new ModuleInfo(this)
{
FileName = module.FileName,
ImageBase = (ulong)module.BaseAddress.ToInt64(),
FileSize = (uint)module.ModuleMemorySize,
TimeStamp = (uint)Math.Round((File.GetCreationTime(module.FileName) - new DateTime(1970, 1, 1)).TotalSeconds),
Version = new VersionInfo()
{
Major = module.FileVersionInfo.FileMajorPart,
Minor = module.FileVersionInfo.FileMinorPart,
Revision = module.FileVersionInfo.FileBuildPart,
Patch = module.FileVersionInfo.FilePrivatePart,
},
});
}
}
modulesCache = modules;
}
}
return(modulesCache);
}
public static void VscxOnRuntimeInstanceLoad(this DkmRuntimeInstance runtime)
{
if (runtime.TagValue != DkmRuntimeInstance.Tag.NativeRuntimeInstance)
{
return;
}
DkmNativeRuntimeInstance nativeRuntime = (DkmNativeRuntimeInstance)runtime;
bool isChrome = false;
bool isTestProcess = false;
// Check if the process is a chrome executable, and if so, attach a CallstackFilter to the DkmProcess.
if (ChromeUtility.IsChromeProcess(nativeRuntime.Process.Path))
{
isChrome = true;
}
#if DEBUG
string fileName = Path.GetFileName(nativeRuntime.Process.Path);
if (fileName.Equals("vistest.exe", StringComparison.CurrentCultureIgnoreCase))
{
isTestProcess = true;
}
#endif
if (isTestProcess || isChrome)
{
DkmProcess process = nativeRuntime.Process;
DebugProcessOptions options = DebugProcessOptions.Create(process.DebugLaunchSettings.OptionsString);
ProcessDebugOptionsDataItem optionsDataItem = new ProcessDebugOptionsDataItem(options);
process.SetDataItem(DkmDataCreationDisposition.CreateAlways, optionsDataItem);
if (isTestProcess || ShouldEnableChildDebugging(nativeRuntime.Process, options))
{
process.SetDataItem(DkmDataCreationDisposition.CreateAlways, new RuntimeBreakpointHandler());
process.SetDataItem(DkmDataCreationDisposition.CreateAlways, new AutoAttachToChildHandler());
}
}
}
void IDkmProcessExecutionNotification.OnProcessPause(DkmProcess process, DkmProcessExecutionCounters processCounters)
{
try
{
ulong?moduleBaseOpt = DebugHelpers.ReadPointerVariable(process, "nullcModuleStartAddress");
// Check if nullc is available
if (moduleBaseOpt == null)
{
return;
}
var processData = DebugHelpers.GetOrCreateDataItem <NullcRemoteProcessDataItem>(process);
if (processData.language == null)
{
processData.compilerId = new DkmCompilerId(DebugHelpers.NullcCompilerGuid, DebugHelpers.NullcLanguageGuid);
processData.language = DkmLanguage.Create("nullc", processData.compilerId);
}
// Create VM runtime and module
if (processData.vmRuntimeInstance == null)
{
DkmRuntimeInstanceId runtimeId = new DkmRuntimeInstanceId(DebugHelpers.NullcVmRuntimeGuid, 0);
processData.vmRuntimeInstance = DkmCustomRuntimeInstance.Create(process, runtimeId, null);
}
if (processData.vmModule == null)
{
DkmModuleId moduleId = new DkmModuleId(Guid.NewGuid(), DebugHelpers.NullcSymbolProviderGuid);
processData.vmModule = DkmModule.Create(moduleId, "nullc.vm.code", processData.compilerId, process.Connection, null);
}
if (processData.vmModuleInstance == null)
{
DkmDynamicSymbolFileId symbolFileId = DkmDynamicSymbolFileId.Create(DebugHelpers.NullcSymbolProviderGuid);
processData.vmModuleInstance = DkmCustomModuleInstance.Create("nullc_vm", "nullc.vm.code", 0, processData.vmRuntimeInstance, null, symbolFileId, DkmModuleFlags.None, DkmModuleMemoryLayout.Unknown, 0, 1, 0, "nullc vm code", false, null, null, null);
processData.vmModuleInstance.SetModule(processData.vmModule, true); // Can use reload?
}
ulong moduleBase = moduleBaseOpt.GetValueOrDefault(0);
uint moduleSize = (uint)(DebugHelpers.ReadPointerVariable(process, "nullcModuleEndAddress").GetValueOrDefault(0) - moduleBase);
// Create JiT runtime and module
if (moduleBase != 0 && moduleSize != 0)
{
if (processData.runtimeInstance == null && processData.nativeRuntimeInstance == null)
{
DkmRuntimeInstanceId runtimeId = new DkmRuntimeInstanceId(DebugHelpers.NullcRuntimeGuid, 0);
if (DebugHelpers.useNativeInterfaces)
{
processData.nativeRuntimeInstance = DebugHelpers.useDefaultRuntimeInstance ? process.GetNativeRuntimeInstance() : DkmNativeRuntimeInstance.Create(process, runtimeId, DkmRuntimeCapabilities.None, process.GetNativeRuntimeInstance(), null);
}
else
{
processData.runtimeInstance = DkmCustomRuntimeInstance.Create(process, runtimeId, null);
}
}
if (processData.module == null)
{
DkmModuleId moduleId = new DkmModuleId(Guid.NewGuid(), DebugHelpers.NullcSymbolProviderGuid);
processData.module = DkmModule.Create(moduleId, "nullc.embedded.code", processData.compilerId, process.Connection, null);
}
if (processData.moduleInstance == null && processData.nativeModuleInstance == null)
{
DkmDynamicSymbolFileId symbolFileId = DkmDynamicSymbolFileId.Create(DebugHelpers.NullcSymbolProviderGuid);
if (DebugHelpers.useNativeInterfaces)
{
processData.nativeModuleInstance = DkmNativeModuleInstance.Create("nullc", "nullc.embedded.code", 0, null, symbolFileId, DkmModuleFlags.None, DkmModuleMemoryLayout.Unknown, 1, "nullc embedded code", processData.nativeRuntimeInstance, moduleBase, moduleSize, Microsoft.VisualStudio.Debugger.Clr.DkmClrHeaderStatus.NativeBinary, false, null, null, null);
processData.nativeModuleInstance.SetModule(processData.module, true); // Can use reload?
}
else
{
processData.moduleInstance = DkmCustomModuleInstance.Create("nullc", "nullc.embedded.code", 0, processData.runtimeInstance, null, symbolFileId, DkmModuleFlags.None, DkmModuleMemoryLayout.Unknown, moduleBase, 1, moduleSize, "nullc embedded code", false, null, null, null);
processData.moduleInstance.SetModule(processData.module, true); // Can use reload?
}
}
}
// Update bytecode
var moduleBytecodeVersion = DebugHelpers.ReadPointerVariable(process, "nullcModuleBytecodeVersion").GetValueOrDefault(0);
if (processData.moduleBytecodeLocation != 0 && moduleBytecodeVersion != processData.moduleBytecodeVersion)
{
processData.moduleBytecodeLocation = 0;
processData.moduleBytecodeSize = 0;
processData.moduleBytecodeRaw = null;
processData.bytecode = null;
}
if (processData.moduleBytecodeLocation == 0)
{
processData.moduleBytecodeLocation = DebugHelpers.ReadPointerVariable(process, "nullcModuleBytecodeLocation").GetValueOrDefault(0);
processData.moduleBytecodeSize = DebugHelpers.ReadPointerVariable(process, "nullcModuleBytecodeSize").GetValueOrDefault(0);
processData.moduleBytecodeVersion = moduleBytecodeVersion;
if (processData.moduleBytecodeLocation != 0)
{
processData.moduleBytecodeRaw = new byte[processData.moduleBytecodeSize];
process.ReadMemory(processData.moduleBytecodeLocation, DkmReadMemoryFlags.None, processData.moduleBytecodeRaw);
processData.bytecode = new NullcBytecode();
processData.bytecode.ReadFrom(processData.moduleBytecodeRaw, DebugHelpers.Is64Bit(process));
// Notify local component about bytecode update
var message = DkmCustomMessage.Create(process.Connection, process, DebugHelpers.NullcReloadSymbolsMessageGuid, 1, null, null);
message.SendHigher();
}
}
}
catch (Exception ex)
{
Console.WriteLine("OnProcessPause failed with: " + ex.ToString());
}
}
