owncast/utils/clientId.go

41 lines
1,000 B
Go
Raw Normal View History

package utils
import (
"crypto/md5" //nolint
"encoding/hex"
"net"
"net/http"
"strings"
log "github.com/sirupsen/logrus"
)
2020-11-13 02:14:59 +03:00
// GenerateClientIDFromRequest generates a client id from the provided request.
func GenerateClientIDFromRequest(req *http.Request) string {
ipAddress := GetIPAddressFromRequest(req)
ipAddressComponents := strings.Split(ipAddress, ":")
ipAddressComponents[len(ipAddressComponents)-1] = ""
clientID := strings.Join(ipAddressComponents, ":") + req.UserAgent()
// Create a MD5 hash of this ip + useragent
b := md5.Sum([]byte(clientID)) // nolint
return hex.EncodeToString(b[:])
}
2020-11-13 02:14:59 +03:00
// GetIPAddressFromRequest returns the IP address from a http request.
func GetIPAddressFromRequest(req *http.Request) string {
ipAddressString := req.RemoteAddr
xForwardedFor := req.Header.Get("X-FORWARDED-FOR")
if xForwardedFor != "" {
return xForwardedFor
}
ip, _, err := net.SplitHostPort(ipAddressString)
if err != nil {
log.Errorln(err)
return ""
}
return ip
}