netann: let SignChannelUpdate take ChannelUpdate interface

netann/channel_update.go

@@ -67,7 +67,7 @@ func ChanUpdSetTimestamp(bestBlockHeight uint32) ChannelUpdateModifier {
 //
 // NOTE: This method modifies the given update.
 func SignChannelUpdate(signer keychain.MessageSignerRing,
-	keyLoc keychain.KeyLocator, update *lnwire.ChannelUpdate1,
+	keyLoc keychain.KeyLocator, update lnwire.ChannelUpdate,
 	mods ...ChannelUpdateModifier) error {
 
 	// Apply the requested changes to the channel update.
@@ -75,16 +75,45 @@ func SignChannelUpdate(signer keychain.MessageSignerRing,
 		modifier(update)
 	}
 
-	// Create the DER-encoded ECDSA signature over the message digest.
+	switch upd := update.(type) {
-	sig, err := SignAnnouncement(signer, keyLoc, update)
+	case *lnwire.ChannelUpdate1:
-	if err != nil {
+		data, err := upd.DataToSign()
-		return err
+		if err != nil {
-	}
+			return err
+		}
 
-	// Parse the DER-encoded signature into a fixed-size 64-byte array.
+		sig, err := signer.SignMessage(keyLoc, data, true)
-	update.Signature, err = lnwire.NewSigFromSignature(sig)
+		if err != nil {
-	if err != nil {
+			return err
-		return err
+		}
+
+		// Parse the DER-encoded signature into a fixed-size 64-byte
+		// array.
+		upd.Signature, err = lnwire.NewSigFromSignature(sig)
+		if err != nil {
+			return err
+		}
+
+	case *lnwire.ChannelUpdate2:
+		data, err := upd.DataToSign()
+		if err != nil {
+			return err
+		}
+
+		sig, err := signer.SignMessageSchnorr(
+			keyLoc, data, false, nil, upd.DigestTag(),
+		)
+		if err != nil {
+			return err
+		}
+
+		upd.Signature, err = lnwire.NewSigFromSignature(sig)
+		if err != nil {
+			return err
+		}
+	default:
+		return fmt.Errorf("unhandled implementation of "+
+			"ChannelUpdate: %T", update)
 	}
 
 	return nil