sofar/mtmediasrv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Strip port from remote address.

Browse Source
This commit is contained in:
Auke Kok 2018-01-13 19:26:54 -08:00
parent 753c767b31
commit 27fecb9e61
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.go
View File

@ -63,12 +63,21 @@ func (s FastCGIServer) ServeHTTP(w http.ResponseWriter, req *http.Request) {
req.Body.Read(header) req.Body.Read(header)
req.Body.Read(version) req.Body.Read(version)
// omit port numbers from peers
ip, _, err := net.SplitHostPort(req.RemoteAddr)
if err != nil {
http.Error(w, err.Error(), 500)
log.Print("Request: unable to identify peer\n")
return
}
remoteip := net.ParseIP(ip).String()
if !bytes.Equal(header, []byte("MTHS")) { if !bytes.Equal(header, []byte("MTHS")) {
log.Print(req.RemoteAddr, ": invalid MTHS header") log.Print(remoteip, ": invalid MTHS header")
return return
} }
if !bytes.Equal(version, []byte{0, 1}) { if !bytes.Equal(version, []byte{0, 1}) {
log.Print(req.RemoteAddr, ": unsupported MTHS version\n") log.Print(remoteip, ": unsupported MTHS version\n")
return return
} }
@ -109,7 +118,7 @@ func (s FastCGIServer) ServeHTTP(w http.ResponseWriter, req *http.Request) {
} }
// log transaction // log transaction
log.Print(req.RemoteAddr, " '", req.UserAgent(), "' ", len(resultarr), "/", len(clientarr), " ", c) log.Print(remoteip, " '", req.UserAgent(), "' ", len(resultarr), "/", len(clientarr), " ", c)
} }
func getHash(path string) (string, error) { func getHash(path string) (string, error) {