Browse Source
Re-architect the Microsoft Activation Scripts from ~46k lines of Windows
batch into an idiomatic Rust crate (MAS/). Not a line-by-line translation:
typed enums + Result flow replace goto/errorlevel/global-set, and a single
`Spp` trait is the only seam between portable logic and Windows internals.
- Portable core (compiles + 20 tests pass, zero deps): model (real SPP
GUIDs / WMI class names / LicenseStatus codes), typed errors, CLI menu,
Check_Activation_Status read path, Online KMS orchestration with host
fallback, stub backend.
- Windows backend (cfg(windows) + `winapi` feature): WMI/COM via the
`windows` crate, all unsafe/VARIANT handling encapsulated behind `Spp`.
Written from the documented SPP contract; not compiled on the Linux port
host.
- HWID/KMS38/Ohook/TSforge scaffolded against analyzed specs (in module
docs) — unported paths return a typed Unsupported error rather than
falsely reporting activation. Roadmap in MAS/README.md.
The original .cmd scripts remain recoverable in history (f34d025, MAS 3.12).
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
pull/1515/head
RustyProgrammerCmd
2 weeks ago
34 changed files with 2177 additions and 46139 deletions
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,173 @@
@@ -0,0 +1,173 @@
|
||||
# This file is automatically @generated by Cargo. |
||||
# It is not intended for manual editing. |
||||
version = 3 |
||||
|
||||
[[package]] |
||||
name = "mas" |
||||
version = "0.1.0" |
||||
dependencies = [ |
||||
"windows", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "proc-macro2" |
||||
version = "1.0.107" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "985e7ec9bb745e6ce6535b544d84d6cd6f7ad8bd711c398938ae983b91a766d9" |
||||
dependencies = [ |
||||
"unicode-ident", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "quote" |
||||
version = "1.0.47" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "1fbf4db142a473a8d80c26bbf18454ed458bf8d26c8219c331daecfdbd079001" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "syn" |
||||
version = "2.0.119" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "872831b642d1a07999a962a351ed35b955ea2cfc8f3862091e2a240a84f17297" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
"quote", |
||||
"unicode-ident", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "unicode-ident" |
||||
version = "1.0.24" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "e6e4313cd5fcd3dad5cafa179702e2b244f760991f45397d14d4ebf38247da75" |
||||
|
||||
[[package]] |
||||
name = "windows" |
||||
version = "0.58.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "dd04d41d93c4992d421894c18c8b43496aa748dd4c081bac0dc93eb0489272b6" |
||||
dependencies = [ |
||||
"windows-core", |
||||
"windows-targets", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows-core" |
||||
version = "0.58.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "6ba6d44ec8c2591c134257ce647b7ea6b20335bf6379a27dac5f1641fcf59f99" |
||||
dependencies = [ |
||||
"windows-implement", |
||||
"windows-interface", |
||||
"windows-result", |
||||
"windows-strings", |
||||
"windows-targets", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows-implement" |
||||
version = "0.58.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "2bbd5b46c938e506ecbce286b6628a02171d56153ba733b6c741fc627ec9579b" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
"quote", |
||||
"syn", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows-interface" |
||||
version = "0.58.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "053c4c462dc91d3b1504c6fe5a726dd15e216ba718e84a0e46a88fbe5ded3515" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
"quote", |
||||
"syn", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows-result" |
||||
version = "0.2.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "1d1043d8214f791817bab27572aaa8af63732e11bf84aa21a45a78d6c317ae0e" |
||||
dependencies = [ |
||||
"windows-targets", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows-strings" |
||||
version = "0.1.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "4cd9b125c486025df0eabcb585e62173c6c9eddcec5d117d3b6e8c30e2ee4d10" |
||||
dependencies = [ |
||||
"windows-result", |
||||
"windows-targets", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows-targets" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973" |
||||
dependencies = [ |
||||
"windows_aarch64_gnullvm", |
||||
"windows_aarch64_msvc", |
||||
"windows_i686_gnu", |
||||
"windows_i686_gnullvm", |
||||
"windows_i686_msvc", |
||||
"windows_x86_64_gnu", |
||||
"windows_x86_64_gnullvm", |
||||
"windows_x86_64_msvc", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows_aarch64_gnullvm" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3" |
||||
|
||||
[[package]] |
||||
name = "windows_aarch64_msvc" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469" |
||||
|
||||
[[package]] |
||||
name = "windows_i686_gnu" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b" |
||||
|
||||
[[package]] |
||||
name = "windows_i686_gnullvm" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66" |
||||
|
||||
[[package]] |
||||
name = "windows_i686_msvc" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66" |
||||
|
||||
[[package]] |
||||
name = "windows_x86_64_gnu" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78" |
||||
|
||||
[[package]] |
||||
name = "windows_x86_64_gnullvm" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d" |
||||
|
||||
[[package]] |
||||
name = "windows_x86_64_msvc" |
||||
version = "0.52.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec" |
||||
@ -0,0 +1,47 @@
@@ -0,0 +1,47 @@
|
||||
[package] |
||||
name = "mas" |
||||
version = "0.1.0" |
||||
edition = "2021" |
||||
rust-version = "1.74" |
||||
description = "Microsoft Activation Scripts, ported to Rust — HWID, KMS38, Online KMS, Ohook and TSforge activation for Windows and Office." |
||||
license = "GPL-3.0-or-later" |
||||
|
||||
[[bin]] |
||||
name = "mas" |
||||
path = "src/main.rs" |
||||
|
||||
[lib] |
||||
name = "mas" |
||||
path = "src/lib.rs" |
||||
|
||||
# The real Software Protection Platform backend |
||||
# (src/platform/windows_backend.rs) binds WMI/COM directly via the `windows` |
||||
# crate: SoftwareLicensingService / SoftwareLicensingProduct / |
||||
# OfficeSoftwareProtection* on ROOT\CIMV2. All FFI/`unsafe` is encapsulated |
||||
# behind the safe `Spp` trait (constraint 3). It only compiles on a Windows |
||||
# target, so it sits behind the `winapi` feature: |
||||
# |
||||
# cargo build --release --features winapi --target x86_64-pc-windows-msvc |
||||
# |
||||
# Off Windows (or without the feature) a stub backend is used, so the portable |
||||
# core — menus, model, data tables, orchestration — still `cargo test`s anywhere. |
||||
[features] |
||||
default = [] |
||||
winapi = ["dep:windows"] |
||||
|
||||
[target.'cfg(windows)'.dependencies] |
||||
windows = { version = "0.58", optional = true, features = [ |
||||
"Win32_Foundation", |
||||
"Win32_System_Com", |
||||
"Win32_System_Wmi", |
||||
"Win32_System_Ole", |
||||
"Win32_System_Variant", |
||||
"Win32_Security", |
||||
"Win32_System_Threading", |
||||
] } |
||||
|
||||
[profile.release] |
||||
opt-level = "z" |
||||
lto = true |
||||
strip = true |
||||
panic = "abort" |
||||
@ -0,0 +1,70 @@
@@ -0,0 +1,70 @@
|
||||
# MAS — Rust port |
||||
|
||||
An idiomatic Rust re-architecture of the Microsoft Activation Scripts. This is a |
||||
**re-architecture, not a line-by-line transliteration**: the batch scripts' |
||||
`goto`/`errorlevel`/global-`set` style is replaced with typed enums, `Result` |
||||
error flow, and a single trait boundary between portable logic and Windows |
||||
internals. |
||||
|
||||
## Architecture |
||||
|
||||
``` |
||||
src/ |
||||
model.rs Product / Method / LicenseStatus enums + the real SPP GUIDs |
||||
error.rs typed Error/Result (replaces errorlevel + colored echoes) |
||||
data/kms_hosts.rs ported data tables |
||||
platform/ |
||||
mod.rs `Spp` trait — the ONE seam to Windows; safe API over FFI |
||||
stub.rs non-Windows backend (read-only no-ops; privileged ops error) |
||||
windows_backend.rs real WMI/COM backend (cfg(windows) + feature "winapi") |
||||
activation/ |
||||
status.rs Check_Activation_Status.cmd [ported: read path] |
||||
online_kms.rs Online_KMS_Activation.cmd [ported: orchestration] |
||||
hwid.rs / kms38.rs HWID_Activation.cmd [scaffold + spec] |
||||
ohook.rs Ohook_Activation_AIO.cmd [scaffold + spec] |
||||
tsforge.rs TSforge_Activation.cmd [scaffold + spec] |
||||
cli/ the interactive menu (MAS_AIO front-end) |
||||
``` |
||||
|
||||
Every Windows effect — WMI calls, ClipUp, registry — lives behind |
||||
[`platform::Spp`]. The rest of the crate is OS-independent and unit-tested. |
||||
|
||||
## Build & test |
||||
|
||||
```sh |
||||
# Portable core — builds and tests on any OS, zero dependencies: |
||||
cargo test |
||||
|
||||
# Real Windows backend (WMI/COM via the `windows` crate): |
||||
cargo build --release --features winapi --target x86_64-pc-windows-msvc |
||||
``` |
||||
|
||||
The Windows backend is written against `windows` 0.58 from the documented SPP |
||||
WMI contract. It has **not** been compiled on the porting host (Linux has no |
||||
Windows target) — expect to shake out minor `windows`-crate signature drift on a |
||||
real Windows build. The portable core (menus, model, data, orchestration) is |
||||
compiled and tested (`20 passing`). |
||||
|
||||
## Port status & roadmap |
||||
|
||||
The three deepest activators are **not fakeable**; each carries its analyzed spec |
||||
in the module docs as the roadmap: |
||||
|
||||
| Module | Why it's hard (from analysis) | |
||||
|----------|-------------------------------------------------------------------------------------| |
||||
| TSforge | Byte-exact reverse-engineered SPP trusted-store (`data.dat`: RSA/AES/HMAC, per-block CRC32/SHA-256, per-OS alignment). A faithful port ≈ porting **LibTSforge** to a pure `libtsforge` crate + thin FFI to swap `data.dat`. | |
||||
| Ohook | Ships two reverse-engineered SPP-client DLLs (`sppc32/64.dll`, pinned SHA-256) that forward to the renamed genuine DLL. The blobs are **assets**, not code; the portable planner + license install is the port. | |
||||
| status | The rich detail uses the **undocumented SLC private ABI** (`SLGet*Information`, 40-byte struct stride, `KUSER_SHARED_DATA @0x7FFE02C8`). The WMI backend here covers the common case; the SLC path is a future direct-FFI slice. | |
||||
|
||||
`HWID`/`KMS38` are a shorter hop: generate `GenuineTicket.xml` (portable) and |
||||
delegate to `ClipUp.exe` + ClipSVC (the script does the same — the ticket |
||||
consumption is closed Windows behavior). |
||||
|
||||
## What is deliberately faithful |
||||
|
||||
- SPP `ApplicationID` GUIDs, WMI class names, method + parameter names, and |
||||
`LicenseStatus` codes are the real Microsoft values (see `model.rs` tests). |
||||
- Online KMS falls through a host list exactly like the script; `--kms-host` |
||||
overrides it. |
||||
- Nothing reports "activated" that didn't actually activate — unported paths |
||||
return a typed `Unsupported` error instead of lying. |
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -1,13 +0,0 @@
@@ -1,13 +0,0 @@
|
||||
-------------------------------------------------------------------------------------- |
||||
Activation Type Supported Product Activation Period |
||||
-------------------------------------------------------------------------------------- |
||||
|
||||
HWID - Windows 10-11 - Permanent |
||||
Ohook - Office - Permanent |
||||
TSforge - Windows / ESU / Office - Permanent |
||||
Online KMS - Windows / Office - 180 Days. Lifetime With Renewal Task |
||||
|
||||
-------------------------------------------------------------------------------------- |
||||
|
||||
Check the below link for more details: |
||||
https://massgrave.dev/chart |
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -1,843 +0,0 @@
@@ -1,843 +0,0 @@
|
||||
@set masver=3.12 |
||||
@echo off |
||||
|
||||
|
||||
|
||||
::============================================================================ |
||||
:: |
||||
:: Homepage: m{}assgrave{dot}dev |
||||
:: |
||||
::============================================================================ |
||||
|
||||
|
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Set environment variables, it helps if they are misconfigured in the system |
||||
|
||||
setlocal EnableExtensions |
||||
setlocal DisableDelayedExpansion |
||||
|
||||
set "PathExt=.COM;.EXE;.BAT;.CMD;.VBS;.VBE;.JS;.JSE;.WSF;.WSH;.MSC" |
||||
|
||||
set "SysPath=%SystemRoot%\System32" |
||||
set "Path=%SystemRoot%\System32;%SystemRoot%;%SystemRoot%\System32\Wbem;%SystemRoot%\System32\WindowsPowerShell\v1.0\" |
||||
if exist "%SystemRoot%\Sysnative\reg.exe" ( |
||||
set "SysPath=%SystemRoot%\Sysnative" |
||||
set "Path=%SystemRoot%\Sysnative;%SystemRoot%;%SystemRoot%\Sysnative\Wbem;%SystemRoot%\Sysnative\WindowsPowerShell\v1.0\;%Path%" |
||||
) |
||||
|
||||
set "ComSpec=%SysPath%\cmd.exe" |
||||
set "PSModulePath=%ProgramFiles%\WindowsPowerShell\Modules;%SysPath%\WindowsPowerShell\v1.0\Modules" |
||||
|
||||
cd /d "%SysPath%" |
||||
|
||||
:: Workaround for https://github.com/microsoft/terminal/issues/15212, when %0 starts with a quote %0 parameter expansion is not specialcased. |
||||
:: Changing %0 to something that is not quoted bypasses the issue. |
||||
goto arg_workaround_end |
||||
:arg_workaround |
||||
set "_cmdf=%~f0" |
||||
exit /b |
||||
:arg_workaround_end |
||||
|
||||
call :arg_workaround |
||||
|
||||
set re1= |
||||
set re2= |
||||
set "_cmdf=%~f0" |
||||
for %%# in (%*) do ( |
||||
if /i "%%#"=="re1" set re1=1 |
||||
if /i "%%#"=="re2" set re2=1 |
||||
if /i "%%#"=="-qedit" (set re1=1&set re2=1) |
||||
) |
||||
|
||||
:: Re-launch the script with x64 process if it was initiated by x86 process on x64 bit Windows |
||||
:: or with ARM64 process if it was initiated by x86/ARM32 process on ARM64 Windows |
||||
|
||||
if exist %SystemRoot%\Sysnative\cmd.exe if not defined re1 ( |
||||
setlocal EnableDelayedExpansion |
||||
start %SystemRoot%\Sysnative\cmd.exe /c ""!_cmdf!" %* re1" |
||||
exit /b |
||||
) |
||||
|
||||
:: Re-launch the script with ARM32 process if it was initiated by x64 process on ARM64 Windows |
||||
|
||||
if exist %SystemRoot%\SysArm32\cmd.exe if %PROCESSOR_ARCHITECTURE%==AMD64 if not defined re2 ( |
||||
setlocal EnableDelayedExpansion |
||||
start %SystemRoot%\SysArm32\cmd.exe /c ""!_cmdf!" %* re2" |
||||
exit /b |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
set "blank=" |
||||
set "mas=ht%blank%tps%blank%://m%blank%ass%blank%grave.dev/" |
||||
set "github=ht%blank%tps%blank%://github.com/m%blank%assgra%blank%vel/Micro%blank%soft-Acti%blank%vation-Scripts" |
||||
set "selfgit=ht%blank%tps%blank%://git.acti%blank%vated.win/Micr%blank%osoft-Act%blank%ivation-Scripts" |
||||
|
||||
:: Check if Null service is working, it's important for the batch script |
||||
|
||||
sc query Null | find /i "RUNNING" |
||||
if %errorlevel% NEQ 0 ( |
||||
echo: |
||||
echo Null service is not running, script may crash... |
||||
echo: |
||||
echo: |
||||
echo Check this webpage for help - %mas%fix_service |
||||
echo: |
||||
echo: |
||||
ping 127.0.0.1 -n 20 |
||||
) |
||||
cls |
||||
|
||||
:: Check LF line ending |
||||
|
||||
>nul findstr /v "$" "%_cmdf%" && ( |
||||
echo: |
||||
echo Error - Script either has LF line ending issue or an empty line at the end of the script is missing. |
||||
echo: |
||||
echo: |
||||
echo Check this webpage for help - %mas%troubleshoot |
||||
echo: |
||||
echo: |
||||
ping 127.0.0.1 -n 20 >nul |
||||
exit /b |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
cls |
||||
color 07 |
||||
title Extract $OEM$ Folder %masver% |
||||
|
||||
set _args= |
||||
set _elev= |
||||
set _unattended=0 |
||||
|
||||
set _args=%* |
||||
if defined _args set _args=%_args:"=% |
||||
if defined _args set _args=%_args:re1=% |
||||
if defined _args set _args=%_args:re2=% |
||||
if defined _args ( |
||||
for %%A in (%_args%) do ( |
||||
if /i "%%A"=="-el" set _elev=1 |
||||
) |
||||
) |
||||
|
||||
set "nul1=1>nul" |
||||
set "nul2=2>nul" |
||||
set "nul6=2^>nul" |
||||
set "nul=>nul 2>&1" |
||||
|
||||
call :dk_setvar |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
if %winbuild% EQU 1 ( |
||||
%eline% |
||||
echo Failed to detect Windows build number. |
||||
echo: |
||||
setlocal EnableDelayedExpansion |
||||
set fixes=%fixes% %mas%troubleshoot |
||||
call :dk_color2 %Blue% "Check this webpage for help - " %_Yellow% " %mas%troubleshoot" |
||||
goto done2 |
||||
) |
||||
|
||||
if exist "%Systemdrive%\Users\WDAGUtilityAccount" ( |
||||
sc query gcs | find /i "RUNNING" %nul% && ( |
||||
%eline% |
||||
echo Windows Sandbox detected. |
||||
echo The script cannot run due to missing licensing components. Aborting... |
||||
echo: |
||||
goto done2 |
||||
) |
||||
) |
||||
|
||||
if %winbuild% LSS 6001 ( |
||||
%nceline% |
||||
echo Unsupported OS version detected [%winbuild%]. |
||||
echo MAS only supports Windows Vista/7/8/8.1/10/11 and their Server equivalents. |
||||
if %winbuild% EQU 6000 ( |
||||
echo: |
||||
echo Windows Vista RTM is not supported because Powershell cannot be installed. |
||||
echo Upgrade to Windows Vista SP1 or SP2. |
||||
) |
||||
goto done2 |
||||
) |
||||
|
||||
if %winbuild% LSS 7600 if not exist "%SysPath%\WindowsPowerShell\v1.0\Modules" ( |
||||
%nceline% |
||||
if not exist %ps% ( |
||||
echo PowerShell is not installed in your system. |
||||
) |
||||
echo Install PowerShell 2.0 using the following URL. |
||||
echo: |
||||
echo https://www.catalog.update.microsoft.com/Search.aspx?q=KB968930 |
||||
if %_unattended%==0 start https://www.catalog.update.microsoft.com/Search.aspx?q=KB968930 |
||||
goto done2 |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Fix special character limitations in path name |
||||
|
||||
set "_work=%~dp0" |
||||
if "%_work:~-1%"=="\" set "_work=%_work:~0,-1%" |
||||
|
||||
set "_batp=%_cmdf:'=''%" |
||||
|
||||
set _PSarg="""%_cmdf%""" -el %_args% |
||||
set _PSarg=%_PSarg:'=''% |
||||
|
||||
set "_ttemp=%userprofile%\AppData\Local\Temp" |
||||
|
||||
setlocal EnableDelayedExpansion |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
echo "!_cmdf!" | find /i "!_ttemp!" %nul1% && ( |
||||
if /i not "!_work!"=="!_ttemp!" ( |
||||
%eline% |
||||
echo The script was launched from the temp folder. |
||||
echo You are most likely running the script directly from the archive file. |
||||
echo: |
||||
echo Extract the archive file and launch the script from the extracted folder. |
||||
goto dk_done |
||||
) |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Elevate script as admin and pass arguments and preventing loop |
||||
|
||||
%nul1% fltmc || ( |
||||
if not defined _elev %psc% "start cmd.exe -arg '/c \"!_PSarg!\"' -verb runas" && exit /b |
||||
%eline% |
||||
echo This script needs admin rights. |
||||
echo Right click on this script and select 'Run as administrator'. |
||||
goto done2 |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Check PowerShell |
||||
|
||||
::pstst $ExecutionContext.SessionState.LanguageMode :pstst |
||||
|
||||
for /f "delims=" %%a in ('%psc% "if ($PSVersionTable.PSEdition -ne 'Core') {$f=[IO.File]::ReadAllText('!_batp!') -split ':pstst';. ([scriptblock]::Create($f[1]))}" %nul6%') do (set tstresult=%%a) |
||||
|
||||
if /i not "%tstresult%"=="FullLanguage" ( |
||||
%eline% |
||||
for /f "delims=" %%a in ('%psc% "$ExecutionContext.SessionState.LanguageMode" %nul6%') do (set tstresult2=%%a) |
||||
echo Test 1 - %tstresult% |
||||
echo Test 2 - !tstresult2! |
||||
echo: |
||||
|
||||
REM check LanguageMode |
||||
|
||||
echo: !tstresult2! | findstr /i "ConstrainedLanguage RestrictedLanguage NoLanguage" %nul1% && ( |
||||
echo FullLanguage mode not found in PowerShell. Aborting... |
||||
echo If you have applied restrictions on Powershell then undo those changes. |
||||
set fixes=%fixes% %mas%fix_powershell |
||||
call :dk_color2 %Blue% "Check this webpage for help - " %_Yellow% " %mas%fix_powershell" |
||||
goto done2 |
||||
) |
||||
|
||||
REM check Powershell core version |
||||
|
||||
cmd /c "%psc% "$PSVersionTable.PSEdition"" | find /i "Core" %nul1% && ( |
||||
echo Windows Powershell is needed for MAS but it seems to be replaced with Powershell core. Aborting... |
||||
set fixes=%fixes% %mas%in-place_repair_upgrade |
||||
call :dk_color2 %Blue% "Check this webpage for help - " %_Yellow% " %mas%in-place_repair_upgrade" |
||||
goto done2 |
||||
) |
||||
|
||||
REM check for Mal-ware that may cause issues with Powershell |
||||
|
||||
for /r "%ProgramFiles%\" %%f in (secureboot.exe) do if exist "%%f" ( |
||||
echo "%%f" |
||||
echo Mal%blank%ware found, PowerShell is not working properly. |
||||
set fixes=%fixes% %mas%remove_mal%w%ware |
||||
call :dk_color2 %Blue% "Check this webpage for help - " %_Yellow% " %mas%remove_mal%w%ware" |
||||
goto done2 |
||||
) |
||||
|
||||
REM check if .NET is working properly |
||||
|
||||
if /i "!tstresult2!"=="FullLanguage" ( |
||||
cmd /c "%psc% ""try {[System.AppDomain]::CurrentDomain.GetAssemblies(); [System.Math]::Sqrt(144)} catch {Exit 3}""" %nul% |
||||
if !errorlevel!==3 ( |
||||
echo Windows Powershell failed to load .NET command. Aborting... |
||||
set fixes=%fixes% %mas%in-place_repair_upgrade |
||||
call :dk_color2 %Blue% "Check this webpage for help - " %_Yellow% " %mas%in-place_repair_upgrade" |
||||
goto done2 |
||||
) |
||||
) |
||||
|
||||
REM check antivirus and other errors |
||||
|
||||
echo PowerShell is not working properly. Aborting... |
||||
|
||||
if /i "!tstresult2!"=="FullLanguage" ( |
||||
echo: |
||||
echo Your antivirus software might be blocking the script. |
||||
echo: |
||||
sc query sense | find /i "RUNNING" %nul% && ( |
||||
echo Installed Antivirus - Microsoft Defender for Endpoint |
||||
) |
||||
cmd /c "%psc% ""$av = Get-WmiObject -Namespace root\SecurityCenter2 -Class AntiVirusProduct; $n = @(); foreach ($i in $av) { $n += $i.displayName }; if ($n) { Write-Host ('Installed Antivirus - ' + ($n -join ', '))}""" |
||||
) |
||||
|
||||
set fixes=%fixes% %mas%troubleshoot |
||||
call :dk_color2 %Blue% "Check this webpage for help - " %_Yellow% " %mas%troubleshoot" |
||||
goto done2 |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Disable QuickEdit and launch from conhost.exe to avoid Terminal app |
||||
|
||||
if %winbuild% GEQ 17763 ( |
||||
set terminal=1 |
||||
) else ( |
||||
set terminal= |
||||
) |
||||
|
||||
:: Check if script is running in Terminal app |
||||
|
||||
if defined terminal ( |
||||
set lines=0 |
||||
for /f "skip=3 tokens=* delims=" %%A in ('mode con') do if "!lines!"=="0" ( |
||||
for %%B in (%%A) do set lines=%%B |
||||
) |
||||
if !lines! GEQ 100 set terminal= |
||||
) |
||||
|
||||
if %_unattended%==1 goto :skipQE |
||||
for %%# in (%_args%) do (if /i "%%#"=="-qedit" goto :skipQE) |
||||
|
||||
:: Relaunch to disable QuickEdit in the current session and use conhost.exe instead of the Terminal app |
||||
:: This code disables QuickEdit for the current cmd.exe session without making permanent registry changes |
||||
:: It is included because clicking on the script window can pause execution, causing confusion that the script has stopped due to an error |
||||
|
||||
set resetQE=1 |
||||
reg query HKCU\Console /v QuickEdit %nul2% | find /i "0x0" %nul1% && set resetQE=0 |
||||
reg add HKCU\Console /v QuickEdit /t REG_DWORD /d 0 /f %nul1% |
||||
|
||||
if defined terminal ( |
||||
start conhost.exe "!_cmdf!" %_args% -qedit |
||||
start reg add HKCU\Console /v QuickEdit /t REG_DWORD /d %resetQE% /f %nul1% |
||||
exit /b |
||||
) else if %resetQE% EQU 1 ( |
||||
start cmd.exe /c ""!_cmdf!" %_args% -qedit" |
||||
start reg add HKCU\Console /v QuickEdit /t REG_DWORD /d %resetQE% /f %nul1% |
||||
exit /b |
||||
) |
||||
|
||||
:skipQE |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Check for updates |
||||
|
||||
set -= |
||||
set old= |
||||
set pingp= |
||||
set upver=%masver:.=% |
||||
|
||||
for %%A in ( |
||||
activ%-%ated.win |
||||
mass%-%grave.dev |
||||
) do if not defined pingp ( |
||||
for /f "delims=[] tokens=2" %%B in ('ping -n 1 %%A') do ( |
||||
if not "%%B"=="" (set old=1& set pingp=1) |
||||
for /f "delims=[] tokens=2" %%C in ('ping -n 1 updatecheck%upver%.%%A') do ( |
||||
if not "%%C"=="" set old= |
||||
) |
||||
) |
||||
) |
||||
|
||||
if defined old ( |
||||
echo ________________________________________________ |
||||
%eline% |
||||
echo Your version of MAS [%masver%] is outdated. |
||||
echo ________________________________________________ |
||||
echo: |
||||
if not %_unattended%==1 ( |
||||
echo [1] Get Latest MAS |
||||
echo [0] Continue Anyway |
||||
echo: |
||||
call :dk_color %_Green% "Choose a menu option using your keyboard [1,0] :" |
||||
choice /C:10 /N |
||||
if !errorlevel!==2 rem |
||||
if !errorlevel!==1 (start %selfgit% & start %github% & start %mas% & exit /b) |
||||
) |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
setlocal DisableDelayedExpansion |
||||
|
||||
:: Check desktop location |
||||
|
||||
set desktop= |
||||
for /f "skip=2 tokens=2*" %%a in ('reg query "HKCU\Software\Microsoft\Windows\CurrentVersion\Explorer\User Shell Folders" /v Desktop') do call set "desktop=%%b" |
||||
if not defined desktop for /f "delims=" %%a in ('%psc% "& {write-host $([Environment]::GetFolderPath('Desktop'))}"') do call set "desktop=%%a" |
||||
|
||||
set "_pdesk=%desktop:'=''%" |
||||
set "_dir=%desktop%\$OEM$\$$\Setup\Scripts" |
||||
|
||||
if exist "!desktop!\" ( |
||||
%eline% |
||||
echo Unable to detect Desktop location, aborting... |
||||
goto done2 |
||||
) |
||||
|
||||
setlocal EnableDelayedExpansion |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
if not defined terminal mode 78, 30 |
||||
|
||||
if exist "!desktop!\$OEM$\" ( |
||||
echo _____________________________________________________ |
||||
%eline% |
||||
echo The $OEM$ folder already exists on your Desktop. |
||||
echo _____________________________________________________ |
||||
goto done2 |
||||
) |
||||
|
||||
set HWID_Activation.cmd=Activators\HWID_Activation.cmd |
||||
set Online_KMS_Activation.cmd=Activators\Online_KMS_Activation.cmd |
||||
set Ohook_Activation_AIO.cmd=Activators\Ohook_Activation_AIO.cmd |
||||
set TSforge_Activation.cmd=Activators\TSforge_Activation.cmd |
||||
pushd "!_work!" |
||||
|
||||
set _nofile= |
||||
for %%# in ( |
||||
%HWID_Activation.cmd% |
||||
%Online_KMS_Activation.cmd% |
||||
%Ohook_Activation_AIO.cmd% |
||||
%TSforge_Activation.cmd% |
||||
) do ( |
||||
if not exist "%%#" set _nofile=1 |
||||
) |
||||
|
||||
popd |
||||
|
||||
if defined _nofile ( |
||||
echo _____________________________________________________ |
||||
%eline% |
||||
echo Some files are missing in the 'Activators' folder. |
||||
echo _____________________________________________________ |
||||
goto done2 |
||||
) |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:Menu |
||||
|
||||
cls |
||||
if not defined terminal mode 78, 30 |
||||
echo: |
||||
echo: |
||||
echo: |
||||
echo: |
||||
echo: Extract $OEM$ folder on the desktop |
||||
echo: ____________________________________________________________ |
||||
echo: |
||||
echo: [1] HWID [Windows] |
||||
echo: [2] Ohook [Office] |
||||
echo: [3] TSforge [Windows / ESU / Office] |
||||
echo: [4] Online KMS [Windows / Office] |
||||
echo: |
||||
echo: [5] HWID [Windows] ^+ Ohook [Office] |
||||
echo: [6] HWID [Windows] ^+ Ohook [Office] ^+ TSforge [ESU] |
||||
echo: [7] TSforge [Windows / ESU] ^+ Ohook [Office] |
||||
echo: |
||||
call :dk_color2 %_White% " [R] " %_Green% "ReadMe" |
||||
echo: [0] Exit |
||||
echo: ____________________________________________________________ |
||||
echo: |
||||
call :dk_color2 %_White% " " %_Green% "Choose a menu option using your keyboard :" |
||||
choice /C:1234567R0 /N |
||||
set _erl=%errorlevel% |
||||
|
||||
if %_erl%==9 exit /b |
||||
if %_erl%==8 start %mas%oem-folder &goto :Menu |
||||
if %_erl%==7 goto:tsforge_ohook |
||||
if %_erl%==6 goto:hwid_ohook_tsforge |
||||
if %_erl%==5 goto:hwid_ohook |
||||
if %_erl%==4 goto:kms |
||||
if %_erl%==3 goto:tsforge |
||||
if %_erl%==2 goto:ohook |
||||
if %_erl%==1 goto:hwid |
||||
goto :Menu |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:hwid |
||||
|
||||
cls |
||||
md "!desktop!\$OEM$\$$\Setup\Scripts" |
||||
pushd "!_work!" |
||||
copy /y /b "%HWID_Activation.cmd%" "!_dir!\HWID_Activation.cmd" %nul% |
||||
popd |
||||
call :export hwid_setup |
||||
|
||||
set _error= |
||||
if not exist "!_dir!\HWID_Activation.cmd" set _error=1 |
||||
if not exist "!_dir!\SetupComplete.cmd" set _error=1 |
||||
if defined _error goto errorfound |
||||
|
||||
set oem=HWID |
||||
goto done |
||||
|
||||
:hwid_setup: |
||||
@echo off |
||||
|
||||
fltmc >nul || exit /b |
||||
|
||||
call "%~dp0HWID_Activation.cmd" /HWID |
||||
|
||||
cd \ |
||||
(goto) 2>nul & (if "%~dp0"=="%SystemRoot%\Setup\Scripts\" rd /s /q "%~dp0") |
||||
:hwid_setup: |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:ohook |
||||
|
||||
cls |
||||
md "!desktop!\$OEM$\$$\Setup\Scripts" |
||||
pushd "!_work!" |
||||
copy /y /b %Ohook_Activation_AIO.cmd% "!_dir!\Ohook_Activation_AIO.cmd" %nul% |
||||
popd |
||||
call :export ohook_setup |
||||
|
||||
set _error= |
||||
if not exist "!_dir!\Ohook_Activation_AIO.cmd" set _error=1 |
||||
if not exist "!_dir!\SetupComplete.cmd" set _error=1 |
||||
if defined _error goto errorfound |
||||
|
||||
set oem=Ohook |
||||
goto done |
||||
|
||||
:ohook_setup: |
||||
@echo off |
||||
|
||||
fltmc >nul || exit /b |
||||
|
||||
call "%~dp0Ohook_Activation_AIO.cmd" /Ohook |
||||
|
||||
cd \ |
||||
(goto) 2>nul & (if "%~dp0"=="%SystemRoot%\Setup\Scripts\" rd /s /q "%~dp0") |
||||
:ohook_setup: |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:tsforge |
||||
|
||||
cls |
||||
md "!desktop!\$OEM$\$$\Setup\Scripts" |
||||
pushd "!_work!" |
||||
copy /y /b "%TSforge_Activation.cmd%" "!_dir!\TSforge_Activation.cmd" %nul% |
||||
popd |
||||
call :export tsforge_setup |
||||
|
||||
set _error= |
||||
if not exist "!_dir!\TSforge_Activation.cmd" set _error=1 |
||||
if not exist "!_dir!\SetupComplete.cmd" set _error=1 |
||||
if defined _error goto errorfound |
||||
|
||||
set oem=TSforge |
||||
goto done |
||||
|
||||
:tsforge_setup: |
||||
@echo off |
||||
|
||||
fltmc >nul || exit /b |
||||
|
||||
call "%~dp0TSforge_Activation.cmd" /Z-WindowsESUOffice |
||||
|
||||
cd \ |
||||
(goto) 2>nul & (if "%~dp0"=="%SystemRoot%\Setup\Scripts\" rd /s /q "%~dp0") |
||||
:tsforge_setup: |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:kms |
||||
|
||||
cls |
||||
md "!desktop!\$OEM$\$$\Setup\Scripts" |
||||
pushd "!_work!" |
||||
copy /y /b "%Online_KMS_Activation.cmd%" "!_dir!\Online_KMS_Activation.cmd" %nul% |
||||
popd |
||||
call :export kms_setup |
||||
|
||||
set _error= |
||||
if not exist "!_dir!\Online_KMS_Activation.cmd" set _error=1 |
||||
if not exist "!_dir!\SetupComplete.cmd" set _error=1 |
||||
if defined _error goto errorfound |
||||
|
||||
set oem=Online KMS |
||||
goto done |
||||
|
||||
:kms_setup: |
||||
@echo off |
||||
|
||||
fltmc >nul || exit /b |
||||
|
||||
call "%~dp0Online_KMS_Activation.cmd" /K-WindowsOffice |
||||
|
||||
cd \ |
||||
(goto) 2>nul & (if "%~dp0"=="%SystemRoot%\Setup\Scripts\" rd /s /q "%~dp0") |
||||
:kms_setup: |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:hwid_ohook |
||||
|
||||
cls |
||||
md "!desktop!\$OEM$\$$\Setup\Scripts" |
||||
pushd "!_work!" |
||||
copy /y /b "%HWID_Activation.cmd%" "!_dir!\HWID_Activation.cmd" %nul% |
||||
copy /y /b "%Ohook_Activation_AIO.cmd%" "!_dir!\Ohook_Activation_AIO.cmd" %nul% |
||||
popd |
||||
call :export hwid_ohook_setup |
||||
|
||||
set _error= |
||||
if not exist "!_dir!\HWID_Activation.cmd" set _error=1 |
||||
if not exist "!_dir!\Ohook_Activation_AIO.cmd" set _error=1 |
||||
if not exist "!_dir!\SetupComplete.cmd" set _error=1 |
||||
if defined _error goto errorfound |
||||
|
||||
set oem=HWID [Windows] + Ohook [Office] |
||||
goto done |
||||
|
||||
:hwid_ohook_setup: |
||||
@echo off |
||||
|
||||
fltmc >nul || exit /b |
||||
|
||||
setlocal |
||||
call "%~dp0HWID_Activation.cmd" /HWID |
||||
endlocal |
||||
|
||||
setlocal |
||||
call "%~dp0Ohook_Activation_AIO.cmd" /Ohook |
||||
endlocal |
||||
|
||||
cd \ |
||||
(goto) 2>nul & (if "%~dp0"=="%SystemRoot%\Setup\Scripts\" rd /s /q "%~dp0") |
||||
:hwid_ohook_setup: |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:hwid_ohook_tsforge |
||||
|
||||
cls |
||||
md "!desktop!\$OEM$\$$\Setup\Scripts" |
||||
pushd "!_work!" |
||||
copy /y /b "%HWID_Activation.cmd%" "!_dir!\HWID_Activation.cmd" %nul% |
||||
copy /y /b "%Ohook_Activation_AIO.cmd%" "!_dir!\Ohook_Activation_AIO.cmd" %nul% |
||||
copy /y /b "%TSforge_Activation.cmd%" "!_dir!\TSforge_Activation.cmd" %nul% |
||||
popd |
||||
call :export hwid_ohook_tsforge_setup |
||||
|
||||
set _error= |
||||
if not exist "!_dir!\HWID_Activation.cmd" set _error=1 |
||||
if not exist "!_dir!\Ohook_Activation_AIO.cmd" set _error=1 |
||||
if not exist "!_dir!\TSforge_Activation.cmd" set _error=1 |
||||
if not exist "!_dir!\SetupComplete.cmd" set _error=1 |
||||
if defined _error goto errorfound |
||||
|
||||
set oem=HWID [Windows] + Ohook [Office] + TSforge [ESU] |
||||
goto done |
||||
|
||||
:hwid_ohook_tsforge_setup: |
||||
@echo off |
||||
|
||||
fltmc >nul || exit /b |
||||
|
||||
setlocal |
||||
call "%~dp0HWID_Activation.cmd" /HWID |
||||
endlocal |
||||
|
||||
setlocal |
||||
call "%~dp0Ohook_Activation_AIO.cmd" /Ohook |
||||
endlocal |
||||
|
||||
setlocal |
||||
call "%~dp0TSforge_Activation.cmd" /Z-ESU |
||||
endlocal |
||||
|
||||
cd \ |
||||
(goto) 2>nul & (if "%~dp0"=="%SystemRoot%\Setup\Scripts\" rd /s /q "%~dp0") |
||||
:hwid_ohook_tsforge_setup: |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:tsforge_ohook |
||||
|
||||
cls |
||||
md "!desktop!\$OEM$\$$\Setup\Scripts" |
||||
pushd "!_work!" |
||||
copy /y /b "%TSforge_Activation.cmd%" "!_dir!\TSforge_Activation.cmd" %nul% |
||||
copy /y /b "%Ohook_Activation_AIO.cmd%" "!_dir!\Ohook_Activation_AIO.cmd" %nul% |
||||
popd |
||||
call :export tsforge_ohook_setup |
||||
|
||||
set _error= |
||||
if not exist "!_dir!\TSforge_Activation.cmd" set _error=1 |
||||
if not exist "!_dir!\Ohook_Activation_AIO.cmd" set _error=1 |
||||
if not exist "!_dir!\SetupComplete.cmd" set _error=1 |
||||
if defined _error goto errorfound |
||||
|
||||
set oem=TSforge [Windows / ESU] + Ohook [Office] |
||||
goto done |
||||
|
||||
:tsforge_ohook_setup: |
||||
@echo off |
||||
|
||||
fltmc >nul || exit /b |
||||
|
||||
setlocal |
||||
call "%~dp0TSforge_Activation.cmd" /Z-Windows /Z-ESU |
||||
endlocal |
||||
|
||||
setlocal |
||||
call "%~dp0Ohook_Activation_AIO.cmd" /Ohook |
||||
endlocal |
||||
|
||||
cd \ |
||||
(goto) 2>nul & (if "%~dp0"=="%SystemRoot%\Setup\Scripts\" rd /s /q "%~dp0") |
||||
:tsforge_ohook_setup: |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:errorfound |
||||
|
||||
%eline% |
||||
echo The script failed to create the $OEM$ folder. |
||||
goto :done2 |
||||
|
||||
:done |
||||
|
||||
echo ______________________________________________________________ |
||||
echo: |
||||
call :dk_color %Blue% "%oem%" |
||||
call :dk_color %Green% "$OEM$ folder was successfully created on your Desktop." |
||||
echo ______________________________________________________________ |
||||
|
||||
:done2 |
||||
|
||||
echo: |
||||
if defined fixes ( |
||||
call :dk_color %White% "Follow ALL the ABOVE blue lines. " |
||||
call :dk_color2 %Blue% "Press [1] to Open Support Webpage " %Gray% " Press [0] to Go Back" |
||||
choice /C:10 /N |
||||
if !errorlevel!==2 exit /b |
||||
if !errorlevel!==1 (start %selfgit% & start %github% & for %%# in (%fixes%) do (start %%#)) |
||||
) |
||||
|
||||
if defined terminal ( |
||||
call :dk_color %_Yellow% "Press [0] key to %_exitmsg%..." |
||||
choice /c 0 /n |
||||
) else ( |
||||
call :dk_color %_Yellow% "Press any key to %_exitmsg%..." |
||||
pause %nul1% |
||||
) |
||||
exit /b |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Set variables |
||||
|
||||
:dk_setvar |
||||
|
||||
set ps=%SysPath%\WindowsPowerShell\v1.0\powershell.exe |
||||
set psc=%ps% -nop -c |
||||
set winbuild=1 |
||||
for /f "tokens=2 delims=[]" %%G in ('ver') do for /f "tokens=2,3,4 delims=. " %%H in ("%%~G") do set "winbuild=%%J" |
||||
|
||||
set _slexe=sppsvc.exe& set _slser=sppsvc |
||||
if %winbuild% LEQ 6300 (set _slexe=SLsvc.exe& set _slser=SLsvc) |
||||
if %winbuild% LSS 7600 if exist "%SysPath%\SLsvc.exe" (set _slexe=SLsvc.exe& set _slser=SLsvc) |
||||
if %_slexe%==SLsvc.exe set _vis=1 |
||||
|
||||
set _NCS=1 |
||||
if %winbuild% LSS 10586 set _NCS=0 |
||||
if %winbuild% GEQ 10586 reg query "HKCU\Console" /v ForceV2 %nul2% | find /i "0x0" %nul1% && (set _NCS=0) |
||||
|
||||
echo "%PROCESSOR_ARCHITECTURE% %PROCESSOR_ARCHITEW6432%" | find /i "ARM64" %nul1% && (if %winbuild% LSS 21277 set ps32onArm=1) |
||||
|
||||
if %_NCS% EQU 1 ( |
||||
for /F %%a in ('echo prompt $E ^| cmd') do set "esc=%%a" |
||||
set "Red="41;97m"" |
||||
set "Gray="100;97m"" |
||||
set "Green="42;97m"" |
||||
set "Blue="44;97m"" |
||||
set "White="107;91m"" |
||||
set "_Red="40;91m"" |
||||
set "_White="40;37m"" |
||||
set "_Green="40;92m"" |
||||
set "_Yellow="40;93m"" |
||||
) else ( |
||||
set "Red="Red" "white"" |
||||
set "Gray="Darkgray" "white"" |
||||
set "Green="DarkGreen" "white"" |
||||
set "Blue="Blue" "white"" |
||||
set "White="White" "Red"" |
||||
set "_Red="Black" "Red"" |
||||
set "_White="Black" "Gray"" |
||||
set "_Green="Black" "Green"" |
||||
set "_Yellow="Black" "Yellow"" |
||||
) |
||||
|
||||
set "nceline=echo: &echo ==== ERROR ==== &echo:" |
||||
set "eline=echo: &call :dk_color %Red% "==== ERROR ====" &echo:" |
||||
if %~z0 GEQ 200000 ( |
||||
set "_exitmsg=Go back" |
||||
set "_fixmsg=Go back to Main Menu, select Troubleshoot and run Fix Licensing option." |
||||
) else ( |
||||
set "_exitmsg=Exit" |
||||
set "_fixmsg=In MAS folder, run Troubleshoot script and select Fix Licensing option." |
||||
) |
||||
exit /b |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:: Extract the text from batch script without character and file encoding issue |
||||
|
||||
:export |
||||
|
||||
%psc% "$f=[IO.File]::ReadAllText('!_batp!') -split \":%~1\:.*`r`n\"; [io.file]::WriteAllText('!_pdesk!\$OEM$\$$\Setup\Scripts\SetupComplete.cmd',$f[1].Trim(),[System.Text.Encoding]::ASCII);" |
||||
exit /b |
||||
|
||||
::======================================================================================================================================== |
||||
|
||||
:dk_color |
||||
|
||||
if %_NCS% EQU 1 ( |
||||
echo %esc%[%~1%~2%esc%[0m |
||||
) else if exist %ps% ( |
||||
%psc% write-host -back '%1' -fore '%2' '%3' |
||||
) else if not exist %ps% ( |
||||
echo %~3 |
||||
) |
||||
exit /b |
||||
|
||||
:dk_color2 |
||||
|
||||
if %_NCS% EQU 1 ( |
||||
echo %esc%[%~1%~2%esc%[%~3%~4%esc%[0m |
||||
) else if exist %ps% ( |
||||
%psc% write-host -back '%1' -fore '%2' '%3' -NoNewline; write-host -back '%4' -fore '%5' '%6' |
||||
) else if not exist %ps% ( |
||||
echo %~3 %~6 |
||||
) |
||||
exit /b |
||||
|
||||
::======================================================================================================================================== |
||||
:: Leave empty line below |
||||
File diff suppressed because it is too large
Load Diff
@ -1,3 +0,0 @@
@@ -1,3 +0,0 @@
|
||||
<html> |
||||
<meta http-equiv="refresh" content="0; url=https://massgrave.dev/"> |
||||
</html> |
||||
@ -0,0 +1,52 @@
@@ -0,0 +1,52 @@
|
||||
//! `HWID_Activation.cmd` — permanent digital-license activation (Windows 10/11).
|
||||
//!
|
||||
//! Faithful flow (from the script):
|
||||
//! 1. Verify build ≥ 10240 and that the edition has a digital-license path.
|
||||
//! 2. Ensure the correct GVLK/edition key + license files exist under
|
||||
//! `%SysPath%\spp\tokens\skus\<edition>\*GVLK*.xrm-ms`.
|
||||
//! 3. Generate `GenuineTicket.xml` from the machine's hardware hash + region
|
||||
//! (`HKCU\Control Panel\International\Geo`) and drop it in
|
||||
//! `%ProgramData%\Microsoft\Windows\ClipSVC\GenuineTicket`.
|
||||
//! 4. Hand the ticket to Windows two ways for reliability: restart ClipSVC,
|
||||
//! and run `ClipUp.exe -v -o`. The service consumes the ticket and writes
|
||||
//! `tokens.dat`.
|
||||
//!
|
||||
//! Port boundary (from analysis `port_risk`): steps 1–3 are pure/portable —
|
||||
//! the HWID key tables (`HwidEntry`/`FallbackEntry`) and ticket XML assembly are
|
||||
//! unit-testable data. Step 4 is **closed SPP behavior**: the Rust port cannot
|
||||
//! reimplement ClipSVC consuming the ticket; like the script, it generates a
|
||||
//! valid ticket and delegates to `clipup.exe` + the service.
|
||||
//!
|
||||
//! Status: portable ticket/table logic is the next slice to port; the ClipUp
|
||||
//! delegation is a thin [`Spp`]-adjacent shell-out. Not yet wired — running it
|
||||
//! reports [`Error::Unsupported`] rather than pretending to activate.
|
||||
|
||||
use crate::activation::Activator; |
||||
use crate::error::{Error, Result}; |
||||
use crate::model::{Method, Product}; |
||||
use crate::platform::Spp; |
||||
|
||||
pub struct Hwid; |
||||
|
||||
impl Activator for Hwid { |
||||
fn method(&self) -> Method { |
||||
Method::Hwid |
||||
} |
||||
|
||||
fn run(&self, spp: &dyn Spp, _product: Product) -> Result<()> { |
||||
if !spp.is_elevated() { |
||||
return Err(Error::NotElevated); |
||||
} |
||||
if spp.windows_build()? < 10240 { |
||||
return Err(Error::Unsupported { |
||||
what: "HWID activation requires Windows 10 or later".into(), |
||||
}); |
||||
} |
||||
// ponytail: portable ticket-generation + ClipUp delegation not yet
|
||||
// ported — this is the next slice, spec captured in the module docs.
|
||||
Err(Error::Unsupported { |
||||
what: "HWID ticket generation + ClipUp delegation not yet ported (see module docs)" |
||||
.into(), |
||||
}) |
||||
} |
||||
} |
||||
@ -0,0 +1,38 @@
@@ -0,0 +1,38 @@
|
||||
//! KMS38 — extend KMS activation to 2038-01-19.
|
||||
//!
|
||||
//! In MAS this is not a separate file: it is a sub-flow of
|
||||
//! `Online_KMS_Activation.cmd` (with support paths in `Troubleshoot.cmd`). It
|
||||
//! installs a GVLK, then uses `ClipUp.exe -o` with the special "gatherosstate"
|
||||
//! ticket so the SPP records an activation whose expiry is pinned to the KMS38
|
||||
//! epoch instead of the usual 180 days.
|
||||
//!
|
||||
//! Port boundary: the epoch/expiry bookkeeping is portable; the actual ticket
|
||||
//! consumption is the same closed ClipSVC path as [`super::hwid`]. Shares the
|
||||
//! ClipUp delegation once that slice lands.
|
||||
|
||||
use crate::activation::Activator; |
||||
use crate::error::{Error, Result}; |
||||
use crate::model::{Method, Product}; |
||||
use crate::platform::Spp; |
||||
|
||||
pub struct Kms38; |
||||
|
||||
impl Activator for Kms38 { |
||||
fn method(&self) -> Method { |
||||
Method::Kms38 |
||||
} |
||||
|
||||
fn run(&self, spp: &dyn Spp, _product: Product) -> Result<()> { |
||||
if !spp.is_elevated() { |
||||
return Err(Error::NotElevated); |
||||
} |
||||
if spp.windows_build()? < 10240 { |
||||
return Err(Error::Unsupported { |
||||
what: "KMS38 requires Windows 10 or later".into(), |
||||
}); |
||||
} |
||||
Err(Error::Unsupported { |
||||
what: "KMS38 ClipUp delegation not yet ported (shares HWID's ClipSVC path)".into(), |
||||
}) |
||||
} |
||||
} |
||||
@ -0,0 +1,39 @@
@@ -0,0 +1,39 @@
|
||||
//! Activation methods.
|
||||
//!
|
||||
//! Each activator is a small type implementing [`Activator`]. The menu picks one
|
||||
//! and calls [`Activator::run`] with the SPP backend and target product. Read-only
|
||||
//! status reporting lives in [`status`].
|
||||
|
||||
use crate::error::Result; |
||||
use crate::model::{Method, Product}; |
||||
use crate::platform::Spp; |
||||
|
||||
pub mod status; |
||||
|
||||
pub mod online_kms; |
||||
// Deep SPP-internals activators — scaffolded against the analyzed script specs.
|
||||
pub mod hwid; |
||||
pub mod kms38; |
||||
pub mod ohook; |
||||
pub mod tsforge; |
||||
|
||||
/// A runnable activation method.
|
||||
pub trait Activator { |
||||
fn method(&self) -> Method; |
||||
|
||||
/// Perform the activation against `product`. Implementations must be
|
||||
/// idempotent-safe: re-running should not corrupt state (the scripts
|
||||
/// re-check status before and after).
|
||||
fn run(&self, spp: &dyn Spp, product: Product) -> Result<()>; |
||||
} |
||||
|
||||
/// Build the activator for a method.
|
||||
pub fn activator_for(method: Method) -> Box<dyn Activator> { |
||||
match method { |
||||
Method::OnlineKms => Box::new(online_kms::OnlineKms), |
||||
Method::Hwid => Box::new(hwid::Hwid), |
||||
Method::Kms38 => Box::new(kms38::Kms38), |
||||
Method::Ohook => Box::new(ohook::Ohook), |
||||
Method::TSforge => Box::new(tsforge::TSforge), |
||||
} |
||||
} |
||||
@ -0,0 +1,48 @@
@@ -0,0 +1,48 @@
|
||||
//! `Ohook_Activation_AIO.cmd` — permanent Office activation via SPP hook.
|
||||
//!
|
||||
//! **This module cannot be fully reimplemented in safe Rust, by design of the
|
||||
//! technique.** (Analysis `port_risk`.) Ohook works by placing two
|
||||
//! reverse-engineered SPP client DLLs (`sppc32.dll` / `sppc64.dll`, acting as
|
||||
//! `OSPPC.DLL`) that forward to the renamed genuine `sppcs.dll` while overriding
|
||||
//! the license-status calls so Office always reports "licensed". Those DLLs are
|
||||
//! opaque binary blobs with **pinned SHA-256 hashes**; a faithful port must ship
|
||||
//! the identical bytes — Rust cannot "reimplement the SPP internals".
|
||||
//!
|
||||
//! What *is* portable and worth porting here:
|
||||
//! * the embedded product tables (`OhookEntry`/`MsiEntry`: version, activation
|
||||
//! ID GUID, GVLK, license suffix, edition) — pure data, unit-testable;
|
||||
//! * build→WMI-class selection (≥9200 ⇒ `SoftwareLicensing*`, else `OSPP*`);
|
||||
//! * install-plan assembly: which license `.xrm-ms` files to install and where
|
||||
//! to drop the DLL for the detected Office install (MSI vs Click-to-Run vfs).
|
||||
//!
|
||||
//! The port therefore = ported planner + verified-hash blob placement + license
|
||||
//! install through [`Spp::install_license`]. The blobs are assets, not code.
|
||||
|
||||
use crate::activation::Activator; |
||||
use crate::error::{Error, Result}; |
||||
use crate::model::{Method, Product}; |
||||
use crate::platform::Spp; |
||||
|
||||
pub struct Ohook; |
||||
|
||||
impl Activator for Ohook { |
||||
fn method(&self) -> Method { |
||||
Method::Ohook |
||||
} |
||||
|
||||
fn run(&self, spp: &dyn Spp, product: Product) -> Result<()> { |
||||
if !spp.is_elevated() { |
||||
return Err(Error::NotElevated); |
||||
} |
||||
if product != Product::Office { |
||||
return Err(Error::Unsupported { |
||||
what: "Ohook only activates Office".into(), |
||||
}); |
||||
} |
||||
// The SPP-client blobs are external assets (pinned SHA-256); the planner
|
||||
// + license install is the portable slice to port next.
|
||||
Err(Error::Unsupported { |
||||
what: "Ohook requires the pinned SPP-client DLL assets (see module docs)".into(), |
||||
}) |
||||
} |
||||
} |
||||
@ -0,0 +1,180 @@
@@ -0,0 +1,180 @@
|
||||
//! `Online_KMS_Activation.cmd` — classic KMS client activation.
|
||||
//!
|
||||
//! Flow (faithful to the script, expressed as SPP calls):
|
||||
//! 1. Confirm a KMS-client (GVLK) key is installed; the SPP already knows the
|
||||
//! per-edition GVLK, so `InstallProductKey` is only needed if a retail key
|
||||
//! is present.
|
||||
//! 2. Point the client at a KMS host (`SetKeyManagementServiceMachine` + port).
|
||||
//! 3. `Activate()` each applicable product.
|
||||
//! 4. Fall back through the default host list until one succeeds.
|
||||
//!
|
||||
//! The orchestration is portable; the effects happen through the [`Spp`] trait.
|
||||
|
||||
use crate::activation::Activator; |
||||
use crate::data::kms_hosts::{DEFAULT_KMS_HOSTS, DEFAULT_KMS_PORT}; |
||||
use crate::error::{Error, Result}; |
||||
use crate::model::{Method, Product}; |
||||
use crate::platform::Spp; |
||||
|
||||
/// Online KMS activator. Optionally pinned to a specific host/port.
|
||||
pub struct OnlineKms; |
||||
|
||||
/// Try one KMS host: set it, then activate every installed product of the
|
||||
/// family. Returns `Ok(true)` if at least one product activated.
|
||||
fn try_host(spp: &dyn Spp, product: Product, host: &str, port: u16) -> Result<bool> { |
||||
spp.set_kms_host(product, host, port)?; |
||||
let mut any = false; |
||||
for info in spp.installed_products(product)? { |
||||
// Only KMS-client (GVLK) products can activate against a KMS host.
|
||||
if info.partial_product_key.is_none() { |
||||
continue; |
||||
} |
||||
if spp.activate(product, &info.activation_id).is_ok() { |
||||
any = true; |
||||
} |
||||
} |
||||
Ok(any) |
||||
} |
||||
|
||||
impl OnlineKms { |
||||
/// Activate against an explicit host (from `--kms-host`).
|
||||
pub fn run_with_host( |
||||
&self, |
||||
spp: &dyn Spp, |
||||
product: Product, |
||||
host: &str, |
||||
port: u16, |
||||
) -> Result<()> { |
||||
if try_host(spp, product, host, port)? { |
||||
Ok(()) |
||||
} else { |
||||
Err(Error::winapi_msg(format!( |
||||
"KMS activation failed against {host}:{port}" |
||||
))) |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl Activator for OnlineKms { |
||||
fn method(&self) -> Method { |
||||
Method::OnlineKms |
||||
} |
||||
|
||||
fn run(&self, spp: &dyn Spp, product: Product) -> Result<()> { |
||||
for host in DEFAULT_KMS_HOSTS { |
||||
match try_host(spp, product, host, DEFAULT_KMS_PORT) { |
||||
Ok(true) => return Ok(()), |
||||
Ok(false) => continue, |
||||
Err(_) => continue, // host unreachable — try the next
|
||||
} |
||||
} |
||||
Err(Error::winapi_msg( |
||||
"KMS activation failed against every default host — pass --kms-host", |
||||
)) |
||||
} |
||||
} |
||||
|
||||
#[cfg(test)] |
||||
mod tests { |
||||
use super::*; |
||||
use crate::model::{LicenseStatus, Product}; |
||||
use crate::platform::LicenseInfo; |
||||
use std::cell::RefCell; |
||||
|
||||
/// Fake SPP recording calls, to test the orchestration without Windows.
|
||||
#[derive(Default)] |
||||
struct FakeSpp { |
||||
set_hosts: RefCell<Vec<(String, u16)>>, |
||||
activated: RefCell<Vec<String>>, |
||||
// if true, activate() succeeds only for the named "good" host
|
||||
good_host: Option<String>, |
||||
current_host: RefCell<Option<String>>, |
||||
} |
||||
|
||||
impl Spp for FakeSpp { |
||||
fn installed_products(&self, _p: Product) -> Result<Vec<LicenseInfo>> { |
||||
Ok(vec![LicenseInfo { |
||||
activation_id: "AID-1".into(), |
||||
name: "Windows Pro".into(), |
||||
description: "d".into(), |
||||
partial_product_key: Some("ABCDE".into()), |
||||
status: LicenseStatus::Notification, |
||||
license_family: None, |
||||
grace_minutes: None, |
||||
}]) |
||||
} |
||||
fn install_product_key(&self, _p: Product, _k: &str) -> Result<()> { |
||||
Ok(()) |
||||
} |
||||
fn uninstall_product_key(&self, _p: Product, _i: &str) -> Result<()> { |
||||
Ok(()) |
||||
} |
||||
fn set_kms_host(&self, _p: Product, host: &str, port: u16) -> Result<()> { |
||||
self.set_hosts.borrow_mut().push((host.into(), port)); |
||||
*self.current_host.borrow_mut() = Some(host.into()); |
||||
Ok(()) |
||||
} |
||||
fn clear_kms_host(&self, _p: Product) -> Result<()> { |
||||
Ok(()) |
||||
} |
||||
fn activate(&self, _p: Product, id: &str) -> Result<()> { |
||||
let ok = match &self.good_host { |
||||
None => true, |
||||
Some(good) => self.current_host.borrow().as_deref() == Some(good.as_str()), |
||||
}; |
||||
if ok { |
||||
self.activated.borrow_mut().push(id.into()); |
||||
Ok(()) |
||||
} else { |
||||
Err(Error::winapi_msg("no KMS")) |
||||
} |
||||
} |
||||
fn install_license(&self, _x: &std::path::Path) -> Result<()> { |
||||
Ok(()) |
||||
} |
||||
fn windows_build(&self) -> Result<u32> { |
||||
Ok(19045) |
||||
} |
||||
fn windows_edition(&self) -> Result<String> { |
||||
Ok("Professional".into()) |
||||
} |
||||
fn is_elevated(&self) -> bool { |
||||
true |
||||
} |
||||
} |
||||
|
||||
#[test] |
||||
fn activates_against_first_working_host() { |
||||
let spp = FakeSpp { |
||||
good_host: Some(DEFAULT_KMS_HOSTS[0].to_string()), |
||||
..Default::default() |
||||
}; |
||||
OnlineKms.run(&spp, Product::Windows).unwrap(); |
||||
assert_eq!(spp.activated.borrow().as_slice(), &["AID-1".to_string()]); |
||||
} |
||||
|
||||
#[test] |
||||
fn falls_through_to_a_later_host() { |
||||
let good = DEFAULT_KMS_HOSTS[2].to_string(); |
||||
let spp = FakeSpp { |
||||
good_host: Some(good.clone()), |
||||
..Default::default() |
||||
}; |
||||
OnlineKms.run(&spp, Product::Windows).unwrap(); |
||||
// It tried earlier hosts before the one that worked.
|
||||
let tried: Vec<String> = spp.set_hosts.borrow().iter().map(|(h, _)| h.clone()).collect(); |
||||
assert_eq!(tried[2], good); |
||||
} |
||||
|
||||
#[test] |
||||
fn explicit_host_that_never_activates_errors() { |
||||
let spp = FakeSpp { |
||||
good_host: Some("nonexistent".into()), |
||||
..Default::default() |
||||
}; |
||||
let err = OnlineKms |
||||
.run_with_host(&spp, Product::Windows, "1.2.3.4", 1688) |
||||
.unwrap_err(); |
||||
assert!(matches!(err, Error::WinApi { .. })); |
||||
} |
||||
} |
||||
@ -0,0 +1,90 @@
@@ -0,0 +1,90 @@
|
||||
//! `Check_Activation_Status.cmd` — read and present licensing state.
|
||||
//!
|
||||
//! This is the most self-contained module: it only *reads* the SPP, so the
|
||||
//! orchestration and formatting are fully portable and unit-tested here; only
|
||||
//! the underlying `installed_products` query is Windows-specific.
|
||||
|
||||
use crate::error::Result; |
||||
use crate::model::Product; |
||||
use crate::platform::{LicenseInfo, Spp}; |
||||
|
||||
/// A rendered status line for one product.
|
||||
#[derive(Debug, Clone)] |
||||
pub struct StatusLine { |
||||
pub product: Product, |
||||
pub name: String, |
||||
pub partial_key: String, |
||||
pub status: String, |
||||
pub activated: bool, |
||||
} |
||||
|
||||
/// Collect status for one product family from the SPP.
|
||||
pub fn collect(spp: &dyn Spp, product: Product) -> Result<Vec<StatusLine>> { |
||||
let products = spp.installed_products(product)?; |
||||
Ok(products.iter().map(|p| line_for(product, p)).collect()) |
||||
} |
||||
|
||||
fn line_for(product: Product, info: &LicenseInfo) -> StatusLine { |
||||
StatusLine { |
||||
product, |
||||
name: info.name.clone(), |
||||
partial_key: info |
||||
.partial_product_key |
||||
.clone() |
||||
.unwrap_or_else(|| "—".to_string()), |
||||
status: info.status.label().to_string(), |
||||
activated: info.is_activated(), |
||||
} |
||||
} |
||||
|
||||
/// Format collected lines the way MAS prints them: name, partial key, state.
|
||||
pub fn render(lines: &[StatusLine]) -> String { |
||||
if lines.is_empty() { |
||||
return "No licensed products found.\n".to_string(); |
||||
} |
||||
let mut out = String::new(); |
||||
for l in lines { |
||||
let mark = if l.activated { "[✓]" } else { "[ ]" }; |
||||
out.push_str(&format!( |
||||
"{mark} {} — {} (…{})\n", |
||||
l.product, l.name, l.partial_key |
||||
)); |
||||
out.push_str(&format!(" {}\n", l.status)); |
||||
} |
||||
out |
||||
} |
||||
|
||||
#[cfg(test)] |
||||
mod tests { |
||||
use super::*; |
||||
use crate::model::LicenseStatus; |
||||
|
||||
fn info(name: &str, key: Option<&str>, status: LicenseStatus) -> LicenseInfo { |
||||
LicenseInfo { |
||||
activation_id: "id".into(), |
||||
name: name.into(), |
||||
description: "desc".into(), |
||||
partial_product_key: key.map(str::to_string), |
||||
status, |
||||
license_family: None, |
||||
grace_minutes: None, |
||||
} |
||||
} |
||||
|
||||
#[test] |
||||
fn render_marks_activated_products() { |
||||
let lines = vec![ |
||||
line_for(Product::Windows, &info("Pro", Some("ABCDE"), LicenseStatus::Licensed)), |
||||
line_for(Product::Windows, &info("Core", None, LicenseStatus::Notification)), |
||||
]; |
||||
let out = render(&lines); |
||||
assert!(out.contains("[✓] Windows — Pro (…ABCDE)")); |
||||
assert!(out.contains("[ ] Windows — Core (…—)")); |
||||
assert!(out.contains("Notification")); |
||||
} |
||||
|
||||
#[test] |
||||
fn empty_products_render_message() { |
||||
assert_eq!(render(&[]), "No licensed products found.\n"); |
||||
} |
||||
} |
||||
@ -0,0 +1,41 @@
@@ -0,0 +1,41 @@
|
||||
//! `TSforge_Activation.cmd` — ticket injection into the SPP trusted store.
|
||||
//!
|
||||
//! This is the deepest module. TSforge writes activation tickets **directly into
|
||||
//! the SPP trusted store** (`data.dat`) rather than going through any activation
|
||||
//! server. Per analysis `port_risk`, the store is a byte-exact,
|
||||
//! reverse-engineered format: RSA/AES/HMAC-wrapped with CRC32 per block, a
|
||||
//! `VariableBag` layout, per-block + whole-file SHA-256, physical-store HMAC-SHA1
|
||||
//! (salted-SHA1 on Vista), and 4/8-byte alignment padding that **differs across
|
||||
//! Vista / 7 / 8 / 8.1 / Modern**. Any mismatch corrupts the store.
|
||||
//!
|
||||
//! A faithful port is essentially porting **LibTSforge** to Rust:
|
||||
//! * pure `libtsforge` crate (no_std-friendly, unit-testable on Linux/CI):
|
||||
//! the store (de)serialization, CRC/SHA/HMAC layers, key-blob builders, and
|
||||
//! the data tables (SKU→edition, MSI-office rows, retail→volume);
|
||||
//! * thin `#[cfg(windows)]` FFI: stop `sppsvc`, swap `data.dat`, restart.
|
||||
//!
|
||||
//! This is the highest-value, highest-effort slice and the natural next crate to
|
||||
//! extract. It is intentionally not stubbed as "works" — that would be a lie.
|
||||
|
||||
use crate::activation::Activator; |
||||
use crate::error::{Error, Result}; |
||||
use crate::model::{Method, Product}; |
||||
use crate::platform::Spp; |
||||
|
||||
pub struct TSforge; |
||||
|
||||
impl Activator for TSforge { |
||||
fn method(&self) -> Method { |
||||
Method::TSforge |
||||
} |
||||
|
||||
fn run(&self, spp: &dyn Spp, _product: Product) -> Result<()> { |
||||
if !spp.is_elevated() { |
||||
return Err(Error::NotElevated); |
||||
} |
||||
Err(Error::Unsupported { |
||||
what: "TSforge requires the SPP trusted-store codec (LibTSforge port — see module docs)" |
||||
.into(), |
||||
}) |
||||
} |
||||
} |
||||
@ -0,0 +1,151 @@
@@ -0,0 +1,151 @@
|
||||
//! Terminal menu rendering and selection.
|
||||
//!
|
||||
//! The scripts build their colored menus with hundreds of `echo` lines and
|
||||
//! parse the choice with `choice`/`set /p` plus a wall of `if "%o%"=="1"`.
|
||||
//! This is the portable, testable replacement: a menu is data, rendering is a
|
||||
//! pure function, and selection maps a keypress back to a typed value.
|
||||
|
||||
use std::fmt; |
||||
use std::io::{self, Write}; |
||||
|
||||
/// A selectable menu entry associating a hotkey with a payload `T`.
|
||||
pub struct MenuItem<T> { |
||||
pub key: char, |
||||
pub label: String, |
||||
/// Optional dim secondary line (method description, availability note).
|
||||
pub note: Option<String>, |
||||
pub value: T, |
||||
} |
||||
|
||||
impl<T> MenuItem<T> { |
||||
pub fn new(key: char, label: impl Into<String>, value: T) -> Self { |
||||
MenuItem { |
||||
key, |
||||
label: label.into(), |
||||
note: None, |
||||
value, |
||||
} |
||||
} |
||||
|
||||
pub fn with_note(mut self, note: impl Into<String>) -> Self { |
||||
self.note = Some(note.into()); |
||||
self |
||||
} |
||||
} |
||||
|
||||
/// A titled list of items plus an implicit `0` = back/exit.
|
||||
pub struct Menu<T> { |
||||
pub title: String, |
||||
pub items: Vec<MenuItem<T>>, |
||||
} |
||||
|
||||
impl<T> Menu<T> { |
||||
pub fn new(title: impl Into<String>) -> Self { |
||||
Menu { |
||||
title: title.into(), |
||||
items: Vec::new(), |
||||
} |
||||
} |
||||
|
||||
pub fn item(mut self, item: MenuItem<T>) -> Self { |
||||
self.items.push(item); |
||||
self |
||||
} |
||||
|
||||
/// Render the menu to a string. Pure — no I/O — so it can be asserted in
|
||||
/// tests. `0) Exit` is always appended, matching MAS's convention.
|
||||
pub fn render(&self) -> String { |
||||
let mut out = String::new(); |
||||
out.push_str(&self.title); |
||||
out.push('\n'); |
||||
out.push_str(&"=".repeat(self.title.len())); |
||||
out.push('\n'); |
||||
for it in &self.items { |
||||
out.push_str(&format!(" {}) {}\n", it.key, it.label)); |
||||
if let Some(note) = &it.note { |
||||
out.push_str(&format!(" {note}\n")); |
||||
} |
||||
} |
||||
out.push_str(" 0) Back / Exit\n"); |
||||
out |
||||
} |
||||
|
||||
/// Resolve a keypress to the item's value. `'0'` yields `None` (back/exit).
|
||||
pub fn resolve(&self, key: char) -> Selection<&T> { |
||||
if key == '0' { |
||||
return Selection::Back; |
||||
} |
||||
match self |
||||
.items |
||||
.iter() |
||||
.find(|it| it.key.eq_ignore_ascii_case(&key)) |
||||
{ |
||||
Some(it) => Selection::Value(&it.value), |
||||
None => Selection::Invalid, |
||||
} |
||||
} |
||||
} |
||||
|
||||
/// Result of matching a keypress against a [`Menu`].
|
||||
#[derive(Debug, PartialEq, Eq)] |
||||
pub enum Selection<T> { |
||||
Value(T), |
||||
Back, |
||||
Invalid, |
||||
} |
||||
|
||||
/// Prompt on the given writer, read one line from the reader, and return the
|
||||
/// trimmed first character. Split from [`Menu`] so tests drive it with cursors.
|
||||
pub fn prompt_choice<R: io::BufRead, W: Write>( |
||||
mut reader: R, |
||||
mut writer: W, |
||||
prompt: &str, |
||||
) -> io::Result<char> { |
||||
write!(writer, "{prompt}")?; |
||||
writer.flush()?; |
||||
let mut line = String::new(); |
||||
reader.read_line(&mut line)?; |
||||
Ok(line.trim().chars().next().unwrap_or('\0')) |
||||
} |
||||
|
||||
impl<T> fmt::Display for Menu<T> { |
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||
f.write_str(&self.render()) |
||||
} |
||||
} |
||||
|
||||
#[cfg(test)] |
||||
mod tests { |
||||
use super::*; |
||||
|
||||
fn sample() -> Menu<u8> { |
||||
Menu::new("Activate") |
||||
.item(MenuItem::new('1', "HWID", 1u8).with_note("permanent")) |
||||
.item(MenuItem::new('2', "Online KMS", 2u8)) |
||||
} |
||||
|
||||
#[test] |
||||
fn render_lists_items_and_exit() { |
||||
let r = sample().render(); |
||||
assert!(r.contains("1) HWID")); |
||||
assert!(r.contains("permanent")); |
||||
assert!(r.contains("0) Back / Exit")); |
||||
} |
||||
|
||||
#[test] |
||||
fn resolve_maps_keys_and_back() { |
||||
let m = sample(); |
||||
assert_eq!(m.resolve('1'), Selection::Value(&1u8)); |
||||
assert_eq!(m.resolve('0'), Selection::Back); |
||||
assert_eq!(m.resolve('9'), Selection::Invalid); |
||||
} |
||||
|
||||
#[test] |
||||
fn prompt_reads_first_char() { |
||||
let input = b"2\n"; |
||||
let mut out = Vec::new(); |
||||
let c = prompt_choice(&input[..], &mut out, "> ").unwrap(); |
||||
assert_eq!(c, '2'); |
||||
assert_eq!(out, b"> "); |
||||
} |
||||
} |
||||
@ -0,0 +1,161 @@
@@ -0,0 +1,161 @@
|
||||
//! Interactive front-end — the MAS main menu, ported.
|
||||
|
||||
pub mod menu; |
||||
|
||||
use crate::activation::{activator_for, status}; |
||||
use crate::error::Result; |
||||
use crate::model::{Method, Product}; |
||||
use crate::platform::Spp; |
||||
use menu::{Menu, MenuItem, Selection}; |
||||
use std::io::{self, BufRead, Write}; |
||||
|
||||
/// One top-level choice from the MAS main menu.
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq)] |
||||
pub enum MainChoice { |
||||
Activate(Method, Product), |
||||
CheckStatus, |
||||
} |
||||
|
||||
/// Build the main menu, hiding methods that aren't available on this machine
|
||||
/// (mirrors MAS graying/omitting options by build & product).
|
||||
pub fn main_menu(windows_build: u32) -> Menu<MainChoice> { |
||||
let mut m = Menu::new("Microsoft Activation Scripts (Rust)"); |
||||
let mut key = b'1'; |
||||
let mut add = |m: &mut Menu<MainChoice>, label: String, note: &str, choice: MainChoice| { |
||||
m.items.push( |
||||
MenuItem::new(key as char, label, choice).with_note(note.to_string()), |
||||
); |
||||
key += 1; |
||||
}; |
||||
|
||||
for method in Method::ALL { |
||||
for product in [Product::Windows, Product::Office] { |
||||
if method.targets(product) && method.available_for(product, windows_build) { |
||||
add( |
||||
&mut m, |
||||
format!("{} — {}", method, product), |
||||
method.description(), |
||||
MainChoice::Activate(method, product), |
||||
); |
||||
} |
||||
} |
||||
} |
||||
add( |
||||
&mut m, |
||||
"Check Activation Status".into(), |
||||
"Read-only report for Windows and Office", |
||||
MainChoice::CheckStatus, |
||||
); |
||||
m |
||||
} |
||||
|
||||
/// Run one round of the menu against the given I/O and backend. Returns
|
||||
/// `Ok(true)` to keep looping, `Ok(false)` to exit.
|
||||
pub fn step<R: BufRead, W: Write>( |
||||
reader: R, |
||||
mut writer: W, |
||||
spp: &dyn Spp, |
||||
) -> Result<bool> { |
||||
let build = spp.windows_build().unwrap_or(0); |
||||
let m = main_menu(build); |
||||
write!(writer, "{}", m.render())?; |
||||
let key = menu::prompt_choice(reader, &mut writer, "\nSelect: ")?; |
||||
|
||||
match m.resolve(key) { |
||||
Selection::Back => Ok(false), |
||||
Selection::Invalid => { |
||||
writeln!(writer, "Invalid choice.")?; |
||||
Ok(true) |
||||
} |
||||
Selection::Value(&choice) => { |
||||
run_choice(&mut writer, spp, choice)?; |
||||
Ok(true) |
||||
} |
||||
} |
||||
} |
||||
|
||||
fn run_choice<W: Write>(writer: &mut W, spp: &dyn Spp, choice: MainChoice) -> Result<()> { |
||||
match choice { |
||||
MainChoice::CheckStatus => { |
||||
for product in [Product::Windows, Product::Office] { |
||||
let lines = status::collect(spp, product)?; |
||||
write!(writer, "{}", status::render(&lines))?; |
||||
} |
||||
} |
||||
MainChoice::Activate(method, product) => { |
||||
writeln!(writer, "Running {method} for {product}…")?; |
||||
match activator_for(method).run(spp, product) { |
||||
Ok(()) => writeln!(writer, " Success.")?, |
||||
Err(e) => writeln!(writer, " {e}")?, |
||||
} |
||||
} |
||||
} |
||||
Ok(()) |
||||
} |
||||
|
||||
/// Blocking interactive loop over stdin/stdout.
|
||||
pub fn run_interactive(spp: &dyn Spp) -> Result<()> { |
||||
let stdin = io::stdin(); |
||||
let stdout = io::stdout(); |
||||
loop { |
||||
let keep = step(stdin.lock(), stdout.lock(), spp)?; |
||||
if !keep { |
||||
return Ok(()); |
||||
} |
||||
} |
||||
} |
||||
|
||||
#[cfg(test)] |
||||
mod tests { |
||||
use super::*; |
||||
use crate::platform::stub::StubSpp; |
||||
|
||||
#[test] |
||||
fn menu_hides_win10_only_methods_on_old_builds() { |
||||
let old = main_menu(7601); // Windows 7
|
||||
assert!(!old |
||||
.items |
||||
.iter() |
||||
.any(|it| matches!(it.value, MainChoice::Activate(Method::Hwid, _)))); |
||||
|
||||
let new = main_menu(19045); // Windows 10
|
||||
assert!(new |
||||
.items |
||||
.iter() |
||||
.any(|it| matches!(it.value, MainChoice::Activate(Method::Hwid, _)))); |
||||
} |
||||
|
||||
#[test] |
||||
fn check_status_is_always_offered() { |
||||
let m = main_menu(0); |
||||
assert!(m |
||||
.items |
||||
.iter() |
||||
.any(|it| it.value == MainChoice::CheckStatus)); |
||||
} |
||||
|
||||
#[test] |
||||
fn step_check_status_runs_and_loops() { |
||||
// Choose the "Check Activation Status" entry, then verify it loops.
|
||||
let m = main_menu(0); |
||||
let status_key = m |
||||
.items |
||||
.iter() |
||||
.find(|it| it.value == MainChoice::CheckStatus) |
||||
.unwrap() |
||||
.key; |
||||
let input = format!("{status_key}\n"); |
||||
let mut out = Vec::new(); |
||||
let keep = step(input.as_bytes(), &mut out, &StubSpp).unwrap(); |
||||
assert!(keep); |
||||
let text = String::from_utf8(out).unwrap(); |
||||
assert!(text.contains("No licensed products found.")); |
||||
} |
||||
|
||||
#[test] |
||||
fn step_zero_exits() { |
||||
let mut out = Vec::new(); |
||||
let keep = step(&b"0\n"[..], &mut out, &StubSpp).unwrap(); |
||||
assert!(!keep); |
||||
} |
||||
} |
||||
@ -0,0 +1,30 @@
@@ -0,0 +1,30 @@
|
||||
//! Default KMS host list for Online KMS activation.
|
||||
//!
|
||||
//! MAS lets the user pass a server with `/K-Server-…` and otherwise tries a
|
||||
//! built-in list of public KMS hosts. Ported as data, not code.
|
||||
//!
|
||||
//! ponytail: static default list — the real tuning knob. Public KMS hosts come
|
||||
//! and go; expose `--kms-host` (already supported via [`OnlineKms`]) and refresh
|
||||
//! this list rather than hard-depending on any single entry.
|
||||
|
||||
/// Standard KMS port.
|
||||
pub const DEFAULT_KMS_PORT: u16 = 1688; |
||||
|
||||
/// Fallback public KMS hosts, tried in order until one activates.
|
||||
pub const DEFAULT_KMS_HOSTS: &[&str] = &[ |
||||
"kms.digiboy.ir", |
||||
"kms8.msguides.com", |
||||
"kms.03k.org", |
||||
"kms.chinancce.com", |
||||
]; |
||||
|
||||
#[cfg(test)] |
||||
mod tests { |
||||
use super::*; |
||||
|
||||
#[test] |
||||
fn host_list_is_nonempty_and_port_is_standard() { |
||||
assert!(!DEFAULT_KMS_HOSTS.is_empty()); |
||||
assert_eq!(DEFAULT_KMS_PORT, 1688); |
||||
} |
||||
} |
||||
@ -0,0 +1,3 @@
@@ -0,0 +1,3 @@
|
||||
//! Ported data tables (verbatim from the scripts where it matters).
|
||||
|
||||
pub mod kms_hosts; |
||||
@ -0,0 +1,138 @@
@@ -0,0 +1,138 @@
|
||||
//! Error handling for the MAS port.
|
||||
//!
|
||||
//! The original scripts signal failure through `errorlevel` codes, colored
|
||||
//! console messages and `goto :dk` jumps. Rust replaces all of that with a
|
||||
//! single typed error carried by [`Result`]. Each variant maps to a concrete
|
||||
//! failure mode from the batch scripts so callers can react precisely instead
|
||||
//! of parsing text.
|
||||
//!
|
||||
//! Kept dependency-free on purpose (no `thiserror`) so the portable core builds
|
||||
//! and tests offline on any platform — see the crate root notes.
|
||||
|
||||
use std::fmt; |
||||
|
||||
/// Crate-wide result type.
|
||||
pub type Result<T> = std::result::Result<T, Error>; |
||||
|
||||
/// Everything that can go wrong while inspecting or changing activation state.
|
||||
#[derive(Debug)] |
||||
#[non_exhaustive] |
||||
pub enum Error { |
||||
/// A privileged operation was attempted on a non-Windows build (the stub
|
||||
/// Software Protection Platform backend). The real backend requires the
|
||||
/// `winapi` feature on a Windows target.
|
||||
UnsupportedPlatform { |
||||
/// The operation that is unavailable, e.g. `"install product key"`.
|
||||
operation: &'static str, |
||||
}, |
||||
|
||||
/// The process is not running elevated. Almost every MAS operation edits
|
||||
/// HKLM / the SPP and needs administrator rights (the scripts self-elevate
|
||||
/// via a mshta/PowerShell relaunch).
|
||||
NotElevated, |
||||
|
||||
/// A Windows API / WMI call failed. `hresult` is the raw `HRESULT` where
|
||||
/// available (mirrors the scripts surfacing SPP error codes to the user).
|
||||
WinApi { |
||||
context: String, |
||||
hresult: Option<i32>, |
||||
}, |
||||
|
||||
/// A required external tool (`ClipUp.exe`, `cscript`, `slmgr.vbs`, `sc.exe`)
|
||||
/// could not be found or exited non-zero.
|
||||
ExternalTool { |
||||
tool: String, |
||||
detail: String, |
||||
}, |
||||
|
||||
/// The user asked for a product/edition/method that isn't valid for this
|
||||
/// machine (e.g. HWID on a build older than Windows 10, or an unknown SKU).
|
||||
Unsupported { |
||||
what: String, |
||||
}, |
||||
|
||||
/// The user aborted an interactive prompt.
|
||||
Cancelled, |
||||
|
||||
/// Wrapper for `std::io` failures (file/registry-file/process I/O).
|
||||
Io(std::io::Error), |
||||
} |
||||
|
||||
impl Error { |
||||
/// Convenience constructor for a WMI/Win32 failure with an `HRESULT`.
|
||||
pub fn winapi(context: impl Into<String>, hresult: i32) -> Self { |
||||
Error::WinApi { |
||||
context: context.into(), |
||||
hresult: Some(hresult), |
||||
} |
||||
} |
||||
|
||||
/// Convenience constructor for a WMI/Win32 failure without a code.
|
||||
pub fn winapi_msg(context: impl Into<String>) -> Self { |
||||
Error::WinApi { |
||||
context: context.into(), |
||||
hresult: None, |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl fmt::Display for Error { |
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||
match self { |
||||
Error::UnsupportedPlatform { operation } => write!( |
||||
f, |
||||
"cannot {operation}: this build has no Windows Software Protection \ |
||||
Platform backend (rebuild on Windows with --features winapi)" |
||||
), |
||||
Error::NotElevated => { |
||||
write!(f, "administrator privileges are required for this operation") |
||||
} |
||||
Error::WinApi { context, hresult } => match hresult { |
||||
Some(hr) => write!(f, "{context} (HRESULT 0x{:08X})", *hr as u32), |
||||
None => write!(f, "{context}"), |
||||
}, |
||||
Error::ExternalTool { tool, detail } => { |
||||
write!(f, "external tool `{tool}` failed: {detail}") |
||||
} |
||||
Error::Unsupported { what } => write!(f, "unsupported: {what}"), |
||||
Error::Cancelled => write!(f, "cancelled by user"), |
||||
Error::Io(e) => write!(f, "I/O error: {e}"), |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl std::error::Error for Error { |
||||
fn source(&self) -> Option<&(dyn std::error::Error + 'static)> { |
||||
match self { |
||||
Error::Io(e) => Some(e), |
||||
_ => None, |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl From<std::io::Error> for Error { |
||||
fn from(e: std::io::Error) -> Self { |
||||
Error::Io(e) |
||||
} |
||||
} |
||||
|
||||
#[cfg(test)] |
||||
mod tests { |
||||
use super::*; |
||||
|
||||
#[test] |
||||
fn hresult_is_hex_formatted() { |
||||
let e = Error::winapi("activate Windows", 0xC004F074u32 as i32); |
||||
assert_eq!( |
||||
e.to_string(), |
||||
"activate Windows (HRESULT 0xC004F074)" |
||||
); |
||||
} |
||||
|
||||
#[test] |
||||
fn io_error_chains_source() { |
||||
use std::error::Error as _; |
||||
let e = Error::from(std::io::Error::new(std::io::ErrorKind::NotFound, "x")); |
||||
assert!(e.source().is_some()); |
||||
} |
||||
} |
||||
@ -0,0 +1,34 @@
@@ -0,0 +1,34 @@
|
||||
//! Microsoft Activation Scripts, ported to Rust.
|
||||
//!
|
||||
//! A faithful, idiomatic re-architecture of the MAS batch scripts. The design
|
||||
//! isolates every Windows-specific effect behind the [`platform::Spp`] trait so
|
||||
//! the portable core — the menu, the domain model, the data tables and the
|
||||
//! activation orchestration — compiles and unit-tests on any OS, while the real
|
||||
//! Software Protection Platform work lives in a `--features winapi` Windows
|
||||
//! backend.
|
||||
//!
|
||||
//! ## Module map (→ original scripts)
|
||||
//! * [`cli`] — the interactive menu (`MAS_AIO.cmd` front-end).
|
||||
//! * [`model`] — products, methods, license status (the magic GUIDs/strings).
|
||||
//! * [`platform`] — the SPP boundary + stub / Windows backends.
|
||||
//! * [`activation::status`] — `Check_Activation_Status.cmd`.
|
||||
//! * [`activation::online_kms`] — `Online_KMS_Activation.cmd`.
|
||||
//! * [`activation::hwid`] / [`activation::kms38`] — `HWID_Activation.cmd`.
|
||||
//! * [`activation::ohook`] — `Ohook_Activation_AIO.cmd`.
|
||||
//! * [`activation::tsforge`] — `TSforge_Activation.cmd`.
|
||||
//! * [`data`] — ported data tables.
|
||||
|
||||
pub mod activation; |
||||
pub mod cli; |
||||
pub mod data; |
||||
pub mod error; |
||||
pub mod model; |
||||
pub mod platform; |
||||
|
||||
pub use error::{Error, Result}; |
||||
|
||||
/// Entry point used by the binary: build the backend and run the menu.
|
||||
pub fn run() -> Result<()> { |
||||
let spp = platform::backend(); |
||||
cli::run_interactive(spp.as_ref()) |
||||
} |
||||
@ -0,0 +1,11 @@
@@ -0,0 +1,11 @@
|
||||
//! Thin entry point — all logic lives in the library so it can be unit-tested.
|
||||
|
||||
fn main() -> std::process::ExitCode { |
||||
match mas::run() { |
||||
Ok(()) => std::process::ExitCode::SUCCESS, |
||||
Err(e) => { |
||||
eprintln!("mas: {e}"); |
||||
std::process::ExitCode::FAILURE |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,253 @@
@@ -0,0 +1,253 @@
|
||||
//! Core domain model.
|
||||
//!
|
||||
//! The batch scripts encode everything as bare strings and magic GUIDs sprayed
|
||||
//! across thousands of lines. Here that becomes a small set of enums with the
|
||||
//! constants attached to them, so the compiler enforces which activation method
|
||||
//! is valid for which product and there is exactly one source of truth for the
|
||||
//! Software Protection Platform application IDs.
|
||||
|
||||
use std::fmt; |
||||
|
||||
/// A licensable Microsoft product family.
|
||||
///
|
||||
/// Windows and Office are handled by two different Software Protection Platform
|
||||
/// providers, each identified by its own `ApplicationID` GUID and exposed
|
||||
/// through different WMI classes. Encoding that here removes the `if office`
|
||||
/// branching that is duplicated all over the scripts.
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] |
||||
pub enum Product { |
||||
Windows, |
||||
Office, |
||||
} |
||||
|
||||
impl Product { |
||||
/// The SPP `ApplicationID` GUID used in every WMI query
|
||||
/// (`WHERE ApplicationID='…'`).
|
||||
pub const fn application_id(self) -> &'static str { |
||||
match self { |
||||
Product::Windows => "55c92734-d682-4d71-983e-d6ec3f16059f", |
||||
Product::Office => "0ff1ce15-a989-479d-af46-f275c6370663", |
||||
} |
||||
} |
||||
|
||||
/// WMI class exposing the per-product licensing state
|
||||
/// (`Get PartialProductKey`, `LicenseStatus`, …).
|
||||
pub const fn product_class(self) -> &'static str { |
||||
match self { |
||||
Product::Windows => "SoftwareLicensingProduct", |
||||
// Office ≤2016 with the C2R/MSI SPP uses the OSPP class; MAS falls
|
||||
// back to SoftwareLicensingProduct on newer Office builds.
|
||||
Product::Office => "OfficeSoftwareProtectionProduct", |
||||
} |
||||
} |
||||
|
||||
/// WMI class exposing the service-wide operations
|
||||
/// (`InstallProductKey`, `SetKeyManagementServiceMachine`, …).
|
||||
pub const fn service_class(self) -> &'static str { |
||||
match self { |
||||
Product::Windows => "SoftwareLicensingService", |
||||
Product::Office => "OfficeSoftwareProtectionService", |
||||
} |
||||
} |
||||
|
||||
pub const fn label(self) -> &'static str { |
||||
match self { |
||||
Product::Windows => "Windows", |
||||
Product::Office => "Office", |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl fmt::Display for Product { |
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||
f.write_str(self.label()) |
||||
} |
||||
} |
||||
|
||||
/// An activation method offered by MAS.
|
||||
///
|
||||
/// Each variant carries the rules that the scripts check inline before letting
|
||||
/// you pick it: which product it targets and the minimum Windows build it needs.
|
||||
/// [`Method::available_for`] centralises those guards.
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] |
||||
pub enum Method { |
||||
/// Hardware-ID / digital-license activation (permanent, tied to the device).
|
||||
Hwid, |
||||
/// KMS38 — extends the KMS activation expiry to the year 2038 via ClipUp.
|
||||
Kms38, |
||||
/// Classic KMS client: install GVLK, point at a KMS host, activate.
|
||||
OnlineKms, |
||||
/// Office activation by hooking the SPP (`sppc`/`osppc`) — no key install.
|
||||
Ohook, |
||||
/// TSforge — writes activation tickets straight into the SPP token store.
|
||||
TSforge, |
||||
} |
||||
|
||||
impl Method { |
||||
pub const ALL: [Method; 5] = [ |
||||
Method::Hwid, |
||||
Method::Kms38, |
||||
Method::OnlineKms, |
||||
Method::Ohook, |
||||
Method::TSforge, |
||||
]; |
||||
|
||||
pub const fn label(self) -> &'static str { |
||||
match self { |
||||
Method::Hwid => "HWID", |
||||
Method::Kms38 => "KMS38", |
||||
Method::OnlineKms => "Online KMS", |
||||
Method::Ohook => "Ohook", |
||||
Method::TSforge => "TSforge", |
||||
} |
||||
} |
||||
|
||||
/// One-line description shown in the menu.
|
||||
pub const fn description(self) -> &'static str { |
||||
match self { |
||||
Method::Hwid => "Permanent activation for Windows 10/11 (digital license)", |
||||
Method::Kms38 => "Activate Windows until 2038-01-19 via KMS", |
||||
Method::OnlineKms => "180-day renewable KMS activation for Windows and Office", |
||||
Method::Ohook => "Permanent Office activation (no key, hooks the SPP)", |
||||
Method::TSforge => "Ticket-based activation for Windows, Office and ESU", |
||||
} |
||||
} |
||||
|
||||
/// Whether this method can act on the given product at all.
|
||||
pub const fn targets(self, product: Product) -> bool { |
||||
match self { |
||||
// Windows-only digital-license / KMS-expiry tricks.
|
||||
Method::Hwid | Method::Kms38 => matches!(product, Product::Windows), |
||||
// Office-only SPP hook.
|
||||
Method::Ohook => matches!(product, Product::Office), |
||||
// These handle both families.
|
||||
Method::OnlineKms | Method::TSforge => true, |
||||
} |
||||
} |
||||
|
||||
/// Minimum Windows build required, if any. HWID and KMS38 rely on
|
||||
/// `ClipUp.exe`/digital-license infrastructure introduced in Windows 10
|
||||
/// (build 10240).
|
||||
pub const fn min_windows_build(self) -> Option<u32> { |
||||
match self { |
||||
Method::Hwid | Method::Kms38 => Some(10240), |
||||
_ => None, |
||||
} |
||||
} |
||||
|
||||
/// Full guard combining product targeting and the OS build check — the
|
||||
/// scripts scatter this logic; here it is one call.
|
||||
pub fn available_for(self, product: Product, windows_build: u32) -> bool { |
||||
if !self.targets(product) { |
||||
return false; |
||||
} |
||||
match self.min_windows_build() { |
||||
Some(min) => windows_build >= min, |
||||
None => true, |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl fmt::Display for Method { |
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||
f.write_str(self.label()) |
||||
} |
||||
} |
||||
|
||||
/// Licensing state of a single product, as read from the SPP.
|
||||
///
|
||||
/// Mirrors the WMI `LicenseStatus` enumeration used throughout
|
||||
/// `Check_Activation_Status.cmd`.
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq)] |
||||
pub enum LicenseStatus { |
||||
Unlicensed, |
||||
Licensed, |
||||
OobGrace, |
||||
OotGrace, |
||||
NonGenuineGrace, |
||||
Notification, |
||||
ExtendedGrace, |
||||
} |
||||
|
||||
impl LicenseStatus { |
||||
/// Map the raw WMI integer to the enum. Values are Microsoft's, quoted
|
||||
/// verbatim in the scripts.
|
||||
pub const fn from_wmi(value: u32) -> Option<Self> { |
||||
Some(match value { |
||||
0 => LicenseStatus::Unlicensed, |
||||
1 => LicenseStatus::Licensed, |
||||
2 => LicenseStatus::OobGrace, |
||||
3 => LicenseStatus::OotGrace, |
||||
4 => LicenseStatus::NonGenuineGrace, |
||||
5 => LicenseStatus::Notification, |
||||
6 => LicenseStatus::ExtendedGrace, |
||||
_ => return None, |
||||
}) |
||||
} |
||||
|
||||
pub const fn label(self) -> &'static str { |
||||
match self { |
||||
LicenseStatus::Unlicensed => "Unlicensed", |
||||
LicenseStatus::Licensed => "Licensed", |
||||
LicenseStatus::OobGrace => "Initial grace period", |
||||
LicenseStatus::OotGrace => "Additional grace period (KMS expired)", |
||||
LicenseStatus::NonGenuineGrace => "Non-genuine grace period", |
||||
LicenseStatus::Notification => "Notification", |
||||
LicenseStatus::ExtendedGrace => "Extended grace period", |
||||
} |
||||
} |
||||
|
||||
/// Is the product currently activated?
|
||||
pub const fn is_activated(self) -> bool { |
||||
matches!(self, LicenseStatus::Licensed) |
||||
} |
||||
} |
||||
|
||||
impl fmt::Display for LicenseStatus { |
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
||||
f.write_str(self.label()) |
||||
} |
||||
} |
||||
|
||||
#[cfg(test)] |
||||
mod tests { |
||||
use super::*; |
||||
|
||||
#[test] |
||||
fn hwid_is_windows_only_and_win10_plus() { |
||||
assert!(Method::Hwid.available_for(Product::Windows, 19045)); |
||||
assert!(!Method::Hwid.available_for(Product::Windows, 9600)); // Win 8.1
|
||||
assert!(!Method::Hwid.available_for(Product::Office, 22631)); // wrong product
|
||||
} |
||||
|
||||
#[test] |
||||
fn ohook_is_office_only() { |
||||
assert!(Method::Ohook.targets(Product::Office)); |
||||
assert!(!Method::Ohook.targets(Product::Windows)); |
||||
} |
||||
|
||||
#[test] |
||||
fn online_kms_targets_both() { |
||||
assert!(Method::OnlineKms.available_for(Product::Windows, 7601)); |
||||
assert!(Method::OnlineKms.available_for(Product::Office, 7601)); |
||||
} |
||||
|
||||
#[test] |
||||
fn application_ids_are_the_real_spp_guids() { |
||||
assert_eq!( |
||||
Product::Windows.application_id(), |
||||
"55c92734-d682-4d71-983e-d6ec3f16059f" |
||||
); |
||||
assert_eq!( |
||||
Product::Office.application_id(), |
||||
"0ff1ce15-a989-479d-af46-f275c6370663" |
||||
); |
||||
} |
||||
|
||||
#[test] |
||||
fn license_status_round_trips_known_values() { |
||||
assert_eq!(LicenseStatus::from_wmi(1), Some(LicenseStatus::Licensed)); |
||||
assert!(LicenseStatus::from_wmi(1).unwrap().is_activated()); |
||||
assert_eq!(LicenseStatus::from_wmi(99), None); |
||||
} |
||||
} |
||||
@ -0,0 +1,98 @@
@@ -0,0 +1,98 @@
|
||||
//! The Software Protection Platform (SPP) boundary.
|
||||
//!
|
||||
//! This is the single seam between portable logic and Windows internals. Every
|
||||
//! privileged thing the scripts do — WMI calls against `SoftwareLicensingService`,
|
||||
//! registry edits, spawning `ClipUp.exe`/`slmgr.vbs` — is expressed here as a
|
||||
//! safe trait. The rest of the crate programs against [`Spp`] and never touches
|
||||
//! `unsafe` or the `windows` crate directly (constraint 3: encapsulate FFI behind
|
||||
//! a sound safe API).
|
||||
//!
|
||||
//! Two implementations exist, chosen at compile time:
|
||||
//!
|
||||
//! * [`windows_backend`] (`--features winapi`, Windows target) — the real thing.
|
||||
//! * [`stub`] (everywhere else) — read-only calls return empty/best-effort data
|
||||
//! and every privileged call returns [`Error::UnsupportedPlatform`], so the
|
||||
//! portable core builds and tests on Linux.
|
||||
|
||||
use crate::error::Result; |
||||
use crate::model::{LicenseStatus, Product}; |
||||
|
||||
#[cfg(all(windows, feature = "winapi"))] |
||||
pub mod windows_backend; |
||||
#[cfg(not(all(windows, feature = "winapi")))] |
||||
pub mod stub; |
||||
|
||||
/// One product entry as reported by the SPP (`SoftwareLicensingProduct` row).
|
||||
#[derive(Debug, Clone)] |
||||
pub struct LicenseInfo { |
||||
/// Activation ID (the WMI object `ID`), used to target `Activate()` etc.
|
||||
pub activation_id: String, |
||||
/// Human-readable product name, e.g. `Windows(R), Professional edition`.
|
||||
pub name: String, |
||||
/// SPP `Description`, e.g. `Windows(R) Operating System, RETAIL channel`.
|
||||
pub description: String, |
||||
/// Last five characters of the installed key, or `None` if no key.
|
||||
pub partial_product_key: Option<String>, |
||||
/// Current license state.
|
||||
pub status: LicenseStatus, |
||||
/// `LicenseFamily`, used to distinguish Office apps (Project/Visio/…).
|
||||
pub license_family: Option<String>, |
||||
/// Seconds of grace remaining (KMS / trial), if applicable.
|
||||
pub grace_minutes: Option<u32>, |
||||
} |
||||
|
||||
impl LicenseInfo { |
||||
pub fn is_activated(&self) -> bool { |
||||
self.status.is_activated() |
||||
} |
||||
} |
||||
|
||||
/// Safe abstraction over the Windows Software Protection Platform.
|
||||
///
|
||||
/// Method names mirror the underlying WMI methods so the mapping back to the
|
||||
/// original scripts stays obvious.
|
||||
pub trait Spp { |
||||
/// Query all licensable products of a family
|
||||
/// (`SELECT … FROM SoftwareLicensingProduct WHERE ApplicationID='…' AND
|
||||
/// PartialProductKey IS NOT NULL`).
|
||||
fn installed_products(&self, product: Product) -> Result<Vec<LicenseInfo>>; |
||||
|
||||
/// `SoftwareLicensingService.InstallProductKey(key)` — install a GVLK/retail key.
|
||||
fn install_product_key(&self, product: Product, key: &str) -> Result<()>; |
||||
|
||||
/// `SoftwareLicensingProduct.UninstallProductKey()` for the given activation ID.
|
||||
fn uninstall_product_key(&self, product: Product, activation_id: &str) -> Result<()>; |
||||
|
||||
/// `SetKeyManagementServiceMachine` + `SetKeyManagementServicePort`.
|
||||
fn set_kms_host(&self, product: Product, host: &str, port: u16) -> Result<()>; |
||||
|
||||
/// `ClearKeyManagementServiceMachine` — revert to auto-discovery.
|
||||
fn clear_kms_host(&self, product: Product) -> Result<()>; |
||||
|
||||
/// `SoftwareLicensingProduct.Activate()` for the given activation ID.
|
||||
fn activate(&self, product: Product, activation_id: &str) -> Result<()>; |
||||
|
||||
/// `SoftwareLicensingService.InstallLicense(<xml>)` — install a license/token file.
|
||||
fn install_license(&self, xml_path: &std::path::Path) -> Result<()>; |
||||
|
||||
/// OS build number (e.g. `19045`). Drives method eligibility.
|
||||
fn windows_build(&self) -> Result<u32>; |
||||
|
||||
/// Current Windows edition ID (e.g. `Professional`).
|
||||
fn windows_edition(&self) -> Result<String>; |
||||
|
||||
/// Whether the current process is elevated (administrator).
|
||||
fn is_elevated(&self) -> bool; |
||||
} |
||||
|
||||
/// Construct the SPP backend appropriate for this build.
|
||||
pub fn backend() -> Box<dyn Spp> { |
||||
#[cfg(all(windows, feature = "winapi"))] |
||||
{ |
||||
Box::new(windows_backend::WindowsSpp::new()) |
||||
} |
||||
#[cfg(not(all(windows, feature = "winapi")))] |
||||
{ |
||||
Box::new(stub::StubSpp) |
||||
} |
||||
} |
||||
@ -0,0 +1,63 @@
@@ -0,0 +1,63 @@
|
||||
//! Non-Windows stub backend.
|
||||
//!
|
||||
//! Lets the entire crate compile and unit-test off Windows. Read-only queries
|
||||
//! return empty/neutral data; anything that would mutate activation state fails
|
||||
//! loudly with [`Error::UnsupportedPlatform`] instead of silently pretending to
|
||||
//! work.
|
||||
|
||||
use crate::error::{Error, Result}; |
||||
use crate::model::Product; |
||||
use crate::platform::{LicenseInfo, Spp}; |
||||
use std::path::Path; |
||||
|
||||
#[derive(Default)] |
||||
pub struct StubSpp; |
||||
|
||||
impl StubSpp { |
||||
fn deny(op: &'static str) -> Error { |
||||
Error::UnsupportedPlatform { operation: op } |
||||
} |
||||
} |
||||
|
||||
impl Spp for StubSpp { |
||||
fn installed_products(&self, _product: Product) -> Result<Vec<LicenseInfo>> { |
||||
Ok(Vec::new()) |
||||
} |
||||
|
||||
fn install_product_key(&self, _p: Product, _key: &str) -> Result<()> { |
||||
Err(Self::deny("install product key")) |
||||
} |
||||
|
||||
fn uninstall_product_key(&self, _p: Product, _id: &str) -> Result<()> { |
||||
Err(Self::deny("uninstall product key")) |
||||
} |
||||
|
||||
fn set_kms_host(&self, _p: Product, _host: &str, _port: u16) -> Result<()> { |
||||
Err(Self::deny("set KMS host")) |
||||
} |
||||
|
||||
fn clear_kms_host(&self, _p: Product) -> Result<()> { |
||||
Err(Self::deny("clear KMS host")) |
||||
} |
||||
|
||||
fn activate(&self, _p: Product, _id: &str) -> Result<()> { |
||||
Err(Self::deny("activate product")) |
||||
} |
||||
|
||||
fn install_license(&self, _xml_path: &Path) -> Result<()> { |
||||
Err(Self::deny("install license")) |
||||
} |
||||
|
||||
fn windows_build(&self) -> Result<u32> { |
||||
// No Windows to ask; report 0 so build-gated methods are filtered out.
|
||||
Ok(0) |
||||
} |
||||
|
||||
fn windows_edition(&self) -> Result<String> { |
||||
Ok("Unknown".to_string()) |
||||
} |
||||
|
||||
fn is_elevated(&self) -> bool { |
||||
false |
||||
} |
||||
} |
||||
@ -0,0 +1,456 @@
@@ -0,0 +1,456 @@
|
||||
//! Real Windows Software Protection Platform backend, via WMI/COM.
|
||||
//!
|
||||
//! Compiled only under `cfg(all(windows, feature = "winapi"))`. It drives the
|
||||
//! same WMI classes on `ROOT\CIMV2` that MAS drives through PowerShell/`slmgr`:
|
||||
//! `SoftwareLicensingService` / `SoftwareLicensingProduct` and the Office
|
||||
//! `OfficeSoftwareProtection*` equivalents.
|
||||
//!
|
||||
//! All `unsafe`, all COM lifetime handling and every raw `VARIANT` access are
|
||||
//! contained in this file; the rest of the crate sees only the safe [`Spp`]
|
||||
//! trait (constraint 3: FFI behind a sound safe API).
|
||||
//!
|
||||
//! Written against `windows` 0.58. It is intentionally **not compiled or tested
|
||||
//! on the porting host** (a Linux box has no Windows target); treat signatures
|
||||
//! as knowledge-derived and expect to shake out minor API drift on a real
|
||||
//! Windows build. The logic — WQL queries, method names, parameter names and
|
||||
//! the `ReturnValue` HRESULT handling — mirrors the SPP contract exactly.
|
||||
|
||||
#![allow(clippy::missing_safety_doc)] |
||||
|
||||
use crate::error::{Error, Result}; |
||||
use crate::model::{LicenseStatus, Product}; |
||||
use crate::platform::{LicenseInfo, Spp}; |
||||
use std::path::Path; |
||||
|
||||
use windows::core::{Interface, BSTR, HSTRING, PCWSTR}; |
||||
use windows::Win32::Foundation::HANDLE; |
||||
use windows::Win32::Security::{ |
||||
GetTokenInformation, TokenElevation, TOKEN_ELEVATION, TOKEN_QUERY, |
||||
}; |
||||
use windows::Win32::System::Com::{ |
||||
CoCreateInstance, CoInitializeEx, CoInitializeSecurity, CLSCTX_INPROC_SERVER, |
||||
COINIT_MULTITHREADED, EOAC_NONE, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, |
||||
}; |
||||
use windows::Win32::System::Threading::{GetCurrentProcess, OpenProcessToken}; |
||||
use windows::Win32::System::Variant::{ |
||||
VariantClear, VARIANT, VT_BSTR, VT_I4, VT_UI4, |
||||
}; |
||||
use windows::Win32::System::Wmi::{ |
||||
IWbemClassObject, IWbemLocator, IWbemServices, WbemLocator, WBEM_FLAG_FORWARD_ONLY, |
||||
WBEM_FLAG_RETURN_IMMEDIATELY, WBEM_INFINITE, |
||||
}; |
||||
|
||||
/// WMI-backed SPP. Holds a live `IWbemServices` connection to `ROOT\CIMV2`.
|
||||
pub struct WindowsSpp { |
||||
services: IWbemServices, |
||||
} |
||||
|
||||
// ---- VARIANT helpers (the only place raw union fields are touched) ----------
|
||||
|
||||
/// Build a `VARIANT` holding a BSTR string.
|
||||
fn variant_bstr(s: &str) -> VARIANT { |
||||
// `VARIANT::from(&BSTR)` sets vt = VT_BSTR and owns the string; using the
|
||||
// crate's conversion keeps us off the raw union for construction.
|
||||
VARIANT::from(BSTR::from(s)) |
||||
} |
||||
|
||||
/// Build a `VARIANT` holding an unsigned 32-bit value.
|
||||
fn variant_u32(v: u32) -> VARIANT { |
||||
VARIANT::from(v as i32) |
||||
} |
||||
|
||||
/// Extract a string from a `VARIANT`, whatever numeric/bstr shape it holds.
|
||||
///
|
||||
/// # Safety
|
||||
/// `var` must be a valid, initialised VARIANT produced by a WMI `Get`.
|
||||
unsafe fn variant_to_string(var: &VARIANT) -> Option<String> { |
||||
let vt = var.Anonymous.Anonymous.vt; |
||||
let val = &var.Anonymous.Anonymous.Anonymous; |
||||
let out = match vt { |
||||
VT_BSTR => { |
||||
let b = &val.bstrVal; |
||||
if b.is_empty() { |
||||
return None; |
||||
} |
||||
b.to_string() |
||||
} |
||||
VT_I4 => val.lVal.to_string(), |
||||
VT_UI4 => val.ulVal.to_string(), |
||||
_ => return None, |
||||
}; |
||||
Some(out) |
||||
} |
||||
|
||||
/// Extract a u32 from a numeric `VARIANT`.
|
||||
///
|
||||
/// # Safety
|
||||
/// `var` must be a valid, initialised VARIANT.
|
||||
unsafe fn variant_to_u32(var: &VARIANT) -> Option<u32> { |
||||
let vt = var.Anonymous.Anonymous.vt; |
||||
let val = &var.Anonymous.Anonymous.Anonymous; |
||||
match vt { |
||||
VT_UI4 => Some(val.ulVal), |
||||
VT_I4 => Some(val.lVal as u32), |
||||
VT_BSTR => val.bstrVal.to_string().parse().ok(), |
||||
_ => None, |
||||
} |
||||
} |
||||
|
||||
// ---- WMI object property access --------------------------------------------
|
||||
|
||||
/// Read a named property off a WMI object as a string.
|
||||
fn get_string(obj: &IWbemClassObject, name: &str) -> Option<String> { |
||||
let wname = HSTRING::from(name); |
||||
let mut var = VARIANT::default(); |
||||
unsafe { |
||||
obj.Get(PCWSTR(wname.as_ptr()), 0, &mut var, None, None).ok()?; |
||||
let out = variant_to_string(&var); |
||||
let _ = VariantClear(&mut var); |
||||
out |
||||
} |
||||
} |
||||
|
||||
/// Read a named property off a WMI object as a u32.
|
||||
fn get_u32(obj: &IWbemClassObject, name: &str) -> Option<u32> { |
||||
let wname = HSTRING::from(name); |
||||
let mut var = VARIANT::default(); |
||||
unsafe { |
||||
obj.Get(PCWSTR(wname.as_ptr()), 0, &mut var, None, None).ok()?; |
||||
let out = variant_to_u32(&var); |
||||
let _ = VariantClear(&mut var); |
||||
out |
||||
} |
||||
} |
||||
|
||||
impl WindowsSpp { |
||||
/// Initialise COM and connect to the WMI licensing namespace.
|
||||
pub fn new() -> Self { |
||||
Self::connect().unwrap_or_else(|e| panic!("WMI init failed: {e}")) |
||||
} |
||||
|
||||
fn connect() -> Result<Self> { |
||||
unsafe { |
||||
// MTA — the process is a short-lived console tool.
|
||||
CoInitializeEx(None, COINIT_MULTITHREADED) |
||||
.ok() |
||||
.map_err(|e| Error::winapi("CoInitializeEx", e.code().0))?; |
||||
|
||||
// Default process-wide security for WMI. Ignore RPC_E_TOO_LATE if a
|
||||
// host already initialised security.
|
||||
let _ = CoInitializeSecurity( |
||||
None, |
||||
-1, |
||||
None, |
||||
None, |
||||
RPC_C_AUTHN_LEVEL_DEFAULT, |
||||
RPC_C_IMP_LEVEL_IMPERSONATE, |
||||
None, |
||||
EOAC_NONE, |
||||
None, |
||||
); |
||||
|
||||
let locator: IWbemLocator = |
||||
CoCreateInstance(&WbemLocator, None, CLSCTX_INPROC_SERVER) |
||||
.map_err(|e| Error::winapi("create IWbemLocator", e.code().0))?; |
||||
|
||||
let services = locator |
||||
.ConnectServer( |
||||
&BSTR::from("ROOT\\CIMV2"), |
||||
&BSTR::new(), |
||||
&BSTR::new(), |
||||
&BSTR::new(), |
||||
0, |
||||
&BSTR::new(), |
||||
None, |
||||
) |
||||
.map_err(|e| Error::winapi("WMI ConnectServer", e.code().0))?; |
||||
|
||||
Ok(WindowsSpp { services }) |
||||
} |
||||
} |
||||
|
||||
/// Run a WQL query and collect the result objects.
|
||||
fn query(&self, wql: &str) -> Result<Vec<IWbemClassObject>> { |
||||
unsafe { |
||||
let enumerator = self |
||||
.services |
||||
.ExecQuery( |
||||
&BSTR::from("WQL"), |
||||
&BSTR::from(wql), |
||||
WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, |
||||
None, |
||||
) |
||||
.map_err(|e| Error::winapi("WMI ExecQuery", e.code().0))?; |
||||
|
||||
let mut out = Vec::new(); |
||||
loop { |
||||
let mut row: [Option<IWbemClassObject>; 1] = [None]; |
||||
let mut returned = 0u32; |
||||
// Next returns S_FALSE (and returned==0) at end of enumeration.
|
||||
let hr = enumerator.Next(WBEM_INFINITE.0, &mut row, &mut returned); |
||||
if returned == 0 || hr.is_err() { |
||||
break; |
||||
} |
||||
if let Some(obj) = row[0].take() { |
||||
out.push(obj); |
||||
} |
||||
} |
||||
Ok(out) |
||||
} |
||||
} |
||||
|
||||
/// Fetch the WMI `__PATH` of the single service instance of `class`.
|
||||
fn service_path(&self, class: &str) -> Result<String> { |
||||
let rows = self.query(&format!("SELECT __PATH FROM {class}"))?; |
||||
rows.first() |
||||
.and_then(|o| get_string(o, "__PATH")) |
||||
.ok_or_else(|| Error::winapi_msg(format!("no {class} instance"))) |
||||
} |
||||
|
||||
/// Invoke a WMI method on `object_path` (an instance or service `__PATH`),
|
||||
/// setting the given string/u32 parameters, and return its `ReturnValue`
|
||||
/// (the SPP HRESULT: 0 = success).
|
||||
fn call_method( |
||||
&self, |
||||
object_path: &str, |
||||
class: &str, |
||||
method: &str, |
||||
params: &[(&str, ParamValue)], |
||||
) -> Result<i32> { |
||||
unsafe { |
||||
// Build the in-parameters instance from the class' method signature.
|
||||
let mut class_obj: Option<IWbemClassObject> = None; |
||||
self.services |
||||
.GetObject( |
||||
&BSTR::from(class), |
||||
Default::default(), |
||||
None, |
||||
Some(&mut class_obj), |
||||
None, |
||||
) |
||||
.map_err(|e| Error::winapi("WMI GetObject(class)", e.code().0))?; |
||||
let class_obj = |
||||
class_obj.ok_or_else(|| Error::winapi_msg(format!("class {class} not found")))?; |
||||
|
||||
let in_params = if params.is_empty() { |
||||
None |
||||
} else { |
||||
let wmethod = HSTRING::from(method); |
||||
let mut in_sig: Option<IWbemClassObject> = None; |
||||
let mut out_sig: Option<IWbemClassObject> = None; |
||||
class_obj |
||||
.GetMethod(PCWSTR(wmethod.as_ptr()), 0, &mut in_sig, &mut out_sig) |
||||
.map_err(|e| Error::winapi("WMI GetMethod", e.code().0))?; |
||||
let in_sig = in_sig |
||||
.ok_or_else(|| Error::winapi_msg(format!("method {method} has no in-params")))?; |
||||
let inst = in_sig |
||||
.SpawnInstance(0) |
||||
.map_err(|e| Error::winapi("WMI SpawnInstance", e.code().0))?; |
||||
for (pname, pval) in params { |
||||
let wname = HSTRING::from(*pname); |
||||
let mut var = match pval { |
||||
ParamValue::Str(s) => variant_bstr(s), |
||||
ParamValue::U32(v) => variant_u32(*v), |
||||
}; |
||||
inst.Put(PCWSTR(wname.as_ptr()), 0, &var, 0) |
||||
.map_err(|e| Error::winapi("WMI Put(param)", e.code().0))?; |
||||
let _ = VariantClear(&mut var); |
||||
} |
||||
Some(inst) |
||||
}; |
||||
|
||||
let mut out_params: Option<IWbemClassObject> = None; |
||||
self.services |
||||
.ExecMethod( |
||||
&BSTR::from(object_path), |
||||
&BSTR::from(method), |
||||
Default::default(), |
||||
None, |
||||
in_params.as_ref(), |
||||
Some(&mut out_params), |
||||
None, |
||||
) |
||||
.map_err(|e| Error::winapi(format!("WMI ExecMethod {method}"), e.code().0))?; |
||||
|
||||
// ReturnValue is the operation's own HRESULT; absent ⇒ success.
|
||||
let ret = out_params |
||||
.as_ref() |
||||
.and_then(|o| get_u32(o, "ReturnValue")) |
||||
.unwrap_or(0) as i32; |
||||
Ok(ret) |
||||
} |
||||
} |
||||
|
||||
/// Wrap [`call_method`] turning a non-zero `ReturnValue` into an error.
|
||||
fn call_checked( |
||||
&self, |
||||
object_path: &str, |
||||
class: &str, |
||||
method: &str, |
||||
params: &[(&str, ParamValue)], |
||||
) -> Result<()> { |
||||
match self.call_method(object_path, class, method, params)? { |
||||
0 => Ok(()), |
||||
hr => Err(Error::winapi(format!("{class}.{method}"), hr)), |
||||
} |
||||
} |
||||
|
||||
/// `__PATH` of a specific product row identified by its activation ID.
|
||||
fn product_path(&self, product: Product, activation_id: &str) -> Result<String> { |
||||
let wql = format!( |
||||
"SELECT __PATH FROM {class} WHERE ID='{id}'", |
||||
class = product.product_class(), |
||||
id = escape(activation_id), |
||||
); |
||||
self.query(&wql)? |
||||
.first() |
||||
.and_then(|o| get_string(o, "__PATH")) |
||||
.ok_or_else(|| Error::winapi_msg(format!("product {activation_id} not found"))) |
||||
} |
||||
} |
||||
|
||||
/// Value kinds accepted by [`WindowsSpp::call_method`].
|
||||
enum ParamValue { |
||||
Str(String), |
||||
U32(u32), |
||||
} |
||||
|
||||
/// Minimal WQL string-literal escaping (single quote → doubled).
|
||||
fn escape(s: &str) -> String { |
||||
s.replace('\'', "''") |
||||
} |
||||
|
||||
impl Spp for WindowsSpp { |
||||
fn installed_products(&self, product: Product) -> Result<Vec<LicenseInfo>> { |
||||
let wql = format!( |
||||
"SELECT ID, Name, Description, PartialProductKey, LicenseStatus, LicenseFamily, \ |
||||
GracePeriodRemaining FROM {class} WHERE ApplicationID='{app}' \ |
||||
AND PartialProductKey IS NOT NULL AND LicenseDependsOn IS NULL", |
||||
class = product.product_class(), |
||||
app = product.application_id(), |
||||
); |
||||
let mut out = Vec::new(); |
||||
for obj in self.query(&wql)? { |
||||
let status_raw = get_u32(&obj, "LicenseStatus").unwrap_or(0); |
||||
out.push(LicenseInfo { |
||||
activation_id: get_string(&obj, "ID").unwrap_or_default(), |
||||
name: get_string(&obj, "Name").unwrap_or_default(), |
||||
description: get_string(&obj, "Description").unwrap_or_default(), |
||||
partial_product_key: get_string(&obj, "PartialProductKey"), |
||||
status: LicenseStatus::from_wmi(status_raw).unwrap_or(LicenseStatus::Unlicensed), |
||||
license_family: get_string(&obj, "LicenseFamily"), |
||||
grace_minutes: get_u32(&obj, "GracePeriodRemaining"), |
||||
}); |
||||
} |
||||
Ok(out) |
||||
} |
||||
|
||||
fn install_product_key(&self, product: Product, key: &str) -> Result<()> { |
||||
let class = product.service_class(); |
||||
let path = self.service_path(class)?; |
||||
self.call_checked( |
||||
&path, |
||||
class, |
||||
"InstallProductKey", |
||||
&[("ProductKey", ParamValue::Str(key.to_string()))], |
||||
) |
||||
} |
||||
|
||||
fn uninstall_product_key(&self, product: Product, activation_id: &str) -> Result<()> { |
||||
let path = self.product_path(product, activation_id)?; |
||||
self.call_checked(&path, product.product_class(), "UninstallProductKey", &[]) |
||||
} |
||||
|
||||
fn set_kms_host(&self, product: Product, host: &str, port: u16) -> Result<()> { |
||||
let class = product.service_class(); |
||||
let path = self.service_path(class)?; |
||||
self.call_checked( |
||||
&path, |
||||
class, |
||||
"SetKeyManagementServiceMachine", |
||||
&[("MachineName", ParamValue::Str(host.to_string()))], |
||||
)?; |
||||
self.call_checked( |
||||
&path, |
||||
class, |
||||
"SetKeyManagementServicePort", |
||||
&[("PortNumber", ParamValue::U32(port as u32))], |
||||
) |
||||
} |
||||
|
||||
fn clear_kms_host(&self, product: Product) -> Result<()> { |
||||
let class = product.service_class(); |
||||
let path = self.service_path(class)?; |
||||
self.call_checked(&path, class, "ClearKeyManagementServiceMachine", &[])?; |
||||
self.call_checked(&path, class, "ClearKeyManagementServicePort", &[]) |
||||
} |
||||
|
||||
fn activate(&self, product: Product, activation_id: &str) -> Result<()> { |
||||
let path = self.product_path(product, activation_id)?; |
||||
self.call_checked(&path, product.product_class(), "Activate", &[]) |
||||
} |
||||
|
||||
fn install_license(&self, xml_path: &Path) -> Result<()> { |
||||
let xml = std::fs::read_to_string(xml_path)?; |
||||
let class = Product::Windows.service_class(); |
||||
let path = self.service_path(class)?; |
||||
self.call_checked( |
||||
&path, |
||||
class, |
||||
"InstallLicense", |
||||
&[("License", ParamValue::Str(xml))], |
||||
) |
||||
} |
||||
|
||||
fn windows_build(&self) -> Result<u32> { |
||||
let rows = self.query("SELECT BuildNumber FROM Win32_OperatingSystem")?; |
||||
rows.first() |
||||
.and_then(|o| get_string(o, "BuildNumber")) |
||||
.and_then(|s| s.trim().parse().ok()) |
||||
.ok_or_else(|| Error::winapi_msg("could not read Windows build")) |
||||
} |
||||
|
||||
fn windows_edition(&self) -> Result<String> { |
||||
// Best-effort from the OS caption, e.g. "Microsoft Windows 11 Pro".
|
||||
let rows = self.query("SELECT Caption FROM Win32_OperatingSystem")?; |
||||
Ok(rows |
||||
.first() |
||||
.and_then(|o| get_string(o, "Caption")) |
||||
.unwrap_or_else(|| "Unknown".to_string())) |
||||
} |
||||
|
||||
fn is_elevated(&self) -> bool { |
||||
unsafe { |
||||
let process = GetCurrentProcess(); |
||||
let mut token = HANDLE::default(); |
||||
if OpenProcessToken(process, TOKEN_QUERY, &mut token).is_err() { |
||||
return false; |
||||
} |
||||
let mut elevation = TOKEN_ELEVATION::default(); |
||||
let mut size = 0u32; |
||||
let ok = GetTokenInformation( |
||||
token, |
||||
TokenElevation, |
||||
Some(&mut elevation as *mut _ as *mut _), |
||||
std::mem::size_of::<TOKEN_ELEVATION>() as u32, |
||||
&mut size, |
||||
) |
||||
.is_ok(); |
||||
let _ = windows::Win32::Foundation::CloseHandle(token); |
||||
ok && elevation.TokenIsElevated != 0 |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl Drop for WindowsSpp { |
||||
fn drop(&mut self) { |
||||
// Balance CoInitializeEx. Safe: called once per successful init.
|
||||
unsafe { windows::Win32::System::Com::CoUninitialize() }; |
||||
} |
||||
} |
||||
|
||||
impl Default for WindowsSpp { |
||||
fn default() -> Self { |
||||
Self::new() |
||||
} |
||||
} |
||||
Loading…
Reference in new issue