diff options
| -rw-r--r-- | Cargo.lock | 30 | ||||
| -rw-r--r-- | Cargo.toml | 4 | ||||
| m--------- | libs/hbb_common | 0 | ||||
| -rw-r--r-- | src/lic.rs | 84 | ||||
| -rw-r--r-- | src/main.rs | 3 |
5 files changed, 119 insertions, 2 deletions
@@ -618,15 +618,19 @@ dependencies = [ name = "hbbs" version = "1.1.3" dependencies = [ + "base64 0.13.0", "clap", "hbb_common", "lazy_static", + "mac_address", + "machine-uid", "reqwest", "rocksdb", "rust-ini", "serde", "serde_derive", "serde_json", + "whoami", ] [[package]] @@ -873,6 +877,15 @@ dependencies = [ ] [[package]] +name = "machine-uid" +version = "0.2.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1f1595709b0a7386bcd56ba34d250d626e5503917d05d32cdccddcd68603e212" +dependencies = [ + "winreg 0.6.2", +] + +[[package]] name = "matches" version = "0.1.8" source = "registry+https://github.com/rust-lang/crates.io-index" @@ -1420,7 +1433,7 @@ dependencies = [ "wasm-bindgen", "wasm-bindgen-futures", "web-sys", - "winreg", + "winreg 0.7.0", ] [[package]] @@ -2073,6 +2086,12 @@ dependencies = [ ] [[package]] +name = "whoami" +version = "0.9.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7884773ab69074615cb8f8425d0e53f11710786158704fca70f53e71b0e05504" + +[[package]] name = "winapi" version = "0.2.8" source = "registry+https://github.com/rust-lang/crates.io-index" @@ -2117,6 +2136,15 @@ checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f" [[package]] name = "winreg" +version = "0.6.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b2986deb581c4fe11b621998a5e53361efe6b48a151178d0cd9eeffa4dc6acc9" +dependencies = [ + "winapi 0.3.9", +] + +[[package]] +name = "winreg" version = "0.7.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "0120db82e8a1e0b9fb3345a539c478767c0048d842860994d96113d5b667bd69" @@ -20,6 +20,10 @@ lazy_static = "1.4" clap = "2.33" rust-ini = "0.16" reqwest = "0.10" +machine-uid = "0.2" +mac_address = "1.1" +whoami = "0.9" +base64 = "0.13" [build-dependencies] hbb_common = { path = "libs/hbb_common" } diff --git a/libs/hbb_common b/libs/hbb_common -Subproject 20bea85903acbd701aed45d195e9206f2ea09ed +Subproject 98257ca5b799ecae53e2fc3ec20bb997a2f6206 @@ -1,6 +1,90 @@ +use hbb_common::{bail, log, sodiumoxide::crypto::sign, ResultType}; +use serde_derive::{Deserialize, Serialize}; +use std::io::prelude::*; +use std::path::Path; + +#[derive(Debug, PartialEq, Default, Serialize, Deserialize, Clone)] +pub struct License { + #[serde(default)] + pub hostname: String, + #[serde(default)] + pub uid: String, + #[serde(default)] + pub mac: String, +} + +const LICENSE_FILE: &'static str = ".license.txt"; + pub fn check_lic(email: &str) -> bool { + let lic = get_lic(); + let path = Path::new(LICENSE_FILE); + if Path::is_file(&path) { + let contents = std::fs::read_to_string(&path).unwrap_or("".to_owned()); + if let Ok(old_lic) = dec_lic(&contents) { + if lic == old_lic { + return true; + } + } + } + + if email.is_empty() { + log::error!("Registered email required."); + return false; + } + + if let Ok(s) = enc_lic(&lic) { + if let Ok(mut f) = std::fs::File::create(path) { + f.write_all(s.as_bytes()).ok(); + f.sync_all().ok(); + } + } true } +pub fn get_lic() -> License { + let hostname = whoami::hostname(); + let uid = machine_uid::get().unwrap_or("".to_owned()); + let mac = if let Ok(Some(ma)) = mac_address::get_mac_address() { + base64::encode_config(ma.bytes(), base64::URL_SAFE_NO_PAD) + } else { + "".to_owned() + }; + License { hostname, uid, mac } +} + +pub fn enc_lic(lic: &License) -> ResultType<String> { + let tmp = serde_json::to_vec::<License>(lic)?; + const SK: &[u64] = &[ + 139, 164, 88, 86, 6, 123, 221, 248, 96, 36, 106, 207, 99, 124, 27, 196, 5, 159, 58, 253, + 238, 94, 3, 184, 237, 236, 122, 59, 205, 95, 6, 189, 88, 168, 68, 104, 60, 5, 163, 198, + 165, 38, 12, 85, 114, 203, 96, 163, 70, 48, 0, 131, 57, 12, 46, 129, 83, 17, 84, 193, 119, + 197, 130, 103, + ]; + let sk: Vec<u8> = SK.iter().map(|x| *x as u8).collect(); + let mut sk_ = [0u8; sign::SECRETKEYBYTES]; + sk_[..].copy_from_slice(&sk); + let sk = sign::SecretKey(sk_); + let tmp = base64::encode_config(sign::sign(&tmp, &sk), base64::URL_SAFE_NO_PAD); + let tmp: String = tmp.chars().rev().collect(); + Ok(tmp) +} + +pub fn dec_lic(s: &str) -> ResultType<License> { + let tmp: String = s.chars().rev().collect(); + const PK: &[u64] = &[ + 88, 168, 68, 104, 60, 5, 163, 198, 165, 38, 12, 85, 114, 203, 96, 163, 70, 48, 0, 131, 57, + 12, 46, 129, 83, 17, 84, 193, 119, 197, 130, 103, + ]; + let pk: Vec<u8> = PK.iter().map(|x| *x as u8).collect(); + let mut pk_ = [0u8; sign::PUBLICKEYBYTES]; + pk_[..].copy_from_slice(&pk); + let pk = sign::PublicKey(pk_); + if let Ok(data) = sign::verify(&base64::decode_config(tmp, base64::URL_SAFE_NO_PAD)?, &pk) { + Ok(serde_json::from_slice::<License>(&data)?) + } else { + bail!("sign:verify failed"); + } +} + pub const EMAIL_ARG: &'static str = "-m, --email=[EMAIL] 'Sets your email address registered with RustDesk'"; diff --git a/src/main.rs b/src/main.rs index 16de822..0b8c919 100644 --- a/src/main.rs +++ b/src/main.rs @@ -20,7 +20,8 @@ fn main() -> ResultType<()> { -C, --change-id=[BOOL(default=Y)] 'Sets if support to change id' {} -k, --key=[KEY] 'Only allow the client with the same key'", - DEFAULT_PORT, lic::EMAIL_ARG + DEFAULT_PORT, + lic::EMAIL_ARG ); let matches = App::new("hbbs") .version(crate::VERSION) |