acdiu/rustdesk-server
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk-server.git synced 2026-05-25 23:59:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

always check license for future charging

Browse Source
This commit is contained in:
rustdesk
2021-04-18 10:56:17 +08:00
parent d87d761ab6
commit 5804afa3e5
1 changed files with 37 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
src/lic.rs
View File

@@ -24,7 +24,7 @@ pub struct Post {
#[serde(default)] #[serde(default)]
version: String, version: String,
#[serde(default)] #[serde(default)]
next_check_time: u32, next_check_time: u64,
} }
const LICENSE_FILE: &'static str = ".license.txt"; const LICENSE_FILE: &'static str = ".license.txt";
@@ -40,12 +40,14 @@ pub fn check_lic(email: &str, version: &str) -> bool {
if Path::is_file(&path) { if Path::is_file(&path) {
let contents = std::fs::read_to_string(&path).unwrap_or("".to_owned()); let contents = std::fs::read_to_string(&path).unwrap_or("".to_owned());
if verify(&contents, &machine) { if verify(&contents, &machine) {
async_check_email(&machine, email, version, 0);
return true; return true;
} }
} }
match check_email(machine, email.to_owned(), version.to_owned()) { match check_email(machine.clone(), email.to_owned(), version.to_owned()) {
Ok(v) => { Ok(v) => {
async_check_email(&machine, email, version, v);
return true; return true;
} }
Err(err) => { Err(err) => {
@@ -55,6 +57,30 @@ pub fn check_lic(email: &str, version: &str) -> bool {
} }
} }
fn async_check_email(machine: &str, email: &str, version: &str, wait: u64) {
let machine = machine.to_owned();
let email = email.to_owned();
let version = version.to_owned();
std::thread::spawn(move || {
let mut wait = wait;
loop {
let machine = machine.clone();
let email = email.clone();
let version = version.clone();
std::thread::sleep(std::time::Duration::from_secs(wait));
match check_email(machine, email, version) {
Ok(v) => {
wait = v;
}
Err(err) => {
log::error!("{}", err);
std::process::exit(-1);
}
}
}
});
}
fn write_lic(lic: &str) { fn write_lic(lic: &str) {
if let Ok(mut f) = std::fs::File::create(LICENSE_FILE) { if let Ok(mut f) = std::fs::File::create(LICENSE_FILE) {
f.write_all(lic.as_bytes()).ok(); f.write_all(lic.as_bytes()).ok();
@@ -62,8 +88,8 @@ fn write_lic(lic: &str) {
} }
} }
fn check_email(machine: String, email: String, version: String) -> ResultType<u32> { fn check_email(machine: String, email: String, version: String) -> ResultType<u64> {
log::info!("Checking email with the server ..."); log::info!("Checking email with the license server ...");
let resp = minreq::post("http://rustdesk.com/api/check-email") let resp = minreq::post("http://rustdesk.com/api/check-email")
.with_body( .with_body(
serde_json::to_string(&Post { serde_json::to_string(&Post {
@@ -84,7 +110,13 @@ fn check_email(machine: String, email: String, version: String) -> ResultType<u3
bail!("Verification failure"); bail!("Verification failure");
} }
write_lic(&p.machine); write_lic(&p.machine);
Ok(p.next_check_time) log::info!("License OK");
let mut wait = p.next_check_time;
if wait == 0 {
wait = 3600 * 24 * 30;
}
Ok(wait)
} else { } else {
bail!("Server error: {}", resp.reason_phrase); bail!("Server error: {}", resp.reason_phrase);
} }