use Google's Protobuf library instead of GoGo

core/crypto/ed25519_test.go

@@ -5,7 +5,9 @@ import (
 	"crypto/rand"
 	"testing"
 
-	pb "github.com/libp2p/go-libp2p/core/crypto/pb"
+	"github.com/libp2p/go-libp2p/core/crypto/pb"
+
+	"google.golang.org/protobuf/proto"
 )
 
 func TestBasicSignAndVerify(t *testing.T) {
@@ -79,15 +81,15 @@ func TestMarshalLoop(t *testing.T) {
 				// Ed25519 private keys used to contain the public key twice.
 				// For backwards-compatibility, we need to continue supporting
 				// that scenario.
-				pbmes := new(pb.PrivateKey)
-				pbmes.Type = priv.Type()
 				data, err := priv.Raw()
 				if err != nil {
 					t.Fatal(err)
 				}
-
-				pbmes.Data = append(data, data[len(data)-ed25519.PublicKeySize:]...)
-				return pbmes.Marshal()
+				data = append(data, data[len(data)-ed25519.PublicKeySize:]...)
+				return proto.Marshal(&pb.PrivateKey{
+					Type: priv.Type().Enum(),
+					Data: data,
+				})
 			},
 		} {
 			t.Run(name, func(t *testing.T) {
@@ -150,18 +152,14 @@ func TestMarshalLoop(t *testing.T) {
 func TestUnmarshalErrors(t *testing.T) {
 	t.Run("PublicKey", func(t *testing.T) {
 		t.Run("Invalid data length", func(t *testing.T) {
-			pbmes := &pb.PublicKey{
-				Type: pb.KeyType_Ed25519,
+			data, err := proto.Marshal(&pb.PublicKey{
+				Type: pb.KeyType_Ed25519.Enum(),
 				Data: []byte{42},
-			}
-
-			data, err := pbmes.Marshal()
+			})
 			if err != nil {
 				t.Fatal(err)
 			}
-
-			_, err = UnmarshalPublicKey(data)
-			if err == nil {
+			if _, err := UnmarshalPublicKey(data); err == nil {
 				t.Fatal("expected an error")
 			}
 		})
@@ -174,16 +172,17 @@ func TestUnmarshalErrors(t *testing.T) {
 				t.Fatal(err)
 			}
 
-			pbmes := new(pb.PrivateKey)
-			pbmes.Type = priv.Type()
 			data, err := priv.Raw()
 			if err != nil {
 				t.Fatal(err)
 			}
-
 			// Append the private key instead of the public key.
-			pbmes.Data = append(data, data[:ed25519.PublicKeySize]...)
-			b, err := pbmes.Marshal()
+			data = append(data, data[:ed25519.PublicKeySize]...)
+
+			b, err := proto.Marshal(&pb.PrivateKey{
+				Type: priv.Type().Enum(),
+				Data: data,
+			})
 			if err != nil {
 				t.Fatal(err)
 			}
@@ -198,12 +197,10 @@ func TestUnmarshalErrors(t *testing.T) {
 		})
 
 		t.Run("Invalid data length", func(t *testing.T) {
-			pbmes := &pb.PrivateKey{
-				Type: pb.KeyType_Ed25519,
+			data, err := proto.Marshal(&pb.PrivateKey{
+				Type: pb.KeyType_Ed25519.Enum(),
 				Data: []byte{42},
-			}
-
-			data, err := pbmes.Marshal()
+			})
 			if err != nil {
 				t.Fatal(err)
 			}

core/crypto/key.go

@@ -12,11 +12,13 @@ import (
 	"fmt"
 	"io"
 
-	pb "github.com/libp2p/go-libp2p/core/crypto/pb"
+	"github.com/libp2p/go-libp2p/core/crypto/pb"
 
-	"github.com/gogo/protobuf/proto"
+	"google.golang.org/protobuf/proto"
 )
 
+//go:generate protoc --go_out=. --go_opt=Mpb/crypto.proto=./pb pb/crypto.proto
+
 const (
 	// RSA is an enum for the supported RSA key type
 	RSA = iota
@@ -194,7 +196,7 @@ func PublicKeyFromProto(pmes *pb.PublicKey) (PubKey, error) {
 
 	switch tpk := pk.(type) {
 	case *RsaPublicKey:
-		tpk.cached, _ = pmes.Marshal()
+		tpk.cached, _ = proto.Marshal(pmes)
 	}
 
 	return pk, nil
@@ -214,14 +216,14 @@ func MarshalPublicKey(k PubKey) ([]byte, error) {
 // PublicKeyToProto converts a public key object into an unserialized
 // protobuf PublicKey message.
 func PublicKeyToProto(k PubKey) (*pb.PublicKey, error) {
-	pbmes := new(pb.PublicKey)
-	pbmes.Type = k.Type()
 	data, err := k.Raw()
 	if err != nil {
 		return nil, err
 	}
-	pbmes.Data = data
-	return pbmes, nil
+	return &pb.PublicKey{
+		Type: k.Type().Enum(),
+		Data: data,
+	}, nil
 }
 
 // UnmarshalPrivateKey converts a protobuf serialized private key into its
@@ -243,15 +245,14 @@ func UnmarshalPrivateKey(data []byte) (PrivKey, error) {
 
 // MarshalPrivateKey converts a key object into its protobuf serialized form.
 func MarshalPrivateKey(k PrivKey) ([]byte, error) {
-	pbmes := new(pb.PrivateKey)
-	pbmes.Type = k.Type()
 	data, err := k.Raw()
 	if err != nil {
 		return nil, err
 	}
-
-	pbmes.Data = data
-	return proto.Marshal(pbmes)
+	return proto.Marshal(&pb.PrivateKey{
+		Type: k.Type().Enum(),
+		Data: data,
+	})
 }
 
 // ConfigDecodeKey decodes from b64 (for config file) to a byte array that can be unmarshalled.