From 45cc770ef4048c0c3589ddf9b853e6822d5811fc Mon Sep 17 00:00:00 2001
From: zonyitoo <zonyitoo@gmail.com>
Date: Mon, 17 Oct 2022 22:29:02 +0800
Subject: [PATCH] ssservice genkey, helper for generate encryption key

---
 Cargo.lock            |  2 ++
 Cargo.toml            |  2 ++
 bin/ssservice.rs      |  7 ++++++-
 src/service/genkey.rs | 44 +++++++++++++++++++++++++++++++++++++++++++
 src/service/mod.rs    |  1 +
 5 files changed, 55 insertions(+), 1 deletion(-)
 create mode 100644 src/service/genkey.rs

diff --git a/Cargo.lock b/Cargo.lock
index b8ea61c3..bbfe7b32 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2062,6 +2062,7 @@ dependencies = [
 name = "shadowsocks-rust"
 version = "1.15.0-alpha.9"
 dependencies = [
+ "base64",
  "build-time",
  "byte_string",
  "byteorder",
@@ -2080,6 +2081,7 @@ dependencies = [
  "mimalloc",
  "num_cpus",
  "qrcode",
+ "rand",
  "rpassword",
  "rpmalloc",
  "serde",
diff --git a/Cargo.toml b/Cargo.toml
index daf3e808..c5aa76ad 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -148,6 +148,7 @@ log4rs = { version = "1.2", optional = true }
 serde = { version = "1.0", features = ["derive"] }
 json5 = "0.4"
 thiserror = "1.0"
+base64 = "0.13"
 
 clap = { version = "4.0", features = ["wrap_help", "suggestions"] }
 cfg-if = "1"
@@ -158,6 +159,7 @@ directories = "4.0"
 xdg = "2.4"
 rpassword = "7.0"
 libc = { version = "0.2", features = ["extra_traits"] }
+rand = "0.8"
 
 futures = "0.3"
 tokio = { version = "1", features = ["rt", "signal"] }
diff --git a/bin/ssservice.rs b/bin/ssservice.rs
index b85088c9..05579740 100644
--- a/bin/ssservice.rs
+++ b/bin/ssservice.rs
@@ -10,7 +10,7 @@
 use std::{env, path::Path, process::ExitCode};
 
 use clap::Command;
-use shadowsocks_rust::service::{local, manager, server};
+use shadowsocks_rust::service::{genkey, local, manager, server};
 
 fn main() -> ExitCode {
     let app = Command::new("shadowsocks")
@@ -36,12 +36,17 @@ fn main() -> ExitCode {
         .subcommand(
             manager::define_command_line_options(Command::new("manager")).about("Shadowsocks Server Manager service"),
         )
+        .subcommand(
+            genkey::define_command_line_options(Command::new("genkey"))
+                .about("Generate shadowsocks encryption key for method"),
+        )
         .get_matches();
 
     match matches.subcommand() {
         Some(("local", matches)) => local::main(matches),
         Some(("server", matches)) => server::main(matches),
         Some(("manager", matches)) => manager::main(matches),
+        Some(("genkey", matches)) => genkey::main(matches),
         _ => unreachable!("expecting a subcommand"),
     }
 }
diff --git a/src/service/genkey.rs b/src/service/genkey.rs
new file mode 100644
index 00000000..b01fdf27
--- /dev/null
+++ b/src/service/genkey.rs
@@ -0,0 +1,44 @@
+//! Generate sufficient key for method
+
+use std::process::ExitCode;
+
+use clap::{builder::PossibleValuesParser, Arg, ArgAction, ArgMatches, Command};
+use rand::RngCore;
+
+use shadowsocks_service::shadowsocks::crypto::{available_ciphers, CipherKind};
+
+/// Defines command line options
+pub fn define_command_line_options(mut app: Command) -> Command {
+    app = app.arg(
+        Arg::new("ENCRYPT_METHOD")
+            .short('m')
+            .long("encrypt-method")
+            .num_args(1)
+            .action(ArgAction::Set)
+            .required(true)
+            .value_parser(PossibleValuesParser::new(available_ciphers()))
+            .help("Server's encryption method"),
+    );
+
+    app
+}
+
+/// Program entrance `main`
+pub fn main(matches: &ArgMatches) -> ExitCode {
+    let method = matches
+        .get_one::<String>("ENCRYPT_METHOD")
+        .map(|x| x.parse::<CipherKind>().expect("method"))
+        .expect("`method` is required");
+
+    let key_len = method.key_len();
+    if key_len > 0 {
+        let mut key = vec![0u8; key_len];
+        let mut rng = rand::thread_rng();
+        rng.fill_bytes(&mut key);
+
+        let encoded_key = base64::encode(&key);
+        println!("{}", encoded_key);
+    }
+
+    ExitCode::SUCCESS
+}
diff --git a/src/service/mod.rs b/src/service/mod.rs
index f5a4ba58..10561ec8 100644
--- a/src/service/mod.rs
+++ b/src/service/mod.rs
@@ -1,5 +1,6 @@
 //! Service launchers
 
+pub mod genkey;
 #[cfg(feature = "local")]
 pub mod local;
 #[cfg(feature = "manager")]