routerrpc: recover from panics during PoC

Author: feelancer21

lnrpc/routerrpc/imputed_cost.go

@@ -41,6 +41,13 @@ func parseImputedCostRestr(restr *lnrpc.ImputedCostRestriction) (
 func (s *Server) XImportImputedCosts(_ context.Context,
 	req *ImportImputedCostsRequest) (*ImportImputedCostsResponse, error) {
 
+	// Recover from any panics during PoC of imputed cost.
+	defer func() {
+		if r := recover(); r != nil {
+			log.Errorf("panic while XImportImputedCosts: %v", r)
+		}
+	}()
+
 	data := make(map[string]*ImputedCostNamespace)
 	names := make(fn.Set[string])
 
@@ -155,6 +162,13 @@ func unmarshalImputedParams(
 func (s *Server) XQueryImputedCosts(_ context.Context,
 	req *QueryImputedCostsRequest) (*QueryImputedCostsResponse, error) {
 
+	// Recover from any panics during PoC of imputed cost.
+	defer func() {
+		if r := recover(); r != nil {
+			log.Errorf("panic while XQueryImputedCosts: %v", r)
+		}
+	}()
+
 	names := fn.NewSet(req.Namespaces...)
 	res := make([]*ImputedCostNamespace, 0, names.Size())
 
@@ -207,6 +221,13 @@ func marshalImputedParams(
 func (s *Server) XDeleteImputedCosts(_ context.Context,
 	req *DeleteImputedCostsRequest) (*DeleteImputedCostsResponse, error) {
 
+	// Recover from any panics during PoC of imputed cost.
+	defer func() {
+		if r := recover(); r != nil {
+			log.Errorf("panic while XDeleteImputedCosts: %v", r)
+		}
+	}()
+
 	if len(req.Namespaces) == 0 {
 		return nil, errors.New("at least one namespace required")
 	}

lnrpc/routerrpc/router_backend.go

@@ -169,6 +169,13 @@ type MissionControl interface {
 func (r *RouterBackend) QueryRoutes(ctx context.Context,
 	in *lnrpc.QueryRoutesRequest) (*lnrpc.QueryRoutesResponse, error) {
 
+	// Recover from any panics during PoC of imputed cost.
+	defer func() {
+		if r := recover(); r != nil {
+			log.Errorf("panic while QueryRoutes: %v", r)
+		}
+	}()
+
 	routeReq, err := r.parseQueryRoutesRequest(in)
 	if err != nil {
 		return nil, err

lnrpc/routerrpc/router_server.go

@@ -365,6 +365,12 @@ func (r *ServerShell) CreateSubServer(configRegistry lnrpc.SubServerConfigDispat
 func (s *Server) SendPaymentV2(req *SendPaymentRequest,
 	stream Router_SendPaymentV2Server) error {
 
+	// Recover from any panics during PoC of imputed cost.
+	defer func() {
+		if r := recover(); r != nil {
+			log.Errorf("panic while SendPaymentV2: %v", r)
+		}
+	}()
 	// Set payment request attempt timeout.
 	if req.TimeoutSeconds == 0 {
 		req.TimeoutSeconds = DefaultPaymentTimeout