Merge pull request #3054 from TheBlueMatt/2024-04-fuzz-bolt11

[rust-lightning] / lightning-invoice / src / de.rs

diff --git a/lightning-invoice/src/de.rs b/lightning-invoice/src/de.rs
index 674518272d01c72d92e9a535d8e066d9077b5fc2..381c7b645f9cc5ae765809f84a469e984c850ea4 100644 (file)
--- a/lightning-invoice/src/de.rs
+++ b/lightning-invoice/src/de.rs
@@ -43,7 +43,11 @@ mod hrp_sm {
        }
 
        impl States {
-               fn next_state(&self, read_symbol: char) -> Result<States, super::Bolt11ParseError> {
+               fn next_state(&self, read_byte: u8) -> Result<States, super::Bolt11ParseError> {
+                       let read_symbol = match char::from_u32(read_byte.into()) {
+                               Some(symb) if symb.is_ascii() => symb,
+                               _ => return Err(super::Bolt11ParseError::MalformedHRP),
+                       };
                        match *self {
                                States::Start => {
                                        if read_symbol == 'l' {
@@ -119,7 +123,7 @@ mod hrp_sm {
                        *range = Some(new_range);
                }
 
-               fn step(&mut self, c: char) -> Result<(), super::Bolt11ParseError> {
+               fn step(&mut self, c: u8) -> Result<(), super::Bolt11ParseError> {
                        let next_state = self.state.next_state(c)?;
                        match next_state {
                                States::ParseCurrencyPrefix => {
@@ -158,7 +162,7 @@ mod hrp_sm {
 
        pub fn parse_hrp(input: &str) -> Result<(&str, &str, &str), super::Bolt11ParseError> {
                let mut sm = StateMachine::new();
-               for c in input.chars() {
+               for c in input.bytes() {
                        sm.step(c)?;
                }