Skip to content

Commit

Permalink
chore(rust): logs adjustments
Browse files Browse the repository at this point in the history
  • Loading branch information
adrianbenavides committed Dec 18, 2024
1 parent 720aed1 commit afbc966
Show file tree
Hide file tree
Showing 9 changed files with 30 additions and 13 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ impl OckamLogFormat {
}

fn format_timestamp(&self, writer: &mut Writer<'_>) -> std::fmt::Result {
let now = chrono::Utc::now().format("%Y-%m-%d %H:%M:%S%.3f");
let now = chrono::Utc::now().format("%Y-%m-%dT%H:%M:%S%.6f");
if writer.has_ansi_escapes() {
let style = Style::new().dimmed();
write!(writer, "{}", style.prefix())?;
Expand Down
1 change: 0 additions & 1 deletion implementations/rust/ockam/ockam_api/src/logs/setup.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@ use opentelemetry::trace::TracerProvider;
use opentelemetry::{global, KeyValue};
use opentelemetry_appender_tracing::layer::OpenTelemetryTracingBridge;
use opentelemetry_otlp::WithExportConfig;
use opentelemetry_sdk as sdk;
use opentelemetry_sdk::export::logs::LogExporter;
use opentelemetry_sdk::export::trace::SpanExporter;
use opentelemetry_sdk::logs::{BatchLogProcessor, LoggerProvider};
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,6 @@ impl NodeManager {

info!(
%alias, %address, ?authorized, ?relay_address,
forwarding_route = ?relay_info.forwarding_route(),
remote_address = ?relay_info.remote_address(),
"relay created"
);
Expand Down
5 changes: 0 additions & 5 deletions implementations/rust/ockam/ockam_api/tests/logging_tracing.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,6 @@ use ockam_api::logs::{

use opentelemetry::global;
use opentelemetry::trace::Tracer;

use opentelemetry_sdk as sdk;
use sdk::testing::logs::*;
use sdk::testing::trace::*;

use opentelemetry_sdk::testing::logs::InMemoryLogsExporter;
use opentelemetry_sdk::testing::trace::InMemorySpanExporter;
use std::fs;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ impl CreateCommand {
// Set node_name so that node can isolate its data in the storage from other nodes
self.get_or_create_identity(&opts, &self.identity).await?;
let _notification_handler = if self.foreground_args.child_process {
// If enabled, the user's terminal will receive notifications
// If enabled, the user's terminal would receive notifications
// from the node after the command exited.
None
} else {
Expand Down
2 changes: 1 addition & 1 deletion implementations/rust/ockam/ockam_command/src/node/show.rs
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,7 @@ pub async fn is_node_up(
.ask_with_timeout::<(), NodeStatus>(ctx, api::query_status(), Duration::from_secs(1))
.await
{
if status.status.is_running() {
if status.process_status.is_running() {
return Ok(true);
}
}
Expand Down
2 changes: 0 additions & 2 deletions implementations/rust/ockam/ockam_command/src/relay/create.rs
Original file line number Diff line number Diff line change
Expand Up @@ -91,8 +91,6 @@ impl Command for CreateCommand {
let alias = cmd.relay_name();
let return_timing = cmd.return_timing();

// let _notification_handler = NotificationHandler::start(&opts.state, opts.terminal.clone());

let node = BackgroundNodeClient::create(ctx, &opts.state, &cmd.to).await?;
let relay_info = {
if at.starts_with(Project::CODE) && cmd.authorized.is_some() {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@ use ockam_vault::{
X25519PublicKey, X25519SecretKeyHandle, X25519_PUBLIC_KEY_LENGTH,
};
use sha2::{Digest, Sha256};
#[cfg(feature = "debugger")]
use tracing::debug;
use Status::*;

/// The number of bytes in a SHA256 digest
Expand Down Expand Up @@ -50,6 +52,9 @@ impl Handshake {

/// Encode the first message, sent from the initiator to the responder
pub(super) async fn encode_message1(&mut self, payload: &[u8]) -> Result<Vec<u8>> {
#[cfg(feature = "debugger")]
debug!("Encoding message 1");

let mut state = self.state.clone();
// output e.pubKey
let e_pub_key = self.get_public_key(state.e()?).await?;
Expand All @@ -70,6 +75,9 @@ impl Handshake {

/// Decode the first message to get the ephemeral public key sent by the initiator
pub(super) async fn decode_message1(&mut self, message1: &[u8]) -> Result<Vec<u8>> {
#[cfg(feature = "debugger")]
debug!("Decoding message 2");

if message1.len() > NOISE_MAX_MESSAGE_SIZE {
return Err(XXError::ExceededMaxMessageLen)?;
}
Expand All @@ -93,6 +101,9 @@ impl Handshake {
/// That message contains: the responder ephemeral public key + a Diffie-Hellman key +
/// an encrypted payload containing the responder identity / signature / credentials
pub(super) async fn encode_message2(&mut self, payload: &[u8]) -> Result<Vec<u8>> {
#[cfg(feature = "debugger")]
debug!("Encoding message 2");

let mut state = self.state.clone();
// output e.pubKey
let e_pub_key = self.get_public_key(state.e()?).await?;
Expand Down Expand Up @@ -126,6 +137,9 @@ impl Handshake {

/// Decode the second message sent by the responder
pub(super) async fn decode_message2(&mut self, message2: &[u8]) -> Result<Vec<u8>> {
#[cfg(feature = "debugger")]
debug!("Decoding message 2");

if message2.len() > NOISE_MAX_MESSAGE_SIZE {
return Err(XXError::ExceededMaxMessageLen)?;
}
Expand Down Expand Up @@ -166,6 +180,9 @@ impl Handshake {
/// That message contains: the initiator static public key (encrypted) + a Diffie-Hellman key +
/// an encrypted payload containing the initiator identity / signature / credentials
pub(super) async fn encode_message3(&mut self, payload: &[u8]) -> Result<Vec<u8>> {
#[cfg(feature = "debugger")]
debug!("Encoding message 3");

let mut state = self.state.clone();
// encrypt s.pubKey
let s_pub_key = self.get_public_key(state.s()?).await?;
Expand All @@ -190,6 +207,9 @@ impl Handshake {

/// Decode the third message sent by the initiator
pub(super) async fn decode_message3(&mut self, message3: &[u8]) -> Result<Vec<u8>> {
#[cfg(feature = "debugger")]
debug!("Decoding message 3");

if message3.len() > NOISE_MAX_MESSAGE_SIZE {
return Err(XXError::ExceededMaxMessageLen)?;
}
Expand Down Expand Up @@ -219,6 +239,9 @@ impl Handshake {
/// Set the final state of the state machine by creating the encryption / decryption keys
/// and return the other party identity
pub(super) async fn set_final_state(&mut self, role: Role) -> Result<()> {
#[cfg(feature = "debugger")]
debug!("Setting final state");

// k1, k2 = HKDF(ck, zerolen, 2)
let mut state = self.state.clone();
let (k1, k2) = self.compute_final_keys(&mut state).await?;
Expand All @@ -239,6 +262,9 @@ impl Handshake {

/// Return the final results of the handshake if we reached the final state
pub(super) fn get_handshake_keys(&self) -> Option<HandshakeKeys> {
#[cfg(feature = "debugger")]
debug!("Getting handshake keys");

match &self.state.status {
Ready(keys) => Some(keys.clone()),
_ => None,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ impl Context {
pub(crate) async fn receiver_next(&mut self) -> Result<Option<RelayMessage>> {
loop {
let relay_msg = if let Some(msg) = self.receiver.recv().await.map(|msg| {
trace!("{}: received new message!", self.address());
trace!(address=%self.address(), "received new message!");

// First we update the mailbox fill metrics
self.mailbox_count.fetch_sub(1, Ordering::Acquire);
Expand Down

0 comments on commit afbc966

Please sign in to comment.