GP-856: Fixed tests for dbgeng, added tests for dbgmodel

This commit is contained in:
Dan 2021-04-20 16:08:07 -04:00
parent a61c2e1400
commit f077adfffb
76 changed files with 917 additions and 225 deletions

View file

@ -971,4 +971,14 @@ public class WrappedDbgModel
return DebugValueType.INVALID;
}
@Override
public int getCurrentScopeFrameIndex() {
return client.getSymbols().getCurrentScopeFrameIndex();
}
@Override
public void setCurrentScopeFrameIndex(int index) {
client.getSymbols().setCurrentScopeFrameIndex(index);
}
}

View file

@ -57,7 +57,7 @@ public class KeyEnumeratorImpl implements KeyEnumeratorInternal {
PointerByReference ppValue = new PointerByReference();
PointerByReference ppMetaData = new PointerByReference();
HRESULT hr = jnaData.GetNext(bref, ppValue, ppMetaData);
if (hr.equals(COMUtilsExtra.E_BOUNDS)) {
if (hr.equals(COMUtilsExtra.E_BOUNDS) || hr.equals(COMUtilsExtra.E_FAIL)) {
//System.err.println("ret null");
return null;
}

View file

@ -906,6 +906,10 @@ public class ModelObjectImpl implements ModelObjectInternal {
String valueString = map.get("BaseAddress").getValueString();
return valueString;
}
if (map.containsKey("UniqueID") && map.containsKey("Id")) {
String valueString = map.get("Id").getValueString();
return valueString;
}
return key;
}

View file

@ -18,18 +18,22 @@ package agent.dbgmodel.model.impl;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.RejectedExecutionException;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.jdom.JDOMException;
import agent.dbgeng.manager.impl.DbgManagerImpl;
import agent.dbgeng.manager.impl.*;
import agent.dbgeng.model.AbstractDbgModel;
import agent.dbgeng.model.iface2.DbgModelTargetObject;
import agent.dbgeng.model.iface2.DbgModelTargetSession;
import agent.dbgmodel.manager.DbgManager2Impl;
import ghidra.async.AsyncUtils;
import ghidra.dbg.DebuggerModelClosedReason;
import ghidra.dbg.agent.AbstractTargetObject;
import ghidra.dbg.agent.AbstractTargetObject.ProxyFactory;
import ghidra.dbg.agent.SpiTargetObject;
import ghidra.dbg.error.DebuggerModelTerminatingException;
import ghidra.dbg.target.TargetObject;
import ghidra.dbg.target.schema.TargetObjectSchema;
import ghidra.dbg.target.schema.XmlSchemaContext;
@ -137,6 +141,10 @@ public class DbgModel2Impl extends AbstractDbgModel
terminate();
return super.close();
}
catch (RejectedExecutionException e) {
reportError(this, "Model is already closing", e);
return AsyncUtils.NIL;
}
catch (Throwable t) {
return CompletableFuture.failedFuture(t);
}
@ -154,6 +162,14 @@ public class DbgModel2Impl extends AbstractDbgModel
return;
}
objectMap.put(object, modelObject);
if (object instanceof DbgProcessImpl) {
DbgProcessImpl impl = (DbgProcessImpl) object;
objectMap.put(impl.getId(), modelObject);
}
if (object instanceof DbgThreadImpl) {
DbgThreadImpl impl = (DbgThreadImpl) object;
objectMap.put(impl.getId(), modelObject);
}
}
@Override
@ -161,4 +177,15 @@ public class DbgModel2Impl extends AbstractDbgModel
return objectMap.get(object);
}
@Override
public <T> CompletableFuture<T> gateFuture(CompletableFuture<T> future) {
return super.gateFuture(future).exceptionally(ex -> {
for (Throwable cause = ex; cause != null; cause = cause.getCause()) {
if (cause instanceof RejectedExecutionException) {
throw new DebuggerModelTerminatingException("dbgeng is terminating", ex);
}
}
return ExceptionUtils.rethrow(ex);
});
}
}

