miss ip.go

polaris1119 · polaris1119 · commit 4db39df11281 · 2014-11-08T11:56:53.000+08:00
diff --git a/websites/code/studygolang/src/util/ip.go b/websites/code/studygolang/src/util/ip.go
@@ -0,0 +1,61 @@
+// Copyright 2014 The StudyGolang Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+// http://studygolang.com
+// Author：polaris	studygolang@gmail.com
+
+package util
+
+import (
+	"encoding/binary"
+	"net"
+	"net/http"
+	"strings"
+)
+
+func Ip(req *http.Request) string {
+	ips := proxy(req)
+	if ips != nil && ips[0] != "" {
+		pos := strings.LastIndex(ips[0], ":")
+		if pos == -1 {
+			return ips[0]
+		} else {
+			return ips[0][:pos]
+		}
+	}
+
+	remoteAddr := req.Header.Get("Remote_addr")
+	if remoteAddr == "" {
+		remoteAddr = req.Header.Get("X-Real-IP")
+		if remoteAddr == "" {
+			remoteAddr = req.RemoteAddr
+		}
+	}
+
+	if remoteAddr == "" {
+		return "127.0.0.1"
+	}
+
+	pos := strings.LastIndex(remoteAddr, ":")
+	if pos == -1 {
+		return remoteAddr
+	} else {
+		return remoteAddr[:pos]
+	}
+}
+
+func proxy(req *http.Request) []string {
+	if ips := req.Header.Get("X-Forwarded-For"); ips != "" {
+		return strings.Split(ips, ",")
+	}
+	return nil
+}
+
+func Ip2long(ipstr string) uint32 {
+	ip := net.ParseIP(ipstr)
+	if ip == nil {
+		return 0
+	}
+	ip = ip.To4()
+	return binary.BigEndian.Uint32(ip)
+}
