summaryrefslogtreecommitdiff
path: root/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'main.go')
-rw-r--r--main.go3
1 files changed, 2 insertions, 1 deletions
diff --git a/main.go b/main.go
index 41f8334..366ebde 100644
--- a/main.go
+++ b/main.go
@@ -21,11 +21,12 @@ type endpoint struct {
// ServeHTTP is http.Handler implementation.
func (ep endpoint) ServeHTTP(w http.ResponseWriter, r *http.Request) {
source, err := url.Parse(r.PostFormValue("source"))
- if err != nil {
+ if err != nil || source.Host == "" {
w.WriteHeader(http.StatusBadRequest)
w.Write([]byte(errSrcInvalid))
return
}
+
if source.Scheme != "http" && source.Scheme != "https" {
w.WriteHeader(http.StatusBadRequest)
w.Write([]byte(errInvalidScheme))