Skip to content

Commit

Permalink
Fix out-of-range panic in NaiveWeek::last_day
Browse files Browse the repository at this point in the history
  • Loading branch information
pitdicker committed May 19, 2023
1 parent 3d9fcd7 commit e843f69
Showing 1 changed file with 22 additions and 5 deletions.
27 changes: 22 additions & 5 deletions src/naive/date.rs
Original file line number Diff line number Diff line change
Expand Up @@ -79,10 +79,13 @@ impl NaiveWeek {
#[inline]
#[must_use]
pub fn first_day(&self) -> NaiveDate {
let start = self.start.num_days_from_monday();
let end = self.date.weekday().num_days_from_monday();
let days = if start > end { 7 - start + end } else { end - start };
self.date - Duration::days(days.into())
let start = self.start.num_days_from_monday() as i32;
let ref_day = self.date.weekday().num_days_from_monday() as i32;
// Calculate the number of days to subtract from `self.date`.
// Do not construct an intermediate date beyond `self.date`, because that may be out of
// range if `date` is close to `NaiveDate::MAX`.
let days = if start > ref_day { start - ref_day - 7 } else { start - ref_day };
self.date.diff_days(days as i64).unwrap()
}

/// Returns a date representing the last day of the week.
Expand All @@ -99,7 +102,13 @@ impl NaiveWeek {
#[inline]
#[must_use]
pub fn last_day(&self) -> NaiveDate {
self.first_day() + Duration::days(6)
let end = self.start.pred().num_days_from_monday() as i32;
let ref_day = self.date.weekday().num_days_from_monday() as i32;
// Calculate the number of days to add to `self.date`.
// Do not construct an intermediate date before `self.date` (like with `first_day()`),
// because that may be out of range if `date` is close to `NaiveDate::MIN`.
let days = if end < ref_day { end - ref_day + 7 } else { end - ref_day };
self.date.diff_days(days as i64).unwrap()
}

/// Returns a [`RangeInclusive<T>`] representing the whole week bounded by
Expand Down Expand Up @@ -3036,6 +3045,14 @@ mod tests {
}
}

#[test]
fn test_naiveweek_min_max() {
let date_max = NaiveDate::MAX;
assert!(date_max.week(Weekday::Mon).first_day() <= date_max);
let date_min = NaiveDate::MIN;
assert!(date_min.week(Weekday::Mon).last_day() >= date_min);
}

#[test]
fn test_weeks_from() {
// tests per: https://github.com/chronotope/chrono/issues/961
Expand Down

0 comments on commit e843f69

Please sign in to comment.