feat: enable tls on grpc server

goto · Dec 18, 2023 · 1e33ede · 1e33ede
1 parent bdd1da6
commit 1e33ede
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 4 deletions.
diff --git a/.env.sample b/.env.sample
@@ -10,6 +10,7 @@ SERVER_WEBSOCKET_WRITE_WAIT_INTERVAL_MS=5000
 SERVER_WEBSOCKET_PINGER_SIZE=1
 
 SERVER_GRPC_PORT=8081
+SERVER_GRPC_TLS_ENABLED=false
 
 WORKER_BUFFER_CHANNEL_SIZE=5
 WORKER_BUFFER_FLUSH_TIMEOUT_MS=5000

diff --git a/.env.test b/.env.test
@@ -11,6 +11,7 @@ SERVER_WEBSOCKET_WRITE_WAIT_INTERVAL_MS=1000
 SERVER_WEBSOCKET_PINGER_SIZE=1
 
 SERVER_GRPC_PORT=8081
+SERVER_GRPC_TLS_ENABLED=false
 
 WORKER_BUFFER_CHANNEL_SIZE=5
 WORKER_BUFFER_FLUSH_TIMEOUT_MS=5000

diff --git a/config/server.go b/config/server.go
@@ -31,7 +31,10 @@ type serverWs struct {
 }
 
 type serverGRPC struct {
-	Port string
+	Port         string
+	TLSEnabled   bool
+	TLSCertPath  string
+	TLSPublicKey string
 }
 
 func serverConfigLoader() {
@@ -71,9 +74,14 @@ func serverWsConfigLoader() {
 }
 
 func serverGRPCConfigLoader() {
-
 	viper.SetDefault("SERVER_GRPC_PORT", "8081")
+	viper.SetDefault("SERVER_GRPC_TLS_ENABLED", false)
+	viper.SetDefault("SERVER_GRPC_TLS_CERT_PATH", "cert/server.crt")
+	viper.SetDefault("SERVER_GRPC_TLS_PUBLIC_KEY", "cert/server.key")
 	ServerGRPC = serverGRPC{
-		Port: util.MustGetString("SERVER_GRPC_PORT"),
+		Port:         util.MustGetString("SERVER_GRPC_PORT"),
+		TLSEnabled:   util.MustGetBool("SERVER_GRPC_TLS_ENABLED"),
+		TLSCertPath:  util.MustGetString("SERVER_GRPC_TLS_CERT_PATH"),
+		TLSPublicKey: util.MustGetString("SERVER_GRPC_TLS_PUBLIC_KEY"),
 	}
 }
diff --git a/services/grpc/service.go b/services/grpc/service.go
@@ -2,7 +2,9 @@ package grpc
 
 import (
 	"context"
+	"crypto/tls"
 	"fmt"
+	"google.golang.org/grpc/credentials"
 	"net"
 
 	pbgrpc "buf.build/gen/go/gotocompany/proton/grpc/go/gotocompany/raccoon/v1beta1/raccoonv1beta1grpc"
@@ -17,7 +19,7 @@ type Service struct {
 }
 
 func NewGRPCService(c collection.Collector) *Service {
-	server := grpc.NewServer()
+	server := newGRPCServer()
 	pbgrpc.RegisterEventServiceServer(server, &Handler{C: c})
 	return &Service{
 		s:         server,
@@ -41,3 +43,24 @@ func (s *Service) Shutdown(context.Context) error {
 	s.s.GracefulStop()
 	return nil
 }
+
+func newGRPCServer() *grpc.Server {
+	if config.ServerGRPC.TLSEnabled {
+		return grpc.NewServer(grpc.Creds(loadTLSCredentials()))
+	}
+	return grpc.NewServer()
+}
+
+func loadTLSCredentials() credentials.TransportCredentials {
+	serverCert, err := tls.LoadX509KeyPair(config.ServerGRPC.TLSCertPath, config.ServerGRPC.TLSPublicKey)
+	if err != nil {
+		panic("failed to load TLS credentials to start grpc server with TLS")
+	}
+
+	config := &tls.Config{
+		Certificates: []tls.Certificate{serverCert},
+		ClientAuth:   tls.NoClientCert,
+	}
+
+	return credentials.NewTLS(config)
+}