From 42a4681b424c7e2ea4312d0e405c5a4c50dd22d0 Mon Sep 17 00:00:00 2001 From: Mikhail Osipov Date: Wed, 16 Dec 2020 22:10:19 +0300 Subject: fix argument check for set, push --- pkg/server/env.go | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'pkg/server') diff --git a/pkg/server/env.go b/pkg/server/env.go index 8f27f70..3ed820e 100644 --- a/pkg/server/env.go +++ b/pkg/server/env.go @@ -15,6 +15,10 @@ func varGet(r *request) { } func varSet(r *request) { + if len(r.args) < 2 { + r.Fatal("not enough args") + } + value := strings.Join(r.args[1:], " ") if err := r.c.s.env.Set(r.args[0], value); err != nil { @@ -23,6 +27,10 @@ func varSet(r *request) { } func varPush(r *request) { + if len(r.args) < 2 { + r.Fatal("not enough args") + } + value := strings.Join(r.args[1:], " ") if err := r.c.s.env.Push(r.args[0], value); err != nil { -- cgit v1.2.3-70-g09d2