[feature] add simple validator

giuem · giuem · commit 8d8073179d08 · 2017-07-25T13:20:51.000+08:00
diff --git a/main.go b/main.go
@@ -21,7 +21,9 @@ func init() {
 
 // ServerHandle handle proxy logic
 func ServerHandle(w http.ResponseWriter, req *http.Request) {
-	if req.Method != "GET" {
+	if req.Method != "GET" ||
+		len(req.Referer()) == 0 || len(req.FormValue("ga")) == 0 {
+		w.WriteHeader(http.StatusBadRequest)
 		return
 	}
 
@@ -30,7 +32,7 @@ func ServerHandle(w http.ResponseWriter, req *http.Request) {
 
 	uid := GetOrSetUUID(w, req)
 
-	w.WriteHeader(http.StatusNoContent)
+	w.WriteHeader(http.StatusOK)
 
 	go SendData(uid, req)
 }

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,9 @@ func init() {`
`21`	`21`
`22`	`22`	`// ServerHandle handle proxy logic`
`23`	`23`	`func ServerHandle(w http.ResponseWriter, req *http.Request) {`
`24`		`- if req.Method != "GET" {`
	`24`	`+ if req.Method != "GET" \|\|`
	`25`	`+ len(req.Referer()) == 0 \|\| len(req.FormValue("ga")) == 0 {`
	`26`	`+ w.WriteHeader(http.StatusBadRequest)`
`25`	`27`	`return`
`26`	`28`	`}`
`27`	`29`
`@@ -30,7 +32,7 @@ func ServerHandle(w http.ResponseWriter, req *http.Request) {`
`30`	`32`
`31`	`33`	`uid := GetOrSetUUID(w, req)`
`32`	`34`
`33`		`- w.WriteHeader(http.StatusNoContent)`
	`35`	`+ w.WriteHeader(http.StatusOK)`
`34`	`36`
`35`	`37`	`go SendData(uid, req)`
`36`	`38`	`}`