View file

@ -46,7 +46,7 @@ public class DbgModel2TargetAvailableImpl extends DbgModel2TargetObjectImpl
this.name = name;
this.changeAttributes(List.of(), List.of(), Map.of(//
PID_ATTRIBUTE_NAME, pid, //
PID_ATTRIBUTE_NAME, (long) pid, //
DISPLAY_ATTRIBUTE_NAME, keyAttachable(pid) + " : " + name.trim() //
), "Initialized");
}
@ -56,7 +56,7 @@ public class DbgModel2TargetAvailableImpl extends DbgModel2TargetObjectImpl
this.pid = pid;
this.changeAttributes(List.of(), List.of(), Map.of(//
PID_ATTRIBUTE_NAME, pid, //
PID_ATTRIBUTE_NAME, (long) pid, //
DISPLAY_ATTRIBUTE_NAME, keyAttachable(pid) //
), "Initialized");
}

View file

@ -34,8 +34,8 @@ import ghidra.async.AsyncUtils;
import ghidra.dbg.DebuggerModelListener;
import ghidra.dbg.agent.DefaultTargetObject;
import ghidra.dbg.target.*;
import ghidra.dbg.target.TargetBreakpointSpecContainer.TargetBreakpointKindSet;
import ghidra.dbg.target.TargetBreakpointSpec.TargetBreakpointKind;
import ghidra.dbg.target.TargetBreakpointSpecContainer.TargetBreakpointKindSet;
import ghidra.dbg.target.TargetExecutionStateful.TargetExecutionState;
import ghidra.dbg.target.schema.TargetObjectSchema;
import ghidra.dbg.util.PathUtils;
@ -247,6 +247,7 @@ public class DbgModel2TargetObjectImpl extends DefaultTargetObject<TargetObject,
attrs.put(TargetEnvironment.ARCH_ATTRIBUTE_NAME, "x86_64");
attrs.put(TargetEnvironment.DEBUGGER_ATTRIBUTE_NAME, "dbgeng");
attrs.put(TargetEnvironment.OS_ATTRIBUTE_NAME, "Windows");
attrs.put(TargetEnvironment.ENDIAN_ATTRIBUTE_NAME, "little");
}
if (proxy instanceof TargetModule) {
//attrs.put(TargetObject.ORDER_ATTRIBUTE_NAME,
@ -293,6 +294,11 @@ public class DbgModel2TargetObjectImpl extends DefaultTargetObject<TargetObject,
@Override
public CompletableFuture<?> fetchChild(final String key) {
/* Would like to do this, but has some very bad effects
return getModel().gateFuture(doFetchChild(key));
}
public CompletableFuture<?> doFetchChild(final String key) {
*/
synchronized (elements) {
if (key.startsWith("[") && key.endsWith("]")) {
String trimKey = key.substring(1, key.length() - 1);
@ -350,7 +356,7 @@ public class DbgModel2TargetObjectImpl extends DefaultTargetObject<TargetObject,
@Override
public void removeListener(DebuggerModelListener l) {
listeners.clear();
listeners.remove(l);
}
@Override

View file

@ -19,6 +19,7 @@ import agent.dbgeng.model.iface2.DbgModelTargetObject;
public interface DbgModel2TargetProxy extends DbgModelTargetObject {
@Override
public DelegateDbgModel2TargetObject getDelegate();
}

View file

@ -26,6 +26,7 @@ import agent.dbgeng.model.iface1.DbgModelSelectableObject;
import agent.dbgeng.model.iface1.DbgModelTargetExecutionStateful;
import agent.dbgeng.model.iface2.*;
import agent.dbgeng.model.impl.DbgModelTargetConnectorContainerImpl;
import agent.dbgeng.model.impl.DbgModelTargetProcessImpl;
import agent.dbgmodel.dbgmodel.main.ModelObject;
import agent.dbgmodel.manager.DbgManager2Impl;
import ghidra.async.AsyncUtils;
@ -146,6 +147,11 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
public void objectSelected(Object object) {
List<String> objPath = findObject(object);
TargetObject obj = getModel().getModelObject(objPath);
if (obj instanceof DbgModelSelectableObject) {
setFocus((DbgModelSelectableObject) obj);
}
/*
getModel().fetchModelValue(objPath, true).thenAccept(obj -> {
if (obj instanceof DbgModelSelectableObject) {
setFocus((DbgModelSelectableObject) obj);
@ -154,6 +160,7 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
Msg.error("Could not set focus on selected object: " + PathUtils.toString(objPath), ex);
return null;
});
*/
}
@Override
@ -262,6 +269,8 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
}
DbgModel2TargetProxy proxy = (DbgModel2TargetProxy) pobj;
DelegateDbgModel2TargetObject delegate = proxy.getDelegate();
Map<String, ? extends TargetObject> existingElements =
delegate.getCachedElements();
xpath.add(0, "Debugger");
DbgManager2Impl manager = (DbgManager2Impl) getManager();
@ -270,9 +279,19 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
String searchKey = obj.getSearchKey();
if (searchKey.equals(info.toString())) {
String elKey = PathUtils.makeKey(searchKey);
DbgModel2TargetProxy proxyElement =
(DbgModel2TargetProxy) DelegateDbgModel2TargetObject
.makeProxy(delegate.getModel(), delegate, elKey, obj);
DbgModel2TargetProxy proxyElement;
if (existingElements.containsKey(searchKey)) {
proxyElement = (DbgModel2TargetProxy) existingElements.get(searchKey);
DelegateDbgModel2TargetObject elementDelegate = proxyElement.getDelegate();
elementDelegate.setModelObject(obj);
}
else {
proxyElement = (DbgModel2TargetProxy) DelegateDbgModel2TargetObject
.makeProxy((DbgModel2Impl) proxy.getModel(), proxy, elKey, obj);
}
//DbgModel2TargetProxy proxyElement =
// (DbgModel2TargetProxy) DelegateDbgModel2TargetObject
// .makeProxy(delegate.getModel(), delegate, elKey, obj);
delegate.changeElements(List.of(), List.of(proxyElement), "Created");
seq.exit(proxyElement);
}
@ -280,6 +299,27 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
}).finish();
}
private CompletableFuture<Void> getObjectAndRemove(Object object,
List<String> ext, Object info) {
List<String> objPath = findObject(object);
if (objPath == null) {
return CompletableFuture.completedFuture(null);
}
List<String> xpath = new ArrayList<>();
xpath.addAll(objPath);
xpath.addAll(ext);
return AsyncUtils.sequence(TypeSpec.cls(Void.class)).then(seq -> {
getModel().fetchModelObject(xpath).handle(seq::next);
}, TypeSpec.cls(TargetObject.class)).then((pobj, seq) -> {
if (pobj == null) {
return;
}
DbgModel2TargetProxy proxy = (DbgModel2TargetProxy) pobj;
DelegateDbgModel2TargetObject delegate = proxy.getDelegate();
delegate.changeElements(List.of(info.toString()), List.of(), "Deleted");
}).finish();
}
@Override
public void sessionRemoved(DebugSessionId sessionId, DbgCause cause) {
getObject(sessionId);
@ -287,23 +327,44 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
@Override
public void processRemoved(DebugProcessId processId, DbgCause cause) {
DbgModelTargetProcess process = (DbgModelTargetProcess) getObject(processId);
if (process == null) {
return;
getObject(processId).thenAccept(object -> {
if (object == null) {
return;
}
DbgModelTargetProcess process = (DbgModelTargetProcess) object.getProxy();
process.setExecutionState(TargetExecutionState.INACTIVE, "Detached");
DbgProcess proc = process.getProcess();
getListeners().fire.event(getProxy(), null, TargetEventType.PROCESS_EXITED,
"Process " + proc.getId() + " exited code=" + proc.getExitCode(), List.of(process));
});
}
@Override
public void processExited(DbgProcess proc, DbgCause cause) {
DbgModelTargetProcess targetProcess =
(DbgModelTargetProcess) getModel().getModelObject(proc);
if (targetProcess != null) {
targetProcess.changeAttributes(List.of(), Map.of( //
TargetExecutionStateful.STATE_ATTRIBUTE_NAME, TargetExecutionState.TERMINATED, //
DbgModelTargetProcessImpl.EXIT_CODE_ATTRIBUTE_NAME, proc.getExitCode() //
), "Exited");
getListeners().fire.event(targetProcess.getProxy(), null,
TargetEventType.PROCESS_EXITED,
"Process " + proc.getId() + " exited code=" + proc.getExitCode(),
List.of(getProxy()));
}
DbgProcess proc = process.getProcess();
getListeners().fire.event(getProxy(), null, TargetEventType.PROCESS_EXITED,
"Process " + proc.getId() + " exited code=" + proc.getExitCode(), List.of(process));
}
@Override
public void threadExited(DebugThreadId threadId, DbgProcess process, DbgCause cause) {
DbgModelTargetThread targetThread = (DbgModelTargetThread) getObject(threadId);
if (targetThread == null) {
return;
}
getListeners().fire.event(getProxy(), targetThread, TargetEventType.THREAD_EXITED,
"Thread " + threadId + " exited", List.of(targetThread));
getObject(threadId).thenAccept(thread -> {
if (thread == null) {
return;
}
DbgModelTargetThread targetThread = (DbgModelTargetThread) thread.getProxy();
getListeners().fire.event(getProxy(), targetThread, TargetEventType.THREAD_EXITED,
"Thread " + threadId + " exited", List.of(targetThread));
});
}
@Override
@ -321,22 +382,37 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
TargetEventType eventType = getEventType(state, cause, reason);
getListeners().fire.event(getProxy(), targetThread, eventType,
"Thread " + thread.getId() + " state changed", List.of(targetThread));
targetThread.threadStateChangedSpecific(state, reason);
});
}
private CompletableFuture<DbgModelTargetObject> stateChanged(Object object, DbgState state,
String reason) {
List<String> objPath = findObject(object);
DbgModelTargetObject obj = (DbgModelTargetObject) getModel().getModelObject(objPath);
if (obj instanceof DbgModelTargetExecutionStateful) {
DbgModelTargetExecutionStateful stateful =
(DbgModelTargetExecutionStateful) obj;
TargetExecutionState execState = stateful.convertState(state);
stateful.setExecutionState(execState, reason);
}
return CompletableFuture.completedFuture(obj);
/*
return AsyncUtils.sequence(TypeSpec.cls(DbgModelTargetObject.class)).then(seq -> {
getModel().fetchModelValue(objPath).handle(seq::next);
}, TypeSpec.cls(Object.class)).then((obj, seq) -> {
if (obj instanceof DbgModelTargetExecutionStateful) {
DbgModelTargetExecutionStateful stateful = (DbgModelTargetExecutionStateful) obj;
TargetExecutionState execState = stateful.convertState(state);
stateful.setExecutionState(execState, reason);
}
seq.exit((DbgModelTargetObject) obj);
}).finish();
}, TypeSpec.cls(Object.class))
.then((obj, seq) -> {
// This is quite possibly redundant
if (obj instanceof DbgModelTargetExecutionStateful) {
DbgModelTargetExecutionStateful stateful =
(DbgModelTargetExecutionStateful) obj;
TargetExecutionState execState = stateful.convertState(state);
stateful.setExecutionState(execState, reason);
}
seq.exit((DbgModelTargetObject) obj);
})
.finish();
*/
}
@Override
@ -356,9 +432,8 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
@Override
public void breakpointDeleted(DbgBreakpointInfo info, DbgCause cause) {
int id = info.getDebugBreakpoint().getId();
bptInfoMap.remove(id);
getObjectRevisited(info.getProc(), List.of("Debug", "Breakpoints"), info);
bptInfoMap.remove((int) info.getNumber());
getObjectAndRemove(info.getProc(), List.of("Debug", "Breakpoints"), info);
}
@Override
@ -372,8 +447,8 @@ public class DbgModel2TargetRootImpl extends DbgModel2DefaultTargetModelRoot
return;
}
DbgModelTargetThread targetThread = getParentProcess().getThreads()
.getTargetThread(getManager().getEventThread());
DbgThread thread = info.getEventThread();
TargetObject targetThread = getModel().getModelObject(thread);
listeners.fire.breakpointHit(bpt.getParent(), targetThread, null, bpt, bpt);
bpt.breakpointHit();
});

