Make rustc_layout_scalar_valid_range attributes work for non-decimal literals

This commit is contained in:
Lukas Wirth 2023-09-29 15:30:47 +02:00
parent dfeff9f40b
commit a943b19e08

View File

@ -119,7 +119,15 @@ fn layout_scalar_valid_range(db: &dyn HirDatabase, def: AdtId) -> (Bound<u128>,
let attr = attrs.by_key(name).tt_values();
for tree in attr {
if let Some(it) = tree.token_trees.first() {
if let Ok(it) = it.to_string().parse() {
let text = it.to_string().replace('_', "");
let base = match text.as_bytes() {
[b'0', b'x', ..] => 16,
[b'0', b'o', ..] => 8,
[b'0', b'b', ..] => 2,
_ => 10,
};
if let Ok(it) = u128::from_str_radix(&text, base) {
return Bound::Included(it);
}
}