feat: catch panics in yamux send/receive loops

Part of libp2p/go-libp2p#1389.

session.go

@@ -10,6 +10,7 @@ import (
  "math"
  "net"
  "os"
+ "runtime/debug"
  "strings"
  "sync"
  "sync/atomic"
@@ -488,7 +489,15 @@ func (s *Session) send() {
  }
 }
 
-func (s *Session) sendLoop() error {
+func (s *Session) sendLoop() (err error) {
+ defer func() {
+ if rerr := recover(); rerr != nil {
+ fmt.Fprintf(os.Stderr, "caught panic: %s\n", err)
+ fmt.Fprintf(os.Stderr, "%s\n", debug.Stack())
+ err = fmt.Errorf("panic in yamux send loop: %s", rerr)
+ }
+ }()
+
  defer close(s.sendDoneCh)
 
  // Extend the write deadline if we've passed the halfway point. This can
@@ -620,7 +629,14 @@ var (
 )
 
 // recvLoop continues to receive data until a fatal error is encountered
-func (s *Session) recvLoop() error {
+func (s *Session) recvLoop() (err error) {
+ defer func() {
+ if rerr := recover(); rerr != nil {
+ fmt.Fprintf(os.Stderr, "caught panic: %s\n", err)
+ fmt.Fprintf(os.Stderr, "%s\n", debug.Stack())
+ err = fmt.Errorf("panic in yamux receive loop: %s", rerr)
+ }
+ }()
  defer close(s.recvDoneCh)
  var hdr header
  for {