betaquant 0.5.9__tar.gz → 0.6.2__tar.gz

{betaquant-0.5.9 → betaquant-0.6.2}/Cargo.lock

@@ -23,20 +23,46 @@ version = "1.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
 
+[[package]]
+name = "base64"
+version = "0.22.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "72b3254f16251a8381aa12e40e3c4d2f0199f8c6508fbecb9d91f575e0fbb8c6"
+
+[[package]]
+name = "base64ct"
+version = "1.8.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2af50177e190e07a26ab74f8b1efbfe2ef87da2116221318cb1c2e82baf7de06"
+
 [[package]]
 name = "betaquant"
-version = "0.5.9"
+version = "0.6.2"
 dependencies = [
  "anyhow",
+ "base64",
+ "ed25519-dalek",
  "log",
  "num-traits",
  "numpy",
  "pyo3",
  "pyo3-log",
  "rayon",
+ "serde",
+ "serde_json",
+ "sha2",
  "thiserror",
 ]
 
+[[package]]
+name = "block-buffer"
+version = "0.10.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3078c7629b62d3f0439517fa394996acacc5cbc91c5a20d8c658e77abd503a71"
+dependencies = [
+ "generic-array",
+]
+
 [[package]]
 name = "cc"
 version = "1.2.62"
@@ -47,6 +73,27 @@ dependencies = [
  "shlex",
 ]
 
+[[package]]
+name = "cfg-if"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9330f8b2ff13f34540b44e946ef35111825727b38d33286ef986142615121801"
+
+[[package]]
+name = "const-oid"
+version = "0.9.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c2459377285ad874054d797f3ccebf984978aa39129f6eafde5cdc8315b612f8"
+
+[[package]]
+name = "cpufeatures"
+version = "0.2.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "59ed5838eebb26a2bb2e58f6d5b5316989ae9d08bab10e0e6d103e656d1b0280"
+dependencies = [
+ "libc",
+]
+
 [[package]]
 name = "crossbeam-deque"
 version = "0.8.6"
@@ -72,24 +119,138 @@ version = "0.8.21"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
 
+[[package]]
+name = "crypto-common"
+version = "0.1.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "78c8292055d1c1df0cce5d180393dc8cce0abec0a7102adb6c7b1eef6016d60a"
+dependencies = [
+ "generic-array",
+ "typenum",
+]
+
+[[package]]
+name = "curve25519-dalek"
+version = "4.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "97fb8b7c4503de7d6ae7b42ab72a5a59857b4c937ec27a3d4539dba95b5ab2be"
+dependencies = [
+ "cfg-if",
+ "cpufeatures",
+ "curve25519-dalek-derive",
+ "digest",
+ "fiat-crypto",
+ "rustc_version",
+ "subtle",
+ "zeroize",
+]
+
+[[package]]
+name = "curve25519-dalek-derive"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f46882e17999c6cc590af592290432be3bce0428cb0d5f8b6715e4dc7b383eb3"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "der"
+version = "0.7.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e7c1832837b905bbfb5101e07cc24c8deddf52f93225eee6ead5f4d63d53ddcb"
+dependencies = [
+ "const-oid",
+ "zeroize",
+]
+
+[[package]]
+name = "digest"
+version = "0.10.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9ed9a281f7bc9b7576e61468ba615a66a5c8cfdff42420a70aa82701a3b1e292"
+dependencies = [
+ "block-buffer",
+ "crypto-common",
+]
+
+[[package]]
+name = "ed25519"
+version = "2.2.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "115531babc129696a58c64a4fef0a8bf9e9698629fb97e9e40767d235cfbcd53"
+dependencies = [
+ "pkcs8",
+ "signature",
+]
+
+[[package]]
+name = "ed25519-dalek"
+version = "2.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "70e796c081cee67dc755e1a36a0a172b897fab85fc3f6bc48307991f64e4eca9"
+dependencies = [
+ "curve25519-dalek",
+ "ed25519",
+ "serde",
+ "sha2",
+ "subtle",
+ "zeroize",
+]
+
 [[package]]
 name = "either"
 version = "1.15.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
 
+[[package]]
+name = "fiat-crypto"
+version = "0.2.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "28dea519a9695b9977216879a3ebfddf92f1c08c05d984f8996aecd6ecdc811d"
+
 [[package]]
 name = "find-msvc-tools"
 version = "0.1.9"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "5baebc0774151f905a1a2cc41989300b1e6fbb29aff0ceffa1064fdd3088d582"
 
