Skip to content

Commit 803a41e

Browse files
committed
Fix padding and adjust comments
1 parent 4c5fb81 commit 803a41e

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

aes-ctr-enc-dec/src/main.rs

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -99,8 +99,10 @@ fn main() -> anyhow::Result<()> {
9999
)?;
100100

101101
let block_size = cipher.block_size();
102-
let padded_size = (input.len() + 2 * block_size - 1) / block_size * block_size;
103-
102+
// Mbed TLS requires the output buffer to be at least
103+
// `ilen + block_size` long.
104+
// Cf. the documentation of `mbedtls_cipher_update()`
105+
let padded_size = input.len() + block_size;
104106
output.resize(padded_size, 0);
105107

106108
cipher
@@ -134,7 +136,10 @@ fn main() -> anyhow::Result<()> {
134136
)?;
135137

136138
let block_size = cipher.block_size();
137-
let padded_size = (input.len() + 2 * block_size - 1) / block_size * block_size;
139+
// Mbed TLS requires the output buffer to be at least
140+
// `ilen + block_size` long.
141+
// Cf. the documentation of `mbedtls_cipher_update()`
142+
let padded_size = input.len() + block_size;
138143
output.resize(padded_size, 0);
139144

140145
cipher

0 commit comments

Comments
 (0)