Merge pull request #15 from ieee0824/fix

fix nil bug
ieee0824 · Oct 12, 2017 · 970fdd9 · 970fdd9
2 parents fc3eba2 + 188cc93
commit 970fdd9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cryptex.go b/cryptex.go
@@ -5,6 +5,7 @@ import (
 	"encoding/base64"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"github.com/ghodss/yaml"
 	"github.com/ieee0824/cryptex/encryptor"
 	"io/ioutil"
@@ -196,6 +197,9 @@ func (c *Cryptex) Decrypt(d *Container) (interface{}, error) {
 }
 
 func (c *Cryptex) Edit(i *Container) (interface{}, error) {
+	if i == nil || i.Values == nil {
+		return nil, fmt.Errorf("container is empty")
+	}
 	p, err := c.decrypt(i.Values)
 	if err != nil {
 		return nil, err