mirror of
https://github.com/gravitl/netmaker.git
synced 2026-04-23 00:17:10 +08:00
Abhishek Kondur
9b4f9a4a50
* NM-240: update api swagger docs * NM-240: remove api doc for debug api * NM-240: generate swagger * NM-240: remove deprecated and unused apis, update jit models * NM-240: deprecate ununsed apis from api docs
76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package controller
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gorilla/mux"
|
|
|
|
"github.com/gravitl/netmaker/netclient/ncutils"
|
|
)
|
|
|
|
func ipHandlers(r *mux.Router) {
|
|
r.HandleFunc("/api/getip", http.HandlerFunc(getPublicIP)).Methods(http.MethodGet)
|
|
}
|
|
|
|
// @Summary Get the current public IP address.
|
|
// @Router /api/getip [get]
|
|
// @Tags IP Service
|
|
// @Produce json
|
|
// @Success 200 {string} string "The public IP address."
|
|
// @Failure 400 {string} string "Invalid IP address or no IP found."
|
|
func getPublicIP(w http.ResponseWriter, r *http.Request) {
|
|
r.Header.Set("Connection", "close")
|
|
ip, err := parseIP(r)
|
|
if err != nil {
|
|
w.WriteHeader(400)
|
|
switch {
|
|
case ip != "":
|
|
_, _ = w.Write([]byte("ip is invalid: " + ip))
|
|
case ip == "":
|
|
_, _ = w.Write([]byte("no ip found"))
|
|
default:
|
|
fmt.Println(err)
|
|
}
|
|
return
|
|
}
|
|
|
|
w.WriteHeader(200)
|
|
_, _ = w.Write([]byte(ip))
|
|
}
|
|
|
|
func parseIP(r *http.Request) (string, error) {
|
|
// Get Public IP from header
|
|
ip := r.Header.Get("X-REAL-IP")
|
|
ipnet := net.ParseIP(ip)
|
|
if ipnet != nil && !ncutils.IpIsPrivate(ipnet) {
|
|
return ip, nil
|
|
}
|
|
|
|
// If above fails, get Public IP from other header instead
|
|
forwardips := r.Header.Get("X-FORWARDED-FOR")
|
|
iplist := strings.Split(forwardips, ",")
|
|
for _, ip := range iplist {
|
|
ipnet := net.ParseIP(ip)
|
|
if ipnet != nil && !ncutils.IpIsPrivate(ipnet) {
|
|
return ip, nil
|
|
}
|
|
}
|
|
|
|
// If above also fails, get Public IP from Remote Address of request
|
|
ip, _, err := net.SplitHostPort(r.RemoteAddr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
ipnet = net.ParseIP(ip)
|
|
if ipnet != nil {
|
|
if ncutils.IpIsPrivate(ipnet) {
|
|
return ip, fmt.Errorf("ip is a private address")
|
|
}
|
|
return ip, nil
|
|
}
|
|
return "", fmt.Errorf("no ip found")
|
|
}
|