Check validity of string

cli/src/commands/account.rs

@@ -73,6 +73,11 @@ impl Account {
         const ITERATIONS: u128 = u16::MAX as u128;
         const ITERATIONS_STR: &str = "65,535";
 
+        // Ensure the vanity string is valid.
+        if !crate::helpers::is_valid_bech32m_string(vanity) {
+            bail!("The vanity string '{vanity}' contains invalid bech32m characters");
+        }
+
         loop {
             // Initialize a timer.
             let timer = std::time::Instant::now();

cli/src/helpers/bech32m.rs

@@ -0,0 +1,28 @@
+// Copyright (C) 2019-2022 Aleo Systems Inc.
+// This file is part of the snarkOS library.
+
+// The snarkOS library is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// The snarkOS library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with the snarkOS library. If not, see <https://www.gnu.org/licenses/>.
+
+const BECH32M_CHARSET: &[u8] = b"qpzry9x8gf2tvdw0s3jn54khce6mua7l1";
+
+/// Returns `true` if the given string is a valid bech32m string.
+pub fn is_valid_bech32m_string(s: &str) -> bool {
+    s.as_bytes().iter().all(|b| BECH32M_CHARSET.contains(b))
+}
+
+#[test]
+fn test_is_valid_bech32m_string() {
+    assert!(is_valid_bech32m_string("qpzry9x8gf2tvdw0s3jn54khce6mua7l1qpzry9x8gf2tvdw0s3jn54khce6mua7l1"));
+    assert!(!is_valid_bech32m_string("qpzry9x8gf2tvdw0s3jn54khce6mua7l1qpzry9x8gf2tvdw0s3jn54khce6mua7l2"));
+}

cli/src/helpers/mod.rs

@@ -14,6 +14,9 @@
 // You should have received a copy of the GNU General Public License
 // along with the snarkOS library. If not, see <https://www.gnu.org/licenses/>.
 
+mod bech32m;
+pub use bech32m::*;
+
 mod log_writer;
 use log_writer::*;