add tracing and expand consts definition

Author: tgauth

sshdconfig/Cargo.lock

@@ -17,6 +17,7 @@ lto = true
 atty = { version = "0.2" }
 chrono = { version = "0.4" }
 clap = { version = "4.5", features = ["derive"] }
+crossterm = { version = "0.27" }
 rust-i18n = { version = "3.1" }
 schemars = "0.9"
 serde = { version = "1.0", features = ["derive"] }

sshdconfig/Cargo.toml

@@ -14,32 +14,34 @@ parseInt = "Parse Integer"
 registry = "Registry"
 
 [get]
-notImplemented = "get not yet implemented for Microsoft.OpenSSH.SSHD/sshd_config"
-defaultShellMustBeString = "shell must be a string"
 defaultShellCmdOptionMustBeString = "cmdOption must be a string"
 defaultShellEscapeArgsMustBe0Or1 = "'%{input}' must be a 0 or 1"
 defaultShellEscapeArgsMustBeDWord = "escapeArguments must be a DWord"
+defaultShellMustBeString = "shell must be a string"
+notImplemented = "get not yet implemented for Microsoft.OpenSSH.SSHD/sshd_config"
 windowsOnly = "Microsoft.OpenSSH.SSHD/Windows is only applicable to Windows"
 
-[set]
-failedToParseInput = "failed to parse input as DefaultShell with error: '%{error}'"
-shellPathMustNotBeRelative = "shell path must not be relative"
-shellPathDoesNotExist = "shell path does not exist: '%{shell}'"
-
 [parser]
 failedToParse = "failed to parse: '%{input}'"
-failedToParseRoot = "failed to parse root: '%{input}'"
-invalidConfig = "invalid config: '%{input}'"
-unknownNodeType = "unknown node type: '%{node}'"
-failedToParseChildNode = "failed to parse child node: '%{input}'"
-missingValueInChildNode = "missing value in child node: '%{input}'"
-missingKeyInChildNode = "missing key in child node: '%{input}'"
 failedToParseAsArray = "value is not an array"
+failedToParseChildNode = "failed to parse child node: '%{input}'"
 failedToParseNode = "failed to parse '%{input}'"
+failedToParseRoot = "failed to parse root: '%{input}'"
+invalidConfig = "invalid config: '%{input}'"
+invalidMultiArgNode = "multi-arg node '%{input}' is not valid"
+invalidValue = "operator is an invalid value for node"
 keyNotFound = "key '%{key}' not found"
 keyNotRepeatable = "key '%{key}' is not repeatable"
-invalidValue = "operator is an invalid value for node"
+missingValueInChildNode = "missing value in child node: '%{input}'"
+missingKeyInChildNode = "missing key in child node: '%{input}'"
 unknownNode = "unknown node: '%{kind}'"
+unknownNodeType = "unknown node type: '%{node}'"
+
+[set]
+failedToParseInput = "failed to parse input as DefaultShell with error: '%{error}'"
+shellPathDoesNotExist = "shell path does not exist: '%{shell}'"
+shellPathMustNotBeRelative = "shell path must not be relative"
 
 [util]
 sshdElevation = "elevated security context required"
+tracingInitError = "Failed to initialize tracing"

sshdconfig/locales/en-us.toml

@@ -8,16 +8,19 @@ use {
     crate::metadata::windows::{DEFAULT_SHELL, DEFAULT_SHELL_CMD_OPTION, DEFAULT_SHELL_ESCAPE_ARGS, REGISTRY_PATH},
 };
 
+use rust_i18n::t;
+use tracing::debug;
+
 use crate::args::Setting;
 use crate::error::SshdConfigError;
