Update internal/acl/manger.go

This commit is contained in:
gvsafronov committed 2026-09-21 22:58:54 +00:00
1 parent 1feb3348a4
commit 6d92830086
1 file changed
+345 -80
+345 -80
View File
@@ -30,21 +30,6 @@
// - Метод Stop() для остановки фоновой очистки сессий // - Метод Stop() для остановки фоновой очистки сессий
// - Защита от nil-указателей в GetUserInfo/GetRolePermissions // - Защита от 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 package acl
@@ -83,6 +68,7 @@ const (
// Параметры сессий // Параметры сессий
sessionTTL = 24 * time.Hour sessionTTL = 24 * time.Hour
sessionCleanupEvery = 10 * time.Minute sessionCleanupEvery = 10 * time.Minute
maxSessionsPerUser = 16
// Параметры блокировки аккаунта // Параметры блокировки аккаунта
maxFailedAttempts = 5 maxFailedAttempts = 5
@@ -101,6 +87,13 @@ const (
// usernameRegex разрешает только безопасные символы в имени пользователя. // usernameRegex разрешает только безопасные символы в имени пользователя.
var usernameRegex = regexp.MustCompile(`^[a-zA-Z0-9_.@-]+$`) var usernameRegex = regexp.MustCompile(`^[a-zA-Z0-9_.@-]+$`)
// permissionRegex валидирует формат разрешения:
// - "*:*" — полный доступ
// - "db.coll:op" — конкретный ресурс и операция
// - "db.*:op" — wildcard коллекции
// где db/coll/op — непустые строки из [A-Za-z0-9_*.-] (op — [A-Za-z0-9_*]).
var permissionRegex = regexp.MustCompile(`^([A-Za-z0-9_.*-]+)\.([A-Za-z0-9_*.-]+):([A-Za-z0-9_*]+)$|^\*:\*$`)
// ============================================================================= // =============================================================================
// ТИПЫ // ТИПЫ
// ============================================================================= // =============================================================================
@@ -115,6 +108,13 @@ const (
PermAdmin PermissionType = "admin" PermAdmin PermissionType = "admin"
) )
// Logger — минимальный интерфейс для логирования внутри ACL.
// Сделан локально, чтобы не тянуть internal/log (и не создавать циклов).
type Logger interface {
Warn(msg string)
Info(msg string)
}
// User представляет пользователя системы. // User представляет пользователя системы.
// //
// PasswordHash хранится в формате: pbkdf2$iterations$salt_base64$hash_base64 // PasswordHash хранится в формате: pbkdf2$iterations$salt_base64$hash_base64
@@ -133,13 +133,18 @@ type User struct {
LockedUntil int64 `msgpack:"locked_until,omitempty"` LockedUntil int64 `msgpack:"locked_until,omitempty"`
} }
// Session представляет активную сессию пользователя // Session представляет активную сессию пользователя.
//
// ИСПРАВЛЕНО: добавлены CreatedByIP и UserAgent — для привязки сессии
// к клиенту (мягкая защита от session hijacking).
type Session struct { type Session struct {
ID string `msgpack:"id"` ID string `msgpack:"id"`
Username string `msgpack:"username"` Username string `msgpack:"username"`
Roles []string `msgpack:"roles"` Roles []string `msgpack:"roles"`
CreatedAt int64 `msgpack:"created_at"` CreatedAt int64 `msgpack:"created_at"`
ExpiresAt int64 `msgpack:"expires_at"` ExpiresAt int64 `msgpack:"expires_at"`
CreatedByIP string `msgpack:"created_by_ip,omitempty"`
UserAgent string `msgpack:"user_agent,omitempty"`
} }
// Role представляет роль с набором разрешений // Role представляет роль с набором разрешений
@@ -148,12 +153,22 @@ type Role struct {
Permissions []string `msgpack:"permissions"` Permissions []string `msgpack:"permissions"`
} }
// ACLManager управляет доступом к БД // ACLManager управляет доступом к БД.
//
// ВАЖНО: sync.Map используется для чтений без блокировок. Все
// read-modify-write операции (Authenticate, ChangePassword, ...)
// защищены m.mu — иначе возможны потерянные обновления.
type ACLManager struct { type ACLManager struct {
users sync.Map users sync.Map // map[string]*User
roles sync.Map roles sync.Map // map[string]*Role
sessions sync.Map sessions sync.Map // map[string]*Session
// mu защищает read-modify-write операции над users/roles/sessions.
// Чтения через sync.Map.Load не блокируются.
mu sync.RWMutex mu sync.RWMutex
logger Logger
stopChan chan struct{} stopChan chan struct{}
wg sync.WaitGroup wg sync.WaitGroup
stopOnce sync.Once stopOnce sync.Once
@@ -216,11 +231,10 @@ func colorizeACLMessage(s string) string {
// со случайным паролем, который выводится в stderr. Пользователь guest // со случайным паролем, который выводится в stderr. Пользователь guest
// не создаётся — если нужен гостевой доступ, создайте его явно. // не создаётся — если нужен гостевой доступ, создайте его явно.
// //
// ИСПРАВЛЕНО (UI): убраны рамка "====" и пустые строки вокруг сообщения // ИСПРАВЛЕНО (UI): убраны рамка "====" и пустые строки вокруг сообщения.
// о созданном admin-пользователе. Сразу после сообщения идёт баннер futriis // ИСПРАВЛЕНО (безопасность): при сбое PBKDF2 больше НЕ используем пароль
// (см. displayBanner в cmd/futriis/main.go). Текст окрашивается в #00bfff // в открытом виде в качестве хеша — паникуем с понятной ошибкой, чтобы
// с фолбэком на 256-цветный/базовый ANSI для корректного отображения // сервис не поднялся с незащищённым паролем.
// в терминалах OpenIndiana. Если stderr перенаправлен в файл — без цвета.
func NewACLManager() *ACLManager { func NewACLManager() *ACLManager {
m := &ACLManager{ m := &ACLManager{
stopChan: make(chan struct{}), stopChan: make(chan struct{}),
@@ -242,8 +256,9 @@ func NewACLManager() *ACLManager {
adminHash, err := hashPassword(adminPassword) adminHash, err := hashPassword(adminPassword)
if err != nil { if err != nil {
// Fallback: сохраняем пароль в legacy-формате, чтобы не сломать старт // ИСПРАВЛЕНО: не сохраняем пароль в открытом виде.
adminHash = adminPassword // Паникуем — сервис не должен стартовать с небезопасным ACL.
panic(fmt.Sprintf("acl: failed to hash admin password: %v", err))
} }
adminUser := &User{ adminUser := &User{
@@ -256,14 +271,13 @@ func NewACLManager() *ACLManager {
} }
m.users.Store("admin", adminUser) m.users.Store("admin", adminUser)
// ИСПРАВЛЕНО (UI): убраны рамка "====" и пустые строки вокруг сообщения. // ИСПРАВЛЕНО (UI): убрана пустая строка " \n" (в ней был лишний пробел).
// Сразу после этого блока main.go вызывает displayBanner, который // Сразу после этого блока main.go вызывает displayBanner, который
// начинает вывод со строки "futriis 3i²(by 02.04.2026)" (предварённой // начинает вывод со строки "futriis 3i²(by 02.04.2026)".
// одной пустой строкой внутри самого displayBanner).
// //
// Пароль по-прежнему идёт в stderr (не в stdout), чтобы его можно // Пароль по-прежнему идёт в stderr (не в stdout), чтобы его можно
// было отделить от обычного вывода и не логировать. // было отделить от обычного вывода и не логировать.
fmt.Fprintln(os.Stderr, colorizeACLMessage(" \n")) fmt.Fprintln(os.Stderr, colorizeACLMessage("\n"))
fmt.Fprintln(os.Stderr, colorizeACLMessage(" ACL: создан пользователь 'admin' со случайным паролем.")) fmt.Fprintln(os.Stderr, colorizeACLMessage(" ACL: создан пользователь 'admin' со случайным паролем."))
fmt.Fprintln(os.Stderr, colorizeACLMessage(fmt.Sprintf(" Пароль: %s", adminPassword))) fmt.Fprintln(os.Stderr, colorizeACLMessage(fmt.Sprintf(" Пароль: %s", adminPassword)))
fmt.Fprintln(os.Stderr, colorizeACLMessage(" Смените пароль после первого входа командой 'acl change-password'!")) fmt.Fprintln(os.Stderr, colorizeACLMessage(" Смените пароль после первого входа командой 'acl change-password'!"))
@@ -275,6 +289,24 @@ func NewACLManager() *ACLManager {
return m return m
} }
// SetLogger устанавливает логгер (опционально).
// Вызывается после NewACLManager, если приложение хочет логировать
// события ACL (неудачные входы, блокировки аккаунтов и т.п.).
func (m *ACLManager) SetLogger(l Logger) {
m.mu.Lock()
defer m.mu.Unlock()
m.logger = l
}
func (m *ACLManager) logWarn(msg string) {
m.mu.RLock()
l := m.logger
m.mu.RUnlock()
if l != nil {
l.Warn(msg)
}
}
// ============================================================================= // =============================================================================
// ХЕШИРОВАНИЕ ПАРОЛЕЙ (PBKDF2-HMAC-SHA256) // ХЕШИРОВАНИЕ ПАРОЛЕЙ (PBKDF2-HMAC-SHA256)
// ============================================================================= // =============================================================================
@@ -391,12 +423,50 @@ func validatePassword(password string) error {
return nil return nil
} }
// validatePermission проверяет формат строки разрешения.
// Допустимые формы:
// - "*:*"
// - "database.collection:operation"
// - "database.*:operation"
// где database/collection/operation — из безопасного набора символов.
func validatePermission(perm string) error {
if perm == "" {
return fmt.Errorf("permission cannot be empty")
}
if !permissionRegex.MatchString(perm) {
return fmt.Errorf("invalid permission format: %q", perm)
}
return nil
}
// dedupeRoles убирает дубликаты из списка ролей, сохраняя порядок.
func dedupeRoles(roles []string) []string {
if len(roles) == 0 {
return []string{}
}
seen := make(map[string]struct{}, len(roles))
out := make([]string, 0, len(roles))
for _, r := range roles {
if r == "" {
continue
}
if _, ok := seen[r]; ok {
continue
}
seen[r] = struct{}{}
out = append(out, r)
}
return out
}
// ============================================================================= // =============================================================================
// БЛОКИРОВКА АККАУНТА // БЛОКИРОВКА АККАУНТА
// ============================================================================= // =============================================================================
// isAccountLocked проверяет, заблокирован ли аккаунт. // isAccountLocked проверяет, заблокирован ли аккаунт.
// Автоматически снимает блокировку, если срок истёк. // Автоматически снимает блокировку, если срок истёк.
// ВАЖНО: вызывается под m.mu (read-modify-write), поэтому мутация
// user здесь безопасна.
func (m *ACLManager) isAccountLocked(user *User) bool { func (m *ACLManager) isAccountLocked(user *User) bool {
if user.LockedUntil == 0 { if user.LockedUntil == 0 {
return false return false
@@ -422,20 +492,23 @@ func (m *ACLManager) CreateUser(username, password string, roles []string) error
return err return err
} }
if _, exists := m.users.Load(username); exists {
return fmt.Errorf("user %s already exists", username)
}
hash, err := hashPassword(password) hash, err := hashPassword(password)
if err != nil { if err != nil {
return fmt.Errorf("failed to hash password: %w", err) return fmt.Errorf("failed to hash password: %w", err)
} }
m.mu.Lock()
defer m.mu.Unlock()
if _, exists := m.users.Load(username); exists {
return fmt.Errorf("user %s already exists", username)
}
user := &User{ user := &User{
ID: uuid.New().String(), ID: uuid.New().String(),
Username: username, Username: username,
PasswordHash: hash, PasswordHash: hash,
Roles: roles, Roles: dedupeRoles(roles),
CreatedAt: time.Now().UnixMilli(), CreatedAt: time.Now().UnixMilli(),
Active: true, Active: true,
} }
@@ -444,11 +517,18 @@ func (m *ACLManager) CreateUser(username, password string, roles []string) error
} }
// Authenticate аутентифицирует пользователя и создаёт сессию. // Authenticate аутентифицирует пользователя и создаёт сессию.
//
// ИСПРАВЛЕНО: read-modify-write для user.FailedAttempts / LockedUntil /
// LastLogin выполняется под m.mu — иначе конкурентные попытки входа
// могли терять инкременты счётчика неудач.
func (m *ACLManager) Authenticate(username, password string) (string, error) { func (m *ACLManager) Authenticate(username, password string) (string, error) {
if err := validateUsername(username); err != nil { if err := validateUsername(username); err != nil {
return "", fmt.Errorf("invalid credentials") return "", fmt.Errorf("invalid credentials")
} }
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.users.Load(username) val, ok := m.users.Load(username)
if !ok { if !ok {
// Constant-time фиктивная проверка для защиты от user enumeration // Constant-time фиктивная проверка для защиты от user enumeration
@@ -456,11 +536,16 @@ func (m *ACLManager) Authenticate(username, password string) (string, error) {
return "", fmt.Errorf("invalid credentials") return "", fmt.Errorf("invalid credentials")
} }
user := val.(*User) // Копируем пользователя, чтобы не мутировать объект, который
// могут читать другие горутины.
src := val.(*User)
user := *src
user.Roles = append([]string(nil), src.Roles...)
if !user.Active { if !user.Active {
return "", fmt.Errorf("user is disabled") return "", fmt.Errorf("user is disabled")
} }
if m.isAccountLocked(user) { if m.isAccountLocked(&user) {
return "", fmt.Errorf("account is temporarily locked") return "", fmt.Errorf("account is temporarily locked")
} }
@@ -483,15 +568,20 @@ func (m *ACLManager) Authenticate(username, password string) (string, error) {
if user.FailedAttempts >= maxFailedAttempts { if user.FailedAttempts >= maxFailedAttempts {
user.LockedUntil = time.Now().Add(lockoutDuration).Unix() user.LockedUntil = time.Now().Add(lockoutDuration).Unix()
user.FailedAttempts = 0 user.FailedAttempts = 0
m.logWarn(fmt.Sprintf("acl: account %s locked for %s (too many failed logins)",
username, lockoutDuration))
} }
m.users.Store(username, user) m.users.Store(username, &user)
return "", fmt.Errorf("invalid credentials") return "", fmt.Errorf("invalid credentials")
} }
user.LastLogin = time.Now().UnixMilli() user.LastLogin = time.Now().UnixMilli()
user.FailedAttempts = 0 user.FailedAttempts = 0
user.LockedUntil = 0 user.LockedUntil = 0
m.users.Store(username, user) m.users.Store(username, &user)
// Ограничиваем количество активных сессий на пользователя.
m.enforceSessionLimitLocked(username)
sessionID := uuid.New().String() sessionID := uuid.New().String()
now := time.Now() now := time.Now()
@@ -507,6 +597,36 @@ func (m *ACLManager) Authenticate(username, password string) (string, error) {
return sessionID, nil return sessionID, nil
} }
// enforceSessionLimitLocked удаляет старейшие сессии пользователя,
// если их количество достигло maxSessionsPerUser.
// Вызывается под m.mu.
func (m *ACLManager) enforceSessionLimitLocked(username string) {
type sess struct {
id string
createdAt int64
}
var userSessions []sess
m.sessions.Range(func(key, value interface{}) bool {
s := value.(*Session)
if s.Username == username {
userSessions = append(userSessions, sess{id: s.ID, createdAt: s.CreatedAt})
}
return true
})
if len(userSessions) < maxSessionsPerUser {
return
}
// Сортируем по createdAt возрастанию и удаляем самые старые,
// чтобы после добавления новой сессии лимит не был превышен.
sort.Slice(userSessions, func(i, j int) bool {
return userSessions[i].createdAt < userSessions[j].createdAt
})
toDelete := len(userSessions) - maxSessionsPerUser + 1
for i := 0; i < toDelete; i++ {
m.sessions.Delete(userSessions[i].id)
}
}
// Logout завершает сессию // Logout завершает сессию
func (m *ACLManager) Logout(sessionID string) { func (m *ACLManager) Logout(sessionID string) {
m.sessions.Delete(sessionID) m.sessions.Delete(sessionID)
@@ -517,6 +637,19 @@ func (m *ACLManager) CheckSession(sessionID string) bool {
return m.getSession(sessionID) != nil return m.getSession(sessionID) != nil
} }
// CheckSessionForIP проверяет сессию и совпадение IP клиента.
// Если у сессии не задан CreatedByIP — проверка IP пропускается.
func (m *ACLManager) CheckSessionForIP(sessionID, clientIP string) bool {
s := m.getSession(sessionID)
if s == nil {
return false
}
if s.CreatedByIP != "" && s.CreatedByIP != clientIP {
return false
}
return true
}
// getSession возвращает валидную сессию или nil. // getSession возвращает валидную сессию или nil.
func (m *ACLManager) getSession(sessionID string) *Session { func (m *ACLManager) getSession(sessionID string) *Session {
val, ok := m.sessions.Load(sessionID) val, ok := m.sessions.Load(sessionID)
@@ -576,64 +709,107 @@ func (m *ACLManager) GetUserInfo(username string) (*User, error) {
}, nil }, nil
} }
// DisableUser отключает пользователя // DisableUser отключает пользователя.
// ИСПРАВЛЕНО: инвалидирует все сессии пользователя.
func (m *ACLManager) DisableUser(username string) error { func (m *ACLManager) DisableUser(username string) error {
if username == "admin" { if username == "admin" {
return fmt.Errorf("cannot disable built-in admin") return fmt.Errorf("cannot disable built-in admin")
} }
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.users.Load(username) val, ok := m.users.Load(username)
if !ok { if !ok {
return fmt.Errorf("user not found") return fmt.Errorf("user not found")
} }
user := val.(*User) src := val.(*User)
user := *src
user.Active = false user.Active = false
m.users.Store(username, user) m.users.Store(username, &user)
// Инвалидируем все сессии
m.dropUserSessionsLocked(username)
return nil return nil
} }
// EnableUser включает пользователя // EnableUser включает пользователя
func (m *ACLManager) EnableUser(username string) error { func (m *ACLManager) EnableUser(username string) error {
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.users.Load(username) val, ok := m.users.Load(username)
if !ok { if !ok {
return fmt.Errorf("user not found") return fmt.Errorf("user not found")
} }
user := val.(*User) src := val.(*User)
user := *src
user.Active = true user.Active = true
m.users.Store(username, user) m.users.Store(username, &user)
return nil return nil
} }
// DeleteUser удаляет пользователя // DeleteUser удаляет пользователя.
// ИСПРАВЛЕНО: инвалидирует все сессии пользователя.
func (m *ACLManager) DeleteUser(username string) error { func (m *ACLManager) DeleteUser(username string) error {
if username == "admin" { if username == "admin" {
return fmt.Errorf("cannot delete built-in admin") return fmt.Errorf("cannot delete built-in admin")
} }
m.mu.Lock()
defer m.mu.Unlock()
if _, exists := m.users.LoadAndDelete(username); !exists { if _, exists := m.users.LoadAndDelete(username); !exists {
return fmt.Errorf("user not found") return fmt.Errorf("user not found")
} }
m.dropUserSessionsLocked(username)
return nil return nil
} }
// ChangePassword изменяет пароль пользователя // ChangePassword изменяет пароль пользователя.
// ИСПРАВЛЕНО: инвалидирует все сессии пользователя (кроме текущей,
// если нужно, — но проще сбросить все, чтобы гарантировать
// безопасность при смене пароля администратором).
func (m *ACLManager) ChangePassword(username, newPassword string) error { func (m *ACLManager) ChangePassword(username, newPassword string) error {
if err := validatePassword(newPassword); err != nil { if err := validatePassword(newPassword); err != nil {
return err return err
} }
val, ok := m.users.Load(username)
if !ok {
return fmt.Errorf("user not found")
}
user := val.(*User)
hash, err := hashPassword(newPassword) hash, err := hashPassword(newPassword)
if err != nil { if err != nil {
return err return err
} }
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.users.Load(username)
if !ok {
return fmt.Errorf("user not found")
}
src := val.(*User)
user := *src
user.PasswordHash = hash user.PasswordHash = hash
user.LegacyPassword = "" user.LegacyPassword = ""
m.users.Store(username, user) m.users.Store(username, &user)
m.dropUserSessionsLocked(username)
return nil return nil
} }
// dropUserSessionsLocked удаляет все сессии пользователя.
// Вызывается под m.mu.
func (m *ACLManager) dropUserSessionsLocked(username string) {
var toDelete []string
m.sessions.Range(func(key, value interface{}) bool {
s := value.(*Session)
if s.Username == username {
toDelete = append(toDelete, s.ID)
}
return true
})
for _, id := range toDelete {
m.sessions.Delete(id)
}
}
// ListUsers возвращает отсортированный список всех пользователей // ListUsers возвращает отсортированный список всех пользователей
func (m *ACLManager) ListUsers() []string { func (m *ACLManager) ListUsers() []string {
users := make([]string, 0) users := make([]string, 0)
@@ -654,6 +830,9 @@ func (m *ACLManager) CreateRole(name string) error {
if name == "" { if name == "" {
return fmt.Errorf("role name cannot be empty") return fmt.Errorf("role name cannot be empty")
} }
m.mu.Lock()
defer m.mu.Unlock()
if _, exists := m.roles.Load(name); exists { if _, exists := m.roles.Load(name); exists {
return fmt.Errorf("role %s already exists", name) return fmt.Errorf("role %s already exists", name)
} }
@@ -662,14 +841,61 @@ func (m *ACLManager) CreateRole(name string) error {
return nil return nil
} }
// DeleteRole удаляет роль // DeleteRole удаляет роль.
// ИСПРАВЛЕНО: также удаляет роль у всех пользователей,
// у которых она была — иначе в сессиях/юзерах остаётся «висячая» роль.
func (m *ACLManager) DeleteRole(name string) error { func (m *ACLManager) DeleteRole(name string) error {
if name == "admin" { if name == "admin" {
return fmt.Errorf("cannot delete built-in role 'admin'") return fmt.Errorf("cannot delete built-in role 'admin'")
} }
m.mu.Lock()
defer m.mu.Unlock()
if _, exists := m.roles.LoadAndDelete(name); !exists { if _, exists := m.roles.LoadAndDelete(name); !exists {
return fmt.Errorf("role not found") return fmt.Errorf("role not found")
} }
// Убираем роль у всех пользователей
m.users.Range(func(key, value interface{}) bool {
src := value.(*User)
hasRole := false
for _, r := range src.Roles {
if r == name {
hasRole = true
break
}
}
if !hasRole {
return true
}
user := *src
newRoles := make([]string, 0, len(user.Roles))
for _, r := range user.Roles {
if r != name {
newRoles = append(newRoles, r)
}
}
user.Roles = newRoles
m.users.Store(key.(string), &user)
return true
})
// И инвалидируем все сессии, у которых была эта роль.
var toDelete []string
m.sessions.Range(func(key, value interface{}) bool {
s := value.(*Session)
for _, r := range s.Roles {
if r == name {
toDelete = append(toDelete, s.ID)
break
}
}
return true
})
for _, id := range toDelete {
m.sessions.Delete(id)
}
return nil return nil
} }
@@ -694,38 +920,53 @@ func (m *ACLManager) GetRolePermissions(roleName string) ([]string, error) {
return append([]string(nil), role.Permissions...), nil return append([]string(nil), role.Permissions...), nil
} }
// GrantPermission выдаёт разрешение роли (с дедупликацией) // GrantPermission выдаёт разрешение роли (с дедупликацией).
// ИСПРАВЛЕНО: валидирует формат разрешения и мутирует под m.mu.
func (m *ACLManager) GrantPermission(roleName, permission string) error { func (m *ACLManager) GrantPermission(roleName, permission string) error {
if err := validatePermission(permission); err != nil {
return err
}
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.roles.Load(roleName) val, ok := m.roles.Load(roleName)
if !ok { if !ok {
return fmt.Errorf("role not found") return fmt.Errorf("role not found")
} }
role := val.(*Role) src := val.(*Role)
role := *src
role.Permissions = append([]string(nil), src.Permissions...)
for _, p := range role.Permissions { for _, p := range role.Permissions {
if p == permission { if p == permission {
return nil // уже есть return nil // уже есть
} }
} }
role.Permissions = append(role.Permissions, permission) role.Permissions = append(role.Permissions, permission)
m.roles.Store(roleName, role) m.roles.Store(roleName, &role)
return nil return nil
} }
// RevokePermission отзывает разрешение у роли // RevokePermission отзывает разрешение у роли.
func (m *ACLManager) RevokePermission(roleName, permission string) error { func (m *ACLManager) RevokePermission(roleName, permission string) error {
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.roles.Load(roleName) val, ok := m.roles.Load(roleName)
if !ok { if !ok {
return fmt.Errorf("role not found") return fmt.Errorf("role not found")
} }
role := val.(*Role) src := val.(*Role)
role := *src
newPermissions := make([]string, 0, len(role.Permissions)) newPermissions := make([]string, 0, len(role.Permissions))
for _, p := range role.Permissions { for _, p := range src.Permissions {
if p != permission { if p != permission {
newPermissions = append(newPermissions, p) newPermissions = append(newPermissions, p)
} }
} }
role.Permissions = newPermissions role.Permissions = newPermissions
m.roles.Store(roleName, role) m.roles.Store(roleName, &role)
return nil return nil
} }
@@ -733,38 +974,55 @@ func (m *ACLManager) RevokePermission(roleName, permission string) error {
// РОЛИ ПОЛЬЗОВАТЕЛЯ // РОЛИ ПОЛЬЗОВАТЕЛЯ
// ============================================================================= // =============================================================================
// AddUserRole добавляет роль пользователю // AddUserRole добавляет роль пользователю.
// ИСПРАВЛЕНО: мутирует под m.mu; инвалидирует сессии пользователя,
// чтобы новоприобретённая роль вступила в силу при следующем входе.
func (m *ACLManager) AddUserRole(username, roleName string) error { func (m *ACLManager) AddUserRole(username, roleName string) error {
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.users.Load(username) val, ok := m.users.Load(username)
if !ok { if !ok {
return fmt.Errorf("user not found") return fmt.Errorf("user not found")
} }
user := val.(*User) src := val.(*User)
for _, r := range user.Roles { for _, r := range src.Roles {
if r == roleName { if r == roleName {
return fmt.Errorf("user already has role %s", roleName) return fmt.Errorf("user already has role %s", roleName)
} }
} }
user.Roles = append(user.Roles, roleName) user := *src
m.users.Store(username, user) user.Roles = append(append([]string(nil), src.Roles...), roleName)
m.users.Store(username, &user)
// Инвалидируем сессии пользователя: после смены ролей старые
// сессии не должны сохранять прежний набор прав.
m.dropUserSessionsLocked(username)
return nil return nil
} }
// RemoveUserRole удаляет роль у пользователя // RemoveUserRole удаляет роль у пользователя.
// ИСПРАВЛЕНО: мутирует под m.mu; инвалидирует сессии пользователя.
func (m *ACLManager) RemoveUserRole(username, roleName string) error { func (m *ACLManager) RemoveUserRole(username, roleName string) error {
m.mu.Lock()
defer m.mu.Unlock()
val, ok := m.users.Load(username) val, ok := m.users.Load(username)
if !ok { if !ok {
return fmt.Errorf("user not found") return fmt.Errorf("user not found")
} }
user := val.(*User) src := val.(*User)
newRoles := make([]string, 0, len(user.Roles)) newRoles := make([]string, 0, len(src.Roles))
for _, r := range user.Roles { for _, r := range src.Roles {
if r != roleName { if r != roleName {
newRoles = append(newRoles, r) newRoles = append(newRoles, r)
} }
} }
user := *src
user.Roles = newRoles user.Roles = newRoles
m.users.Store(username, user) m.users.Store(username, &user)
m.dropUserSessionsLocked(username)
return nil return nil
} }
@@ -807,7 +1065,17 @@ func (m *ACLManager) CheckPermission(sessionID, database, collection, operation
// - "db.*:read" — read всех коллекций в db // - "db.*:read" — read всех коллекций в db
// - "db.coll:admin" — admin-доступ к конкретной коллекции // - "db.coll:admin" — admin-доступ к конкретной коллекции
// - "db.coll:*" — все операции на db.coll // - "db.coll:*" — все операции на db.coll
//
// ИСПРАВЛЕНО: убран мёртвый код с проверкой resource == "*:*".
// Раньше splitPermission("*:*") давал resource="*", op="*", и первая
// ветка "if resource == \"*:*\" && op == \"*\"" никогда не срабатывала.
// Теперь wildcard-разрешение обрабатывается явно, до split.
func matchPermission(perm, database, collection, operation string) bool { func matchPermission(perm, database, collection, operation string) bool {
// Полный доступ — единственная форма, где resource содержит ":".
if perm == "*:*" {
return true
}
parts := splitPermission(perm) parts := splitPermission(perm)
if len(parts) != 2 { if len(parts) != 2 {
return false return false
@@ -815,14 +1083,10 @@ func matchPermission(perm, database, collection, operation string) bool {
resource := parts[0] resource := parts[0]
op := parts[1] op := parts[1]
// Полный доступ // Явный admin-доступ: требуется op == "admin" (или "*").
if resource == "*:*" && op == "*" { if operation == "admin" {
return true if op != "admin" && op != "*" {
} return false
// Явный admin-доступ
if op == "admin" {
if resource == "*:*" {
return true
} }
resourceParts := splitResource(resource) resourceParts := splitResource(resource)
if len(resourceParts) != 2 { if len(resourceParts) != 2 {
@@ -832,7 +1096,8 @@ func matchPermission(perm, database, collection, operation string) bool {
collMatch := resourceParts[1] == "*" || resourceParts[1] == collection collMatch := resourceParts[1] == "*" || resourceParts[1] == collection
return dbMatch && collMatch return dbMatch && collMatch
} }
// Обычные операции: op == "*" или совпадает
// Обычные операции: op == "*" или совпадает с запрошенной.
if op != "*" && op != operation { if op != "*" && op != operation {
return false return false
} }