Fixes intelsdi-x#260: Adds URL checking

tiffanyfay · Feb 24, 2016 · 2fe0dde · 2fe0dde
1 parent 3b1ae75
commit 2fe0dde
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Godeps/Godeps.json b/Godeps/Godeps.json
diff --git a/mgmt/rest/client/client.go b/mgmt/rest/client/client.go
@@ -35,6 +35,7 @@ import (
 	"path/filepath"
 	"strings"
 
+	"github.com/asaskevich/govalidator"
 	"github.com/intelsdi-x/snap/mgmt/rest/rbody"
 )
 
@@ -111,6 +112,11 @@ func (c *Client) do(method, path string, ct contentType, body ...[]byte) (*rbody
 		rsp *http.Response
 		err error
 	)
+	// URL validation
+	validURL := govalidator.IsURL(c.prefix + path)
+	if !validURL {
+		return nil, errors.New("URL is not in the format of http(s)://<ip>:<port>")
+	}
 	switch method {
 	case "GET":
 		rsp, err = c.http.Get(c.prefix + path)