zishuo/vnts
1
0
Fork 0
mirror of https://github.com/lbl8603/vnts.git synced 2025-12-24 12:47:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

修改校验

Browse Source
This commit is contained in:
lbl8603
2024-07-20 14:57:19 +08:00
parent 6d2bddd755
commit 23966d3138
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/core/server/web/service/mod.rs
View File

@@ -1,4 +1,4 @@
use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4, ToSocketAddrs};
use std::str::FromStr;
use std::sync::Arc;
use std::time::{Duration, Instant};
@@ -165,10 +165,17 @@ impl VntsWebService {
Ok(wg_data)
}
fn check_wg_config(config: &CreateWgConfig) -> anyhow::Result<([u8; 32], [u8; 32])> {
let addr = SocketAddr::from_str(&config.vnts_endpoint).context("服务器地址错误")?;
if addr.ip().is_unspecified() || addr.port() == 0 {
Err(anyhow!("服务端地址错误"))?
match config.vnts_endpoint.to_socket_addrs() {
Ok(mut addr) => {
if let Some(addr) = addr.next() {
if addr.ip().is_unspecified() || addr.port() == 0 {
Err(anyhow!("服务端地址错误"))?
}
}
}
Err(e) => Err(anyhow!("服务端地址解析失败:{}", e))?,
}
let private_key = general_purpose::STANDARD
.decode(&config.private_key)
.context("私钥错误")?;