gitee.com/zhaochuninhefei/gmgo@v0.0.31-0.20240209061119-069254a02979/grpc/grpc_test/grpc_test.go

// Copyright (c) 2022 zhaochun
// gmgo is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
//          http://license.coscl.org.cn/MulanPSL2
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

/*
grpc_test 是对`gitee.com/zhaochuninhefei/gmgo/grpc`的测试包
*/

package grpc_test

import (
	"errors"
	"fmt"
	"io/ioutil"
	"log"
	"net"
	"testing"
	"time"

	"gitee.com/zhaochuninhefei/gmgo/gmtls"
	"gitee.com/zhaochuninhefei/gmgo/grpc"
	"gitee.com/zhaochuninhefei/gmgo/grpc/credentials"
	"gitee.com/zhaochuninhefei/gmgo/grpc/grpc_test/echo"
	"gitee.com/zhaochuninhefei/gmgo/net/context"
	"gitee.com/zhaochuninhefei/gmgo/x509"
	"gitee.com/zhaochuninhefei/zcgolog/zclog"
)

//goland:noinspection GoSnakeCaseUsage
const (
	port    = ":50051"
	address = "localhost:50051"

	sm2_ca       = "testdata/sm2_ca.cert"
	sm2_signCert = "testdata/sm2_sign.cert"
	sm2_signKey  = "testdata/sm2_sign.key"
	sm2_userCert = "testdata/sm2_user.cert"
	sm2_userKey  = "testdata/sm2_user.key"

	ecdsa_ca       = "testdata/ecdsa_ca.cert"
	ecdsa_signCert = "testdata/ecdsa_sign.cert"
	ecdsa_signKey  = "testdata/ecdsa_sign.key"
	ecdsa_userCert = "testdata/ecdsa_user.cert"
	ecdsa_userKey  = "testdata/ecdsa_user.key"

	ecdsaext_ca       = "testdata/ecdsaext_ca.cert"
	ecdsaext_signCert = "testdata/ecdsaext_sign.cert"
	ecdsaext_signKey  = "testdata/ecdsaext_sign.key"
	ecdsaext_userCert = "testdata/ecdsaext_user.cert"
	ecdsaext_userKey  = "testdata/ecdsaext_user.key"
)

func TestMain(m *testing.M) {
	zcgologConfig := &zclog.Config{
		LogLevelGlobal: zclog.LOG_LEVEL_DEBUG,
	}
	zclog.InitLogger(zcgologConfig)
	go serverRun()
	time.Sleep(1000000)
	m.Run()
}

var end chan bool

func Test_credentials_sm2(t *testing.T) {
	end = make(chan bool, 64)
	go clientRun("sm2")
	<-end
}

func Test_credentials_ecdsa(t *testing.T) {
	end = make(chan bool, 64)
	go clientRun("ecdsa")
	<-end
}

func Test_credentials_ecdsaext(t *testing.T) {
	end = make(chan bool, 64)
	go clientRun("ecdsaext")
	<-end
}

func serverRun() {
	// 准备3份服务端证书, 分别是sm2, ecdsa, ecdsaext
	var certs []gmtls.Certificate
	sm2SignCert, err := gmtls.LoadX509KeyPair(sm2_signCert, sm2_signKey)
	if err != nil {
		log.Fatal(err)
	}
	certs = append(certs, sm2SignCert)
	ecdsaSignCert, err := gmtls.LoadX509KeyPair(ecdsa_signCert, ecdsa_signKey)
	if err != nil {
		log.Fatal(err)
	}
	certs = append(certs, ecdsaSignCert)
	ecdsaextSignCert, err := gmtls.LoadX509KeyPair(ecdsaext_signCert, ecdsaext_signKey)
	if err != nil {
		log.Fatal(err)
	}
	certs = append(certs, ecdsaextSignCert)

	// 准备CA证书池，导入颁发客户端证书的CA证书
	certPool := x509.NewCertPool()
	sm2CaCert, err := ioutil.ReadFile(sm2_ca)
	if err != nil {
		log.Fatal(err)
	}
	certPool.AppendCertsFromPEM(sm2CaCert)
	ecdsaCaCert, err := ioutil.ReadFile(ecdsa_ca)
	if err != nil {
		log.Fatal(err)
	}
	certPool.AppendCertsFromPEM(ecdsaCaCert)
	ecdsaextCaCert, err := ioutil.ReadFile(ecdsaext_ca)
	if err != nil {
		log.Fatal(err)
	}
	certPool.AppendCertsFromPEM(ecdsaextCaCert)

	// 创建gmtls配置
	config := &gmtls.Config{
		Certificates: certs,
		ClientAuth:   gmtls.RequireAndVerifyClientCert,
		ClientCAs:    certPool,
	}

	// 创建grpc服务端
	creds := credentials.NewTLS(config)
	s := grpc.NewServer(grpc.Creds(creds))
	echo.RegisterEchoServer(s, &server{})

	// 开启tcp监听端口
	lis, err := net.Listen("tcp", port)
	if err != nil {
		log.Fatalf("fail to listen: %v", err)
	}
	// 启动grpc服务
	err = s.Serve(lis)
	if err != nil {
		log.Fatalf("Serve: %v", err)
	}
}