View file

@ -99,6 +99,8 @@ public class DelegateDbgModel2TargetObject extends DbgModel2TargetObjectImpl imp
return DbgModelTargetRegisterBank.class;
case "TTD":
return DbgModelTargetTTD.class;
case "Debug":
return DbgModelTargetDebugContainer.class;
}
if (parentName != null) {
switch (parentName) {
@ -287,6 +289,7 @@ public class DelegateDbgModel2TargetObject extends DbgModel2TargetObjectImpl imp
requestAttributes(false);
return;
}
/*
if (proxy instanceof DbgModelTargetRegisterBank) {
requestAttributes(false).thenAccept(__ -> {
DbgModelTargetRegisterBank bank = (DbgModelTargetRegisterBank) proxy;
@ -296,6 +299,7 @@ public class DelegateDbgModel2TargetObject extends DbgModel2TargetObjectImpl imp
});
return;
}
*/
if (proxy instanceof DbgModelTargetProcessContainer || //
proxy instanceof DbgModelTargetThreadContainer || //
@ -323,6 +327,7 @@ public class DelegateDbgModel2TargetObject extends DbgModel2TargetObjectImpl imp
requestAttributes(false);
return;
}
/*
if (proxy instanceof DbgModelTargetRegisterBank) {
requestAttributes(false).thenAccept(__ -> {
DbgModelTargetRegisterBank bank = (DbgModelTargetRegisterBank) proxy;
@ -332,11 +337,16 @@ public class DelegateDbgModel2TargetObject extends DbgModel2TargetObjectImpl imp
});
return;
}
*/
if (proxy instanceof DbgModelTargetRegister || //
proxy instanceof DbgModelTargetStackFrame) {
requestAttributes(false);
return;
}
if (proxy.getName().equals("Debug")) {
requestAttributes(false);
return;
}
}
public void onRunning() {

View file

@ -5,6 +5,7 @@
<interface name="Attacher" />
<interface name="EventScope" />
<interface name="Launcher" />
<interface name="ActiveScope" />
<interface name="FocusScope" />
<interface name="Aggregate" />
<element schema="VOID" />
@ -153,7 +154,7 @@
<schema name="Available" elementResync="NEVER" attributeResync="NEVER">
<interface name="Attachable" />
<element schema="VOID" />
<attribute name="_pid" schema="INT" hidden="yes" />
<attribute name="_pid" schema="LONG" hidden="yes" required="yes"/>
<attribute name="_modified" schema="BOOL" hidden="yes" />
<attribute name="_display" schema="STRING" hidden="yes" />
<attribute name="_kind" schema="STRING" fixed="yes" hidden="yes" />
@ -169,7 +170,7 @@
<element schema="VOID" />
<attribute name="_os" schema="STRING" hidden="yes" />
<attribute name="_debugger" schema="STRING" hidden="yes" />
<attribute name="_endian" schema="STRING" hidden="yes" />
<!-- attribute name="_endian" schema="STRING" hidden="yes" /-->
<attribute name="_arch" schema="STRING" hidden="yes" />
<attribute name="_modified" schema="BOOL" hidden="yes" />
<attribute name="_display" schema="STRING" hidden="yes" />
@ -285,6 +286,7 @@
<attribute schema="ANY" />
</schema>
<schema name="DebugContainer" canonical="yes" elementResync="NEVER" attributeResync="NEVER">
<interface name="Aggregate" />
<element schema="OBJECT" />
<attribute name="_modified" schema="BOOL" hidden="yes" />
<attribute name="_display" schema="STRING" hidden="yes" />
@ -515,7 +517,7 @@
<attribute name="Disposition" schema="OBJECT" />
<attribute name="Pending" schema="OBJECT" />
<attribute name="Times" schema="OBJECT" />
<attribute schema="VOID" />
<attribute schema="ANY" />
</schema>
<schema name="StackFrame" elementResync="NEVER" attributeResync="NEVER">
<interface name="StackFrame" />

View file

@ -16,8 +16,16 @@
package agent.dbgmodel.model.invm;
import agent.dbgeng.model.AbstractModelForDbgengBreakpointsTest;
import ghidra.dbg.util.PathPattern;
import ghidra.dbg.util.PathUtils;
public class InVmModelForDbgmodelBreakpointsTest extends AbstractModelForDbgengBreakpointsTest {
@Override
protected PathPattern getBreakPattern() {
return new PathPattern(PathUtils.parse("Sessions[0x0].Processes[].Debug.Breakpoints[]"));
}
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();

View file

@ -16,8 +16,17 @@
package agent.dbgmodel.model.invm;
import agent.dbgeng.model.AbstractModelForDbgengFrameActivationTest;
import ghidra.dbg.util.PathPattern;
import ghidra.dbg.util.PathUtils;
public class InVmModelForDbgmodelFrameActivationTest
extends AbstractModelForDbgengFrameActivationTest {
protected PathPattern getStackPattern() {
return new PathPattern(
PathUtils.parse("Sessions[0x0].Processes[].Threads[].Stack.Frames[]"));
}
public class InVmModelForDbgmodelFrameFocusTest extends AbstractModelForDbgengFrameActivationTest {
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();

View file

@ -15,21 +15,74 @@
*/
package agent.dbgmodel.model.invm;
import static org.junit.Assert.assertNull;
import static org.junit.Assume.assumeTrue;
import java.util.List;
import org.junit.Ignore;
import org.junit.Test;
import agent.dbgeng.model.AbstractModelForDbgengInterpreterTest;
import agent.dbgeng.model.WindowsSpecimen;
import agent.dbgeng.model.iface2.DbgModelTargetProcess;
import ghidra.dbg.target.TargetInterpreter;
import ghidra.dbg.target.TargetProcess;
import ghidra.dbg.test.AbstractDebuggerModelTest;
import ghidra.dbg.test.ProvidesTargetViaLaunchSpecimen;
import ghidra.dbg.util.PathUtils;
public class InVmModelForDbgmodelInterpreterTest extends AbstractModelForDbgengInterpreterTest {
public class InVmModelForDbgmodelInterpreterTest extends AbstractModelForDbgengInterpreterTest
implements ProvidesTargetViaLaunchSpecimen {
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
@Override
public AbstractDebuggerModelTest getTest() {
return this;
}
@Override
protected List<String> seedPath() {
return PathUtils.parse("");
}
@Override
public List<String> getExpectedInterpreterPath() {
return PathUtils.parse("Sessions[0x0]");
}
@Override
protected void ensureInterpreterAvailable() throws Throwable {
obtainTarget();
}
@Override
@Ignore
@Test
public void testAttachViaInterpreterShowsInProcessContainer() throws Throwable {
super.testAttachViaInterpreterShowsInProcessContainer();
}
@Override
@Test
public void testLaunchViaInterpreterShowsInProcessContainer() throws Throwable {
assumeTrue(m.hasProcessContainer());
m.build();
DbgModelTargetProcess initialTarget = (DbgModelTargetProcess) obtainTarget();
DebuggerTestSpecimen specimen = WindowsSpecimen.NOTEPAD;
assertNull(getProcessRunning(specimen, this));
TargetInterpreter interpreter = findInterpreter();
for (String line : specimen.getLaunchScript()) {
waitOn(interpreter.execute(line));
}
TargetProcess process = retryForProcessRunning(specimen, this);
initialTarget.detach();
runTestKillViaInterpreter(process, interpreter);
}
}

View file

@ -0,0 +1,62 @@
/* ###
* IP: GHIDRA
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package agent.dbgmodel.model.invm;
import java.util.List;
import agent.dbgeng.model.AbstractModelForDbgengProcessActivationTest;
import ghidra.dbg.target.TargetInterpreter;
import ghidra.dbg.target.TargetObject;
import ghidra.dbg.util.PathPattern;
import ghidra.dbg.util.PathUtils;
public class InVmModelForDbgmodelProcessActivationTest
extends AbstractModelForDbgengProcessActivationTest {
protected PathPattern getProcessPattern() {
return new PathPattern(PathUtils.parse("Sessions[0x0].Processes[]"));
}
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
@Override
public List<String> getExpectedSessionPath() {
return PathUtils.parse("Sessions[0x0]");
}
public String getIdFromCapture(String line) {
return "0x" + line.split("\\s+")[3];
}
@Override
protected void activateViaInterpreter(TargetObject obj, TargetInterpreter interpreter)
throws Throwable {
String processId = obj.getName();
processId = processId.substring(3, processId.length() - 1);
String output = waitOn(interpreter.executeCapture("|"));
String[] lines = output.split("\n");
for (String l : lines) {
if (l.contains(processId)) {
processId = l.split("\\s+")[1];
break;
}
}
waitOn(interpreter.execute("|" + processId + " s"));
}
}

View file

@ -1,25 +0,0 @@
/* ###
* IP: GHIDRA
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package agent.dbgmodel.model.invm;
import agent.dbgeng.model.AbstractModelForDbgengProcessActivationTest;
public class InVmModelForDbgmodelProcessFocusTest extends AbstractModelForDbgengProcessActivationTest {
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
}

View file

@ -15,6 +15,9 @@
*/
package agent.dbgmodel.model.invm;
import org.junit.Ignore;
import org.junit.Test;
import agent.dbgeng.model.AbstractModelForDbgengRootAttacherTest;
public class InVmModelForDbgmodelRootAttacherTest extends AbstractModelForDbgengRootAttacherTest {
@ -22,4 +25,12 @@ public class InVmModelForDbgmodelRootAttacherTest extends AbstractModelForDbgeng
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
@Override
@Ignore
@Test
// Takes forever - passes w/ OTE on Memory in tear down
public void testAttachByPidThenResumeInterrupt() throws Throwable {
super.testAttachByPidThenResumeInterrupt();
}
}

View file

@ -15,7 +15,14 @@
*/
package agent.dbgmodel.model.invm;
import java.util.*;
import agent.dbgeng.model.AbstractModelForDbgengScenarioMemoryTest;
import ghidra.dbg.target.TargetModule;
import ghidra.dbg.target.TargetProcess;
import ghidra.dbg.util.PathUtils;
import ghidra.program.model.address.Address;
import ghidra.program.model.address.AddressRange;
public class InVmModelForDbgmodelScenarioMemoryTest
extends AbstractModelForDbgengScenarioMemoryTest {
@ -23,4 +30,17 @@ public class InVmModelForDbgmodelScenarioMemoryTest
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
@Override
protected Address getAddressToWrite(TargetProcess process) throws Throwable {
// It seems this is the only test case that exercises module symbols.
List<String> modulePath = PathUtils.extend(process.getPath(),
PathUtils.parse("Modules"));
Map<List<String>, TargetModule> modules = m.findAll(TargetModule.class, modulePath, true);
Collection<TargetModule> values = modules.values();
TargetModule test = (TargetModule) values.toArray()[0];
AddressRange range =
(AddressRange) test.fetchAttribute(TargetModule.RANGE_ATTRIBUTE_NAME).get();
return range.getMinAddress().add(0x15000);
}
}

View file

@ -16,10 +16,20 @@
package agent.dbgmodel.model.invm;
import agent.dbgeng.model.AbstractModelForDbgengScenarioStackTest;
import ghidra.dbg.target.TargetProcess;
import ghidra.program.model.address.Address;
public class InVmModelForDbgmodelScenarioStackTest extends AbstractModelForDbgengScenarioStackTest {
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
@Override
protected void postLaunch(TargetProcess process) throws Throwable {
}
@Override
protected void validateFramePC(int index, Address pc) {
}
}

View file

@ -23,4 +23,5 @@ public class InVmModelForDbgmodelScenarioX64RegistersTest
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
}

View file

@ -20,7 +20,7 @@ import org.junit.Ignore;
import agent.dbgeng.model.AbstractModelForDbgengSessionActivationTest;
@Ignore("Don't know how to make multiple sessions")
public class InVmModelForDbgmodelSessionFocusTest extends AbstractModelForDbgengSessionActivationTest {
public class InVmModelForDbgmodelSessionActivationTest extends AbstractModelForDbgengSessionActivationTest {
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();

View file

@ -22,4 +22,5 @@ public class InVmModelForDbgmodelSteppableTest extends AbstractModelForDbgengSte
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
}

View file

@ -0,0 +1,63 @@
/* ###
* IP: GHIDRA
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package agent.dbgmodel.model.invm;
import java.util.List;
import agent.dbgeng.model.AbstractModelForDbgengThreadActivationTest;
import ghidra.dbg.target.TargetInterpreter;
import ghidra.dbg.target.TargetObject;
import ghidra.dbg.util.PathPattern;
import ghidra.dbg.util.PathUtils;
public class InVmModelForDbgmodelThreadActivationTest
extends AbstractModelForDbgengThreadActivationTest {
protected PathPattern getThreadPattern() {
return new PathPattern(PathUtils.parse("Sessions[0x0].Processes[].Threads[]"));
}
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
@Override
public List<String> getExpectedSessionPath() {
return PathUtils.parse("Sessions[0x0]");
}
public String getIdFromCapture(String line) {
return "0x" + line.split("\\s+")[3].split("\\.")[1];
}
@Override
protected void activateViaInterpreter(TargetObject obj, TargetInterpreter interpreter)
throws Throwable {
String threadId = obj.getName();
threadId = threadId.substring(3, threadId.length() - 1);
String output = waitOn(interpreter.executeCapture("~"));
String[] lines = output.split("\n");
for (String l : lines) {
if (l.contains(threadId)) {
threadId = l.split("\\s+")[1];
break;
}
}
waitOn(interpreter.execute("~" + threadId + " s"));
}
}

View file

@ -1,25 +0,0 @@
/* ###
* IP: GHIDRA
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package agent.dbgmodel.model.invm;
import agent.dbgeng.model.AbstractModelForDbgengThreadActivationTest;
public class InVmModelForDbgmodelThreadFocusTest extends AbstractModelForDbgengThreadActivationTest {
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
}

View file

@ -15,11 +15,45 @@
*/
package agent.dbgmodel.model.invm;
import java.util.List;
import java.util.Map;
import org.junit.Ignore;
import org.junit.Test;
import agent.dbgeng.model.AbstractModelForDbgengX64RegistersTest;
import ghidra.dbg.util.PathUtils;
public class InVmModelForDbgmodelX64RegistersTest extends AbstractModelForDbgengX64RegistersTest {
public final Map<String, byte[]> REG_VALSX = Map.ofEntries(
Map.entry("rax", arr("0123456789abcdef")),
Map.entry("rdx", arr("fedcba9876543210")));
@Override
public ModelHost modelHost() throws Throwable {
return new InVmDbgmodelModelHost();
}
@Override
public boolean isRegisterBankAlsoContainer() {
return false;
}
@Override
public List<String> getExpectedRegisterBankPath(List<String> threadPath) {
return PathUtils.extend(threadPath, List.of("Registers", "User"));
}
@Override
public Map<String, byte[]> getRegisterWrites() {
return REG_VALSX;
}
@Override
@Ignore
@Test
public void testRegistersHaveExpectedSizes() throws Throwable {
super.testRegistersHaveExpectedSizes();
}
}