13 Commits

Author SHA1 Message Date
8f6e1ddc53 2024 edition + cargo update
All checks were successful
Docker Image CI / test (push) Successful in 15m43s
Docker Image CI / push (push) Successful in 33m17s
2025-03-11 15:28:41 +03:00
67f925f160 remove unused code for vars
All checks were successful
Docker Image CI / test (push) Successful in 2m56s
Docker Image CI / push (push) Successful in 22m30s
2024-10-28 11:35:18 +03:00
7a22af7598 vars with Arc<str>
All checks were successful
Docker Image CI / test (push) Successful in 3m0s
Docker Image CI / push (push) Successful in 21m33s
2024-10-28 00:43:18 +03:00
73435ae70a ignore /.idea
Some checks failed
Docker Image CI / test (push) Successful in 3m10s
Docker Image CI / push (push) Failing after 11m32s
2024-10-27 20:19:57 +03:00
9bf2103c47 app config refactor
All checks were successful
Docker Image CI / test (push) Successful in 3m26s
Docker Image CI / push (push) Successful in 22m31s
2024-10-25 12:02:53 +03:00
1179840114 arg --openssl to --with-openssl
All checks were successful
Docker Image CI / test (push) Successful in 3m3s
Docker Image CI / push (push) Successful in 22m24s
2024-10-25 10:43:07 +03:00
5e3ec48259 ovpn config refactor
All checks were successful
Docker Image CI / test (push) Successful in 3m7s
Docker Image CI / push (push) Successful in 23m5s
2024-10-25 09:25:06 +03:00
cdac6d2aa5 openssl providers in separate files
All checks were successful
Docker Image CI / test (push) Successful in 2m34s
Docker Image CI / push (push) Successful in 14m47s
2024-10-22 16:23:53 +03:00
f5e207654c upd Dockerfile 2024-10-22 16:13:44 +03:00
232ad335fa upd tokio 2024-10-22 16:05:39 +03:00
c7677bdb70 refactor 2024-10-22 15:45:36 +03:00
fa9c1ecb2c + san email 2024-10-22 14:18:55 +03:00
8c1add6ff1 +internal openssl cert provider 2024-10-22 00:41:06 +03:00
14 changed files with 1877 additions and 389 deletions

2
.gitignore vendored
View File

@@ -1,3 +1,5 @@
/target
/.env.ps1
/.vscode
/.idea

952
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@@ -1,18 +1,22 @@
[package]
name = "peazyrsa"
version = "0.1.0"
edition = "2021"
edition = "2024"
[dependencies]
anyhow = "1.0.90"
async-stream = "0.3.6"
chrono = "0.4.38"
clap = { version = "4.5.20", features = ["derive"] }
encoding = "0.2.33"
futures = "0.3.31"
futures-core = "0.3.31"
futures-util = "0.3.31"
lazy_static = "1.5.0"
openssl = { version="0.10.68" }
regex = "1.11.0"
tokio = { version = "1.40.0", features = ["fs", "rt", "process", "macros", "io-util"] }
tera = "1.20.0"
tokio = { version = "1.41.0", features = ["fs", "rt", "process", "macros", "io-util"] }
[profile.release]
opt-level = 3

View File

@@ -1,5 +1,6 @@
FROM lukemathwalker/cargo-chef:latest-rust-1 AS chef
WORKDIR /app
RUN apt --no-install-recommends update && apt install -y libssl-dev
FROM chef AS planner
COPY . .

69
src/arcstr.rs Normal file
View File

@@ -0,0 +1,69 @@
use std::ffi::OsStr;
use std::fmt::{Display, Formatter};
use std::ops::Deref;
use std::path::Path;
use std::sync::Arc;
#[derive(Debug, Clone, PartialEq, Ord, PartialOrd, Eq)]
pub(crate) struct ArcStr {
inner: Arc<str>,
}
impl Deref for ArcStr {
type Target = str;
fn deref(&self) -> &Self::Target {
self.inner.deref()
}
}
impl From<&str> for ArcStr {
fn from(value: &str) -> Self {
Self {
inner: Arc::from(value),
}
}
}
#[allow(suspicious_double_ref_op)]
impl From<&&str> for ArcStr {
fn from(value: &&str) -> Self {
Self {
inner: Arc::from(value.deref()),
}
}
}
impl AsRef<OsStr> for ArcStr {
fn as_ref(&self) -> &OsStr {
self.inner.as_ref().as_ref()
}
}
impl AsRef<Path> for ArcStr {
fn as_ref(&self) -> &Path {
self.inner.as_ref().as_ref()
}
}
impl AsRef<ArcStr> for ArcStr {
fn as_ref(&self) -> &ArcStr {
self
}
}
impl Display for ArcStr {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.deref())
}
}
impl ArcStr {
pub(crate) fn as_path(&self) -> &Path {
self.as_ref()
}
pub(crate) fn as_str(&self) -> &str {
self.inner.deref()
}
}

View File

