94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
var (
|
|
ctx = context.Background()
|
|
redisClient *redis.Client
|
|
hashName = "bl:sendnrw"
|
|
)
|
|
|
|
func main() {
|
|
// Redis Client initialisieren
|
|
redisClient = redis.NewClient(&redis.Options{
|
|
Addr: "10.10.5.249:6379", // ggf. anpassen
|
|
Password: "", // falls Passwort benötigt wird
|
|
DB: 0, // Standard DB
|
|
})
|
|
|
|
// Testverbindung
|
|
if err := redisClient.Ping(ctx).Err(); err != nil {
|
|
log.Fatalf("Keine Verbindung zu Redis: %v", err)
|
|
}
|
|
|
|
http.HandleFunc("/", logIPHandler)
|
|
|
|
log.Println("Server startet auf :8082")
|
|
if err := http.ListenAndServe(":8082", nil); err != nil {
|
|
log.Fatal("ListenAndServe: ", err)
|
|
}
|
|
}
|
|
|
|
func logIPHandler(w http.ResponseWriter, r *http.Request) {
|
|
ip := getClientIP(r)
|
|
if ip == "" {
|
|
http.Error(w, "Ungültige IP", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
ipKey := ip + "/32"
|
|
|
|
// IP in Redis Hash speichern
|
|
err := redisClient.HSet(ctx, hashName, ipKey, 1).Err()
|
|
if err != nil {
|
|
log.Printf("Fehler beim Schreiben in Redis: %v", err)
|
|
http.Error(w, "Interner Fehler", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
log.Printf("IP %s in Redis gespeichert", ipKey)
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
w.Write([]byte("IP erfasst: " + ipKey + "\n"))
|
|
}
|
|
|
|
func getClientIP(r *http.Request) string {
|
|
// X-Forwarded-For prüfen
|
|
xff := r.Header.Get("X-Forwarded-For")
|
|
if xff != "" {
|
|
parts := strings.Split(xff, ",")
|
|
ip := strings.TrimSpace(parts[0])
|
|
if isValidIP(ip) {
|
|
return ip
|
|
}
|
|
}
|
|
|
|
// X-Real-IP prüfen
|
|
xrip := r.Header.Get("X-Real-Ip")
|
|
if xrip != "" && isValidIP(xrip) {
|
|
return xrip
|
|
}
|
|
|
|
// RemoteAddr als Fallback
|
|
host, _, err := net.SplitHostPort(r.RemoteAddr)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
if isValidIP(host) {
|
|
return host
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func isValidIP(ip string) bool {
|
|
return net.ParseIP(ip) != nil
|
|
}
|