feat: sanitize and set node alias

What this commit does:
Implements a method `set_node_alias` on NodeBuilder to allow
callers customize/set the value of the node alias. This method
sanitizes the user-provided alias by ensuring the following:
 + Node alias is UTF-8-encoded String
 + Node alias is non-empty
 + Node alias cannot exceed 32 bytes
 + Node alias is only valid up to the first null byte. Every
   character after the null byte is discraded

Additionally, a test case is provided to cover sanitizing
empty node alias, as well as an alias with emojis (copied and
modified from rust-lightning) and a sandwiched null byte.

Author: enigbe

src/builder.rs

@@ -109,7 +109,7 @@ impl Default for LiquiditySourceConfig {
 /// An error encountered during building a [`Node`].
 ///
 /// [`Node`]: crate::Node
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, PartialEq)]
 pub enum BuildError {
 	/// The given seed bytes are invalid, e.g., have invalid length.
 	InvalidSeedBytes,
@@ -139,6 +139,8 @@ pub enum BuildError {
 	WalletSetupFailed,
 	/// We failed to setup the logger.
 	LoggerSetupFailed,
+	/// The provided alias is invalid
+	InvalidNodeAlias(String),
 }
 
 impl fmt::Display for BuildError {
@@ -159,6 +161,9 @@ impl fmt::Display for BuildError {
 			Self::KVStoreSetupFailed => write!(f, "Failed to setup KVStore."),
 			Self::WalletSetupFailed => write!(f, "Failed to setup onchain wallet."),
 			Self::LoggerSetupFailed => write!(f, "Failed to setup the logger."),
+			Self::InvalidNodeAlias(ref reason) => {
+				write!(f, "Given node alias is invalid: {}", reason)
+			},
 		}
 	}
 }
@@ -309,6 +314,17 @@ impl NodeBuilder {
 		self
 	}
 
+	/// Sets the alias the [`Node`] will use in its announcement. The provided
+	/// alias must be a valid UTF-8 string.
+	pub fn set_node_alias<T: Into<String>>(
+		&mut self, node_alias: T,
+	) -> Result<&mut Self, BuildError> {
+		let node_alias = sanitize_alias(node_alias).map_err(|e| e)?;
+
+		self.config.node_alias = Some(node_alias);
+		Ok(self)
+	}
+
 	/// Builds a [`Node`] instance with a [`SqliteStore`] backend and according to the options
 	/// previously configured.
 	pub fn build(&self) -> Result<Node, BuildError> {
@@ -1050,3 +1066,68 @@ fn seed_bytes_from_config(
 		},
 	}
 }
+
+/// Sanitize the user-provided node alias to ensure that it is a valid protocol-specified UTF-8 string.
+fn sanitize_alias<T: Into<String>>(node_alias: T) -> Result<String, BuildError> {
+	// Alias is convertible into UTF-8 encoded string
+	let node_alias: String = node_alias.into();
+	let alias = node_alias.trim();
+
+	// Alias is non-empty
+	if alias.is_empty() {
+		return Err(BuildError::InvalidNodeAlias("Node alias cannot be empty.".to_string()));
+	}
+
+	// Alias valid up to first null byte
+	let first_null = alias.as_bytes().iter().position(|b| *b == 0).unwrap_or(alias.len());
+	let actual_alias = alias.split_at(first_null).0;
+
+	// Alias must be 32-bytes long or less
+	if actual_alias.as_bytes().len() > 32 {
+		return Err(BuildError::InvalidNodeAlias("Node alias cannot exceed 32 bytes.".to_string()));
+	}
+
+	Ok(actual_alias.to_string())
+}
+
+#[cfg(test)]
+mod tests {
+	use crate::{BuildError, Node};
+
+	use super::NodeBuilder;
+
+	fn create_node_with_alias<T: Into<String>>(alias: T) -> Result<Node, BuildError> {
+		NodeBuilder::new().set_node_alias(&alias.into())?.build()
+	}
+
+	#[test]
+	fn empty_node_alias() {
+		// Empty node alias
+		let alias = "";
+		let node = create_node_with_alias(alias);
+		assert_eq!(
+			node.err().unwrap(),
+			BuildError::InvalidNodeAlias("Node alias cannot be empty.".to_string())
+		);
+	}
+
+	#[test]
+	fn node_alias_with_sandwiched_null() {
+		// Alias with emojis
+		let expected_alias = "I\u{1F496}LDK-Node!";
+		let user_provided_alias = "I\u{1F496}LDK-Node!\0\u{26A1}";
+		let node = create_node_with_alias(user_provided_alias).unwrap();
+
+		assert_eq!(expected_alias, node.config().node_alias.unwrap());
+	}
+
+	#[test]
+	fn node_alias_longer_than_32_bytes() {
+		let alias = "This is a string longer than thirty-two bytes!"; // 46 bytes
+		let node = create_node_with_alias(alias);
+		assert_eq!(
+			node.err().unwrap(),
+			BuildError::InvalidNodeAlias("Node alias cannot exceed 32 bytes.".to_string())
+		);
+	}
+}

src/config.rs

@@ -163,6 +163,11 @@ pub struct Config {
 	/// **Note:** If unset, default parameters will be used, and you will be able to override the
 	/// parameters on a per-payment basis in the corresponding method calls.
 	pub sending_parameters: Option<SendingParameters>,
+	/// The node alias to be used in announcements.
+	///
+	/// **Note**: This is required if, alongside a valid public socket address, node announcements
+	/// are to be broadcast.
+	pub node_alias: Option<String>,
 }
 
 impl Default for Config {
@@ -180,6 +185,7 @@ impl Default for Config {
 			log_level: DEFAULT_LOG_LEVEL,
 			anchor_channels_config: Some(AnchorChannelsConfig::default()),
 			sending_parameters: None,
+			node_alias: None,
 		}
 	}
 }