1use thiserror::Error;
4
5#[derive(Copy, Clone, Debug, Eq, PartialEq, Error)]
7#[error("invalid date string")]
8pub struct DateParseError;
9
10use super::approx::approxidate_careful_at;
11use super::compat::{self, time_t, tm};
12use super::tm::{
13 empty_tm, init_tm_unknown, is_date_known, local_tzoffset, match_string, maybeiso8601, nodate,
14 parse_timestamp_prefix, process_wall_clock_sec, skip_alpha, tm_to_time_t, TIMESTAMP_MAX,
15};
16
17struct TzName {
18 name: &'static str,
19 offset_hours: i32,
20 dst: i32,
21}
22
23const TIMEZONE_NAMES: &[TzName] = &[
24 TzName {
25 name: "IDLW",
26 offset_hours: -12,
27 dst: 0,
28 },
29 TzName {
30 name: "NT",
31 offset_hours: -11,
32 dst: 0,
33 },
34 TzName {
35 name: "CAT",
36 offset_hours: -10,
37 dst: 0,
38 },
39 TzName {
40 name: "HST",
41 offset_hours: -10,
42 dst: 0,
43 },
44 TzName {
45 name: "HDT",
46 offset_hours: -10,
47 dst: 1,
48 },
49 TzName {
50 name: "YST",
51 offset_hours: -9,
52 dst: 0,
53 },
54 TzName {
55 name: "YDT",
56 offset_hours: -9,
57 dst: 1,
58 },
59 TzName {
60 name: "PST",
61 offset_hours: -8,
62 dst: 0,
63 },
64 TzName {
65 name: "PDT",
66 offset_hours: -8,
67 dst: 1,
68 },
69 TzName {
70 name: "MST",
71 offset_hours: -7,
72 dst: 0,
73 },
74 TzName {
75 name: "MDT",
76 offset_hours: -7,
77 dst: 1,
78 },
79 TzName {
80 name: "CST",
81 offset_hours: -6,
82 dst: 0,
83 },
84 TzName {
85 name: "CDT",
86 offset_hours: -6,
87 dst: 1,
88 },
89 TzName {
90 name: "EST",
91 offset_hours: -5,
92 dst: 0,
93 },
94 TzName {
95 name: "EDT",
96 offset_hours: -5,
97 dst: 1,
98 },
99 TzName {
100 name: "AST",
101 offset_hours: -3,
102 dst: 0,
103 },
104 TzName {
105 name: "ADT",
106 offset_hours: -3,
107 dst: 1,
108 },
109 TzName {
110 name: "WAT",
111 offset_hours: -1,
112 dst: 0,
113 },
114 TzName {
115 name: "GMT",
116 offset_hours: 0,
117 dst: 0,
118 },
119 TzName {
120 name: "UTC",
121 offset_hours: 0,
122 dst: 0,
123 },
124 TzName {
125 name: "Z",
126 offset_hours: 0,
127 dst: 0,
128 },
129 TzName {
130 name: "WET",
131 offset_hours: 0,
132 dst: 0,
133 },
134 TzName {
135 name: "BST",
136 offset_hours: 0,
137 dst: 1,
138 },
139 TzName {
140 name: "CET",
141 offset_hours: 1,
142 dst: 0,
143 },
144 TzName {
145 name: "MET",
146 offset_hours: 1,
147 dst: 0,
148 },
149 TzName {
150 name: "MEWT",
151 offset_hours: 1,
152 dst: 0,
153 },
154 TzName {
155 name: "MEST",
156 offset_hours: 1,
157 dst: 1,
158 },
159 TzName {
160 name: "CEST",
161 offset_hours: 1,
162 dst: 1,
163 },
164 TzName {
165 name: "MESZ",
166 offset_hours: 1,
167 dst: 1,
168 },
169 TzName {
170 name: "FWT",
171 offset_hours: 1,
172 dst: 0,
173 },
174 TzName {
175 name: "FST",
176 offset_hours: 1,
177 dst: 1,
178 },
179 TzName {
180 name: "EET",
181 offset_hours: 2,
182 dst: 0,
183 },
184 TzName {
185 name: "EEST",
186 offset_hours: 2,
187 dst: 1,
188 },
189 TzName {
190 name: "WAST",
191 offset_hours: 7,
192 dst: 0,
193 },
194 TzName {
195 name: "WADT",
196 offset_hours: 7,
197 dst: 1,
198 },
199 TzName {
200 name: "CCT",
201 offset_hours: 8,
202 dst: 0,
203 },
204 TzName {
205 name: "JST",
206 offset_hours: 9,
207 dst: 0,
208 },
209 TzName {
210 name: "EAST",
211 offset_hours: 10,
212 dst: 0,
213 },
214 TzName {
215 name: "EADT",
216 offset_hours: 10,
217 dst: 1,
218 },
219 TzName {
220 name: "GST",
221 offset_hours: 10,
222 dst: 0,
223 },
224 TzName {
225 name: "NZT",
226 offset_hours: 12,
227 dst: 0,
228 },
229 TzName {
230 name: "NZST",
231 offset_hours: 12,
232 dst: 0,
233 },
234 TzName {
235 name: "NZDT",
236 offset_hours: 12,
237 dst: 1,
238 },
239 TzName {
240 name: "IDLE",
241 offset_hours: 12,
242 dst: 0,
243 },
244];
245
246const MONTH_NAME_08: &str = match core::str::from_utf8(&[65, 117, 103, 117, 115, 116]) {
247 Ok(name) => name,
248 Err(_) => panic!(),
249};
250
251pub(crate) const MONTH_NAMES: [&str; 12] = [
252 "January",
253 "February",
254 "March",
255 "April",
256 "May",
257 "June",
258 "July",
259 MONTH_NAME_08,
260 "September",
261 "October",
262 "November",
263 "December",
264];
265
266pub(crate) const WEEKDAY_NAMES: [&str; 7] = [
267 "Sundays",
268 "Mondays",
269 "Tuesdays",
270 "Wednesdays",
271 "Thursdays",
272 "Fridays",
273 "Saturdays",
274];
275
276pub fn date_string(date: u64, offset: i32) -> String {
278 let mut sign = '+';
279 let mut o = offset;
280 if o < 0 {
281 o = -o;
282 sign = '-';
283 }
284 format!("{} {}{:02}{:02}", date, sign, o / 60, o % 60)
285}
286
287pub fn parse_date(date: &str) -> Result<String, DateParseError> {
289 parse_date_at(date, process_wall_clock_sec())
290}
291
292pub fn parse_date_at(date: &str, now_sec: i64) -> Result<String, DateParseError> {
294 if let Ok((ts, off)) = parse_date_basic_at(date, now_sec) {
295 return Ok(date_string(ts, off));
296 }
297 let mut err = 0;
298 let ts = approxidate_careful_at(date, now_sec, Some(&mut err));
299 if err != 0 {
300 return Err(DateParseError);
301 }
302 let off = tz_hhmm_to_offset_minutes(local_tzoffset(ts));
303 Ok(date_string(ts, off))
304}
305
306fn tz_hhmm_to_offset_minutes(hhmm: i32) -> i32 {
307 let sign = if hhmm < 0 { -1 } else { 1 };
308 let abs = hhmm.unsigned_abs();
309 sign * ((abs / 100) * 60 + (abs % 100)) as i32
310}
311
312pub fn parse_date_basic_at(date: &str, now_sec: i64) -> Result<(u64, i32), DateParseError> {
315 let bytes = date.as_bytes();
316 let mut tm = init_tm_unknown();
317 let mut offset: i32 = -1;
318 let mut tm_gmt = 0i32;
319 let mut i = 0usize;
320
321 if bytes.first() == Some(&b'@') {
322 if let Some((ts, off)) = match_object_header_date(&bytes[1..]) {
323 return Ok((ts, off));
324 }
325 }
326
327 while i < bytes.len() {
328 let c = bytes[i];
329 if c == 0 || c == b'\n' {
330 break;
331 }
332 let mut m = 0usize;
333 if c.is_ascii_alphabetic() {
334 m = match_alpha(&bytes[i..], &mut tm, &mut offset);
335 } else if c.is_ascii_digit() {
336 m = match_digit(&bytes[i..], &mut tm, &mut offset, &mut tm_gmt, now_sec);
337 } else if (c == b'-' || c == b'+') && bytes.get(i + 1).is_some_and(|x| x.is_ascii_digit()) {
338 m = match_tz(&bytes[i..], &mut offset);
339 }
340 if m == 0 {
341 m = 1;
342 }
343 i += m;
344 }
345
346 let tts = tm_to_time_t(&tm);
347 if tts < 0 {
348 return Err(DateParseError);
349 }
350 let mut ts = tts as u64;
351
352 if offset == -1 {
353 tm.tm_isdst = -1;
354 let temp_time = unsafe { compat::mktime(&mut tm) };
355 let tt = ts as i128;
356 let tloc = temp_time as i128;
357 offset = if tt > tloc {
358 ((tt - tloc) / 60) as i32
359 } else {
360 -(((tloc - tt) / 60) as i32)
361 };
362 }
363
364 if tm_gmt == 0 {
365 if offset > 0 && (offset as i64) * 60 > ts as i64 {
366 return Err(DateParseError);
367 }
368 if offset < 0 && (-(offset as i128)) * 60 > (TIMESTAMP_MAX as i128 - ts as i128) {
369 return Err(DateParseError);
370 }
371 let ts128 = ts as i128;
373 let adj = (offset as i128) * 60;
374 let new_ts = ts128 - adj;
375 if new_ts < 0 {
376 return Err(DateParseError);
377 }
378 ts = new_ts as u64;
379 }
380
381 Ok((ts, offset))
382}
383
384fn match_object_header_date(date: &[u8]) -> Option<(u64, i32)> {
385 if date.is_empty() || !date[0].is_ascii_digit() {
386 return None;
387 }
388 let (stamp, mut rest) = parse_timestamp_prefix(date);
389 if rest >= date.len() || date[rest] != b' ' {
390 return None;
391 }
392 if stamp == u64::MAX {
393 return None;
394 }
395 rest += 1;
396 if rest >= date.len() || (date[rest] != b'+' && date[rest] != b'-') {
397 return None;
398 }
399 let sign = date[rest];
400 rest += 1;
401 if rest + 4 > date.len() {
402 return None;
403 }
404 let tz_digits = std::str::from_utf8(&date[rest..rest + 4]).ok()?;
405 let ofs_raw: i32 = tz_digits.parse().ok()?;
406 let mut ofs = (ofs_raw / 100) * 60 + (ofs_raw % 100);
407 if sign == b'-' {
408 ofs = -ofs;
409 }
410 let end = rest + 4;
411 if end < date.len() && date[end] != b'\n' && date[end] != 0 {
412 return None;
413 }
414 Some((stamp, ofs))
415}
416
417fn match_tz(date: &[u8], offp: &mut i32) -> usize {
419 if date.is_empty() || (date[0] != b'+' && date[0] != b'-') {
420 return 0;
421 }
422 let (hour_ul, n) = parse_timestamp_prefix(&date[1..]);
423 let mut end = 1 + n;
424 let mut min: i32 = 0;
425 let mut hour: i32 = hour_ul as i32;
426 if n == 4 {
427 min = hour % 100;
428 hour /= 100;
429 } else if n != 2 {
430 min = 99;
431 } else if end < date.len() && date[end] == b':' {
432 let (m2, n2) = parse_timestamp_prefix(&date[end + 1..]);
433 if n2 == 0 {
434 min = 99;
435 } else {
436 min = m2 as i32;
437 end += 1 + n2;
438 if end - 1 != 5 {
439 min = 99;
440 }
441 }
442 }
443 if min < 60 && hour < 24 {
444 let mut off = hour * 60 + min;
445 if date[0] == b'-' {
446 off = -off;
447 }
448 *offp = off;
449 }
450 end
451}
452
453fn parse_long_prefix(s: &[u8]) -> (i64, usize) {
455 if s.is_empty() {
456 return (0, 0);
457 }
458 let mut i = 0usize;
459 let neg = s[0] == b'-';
460 if s[0] == b'+' || s[0] == b'-' {
461 i = 1;
462 }
463 let start = i;
464 while i < s.len() && s[i].is_ascii_digit() {
465 i += 1;
466 }
467 if i == start {
468 return (0, 0);
469 }
470 let Ok(slice) = std::str::from_utf8(&s[start..i]) else {
471 return (0, 0);
472 };
473 let Ok(v) = slice.parse::<i64>() else {
474 return (0, 0);
475 };
476 let v = if neg { -v } else { v };
477 (v, i)
478}
479
480fn parse_uint_suffix(s: &[u8]) -> (u64, usize) {
481 parse_timestamp_prefix(s)
482}
483
484fn set_date(year: i32, month: i32, day: i32, now_tm: Option<&tm>, now: i64, tm: &mut tm) -> i32 {
486 if !(month > 0 && month < 13 && day > 0 && day < 32) {
487 return -1;
488 }
489 let Some(nt) = now_tm else {
490 tm.tm_mon = month - 1;
491 tm.tm_mday = day;
492 if year == -1 {
493 return 1;
494 }
495 if (1970..2100).contains(&year) {
496 tm.tm_year = year - 1900;
497 } else if (70..100).contains(&year) {
498 tm.tm_year = year;
499 } else if year < 38 {
500 tm.tm_year = year + 100;
501 } else {
502 return -1;
503 }
504 return 0;
505 };
506 let mut check = *tm;
507 check.tm_mon = month - 1;
508 check.tm_mday = day;
509 if year == -1 {
510 check.tm_year = nt.tm_year;
511 } else if (1970..2100).contains(&year) {
512 check.tm_year = year - 1900;
513 } else if (70..100).contains(&year) {
514 check.tm_year = year;
515 } else if year < 38 {
516 check.tm_year = year + 100;
517 } else {
518 return -1;
519 }
520 let specified = tm_to_time_t(&check);
521 if specified >= 0 && now + 10 * 24 * 3600 < specified {
522 return -1;
523 }
524 tm.tm_mon = check.tm_mon;
525 tm.tm_mday = check.tm_mday;
526 if year != -1 {
527 tm.tm_year = check.tm_year;
528 }
529 0
530}
531
532fn set_time(hour: i64, minute: i64, second: i64, tm: &mut tm) -> i32 {
533 if (0..=24).contains(&hour) && (0..60).contains(&minute) && (0..=60).contains(&second) {
534 tm.tm_hour = hour as i32;
535 tm.tm_min = minute as i32;
536 tm.tm_sec = second as i32;
537 0
538 } else {
539 -1
540 }
541}
542
543pub(crate) fn match_multi_number(
545 num: u64,
546 date: &[u8],
547 sep_i: usize,
548 tm: &mut tm,
549 now_in: i64,
550) -> usize {
551 let Some(&c) = date.get(sep_i) else {
552 return 0;
553 };
554 if !matches!(c, b':' | b'-' | b'/' | b'.') {
555 return 0;
556 }
557
558 let (num2, n2) = parse_long_prefix(&date[sep_i + 1..]);
559 if n2 == 0 {
560 return 0;
561 }
562 let mut pos = sep_i + 1 + n2;
563 let mut num3: i64 = -1;
564 if pos < date.len() && date[pos] == c && pos + 1 < date.len() && date[pos + 1].is_ascii_digit()
565 {
566 let (n3, rel) = parse_long_prefix(&date[pos + 1..]);
567 num3 = n3;
568 pos += 1 + rel;
569 }
570
571 match c {
572 b':' => {
573 let mut n3 = num3;
574 if n3 < 0 {
575 n3 = 0;
576 }
577 if set_time(num as i64, num2, n3, tm) == 0 {
578 if pos < date.len()
579 && date[pos] == b'.'
580 && pos + 1 < date.len()
581 && date[pos + 1].is_ascii_digit()
582 && is_date_known(tm)
583 {
584 let (_, rel) = parse_long_prefix(&date[pos + 1..]);
585 pos += 1 + rel;
586 }
587 } else {
588 return 0;
589 }
590 }
591 b'-' | b'/' | b'.' => {
592 let now = if now_in == 0 {
593 process_wall_clock_sec()
594 } else {
595 now_in
596 };
597 let mut now_tm = empty_tm();
598 let refuse_future: Option<&tm> = if compat::gmtime(now as time_t, &mut now_tm) {
599 Some(&now_tm)
600 } else {
601 None
602 };
603
604 let y = num as i32;
605 let m = num2 as i32;
606 let d = if num3 < 0 { 0 } else { num3 as i32 };
607
608 if num > 70 {
609 if set_date(y, m, d, None, now, tm) == 0 {
610 return pos;
611 }
612 if set_date(y, d, m, None, now, tm) == 0 {
613 return pos;
614 }
615 }
616 if c != b'.' && set_date(d, y, m, refuse_future, now, tm) == 0 {
617 return pos;
618 }
619 if set_date(d, m, y, refuse_future, now, tm) == 0 {
620 return pos;
621 }
622 if c == b'.' && set_date(d, y, m, refuse_future, now, tm) == 0 {
623 return pos;
624 }
625 return 0;
626 }
627 _ => return 0,
628 }
629 pos
630}
631
632fn match_alpha(date: &[u8], tm: &mut tm, offset: &mut i32) -> usize {
633 for (i, name) in MONTH_NAMES.iter().enumerate() {
634 let m = match_string(date, name);
635 if m >= 3 {
636 tm.tm_mon = i as i32;
637 return m;
638 }
639 }
640
641 for (i, name) in WEEKDAY_NAMES.iter().enumerate() {
642 let m = match_string(date, name);
643 if m >= 3 {
644 tm.tm_wday = i as i32;
645 return m;
646 }
647 }
648
649 for tz in TIMEZONE_NAMES {
650 let m = match_string(date, tz.name);
651 if m >= 3 || m == tz.name.len() {
652 let off = tz.offset_hours + tz.dst;
653 if *offset == -1 {
654 *offset = 60 * off;
655 }
656 return m;
657 }
658 }
659
660 if match_string(date, "PM") == 2 {
661 tm.tm_hour = (tm.tm_hour % 12) + 12;
662 return 2;
663 }
664
665 if match_string(date, "AM") == 2 {
666 tm.tm_hour %= 12;
667 return 2;
668 }
669
670 if date.first() == Some(&b'T')
671 && date.get(1).is_some_and(|b| b.is_ascii_digit())
672 && tm.tm_hour == -1
673 {
674 tm.tm_min = 0;
675 tm.tm_sec = 0;
676 return 1;
677 }
678
679 skip_alpha(date)
680}
681
682fn match_digit(
683 date: &[u8],
684 tm: &mut tm,
685 offset: &mut i32,
686 tm_gmt: &mut i32,
687 now_sec: i64,
688) -> usize {
689 let (num, n) = parse_timestamp_prefix(date);
690 if n == 0 {
691 return 0;
692 }
693 let end = n;
694
695 if num >= 100_000_000 && nodate(tm) && compat::gmtime(num as time_t, tm) {
696 *tm_gmt = 1;
697 return end;
698 }
699
700 if let Some(&sep) = date.get(end) {
701 if matches!(sep, b':' | b'.' | b'/' | b'-')
702 && date.get(end + 1).is_some_and(|b| b.is_ascii_digit())
703 {
704 let m = match_multi_number(num, date, end, tm, now_sec);
705 if m != 0 {
706 return m;
707 }
708 }
709 }
710
711 let mut n_digits = 0usize;
712 loop {
713 n_digits += 1;
714 if n_digits >= date.len() || !date[n_digits].is_ascii_digit() {
715 break;
716 }
717 }
718
719 if n_digits == 8 || n_digits == 6 {
720 let num1 = (num / 10000) as i32;
721 let num2 = ((num % 10000) / 100) as i32;
722 let num3 = (num % 100) as i32;
723 if n_digits == 8 {
724 let _ = set_date(num1, num2, num3, None, now_sec, tm);
725 } else if set_time(num1 as i64, num2 as i64, num3 as i64, tm) == 0
726 && date.get(end) == Some(&b'.')
727 && date.get(end + 1).is_some_and(|b| b.is_ascii_digit())
728 {
729 let (_, rel) = parse_uint_suffix(&date[end + 1..]);
730 return end + 1 + rel;
731 }
732 return end;
733 }
734
735 if maybeiso8601(tm) {
736 let mut num1 = num as u32;
737 let mut num2: u32 = 0;
738 if n_digits == 4 {
739 num1 = (num / 100) as u32;
740 num2 = (num % 100) as u32;
741 }
742 if (n_digits == 4 || n_digits == 2)
743 && !nodate(tm)
744 && set_time(num1 as i64, num2 as i64, 0, tm) == 0
745 {
746 return n_digits;
747 }
748 tm.tm_min = -1;
749 tm.tm_sec = -1;
750 }
751
752 if n_digits == 4 {
753 if num <= 1400 && *offset == -1 {
754 let minutes = (num % 100) as u32;
755 let hours = (num / 100) as u32;
756 *offset = (hours * 60 + minutes) as i32;
757 } else if num > 1900 && num < 2100 {
758 tm.tm_year = (num as i32) - 1900;
759 }
760 return n_digits;
761 }
762
763 if n_digits > 2 {
764 return n_digits;
765 }
766
767 if num > 0 && num < 32 && tm.tm_mday < 0 {
768 tm.tm_mday = num as i32;
769 return n_digits;
770 }
771
772 if n_digits == 2 && tm.tm_year < 0 {
773 if num < 10 && tm.tm_mday >= 0 {
774 tm.tm_year = (num as i32) + 100;
775 return n_digits;
776 }
777 if num >= 70 {
778 tm.tm_year = num as i32;
779 return n_digits;
780 }
781 }
782
783 if num > 0 && num < 13 && tm.tm_mon < 0 {
784 tm.tm_mon = (num as i32) - 1;
785 }
786
787 n_digits
788}