@@ -1,20 +1,17 @@
use anyhow::{anyhow, Result};
use anyhow::{Context, Result};
use std::{path::PathBuf, sync::Arc};
use crate::common::{is_file_exist, read_file, write_file, AppConfig};
use crate::crypto::ICryptoProvider;
use crate::common::AppConfig;
use crate::crypto_provider::ICryptoProvider;
pub(crate) struct Certs<T>
where
T: ICryptoProvider,
{
pub(crate) encoding: String,
pub(crate) ca_file: PathBuf,
pub(crate) key_file: PathBuf,
pub(crate) cert_file: PathBuf,
pub(crate) config_file: PathBuf,
pub(crate) template_file: PathBuf,
pub(crate) provider: Arc<T>,
}
@@ -23,26 +20,16 @@ where
T: ICryptoProvider,
{
pub(crate) fn new(cfg: &AppConfig, provider: T) -> Self {
let base_dir = PathBuf::from(&cfg.base_directory);
let keys_dir = base_dir.clone().join(cfg.keys_subdir.clone());
let config_dir = base_dir.clone().join(cfg.config_subdir.clone());
let name = cfg.name.clone();
let provider = Arc::new(provider);
Certs {
encoding: cfg.encoding.clone(),
ca_file: keys_dir.join(cfg.ca_filename.clone()),
key_file: keys_dir.join(format!("{}.key", &name)),
cert_file: keys_dir.join(format!("{}.crt", &name)),
config_file: config_dir.join(format!("{}.ovpn", &name)),
template_file: base_dir.clone().join(cfg.template_file.clone()),
provider: Arc::new(provider),
ca_file: cfg.ca_filepath.clone(),
key_file: cfg.key_filepath.clone(),
cert_file: cfg.cert_filepath.clone(),
provider,
}
}
async fn is_config_exists(&self) -> bool {
is_file_exist(&self.config_file).await
}
pub(crate) async fn request(&self) -> Result<()> {
self.provider.request().await
}
@@ -51,39 +38,10 @@ where
self.provider.sign().await
}
pub(crate) async fn build_client_config(&self) -> Result<bool> {
if self.is_config_exists().await {
return Ok(false);
}
pub(crate) async fn build_all(&self) -> Result<()> {
self.request().await.context("request")?;
self.sign().await.context("sign")?;
self.request().await?;
self.sign().await?;
let (template_file, ca_file, cert_file, key_file) = (
self.template_file.clone(),
self.ca_file.clone(),
self.cert_file.clone(),
self.key_file.clone(),
);
let enc = self.encoding.clone();
let (enc1, enc2, enc3, enc4) = (enc.clone(), enc.clone(), enc.clone(), enc.clone());
if let (Ok(Ok(template)), Ok(Ok(ca)), Ok(Ok(cert)), Ok(Ok(key))) = tokio::join!(
tokio::spawn(read_file(template_file, enc1)),
tokio::spawn(read_file(ca_file, enc2)),
tokio::spawn(read_file(cert_file, enc3)),
tokio::spawn(read_file(key_file, enc4))
) {
let text = template
.replace("{{ca}}", ca.trim())
.replace("{{cert}}", cert.trim())
.replace("{{key}}", key.trim());
write_file(&self.config_file, text, &self.encoding).await?;
Ok(true)
} else {
Err(anyhow!("files read error"))
}
Ok(())
}
}

View File

