discovery+switch: apply zero forwarding policy updates in-memory as well as on disk

In this commit, we fix a bug where if a user updates a forwarding policy to be
zero, the update will be applied to the policy correctly on-disk, but not
in-memory.

We solve this issue by having the gossiper return the list of on-disk updated
policies and passing these policies to the switch, so the switch can assume
that zero-valued fields are intentional and not just uninitialized.

discovery/gossiper.go

@@ -93,7 +93,7 @@ type chanPolicyUpdateRequest struct {
 	targetChans []wire.OutPoint
 	newSchema   routing.ChannelPolicy
 
-	errResp chan error
+	chanPolicies chan updatedChanPolicies
 }
 
 // Config defines the configuration for the service. ALL elements within the
@@ -338,27 +338,40 @@ func New(cfg Config, selfKey *btcec.PublicKey) *AuthenticatedGossiper {
 	return gossiper
 }
 
+// updatedChanPolicies is a set of channel policies that have been successfully
+// updated and written to disk, or an error if the policy update failed. This
+// struct's map field is intended to be used for updating channel policies on
+// the link layer.
+type updatedChanPolicies struct {
+	chanPolicies map[wire.OutPoint]*channeldb.ChannelEdgePolicy
+	err          error
+}
+
 // PropagateChanPolicyUpdate signals the AuthenticatedGossiper to update the
 // channel forwarding policies for the specified channels. If no channels are
 // specified, then the update will be applied to all outgoing channels from the
 // source node. Policy updates are done in two stages: first, the
 // AuthenticatedGossiper ensures the update has been committed by dependent
-// sub-systems, then it signs and broadcasts new updates to the network.
+// sub-systems, then it signs and broadcasts new updates to the network. A
+// mapping between outpoints and updated channel policies is returned, which is
+// used to update the forwarding policies of the underlying links.
 func (d *AuthenticatedGossiper) PropagateChanPolicyUpdate(
-	newSchema routing.ChannelPolicy, chanPoints ...wire.OutPoint) error {
+	newSchema routing.ChannelPolicy, chanPoints ...wire.OutPoint) (
+	map[wire.OutPoint]*channeldb.ChannelEdgePolicy, error) {
 
-	errChan := make(chan error, 1)
+	chanPolicyChan := make(chan updatedChanPolicies, 1)
 	policyUpdate := &chanPolicyUpdateRequest{
-		targetChans: chanPoints,
-		newSchema:   newSchema,
-		errResp:     errChan,
+		targetChans:  chanPoints,
+		newSchema:    newSchema,
+		chanPolicies: chanPolicyChan,
 	}
 
 	select {
 	case d.chanPolicyUpdates <- policyUpdate:
-		return <-errChan
+		updatedPolicies := <-chanPolicyChan
+		return updatedPolicies.chanPolicies, updatedPolicies.err
 	case <-d.quit:
-		return fmt.Errorf("AuthenticatedGossiper shutting down")
+		return nil, fmt.Errorf("AuthenticatedGossiper shutting down")
 	}
 }
 
