1use rucc_base::{Interner, Symbol};
36use rucc_diag::{Diagnostic, Span};
37use rucc_session::Std;
38use rucc_target::TargetInfo;
39
40use crate::keyword::{Keyword, Keywords};
41use crate::literal::{CharConstant, LiteralError, StringLiteral};
42use crate::number::{FloatConstant, IntConstant, IntError};
43use crate::remarks::Remarks;
44use crate::token::{PpToken, PpTokenKind, Punct, TokenFlags};
45
46#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
51pub enum TokenKind {
52 Keyword(Keyword),
54 Ident,
56 Int,
58 Float,
60 Char,
62 Str,
65 Punct(Punct),
67 Eof,
69}
70
71#[derive(Debug, Clone, Copy, PartialEq, Eq)]
76pub struct Token {
77 pub kind: TokenKind,
79 pub flags: TokenFlags,
82 pub value: u32,
85 pub span: Span,
87}
88
89impl Token {
90 #[inline]
92 #[must_use]
93 pub const fn is_eof(self) -> bool {
94 matches!(self.kind, TokenKind::Eof)
95 }
96
97 #[inline]
99 #[must_use]
100 pub const fn keyword(self) -> Option<Keyword> {
101 match self.kind {
102 TokenKind::Keyword(word) => Some(word),
103 _ => None,
104 }
105 }
106
107 #[inline]
109 #[must_use]
110 pub const fn punct(self) -> Option<Punct> {
111 match self.kind {
112 TokenKind::Punct(punct) => Some(punct),
113 _ => None,
114 }
115 }
116
117 #[inline]
119 #[must_use]
120 pub const fn ident(self) -> Option<Symbol> {
121 match self.kind {
122 TokenKind::Ident => Some(Symbol::from_raw(self.value)),
123 _ => None,
124 }
125 }
126}
127
128#[derive(Debug, Default)]
130pub struct Tokens {
131 pub tokens: Vec<Token>,
133 pub ints: Vec<IntConstant>,
135 pub floats: Vec<FloatConstant>,
137 pub chars: Vec<CharConstant>,
139 pub strings: Vec<StringLiteral>,
141 pub pragmas: Vec<Pragma>,
143}
144
145#[derive(Debug, Clone)]
155pub struct Pragma {
156 pub before: u32,
158 pub tokens: Vec<Token>,
160 pub span: Span,
162}
163
164impl Tokens {
165 #[must_use]
167 pub fn int(&self, token: Token) -> Option<&IntConstant> {
168 match token.kind {
169 TokenKind::Int => self.ints.get(token.value as usize),
170 _ => None,
171 }
172 }
173
174 #[must_use]
176 pub fn float(&self, token: Token) -> Option<&FloatConstant> {
177 match token.kind {
178 TokenKind::Float => self.floats.get(token.value as usize),
179 _ => None,
180 }
181 }
182
183 #[must_use]
185 pub fn character(&self, token: Token) -> Option<&CharConstant> {
186 match token.kind {
187 TokenKind::Char => self.chars.get(token.value as usize),
188 _ => None,
189 }
190 }
191
192 #[must_use]
194 pub fn string(&self, token: Token) -> Option<&StringLiteral> {
195 match token.kind {
196 TokenKind::Str => self.strings.get(token.value as usize),
197 _ => None,
198 }
199 }
200}
201
202#[derive(Debug, Clone, Copy)]
204pub struct Convert<'a> {
205 pub keywords: &'a Keywords,
207 pub interner: &'a Interner,
209 pub target: &'a TargetInfo,
211 pub std: Std,
213 pub gnu: bool,
217 pub pedantic: bool,
220}
221
222#[must_use]
228pub fn convert(pp: &[PpToken], cx: &Convert<'_>) -> (Tokens, Vec<Diagnostic>) {
229 let mut out = Tokens { tokens: Vec::with_capacity(pp.len()), ..Tokens::default() };
230 let mut diagnostics = Vec::new();
231 let mut index = 0;
232 while index < pp.len() {
233 index = one(pp, index, cx, &mut out, &mut diagnostics);
234 }
235 if out.tokens.last().is_none_or(|last| !last.is_eof()) {
236 let end =
239 out.tokens.last().map_or(Span::new(0, 0), |last| Span::new(last.span.hi, last.span.hi));
240 out.tokens.push(Token {
241 kind: TokenKind::Eof,
242 flags: TokenFlags::EMPTY,
243 value: 0,
244 span: end,
245 });
246 }
247 (out, diagnostics)
248}
249
250fn one(
256 pp: &[PpToken],
257 index: usize,
258 cx: &Convert<'_>,
259 out: &mut Tokens,
260 diagnostics: &mut Vec<Diagnostic>,
261) -> usize {
262 let token = pp[index];
263 let mut index = index + 1;
264 match token.kind {
265 PpTokenKind::Ident => out.tokens.push(identifier(token, cx)),
266 PpTokenKind::Number => {
267 out.tokens.push(number(token, cx, &mut out.ints, &mut out.floats, diagnostics));
268 }
269 PpTokenKind::CharConst => {
270 out.tokens.push(char_const(token, cx, &mut out.chars, diagnostics));
271 }
272 PpTokenKind::StringLit => {
273 let start = index - 1;
277 while pp.get(index).is_some_and(|next| next.kind == PpTokenKind::StringLit) {
278 index += 1;
279 }
280 let run = &pp[start..index];
281 out.tokens.push(string_lit(run, cx, &mut out.strings, diagnostics));
282 }
283 PpTokenKind::Punct(Punct::Hash)
287 if token.flags.has(TokenFlags::START_OF_LINE)
288 && pp.get(index).is_some_and(|next| is_pragma(*next, cx)) =>
289 {
290 index += 1;
291 let before = u32::try_from(out.tokens.len()).unwrap_or(u32::MAX);
292 let mut line = Tokens::default();
293 while pp.get(index).is_some_and(|next| {
294 !matches!(next.kind, PpTokenKind::Eof) && !next.flags.has(TokenFlags::START_OF_LINE)
295 }) {
296 index = one(pp, index, cx, out, diagnostics);
297 line.tokens.push(out.tokens.pop().expect("one token out"));
298 }
299 out.pragmas.push(Pragma { before, tokens: line.tokens, span: token.span });
300 }
301 PpTokenKind::Punct(punct) => out.tokens.push(Token {
302 kind: TokenKind::Punct(punct),
303 flags: token.flags,
304 value: 0,
305 span: token.span,
306 }),
307 PpTokenKind::Eof => out.tokens.push(Token {
308 kind: TokenKind::Eof,
309 flags: token.flags,
310 value: 0,
311 span: token.span,
312 }),
313 PpTokenKind::Other | PpTokenKind::HeaderName => {
318 let text = spelling(token, cx);
319 diagnostics.push(Diagnostic::error(format!("stray '{text}' in program"), token.span));
320 }
321 }
322 index
323}
324
325fn is_pragma(token: PpToken, cx: &Convert<'_>) -> bool {
328 token.kind == PpTokenKind::Ident && spelling(token, cx) == "pragma"
329}
330
331fn spelling<'a>(token: PpToken, cx: &Convert<'a>) -> &'a str {
333 token.value.map_or("", |symbol| cx.interner.resolve(symbol))
334}
335
336fn identifier(token: PpToken, cx: &Convert<'_>) -> Token {
338 let symbol = token.value.expect("an identifier carries its spelling");
339 let kind = match cx.keywords.get(symbol) {
340 Some(word) => TokenKind::Keyword(word),
341 None => TokenKind::Ident,
342 };
343 Token { kind, flags: token.flags, value: symbol.raw(), span: token.span }
344}
345
346fn number(
348 token: PpToken,
349 cx: &Convert<'_>,
350 ints: &mut Vec<IntConstant>,
351 floats: &mut Vec<FloatConstant>,
352 diagnostics: &mut Vec<Diagnostic>,
353) -> Token {
354 let text = spelling(token, cx);
355 match crate::number::integer(text, cx.std, cx.target) {
358 Ok(value) => {
359 report(value.remarks, None, token.span, cx, diagnostics);
360 ints.push(value);
361 let index = u32::try_from(ints.len() - 1).expect("that many constants in one file");
362 Token { kind: TokenKind::Int, flags: token.flags, value: index, span: token.span }
363 }
364 Err(IntError::Floating) => match crate::number::floating(text, cx.std, cx.target) {
365 Ok(value) => {
366 report(value.remarks, Some(value.ty.name()), token.span, cx, diagnostics);
367 floats.push(value);
368 let index =
369 u32::try_from(floats.len() - 1).expect("that many constants in one file");
370 Token { kind: TokenKind::Float, flags: token.flags, value: index, span: token.span }
371 }
372 Err(error) => {
373 diagnostics.push(Diagnostic::error(error.message(), token.span));
374 floats.push(zero_float(cx));
377 let index =
378 u32::try_from(floats.len() - 1).expect("that many constants in one file");
379 Token { kind: TokenKind::Float, flags: token.flags, value: index, span: token.span }
380 }
381 },
382 Err(error) => {
383 diagnostics.push(Diagnostic::error(error.message(), token.span));
384 ints.push(IntConstant {
385 value: 0,
386 ty: crate::number::IntConstantType::Standard(rucc_types::IntKind::Int),
387 remarks: Remarks::NONE,
388 });
389 let index = u32::try_from(ints.len() - 1).expect("that many constants in one file");
390 Token { kind: TokenKind::Int, flags: token.flags, value: index, span: token.span }
391 }
392 }
393}
394
395fn zero_float(cx: &Convert<'_>) -> FloatConstant {
397 let ty = crate::number::FloatConstantType::Double;
398 FloatConstant {
399 value: rucc_base::float::Float::zero(ty.format(cx.target), false),
400 ty,
401 imaginary: false,
402 remarks: Remarks::NONE,
403 }
404}
405
406fn char_const(
408 token: PpToken,
409 cx: &Convert<'_>,
410 chars: &mut Vec<CharConstant>,
411 diagnostics: &mut Vec<Diagnostic>,
412) -> Token {
413 let text = spelling(token, cx);
414 let value = match crate::literal::character(text, cx.std, cx.gnu, cx.target) {
415 Ok(value) => {
416 report(value.remarks, None, token.span, cx, diagnostics);
417 value
418 }
419 Err(error) => {
420 diagnostics.push(Diagnostic::error(error.message(), token.span));
421 CharConstant {
422 value: 0,
423 encoding: crate::literal::Encoding::Plain,
424 remarks: Remarks::NONE,
425 }
426 }
427 };
428 chars.push(value);
429 let index = u32::try_from(chars.len() - 1).expect("that many constants in one file");
430 Token { kind: TokenKind::Char, flags: token.flags, value: index, span: token.span }
431}
432
433fn string_lit(
435 run: &[PpToken],
436 cx: &Convert<'_>,
437 strings: &mut Vec<StringLiteral>,
438 diagnostics: &mut Vec<Diagnostic>,
439) -> Token {
440 let first = run[0];
441 let span = first.span.to(run[run.len() - 1].span);
442 let texts: Vec<&str> = run.iter().map(|token| spelling(*token, cx)).collect();
443 let value = match crate::literal::strings(&texts, cx.std, cx.gnu, cx.target) {
444 Ok(value) => {
445 report(value.remarks, None, span, cx, diagnostics);
446 value
447 }
448 Err(error) => {
449 diagnostics.push(Diagnostic::error(error.message(), span));
450 let encoding = if error == LiteralError::MixedEncodings {
453 crate::literal::Encoding::Plain
454 } else {
455 crate::literal::Encoding::read_prefix(texts[0])
456 };
457 StringLiteral { elements: Vec::new(), encoding, remarks: Remarks::NONE }
458 }
459 };
460 strings.push(value);
461 let index = u32::try_from(strings.len() - 1).expect("that many literals in one file");
462 Token { kind: TokenKind::Str, flags: first.flags, value: index, span }
463}
464
465fn report(
471 remarks: Remarks,
472 type_name: Option<&str>,
473 span: Span,
474 cx: &Convert<'_>,
475 diagnostics: &mut Vec<Diagnostic>,
476) {
477 if remarks.is_none() {
478 return;
479 }
480
481 let always: [(Remarks, &str); 6] = [
484 (Remarks::MULTICHARACTER, "multi-character character constant"),
485 (Remarks::TOO_LONG, "character constant too long for its type"),
486 (Remarks::UNKNOWN_ESCAPE, "unknown escape sequence"),
487 (Remarks::HEX_ESCAPE_OUT_OF_RANGE, "hex escape sequence out of range"),
488 (Remarks::OCTAL_ESCAPE_OUT_OF_RANGE, "octal escape sequence out of range"),
489 (Remarks::UNSIGNED, "integer constant is so large that it is unsigned"),
490 ];
491 for (remark, message) in always {
492 if remarks.has(remark) {
493 diagnostics.push(Diagnostic::warning(message, span));
494 }
495 }
496 if remarks.has(Remarks::OUT_OF_RANGE) {
497 let ty = type_name.unwrap_or("double");
498 diagnostics
499 .push(Diagnostic::warning(format!("floating constant exceeds range of '{ty}'"), span));
500 }
501 if remarks.has(Remarks::TRUNCATED) {
502 diagnostics.push(Diagnostic::warning("floating constant truncated to zero", span));
503 }
504
505 if !cx.pedantic {
506 return;
507 }
508 let pedantic: [(Remarks, &str); 9] = [
511 (Remarks::NON_ISO_ESCAPE, "non-ISO-standard escape sequence"),
512 (Remarks::DOUBLE_SUFFIX, "suffix for double constant is a GCC extension"),
513 (Remarks::IMAGINARY, "imaginary constants are a GCC extension"),
514 (Remarks::BINARY, "binary constants are a C23 feature or GCC extension"),
515 (Remarks::EXTENDED_SUFFIX, "non-standard suffix on floating constant"),
516 (Remarks::HEX_FLOAT, "use of C99 hexadecimal floating constant"),
517 (Remarks::LONG_LONG, "use of C99 long long integer constant"),
518 (Remarks::SEPARATORS, "digit separators are a C23 feature"),
519 (Remarks::BIT_INT, "'_BitInt' constants are a C23 feature"),
520 ];
521 for (remark, message) in pedantic {
522 if remarks.has(remark) {
523 diagnostics.push(Diagnostic::warning(message, span));
524 }
525 }
526 if remarks.has(Remarks::UCN) {
527 diagnostics.push(Diagnostic::warning(
528 "universal character names are only valid in C++ and C99",
529 span,
530 ));
531 }
532}
533
534#[cfg(test)]
535mod tests {
536 use rucc_target::Triple;
537
538 use super::*;
539 use crate::lexer::{Options, tokenize};
540
541 struct Fixture {
544 interner: Interner,
545 keywords: Keywords,
546 target: TargetInfo,
547 std: Std,
548 gnu: bool,
549 pedantic: bool,
550 }
551
552 impl Fixture {
553 fn new(std: Std) -> Fixture {
554 let mut interner = Interner::new();
555 let keywords = Keywords::new(&mut interner, std, true);
556 let target =
557 TargetInfo::new("x86_64-unknown-linux-gnu".parse::<Triple>().expect("a triple"));
558 Fixture { interner, keywords, target, std, gnu: false, pedantic: false }
559 }
560
561 fn run(&mut self, src: &str) -> (Tokens, Vec<String>) {
563 let (pp, lex_diagnostics) =
564 tokenize(src.as_bytes(), 0, Options::new(), &mut self.interner);
565 assert!(lex_diagnostics.is_empty(), "the scanner disliked the source: {src}");
566 let cx = Convert {
567 keywords: &self.keywords,
568 interner: &self.interner,
569 target: &self.target,
570 std: self.std,
571 gnu: self.gnu,
572 pedantic: self.pedantic,
573 };
574 let (tokens, diagnostics) = convert(&pp, &cx);
575 (tokens, diagnostics.iter().map(|d| d.message.clone()).collect())
576 }
577 }
578
579 fn kinds(src: &str) -> Vec<TokenKind> {
580 Fixture::new(Std::C23).run(src).0.tokens.iter().map(|t| t.kind).collect()
581 }
582
583 #[test]
584 fn a_token_is_sixteen_bytes() {
585 assert_eq!(size_of::<Token>(), 16);
588 }
589
590 #[test]
591 fn a_declaration_converts_into_keywords_an_identifier_and_a_constant() {
592 assert_eq!(
593 kinds("int x = 1;"),
594 vec![
595 TokenKind::Keyword(Keyword::Int),
596 TokenKind::Ident,
597 TokenKind::Punct(Punct::Eq),
598 TokenKind::Int,
599 TokenKind::Punct(Punct::Semi),
600 TokenKind::Eof,
601 ]
602 );
603 }
604
605 #[test]
607 fn the_dialect_decides_which_identifiers_are_keywords() {
608 let mut c89 = Fixture::new(Std::C89);
609 let (tokens, _) = c89.run("restrict");
610 assert_eq!(tokens.tokens[0].kind, TokenKind::Ident);
611 let mut c99 = Fixture::new(Std::C99);
612 let (tokens, _) = c99.run("restrict");
613 assert_eq!(tokens.tokens[0].kind, TokenKind::Keyword(Keyword::Restrict));
614 }
615
616 #[test]
617 fn a_number_becomes_whichever_kind_of_constant_it_is() {
618 let mut fixture = Fixture::new(Std::C23);
619 let (tokens, diagnostics) = fixture.run("1 2.5 0x1p3 1u");
620 assert!(diagnostics.is_empty());
621 let kinds: Vec<_> = tokens.tokens.iter().map(|t| t.kind).collect();
622 assert_eq!(
623 kinds,
624 vec![
625 TokenKind::Int,
626 TokenKind::Float,
627 TokenKind::Float,
628 TokenKind::Int,
629 TokenKind::Eof
630 ]
631 );
632 assert_eq!(tokens.int(tokens.tokens[0]).expect("an integer").value, 1);
633 assert!(tokens.float(tokens.tokens[1]).is_some());
634 assert_eq!(tokens.tokens[2].value, 1);
637 assert_eq!(tokens.tokens[3].value, 1);
638 assert_eq!(tokens.int(tokens.tokens[3]).expect("an integer").value, 1);
639 assert!(tokens.float(tokens.tokens[0]).is_none());
641 assert!(tokens.string(tokens.tokens[0]).is_none());
642 }
643
644 #[test]
647 fn adjacent_string_literals_become_one_token() {
648 let mut fixture = Fixture::new(Std::C23);
649 let (tokens, diagnostics) = fixture.run(r#"char *s = "a" "b" L"c";"#);
650 assert!(diagnostics.is_empty(), "{diagnostics:?}");
651 let literal = tokens
652 .tokens
653 .iter()
654 .find(|t| t.kind == TokenKind::Str)
655 .copied()
656 .expect("a string literal");
657 let value = tokens.string(literal).expect("the literal");
658 assert_eq!(value.elements, vec![0x61, 0x62, 0x63]);
659 assert_eq!(value.encoding, crate::literal::Encoding::Wide);
660 assert_eq!(tokens.tokens.iter().filter(|t| t.kind == TokenKind::Str).count(), 1);
661 assert_eq!(literal.span.lo, 10);
663 assert_eq!(literal.span.hi, 22);
664 }
665
666 #[test]
667 fn a_character_constant_carries_its_value_and_its_warning() {
668 let mut fixture = Fixture::new(Std::C23);
669 let (tokens, diagnostics) = fixture.run("'ab'");
670 assert_eq!(diagnostics, vec!["multi-character character constant".to_owned()]);
671 assert_eq!(tokens.character(tokens.tokens[0]).expect("a constant").value, 0x6162);
672 }
673
674 #[test]
677 fn the_warnings_that_need_no_flag_are_given_without_one() {
678 let mut fixture = Fixture::new(Std::C17);
679 let (_, diagnostics) = fixture.run(r"'abcde' '\q' '\x1ff' '\400' 1e400 1e-400");
680 assert_eq!(
681 diagnostics,
682 vec![
683 "character constant too long for its type".to_owned(),
684 "unknown escape sequence".to_owned(),
685 "hex escape sequence out of range".to_owned(),
686 "octal escape sequence out of range".to_owned(),
687 "floating constant exceeds range of 'double'".to_owned(),
688 "floating constant truncated to zero".to_owned(),
689 ]
690 );
691 }
692
693 #[test]
695 fn the_warnings_that_need_pedantic_wait_for_it() {
696 let mut quiet = Fixture::new(Std::C17);
697 let (_, diagnostics) = quiet.run(r"1.0d 1.0i 0b1010 '\e'");
698 assert!(diagnostics.is_empty(), "{diagnostics:?}");
699
700 let mut loud = Fixture::new(Std::C17);
701 loud.pedantic = true;
702 let (_, diagnostics) = loud.run(r"1.0d 1.0i 0b1010 '\e'");
703 assert_eq!(
704 diagnostics,
705 vec![
706 "suffix for double constant is a GCC extension".to_owned(),
707 "imaginary constants are a GCC extension".to_owned(),
708 "binary constants are a C23 feature or GCC extension".to_owned(),
709 "non-ISO-standard escape sequence".to_owned(),
710 ]
711 );
712 }
713
714 #[test]
715 fn the_overflow_warning_names_the_type_the_constant_actually_has() {
716 let mut fixture = Fixture::new(Std::C23);
717 let (_, diagnostics) = fixture.run("1e400f");
718 assert_eq!(diagnostics, vec!["floating constant exceeds range of 'float'".to_owned()]);
719 }
720
721 #[test]
724 fn a_constant_that_will_not_convert_still_leaves_a_token_behind() {
725 let mut fixture = Fixture::new(Std::C23);
726 let (tokens, diagnostics) = fixture.run("int x = 1.2.3;");
727 assert_eq!(diagnostics.len(), 1);
728 let kinds: Vec<_> = tokens.tokens.iter().map(|t| t.kind).collect();
729 assert_eq!(
730 kinds,
731 vec![
732 TokenKind::Keyword(Keyword::Int),
733 TokenKind::Ident,
734 TokenKind::Punct(Punct::Eq),
735 TokenKind::Float,
736 TokenKind::Punct(Punct::Semi),
737 TokenKind::Eof,
738 ]
739 );
740
741 let mut fixture = Fixture::new(Std::C23);
742 let (tokens, diagnostics) = fixture.run("int x = 42ux;");
743 assert_eq!(diagnostics, vec!["invalid suffix on integer constant".to_owned()]);
744 assert_eq!(tokens.int(tokens.tokens[3]).expect("a stand in").value, 0);
745 }
746
747 #[test]
748 fn a_run_of_literals_with_two_prefixes_is_refused_the_way_gcc_refuses_it() {
749 let mut fixture = Fixture::new(Std::C23);
750 let (tokens, diagnostics) = fixture.run(r#"u"a" L"b""#);
751 assert_eq!(
752 diagnostics,
753 vec!["unsupported non-standard concatenation of string literals".to_owned()]
754 );
755 assert!(tokens.string(tokens.tokens[0]).expect("a stand in").elements.is_empty());
756 }
757
758 #[test]
761 fn a_stray_byte_is_an_error_here_and_nowhere_earlier() {
762 let mut fixture = Fixture::new(Std::C23);
763 let (tokens, diagnostics) = fixture.run("a ` b");
764 assert_eq!(diagnostics, vec!["stray '`' in program".to_owned()]);
765 let kinds: Vec<_> = tokens.tokens.iter().map(|t| t.kind).collect();
766 assert_eq!(kinds, vec![TokenKind::Ident, TokenKind::Ident, TokenKind::Eof]);
767 }
768
769 #[test]
770 fn the_stream_always_ends_in_end_of_file() {
771 let mut fixture = Fixture::new(Std::C23);
772 let (tokens, _) = fixture.run("");
773 assert_eq!(tokens.tokens.len(), 1);
774 assert!(tokens.tokens[0].is_eof());
775 let (tokens, _) = convert(
777 &[],
778 &Convert {
779 keywords: &fixture.keywords,
780 interner: &fixture.interner,
781 target: &fixture.target,
782 std: fixture.std,
783 gnu: false,
784 pedantic: false,
785 },
786 );
787 assert_eq!(tokens.tokens.len(), 1);
788 assert!(tokens.tokens[0].is_eof());
789 }
790
791 #[test]
792 fn a_token_says_what_it_is_without_the_caller_matching_on_the_kind() {
793 let mut fixture = Fixture::new(Std::C23);
794 let (tokens, _) = fixture.run("int x;");
795 assert_eq!(tokens.tokens[0].keyword(), Some(Keyword::Int));
796 assert_eq!(tokens.tokens[0].ident(), None);
797 assert!(tokens.tokens[1].ident().is_some());
798 assert_eq!(tokens.tokens[2].punct(), Some(Punct::Semi));
799 assert_eq!(tokens.tokens[2].keyword(), None);
800 }
801
802 #[test]
805 fn a_pragma_line_leaves_the_stream_and_is_kept_beside_it() {
806 let mut fixture = Fixture::new(Std::C23);
807 let (tokens, diagnostics) = fixture.run("int a;\n#pragma pack(4)\nint b;");
808 assert!(diagnostics.is_empty(), "{diagnostics:?}");
809 let kinds: Vec<_> = tokens.tokens.iter().map(|t| t.kind).collect();
810 assert_eq!(
811 kinds,
812 vec![
813 TokenKind::Keyword(Keyword::Int),
814 TokenKind::Ident,
815 TokenKind::Punct(Punct::Semi),
816 TokenKind::Keyword(Keyword::Int),
817 TokenKind::Ident,
818 TokenKind::Punct(Punct::Semi),
819 TokenKind::Eof,
820 ]
821 );
822 assert_eq!(tokens.pragmas.len(), 1);
823 let pragma = &tokens.pragmas[0];
824 assert_eq!(pragma.before, 3);
827 let kinds: Vec<_> = pragma.tokens.iter().map(|t| t.kind).collect();
828 assert_eq!(
829 kinds,
830 vec![
831 TokenKind::Ident,
832 TokenKind::Punct(Punct::LParen),
833 TokenKind::Int,
834 TokenKind::Punct(Punct::RParen),
835 ]
836 );
837 }
838
839 #[test]
842 fn a_pragma_at_either_end_of_the_file_is_still_a_line() {
843 let mut fixture = Fixture::new(Std::C23);
844 let (tokens, diagnostics) = fixture.run("#pragma once\nint a;\n#pragma GCC poison x");
845 assert!(diagnostics.is_empty(), "{diagnostics:?}");
846 assert_eq!(tokens.tokens.len(), 4);
847 assert_eq!(tokens.pragmas.len(), 2);
848 assert_eq!(tokens.pragmas[0].before, 0);
849 assert_eq!(tokens.pragmas[0].tokens.len(), 1);
850 assert_eq!(tokens.pragmas[1].before, 3);
851 assert_eq!(tokens.pragmas[1].tokens.len(), 3);
852 }
853
854 #[test]
857 fn a_hash_that_is_not_a_pragma_is_left_where_it_is() {
858 let mut fixture = Fixture::new(Std::C23);
859 let (tokens, _) = fixture.run("#define x\nint pragma;\n# pragma");
860 assert_eq!(tokens.tokens[0].kind, TokenKind::Punct(Punct::Hash));
861 assert_eq!(tokens.pragmas.len(), 1);
864 }
865
866 #[test]
869 fn the_two_extra_spellings_of_the_wide_integer_are_keywords() {
870 let kinds = kinds("__int128_t a; __uint128_t b;");
871 assert_eq!(kinds[0], TokenKind::Keyword(Keyword::Int128T));
872 assert_eq!(kinds[3], TokenKind::Keyword(Keyword::UInt128T));
873 let mut c89 = Fixture::new(Std::C89);
875 let (tokens, _) = c89.run("__uint128_t");
876 assert_eq!(tokens.tokens[0].kind, TokenKind::Keyword(Keyword::UInt128T));
877 }
878
879 #[test]
882 fn the_flags_come_through_from_the_preprocessing_token() {
883 let mut fixture = Fixture::new(Std::C23);
884 let (tokens, _) = fixture.run("a\n b");
885 assert!(tokens.tokens[0].flags.has(TokenFlags::START_OF_LINE));
886 assert!(tokens.tokens[1].flags.has(TokenFlags::START_OF_LINE));
887 assert!(tokens.tokens[1].flags.has(TokenFlags::LEADING_SPACE));
888 }
889}