handle nil mix tx better in cm message

decred · Aug 8, 2023 · be7d77b · be7d77b
1 parent 48b857a
commit be7d77b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/wire/msgmixcm.go b/wire/msgmixcm.go
@@ -105,6 +105,10 @@ func (msg *MsgMixCM) writeMessageNoSignature(op string, w io.Writer, pver uint32
 		return err
 	}
 
+	if msg.Mix == nil {
+		msg := fmt.Sprintf("nil mix transaction")
+		return messageError(op, ErrInvalidMsg, msg)
+	}
 	err = msg.Mix.BtcEncode(w, pver)
 	if err != nil {
 		return err
@@ -201,6 +205,10 @@ func (msg *MsgMixCM) GetRun() uint32 {
 func NewMsgMixCM(identity [33]byte, sid [32]byte, expiry int64, run uint32,
 	mix *MsgTx, seenDCs []chainhash.Hash) *MsgMixCM {
 
+	if mix == nil {
+		mix = NewMsgTx()
+	}
+
 	return &MsgMixCM{
 		Identity:  identity,
 		SessionID: sid,