Skip to content

Commit a058029

Browse files
committed
chore: Coding style
1 parent 9fd4520 commit a058029

File tree

1 file changed

+8
-11
lines changed

1 file changed

+8
-11
lines changed

dstack-util/src/system_setup.rs

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -958,17 +958,14 @@ impl Stage1<'_> {
958958
macro_rules! const_pad {
959959
($s:expr, $len:expr) => {
960960
const {
961+
assert!($s.len() <= $len, "The s is too long");
961962
let mut padded: [u8; $len] = [0; $len];
962-
if $s.len() > $len {
963-
padded
964-
} else {
965-
let mut i = 0;
966-
while i < $s.len() {
967-
padded[i] = $s[i];
968-
i += 1;
969-
}
970-
padded
963+
let mut i = 0;
964+
while i < $s.len() {
965+
padded[i] = $s[i];
966+
i += 1;
971967
}
968+
padded
972969
}
973970
};
974971
}
@@ -982,7 +979,7 @@ fn validate_luks2_headers(mut reader: impl std::io::Read) -> Result<()> {
982979
}
983980

984981
fn validate_single_luks2_header(mut reader: impl std::io::Read, hdr_ind: u64) -> Result<()> {
985-
let mut hdr_data = vec![0; 4096];
982+
let mut hdr_data = vec![0u8; 4096];
986983
reader
987984
.read_exact(&mut hdr_data)
988985
.context("Failed to read LUKS header")?;
@@ -1033,7 +1030,7 @@ fn validate_single_luks2_header(mut reader: impl std::io::Read, hdr_ind: u64) ->
10331030

10341031
// Check JSON
10351032
let json_size = hdr_size - 4096;
1036-
let mut jsn_data = vec![0; json_size as usize];
1033+
let mut jsn_data = vec![0u8; json_size as usize];
10371034
reader
10381035
.read_exact(&mut jsn_data)
10391036
.context("Failed to read LUKS JSON")?;

0 commit comments

Comments
 (0)