Check number of arguments

go/cmd/gitaly-receive-pack/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"encoding/json"
+	"fmt"
 	"os"
 
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/handler"
@@ -19,6 +20,10 @@ func main() {
 		logger.Fatal("preparation failed", err)
 	}
 
+	if n := len(os.Args); n != 3 {
+		logger.Fatal("wrong number of arguments", fmt.Errorf("expected 2 arguments, got %v", os.Args))
+	}
+
 	var request pb.SSHReceivePackRequest
 	if err := json.Unmarshal([]byte(os.Args[2]), &request); err != nil {
 		logger.Fatal("unmarshaling request json failed", err)

go/cmd/gitaly-upload-pack/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"encoding/json"
+	"fmt"
 	"os"
 
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/handler"
@@ -19,6 +20,10 @@ func main() {
 		logger.Fatal("preparation failed", err)
 	}
 
+	if n := len(os.Args); n != 3 {
+		logger.Fatal("wrong number of arguments", fmt.Errorf("expected 2 arguments, got %v", os.Args))
+	}
+
 	var request pb.SSHUploadPackRequest
 	if err := json.Unmarshal([]byte(os.Args[2]), &request); err != nil {
 		logger.Fatal("unmarshaling request json failed", err)