+[[package]]
+name = "generic-array"
+version = "0.14.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "85649ca51fd72272d7821adaf274ad91c288277713d9c18820d8499a7ff69e9a"
+dependencies = [
+ "typenum",
+ "version_check",
+]
+
+[[package]]
+name = "getrandom"
+version = "0.2.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ff2abc00be7fca6ebc474524697ae276ad847ad0a6b3faa4bcb027e9a4614ad0"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "wasi",
+]
+
 [[package]]
 name = "heck"
 version = "0.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
 
+[[package]]
+name = "itoa"
+version = "1.0.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8f42a60cbdf9a97f5d2305f08a87dc4e09308d1276d28c869c684d7777685682"
+
 [[package]]
 name = "libc"
 version = "0.2.180"
@@ -112,6 +273,12 @@ dependencies = [
  "rawpointer",
 ]
 
+[[package]]
+name = "memchr"
+version = "2.8.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6b947ae49db0d222b1dbc6b113ce7248a3fc3a6ca21b696717bfc000ba4484d8"
+
 [[package]]
 name = "ndarray"
 version = "0.17.2"
@@ -176,6 +343,16 @@ version = "1.21.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
 
+[[package]]
+name = "pkcs8"
+version = "0.10.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f950b2377845cebe5cf8b5165cb3cc1a5e0fa5cfa3e1f7f55707d8fd82e0a7b7"
+dependencies = [
+ "der",
+ "spki",
+]
+
 [[package]]
 name = "portable-atomic"
 version = "1.13.1"
@@ -288,6 +465,15 @@ dependencies = [
  "proc-macro2",
 ]
 
+[[package]]
+name = "rand_core"
+version = "0.6.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c"
+dependencies = [
+ "getrandom",
+]
+
 [[package]]
 name = "rawpointer"
 version = "0.2.1"
@@ -320,18 +506,112 @@ version = "2.1.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "357703d41365b4b27c590e3ed91eabb1b663f07c4c084095e60cbed4362dff0d"
 
+[[package]]
+name = "rustc_version"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cfcb3a22ef46e85b45de6ee7e79d063319ebb6594faafcf1c225ea92ab6e9b92"
+dependencies = [
+ "semver",
+]
+
 [[package]]
 name = "rustversion"
 version = "1.0.22"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b39cdef0fa800fc44525c84ccb54a029961a8215f9619753635a9c0d2538d46d"
 
+[[package]]
+name = "semver"
+version = "1.0.28"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8a7852d02fc848982e0c167ef163aaff9cd91dc640ba85e263cb1ce46fae51cd"
+
+[[package]]
+name = "serde"
+version = "1.0.228"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9a8e94ea7f378bd32cbbd37198a4a91436180c5bb472411e48b5ec2e2124ae9e"
+dependencies = [
+ "serde_core",
+ "serde_derive",
+]
+
+[[package]]
+name = "serde_core"
+version = "1.0.228"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "41d385c7d4ca58e59fc732af25c3983b67ac852c1a25000afe1175de458b67ad"
+dependencies = [
+ "serde_derive",
+]
+
+[[package]]
+name = "serde_derive"
+version = "1.0.228"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d540f220d3187173da220f885ab66608367b6574e925011a9353e4badda91d79"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "serde_json"
+version = "1.0.150"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e8014e44b4736ed0538adeecded0fce2a272f22dc9578a7eb6b2d9993c74cfb9"
+dependencies = [
+ "itoa",
+ "memchr",
+ "serde",
+ "serde_core",
+ "zmij",
+]
+
+[[package]]
+name = "sha2"
+version = "0.10.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a7507d819769d01a365ab707794a4084392c824f54a7a6a7862f8c3d0892b283"
+dependencies = [
+ "cfg-if",
+ "cpufeatures",
+ "digest",
+]
+
 [[package]]
 name = "shlex"
 version = "1.3.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "0fda2ff0d084019ba4d7c6f371c95d8fd75ce3524c3cb8fb653a3023f6323e64"
 
