Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a029a38787 | ||
|
|
4a83ab6bd6 | ||
|
|
2c09ab8f87 |
204
main.go
204
main.go
@@ -1,15 +1,17 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"encoding/json"
|
||||||
|
"html/template"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"crypto/rand"
|
"time"
|
||||||
"html/template"
|
|
||||||
"log"
|
|
||||||
"net/http"
|
|
||||||
"encoding/json"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -18,33 +20,70 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
debug = false
|
debug = false
|
||||||
templates = make(map[string]*template.Template)
|
templates = make(map[string]*template.Template)
|
||||||
AppVersion = "development"
|
AppVersion = "development"
|
||||||
counterFile = "/data/counter.txt"
|
counterFile = "/data/counter.txt"
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type responseWriter struct {
|
||||||
|
http.ResponseWriter
|
||||||
|
statusCode int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rw *responseWriter) WriteHeader(code int) {
|
||||||
|
rw.statusCode = code
|
||||||
|
rw.ResponseWriter.WriteHeader(code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func newResponseWriter(w http.ResponseWriter) *responseWriter {
|
||||||
|
return &responseWriter{w, http.StatusOK} // Default 200 OK
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoggingMiddleware(next http.Handler) http.Handler {
|
||||||
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
start := time.Now()
|
||||||
|
rw := newResponseWriter(w)
|
||||||
|
next.ServeHTTP(rw, r)
|
||||||
|
duration := time.Since(start)
|
||||||
|
clientIP := getClientIP(r)
|
||||||
|
userAgent := r.UserAgent()
|
||||||
|
// Format: IP - - [Datum] "Method Path Proto" Status Duration User-Agent
|
||||||
|
// "Combined Log Format"
|
||||||
|
log.Printf("%s - - [%s] \"%s %s %s\" %d %v\n",
|
||||||
|
clientIP,
|
||||||
|
time.Now().Format("02/Jan/2006:15:04:05 -0700"),
|
||||||
|
r.Method,
|
||||||
|
r.URL.Path,
|
||||||
|
r.Proto,
|
||||||
|
rw.statusCode,
|
||||||
|
duration,
|
||||||
|
userAgent,
|
||||||
|
)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func getClientIP(r *http.Request) string {
|
func getClientIP(r *http.Request) string {
|
||||||
// 1. Prüfe den X-Forwarded-For Header (Standard für Proxies)
|
// 1. Prüfe den X-Forwarded-For Header (Standard für Proxies)
|
||||||
xForwardedFor := r.Header.Get("X-Forwarded-For")
|
xForwardedFor := r.Header.Get("X-Forwarded-For")
|
||||||
if xForwardedFor != "" {
|
if xForwardedFor != "" {
|
||||||
// Der Header kann eine Liste von IPs sein (Client, Proxy1, Proxy2)
|
// Der Header kann eine Liste von IPs sein (Client, Proxy1, Proxy2)
|
||||||
// Die erste IP in der Liste ist die echte Client-IP
|
// Die erste IP in der Liste ist die echte Client-IP
|
||||||
ips := strings.Split(xForwardedFor, ",")
|
ips := strings.Split(xForwardedFor, ",")
|
||||||
return strings.TrimSpace(ips[0])
|
return strings.TrimSpace(ips[0])
|
||||||
}
|
}
|
||||||
|
|
||||||
// 2. Fallback auf X-Real-IP (oft von Traefik/Nginx gesetzt)
|
// 2. Fallback auf X-Real-IP (oft von Traefik/Nginx gesetzt)
|
||||||
xRealIP := r.Header.Get("X-Real-IP")
|
xRealIP := r.Header.Get("X-Real-IP")
|
||||||
if xRealIP != "" {
|
if xRealIP != "" {
|
||||||
return xRealIP
|
return xRealIP
|
||||||
}
|
}
|
||||||
|
|
||||||
// 3. Letzter Ausweg: Die direkte IP (wird in deinem Fall die Traefik-IP sein)
|
// 3. Letzter Ausweg: Die direkte IP (wird in deinem Fall die Traefik-IP sein)
|
||||||
// RemoteAddr enthält oft auch den Port (z.B. "127.0.0.1:1234")
|
// RemoteAddr enthält oft auch den Port (z.B. "127.0.0.1:1234")
|
||||||
ip, _, _ := net.SplitHostPort(r.RemoteAddr)
|
ip, _, _ := net.SplitHostPort(r.RemoteAddr)
|
||||||
return ip
|
return ip
|
||||||
}
|
}
|
||||||
|
|
||||||
// Diese Funktion wird nur intern aufgerufen, wenn der Mutex bereits gesperrt ist
|
// Diese Funktion wird nur intern aufgerufen, wenn der Mutex bereits gesperrt ist
|
||||||
@@ -69,37 +108,39 @@ func IncrementPasswordCount() {
|
|||||||
mu.Lock()
|
mu.Lock()
|
||||||
defer mu.Unlock()
|
defer mu.Unlock()
|
||||||
|
|
||||||
// Wir rufen jetzt die interne Funktion auf, die NICHT versucht,
|
// Wir rufen jetzt die interne Funktion auf, die NICHT versucht,
|
||||||
// den Mutex erneut zu sperren
|
// den Mutex erneut zu sperren
|
||||||
count := getCount()
|
count := getCount()
|
||||||
count++
|
count++
|
||||||
os.WriteFile(counterFile, []byte(strconv.Itoa(count)), 0644)
|
os.WriteFile(counterFile, []byte(strconv.Itoa(count)), 0644)
|
||||||
}
|
}
|
||||||
|
|
||||||
func loadTemplates() {
|
func loadTemplates() {
|
||||||
// 1. FuncMap definieren
|
// 1. FuncMap definieren
|
||||||
funcMap := template.FuncMap{
|
funcMap := template.FuncMap{
|
||||||
"getAppVersion": func() string { return AppVersion },
|
"getAppVersion": func() string { return AppVersion },
|
||||||
"getPassCount": func() int { return GetPasswordCount() },
|
"getPassCount": func() int { return GetPasswordCount() },
|
||||||
}
|
}
|
||||||
|
|
||||||
// 2. Templates mit FuncMap laden
|
// 2. Templates mit FuncMap laden
|
||||||
// Wir nutzen New("base.html"), da base.html meist das Haupt-Layout definiert
|
// Wir nutzen New("base.html"), da base.html meist das Haupt-Layout definiert
|
||||||
templates["index.html"] = template.Must(template.New("base.html").Funcs(funcMap).ParseFiles(
|
templates["index.html"] = template.Must(template.New("base.html").Funcs(funcMap).ParseFiles(
|
||||||
"templates/base.html",
|
"templates/base.html",
|
||||||
"templates/index.html",
|
"templates/index.html",
|
||||||
))
|
))
|
||||||
|
|
||||||
templates["help.html"] = template.Must(template.New("base.html").Funcs(funcMap).ParseFiles(
|
templates["help.html"] = template.Must(template.New("base.html").Funcs(funcMap).ParseFiles(
|
||||||
"templates/base.html",
|
"templates/base.html",
|
||||||
"templates/help.html",
|
"templates/help.html",
|
||||||
))
|
))
|
||||||
|
|
||||||
log.Printf("Alle Templates erfolgreich geladen")
|
log.Printf("Alle Templates erfolgreich geladen")
|
||||||
}
|
}
|
||||||
|
|
||||||
func generatePassword() string {
|
func generatePassword() string {
|
||||||
if debug { log.Printf("called generatePassword\n") }
|
if debug {
|
||||||
|
log.Printf("called generatePassword\n")
|
||||||
|
}
|
||||||
password := make([]byte, passwordLength)
|
password := make([]byte, passwordLength)
|
||||||
_, err := rand.Read(password)
|
_, err := rand.Read(password)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -113,37 +154,37 @@ func generatePassword() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func passwordHandler(w http.ResponseWriter, r *http.Request) {
|
func passwordHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
clientIP := getClientIP(r)
|
if debug {
|
||||||
log.Printf("passwordHandler aufrerufen von IP: %s\n", clientIP)
|
log.Printf("called passwordHandler\n")
|
||||||
if debug { log.Printf("called passwordHandler\n") }
|
}
|
||||||
password := generatePassword()
|
password := generatePassword()
|
||||||
currentCount := GetPasswordCount()
|
currentCount := GetPasswordCount()
|
||||||
response := map[string]interface{}{
|
response := map[string]interface{}{
|
||||||
"password": password,
|
"password": password,
|
||||||
"count": currentCount,
|
"count": currentCount,
|
||||||
}
|
}
|
||||||
w.Header().Set("Content-Type", "application/json")
|
w.Header().Set("Content-Type", "application/json")
|
||||||
err := json.NewEncoder(w).Encode(response)
|
err := json.NewEncoder(w).Encode(response)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Fehler beim Senden des JSON: %v", err)
|
log.Printf("Fehler beim Senden des JSON: %v", err)
|
||||||
http.Error(w, "Interner Fehler", http.StatusInternalServerError)
|
http.Error(w, "Interner Fehler", http.StatusInternalServerError)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func passwordAPIHandler(w http.ResponseWriter, r *http.Request) {
|
func passwordAPIHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
clientIP := getClientIP(r)
|
if debug {
|
||||||
log.Printf("passwordAPIHandler aufrerufen von IP: %s\n", clientIP)
|
log.Printf("called passwordHandler\n")
|
||||||
if debug { log.Printf("called passwordHandler\n") }
|
}
|
||||||
password := generatePassword()
|
password := generatePassword()
|
||||||
w.Header().Set("Content-Type", "text/plain")
|
w.Header().Set("Content-Type", "text/plain")
|
||||||
w.Write([]byte(password))
|
w.Write([]byte(password))
|
||||||
}
|
}
|
||||||
|
|
||||||
func indexHandler(w http.ResponseWriter, r *http.Request) {
|
func indexHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
clientIP := getClientIP(r)
|
if debug {
|
||||||
log.Printf("indexHandler aufrerufen von IP: %s\n", clientIP)
|
log.Printf("call indexHandler: Request %s %s\n", r.Method, r.URL)
|
||||||
if debug { log.Printf("call indexHandler: Request %s %s\n", r.Method, r.URL) }
|
}
|
||||||
password := generatePassword()
|
password := generatePassword()
|
||||||
//password := "load..."
|
//password := "load..."
|
||||||
data := struct {
|
data := struct {
|
||||||
@@ -151,7 +192,9 @@ func indexHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
}{
|
}{
|
||||||
Password: password,
|
Password: password,
|
||||||
}
|
}
|
||||||
if debug { log.Printf("prepare template for index\n") }
|
if debug {
|
||||||
|
log.Printf("prepare template for index\n")
|
||||||
|
}
|
||||||
err := templates["index.html"].ExecuteTemplate(w, "base.html", data)
|
err := templates["index.html"].ExecuteTemplate(w, "base.html", data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Fehler beim Rendern des Templates: %v", err)
|
log.Printf("Fehler beim Rendern des Templates: %v", err)
|
||||||
@@ -160,9 +203,9 @@ func indexHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func helpHandler(w http.ResponseWriter, r *http.Request) {
|
func helpHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
clientIP := getClientIP(r)
|
if debug {
|
||||||
log.Printf("helpHandler aufrerufen von IP: %s\n", clientIP)
|
log.Printf("call helpHandler\n")
|
||||||
if debug { log.Printf("call helpHandler\n") }
|
}
|
||||||
err := templates["help.html"].ExecuteTemplate(w, "base.html", nil)
|
err := templates["help.html"].ExecuteTemplate(w, "base.html", nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Fehler beim Rendern des Templates: %v", err)
|
log.Printf("Fehler beim Rendern des Templates: %v", err)
|
||||||
@@ -172,14 +215,19 @@ func helpHandler(w http.ResponseWriter, r *http.Request) {
|
|||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
loadTemplates()
|
loadTemplates()
|
||||||
fs := http.FileServer(http.Dir("static"))
|
mux := http.NewServeMux()
|
||||||
http.Handle("/static/", http.StripPrefix("/static/", fs))
|
|
||||||
|
|
||||||
http.HandleFunc("/", indexHandler)
|
fs := http.FileServer(http.Dir("static"))
|
||||||
http.HandleFunc("/api/password", passwordAPIHandler)
|
mux.Handle("/static/", http.StripPrefix("/static/", fs))
|
||||||
http.HandleFunc("/json/password", passwordHandler)
|
|
||||||
http.HandleFunc("/help", helpHandler)
|
mux.HandleFunc("/", indexHandler)
|
||||||
|
mux.HandleFunc("/api/password", passwordAPIHandler)
|
||||||
|
mux.HandleFunc("/json/password", passwordHandler)
|
||||||
|
mux.HandleFunc("/help", helpHandler)
|
||||||
|
|
||||||
|
loggingRouter := LoggingMiddleware(mux)
|
||||||
|
|
||||||
log.Println("Server läuft auf http://localhost:8080")
|
log.Println("Server läuft auf http://localhost:8080")
|
||||||
log.Fatal(http.ListenAndServe(":8080", nil))
|
|
||||||
|
log.Fatal(http.ListenAndServe(":8080", loggingRouter))
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user