func clientRun(certType string) {
	// 创建客户端本地的证书池
	caPool := x509.NewCertPool()
	// ca证书
	var cacert []byte
	// 客户端证书
	var cert gmtls.Certificate
	// 客户端优先曲线列表
	var curvePreference []gmtls.CurveID
	// 客户端优先密码套件列表
	var cipherSuitesPrefer []uint16
	// 客户端优先签名算法
	var sigAlgPrefer []gmtls.SignatureScheme
	var err error
	switch certType {
	case "sm2":
		// 读取sm2 ca证书
		cacert, err = ioutil.ReadFile(sm2_ca)
		// 读取User证书与私钥，作为客户端的证书与私钥，一般用作密钥交换证书。
		// 但如果服务端要求查看客户端证书(双向tls通信)则也作为客户端身份验证用证书，
		// 此时该证书应该由第三方ca机构颁发签名。
		cert, err = gmtls.LoadX509KeyPair(sm2_userCert, sm2_userKey)
		curvePreference = append(curvePreference, gmtls.Curve256Sm2)
		cipherSuitesPrefer = append(cipherSuitesPrefer, gmtls.TLS_SM4_GCM_SM3)
	case "ecdsa":
		// 读取ecdsa ca证书
		cacert, err = ioutil.ReadFile(ecdsa_ca)
		// 读取User证书与私钥，作为客户端的证书与私钥，一般用作密钥交换证书。
		// 但如果服务端要求查看客户端证书(双向tls通信)则也作为客户端身份验证用证书，
		// 此时该证书应该由第三方ca机构颁发签名。
		cert, err = gmtls.LoadX509KeyPair(ecdsa_userCert, ecdsa_userKey)
		curvePreference = append(curvePreference, gmtls.CurveP256)
		cipherSuitesPrefer = append(cipherSuitesPrefer, gmtls.TLS_AES_128_GCM_SHA256)
		sigAlgPrefer = append(sigAlgPrefer, gmtls.ECDSAWithP256AndSHA256)
	case "ecdsaext":
		// 读取ecdsaext ca证书
		cacert, err = ioutil.ReadFile(ecdsaext_ca)
		// 读取User证书与私钥，作为客户端的证书与私钥，一般用作密钥交换证书。
		// 但如果服务端要求查看客户端证书(双向tls通信)则也作为客户端身份验证用证书，
		// 此时该证书应该由第三方ca机构颁发签名。
		cert, err = gmtls.LoadX509KeyPair(ecdsaext_userCert, ecdsaext_userKey)
		curvePreference = append(curvePreference, gmtls.CurveP256)
		cipherSuitesPrefer = append(cipherSuitesPrefer, gmtls.TLS_AES_128_GCM_SHA256)
		sigAlgPrefer = append(sigAlgPrefer, gmtls.ECDSAEXTWithP256AndSHA256)
	default:
		err = errors.New("目前只支持sm2/ecdsa/ecdsaext")
	}
	if err != nil {
		zclog.Fatal(err)
	}
	// 将ca证书作为根证书加入证书池
	// 即，客户端相信持有该ca颁发的证书的服务端
	caPool.AppendCertsFromPEM(cacert)

	// 定义gmtls配置
	config := &gmtls.Config{
		RootCAs:      caPool,
		Certificates: []gmtls.Certificate{cert},
		// 因为相关证书是由`x509/x509_test.go`的`TestCreateCertFromCA`生成的，
		// 指定了SAN包含"server.test.com"
		ServerName:         "server.test.com",
		CurvePreferences:   curvePreference,
		PreferCipherSuites: cipherSuitesPrefer,
		SignAlgPrefer:      sigAlgPrefer,
		ClientAuth:         gmtls.RequireAndVerifyClientCert,
	}
	creds := credentials.NewTLS(config)
	conn, err := grpc.Dial(address, grpc.WithTransportCredentials(creds))
	if err != nil {
		log.Fatalf("cannot to connect: %v", err)
	}
	defer func(conn *grpc.ClientConn) {
		_ = conn.Close()
	}(conn)
	c := echo.NewEchoClient(conn)
	echoInClient(c)
	end <- true
}

// 客户端echo处理
func echoInClient(c echo.EchoClient) {
	msgClient := "hello, this is client."
	fmt.Printf("客户端发出消息: %s\n", msgClient)
	r, err := c.Echo(context.Background(), &echo.EchoRequest{Req: msgClient})
	if err != nil {
		log.Fatalf("failed to echo: %v", err)
	}
	msgServer := r.Result
	fmt.Printf("客户端收到消息: %s\n", msgServer)
}

type server struct{}

// Echo 服务端echo处理
//goland:noinspection GoUnusedParameter
func (s *server) Echo(ctx context.Context, req *echo.EchoRequest) (*echo.EchoResponse, error) {
	msgClient := req.Req
	fmt.Printf("服务端接收到消息: %s\n", msgClient)
	msgServer := "hello,this is server."
	fmt.Printf("服务端返回消息: %s\n", msgServer)
	return &echo.EchoResponse{Result: msgServer}, nil
}