print recovered panics to stderr (#33)

libp2p · Apr 14, 2022 · 3e24327 · 3e24327
1 parent 93c76c2
commit 3e24327
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/protoio/uvarint_reader.go b/protoio/uvarint_reader.go
@@ -36,6 +36,8 @@ import (
 	"bufio"
 	"fmt"
 	"io"
+	"os"
+	"runtime/debug"
 
 	"github.com/gogo/protobuf/proto"
 
@@ -60,6 +62,7 @@ func NewDelimitedReader(r io.Reader, maxSize int) ReadCloser {
 func (ur *uvarintReader) ReadMsg(msg proto.Message) (err error) {
 	defer func() {
 		if rerr := recover(); rerr != nil {
+			fmt.Fprintf(os.Stderr, "caught panic: %s\n%s\n", rerr, debug.Stack())
 			err = fmt.Errorf("panic reading message: %s", rerr)
 		}
 	}()

diff --git a/protoio/uvarint_writer.go b/protoio/uvarint_writer.go
@@ -35,6 +35,8 @@ package protoio
 import (
 	"fmt"
 	"io"
+	"os"
+	"runtime/debug"
 
 	"github.com/gogo/protobuf/proto"
 
@@ -54,6 +56,7 @@ func NewDelimitedWriter(w io.Writer) WriteCloser {
 func (uw *uvarintWriter) WriteMsg(msg proto.Message) (err error) {
 	defer func() {
 		if rerr := recover(); rerr != nil {
+			fmt.Fprintf(os.Stderr, "caught panic: %s\n%s\n", rerr, debug.Stack())
 			err = fmt.Errorf("panic reading message: %s", rerr)
 		}
 	}()