@@ -895,13 +908,17 @@ func (d *AuthenticatedGossiper) networkHandler() {
 			// First, we'll now create new fully signed updates for
 			// the affected channels and also update the underlying
 			// graph with the new state.
-			newChanUpdates, err := d.processChanPolicyUpdate(
+			chanPolicies, newChanUpdates, err := d.processChanPolicyUpdate(
 				policyUpdate,
 			)
+			update := updatedChanPolicies{
+				chanPolicies,
+				err,
+			}
+			policyUpdate.chanPolicies <- update
 			if err != nil {
 				log.Errorf("Unable to craft policy updates: %v",
 					err)
-				policyUpdate.errResp <- err
 				continue
 			}
 
@@ -910,8 +927,6 @@ func (d *AuthenticatedGossiper) networkHandler() {
 			// start of the next epoch.
 			announcements.AddMsgs(newChanUpdates...)
 
-			policyUpdate.errResp <- nil
-
 		case announcement := <-d.networkMsgs:
 			// We should only broadcast this message forward if it
 			// originated from us or it wasn't received as part of
@@ -1244,7 +1259,9 @@ func (d *AuthenticatedGossiper) retransmitStaleChannels() error {
 //
 // TODO(roasbeef): generalize into generic for any channel update
 func (d *AuthenticatedGossiper) processChanPolicyUpdate(
-	policyUpdate *chanPolicyUpdateRequest) ([]networkMsg, error) {
+	policyUpdate *chanPolicyUpdateRequest) (
+	map[wire.OutPoint]*channeldb.ChannelEdgePolicy, []networkMsg, error) {
+
 	// First, we'll construct a set of all the channels that need to be
 	// updated.
 	chansToUpdate := make(map[wire.OutPoint]struct{})
@@ -1252,6 +1269,10 @@ func (d *AuthenticatedGossiper) processChanPolicyUpdate(
 		chansToUpdate[chanPoint] = struct{}{}
 	}
 
+	// Next, we'll create a mapping from outpoint to edge policy that will
+	// be used by each edge's underlying link to update its policy.
+	chanPolicies := make(map[wire.OutPoint]*channeldb.ChannelEdgePolicy)
+
 	haveChanFilter := len(chansToUpdate) != 0
 	if haveChanFilter {
 		log.Infof("Updating routing policies for chan_points=%v",
@@ -1295,7 +1316,7 @@ func (d *AuthenticatedGossiper) processChanPolicyUpdate(
 		return nil
 	})
 	if err != nil {
-		return nil, err
+		return nil, nil, err
 	}
 
 	// With the set of edges we need to update retrieved, we'll now re-sign
@@ -1309,9 +1330,13 @@ func (d *AuthenticatedGossiper) processChanPolicyUpdate(
 			edgeInfo.info, edgeInfo.edge,
 		)
 		if err != nil {
-			return nil, err
+			return nil, nil, err
 		}
 
+		// Since the update succeeded, add the edge to our policy
+		// mapping.
+		chanPolicies[edgeInfo.info.ChannelPoint] = edgeInfo.edge
+
 		// We'll avoid broadcasting any updates for private channels to
 		// avoid directly giving away their existence. Instead, we'll
 		// send the update directly to the remote party.
@@ -1340,7 +1365,7 @@ func (d *AuthenticatedGossiper) processChanPolicyUpdate(
 		})
 	}
 
-	return chanUpdates, nil
+	return chanPolicies, chanUpdates, nil
 }
 
 // processRejectedEdge examines a rejected edge to see if we can extract any

discovery/gossiper_test.go

@@ -3391,8 +3391,14 @@ func TestPropagateChanPolicyUpdate(t *testing.T) {
 	// the channel ann proof from the first channel in order to have it be
 	// marked as private channel.
 	firstChanID := channelsToAnnounce[0].localChanAnn.ShortChannelID
-	for _, batch := range channelsToAnnounce {
-		sendLocalMsg(t, ctx, batch.localChanAnn, localKey)
+	for i, batch := range channelsToAnnounce {
+		// channelPoint ensures that each channel policy in the map
+		// returned by PropagateChanPolicyUpdate has a unique key. Since
+		// the map is keyed by wire.OutPoint, we want to ensure that
+		// each channel has a unique channel point.
+		channelPoint := ChannelPoint(wire.OutPoint{Index: uint32(i)})
+
+		sendLocalMsg(t, ctx, batch.localChanAnn, localKey, channelPoint)
 		sendLocalMsg(t, ctx, batch.chanUpdAnn1, localKey)
 		sendLocalMsg(t, ctx, batch.nodeAnn1, localKey)
 
@@ -3430,11 +3436,19 @@ out:
 	newPolicy := routing.ChannelPolicy{
 		TimeLockDelta: newTimeLockDelta,
 	}
-	err = ctx.gossiper.PropagateChanPolicyUpdate(newPolicy)
+	newChanPolicies, err := ctx.gossiper.PropagateChanPolicyUpdate(newPolicy)
 	if err != nil {
 		t.Fatalf("unable to chan policies: %v", err)
 	}
 
+	// Ensure that the updated channel policies are as expected.
+	for _, dbPolicy := range newChanPolicies {
+		if dbPolicy.TimeLockDelta != uint16(newPolicy.TimeLockDelta) {
+			t.Fatalf("wrong delta: expected %v, got %v",
+				newPolicy.TimeLockDelta, dbPolicy.TimeLockDelta)
+		}
+	}
+
 	// Two channel updates should now be broadcast, with neither of them
 	// being the channel our first private channel.
 	for i := 0; i < numChannels-1; i++ {