1use std::path::Path;
14
15use rucc_base::Interner;
16use rucc_codegen::pipeline::{self, Machine};
17use rucc_diag::{Diagnostic, Severity, Span};
18use rucc_lex::{Convert, Keywords, PpToken, convert};
19use rucc_sema::{Checker, Context as CheckContext};
20use rucc_session::{EmitKind, FileSystem, Options, Session};
21use rucc_target::TargetInfo;
22
23use crate::preprocess::render;
24
25#[derive(Debug, Clone, PartialEq, Eq, Default)]
32pub enum Artifact {
33 #[default]
36 Nothing,
37 Text(String),
39 Object(Vec<u8>),
41}
42
43impl Artifact {
44 #[must_use]
46 pub fn bytes(&self) -> &[u8] {
47 match self {
48 Artifact::Nothing => &[],
49 Artifact::Text(text) => text.as_bytes(),
50 Artifact::Object(bytes) => bytes,
51 }
52 }
53}
54
55#[derive(Debug, Clone, PartialEq, Eq)]
57pub struct Compiled {
58 pub artifact: Artifact,
60 pub messages: Vec<String>,
62 pub errors: u32,
64}
65
66impl Compiled {
67 #[must_use]
69 pub fn failed(&self) -> bool {
70 self.errors > 0
71 }
72
73 #[must_use]
78 pub fn text(&self) -> &str {
79 match &self.artifact {
80 Artifact::Text(text) => text,
81 _ => "",
82 }
83 }
84}
85
86#[must_use]
99pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
100 let mut sess = Session::new(opts.clone());
101 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
105 let mut diagnostics: Vec<Diagnostic> = Vec::new();
106
107 let bytes = match fs.read(Path::new(name)) {
108 Ok(bytes) => bytes,
109 Err(e) => return failure(format!("{name}: {e}")),
110 };
111 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
112 return failure(format!("{name}: the source map has no room left for this file"));
113 };
114
115 let mut pp = rucc_pp::Preprocessor::new();
119 let predef = rucc_pp::Predef::for_options(opts);
120 let expanded: Vec<PpToken> = {
121 let mut cx = rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
122 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
123 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
124 return failure(format!("{name}: the source map has no room for the built in macros"));
125 }
126 pp.run(file, &mut cx).iter().map(|token| token.to_pp()).collect()
127 };
128 diagnostics.extend(pp.take_diagnostics());
129
130 let cx = Convert {
133 keywords: &keywords,
134 interner: &sess.interner,
135 target: &sess.target,
136 std: opts.std,
137 gnu: opts.gnu_extensions,
138 pedantic: opts.pedantic,
139 };
140 let (tokens, complaints) = convert(&expanded, &cx);
141 diagnostics.extend(complaints);
142
143 let parsed = rucc_parse::parse(
144 &tokens,
145 rucc_parse::Context {
146 interner: &sess.interner,
147 std: opts.std,
148 gnu: opts.gnu_extensions,
149 pedantic: opts.pedantic,
150 error_limit: opts.error_limit as usize,
151 },
152 );
153 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
154 diagnostics.extend(parsed.diagnostics);
155
156 let mut artifact = Artifact::Nothing;
157 if !parse_failed {
158 let mut checker = Checker::new(
159 &parsed.ast,
160 CheckContext {
161 names: &sess.interner,
162 target: &sess.target,
163 std: opts.std,
164 gnu: opts.gnu_extensions,
165 pedantic: opts.pedantic,
166 error_limit: opts.error_limit as usize,
167 },
168 );
169 checker.check_unit();
170 let checked = checker.finish();
171 if !checked.failed() {
172 match opts.emit {
173 EmitKind::Tast => {
174 artifact = Artifact::Text(rucc_sema::print(
175 &checked.tast,
176 &checked.types,
177 &sess.interner,
178 ));
179 }
180 EmitKind::Ir
181 | EmitKind::MirFinal
182 | EmitKind::Asm
183 | EmitKind::Object
184 | EmitKind::Executable => {
185 let mut lowered = rucc_lower::lower(
186 name,
187 rucc_lower::Context {
188 tast: &checked.tast,
189 types: &checked.types,
190 target: &sess.target,
191 names: &mut sess.interner,
192 },
193 );
194 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
198 if !failed {
199 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
204 for error in errors {
205 diagnostics.push(internal(&format!("invalid IR, {error}")));
206 }
207 } else if opts.emit == EmitKind::Ir {
208 artifact =
209 Artifact::Text(rucc_ir::print(&lowered.module, &sess.interner));
210 } else {
211 match generate(
214 &mut lowered.module,
215 &mut sess.interner,
216 &sess.target,
217 opts,
218 ) {
219 Ok(made) => artifact = made,
220 Err(complaints) => diagnostics.extend(complaints),
221 }
222 }
223 }
224 diagnostics.extend(lowered.diagnostics);
225 }
226 _ => {}
227 }
228 }
229 diagnostics.extend(checked.diagnostics);
230 }
231
232 let mut messages = Vec::with_capacity(diagnostics.len());
233 let mut errors = 0;
234 for diag in &diagnostics {
235 if diag.severity.is_fatal()
236 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
237 {
238 errors += 1;
239 }
240 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
241 }
242 if errors > 0 {
243 artifact = Artifact::Nothing;
245 }
246 Compiled { artifact, messages, errors }
247}
248
249#[must_use]
259pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
260 let mut sess = Session::new(opts.clone());
261 if opts.emit != EmitKind::Ir {
262 return failure(format!(
263 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
264 the C in front of it became",
265 opts.emit.as_str()
266 ));
267 }
268 let bytes = match fs.read(Path::new(name)) {
269 Ok(bytes) => bytes,
270 Err(e) => return failure(format!("{name}: {e}")),
271 };
272 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
273 return failure(format!("{name}: this is not text, so it is not IR"));
274 };
275
276 let module = match rucc_ir::parse(text, &mut sess.interner) {
277 Ok(module) => module,
278 Err(error) => {
279 return failure(format!("{name}:{}: {}", error.line, error.message));
280 }
281 };
282 let mut diagnostics: Vec<Diagnostic> = Vec::new();
283 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
284 for error in errors {
285 diagnostics.push(invalid(&format!("invalid IR, {error}")));
286 }
287 }
288 let mut messages = Vec::with_capacity(diagnostics.len());
289 for diag in &diagnostics {
290 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
291 }
292 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
293 let artifact = if errors > 0 {
294 Artifact::Nothing
295 } else {
296 Artifact::Text(rucc_ir::print(&module, &sess.interner))
297 };
298 Compiled { artifact, messages, errors }
299}
300
301fn generate(
320 module: &mut rucc_ir::Module,
321 names: &mut Interner,
322 target: &TargetInfo,
323 opts: &Options,
324) -> Result<Artifact, Vec<Diagnostic>> {
325 let Some(machine) = Machine::for_target(target) else {
326 return Err(vec![unsupported(&format!(
327 "there is no back end for {} in this compiler yet, so there is nothing to generate",
328 target.triple
329 ))]);
330 };
331 let flags = pipeline::Flags { frame_pointer: opts.frame_pointer, red_zone: opts.red_zone };
332
333 let mut funcs = Vec::new();
334 let mut complaints = Vec::new();
335 for id in module.funcs() {
336 if module[id].is_declaration() {
337 continue;
338 }
339 match pipeline::compile(&mut module[id], names, &machine, flags) {
340 Ok(func) => funcs.push(func),
341 Err(why) => {
342 let name = names.resolve(module[id].name).to_owned();
343 let span = why.inst().map_or(Span::DUMMY, |inst| module[id].span(inst));
346 let said = format!("cannot generate code for '{name}': {why}");
347 complaints.push(unsupported_at(&said, span));
348 }
349 }
350 }
351 if !complaints.is_empty() {
352 return Err(complaints);
353 }
354 let globals = match opts.emit {
358 EmitKind::Asm | EmitKind::Object | EmitKind::Executable => {
359 rucc_asm::globals(module, names).map_err(refused)?
360 }
361 _ => rucc_asm::Globals::default(),
362 };
363 match opts.emit {
367 EmitKind::Asm => {
368 rucc_asm::print(&funcs, &globals, names, target).map(Artifact::Text).map_err(refused)
369 }
370 EmitKind::Object | EmitKind::Executable => {
373 let text = rucc_asm::assemble(&funcs, names, target).map_err(refused)?;
374 let data = globals.image();
375 rucc_object::write(&text, &data, target).map(Artifact::Object).map_err(
378 |why| match why {
379 rucc_object::Error::Format { .. } => vec![unsupported(&why.to_string())],
380 rucc_object::Error::Refused { .. } => vec![internal(&why.to_string())],
381 },
382 )
383 }
384 _ => Ok(Artifact::Text(rucc_mir::print(&funcs, names, target.regs))),
385 }
386}
387
388fn refused(why: rucc_asm::Error) -> Vec<Diagnostic> {
394 match why {
395 rucc_asm::Error::Thread { .. } => vec![unsupported(&why.to_string())],
396 _ => vec![internal(&why.to_string())],
397 }
398}
399
400fn unsupported(message: &str) -> Diagnostic {
406 unsupported_at(message, Span::DUMMY)
407}
408
409fn unsupported_at(message: &str, span: Span) -> Diagnostic {
415 Diagnostic::error(message.to_owned(), span)
416 .with_code("E0653")
417 .note("this construct is not lowered yet, see https://github.com/tamnd/rucc/issues", span)
418}
419
420fn invalid(message: &str) -> Diagnostic {
422 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
423}
424
425fn internal(message: &str) -> Diagnostic {
427 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
428 .with_code("E0652")
429 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
430}
431
432fn failure(message: String) -> Compiled {
435 Compiled {
436 artifact: Artifact::Nothing,
437 messages: vec![format!("rucc: error: {message}")],
438 errors: 1,
439 }
440}
441
442#[cfg(test)]
443mod tests {
444 use rucc_session::{MemoryFileSystem, Std};
445 use rucc_target::Triple;
446
447 use super::*;
448
449 fn options() -> Options {
450 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
451 opts.emit = EmitKind::Tast;
452 opts
453 }
454
455 fn run(opts: &Options, source: &str) -> Compiled {
456 let mut fs = MemoryFileSystem::new();
457 fs.insert("/main.c", source.to_owned().into_bytes());
458 compile(opts, "/main.c", &fs)
459 }
460
461 fn freestanding() -> Options {
465 let mut opts = options();
466 opts.hosted = false;
467 opts.search.push_system(rucc_session::runtime::DIR);
468 opts
469 }
470
471 fn shipped(source: &str) -> String {
473 let result = run(&freestanding(), source);
474 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
475 result.text().to_owned()
476 }
477
478 fn tast(source: &str) -> String {
480 let result = run(&options(), source);
481 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
482 result.text().to_owned()
483 }
484
485 #[test]
486 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
487 let text = shipped(concat!(
488 "#include <stdarg.h>\n",
489 "int sum(int n, ...) {\n",
490 " va_list ap, copy;\n",
491 " va_start(ap, n);\n",
492 " va_copy(copy, ap);\n",
493 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
494 " va_end(ap);\n",
495 " va_end(copy);\n",
496 " return total;\n",
497 "}\n",
498 ));
499 assert!(text.contains("va-start"), "{text}");
500 assert!(text.contains("va-copy"), "{text}");
501 assert!(text.contains("va-arg"), "{text}");
502 assert!(text.contains("va-end"), "{text}");
503 }
504
505 #[test]
509 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
510 let text = shipped(concat!(
511 "#define __need___va_list\n",
512 "#include <stdarg.h>\n",
513 "int vprint(const char *f, __gnuc_va_list ap);\n",
514 "#ifdef va_start\n",
515 "#error va_start should not be defined\n",
516 "#endif\n",
517 "#ifdef _VA_LIST_DEFINED\n",
518 "#error va_list should not have been made\n",
519 "#endif\n",
520 ));
521 assert!(text.contains("vprint"), "{text}");
522 }
523
524 #[test]
527 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
528 let text = shipped(concat!(
529 "#define __need_size_t\n",
530 "#include <stddef.h>\n",
531 "#ifdef offsetof\n",
532 "#error offsetof should not be defined yet\n",
533 "#endif\n",
534 "#define __need_ptrdiff_t\n",
535 "#include <stddef.h>\n",
536 "#include <stddef.h>\n",
537 "size_t a;\n",
538 "ptrdiff_t b;\n",
539 "wchar_t c;\n",
540 "max_align_t d;\n",
541 "void *e = NULL;\n",
542 "struct P { int x; long y; };\n",
543 "size_t f = offsetof(struct P, y);\n",
544 ));
545 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
546 assert!(text.contains("decl #1 b : long"), "{text}");
547 }
548
549 #[test]
550 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
551 let text = shipped(concat!(
552 "#include <limits.h>\n",
553 "#include <float.h>\n",
554 "int bits = CHAR_BIT;\n",
555 "long big = LONG_MAX;\n",
556 "int low = INT_MIN;\n",
557 "int radix = FLT_RADIX;\n",
558 "int digits = DBL_MANT_DIG;\n",
559 ));
560 assert!(text.contains("const 8 : int"), "{text}");
561 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
562 assert!(text.contains("const 2 : int"), "{text}");
563 assert!(text.contains("const 53 : int"), "{text}");
564 }
565
566 #[test]
570 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
571 let text = shipped(concat!(
572 "#include <stdint.h>\n",
573 "int64_t a = INT64_C(1);\n",
574 "uint_least16_t b;\n",
575 "intptr_t c;\n",
576 "uintmax_t d = UINTMAX_MAX;\n",
577 "int wide = sizeof(int_fast64_t);\n",
578 ));
579 assert!(text.contains("decl #0 a : long"), "{text}");
580 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
581 assert!(text.contains("decl #2 c : long"), "{text}");
582 }
583
584 #[test]
585 fn the_three_formality_headers_still_have_to_work() {
586 let text = shipped(concat!(
587 "#include <stdbool.h>\n",
588 "#include <stdalign.h>\n",
589 "#include <iso646.h>\n",
590 "#include <stdnoreturn.h>\n",
591 "int t = true and not false;\n",
592 "_Alignas(16) char buf[16];\n",
593 "int a = alignof(long);\n",
594 ));
595 assert!(text.contains("decl #0 t : int"), "{text}");
596 assert!(text.contains("const 8 : unsigned long"), "{text}");
597 }
598
599 #[test]
602 fn every_shipped_header_can_be_included_twice() {
603 let mut source = String::new();
604 for _ in 0..2 {
605 for name in rucc_session::runtime::names() {
606 source.push_str(&format!("#include <{name}>\n"));
607 }
608 }
609 source.push_str("int x;\n");
610 let text = shipped(&source);
611 assert!(text.starts_with("decl #0 x : int"), "{text}");
612 }
613
614 #[test]
615 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
616 let fs = MemoryFileSystem::new();
617 let result = compile(&options(), "/nope.c", &fs);
618 assert!(result.failed());
619 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
620 assert!(result.text().is_empty());
621 }
622
623 #[test]
624 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
625 let text = tast("int x = 1;\n");
626 let expected = "\
627decl #0 x : int object external static defined
628 init
629 +0
630 const 1 : int
631";
632 assert_eq!(text, expected);
633 }
634
635 #[test]
636 fn the_macros_are_expanded_before_anything_is_parsed() {
637 let text = tast("#define N 2\nint a[N];\n");
641 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
642 }
643
644 #[test]
650 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
651 let text = tast(concat!(
652 "#pragma pack(4)\n",
653 "struct s { int a; };\n",
654 "#pragma pack()\n",
655 "int b;\n",
656 "_Pragma(\"GCC visibility push(default)\") int c;\n",
657 ));
658 assert!(text.contains("decl #0 b : int"), "{text}");
659 assert!(text.contains("decl #1 c : int"), "{text}");
660 }
661
662 #[test]
670 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
671 tast(concat!(
672 "struct A { char c; int i; } __attribute__((packed));\n",
673 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
674 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
675 "struct B { char c; int i; } __attribute__((aligned));\n",
678 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
679 "struct C { char c; int i __attribute__((packed)); };\n",
680 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
681 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
682 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
683 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
684 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
685 "struct E { char c; _Alignas(8) int i; };\n",
686 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
687 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
688 "struct F { char c; int i __attribute__((aligned(8))); };\n",
689 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
690 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
693 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
694 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
695 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
696 "struct I { [[gnu::packed]] char c; int i; };\n",
699 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
700 "struct J { char c; [[gnu::packed]] int i; };\n",
701 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
702 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
703 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
704 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
705 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
706 "union L { char c; int i; } __attribute__((packed));\n",
707 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
708 "struct O { char c; int i; } __attribute__((__packed__));\n",
712 "_Static_assert(sizeof(struct O) == 5 && _Alignof(struct O) == 1, \"O\");\n",
713 "struct P { char c; int i; } __attribute__((__aligned__(8)));\n",
714 "_Static_assert(sizeof(struct P) == 8 && _Alignof(struct P) == 8, \"P\");\n",
715 ));
716 }
717
718 #[test]
728 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
729 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
731 assert_eq!(
732 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
733 1
734 );
735 assert_eq!(
736 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
737 1
738 );
739 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
740 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
742 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
743 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
745 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
746 }
747
748 fn bit_field_byte(record: &str) -> u64 {
750 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
751 let body = body(&source);
752 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
753 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
754 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
755 }
756
757 #[test]
763 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
764 tast(concat!(
765 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
766 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
767 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
768 "struct b { char c; __attribute__((packed)) int i; };\n",
769 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
770 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
771 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
772 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
773 ));
774 }
775
776 #[test]
782 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
783 tast(concat!(
784 "#pragma pack(1)\n",
785 "struct A { char c; int i; };\n",
786 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
787 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
788 "#pragma pack()\n",
789 "struct B { char c; int i; };\n",
790 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
791 "#pragma pack(2)\n",
792 "struct C { char c; int i; double d; };\n",
793 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
794 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
795 "struct K { char c; int i __attribute__((aligned(8))); };\n",
797 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
798 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
799 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
801 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
802 "#pragma pack()\n",
803 "#pragma pack(push, 1)\n",
804 "struct D { char c; short s; };\n",
805 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
806 "#pragma pack(pop)\n",
807 "struct E { char c; short s; };\n",
808 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
809 "struct H { char c;\n",
811 "#pragma pack(1)\n",
812 " int i; };\n",
813 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
814 "#pragma pack(1)\n",
815 "struct I { char c;\n",
816 "#pragma pack()\n",
817 " int i; };\n",
818 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
819 "#pragma pack()\n",
820 "#pragma pack(push, 8)\n",
822 "#pragma pack(push, 1)\n",
823 "struct P { char c; int i; };\n",
824 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
825 "#pragma pack(pop)\n",
826 "struct Q { char c; int i; };\n",
827 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
828 "#pragma pack(pop)\n",
829 "#pragma pack(16)\n",
831 "struct R { char c; int i; };\n",
832 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
833 "#pragma pack()\n",
834 "#pragma pack(1)\n",
835 "struct S { char c; int i : 5; int j : 20; };\n",
836 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
837 "union T { char c; int i; };\n",
838 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
839 "#pragma pack()\n",
840 ));
841 }
842
843 #[test]
847 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
848 let result = run(
849 &options(),
850 concat!(
851 "#pragma pack 4\n",
852 "#pragma pack(pop)\n",
853 "#pragma pack(3)\n",
854 "#pragma pack(1) junk\n",
855 "#pragma pack(push, 1\n",
856 "#pragma pack(x)\n",
857 "#pragma pack(0)\n",
860 "#pragma pack(push)\n",
861 "struct s { char c; int i; };\n",
862 "#pragma pack(pop)\n",
863 "#pragma pack(pop, foo)\n",
864 ),
865 );
866 let expected = [
867 "missing `(` after `#pragma pack` - ignored",
868 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
869 "alignment must be a small power of two, not 3",
870 "junk at end of `#pragma pack`",
871 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
872 "unknown action `x` for `#pragma pack` - ignored",
873 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
874 ];
875 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
876 for (message, want) in result.messages.iter().zip(expected) {
877 assert!(message.contains(want), "expected {want:?} in {message:?}");
878 }
879 }
880
881 #[test]
885 fn the_wide_integer_answers_to_all_three_of_its_names() {
886 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
887 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
888 assert!(text.contains("decl #1 b : __int128"), "{text}");
889 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
890 }
891
892 #[test]
893 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
894 let text = tast("long f(int a, long b) { return a + b; }\n");
898 assert!(text.contains("convert arithmetic"), "{text}");
899 }
900
901 #[test]
902 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
903 for source in [
904 "#error stop\n",
905 "int f(void) { return 1 + ; }\n",
906 "int f(void) { return undeclared; }\n",
907 ] {
908 let result = run(&options(), source);
909 assert!(result.failed(), "expected this to fail:\n{source}");
910 assert!(
911 result.text().is_empty(),
912 "a file that did not compile wrote a tree:\n{source}"
913 );
914 }
915 }
916
917 #[test]
918 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
919 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
923 assert_eq!(result.errors, 1, "{:?}", result.messages);
924 }
925
926 #[test]
927 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
928 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
932 assert_eq!(result.errors, 1, "{:?}", result.messages);
933 }
934
935 #[test]
936 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
937 let source = "int f(void) { char c = 300; return c; }\n";
938 let plain = run(&options(), source);
939 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
940 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
941 assert!(!plain.text().is_empty(), "a warning is not a reason to write nothing");
942
943 let mut opts = options();
944 opts.warnings_are_errors = true;
945 let strict = run(&opts, source);
946 assert!(strict.failed());
947 assert!(strict.text().is_empty(), "and under -Werror it is a reason to write nothing");
948 for message in &strict.messages {
949 assert!(!message.contains("warning:"), "{message}");
950 }
951 }
952
953 #[test]
954 fn the_dialect_reaches_the_keywords_and_the_checking() {
955 let source = "typeof(1) x;\n";
958 let mut opts = options();
959 opts.std = Std::C23;
960 opts.gnu_extensions = false;
961 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
962
963 opts.std = Std::C17;
964 assert!(run(&opts, source).failed());
965 }
966
967 #[test]
968 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
969 let mut opts = options();
970 opts.emit = EmitKind::Object;
971 let result = run(&opts, "int x = 1;\n");
972 assert!(!result.failed(), "{:?}", result.messages);
973 assert!(result.text().is_empty());
974 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
977 }
978
979 fn mir(source: &str) -> String {
981 let mut opts = options();
982 opts.emit = EmitKind::MirFinal;
983 let result = run(&opts, source);
984 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
985 result.text().to_owned()
986 }
987
988 #[test]
994 fn a_function_goes_from_c_to_instructions_with_real_registers_in_them() {
995 let text = mir("int add(int a, int b) { return a + b; }\n");
996 assert!(text.starts_with("mfunc @add {"), "{text}");
997 assert!(text.contains("x64.add_rr_32"), "{text}");
998 assert!(text.contains("x64.ret"), "{text}");
999 assert!(!text.contains('%'), "{text}");
1002 }
1003
1004 #[test]
1006 fn a_function_with_no_body_produces_no_machine_function() {
1007 let text = mir("int g(int);\nint f(int a) { return g(a); }\n");
1008 assert_eq!(text.matches("mfunc @").count(), 1, "{text}");
1009 assert!(text.contains("mfunc @f {"), "{text}");
1010 assert!(text.contains("x64.call"), "{text}");
1011 }
1012
1013 #[test]
1015 fn every_definition_in_the_file_is_generated_and_they_keep_their_order() {
1016 let text = mir("int a(int x) { return x; }\nint b(int x) { return x; }\n");
1017 let first = text.find("mfunc @a").expect("the first function");
1018 let second = text.find("mfunc @b").expect("the second function");
1019 assert!(first < second, "{text}");
1020 }
1021
1022 #[test]
1024 fn the_target_decides_which_convention_the_generated_code_follows() {
1025 let mut opts = options();
1026 opts.emit = EmitKind::MirFinal;
1027 let linux = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1028 assert!(linux.contains("$rdi"), "{linux}");
1029
1030 opts.target = "x86_64-pc-windows-msvc".parse::<Triple>().unwrap();
1031 let windows = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1032 assert!(windows.contains("$rcx"), "{windows}");
1033 assert!(!windows.contains("$rdi"), "{windows}");
1034 }
1035
1036 #[test]
1038 fn a_target_this_has_no_back_end_for_is_reported_rather_than_generated() {
1039 let mut opts = options();
1040 opts.emit = EmitKind::MirFinal;
1041 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
1042 let result = run(&opts, "int f(int a) { return a; }\n");
1043 assert!(result.failed());
1044 assert!(result.messages[0].contains("no back end for aarch64"), "{:?}", result.messages);
1045 assert!(result.text().is_empty());
1046 }
1047
1048 #[test]
1055 fn a_construct_the_back_end_cannot_reach_yet_is_reported_against_its_function() {
1056 let mut opts = options();
1057 opts.emit = EmitKind::MirFinal;
1058 let source = "long double a(long double x) { return x; }\n\
1059 long double b(long double x) { return x; }\n";
1060 let result = run(&opts, source);
1061 assert!(result.failed());
1062 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
1063 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
1064 assert!(result.messages[0].contains("x87 stack"), "{:?}", result);
1065 assert!(result.messages[1].contains("cannot generate code for 'b'"), "{:?}", result);
1066 assert!(result.text().is_empty());
1067 }
1068
1069 #[test]
1076 fn an_opcode_with_no_name_in_the_rule_language_is_named_by_its_own_spelling() {
1077 let mut opts = options();
1078 opts.emit = EmitKind::MirFinal;
1079 let result = run(&opts, "int f(int a) {\n __int128 wide = a;\n return (int) wide;\n}\n");
1080 assert!(result.failed());
1081 assert!(
1082 result.messages[0].contains("no rule lowers a `sext` producing a `i128`"),
1083 "{result:?}"
1084 );
1085 assert!(result.messages[0].contains(":2:"), "the line the widening is on: {result:?}");
1086 assert!(!result.messages[0].contains("this instruction"), "{result:?}");
1087 }
1088
1089 #[test]
1091 fn the_note_on_unfinished_work_points_at_the_issues_rather_than_at_the_plan() {
1092 let mut opts = options();
1093 opts.emit = EmitKind::MirFinal;
1094 let result = run(&opts, "int f(int a) { __int128 wide = a; return (int) wide; }\n");
1095 assert!(result.failed());
1096 let note = result.messages.iter().find(|line| line.contains("note:")).expect("a note");
1097 assert!(note.contains("https://github.com/tamnd/rucc/issues"), "{note}");
1098 assert!(!note.contains("spec/17-milestones.md"), "{note}");
1099 }
1100
1101 #[test]
1103 fn the_frame_flags_on_the_command_line_reach_the_generated_frame() {
1104 let source = "int f(int a) { return a; }\n";
1105 assert!(!mir(source).contains("$rbp"), "a leaf needs no frame pointer by default");
1106
1107 let mut opts = options();
1108 opts.emit = EmitKind::MirFinal;
1109 opts.frame_pointer = true;
1110 let kept = run(&opts, source).text().to_owned();
1111 assert!(kept.contains("x64.push_64 $rbp"), "{kept}");
1112 }
1113
1114 fn asm(source: &str) -> String {
1116 let mut opts = options();
1117 opts.emit = EmitKind::Asm;
1118 let result = run(&opts, source);
1119 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1120 result.text().to_owned()
1121 }
1122
1123 #[test]
1130 fn a_function_goes_from_c_to_assembly_an_assembler_would_take() {
1131 let text = asm("int add(int a, int b) { return a + b; }\n");
1132 assert!(text.contains("\t.globl\tadd\n"), "{text}");
1133 assert!(text.contains("\t.type\tadd, @function\n"), "{text}");
1134 assert!(text.contains("\nadd:\n"), "{text}");
1135 assert!(text.contains("\taddl\t"), "{text}");
1136 assert!(text.contains("\tret\n"), "{text}");
1137 assert!(text.contains("\t.size\tadd, .-add\n"), "{text}");
1138 assert!(text.contains(".note.GNU-stack"), "{text}");
1141 }
1142
1143 #[test]
1149 fn a_call_through_a_function_pointer_goes_through_the_register_it_is_in() {
1150 let text = asm("int g(int);\nint f(int (*p)(int), int a) { return p(a) + g(a); }\n");
1151 assert!(text.contains("\tcall\t*%"), "{text}");
1152 assert!(text.contains("\tcall\tg\n"), "{text}");
1153 assert!(text.contains("%rdi"), "{text}");
1157 }
1158
1159 #[test]
1161 fn the_address_of_a_global_is_read_from_the_instruction_pointer() {
1162 let text = asm("extern int counter;\nint f(void) { return counter; }\n");
1163 assert!(text.contains("\tleaq\tcounter(%rip), "), "{text}");
1164 }
1165
1166 #[test]
1168 fn a_cast_between_a_pointer_and_an_integer_leaves_the_value_where_it_is() {
1169 let text = asm("long f(void *p) { return (long)p; }\n");
1170 for line in text.lines().filter(|line| line.starts_with('\t') && !line.contains('.')) {
1175 let mnemonic = line.split_whitespace().next().unwrap_or("");
1176 assert!(matches!(mnemonic, "movq" | "ret"), "{line} in\n{text}");
1177 }
1178 }
1179
1180 #[test]
1184 fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
1185 let six = "long a, long b, long c, long d, long e, long f";
1186 let text = asm(&format!("long f({six}, long g, long h) {{ return g + h; }}\n"));
1187
1188 assert!(text.contains("\tmovq\t8(%rsp), "), "{text}");
1192 assert!(text.contains("\tmovq\t16(%rsp), "), "{text}");
1193
1194 let narrow = asm(&format!("int f({six}, int g) {{ return g; }}\n"));
1198 assert!(narrow.contains("\tmovl\t8(%rsp), "), "{narrow}");
1199 let eight =
1200 "double a, double b, double c, double d, double e, double f, double g, double h";
1201 let float = asm(&format!("double f({eight}, double i) {{ return i; }}\n"));
1202 assert!(float.contains("\tmovsd\t8(%rsp), "), "{float}");
1203 }
1204
1205 #[test]
1208 fn a_call_writes_the_arguments_with_no_register_left_at_the_stack_pointer() {
1209 let six = "1, 2, 3, 4, 5, 6";
1210 let decl = "long g(long, long, long, long, long, long, long, long);\n";
1211 let text = asm(&format!("{decl}long f(void) {{ return g({six}, 7, 8); }}\n"));
1212
1213 assert!(text.contains("\tmovq\t%"), "{text}");
1214 assert!(text.contains(", (%rsp)\n"), "{text}");
1215 assert!(text.contains(", 8(%rsp)\n"), "{text}");
1216 assert!(text.contains("\tsubq\t$"), "{text}");
1218
1219 let narrow = "int g(int, int, int, int, int, int, int);\n";
1221 let text = asm(&format!("{narrow}int f(void) {{ return g({six}, 7); }}\n"));
1222 assert!(text.contains("\tmovl\t%"), "{text}");
1223 assert!(text.contains(", (%rsp)\n"), "{text}");
1224 }
1225
1226 #[test]
1229 fn a_variadic_call_counts_registers_and_not_arguments() {
1230 let nine = "1., 2., 3., 4., 5., 6., 7., 8., 9.";
1231 let decl = "int g(int, ...);\n";
1232 let text = asm(&format!("{decl}int f(void) {{ return g(0, {nine}); }}\n"));
1233
1234 assert!(text.contains("\tmovl\t$8, "), "eight registers, not nine: {text}");
1235 assert!(text.contains("\tmovsd\t%"), "{text}");
1236 assert!(text.contains(", (%rsp)\n"), "{text}");
1237 }
1238
1239 #[test]
1242 fn a_realigned_frame_reads_them_through_the_frame_pointer() {
1243 let six = "long a, long b, long c, long d, long e, long f";
1244 let body = "_Alignas(32) long wide[4]; wide[0] = g; return wide[0];";
1245 let text = asm(&format!("long f({six}, long g) {{ {body} }}\n"));
1246
1247 assert!(text.contains("\tandq\t$-32, %rsp"), "{text}");
1251 assert!(text.contains("\tmovq\t16(%rbp), "), "{text}");
1252 assert!(!text.contains("\tmovq\t16(%rsp), "), "{text}");
1253 }
1254
1255 #[test]
1257 fn the_target_decides_how_the_assembly_is_spelled() {
1258 let mut opts = options();
1259 opts.emit = EmitKind::Asm;
1260 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
1261 let text = run(&opts, "int f(void) { return 0; }\n").text().to_owned();
1262 assert!(text.contains("__TEXT,__text"), "{text}");
1263 assert!(text.contains("\n_f:\n"), "{text}");
1264 assert!(!text.contains(".note.GNU-stack"), "{text}");
1265 }
1266
1267 fn obj(source: &str) -> Vec<u8> {
1269 let mut opts = options();
1270 opts.emit = EmitKind::Object;
1271 let result = run(&opts, source);
1272 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1273 match result.artifact {
1274 Artifact::Object(bytes) => bytes,
1275 other => panic!("expected an object, got {other:?}"),
1276 }
1277 }
1278
1279 #[test]
1285 fn a_function_goes_from_c_to_an_object_a_linker_would_take() {
1286 let bytes = obj("int add(int a, int b) { return a + b; }\n");
1287 assert_eq!(&bytes[..4], b"\x7fELF", "an object file starts by saying it is one");
1288 let text = asm("int add(int a, int b) { return a + b; }\n");
1289 assert!(
1290 text.contains("\taddl\t"),
1291 "and the listing of it is the same instructions:\n{text}"
1292 );
1293 }
1294
1295 #[test]
1297 fn a_variable_goes_from_c_to_the_section_it_belongs_in() {
1298 let text = asm("int counter = 42;\nstatic int hidden;\nconst int fixed = 7;\n");
1299 assert!(text.contains("\t.data\n\t.globl\tcounter\n"), "{text}");
1300 assert!(text.contains("\ncounter:\n\t.long\t42\n"), "{text}");
1301 assert!(text.contains("\t.size\tcounter, .-counter\n"), "{text}");
1302 assert!(text.contains("\t.bss\n\t.p2align\t2\n"), "{text}");
1305 assert!(text.contains("\nhidden:\n\t.space\t4\n"), "{text}");
1306 assert!(!text.contains(".globl\thidden"), "{text}");
1307 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
1310 }
1311
1312 #[test]
1314 fn a_string_literal_is_a_variable_with_a_name_no_program_could_write() {
1315 let text = asm("const char *f(void) { return \"hi\"; }\n");
1316 assert!(text.contains("\t.ascii\t\"hi\\000\"\n"), "{text}");
1317 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
1318 let label = text
1319 .lines()
1320 .find(|line| line.starts_with(".Lstr"))
1321 .unwrap_or_else(|| panic!("a label for the literal in\n{text}"));
1322 assert!(!text.contains(&format!(".globl\t{}", label.trim_end_matches(':'))), "{text}");
1323 }
1324
1325 #[test]
1327 fn an_address_in_an_initializer_is_left_to_the_linker() {
1328 let source = "int counter;\nint *p = &counter;\n";
1329 let text = asm(source);
1330 assert!(text.contains("\np:\n\t.quad\tcounter\n"), "{text}");
1331 let bytes = obj(source);
1334 assert!(bytes.windows(8).any(|w| w == b"counter\0"), "the object has to name it");
1335 }
1336
1337 #[test]
1339 fn a_thread_local_variable_is_reported_as_work_that_is_not_done() {
1340 let mut opts = options();
1341 opts.emit = EmitKind::Asm;
1342 let result = run(&opts, "_Thread_local int x = 1;\n");
1343 assert!(result.failed(), "every thread sharing one variable is worse than a message");
1344 assert!(result.messages.iter().any(|m| m.contains("thread-local")), "{:?}", result);
1345 assert!(!result.messages.iter().any(|m| m.contains("internal")), "{:?}", result);
1347 }
1348
1349 #[test]
1351 fn the_object_and_the_listing_are_two_spellings_of_one_compilation() {
1352 let source = "int callee(void); int g(void) { return callee(); }\n";
1356 let bytes = obj(source);
1357 assert!(
1358 bytes.windows(7).any(|w| w == b"callee\0"),
1359 "the object has to name the callee for the linker to find it"
1360 );
1361 let text = asm(source);
1362 assert!(text.contains("\tcall\tcallee\n"), "{text}");
1363 }
1364
1365 #[test]
1371 fn compiling_for_an_executable_produces_an_object_and_not_a_dump() {
1372 let mut opts = options();
1373 opts.emit = EmitKind::Executable;
1375 let result = run(&opts, "int main(void) { return 0; }\n");
1376 assert_eq!(result.messages, Vec::<String>::new());
1377 match result.artifact {
1378 Artifact::Object(bytes) => assert_eq!(&bytes[..4], b"\x7fELF"),
1379 other => panic!("expected an object, got {other:?}"),
1380 }
1381 }
1382
1383 #[test]
1385 fn a_platform_with_no_object_writer_is_said_so_rather_than_written_as_elf() {
1386 let mut opts = options();
1387 opts.emit = EmitKind::Object;
1388 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
1389 let result = run(&opts, "int f(void) { return 0; }\n");
1390 assert!(result.failed(), "an object nobody can read is worse than a message");
1391 assert!(
1392 result.messages.iter().any(|m| m.contains("no object writer")),
1393 "{:?}",
1394 result.messages
1395 );
1396 }
1397
1398 fn ir(source: &str) -> String {
1400 let mut opts = options();
1401 opts.emit = EmitKind::Ir;
1402 let result = run(&opts, source);
1403 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1404 result.text().to_owned()
1405 }
1406
1407 fn body(source: &str) -> String {
1409 let text = ir(source);
1410 let (_, rest) = text.split_once("{\n").expect("a function definition");
1411 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
1412 body.to_owned()
1413 }
1414
1415 #[test]
1423 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
1424 let text = ir(concat!(
1425 "int g;\n",
1426 "int a = __builtin_constant_p(1);\n",
1427 "int b = __builtin_constant_p(g);\n",
1428 "int c = __builtin_constant_p(\"abc\");\n",
1429 "int d = __builtin_constant_p(&g);\n",
1430 "int e = __builtin_constant_p(1.5);\n",
1431 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
1432 ));
1433 assert!(text.contains("global @a : i32 = 1,"), "{text}");
1434 assert!(text.contains("global @b : i32 = 0,"), "{text}");
1435 assert!(text.contains("global @c : i32 = 1,"), "{text}");
1436 assert!(text.contains("global @d : i32 = 0,"), "{text}");
1437 assert!(text.contains("global @e : i32 = 1,"), "{text}");
1438 assert!(text.contains("global @h : i32 = 11,"), "{text}");
1439 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
1440
1441 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
1445 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
1446 }
1447
1448 #[test]
1457 fn a_call_to_a_library_builtin_reaches_the_library_function() {
1458 let text = body("void f(void) { __builtin_abort(); }\n");
1459 assert_eq!(text, "block0:\n call @abort() : ()\n return\n");
1460
1461 let text = ir("int f(const char *s) { return __builtin_puts(s) + __builtin_strlen(s); }\n");
1464 assert!(text.contains("call @puts(%0) : (ptr) -> i32"), "{text}");
1465 assert!(text.contains("call @strlen(%0) : (ptr) -> i64"), "{text}");
1466 assert!(!text.contains("__builtin_"), "the prefix is not part of any name here:\n{text}");
1467 }
1468
1469 #[test]
1481 fn the_hint_builtins_are_their_first_argument_and_the_hint_leaves_no_trace() {
1482 let text = ir(concat!(
1483 "long a = __builtin_expect(7, 1);\n",
1484 "long b = __builtin_expect_with_probability(9, 1, 0.9);\n",
1485 "unsigned long c = sizeof(__builtin_expect((char)1, 1));\n",
1486 ));
1487 assert!(text.contains("global @a : i64 = 7,"), "{text}");
1488 assert!(text.contains("global @b : i64 = 9,"), "{text}");
1489 assert!(text.contains("global @c : i64 = 8,"), "{text}");
1490 assert!(!text.contains("__builtin_expect"), "it is not a call to anything:\n{text}");
1491
1492 let text = body("long f(char c) { return __builtin_expect(c, 1); }\n");
1495 assert!(text.contains("sext"), "{text}");
1496
1497 let one = "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 1\n %2 = sext.i64 %1\n return %0\n";
1501 assert_eq!(body("int f(void) { int i = 0; __builtin_expect(1, i++); return i; }\n"), one);
1502 let source = "int g(void) { int i = 0; __builtin_expect_with_probability(1, i++, 0.5); return i; }\n";
1503 assert_eq!(body(source), one);
1504 }
1505
1506 #[test]
1518 fn a_promise_that_control_does_not_arrive_writes_no_instruction() {
1519 let promised = "int f(int x) { if (x) return 1; __builtin_unreachable(); }\n";
1520 let text = ir(promised);
1521 assert!(text.contains(" unreachable_hint\n"), "{text}");
1522 assert!(!text.contains("call"), "it is not a call to anything:\n{text}");
1523
1524 let after = body("int g(int x) { __builtin_unreachable(); return x; }\n");
1528 assert!(after.contains("return"), "{after}");
1529
1530 let text = asm(promised);
1533 let mine = text.split_once("\nf:\n").expect("a definition").1;
1534 let mine = mine.split_once("\t.size").expect("a definition").0;
1535 let plain = asm("int f(int x) { if (x) return 1; }\n");
1536 let plain = plain.split_once("\nf:\n").expect("a definition").1;
1537 let plain = plain.split_once("\t.size").expect("a definition").0;
1538 assert_eq!(mine, plain);
1539 assert!(mine.trim_end().ends_with("ret"), "{mine}");
1540 assert!(!mine.contains("ud2"), "{mine}");
1541 }
1542
1543 #[test]
1550 fn a_library_builtin_is_diagnosed_under_the_name_the_program_wrote() {
1551 let mut opts = options();
1552 opts.emit = EmitKind::Ir;
1553 let messages = run(&opts, "void f(void) { __builtin_abort(1); }\n").messages;
1554 assert!(
1555 messages.iter().any(|m| m.contains("__builtin_abort")),
1556 "expected the written name in {messages:?}"
1557 );
1558 }
1559
1560 #[test]
1568 fn a_builtin_nothing_lowers_is_refused_by_name() {
1569 let mut opts = options();
1570 opts.emit = EmitKind::Ir;
1571 for (builtin, call) in [
1572 ("__builtin_clz", "__builtin_clz(1u)"),
1573 ("__builtin_alloca", "(int)(long)__builtin_alloca(8)"),
1574 ("__atomic_load_n", "__atomic_load_n(&counter, 0)"),
1575 ("__sync_fetch_and_add", "(int)__sync_fetch_and_add(&counter, 1)"),
1576 ] {
1577 let source = format!("int counter;\nint f(void) {{ return {call}; }}\n");
1578 let messages = run(&opts, &source).messages;
1579 let named = messages.iter().any(|m| m.contains(builtin) && m.contains("E0686"));
1580 assert!(named, "expected {builtin} to be refused by name in {messages:?}");
1581 }
1582 }
1583
1584 #[test]
1592 fn what_is_refused_is_the_call_and_not_the_name() {
1593 let text = ir("unsigned long n = sizeof(__builtin_clz(1u));\n");
1594 assert!(text.contains("global @n : i64 = 4,"), "{text}");
1595
1596 let text = ir(
1597 "int __builtin_clz(unsigned x) { return 1; }\nint f(void) { return __builtin_clz(2u); }\n",
1598 );
1599 assert!(text.contains("call @__builtin_clz"), "{text}");
1600 }
1601
1602 #[test]
1607 fn a_static_function_nothing_refers_to_is_not_emitted() {
1608 let text = ir("static int dropped(void) { return 1; }\n\
1609 static int kept(void) { return 2; }\n\
1610 int main(void) { return kept(); }\n");
1611 assert!(text.contains("func @kept"), "{text}");
1612 assert!(!text.contains("dropped"), "{text}");
1613 }
1614
1615 #[test]
1621 fn two_static_functions_that_only_call_each_other_are_both_dropped() {
1622 let text = ir("static int ping(void);\n\
1623 static int pong(void) { return ping(); }\n\
1624 static int ping(void) { return pong(); }\n\
1625 int main(void) { return 0; }\n");
1626 assert!(!text.contains("ping"), "{text}");
1627 assert!(!text.contains("pong"), "{text}");
1628 }
1629
1630 #[test]
1636 fn naming_a_static_function_anywhere_keeps_it() {
1637 let text = ir("static int by_address(void) { return 1; }\n\
1638 static int in_an_image(void) { return 2; }\n\
1639 static int deeper(void) { return 3; }\n\
1640 static int reaches_deeper(void) { return deeper(); }\n\
1641 static int (*table[1])(void) = {in_an_image};\n\
1642 int main(void) {\n\
1643 int (*p)(void) = by_address;\n\
1644 return p() + table[0]() + reaches_deeper();\n\
1645 }\n");
1646 for kept in ["by_address", "in_an_image", "deeper", "reaches_deeper"] {
1647 assert!(text.contains(&format!("func @{kept}")), "expected {kept} in:\n{text}");
1648 }
1649 }
1650
1651 #[test]
1657 fn an_attribute_keeps_a_static_function_nothing_refers_to() {
1658 for attribute in ["used", "retain", "constructor", "destructor", "__used__"] {
1659 let source = format!(
1660 "__attribute__(({attribute})) static int kept(void) {{ return 1; }}\n\
1661 int main(void) {{ return 0; }}\n"
1662 );
1663 let text = ir(&source);
1664 assert!(text.contains("func @kept"), "for {attribute}:\n{text}");
1665 }
1666 }
1667
1668 #[test]
1671 fn a_function_anything_could_call_is_emitted_without_being_called() {
1672 let text =
1673 ir("int nobody_here_calls_it(void) { return 1; }\nint main(void) { return 0; }\n");
1674 assert!(text.contains("func @nobody_here_calls_it"), "{text}");
1675 }
1676
1677 #[test]
1684 fn a_classification_c_has_an_operator_for_is_that_operator() {
1685 for (builtin, operator) in [
1686 ("__builtin_isgreater", "binary >"),
1687 ("__builtin_isgreaterequal", "binary >="),
1688 ("__builtin_isless", "binary <"),
1689 ("__builtin_islessequal", "binary <="),
1690 ] {
1691 let source = format!("int f(double x, double y) {{ return {builtin}(x, y); }}\n");
1692 let text = tast(&source);
1693 assert!(text.contains(&format!("{operator} : int")), "for {builtin}:\n{text}");
1694 }
1695 }
1696
1697 #[test]
1706 fn the_classification_builtins_are_comparisons_and_not_calls() {
1707 let text = body("int f(double x, double y) { return __builtin_isunordered(x, y); }\n");
1708 assert_eq!(
1709 text,
1710 "block0(%0: f64, %1: f64):\n %2 = fcmp uno %0, %1\n %3 = zext.i32 \
1711 %2\n return %3\n"
1712 );
1713
1714 let text = body("int f(double x, double y) { return __builtin_islessgreater(x, y); }\n");
1716 assert!(text.contains("fcmp one %0, %1"), "{text}");
1717
1718 let text = body("int f(double x) { return __builtin_isnan(x); }\n");
1719 assert!(text.contains("fcmp uno %0, %0"), "{text}");
1720
1721 let text = body("int f(double x) { return __builtin_isinf(x); }\n");
1722 assert!(text.contains("fconst.f64 0x7ff0000000000000"), "{text}");
1723 assert!(text.contains("fconst.f64 0xfff0000000000000"), "{text}");
1724 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
1725 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
1726 assert!(text.contains("%5 = or %3, %4"), "{text}");
1727
1728 let text = body("int f(double x) { return __builtin_isfinite(x); }\n");
1731 assert!(text.contains("%3 = fcmp olt %2, %0"), "{text}");
1732 assert!(text.contains("%4 = fcmp olt %0, %1"), "{text}");
1733 assert!(text.contains("%5 = and %3, %4"), "{text}");
1734
1735 let text = body("int f(double x) { return __builtin_signbit(x); }\n");
1736 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
1737 assert!(text.contains("icmp slt %1, %2"), "{text}");
1738
1739 let text = body("int f(long double x) { return __builtin_signbitl(x); }\n");
1742 assert!(text.contains("%1 = bitcast.i80 %0"), "{text}");
1743
1744 let text = body("double g(void);\nint f(void) { return __builtin_isnan(g()); }\n");
1747 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
1748 }
1749
1750 #[test]
1757 fn a_classification_spelling_that_names_a_width_converts_before_it_asks() {
1758 let text = ir(concat!(
1759 "int a = __builtin_isinff(1e300);\n",
1760 "int b = __builtin_isinf(1e300);\n",
1761 "int c = __builtin_isnan(0.0);\n",
1765 "int d = __builtin_signbit(-0.0);\n",
1766 "int e = __builtin_islessgreater(1.0, 2.0);\n",
1767 ));
1768 assert!(text.contains("global @a : i32 = 1,"), "{text}");
1769 assert!(text.contains("global @b : i32 = 0,"), "{text}");
1770 assert!(text.contains("global @c : i32 = 0,"), "{text}");
1771 assert!(text.contains("global @d : i32 = 1,"), "{text}");
1772 assert!(text.contains("global @e : i32 = 1,"), "{text}");
1773 }
1774
1775 #[test]
1777 fn a_classification_builtin_refuses_an_argument_that_is_not_floating_point() {
1778 let mut opts = options();
1779 opts.emit = EmitKind::Ir;
1780 let source = concat!(
1781 "int a(int x) { return __builtin_isnan(x); }\n",
1782 "int b(int x, int y) { return __builtin_isunordered(x, y); }\n",
1783 "int c(double x) { return __builtin_isnan(x, x); }\n",
1784 );
1785 let messages = run(&opts, source).messages;
1786 assert_eq!(
1787 messages,
1788 [
1789 "/main.c:1:23: error: non-floating-point argument in call to function \
1790 '__builtin_isnan' [E0685]",
1791 "/main.c:2:30: error: non-floating-point arguments in call to function \
1792 '__builtin_isunordered' [E0685]",
1793 "/main.c:3:26: error: too many arguments to function '__builtin_isnan' [E0511]",
1794 ]
1795 );
1796 }
1797
1798 #[test]
1806 fn a_builtin_whose_answer_is_a_constant_is_one_and_not_a_call() {
1807 let text = ir(concat!(
1808 "double a = __builtin_inf();\n",
1809 "float b = __builtin_huge_valf();\n",
1810 "long double c = __builtin_infl();\n",
1811 "double d = __builtin_huge_val();\n",
1812 ));
1813 assert!(text.contains("global @a : f64 = 0x7ff0000000000000,"), "{text}");
1814 assert!(text.contains("global @b : f32 = 0x7f800000,"), "{text}");
1815 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
1816 assert!(text.contains("global @d : f64 = 0x7ff0000000000000,"), "{text}");
1817 assert!(!text.contains("call"), "{text}");
1818 }
1819
1820 #[test]
1829 fn a_nan_is_written_with_the_payload_the_program_asked_for() {
1830 let text = ir(concat!(
1831 "double a = __builtin_nan(\"\");\n",
1832 "double b = __builtin_nan(\"0x1\");\n",
1833 "double c = __builtin_nan(\"010\");\n",
1835 "double d = __builtin_nans(\"\");\n",
1836 "double e = __builtin_nans(\"0x1\");\n",
1837 "float f = __builtin_nanf(\"0x1\");\n",
1838 "float g = __builtin_nansf(\"\");\n",
1839 "long double h = __builtin_nansl(\"\");\n",
1840 ));
1841 assert!(text.contains("global @a : f64 = 0x7ff8000000000000,"), "{text}");
1842 assert!(text.contains("global @b : f64 = 0x7ff8000000000001,"), "{text}");
1843 assert!(text.contains("global @c : f64 = 0x7ff8000000000008,"), "{text}");
1844 assert!(text.contains("global @d : f64 = 0x7ff4000000000000,"), "{text}");
1845 assert!(text.contains("global @e : f64 = 0x7ff0000000000001,"), "{text}");
1846 assert!(text.contains("global @f : f32 = 0x7fc00001,"), "{text}");
1847 assert!(text.contains("global @g : f32 = 0x7fa00000,"), "{text}");
1848 assert!(text.contains("f80 0x7fffa000000000000000"), "{text}");
1849
1850 let text = ir(concat!(
1853 "double f(const char *p) { return __builtin_nan(p); }\n",
1854 "double g(void) { return __builtin_nans(\"1x\"); }\n",
1855 ));
1856 assert_eq!(text.matches("call @nan(").count(), 1, "{text}");
1857 assert_eq!(text.matches("call @nans(").count(), 1, "{text}");
1858 }
1859
1860 #[test]
1868 fn the_length_and_the_order_of_a_string_literal_are_known_here() {
1869 let text = ir(concat!(
1870 "unsigned long a = __builtin_strlen(\"hello\");\n",
1871 "unsigned long b = __builtin_strlen(\"a\\0bc\");\n",
1872 "int c = __builtin_strcmp(\"X\", \"X\\376\") < 0;\n",
1873 "int d = __builtin_strcmp(\"abc\", \"abc\");\n",
1874 "int e = __builtin_strcmp(\"abc\", \"ab\") > 0;\n",
1875 ));
1876 assert!(text.contains("global @a : i64 = 5,"), "{text}");
1877 assert!(text.contains("global @b : i64 = 1,"), "{text}");
1878 assert!(text.contains("global @c : i32 = 1,"), "{text}");
1879 assert!(text.contains("global @d : i32 = 0,"), "{text}");
1880 assert!(text.contains("global @e : i32 = 1,"), "{text}");
1881 assert!(!text.contains("call"), "{text}");
1882
1883 let text = ir("unsigned long f(const char *p) { return __builtin_strlen(p); }\n");
1885 assert!(text.contains("call @strlen("), "{text}");
1886 }
1887
1888 #[test]
1895 fn a_sign_builtin_is_a_mask_over_the_bits_and_not_a_call() {
1896 let text = body("double f(double x) { return __builtin_fabs(x); }\n");
1897 assert!(text.contains("bitcast.i64 %0"), "{text}");
1898 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
1899 assert!(text.contains("and %1, %2"), "{text}");
1900 assert!(text.contains("bitcast.f64 %3"), "{text}");
1901 assert!(!text.contains("call"), "{text}");
1902
1903 let text = body("double f(double x, double y) { return __builtin_copysign(x, y); }\n");
1904 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
1905 assert!(text.contains("%8 = or %4, %7"), "{text}");
1906 assert!(!text.contains("call"), "{text}");
1907
1908 let text = body("long double f(long double x) { return __builtin_fabsl(x); }\n");
1911 assert!(text.contains("bitcast.i80 %0"), "{text}");
1912 assert!(text.contains("bitcast.f80"), "{text}");
1913
1914 let text = body("double f(float x) { return __builtin_fabs(x); }\n");
1917 assert!(text.contains("fpext.f64 %0"), "{text}");
1918 assert!(text.contains("bitcast.i64 %1"), "{text}");
1919 }
1920
1921 #[test]
1930 fn the_sign_builtins_answer_a_zero_and_a_nan_the_way_the_bits_say() {
1931 let text = ir(concat!(
1932 "double a = __builtin_fabs(-3.5);\n",
1933 "double b = __builtin_copysign(1.0, -0.0);\n",
1934 "double c = __builtin_copysign(0.0, -2.0);\n",
1935 "double d = __builtin_copysign(-__builtin_nan(\"\"), 1.0);\n",
1937 "double e = __builtin_fabs(-__builtin_nan(\"0x1\"));\n",
1938 "float g = __builtin_copysignf(-0.0f, 2.0f);\n",
1939 "long double h = __builtin_copysignl(1.0L, -1.0L);\n",
1940 "long double i = __builtin_fabsl(-__builtin_infl());\n",
1941 ));
1942 assert!(text.contains("global @a : f64 = 0x400c000000000000,"), "{text}");
1943 assert!(text.contains("global @b : f64 = 0xbff0000000000000,"), "{text}");
1944 assert!(text.contains("global @c : f64 = 0x8000000000000000,"), "{text}");
1945 assert!(text.contains("global @d : f64 = 0x7ff8000000000000,"), "{text}");
1946 assert!(text.contains("global @e : f64 = 0x7ff8000000000001,"), "{text}");
1947 assert!(text.contains("global @g : f32 = 0x0,"), "{text}");
1948 assert!(text.contains("f80 0xbfff8000000000000000"), "{text}");
1949 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
1950 }
1951
1952 #[test]
1959 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
1960 let text = ir(concat!(
1961 "constexpr int side = 4;\n",
1962 "constexpr int wider = side + 1;\n",
1963 "constexpr double half = 1.5;\n",
1964 "struct point { int x; int y; };\n",
1965 "constexpr struct point origin = { 5, 6 };\n",
1966 "int square[side * side];\n",
1967 "int rectangle[wider];\n",
1968 "int rounded[(int)half * 2];\n",
1969 "int across[origin.y];\n",
1970 "enum named { four = side };\n",
1971 "int e = four;\n",
1972 ));
1973 assert!(text.contains("global @square : bytes 64 ="), "{text}");
1974 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
1975 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
1976 assert!(text.contains("global @across : bytes 24 ="), "{text}");
1977 assert!(text.contains("global @e : i32 = 4,"), "{text}");
1978
1979 let mut opts = options();
1982 opts.emit = EmitKind::Ir;
1983 let konst = "const int n = 1;\nint a[n];\n";
1984 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
1985 assert_eq!(run(&opts, konst).messages, [message]);
1986
1987 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
1989 assert_eq!(run(&opts, subscript).messages, [message]);
1990
1991 let address = "constexpr int c = 3;\nint *p = &c;\n";
1993 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
1994 pointer target type [E0514]";
1995 assert_eq!(run(&opts, address).messages, [warning]);
1996 }
1997
1998 #[test]
2007 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
2008 let mut opts = options();
2011 opts.std = Std::C17;
2012 let source = concat!(
2013 "int add(a, b)\n",
2014 "int a;\n",
2015 "int b;\n",
2016 "{ return a + b; }\n",
2017 "int promoted(c)\n",
2018 "char c;\n",
2019 "{ return c; }\n",
2020 "int narrow(char);\n",
2021 "int narrow(c)\n",
2022 "char c;\n",
2023 "{ return c; }\n",
2024 "int first(a)\n",
2025 "int a[4];\n",
2026 "{ return a[0]; }\n",
2027 );
2028 let result = run(&opts, source);
2029 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2030 let text = result.text();
2031 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
2032 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
2033 assert!(text.contains("c : char object automatic defined"), "{text}");
2035 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
2036 assert!(text.contains("first : int(int *) function external defined"), "{text}");
2038 }
2039
2040 #[test]
2047 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
2048 let mut opts = options();
2049 opts.std = Std::C17;
2050 for (source, message) in [
2051 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
2052 (
2053 "int f(a)\nint a;\nint b;\n{ return a; }\n",
2054 "3:5: error: declaration for parameter 'b' but no such parameter",
2055 ),
2056 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
2057 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
2058 (
2059 "int f(a)\nstatic int a;\n{ return a; }\n",
2060 "2:12: error: storage class specified for parameter 'a'",
2061 ),
2062 (
2063 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
2064 "2:7: error: argument 'a' doesn't match prototype",
2065 ),
2066 ] {
2067 let result = run(&opts, source);
2068 assert!(result.failed(), "expected this to fail:\n{source}");
2069 assert!(result.messages[0].contains(message), "{:?}", result.messages);
2070 }
2071
2072 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
2075 let mut older = options();
2076 older.std = Std::C89;
2077 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
2078 let result = run(&opts, implicit);
2079 assert!(
2080 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
2081 "{:?}",
2082 result.messages
2083 );
2084
2085 let mut newer = options();
2089 newer.std = Std::C23;
2090 let plain = "int f(a)\nint a;\n{ return a; }\n";
2091 let result = run(&newer, plain);
2092 assert!(!result.failed(), "{:?}", result.messages);
2093 assert_eq!(
2094 result.messages,
2095 ["/main.c:1:5: warning: old-style function definition [E0412]"]
2096 );
2097 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
2098 }
2099
2100 #[test]
2107 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
2108 let text = ir(concat!(
2109 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
2110 "struct brim { char buf[9223372036854775807L]; };\n",
2111 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
2112 "unsigned long h = sizeof(struct huge_struct);\n",
2113 "unsigned long b = sizeof(struct brim);\n",
2114 "unsigned long y = sizeof(struct bitty);\n",
2115 ));
2116 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
2117 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
2118 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
2119
2120 let mut opts = options();
2121 opts.emit = EmitKind::Ir;
2122 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
2123 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
2124 assert_eq!(run(&opts, over).messages, [message]);
2125 let array = "struct wide { short buf[1L << 62]; };\n";
2126 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
2127 maximum object size '9223372036854775807' [E0537]";
2128 assert_eq!(run(&opts, array).messages[0], message);
2129 }
2130
2131 fn compile_bytes(source: &[u8]) -> Compiled {
2136 let mut opts = options();
2137 opts.emit = EmitKind::Ir;
2138 let mut fs = MemoryFileSystem::new();
2139 fs.insert("/main.c", source.to_vec());
2140 compile(&opts, "/main.c", &fs)
2141 }
2142
2143 #[test]
2150 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
2151 let mut source = b"char s[] = \"a".to_vec();
2152 source.push(0xff);
2153 source.extend_from_slice(b"b\";\nchar c = '");
2154 source.push(0xff);
2155 source.extend_from_slice(b"';\n");
2156 let result = compile_bytes(&source);
2157 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
2158 assert!(result.text().contains(r#"bytes "a\ffb\00""#), "{}", result.text());
2159 assert!(result.text().contains("global @c : i8 = -1,"), "{}", result.text());
2161
2162 let mut stray = b"int a".to_vec();
2163 stray.push(0xff);
2164 stray.extend_from_slice(b" = 1;\n");
2165 let result = compile_bytes(&stray);
2166 assert!(
2167 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
2168 "{:?}",
2169 result.messages
2170 );
2171 }
2172
2173 #[test]
2174 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
2175 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
2176 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
2177 let expected = "\
2178func @add(i32, i32) -> i32, linkage(external) {
2179block0(%0: i32, %1: i32):
2180 %2 = add.nsw %0, %1
2181 return %2
2182}
2183";
2184 assert!(text.contains(expected), "{text}");
2185 }
2186
2187 #[test]
2188 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
2189 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
2190 assert!(!text.contains("alloca"), "{text}");
2191 assert!(!text.contains("load"), "{text}");
2192 assert!(!text.contains("store"), "{text}");
2193 }
2194
2195 #[test]
2196 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
2197 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
2198 let expected = "\
2199block0:
2200 %0 = alloca, size 4, align 4
2201 %1 = iconst.i32 1
2202 store %1 -> %0, align 4
2203 %2 = call @g(%0) : (ptr) -> i32
2204 return %2
2205";
2206 assert_eq!(text, expected);
2207 }
2208
2209 #[test]
2210 fn a_loop_carries_what_it_changes_as_block_parameters() {
2211 let text = body(
2214 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
2215 return total;\n}\n",
2216 );
2217 assert!(!text.contains("alloca"), "{text}");
2218 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
2219 assert!(text.contains("jump block1("), "{text}");
2220 }
2221
2222 #[test]
2223 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
2224 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
2225 assert!(text.contains("icmp slt %0, %1"), "{text}");
2226 assert!(!text.contains("zext"), "{text}");
2227 }
2228
2229 #[test]
2230 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
2231 let text = body("int f(int a, int b) { return a && b; }\n");
2232 let expected = "\
2233block0(%0: i32, %1: i32):
2234 %2 = iconst.i32 0
2235 %3 = icmp ne %0, %2
2236 %4 = iconst.i1 0
2237 br_if %3, block1, block2(%4)
2238
2239block1:
2240 %5 = iconst.i32 0
2241 %6 = icmp ne %1, %5
2242 jump block2(%6)
2243
2244block2(%7: i1):
2245 %8 = zext.i32 %7
2246 return %8
2247";
2248 assert_eq!(text, expected);
2249 }
2250
2251 #[test]
2252 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
2253 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
2254 assert!(!text.contains("block3"), "{text}");
2257 assert!(!text.contains("iconst.i32 3"), "{text}");
2258 }
2259
2260 #[test]
2261 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
2262 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
2263 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
2264 assert!(body("int f(void) { }\n").contains("unreachable"));
2265 }
2266
2267 #[test]
2268 fn a_structure_is_copied_rather_than_held_in_a_value() {
2269 let text = body(
2270 "struct point { int x, y; };\n\
2271 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
2272 );
2273 assert!(text.contains("memcpy"), "{text}");
2274 }
2275
2276 #[test]
2277 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
2278 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
2279 assert!(text.contains("memset"), "{text}");
2280 }
2281
2282 #[test]
2283 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
2284 let text = body(
2285 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
2286 default: r = 4; } return r; }\n",
2287 );
2288 let expected = "\
2289block0(%0: i32):
2290 %1 = iconst.i32 0
2291 switch %0, block1, [1 => block2, 2 => block3(%1)]
2292
2293block1:
2294 %2 = iconst.i32 4
2295 jump block4(%2)
2296
2297block2:
2298 %3 = iconst.i32 1
2299 jump block3(%3)
2300
2301block3(%4: i32):
2302 %5 = iconst.i32 2
2303 %6 = add.nsw %4, %5
2304 jump block4(%6)
2305
2306block4(%7: i32):
2307 return %7
2308";
2309 assert_eq!(text, expected);
2310 }
2311
2312 #[test]
2313 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
2314 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
2317 assert!(text.contains("%2 = sub %0, %1"), "{text}");
2318 assert!(text.contains("icmp ule"), "{text}");
2319 assert!(!text.contains("switch"), "{text}");
2320 }
2321
2322 #[test]
2323 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
2324 let text = body(
2325 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
2326 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
2327 );
2328 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
2331 assert!(text.contains("block5:\n jump block7("), "{text}");
2332 assert!(text.contains("block6:\n jump block8("), "{text}");
2333 }
2334
2335 #[test]
2336 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
2337 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
2338 }
2339
2340 #[test]
2341 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
2342 let text = body(
2347 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
2348 return n; }\n",
2349 );
2350 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
2353 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
2354 assert!(text.contains("block5:\n jump block3("), "{text}");
2355 }
2356
2357 #[test]
2358 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
2359 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
2362 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
2363 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
2364 assert!(text.contains("br_if %7, block3, block4"), "{text}");
2365 }
2366
2367 #[test]
2368 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
2369 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
2370 assert!(!text.contains("alloca"), "{text}");
2372 assert!(text.contains("block3(%4: i32):\n return %4"), "{text}");
2373 assert_eq!(text.matches("jump block3(").count(), 2, "{text}");
2374 }
2375
2376 #[test]
2377 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
2378 let text =
2379 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
2380 assert!(!text.contains("alloca"), "{text}");
2381 assert!(text.contains("block1(%2: i32):"), "{text}");
2382 assert!(text.contains("jump block1(%5)"), "{text}");
2383 }
2384
2385 #[test]
2386 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
2387 assert_eq!(
2390 body("int f(int x) { return x; spare: return 0; }\n"),
2391 "block0(%0: i32):\n return %0\n"
2392 );
2393 }
2394
2395 #[test]
2396 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
2397 let text = body(
2398 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
2399 );
2400 assert_eq!(
2403 text,
2404 "\
2405block0(%0: ptr):
2406 %1 = load.i8 %0, align 1
2407 %2 = iconst.i8 3
2408 %3 = ashr %1, %2
2409 %4 = sext.i32 %3
2410 return %4
2411"
2412 );
2413 }
2414
2415 #[test]
2416 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
2417 let text =
2421 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
2422 assert_eq!(
2423 text,
2424 "\
2425block0(%0: ptr, %1: i32):
2426 %2 = iconst.i32 16777215
2427 %3 = and %1, %2
2428 %4 = trunc.i16 %3
2429 store %4 -> %0, align 2
2430 %5 = iconst.i32 16
2431 %6 = lshr %3, %5
2432 %7 = trunc.i8 %6
2433 %8 = iconst.i64 2
2434 %9 = ptr_add %0, %8
2435 store %7 -> %9, align 1
2436 return
2437"
2438 );
2439 }
2440
2441 #[test]
2442 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
2443 let text =
2444 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
2445 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
2448 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
2449 }
2450
2451 #[test]
2452 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
2453 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
2456 assert_eq!(text.matches("ashr").count(), 0, "{text}");
2457 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
2458 }
2459
2460 #[test]
2461 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
2462 let text = body(
2466 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
2467 );
2468 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
2469 }
2470
2471 #[test]
2472 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
2473 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
2476 assert!(
2477 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
2478 "{text}"
2479 );
2480 }
2481
2482 #[test]
2483 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
2484 let text = ir(concat!(
2489 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
2490 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
2491 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
2492 "char s[2] = \"hi\";\n",
2493 ));
2494 assert!(
2495 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
2496 "{text}"
2497 );
2498 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
2499 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
2500 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
2503 }
2504
2505 #[test]
2506 fn a_definition_takes_a_parameter_it_left_unnamed() {
2507 let text = ir("int f(int a, int) { return a; }\n");
2511 assert!(text.contains("func @f(i32, i32) -> i32"), "{text}");
2512 assert!(text.contains("block0(%0: i32, %1: i32):"), "{text}");
2513
2514 let text = ir("int g(int, int n) { return n; }\n");
2517 assert!(text.contains("block0(%0: i32, %1: i32):\n return %1\n"), "{text}");
2518 }
2519
2520 #[test]
2521 fn an_assignment_of_a_structure_is_the_object_it_wrote() {
2522 let text = body(concat!(
2527 "struct s { int f; int g; };\n",
2528 "void h(struct s *a, struct s *c, struct s *d, struct s *e)\n",
2529 "{ *d = *e = a[0] = *c; }\n",
2530 ));
2531 assert_eq!(text.matches("memcpy").count(), 3, "{text}");
2532 assert!(text.contains("memcpy %8, %1, size 8, align 4\n"), "{text}");
2533 assert!(text.contains("memcpy %3, %8, size 8, align 4\n"), "{text}");
2534 assert!(text.contains("memcpy %2, %3, size 8, align 4\n"), "{text}");
2535 }
2536
2537 #[test]
2538 fn a_string_literal_stops_at_the_end_of_the_array_it_is_filling() {
2539 let mut opts = options();
2544 opts.emit = EmitKind::Ir;
2545 let result = run(
2546 &opts,
2547 concat!(
2548 "const char a[2][3] = { \"1234\", \"xyz\" };\n",
2549 "static const char b[3][5] = { \"12345\", \"678\", \"9\" };\n",
2550 "union u { struct { char x[4]; char y[4]; }; struct { char z[8]; }; };\n",
2551 "const union u c = { { \"1234\", \"567\" } };\n",
2552 ),
2553 );
2554 let text = result.text();
2555 assert_eq!(
2556 result.messages,
2557 ["/main.c:1:24: warning: initializer-string for array of 'const char' is too long \
2558 (5 chars into 3 available) [E0637]"]
2559 );
2560 assert!(text.contains("global @a : bytes 6 = { bytes \"123\", bytes \"xyz\" }"), "{text}");
2561 assert!(
2562 text.contains(
2563 "global @b : bytes 15 = { bytes \"12345\", bytes \"678\\00\", zero 1, \
2564 bytes \"9\\00\", zero 3 }"
2565 ),
2566 "{text}"
2567 );
2568 assert!(
2571 text.contains("global @c : bytes 8 = { bytes \"1234\", bytes \"567\\00\" }"),
2572 "{text}"
2573 );
2574 }
2575
2576 #[test]
2577 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
2578 let text = body(concat!(
2582 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
2583 "void g(struct v *);\n",
2584 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
2585 ));
2586 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
2587 }
2588
2589 #[test]
2590 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
2591 let text = ir(concat!(
2596 "struct s { int x; };\n",
2597 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
2598 "int n = (int){ 7 };\n",
2599 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
2600 ));
2601 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
2602 assert!(text.contains("global @n : i32 = 7,"), "{text}");
2603 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
2606 }
2607
2608 #[test]
2609 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
2610 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
2614 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
2615 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
2616 }
2617
2618 #[test]
2619 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
2620 let text = ir("unsigned char foo[1][0];\n");
2624 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
2625 }
2626
2627 #[test]
2628 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
2629 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
2632 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
2633 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
2634 }
2635
2636 #[test]
2637 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
2638 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
2642 assert!(
2643 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
2644 "{text}"
2645 );
2646 }
2647
2648 #[test]
2649 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
2650 let text = body(
2655 "\
2656struct s { int a, b; };
2657struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
2658",
2659 );
2660 assert!(text.contains("block3(%7: ptr)"), "{text}");
2662 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
2663 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
2664 }
2665
2666 #[test]
2667 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
2668 let text = ir("\
2672struct pair { int a, b; };
2673struct pair make(int a, int b);
2674struct pair twice(struct pair p) { return make(p.a, p.b); }
2675");
2676 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
2677 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
2678 }
2679
2680 #[test]
2681 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
2682 let text = ir("\
2686struct big { double v[8]; };
2687struct big grow(struct big b);
2688struct big twice(struct big b) { return grow(grow(b)); }
2689");
2690 assert!(
2691 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
2692 "{text}"
2693 );
2694 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
2695 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
2698 }
2699
2700 #[test]
2701 fn a_structure_passed_to_a_variadic_function_says_so_at_the_call() {
2702 let text = ir("\
2707struct big { double v[8]; };
2708struct pair { int a, b; };
2709int p(const char *, ...);
2710int f(struct big b, struct pair q) { return p(\"\", 1, b, q); }
2711");
2712 assert!(
2713 text.contains("call @p(%4, %5, %2 byval(64, align 8), %6) : (ptr, ...) -> i32"),
2714 "{text}"
2715 );
2716 }
2717
2718 #[test]
2719 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
2720 let body = body(
2723 "\
2724struct pair { int a, b; };
2725struct pair make(int a, int b);
2726int second(void) { return make(1, 2).b; }
2727",
2728 );
2729 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
2730 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
2731 }
2732
2733 #[test]
2734 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
2735 let source = "\
2739struct hfa { float x, y, z; };
2740int take(struct hfa h);
2741int give(struct hfa h) { return take(h); }
2742";
2743 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
2744 let mut opts = options();
2745 opts.emit = EmitKind::Ir;
2746 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
2747 let result = run(&opts, source);
2748 assert_eq!(result.messages, Vec::<String>::new());
2749 assert!(result.text().contains("func @take(f32, f32, f32) -> i32"), "{}", result.text());
2750 }
2751
2752 #[test]
2753 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
2754 let source = "\
2757int use(int *);
2758void f(int n) {
2759 {
2760 int a[n];
2761 use(a);
2762 }
2763 use(0);
2764}
2765";
2766 let body = body(source);
2767 assert!(body.contains("mul.nsw"), "{body}");
2768 assert!(body.contains("stacksave"), "{body}");
2769 assert!(body.contains("alloca %"), "{body}");
2770 assert!(body.contains("stackrestore"), "{body}");
2771 }
2772
2773 #[test]
2774 fn a_goto_out_of_the_scope_of_one_gives_its_stack_back_on_the_way() {
2775 let source = "\
2780int use(int *);
2781int f(int n) {
2782 {
2783 int a[n];
2784 if (use(a)) goto out;
2785 use(0);
2786 }
2787out:
2788 return 0;
2789}
2790";
2791 let body = body(source);
2792 assert_eq!(body.matches("stackrestore").count(), 2, "{body}");
2794 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
2795 assert!(after.starts_with(" %4\n jump block"), "{body}");
2796 }
2797
2798 #[test]
2799 fn a_goto_to_a_label_the_array_is_still_alive_at_leaves_the_stack_alone() {
2800 let source = "\
2804int use(int *);
2805int f(int n) {
2806 int a[n];
2807again:
2808 if (use(a)) goto again;
2809 return 0;
2810}
2811";
2812 let body = body(source);
2813 assert!(body.contains("stacksave"), "{body}");
2814 assert!(!body.contains("stackrestore"), "{body}");
2815 }
2816
2817 #[test]
2818 fn a_goto_back_to_a_label_in_front_of_one_gives_it_back_every_time_round() {
2819 let source = "\
2824int use(int *);
2825int f(int n) {
2826again:
2827 {
2828 int a[n];
2829 if (use(a)) goto again;
2830 }
2831 return 0;
2832}
2833";
2834 let body = body(source);
2835 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
2836 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
2837 assert!(after.starts_with(" %4\n jump block1\n"), "{body}");
2838 }
2839
2840 #[test]
2841 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
2842 let source = "\
2848int f(void);
2849void t(void) {
2850 int count = 10;
2851 for (; count--;) {
2852 int b[f()];
2853 int i;
2854 for (i = 0; i < f(); i++) {
2855 b[i] = count;
2856 }
2857 }
2858}
2859";
2860 let body = body(source);
2861 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
2865 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
2866 let (next, _) = after.split_once("\n\n").expect("a block after the restore");
2867 assert!(next.contains("jump block1("), "{body}");
2868 }
2869
2870 #[test]
2871 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
2872 let source = "\
2875unsigned long f(int n) {
2876 int a[n];
2877 n = 0;
2878 return sizeof a;
2879}
2880";
2881 let body = body(source);
2882 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
2884 }
2885
2886 #[test]
2887 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
2888 let source = "\
2891int use(int);
2892int f(int x) {
2893 return ({
2894 int t = use(x);
2895 t * t;
2896 });
2897}
2898";
2899 let expected = "\
2900block0(%0: i32):
2901 %1 = call @use(%0) : (i32) -> i32
2902 %2 = mul.nsw %1, %1
2903 return %2
2904";
2905 assert_eq!(body(source), expected);
2906 }
2907
2908 #[test]
2909 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
2910 let source = "int f(int x) { return ({ return x; 0; }); }\n";
2914 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
2915 }
2916
2917 #[test]
2918 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
2919 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
2923 let expected = "\
2924block0(%0: ptr):
2925 %1 = va_arg.f64 %0
2926 %2 = va_arg.f64 %0
2927 %3 = fadd %1, %2
2928 return %3
2929";
2930 assert_eq!(body(source), expected);
2931 }
2932
2933 #[test]
2934 fn one_that_reads_a_structure_answers_where_the_object_is() {
2935 let source = "\
2942struct s { int a; long b; };
2943long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.b; }
2944";
2945 let expected = "\
2946block0(%0: ptr):
2947 %1 = alloca, size 16, align 8
2948 %2 = va_object %0, size 16, align 8
2949 memcpy %1, %2, size 16, align 8
2950 %3 = iconst.i64 8
2951 %4 = ptr_add %1, %3
2952 %5 = load.i64 %4, align 8
2953 return %5
2954";
2955 assert_eq!(body(source), expected);
2956 }
2957
2958 #[test]
2959 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
2960 let source = "\
2964int f(int c) {
2965 void *p = c ? &&one : &&two;
2966 goto *p;
2967one:
2968 return 1;
2969two:
2970 return 2;
2971}
2972";
2973 let expected = "\
2974block0(%0: i32):
2975 %1 = iconst.i32 0
2976 %2 = icmp ne %0, %1
2977 br_if %2, block1, block2
2978
2979block1:
2980 %3 = block_addr block3
2981 jump block4(%3)
2982
2983block2:
2984 %4 = block_addr block5
2985 jump block4(%4)
2986
2987block3:
2988 %5 = iconst.i32 1
2989 return %5
2990
2991block4(%6: ptr):
2992 indirect_br %6, block3, block5
2993
2994block5:
2995 %7 = iconst.i32 2
2996 return %7
2997";
2998 assert_eq!(body(source), expected);
2999 }
3000
3001 #[test]
3002 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
3003 let source = "void **next(void);
3006void f(void) { goto *next(); }
3007";
3008 let expected = "\
3009block0:
3010 %0 = call @next() : () -> ptr
3011 unreachable
3012";
3013 assert_eq!(body(source), expected);
3014 }
3015
3016 #[test]
3017 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
3018 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
3021 let expected = "\
3022block0:
3023 inline_asm.volatile \"mfence\", \"\", \"memory\"()
3024 return
3025";
3026 assert_eq!(body(source), expected);
3027 }
3028
3029 #[test]
3030 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
3031 let source = "\
3034int f(int x, int y) {
3035 int r;
3036 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
3037 return r + y;
3038}
3039";
3040 let expected = "\
3041block0(%0: i32, %1: i32):
3042 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
3043 %4 = add.nsw %2, %3
3044 return %4
3045";
3046 assert_eq!(body(source), expected);
3047 }
3048
3049 #[test]
3050 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
3051 let source = "\
3056struct pair { int a, b; };
3057int f(int x) {
3058 int slot = x;
3059 struct pair p = { x, x };
3060 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
3061 return slot + p.a;
3062}
3063";
3064 let text = body(source);
3065 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
3066 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
3067 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
3068 }
3069
3070 #[test]
3071 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
3072 let source = "\
3077int f(int x) {
3078 int r = 7;
3079 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
3080 return r;
3081away:
3082 return r;
3083}
3084";
3085 let expected = "\
3086block0(%0: i32):
3087 %1 = iconst.i32 7
3088 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
3089
3090block1:
3091 return %2
3092
3093block2:
3094 return %1
3095";
3096 assert_eq!(body(source), expected);
3097 }
3098
3099 #[test]
3100 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
3101 let mut opts = options();
3105 opts.emit = EmitKind::Ir;
3106 for (source, expected) in [
3107 (
3108 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
3109 "output operand constraint lacks '='",
3110 ),
3111 (
3112 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
3113 "lvalue required in 'asm' statement",
3114 ),
3115 (
3116 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
3117 "read-only variable 'g' used as 'asm' output",
3118 ),
3119 (
3120 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
3121 "input operand constraint contains '='",
3122 ),
3123 (
3124 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
3125 "memory input 0 is not directly addressable",
3126 ),
3127 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
3128 (
3129 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
3130 "duplicate asm operand name 'a'",
3131 ),
3132 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
3133 ] {
3134 let result = run(&opts, source);
3135 assert!(result.failed(), "expected this to be reported:\n{source}");
3136 assert!(
3137 result.messages.iter().any(|m| m.contains(expected)),
3138 "{expected}\n{:?}",
3139 result.messages
3140 );
3141 }
3142 }
3143
3144 #[test]
3145 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
3146 let mut opts = options();
3147 opts.emit = EmitKind::Ir;
3148 for source in [
3149 "int f(int n) { void *p = &&out; if (n) goto *p; { int a[n]; out: return 1; } }\n",
3150 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
3151 ] {
3152 let result = run(&opts, source);
3153 assert!(result.failed(), "expected this to be reported:\n{source}");
3154 assert!(
3155 result.messages.iter().any(|m| m.contains("not supported yet")),
3156 "{:?}",
3157 result.messages
3158 );
3159 }
3160 }
3161
3162 fn round_trip(source: &str) -> (String, String) {
3164 let printed = ir(source);
3165 let mut opts = options();
3166 opts.emit = EmitKind::Ir;
3167 let mut fs = MemoryFileSystem::new();
3168 fs.insert("/main.ir", printed.clone().into_bytes());
3169 let result = compile_ir(&opts, "/main.ir", &fs);
3170 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
3171 (printed, result.text().to_owned())
3172 }
3173
3174 #[test]
3175 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
3176 let (printed, again) = round_trip(
3180 "struct point { int x, y; };\n static const char greeting[] = \"hi\";\n int puts(const char *);\n int f(int n) { struct point p = { n, 1 }; puts(greeting); return p.x; }\n",
3181 );
3182 assert_eq!(printed, again);
3183 }
3184
3185 #[test]
3186 fn ir_that_is_not_ir_says_which_line_stopped_it() {
3187 let mut opts = options();
3188 opts.emit = EmitKind::Ir;
3189 let mut fs = MemoryFileSystem::new();
3190 let text = "\
3191; ModuleID = 'a.c'
3192; format 0
3193target triple = \"x86_64-unknown-linux-gnu\"
3194target datalayout = \"e-p:64:64-i64:64-S128\"
3195
3196func @f(), linkage(external) {
3197block0:
3198 frobnicate
3199}
3200";
3201 fs.insert("/main.ir", text.as_bytes().to_vec());
3202 let result = compile_ir(&opts, "/main.ir", &fs);
3203 assert!(result.failed());
3204 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
3205 }
3206
3207 #[test]
3208 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
3209 let mut opts = options();
3212 opts.emit = EmitKind::Ir;
3213 let mut fs = MemoryFileSystem::new();
3214 let text = "\
3215; ModuleID = 'a.c'
3216; format 0
3217target triple = \"x86_64-unknown-linux-gnu\"
3218target datalayout = \"e-p:64:64-i64:64-S128\"
3219
3220func @f(), linkage(external) {
3221block0:
3222 %0 = iconst.i32 1
3223 return %0
3224}
3225";
3226 fs.insert("/main.ir", text.as_bytes().to_vec());
3227 let result = compile_ir(&opts, "/main.ir", &fs);
3228 assert!(result.failed());
3229 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
3230 }
3231
3232 #[test]
3233 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
3234 let mut fs = MemoryFileSystem::new();
3236 fs.insert("/main.ir", Vec::new());
3237 let result = compile_ir(&options(), "/main.ir", &fs);
3238 assert!(result.failed());
3239 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
3240 }
3241
3242 #[test]
3243 fn the_printed_ir_reads_back_as_the_same_module() {
3244 let text = ir("\
3247struct point { int x, y; };
3248static const char greeting[] = \"hi\";
3249int table[4] = { 1, 2, 3 };
3250int puts(const char *);
3251double half(double x) { return x / 2.0; }
3252int f(int n) {
3253 int total = 0;
3254 for (int i = 0; i < n; i++) {
3255 if (i == 3) continue;
3256 total += table[i];
3257 }
3258 switch (n) {
3259 case 0: total = 1;
3260 case 1: total++; break;
3261 default: total = -total;
3262 }
3263 struct point p = { total, 1 };
3264 int *q = &p.y;
3265 puts(greeting);
3266 return p.x + *q;
3267}
3268int dispatch(int c) {
3269 void *p = c ? &&one : &&two;
3270 goto *p;
3271one:
3272 return 1;
3273two:
3274 return 2;
3275}
3276int assembly(int x, int *p) {
3277 int r;
3278 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
3279 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
3280 return r;
3281away:
3282 return 0;
3283}
3284");
3285 let mut names = Interner::new();
3286 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
3287 assert_eq!(rucc_ir::print(&module, &names), text);
3288 }
3289}