+[[package]]
+name = "signature"
+version = "2.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "77549399552de45a898a580c1b41d445bf730df867cc44e6c0233bbc4b8329de"
+dependencies = [
+ "rand_core",
+]
+
+[[package]]
+name = "spki"
+version = "0.7.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d91ed6c858b01f942cd56b37a94b3e0a1798290327d1236e4d9cf4eaca44d29d"
+dependencies = [
+ "base64ct",
+ "der",
+]
+
+[[package]]
+name = "subtle"
+version = "2.6.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "13c2bddecc57b384dee18652358fb23172facb8a2c51ccc10d74c157bdea3292"
+
 [[package]]
 name = "syn"
 version = "2.0.114"
@@ -369,8 +649,38 @@ dependencies = [
  "syn",
 ]
 
+[[package]]
+name = "typenum"
+version = "1.20.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b6f5e870be6c3b371b77fe0ee0bafb859fa4964b4404c27de1d380043c4dda20"
+
 [[package]]
 name = "unicode-ident"
 version = "1.0.22"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "9312f7c4f6ff9069b165498234ce8be658059c6728633667c526e27dc2cf1df5"
+
+[[package]]
+name = "version_check"
+version = "0.9.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0b928f33d975fc6ad9f86c8f283853ad26bdd5b10b7f1542aa2fa15e2289105a"
+
+[[package]]
+name = "wasi"
+version = "0.11.1+wasi-snapshot-preview1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ccf3ec651a847eb01de73ccad15eb7d99f80485de043efb2f370cd654f4ea44b"
+
+[[package]]
+name = "zeroize"
+version = "1.8.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b97154e67e32c85465826e8bcc1c59429aaaf107c1e4a9e53c8d8ccd5eff88d0"
+
+[[package]]
+name = "zmij"
+version = "1.0.21"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b8848ee67ecc8aedbaf3e4122217aff892639231befc6a1b58d29fff4c2cabaa"

{betaquant-0.5.9 → betaquant-0.6.2}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "betaquant"
-version = "0.5.9"
+version = "0.6.2"
 edition = "2024"
 authors = ["ZhaoJun"]
 readme = "README.md"
@@ -19,6 +19,11 @@ numpy = "0.28"
 num-traits = "0.2"
 rayon = "1.11"
 log = "0.4"
+ed25519-dalek = "2"
+base64 = "0.22"
+serde = { version = "1", features = ["derive"] }
+serde_json = "1"
+sha2 = "0.10"
 
 
 [build-dependencies]

