Go (net/http)

Język Go został zaprojektowany z myślą o ultra-szybkich usługach sieciowych. Ponieważ ProxyTracer odpowiada w czasie poniżej 10 ms, wywołanie go w ramach standardowego middleware net/http nie spowoduje zauważalnego wąskiego gardła w gorutynach.

Middleware

Ta implementacja bezpiecznie pobiera adres IP z nagłówków X-Forwarded-For, używa natywnego klienta http.Client w Go ze ścisłym limitem czasu i natychmiast odrzuca złośliwe połączenia, zanim dotrą do ciężkiej logiki biznesowej.


package middleware
 
import (
    "encoding/json"
    "log"
    "net/http"
    "os"
    "strings"
    "time"
)
 
// ProxyTracerResponse mapuje dokładnie odpowiedź JSON {"proxy": true|false}
type ProxyTracerResponse struct {
    Proxy bool `json:"proxy"`
}
 
func ProxyTracer(next http.Handler) http.Handler {
    apiKey := os.Getenv("PROXYTRACER_API_KEY")
    
    // Zainicjalizuj klienta wielokrotnego użytku ze ścisłym limitem czasu 500 ms
    client := &http.Client{
        Timeout: 500 * time.Millisecond,
    }
 
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 1. Bezpiecznie pobierz prawdziwy adres IP zza load balancerów
        ip := r.Header.Get("X-Forwarded-For")
        if ip == "" {
            ip = strings.Split(r.RemoteAddr, ":")[0]
        } else {
            ip = strings.Split(ip, ",")[0]
        }
        ip = strings.TrimSpace(ip)
 
        // Przepuść żądanie w środowisku deweloperskim (localhost)
        if ip == "127.0.0.1" || ip == "::1" || ip == "" {
            next.ServeHTTP(w, r)
            return
        }
 
        // 2. Odpytaj API ProxyTracer
        req, err := http.NewRequest("GET", "https://api.proxytracer.com/v1/check/"+ip, nil)
        if err == nil {
            req.Header.Set("Authorization", "Bearer "+apiKey)
            resp, err := client.Do(req)
 
            if err == nil && resp.StatusCode == http.StatusOK {
                defer resp.Body.Close()
                
                var ptResp ProxyTracerResponse
                if err := json.NewDecoder(resp.Body).Decode(&ptResp); err == nil {
                    
                    // 3. Odrzuć połączenie, jeśli wykryto proxy/VPN
                    if ptResp.Proxy {
                        w.Header().Set("Content-Type", "application/json")
                        w.WriteHeader(http.StatusForbidden)
                        w.Write([]byte(`{"error": "Access Denied: VPN or Proxy detected."}`))
                        return // Natychmiast przerywa łańcuch procedur obsługi (short-circuit)
                    }
                }
            } else {
                // Fail-open: Zaloguj błąd i przepuść żądanie, aby zapewnić dostępność usługi
                log.Printf("ProxyTracer API error: %v", err)
            }
        }
 
        // 4. Ruch jest czysty, przejdź do kolejnego handlera
        next.ServeHTTP(w, r)
    })
}

Użycie

Po prostu opakuj główny router lub wybrane wrażliwe endpointy w middleware:


func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/api/secure-data", secureHandler)
 
    // Opakuj cały multiplekser (router) w middleware ProxyTracer
    log.Fatal(http.ListenAndServe(":8080", middleware.ProxyTracer(mux)))
}