diff options
| author | Jacob Vosmaer <jacob@gitlab.com> | 2017-05-03 16:10:29 +0200 |
|---|---|---|
| committer | Jacob Vosmaer <jacob@gitlab.com> | 2017-05-23 15:26:35 +0200 |
| commit | e3dc9c769f7ba74189b00a63def7cdb2a99f57d9 (patch) | |
| tree | 6c5a8acc7a543d566efc80074e62c0583b1f2671 | |
| parent | 33d583bf077b5f7b7fbb7eda2afb209944caf79e (diff) | |
| download | gitlab-shell-e3dc9c769f7ba74189b00a63def7cdb2a99f57d9.tar.gz | |
Check number of arguments
| -rw-r--r-- | go/cmd/gitaly-receive-pack/main.go | 5 | ||||
| -rw-r--r-- | go/cmd/gitaly-upload-pack/main.go | 5 |
2 files changed, 10 insertions, 0 deletions
diff --git a/go/cmd/gitaly-receive-pack/main.go b/go/cmd/gitaly-receive-pack/main.go index 11a0f7b..9e04347 100644 --- a/go/cmd/gitaly-receive-pack/main.go +++ b/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) diff --git a/go/cmd/gitaly-upload-pack/main.go b/go/cmd/gitaly-upload-pack/main.go index 80ead2b..028c623 100644 --- a/go/cmd/gitaly-upload-pack/main.go +++ b/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) |