diff --git a/src/logger/pretty_logger.rs b/src/logger/pretty_logger.rs index 7da688a..d843870 100644 --- a/src/logger/pretty_logger.rs +++ b/src/logger/pretty_logger.rs @@ -52,7 +52,7 @@ pub fn log(message: &Record) { impl PrettyLogger { pub fn init() { - if !unsafe { crate::logger::IS_DEFAULT_LOGGER } { + if unsafe { crate::logger::IS_DEFAULT_LOGGER } { crate::logger::set_logger_box(Box::new(PrettyLogger { output: None })); } } diff --git a/ush-obfuscate/src/lib.rs b/ush-obfuscate/src/lib.rs index e382eb1..cb37a22 100644 --- a/ush-obfuscate/src/lib.rs +++ b/ush-obfuscate/src/lib.rs @@ -48,7 +48,7 @@ pub fn file_symbol(_input: TokenStream) -> TokenStream { // Return as a string literal let output = quote! { - obfuscate::symbol!(#concatted) + obfuscate::sym!(#concatted) }; output.into() } diff --git a/ush-payload/src/main.rs b/ush-payload/src/main.rs index 688ae83..e4f097c 100644 --- a/ush-payload/src/main.rs +++ b/ush-payload/src/main.rs @@ -1,11 +1,13 @@ #![macro_use] extern crate unshell; -use unshell::{info, obfuscate::sym, tree::Tree}; +use unshell::{info, logger::PrettyLogger, obfuscate::sym, tree::Tree}; fn main() { + PrettyLogger::init(); + let mut manager = Tree::new(); - manager.init_logger(); + // manager.init_logger(); println!("{}", sym!("TEST"));