diff options
Diffstat (limited to 'cmd/serv.go')
| -rw-r--r-- | cmd/serv.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/cmd/serv.go b/cmd/serv.go index 1b41a5a078..fb1e581134 100644 --- a/cmd/serv.go +++ b/cmd/serv.go @@ -26,7 +26,6 @@ import ( "github.com/dgrijalva/jwt-go" "github.com/kballard/go-shellquote" - "github.com/unknwon/com" "github.com/urfave/cli" ) @@ -105,7 +104,10 @@ func runServ(c *cli.Context) error { if len(keys) != 2 || keys[0] != "key" { fail("Key ID format error", "Invalid key argument: %s", c.Args()[0]) } - keyID := com.StrTo(keys[1]).MustInt64() + keyID, err := strconv.ParseInt(keys[1], 10, 64) + if err != nil { + fail("Key ID format error", "Invalid key argument: %s", c.Args()[1]) + } cmd := os.Getenv("SSH_ORIGINAL_COMMAND") if len(cmd) == 0 { |