DB: Prevent encoding a nil message (#4470)

* Prevent encoding a nil message
* Merge refs/heads/master into prevent-saving-nil-msg
This commit is contained in:
Preston Van Loon
2020-01-09 17:38:53 -08:00
committed by prylabs-bulldozer[bot]
parent 6de485c27e
commit eae2268dd1

View File

@@ -1,6 +1,8 @@
package kv
import (
"errors"
"github.com/gogo/protobuf/proto"
"github.com/golang/snappy"
)
@@ -17,6 +19,9 @@ func decode(data []byte, dst proto.Message) error {
}
func encode(msg proto.Message) ([]byte, error) {
if msg == nil {
return nil, errors.New("cannot encode nil message")
}
enc, err := proto.Marshal(msg)
if err != nil {
return nil, err