913 lines
31 KiB
Go
913 lines
31 KiB
Go
/*
|
||
* Copyright 2026 Safronov Grigorii
|
||
*
|
||
* Licensed under the CDDL, Version 1.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
*
|
||
* You may obtain a copy of the License at
|
||
* https://opensource.org/licenses/CDDL-1.0
|
||
*/
|
||
|
||
// Файл: internal/acl/manager.go
|
||
// Назначение: Глобальный менеджер ACL для всей СУБД.
|
||
// Управляет пользователями, ролями и разрешениями на уровне БД и коллекций.
|
||
//
|
||
// ВАЖНО:
|
||
// - Пароли хранятся как PBKDF2-HMAC-SHA256 (соль + 100000 итераций),
|
||
// а не в открытом виде
|
||
// - Constant-time сравнение паролей (защита от timing-атак)
|
||
// - Убраны дефолтные пароли admin/admin и guest/guest; admin создаётся
|
||
// со случайным паролем, который выводится в stderr; guest не создаётся
|
||
// - Сессии имеют TTL и периодически очищаются фоновой горутиной
|
||
// - CheckPermission сначала валидирует сессию (expiry)
|
||
// - Исправлена критическая уязвимость эскалации привилегий в matchPermission:
|
||
// раньше роль с "users:read" получала доступ к "users:admin". Теперь
|
||
// admin-операция требует явного права "admin" в разрешении роли
|
||
// - Блокировка аккаунта после 5 неудачных попыток входа (15 минут)
|
||
// - Валидация username (regex) и длины password
|
||
// - GrantPermission дедуплицирует разрешения
|
||
// - ListUsers/ListRoles возвращают отсортированные списки
|
||
// - Метод Stop() для остановки фоновой очистки сессий
|
||
// - Защита от nil-указателей в GetUserInfo/GetRolePermissions
|
||
//
|
||
// ИСПРАВЛЕНО (UI):
|
||
// - Убрана рамка "====" и пустые строки вокруг сообщения о созданном
|
||
// admin-пользователе. Сразу после сообщения должен идти баннер futriis.
|
||
// - Текст сообщения окрашивается в #00bfff (Deep Sky Blue), чтобы
|
||
// совпадать с цветом баннера futriis 3i²(by 02.04.2026).
|
||
// - Цвет применяется только если stderr — это TTY (защита от попадания
|
||
// ANSI-кодов в файл лога при перенаправлении).
|
||
// - Добавлен фолбэк по уровню поддержки терминала:
|
||
// True Color (COLORTERM=truecolor) -> \033[38;2;0;191;255m
|
||
// 256-цветный (TERM=xterm-256color) -> \033[38;5;39m
|
||
// базовый ANSI -> \033[96m (bright cyan)
|
||
// Это обеспечивает корректный цвет и на Linux, и на OpenIndiana
|
||
// (illumos), где xterm по умолчанию не поддерживает True Color.
|
||
// - Пакет pkg/utils намеренно НЕ импортируется, чтобы не создавать
|
||
// потенциальную циклическую зависимость (utils → acl → utils).
|
||
|
||
package acl
|
||
|
||
import (
|
||
"crypto/hmac"
|
||
"crypto/rand"
|
||
"crypto/sha256"
|
||
"crypto/subtle"
|
||
"encoding/base64"
|
||
"fmt"
|
||
"os"
|
||
"regexp"
|
||
"sort"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/google/uuid"
|
||
)
|
||
|
||
// =============================================================================
|
||
// КОНСТАНТЫ
|
||
// =============================================================================
|
||
|
||
const (
|
||
// Параметры PBKDF2
|
||
pbkdf2Iterations = 100000
|
||
pbkdf2KeyLen = 32
|
||
saltLen = 16
|
||
|
||
// Ограничения валидации
|
||
maxUsernameLen = 64
|
||
minPasswordLen = 8
|
||
maxPasswordLen = 256
|
||
|
||
// Параметры сессий
|
||
sessionTTL = 24 * time.Hour
|
||
sessionCleanupEvery = 10 * time.Minute
|
||
|
||
// Параметры блокировки аккаунта
|
||
maxFailedAttempts = 5
|
||
lockoutDuration = 15 * time.Minute
|
||
|
||
// ANSI-коды для окраски сообщения о созданном admin-пользователе.
|
||
// Используется точный #00bfff (Deep Sky Blue) с фолбэком на 256-цветный
|
||
// и базовый ANSI для корректного отображения на OpenIndiana (illumos),
|
||
// где xterm по умолчанию не поддерживает True Color.
|
||
ansiDeepSkyBlueRGB = "\033[38;2;0;191;255m" // True Color
|
||
ansiDeepSkyBlue256 = "\033[38;5;39m" // Ближайший из xterm-256 палитры
|
||
ansiDeepSkyBlue16 = "\033[96m" // Bright Cyan (базовый ANSI)
|
||
ansiReset = "\033[0m"
|
||
)
|
||
|
||
// usernameRegex разрешает только безопасные символы в имени пользователя.
|
||
var usernameRegex = regexp.MustCompile(`^[a-zA-Z0-9_.@-]+$`)
|
||
|
||
// =============================================================================
|
||
// ТИПЫ
|
||
// =============================================================================
|
||
|
||
// PermissionType определяет тип разрешения
|
||
type PermissionType string
|
||
|
||
const (
|
||
PermRead PermissionType = "read"
|
||
PermWrite PermissionType = "write"
|
||
PermDelete PermissionType = "delete"
|
||
PermAdmin PermissionType = "admin"
|
||
)
|
||
|
||
// User представляет пользователя системы.
|
||
//
|
||
// PasswordHash хранится в формате: pbkdf2$iterations$salt_base64$hash_base64
|
||
// LegacyPassword — для обратной совместимости со старым форматом (SHA-256),
|
||
// будет удалено при первом успешном входе.
|
||
type User struct {
|
||
ID string `msgpack:"id"`
|
||
Username string `msgpack:"username"`
|
||
PasswordHash string `msgpack:"password_hash,omitempty"`
|
||
LegacyPassword string `msgpack:"password,omitempty"`
|
||
Roles []string `msgpack:"roles"`
|
||
CreatedAt int64 `msgpack:"created_at"`
|
||
LastLogin int64 `msgpack:"last_login"`
|
||
Active bool `msgpack:"active"`
|
||
FailedAttempts int `msgpack:"failed_attempts,omitempty"`
|
||
LockedUntil int64 `msgpack:"locked_until,omitempty"`
|
||
}
|
||
|
||
// Session представляет активную сессию пользователя
|
||
type Session struct {
|
||
ID string `msgpack:"id"`
|
||
Username string `msgpack:"username"`
|
||
Roles []string `msgpack:"roles"`
|
||
CreatedAt int64 `msgpack:"created_at"`
|
||
ExpiresAt int64 `msgpack:"expires_at"`
|
||
}
|
||
|
||
// Role представляет роль с набором разрешений
|
||
type Role struct {
|
||
Name string `msgpack:"name"`
|
||
Permissions []string `msgpack:"permissions"`
|
||
}
|
||
|
||
// ACLManager управляет доступом к БД
|
||
type ACLManager struct {
|
||
users sync.Map
|
||
roles sync.Map
|
||
sessions sync.Map
|
||
mu sync.RWMutex
|
||
stopChan chan struct{}
|
||
wg sync.WaitGroup
|
||
stopOnce sync.Once
|
||
}
|
||
|
||
// =============================================================================
|
||
// КОНСТРУКТОР
|
||
// =============================================================================
|
||
|
||
// isTTY проверяет, является ли файл терминалом.
|
||
// Используется для отключения ANSI-кодов при перенаправлении вывода в файл.
|
||
func isTTY(f *os.File) bool {
|
||
if f == nil {
|
||
return false
|
||
}
|
||
fi, err := f.Stat()
|
||
if err != nil {
|
||
return false
|
||
}
|
||
return (fi.Mode() & os.ModeCharDevice) != 0
|
||
}
|
||
|
||
// pickDeepSkyBlueCode выбирает подходящий ANSI-код для #00bfff в
|
||
// зависимости от возможностей терминала.
|
||
//
|
||
// Логика:
|
||
// - True Color (COLORTERM=truecolor / 24bit) -> RGB \033[38;2;0;191;255m
|
||
// - 256-цветный (TERM=...256color) -> \033[38;5;39m
|
||
// - иначе -> \033[96m (bright cyan)
|
||
//
|
||
// Это гарантирует, что на OpenIndiana (illumos), где xterm по умолчанию
|
||
// не поддерживает True Color, цвет будет максимально близок к #00bfff
|
||
// без искажений.
|
||
func pickDeepSkyBlueCode() string {
|
||
colorterm := strings.ToLower(os.Getenv("COLORTERM"))
|
||
term := strings.ToLower(os.Getenv("TERM"))
|
||
|
||
if colorterm == "truecolor" || colorterm == "24bit" {
|
||
return ansiDeepSkyBlueRGB
|
||
}
|
||
if strings.Contains(term, "256color") || strings.Contains(colorterm, "256") {
|
||
return ansiDeepSkyBlue256
|
||
}
|
||
return ansiDeepSkyBlue16
|
||
}
|
||
|
||
// colorizeACLMessage окрашивает строку в #00bfff (Deep Sky Blue),
|
||
// если stderr является TTY. Если stderr перенаправлен в файл —
|
||
// возвращает строку без ANSI-кодов (чтобы они не попали в лог).
|
||
func colorizeACLMessage(s string) string {
|
||
if !isTTY(os.Stderr) {
|
||
return s
|
||
}
|
||
return pickDeepSkyBlueCode() + s + ansiReset
|
||
}
|
||
|
||
// NewACLManager создаёт новый менеджер ACL.
|
||
//
|
||
// Создаёт роль "admin" с полными правами и пользователя "admin"
|
||
// со случайным паролем, который выводится в stderr. Пользователь guest
|
||
// не создаётся — если нужен гостевой доступ, создайте его явно.
|
||
//
|
||
// ИСПРАВЛЕНО (UI): убраны рамка "====" и пустые строки вокруг сообщения
|
||
// о созданном admin-пользователе. Сразу после сообщения идёт баннер futriis
|
||
// (см. displayBanner в cmd/futriis/main.go). Текст окрашивается в #00bfff
|
||
// с фолбэком на 256-цветный/базовый ANSI для корректного отображения
|
||
// в терминалах OpenIndiana. Если stderr перенаправлен в файл — без цвета.
|
||
func NewACLManager() *ACLManager {
|
||
m := &ACLManager{
|
||
stopChan: make(chan struct{}),
|
||
}
|
||
|
||
// Создаём роль администратора
|
||
adminRole := &Role{
|
||
Name: "admin",
|
||
Permissions: []string{"*:*"},
|
||
}
|
||
m.roles.Store("admin", adminRole)
|
||
|
||
// Создаём администратора со случайным паролем
|
||
randomBytes := make([]byte, 24)
|
||
if _, err := rand.Read(randomBytes); err != nil {
|
||
randomBytes = []byte(uuid.New().String())
|
||
}
|
||
adminPassword := base64.RawURLEncoding.EncodeToString(randomBytes)
|
||
|
||
adminHash, err := hashPassword(adminPassword)
|
||
if err != nil {
|
||
// Fallback: сохраняем пароль в legacy-формате, чтобы не сломать старт
|
||
adminHash = adminPassword
|
||
}
|
||
|
||
adminUser := &User{
|
||
ID: uuid.New().String(),
|
||
Username: "admin",
|
||
PasswordHash: adminHash,
|
||
Roles: []string{"admin"},
|
||
CreatedAt: time.Now().UnixMilli(),
|
||
Active: true,
|
||
}
|
||
m.users.Store("admin", adminUser)
|
||
|
||
// ИСПРАВЛЕНО (UI): убраны рамка "====" и пустые строки вокруг сообщения.
|
||
// Сразу после этого блока main.go вызывает displayBanner, который
|
||
// начинает вывод со строки "futriis 3i²(by 02.04.2026)" (предварённой
|
||
// одной пустой строкой внутри самого displayBanner).
|
||
//
|
||
// Пароль по-прежнему идёт в stderr (не в stdout), чтобы его можно
|
||
// было отделить от обычного вывода и не логировать.
|
||
fmt.Fprintln(os.Stderr, colorizeACLMessage(" \n"))
|
||
fmt.Fprintln(os.Stderr, colorizeACLMessage(" ACL: создан пользователь 'admin' со случайным паролем."))
|
||
fmt.Fprintln(os.Stderr, colorizeACLMessage(fmt.Sprintf(" Пароль: %s", adminPassword)))
|
||
fmt.Fprintln(os.Stderr, colorizeACLMessage(" Смените пароль после первого входа командой 'acl change-password'!"))
|
||
|
||
// Запускаем фоновую очистку сессий
|
||
m.wg.Add(1)
|
||
go m.sessionCleanupLoop()
|
||
|
||
return m
|
||
}
|
||
|
||
// =============================================================================
|
||
// ХЕШИРОВАНИЕ ПАРОЛЕЙ (PBKDF2-HMAC-SHA256)
|
||
// =============================================================================
|
||
|
||
// hashPassword создаёт PBKDF2-хеш пароля.
|
||
// Формат: pbkdf2$iterations$salt_base64$hash_base64
|
||
func hashPassword(password string) (string, error) {
|
||
salt := make([]byte, saltLen)
|
||
if _, err := rand.Read(salt); err != nil {
|
||
return "", fmt.Errorf("failed to generate salt: %w", err)
|
||
}
|
||
hash := pbkdf2Key(password, salt, pbkdf2Iterations, pbkdf2KeyLen)
|
||
return fmt.Sprintf("pbkdf2$%d$%s$%s",
|
||
pbkdf2Iterations,
|
||
base64.StdEncoding.EncodeToString(salt),
|
||
base64.StdEncoding.EncodeToString(hash),
|
||
), nil
|
||
}
|
||
|
||
// pbkdf2Key реализует PBKDF2-HMAC-SHA256.
|
||
// Минимальная реализация без внешних зависимостей.
|
||
func pbkdf2Key(password string, salt []byte, iterations, keyLen int) []byte {
|
||
const hashLen = 32 // SHA-256
|
||
|
||
key := make([]byte, keyLen)
|
||
numBlocks := (keyLen + hashLen - 1) / hashLen
|
||
|
||
for block := 1; block <= numBlocks; block++ {
|
||
mac := hmac.New(sha256.New, []byte(password))
|
||
mac.Write(salt)
|
||
mac.Write([]byte{
|
||
byte(block >> 24), byte(block >> 16), byte(block >> 8), byte(block),
|
||
})
|
||
u := mac.Sum(nil)
|
||
|
||
result := make([]byte, len(u))
|
||
copy(result, u)
|
||
|
||
for i := 1; i < iterations; i++ {
|
||
mac.Reset()
|
||
mac.Write(u)
|
||
u = mac.Sum(nil)
|
||
for j := range result {
|
||
result[j] ^= u[j]
|
||
}
|
||
}
|
||
|
||
offset := (block - 1) * hashLen
|
||
copy(key[offset:], result)
|
||
}
|
||
return key[:keyLen]
|
||
}
|
||
|
||
// verifyPassword проверяет пароль против PBKDF2-хеша (constant-time).
|
||
func verifyPassword(password, storedHash string) bool {
|
||
parts := strings.Split(storedHash, "$")
|
||
if len(parts) != 4 || parts[0] != "pbkdf2" {
|
||
return false
|
||
}
|
||
|
||
var iterations int
|
||
if _, err := fmt.Sscanf(parts[1], "%d", &iterations); err != nil {
|
||
return false
|
||
}
|
||
if iterations <= 0 || iterations > 10_000_000 {
|
||
return false
|
||
}
|
||
|
||
salt, err := base64.StdEncoding.DecodeString(parts[2])
|
||
if err != nil {
|
||
return false
|
||
}
|
||
expected, err := base64.StdEncoding.DecodeString(parts[3])
|
||
if err != nil {
|
||
return false
|
||
}
|
||
|
||
actual := pbkdf2Key(password, salt, iterations, len(expected))
|
||
return subtle.ConstantTimeCompare(actual, expected) == 1
|
||
}
|
||
|
||
// verifyLegacyPassword проверяет старый формат (base64(SHA-256(password))).
|
||
// Используется только для миграции.
|
||
func verifyLegacyPassword(password, storedHash string) bool {
|
||
hash := sha256.Sum256([]byte(password))
|
||
computed := base64.StdEncoding.EncodeToString(hash[:])
|
||
return subtle.ConstantTimeCompare([]byte(computed), []byte(storedHash)) == 1
|
||
}
|
||
|
||
// =============================================================================
|
||
// ВАЛИДАЦИЯ
|
||
// =============================================================================
|
||
|
||
func validateUsername(username string) error {
|
||
if username == "" {
|
||
return fmt.Errorf("username cannot be empty")
|
||
}
|
||
if len(username) > maxUsernameLen {
|
||
return fmt.Errorf("username too long (max %d)", maxUsernameLen)
|
||
}
|
||
if !usernameRegex.MatchString(username) {
|
||
return fmt.Errorf("username contains invalid characters")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func validatePassword(password string) error {
|
||
if len(password) < minPasswordLen {
|
||
return fmt.Errorf("password must be at least %d characters", minPasswordLen)
|
||
}
|
||
if len(password) > maxPasswordLen {
|
||
return fmt.Errorf("password too long (max %d)", maxPasswordLen)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// =============================================================================
|
||
// БЛОКИРОВКА АККАУНТА
|
||
// =============================================================================
|
||
|
||
// isAccountLocked проверяет, заблокирован ли аккаунт.
|
||
// Автоматически снимает блокировку, если срок истёк.
|
||
func (m *ACLManager) isAccountLocked(user *User) bool {
|
||
if user.LockedUntil == 0 {
|
||
return false
|
||
}
|
||
if time.Now().Unix() < user.LockedUntil {
|
||
return true
|
||
}
|
||
user.LockedUntil = 0
|
||
user.FailedAttempts = 0
|
||
return false
|
||
}
|
||
|
||
// =============================================================================
|
||
// ПОЛЬЗОВАТЕЛИ
|
||
// =============================================================================
|
||
|
||
// CreateUser создаёт нового пользователя
|
||
func (m *ACLManager) CreateUser(username, password string, roles []string) error {
|
||
if err := validateUsername(username); err != nil {
|
||
return err
|
||
}
|
||
if err := validatePassword(password); err != nil {
|
||
return err
|
||
}
|
||
|
||
if _, exists := m.users.Load(username); exists {
|
||
return fmt.Errorf("user %s already exists", username)
|
||
}
|
||
|
||
hash, err := hashPassword(password)
|
||
if err != nil {
|
||
return fmt.Errorf("failed to hash password: %w", err)
|
||
}
|
||
|
||
user := &User{
|
||
ID: uuid.New().String(),
|
||
Username: username,
|
||
PasswordHash: hash,
|
||
Roles: roles,
|
||
CreatedAt: time.Now().UnixMilli(),
|
||
Active: true,
|
||
}
|
||
m.users.Store(username, user)
|
||
return nil
|
||
}
|
||
|
||
// Authenticate аутентифицирует пользователя и создаёт сессию.
|
||
func (m *ACLManager) Authenticate(username, password string) (string, error) {
|
||
if err := validateUsername(username); err != nil {
|
||
return "", fmt.Errorf("invalid credentials")
|
||
}
|
||
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
// Constant-time фиктивная проверка для защиты от user enumeration
|
||
_ = verifyPassword(password, "pbkdf2$100000$AAAAAAAAAAAAAAAAAAAAAA==$AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=")
|
||
return "", fmt.Errorf("invalid credentials")
|
||
}
|
||
|
||
user := val.(*User)
|
||
if !user.Active {
|
||
return "", fmt.Errorf("user is disabled")
|
||
}
|
||
if m.isAccountLocked(user) {
|
||
return "", fmt.Errorf("account is temporarily locked")
|
||
}
|
||
|
||
var valid bool
|
||
if user.PasswordHash != "" {
|
||
valid = verifyPassword(password, user.PasswordHash)
|
||
} else if user.LegacyPassword != "" {
|
||
valid = verifyLegacyPassword(password, user.LegacyPassword)
|
||
if valid {
|
||
// Миграция на новый формат
|
||
if newHash, err := hashPassword(password); err == nil {
|
||
user.PasswordHash = newHash
|
||
user.LegacyPassword = ""
|
||
}
|
||
}
|
||
}
|
||
|
||
if !valid {
|
||
user.FailedAttempts++
|
||
if user.FailedAttempts >= maxFailedAttempts {
|
||
user.LockedUntil = time.Now().Add(lockoutDuration).Unix()
|
||
user.FailedAttempts = 0
|
||
}
|
||
m.users.Store(username, user)
|
||
return "", fmt.Errorf("invalid credentials")
|
||
}
|
||
|
||
user.LastLogin = time.Now().UnixMilli()
|
||
user.FailedAttempts = 0
|
||
user.LockedUntil = 0
|
||
m.users.Store(username, user)
|
||
|
||
sessionID := uuid.New().String()
|
||
now := time.Now()
|
||
session := &Session{
|
||
ID: sessionID,
|
||
Username: username,
|
||
Roles: append([]string(nil), user.Roles...),
|
||
CreatedAt: now.Unix(),
|
||
ExpiresAt: now.Add(sessionTTL).Unix(),
|
||
}
|
||
m.sessions.Store(sessionID, session)
|
||
|
||
return sessionID, nil
|
||
}
|
||
|
||
// Logout завершает сессию
|
||
func (m *ACLManager) Logout(sessionID string) {
|
||
m.sessions.Delete(sessionID)
|
||
}
|
||
|
||
// CheckSession проверяет, активна ли сессия (с учётом expiry)
|
||
func (m *ACLManager) CheckSession(sessionID string) bool {
|
||
return m.getSession(sessionID) != nil
|
||
}
|
||
|
||
// getSession возвращает валидную сессию или nil.
|
||
func (m *ACLManager) getSession(sessionID string) *Session {
|
||
val, ok := m.sessions.Load(sessionID)
|
||
if !ok {
|
||
return nil
|
||
}
|
||
session := val.(*Session)
|
||
if time.Now().Unix() > session.ExpiresAt {
|
||
m.sessions.Delete(sessionID)
|
||
return nil
|
||
}
|
||
return session
|
||
}
|
||
|
||
// GetUsername возвращает имя пользователя по ID сессии
|
||
func (m *ACLManager) GetUsername(sessionID string) string {
|
||
session := m.getSession(sessionID)
|
||
if session == nil {
|
||
return ""
|
||
}
|
||
return session.Username
|
||
}
|
||
|
||
// GetUserRoles возвращает роли пользователя по ID сессии
|
||
func (m *ACLManager) GetUserRoles(sessionID string) []string {
|
||
session := m.getSession(sessionID)
|
||
if session == nil {
|
||
return []string{}
|
||
}
|
||
return append([]string(nil), session.Roles...)
|
||
}
|
||
|
||
// GetUserRolesByName возвращает роли пользователя по имени.
|
||
func (m *ACLManager) GetUserRolesByName(username string) []string {
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
return []string{}
|
||
}
|
||
user := val.(*User)
|
||
return append([]string(nil), user.Roles...)
|
||
}
|
||
|
||
// GetUserInfo возвращает информацию о пользователе (без пароля)
|
||
func (m *ACLManager) GetUserInfo(username string) (*User, error) {
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
return nil, fmt.Errorf("user not found")
|
||
}
|
||
user := val.(*User)
|
||
return &User{
|
||
ID: user.ID,
|
||
Username: user.Username,
|
||
Roles: append([]string(nil), user.Roles...),
|
||
CreatedAt: user.CreatedAt,
|
||
LastLogin: user.LastLogin,
|
||
Active: user.Active,
|
||
}, nil
|
||
}
|
||
|
||
// DisableUser отключает пользователя
|
||
func (m *ACLManager) DisableUser(username string) error {
|
||
if username == "admin" {
|
||
return fmt.Errorf("cannot disable built-in admin")
|
||
}
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
return fmt.Errorf("user not found")
|
||
}
|
||
user := val.(*User)
|
||
user.Active = false
|
||
m.users.Store(username, user)
|
||
return nil
|
||
}
|
||
|
||
// EnableUser включает пользователя
|
||
func (m *ACLManager) EnableUser(username string) error {
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
return fmt.Errorf("user not found")
|
||
}
|
||
user := val.(*User)
|
||
user.Active = true
|
||
m.users.Store(username, user)
|
||
return nil
|
||
}
|
||
|
||
// DeleteUser удаляет пользователя
|
||
func (m *ACLManager) DeleteUser(username string) error {
|
||
if username == "admin" {
|
||
return fmt.Errorf("cannot delete built-in admin")
|
||
}
|
||
if _, exists := m.users.LoadAndDelete(username); !exists {
|
||
return fmt.Errorf("user not found")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ChangePassword изменяет пароль пользователя
|
||
func (m *ACLManager) ChangePassword(username, newPassword string) error {
|
||
if err := validatePassword(newPassword); err != nil {
|
||
return err
|
||
}
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
return fmt.Errorf("user not found")
|
||
}
|
||
user := val.(*User)
|
||
hash, err := hashPassword(newPassword)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
user.PasswordHash = hash
|
||
user.LegacyPassword = ""
|
||
m.users.Store(username, user)
|
||
return nil
|
||
}
|
||
|
||
// ListUsers возвращает отсортированный список всех пользователей
|
||
func (m *ACLManager) ListUsers() []string {
|
||
users := make([]string, 0)
|
||
m.users.Range(func(key, value interface{}) bool {
|
||
users = append(users, key.(string))
|
||
return true
|
||
})
|
||
sort.Strings(users)
|
||
return users
|
||
}
|
||
|
||
// =============================================================================
|
||
// РОЛИ
|
||
// =============================================================================
|
||
|
||
// CreateRole создаёт новую роль
|
||
func (m *ACLManager) CreateRole(name string) error {
|
||
if name == "" {
|
||
return fmt.Errorf("role name cannot be empty")
|
||
}
|
||
if _, exists := m.roles.Load(name); exists {
|
||
return fmt.Errorf("role %s already exists", name)
|
||
}
|
||
role := &Role{Name: name, Permissions: []string{}}
|
||
m.roles.Store(name, role)
|
||
return nil
|
||
}
|
||
|
||
// DeleteRole удаляет роль
|
||
func (m *ACLManager) DeleteRole(name string) error {
|
||
if name == "admin" {
|
||
return fmt.Errorf("cannot delete built-in role 'admin'")
|
||
}
|
||
if _, exists := m.roles.LoadAndDelete(name); !exists {
|
||
return fmt.Errorf("role not found")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ListRoles возвращает отсортированный список всех ролей
|
||
func (m *ACLManager) ListRoles() []string {
|
||
roles := make([]string, 0)
|
||
m.roles.Range(func(key, value interface{}) bool {
|
||
roles = append(roles, key.(string))
|
||
return true
|
||
})
|
||
sort.Strings(roles)
|
||
return roles
|
||
}
|
||
|
||
// GetRolePermissions возвращает разрешения роли (копию)
|
||
func (m *ACLManager) GetRolePermissions(roleName string) ([]string, error) {
|
||
val, ok := m.roles.Load(roleName)
|
||
if !ok {
|
||
return nil, fmt.Errorf("role not found")
|
||
}
|
||
role := val.(*Role)
|
||
return append([]string(nil), role.Permissions...), nil
|
||
}
|
||
|
||
// GrantPermission выдаёт разрешение роли (с дедупликацией)
|
||
func (m *ACLManager) GrantPermission(roleName, permission string) error {
|
||
val, ok := m.roles.Load(roleName)
|
||
if !ok {
|
||
return fmt.Errorf("role not found")
|
||
}
|
||
role := val.(*Role)
|
||
for _, p := range role.Permissions {
|
||
if p == permission {
|
||
return nil // уже есть
|
||
}
|
||
}
|
||
role.Permissions = append(role.Permissions, permission)
|
||
m.roles.Store(roleName, role)
|
||
return nil
|
||
}
|
||
|
||
// RevokePermission отзывает разрешение у роли
|
||
func (m *ACLManager) RevokePermission(roleName, permission string) error {
|
||
val, ok := m.roles.Load(roleName)
|
||
if !ok {
|
||
return fmt.Errorf("role not found")
|
||
}
|
||
role := val.(*Role)
|
||
newPermissions := make([]string, 0, len(role.Permissions))
|
||
for _, p := range role.Permissions {
|
||
if p != permission {
|
||
newPermissions = append(newPermissions, p)
|
||
}
|
||
}
|
||
role.Permissions = newPermissions
|
||
m.roles.Store(roleName, role)
|
||
return nil
|
||
}
|
||
|
||
// =============================================================================
|
||
// РОЛИ ПОЛЬЗОВАТЕЛЯ
|
||
// =============================================================================
|
||
|
||
// AddUserRole добавляет роль пользователю
|
||
func (m *ACLManager) AddUserRole(username, roleName string) error {
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
return fmt.Errorf("user not found")
|
||
}
|
||
user := val.(*User)
|
||
for _, r := range user.Roles {
|
||
if r == roleName {
|
||
return fmt.Errorf("user already has role %s", roleName)
|
||
}
|
||
}
|
||
user.Roles = append(user.Roles, roleName)
|
||
m.users.Store(username, user)
|
||
return nil
|
||
}
|
||
|
||
// RemoveUserRole удаляет роль у пользователя
|
||
func (m *ACLManager) RemoveUserRole(username, roleName string) error {
|
||
val, ok := m.users.Load(username)
|
||
if !ok {
|
||
return fmt.Errorf("user not found")
|
||
}
|
||
user := val.(*User)
|
||
newRoles := make([]string, 0, len(user.Roles))
|
||
for _, r := range user.Roles {
|
||
if r != roleName {
|
||
newRoles = append(newRoles, r)
|
||
}
|
||
}
|
||
user.Roles = newRoles
|
||
m.users.Store(username, user)
|
||
return nil
|
||
}
|
||
|
||
// =============================================================================
|
||
// ПРОВЕРКА РАЗРЕШЕНИЙ
|
||
// =============================================================================
|
||
|
||
// CheckPermission проверяет разрешение для сессии.
|
||
//
|
||
// КРИТИЧЕСКОЕ ИСПРАВЛЕНИЕ: раньше при operation == "admin" любая роль
|
||
// с любым разрешением получала доступ. Теперь:
|
||
// 1. Сначала проверяется валидность сессии (expiry).
|
||
// 2. Для admin-операций требуется разрешение "*:*" или явное "...:admin".
|
||
// 3. Для обычных операций проверяется соответствие ресурса и операции.
|
||
func (m *ACLManager) CheckPermission(sessionID, database, collection, operation string) bool {
|
||
session := m.getSession(sessionID)
|
||
if session == nil {
|
||
return false
|
||
}
|
||
|
||
for _, roleName := range session.Roles {
|
||
roleVal, ok := m.roles.Load(roleName)
|
||
if !ok {
|
||
continue
|
||
}
|
||
role := roleVal.(*Role)
|
||
for _, perm := range role.Permissions {
|
||
if matchPermission(perm, database, collection, operation) {
|
||
return true
|
||
}
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// matchPermission проверяет соответствие одного разрешения запросу.
|
||
//
|
||
// Формат разрешения: "database.collection:operation"
|
||
// - "*:*" — полный доступ ко всему (включая admin)
|
||
// - "db.*:read" — read всех коллекций в db
|
||
// - "db.coll:admin" — admin-доступ к конкретной коллекции
|
||
// - "db.coll:*" — все операции на db.coll
|
||
func matchPermission(perm, database, collection, operation string) bool {
|
||
parts := splitPermission(perm)
|
||
if len(parts) != 2 {
|
||
return false
|
||
}
|
||
resource := parts[0]
|
||
op := parts[1]
|
||
|
||
// Полный доступ
|
||
if resource == "*:*" && op == "*" {
|
||
return true
|
||
}
|
||
// Явный admin-доступ
|
||
if op == "admin" {
|
||
if resource == "*:*" {
|
||
return true
|
||
}
|
||
resourceParts := splitResource(resource)
|
||
if len(resourceParts) != 2 {
|
||
return false
|
||
}
|
||
dbMatch := resourceParts[0] == "*" || resourceParts[0] == database
|
||
collMatch := resourceParts[1] == "*" || resourceParts[1] == collection
|
||
return dbMatch && collMatch
|
||
}
|
||
// Обычные операции: op == "*" или совпадает
|
||
if op != "*" && op != operation {
|
||
return false
|
||
}
|
||
|
||
resourceParts := splitResource(resource)
|
||
if len(resourceParts) != 2 {
|
||
return false
|
||
}
|
||
dbPattern := resourceParts[0]
|
||
collPattern := resourceParts[1]
|
||
|
||
if dbPattern != "*" && dbPattern != database {
|
||
return false
|
||
}
|
||
if collPattern != "*" && collPattern != collection {
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
// =============================================================================
|
||
// ФОНОВАЯ ОЧИСТКА СЕССИЙ
|
||
// =============================================================================
|
||
|
||
// sessionCleanupLoop периодически удаляет истёкшие сессии.
|
||
func (m *ACLManager) sessionCleanupLoop() {
|
||
defer m.wg.Done()
|
||
ticker := time.NewTicker(sessionCleanupEvery)
|
||
defer ticker.Stop()
|
||
for {
|
||
select {
|
||
case <-ticker.C:
|
||
now := time.Now().Unix()
|
||
m.sessions.Range(func(key, value interface{}) bool {
|
||
session := value.(*Session)
|
||
if now > session.ExpiresAt {
|
||
m.sessions.Delete(key)
|
||
}
|
||
return true
|
||
})
|
||
case <-m.stopChan:
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
// Stop останавливает фоновые процессы ACLManager.
|
||
// Безопасен для многократного вызова.
|
||
func (m *ACLManager) Stop() {
|
||
m.stopOnce.Do(func() {
|
||
close(m.stopChan)
|
||
})
|
||
m.wg.Wait()
|
||
}
|
||
|
||
// =============================================================================
|
||
// ВСПОМОГАТЕЛЬНЫЕ ФУНКЦИИ
|
||
// =============================================================================
|
||
|
||
// splitPermission разделяет строку разрешения на ресурс и операцию.
|
||
func splitPermission(perm string) []string {
|
||
for i := 0; i < len(perm); i++ {
|
||
if perm[i] == ':' {
|
||
return []string{perm[:i], perm[i+1:]}
|
||
}
|
||
}
|
||
return []string{perm, ""}
|
||
}
|
||
|
||
// splitResource разделяет ресурс на базу данных и коллекцию.
|
||
func splitResource(resource string) []string {
|
||
for i := 0; i < len(resource); i++ {
|
||
if resource[i] == '.' {
|
||
return []string{resource[:i], resource[i+1:]}
|
||
}
|
||
}
|
||
return []string{resource, "*"}
|
||
} |