Prevent failure index underflow

2 years ago · 8e4521e033
parent 4a84505a83
commit 8e4521e033
1 changed files with 6 additions and 0 deletions
--- a/payment.go
+++ b/payment.go
@ -51,6 +51,12 @@ func (r *regolancer) pay(ctx context.Context, invoice *lnrpc.AddInvoiceResponse,
 			return fmt.Errorf("error: %s @ %d", result.Failure.Code.String(),
 				result.Failure.FailureSourceIndex)
 		}
+		if result.Failure.FailureSourceIndex == 0 {
+			log.Print(errColorF("%s (unexpected hop index %d, should be greater than 0)", result.Failure.Code.String(),
+				result.Failure.FailureSourceIndex))
+			return fmt.Errorf("error: %s @ %d", result.Failure.Code.String(),
+				result.Failure.FailureSourceIndex)
+		}
 		nodeCtx, cancel := context.WithTimeout(ctx, time.Minute)
 		defer cancel()
 		node1, err := r.getNodeInfo(nodeCtx, route.Hops[result.Failure.FailureSourceIndex-1].PubKey)