Исправления в ветке main

This commit is contained in:
2025-11-09 14:02:35 +06:00
parent 487f3d484c
commit a78e1d24bf
9 changed files with 55 additions and 17 deletions

View File

@@ -8,6 +8,8 @@ import (
"fmt"
"net/http"
"os"
"strconv"
"strings"
"time"
"github.com/direct-dev-ru/linux-command-gpt/config"
@@ -172,9 +174,16 @@ func setCSRFCookie(w http.ResponseWriter, token string) {
MaxAge: CSRFTokenLifetimeSeconds, // Минимум 12 часов в секундах
}
// Добавляем домен если указан
// Устанавливаем Domain только если это не IP адрес и не 0.0.0.0
// При доступе по IP адресу не устанавливаем Domain, иначе cookie не будет работать
if config.AppConfig.Server.Domain != "" {
cookie.Domain = config.AppConfig.Server.Domain
domain := config.AppConfig.Server.Domain
// Проверяем, не является ли домен IP адресом или 0.0.0.0
if !isIPAddress(domain) && domain != "0.0.0.0" && domain != "::" && domain != "::1" {
cookie.Domain = domain
}
// Если domain пустой, 0.0.0.0 или IP адрес - не устанавливаем Domain
// Браузер автоматически применит cookie к текущему хосту
}
http.SetCookie(w, cookie)
@@ -192,9 +201,12 @@ func СlearCSRFCookie(w http.ResponseWriter) {
MaxAge: -1,
}
// Добавляем домен если указан
// Устанавливаем Domain только если это не IP адрес
if config.AppConfig.Server.Domain != "" {
cookie.Domain = config.AppConfig.Server.Domain
domain := config.AppConfig.Server.Domain
if !isIPAddress(domain) && domain != "0.0.0.0" && domain != "::" && domain != "::1" {
cookie.Domain = domain
}
}
http.SetCookie(w, cookie)
@@ -270,3 +282,17 @@ func InitCSRFManager() error {
func GetCSRFManager() *CSRFManager {
return csrfManager
}
// isIPAddress проверяет, является ли строка IPv4 адресом
func isIPAddress(s string) bool {
parts := strings.Split(s, ".")
if len(parts) != 4 {
return false
}
for _, part := range parts {
if num, err := strconv.Atoi(part); err != nil || num < 0 || num > 255 {
return false
}
}
return true
}