{betaquant-0.5.9 → betaquant-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: betaquant
-Version: 0.5.9
+Version: 0.6.2
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
@@ -28,6 +28,64 @@ Project-URL: repository, https://home.zhaojun.com
 pip install betaquant
 ```
 
+## 授权（License Token）
+
+betaquant 的算子受许可证（license）保护，**使用前需持有有效的 license token**。
+
+
+### 取得机器指纹（机器绑定用）
+
+```python
+import betaquant
+
+machine = betaquant.machine_fingerprint()   # 跨 Linux/Windows，机器绑定以它为准
+print(machine)
+```
+
+把该指纹提交给签发方，换取绑定到本机的 token。
+
+### 启用 token：算子入口自动鉴权（推荐）
+
+调用 `set_license_token` 把 token 设入进程级状态后，**每个算子入口都会自动鉴权**：
+未授权 / 过期 / 未设 token 时算子直接抛 `ValueError`，无需在每次调用前手写判断。
+
+
+```python
+import betaquant
+
+token = "<从签发方取得的 license key>"
+betaquant.set_license_token(token)   # 启动时一次
+
+result = betaquant.ts_ma(data, 3)     # 已授权 → 正常计算
+# 若 token 未设置 / 过期 / 该算子不在 allow，betaquant.ts_ma(...) 直接抛 ValueError
+```
+
+> 不调用 `set_license_token` 而直接调算子会抛 `ValueError`。这是与早期版本的关键区别：
+> 鉴权不再依赖调用方自觉判断，而是内嵌在每个算子入口。
+
+### 手动校验（可选）
+
+也可显式验签或单独判断某算子是否授权（不改变进程级 token 状态）：
+
+```python
+# 验签 + 校验（失败 / 过期 / 机器不符抛 ValueError），返回声明 dict
+claims = betaquant.verify_license(token, machine=machine)
+
+# 判断某算子是否被授权（token 无效 / 过期直接返回 False）
+if betaquant.license_allows(token, "ts_ma", machine=machine):
+    ...
+```
+
+授权语义（**默认拒绝**）：算子必须显式出现在 `allow`（或 `allow` 含 `"*"`）才放行；
+`deny` 优先级高于 `allow`。`allow=[]` 表示全部拒绝，并非不限制。
+
+> 过期判定使用本机系统时间，离线方案对“改系统时间”无法根治，属已知限制：用短有效期
+> + 定期换发缓解。
+
+更完整的端到端示例（签发 → 启用 token → 算子硬门禁 → 过期续期 → 机器绑定）见
+`encryptor/license/examples/demo.py`。签发方封装见 `encryptor/license`
+（`LicenseService`）。
+
 ## 使用
 
 ### 上下文设置

{betaquant-0.5.9 → betaquant-0.6.2}/README.md

@@ -12,6 +12,64 @@
 pip install betaquant
 ```
 
+## 授权（License Token）
+
+betaquant 的算子受许可证（license）保护，**使用前需持有有效的 license token**。
+
+
+### 取得机器指纹（机器绑定用）
+
+```python
+import betaquant
+
+machine = betaquant.machine_fingerprint()   # 跨 Linux/Windows，机器绑定以它为准
+print(machine)
+```
+
+把该指纹提交给签发方，换取绑定到本机的 token。
+
+### 启用 token：算子入口自动鉴权（推荐）
+
+调用 `set_license_token` 把 token 设入进程级状态后，**每个算子入口都会自动鉴权**：
+未授权 / 过期 / 未设 token 时算子直接抛 `ValueError`，无需在每次调用前手写判断。
+
+
+```python
+import betaquant
+
+token = "<从签发方取得的 license key>"
+betaquant.set_license_token(token)   # 启动时一次
+
+result = betaquant.ts_ma(data, 3)     # 已授权 → 正常计算
+# 若 token 未设置 / 过期 / 该算子不在 allow，betaquant.ts_ma(...) 直接抛 ValueError
+```
+
+> 不调用 `set_license_token` 而直接调算子会抛 `ValueError`。这是与早期版本的关键区别：
+> 鉴权不再依赖调用方自觉判断，而是内嵌在每个算子入口。
+
+### 手动校验（可选）
+
+也可显式验签或单独判断某算子是否授权（不改变进程级 token 状态）：
+
+```python
+# 验签 + 校验（失败 / 过期 / 机器不符抛 ValueError），返回声明 dict
+claims = betaquant.verify_license(token, machine=machine)
+
+# 判断某算子是否被授权（token 无效 / 过期直接返回 False）
+if betaquant.license_allows(token, "ts_ma", machine=machine):
+    ...
+```
+
+授权语义（**默认拒绝**）：算子必须显式出现在 `allow`（或 `allow` 含 `"*"`）才放行；
+`deny` 优先级高于 `allow`。`allow=[]` 表示全部拒绝，并非不限制。
+
+> 过期判定使用本机系统时间，离线方案对“改系统时间”无法根治，属已知限制：用短有效期
+> + 定期换发缓解。
+
+更完整的端到端示例（签发 → 启用 token → 算子硬门禁 → 过期续期 → 机器绑定）见
+`encryptor/license/examples/demo.py`。签发方封装见 `encryptor/license`
+（`LicenseService`）。
+
 ## 使用
 
 ### 上下文设置

{betaquant-0.5.9 → betaquant-0.6.2}/build.rs

@@ -218,6 +218,13 @@ fn build_py_bindings(functions: &[TaFunc]) -> Result<()> {
     write!(code, "{}", py_args)?;
     writeln!(code, "  ) -> PyResult<()> {{")?;
 
+    // license 硬门禁：算子入口先做鉴权，未授权 / 过期 / 未设置 token 直接抛 ValueError。
+    writeln!(
+      code,
+      "    crate::license::require_for(\"{}\").map_err(|e| PyValueError::new_err(e.to_string()))?;",
+      py_func_name
+    )?;
+
     writeln!(code, "    let ctx = ctx(py);")?;
 
     let arrays: Vec<(&String, &TaType)> = func

betaquant-0.6.2/python/betaquant/algo/__init__.py

@@ -0,0 +1,14 @@
+# Copyright 2026 MSD-RS Project LiJia
+# SPDX-License-Identifier: BSD-2-Clause
+
+from .manual import ts_ema
+from .algo_gen import *
+from ._algo import (
+    set_ctx,
+    reset_ctx,
+    verify_license,
+    license_allows,
+    machine_fingerprint,
+    set_license_token,
+    clear_license_token,
+)