@@ -3,7 +3,7 @@ use async_stream::stream;
use clap::Parser;
use encoding::{label::encoding_from_whatwg_label, EncoderTrap};
use std::{
collections::BTreeMap,
fmt::Display,
path::{Path, PathBuf},
};
use tokio::{
@@ -13,7 +13,32 @@ use tokio::{
use futures_core::stream::Stream;
pub(crate) type VarsMap = BTreeMap<String, String>;
pub(crate) const UTF8_STR: &str = "utf8";
pub(crate) const DEFAULT_ENCODING: &str = UTF8_STR;
#[derive(Debug, Clone, PartialEq)]
pub enum OpenSSLProviderArg {
Internal,
ExternalBin(String),
}
impl From<Option<&String>> for OpenSSLProviderArg {
fn from(value: Option<&String>) -> Self {
match value {
Some(x) => OpenSSLProviderArg::ExternalBin(x.clone()),
_ => OpenSSLProviderArg::Internal,
}
}
}
impl Display for OpenSSLProviderArg {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
OpenSSLProviderArg::ExternalBin(x) => write!(f, "{}", x),
_ => write!(f, "internal"),
}
}
}
#[derive(Parser)]
#[command(author, version, about, long_about = None)]
@@ -42,12 +67,12 @@ pub(crate) struct Args {
pub(crate) config_dir: String,
/// valid days
#[arg(long, default_value = "30650")]
#[arg(long, default_value = "3650")]
pub(crate) days: u32,
/// openssl binary
#[arg(long, default_value = "openssl")]
pub(crate) openssl: String,
/// use openssl binary
#[arg(long = "with-openssl", short)]
pub(crate) openssl: Option<String>,
/// template file
#[arg(long, default_value = "template.ovpn")]
@@ -57,35 +82,54 @@ pub(crate) struct Args {
pub(crate) struct AppConfig {
pub(crate) encoding: String,
pub(crate) req_days: u32,
pub(crate) keys_subdir: String,
pub(crate) config_subdir: String,
pub(crate) template_file: String,
pub(crate) template_file: PathBuf,
pub(crate) openssl_default_cnf: String,
pub(crate) openssl_cnf_env: String,
pub(crate) ca_filename: String,
pub(crate) ca_filepath: PathBuf,
pub(crate) ca_key_filepath: PathBuf,
pub(crate) default_email_domain: String,
pub(crate) openssl: String,
pub(crate) base_directory: String,
pub(crate) openssl: OpenSSLProviderArg,
pub(crate) base_directory: PathBuf,
pub(crate) email: String,
pub(crate) name: String,
pub(crate) conf_filepath: PathBuf,
pub(crate) req_filepath: PathBuf,
pub(crate) key_filepath: PathBuf,
pub(crate) cert_filepath: PathBuf,
}
impl Default for AppConfig {
fn default() -> Self {
let name = String::from("user");
let base_directory: PathBuf = ".".into();
let keys_directory = base_directory.join("keys");
let config_directory = base_directory.join("config");
let template_file = base_directory.join("template.ovpn");
let openssl_default_cnf = String::from("openssl-1.0.0.cnf");
let ca_filepath = keys_directory.join("ca.crt");
let ca_key_filepath = keys_directory.join("ca.key");
let req_filepath = keys_directory.join(format!("{}.csr", &name));
let key_filepath = keys_directory.join(format!("{}.key", &name));
let cert_filepath = keys_directory.join(format!("{}.crt", &name));
let conf_filepath = config_directory.join(format!("{}.ovpn", &name));
Self {
encoding: "cp866".into(),
encoding: DEFAULT_ENCODING.into(),
req_days: 30650,
keys_subdir: "keys".into(),
config_subdir: "config".into(),
template_file: "template.ovpn".into(),
openssl_default_cnf: "openssl-1.0.0.cnf".into(),
template_file,
conf_filepath,
req_filepath,
key_filepath,
cert_filepath,
openssl_default_cnf,
openssl_cnf_env: "KEY_CONFIG".into(),
ca_filename: "ca.crt".into(),
ca_filepath,
ca_key_filepath,
default_email_domain: "example.com".into(),
openssl: "openssl".into(),
base_directory: ".".into(),
openssl: OpenSSLProviderArg::Internal,
base_directory,
email: "name@example.com".into(),
name: "user".into(),
name,
}
}
}
@@ -94,11 +138,25 @@ impl From<&Args> for AppConfig {
fn from(args: &Args) -> Self {
let defaults = Self::default();
let name = args.name.clone();
let base_directory = args
.directory
.as_ref()
.unwrap_or(&defaults.base_directory)
.map(PathBuf::from)
.unwrap_or(defaults.base_directory)
.clone();
let keys_directory = base_directory.join(&args.keys_dir);
let config_directory = base_directory.join(&args.config_dir);
let template_file = base_directory.join(&args.template_file);
let openssl_default_cnf = String::from("openssl-1.0.0.cnf");
let ca_filepath = keys_directory.join("ca.crt");
let ca_key_filepath = keys_directory.join("ca.key");
let req_filepath = keys_directory.join(format!("{}.csr", &name));
let key_filepath = keys_directory.join(format!("{}.key", &name));
let cert_filepath = keys_directory.join(format!("{}.crt", &name));
let conf_filepath = config_directory.join(format!("{}.ovpn", &name));
let email = args.email.clone().unwrap_or(format!(
"{}@{}",
&args.name,
@@ -110,75 +168,74 @@ impl From<&Args> for AppConfig {
defaults.encoding.clone()
};
let name = args.name.clone();
let openssl = args.openssl.clone();
let template_file = args.template_file.clone();
let openssl: OpenSSLProviderArg = args.openssl.as_ref().into();
let req_days = args.days;
let keys_subdir = args.keys_dir.clone();
let config_subdir = args.config_dir.clone();
Self {
base_directory,
template_file,
openssl_default_cnf,
ca_filepath,
ca_key_filepath,
req_filepath,
key_filepath,
cert_filepath,
conf_filepath,
email,
encoding,
name,
openssl,
template_file,
req_days,
keys_subdir,
config_subdir,
..defaults
}
}
}
pub(crate) async fn is_file_exist(filepath: &PathBuf) -> bool {
let metadata = tokio::fs::metadata(&filepath).await;
if metadata.is_err() {
return false;
match tokio::fs::metadata(&filepath).await {
Ok(x) => x.is_file(),
_ => false,
}
}
if !metadata.unwrap().is_file() {
return false;
}
true
}
pub(crate) async fn read_file<'a, S, P>(filepath: P, encoding: S) -> Result<String>
pub(crate) async fn read_file<P>(filepath: P, encoding: &str) -> Result<String>
where
S: AsRef<str> + std::cmp::PartialEq<&'a str>,
P: AsRef<Path>,
{
let filepath = PathBuf::from(filepath.as_ref());
if encoding == "utf8" {
if encoding == UTF8_STR {
return Ok(fs::read_to_string(filepath).await?);
}
let enc = encoding_from_whatwg_label(encoding.as_ref()).ok_or(anyhow!("encoding not found"))?;
let enc = encoding_from_whatwg_label(encoding).ok_or(anyhow!("encoding not found"))?;
let bytes = fs::read(filepath).await?;
enc.decode(&bytes, encoding::DecoderTrap::Ignore)
.map_err(|_| anyhow!("could not read file"))
}
pub(crate) async fn write_file(filepath: &PathBuf, text: String, encoding: &str) -> Result<()> {
if encoding == "utf8" {
pub(crate) async fn write_file<P: AsRef<Path>>(
filepath: P,
text: &str,
encoding: &str,
) -> Result<()> {
if encoding == UTF8_STR {
return Ok(fs::write(filepath, text).await?);
}
let enc = encoding_from_whatwg_label(encoding).ok_or(anyhow!("encoding not found"))?;
let mut bytes = Vec::new();
enc.encode_to(&text, EncoderTrap::Ignore, &mut bytes)
.map_err(|_| anyhow!("can't encode"))?;
enc.encode_to(text, EncoderTrap::Ignore, &mut bytes)
.map_err(|e| anyhow!("can't encode: {:?}", e))?;
fs::write(filepath, bytes).await.context("can't write file")
}
pub(crate) async fn read_file_by_lines(
filepath: &PathBuf,
pub(crate) async fn read_file_by_lines<P: AsRef<Path>>(
filepath: P,
encoding: &str,
) -> Result<Box<dyn Stream<Item = String>>> {
Ok(if encoding == "utf8" {
Ok(if encoding == UTF8_STR {
let f = File::open(filepath).await?;
let reader = BufReader::new(f);
let mut lines = reader.lines();

6
src/crypto_provider.rs Normal file
View File

@@ -0,0 +1,6 @@
use anyhow::Result;
pub(crate) trait ICryptoProvider {
async fn request(&self) -> Result<()>;
async fn sign(&self) -> Result<()>;
}

View File

@@ -1,36 +1,71 @@
use anyhow::{anyhow, Result};
use anyhow::{anyhow, Context, Result};
use clap::Parser;
use common::{is_file_exist, OpenSSLProviderArg};
use crypto_provider::ICryptoProvider;
mod arcstr;
mod certs;
mod common;
mod crypto;
mod crypto_provider;
mod openssl;
mod ovpn;
mod vars;
use crate::certs::Certs;
use crate::common::{AppConfig, Args};
use crate::crypto::make_certs_provider;
use crate::vars::VarsFile;
use crate::openssl::{external::OpenSSLExternalProvider, internal::OpenSSLInternalProvider};
use crate::ovpn::OvpnConfig;
use crate::vars::{VarsFile, VarsMap};
async fn build_client_with<T: ICryptoProvider>(config: &AppConfig, provider: T) -> Result<String> {
let config_file = config.conf_filepath.clone();
let config_file_str = config_file
.to_str()
.ok_or(anyhow!("config file exist err"))?
.to_string();
if is_file_exist(&config_file).await {
return Err(anyhow!("Config file exist: {}", &config_file_str));
}
let certs = Certs::new(config, provider);
OvpnConfig::try_from_certs(certs, config)
.await?
.render()?
.to_file(&config_file)
.await?;
Ok(config_file_str)
}
async fn build_client_config(config: &AppConfig, vars: VarsMap) -> Result<String> {
match config.openssl {
OpenSSLProviderArg::Internal => {
let provider = OpenSSLInternalProvider::try_from_cfg(config, vars)?;
build_client_with(config, provider).await
}
OpenSSLProviderArg::ExternalBin(_) => {
let provider = OpenSSLExternalProvider::from_cfg(config, vars);
build_client_with(config, provider).await
}
}
}
#[tokio::main(flavor = "current_thread")]
async fn main() -> Result<()> {
let args = Args::parse();
let config = AppConfig::from(&args);
let mut vars = VarsFile::from_config(&config).await?;
vars.parse().await?;
let vars = VarsFile::from_config(&config)
.await
.context("vars from config")?
.parse()
.await
.context("parse vars")?;
println!("found vars: {}", vars.filepath.to_str().expect("fff"));
println!("loaded: {:#?}", &vars.vars);
println!("loaded: {:#?}", &vars);
let provider = make_certs_provider(&config, vars.vars.unwrap());
let certs = Certs::new(&config, provider);
let created = certs.build_client_config().await?;
let config_file = build_client_config(&config, vars).await?;
println!("created: {}", &config_file);
let result_file = certs.config_file.to_str().unwrap();
if created {
println!("created: {result_file}");
Ok(())
} else {
Err(anyhow!("file exists: {result_file}"))
}
}

View File

@@ -1,20 +1,14 @@
use anyhow::{anyhow, Result};
use std::ops::Deref;
use std::path::PathBuf;
use tokio::process::Command;
use crate::common::{is_file_exist, AppConfig, VarsMap};
use crate::common::{is_file_exist, AppConfig};
use crate::crypto_provider::ICryptoProvider;
use crate::vars::{IStrMap, VarsMap};
pub(crate) trait ICryptoProvider {
async fn request(&self) -> Result<()>;
async fn sign(&self) -> Result<()>;
}
pub(crate) fn make_certs_provider(cfg: &AppConfig, vars: VarsMap) -> impl ICryptoProvider {
OpenSSLProvider::from_cfg(cfg, vars)
}
pub(crate) struct OpenSSLProvider {
pub(crate) struct OpenSSLExternalProvider {
vars: VarsMap,
base_dir: PathBuf,
openssl_cnf: PathBuf,
@@ -26,7 +20,7 @@ pub(crate) struct OpenSSLProvider {
req_days: u32,
}
impl OpenSSLProvider {
impl OpenSSLExternalProvider {
async fn is_ca_exists(&self) -> bool {
is_file_exist(&self.ca_file).await
}
@@ -41,37 +35,33 @@ impl OpenSSLProvider {
pub(crate) fn from_cfg(cfg: &AppConfig, vars: VarsMap) -> Self {
let base_dir = PathBuf::from(&cfg.base_directory);
let keys_dir = base_dir.clone().join(cfg.keys_subdir.clone());
let name = cfg.name.clone();
let mut vars = vars;
vars.insert("KEY_CN".into(), name.clone());
vars.insert("KEY_NAME".into(), name.clone());
vars.insert("KEY_EMAIL".into(), cfg.email.clone());
vars.insert("KEY_CN", &cfg.name);
vars.insert("KEY_NAME", &cfg.name);
vars.insert("KEY_EMAIL", &cfg.email);
let ca_file = keys_dir.join(cfg.ca_filename.clone());
let req_file = keys_dir.join(format!("{}.csr", &name));
let key_file = keys_dir.join(format!("{}.key", &name));
let cert_file = keys_dir.join(format!("{}.crt", &name));
let openssl_cnf = base_dir.clone().join(
std::env::var(cfg.openssl_cnf_env.clone()).unwrap_or(cfg.openssl_default_cnf.clone()),
let openssl_cnf = base_dir.join(
std::env::var(&cfg.openssl_cnf_env)
.as_ref()
.unwrap_or(&cfg.openssl_default_cnf),
);
Self {
vars,
base_dir,
openssl_cnf,
openssl: cfg.openssl.clone(),
ca_file,
req_file,
key_file,
cert_file,
openssl: cfg.openssl.to_string(),
ca_file: cfg.ca_filepath.clone(),
req_file: cfg.req_filepath.clone(),
key_file: cfg.key_filepath.clone(),
cert_file: cfg.cert_filepath.clone(),
req_days: cfg.req_days,
}
}
}
impl ICryptoProvider for OpenSSLProvider {
impl ICryptoProvider for OpenSSLExternalProvider {
async fn request(&self) -> Result<()> {
if self.is_req_exists().await {
return Ok(());
@@ -98,7 +88,7 @@ impl ICryptoProvider for OpenSSLProvider {
"-batch",
])
.current_dir(&self.base_dir)
.envs(&self.vars)
.envs(self.vars.deref())
.status()
.await?;
@@ -134,7 +124,7 @@ impl ICryptoProvider for OpenSSLProvider {
"-batch",
])
.current_dir(&self.base_dir)
.envs(&self.vars)
.envs(self.vars.deref())
.status()
.await?;

331
src/openssl/internal.rs Normal file
View File

@@ -0,0 +1,331 @@
use anyhow::{anyhow, Context, Result};
use openssl::{
asn1::Asn1Time,
conf::{Conf, ConfMethod},
hash::MessageDigest,
pkey::{PKey, Private},
rsa::Rsa,
stack::Stack,
x509::{
extension::{ExtendedKeyUsage, KeyUsage, SubjectAlternativeName},
X509Extension, X509Name, X509NameBuilder, X509Req, X509ReqBuilder, X509,
},
};
use std::path::{Path, PathBuf};
use tokio::fs;
use crate::{
common::{is_file_exist, read_file, AppConfig},
crypto_provider::ICryptoProvider,
};
use lazy_static::lazy_static;
use std::collections::HashMap;
use crate::vars::{IStrMap, VarsMap};
use chrono::{Datelike, Days, Timelike, Utc};
lazy_static! {
static ref KEYMAP: HashMap<&'static str, &'static str> = {
let mut m = HashMap::new();
m.insert("C", "KEY_COUNTRY");
m.insert("ST", "KEY_PROVINCE");
m.insert("O", "KEY_ORG");
m.insert("OU", "KEY_OU");
m.insert("CN", "KEY_CN");
m.insert("name", "KEY_NAME");
m
};
}
trait ToPemX {
fn to_pem_x(&self) -> Result<Vec<u8>>;
}
impl ToPemX for X509 {
fn to_pem_x(&self) -> Result<Vec<u8>> {
Ok(self.to_pem()?)
}
}
impl ToPemX for X509Req {
fn to_pem_x(&self) -> Result<Vec<u8>> {
Ok(self.to_pem()?)
}
}
impl ToPemX for PKey<Private> {
fn to_pem_x(&self) -> Result<Vec<u8>> {
Ok(self.private_key_to_pem_pkcs8()?)
}
}
struct Pem<'a, T: ToPemX>(&'a T);
trait WritePem {
async fn write<T: AsRef<Path>>(&self, path: T) -> Result<()>;
}
impl<'a, P: ToPemX> WritePem for Pem<'a, P> {
async fn write<T: AsRef<Path>>(&self, path: T) -> Result<()> {
let pem = self.0.to_pem_x().context("to_pem()")?;
fs::write(path, pem).await.context("write pem")?;
Ok(())
}
}
fn get_time_str_x509(days: u32) -> Result<String> {
let dt = Utc::now();
let dt = dt
.checked_add_days(Days::new(days as u64))
.ok_or(anyhow!("failed to add days"))
.context("checked_add_days")?;
let year = dt.year() % 10000;
let month = dt.month() % 13;
let day = dt.day() % 32;
let hour = dt.hour() % 25;
let minute = dt.minute() % 60;
let second = dt.second() % 60;
let s = format!("{year:04}{month:02}{day:02}{hour:02}{minute:02}{second:02}Z");
Ok(s)
}
pub(crate) struct OpenSSLInternalProvider {
vars: VarsMap,
ca_file: PathBuf,
ca_key_file: PathBuf,
req_file: PathBuf,
key_file: PathBuf,
cert_file: PathBuf,
req_days: u32,
key_size: u32,
encoding: String,
}
impl OpenSSLInternalProvider {
async fn is_ca_exists(&self) -> bool {
is_file_exist(&self.ca_file).await
}
async fn is_cert_exists(&self) -> bool {
is_file_exist(&self.cert_file).await
}
async fn is_req_exists(&self) -> bool {
is_file_exist(&self.req_file).await
}
pub(crate) fn try_from_cfg(cfg: &AppConfig, vars: VarsMap) -> Result<Self> {
let mut vars = vars;
vars.insert("KEY_CN", &cfg.name);
vars.insert("KEY_NAME", &cfg.name);
vars.insert("KEY_EMAIL", &cfg.email);
let key_size_s = vars.get(&"KEY_SIZE").unwrap_or("2048");
let key_size: u32 = key_size_s.parse().context("parse key size error")?;
let encoding = cfg.encoding.clone();
Ok(Self {
vars,
ca_file: cfg.ca_filepath.clone(),
ca_key_file: cfg.ca_key_filepath.clone(),
req_file: cfg.req_filepath.clone(),
key_file: cfg.key_filepath.clone(),
cert_file: cfg.cert_filepath.clone(),
req_days: cfg.req_days,
key_size,
encoding,
})
}
fn generate_key_pair(&self) -> Result<(Rsa<Private>, PKey<Private>)> {
let rsa = Rsa::generate(self.key_size)?;
let pkey = PKey::from_rsa(rsa.clone())?;
Ok((rsa, pkey))
}
async fn get_ca_cert(&self) -> Result<X509> {
let text = read_file(&self.ca_file, &self.encoding).await?;
Ok(X509::from_pem(text.as_bytes())?)
}
async fn get_ca_key(&self) -> Result<PKey<Private>> {
let text = read_file(&self.ca_key_file, &self.encoding).await?;
Ok(PKey::from_rsa(Rsa::private_key_from_pem(text.as_bytes())?)?)
}
async fn get_key(&self) -> Result<(Rsa<Private>, PKey<Private>)> {
let text = read_file(&self.key_file, &self.encoding).await?;
let rsa = Rsa::private_key_from_pem(text.as_bytes())?;
let pkey = PKey::from_rsa(rsa.clone())?;
Ok((rsa, pkey))
}
async fn get_req(&self) -> Result<X509Req> {
let text = read_file(&self.req_file, &self.encoding).await?;
Ok(X509Req::from_pem(text.as_bytes())?)
}
async fn ensure_key(&self) -> Result<(Rsa<Private>, PKey<Private>)> {
if is_file_exist(&self.key_file).await {
self.get_key().await
} else {
let (rsa, pkey) = self.generate_key_pair()?;
Pem(&pkey)
.write(&self.key_file)
.await
.context("key write pem")?;
Ok((rsa, pkey))
}
}
fn build_x509_name(&self) -> Result<X509Name> {
let mut name_builder =
X509NameBuilder::new().context("Failed to create X509 name builder")?;
for (&key, &var) in KEYMAP.iter() {
let value = self
.vars
.get(&var)
.ok_or(anyhow!("variable not set: {}", var))?;
name_builder.append_entry_by_text(key, value)?;
}
Ok(name_builder.build())
}
fn gen_x509_extensions(
context: &openssl::x509::X509v3Context,
vars: &VarsMap,
) -> Result<Vec<X509Extension>> {
let key_usage = KeyUsage::new()
.key_agreement()
.digital_signature()
.build()?;
let key_extended_ext = ExtendedKeyUsage::new().client_auth().build()?;
let mut san_extension = SubjectAlternativeName::new();
if let Some(name) = vars.get(&"KEY_NAME") {
san_extension.dns(name);
}
if let Some(email) = vars.get(&"KEY_EMAIL") {
san_extension.email(email);
}
let san_ext = san_extension.build(context).context("build san")?;
Ok(vec![san_ext, key_usage, key_extended_ext])
}
fn gen_x509_extensions_stack(
context: &openssl::x509::X509v3Context,
vars: &VarsMap,
) -> Result<Stack<X509Extension>> {
let mut stack = Stack::new()?;
for extension in Self::gen_x509_extensions(context, vars)?.into_iter() {
stack.push(extension).context("push ext")?;
}
Ok(stack)
}
}
impl ICryptoProvider for OpenSSLInternalProvider {
async fn request(&self) -> Result<()> {
if self.is_req_exists().await {
return Ok(());
}
if !self.is_ca_exists().await {
return Err(anyhow!(
"ca file not found: {}",
&self.ca_file.to_str().unwrap()
));
}
let (_, pkey) = self.ensure_key().await?;
let name = self.build_x509_name()?;
let conf = Conf::new(ConfMethod::default()).context("conf new")?;
// Create certificate signing request (CSR)
let mut csr_builder = X509ReqBuilder::new()?;
csr_builder.set_version(2).context("set version")?;
csr_builder.set_pubkey(&pkey).context("set pubkey")?;
csr_builder
.set_subject_name(&name)
.context("set subject name")?;
let context = csr_builder.x509v3_context(Some(&conf));
let extensions = Self::gen_x509_extensions_stack(&context, &self.vars)?;
csr_builder.add_extensions(&extensions)?;
csr_builder.sign(&pkey, MessageDigest::sha512())?;
let csr = csr_builder.build();
Pem(&csr)
.write(&self.req_file)
.await
.context("req write pem")?;
Ok(())
}
async fn sign(&self) -> Result<()> {
if self.is_cert_exists().await {
return Ok(());
}
if !self.is_ca_exists().await {
return Err(anyhow!(
"ca file not found: {}",
&self.ca_file.to_str().unwrap()
));
}
if !self.is_req_exists().await {
return Err(anyhow!(
"csr file not found: {}",
&self.req_file.to_str().unwrap()
));
}
let ca_key = self.get_ca_key().await?;
let ca_cert = self.get_ca_cert().await?;
let req = self.get_req().await?;
let pub_key = req.public_key()?;
let subject_name = req.subject_name();
let mut builder = openssl::x509::X509Builder::new().context("new builder")?;
let not_before = Asn1Time::days_from_now(0).context("days_from_now 0")?;
let na_s = get_time_str_x509(self.req_days).context("na_s get_time_str_x509")?;
let not_after = Asn1Time::from_str_x509(&na_s)
.context(format!("not_after from_str_x509: {}", &na_s))?;
builder.set_version(2).context("set version")?;
builder
.set_not_before(&not_before)
.context("set not_before")?;
builder.set_not_after(&not_after).context("set not_after")?;
builder
.set_issuer_name(ca_cert.issuer_name())
.context("set_issuer_name")?;
builder.set_pubkey(&pub_key).context("set_pubkey")?;
builder
.set_subject_name(subject_name)
.context("set_subject_name")?;
let context = builder.x509v3_context(Some(&ca_cert), None);
for extension in Self::gen_x509_extensions(&context, &self.vars)? {
builder.append_extension(extension).context("append ext")?;
}
builder
.sign(&ca_key, MessageDigest::sha512())
.context("builder.sign")?;
let cert = builder.build();
Pem(&cert)
.write(&self.cert_file)
.await
.context("cert.to_pem()")?;
Ok(())
}
}

2
src/openssl/mod.rs Normal file
View File

@@ -0,0 +1,2 @@
pub(crate) mod external;
pub(crate) mod internal;

183
src/ovpn.rs Normal file
View File

@@ -0,0 +1,183 @@
use std::path::{Path, PathBuf};
use crate::{
certs::Certs,
common::{read_file, write_file, AppConfig, DEFAULT_ENCODING},
crypto_provider::ICryptoProvider,
};
use anyhow::{anyhow, Context, Ok, Result};
use tera::Tera;
pub(crate) struct OvpnConfig {
pub(crate) name: String,
pub(crate) encoding: String,
pub(crate) email: Option<String>,
pub(crate) template: String,
pub(crate) template_filename: Option<String>,
pub(crate) ca_cert: String,
pub(crate) cert: String,
pub(crate) key: String,
}
impl From<&OvpnConfig> for tera::Context {
fn from(value: &OvpnConfig) -> Self {
let mut context = tera::Context::new();
context.insert("name", &value.name);
context.insert("encoding", &value.encoding);
context.insert("email", &value.email);
context.insert("template_filename", &value.template_filename);
context.insert("ca", &value.ca_cert);
context.insert("cert", &value.cert);
context.insert("key", &value.key);
context
}
}
#[derive(Debug, Default, Clone)]
pub(crate) struct OvpnConfigBuilder {
name: String,
encoding: Option<String>,
email: Option<String>,
template_file: PathBuf,
ca_cert_file: Option<PathBuf>,
cert_file: Option<PathBuf>,
key_file: Option<PathBuf>,
}
impl OvpnConfigBuilder {
pub(crate) fn new<P: Into<PathBuf>>(name: String, template_file: P) -> Self {
Self {
name,
template_file: template_file.into(),
..Default::default()
}
}
pub(crate) fn with_encoding(&mut self, encoding: String) -> &mut Self {
self.encoding = Some(encoding);
self
}
pub(crate) fn with_email(&mut self, email: String) -> &mut Self {
self.email = Some(email);
self
}
pub(crate) fn with_ca_cert_file<P: Into<PathBuf>>(&mut self, ca_cert_file: P) -> &mut Self {
self.ca_cert_file = Some(ca_cert_file.into());
self
}
pub(crate) fn with_cert_file<P: Into<PathBuf>>(&mut self, cert_file: P) -> &mut Self {
self.cert_file = Some(cert_file.into());
self
}
pub(crate) fn with_key_file<P: Into<PathBuf>>(&mut self, key_file: P) -> &mut Self {
self.key_file = Some(key_file.into());
self
}
pub(crate) async fn build(self) -> Result<OvpnConfig> {
let name = self.name;
let encoding = self.encoding.unwrap_or(DEFAULT_ENCODING.into());
let template = read_file(&self.template_file, &encoding)
.await
.context("template file read error")?;
let ca_cert_file = self.ca_cert_file.ok_or(anyhow!("ca cert file not set"))?;
let cert_file = self.cert_file.ok_or(anyhow!("cert file not set"))?;
let key_file = self.key_file.ok_or(anyhow!("key file not set"))?;
let ca_cert = read_file(ca_cert_file, &encoding)
.await
.context("ca cert file read error")?
.trim_end()
.to_string();
let cert = read_file(cert_file, &encoding)
.await
.context("cert file read error")?
.trim_end()
.to_string();
let key = read_file(key_file, &encoding)
.await
.context("key file read error")?
.trim_end()
.to_string();
let template_filename = self
.template_file
.file_name()
.ok_or(anyhow!("template_file filename"))?
.to_str()
.ok_or(anyhow!("template_file to_str"))?
.to_string();
Ok(OvpnConfig {
name,
encoding,
email: self.email,
template,
ca_cert,
cert,
key,
template_filename: Some(template_filename),
})
}
}
pub(crate) struct OvpnConfigRendered {
content: String,
encoding: String,
}
impl OvpnConfigRendered {
pub async fn to_file<P: AsRef<Path>>(&self, filepath: P) -> Result<()> {
write_file(filepath, &self.content, &self.encoding)
.await
.context("ovpn config file write error")
}
}
impl TryFrom<OvpnConfig> for OvpnConfigRendered {
type Error = anyhow::Error;
fn try_from(value: OvpnConfig) -> Result<Self> {
let mut tera = Tera::default();
tera.add_raw_template("template", &value.template)
.context("raw template add error")?;
let context: tera::Context = (&value).into();
let content = tera
.render("template", &context)
.context("config render error")?;
Ok(OvpnConfigRendered {
content,
encoding: value.encoding,
})
}
}
impl OvpnConfig {
pub(crate) fn render(self) -> Result<OvpnConfigRendered> {
OvpnConfigRendered::try_from(self)
}
pub(crate) async fn try_from_certs<T: ICryptoProvider>(
certs: Certs<T>,
config: &AppConfig,
) -> Result<Self> {
let base_dir = PathBuf::from(&config.base_directory);
let template_file = base_dir.join(&config.template_file);
certs.build_all().await.context("certs build all error")?;
let mut builder = OvpnConfigBuilder::new(config.name.clone(), template_file);
builder
.with_encoding(config.encoding.clone())
.with_ca_cert_file(certs.ca_file.clone())
.with_email(config.email.clone())
.with_cert_file(certs.cert_file.clone())
.with_key_file(certs.key_file.clone());
Ok(builder.build().await.context("ovpn config build error")?)
}
}

View File

@@ -3,38 +3,84 @@ use regex::Regex;
use std::{path::PathBuf, pin::Pin};
use tokio::pin;
use crate::arcstr::ArcStr;
use crate::common::{read_file_by_lines, AppConfig};
use futures_util::stream::StreamExt;
use std::collections::BTreeMap;
use std::fmt::{Debug, Formatter};
use std::ops::Deref;
use std::path::Path;
use crate::common::{read_file_by_lines, AppConfig, VarsMap};
pub(crate) type ArcVarsMap = BTreeMap<ArcStr, ArcStr>;
#[derive(Clone, PartialEq, Ord, PartialOrd, Eq)]
pub(crate) struct VarsMap(ArcVarsMap);
pub(crate) trait IStrMap {
fn get<S: AsRef<str> + Clone>(&self, key: &S) -> Option<&str>;
fn insert<S: AsRef<str> + Clone>(&mut self, key: S, value: S) -> Option<S>;
}
impl IStrMap for VarsMap {
fn get<S: AsRef<str> + Clone>(&self, key: &S) -> Option<&str> {
self.0.get(&ArcStr::from(key.as_ref())).map(|x| x.as_str())
}
fn insert<S: AsRef<str> + Clone>(&mut self, key: S, value: S) -> Option<S> {
let key = ArcStr::from(key.as_ref());
let value2 = ArcStr::from(value.clone().as_ref());
self.0.insert(key, value2).and(Some(value))
}
}
impl Deref for VarsMap {
type Target = BTreeMap<ArcStr, ArcStr>;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl Debug for VarsMap {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
let debug_map: BTreeMap<_, _> =
self.iter().map(|(k, v)| (k.as_str(), v.as_str())).collect();
debug_map.fmt(f)
}
}
pub(crate) struct VarsFile {
pub(crate) filepath: PathBuf,
pub(crate) vars: Option<VarsMap>,
pub(crate) encoding: String,
pub(crate) filepath: ArcStr,
pub(crate) encoding: ArcStr,
}
impl VarsFile {
async fn from_file(filepath: &PathBuf, encoding: String) -> Result<Self> {
let metadata = tokio::fs::metadata(&filepath).await.context(format!(
"file not found {}",
filepath.to_str().expect("str")
))?;
async fn from_file<P>(filepath: P, encoding: ArcStr) -> Result<Self>
where
P: AsRef<Path>,
{
let filepath_str = filepath
.as_ref()
.to_str()
.ok_or(anyhow!("filepath to str"))?;
let metadata = tokio::fs::metadata(filepath_str)
.await
.context(format!("config file not found: {}", &filepath_str))?;
if !metadata.is_file() {
Err(anyhow!("{} is not a file", filepath.to_str().expect("str")))?
Err(anyhow!("config is not a file {}", &filepath_str))?;
}
Ok(VarsFile {
filepath: filepath.to_path_buf(),
vars: None,
filepath: ArcStr::from(filepath_str),
encoding,
})
}
async fn from_dir(dir: PathBuf, encoding: String) -> Result<Self> {
let filepath = dir.join("vars");
let err_context = format!(
"vars or vars.bat file not found in {}",
dir.to_str().expect("str")
);
async fn from_dir<P>(dir: P, encoding: ArcStr) -> Result<Self>
where
P: AsRef<Path>,
{
let filepath = dir.as_ref().join("vars");
let err_context = "vars or vars.bat file not found";
match Self::from_file(&filepath, encoding.clone()).await {
Ok(res) => Ok(res),
@@ -47,50 +93,38 @@ impl VarsFile {
pub(crate) async fn from_config(config: &AppConfig) -> Result<Self> {
Self::from_dir(
PathBuf::from(&config.base_directory),
config.encoding.clone(),
ArcStr::from(config.encoding.as_str()),
)
.await
}
pub(crate) async fn parse(&mut self) -> Result<()> {
let mut result = VarsMap::new();
let lines = read_file_by_lines(&self.filepath, &self.encoding).await?;
pub(crate) async fn parse(&self) -> Result<VarsMap> {
let mut result = ArcVarsMap::new();
result.insert("__file__".into(), self.filepath.clone());
result.insert("__encoding__".into(), self.encoding.clone());
let lines = read_file_by_lines(&self.filepath.as_path(), &self.encoding)
.await
.context("vars read error")?;
let lines = Pin::from(lines);
pin!(lines);
let re_v2 =
Regex::new(r#"^(export|set)\s\b(?P<key>[\w\d_]+)\b=\s?"?(?P<value>[^\#]+?)"?$"#)
.context("regex v2")?;
let re_v3 = Regex::new(r"^set_var\s(?P<key1>[\w\d_]+)\s+(?P<value1>[^\#]+?)$")
.context("regex v3")?;
let re_v2 = Regex::new(r#"^(export|set)\s\b(?P<key>[\w\d_]+)\b=\s?"?(?P<value>[^#]+?)"?$"#)
.context("regex v2")
.context("re_v2 error")?;
let re_v3 = Regex::new(r"^set_var\s(?P<key1>[\w\d_]+)\s+(?P<value1>[^#]+?)$")
.context("regex v3")
.context("re_v3 error")?;
while let Some(line) = lines.next().await {
if let Some(caps) = re_v2.captures(line.as_str()) {
result.insert(caps["key"].to_string(), caps["value"].to_string());
for re in [&re_v2, &re_v3].iter() {
if let Some(caps) = re.captures(line.as_str()) {
let [key, value] = [&caps["key"], &caps["value"]].map(ArcStr::from);
result.insert(key, value);
continue;
}
if let Some(caps) = re_v3.captures(line.as_str()) {
result.insert(caps["key"].to_string(), caps["value"].to_string());
};
}
self.vars = Some(result);
Ok(())
}
#[allow(dead_code)]
fn apply(&self) -> Result<()> {
if let Some(vars) = self.vars.clone() {
for (key, value) in vars.iter() {
unsafe {
std::env::set_var(key, value);
}
}
} else {
Err(anyhow!("vars not parsed"))?
}
Ok(())
Ok(VarsMap(result))
}
}