RGBCube/uutils-coreutils
1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-07-28 03:27:44 +00:00
Code Issues Activity

nl: handle line number overflow

Browse source
This commit is contained in:
Daniel Hofstetter 2023-08-15 13:07:22 +02:00
parent 15c27cd545
commit d604f709ce
2 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/uu/nl/src/nl.rs
View file

@ -344,7 +344,10 @@ fn nl<T: Read>(reader: &mut BufReader<T>, settings: &Settings) -> UResult<()> {
line
);
// update line number for the potential next line
line_no += settings.line_increment;
match line_no.checked_add(settings.line_increment) {
Some(new_line_no) => line_no = new_line_no,
None => return Err(USimpleError::new(1, "line number overflow")),
}
} else {
let spaces = " ".repeat(settings.number_width + 1);
println!("{spaces}{line}");

18
tests/by-util/test_nl.rs
View file

@ -464,3 +464,21 @@ fn test_invalid_regex_numbering() {
.stderr_contains("invalid regular expression");
}
}
#[test]
fn test_line_number_overflow() {
new_ucmd!()
.arg(format!("--starting-line-number={}", i64::MAX))
.pipe_in("a\nb")
.fails()
.stdout_is(format!("{}\ta\n", i64::MAX))
.stderr_is("nl: line number overflow\n");
new_ucmd!()
.arg(format!("--starting-line-number={}", i64::MIN))
.arg("--line-increment=-1")
.pipe_in("a\nb")
.fails()
.stdout_is(format!("{}\ta\n", i64::MIN))
.stderr_is("nl: line number overflow\n");
}