Skip to content

Commit 46cbed5

Browse files
committed
fix(bolt12): Make CurrencyCode a validated wrapper type
Convert CurrencyCode from type alias to struct with validation, ensuring ISO 4217 compliance (3 ASCII uppercase letters) at construction time.
1 parent 78fee88 commit 46cbed5

File tree

4 files changed

+145
-10
lines changed

4 files changed

+145
-10
lines changed

lightning/src/offers/invoice_request.rs

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1468,7 +1468,7 @@ mod tests {
14681468
#[cfg(c_bindings)]
14691469
use crate::offers::offer::OfferWithExplicitMetadataBuilder as OfferBuilder;
14701470
use crate::offers::offer::{
1471-
Amount, ExperimentalOfferTlvStreamRef, OfferTlvStreamRef, Quantity,
1471+
Amount, CurrencyCode, ExperimentalOfferTlvStreamRef, OfferTlvStreamRef, Quantity,
14721472
};
14731473
use crate::offers::parse::{Bolt12ParseError, Bolt12SemanticError};
14741474
use crate::offers::payer::PayerTlvStreamRef;
@@ -1997,7 +1997,10 @@ mod tests {
19971997
assert_eq!(tlv_stream.amount, None);
19981998

19991999
let invoice_request = OfferBuilder::new(recipient_pubkey())
2000-
.amount(Amount::Currency { iso4217_code: *b"USD", amount: 10 })
2000+
.amount(Amount::Currency {
2001+
iso4217_code: CurrencyCode::new(*b"USD").unwrap(),
2002+
amount: 10,
2003+
})
20012004
.build_unchecked()
20022005
.request_invoice(&expanded_key, nonce, &secp_ctx, payment_id)
20032006
.unwrap()
@@ -2372,7 +2375,10 @@ mod tests {
23722375

23732376
let invoice_request = OfferBuilder::new(recipient_pubkey())
23742377
.description("foo".to_string())
2375-
.amount(Amount::Currency { iso4217_code: *b"USD", amount: 1000 })
2378+
.amount(Amount::Currency {
2379+
iso4217_code: CurrencyCode::new(*b"USD").unwrap(),
2380+
amount: 1000,
2381+
})
23762382
.build_unchecked()
23772383
.request_invoice(&expanded_key, nonce, &secp_ctx, payment_id)
23782384
.unwrap()

lightning/src/offers/merkle.rs

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,7 @@ mod tests {
287287
use crate::ln::inbound_payment::ExpandedKey;
288288
use crate::offers::invoice_request::{InvoiceRequest, UnsignedInvoiceRequest};
289289
use crate::offers::nonce::Nonce;
290-
use crate::offers::offer::{Amount, OfferBuilder};
290+
use crate::offers::offer::{Amount, CurrencyCode, OfferBuilder};
291291
use crate::offers::parse::Bech32Encode;
292292
use crate::offers::signer::Metadata;
293293
use crate::offers::test_utils::recipient_pubkey;
@@ -355,7 +355,10 @@ mod tests {
355355
// BOLT 12 test vectors
356356
let invoice_request = OfferBuilder::new(recipient_pubkey)
357357
.description("A Mathematical Treatise".into())
358-
.amount(Amount::Currency { iso4217_code: *b"USD", amount: 100 })
358+
.amount(Amount::Currency {
359+
iso4217_code: CurrencyCode::new(*b"USD").unwrap(),
360+
amount: 100,
361+
})
359362
.build_unchecked()
360363
// Override the payer metadata and signing pubkey to match the test vectors
361364
.request_invoice(&expanded_key, nonce, &secp_ctx, payment_id)

lightning/src/offers/offer.rs

Lines changed: 129 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -999,7 +999,9 @@ impl OfferContents {
999999
let (currency, amount) = match &self.amount {
10001000
None => (None, None),
10011001
Some(Amount::Bitcoin { amount_msats }) => (None, Some(*amount_msats)),
1002-
Some(Amount::Currency { iso4217_code, amount }) => (Some(iso4217_code), Some(*amount)),
1002+
Some(Amount::Currency { iso4217_code, amount }) => {
1003+
(Some(iso4217_code.as_bytes()), Some(*amount))
1004+
},
10031005
};
10041006

10051007
let features = {
@@ -1076,7 +1078,62 @@ pub enum Amount {
10761078
}
10771079

10781080
/// An ISO 4217 three-letter currency code (e.g., USD).
1079-
pub type CurrencyCode = [u8; 3];
1081+
///
1082+
/// Currency codes must be exactly 3 ASCII uppercase letters.
1083+
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
1084+
pub struct CurrencyCode([u8; 3]);
1085+
1086+
impl CurrencyCode {
1087+
/// Creates a new `CurrencyCode` from a 3-byte array.
1088+
///
1089+
/// Returns an error if the bytes are not valid UTF-8 or not all ASCII uppercase.
1090+
pub fn new(code: [u8; 3]) -> Result<Self, CurrencyCodeError> {
1091+
let currency_str =
1092+
core::str::from_utf8(&code).map_err(|_| CurrencyCodeError::InvalidUtf8)?;
1093+
1094+
if !currency_str.chars().all(|c| c.is_ascii_uppercase()) {
1095+
return Err(CurrencyCodeError::NotAsciiUppercase { code: currency_str.to_string() });
1096+
}
1097+
1098+
Ok(Self(code))
1099+
}
1100+
1101+
/// Returns the currency code as a byte array.
1102+
pub fn as_bytes(&self) -> &[u8; 3] {
1103+
&self.0
1104+
}
1105+
1106+
/// Returns the currency code as a string slice.
1107+
pub fn as_str(&self) -> &str {
1108+
unsafe { core::str::from_utf8_unchecked(&self.0) }
1109+
}
1110+
}
1111+
1112+
impl FromStr for CurrencyCode {
1113+
type Err = CurrencyCodeError;
1114+
1115+
fn from_str(s: &str) -> Result<Self, Self::Err> {
1116+
if s.len() != 3 {
1117+
return Err(CurrencyCodeError::InvalidLength { actual: s.len() });
1118+
}
1119+
1120+
let mut code = [0u8; 3];
1121+
code.copy_from_slice(s.as_bytes());
1122+
Self::new(code)
1123+
}
1124+
}
1125+
1126+
impl AsRef<[u8]> for CurrencyCode {
1127+
fn as_ref(&self) -> &[u8] {
1128+
&self.0
1129+
}
1130+
}
1131+
1132+
impl core::fmt::Display for CurrencyCode {
1133+
fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
1134+
f.write_str(self.as_str())
1135+
}
1136+
}
10801137

10811138
/// Quantity of items supported by an [`Offer`].
10821139
#[derive(Clone, Copy, Debug, PartialEq)]
@@ -1115,7 +1172,7 @@ const OFFER_ISSUER_ID_TYPE: u64 = 22;
11151172
tlv_stream!(OfferTlvStream, OfferTlvStreamRef<'a>, OFFER_TYPES, {
11161173
(2, chains: (Vec<ChainHash>, WithoutLength)),
11171174
(OFFER_METADATA_TYPE, metadata: (Vec<u8>, WithoutLength)),
1118-
(6, currency: CurrencyCode),
1175+
(6, currency: [u8; 3]),
11191176
(8, amount: (u64, HighZeroBytesDroppedBigSize)),
11201177
(10, description: (String, WithoutLength)),
11211178
(12, features: (OfferFeatures, WithoutLength)),
@@ -1209,7 +1266,11 @@ impl TryFrom<FullOfferTlvStream> for OfferContents {
12091266
},
12101267
(None, Some(amount_msats)) => Some(Amount::Bitcoin { amount_msats }),
12111268
(Some(_), None) => return Err(Bolt12SemanticError::MissingAmount),
1212-
(Some(iso4217_code), Some(amount)) => Some(Amount::Currency { iso4217_code, amount }),
1269+
(Some(currency_bytes), Some(amount)) => {
1270+
let iso4217_code = CurrencyCode::new(currency_bytes)
1271+
.map_err(|_| Bolt12SemanticError::InvalidCurrencyCode)?;
1272+
Some(Amount::Currency { iso4217_code, amount })
1273+
},
12131274
};
12141275

12151276
if amount.is_some() && description.is_none() {
@@ -1256,6 +1317,37 @@ impl core::fmt::Display for Offer {
12561317
}
12571318
}
12581319

1320+
/// Errors that can occur when creating or parsing a `CurrencyCode`
1321+
#[derive(Clone, Debug, PartialEq, Eq)]
1322+
pub enum CurrencyCodeError {
1323+
/// The currency code must be exactly 3 bytes
1324+
InvalidLength {
1325+
/// The actual length of the currency code
1326+
actual: usize
1327+
},
1328+
/// The currency code contains invalid UTF-8
1329+
InvalidUtf8,
1330+
/// The currency code must be all ASCII uppercase letters
1331+
NotAsciiUppercase {
1332+
/// The actual currency code
1333+
code: String
1334+
},
1335+
}
1336+
1337+
impl core::fmt::Display for CurrencyCodeError {
1338+
fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
1339+
match self {
1340+
Self::InvalidLength { actual } => {
1341+
write!(f, "Currency code must be 3 bytes, got {}", actual)
1342+
},
1343+
Self::InvalidUtf8 => write!(f, "Currency code contains invalid UTF-8"),
1344+
Self::NotAsciiUppercase { code } => {
1345+
write!(f, "Currency code '{}' must be all ASCII uppercase", code)
1346+
},
1347+
}
1348+
}
1349+
}
1350+
12591351
#[cfg(test)]
12601352
mod tests {
12611353
#[cfg(not(c_bindings))]
@@ -1273,6 +1365,7 @@ mod tests {
12731365
use crate::ln::inbound_payment::ExpandedKey;
12741366
use crate::ln::msgs::{DecodeError, MAX_VALUE_MSAT};
12751367
use crate::offers::nonce::Nonce;
1368+
use crate::offers::offer::CurrencyCode;
12761369
use crate::offers::parse::{Bolt12ParseError, Bolt12SemanticError};
12771370
use crate::offers::test_utils::*;
12781371
use crate::types::features::OfferFeatures;
@@ -1541,7 +1634,8 @@ mod tests {
15411634
#[test]
15421635
fn builds_offer_with_amount() {
15431636
let bitcoin_amount = Amount::Bitcoin { amount_msats: 1000 };
1544-
let currency_amount = Amount::Currency { iso4217_code: *b"USD", amount: 10 };
1637+
let currency_amount =
1638+
Amount::Currency { iso4217_code: CurrencyCode::new(*b"USD").unwrap(), amount: 10 };
15451639

15461640
let offer = OfferBuilder::new(pubkey(42)).amount_msats(1000).build().unwrap();
15471641
let tlv_stream = offer.as_tlv_stream();
@@ -1820,6 +1914,36 @@ mod tests {
18201914
Bolt12ParseError::InvalidSemantics(Bolt12SemanticError::InvalidAmount)
18211915
),
18221916
}
1917+
1918+
let mut tlv_stream = offer.as_tlv_stream();
1919+
tlv_stream.0.amount = Some(1000);
1920+
tlv_stream.0.currency = Some(b"\xFF\xFE\xFD"); // invalid UTF-8 bytes
1921+
1922+
let mut encoded_offer = Vec::new();
1923+
tlv_stream.write(&mut encoded_offer).unwrap();
1924+
1925+
match Offer::try_from(encoded_offer) {
1926+
Ok(_) => panic!("expected error"),
1927+
Err(e) => assert_eq!(
1928+
e,
1929+
Bolt12ParseError::InvalidSemantics(Bolt12SemanticError::InvalidCurrencyCode)
1930+
),
1931+
}
1932+
1933+
let mut tlv_stream = offer.as_tlv_stream();
1934+
tlv_stream.0.amount = Some(1000);
1935+
tlv_stream.0.currency = Some(b"usd"); // invalid ISO 4217 code
1936+
1937+
let mut encoded_offer = Vec::new();
1938+
tlv_stream.write(&mut encoded_offer).unwrap();
1939+
1940+
match Offer::try_from(encoded_offer) {
1941+
Ok(_) => panic!("expected error"),
1942+
Err(e) => assert_eq!(
1943+
e,
1944+
Bolt12ParseError::InvalidSemantics(Bolt12SemanticError::InvalidCurrencyCode)
1945+
),
1946+
}
18231947
}
18241948

18251949
#[test]

lightning/src/offers/parse.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,8 @@ pub enum Bolt12SemanticError {
149149
MissingAmount,
150150
/// The amount exceeded the total bitcoin supply or didn't match an expected amount.
151151
InvalidAmount,
152+
/// The currency code did not contain valid ASCII uppercase letters.
153+
InvalidCurrencyCode,
152154
/// An amount was provided but was not sufficient in value.
153155
InsufficientAmount,
154156
/// An amount was provided but was not expected.

0 commit comments

Comments
 (0)