1use std::path::Path;
14
15use rucc_base::Interner;
16use rucc_codegen::coverage::Fired;
17use rucc_codegen::pipeline::{self, Machine};
18use rucc_diag::{Diagnostic, Severity, Span};
19use rucc_lex::{Convert, Keywords, PpToken, convert};
20use rucc_sema::{Checker, Context as CheckContext};
21use rucc_session::{EmitKind, FileSystem, Options, Session};
22use rucc_target::TargetInfo;
23
24use crate::preprocess::render;
25
26#[derive(Debug, Clone, PartialEq, Eq, Default)]
33pub enum Artifact {
34 #[default]
37 Nothing,
38 Text(String),
40 Object(Vec<u8>),
42}
43
44impl Artifact {
45 #[must_use]
47 pub fn bytes(&self) -> &[u8] {
48 match self {
49 Artifact::Nothing => &[],
50 Artifact::Text(text) => text.as_bytes(),
51 Artifact::Object(bytes) => bytes,
52 }
53 }
54}
55
56#[derive(Debug, Clone, PartialEq, Eq)]
58pub struct Compiled {
59 pub artifact: Artifact,
61 pub messages: Vec<String>,
63 pub errors: u32,
65 pub fired: Fired,
71 pub dumps: Vec<rucc_opt::Dump>,
77 pub remarks: String,
83}
84
85impl Compiled {
86 #[must_use]
88 pub fn failed(&self) -> bool {
89 self.errors > 0
90 }
91
92 #[must_use]
97 pub fn text(&self) -> &str {
98 match &self.artifact {
99 Artifact::Text(text) => text,
100 _ => "",
101 }
102 }
103}
104
105#[must_use]
118pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
119 let mut sess = Session::new(opts.clone());
120 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
124 let mut diagnostics: Vec<Diagnostic> = Vec::new();
125 let mut fired = Fired::new();
127 let mut dumps = Vec::new();
129 let mut remarks = String::new();
130
131 let bytes = match fs.read(Path::new(name)) {
132 Ok(bytes) => bytes,
133 Err(e) => return failure(format!("{name}: {e}")),
134 };
135 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
136 return failure(format!("{name}: the source map has no room left for this file"));
137 };
138
139 let mut pp = rucc_pp::Preprocessor::new();
143 let predef = rucc_pp::Predef::for_options(opts);
144 let expanded: Vec<PpToken> = {
145 let mut cx = rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
146 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
147 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
148 return failure(format!("{name}: the source map has no room for the built in macros"));
149 }
150 pp.run(file, &mut cx).iter().map(|token| token.to_pp()).collect()
151 };
152 diagnostics.extend(pp.take_diagnostics());
153
154 let cx = Convert {
157 keywords: &keywords,
158 interner: &sess.interner,
159 target: &sess.target,
160 std: opts.std,
161 gnu: opts.gnu_extensions,
162 pedantic: opts.pedantic,
163 };
164 let (tokens, complaints) = convert(&expanded, &cx);
165 diagnostics.extend(complaints);
166
167 let parsed = rucc_parse::parse(
168 &tokens,
169 rucc_parse::Context {
170 interner: &sess.interner,
171 std: opts.std,
172 gnu: opts.gnu_extensions,
173 pedantic: opts.pedantic,
174 error_limit: opts.error_limit as usize,
175 },
176 );
177 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
178 diagnostics.extend(parsed.diagnostics);
179
180 let mut artifact = Artifact::Nothing;
181 let mut instrumented = Instrumented::default();
184 if !parse_failed {
185 let mut checker = Checker::new(
186 &parsed.ast,
187 CheckContext {
188 names: &sess.interner,
189 target: &sess.target,
190 std: opts.std,
191 gnu: opts.gnu_extensions,
192 pedantic: opts.pedantic,
193 error_limit: opts.error_limit as usize,
194 builtins: opts.builtins && opts.hosted,
197 no_builtin: &opts.no_builtin,
198 },
199 );
200 checker.check_unit();
201 let checked = checker.finish();
202 if !checked.failed() {
203 match opts.emit {
204 EmitKind::Tast => {
205 artifact = Artifact::Text(rucc_sema::print(
206 &checked.tast,
207 &checked.types,
208 &sess.interner,
209 ));
210 }
211 EmitKind::TypeGranules => {
215 artifact = Artifact::Text(rucc_types::granule_report(
216 &checked.types,
217 &sess.interner,
218 &sess.target,
219 ));
220 }
221 EmitKind::Ir
222 | EmitKind::MirFinal
223 | EmitKind::Asm
224 | EmitKind::Object
225 | EmitKind::Executable
226 | EmitKind::SafetySummary => {
227 let mut lowered = rucc_lower::lower(
228 name,
229 rucc_lower::Context {
230 tast: &checked.tast,
231 types: &checked.types,
232 target: &sess.target,
233 names: &mut sess.interner,
234 },
235 );
236 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
240 if !failed {
241 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
246 for error in errors {
247 diagnostics.push(internal(&format!("invalid IR, {error}")));
248 }
249 } else if let Err(complaints) =
250 instrument(&mut lowered.module, &mut sess.interner, opts)
251 .map(|done| instrumented = done)
252 {
253 diagnostics.extend(complaints);
254 } else if let Err(complaints) = optimize(
255 &mut lowered.module,
256 &sess.interner,
257 opts,
258 name,
259 &mut dumps,
260 &mut remarks,
261 ) {
262 diagnostics.extend(complaints);
263 } else if opts.emit == EmitKind::SafetySummary {
264 artifact = Artifact::Text(
269 rucc_safety::summarize(
270 &lowered.module,
271 &sess.interner,
272 name,
273 opts.safety.as_str(),
274 instrumented.checks,
275 instrumented.interposed,
276 instrumented.crossings,
277 )
278 .render(),
279 );
280 } else if opts.emit == EmitKind::Ir {
281 artifact =
286 Artifact::Text(rucc_ir::print(&lowered.module, &sess.interner));
287 } else {
288 match generate(
291 &mut lowered.module,
292 &mut sess.interner,
293 &sess.target,
294 opts,
295 &mut fired,
296 ) {
297 Ok(made) => artifact = made,
298 Err(complaints) => diagnostics.extend(complaints),
299 }
300 }
301 }
302 diagnostics.extend(lowered.diagnostics);
303 }
304 _ => {}
305 }
306 }
307 diagnostics.extend(checked.diagnostics);
308 }
309
310 let mut messages = Vec::with_capacity(diagnostics.len());
311 let mut errors = 0;
312 for diag in &diagnostics {
313 if !opts.warnings && diag.severity == Severity::Warning {
317 continue;
318 }
319 if diag.severity.is_fatal()
320 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
321 {
322 errors += 1;
323 }
324 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
325 }
326 if errors > 0 {
327 artifact = Artifact::Nothing;
329 }
330 Compiled { artifact, messages, errors, fired, dumps, remarks }
333}
334
335#[must_use]
345pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
346 let mut sess = Session::new(opts.clone());
347 if opts.emit != EmitKind::Ir {
348 return failure(format!(
349 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
350 the C in front of it became",
351 opts.emit.as_str()
352 ));
353 }
354 let bytes = match fs.read(Path::new(name)) {
355 Ok(bytes) => bytes,
356 Err(e) => return failure(format!("{name}: {e}")),
357 };
358 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
359 return failure(format!("{name}: this is not text, so it is not IR"));
360 };
361
362 let module = match rucc_ir::parse(text, &mut sess.interner) {
363 Ok(module) => module,
364 Err(error) => {
365 return failure(format!("{name}:{}: {}", error.line, error.message));
366 }
367 };
368 let mut diagnostics: Vec<Diagnostic> = Vec::new();
369 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
370 for error in errors {
371 diagnostics.push(invalid(&format!("invalid IR, {error}")));
372 }
373 }
374 let mut messages = Vec::with_capacity(diagnostics.len());
375 for diag in &diagnostics {
376 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
377 }
378 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
379 let artifact = if errors > 0 {
380 Artifact::Nothing
381 } else {
382 Artifact::Text(rucc_ir::print(&module, &sess.interner))
383 };
384 Compiled {
386 artifact,
387 messages,
388 errors,
389 fired: Fired::new(),
390 dumps: Vec::new(),
391 remarks: String::new(),
392 }
393}
394
395fn instrument(
418 module: &mut rucc_ir::Module,
419 names: &mut Interner,
420 opts: &Options,
421) -> Result<Instrumented, Vec<Diagnostic>> {
422 if !opts.safety.instruments() {
423 return Ok(Instrumented::default());
424 }
425 let checks = rucc_safety::run(module);
426 let interposed = rucc_safety::redirect(module, names);
431 let crossings = rucc_safety::witness(module, names);
434 match rucc_ir::verify(module, names) {
435 Ok(()) => Ok(Instrumented { checks, interposed, crossings }),
436 Err(errors) => Err(errors
437 .iter()
438 .map(|e| internal(&format!("invalid IR after check insertion, {e}")))
439 .collect()),
440 }
441}
442
443#[derive(Clone, Copy, Debug, Default)]
449struct Instrumented {
450 checks: rucc_safety::Counts,
452 interposed: usize,
454 crossings: rucc_safety::Sites,
456}
457
458fn optimize(
470 module: &mut rucc_ir::Module,
471 names: &Interner,
472 opts: &Options,
473 file: &str,
474 dumps: &mut Vec<rucc_opt::Dump>,
475 remarks: &mut String,
476) -> Result<(), Vec<Diagnostic>> {
477 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
478 settings.toggles.clone_from(&opts.passes);
479 settings.fuel = opts.pass_fuel.iter().cloned().collect();
480 settings.global_fuel = opts.pass_fuel_global;
481 settings.verify |= opts.verify_each;
482 for (on, spec) in &opts.pass_gates {
483 if let Err(why) = settings.gates.add(*on, spec) {
486 return Err(vec![internal(&why)]);
487 }
488 }
489 for spec in &opts.dump_ir {
490 if let Err(why) = settings.dumps.add(spec) {
493 return Err(vec![internal(&why)]);
494 }
495 }
496 let mut wants = rucc_opt::Wants::none();
497 for spec in &opts.opt_info {
498 if let Err(why) = wants.add(spec) {
501 return Err(vec![internal(&why)]);
502 }
503 }
504 let report = rucc_opt::run(module, names, &settings);
505 remarks.push_str(&rucc_opt::optinfo::render(file, &report, names, wants));
506 dumps.extend(report.dumps);
507 match report.broke.is_empty() {
508 true => Ok(()),
509 false => Err(report.broke.iter().map(|why| internal(why)).collect()),
510 }
511}
512
513fn generate(
532 module: &mut rucc_ir::Module,
533 names: &mut Interner,
534 target: &TargetInfo,
535 opts: &Options,
536 fired: &mut Fired,
537) -> Result<Artifact, Vec<Diagnostic>> {
538 let Some(machine) = Machine::for_target(target) else {
539 return Err(vec![unsupported(&format!(
540 "there is no back end for {} in this compiler yet, so there is nothing to generate",
541 target.triple
542 ))]);
543 };
544 let flags = pipeline::Flags { frame_pointer: opts.frame_pointer, red_zone: opts.red_zone };
545
546 if opts.safety.instruments() {
555 rucc_safety::lower(module, names);
556 if let Err(errors) = rucc_ir::verify(module, names) {
557 return Err(errors
558 .iter()
559 .map(|e| internal(&format!("invalid IR after check lowering, {e}")))
560 .collect());
561 }
562 }
563
564 let mut funcs = Vec::new();
565 let mut complaints = Vec::new();
566 for id in module.funcs() {
567 if module[id].is_declaration() {
568 continue;
569 }
570 match pipeline::compile_recording(&mut module[id], names, &machine, flags, fired) {
571 Ok(func) => funcs.push(func),
572 Err(why) => {
573 let name = names.resolve(module[id].name).to_owned();
574 let span = why.inst().map_or(Span::DUMMY, |inst| module[id].span(inst));
577 let said = format!("cannot generate code for '{name}': {why}");
578 complaints.push(unsupported_at(&said, span));
579 }
580 }
581 }
582 if !complaints.is_empty() {
583 return Err(complaints);
584 }
585 let (globals, aliases) = match opts.emit {
591 EmitKind::Asm | EmitKind::Object | EmitKind::Executable => (
592 rucc_asm::globals(module, names).map_err(refused)?,
593 rucc_asm::aliases(module, names).map_err(refused)?,
594 ),
595 _ => (rucc_asm::Globals::default(), Vec::new()),
596 };
597 match opts.emit {
601 EmitKind::Asm => rucc_asm::print(&funcs, &globals, &aliases, names, target)
602 .map(Artifact::Text)
603 .map_err(refused),
604 EmitKind::Object | EmitKind::Executable => {
607 let text = rucc_asm::assemble(&funcs, names, target).map_err(refused)?;
608 let data = globals.image();
609 rucc_object::write(&text, &data, &aliases, target).map(Artifact::Object).map_err(
612 |why| match why {
613 rucc_object::Error::Format { .. } => vec![unsupported(&why.to_string())],
614 rucc_object::Error::Refused { .. } => vec![internal(&why.to_string())],
615 },
616 )
617 }
618 _ => Ok(Artifact::Text(rucc_mir::print(&funcs, names, target.regs))),
619 }
620}
621
622fn refused(why: rucc_asm::Error) -> Vec<Diagnostic> {
628 match why {
629 rucc_asm::Error::Thread { .. } | rucc_asm::Error::IFunc { .. } => {
630 vec![unsupported(&why.to_string())]
631 }
632 _ => vec![internal(&why.to_string())],
633 }
634}
635
636fn unsupported(message: &str) -> Diagnostic {
642 unsupported_at(message, Span::DUMMY)
643}
644
645fn unsupported_at(message: &str, span: Span) -> Diagnostic {
651 Diagnostic::error(message.to_owned(), span)
652 .with_code("E0653")
653 .note("this construct is not lowered yet, see https://github.com/tamnd/rucc/issues", span)
654}
655
656fn invalid(message: &str) -> Diagnostic {
658 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
659}
660
661fn internal(message: &str) -> Diagnostic {
663 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
664 .with_code("E0652")
665 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
666}
667
668fn failure(message: String) -> Compiled {
671 Compiled {
672 artifact: Artifact::Nothing,
673 messages: vec![format!("rucc: error: {message}")],
674 errors: 1,
675 fired: Fired::new(),
676 dumps: Vec::new(),
677 remarks: String::new(),
678 }
679}
680
681#[cfg(test)]
682mod tests {
683 use rucc_session::{MemoryFileSystem, Std};
684 use rucc_target::Triple;
685
686 use super::*;
687
688 fn options() -> Options {
689 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
690 opts.emit = EmitKind::Tast;
691 opts
692 }
693
694 fn run(opts: &Options, source: &str) -> Compiled {
695 let mut fs = MemoryFileSystem::new();
696 fs.insert("/main.c", source.to_owned().into_bytes());
697 compile(opts, "/main.c", &fs)
698 }
699
700 fn freestanding() -> Options {
704 let mut opts = options();
705 opts.hosted = false;
706 opts.search.push_system(rucc_session::runtime::DIR);
707 opts
708 }
709
710 fn shipped(source: &str) -> String {
712 let result = run(&freestanding(), source);
713 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
714 result.text().to_owned()
715 }
716
717 fn tast(source: &str) -> String {
719 let result = run(&options(), source);
720 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
721 result.text().to_owned()
722 }
723
724 #[test]
725 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
726 let text = shipped(concat!(
727 "#include <stdarg.h>\n",
728 "int sum(int n, ...) {\n",
729 " va_list ap, copy;\n",
730 " va_start(ap, n);\n",
731 " va_copy(copy, ap);\n",
732 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
733 " va_end(ap);\n",
734 " va_end(copy);\n",
735 " return total;\n",
736 "}\n",
737 ));
738 assert!(text.contains("va-start"), "{text}");
739 assert!(text.contains("va-copy"), "{text}");
740 assert!(text.contains("va-arg"), "{text}");
741 assert!(text.contains("va-end"), "{text}");
742 }
743
744 #[test]
748 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
749 let text = shipped(concat!(
750 "#define __need___va_list\n",
751 "#include <stdarg.h>\n",
752 "int vprint(const char *f, __gnuc_va_list ap);\n",
753 "#ifdef va_start\n",
754 "#error va_start should not be defined\n",
755 "#endif\n",
756 "#ifdef _VA_LIST_DEFINED\n",
757 "#error va_list should not have been made\n",
758 "#endif\n",
759 ));
760 assert!(text.contains("vprint"), "{text}");
761 }
762
763 #[test]
766 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
767 let text = shipped(concat!(
768 "#define __need_size_t\n",
769 "#include <stddef.h>\n",
770 "#ifdef offsetof\n",
771 "#error offsetof should not be defined yet\n",
772 "#endif\n",
773 "#define __need_ptrdiff_t\n",
774 "#include <stddef.h>\n",
775 "#include <stddef.h>\n",
776 "size_t a;\n",
777 "ptrdiff_t b;\n",
778 "wchar_t c;\n",
779 "max_align_t d;\n",
780 "void *e = NULL;\n",
781 "struct P { int x; long y; };\n",
782 "size_t f = offsetof(struct P, y);\n",
783 ));
784 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
785 assert!(text.contains("decl #1 b : long"), "{text}");
786 }
787
788 #[test]
789 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
790 let text = shipped(concat!(
791 "#include <limits.h>\n",
792 "#include <float.h>\n",
793 "int bits = CHAR_BIT;\n",
794 "long big = LONG_MAX;\n",
795 "int low = INT_MIN;\n",
796 "int radix = FLT_RADIX;\n",
797 "int digits = DBL_MANT_DIG;\n",
798 ));
799 assert!(text.contains("const 8 : int"), "{text}");
800 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
801 assert!(text.contains("const 2 : int"), "{text}");
802 assert!(text.contains("const 53 : int"), "{text}");
803 }
804
805 #[test]
809 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
810 let text = shipped(concat!(
811 "#include <stdint.h>\n",
812 "int64_t a = INT64_C(1);\n",
813 "uint_least16_t b;\n",
814 "intptr_t c;\n",
815 "uintmax_t d = UINTMAX_MAX;\n",
816 "int wide = sizeof(int_fast64_t);\n",
817 ));
818 assert!(text.contains("decl #0 a : long"), "{text}");
819 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
820 assert!(text.contains("decl #2 c : long"), "{text}");
821 }
822
823 #[test]
824 fn the_three_formality_headers_still_have_to_work() {
825 let text = shipped(concat!(
826 "#include <stdbool.h>\n",
827 "#include <stdalign.h>\n",
828 "#include <iso646.h>\n",
829 "#include <stdnoreturn.h>\n",
830 "int t = true and not false;\n",
831 "_Alignas(16) char buf[16];\n",
832 "int a = alignof(long);\n",
833 ));
834 assert!(text.contains("decl #0 t : int"), "{text}");
835 assert!(text.contains("const 8 : unsigned long"), "{text}");
836 }
837
838 #[test]
841 fn every_shipped_header_can_be_included_twice() {
842 let mut source = String::new();
843 for _ in 0..2 {
844 for name in rucc_session::runtime::names() {
845 source.push_str(&format!("#include <{name}>\n"));
846 }
847 }
848 source.push_str("int x;\n");
849 let text = shipped(&source);
850 assert!(text.starts_with("decl #0 x : int"), "{text}");
851 }
852
853 #[test]
854 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
855 let fs = MemoryFileSystem::new();
856 let result = compile(&options(), "/nope.c", &fs);
857 assert!(result.failed());
858 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
859 assert!(result.text().is_empty());
860 }
861
862 #[test]
863 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
864 let text = tast("int x = 1;\n");
865 let expected = "\
866decl #0 x : int object external static defined
867 init
868 +0
869 const 1 : int
870";
871 assert_eq!(text, expected);
872 }
873
874 #[test]
875 fn the_macros_are_expanded_before_anything_is_parsed() {
876 let text = tast("#define N 2\nint a[N];\n");
880 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
881 }
882
883 #[test]
889 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
890 let text = tast(concat!(
891 "#pragma pack(4)\n",
892 "struct s { int a; };\n",
893 "#pragma pack()\n",
894 "int b;\n",
895 "_Pragma(\"GCC visibility push(default)\") int c;\n",
896 ));
897 assert!(text.contains("decl #0 b : int"), "{text}");
898 assert!(text.contains("decl #1 c : int"), "{text}");
899 }
900
901 #[test]
909 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
910 tast(concat!(
911 "struct A { char c; int i; } __attribute__((packed));\n",
912 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
913 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
914 "struct B { char c; int i; } __attribute__((aligned));\n",
917 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
918 "struct C { char c; int i __attribute__((packed)); };\n",
919 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
920 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
921 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
922 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
923 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
924 "struct E { char c; _Alignas(8) int i; };\n",
925 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
926 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
927 "struct F { char c; int i __attribute__((aligned(8))); };\n",
928 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
929 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
932 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
933 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
934 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
935 "struct I { [[gnu::packed]] char c; int i; };\n",
938 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
939 "struct J { char c; [[gnu::packed]] int i; };\n",
940 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
941 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
942 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
943 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
944 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
945 "union L { char c; int i; } __attribute__((packed));\n",
946 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
947 "struct O { char c; int i; } __attribute__((__packed__));\n",
951 "_Static_assert(sizeof(struct O) == 5 && _Alignof(struct O) == 1, \"O\");\n",
952 "struct P { char c; int i; } __attribute__((__aligned__(8)));\n",
953 "_Static_assert(sizeof(struct P) == 8 && _Alignof(struct P) == 8, \"P\");\n",
954 ));
955 }
956
957 #[test]
966 fn the_aligned_attribute_on_a_declaration_raises_what_that_one_object_is_aligned_to() {
967 tast(concat!(
968 "int v __attribute__((aligned(64)));\n",
969 "_Static_assert(__alignof__(v) == 64, \"v\");\n",
970 "__attribute__((aligned(32))) int w;\n",
973 "_Static_assert(__alignof__(w) == 32, \"w\");\n",
974 "[[gnu::aligned(16)]] int x;\n",
975 "_Static_assert(__alignof__(x) == 16, \"x\");\n",
976 "int y __attribute__((aligned(2)));\n",
979 "_Static_assert(__alignof__(y) == 4, \"y\");\n",
980 "void f(void) { int a __attribute__((aligned(128)));\n",
982 "_Static_assert(__alignof__(a) == 128, \"a\"); (void)a; }\n",
983 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
986 "void g(void) __attribute__((aligned(256)));\n",
989 "void g(void) {}\n",
990 "_Static_assert(__alignof__(g) == 256, \"g\");\n",
991 ));
992 }
993
994 #[test]
998 fn what_a_declaration_asked_to_be_aligned_to_is_what_the_assembler_is_told() {
999 let text = asm(concat!(
1000 "int v __attribute__((aligned(64)));\n",
1001 "void g(void) __attribute__((aligned(256)));\n",
1002 "void g(void) {}\n",
1003 "void plain(void) {}\n",
1004 ));
1005 assert!(text.contains("\t.p2align\t6\n\t.type\tv, @object\n"), "{text}");
1006 assert!(text.contains("\t.p2align\t8, 0x90\n\t.globl\tg\n"), "{text}");
1007 assert!(text.contains("\t.p2align\t4, 0x90\n\t.globl\tplain\n"), "{text}");
1008 }
1009
1010 #[test]
1019 fn an_aligned_typedef_says_what_an_object_of_it_is_aligned_to_and_may_lower_it() {
1020 tast(concat!(
1021 "typedef int L __attribute__((aligned(2)));\n",
1022 "_Static_assert(__alignof__(L) == 2, \"L\");\n",
1023 "_Static_assert(_Alignof(L) == 2, \"L alignof\");\n",
1024 "_Static_assert(sizeof(L) == 4, \"L size\");\n",
1026 "struct T { char c; L x; };\n",
1027 "_Static_assert(sizeof(struct T) == 6, \"T\");\n",
1028 "_Static_assert(__builtin_offsetof(struct T, x) == 2, \"T.x\");\n",
1029 "typedef int H __attribute__((aligned(16)));\n",
1031 "_Static_assert(__alignof__(H) == 16, \"H\");\n",
1032 "_Static_assert(sizeof(H) == 4, \"H size\");\n",
1033 "struct U { char c; H x; };\n",
1034 "_Static_assert(sizeof(struct U) == 32, \"U\");\n",
1035 "_Static_assert(__builtin_offsetof(struct U, x) == 16, \"U.x\");\n",
1036 "typedef L M __attribute__((aligned(8)));\n",
1039 "_Static_assert(__alignof__(M) == 8, \"M\");\n",
1040 "typedef L N;\n",
1043 "_Static_assert(__alignof__(N) == 2, \"N\");\n",
1044 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1046 ));
1047 let text = asm(concat!(
1048 "typedef int L __attribute__((aligned(2)));\n",
1049 "typedef int H __attribute__((aligned(16)));\n",
1050 "L low;\n",
1051 "H high;\n",
1052 ));
1053 assert!(text.contains("\t.p2align\t1\n\t.type\tlow, @object\n"), "{text}");
1054 assert!(text.contains("\t.p2align\t4\n\t.type\thigh, @object\n"), "{text}");
1055 }
1056
1057 #[test]
1065 fn the_vector_size_attribute_builds_a_type_of_lanes_and_measures_it_in_bytes() {
1066 tast(concat!(
1067 "typedef int __attribute__((vector_size(16))) v4si;\n",
1068 "_Static_assert(sizeof(v4si) == 16 && _Alignof(v4si) == 16, \"v4si\");\n",
1069 "typedef char __attribute__((vector_size(16))) v16qi;\n",
1070 "_Static_assert(sizeof(v16qi) == 16, \"v16qi\");\n",
1071 "typedef int __attribute__((vector_size(4))) v1si;\n",
1074 "_Static_assert(sizeof(v1si) == 4, \"v1si\");\n",
1075 "typedef float __attribute__((__vector_size__(8))) v2sf;\n",
1077 "_Static_assert(sizeof(v2sf) == 8, \"v2sf\");\n",
1078 "typedef short [[gnu::vector_size(8)]] v4hi;\n",
1079 "_Static_assert(sizeof(v4hi) == 8, \"v4hi\");\n",
1080 "v4si g;\n",
1083 "_Static_assert(sizeof(g[0]) == 4, \"lane\");\n",
1084 "_Static_assert(sizeof(g + g) == 16, \"whole\");\n",
1085 "_Static_assert(sizeof(g + 1) == 16, \"broadcast\");\n",
1088 "_Static_assert(sizeof(v4si[3]) == 48, \"array\");\n",
1090 ));
1091 }
1092
1093 #[test]
1103 fn a_vector_is_written_whole_into_an_array_of_them_and_named_by_a_type_name() {
1104 tast(concat!(
1105 "typedef int __attribute__((vector_size(8))) v2si;\n",
1106 "v2si table[] = { (v2si){ 1, 2 }, (v2si){ 3, 4 } };\n",
1107 "_Static_assert(sizeof(table) == 16, \"two of them and not eight lanes\");\n",
1108 "v2si written = (int __attribute__((vector_size(8)))){ 5, 6 };\n",
1110 "_Static_assert(sizeof((int __attribute__((vector_size(16)))){ 0 }) == 16, \"named\");\n",
1111 "v2si lanes[2] = { 1, 2, 3, 4 };\n",
1114 "_Static_assert(sizeof(lanes) == 16, \"still elided\");\n",
1115 ));
1116 }
1117
1118 #[test]
1126 fn a_lane_is_assignable_and_a_shift_takes_a_count_of_its_own_lane() {
1127 let result = run(
1128 &options(),
1129 concat!(
1130 "typedef int __attribute__((vector_size(16))) v4si;\n",
1131 "typedef unsigned __attribute__((vector_size(16))) v4ui;\n",
1132 "void write(v4si *out, v4ui a, v4si b, int n) {\n",
1133 " v4si v = { 1, 2, 3, 4 };\n",
1134 " v[0] = n;\n",
1135 " v[1] += n;\n",
1136 " v[2]++;\n",
1137 " *&v[3] = n;\n",
1138 " v4ui shifted = a >> b;\n",
1140 " shifted <<= b;\n",
1141 " *out = v + (v4si)shifted + (1 << b);\n",
1144 "}\n",
1145 "void refused(const v4si c) {\n",
1148 " c[0] = 1;\n",
1149 "}\n",
1150 ),
1151 );
1152 assert_eq!(result.messages.len(), 1, "{:?}", result.messages);
1153 assert!(result.messages[0].contains("assignment of read-only"), "{:?}", result.messages);
1154 }
1155
1156 #[test]
1163 fn a_record_that_asks_for_the_other_byte_order_is_refused_rather_than_laid_out_in_this_one() {
1164 let opts = options();
1165 let big = "struct s { int i; } __attribute__((scalar_storage_order(\"big-endian\")));\n";
1166 assert_eq!(
1167 run(&opts, big).messages,
1168 ["/main.c:1:36: error: 'scalar_storage_order' is not implemented yet [E0688]\n\
1169 /main.c:1:36: note: every scalar in this record would be read in the wrong byte \
1170 order"]
1171 );
1172
1173 let armoured =
1174 "struct s { int i; } __attribute__((__scalar_storage_order__(\"little-endian\")));\n";
1175 let messages = run(&opts, armoured).messages;
1176 assert!(messages[0].contains("[E0688]"), "{messages:?}");
1177
1178 let front = "struct __attribute__((scalar_storage_order(\"big-endian\"))) s { int i; };\n";
1181 assert!(run(&opts, front).messages[0].contains("[E0688]"), "{front}");
1182 let standard = "struct s { int i; } [[gnu::scalar_storage_order(\"big-endian\")]];\n";
1183 assert!(run(&opts, standard).messages[0].contains("[E0688]"), "{standard}");
1184 }
1185
1186 #[test]
1196 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
1197 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
1199 assert_eq!(
1200 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
1201 1
1202 );
1203 assert_eq!(
1204 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
1205 1
1206 );
1207 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
1208 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
1210 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
1211 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
1213 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
1214 }
1215
1216 fn bit_field_byte(record: &str) -> u64 {
1218 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
1219 let body = body(&source);
1220 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
1221 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
1222 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
1223 }
1224
1225 #[test]
1231 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
1232 tast(concat!(
1233 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
1234 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
1235 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
1236 "struct b { char c; __attribute__((packed)) int i; };\n",
1237 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
1238 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
1239 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
1240 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
1241 ));
1242 }
1243
1244 #[test]
1250 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
1251 tast(concat!(
1252 "#pragma pack(1)\n",
1253 "struct A { char c; int i; };\n",
1254 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1255 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1256 "#pragma pack()\n",
1257 "struct B { char c; int i; };\n",
1258 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
1259 "#pragma pack(2)\n",
1260 "struct C { char c; int i; double d; };\n",
1261 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
1262 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
1263 "struct K { char c; int i __attribute__((aligned(8))); };\n",
1265 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
1266 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
1267 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
1269 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
1270 "#pragma pack()\n",
1271 "#pragma pack(push, 1)\n",
1272 "struct D { char c; short s; };\n",
1273 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
1274 "#pragma pack(pop)\n",
1275 "struct E { char c; short s; };\n",
1276 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
1277 "struct H { char c;\n",
1279 "#pragma pack(1)\n",
1280 " int i; };\n",
1281 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
1282 "#pragma pack(1)\n",
1283 "struct I { char c;\n",
1284 "#pragma pack()\n",
1285 " int i; };\n",
1286 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1287 "#pragma pack()\n",
1288 "#pragma pack(push, 8)\n",
1290 "#pragma pack(push, 1)\n",
1291 "struct P { char c; int i; };\n",
1292 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
1293 "#pragma pack(pop)\n",
1294 "struct Q { char c; int i; };\n",
1295 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
1296 "#pragma pack(pop)\n",
1297 "#pragma pack(16)\n",
1299 "struct R { char c; int i; };\n",
1300 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
1301 "#pragma pack()\n",
1302 "#pragma pack(1)\n",
1303 "struct S { char c; int i : 5; int j : 20; };\n",
1304 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
1305 "union T { char c; int i; };\n",
1306 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
1307 "#pragma pack()\n",
1308 ));
1309 }
1310
1311 #[test]
1315 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
1316 let result = run(
1317 &options(),
1318 concat!(
1319 "#pragma pack 4\n",
1320 "#pragma pack(pop)\n",
1321 "#pragma pack(3)\n",
1322 "#pragma pack(1) junk\n",
1323 "#pragma pack(push, 1\n",
1324 "#pragma pack(x)\n",
1325 "#pragma pack(0)\n",
1328 "#pragma pack(push)\n",
1329 "struct s { char c; int i; };\n",
1330 "#pragma pack(pop)\n",
1331 "#pragma pack(pop, foo)\n",
1332 ),
1333 );
1334 let expected = [
1335 "missing `(` after `#pragma pack` - ignored",
1336 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
1337 "alignment must be a small power of two, not 3",
1338 "junk at end of `#pragma pack`",
1339 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
1340 "unknown action `x` for `#pragma pack` - ignored",
1341 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
1342 ];
1343 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
1344 for (message, want) in result.messages.iter().zip(expected) {
1345 assert!(message.contains(want), "expected {want:?} in {message:?}");
1346 }
1347 }
1348
1349 #[test]
1353 fn the_wide_integer_answers_to_all_three_of_its_names() {
1354 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
1355 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
1356 assert!(text.contains("decl #1 b : __int128"), "{text}");
1357 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
1358 }
1359
1360 #[test]
1361 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
1362 let text = tast("long f(int a, long b) { return a + b; }\n");
1366 assert!(text.contains("convert arithmetic"), "{text}");
1367 }
1368
1369 #[test]
1370 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
1371 for source in [
1372 "#error stop\n",
1373 "int f(void) { return 1 + ; }\n",
1374 "int f(void) { return undeclared; }\n",
1375 ] {
1376 let result = run(&options(), source);
1377 assert!(result.failed(), "expected this to fail:\n{source}");
1378 assert!(
1379 result.text().is_empty(),
1380 "a file that did not compile wrote a tree:\n{source}"
1381 );
1382 }
1383 }
1384
1385 #[test]
1386 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
1387 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
1391 assert_eq!(result.errors, 1, "{:?}", result.messages);
1392 }
1393
1394 #[test]
1395 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
1396 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
1400 assert_eq!(result.errors, 1, "{:?}", result.messages);
1401 }
1402
1403 #[test]
1404 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
1405 let source = "int f(void) { char c = 300; return c; }\n";
1406 let plain = run(&options(), source);
1407 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
1408 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
1409 assert!(!plain.text().is_empty(), "a warning is not a reason to write nothing");
1410
1411 let mut opts = options();
1412 opts.warnings_are_errors = true;
1413 let strict = run(&opts, source);
1414 assert!(strict.failed());
1415 assert!(strict.text().is_empty(), "and under -Werror it is a reason to write nothing");
1416 for message in &strict.messages {
1417 assert!(!message.contains("warning:"), "{message}");
1418 }
1419 }
1420
1421 #[test]
1422 fn w_drops_the_warning_before_werror_can_promote_it() {
1423 let source = "int f(void) { char c = 300; return c; }\n";
1424 let mut opts = options();
1425 opts.warnings = false;
1426 let quiet = run(&opts, source);
1427 assert_eq!(quiet.messages, Vec::<String>::new());
1428 assert_eq!(quiet.errors, 0);
1429 assert!(!quiet.text().is_empty(), "and the file still compiles");
1430
1431 opts.warnings_are_errors = true;
1434 let both = run(&opts, source);
1435 assert_eq!(both.messages, Vec::<String>::new());
1436 assert!(!both.failed(), "-w -Werror is not an error about a warning nobody saw");
1437 }
1438
1439 #[test]
1440 fn the_dialect_reaches_the_keywords_and_the_checking() {
1441 let source = "typeof(1) x;\n";
1444 let mut opts = options();
1445 opts.std = Std::C23;
1446 opts.gnu_extensions = false;
1447 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
1448
1449 opts.std = Std::C17;
1450 assert!(run(&opts, source).failed());
1451 }
1452
1453 #[test]
1454 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
1455 let mut opts = options();
1456 opts.emit = EmitKind::Object;
1457 let result = run(&opts, "int x = 1;\n");
1458 assert!(!result.failed(), "{:?}", result.messages);
1459 assert!(result.text().is_empty());
1460 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
1463 }
1464
1465 fn mir(source: &str) -> String {
1467 let mut opts = options();
1468 opts.emit = EmitKind::MirFinal;
1469 let result = run(&opts, source);
1470 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1471 result.text().to_owned()
1472 }
1473
1474 #[test]
1480 fn a_function_goes_from_c_to_instructions_with_real_registers_in_them() {
1481 let text = mir("int add(int a, int b) { return a + b; }\n");
1482 assert!(text.starts_with("mfunc @add {"), "{text}");
1483 assert!(text.contains("x64.add_rr_32"), "{text}");
1484 assert!(text.contains("x64.ret"), "{text}");
1485 assert!(!text.contains('%'), "{text}");
1488 }
1489
1490 #[test]
1492 fn a_function_with_no_body_produces_no_machine_function() {
1493 let text = mir("int g(int);\nint f(int a) { return g(a); }\n");
1494 assert_eq!(text.matches("mfunc @").count(), 1, "{text}");
1495 assert!(text.contains("mfunc @f {"), "{text}");
1496 assert!(text.contains("x64.call"), "{text}");
1497 }
1498
1499 #[test]
1501 fn every_definition_in_the_file_is_generated_and_they_keep_their_order() {
1502 let text = mir("int a(int x) { return x; }\nint b(int x) { return x; }\n");
1503 let first = text.find("mfunc @a").expect("the first function");
1504 let second = text.find("mfunc @b").expect("the second function");
1505 assert!(first < second, "{text}");
1506 }
1507
1508 #[test]
1510 fn the_target_decides_which_convention_the_generated_code_follows() {
1511 let mut opts = options();
1512 opts.emit = EmitKind::MirFinal;
1513 let linux = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1514 assert!(linux.contains("$rdi"), "{linux}");
1515
1516 opts.target = "x86_64-pc-windows-msvc".parse::<Triple>().unwrap();
1517 let windows = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1518 assert!(windows.contains("$rcx"), "{windows}");
1519 assert!(!windows.contains("$rdi"), "{windows}");
1520 }
1521
1522 #[test]
1524 fn a_target_this_has_no_back_end_for_is_reported_rather_than_generated() {
1525 let mut opts = options();
1526 opts.emit = EmitKind::MirFinal;
1527 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
1528 let result = run(&opts, "int f(int a) { return a; }\n");
1529 assert!(result.failed());
1530 assert!(result.messages[0].contains("no back end for aarch64"), "{:?}", result.messages);
1531 assert!(result.text().is_empty());
1532 }
1533
1534 #[test]
1541 fn a_construct_the_back_end_cannot_reach_yet_is_reported_against_its_function() {
1542 let mut opts = options();
1543 opts.emit = EmitKind::MirFinal;
1544 let source = "void a(int n) { int v[n]; v[0] = 1; }\n\
1545 void b(int n) { int v[n]; v[0] = 1; }\n";
1546 let result = run(&opts, source);
1547 assert!(result.failed());
1548 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
1549 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
1550 assert!(result.messages[0].contains("no rule lowers a `stacksave`"), "{:?}", result);
1551 assert!(result.messages[1].contains("cannot generate code for 'b'"), "{:?}", result);
1552 assert!(result.text().is_empty());
1553 }
1554
1555 #[test]
1562 fn an_opcode_with_no_name_in_the_rule_language_is_named_by_its_own_spelling() {
1563 let mut opts = options();
1564 opts.emit = EmitKind::MirFinal;
1565 let result = run(&opts, "int f(int a) {\n __int128 wide = a;\n return (int) wide;\n}\n");
1566 assert!(result.failed());
1567 assert!(
1568 result.messages[0].contains("no rule lowers a `sext` producing a `i128`"),
1569 "{result:?}"
1570 );
1571 assert!(result.messages[0].contains(":2:"), "the line the widening is on: {result:?}");
1572 assert!(!result.messages[0].contains("this instruction"), "{result:?}");
1573 }
1574
1575 #[test]
1577 fn the_note_on_unfinished_work_points_at_the_issues_rather_than_at_the_plan() {
1578 let mut opts = options();
1579 opts.emit = EmitKind::MirFinal;
1580 let result = run(&opts, "int f(int a) { __int128 wide = a; return (int) wide; }\n");
1581 assert!(result.failed());
1582 let note = result.messages.iter().find(|line| line.contains("note:")).expect("a note");
1583 assert!(note.contains("https://github.com/tamnd/rucc/issues"), "{note}");
1584 assert!(!note.contains("spec/17-milestones.md"), "{note}");
1585 }
1586
1587 #[test]
1589 fn the_frame_flags_on_the_command_line_reach_the_generated_frame() {
1590 let source = "int f(int a) { return a; }\n";
1591 assert!(!mir(source).contains("$rbp"), "a leaf needs no frame pointer by default");
1592
1593 let mut opts = options();
1594 opts.emit = EmitKind::MirFinal;
1595 opts.frame_pointer = true;
1596 let kept = run(&opts, source).text().to_owned();
1597 assert!(kept.contains("x64.push_64 $rbp"), "{kept}");
1598 }
1599
1600 fn asm(source: &str) -> String {
1602 let mut opts = options();
1603 opts.emit = EmitKind::Asm;
1604 let result = run(&opts, source);
1605 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1606 result.text().to_owned()
1607 }
1608
1609 #[test]
1616 fn a_function_goes_from_c_to_assembly_an_assembler_would_take() {
1617 let text = asm("int add(int a, int b) { return a + b; }\n");
1618 assert!(text.contains("\t.globl\tadd\n"), "{text}");
1619 assert!(text.contains("\t.type\tadd, @function\n"), "{text}");
1620 assert!(text.contains("\nadd:\n"), "{text}");
1621 assert!(text.contains("\taddl\t"), "{text}");
1622 assert!(text.contains("\tret\n"), "{text}");
1623 assert!(text.contains("\t.size\tadd, .-add\n"), "{text}");
1624 assert!(text.contains(".note.GNU-stack"), "{text}");
1627 }
1628
1629 #[test]
1635 fn a_call_through_a_function_pointer_goes_through_the_register_it_is_in() {
1636 let text = asm("int g(int);\nint f(int (*p)(int), int a) { return p(a) + g(a); }\n");
1637 assert!(text.contains("\tcall\t*%"), "{text}");
1638 assert!(text.contains("\tcall\tg\n"), "{text}");
1639 assert!(text.contains("%rdi"), "{text}");
1643 }
1644
1645 #[test]
1647 fn the_address_of_a_global_is_read_from_the_instruction_pointer() {
1648 let text = asm("extern int counter;\nint f(void) { return counter; }\n");
1649 assert!(text.contains("\tleaq\tcounter(%rip), "), "{text}");
1650 }
1651
1652 #[test]
1654 fn a_cast_between_a_pointer_and_an_integer_leaves_the_value_where_it_is() {
1655 let text = asm("long f(void *p) { return (long)p; }\n");
1656 for line in text.lines().filter(|line| line.starts_with('\t') && !line.contains('.')) {
1661 let mnemonic = line.split_whitespace().next().unwrap_or("");
1662 assert!(matches!(mnemonic, "movq" | "ret"), "{line} in\n{text}");
1663 }
1664 }
1665
1666 #[test]
1670 fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
1671 let six = "long a, long b, long c, long d, long e, long f";
1672 let text = asm(&format!("long f({six}, long g, long h) {{ return g + h; }}\n"));
1673
1674 assert!(text.contains("\tmovq\t8(%rsp), "), "{text}");
1678 assert!(text.contains("\tmovq\t16(%rsp), "), "{text}");
1679
1680 let narrow = asm(&format!("int f({six}, int g) {{ return g; }}\n"));
1684 assert!(narrow.contains("\tmovl\t8(%rsp), "), "{narrow}");
1685 let eight =
1686 "double a, double b, double c, double d, double e, double f, double g, double h";
1687 let float = asm(&format!("double f({eight}, double i) {{ return i; }}\n"));
1688 assert!(float.contains("\tmovsd\t8(%rsp), "), "{float}");
1689 }
1690
1691 #[test]
1694 fn a_call_writes_the_arguments_with_no_register_left_at_the_stack_pointer() {
1695 let six = "1, 2, 3, 4, 5, 6";
1696 let decl = "long g(long, long, long, long, long, long, long, long);\n";
1697 let text = asm(&format!("{decl}long f(void) {{ return g({six}, 7, 8); }}\n"));
1698
1699 assert!(text.contains("\tmovq\t%"), "{text}");
1700 assert!(text.contains(", (%rsp)\n"), "{text}");
1701 assert!(text.contains(", 8(%rsp)\n"), "{text}");
1702 assert!(text.contains("\tsubq\t$"), "{text}");
1704
1705 let narrow = "int g(int, int, int, int, int, int, int);\n";
1707 let text = asm(&format!("{narrow}int f(void) {{ return g({six}, 7); }}\n"));
1708 assert!(text.contains("\tmovl\t%"), "{text}");
1709 assert!(text.contains(", (%rsp)\n"), "{text}");
1710 }
1711
1712 #[test]
1715 fn a_variadic_call_counts_registers_and_not_arguments() {
1716 let nine = "1., 2., 3., 4., 5., 6., 7., 8., 9.";
1717 let decl = "int g(int, ...);\n";
1718 let text = asm(&format!("{decl}int f(void) {{ return g(0, {nine}); }}\n"));
1719
1720 assert!(text.contains("\tmovl\t$8, "), "eight registers, not nine: {text}");
1721 assert!(text.contains("\tmovsd\t%"), "{text}");
1722 assert!(text.contains(", (%rsp)\n"), "{text}");
1723 }
1724
1725 #[test]
1730 fn a_variadic_function_writes_the_argument_registers_it_was_handed_into_its_frame() {
1731 let body =
1732 "__builtin_va_list ap; __builtin_va_start(ap, n); __builtin_va_end(ap); return n;";
1733 let text = asm(&format!("int f(int n, ...) {{ {body} }}\n"));
1734
1735 let stores = |mnemonic: &str| text.matches(&format!("\t{mnemonic}\t%")).count();
1738 assert!(text.contains(", 8(%r"), "the second slot, not the first: {text}");
1739 assert!(!text.contains(", 0(%r"), "{text}");
1740 assert_eq!(stores("movsd"), 8, "every vector register: {text}");
1741
1742 assert!(text.contains("\tsubq\t$"), "{text}");
1744 }
1745
1746 #[test]
1749 fn va_start_writes_the_four_fields_the_psabi_describes() {
1750 let start = "__builtin_va_list ap; __builtin_va_start(ap, d);";
1751 let params = "int a, int b, int c, double d";
1752 let text = asm(&format!("int f({params}, ...) {{ {start} return a; }}\n"));
1753
1754 assert!(text.contains(" movl $24, "), "{text}");
1758 assert!(text.contains(" movl $64, "), "{text}");
1759 assert!(text.contains(", 8(%r"), "{text}");
1763 assert!(text.contains(", 16(%r"), "{text}");
1764 let frame: u32 = text
1765 .lines()
1766 .find_map(|line| line.trim().strip_prefix("subq $")?.split(',').next()?.parse().ok())
1767 .expect("a variadic function takes a frame for the save area");
1768 let above = |line: &str| {
1769 let at: u32 = line.trim().strip_prefix("leaq ")?.split('(').next()?.parse().ok()?;
1770 Some(at > frame)
1771 };
1772 assert!(text.lines().filter_map(above).any(|it| it), "{frame}: {text}");
1773 }
1774
1775 #[test]
1778 fn va_arg_branches_on_whether_the_argument_is_still_in_the_save_area() {
1779 let read = "__builtin_va_list ap; __builtin_va_start(ap, n);";
1780 let ints = format!("int f(int n, ...) {{ {read} return __builtin_va_arg(ap, int); }}\n");
1781 let text = asm(&ints);
1782
1783 assert!(text.contains("$40, "), "{text}");
1786 assert!(text.contains(" cmpl "), "{text}");
1787 assert!(text.contains(" setbe "), "unsigned, since an offset is a count of bytes: {text}");
1788
1789 let arg = "__builtin_va_arg(ap, double)";
1790 let text = asm(&format!("double f(int n, ...) {{ {read} return {arg}; }}\n"));
1791 assert!(text.contains("$160, "), "the last vector slot: {text}");
1792 }
1793
1794 #[test]
1797 fn a_structure_assignment_is_a_move_for_each_word_of_it() {
1798 let decl = "struct pair { long a, b; };\n";
1799 let body = "struct pair p = *q; return p.a + p.b;";
1800 let text = asm(&format!("{decl}long f(struct pair *q) {{ {body} }}\n"));
1801
1802 assert!(!text.contains("memcpy"), "nothing calls the library: {text}");
1803 assert!(!text.contains("\tcall"), "{text}");
1804 assert!(text.matches("\tmovq\t").count() >= 4, "two words each way: {text}");
1806 }
1807
1808 #[test]
1811 fn how_wide_a_word_of_a_copy_is_follows_the_alignment() {
1812 let decl = "struct bytes { char a[8]; };\n";
1813 let body = "struct bytes p = *q; return p.a[0];";
1814 let text = asm(&format!("{decl}int f(struct bytes *q) {{ {body} }}\n"));
1815
1816 assert!(text.matches("\tmovb\t").count() >= 16, "a byte at a time: {text}");
1818 }
1819
1820 #[test]
1823 fn the_part_of_an_initialiser_that_names_nothing_is_stored_as_zero() {
1824 let decl = "struct wide { long a, b, c; };\n";
1825 let text = asm(&format!("{decl}long f(void) {{ struct wide w = {{ 7 }}; return w.c; }}\n"));
1826
1827 assert!(!text.contains("memset"), "nothing calls the library: {text}");
1828 assert!(text.contains("\tmovq\t$0, ") || text.contains("$0, %"), "the zero: {text}");
1829 }
1830
1831 #[test]
1834 fn a_copy_too_large_to_unroll_calls_the_runtime() {
1835 let decl = "struct huge { char a[4096]; };\n";
1836 let mut opts = options();
1837 opts.emit = EmitKind::Asm;
1838 let source = format!("{decl}void f(struct huge *p, struct huge *q) {{ *p = *q; }}\n");
1839 let result = run(&opts, &source);
1840 assert!(!result.failed(), "{:?}", result.messages);
1841 let text = result.text();
1842 assert!(text.contains("call") && text.contains("memcpy"), "{text}");
1843 assert!(text.contains("4096"), "the size travels: {text}");
1846 }
1847
1848 #[test]
1851 fn a_realigned_frame_reads_them_through_the_frame_pointer() {
1852 let six = "long a, long b, long c, long d, long e, long f";
1853 let body = "_Alignas(32) long wide[4]; wide[0] = g; return wide[0];";
1854 let text = asm(&format!("long f({six}, long g) {{ {body} }}\n"));
1855
1856 assert!(text.contains("\tandq\t$-32, %rsp"), "{text}");
1860 assert!(text.contains("\tmovq\t16(%rbp), "), "{text}");
1861 assert!(!text.contains("\tmovq\t16(%rsp), "), "{text}");
1862 }
1863
1864 #[test]
1866 fn the_target_decides_how_the_assembly_is_spelled() {
1867 let mut opts = options();
1868 opts.emit = EmitKind::Asm;
1869 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
1870 let text = run(&opts, "int f(void) { return 0; }\n").text().to_owned();
1871 assert!(text.contains("__TEXT,__text"), "{text}");
1872 assert!(text.contains("\n_f:\n"), "{text}");
1873 assert!(!text.contains(".note.GNU-stack"), "{text}");
1874 }
1875
1876 fn obj(source: &str) -> Vec<u8> {
1878 let mut opts = options();
1879 opts.emit = EmitKind::Object;
1880 let result = run(&opts, source);
1881 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1882 match result.artifact {
1883 Artifact::Object(bytes) => bytes,
1884 other => panic!("expected an object, got {other:?}"),
1885 }
1886 }
1887
1888 #[test]
1894 fn a_function_goes_from_c_to_an_object_a_linker_would_take() {
1895 let bytes = obj("int add(int a, int b) { return a + b; }\n");
1896 assert_eq!(&bytes[..4], b"\x7fELF", "an object file starts by saying it is one");
1897 let text = asm("int add(int a, int b) { return a + b; }\n");
1898 assert!(
1899 text.contains("\taddl\t"),
1900 "and the listing of it is the same instructions:\n{text}"
1901 );
1902 }
1903
1904 #[test]
1906 fn a_variable_goes_from_c_to_the_section_it_belongs_in() {
1907 let text = asm("int counter = 42;\nstatic int hidden;\nconst int fixed = 7;\n");
1908 assert!(text.contains("\t.data\n\t.globl\tcounter\n"), "{text}");
1909 assert!(text.contains("\ncounter:\n\t.long\t42\n"), "{text}");
1910 assert!(text.contains("\t.size\tcounter, .-counter\n"), "{text}");
1911 assert!(text.contains("\t.bss\n\t.p2align\t2\n"), "{text}");
1914 assert!(text.contains("\nhidden:\n\t.space\t4\n"), "{text}");
1915 assert!(!text.contains(".globl\thidden"), "{text}");
1916 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
1919 }
1920
1921 #[test]
1928 fn a_bit_field_initializer_writes_every_byte_of_the_value_and_not_only_the_ones_that_are_set() {
1929 let text = asm("struct s { unsigned f : 20; } x = { 0x12300 };\n");
1930 assert!(text.contains("\t.data\n"), "there is something to write: {text}");
1931 assert!(text.contains("\nx:\n\t.ascii\t\"\\000#\\001\"\n"), "and it is the value: {text}");
1932
1933 let text = asm("struct s { unsigned a : 8; unsigned b : 8; } x = { 0, 3 };\n");
1936 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\003\"\n"), "{text}");
1937
1938 let text = asm("struct s { unsigned long long f : 40; } x = { 0x100000 };\n");
1941 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\000\\020\"\n\t.space\t5\n"), "{text}");
1942
1943 let text = asm("struct s { unsigned f : 20; } x = { 0 };\n");
1945 assert!(text.contains("\t.bss\n"), "an object of zeroes is zeroes: {text}");
1946 assert!(text.contains("\nx:\n\t.space\t4\n"), "{text}");
1947 }
1948
1949 #[test]
1951 fn a_string_literal_is_a_variable_with_a_name_no_program_could_write() {
1952 let text = asm("const char *f(void) { return \"hi\"; }\n");
1953 assert!(text.contains("\t.ascii\t\"hi\\000\"\n"), "{text}");
1954 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
1955 let label = text
1956 .lines()
1957 .find(|line| line.starts_with(".Lstr"))
1958 .unwrap_or_else(|| panic!("a label for the literal in\n{text}"));
1959 assert!(!text.contains(&format!(".globl\t{}", label.trim_end_matches(':'))), "{text}");
1960 }
1961
1962 #[test]
1964 fn an_address_in_an_initializer_is_left_to_the_linker() {
1965 let source = "int counter;\nint *p = &counter;\n";
1966 let text = asm(source);
1967 assert!(text.contains("\np:\n\t.quad\tcounter\n"), "{text}");
1968 let bytes = obj(source);
1971 assert!(bytes.windows(8).any(|w| w == b"counter\0"), "the object has to name it");
1972 }
1973
1974 #[test]
1976 fn a_thread_local_variable_is_reported_as_work_that_is_not_done() {
1977 let mut opts = options();
1978 opts.emit = EmitKind::Asm;
1979 let result = run(&opts, "_Thread_local int x = 1;\n");
1980 assert!(result.failed(), "every thread sharing one variable is worse than a message");
1981 assert!(result.messages.iter().any(|m| m.contains("thread-local")), "{:?}", result);
1982 assert!(!result.messages.iter().any(|m| m.contains("internal")), "{:?}", result);
1984 }
1985
1986 #[test]
1988 fn the_object_and_the_listing_are_two_spellings_of_one_compilation() {
1989 let source = "int callee(void); int g(void) { return callee(); }\n";
1993 let bytes = obj(source);
1994 assert!(
1995 bytes.windows(7).any(|w| w == b"callee\0"),
1996 "the object has to name the callee for the linker to find it"
1997 );
1998 let text = asm(source);
1999 assert!(text.contains("\tcall\tcallee\n"), "{text}");
2000 }
2001
2002 #[test]
2008 fn compiling_for_an_executable_produces_an_object_and_not_a_dump() {
2009 let mut opts = options();
2010 opts.emit = EmitKind::Executable;
2012 let result = run(&opts, "int main(void) { return 0; }\n");
2013 assert_eq!(result.messages, Vec::<String>::new());
2014 match result.artifact {
2015 Artifact::Object(bytes) => assert_eq!(&bytes[..4], b"\x7fELF"),
2016 other => panic!("expected an object, got {other:?}"),
2017 }
2018 }
2019
2020 #[test]
2022 fn a_platform_with_no_object_writer_is_said_so_rather_than_written_as_elf() {
2023 let mut opts = options();
2024 opts.emit = EmitKind::Object;
2025 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2026 let result = run(&opts, "int f(void) { return 0; }\n");
2027 assert!(result.failed(), "an object nobody can read is worse than a message");
2028 assert!(
2029 result.messages.iter().any(|m| m.contains("no object writer")),
2030 "{:?}",
2031 result.messages
2032 );
2033 }
2034
2035 fn ir(source: &str) -> String {
2037 let mut opts = options();
2038 opts.emit = EmitKind::Ir;
2039 let result = run(&opts, source);
2040 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2041 result.text().to_owned()
2042 }
2043
2044 fn errors(source: &str) -> Vec<String> {
2046 let mut opts = options();
2047 opts.emit = EmitKind::Ir;
2048 let result = run(&opts, source);
2049 assert!(result.failed(), "expected this to be refused:\n{source}");
2050 result.messages
2051 }
2052
2053 fn body(source: &str) -> String {
2055 let text = ir(source);
2056 let (_, rest) = text.split_once("{\n").expect("a function definition");
2057 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
2058 body.to_owned()
2059 }
2060
2061 fn safe_ir(tier: rucc_session::Safety, source: &str) -> String {
2063 let mut opts = options();
2064 opts.emit = EmitKind::Ir;
2065 opts.safety = tier;
2066 let result = run(&opts, source);
2067 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2068 result.text().to_owned()
2069 }
2070
2071 const READS_THROUGH_A_POINTER: &str = "int read(int *p) { return p[1]; }\n";
2072
2073 #[test]
2074 fn a_build_that_did_not_ask_for_the_monitor_is_compiled_the_way_it_always_was() {
2075 let text = ir(READS_THROUGH_A_POINTER);
2079 assert!(!text.contains("check_"), "{text}");
2080 assert!(!text.contains("cap_of"), "{text}");
2081 }
2082
2083 #[test]
2084 fn asking_for_a_tier_puts_the_checks_in_before_the_optimizer_sees_them() {
2085 let text = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2086 assert!(text.contains("cap_of"), "{text}");
2087 assert!(text.contains("check_bounds"), "{text}");
2088 assert!(text.contains("check_live"), "{text}");
2089 assert!(text.contains("check_deriv"), "{text}");
2091 }
2092
2093 #[test]
2094 fn the_three_tiers_that_are_not_off_all_check_the_same_accesses_so_far() {
2095 let detect = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2099 for tier in [rucc_session::Safety::Enforce, rucc_session::Safety::Kernel] {
2100 assert_eq!(safe_ir(tier, READS_THROUGH_A_POINTER), detect, "{tier}");
2101 }
2102 }
2103
2104 fn summary(tier: rucc_session::Safety, source: &str) -> String {
2106 let mut opts = options();
2107 opts.emit = EmitKind::SafetySummary;
2108 opts.safety = tier;
2109 let result = run(&opts, source);
2110 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2111 result.text().to_owned()
2112 }
2113
2114 #[test]
2115 fn the_summary_counts_the_checks_that_went_in_and_the_ones_still_standing() {
2116 let text = summary(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2117 assert!(text.contains("\"tier\": \"detect\""), "{text}");
2118 assert!(
2120 text.contains("\"bounds\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"),
2121 "{text}"
2122 );
2123 assert!(
2124 text.contains(
2125 "\"derivation\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"
2126 ),
2127 "{text}"
2128 );
2129 }
2130
2131 #[test]
2132 fn a_build_without_the_monitor_summarises_as_a_build_with_no_checks_in_it() {
2133 let text = summary(rucc_session::Safety::Off, READS_THROUGH_A_POINTER);
2137 assert!(text.contains("\"tier\": \"off\""), "{text}");
2138 assert!(
2139 text.contains("\"bounds\": { \"emitted\": 0, \"remaining\": 0, \"discharged\": 0 }"),
2140 "{text}"
2141 );
2142 }
2143
2144 #[test]
2145 fn a_call_the_boundary_models_is_counted_apart_from_one_it_does_not() {
2146 let text = summary(
2147 rucc_session::Safety::Detect,
2148 "void *memcpy(void *, const void *, unsigned long);\n\
2149 int puts(const char *);\n\
2150 void f(char *d, char *s) { memcpy(d, s, 4); puts(d); }\n",
2151 );
2152 assert!(text.contains("\"interposed\": 1"), "{text}");
2153 assert!(text.contains("\"puts\""), "{text}");
2154 assert!(!text.contains("__rucc_wrap_memcpy\""), "{text}");
2158 }
2159
2160 #[test]
2161 fn the_two_directions_a_pointer_crosses_the_boundary_are_counted_apart() {
2162 let text = summary(
2166 rucc_session::Safety::Detect,
2167 "void *notes_open(void);\n\
2168 char *f(char *p) { char *q = notes_open(); return q ? q : p; }\n",
2169 );
2170 assert!(text.contains("\"crossings\": { \"entered\": 1, \"returned\": 1 }"), "{text}");
2171 assert!(text.contains("\"notes_open\""), "{text}");
2172 }
2173
2174 #[test]
2175 fn a_static_function_nobody_takes_the_address_of_is_not_a_crossing() {
2176 let text = summary(
2179 rucc_session::Safety::Detect,
2180 "static int len(const char *p) { return p ? 1 : 0; }\n\
2181 int f(void) { return len(\"x\"); }\n",
2182 );
2183 assert!(text.contains("\"crossings\": { \"entered\": 0, \"returned\": 0 }"), "{text}");
2184 }
2185
2186 fn granules(source: &str) -> String {
2188 let mut opts = options();
2189 opts.emit = EmitKind::TypeGranules;
2190 let result = run(&opts, source);
2191 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2192 result.text().to_owned()
2193 }
2194
2195 #[test]
2196 fn the_granule_report_names_every_record_and_both_keyings() {
2197 let text = granules(
2198 "struct hot { char *p; int a; int b; };\n\
2199 int f(struct hot *h) { return h->a; }\n",
2200 );
2201 assert!(text.contains("struct hot"), "{text}");
2202 assert!(text.contains("every type distinct"), "{text}");
2205 assert!(text.contains("every pointer one type"), "{text}");
2206 assert!(text.contains("budget"), "{text}");
2207 }
2208
2209 #[test]
2210 fn a_record_nothing_uses_is_still_measured() {
2211 let text = granules("struct unused { long a; double b; };\nint f(void) { return 0; }\n");
2214 assert!(text.contains("struct unused"), "{text}");
2215 }
2216
2217 #[test]
2218 fn the_granule_report_stops_before_anything_is_lowered() {
2219 let text = granules(
2223 "struct wide { long double d; };\n\
2224 long double f(long double x) { return x * x; }\n",
2225 );
2226 assert!(text.contains("struct wide"), "{text}");
2227 }
2228
2229 #[test]
2230 fn a_witness_reaches_the_assembler_as_a_call_to_the_runtime() {
2231 let text = safe_asm(rucc_session::Safety::Detect, "char *f(char *p) { return p; }\n");
2234 assert!(text.contains("\tcall\t__rucc_cap_witness\n"), "{text}");
2235 }
2236
2237 #[test]
2238 fn a_pointer_turned_into_an_integer_is_on_the_trust_set() {
2239 let text = summary(
2240 rucc_session::Safety::Detect,
2241 "unsigned long f(int *p) { return (unsigned long) p; }\n",
2242 );
2243 assert!(text.contains("\"exposed\": 1"), "{text}");
2244 }
2245
2246 fn safe_asm(tier: rucc_session::Safety, source: &str) -> String {
2248 let mut opts = options();
2249 opts.emit = EmitKind::Asm;
2250 opts.safety = tier;
2251 let result = run(&opts, source);
2252 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2253 result.text().to_owned()
2254 }
2255
2256 #[test]
2257 fn a_check_reaches_the_assembler_as_a_call_to_the_runtime() {
2258 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2259 assert!(text.contains("\tcall\t__rucc_check_bounds\n"), "{text}");
2260 assert!(text.contains("\tcall\t__rucc_check_live\n"), "{text}");
2261 assert!(text.contains("\tcall\t__rucc_check_deriv\n"), "{text}");
2262 }
2263
2264 #[test]
2265 fn every_check_that_reached_the_assembler_has_a_row_describing_it() {
2266 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2270 let section = format!("\t.section\t{},", rucc_safety::SECTION);
2271 assert_eq!(text.matches(§ion).count(), 3, "{text}");
2272 for index in 0..3 {
2273 let name = format!("__rucc_safety_desc_{index}");
2274 assert!(text.contains(&format!("{name}:\n")), "{text}");
2277 assert!(text.contains(&format!("{name}(%rip)")), "{text}");
2278 }
2279 assert!(!text.contains("__rucc_safety_desc_3"), "{text}");
2280 }
2281
2282 #[test]
2290 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
2291 let text = ir(concat!(
2292 "int g;\n",
2293 "int a = __builtin_constant_p(1);\n",
2294 "int b = __builtin_constant_p(g);\n",
2295 "int c = __builtin_constant_p(\"abc\");\n",
2296 "int d = __builtin_constant_p(&g);\n",
2297 "int e = __builtin_constant_p(1.5);\n",
2298 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
2299 ));
2300 assert!(text.contains("global @a : i32 = 1,"), "{text}");
2301 assert!(text.contains("global @b : i32 = 0,"), "{text}");
2302 assert!(text.contains("global @c : i32 = 1,"), "{text}");
2303 assert!(text.contains("global @d : i32 = 0,"), "{text}");
2304 assert!(text.contains("global @e : i32 = 1,"), "{text}");
2305 assert!(text.contains("global @h : i32 = 11,"), "{text}");
2306 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
2307
2308 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
2312 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
2313 }
2314
2315 #[test]
2324 fn a_call_to_a_library_builtin_reaches_the_library_function() {
2325 let text = body("void f(void) { __builtin_abort(); }\n");
2326 assert_eq!(text, "block0:\n call @abort() : ()\n return\n");
2327
2328 let text = ir("int f(const char *s) { return __builtin_puts(s) + __builtin_strlen(s); }\n");
2331 assert!(text.contains("call @puts(%0) : (ptr) -> i32"), "{text}");
2332 assert!(text.contains("call @strlen(%0) : (ptr) -> i64"), "{text}");
2333 assert!(!text.contains("__builtin_"), "the prefix is not part of any name here:\n{text}");
2334 }
2335
2336 #[test]
2347 fn the_absolute_value_family_is_the_magnitude_and_not_a_call() {
2348 let text = body(concat!(
2349 "long long llabs(long long);\n",
2350 "long long f(long long x) { return llabs(x); }\n",
2351 ));
2352 assert!(text.contains("%1 = iconst.i64 63"), "{text}");
2353 assert!(text.contains("%2 = ashr %0, %1"), "{text}");
2354 assert!(text.contains("%3 = xor %0, %2"), "{text}");
2355 assert!(text.contains("%4 = sub %3, %2"), "{text}");
2356 assert!(!text.contains("call"), "the call does not happen:\n{text}");
2357
2358 let text = body("int abs(int);\nint f(int x) { return abs(x); }\n");
2361 assert!(text.contains("iconst.i32 31"), "{text}");
2362 let text = body("long labs(long);\nlong f(long x) { return labs(x); }\n");
2363 assert!(text.contains("iconst.i64 63"), "{text}");
2364
2365 let text = body("long long f(long long x) { return __builtin_llabs(x); }\n");
2368 assert!(!text.contains("call"), "{text}");
2369
2370 let text = ir(concat!(
2372 "long long llabs(long long b);\n",
2373 "long long g(long long x) { return llabs(x); }\n",
2374 "long long llabs(long long b) { return 7; }\n",
2375 ));
2376 assert!(!text.contains("call @llabs"), "{text}");
2377 }
2378
2379 #[test]
2386 fn a_byte_swap_is_arithmetic_and_not_a_call() {
2387 let text = body("unsigned f(unsigned x) { return __builtin_bswap32(x); }\n");
2388 assert_eq!(text, "block0(%0: i32):\n %1 = bswap %0\n return %1\n");
2389
2390 let text = body("unsigned f(unsigned char c) { return __builtin_bswap32(c); }\n");
2393 assert!(text.contains("zext.i32 %0"), "widened first: {text}");
2394 assert!(text.contains("bswap %1"), "and swapped at four bytes: {text}");
2395 }
2396
2397 #[test]
2403 fn the_byte_swaps_reverse_at_the_width_their_name_says() {
2404 for (name, ty, width) in [
2405 ("__builtin_bswap16", "unsigned short", "i16"),
2406 ("__builtin_bswap32", "unsigned", "i32"),
2407 ("__builtin_bswap64", "unsigned long long", "i64"),
2408 ] {
2409 let source = format!("{ty} f({ty} x) {{ return {name}(x); }}\n");
2410 let text = body(&source);
2411 assert_eq!(
2412 text,
2413 format!("block0(%0: {width}):\n %1 = bswap %0\n return %1\n"),
2414 "{name}"
2415 );
2416 }
2417 }
2418
2419 #[test]
2426 fn the_bit_counts_are_instructions_and_not_calls() {
2427 let text = body("int f(unsigned x) { return __builtin_clz(x); }\n");
2428 assert_eq!(text, "block0(%0: i32):\n %1 = ctlz %0\n return %1\n");
2429
2430 let text = body("int f(unsigned x) { return __builtin_ctz(x); }\n");
2431 assert_eq!(text, "block0(%0: i32):\n %1 = cttz %0\n return %1\n");
2432
2433 let text = body("int f(unsigned x) { return __builtin_popcount(x); }\n");
2434 assert_eq!(text, "block0(%0: i32):\n %1 = ctpop %0\n return %1\n");
2435 }
2436
2437 #[test]
2446 fn the_bit_counts_ask_about_the_width_their_name_says() {
2447 let text = body("int f(unsigned long long x) { return __builtin_clzll(x); }\n");
2448 assert!(text.starts_with("block0(%0: i64):"), "counted at eight bytes: {text}");
2449 assert!(text.contains("%1 = ctlz %0"), "{text}");
2450 assert!(text.contains("trunc.i32 %1"), "and answered in an int: {text}");
2451
2452 let text = body("int f(unsigned long long x) { return __builtin_clz(x); }\n");
2455 assert!(text.contains("trunc.i32 %0"), "narrowed to what was asked about: {text}");
2456 assert!(text.contains("ctlz %1"), "and counted there: {text}");
2457
2458 let text = body("int f(unsigned long x) { return __builtin_popcountl(x); }\n");
2459 assert!(text.contains("%1 = ctpop %0"), "{text}");
2460 assert!(!text.contains("call"), "{text}");
2461 }
2462
2463 #[test]
2468 fn a_parity_is_the_low_bit_of_the_set_bit_count() {
2469 let text = body("int f(unsigned x) { return __builtin_parity(x); }\n");
2470 assert!(text.contains("%1 = ctpop %0"), "{text}");
2471 assert!(text.contains("iconst.i32 1"), "{text}");
2472 assert!(text.contains("and %1, %2"), "the low bit of it: {text}");
2473 }
2474
2475 #[test]
2481 fn the_first_set_bit_is_one_based_and_zero_for_a_zero() {
2482 let text = body("int f(int x) { return __builtin_ffs(x); }\n");
2483 assert!(text.contains("%1 = cttz %0"), "{text}");
2484 assert!(text.contains("%4 = add %1, %2"), "one more than the count: {text}");
2485 assert!(text.contains("%5 = icmp ne %0, %3"), "whether there was a bit at all: {text}");
2486 assert!(text.contains("%7 = sub %3, %6"), "spread to a mask: {text}");
2487 assert!(text.contains("%8 = and %4, %7"), "and kept only then: {text}");
2488 assert!(!text.contains("br_if"), "no branch: {text}");
2489 }
2490
2491 #[test]
2501 fn an_overflow_check_is_arithmetic_and_not_a_call() {
2502 let text =
2503 body("int f(int a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
2504 assert!(text.contains("%3, %4 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
2505 assert!(text.contains("store %3 -> %2"), "{text}");
2506 assert!(!text.contains("call"), "{text}");
2507
2508 let text =
2509 body("int f(int a, int b, int *r) { return __builtin_sub_overflow(a, b, r); }\n");
2510 assert!(text.contains("ssub_overflow.(i32, i1) %0, %1"), "{text}");
2511
2512 let text =
2513 body("int f(int a, int b, int *r) { return __builtin_mul_overflow(a, b, r); }\n");
2514 assert!(text.contains("smul_overflow.(i32, i1) %0, %1"), "{text}");
2515
2516 let text = body(
2519 "int f(unsigned a, unsigned b, unsigned *r) { return __builtin_add_overflow(a, b, r); }\n",
2520 );
2521 assert!(text.contains("uadd_overflow.(i32, i1) %0, %1"), "{text}");
2522 }
2523
2524 #[test]
2532 fn an_overflow_check_is_done_at_a_type_that_holds_every_operand() {
2533 let text = body(
2534 "int f(unsigned a, int b, long long *r) { return __builtin_add_overflow(a, b, r); }\n",
2535 );
2536 assert!(text.contains("%3 = zext.i64 %0"), "the unsigned operand keeps its value: {text}");
2537 assert!(text.contains("%4 = sext.i64 %1"), "and so does the signed one: {text}");
2538 assert!(text.contains("sadd_overflow.(i64, i1) %3, %4"), "{text}");
2539
2540 let text = body(
2543 "int f(long long a, long long b, long long *r) { return __builtin_mul_overflow(a, b, r); }\n",
2544 );
2545 assert!(text.contains("smul_overflow.(i64, i1) %0, %1"), "{text}");
2546 assert!(!text.contains("sext."), "{text}");
2547 assert!(!text.contains("zext.i64"), "{text}");
2549 }
2550
2551 #[test]
2559 fn an_overflow_check_writes_the_wrapped_answer_whether_or_not_it_fit() {
2560 let text =
2561 body("int f(int a, int b, char *r) { return __builtin_sub_overflow(a, b, r); }\n");
2562 assert!(text.contains("%3, %4 = ssub_overflow.(i32, i1) %0, %1"), "{text}");
2563 assert!(text.contains("%5 = trunc.i8 %3"), "narrowed to where it goes: {text}");
2564 assert!(text.contains("%6 = sext.i32 %5"), "and back: {text}");
2565 assert!(text.contains("%7 = icmp ne %6, %3"), "which is whether it fit: {text}");
2566 assert!(text.contains("store %5 -> %2"), "the narrowed value is stored either way: {text}");
2567 assert!(text.contains("%8 = or %4, %7"), "and either bit is an overflow: {text}");
2568 }
2569
2570 #[test]
2577 fn a_call_needing_more_than_sixty_four_bits_says_so() {
2578 let refused = concat!(
2579 "int f(unsigned long long a, long long b, long long *r) {\n",
2580 " return __builtin_add_overflow(a, b, r);\n",
2581 "}\n",
2582 );
2583 let messages = errors(refused);
2584 assert_eq!(messages.len(), 1, "{messages:?}");
2585 assert!(messages[0].contains("E0694"), "{messages:?}");
2586 assert!(messages[0].contains("wider than 64 bits"), "{messages:?}");
2587 }
2588
2589 #[test]
2592 fn an_overflow_check_over_something_that_is_not_an_integer_says_so() {
2593 let messages =
2594 errors("int f(double a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
2595 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
2596
2597 let messages =
2598 errors("int f(int a, int b, double *r) { return __builtin_add_overflow(a, b, r); }\n");
2599 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
2600 }
2601
2602 #[test]
2613 fn an_ordered_access_is_ordered_in_the_ir() {
2614 let text = body("int f(int *p) { return __atomic_load_n(p, 0); }\n");
2615 assert!(text.contains("atomic_load.i32 %0, align 4, relaxed"), "{text}");
2616
2617 let text = body("long f(long *p) { return __atomic_load_n(p, 2); }\n");
2618 assert!(text.contains("atomic_load.i64 %0, align 8, acquire"), "{text}");
2619
2620 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
2621 assert!(text.contains("atomic_store %1 -> %0, align 4, release"), "{text}");
2622
2623 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
2624 assert!(text.contains("atomic_store %1 -> %0, align 4, seq_cst"), "{text}");
2625
2626 let text = body("void f(char *p, int v) { __atomic_store_n(p, v, 0); }\n");
2629 assert!(text.contains("trunc.i8 %1"), "{text}");
2630 assert!(text.contains("atomic_store %2 -> %0, align 1, relaxed"), "{text}");
2631 }
2632
2633 #[test]
2642 fn an_ordered_access_is_the_plain_instruction_on_this_machine() {
2643 let text = asm("int f(int *p) { return __atomic_load_n(p, 5); }\n");
2644 assert!(text.contains("movl\t(%rdi), %eax"), "{text}");
2645 assert!(!text.contains("mfence"), "a load needs no barrier here: {text}");
2646
2647 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
2648 assert!(text.contains("movl\t%esi, (%rdi)"), "{text}");
2649 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
2650
2651 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
2652 let (before, after) = text.split_once("mfence").expect("a barrier: {text}");
2653 assert!(before.contains("movl\t%esi, (%rdi)"), "the store comes first: {text}");
2654 assert!(!after.contains("movl"), "and nothing else is between them: {text}");
2655 }
2656
2657 #[test]
2667 fn a_barrier_is_one_instruction_at_the_strongest_ordering_and_none_below_it() {
2668 assert!(asm("void f(void) { __atomic_thread_fence(5); }\n").contains("mfence"));
2669 assert!(asm("void f(void) { __sync_synchronize(); }\n").contains("mfence"));
2670
2671 for weaker in ["1", "2", "3", "4"] {
2672 let source = format!("void f(void) {{ __atomic_thread_fence({weaker}); }}\n");
2673 assert!(!asm(&source).contains("mfence"), "{weaker} costs nothing here");
2674 }
2675 }
2676
2677 #[test]
2689 fn a_memory_order_an_operation_cannot_carry_is_read_as_the_strongest() {
2690 let mut opts = options();
2691 opts.emit = EmitKind::Ir;
2692
2693 let acquire_store = run(&opts, "void f(int *p, int v) { __atomic_store_n(p, v, 2); }\n");
2694 assert!(acquire_store.text().contains("seq_cst"), "{:?}", acquire_store.text());
2695 assert!(acquire_store.messages[0].contains("[W0333]"), "{:?}", acquire_store.messages);
2696
2697 let nonsense = run(&opts, "int f(int *p) { return __atomic_load_n(p, 99); }\n");
2698 assert!(nonsense.text().contains("seq_cst"), "{:?}", nonsense.text());
2699 assert!(nonsense.messages[0].contains("[W0333]"), "{:?}", nonsense.messages);
2700
2701 let computed = run(&opts, "int f(int *p, int n) { return __atomic_load_n(p, n); }\n");
2702 assert!(computed.text().contains("seq_cst"), "{:?}", computed.text());
2703 assert_eq!(computed.messages, Vec::<String>::new(), "a computed order is not a mistake");
2704 }
2705
2706 #[test]
2718 fn a_conversion_between_a_float_and_the_widest_unsigned_integer_is_written_without_a_branch() {
2719 let text = asm("double f(unsigned long long x) { return (double)x; }\n");
2720 assert!(text.contains("cvtsi2sdq"), "the signed conversion is what runs: {text}");
2721 assert!(text.contains("shrq"), "with the value halved first: {text}");
2722 assert!(text.contains("addsd"), "and doubled after: {text}");
2723 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
2724
2725 let text = asm("unsigned long long f(double d) { return (unsigned long long)d; }\n");
2726 assert!(text.contains("cvttsd2siq"), "the signed conversion is what runs: {text}");
2727 assert!(text.contains("subsd"), "with half the range taken off first: {text}");
2728 assert!(text.contains("shlq\t$63"), "and the top bit put back: {text}");
2729 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
2730 }
2731
2732 #[test]
2743 fn a_plain_name_the_program_took_is_the_programs_own_function() {
2744 let taken = concat!(
2745 "static long long llabs(long long b) { return 7; }\n",
2746 "long long f(long long x) { return llabs(x); }\n",
2747 );
2748 assert!(ir(taken).contains("call @llabs"), "a static definition is the program's own");
2749
2750 let retyped = concat!("int llabs(int b);\n", "int f(int x) { return llabs(x); }\n",);
2751 assert!(ir(retyped).contains("call @llabs"), "another type is another function");
2752
2753 let plain = concat!(
2754 "long long llabs(long long b);\n",
2755 "long long f(long long x) { return llabs(x); }\n",
2756 );
2757 let mut opts = options();
2758 opts.emit = EmitKind::Ir;
2759 assert!(!run(&opts, plain).text().contains("call @llabs"), "the library's by default");
2760
2761 opts.builtins = false;
2762 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin");
2763
2764 opts.builtins = true;
2765 opts.no_builtin = vec!["llabs".to_owned()];
2766 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin-llabs");
2767 let one = "long labs(long b);\nlong f(long x) { return labs(x); }\n";
2768 assert!(!run(&opts, one).text().contains("call @labs"), "one name and not the family");
2769
2770 opts.no_builtin = Vec::new();
2773 opts.builtins = false;
2774 let prefixed = "long long f(long long x) { return __builtin_llabs(x); }\n";
2775 assert!(!run(&opts, prefixed).text().contains("call @llabs"), "the prefix is a promise");
2776 }
2777
2778 #[test]
2790 fn the_hint_builtins_are_their_first_argument_and_the_hint_leaves_no_trace() {
2791 let text = ir(concat!(
2792 "long a = __builtin_expect(7, 1);\n",
2793 "long b = __builtin_expect_with_probability(9, 1, 0.9);\n",
2794 "unsigned long c = sizeof(__builtin_expect((char)1, 1));\n",
2795 ));
2796 assert!(text.contains("global @a : i64 = 7,"), "{text}");
2797 assert!(text.contains("global @b : i64 = 9,"), "{text}");
2798 assert!(text.contains("global @c : i64 = 8,"), "{text}");
2799 assert!(!text.contains("__builtin_expect"), "it is not a call to anything:\n{text}");
2800
2801 let text = body("long f(char c) { return __builtin_expect(c, 1); }\n");
2804 assert!(text.contains("sext"), "{text}");
2805
2806 let one = "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 1\n %2 = sext.i64 %1\n return %0\n";
2810 assert_eq!(body("int f(void) { int i = 0; __builtin_expect(1, i++); return i; }\n"), one);
2811 let source = "int g(void) { int i = 0; __builtin_expect_with_probability(1, i++, 0.5); return i; }\n";
2812 assert_eq!(body(source), one);
2813 }
2814
2815 #[test]
2827 fn a_promise_that_control_does_not_arrive_writes_no_instruction() {
2828 let promised = "int f(int x) { if (x) return 1; __builtin_unreachable(); }\n";
2829 let text = ir(promised);
2830 assert!(text.contains(" unreachable_hint\n"), "{text}");
2831 assert!(!text.contains("call"), "it is not a call to anything:\n{text}");
2832
2833 let after = body("int g(int x) { __builtin_unreachable(); return x; }\n");
2837 assert!(after.contains("return"), "{after}");
2838
2839 let text = asm(promised);
2842 let mine = text.split_once("\nf:\n").expect("a definition").1;
2843 let mine = mine.split_once("\t.size").expect("a definition").0;
2844 let plain = asm("int f(int x) { if (x) return 1; }\n");
2845 let plain = plain.split_once("\nf:\n").expect("a definition").1;
2846 let plain = plain.split_once("\t.size").expect("a definition").0;
2847 assert_eq!(mine, plain);
2848 assert!(mine.trim_end().ends_with("ret"), "{mine}");
2849 assert!(!mine.contains("ud2"), "{mine}");
2850 }
2851
2852 #[test]
2859 fn a_library_builtin_is_diagnosed_under_the_name_the_program_wrote() {
2860 let mut opts = options();
2861 opts.emit = EmitKind::Ir;
2862 let messages = run(&opts, "void f(void) { __builtin_abort(1); }\n").messages;
2863 assert!(
2864 messages.iter().any(|m| m.contains("__builtin_abort")),
2865 "expected the written name in {messages:?}"
2866 );
2867 }
2868
2869 #[test]
2877 fn a_builtin_nothing_lowers_is_refused_by_name() {
2878 let mut opts = options();
2879 opts.emit = EmitKind::Ir;
2880 for (builtin, call) in [
2881 ("__builtin_return_address", "(int)(long)__builtin_return_address(0)"),
2882 ("__builtin_alloca", "(int)(long)__builtin_alloca(8)"),
2883 ("__atomic_exchange_n", "__atomic_exchange_n(&counter, 1, 0)"),
2884 ("__sync_fetch_and_add", "(int)__sync_fetch_and_add(&counter, 1)"),
2885 ] {
2886 let source = format!("int counter;\nint f(void) {{ return {call}; }}\n");
2887 let messages = run(&opts, &source).messages;
2888 let named = messages.iter().any(|m| m.contains(builtin) && m.contains("E0686"));
2889 assert!(named, "expected {builtin} to be refused by name in {messages:?}");
2890 }
2891 }
2892
2893 #[test]
2901 fn what_is_refused_is_the_call_and_not_the_name() {
2902 let text = ir("unsigned long n = sizeof(__builtin_return_address(0));\n");
2903 assert!(text.contains("global @n : i64 = 8,"), "{text}");
2904
2905 let text = ir(concat!(
2906 "void *__builtin_return_address(unsigned x) { return 0; }\n",
2907 "void *f(void) { return __builtin_return_address(0); }\n",
2908 ));
2909 assert!(text.contains("call @__builtin_return_address"), "{text}");
2910 }
2911
2912 #[test]
2917 fn a_static_function_nothing_refers_to_is_not_emitted() {
2918 let text = ir("static int dropped(void) { return 1; }\n\
2919 static int kept(void) { return 2; }\n\
2920 int main(void) { return kept(); }\n");
2921 assert!(text.contains("func @kept"), "{text}");
2922 assert!(!text.contains("dropped"), "{text}");
2923 }
2924
2925 #[test]
2931 fn two_static_functions_that_only_call_each_other_are_both_dropped() {
2932 let text = ir("static int ping(void);\n\
2933 static int pong(void) { return ping(); }\n\
2934 static int ping(void) { return pong(); }\n\
2935 int main(void) { return 0; }\n");
2936 assert!(!text.contains("ping"), "{text}");
2937 assert!(!text.contains("pong"), "{text}");
2938 }
2939
2940 #[test]
2946 fn naming_a_static_function_anywhere_keeps_it() {
2947 let text = ir("static int by_address(void) { return 1; }\n\
2948 static int in_an_image(void) { return 2; }\n\
2949 static int deeper(void) { return 3; }\n\
2950 static int reaches_deeper(void) { return deeper(); }\n\
2951 static int (*table[1])(void) = {in_an_image};\n\
2952 int main(void) {\n\
2953 int (*p)(void) = by_address;\n\
2954 return p() + table[0]() + reaches_deeper();\n\
2955 }\n");
2956 for kept in ["by_address", "in_an_image", "deeper", "reaches_deeper"] {
2957 assert!(text.contains(&format!("func @{kept}")), "expected {kept} in:\n{text}");
2958 }
2959 }
2960
2961 #[test]
2967 fn an_attribute_keeps_a_static_function_nothing_refers_to() {
2968 for attribute in ["used", "retain", "constructor", "destructor", "__used__"] {
2969 let source = format!(
2970 "__attribute__(({attribute})) static int kept(void) {{ return 1; }}\n\
2971 int main(void) {{ return 0; }}\n"
2972 );
2973 let text = ir(&source);
2974 assert!(text.contains("func @kept"), "for {attribute}:\n{text}");
2975 }
2976 }
2977
2978 #[test]
2981 fn a_function_anything_could_call_is_emitted_without_being_called() {
2982 let text =
2983 ir("int nobody_here_calls_it(void) { return 1; }\nint main(void) { return 0; }\n");
2984 assert!(text.contains("func @nobody_here_calls_it"), "{text}");
2985 }
2986
2987 #[test]
2994 fn a_classification_c_has_an_operator_for_is_that_operator() {
2995 for (builtin, operator) in [
2996 ("__builtin_isgreater", "binary >"),
2997 ("__builtin_isgreaterequal", "binary >="),
2998 ("__builtin_isless", "binary <"),
2999 ("__builtin_islessequal", "binary <="),
3000 ] {
3001 let source = format!("int f(double x, double y) {{ return {builtin}(x, y); }}\n");
3002 let text = tast(&source);
3003 assert!(text.contains(&format!("{operator} : int")), "for {builtin}:\n{text}");
3004 }
3005 }
3006
3007 #[test]
3016 fn the_classification_builtins_are_comparisons_and_not_calls() {
3017 let text = body("int f(double x, double y) { return __builtin_isunordered(x, y); }\n");
3018 assert_eq!(
3019 text,
3020 "block0(%0: f64, %1: f64):\n %2 = fcmp uno %0, %1\n %3 = zext.i32 \
3021 %2\n return %3\n"
3022 );
3023
3024 let text = body("int f(double x, double y) { return __builtin_islessgreater(x, y); }\n");
3026 assert!(text.contains("fcmp one %0, %1"), "{text}");
3027
3028 let text = body("int f(double x) { return __builtin_isnan(x); }\n");
3029 assert!(text.contains("fcmp uno %0, %0"), "{text}");
3030
3031 let text = body("int f(double x) { return __builtin_isinf(x); }\n");
3032 assert!(text.contains("fconst.f64 0x7ff0000000000000"), "{text}");
3033 assert!(text.contains("fconst.f64 0xfff0000000000000"), "{text}");
3034 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
3035 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
3036 assert!(text.contains("%5 = or %3, %4"), "{text}");
3037
3038 let text = body("int f(double x) { return __builtin_isfinite(x); }\n");
3041 assert!(text.contains("%3 = fcmp olt %2, %0"), "{text}");
3042 assert!(text.contains("%4 = fcmp olt %0, %1"), "{text}");
3043 assert!(text.contains("%5 = and %3, %4"), "{text}");
3044
3045 let text = body("int f(double x) { return __builtin_signbit(x); }\n");
3046 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
3047 assert!(text.contains("icmp slt %1, %2"), "{text}");
3048
3049 let text = body("int f(long double x) { return __builtin_signbitl(x); }\n");
3052 assert!(text.contains("%1 = bitcast.i80 %0"), "{text}");
3053
3054 let text = body("double g(void);\nint f(void) { return __builtin_isnan(g()); }\n");
3057 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
3058 }
3059
3060 #[test]
3067 fn a_classification_spelling_that_names_a_width_converts_before_it_asks() {
3068 let text = ir(concat!(
3069 "int a = __builtin_isinff(1e300);\n",
3070 "int b = __builtin_isinf(1e300);\n",
3071 "int c = __builtin_isnan(0.0);\n",
3075 "int d = __builtin_signbit(-0.0);\n",
3076 "int e = __builtin_islessgreater(1.0, 2.0);\n",
3077 ));
3078 assert!(text.contains("global @a : i32 = 1,"), "{text}");
3079 assert!(text.contains("global @b : i32 = 0,"), "{text}");
3080 assert!(text.contains("global @c : i32 = 0,"), "{text}");
3081 assert!(text.contains("global @d : i32 = 1,"), "{text}");
3082 assert!(text.contains("global @e : i32 = 1,"), "{text}");
3083 }
3084
3085 #[test]
3087 fn a_classification_builtin_refuses_an_argument_that_is_not_floating_point() {
3088 let mut opts = options();
3089 opts.emit = EmitKind::Ir;
3090 let source = concat!(
3091 "int a(int x) { return __builtin_isnan(x); }\n",
3092 "int b(int x, int y) { return __builtin_isunordered(x, y); }\n",
3093 "int c(double x) { return __builtin_isnan(x, x); }\n",
3094 );
3095 let messages = run(&opts, source).messages;
3096 assert_eq!(
3097 messages,
3098 [
3099 "/main.c:1:23: error: non-floating-point argument in call to function \
3100 '__builtin_isnan' [E0685]",
3101 "/main.c:2:30: error: non-floating-point arguments in call to function \
3102 '__builtin_isunordered' [E0685]",
3103 "/main.c:3:26: error: too many arguments to function '__builtin_isnan' [E0511]",
3104 ]
3105 );
3106 }
3107
3108 #[test]
3117 fn the_last_three_classification_builtins_are_comparisons_and_not_calls() {
3118 let text = body("int f(double x) { return __builtin_isnormal(x); }\n");
3119 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
3123 assert!(text.contains("%2 = iconst.i64 9223372036854775807"), "{text}");
3124 assert!(text.contains("%3 = and %1, %2"), "{text}");
3125 assert!(text.contains("%4 = iconst.i64 4503599627370496"), "{text}");
3126 assert!(text.contains("%5 = iconst.i64 9218868437227405312"), "{text}");
3127 assert!(text.contains("%6 = icmp uge %3, %4"), "{text}");
3128 assert!(text.contains("%7 = icmp ult %3, %5"), "{text}");
3129 assert!(text.contains("%8 = and %6, %7"), "{text}");
3130
3131 let text = body("int f(long double x) { return __builtin_isnormal(x); }\n");
3135 assert!(text.contains("%4 = iconst.i80 27670116110564327424"), "{text}");
3136 assert!(text.contains("%5 = iconst.i80 604453686435277732577280"), "{text}");
3137
3138 let text = body("int f(double x) { return __builtin_isinf_sign(x); }\n");
3139 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
3140 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
3141 assert!(text.contains("%7 = sub %5, %6"), "{text}");
3142
3143 let text = body("int f(double x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n");
3144 assert!(text.contains("fcmp uno %0, %0"), "{text}");
3145 assert!(text.contains("fcmp oeq %0, %6"), "{text}");
3146 assert_eq!(text.matches(" = zext.i32 ").count(), 4, "{text}");
3150 assert_eq!(text.matches(" = xor ").count(), 4, "{text}");
3151 assert!(!text.contains("call"), "{text}");
3152
3153 let text = body(concat!(
3156 "double g(void);\n",
3157 "int f(void) { return __builtin_fpclassify(0, 1, 2, 3, 4, g()); }\n",
3158 ));
3159 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
3160 }
3161
3162 #[test]
3169 fn the_last_three_classification_builtins_fold_where_their_operand_is_a_constant() {
3170 let text = ir(concat!(
3171 "int a = __builtin_isnormal(1.0);\n",
3172 "int b = __builtin_isnormal(0.0);\n",
3173 "int c = __builtin_isnormal(1.0 / 0.0);\n",
3174 "int d = __builtin_isinf_sign(-1.0 / 0.0);\n",
3175 "int e = __builtin_isinf_sign(1.0);\n",
3176 "int g = __builtin_fpclassify(0, 1, 2, 3, 4, 0.0);\n",
3177 "int h = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0);\n",
3178 "int i = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0 / 0.0);\n",
3179 ));
3180 assert!(text.contains("global @a : i32 = 1,"), "{text}");
3181 assert!(text.contains("global @b : i32 = 0,"), "{text}");
3182 assert!(text.contains("global @c : i32 = 0,"), "{text}");
3183 assert!(text.contains("global @d : i32 = -1,"), "{text}");
3184 assert!(text.contains("global @e : i32 = 0,"), "{text}");
3185 assert!(text.contains("global @g : i32 = 4,"), "{text}");
3186 assert!(text.contains("global @h : i32 = 2,"), "{text}");
3187 assert!(text.contains("global @i : i32 = 1,"), "{text}");
3188 }
3189
3190 #[test]
3196 fn fpclassify_refuses_an_answer_that_is_not_an_integer_constant() {
3197 let mut opts = options();
3198 opts.emit = EmitKind::Ir;
3199 let source = concat!(
3200 "int a(double x, int n) { return __builtin_fpclassify(0, 1, n, 3, 4, x); }\n",
3201 "int b(double x) { return __builtin_fpclassify(0, 1, 2, 3, x); }\n",
3202 "int c(int x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n",
3203 );
3204 let messages = run(&opts, source).messages;
3205 assert_eq!(
3206 messages,
3207 [
3208 "/main.c:1:60: error: non-const integer argument 3 in call to function \
3209 '__builtin_fpclassify' [E0687]",
3210 "/main.c:2:26: error: too few arguments to function '__builtin_fpclassify' \
3211 [E0511]",
3212 "/main.c:3:23: error: non-floating-point argument in call to function \
3213 '__builtin_fpclassify' [E0685]",
3214 ]
3215 );
3216 }
3217
3218 #[test]
3226 fn a_builtin_whose_answer_is_a_constant_is_one_and_not_a_call() {
3227 let text = ir(concat!(
3228 "double a = __builtin_inf();\n",
3229 "float b = __builtin_huge_valf();\n",
3230 "long double c = __builtin_infl();\n",
3231 "double d = __builtin_huge_val();\n",
3232 ));
3233 assert!(text.contains("global @a : f64 = 0x7ff0000000000000,"), "{text}");
3234 assert!(text.contains("global @b : f32 = 0x7f800000,"), "{text}");
3235 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
3236 assert!(text.contains("global @d : f64 = 0x7ff0000000000000,"), "{text}");
3237 assert!(!text.contains("call"), "{text}");
3238 }
3239
3240 #[test]
3249 fn a_nan_is_written_with_the_payload_the_program_asked_for() {
3250 let text = ir(concat!(
3251 "double a = __builtin_nan(\"\");\n",
3252 "double b = __builtin_nan(\"0x1\");\n",
3253 "double c = __builtin_nan(\"010\");\n",
3255 "double d = __builtin_nans(\"\");\n",
3256 "double e = __builtin_nans(\"0x1\");\n",
3257 "float f = __builtin_nanf(\"0x1\");\n",
3258 "float g = __builtin_nansf(\"\");\n",
3259 "long double h = __builtin_nansl(\"\");\n",
3260 ));
3261 assert!(text.contains("global @a : f64 = 0x7ff8000000000000,"), "{text}");
3262 assert!(text.contains("global @b : f64 = 0x7ff8000000000001,"), "{text}");
3263 assert!(text.contains("global @c : f64 = 0x7ff8000000000008,"), "{text}");
3264 assert!(text.contains("global @d : f64 = 0x7ff4000000000000,"), "{text}");
3265 assert!(text.contains("global @e : f64 = 0x7ff0000000000001,"), "{text}");
3266 assert!(text.contains("global @f : f32 = 0x7fc00001,"), "{text}");
3267 assert!(text.contains("global @g : f32 = 0x7fa00000,"), "{text}");
3268 assert!(text.contains("f80 0x7fffa000000000000000"), "{text}");
3269
3270 let text = ir(concat!(
3273 "double f(const char *p) { return __builtin_nan(p); }\n",
3274 "double g(void) { return __builtin_nans(\"1x\"); }\n",
3275 ));
3276 assert_eq!(text.matches("call @nan(").count(), 1, "{text}");
3277 assert_eq!(text.matches("call @nans(").count(), 1, "{text}");
3278 }
3279
3280 #[test]
3288 fn the_length_and_the_order_of_a_string_literal_are_known_here() {
3289 let text = ir(concat!(
3290 "unsigned long a = __builtin_strlen(\"hello\");\n",
3291 "unsigned long b = __builtin_strlen(\"a\\0bc\");\n",
3292 "int c = __builtin_strcmp(\"X\", \"X\\376\") < 0;\n",
3293 "int d = __builtin_strcmp(\"abc\", \"abc\");\n",
3294 "int e = __builtin_strcmp(\"abc\", \"ab\") > 0;\n",
3295 ));
3296 assert!(text.contains("global @a : i64 = 5,"), "{text}");
3297 assert!(text.contains("global @b : i64 = 1,"), "{text}");
3298 assert!(text.contains("global @c : i32 = 1,"), "{text}");
3299 assert!(text.contains("global @d : i32 = 0,"), "{text}");
3300 assert!(text.contains("global @e : i32 = 1,"), "{text}");
3301 assert!(!text.contains("call"), "{text}");
3302
3303 let text = ir("unsigned long f(const char *p) { return __builtin_strlen(p); }\n");
3305 assert!(text.contains("call @strlen("), "{text}");
3306 }
3307
3308 #[test]
3315 fn a_sign_builtin_is_a_mask_over_the_bits_and_not_a_call() {
3316 let text = body("double f(double x) { return __builtin_fabs(x); }\n");
3317 assert!(text.contains("bitcast.i64 %0"), "{text}");
3318 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
3319 assert!(text.contains("and %1, %2"), "{text}");
3320 assert!(text.contains("bitcast.f64 %3"), "{text}");
3321 assert!(!text.contains("call"), "{text}");
3322
3323 let text = body("double f(double x, double y) { return __builtin_copysign(x, y); }\n");
3324 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
3325 assert!(text.contains("%8 = or %4, %7"), "{text}");
3326 assert!(!text.contains("call"), "{text}");
3327
3328 let text = body("long double f(long double x) { return __builtin_fabsl(x); }\n");
3331 assert!(text.contains("bitcast.i80 %0"), "{text}");
3332 assert!(text.contains("bitcast.f80"), "{text}");
3333
3334 let text = body("double f(float x) { return __builtin_fabs(x); }\n");
3337 assert!(text.contains("fpext.f64 %0"), "{text}");
3338 assert!(text.contains("bitcast.i64 %1"), "{text}");
3339 }
3340
3341 #[test]
3350 fn the_sign_builtins_answer_a_zero_and_a_nan_the_way_the_bits_say() {
3351 let text = ir(concat!(
3352 "double a = __builtin_fabs(-3.5);\n",
3353 "double b = __builtin_copysign(1.0, -0.0);\n",
3354 "double c = __builtin_copysign(0.0, -2.0);\n",
3355 "double d = __builtin_copysign(-__builtin_nan(\"\"), 1.0);\n",
3357 "double e = __builtin_fabs(-__builtin_nan(\"0x1\"));\n",
3358 "float g = __builtin_copysignf(-0.0f, 2.0f);\n",
3359 "long double h = __builtin_copysignl(1.0L, -1.0L);\n",
3360 "long double i = __builtin_fabsl(-__builtin_infl());\n",
3361 ));
3362 assert!(text.contains("global @a : f64 = 0x400c000000000000,"), "{text}");
3363 assert!(text.contains("global @b : f64 = 0xbff0000000000000,"), "{text}");
3364 assert!(text.contains("global @c : f64 = 0x8000000000000000,"), "{text}");
3365 assert!(text.contains("global @d : f64 = 0x7ff8000000000000,"), "{text}");
3366 assert!(text.contains("global @e : f64 = 0x7ff8000000000001,"), "{text}");
3367 assert!(text.contains("global @g : f32 = 0x0,"), "{text}");
3368 assert!(text.contains("f80 0xbfff8000000000000000"), "{text}");
3369 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
3370 }
3371
3372 #[test]
3379 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
3380 let text = ir(concat!(
3381 "constexpr int side = 4;\n",
3382 "constexpr int wider = side + 1;\n",
3383 "constexpr double half = 1.5;\n",
3384 "struct point { int x; int y; };\n",
3385 "constexpr struct point origin = { 5, 6 };\n",
3386 "int square[side * side];\n",
3387 "int rectangle[wider];\n",
3388 "int rounded[(int)half * 2];\n",
3389 "int across[origin.y];\n",
3390 "enum named { four = side };\n",
3391 "int e = four;\n",
3392 ));
3393 assert!(text.contains("global @square : bytes 64 ="), "{text}");
3394 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
3395 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
3396 assert!(text.contains("global @across : bytes 24 ="), "{text}");
3397 assert!(text.contains("global @e : i32 = 4,"), "{text}");
3398
3399 let mut opts = options();
3402 opts.emit = EmitKind::Ir;
3403 let konst = "const int n = 1;\nint a[n];\n";
3404 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
3405 assert_eq!(run(&opts, konst).messages, [message]);
3406
3407 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
3409 assert_eq!(run(&opts, subscript).messages, [message]);
3410
3411 let address = "constexpr int c = 3;\nint *p = &c;\n";
3413 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
3414 pointer target type [E0514]";
3415 assert_eq!(run(&opts, address).messages, [warning]);
3416 }
3417
3418 #[test]
3427 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
3428 let mut opts = options();
3431 opts.std = Std::C17;
3432 let source = concat!(
3433 "int add(a, b)\n",
3434 "int a;\n",
3435 "int b;\n",
3436 "{ return a + b; }\n",
3437 "int promoted(c)\n",
3438 "char c;\n",
3439 "{ return c; }\n",
3440 "int narrow(char);\n",
3441 "int narrow(c)\n",
3442 "char c;\n",
3443 "{ return c; }\n",
3444 "int first(a)\n",
3445 "int a[4];\n",
3446 "{ return a[0]; }\n",
3447 );
3448 let result = run(&opts, source);
3449 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3450 let text = result.text();
3451 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
3452 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
3453 assert!(text.contains("c : char object automatic defined"), "{text}");
3455 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
3456 assert!(text.contains("first : int(int *) function external defined"), "{text}");
3458 }
3459
3460 #[test]
3467 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
3468 let mut opts = options();
3469 opts.std = Std::C17;
3470 for (source, message) in [
3471 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
3472 (
3473 "int f(a)\nint a;\nint b;\n{ return a; }\n",
3474 "3:5: error: declaration for parameter 'b' but no such parameter",
3475 ),
3476 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
3477 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
3478 (
3479 "int f(a)\nstatic int a;\n{ return a; }\n",
3480 "2:12: error: storage class specified for parameter 'a'",
3481 ),
3482 (
3483 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
3484 "2:7: error: argument 'a' doesn't match prototype",
3485 ),
3486 ] {
3487 let result = run(&opts, source);
3488 assert!(result.failed(), "expected this to fail:\n{source}");
3489 assert!(result.messages[0].contains(message), "{:?}", result.messages);
3490 }
3491
3492 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
3495 let mut older = options();
3496 older.std = Std::C89;
3497 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
3498 let result = run(&opts, implicit);
3499 assert!(
3500 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
3501 "{:?}",
3502 result.messages
3503 );
3504
3505 let mut newer = options();
3509 newer.std = Std::C23;
3510 let plain = "int f(a)\nint a;\n{ return a; }\n";
3511 let result = run(&newer, plain);
3512 assert!(!result.failed(), "{:?}", result.messages);
3513 assert_eq!(
3514 result.messages,
3515 ["/main.c:1:5: warning: old-style function definition [E0412]"]
3516 );
3517 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
3518 }
3519
3520 #[test]
3527 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
3528 let text = ir(concat!(
3529 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
3530 "struct brim { char buf[9223372036854775807L]; };\n",
3531 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
3532 "unsigned long h = sizeof(struct huge_struct);\n",
3533 "unsigned long b = sizeof(struct brim);\n",
3534 "unsigned long y = sizeof(struct bitty);\n",
3535 ));
3536 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
3537 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
3538 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
3539
3540 let mut opts = options();
3541 opts.emit = EmitKind::Ir;
3542 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
3543 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
3544 assert_eq!(run(&opts, over).messages, [message]);
3545 let array = "struct wide { short buf[1L << 62]; };\n";
3546 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
3547 maximum object size '9223372036854775807' [E0537]";
3548 assert_eq!(run(&opts, array).messages[0], message);
3549 }
3550
3551 fn compile_bytes(source: &[u8]) -> Compiled {
3556 let mut opts = options();
3557 opts.emit = EmitKind::Ir;
3558 let mut fs = MemoryFileSystem::new();
3559 fs.insert("/main.c", source.to_vec());
3560 compile(&opts, "/main.c", &fs)
3561 }
3562
3563 #[test]
3570 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
3571 let mut source = b"char s[] = \"a".to_vec();
3572 source.push(0xff);
3573 source.extend_from_slice(b"b\";\nchar c = '");
3574 source.push(0xff);
3575 source.extend_from_slice(b"';\n");
3576 let result = compile_bytes(&source);
3577 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
3578 assert!(result.text().contains(r#"bytes "a\ffb\00""#), "{}", result.text());
3579 assert!(result.text().contains("global @c : i8 = -1,"), "{}", result.text());
3581
3582 let mut stray = b"int a".to_vec();
3583 stray.push(0xff);
3584 stray.extend_from_slice(b" = 1;\n");
3585 let result = compile_bytes(&stray);
3586 assert!(
3587 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
3588 "{:?}",
3589 result.messages
3590 );
3591 }
3592
3593 #[test]
3594 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
3595 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
3596 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
3597 let expected = "\
3598func @add(i32, i32) -> i32, linkage(external) {
3599block0(%0: i32, %1: i32):
3600 %2 = add.nsw %0, %1
3601 return %2
3602}
3603";
3604 assert!(text.contains(expected), "{text}");
3605 }
3606
3607 #[test]
3608 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
3609 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
3610 assert!(!text.contains("alloca"), "{text}");
3611 assert!(!text.contains("load"), "{text}");
3612 assert!(!text.contains("store"), "{text}");
3613 }
3614
3615 #[test]
3616 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
3617 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
3618 let expected = "\
3619block0:
3620 %0 = alloca, size 4, align 4
3621 %1 = iconst.i32 1
3622 store %1 -> %0, align 4
3623 %2 = call @g(%0) : (ptr) -> i32
3624 return %2
3625";
3626 assert_eq!(text, expected);
3627 }
3628
3629 #[test]
3630 fn a_loop_carries_what_it_changes_as_block_parameters() {
3631 let text = body(
3634 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
3635 return total;\n}\n",
3636 );
3637 assert!(!text.contains("alloca"), "{text}");
3638 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
3639 assert!(text.contains("jump block1("), "{text}");
3640 }
3641
3642 #[test]
3643 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
3644 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
3645 assert!(text.contains("icmp slt %0, %1"), "{text}");
3646 assert!(!text.contains("zext"), "{text}");
3647 }
3648
3649 #[test]
3650 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
3651 let text = body("int f(int a, int b) { return a && b; }\n");
3652 let expected = "\
3653block0(%0: i32, %1: i32):
3654 %2 = iconst.i32 0
3655 %3 = icmp ne %0, %2
3656 %4 = iconst.i1 0
3657 br_if %3, block1, block2(%4)
3658
3659block1:
3660 %5 = iconst.i32 0
3661 %6 = icmp ne %1, %5
3662 jump block2(%6)
3663
3664block2(%7: i1):
3665 %8 = zext.i32 %7
3666 return %8
3667";
3668 assert_eq!(text, expected);
3669 }
3670
3671 #[test]
3672 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
3673 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
3674 assert!(!text.contains("block3"), "{text}");
3677 assert!(!text.contains("iconst.i32 3"), "{text}");
3678 }
3679
3680 #[test]
3681 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
3682 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
3683 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
3684 assert!(body("int f(void) { }\n").contains("unreachable"));
3685 }
3686
3687 #[test]
3688 fn a_structure_is_copied_rather_than_held_in_a_value() {
3689 let text = body(
3690 "struct point { int x, y; };\n\
3691 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
3692 );
3693 assert!(text.contains("memcpy"), "{text}");
3694 }
3695
3696 #[test]
3697 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
3698 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
3699 assert!(text.contains("memset"), "{text}");
3700 }
3701
3702 #[test]
3703 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
3704 let text = body(
3705 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
3706 default: r = 4; } return r; }\n",
3707 );
3708 let expected = "\
3709block0(%0: i32):
3710 %1 = iconst.i32 0
3711 switch %0, block1, [1 => block2, 2 => block3(%1)]
3712
3713block1:
3714 %2 = iconst.i32 4
3715 jump block4(%2)
3716
3717block2:
3718 %3 = iconst.i32 1
3719 jump block3(%3)
3720
3721block3(%4: i32):
3722 %5 = iconst.i32 2
3723 %6 = add.nsw %4, %5
3724 jump block4(%6)
3725
3726block4(%7: i32):
3727 return %7
3728";
3729 assert_eq!(text, expected);
3730 }
3731
3732 #[test]
3733 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
3734 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
3737 assert!(text.contains("%2 = sub %0, %1"), "{text}");
3738 assert!(text.contains("icmp ule"), "{text}");
3739 assert!(!text.contains("switch"), "{text}");
3740 }
3741
3742 #[test]
3743 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
3744 let text = body(
3745 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
3746 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
3747 );
3748 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
3751 assert!(text.contains("block5:\n jump block7("), "{text}");
3752 assert!(text.contains("block6:\n jump block8("), "{text}");
3753 }
3754
3755 #[test]
3756 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
3757 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
3758 }
3759
3760 #[test]
3761 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
3762 let text = body(
3767 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
3768 return n; }\n",
3769 );
3770 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
3773 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
3774 assert!(text.contains("block4:\n jump block3("), "{text}");
3775 }
3776
3777 #[test]
3778 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
3779 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
3782 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
3783 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
3784 assert!(text.contains("br_if %6, block2, block3"), "{text}");
3785 }
3786
3787 #[test]
3788 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
3789 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
3790 assert!(!text.contains("alloca"), "{text}");
3794 assert!(text.contains("block2(%4: i32):\n return %4"), "{text}");
3795 assert_eq!(text.matches("jump block2(").count(), 2, "{text}");
3796 }
3797
3798 #[test]
3799 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
3800 let text =
3801 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
3802 assert!(!text.contains("alloca"), "{text}");
3803 assert!(text.contains("block1(%2: i32):"), "{text}");
3804 assert!(text.contains("jump block1(%5)"), "{text}");
3805 }
3806
3807 #[test]
3808 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
3809 assert_eq!(
3812 body("int f(int x) { return x; spare: return 0; }\n"),
3813 "block0(%0: i32):\n return %0\n"
3814 );
3815 }
3816
3817 #[test]
3818 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
3819 let text = body(
3820 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
3821 );
3822 assert_eq!(
3825 text,
3826 "\
3827block0(%0: ptr):
3828 %1 = load.i8 %0, align 1
3829 %2 = iconst.i8 3
3830 %3 = ashr %1, %2
3831 %4 = sext.i32 %3
3832 return %4
3833"
3834 );
3835 }
3836
3837 #[test]
3838 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
3839 let text =
3843 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
3844 assert_eq!(
3845 text,
3846 "\
3847block0(%0: ptr, %1: i32):
3848 %2 = iconst.i32 16777215
3849 %3 = and %1, %2
3850 %4 = trunc.i16 %3
3851 store %4 -> %0, align 2
3852 %5 = iconst.i32 16
3853 %6 = lshr %3, %5
3854 %7 = trunc.i8 %6
3855 %8 = iconst.i64 2
3856 %9 = ptr_add %0, %8
3857 store %7 -> %9, align 1
3858 return
3859"
3860 );
3861 }
3862
3863 #[test]
3864 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
3865 let text =
3866 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
3867 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
3870 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
3871 }
3872
3873 #[test]
3874 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
3875 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
3878 assert_eq!(text.matches("ashr").count(), 0, "{text}");
3879 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
3880 }
3881
3882 #[test]
3883 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
3884 let text = body(
3888 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
3889 );
3890 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
3891 }
3892
3893 #[test]
3894 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
3895 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
3898 assert!(
3899 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
3900 "{text}"
3901 );
3902 }
3903
3904 #[test]
3905 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
3906 let text = ir(concat!(
3911 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
3912 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
3913 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
3914 "char s[2] = \"hi\";\n",
3915 ));
3916 assert!(
3917 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
3918 "{text}"
3919 );
3920 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
3921 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
3922 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
3925 }
3926
3927 #[test]
3928 fn a_definition_takes_a_parameter_it_left_unnamed() {
3929 let text = ir("int f(int a, int) { return a; }\n");
3933 assert!(text.contains("func @f(i32, i32) -> i32"), "{text}");
3934 assert!(text.contains("block0(%0: i32, %1: i32):"), "{text}");
3935
3936 let text = ir("int g(int, int n) { return n; }\n");
3939 assert!(text.contains("block0(%0: i32, %1: i32):\n return %1\n"), "{text}");
3940 }
3941
3942 #[test]
3943 fn an_assignment_of_a_structure_is_the_object_it_wrote() {
3944 let text = body(concat!(
3949 "struct s { int f; int g; };\n",
3950 "void h(struct s *a, struct s *c, struct s *d, struct s *e)\n",
3951 "{ *d = *e = a[0] = *c; }\n",
3952 ));
3953 assert_eq!(text.matches("memcpy").count(), 3, "{text}");
3954 assert!(text.contains("memcpy %8, %1, size 8, align 4\n"), "{text}");
3955 assert!(text.contains("memcpy %3, %8, size 8, align 4\n"), "{text}");
3956 assert!(text.contains("memcpy %2, %3, size 8, align 4\n"), "{text}");
3957 }
3958
3959 #[test]
3960 fn a_string_literal_stops_at_the_end_of_the_array_it_is_filling() {
3961 let mut opts = options();
3966 opts.emit = EmitKind::Ir;
3967 let result = run(
3968 &opts,
3969 concat!(
3970 "const char a[2][3] = { \"1234\", \"xyz\" };\n",
3971 "static const char b[3][5] = { \"12345\", \"678\", \"9\" };\n",
3972 "union u { struct { char x[4]; char y[4]; }; struct { char z[8]; }; };\n",
3973 "const union u c = { { \"1234\", \"567\" } };\n",
3974 ),
3975 );
3976 let text = result.text();
3977 assert_eq!(
3978 result.messages,
3979 ["/main.c:1:24: warning: initializer-string for array of 'const char' is too long \
3980 (5 chars into 3 available) [E0637]"]
3981 );
3982 assert!(text.contains("global @a : bytes 6 = { bytes \"123\", bytes \"xyz\" }"), "{text}");
3983 assert!(
3984 text.contains(
3985 "global @b : bytes 15 = { bytes \"12345\", bytes \"678\\00\", zero 1, \
3986 bytes \"9\\00\", zero 3 }"
3987 ),
3988 "{text}"
3989 );
3990 assert!(
3993 text.contains("global @c : bytes 8 = { bytes \"1234\", bytes \"567\\00\" }"),
3994 "{text}"
3995 );
3996 }
3997
3998 #[test]
3999 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
4000 let text = body(concat!(
4004 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
4005 "void g(struct v *);\n",
4006 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
4007 ));
4008 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
4009 }
4010
4011 #[test]
4012 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
4013 let text = ir(concat!(
4018 "struct s { int x; };\n",
4019 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
4020 "int n = (int){ 7 };\n",
4021 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
4022 ));
4023 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
4024 assert!(text.contains("global @n : i32 = 7,"), "{text}");
4025 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
4028 }
4029
4030 #[test]
4031 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
4032 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
4036 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
4037 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
4038 }
4039
4040 #[test]
4041 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
4042 let text = ir("unsigned char foo[1][0];\n");
4046 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
4047 }
4048
4049 #[test]
4050 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
4051 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
4054 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
4055 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
4056 }
4057
4058 #[test]
4059 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
4060 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
4064 assert!(
4065 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
4066 "{text}"
4067 );
4068 }
4069
4070 #[test]
4071 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
4072 let text = body(
4077 "\
4078struct s { int a, b; };
4079struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
4080",
4081 );
4082 assert!(text.contains("block3(%7: ptr)"), "{text}");
4084 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
4085 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
4086 }
4087
4088 #[test]
4089 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
4090 let text = ir("\
4094struct pair { int a, b; };
4095struct pair make(int a, int b);
4096struct pair twice(struct pair p) { return make(p.a, p.b); }
4097");
4098 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
4099 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
4100 }
4101
4102 #[test]
4103 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
4104 let text = ir("\
4108struct big { double v[8]; };
4109struct big grow(struct big b);
4110struct big twice(struct big b) { return grow(grow(b)); }
4111");
4112 assert!(
4113 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
4114 "{text}"
4115 );
4116 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
4117 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
4120 }
4121
4122 #[test]
4123 fn a_structure_passed_to_a_variadic_function_says_so_at_the_call() {
4124 let text = ir("\
4129struct big { double v[8]; };
4130struct pair { int a, b; };
4131int p(const char *, ...);
4132int f(struct big b, struct pair q) { return p(\"\", 1, b, q); }
4133");
4134 assert!(
4135 text.contains("call @p(%4, %5, %2 byval(64, align 8), %6) : (ptr, ...) -> i32"),
4136 "{text}"
4137 );
4138 }
4139
4140 #[test]
4141 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
4142 let body = body(
4145 "\
4146struct pair { int a, b; };
4147struct pair make(int a, int b);
4148int second(void) { return make(1, 2).b; }
4149",
4150 );
4151 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
4152 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
4153 }
4154
4155 #[test]
4156 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
4157 let source = "\
4161struct hfa { float x, y, z; };
4162int take(struct hfa h);
4163int give(struct hfa h) { return take(h); }
4164";
4165 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
4166 let mut opts = options();
4167 opts.emit = EmitKind::Ir;
4168 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
4169 let result = run(&opts, source);
4170 assert_eq!(result.messages, Vec::<String>::new());
4171 assert!(result.text().contains("func @take(f32, f32, f32) -> i32"), "{}", result.text());
4172 }
4173
4174 #[test]
4175 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
4176 let source = "\
4179int use(int *);
4180void f(int n) {
4181 {
4182 int a[n];
4183 use(a);
4184 }
4185 use(0);
4186}
4187";
4188 let body = body(source);
4189 assert!(body.contains("mul.nsw"), "{body}");
4190 assert!(body.contains("stacksave"), "{body}");
4191 assert!(body.contains("alloca %"), "{body}");
4192 assert!(body.contains("stackrestore"), "{body}");
4193 }
4194
4195 #[test]
4196 fn a_goto_out_of_the_scope_of_one_gives_its_stack_back_on_the_way() {
4197 let source = "\
4202int use(int *);
4203int f(int n) {
4204 {
4205 int a[n];
4206 if (use(a)) goto out;
4207 use(0);
4208 }
4209out:
4210 return 0;
4211}
4212";
4213 let body = body(source);
4214 assert_eq!(body.matches("stackrestore").count(), 2, "{body}");
4216 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
4217 assert!(after.starts_with(" %4\n jump block"), "{body}");
4218 }
4219
4220 #[test]
4221 fn a_goto_to_a_label_the_array_is_still_alive_at_leaves_the_stack_alone() {
4222 let source = "\
4226int use(int *);
4227int f(int n) {
4228 int a[n];
4229again:
4230 if (use(a)) goto again;
4231 return 0;
4232}
4233";
4234 let body = body(source);
4235 assert!(body.contains("stacksave"), "{body}");
4236 assert!(!body.contains("stackrestore"), "{body}");
4237 }
4238
4239 #[test]
4240 fn a_goto_back_to_a_label_in_front_of_one_gives_it_back_every_time_round() {
4241 let source = "\
4246int use(int *);
4247int f(int n) {
4248again:
4249 {
4250 int a[n];
4251 if (use(a)) goto again;
4252 }
4253 return 0;
4254}
4255";
4256 let body = body(source);
4257 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
4258 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
4259 assert!(after.starts_with(" %4\n jump block1\n"), "{body}");
4260 }
4261
4262 #[test]
4263 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
4264 let source = "\
4270int f(void);
4271void t(void) {
4272 int count = 10;
4273 for (; count--;) {
4274 int b[f()];
4275 int i;
4276 for (i = 0; i < f(); i++) {
4277 b[i] = count;
4278 }
4279 }
4280}
4281";
4282 let body = body(source);
4283 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
4287 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
4288 let next = after.split("\n\n").next().expect("the block the restore is in");
4291 assert!(next.contains("jump block1("), "{body}");
4292 }
4293
4294 #[test]
4295 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
4296 let source = "\
4299unsigned long f(int n) {
4300 int a[n];
4301 n = 0;
4302 return sizeof a;
4303}
4304";
4305 let body = body(source);
4306 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
4308 }
4309
4310 #[test]
4311 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
4312 let source = "\
4315int use(int);
4316int f(int x) {
4317 return ({
4318 int t = use(x);
4319 t * t;
4320 });
4321}
4322";
4323 let expected = "\
4324block0(%0: i32):
4325 %1 = call @use(%0) : (i32) -> i32
4326 %2 = mul.nsw %1, %1
4327 return %2
4328";
4329 assert_eq!(body(source), expected);
4330 }
4331
4332 #[test]
4333 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
4334 let source = "int f(int x) { return ({ return x; 0; }); }\n";
4338 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
4339 }
4340
4341 #[test]
4342 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
4343 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
4347 let expected = "\
4348block0(%0: ptr):
4349 %1 = va_arg.f64 %0
4350 %2 = va_arg.f64 %0
4351 %3 = fadd %1, %2
4352 return %3
4353";
4354 assert_eq!(body(source), expected);
4355 }
4356
4357 #[test]
4358 fn one_that_reads_a_structure_answers_where_the_object_is() {
4359 let source = "\
4373struct s { int a; long b; };
4374long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.b; }
4375";
4376 let expected = "\
4377block0(%0: ptr):
4378 %1 = alloca, size 16, align 16
4379 %2 = va_object %0, size 16, align 8, in(int 8 at 0, int 8 at 8)
4380 memcpy %1, %2, size 16, align 8
4381 %3 = iconst.i64 8
4382 %4 = ptr_add %1, %3
4383 %5 = load.i64 %4, align 8
4384 return %5
4385";
4386 assert_eq!(body(source), expected);
4387 }
4388
4389 #[test]
4393 fn the_classification_says_which_registers_the_object_arrived_in() {
4394 let source = "\
4395struct s { double a; double b; };
4396double f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a; }
4397";
4398 assert!(
4399 body(source)
4400 .contains("va_object %0, size 16, align 8, in(float f64 at 0, float f64 at 8)"),
4401 "{}",
4402 body(source)
4403 );
4404
4405 let big = "\
4406struct s { long a[4]; };
4407long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a[0]; }
4408";
4409 assert!(body(big).contains("va_object %0, size 32, align 8\n"), "{}", body(big));
4410 }
4411
4412 #[test]
4413 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
4414 let source = "\
4418int f(int c) {
4419 void *p = c ? &&one : &&two;
4420 goto *p;
4421one:
4422 return 1;
4423two:
4424 return 2;
4425}
4426";
4427 let expected = "\
4428block0(%0: i32):
4429 %1 = iconst.i32 0
4430 %2 = icmp ne %0, %1
4431 br_if %2, block1, block2
4432
4433block1:
4434 %3 = block_addr block3
4435 jump block4(%3)
4436
4437block2:
4438 %4 = block_addr block5
4439 jump block4(%4)
4440
4441block3:
4442 %5 = iconst.i32 1
4443 return %5
4444
4445block4(%6: ptr):
4446 indirect_br %6, block3, block5
4447
4448block5:
4449 %7 = iconst.i32 2
4450 return %7
4451";
4452 assert_eq!(body(source), expected);
4453 }
4454
4455 #[test]
4456 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
4457 let source = "void **next(void);
4460void f(void) { goto *next(); }
4461";
4462 let expected = "\
4463block0:
4464 %0 = call @next() : () -> ptr
4465 unreachable
4466";
4467 assert_eq!(body(source), expected);
4468 }
4469
4470 #[test]
4471 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
4472 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
4475 let expected = "\
4476block0:
4477 inline_asm.volatile \"mfence\", \"\", \"memory\"()
4478 return
4479";
4480 assert_eq!(body(source), expected);
4481 }
4482
4483 #[test]
4484 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
4485 let source = "\
4488int f(int x, int y) {
4489 int r;
4490 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
4491 return r + y;
4492}
4493";
4494 let expected = "\
4495block0(%0: i32, %1: i32):
4496 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
4497 %4 = add.nsw %2, %3
4498 return %4
4499";
4500 assert_eq!(body(source), expected);
4501 }
4502
4503 #[test]
4504 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
4505 let source = "\
4510struct pair { int a, b; };
4511int f(int x) {
4512 int slot = x;
4513 struct pair p = { x, x };
4514 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
4515 return slot + p.a;
4516}
4517";
4518 let text = body(source);
4519 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
4520 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
4521 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
4522 }
4523
4524 #[test]
4525 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
4526 let source = "\
4531int f(int x) {
4532 int r = 7;
4533 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
4534 return r;
4535away:
4536 return r;
4537}
4538";
4539 let expected = "\
4540block0(%0: i32):
4541 %1 = iconst.i32 7
4542 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
4543
4544block1:
4545 return %2
4546
4547block2:
4548 return %1
4549";
4550 assert_eq!(body(source), expected);
4551 }
4552
4553 #[test]
4554 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
4555 let mut opts = options();
4559 opts.emit = EmitKind::Ir;
4560 for (source, expected) in [
4561 (
4562 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
4563 "output operand constraint lacks '='",
4564 ),
4565 (
4566 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
4567 "lvalue required in 'asm' statement",
4568 ),
4569 (
4570 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
4571 "read-only variable 'g' used as 'asm' output",
4572 ),
4573 (
4574 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
4575 "input operand constraint contains '='",
4576 ),
4577 (
4578 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
4579 "memory input 0 is not directly addressable",
4580 ),
4581 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
4582 (
4583 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
4584 "duplicate asm operand name 'a'",
4585 ),
4586 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
4587 ] {
4588 let result = run(&opts, source);
4589 assert!(result.failed(), "expected this to be reported:\n{source}");
4590 assert!(
4591 result.messages.iter().any(|m| m.contains(expected)),
4592 "{expected}\n{:?}",
4593 result.messages
4594 );
4595 }
4596 }
4597
4598 #[test]
4599 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
4600 let mut opts = options();
4601 opts.emit = EmitKind::Ir;
4602 for source in [
4603 "int f(int n) { void *p = &&out; if (n) goto *p; { int a[n]; out: return 1; } }\n",
4604 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
4605 ] {
4606 let result = run(&opts, source);
4607 assert!(result.failed(), "expected this to be reported:\n{source}");
4608 assert!(
4609 result.messages.iter().any(|m| m.contains("not supported yet")),
4610 "{:?}",
4611 result.messages
4612 );
4613 }
4614 }
4615
4616 fn round_trip(source: &str) -> (String, String) {
4618 let printed = ir(source);
4619 let mut opts = options();
4620 opts.emit = EmitKind::Ir;
4621 let mut fs = MemoryFileSystem::new();
4622 fs.insert("/main.ir", printed.clone().into_bytes());
4623 let result = compile_ir(&opts, "/main.ir", &fs);
4624 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
4625 (printed, result.text().to_owned())
4626 }
4627
4628 #[test]
4629 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
4630 let (printed, again) = round_trip(
4634 "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",
4635 );
4636 assert_eq!(printed, again);
4637 }
4638
4639 #[test]
4640 fn ir_that_is_not_ir_says_which_line_stopped_it() {
4641 let mut opts = options();
4642 opts.emit = EmitKind::Ir;
4643 let mut fs = MemoryFileSystem::new();
4644 let text = "\
4645; ModuleID = 'a.c'
4646; format 0
4647target triple = \"x86_64-unknown-linux-gnu\"
4648target datalayout = \"e-p:64:64-i64:64-S128\"
4649
4650func @f(), linkage(external) {
4651block0:
4652 frobnicate
4653}
4654";
4655 fs.insert("/main.ir", text.as_bytes().to_vec());
4656 let result = compile_ir(&opts, "/main.ir", &fs);
4657 assert!(result.failed());
4658 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
4659 }
4660
4661 #[test]
4662 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
4663 let mut opts = options();
4666 opts.emit = EmitKind::Ir;
4667 let mut fs = MemoryFileSystem::new();
4668 let text = "\
4669; ModuleID = 'a.c'
4670; format 0
4671target triple = \"x86_64-unknown-linux-gnu\"
4672target datalayout = \"e-p:64:64-i64:64-S128\"
4673
4674func @f(), linkage(external) {
4675block0:
4676 %0 = iconst.i32 1
4677 return %0
4678}
4679";
4680 fs.insert("/main.ir", text.as_bytes().to_vec());
4681 let result = compile_ir(&opts, "/main.ir", &fs);
4682 assert!(result.failed());
4683 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
4684 }
4685
4686 #[test]
4687 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
4688 let mut fs = MemoryFileSystem::new();
4690 fs.insert("/main.ir", Vec::new());
4691 let result = compile_ir(&options(), "/main.ir", &fs);
4692 assert!(result.failed());
4693 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
4694 }
4695
4696 #[test]
4697 fn the_printed_ir_reads_back_as_the_same_module() {
4698 let text = ir("\
4701struct point { int x, y; };
4702static const char greeting[] = \"hi\";
4703int table[4] = { 1, 2, 3 };
4704int puts(const char *);
4705double half(double x) { return x / 2.0; }
4706int f(int n) {
4707 int total = 0;
4708 for (int i = 0; i < n; i++) {
4709 if (i == 3) continue;
4710 total += table[i];
4711 }
4712 switch (n) {
4713 case 0: total = 1;
4714 case 1: total++; break;
4715 default: total = -total;
4716 }
4717 struct point p = { total, 1 };
4718 int *q = &p.y;
4719 puts(greeting);
4720 return p.x + *q;
4721}
4722int dispatch(int c) {
4723 void *p = c ? &&one : &&two;
4724 goto *p;
4725one:
4726 return 1;
4727two:
4728 return 2;
4729}
4730int assembly(int x, int *p) {
4731 int r;
4732 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
4733 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
4734 return r;
4735away:
4736 return 0;
4737}
4738");
4739 let mut names = Interner::new();
4740 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
4741 assert_eq!(rucc_ir::print(&module, &names), text);
4742 }
4743}