Skip to content

Commit dfb19ae

Browse files
authored
fix wrong use of WorkflowTraceUtil in two probes (#468)
1 parent 6076dbc commit dfb19ae

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

TLS-Server-Scanner/src/main/java/de/rub/nds/tlsscanner/serverscanner/probe/MacProbe.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
import de.rub.nds.tlsattacker.core.state.State;
3535
import de.rub.nds.tlsattacker.core.workflow.ParallelExecutor;
3636
import de.rub.nds.tlsattacker.core.workflow.WorkflowTrace;
37+
import de.rub.nds.tlsattacker.core.workflow.WorkflowTraceConfigurationUtil;
3738
import de.rub.nds.tlsattacker.core.workflow.WorkflowTraceResultUtil;
3839
import de.rub.nds.tlsattacker.core.workflow.action.GenericReceiveAction;
3940
import de.rub.nds.tlsattacker.core.workflow.action.ReceiveAction;
@@ -228,7 +229,7 @@ private WorkflowTrace getVerifyDataTrace(Config config, int xorPosition) {
228229
WorkflowTraceType.DYNAMIC_HANDSHAKE, RunningModeType.CLIENT);
229230
FinishedMessage lastSendMessage =
230231
(FinishedMessage)
231-
WorkflowTraceResultUtil.getLastSentMessage(
232+
WorkflowTraceConfigurationUtil.getLastStaticConfiguredSendMessage(
232233
trace, HandshakeMessageType.FINISHED);
233234
lastSendMessage.setVerifyData(Modifiable.xor(new byte[] {01}, xorPosition));
234235
return trace;

TLS-Server-Scanner/src/main/java/de/rub/nds/tlsscanner/serverscanner/probe/invalidcurve/trace/InvalidCurveWorkflowGenerator.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
import de.rub.nds.tlsattacker.core.protocol.message.extension.ExtensionMessage;
2222
import de.rub.nds.tlsattacker.core.protocol.message.extension.KeyShareExtensionMessage;
2323
import de.rub.nds.tlsattacker.core.workflow.WorkflowTrace;
24-
import de.rub.nds.tlsattacker.core.workflow.WorkflowTraceResultUtil;
24+
import de.rub.nds.tlsattacker.core.workflow.WorkflowTraceConfigurationUtil;
2525
import de.rub.nds.tlsattacker.core.workflow.action.ChangeDefaultPreMasterSecretAction;
2626
import de.rub.nds.tlsattacker.core.workflow.action.GenericReceiveAction;
2727
import de.rub.nds.tlsattacker.core.workflow.action.ReceiveAction;
@@ -72,7 +72,7 @@ private static WorkflowTrace prepareRegularTrace(
7272

7373
ClientHelloMessage clientHello =
7474
(ClientHelloMessage)
75-
WorkflowTraceResultUtil.getFirstSentMessage(
75+
WorkflowTraceConfigurationUtil.getFirstStaticConfiguredSendMessage(
7676
trace, HandshakeMessageType.CLIENT_HELLO);
7777
KeyShareExtensionMessage ksExt;
7878
for (ExtensionMessage ext : clientHello.getExtensions()) {
@@ -96,7 +96,7 @@ private static WorkflowTrace prepareRegularTrace(
9696

9797
ECDHClientKeyExchangeMessage message =
9898
(ECDHClientKeyExchangeMessage)
99-
WorkflowTraceResultUtil.getFirstSentMessage(
99+
WorkflowTraceConfigurationUtil.getFirstStaticConfiguredSendMessage(
100100
trace, HandshakeMessageType.CLIENT_KEY_EXCHANGE);
101101
message.setPublicKey(serializedPublicKey);
102102
message.prepareComputations();
@@ -155,7 +155,7 @@ private static WorkflowTrace prepareRenegotiationTrace(
155155
RunningModeType.CLIENT);
156156
ECDHClientKeyExchangeMessage message =
157157
(ECDHClientKeyExchangeMessage)
158-
WorkflowTraceResultUtil.getLastSentMessage(
158+
WorkflowTraceConfigurationUtil.getLastStaticConfiguredSendMessage(
159159
trace, HandshakeMessageType.CLIENT_KEY_EXCHANGE);
160160
message.setPublicKey(serializedPublicKey);
161161
message.prepareComputations();

0 commit comments

Comments
 (0)