feat(share/p2p/shrex): add recovery middleware to shrex-nd and shrex-…

…eds servers (#3328)

Add recovery middleware to shrex-eds and shrex-nd server implementation

share/p2p/recovery.go

@@ -0,0 +1,21 @@
+package p2p
+
+import (
+	"fmt"
+
+	"github.com/libp2p/go-libp2p/core/network"
+)
+
+// RecoveryMiddleware is a middleware that recovers from panics in the handler.
+func RecoveryMiddleware(handler network.StreamHandler) network.StreamHandler {
+	return func(stream network.Stream) {
+		defer func() {
+			r := recover()
+			if r != nil {
+				err := fmt.Errorf("PANIC while handling request: %s", r)
+				log.Error(err)
+			}
+		}()
+		handler(stream)
+	}
+}

share/p2p/shrexeds/server.go

@@ -52,7 +52,10 @@ func NewServer(params *Parameters, host host.Host, store *eds.Store) (*Server, e
 
 func (s *Server) Start(context.Context) error {
 	s.ctx, s.cancel = context.WithCancel(context.Background())
-	s.host.SetStreamHandler(s.protocolID, s.middleware.RateLimitHandler(s.handleStream))
+	handler := s.handleStream
+	withRateLimit := s.middleware.RateLimitHandler(handler)
+	withRecovery := p2p.RecoveryMiddleware(withRateLimit)
+	s.host.SetStreamHandler(s.protocolID, withRecovery)
 	return nil
 }
 

share/p2p/shrexnd/server.go

@@ -54,7 +54,10 @@ func NewServer(params *Parameters, host host.Host, store *eds.Store) (*Server, e
 	ctx, cancel := context.WithCancel(context.Background())
 	srv.cancel = cancel
 
-	srv.handler = srv.middleware.RateLimitHandler(srv.streamHandler(ctx))
+	handler := srv.streamHandler(ctx)
+	withRateLimit := srv.middleware.RateLimitHandler(handler)
+	withRecovery := p2p.RecoveryMiddleware(withRateLimit)
+	srv.handler = withRecovery
 	return srv, nil
 }