wacl 0.1.1

Access Control List and authentication with SHA-256 password hashing
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

//! 用户句柄（对标 libs/server/ACL/UserHandle.cs）

use std::sync::Arc;

use parking_lot::RwLock;

use super::user::User;

/// 用户引用句柄：读侧无锁快照，写侧 CAS 换新
pub struct UserHandle {
  /// 当前指向的用户（对标 C# UserHandle.user 引用）
  user: RwLock<Arc<User>>,
}

impl UserHandle {
  /// 构造句柄（对标 C# UserHandle 构造；user 为空即 panic，与
  /// ArgumentNullException 语义一致——此处由类型系统保证非空）
  pub fn new(user: Arc<User>) -> Self {
    Self {
      user: RwLock::new(user),
    }
  }

  /// 取当前用户（最新版本，对标 C# UserHandle.User 属性）
  pub fn user(&self) -> Arc<User> {
    self.user.read().clone()
  }

  /// CAS 换新：仅当当前用户与 `expected` 同一实例时替换为 `new_user`
  ///
  /// libs/server/ACL/UserHandle.cs:TrySetUser
  pub fn try_set_user(&self, new_user: Arc<User>, expected: &Arc<User>) -> bool {
    let mut slot = self.user.write();
    if Arc::ptr_eq(&slot, expected) {
      *slot = new_user;
      true
    } else {
      false
    }
  }
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn try_set_user_swaps_only_matching_instance() {
    let u1 = Arc::new(User::new("a".into()));
    let u2 = Arc::new(User::new("a".into()));
    let h = UserHandle::new(u1.clone());
    assert!(Arc::ptr_eq(&h.user(), &u1));

    // 期望实例不符 → 替换失败
    assert!(!h.try_set_user(u2.clone(), &u2));
    assert!(Arc::ptr_eq(&h.user(), &u1));

    // 期望实例相符 → 替换成功
    assert!(h.try_set_user(u2.clone(), &u1));
    assert!(Arc::ptr_eq(&h.user(), &u2));
  }
}