tonarino/innernet
1
1
Fork 0
mirror of https://github.com/tonarino/innernet.git synced 2024-09-20 14:56:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

server: validate hostname in 'new'

Browse source 
fixes #164
This commit is contained in:
Jake McGinty 2021-11-05 12:36:35 +09:00
parent 17dd26921f
commit d7cf24c63c
2 changed files with 11 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
server/src/initialize.rs
View file

@ -5,7 +5,7 @@ use dialoguer::{theme::ColorfulTheme, Input};
use indoc::printdoc;
use publicip::Preference;
use rusqlite::{params, Connection};
use shared::{prompts, CidrContents, Endpoint, PeerContents, PERSISTENT_KEEPALIVE_INTERVAL_SECS};
use shared::{CidrContents, Endpoint, PERSISTENT_KEEPALIVE_INTERVAL_SECS, PeerContents, prompts};
use wireguard_control::KeyPair;
fn create_database<P: AsRef<Path>>(
@ -26,7 +26,7 @@ fn create_database<P: AsRef<Path>>(
pub struct InitializeOpts {
/// The network name (ex: evilcorp)
#[structopt(long)]
pub network_name: Option<InterfaceName>,
pub network_name: Option<Interface>,
/// The network CIDR (ex: 10.42.0.0/16)
#[structopt(long)]
@ -122,7 +122,7 @@ pub fn init_wizard(conf: &ServerConfig, opts: InitializeOpts) -> Result<(), Erro
\n"
);
let name: InterfaceName = if let Some(name) = opts.network_name {
let name: Interface = if let Some(name) = opts.network_name {
name
} else {
Input::with_theme(&theme)

19
shared/src/types.rs
View file

@ -3,16 +3,7 @@ use ipnetwork::IpNetwork;
use lazy_static::lazy_static;
use regex::Regex;
use serde::{Deserialize, Serialize};
use std::{
fmt::{self, Display, Formatter},
io,
net::{IpAddr, SocketAddr, ToSocketAddrs},
ops::{Deref, DerefMut},
path::Path,
str::FromStr,
time::{Duration, SystemTime},
vec,
};
use std::{fmt::{self, Display, Formatter}, io, net::{IpAddr, SocketAddr, ToSocketAddrs}, ops::{Deref, DerefMut}, path::Path, str::FromStr, time::{Duration, SystemTime}, vec};
use structopt::StructOpt;
use url::Host;
use wireguard_control::{
@ -22,7 +13,7 @@ use wireguard_control::{
use crate::wg::PeerInfoExt;
#[derive(Debug, Clone)]
#[derive(Debug, Clone, PartialEq)]
pub struct Interface {
name: InterfaceName,
}
@ -49,6 +40,12 @@ impl Deref for Interface {
}
}
impl Display for Interface {
fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
f.write_str(&self.name.to_string())
}
}
#[derive(Clone, Debug, PartialEq)]
/// An external endpoint that supports both IP and domain name hosts.
pub struct Endpoint {