-use rust_i18n::t;
 
 /// Invoke the get command.
 ///
 /// # Errors
 ///
 /// This function will return an error if the desired settings cannot be retrieved.
 pub fn invoke_get(setting: &Setting) -> Result<(), SshdConfigError> {
+    debug!("Get setting: {:?}", setting);
     match *setting {
         Setting::SshdConfig => Err(SshdConfigError::NotImplemented(t!("get.notImplemented").to_string())),
         Setting::WindowsGlobal => get_default_shell()
@@ -78,4 +81,4 @@ fn get_default_shell() -> Result<(), SshdConfigError> {
 #[cfg(not(windows))]
 fn get_default_shell() -> Result<(), SshdConfigError> {
     Err(SshdConfigError::InvalidInput(t!("get.windowsOnly").to_string()))
-}
+}

sshdconfig/src/get.rs

@@ -4,12 +4,15 @@
 use clap::{Parser};
 use rust_i18n::i18n;
 use schemars::schema_for;
+use std::process::exit;
+use tracing::{debug, error};
 
 use args::{Args, Command, DefaultShell, Setting};
 use export::invoke_export;
 use get::invoke_get;
 use parser::SshdConfigParser;
 use set::invoke_set;
+use util::enable_tracing;
 
 mod args;
 mod error;
@@ -22,14 +25,25 @@ mod util;
 
 i18n!("locales", fallback = "en-us");
 
+const EXIT_SUCCESS: i32 = 0;
+const EXIT_FAILURE: i32 = 1;
+
 fn main() {
+    enable_tracing();
+
     let args = Args::parse();
 
     let result = match &args.command {
-        Command::Export => invoke_export(),
-        Command::Get { setting } => invoke_get(setting),
-        Command::Set { input } => invoke_set(input),
+        Command::Export => {
+            debug!("Export command");
+            invoke_export()
+        },
+        Command::Get { setting } => {
+            debug!("Get command: setting={:?}", setting);
+            invoke_get(setting)
+        },
         Command::Schema { setting } => {
+            debug!("Schema command: setting={:?}", setting);
             let schema = match setting {
                 Setting::SshdConfig => {
                     schema_for!(SshdConfigParser)
@@ -40,11 +54,17 @@ fn main() {
             };
             println!("{}", serde_json::to_string(&schema).unwrap());
             Ok(())
-        }
+        },
+        Command::Set { input } => {
+            debug!("Set command: input={}", input);
+            invoke_set(input)
+        },
     };
 
     if let Err(e) = result {
-        eprintln!("{e}");
-        std::process::exit(1);
+        error!("{e}");
+        exit(EXIT_FAILURE);
     }
+
+    exit(EXIT_SUCCESS);
 }

sshdconfig/src/main.rs

@@ -1,42 +1,44 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 
-use clap::ValueEnum;
-
-// TODO: ensure lists are complete
-
-// keywords that can be repeated over multiple lines and should be represented as arrays
-pub const REPEATABLE_KEYWORDS: [&str; 6] = [
-    "hostkey",
-    "include",
-    "listenaddress",
-    "port",
-    "setenv",
-    "subsystem"
-];
-
-#[derive(Clone, Debug, Eq, PartialEq, ValueEnum)]
-pub enum RepeatableKeyword {
-    HostKey,
-    Include,
-    ListenAddress,
-    Port,
-    SetEnv,
-    Subsystem,
-}
-
-// keywords that can have multiple argments per line and should be represented as arrays
-// but cannot be repeated over multiple lines, as subsequent entries are ignored
-pub const MULTI_ARG_KEYWORDS: [&str; 7] = [
+// keywords that can have multiple argments per line but cannot be repeated over multiple lines,
+// as subsequent entries are ignored, should be represented as arrays
+pub const MULTI_ARG_KEYWORDS: [&str; 16] = [
+    "authenticationmethods",
+    "authorizedkeysfile",
     "casignaturealgorithms",
+    "channeltimeout",
     "ciphers",
     "hostbasedacceptedalgorithms",
     "hostkeyalgorithms",
+    "ipqos",
     "kexalgorithms",
     "macs",
+    "permitlisten",
+    "permitopen",
+    "permituserenvironment",
+    "persourcepenalties",
+    "persourcepenaltyexemptlist",
     "pubkeyacceptedalgorithms"
 ];
 
+// keywords that can be repeated over multiple lines and should be represented as arrays.
+// note that some keywords can be both multi-arg and repeatable, in which case they only need to be listed here
+pub const REPEATABLE_KEYWORDS: [&str; 12] = [
+    "acceptenv",
+    "allowgroups",
+    "allowusers",
+    "denygroups",
+    "denyusers",
+    "hostkey",
+    "include",
+    "listenaddress",
+    "match",
+    "port",
+    "setenv",
+    "subsystem"
+];
+
 #[cfg(windows)]
 pub mod windows {
     pub const REGISTRY_PATH: &str = "HKLM\\SOFTWARE\\OpenSSH";