add to tx hash sessions map in proper place

jrick · Apr 22, 2024 · 9deaf20 · 9deaf20
1 parent 50ab0a5
commit 9deaf20
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mixing/mixpool/mixpool.go b/mixing/mixpool/mixpool.go
@@ -627,7 +627,6 @@ Loop:
 			if r.CMs != nil {
 				r.CMs = append(r.CMs, msg)
 			}
-			p.sessionsByTxHash[msg.Mix.TxHash()] = ses
 		case *wire.MsgMixSecrets:
 			if r.RSs != nil {
 				r.RSs = append(r.RSs, msg)
@@ -1065,6 +1064,10 @@ func (p *Pool) acceptEntry(msg mixing.Message, msgtype msgtype, hash *chainhash.
 	p.pool[*hash] = e
 	p.messagesByIdentity[*id] = append(p.messagesByIdentity[*id], *hash)
 
+	if cm, ok := msg.(*wire.MsgMixConfirm); ok {
+		p.sessionsByTxHash[cm.Mix.TxHash()] = ses
+	}
+
 	rs.incrementCountFor(msgtype)
 	ses.bc.signal()