fix argument check for set, push

1 files changed, 8 insertions, 0 deletions

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 {