1use std::path::Path;
14
15use rucc_base::Interner;
16use rucc_codegen::coverage::Fired;
17use rucc_codegen::elsewhere::Elsewhere;
18use rucc_codegen::pipeline::{self, Machine};
19use rucc_diag::{Diagnostic, Severity, Span};
20use rucc_ir::{Pic as IrPic, Visibility as IrVisibility};
21use rucc_lex::{Convert, Keywords, PpToken, convert};
22use rucc_sema::{Checker, Context as CheckContext};
23use rucc_session::{EmitKind, FileSystem, Options, Pic, Session, Visibility};
24use rucc_target::TargetInfo;
25use rucc_tuple::ObjectFormat;
26
27use crate::preprocess::render;
28
29#[derive(Debug, Clone, PartialEq, Eq, Default)]
36pub enum Artifact {
37 #[default]
40 Nothing,
41 Text(String),
43 Object(Vec<u8>),
45}
46
47impl Artifact {
48 #[must_use]
50 pub fn bytes(&self) -> &[u8] {
51 match self {
52 Artifact::Nothing => &[],
53 Artifact::Text(text) => text.as_bytes(),
54 Artifact::Object(bytes) => bytes,
55 }
56 }
57}
58
59#[derive(Debug, Clone, PartialEq, Eq)]
61pub struct Compiled {
62 pub artifact: Artifact,
64 pub messages: Vec<String>,
66 pub errors: u32,
68 pub fired: Fired,
74 pub dumps: Vec<rucc_opt::Dump>,
80 pub remarks: String,
86 pub deps: Vec<rucc_pp::Dependency>,
91 pub temps: Temps,
98}
99
100#[derive(Debug, Clone, PartialEq, Eq, Default)]
107pub struct Temps {
108 pub preprocessed: Option<String>,
110 pub assembly: Option<String>,
112}
113
114impl Compiled {
115 #[must_use]
117 pub fn failed(&self) -> bool {
118 self.errors > 0
119 }
120
121 #[must_use]
126 pub fn text(&self) -> &str {
127 match &self.artifact {
128 Artifact::Text(text) => text,
129 _ => "",
130 }
131 }
132}
133
134#[must_use]
147pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
148 let mut sess = Session::new(opts.clone());
149 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
153 let mut diagnostics: Vec<Diagnostic> = Vec::new();
154 let mut fired = Fired::new();
156 let mut dumps = Vec::new();
158 let mut remarks = String::new();
159 let mut temps = Temps::default();
161
162 let bytes = match fs.read(Path::new(name)) {
163 Ok(bytes) => bytes,
164 Err(e) => return failure(format!("{name}: {e}")),
165 };
166 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
167 return failure(format!("{name}: the source map has no room left for this file"));
168 };
169
170 let mut pp = rucc_pp::Preprocessor::new();
174 let predef = rucc_pp::Predef::for_options(opts);
175 let expanded: Vec<PpToken> = {
176 let mut tokens = Vec::new();
177 {
182 let mut cx =
183 rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
184 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
185 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
186 return failure(format!(
187 "{name}: the source map has no room for the built in macros"
188 ));
189 }
190 if pp.preinclude(&opts.preincludes, &mut tokens, &mut cx).is_err() {
191 return failure(format!("{name}: the source map has no room for the command line"));
192 }
193 tokens.append(&mut pp.run(file, &mut cx));
194 }
195 if opts.save_temps.wanted() {
196 temps.preprocessed = Some(rucc_pp::print(
197 file,
198 &tokens,
199 pp.line_directives(),
200 &sess.sources,
201 &sess.interner,
202 rucc_pp::PrintOptions { line_markers: opts.line_markers },
203 ));
204 }
205 tokens.iter().map(|token| token.to_pp()).collect()
206 };
207 diagnostics.extend(pp.take_diagnostics());
208 let deps = pp.dependencies().to_vec();
211
212 let cx = Convert {
215 keywords: &keywords,
216 interner: &sess.interner,
217 target: &sess.target,
218 std: opts.std,
219 gnu: opts.gnu_extensions,
220 pedantic: opts.pedantic,
221 };
222 let (tokens, complaints) = convert(&expanded, &cx);
223 diagnostics.extend(complaints);
224
225 let parsed = rucc_parse::parse(
226 &tokens,
227 rucc_parse::Context {
228 interner: &sess.interner,
229 std: opts.std,
230 gnu: opts.gnu_extensions,
231 pedantic: opts.pedantic,
232 error_limit: opts.error_limit as usize,
233 },
234 );
235 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
236 diagnostics.extend(parsed.diagnostics);
237
238 let mut artifact = Artifact::Nothing;
239 let mut instrumented = Instrumented::default();
242 if !parse_failed {
243 let mut checker = Checker::new(
244 &parsed.ast,
245 CheckContext {
246 names: &sess.interner,
247 target: &sess.target,
248 std: opts.std,
249 gnu: opts.gnu_extensions,
250 pedantic: opts.pedantic,
251 permissive: opts.permissive,
252 gnu89_inline: opts.gnu89_inline,
253 error_limit: opts.error_limit as usize,
254 builtins: opts.builtins && opts.hosted,
257 no_builtin: &opts.no_builtin,
258 },
259 );
260 checker.check_unit();
261 let checked = checker.finish();
262 if !checked.failed() {
263 match opts.emit {
264 EmitKind::Tast => {
265 artifact = Artifact::Text(rucc_sema::print(
266 &checked.tast,
267 &checked.types,
268 &sess.interner,
269 ));
270 }
271 EmitKind::TypeGranules => {
275 artifact = Artifact::Text(rucc_types::granule_report(
276 &checked.types,
277 &sess.interner,
278 &sess.target,
279 ));
280 }
281 EmitKind::Ir
282 | EmitKind::MirFinal
283 | EmitKind::Asm
284 | EmitKind::Object
285 | EmitKind::Executable
286 | EmitKind::SafetySummary => {
287 let mut lowered = rucc_lower::lower(
288 name,
289 rucc_lower::Context {
290 tast: &checked.tast,
291 types: &checked.types,
292 target: &sess.target,
293 names: &mut sess.interner,
294 visibility: match opts.visibility {
295 Visibility::Default => IrVisibility::Default,
296 Visibility::Hidden => IrVisibility::Hidden,
297 Visibility::Protected => IrVisibility::Protected,
298 },
299 },
300 );
301 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
305 if !failed {
306 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
311 for error in errors {
312 diagnostics.push(internal(&format!("invalid IR, {error}")));
313 }
314 } else if let Err(complaints) =
315 instrument(&mut lowered.module, &mut sess.interner, opts)
316 .map(|done| instrumented = done)
317 {
318 diagnostics.extend(complaints);
319 } else if let Err(complaints) = optimize(
320 &mut lowered.module,
321 &sess.interner,
322 &sess.target,
323 opts,
324 name,
325 &mut dumps,
326 &mut remarks,
327 ) {
328 diagnostics.extend(complaints);
329 } else if opts.emit == EmitKind::SafetySummary {
330 artifact = Artifact::Text(
335 rucc_safety::summarize(
336 &lowered.module,
337 &sess.interner,
338 name,
339 opts.safety.as_str(),
340 instrumented.checks,
341 instrumented.interposed,
342 instrumented.crossings,
343 )
344 .render(),
345 );
346 } else if opts.emit == EmitKind::Ir {
347 artifact =
352 Artifact::Text(rucc_ir::print(&lowered.module, &sess.interner));
353 } else {
354 match generate(
357 &mut lowered.module,
358 &mut sess.interner,
359 &sess.target,
360 opts,
361 &mut fired,
362 &mut temps.assembly,
363 ) {
364 Ok(made) => artifact = made,
365 Err(complaints) => diagnostics.extend(complaints),
366 }
367 }
368 }
369 diagnostics.extend(lowered.diagnostics);
370 }
371 _ => {}
372 }
373 }
374 diagnostics.extend(checked.diagnostics);
375 }
376
377 let mut messages = Vec::with_capacity(diagnostics.len());
378 let mut errors = 0;
379 for diag in &diagnostics {
380 if !opts.warnings && diag.severity == Severity::Warning {
384 continue;
385 }
386 if diag.severity.is_fatal()
387 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
388 {
389 errors += 1;
390 }
391 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
392 }
393 if errors > 0 {
394 artifact = Artifact::Nothing;
396 }
397 Compiled { artifact, messages, errors, fired, dumps, remarks, deps, temps }
400}
401
402#[must_use]
412pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
413 let mut sess = Session::new(opts.clone());
414 if opts.emit != EmitKind::Ir {
415 return failure(format!(
416 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
417 the C in front of it became",
418 opts.emit.as_str()
419 ));
420 }
421 let bytes = match fs.read(Path::new(name)) {
422 Ok(bytes) => bytes,
423 Err(e) => return failure(format!("{name}: {e}")),
424 };
425 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
426 return failure(format!("{name}: this is not text, so it is not IR"));
427 };
428
429 let module = match rucc_ir::parse(text, &mut sess.interner) {
430 Ok(module) => module,
431 Err(error) => {
432 return failure(format!("{name}:{}: {}", error.line, error.message));
433 }
434 };
435 let mut diagnostics: Vec<Diagnostic> = Vec::new();
436 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
437 for error in errors {
438 diagnostics.push(invalid(&format!("invalid IR, {error}")));
439 }
440 }
441 let mut messages = Vec::with_capacity(diagnostics.len());
442 for diag in &diagnostics {
443 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
444 }
445 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
446 let artifact = if errors > 0 {
447 Artifact::Nothing
448 } else {
449 Artifact::Text(rucc_ir::print(&module, &sess.interner))
450 };
451 Compiled {
453 artifact,
454 messages,
455 errors,
456 fired: Fired::new(),
457 dumps: Vec::new(),
458 remarks: String::new(),
459 deps: Vec::new(),
460 temps: Temps::default(),
461 }
462}
463
464fn instrument(
487 module: &mut rucc_ir::Module,
488 names: &mut Interner,
489 opts: &Options,
490) -> Result<Instrumented, Vec<Diagnostic>> {
491 if !opts.safety.instruments() {
492 return Ok(Instrumented::default());
493 }
494 let checks = rucc_safety::run(module);
495 let interposed = rucc_safety::redirect(module, names);
500 let crossings = rucc_safety::witness(module, names);
503 match rucc_ir::verify(module, names) {
504 Ok(()) => Ok(Instrumented { checks, interposed, crossings }),
505 Err(errors) => Err(errors
506 .iter()
507 .map(|e| internal(&format!("invalid IR after check insertion, {e}")))
508 .collect()),
509 }
510}
511
512#[derive(Clone, Copy, Debug, Default)]
518struct Instrumented {
519 checks: rucc_safety::Counts,
521 interposed: usize,
523 crossings: rucc_safety::Sites,
525}
526
527fn optimize(
539 module: &mut rucc_ir::Module,
540 names: &Interner,
541 target: &TargetInfo,
542 opts: &Options,
543 file: &str,
544 dumps: &mut Vec<rucc_opt::Dump>,
545 remarks: &mut String,
546) -> Result<(), Vec<Diagnostic>> {
547 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
548 settings.interposition = match opts.interposition {
554 true => replaceable(target, opts),
555 false => IrPic::Executable,
556 };
557 settings.toggles.clone_from(&opts.passes);
558 settings.fuel = opts.pass_fuel.iter().cloned().collect();
559 settings.global_fuel = opts.pass_fuel_global;
560 settings.verify |= opts.verify_each;
561 for (on, spec) in &opts.pass_gates {
562 if let Err(why) = settings.gates.add(*on, spec) {
565 return Err(vec![internal(&why)]);
566 }
567 }
568 for spec in &opts.dump_ir {
569 if let Err(why) = settings.dumps.add(spec) {
572 return Err(vec![internal(&why)]);
573 }
574 }
575 let mut wants = rucc_opt::Wants::none();
576 for spec in &opts.opt_info {
577 if let Err(why) = wants.add(spec) {
580 return Err(vec![internal(&why)]);
581 }
582 }
583 let report = rucc_opt::run(module, names, &settings);
584 remarks.push_str(&rucc_opt::optinfo::render(file, &report, names, wants));
585 dumps.extend(report.dumps);
586 match report.broke.is_empty() {
587 true => Ok(()),
588 false => Err(report.broke.iter().map(|why| internal(why)).collect()),
589 }
590}
591
592fn replaceable(target: &TargetInfo, opts: &Options) -> IrPic {
626 match (target.tuple.os().object_format(), opts.pic) {
627 (Some(ObjectFormat::Elf), Pic::Library) => IrPic::Library,
628 _ => IrPic::Executable,
629 }
630}
631
632fn generate(
633 module: &mut rucc_ir::Module,
634 names: &mut Interner,
635 target: &TargetInfo,
636 opts: &Options,
637 fired: &mut Fired,
638 assembly: &mut Option<String>,
639) -> Result<Artifact, Vec<Diagnostic>> {
640 let Some(machine) = Machine::for_target(target) else {
641 return Err(vec![unsupported(&format!(
642 "there is no back end for {} in this compiler yet, so there is nothing to generate",
643 target.tuple
644 ))]);
645 };
646 let flags = pipeline::Flags { frame_pointer: opts.frame_pointer, red_zone: opts.red_zone };
647
648 if opts.safety.instruments() {
657 rucc_safety::lower(module, names);
658 if let Err(errors) = rucc_ir::verify(module, names) {
659 return Err(errors
660 .iter()
661 .map(|e| internal(&format!("invalid IR after check lowering, {e}")))
662 .collect());
663 }
664 }
665
666 let elsewhere = Elsewhere::of(module, replaceable(target, opts));
674
675 let mut funcs = Vec::new();
676 let mut complaints = Vec::new();
677 for id in module.funcs() {
678 if module[id].is_declaration() {
679 continue;
680 }
681 match pipeline::compile_recording(
682 &mut module[id],
683 names,
684 &machine,
685 &elsewhere,
686 flags,
687 fired,
688 ) {
689 Ok(func) => funcs.push(func),
690 Err(why) => {
691 let name = names.resolve(module[id].name).to_owned();
692 let span = why.inst().map_or(Span::DUMMY, |inst| module[id].span(inst));
695 let said = format!("cannot generate code for '{name}': {why}");
696 complaints.push(unsupported_at(&said, span));
697 }
698 }
699 }
700 if !complaints.is_empty() {
701 return Err(complaints);
702 }
703 let (globals, aliases) = match opts.emit {
709 EmitKind::Asm | EmitKind::Object | EmitKind::Executable => (
710 rucc_asm::globals(module, names).map_err(refused)?,
711 rucc_asm::aliases(module, names).map_err(refused)?,
712 ),
713 _ => (rucc_asm::Globals::default(), Vec::new()),
714 };
715 let unwind = opts.unwinds();
719 match opts.emit {
720 EmitKind::Asm => rucc_asm::print(&funcs, &globals, &aliases, names, target, unwind)
721 .map(Artifact::Text)
722 .map_err(refused),
723 EmitKind::Object | EmitKind::Executable => {
726 if opts.save_temps.wanted() {
727 *assembly = Some(
728 rucc_asm::print(&funcs, &globals, &aliases, names, target, unwind)
729 .map_err(refused)?,
730 );
731 }
732 let text = rucc_asm::assemble(&funcs, names, target, unwind).map_err(refused)?;
733 let data = globals.image();
734 rucc_object::write(&text, &data, &aliases, target).map(Artifact::Object).map_err(
737 |why| match why {
738 rucc_object::Error::Format { .. } => vec![unsupported(&why.to_string())],
739 rucc_object::Error::Refused { .. } => vec![internal(&why.to_string())],
740 },
741 )
742 }
743 _ => Ok(Artifact::Text(rucc_mir::print(&funcs, names, target.regs))),
744 }
745}
746
747fn refused(why: rucc_asm::Error) -> Vec<Diagnostic> {
753 match why {
754 rucc_asm::Error::Thread { .. } | rucc_asm::Error::IFunc { .. } => {
755 vec![unsupported(&why.to_string())]
756 }
757 _ => vec![internal(&why.to_string())],
758 }
759}
760
761fn unsupported(message: &str) -> Diagnostic {
767 unsupported_at(message, Span::DUMMY)
768}
769
770fn unsupported_at(message: &str, span: Span) -> Diagnostic {
776 Diagnostic::error(message.to_owned(), span)
777 .with_code("E0653")
778 .note("this construct is not lowered yet, see https://github.com/tamnd/rucc/issues", span)
779}
780
781fn invalid(message: &str) -> Diagnostic {
783 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
784}
785
786fn internal(message: &str) -> Diagnostic {
788 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
789 .with_code("E0652")
790 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
791}
792
793fn failure(message: String) -> Compiled {
796 Compiled {
797 artifact: Artifact::Nothing,
798 messages: vec![format!("rucc: error: {message}")],
799 errors: 1,
800 fired: Fired::new(),
801 dumps: Vec::new(),
802 remarks: String::new(),
803 deps: Vec::new(),
804 temps: Temps::default(),
805 }
806}
807
808#[cfg(test)]
809mod tests {
810 use rucc_session::{MemoryFileSystem, Std};
811 use rucc_target::Triple;
812
813 use super::*;
814
815 fn options() -> Options {
816 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
817 opts.emit = EmitKind::Tast;
818 opts
819 }
820
821 fn run(opts: &Options, source: &str) -> Compiled {
822 let mut fs = MemoryFileSystem::new();
823 fs.insert("/main.c", source.to_owned().into_bytes());
824 compile(opts, "/main.c", &fs)
825 }
826
827 fn freestanding() -> Options {
831 let mut opts = options();
832 opts.hosted = false;
833 opts.search.push_system(rucc_session::runtime::DIR);
834 opts
835 }
836
837 fn shipped(source: &str) -> String {
839 let result = run(&freestanding(), source);
840 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
841 result.text().to_owned()
842 }
843
844 fn tast(source: &str) -> String {
846 let result = run(&options(), source);
847 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
848 result.text().to_owned()
849 }
850
851 #[test]
852 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
853 let text = shipped(concat!(
854 "#include <stdarg.h>\n",
855 "int sum(int n, ...) {\n",
856 " va_list ap, copy;\n",
857 " va_start(ap, n);\n",
858 " va_copy(copy, ap);\n",
859 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
860 " va_end(ap);\n",
861 " va_end(copy);\n",
862 " return total;\n",
863 "}\n",
864 ));
865 assert!(text.contains("va-start"), "{text}");
866 assert!(text.contains("va-copy"), "{text}");
867 assert!(text.contains("va-arg"), "{text}");
868 assert!(text.contains("va-end"), "{text}");
869 }
870
871 #[test]
875 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
876 let text = shipped(concat!(
877 "#define __need___va_list\n",
878 "#include <stdarg.h>\n",
879 "int vprint(const char *f, __gnuc_va_list ap);\n",
880 "#ifdef va_start\n",
881 "#error va_start should not be defined\n",
882 "#endif\n",
883 "#ifdef _VA_LIST_DEFINED\n",
884 "#error va_list should not have been made\n",
885 "#endif\n",
886 ));
887 assert!(text.contains("vprint"), "{text}");
888 }
889
890 #[test]
893 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
894 let text = shipped(concat!(
895 "#define __need_size_t\n",
896 "#include <stddef.h>\n",
897 "#ifdef offsetof\n",
898 "#error offsetof should not be defined yet\n",
899 "#endif\n",
900 "#define __need_ptrdiff_t\n",
901 "#include <stddef.h>\n",
902 "#include <stddef.h>\n",
903 "size_t a;\n",
904 "ptrdiff_t b;\n",
905 "wchar_t c;\n",
906 "max_align_t d;\n",
907 "void *e = NULL;\n",
908 "struct P { int x; long y; };\n",
909 "size_t f = offsetof(struct P, y);\n",
910 ));
911 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
912 assert!(text.contains("decl #1 b : long"), "{text}");
913 }
914
915 #[test]
916 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
917 let text = shipped(concat!(
918 "#include <limits.h>\n",
919 "#include <float.h>\n",
920 "int bits = CHAR_BIT;\n",
921 "long big = LONG_MAX;\n",
922 "int low = INT_MIN;\n",
923 "int radix = FLT_RADIX;\n",
924 "int digits = DBL_MANT_DIG;\n",
925 ));
926 assert!(text.contains("const 8 : int"), "{text}");
927 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
928 assert!(text.contains("const 2 : int"), "{text}");
929 assert!(text.contains("const 53 : int"), "{text}");
930 }
931
932 #[test]
936 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
937 let text = shipped(concat!(
938 "#include <stdint.h>\n",
939 "int64_t a = INT64_C(1);\n",
940 "uint_least16_t b;\n",
941 "intptr_t c;\n",
942 "uintmax_t d = UINTMAX_MAX;\n",
943 "int wide = sizeof(int_fast64_t);\n",
944 ));
945 assert!(text.contains("decl #0 a : long"), "{text}");
946 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
947 assert!(text.contains("decl #2 c : long"), "{text}");
948 }
949
950 #[test]
951 fn the_three_formality_headers_still_have_to_work() {
952 let text = shipped(concat!(
953 "#include <stdbool.h>\n",
954 "#include <stdalign.h>\n",
955 "#include <iso646.h>\n",
956 "#include <stdnoreturn.h>\n",
957 "int t = true and not false;\n",
958 "_Alignas(16) char buf[16];\n",
959 "int a = alignof(long);\n",
960 ));
961 assert!(text.contains("decl #0 t : int"), "{text}");
962 assert!(text.contains("const 8 : unsigned long"), "{text}");
963 }
964
965 #[test]
968 fn every_shipped_header_can_be_included_twice() {
969 let mut source = String::new();
970 for _ in 0..2 {
971 for name in rucc_session::runtime::names() {
972 source.push_str(&format!("#include <{name}>\n"));
973 }
974 }
975 source.push_str("int x;\n");
976 let text = shipped(&source);
977 assert!(text.starts_with("decl #0 x : int"), "{text}");
978 }
979
980 #[test]
981 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
982 let fs = MemoryFileSystem::new();
983 let result = compile(&options(), "/nope.c", &fs);
984 assert!(result.failed());
985 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
986 assert!(result.text().is_empty());
987 }
988
989 #[test]
990 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
991 let text = tast("int x = 1;\n");
992 let expected = "\
993decl #0 x : int object external static defined
994 init
995 +0
996 const 1 : int
997";
998 assert_eq!(text, expected);
999 }
1000
1001 #[test]
1002 fn the_macros_are_expanded_before_anything_is_parsed() {
1003 let text = tast("#define N 2\nint a[N];\n");
1007 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
1008 }
1009
1010 #[test]
1016 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
1017 let text = tast(concat!(
1018 "#pragma pack(4)\n",
1019 "struct s { int a; };\n",
1020 "#pragma pack()\n",
1021 "int b;\n",
1022 "_Pragma(\"GCC visibility push(default)\") int c;\n",
1023 ));
1024 assert!(text.contains("decl #0 b : int"), "{text}");
1025 assert!(text.contains("decl #1 c : int"), "{text}");
1026 }
1027
1028 #[test]
1036 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
1037 tast(concat!(
1038 "struct A { char c; int i; } __attribute__((packed));\n",
1039 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1040 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1041 "struct B { char c; int i; } __attribute__((aligned));\n",
1044 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
1045 "struct C { char c; int i __attribute__((packed)); };\n",
1046 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
1047 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
1048 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
1049 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
1050 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
1051 "struct E { char c; _Alignas(8) int i; };\n",
1052 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
1053 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
1054 "struct F { char c; int i __attribute__((aligned(8))); };\n",
1055 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
1056 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
1059 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
1060 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
1061 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
1062 "struct I { [[gnu::packed]] char c; int i; };\n",
1065 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1066 "struct J { char c; [[gnu::packed]] int i; };\n",
1067 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
1068 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
1069 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
1070 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
1071 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
1072 "union L { char c; int i; } __attribute__((packed));\n",
1073 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
1074 "struct O { char c; int i; } __attribute__((__packed__));\n",
1078 "_Static_assert(sizeof(struct O) == 5 && _Alignof(struct O) == 1, \"O\");\n",
1079 "struct P { char c; int i; } __attribute__((__aligned__(8)));\n",
1080 "_Static_assert(sizeof(struct P) == 8 && _Alignof(struct P) == 8, \"P\");\n",
1081 ));
1082 }
1083
1084 #[test]
1093 fn the_aligned_attribute_on_a_declaration_raises_what_that_one_object_is_aligned_to() {
1094 tast(concat!(
1095 "int v __attribute__((aligned(64)));\n",
1096 "_Static_assert(__alignof__(v) == 64, \"v\");\n",
1097 "__attribute__((aligned(32))) int w;\n",
1100 "_Static_assert(__alignof__(w) == 32, \"w\");\n",
1101 "[[gnu::aligned(16)]] int x;\n",
1102 "_Static_assert(__alignof__(x) == 16, \"x\");\n",
1103 "int y __attribute__((aligned(2)));\n",
1106 "_Static_assert(__alignof__(y) == 4, \"y\");\n",
1107 "void f(void) { int a __attribute__((aligned(128)));\n",
1109 "_Static_assert(__alignof__(a) == 128, \"a\"); (void)a; }\n",
1110 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1113 "void g(void) __attribute__((aligned(256)));\n",
1116 "void g(void) {}\n",
1117 "_Static_assert(__alignof__(g) == 256, \"g\");\n",
1118 ));
1119 }
1120
1121 #[test]
1125 fn what_a_declaration_asked_to_be_aligned_to_is_what_the_assembler_is_told() {
1126 let text = asm(concat!(
1127 "int v __attribute__((aligned(64)));\n",
1128 "void g(void) __attribute__((aligned(256)));\n",
1129 "void g(void) {}\n",
1130 "void plain(void) {}\n",
1131 ));
1132 assert!(text.contains("\t.p2align\t6\n\t.type\tv, @object\n"), "{text}");
1133 assert!(text.contains("\t.p2align\t8, 0x90\n\t.globl\tg\n"), "{text}");
1134 assert!(text.contains("\t.p2align\t4, 0x90\n\t.globl\tplain\n"), "{text}");
1135 }
1136
1137 #[test]
1146 fn an_aligned_typedef_says_what_an_object_of_it_is_aligned_to_and_may_lower_it() {
1147 tast(concat!(
1148 "typedef int L __attribute__((aligned(2)));\n",
1149 "_Static_assert(__alignof__(L) == 2, \"L\");\n",
1150 "_Static_assert(_Alignof(L) == 2, \"L alignof\");\n",
1151 "_Static_assert(sizeof(L) == 4, \"L size\");\n",
1153 "struct T { char c; L x; };\n",
1154 "_Static_assert(sizeof(struct T) == 6, \"T\");\n",
1155 "_Static_assert(__builtin_offsetof(struct T, x) == 2, \"T.x\");\n",
1156 "typedef int H __attribute__((aligned(16)));\n",
1158 "_Static_assert(__alignof__(H) == 16, \"H\");\n",
1159 "_Static_assert(sizeof(H) == 4, \"H size\");\n",
1160 "struct U { char c; H x; };\n",
1161 "_Static_assert(sizeof(struct U) == 32, \"U\");\n",
1162 "_Static_assert(__builtin_offsetof(struct U, x) == 16, \"U.x\");\n",
1163 "typedef L M __attribute__((aligned(8)));\n",
1166 "_Static_assert(__alignof__(M) == 8, \"M\");\n",
1167 "typedef L N;\n",
1170 "_Static_assert(__alignof__(N) == 2, \"N\");\n",
1171 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1173 ));
1174 let text = asm(concat!(
1175 "typedef int L __attribute__((aligned(2)));\n",
1176 "typedef int H __attribute__((aligned(16)));\n",
1177 "L low;\n",
1178 "H high;\n",
1179 ));
1180 assert!(text.contains("\t.p2align\t1\n\t.type\tlow, @object\n"), "{text}");
1181 assert!(text.contains("\t.p2align\t4\n\t.type\thigh, @object\n"), "{text}");
1182 }
1183
1184 #[test]
1192 fn the_vector_size_attribute_builds_a_type_of_lanes_and_measures_it_in_bytes() {
1193 tast(concat!(
1194 "typedef int __attribute__((vector_size(16))) v4si;\n",
1195 "_Static_assert(sizeof(v4si) == 16 && _Alignof(v4si) == 16, \"v4si\");\n",
1196 "typedef char __attribute__((vector_size(16))) v16qi;\n",
1197 "_Static_assert(sizeof(v16qi) == 16, \"v16qi\");\n",
1198 "typedef int __attribute__((vector_size(4))) v1si;\n",
1201 "_Static_assert(sizeof(v1si) == 4, \"v1si\");\n",
1202 "typedef float __attribute__((__vector_size__(8))) v2sf;\n",
1204 "_Static_assert(sizeof(v2sf) == 8, \"v2sf\");\n",
1205 "typedef short [[gnu::vector_size(8)]] v4hi;\n",
1206 "_Static_assert(sizeof(v4hi) == 8, \"v4hi\");\n",
1207 "v4si g;\n",
1210 "_Static_assert(sizeof(g[0]) == 4, \"lane\");\n",
1211 "_Static_assert(sizeof(g + g) == 16, \"whole\");\n",
1212 "_Static_assert(sizeof(g + 1) == 16, \"broadcast\");\n",
1215 "_Static_assert(sizeof(v4si[3]) == 48, \"array\");\n",
1217 ));
1218 }
1219
1220 #[test]
1230 fn a_vector_is_written_whole_into_an_array_of_them_and_named_by_a_type_name() {
1231 tast(concat!(
1232 "typedef int __attribute__((vector_size(8))) v2si;\n",
1233 "v2si table[] = { (v2si){ 1, 2 }, (v2si){ 3, 4 } };\n",
1234 "_Static_assert(sizeof(table) == 16, \"two of them and not eight lanes\");\n",
1235 "v2si written = (int __attribute__((vector_size(8)))){ 5, 6 };\n",
1237 "_Static_assert(sizeof((int __attribute__((vector_size(16)))){ 0 }) == 16, \"named\");\n",
1238 "v2si lanes[2] = { 1, 2, 3, 4 };\n",
1241 "_Static_assert(sizeof(lanes) == 16, \"still elided\");\n",
1242 ));
1243 }
1244
1245 #[test]
1253 fn a_lane_is_assignable_and_a_shift_takes_a_count_of_its_own_lane() {
1254 let result = run(
1255 &options(),
1256 concat!(
1257 "typedef int __attribute__((vector_size(16))) v4si;\n",
1258 "typedef unsigned __attribute__((vector_size(16))) v4ui;\n",
1259 "void write(v4si *out, v4ui a, v4si b, int n) {\n",
1260 " v4si v = { 1, 2, 3, 4 };\n",
1261 " v[0] = n;\n",
1262 " v[1] += n;\n",
1263 " v[2]++;\n",
1264 " *&v[3] = n;\n",
1265 " v4ui shifted = a >> b;\n",
1267 " shifted <<= b;\n",
1268 " *out = v + (v4si)shifted + (1 << b);\n",
1271 "}\n",
1272 "void refused(const v4si c) {\n",
1275 " c[0] = 1;\n",
1276 "}\n",
1277 ),
1278 );
1279 assert_eq!(result.messages.len(), 1, "{:?}", result.messages);
1280 assert!(result.messages[0].contains("assignment of read-only"), "{:?}", result.messages);
1281 }
1282
1283 #[test]
1290 fn a_record_that_asks_for_the_other_byte_order_is_refused_rather_than_laid_out_in_this_one() {
1291 let opts = options();
1292 let big = "struct s { int i; } __attribute__((scalar_storage_order(\"big-endian\")));\n";
1293 assert_eq!(
1294 run(&opts, big).messages,
1295 ["/main.c:1:36: error: 'scalar_storage_order' is not implemented yet [E0688]\n\
1296 /main.c:1:36: note: every scalar in this record would be read in the wrong byte \
1297 order"]
1298 );
1299
1300 let armoured =
1301 "struct s { int i; } __attribute__((__scalar_storage_order__(\"little-endian\")));\n";
1302 let messages = run(&opts, armoured).messages;
1303 assert!(messages[0].contains("[E0688]"), "{messages:?}");
1304
1305 let front = "struct __attribute__((scalar_storage_order(\"big-endian\"))) s { int i; };\n";
1308 assert!(run(&opts, front).messages[0].contains("[E0688]"), "{front}");
1309 let standard = "struct s { int i; } [[gnu::scalar_storage_order(\"big-endian\")]];\n";
1310 assert!(run(&opts, standard).messages[0].contains("[E0688]"), "{standard}");
1311 }
1312
1313 #[test]
1323 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
1324 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
1326 assert_eq!(
1327 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
1328 1
1329 );
1330 assert_eq!(
1331 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
1332 1
1333 );
1334 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
1335 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
1337 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
1338 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
1340 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
1341 }
1342
1343 fn bit_field_byte(record: &str) -> u64 {
1345 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
1346 let body = body(&source);
1347 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
1348 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
1349 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
1350 }
1351
1352 #[test]
1358 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
1359 tast(concat!(
1360 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
1361 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
1362 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
1363 "struct b { char c; __attribute__((packed)) int i; };\n",
1364 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
1365 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
1366 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
1367 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
1368 ));
1369 }
1370
1371 #[test]
1377 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
1378 tast(concat!(
1379 "#pragma pack(1)\n",
1380 "struct A { char c; int i; };\n",
1381 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1382 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1383 "#pragma pack()\n",
1384 "struct B { char c; int i; };\n",
1385 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
1386 "#pragma pack(2)\n",
1387 "struct C { char c; int i; double d; };\n",
1388 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
1389 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
1390 "struct K { char c; int i __attribute__((aligned(8))); };\n",
1392 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
1393 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
1394 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
1396 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
1397 "#pragma pack()\n",
1398 "#pragma pack(push, 1)\n",
1399 "struct D { char c; short s; };\n",
1400 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
1401 "#pragma pack(pop)\n",
1402 "struct E { char c; short s; };\n",
1403 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
1404 "struct H { char c;\n",
1406 "#pragma pack(1)\n",
1407 " int i; };\n",
1408 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
1409 "#pragma pack(1)\n",
1410 "struct I { char c;\n",
1411 "#pragma pack()\n",
1412 " int i; };\n",
1413 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1414 "#pragma pack()\n",
1415 "#pragma pack(push, 8)\n",
1417 "#pragma pack(push, 1)\n",
1418 "struct P { char c; int i; };\n",
1419 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
1420 "#pragma pack(pop)\n",
1421 "struct Q { char c; int i; };\n",
1422 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
1423 "#pragma pack(pop)\n",
1424 "#pragma pack(16)\n",
1426 "struct R { char c; int i; };\n",
1427 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
1428 "#pragma pack()\n",
1429 "#pragma pack(1)\n",
1430 "struct S { char c; int i : 5; int j : 20; };\n",
1431 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
1432 "union T { char c; int i; };\n",
1433 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
1434 "#pragma pack()\n",
1435 ));
1436 }
1437
1438 #[test]
1442 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
1443 let result = run(
1444 &options(),
1445 concat!(
1446 "#pragma pack 4\n",
1447 "#pragma pack(pop)\n",
1448 "#pragma pack(3)\n",
1449 "#pragma pack(1) junk\n",
1450 "#pragma pack(push, 1\n",
1451 "#pragma pack(x)\n",
1452 "#pragma pack(0)\n",
1455 "#pragma pack(push)\n",
1456 "struct s { char c; int i; };\n",
1457 "#pragma pack(pop)\n",
1458 "#pragma pack(pop, foo)\n",
1459 ),
1460 );
1461 let expected = [
1462 "missing `(` after `#pragma pack` - ignored",
1463 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
1464 "alignment must be a small power of two, not 3",
1465 "junk at end of `#pragma pack`",
1466 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
1467 "unknown action `x` for `#pragma pack` - ignored",
1468 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
1469 ];
1470 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
1471 for (message, want) in result.messages.iter().zip(expected) {
1472 assert!(message.contains(want), "expected {want:?} in {message:?}");
1473 }
1474 }
1475
1476 #[test]
1480 fn the_wide_integer_answers_to_all_three_of_its_names() {
1481 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
1482 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
1483 assert!(text.contains("decl #1 b : __int128"), "{text}");
1484 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
1485 }
1486
1487 #[test]
1488 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
1489 let text = tast("long f(int a, long b) { return a + b; }\n");
1493 assert!(text.contains("convert arithmetic"), "{text}");
1494 }
1495
1496 #[test]
1497 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
1498 for source in [
1499 "#error stop\n",
1500 "int f(void) { return 1 + ; }\n",
1501 "int f(void) { return undeclared; }\n",
1502 ] {
1503 let result = run(&options(), source);
1504 assert!(result.failed(), "expected this to fail:\n{source}");
1505 assert!(
1506 result.text().is_empty(),
1507 "a file that did not compile wrote a tree:\n{source}"
1508 );
1509 }
1510 }
1511
1512 #[test]
1513 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
1514 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
1518 assert_eq!(result.errors, 1, "{:?}", result.messages);
1519 }
1520
1521 #[test]
1522 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
1523 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
1527 assert_eq!(result.errors, 1, "{:?}", result.messages);
1528 }
1529
1530 #[test]
1531 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
1532 let source = "int f(void) { char c = 300; return c; }\n";
1533 let plain = run(&options(), source);
1534 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
1535 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
1536 assert!(!plain.text().is_empty(), "a warning is not a reason to write nothing");
1537
1538 let mut opts = options();
1539 opts.warnings_are_errors = true;
1540 let strict = run(&opts, source);
1541 assert!(strict.failed());
1542 assert!(strict.text().is_empty(), "and under -Werror it is a reason to write nothing");
1543 for message in &strict.messages {
1544 assert!(!message.contains("warning:"), "{message}");
1545 }
1546 }
1547
1548 #[test]
1549 fn w_drops_the_warning_before_werror_can_promote_it() {
1550 let source = "int f(void) { char c = 300; return c; }\n";
1551 let mut opts = options();
1552 opts.warnings = false;
1553 let quiet = run(&opts, source);
1554 assert_eq!(quiet.messages, Vec::<String>::new());
1555 assert_eq!(quiet.errors, 0);
1556 assert!(!quiet.text().is_empty(), "and the file still compiles");
1557
1558 opts.warnings_are_errors = true;
1561 let both = run(&opts, source);
1562 assert_eq!(both.messages, Vec::<String>::new());
1563 assert!(!both.failed(), "-w -Werror is not an error about a warning nobody saw");
1564 }
1565
1566 #[test]
1567 fn the_dialect_reaches_the_keywords_and_the_checking() {
1568 let source = "typeof(1) x;\n";
1571 let mut opts = options();
1572 opts.std = Std::C23;
1573 opts.gnu_extensions = false;
1574 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
1575
1576 opts.std = Std::C17;
1577 assert!(run(&opts, source).failed());
1578 }
1579
1580 #[test]
1581 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
1582 let mut opts = options();
1583 opts.emit = EmitKind::Object;
1584 let result = run(&opts, "int x = 1;\n");
1585 assert!(!result.failed(), "{:?}", result.messages);
1586 assert!(result.text().is_empty());
1587 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
1590 }
1591
1592 fn mir(source: &str) -> String {
1594 let mut opts = options();
1595 opts.emit = EmitKind::MirFinal;
1596 let result = run(&opts, source);
1597 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1598 result.text().to_owned()
1599 }
1600
1601 #[test]
1607 fn a_function_goes_from_c_to_instructions_with_real_registers_in_them() {
1608 let text = mir("int add(int a, int b) { return a + b; }\n");
1609 assert!(text.starts_with("mfunc @add {"), "{text}");
1610 assert!(text.contains("x64.add_rr_32"), "{text}");
1611 assert!(text.contains("x64.ret"), "{text}");
1612 assert!(!text.contains('%'), "{text}");
1615 }
1616
1617 #[test]
1619 fn a_function_with_no_body_produces_no_machine_function() {
1620 let text = mir("int g(int);\nint f(int a) { return g(a); }\n");
1621 assert_eq!(text.matches("mfunc @").count(), 1, "{text}");
1622 assert!(text.contains("mfunc @f {"), "{text}");
1623 assert!(text.contains("x64.call"), "{text}");
1624 }
1625
1626 #[test]
1628 fn every_definition_in_the_file_is_generated_and_they_keep_their_order() {
1629 let text = mir("int a(int x) { return x; }\nint b(int x) { return x; }\n");
1630 let first = text.find("mfunc @a").expect("the first function");
1631 let second = text.find("mfunc @b").expect("the second function");
1632 assert!(first < second, "{text}");
1633 }
1634
1635 #[test]
1637 fn the_target_decides_which_convention_the_generated_code_follows() {
1638 let mut opts = options();
1639 opts.emit = EmitKind::MirFinal;
1640 let linux = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1641 assert!(linux.contains("$rdi"), "{linux}");
1642
1643 opts.target = "x86_64-pc-windows-msvc".parse::<Triple>().unwrap();
1644 let windows = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1645 assert!(windows.contains("$rcx"), "{windows}");
1646 assert!(!windows.contains("$rdi"), "{windows}");
1647 }
1648
1649 #[test]
1651 fn a_target_this_has_no_back_end_for_is_reported_rather_than_generated() {
1652 let mut opts = options();
1653 opts.emit = EmitKind::MirFinal;
1654 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
1655 let result = run(&opts, "int f(int a) { return a; }\n");
1656 assert!(result.failed());
1657 assert!(result.messages[0].contains("no back end for aarch64"), "{:?}", result.messages);
1658 assert!(result.text().is_empty());
1659 }
1660
1661 #[test]
1668 fn a_construct_the_back_end_cannot_reach_yet_is_reported_against_its_function() {
1669 let mut opts = options();
1670 opts.emit = EmitKind::MirFinal;
1671 let source = "void a(int n) { int v[n]; v[0] = 1; }\n\
1672 void b(int n) { int v[n]; v[0] = 1; }\n";
1673 let result = run(&opts, source);
1674 assert!(result.failed());
1675 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
1676 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
1677 assert!(result.messages[0].contains("no rule lowers a `stacksave`"), "{:?}", result);
1678 assert!(result.messages[1].contains("cannot generate code for 'b'"), "{:?}", result);
1679 assert!(result.text().is_empty());
1680 }
1681
1682 #[test]
1689 fn an_opcode_with_no_name_in_the_rule_language_is_named_by_its_own_spelling() {
1690 let mut opts = options();
1691 opts.emit = EmitKind::MirFinal;
1692 let result = run(&opts, "int f(int a) {\n __int128 wide = a;\n return (int) wide;\n}\n");
1693 assert!(result.failed());
1694 assert!(
1695 result.messages[0].contains("no rule lowers a `sext` producing a `i128`"),
1696 "{result:?}"
1697 );
1698 assert!(result.messages[0].contains(":2:"), "the line the widening is on: {result:?}");
1699 assert!(!result.messages[0].contains("this instruction"), "{result:?}");
1700 }
1701
1702 #[test]
1704 fn the_note_on_unfinished_work_points_at_the_issues_rather_than_at_the_plan() {
1705 let mut opts = options();
1706 opts.emit = EmitKind::MirFinal;
1707 let result = run(&opts, "int f(int a) { __int128 wide = a; return (int) wide; }\n");
1708 assert!(result.failed());
1709 let note = result.messages.iter().find(|line| line.contains("note:")).expect("a note");
1710 assert!(note.contains("https://github.com/tamnd/rucc/issues"), "{note}");
1711 assert!(!note.contains("spec/17-milestones.md"), "{note}");
1712 }
1713
1714 #[test]
1716 fn the_frame_flags_on_the_command_line_reach_the_generated_frame() {
1717 let source = "int f(int a) { return a; }\n";
1718 assert!(!mir(source).contains("$rbp"), "a leaf needs no frame pointer by default");
1719
1720 let mut opts = options();
1721 opts.emit = EmitKind::MirFinal;
1722 opts.frame_pointer = true;
1723 let kept = run(&opts, source).text().to_owned();
1724 assert!(kept.contains("x64.push_64 $rbp"), "{kept}");
1725 }
1726
1727 fn asm(source: &str) -> String {
1729 let mut opts = options();
1730 opts.emit = EmitKind::Asm;
1731 let result = run(&opts, source);
1732 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1733 result.text().to_owned()
1734 }
1735
1736 #[test]
1743 fn a_function_goes_from_c_to_assembly_an_assembler_would_take() {
1744 let text = asm("int add(int a, int b) { return a + b; }\n");
1745 assert!(text.contains("\t.globl\tadd\n"), "{text}");
1746 assert!(text.contains("\t.type\tadd, @function\n"), "{text}");
1747 assert!(text.contains("\nadd:\n"), "{text}");
1748 assert!(text.contains("\taddl\t"), "{text}");
1749 assert!(text.contains("\tret\n"), "{text}");
1750 assert!(text.contains("\t.size\tadd, .-add\n"), "{text}");
1751 assert!(text.contains(".note.GNU-stack"), "{text}");
1754 }
1755
1756 #[test]
1762 fn a_call_through_a_function_pointer_goes_through_the_register_it_is_in() {
1763 let text = asm("int g(int);\nint f(int (*p)(int), int a) { return p(a) + g(a); }\n");
1764 assert!(text.contains("\tcall\t*%"), "{text}");
1765 assert!(text.contains("\tcall\tg\n"), "{text}");
1766 assert!(text.contains("%rdi"), "{text}");
1770 }
1771
1772 #[test]
1774 fn the_address_of_a_global_is_read_from_the_instruction_pointer() {
1775 let text = asm("extern int counter;\nint f(void) { return counter; }\n");
1776 assert!(text.contains("\tleaq\tcounter(%rip), "), "{text}");
1777 }
1778
1779 #[test]
1781 fn a_cast_between_a_pointer_and_an_integer_leaves_the_value_where_it_is() {
1782 let text = asm("long f(void *p) { return (long)p; }\n");
1783 for line in text.lines().filter(|line| line.starts_with('\t') && !line.contains('.')) {
1788 let mnemonic = line.split_whitespace().next().unwrap_or("");
1789 assert!(matches!(mnemonic, "movq" | "ret"), "{line} in\n{text}");
1790 }
1791 }
1792
1793 #[test]
1797 fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
1798 let six = "long a, long b, long c, long d, long e, long f";
1799 let text = asm(&format!("long f({six}, long g, long h) {{ return g + h; }}\n"));
1800
1801 assert!(text.contains("\tmovq\t8(%rsp), "), "{text}");
1805 assert!(text.contains("\tmovq\t16(%rsp), "), "{text}");
1806
1807 let narrow = asm(&format!("int f({six}, int g) {{ return g; }}\n"));
1811 assert!(narrow.contains("\tmovl\t8(%rsp), "), "{narrow}");
1812 let eight =
1813 "double a, double b, double c, double d, double e, double f, double g, double h";
1814 let float = asm(&format!("double f({eight}, double i) {{ return i; }}\n"));
1815 assert!(float.contains("\tmovsd\t8(%rsp), "), "{float}");
1816 }
1817
1818 #[test]
1821 fn a_call_writes_the_arguments_with_no_register_left_at_the_stack_pointer() {
1822 let six = "1, 2, 3, 4, 5, 6";
1823 let decl = "long g(long, long, long, long, long, long, long, long);\n";
1824 let text = asm(&format!("{decl}long f(void) {{ return g({six}, 7, 8); }}\n"));
1825
1826 assert!(text.contains("\tmovq\t%"), "{text}");
1827 assert!(text.contains(", (%rsp)\n"), "{text}");
1828 assert!(text.contains(", 8(%rsp)\n"), "{text}");
1829 assert!(text.contains("\tsubq\t$"), "{text}");
1831
1832 let narrow = "int g(int, int, int, int, int, int, int);\n";
1834 let text = asm(&format!("{narrow}int f(void) {{ return g({six}, 7); }}\n"));
1835 assert!(text.contains("\tmovl\t%"), "{text}");
1836 assert!(text.contains(", (%rsp)\n"), "{text}");
1837 }
1838
1839 #[test]
1842 fn a_variadic_call_counts_registers_and_not_arguments() {
1843 let nine = "1., 2., 3., 4., 5., 6., 7., 8., 9.";
1844 let decl = "int g(int, ...);\n";
1845 let text = asm(&format!("{decl}int f(void) {{ return g(0, {nine}); }}\n"));
1846
1847 assert!(text.contains("\tmovl\t$8, "), "eight registers, not nine: {text}");
1848 assert!(text.contains("\tmovsd\t%"), "{text}");
1849 assert!(text.contains(", (%rsp)\n"), "{text}");
1850 }
1851
1852 #[test]
1857 fn a_variadic_function_writes_the_argument_registers_it_was_handed_into_its_frame() {
1858 let body =
1859 "__builtin_va_list ap; __builtin_va_start(ap, n); __builtin_va_end(ap); return n;";
1860 let text = asm(&format!("int f(int n, ...) {{ {body} }}\n"));
1861
1862 let stores = |mnemonic: &str| text.matches(&format!("\t{mnemonic}\t%")).count();
1865 assert!(text.contains(", 8(%r"), "the second slot, not the first: {text}");
1866 assert!(!text.contains(", 0(%r"), "{text}");
1867 assert_eq!(stores("movsd"), 8, "every vector register: {text}");
1868
1869 assert!(text.contains("\tsubq\t$"), "{text}");
1871 }
1872
1873 #[test]
1876 fn va_start_writes_the_four_fields_the_psabi_describes() {
1877 let start = "__builtin_va_list ap; __builtin_va_start(ap, d);";
1878 let params = "int a, int b, int c, double d";
1879 let text = asm(&format!("int f({params}, ...) {{ {start} return a; }}\n"));
1880
1881 assert!(text.contains(" movl $24, "), "{text}");
1885 assert!(text.contains(" movl $64, "), "{text}");
1886 assert!(text.contains(", 8(%r"), "{text}");
1890 assert!(text.contains(", 16(%r"), "{text}");
1891 let frame: u32 = text
1892 .lines()
1893 .find_map(|line| line.trim().strip_prefix("subq $")?.split(',').next()?.parse().ok())
1894 .expect("a variadic function takes a frame for the save area");
1895 let above = |line: &str| {
1896 let at: u32 = line.trim().strip_prefix("leaq ")?.split('(').next()?.parse().ok()?;
1897 Some(at > frame)
1898 };
1899 assert!(text.lines().filter_map(above).any(|it| it), "{frame}: {text}");
1900 }
1901
1902 #[test]
1905 fn va_arg_branches_on_whether_the_argument_is_still_in_the_save_area() {
1906 let read = "__builtin_va_list ap; __builtin_va_start(ap, n);";
1907 let ints = format!("int f(int n, ...) {{ {read} return __builtin_va_arg(ap, int); }}\n");
1908 let text = asm(&ints);
1909
1910 assert!(text.contains("$40, "), "{text}");
1913 assert!(text.contains(" cmpl "), "{text}");
1914 assert!(text.contains(" setbe "), "unsigned, since an offset is a count of bytes: {text}");
1915
1916 let arg = "__builtin_va_arg(ap, double)";
1917 let text = asm(&format!("double f(int n, ...) {{ {read} return {arg}; }}\n"));
1918 assert!(text.contains("$160, "), "the last vector slot: {text}");
1919 }
1920
1921 #[test]
1924 fn a_structure_assignment_is_a_move_for_each_word_of_it() {
1925 let decl = "struct pair { long a, b; };\n";
1926 let body = "struct pair p = *q; return p.a + p.b;";
1927 let text = asm(&format!("{decl}long f(struct pair *q) {{ {body} }}\n"));
1928
1929 assert!(!text.contains("memcpy"), "nothing calls the library: {text}");
1930 assert!(!text.contains("\tcall"), "{text}");
1931 assert!(text.matches("\tmovq\t").count() >= 4, "two words each way: {text}");
1933 }
1934
1935 #[test]
1938 fn how_wide_a_word_of_a_copy_is_follows_the_alignment() {
1939 let decl = "struct bytes { char a[8]; };\n";
1940 let body = "struct bytes p = *q; return p.a[0];";
1941 let text = asm(&format!("{decl}int f(struct bytes *q) {{ {body} }}\n"));
1942
1943 assert!(text.matches("\tmovb\t").count() >= 16, "a byte at a time: {text}");
1945 }
1946
1947 #[test]
1950 fn the_part_of_an_initialiser_that_names_nothing_is_stored_as_zero() {
1951 let decl = "struct wide { long a, b, c; };\n";
1952 let text = asm(&format!("{decl}long f(void) {{ struct wide w = {{ 7 }}; return w.c; }}\n"));
1953
1954 assert!(!text.contains("memset"), "nothing calls the library: {text}");
1955 assert!(text.contains("\tmovq\t$0, ") || text.contains("$0, %"), "the zero: {text}");
1956 }
1957
1958 #[test]
1961 fn a_copy_too_large_to_unroll_calls_the_runtime() {
1962 let decl = "struct huge { char a[4096]; };\n";
1963 let mut opts = options();
1964 opts.emit = EmitKind::Asm;
1965 let source = format!("{decl}void f(struct huge *p, struct huge *q) {{ *p = *q; }}\n");
1966 let result = run(&opts, &source);
1967 assert!(!result.failed(), "{:?}", result.messages);
1968 let text = result.text();
1969 assert!(text.contains("call") && text.contains("memcpy"), "{text}");
1970 assert!(text.contains("4096"), "the size travels: {text}");
1973 }
1974
1975 #[test]
1978 fn a_realigned_frame_reads_them_through_the_frame_pointer() {
1979 let six = "long a, long b, long c, long d, long e, long f";
1980 let body = "_Alignas(32) long wide[4]; wide[0] = g; return wide[0];";
1981 let text = asm(&format!("long f({six}, long g) {{ {body} }}\n"));
1982
1983 assert!(text.contains("\tandq\t$-32, %rsp"), "{text}");
1987 assert!(text.contains("\tmovq\t16(%rbp), "), "{text}");
1988 assert!(!text.contains("\tmovq\t16(%rsp), "), "{text}");
1989 }
1990
1991 #[test]
1993 fn the_target_decides_how_the_assembly_is_spelled() {
1994 let mut opts = options();
1995 opts.emit = EmitKind::Asm;
1996 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
1997 let text = run(&opts, "int f(void) { return 0; }\n").text().to_owned();
1998 assert!(text.contains("__TEXT,__text"), "{text}");
1999 assert!(text.contains("\n_f:\n"), "{text}");
2000 assert!(!text.contains(".note.GNU-stack"), "{text}");
2001 }
2002
2003 fn obj(source: &str) -> Vec<u8> {
2005 let mut opts = options();
2006 opts.emit = EmitKind::Object;
2007 let result = run(&opts, source);
2008 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2009 match result.artifact {
2010 Artifact::Object(bytes) => bytes,
2011 other => panic!("expected an object, got {other:?}"),
2012 }
2013 }
2014
2015 #[test]
2021 fn a_function_goes_from_c_to_an_object_a_linker_would_take() {
2022 let bytes = obj("int add(int a, int b) { return a + b; }\n");
2023 assert_eq!(&bytes[..4], b"\x7fELF", "an object file starts by saying it is one");
2024 let text = asm("int add(int a, int b) { return a + b; }\n");
2025 assert!(
2026 text.contains("\taddl\t"),
2027 "and the listing of it is the same instructions:\n{text}"
2028 );
2029 }
2030
2031 #[test]
2033 fn a_variable_goes_from_c_to_the_section_it_belongs_in() {
2034 let text = asm("int counter = 42;\nstatic int hidden;\nconst int fixed = 7;\n");
2035 assert!(text.contains("\t.data\n\t.globl\tcounter\n"), "{text}");
2036 assert!(text.contains("\ncounter:\n\t.long\t42\n"), "{text}");
2037 assert!(text.contains("\t.size\tcounter, .-counter\n"), "{text}");
2038 assert!(text.contains("\t.bss\n\t.p2align\t2\n"), "{text}");
2041 assert!(text.contains("\nhidden:\n\t.space\t4\n"), "{text}");
2042 assert!(!text.contains(".globl\thidden"), "{text}");
2043 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2046 }
2047
2048 #[test]
2055 fn a_bit_field_initializer_writes_every_byte_of_the_value_and_not_only_the_ones_that_are_set() {
2056 let text = asm("struct s { unsigned f : 20; } x = { 0x12300 };\n");
2057 assert!(text.contains("\t.data\n"), "there is something to write: {text}");
2058 assert!(text.contains("\nx:\n\t.ascii\t\"\\000#\\001\"\n"), "and it is the value: {text}");
2059
2060 let text = asm("struct s { unsigned a : 8; unsigned b : 8; } x = { 0, 3 };\n");
2063 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\003\"\n"), "{text}");
2064
2065 let text = asm("struct s { unsigned long long f : 40; } x = { 0x100000 };\n");
2068 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\000\\020\"\n\t.space\t5\n"), "{text}");
2069
2070 let text = asm("struct s { unsigned f : 20; } x = { 0 };\n");
2072 assert!(text.contains("\t.bss\n"), "an object of zeroes is zeroes: {text}");
2073 assert!(text.contains("\nx:\n\t.space\t4\n"), "{text}");
2074 }
2075
2076 #[test]
2078 fn a_string_literal_is_a_variable_with_a_name_no_program_could_write() {
2079 let text = asm("const char *f(void) { return \"hi\"; }\n");
2080 assert!(text.contains("\t.ascii\t\"hi\\000\"\n"), "{text}");
2081 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2082 let label = text
2083 .lines()
2084 .find(|line| line.starts_with(".Lstr"))
2085 .unwrap_or_else(|| panic!("a label for the literal in\n{text}"));
2086 assert!(!text.contains(&format!(".globl\t{}", label.trim_end_matches(':'))), "{text}");
2087 }
2088
2089 #[test]
2091 fn an_address_in_an_initializer_is_left_to_the_linker() {
2092 let source = "int counter;\nint *p = &counter;\n";
2093 let text = asm(source);
2094 assert!(text.contains("\np:\n\t.quad\tcounter\n"), "{text}");
2095 let bytes = obj(source);
2098 assert!(bytes.windows(8).any(|w| w == b"counter\0"), "the object has to name it");
2099 }
2100
2101 #[test]
2110 fn a_constant_holding_an_address_goes_in_the_section_the_loader_may_write_once() {
2111 let text = asm("static void a(void) {}\nstatic void b(void) {}\n\
2114 struct m { void (*x)(void); void (*y)(void); };\n\
2115 const struct m t = { a, b };\n");
2116 assert!(text.contains("\t.section\t.data.rel.ro.local,\"aw\",@progbits\n"), "{text}");
2117 assert!(text.contains("\nt:\n\t.quad\ta\n\t.quad\tb\n"), "{text}");
2118
2119 let text =
2122 asm("void a(void);\nstruct m { void (*x)(void); };\nconst struct m t = { a };\n");
2123 assert!(text.contains("\t.section\t.data.rel.ro,\"aw\",@progbits\n"), "{text}");
2124
2125 let text = asm("const int fixed = 7;\n");
2127 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2128 }
2129
2130 #[test]
2132 fn a_thread_local_variable_is_reported_as_work_that_is_not_done() {
2133 let mut opts = options();
2134 opts.emit = EmitKind::Asm;
2135 let result = run(&opts, "_Thread_local int x = 1;\n");
2136 assert!(result.failed(), "every thread sharing one variable is worse than a message");
2137 assert!(result.messages.iter().any(|m| m.contains("thread-local")), "{:?}", result);
2138 assert!(!result.messages.iter().any(|m| m.contains("internal")), "{:?}", result);
2140 }
2141
2142 #[test]
2144 fn the_object_and_the_listing_are_two_spellings_of_one_compilation() {
2145 let source = "int callee(void); int g(void) { return callee(); }\n";
2149 let bytes = obj(source);
2150 assert!(
2151 bytes.windows(7).any(|w| w == b"callee\0"),
2152 "the object has to name the callee for the linker to find it"
2153 );
2154 let text = asm(source);
2155 assert!(text.contains("\tcall\tcallee\n"), "{text}");
2156 }
2157
2158 #[test]
2164 fn compiling_for_an_executable_produces_an_object_and_not_a_dump() {
2165 let mut opts = options();
2166 opts.emit = EmitKind::Executable;
2168 let result = run(&opts, "int main(void) { return 0; }\n");
2169 assert_eq!(result.messages, Vec::<String>::new());
2170 match result.artifact {
2171 Artifact::Object(bytes) => assert_eq!(&bytes[..4], b"\x7fELF"),
2172 other => panic!("expected an object, got {other:?}"),
2173 }
2174 }
2175
2176 #[test]
2178 fn a_platform_with_no_object_writer_is_said_so_rather_than_written_as_elf() {
2179 let mut opts = options();
2180 opts.emit = EmitKind::Object;
2181 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2182 let result = run(&opts, "int f(void) { return 0; }\n");
2183 assert!(result.failed(), "an object nobody can read is worse than a message");
2184 assert!(
2185 result.messages.iter().any(|m| m.contains("no object writer")),
2186 "{:?}",
2187 result.messages
2188 );
2189 }
2190
2191 fn ir(source: &str) -> String {
2193 let mut opts = options();
2194 opts.emit = EmitKind::Ir;
2195 let result = run(&opts, source);
2196 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2197 result.text().to_owned()
2198 }
2199
2200 fn errors(source: &str) -> Vec<String> {
2202 let mut opts = options();
2203 opts.emit = EmitKind::Ir;
2204 let result = run(&opts, source);
2205 assert!(result.failed(), "expected this to be refused:\n{source}");
2206 result.messages
2207 }
2208
2209 fn body(source: &str) -> String {
2211 let text = ir(source);
2212 let (_, rest) = text.split_once("{\n").expect("a function definition");
2213 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
2214 body.to_owned()
2215 }
2216
2217 #[test]
2225 fn gnu89_inline_is_what_decides_whether_a_bare_inline_definition_reaches_the_module() {
2226 let source = "inline int f(int x) { return x + 1; }\n";
2227 let with = |flag: bool| {
2228 let mut opts = options();
2229 opts.emit = EmitKind::Ir;
2230 opts.gnu89_inline = flag;
2231 let result = run(&opts, source);
2232 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
2233 result.text().to_owned()
2234 };
2235
2236 assert!(!with(false).contains("block0"), "no body: {}", with(false));
2239
2240 assert!(with(true).contains("block0"), "a body: {}", with(true));
2243 }
2244
2245 #[test]
2259 fn lowering_says_nothing_about_the_type_an_access_went_through() {
2260 let source = "\
2263struct s { int a; float b; };\n\
2264union u { int i; float f; };\n\
2265int scalar(int *p) { return *p; }\n\
2266float member(struct s *p) { p->a = 1; return p->b; }\n\
2267int element(int *a, long i) { return a[i]; }\n\
2268float through_a_union(union u *p) { p->i = 1; return p->f; }\n";
2269 assert!(!ir(source).contains("tbaa"), "{}", ir(source));
2270 }
2271
2272 #[test]
2280 fn a_bare_return_from_a_function_that_promised_a_value_gives_back_a_zero() {
2281 let mut opts = options();
2282 opts.emit = EmitKind::Ir;
2283 opts.std = Std::C89;
2284 let compiled = |source: &str| {
2285 let result = run(&opts, source);
2286 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
2287 result.text().to_owned()
2288 };
2289
2290 let text = compiled("int f(int x) { if (x) return; return 3; }\n");
2291 assert!(text.contains("iconst.i32 0\n return"), "zero goes back: {text}");
2292 assert!(!text.contains("unreachable"), "the branch that reached it is kept: {text}");
2293
2294 let text = compiled("double f(int x) { if (x) return; return 1.0; }\n");
2296 assert!(text.contains("fconst.f64 0x0\n return"), "a float zero goes back: {text}");
2297 }
2298
2299 #[test]
2307 fn a_call_to_a_name_nothing_declared_declares_it_as_c89_said_to() {
2308 let mut opts = options();
2309 opts.emit = EmitKind::Ir;
2310 opts.std = Std::C89;
2311 let compiled = |source: &str| {
2312 let result = run(&opts, source);
2313 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
2314 result.text().to_owned()
2315 };
2316
2317 let text = compiled("int f(void) { return g(); }\n");
2319 assert!(text.contains("call @g"), "the call is to the name that was written: {text}");
2320 assert!(text.contains("i32"), "and it gives back an int: {text}");
2321
2322 let text = compiled("int f(char c) { return g(c); }\n");
2325 assert!(text.contains("sext.i32"), "the argument is promoted: {text}");
2326
2327 let mut opts = options();
2330 opts.std = Std::C89;
2331 let said = run(&opts, "int f(void) { return h; }\n").messages.join("\n");
2332 assert!(said.contains("'h' undeclared"), "not a call, so not declared: {said}");
2333 }
2334
2335 #[test]
2345 fn a_name_called_before_it_is_defined_still_gets_its_definition() {
2346 let mut opts = options();
2347 opts.emit = EmitKind::Ir;
2348 opts.std = Std::C89;
2349 let text = run(&opts, "int f(void) { return dummy(); }\ndummy () { return 7; }\n")
2350 .text()
2351 .to_owned();
2352 assert!(text.contains("func @f()"), "the caller is there: {text}");
2353 assert!(text.contains("func @dummy"), "and so is what it calls: {text}");
2354 assert!(text.contains("iconst.i32 7"), "with the body it was given: {text}");
2355 }
2356
2357 #[test]
2365 fn an_old_style_parameter_is_converted_from_what_the_call_promoted_it_to() {
2366 let mut opts = options();
2367 opts.emit = EmitKind::Ir;
2368 opts.std = Std::C89;
2369 let compiled = |source: &str| run(&opts, source).text().to_owned();
2370
2371 let text = compiled("f (c) unsigned char c; { return c; }\n");
2372 assert!(text.contains("func @f(i32"), "an int arrives: {text}");
2373 assert!(text.contains("trunc.i8"), "and is cut down to what was declared: {text}");
2374 assert!(text.contains("zext.i32"), "then read back unsigned: {text}");
2375
2376 let text = compiled("f (s) short s; { return s; }\n");
2378 assert!(text.contains("trunc.i16"), "cut down: {text}");
2379 assert!(text.contains("sext.i32"), "and read back signed: {text}");
2380
2381 let text = compiled("f (x) float x; { return x * 2; }\n");
2384 assert!(text.contains("func @f(f64"), "a double arrives: {text}");
2385 assert!(text.contains("fptrunc.f32"), "and is narrowed to the float: {text}");
2386
2387 let text = compiled("int f(unsigned char c) { return c; }\n");
2390 assert!(text.contains("func @f(i8)"), "the declared type arrives: {text}");
2391 assert!(!text.contains("trunc"), "so there is nothing to cut down: {text}");
2392 }
2393
2394 #[test]
2403 fn the_rules_gcc_promoted_are_decided_by_the_dialect_and_by_fpermissive() {
2404 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
2406 let cases = [
2407 ("static counted;\n", ["", "error", "warning", "error"]),
2408 ("int f(void) { return g(); }\n", ["", "error", "warning", "error"]),
2409 ("int f(x) { return x; }\n", ["", "error", "warning", "error"]),
2410 ("int *p;\nvoid h(void) { p = 1; }\n", ["warning", "error", "warning", "error"]),
2411 (
2412 "char *q;\nint *r;\nvoid k(void) { r = q; }\n",
2413 ["warning", "error", "warning", "error"],
2414 ),
2415 ("int f(void) { return; }\n", ["", "error", "warning", "error"]),
2416 ("void g(void) { return 1; }\n", ["warning", "error", "warning", "error"]),
2417 ];
2418
2419 for (source, wanted) in cases {
2420 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
2421 let mut opts = options();
2422 opts.std = std;
2423 opts.permissive = permissive;
2424 let said = run(&opts, source).messages.join("\n");
2425 let severity = if said.contains(": error: ") {
2426 "error"
2427 } else if said.contains(": warning: ") {
2428 "warning"
2429 } else {
2430 ""
2431 };
2432 let how = if permissive { " -fpermissive" } else { "" };
2433 assert_eq!(
2434 severity,
2435 wanted,
2436 "under -std={}{how}, {source} was answered with `{said}`",
2437 std.as_str()
2438 );
2439 if wanted.is_empty() {
2440 assert!(said.is_empty(), "nothing to say, but said `{said}`");
2441 }
2442 }
2443 }
2444 }
2445
2446 #[test]
2455 fn the_three_variadic_builtins_answer_a_bad_list_the_way_a_call_answers_a_bad_argument() {
2456 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
2457 let cases = [
2458 (
2459 "int f(int n, ...) { char *p; return __builtin_va_arg(p, int); }\n",
2460 "first argument to 'va_arg' not of type 'va_list'",
2461 ["error", "error", "error", "error"],
2462 ),
2463 (
2464 "void f(int n, ...) { char *p; __builtin_va_start(p, n); }\n",
2465 "passing argument 1 of '__builtin_va_start' from incompatible pointer type",
2466 ["warning", "error", "warning", "error"],
2467 ),
2468 (
2469 "void f(int n, ...) { int x; __builtin_va_end(x); }\n",
2470 "passing argument 1 of '__builtin_va_end' makes pointer from integer without a \
2471 cast",
2472 ["warning", "error", "warning", "error"],
2473 ),
2474 (
2475 "void f(int n, ...) { __builtin_va_list a; char *p; __builtin_va_copy(a, p); }\n",
2476 "passing argument 2 of '__builtin_va_copy' from incompatible pointer type",
2477 ["warning", "error", "warning", "error"],
2478 ),
2479 ];
2480
2481 for (source, message, wanted) in cases {
2482 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
2483 let mut opts = options();
2484 opts.std = std;
2485 opts.permissive = permissive;
2486 let said = run(&opts, source).messages.join("\n");
2487 let how = if permissive { " -fpermissive" } else { "" };
2488 assert!(
2489 said.contains(&format!(": {wanted}: {message}")),
2490 "under -std={}{how}, {source} was answered with `{said}`",
2491 std.as_str()
2492 );
2493 }
2494 }
2495 }
2496
2497 fn safe_ir(tier: rucc_session::Safety, source: &str) -> String {
2499 let mut opts = options();
2500 opts.emit = EmitKind::Ir;
2501 opts.safety = tier;
2502 let result = run(&opts, source);
2503 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2504 result.text().to_owned()
2505 }
2506
2507 const READS_THROUGH_A_POINTER: &str = "int read(int *p) { return p[1]; }\n";
2508
2509 #[test]
2510 fn a_build_that_did_not_ask_for_the_monitor_is_compiled_the_way_it_always_was() {
2511 let text = ir(READS_THROUGH_A_POINTER);
2515 assert!(!text.contains("check_"), "{text}");
2516 assert!(!text.contains("cap_of"), "{text}");
2517 }
2518
2519 #[test]
2520 fn asking_for_a_tier_puts_the_checks_in_before_the_optimizer_sees_them() {
2521 let text = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2522 assert!(text.contains("cap_of"), "{text}");
2523 assert!(text.contains("check_bounds"), "{text}");
2524 assert!(text.contains("check_live"), "{text}");
2525 assert!(text.contains("check_deriv"), "{text}");
2527 }
2528
2529 #[test]
2530 fn the_three_tiers_that_are_not_off_all_check_the_same_accesses_so_far() {
2531 let detect = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2535 for tier in [rucc_session::Safety::Enforce, rucc_session::Safety::Kernel] {
2536 assert_eq!(safe_ir(tier, READS_THROUGH_A_POINTER), detect, "{tier}");
2537 }
2538 }
2539
2540 fn summary(tier: rucc_session::Safety, source: &str) -> String {
2542 let mut opts = options();
2543 opts.emit = EmitKind::SafetySummary;
2544 opts.safety = tier;
2545 let result = run(&opts, source);
2546 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2547 result.text().to_owned()
2548 }
2549
2550 #[test]
2551 fn the_summary_counts_the_checks_that_went_in_and_the_ones_still_standing() {
2552 let text = summary(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2553 assert!(text.contains("\"tier\": \"detect\""), "{text}");
2554 assert!(
2556 text.contains("\"bounds\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"),
2557 "{text}"
2558 );
2559 assert!(
2560 text.contains(
2561 "\"derivation\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"
2562 ),
2563 "{text}"
2564 );
2565 }
2566
2567 #[test]
2568 fn a_build_without_the_monitor_summarises_as_a_build_with_no_checks_in_it() {
2569 let text = summary(rucc_session::Safety::Off, READS_THROUGH_A_POINTER);
2573 assert!(text.contains("\"tier\": \"off\""), "{text}");
2574 assert!(
2575 text.contains("\"bounds\": { \"emitted\": 0, \"remaining\": 0, \"discharged\": 0 }"),
2576 "{text}"
2577 );
2578 }
2579
2580 #[test]
2581 fn a_call_the_boundary_models_is_counted_apart_from_one_it_does_not() {
2582 let text = summary(
2583 rucc_session::Safety::Detect,
2584 "void *memcpy(void *, const void *, unsigned long);\n\
2585 int puts(const char *);\n\
2586 void f(char *d, char *s) { memcpy(d, s, 4); puts(d); }\n",
2587 );
2588 assert!(text.contains("\"interposed\": 1"), "{text}");
2589 assert!(text.contains("\"puts\""), "{text}");
2590 assert!(!text.contains("__rucc_wrap_memcpy\""), "{text}");
2594 }
2595
2596 #[test]
2597 fn the_two_directions_a_pointer_crosses_the_boundary_are_counted_apart() {
2598 let text = summary(
2602 rucc_session::Safety::Detect,
2603 "void *notes_open(void);\n\
2604 char *f(char *p) { char *q = notes_open(); return q ? q : p; }\n",
2605 );
2606 assert!(text.contains("\"crossings\": { \"entered\": 1, \"returned\": 1 }"), "{text}");
2607 assert!(text.contains("\"notes_open\""), "{text}");
2608 }
2609
2610 #[test]
2611 fn a_static_function_nobody_takes_the_address_of_is_not_a_crossing() {
2612 let text = summary(
2615 rucc_session::Safety::Detect,
2616 "static int len(const char *p) { return p ? 1 : 0; }\n\
2617 int f(void) { return len(\"x\"); }\n",
2618 );
2619 assert!(text.contains("\"crossings\": { \"entered\": 0, \"returned\": 0 }"), "{text}");
2620 }
2621
2622 fn granules(source: &str) -> String {
2624 let mut opts = options();
2625 opts.emit = EmitKind::TypeGranules;
2626 let result = run(&opts, source);
2627 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2628 result.text().to_owned()
2629 }
2630
2631 #[test]
2632 fn the_granule_report_names_every_record_and_both_keyings() {
2633 let text = granules(
2634 "struct hot { char *p; int a; int b; };\n\
2635 int f(struct hot *h) { return h->a; }\n",
2636 );
2637 assert!(text.contains("struct hot"), "{text}");
2638 assert!(text.contains("every type distinct"), "{text}");
2641 assert!(text.contains("every pointer one type"), "{text}");
2642 assert!(text.contains("budget"), "{text}");
2643 }
2644
2645 #[test]
2646 fn a_record_nothing_uses_is_still_measured() {
2647 let text = granules("struct unused { long a; double b; };\nint f(void) { return 0; }\n");
2650 assert!(text.contains("struct unused"), "{text}");
2651 }
2652
2653 #[test]
2654 fn the_granule_report_stops_before_anything_is_lowered() {
2655 let text = granules(
2659 "struct wide { long double d; };\n\
2660 long double f(long double x) { return x * x; }\n",
2661 );
2662 assert!(text.contains("struct wide"), "{text}");
2663 }
2664
2665 #[test]
2666 fn a_witness_reaches_the_assembler_as_a_call_to_the_runtime() {
2667 let text = safe_asm(rucc_session::Safety::Detect, "char *f(char *p) { return p; }\n");
2670 assert!(text.contains("\tcall\t__rucc_cap_witness\n"), "{text}");
2671 }
2672
2673 #[test]
2674 fn a_pointer_turned_into_an_integer_is_on_the_trust_set() {
2675 let text = summary(
2676 rucc_session::Safety::Detect,
2677 "unsigned long f(int *p) { return (unsigned long) p; }\n",
2678 );
2679 assert!(text.contains("\"exposed\": 1"), "{text}");
2680 }
2681
2682 fn safe_asm(tier: rucc_session::Safety, source: &str) -> String {
2684 let mut opts = options();
2685 opts.emit = EmitKind::Asm;
2686 opts.safety = tier;
2687 let result = run(&opts, source);
2688 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2689 result.text().to_owned()
2690 }
2691
2692 #[test]
2693 fn a_check_reaches_the_assembler_as_a_call_to_the_runtime() {
2694 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2695 assert!(text.contains("\tcall\t__rucc_check_bounds\n"), "{text}");
2696 assert!(text.contains("\tcall\t__rucc_check_live\n"), "{text}");
2697 assert!(text.contains("\tcall\t__rucc_check_deriv\n"), "{text}");
2698 }
2699
2700 #[test]
2701 fn every_check_that_reached_the_assembler_has_a_row_describing_it() {
2702 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2706 let section = format!("\t.section\t{},", rucc_safety::SECTION);
2707 assert_eq!(text.matches(§ion).count(), 3, "{text}");
2708 for index in 0..3 {
2709 let name = format!("__rucc_safety_desc_{index}");
2710 assert!(text.contains(&format!("{name}:\n")), "{text}");
2713 assert!(text.contains(&format!("{name}(%rip)")), "{text}");
2714 }
2715 assert!(!text.contains("__rucc_safety_desc_3"), "{text}");
2716 }
2717
2718 #[test]
2726 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
2727 let text = ir(concat!(
2728 "int g;\n",
2729 "int a = __builtin_constant_p(1);\n",
2730 "int b = __builtin_constant_p(g);\n",
2731 "int c = __builtin_constant_p(\"abc\");\n",
2732 "int d = __builtin_constant_p(&g);\n",
2733 "int e = __builtin_constant_p(1.5);\n",
2734 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
2735 ));
2736 assert!(text.contains("global @a : i32 = 1,"), "{text}");
2737 assert!(text.contains("global @b : i32 = 0,"), "{text}");
2738 assert!(text.contains("global @c : i32 = 1,"), "{text}");
2739 assert!(text.contains("global @d : i32 = 0,"), "{text}");
2740 assert!(text.contains("global @e : i32 = 1,"), "{text}");
2741 assert!(text.contains("global @h : i32 = 11,"), "{text}");
2742 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
2743
2744 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
2748 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
2749 }
2750
2751 #[test]
2760 fn a_call_to_a_library_builtin_reaches_the_library_function() {
2761 let text = body("void f(void) { __builtin_abort(); }\n");
2762 assert_eq!(text, "block0:\n call @abort() : ()\n return\n");
2763
2764 let text = ir("int f(const char *s) { return __builtin_puts(s) + __builtin_strlen(s); }\n");
2767 assert!(text.contains("call @puts(%0) : (ptr) -> i32"), "{text}");
2768 assert!(text.contains("call @strlen(%0) : (ptr) -> i64"), "{text}");
2769 assert!(!text.contains("__builtin_"), "the prefix is not part of any name here:\n{text}");
2770 }
2771
2772 #[test]
2783 fn the_absolute_value_family_is_the_magnitude_and_not_a_call() {
2784 let text = body(concat!(
2785 "long long llabs(long long);\n",
2786 "long long f(long long x) { return llabs(x); }\n",
2787 ));
2788 assert!(text.contains("%1 = iconst.i64 63"), "{text}");
2789 assert!(text.contains("%2 = ashr %0, %1"), "{text}");
2790 assert!(text.contains("%3 = xor %0, %2"), "{text}");
2791 assert!(text.contains("%4 = sub %3, %2"), "{text}");
2792 assert!(!text.contains("call"), "the call does not happen:\n{text}");
2793
2794 let text = body("int abs(int);\nint f(int x) { return abs(x); }\n");
2797 assert!(text.contains("iconst.i32 31"), "{text}");
2798 let text = body("long labs(long);\nlong f(long x) { return labs(x); }\n");
2799 assert!(text.contains("iconst.i64 63"), "{text}");
2800
2801 let text = body("long long f(long long x) { return __builtin_llabs(x); }\n");
2804 assert!(!text.contains("call"), "{text}");
2805
2806 let text = ir(concat!(
2808 "long long llabs(long long b);\n",
2809 "long long g(long long x) { return llabs(x); }\n",
2810 "long long llabs(long long b) { return 7; }\n",
2811 ));
2812 assert!(!text.contains("call @llabs"), "{text}");
2813 }
2814
2815 #[test]
2822 fn a_byte_swap_is_arithmetic_and_not_a_call() {
2823 let text = body("unsigned f(unsigned x) { return __builtin_bswap32(x); }\n");
2824 assert_eq!(text, "block0(%0: i32):\n %1 = bswap %0\n return %1\n");
2825
2826 let text = body("unsigned f(unsigned char c) { return __builtin_bswap32(c); }\n");
2829 assert!(text.contains("zext.i32 %0"), "widened first: {text}");
2830 assert!(text.contains("bswap %1"), "and swapped at four bytes: {text}");
2831 }
2832
2833 #[test]
2839 fn the_byte_swaps_reverse_at_the_width_their_name_says() {
2840 for (name, ty, width) in [
2841 ("__builtin_bswap16", "unsigned short", "i16"),
2842 ("__builtin_bswap32", "unsigned", "i32"),
2843 ("__builtin_bswap64", "unsigned long long", "i64"),
2844 ] {
2845 let source = format!("{ty} f({ty} x) {{ return {name}(x); }}\n");
2846 let text = body(&source);
2847 assert_eq!(
2848 text,
2849 format!("block0(%0: {width}):\n %1 = bswap %0\n return %1\n"),
2850 "{name}"
2851 );
2852 }
2853 }
2854
2855 #[test]
2862 fn the_bit_counts_are_instructions_and_not_calls() {
2863 let text = body("int f(unsigned x) { return __builtin_clz(x); }\n");
2864 assert_eq!(text, "block0(%0: i32):\n %1 = ctlz %0\n return %1\n");
2865
2866 let text = body("int f(unsigned x) { return __builtin_ctz(x); }\n");
2867 assert_eq!(text, "block0(%0: i32):\n %1 = cttz %0\n return %1\n");
2868
2869 let text = body("int f(unsigned x) { return __builtin_popcount(x); }\n");
2870 assert_eq!(text, "block0(%0: i32):\n %1 = ctpop %0\n return %1\n");
2871 }
2872
2873 #[test]
2882 fn the_bit_counts_ask_about_the_width_their_name_says() {
2883 let text = body("int f(unsigned long long x) { return __builtin_clzll(x); }\n");
2884 assert!(text.starts_with("block0(%0: i64):"), "counted at eight bytes: {text}");
2885 assert!(text.contains("%1 = ctlz %0"), "{text}");
2886 assert!(text.contains("trunc.i32 %1"), "and answered in an int: {text}");
2887
2888 let text = body("int f(unsigned long long x) { return __builtin_clz(x); }\n");
2891 assert!(text.contains("trunc.i32 %0"), "narrowed to what was asked about: {text}");
2892 assert!(text.contains("ctlz %1"), "and counted there: {text}");
2893
2894 let text = body("int f(unsigned long x) { return __builtin_popcountl(x); }\n");
2895 assert!(text.contains("%1 = ctpop %0"), "{text}");
2896 assert!(!text.contains("call"), "{text}");
2897 }
2898
2899 #[test]
2904 fn a_parity_is_the_low_bit_of_the_set_bit_count() {
2905 let text = body("int f(unsigned x) { return __builtin_parity(x); }\n");
2906 assert!(text.contains("%1 = ctpop %0"), "{text}");
2907 assert!(text.contains("iconst.i32 1"), "{text}");
2908 assert!(text.contains("and %1, %2"), "the low bit of it: {text}");
2909 }
2910
2911 #[test]
2917 fn the_first_set_bit_is_one_based_and_zero_for_a_zero() {
2918 let text = body("int f(int x) { return __builtin_ffs(x); }\n");
2919 assert!(text.contains("%1 = cttz %0"), "{text}");
2920 assert!(text.contains("%4 = add %1, %2"), "one more than the count: {text}");
2921 assert!(text.contains("%5 = icmp ne %0, %3"), "whether there was a bit at all: {text}");
2922 assert!(text.contains("%7 = sub %3, %6"), "spread to a mask: {text}");
2923 assert!(text.contains("%8 = and %4, %7"), "and kept only then: {text}");
2924 assert!(!text.contains("br_if"), "no branch: {text}");
2925 }
2926
2927 #[test]
2937 fn an_overflow_check_is_arithmetic_and_not_a_call() {
2938 let text =
2939 body("int f(int a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
2940 assert!(text.contains("%3, %4 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
2941 assert!(text.contains("store %3 -> %2"), "{text}");
2942 assert!(!text.contains("call"), "{text}");
2943
2944 let text =
2945 body("int f(int a, int b, int *r) { return __builtin_sub_overflow(a, b, r); }\n");
2946 assert!(text.contains("ssub_overflow.(i32, i1) %0, %1"), "{text}");
2947
2948 let text =
2949 body("int f(int a, int b, int *r) { return __builtin_mul_overflow(a, b, r); }\n");
2950 assert!(text.contains("smul_overflow.(i32, i1) %0, %1"), "{text}");
2951
2952 let text = body(
2955 "int f(unsigned a, unsigned b, unsigned *r) { return __builtin_add_overflow(a, b, r); }\n",
2956 );
2957 assert!(text.contains("uadd_overflow.(i32, i1) %0, %1"), "{text}");
2958 }
2959
2960 #[test]
2968 fn an_overflow_check_is_done_at_a_type_that_holds_every_operand() {
2969 let text = body(
2970 "int f(unsigned a, int b, long long *r) { return __builtin_add_overflow(a, b, r); }\n",
2971 );
2972 assert!(text.contains("%3 = zext.i64 %0"), "the unsigned operand keeps its value: {text}");
2973 assert!(text.contains("%4 = sext.i64 %1"), "and so does the signed one: {text}");
2974 assert!(text.contains("sadd_overflow.(i64, i1) %3, %4"), "{text}");
2975
2976 let text = body(
2979 "int f(long long a, long long b, long long *r) { return __builtin_mul_overflow(a, b, r); }\n",
2980 );
2981 assert!(text.contains("smul_overflow.(i64, i1) %0, %1"), "{text}");
2982 assert!(!text.contains("sext."), "{text}");
2983 assert!(!text.contains("zext.i64"), "{text}");
2985 }
2986
2987 #[test]
2995 fn an_overflow_check_writes_the_wrapped_answer_whether_or_not_it_fit() {
2996 let text =
2997 body("int f(int a, int b, char *r) { return __builtin_sub_overflow(a, b, r); }\n");
2998 assert!(text.contains("%3, %4 = ssub_overflow.(i32, i1) %0, %1"), "{text}");
2999 assert!(text.contains("%5 = trunc.i8 %3"), "narrowed to where it goes: {text}");
3000 assert!(text.contains("%6 = sext.i32 %5"), "and back: {text}");
3001 assert!(text.contains("%7 = icmp ne %6, %3"), "which is whether it fit: {text}");
3002 assert!(text.contains("store %5 -> %2"), "the narrowed value is stored either way: {text}");
3003 assert!(text.contains("%8 = or %4, %7"), "and either bit is an overflow: {text}");
3004 }
3005
3006 #[test]
3013 fn a_call_needing_more_than_sixty_four_bits_says_so() {
3014 let refused = concat!(
3015 "int f(unsigned long long a, long long b, long long *r) {\n",
3016 " return __builtin_add_overflow(a, b, r);\n",
3017 "}\n",
3018 );
3019 let messages = errors(refused);
3020 assert_eq!(messages.len(), 1, "{messages:?}");
3021 assert!(messages[0].contains("E0694"), "{messages:?}");
3022 assert!(messages[0].contains("wider than 64 bits"), "{messages:?}");
3023 }
3024
3025 #[test]
3028 fn an_overflow_check_over_something_that_is_not_an_integer_says_so() {
3029 let messages =
3030 errors("int f(double a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3031 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3032
3033 let messages =
3034 errors("int f(int a, int b, double *r) { return __builtin_add_overflow(a, b, r); }\n");
3035 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3036 }
3037
3038 #[test]
3049 fn an_ordered_access_is_ordered_in_the_ir() {
3050 let text = body("int f(int *p) { return __atomic_load_n(p, 0); }\n");
3051 assert!(text.contains("atomic_load.i32 %0, align 4, relaxed"), "{text}");
3052
3053 let text = body("long f(long *p) { return __atomic_load_n(p, 2); }\n");
3054 assert!(text.contains("atomic_load.i64 %0, align 8, acquire"), "{text}");
3055
3056 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
3057 assert!(text.contains("atomic_store %1 -> %0, align 4, release"), "{text}");
3058
3059 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
3060 assert!(text.contains("atomic_store %1 -> %0, align 4, seq_cst"), "{text}");
3061
3062 let text = body("void f(char *p, int v) { __atomic_store_n(p, v, 0); }\n");
3065 assert!(text.contains("trunc.i8 %1"), "{text}");
3066 assert!(text.contains("atomic_store %2 -> %0, align 1, relaxed"), "{text}");
3067 }
3068
3069 #[test]
3078 fn an_ordered_access_is_the_plain_instruction_on_this_machine() {
3079 let text = asm("int f(int *p) { return __atomic_load_n(p, 5); }\n");
3080 assert!(text.contains("movl\t(%rdi), %eax"), "{text}");
3081 assert!(!text.contains("mfence"), "a load needs no barrier here: {text}");
3082
3083 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
3084 assert!(text.contains("movl\t%esi, (%rdi)"), "{text}");
3085 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
3086
3087 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
3088 let (before, after) = text.split_once("mfence").expect("a barrier: {text}");
3089 assert!(before.contains("movl\t%esi, (%rdi)"), "the store comes first: {text}");
3090 assert!(!after.contains("movl"), "and nothing else is between them: {text}");
3091 }
3092
3093 #[test]
3103 fn a_barrier_is_one_instruction_at_the_strongest_ordering_and_none_below_it() {
3104 assert!(asm("void f(void) { __atomic_thread_fence(5); }\n").contains("mfence"));
3105 assert!(asm("void f(void) { __sync_synchronize(); }\n").contains("mfence"));
3106
3107 for weaker in ["1", "2", "3", "4"] {
3108 let source = format!("void f(void) {{ __atomic_thread_fence({weaker}); }}\n");
3109 assert!(!asm(&source).contains("mfence"), "{weaker} costs nothing here");
3110 }
3111 }
3112
3113 #[test]
3119 fn a_compare_and_exchange_is_one_instruction_answering_two_things() {
3120 let text =
3123 body("int f(int *p, int e, int d) { return __sync_val_compare_and_swap(p, e, d); }\n");
3124 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
3125 assert!(text.contains("return %3"), "the value it found: {text}");
3126
3127 let text =
3128 body("int f(int *p, int e, int d) { return __sync_bool_compare_and_swap(p, e, d); }\n");
3129 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
3130 assert!(text.contains("zext.i32 %4"), "whether it happened: {text}");
3131
3132 let text = body(
3135 "int f(int *p, int *e, int d) { return __atomic_compare_exchange_n(p, e, d, 0, 4, 2); }\n",
3136 );
3137 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3138 assert!(text.contains("%4, %5 = cmpxchg.(i32, i1) %0, %3, %2, align 4, acq_rel"), "{text}");
3139 assert!(text.contains("br_if %5, block2, block1"), "{text}");
3140 assert!(text.contains("store %4 -> %1, align 4"), "{text}");
3141
3142 let text = body(
3145 "int f(int *p, int *e, int *d) { return __atomic_compare_exchange(p, e, d, 0, 5, 5); }\n",
3146 );
3147 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3148 assert!(text.contains("%4 = load.i32 %2, align 4"), "{text}");
3149 assert!(text.contains("%5, %6 = cmpxchg.(i32, i1) %0, %3, %4, align 4, seq_cst"), "{text}");
3150 }
3151
3152 #[test]
3159 fn a_compare_and_exchange_is_a_locked_instruction_at_the_width_of_the_object() {
3160 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
3161 for (ty, suffix, reg) in widths {
3162 let source = format!(
3163 "int f({ty} *p, {ty} e, {ty} d) {{ return __sync_bool_compare_and_swap(p, e, d); }}\n"
3164 );
3165 let text = asm(&source);
3166 assert!(text.contains("\tlock\n"), "{ty}: {text}");
3167 assert!(text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3168 assert!(text.contains("sete\t"), "{ty}: {text}");
3169 }
3170 let source =
3171 "int f(long *p, long e, long d) { return __sync_bool_compare_and_swap(p, e, d); }\n";
3172 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
3173
3174 for order in ["0", "2", "3", "4", "5"] {
3178 let call = format!("__atomic_compare_exchange_n(p, e, d, 0, {order}, 0)");
3179 let source = format!("int f(int *p, int *e, int d) {{ return {call}; }}\n");
3180 let text = asm(&source);
3181 assert!(text.contains("cmpxchgl\t"), "{order}: {text}");
3182 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
3183 }
3184 }
3185
3186 #[test]
3198 fn a_read_modify_write_is_one_instruction_and_the_arithmetic_a_name_asks_for() {
3199 let text = body("int f(int *p, int v) { return __atomic_fetch_add(p, v, 5); }\n");
3200 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
3201 assert!(text.contains("return %2"), "the value that was there: {text}");
3202
3203 let text = body("int f(int *p, int v) { return __atomic_add_fetch(p, v, 5); }\n");
3204 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
3205 assert!(text.contains("%3 = add %2, %1"), "and the value afterwards: {text}");
3206
3207 let text = body("int f(int *p, int v) { return __atomic_sub_fetch(p, v, 5); }\n");
3208 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
3209 assert!(text.contains("%3 = sub %2, %1"), "{text}");
3210
3211 let text = body("int f(int *p, int v) { return __sync_fetch_and_sub(p, v); }\n");
3213 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
3214
3215 let text = body("int f(int *p, int v) { return __atomic_exchange_n(p, v, 5); }\n");
3218 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, seq_cst"), "{text}");
3219
3220 let text = body("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
3221 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, acquire"), "{text}");
3222
3223 let text = body("void f(int *p) { __sync_lock_release(p); }\n");
3226 assert!(text.contains("release"), "{text}");
3227 assert!(text.contains("%1 = iconst.i32 0"), "{text}");
3228
3229 let text = body("void f(int *p, int guard) { __sync_lock_release(p, guard); }\n");
3233 assert!(text.contains("%2 = iconst.i32 0"), "{text}");
3234 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
3235
3236 let text = body("int f(int *p, int v) { return __atomic_fetch_and(p, v, 5); }\n");
3239 assert!(text.contains("%2 = atomic_rmw.i32 and %0, %1, align 4, seq_cst"), "{text}");
3240
3241 let text = body("int f(int *p, int v) { return __sync_or_and_fetch(p, v); }\n");
3242 assert!(text.contains("%2 = atomic_rmw.i32 or %0, %1, align 4, seq_cst"), "{text}");
3243 assert!(text.contains("%3 = or %2, %1"), "and the value afterwards: {text}");
3244
3245 let text = body("int f(int *p, int v) { return __atomic_nand_fetch(p, v, 5); }\n");
3248 assert!(text.contains("%2 = atomic_rmw.i32 nand %0, %1, align 4, seq_cst"), "{text}");
3249 assert!(text.contains("%3 = and %2, %1"), "{text}");
3250 assert!(text.contains("%4 = iconst.i32 -1"), "{text}");
3251 assert!(text.contains("%5 = xor %3, %4"), "{text}");
3252 }
3253
3254 #[test]
3265 fn a_bitwise_read_modify_write_is_a_loop_around_the_compare_and_exchange() {
3266 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
3267 for (ty, suffix, reg) in widths {
3268 for (name, call, insn) in [
3269 ("and", "__atomic_fetch_and(p, v, 5)", "and"),
3270 ("or", "__sync_fetch_and_or(p, v)", "or"),
3271 ("xor", "__atomic_xor_fetch(p, v, 5)", "xor"),
3272 ] {
3273 let source = format!("{ty} f({ty} *p, {ty} v) {{ return {call}; }}\n");
3274 let text = asm(&source);
3275 assert!(text.contains("\tlock\n"), "{ty} {name}: {text}");
3276 assert!(
3277 text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")),
3278 "{ty} {name}: {text}"
3279 );
3280 assert!(text.contains(&format!("{insn}{suffix}\t")), "{ty} {name}: {text}");
3281 assert!(!text.contains("\txadd"), "{ty} {name} is not an add: {text}");
3283 assert!(!text.contains("\txchg"), "{ty} {name} is not an exchange: {text}");
3284 }
3285 }
3286 let source = "long f(long *p, long v) { return __atomic_fetch_or(p, v, 5); }\n";
3287 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
3288
3289 let text = asm("int f(int *p, int v) { return __sync_fetch_and_nand(p, v); }\n");
3293 assert!(text.contains("cmpxchgl\t"), "{text}");
3294 assert!(text.contains("andl\t"), "{text}");
3295 assert!(text.contains("notl\t"), "{text}");
3296 }
3297
3298 #[test]
3307 fn an_access_through_a_second_pointer_is_the_same_access_and_one_more() {
3308 let text = body("void f(int *p, int *r) { __atomic_load(p, r, 5); }\n");
3309 assert!(text.contains("%2 = atomic_load.i32 %0, align 4, seq_cst"), "{text}");
3310 assert!(text.contains("store %2 -> %1, align 4"), "and out through the place: {text}");
3311
3312 let text = body("void f(int *p, int *v) { __atomic_store(p, v, 3); }\n");
3313 assert!(text.contains("%2 = load.i32 %1, align 4"), "in through the place: {text}");
3314 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
3315
3316 let text = body("void f(int *p, int *v, int *r) { __atomic_exchange(p, v, r, 5); }\n");
3319 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3320 assert!(text.contains("%4 = atomic_rmw.i32 xchg %0, %3, align 4, seq_cst"), "{text}");
3321 assert!(text.contains("store %4 -> %2, align 4"), "{text}");
3322 }
3323
3324 #[test]
3335 fn a_flag_is_an_exchange_of_one_byte_and_a_store_of_a_zero_over_the_same_byte() {
3336 for pointer in ["char", "int", "void"] {
3337 let source = format!("int f({pointer} *p) {{ return __atomic_test_and_set(p, 5); }}\n");
3338 let text = body(&source);
3339 assert!(text.contains("%1 = iconst.i8 1"), "{pointer}: {text}");
3340 assert!(
3341 text.contains("%2 = atomic_rmw.i8 xchg %0, %1, align 1, seq_cst"),
3342 "{pointer}: {text}"
3343 );
3344 assert!(text.contains("%4 = icmp ne %2, %3"), "{pointer}: {text}");
3345
3346 let source = format!("void f({pointer} *p) {{ __atomic_clear(p, 3); }}\n");
3347 let text = body(&source);
3348 assert!(text.contains("atomic_store %2 -> %0, align 1, release"), "{pointer}: {text}");
3349 }
3350
3351 let text = asm("int f(int *p) { return __atomic_test_and_set(p, 5); }\n");
3354 assert!(text.contains("xchgb\t%al, (%rdi)"), "{text}");
3355 assert!(text.contains("setne\t"), "{text}");
3356 }
3357
3358 #[test]
3366 fn a_read_modify_write_is_an_exchange_or_a_locked_add_at_the_width_of_the_object() {
3367 let widths = [("char", "b", "%sil"), ("short", "w", "%si"), ("int", "l", "%esi")];
3368 for (ty, suffix, reg) in widths {
3369 let source =
3370 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_fetch_add(p, v, 5); }}\n");
3371 let text = asm(&source);
3372 assert!(text.contains("\tlock\n"), "{ty}: {text}");
3373 assert!(text.contains(&format!("xadd{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3374
3375 let source =
3376 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_exchange_n(p, v, 5); }}\n");
3377 let text = asm(&source);
3378 assert!(text.contains(&format!("xchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3379 assert!(!text.contains("\tlock\n"), "an exchange is locked already: {ty}: {text}");
3380 }
3381 let source = "long f(long *p, long v) { return __atomic_fetch_add(p, v, 5); }\n";
3382 assert!(asm(source).contains("xaddq\t%rsi, (%rdi)"), "{}", asm(source));
3383
3384 let source = "int f(int *p, int v) { return __atomic_fetch_sub(p, v, 5); }\n";
3387 let text = asm(source);
3388 assert!(text.contains("negl\t"), "{text}");
3389 assert!(text.contains("xaddl\t"), "{text}");
3390
3391 for order in ["0", "2", "3", "4", "5"] {
3394 let source =
3395 format!("int f(int *p, int v) {{ return __atomic_fetch_add(p, v, {order}); }}\n");
3396 let text = asm(&source);
3397 assert!(text.contains("xaddl\t"), "{order}: {text}");
3398 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
3399 }
3400
3401 let text = asm("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
3405 assert!(text.contains("xchgl\t%esi, (%rdi)"), "{text}");
3406 let text = asm("void f(int *p) { __sync_lock_release(p); }\n");
3411 assert!(text.contains("movl\t$0, %eax"), "{text}");
3412 assert!(text.contains("movl\t%eax, (%rdi)"), "{text}");
3413 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
3414 }
3415
3416 #[test]
3428 fn the_lock_free_questions_are_answered_as_constants() {
3429 for size in ["1", "2", "4", "8"] {
3430 let source =
3431 format!("int f(void) {{ return __atomic_always_lock_free({size}, 0); }}\n");
3432 let text = asm(&source);
3433 assert!(text.contains("movb\t$1, %al"), "{size} bytes is lock free: {text}");
3434 assert!(!text.contains("call"), "and is not a call: {text}");
3435 }
3436 for size in ["3", "16", "sizeof(long double)"] {
3437 let source = format!("int f(void) {{ return __atomic_is_lock_free({size}, 0); }}\n");
3438 let text = asm(&source);
3439 assert!(text.contains("movb\t$0, %al"), "{size} bytes is not: {text}");
3440 assert!(!text.contains("call"), "and is not a call either: {text}");
3441 }
3442
3443 let text = asm("int f(int n) { return __atomic_is_lock_free(n, 0); }\n");
3447 assert!(text.contains("movb\t$0, %al"), "a size nobody knows is not lock free: {text}");
3448 let text = asm("int f(int *p) { return __atomic_always_lock_free(8, p); }\n");
3449 assert!(text.contains("movb\t$0, %al"), "eight bytes at four is not: {text}");
3450 let text = asm("int f(long *p) { return __atomic_always_lock_free(8, p); }\n");
3451 assert!(text.contains("movb\t$1, %al"), "and at eight it is: {text}");
3452 }
3453
3454 #[test]
3466 fn a_memory_order_an_operation_cannot_carry_is_read_as_the_strongest() {
3467 let mut opts = options();
3468 opts.emit = EmitKind::Ir;
3469
3470 let acquire_store = run(&opts, "void f(int *p, int v) { __atomic_store_n(p, v, 2); }\n");
3471 assert!(acquire_store.text().contains("seq_cst"), "{:?}", acquire_store.text());
3472 assert!(acquire_store.messages[0].contains("[W0333]"), "{:?}", acquire_store.messages);
3473
3474 let nonsense = run(&opts, "int f(int *p) { return __atomic_load_n(p, 99); }\n");
3475 assert!(nonsense.text().contains("seq_cst"), "{:?}", nonsense.text());
3476 assert!(nonsense.messages[0].contains("[W0333]"), "{:?}", nonsense.messages);
3477
3478 let computed = run(&opts, "int f(int *p, int n) { return __atomic_load_n(p, n); }\n");
3479 assert!(computed.text().contains("seq_cst"), "{:?}", computed.text());
3480 assert_eq!(computed.messages, Vec::<String>::new(), "a computed order is not a mistake");
3481 }
3482
3483 #[test]
3495 fn a_conversion_between_a_float_and_the_widest_unsigned_integer_is_written_without_a_branch() {
3496 let text = asm("double f(unsigned long long x) { return (double)x; }\n");
3497 assert!(text.contains("cvtsi2sdq"), "the signed conversion is what runs: {text}");
3498 assert!(text.contains("shrq"), "with the value halved first: {text}");
3499 assert!(text.contains("addsd"), "and doubled after: {text}");
3500 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
3501
3502 let text = asm("unsigned long long f(double d) { return (unsigned long long)d; }\n");
3503 assert!(text.contains("cvttsd2siq"), "the signed conversion is what runs: {text}");
3504 assert!(text.contains("subsd"), "with half the range taken off first: {text}");
3505 assert!(text.contains("shlq\t$63"), "and the top bit put back: {text}");
3506 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
3507 }
3508
3509 #[test]
3520 fn a_plain_name_the_program_took_is_the_programs_own_function() {
3521 let taken = concat!(
3522 "static long long llabs(long long b) { return 7; }\n",
3523 "long long f(long long x) { return llabs(x); }\n",
3524 );
3525 assert!(ir(taken).contains("call @llabs"), "a static definition is the program's own");
3526
3527 let retyped = concat!("int llabs(int b);\n", "int f(int x) { return llabs(x); }\n",);
3528 assert!(ir(retyped).contains("call @llabs"), "another type is another function");
3529
3530 let plain = concat!(
3531 "long long llabs(long long b);\n",
3532 "long long f(long long x) { return llabs(x); }\n",
3533 );
3534 let mut opts = options();
3535 opts.emit = EmitKind::Ir;
3536 assert!(!run(&opts, plain).text().contains("call @llabs"), "the library's by default");
3537
3538 opts.builtins = false;
3539 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin");
3540
3541 opts.builtins = true;
3542 opts.no_builtin = vec!["llabs".to_owned()];
3543 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin-llabs");
3544 let one = "long labs(long b);\nlong f(long x) { return labs(x); }\n";
3545 assert!(!run(&opts, one).text().contains("call @labs"), "one name and not the family");
3546
3547 opts.no_builtin = Vec::new();
3550 opts.builtins = false;
3551 let prefixed = "long long f(long long x) { return __builtin_llabs(x); }\n";
3552 assert!(!run(&opts, prefixed).text().contains("call @llabs"), "the prefix is a promise");
3553 }
3554
3555 #[test]
3568 fn the_hint_builtins_are_their_first_argument_and_the_hint_leaves_no_trace() {
3569 let text = ir(concat!(
3570 "long a = __builtin_expect(7, 1);\n",
3571 "long b = __builtin_expect_with_probability(9, 1, 0.9);\n",
3572 "unsigned long c = sizeof(__builtin_expect((char)1, 1));\n",
3573 ));
3574 assert!(text.contains("global @a : i64 = 7,"), "{text}");
3575 assert!(text.contains("global @b : i64 = 9,"), "{text}");
3576 assert!(text.contains("global @c : i64 = 8,"), "{text}");
3577 assert!(!text.contains("__builtin_expect"), "it is not a call to anything:\n{text}");
3578
3579 let text = body("long f(char c) { return __builtin_expect(c, 1); }\n");
3582 assert!(text.contains("sext"), "{text}");
3583
3584 let one = "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 1\n %2 = sext.i64 %1\n return %0\n";
3588 assert_eq!(body("int f(void) { int i = 0; __builtin_expect(1, i++); return i; }\n"), one);
3589 let source = "int g(void) { int i = 0; __builtin_expect_with_probability(1, i++, 0.5); return i; }\n";
3590 assert_eq!(body(source), one);
3591
3592 let kept = body("int f(int n) { int i = 0; __builtin_expect(n, i++); return i; }\n");
3597 assert!(kept.contains("add.nsw"), "the hint still runs: {kept}");
3598 assert!(kept.ends_with("return %3\n"), "and the answer is what it left behind: {kept}");
3599 let both = "int g(int n) { int i = 0; __builtin_expect_with_probability(n, i++, 0.5); return i; }\n";
3600 assert!(body(both).contains("add.nsw"), "and so does the one with three arguments");
3601 }
3602
3603 #[test]
3615 fn a_promise_that_control_does_not_arrive_writes_no_instruction() {
3616 let promised = "int f(int x) { if (x) return 1; __builtin_unreachable(); }\n";
3617 let text = ir(promised);
3618 assert!(text.contains(" unreachable_hint\n"), "{text}");
3619 assert!(!text.contains("call"), "it is not a call to anything:\n{text}");
3620
3621 let after = body("int g(int x) { __builtin_unreachable(); return x; }\n");
3625 assert!(after.contains("return"), "{after}");
3626
3627 let text = asm(promised);
3630 let mine = text.split_once("\nf:\n").expect("a definition").1;
3631 let mine = mine.split_once("\t.size").expect("a definition").0;
3632 let plain = asm("int f(int x) { if (x) return 1; }\n");
3633 let plain = plain.split_once("\nf:\n").expect("a definition").1;
3634 let plain = plain.split_once("\t.size").expect("a definition").0;
3635 assert_eq!(mine, plain);
3636 let last = mine.lines().rfind(|line| !line.trim_start().starts_with('.'));
3639 assert_eq!(last.map(str::trim), Some("ret"), "{mine}");
3640 assert!(!mine.contains("ud2"), "{mine}");
3641 }
3642
3643 #[test]
3650 fn a_library_builtin_is_diagnosed_under_the_name_the_program_wrote() {
3651 let mut opts = options();
3652 opts.emit = EmitKind::Ir;
3653 let messages = run(&opts, "void f(void) { __builtin_abort(1); }\n").messages;
3654 assert!(
3655 messages.iter().any(|m| m.contains("__builtin_abort")),
3656 "expected the written name in {messages:?}"
3657 );
3658 }
3659
3660 #[test]
3668 fn a_builtin_nothing_lowers_is_refused_by_name() {
3669 let mut opts = options();
3670 opts.emit = EmitKind::Ir;
3671 for (builtin, call) in [
3672 ("__builtin_return_address", "(int)(long)__builtin_return_address(0)"),
3673 ("__builtin_alloca", "(int)(long)__builtin_alloca(8)"),
3674 ("__atomic_signal_fence", "(__atomic_signal_fence(5), 0)"),
3675 ] {
3676 let source = format!("int counter;\nint f(void) {{ return {call}; }}\n");
3677 let messages = run(&opts, &source).messages;
3678 let named = messages.iter().any(|m| m.contains(builtin) && m.contains("E0686"));
3679 assert!(named, "expected {builtin} to be refused by name in {messages:?}");
3680 }
3681 }
3682
3683 #[test]
3691 fn what_is_refused_is_the_call_and_not_the_name() {
3692 let text = ir("unsigned long n = sizeof(__builtin_return_address(0));\n");
3693 assert!(text.contains("global @n : i64 = 8,"), "{text}");
3694
3695 let text = ir(concat!(
3696 "void *__builtin_return_address(unsigned x) { return 0; }\n",
3697 "void *f(void) { return __builtin_return_address(0); }\n",
3698 ));
3699 assert!(text.contains("call @__builtin_return_address"), "{text}");
3700 }
3701
3702 #[test]
3707 fn a_static_function_nothing_refers_to_is_not_emitted() {
3708 let text = ir("static int dropped(void) { return 1; }\n\
3709 static int kept(void) { return 2; }\n\
3710 int main(void) { return kept(); }\n");
3711 assert!(text.contains("func @kept"), "{text}");
3712 assert!(!text.contains("dropped"), "{text}");
3713 }
3714
3715 #[test]
3721 fn two_static_functions_that_only_call_each_other_are_both_dropped() {
3722 let text = ir("static int ping(void);\n\
3723 static int pong(void) { return ping(); }\n\
3724 static int ping(void) { return pong(); }\n\
3725 int main(void) { return 0; }\n");
3726 assert!(!text.contains("ping"), "{text}");
3727 assert!(!text.contains("pong"), "{text}");
3728 }
3729
3730 #[test]
3736 fn naming_a_static_function_anywhere_keeps_it() {
3737 let text = ir("static int by_address(void) { return 1; }\n\
3738 static int in_an_image(void) { return 2; }\n\
3739 static int deeper(void) { return 3; }\n\
3740 static int reaches_deeper(void) { return deeper(); }\n\
3741 static int (*table[1])(void) = {in_an_image};\n\
3742 int main(void) {\n\
3743 int (*p)(void) = by_address;\n\
3744 return p() + table[0]() + reaches_deeper();\n\
3745 }\n");
3746 for kept in ["by_address", "in_an_image", "deeper", "reaches_deeper"] {
3747 assert!(text.contains(&format!("func @{kept}")), "expected {kept} in:\n{text}");
3748 }
3749 }
3750
3751 #[test]
3757 fn an_attribute_keeps_a_static_function_nothing_refers_to() {
3758 for attribute in ["used", "retain", "constructor", "destructor", "__used__"] {
3759 let source = format!(
3760 "__attribute__(({attribute})) static int kept(void) {{ return 1; }}\n\
3761 int main(void) {{ return 0; }}\n"
3762 );
3763 let text = ir(&source);
3764 assert!(text.contains("func @kept"), "for {attribute}:\n{text}");
3765 }
3766 }
3767
3768 #[test]
3771 fn a_function_anything_could_call_is_emitted_without_being_called() {
3772 let text =
3773 ir("int nobody_here_calls_it(void) { return 1; }\nint main(void) { return 0; }\n");
3774 assert!(text.contains("func @nobody_here_calls_it"), "{text}");
3775 }
3776
3777 #[test]
3784 fn a_classification_c_has_an_operator_for_is_that_operator() {
3785 for (builtin, operator) in [
3786 ("__builtin_isgreater", "binary >"),
3787 ("__builtin_isgreaterequal", "binary >="),
3788 ("__builtin_isless", "binary <"),
3789 ("__builtin_islessequal", "binary <="),
3790 ] {
3791 let source = format!("int f(double x, double y) {{ return {builtin}(x, y); }}\n");
3792 let text = tast(&source);
3793 assert!(text.contains(&format!("{operator} : int")), "for {builtin}:\n{text}");
3794 }
3795 }
3796
3797 #[test]
3806 fn the_classification_builtins_are_comparisons_and_not_calls() {
3807 let text = body("int f(double x, double y) { return __builtin_isunordered(x, y); }\n");
3808 assert_eq!(
3809 text,
3810 "block0(%0: f64, %1: f64):\n %2 = fcmp uno %0, %1\n %3 = zext.i32 \
3811 %2\n return %3\n"
3812 );
3813
3814 let text = body("int f(double x, double y) { return __builtin_islessgreater(x, y); }\n");
3816 assert!(text.contains("fcmp one %0, %1"), "{text}");
3817
3818 let text = body("int f(double x) { return __builtin_isnan(x); }\n");
3819 assert!(text.contains("fcmp uno %0, %0"), "{text}");
3820
3821 let text = body("int f(double x) { return __builtin_isinf(x); }\n");
3822 assert!(text.contains("fconst.f64 0x7ff0000000000000"), "{text}");
3823 assert!(text.contains("fconst.f64 0xfff0000000000000"), "{text}");
3824 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
3825 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
3826 assert!(text.contains("%5 = or %3, %4"), "{text}");
3827
3828 let text = body("int f(double x) { return __builtin_isfinite(x); }\n");
3831 assert!(text.contains("%3 = fcmp olt %2, %0"), "{text}");
3832 assert!(text.contains("%4 = fcmp olt %0, %1"), "{text}");
3833 assert!(text.contains("%5 = and %3, %4"), "{text}");
3834
3835 let text = body("int f(double x) { return __builtin_signbit(x); }\n");
3836 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
3837 assert!(text.contains("icmp slt %1, %2"), "{text}");
3838
3839 let text = body("int f(long double x) { return __builtin_signbitl(x); }\n");
3842 assert!(text.contains("%1 = bitcast.i80 %0"), "{text}");
3843
3844 let text = body("double g(void);\nint f(void) { return __builtin_isnan(g()); }\n");
3847 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
3848 }
3849
3850 #[test]
3857 fn a_classification_spelling_that_names_a_width_converts_before_it_asks() {
3858 let text = ir(concat!(
3859 "int a = __builtin_isinff(1e300);\n",
3860 "int b = __builtin_isinf(1e300);\n",
3861 "int c = __builtin_isnan(0.0);\n",
3865 "int d = __builtin_signbit(-0.0);\n",
3866 "int e = __builtin_islessgreater(1.0, 2.0);\n",
3867 ));
3868 assert!(text.contains("global @a : i32 = 1,"), "{text}");
3869 assert!(text.contains("global @b : i32 = 0,"), "{text}");
3870 assert!(text.contains("global @c : i32 = 0,"), "{text}");
3871 assert!(text.contains("global @d : i32 = 1,"), "{text}");
3872 assert!(text.contains("global @e : i32 = 1,"), "{text}");
3873 }
3874
3875 #[test]
3877 fn a_classification_builtin_refuses_an_argument_that_is_not_floating_point() {
3878 let mut opts = options();
3879 opts.emit = EmitKind::Ir;
3880 let source = concat!(
3881 "int a(int x) { return __builtin_isnan(x); }\n",
3882 "int b(int x, int y) { return __builtin_isunordered(x, y); }\n",
3883 "int c(double x) { return __builtin_isnan(x, x); }\n",
3884 );
3885 let messages = run(&opts, source).messages;
3886 assert_eq!(
3887 messages,
3888 [
3889 "/main.c:1:23: error: non-floating-point argument in call to function \
3890 '__builtin_isnan' [E0685]",
3891 "/main.c:2:30: error: non-floating-point arguments in call to function \
3892 '__builtin_isunordered' [E0685]",
3893 "/main.c:3:26: error: too many arguments to function '__builtin_isnan' [E0511]",
3894 ]
3895 );
3896 }
3897
3898 #[test]
3907 fn the_last_three_classification_builtins_are_comparisons_and_not_calls() {
3908 let text = body("int f(double x) { return __builtin_isnormal(x); }\n");
3909 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
3913 assert!(text.contains("%2 = iconst.i64 9223372036854775807"), "{text}");
3914 assert!(text.contains("%3 = and %1, %2"), "{text}");
3915 assert!(text.contains("%4 = iconst.i64 4503599627370496"), "{text}");
3916 assert!(text.contains("%5 = iconst.i64 9218868437227405312"), "{text}");
3917 assert!(text.contains("%6 = icmp uge %3, %4"), "{text}");
3918 assert!(text.contains("%7 = icmp ult %3, %5"), "{text}");
3919 assert!(text.contains("%8 = and %6, %7"), "{text}");
3920
3921 let text = body("int f(long double x) { return __builtin_isnormal(x); }\n");
3925 assert!(text.contains("%4 = iconst.i80 27670116110564327424"), "{text}");
3926 assert!(text.contains("%5 = iconst.i80 604453686435277732577280"), "{text}");
3927
3928 let text = body("int f(double x) { return __builtin_isinf_sign(x); }\n");
3929 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
3930 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
3931 assert!(text.contains("%7 = sub %5, %6"), "{text}");
3932
3933 let text = body("int f(double x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n");
3934 assert!(text.contains("fcmp uno %0, %0"), "{text}");
3935 assert!(text.contains("fcmp oeq %0, %6"), "{text}");
3936 assert_eq!(text.matches(" = zext.i32 ").count(), 4, "{text}");
3940 assert_eq!(text.matches(" = xor ").count(), 4, "{text}");
3941 assert!(!text.contains("call"), "{text}");
3942
3943 let text = body(concat!(
3946 "double g(void);\n",
3947 "int f(void) { return __builtin_fpclassify(0, 1, 2, 3, 4, g()); }\n",
3948 ));
3949 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
3950 }
3951
3952 #[test]
3959 fn the_last_three_classification_builtins_fold_where_their_operand_is_a_constant() {
3960 let text = ir(concat!(
3961 "int a = __builtin_isnormal(1.0);\n",
3962 "int b = __builtin_isnormal(0.0);\n",
3963 "int c = __builtin_isnormal(1.0 / 0.0);\n",
3964 "int d = __builtin_isinf_sign(-1.0 / 0.0);\n",
3965 "int e = __builtin_isinf_sign(1.0);\n",
3966 "int g = __builtin_fpclassify(0, 1, 2, 3, 4, 0.0);\n",
3967 "int h = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0);\n",
3968 "int i = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0 / 0.0);\n",
3969 ));
3970 assert!(text.contains("global @a : i32 = 1,"), "{text}");
3971 assert!(text.contains("global @b : i32 = 0,"), "{text}");
3972 assert!(text.contains("global @c : i32 = 0,"), "{text}");
3973 assert!(text.contains("global @d : i32 = -1,"), "{text}");
3974 assert!(text.contains("global @e : i32 = 0,"), "{text}");
3975 assert!(text.contains("global @g : i32 = 4,"), "{text}");
3976 assert!(text.contains("global @h : i32 = 2,"), "{text}");
3977 assert!(text.contains("global @i : i32 = 1,"), "{text}");
3978 }
3979
3980 #[test]
3986 fn fpclassify_refuses_an_answer_that_is_not_an_integer_constant() {
3987 let mut opts = options();
3988 opts.emit = EmitKind::Ir;
3989 let source = concat!(
3990 "int a(double x, int n) { return __builtin_fpclassify(0, 1, n, 3, 4, x); }\n",
3991 "int b(double x) { return __builtin_fpclassify(0, 1, 2, 3, x); }\n",
3992 "int c(int x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n",
3993 );
3994 let messages = run(&opts, source).messages;
3995 assert_eq!(
3996 messages,
3997 [
3998 "/main.c:1:60: error: non-const integer argument 3 in call to function \
3999 '__builtin_fpclassify' [E0687]",
4000 "/main.c:2:26: error: too few arguments to function '__builtin_fpclassify' \
4001 [E0511]",
4002 "/main.c:3:23: error: non-floating-point argument in call to function \
4003 '__builtin_fpclassify' [E0685]",
4004 ]
4005 );
4006 }
4007
4008 #[test]
4016 fn a_builtin_whose_answer_is_a_constant_is_one_and_not_a_call() {
4017 let text = ir(concat!(
4018 "double a = __builtin_inf();\n",
4019 "float b = __builtin_huge_valf();\n",
4020 "long double c = __builtin_infl();\n",
4021 "double d = __builtin_huge_val();\n",
4022 ));
4023 assert!(text.contains("global @a : f64 = 0x7ff0000000000000,"), "{text}");
4024 assert!(text.contains("global @b : f32 = 0x7f800000,"), "{text}");
4025 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
4026 assert!(text.contains("global @d : f64 = 0x7ff0000000000000,"), "{text}");
4027 assert!(!text.contains("call"), "{text}");
4028 }
4029
4030 #[test]
4039 fn a_nan_is_written_with_the_payload_the_program_asked_for() {
4040 let text = ir(concat!(
4041 "double a = __builtin_nan(\"\");\n",
4042 "double b = __builtin_nan(\"0x1\");\n",
4043 "double c = __builtin_nan(\"010\");\n",
4045 "double d = __builtin_nans(\"\");\n",
4046 "double e = __builtin_nans(\"0x1\");\n",
4047 "float f = __builtin_nanf(\"0x1\");\n",
4048 "float g = __builtin_nansf(\"\");\n",
4049 "long double h = __builtin_nansl(\"\");\n",
4050 ));
4051 assert!(text.contains("global @a : f64 = 0x7ff8000000000000,"), "{text}");
4052 assert!(text.contains("global @b : f64 = 0x7ff8000000000001,"), "{text}");
4053 assert!(text.contains("global @c : f64 = 0x7ff8000000000008,"), "{text}");
4054 assert!(text.contains("global @d : f64 = 0x7ff4000000000000,"), "{text}");
4055 assert!(text.contains("global @e : f64 = 0x7ff0000000000001,"), "{text}");
4056 assert!(text.contains("global @f : f32 = 0x7fc00001,"), "{text}");
4057 assert!(text.contains("global @g : f32 = 0x7fa00000,"), "{text}");
4058 assert!(text.contains("f80 0x7fffa000000000000000"), "{text}");
4059
4060 let text = ir(concat!(
4063 "double f(const char *p) { return __builtin_nan(p); }\n",
4064 "double g(void) { return __builtin_nans(\"1x\"); }\n",
4065 ));
4066 assert_eq!(text.matches("call @nan(").count(), 1, "{text}");
4067 assert_eq!(text.matches("call @nans(").count(), 1, "{text}");
4068 }
4069
4070 #[test]
4078 fn the_length_and_the_order_of_a_string_literal_are_known_here() {
4079 let text = ir(concat!(
4080 "unsigned long a = __builtin_strlen(\"hello\");\n",
4081 "unsigned long b = __builtin_strlen(\"a\\0bc\");\n",
4082 "int c = __builtin_strcmp(\"X\", \"X\\376\") < 0;\n",
4083 "int d = __builtin_strcmp(\"abc\", \"abc\");\n",
4084 "int e = __builtin_strcmp(\"abc\", \"ab\") > 0;\n",
4085 ));
4086 assert!(text.contains("global @a : i64 = 5,"), "{text}");
4087 assert!(text.contains("global @b : i64 = 1,"), "{text}");
4088 assert!(text.contains("global @c : i32 = 1,"), "{text}");
4089 assert!(text.contains("global @d : i32 = 0,"), "{text}");
4090 assert!(text.contains("global @e : i32 = 1,"), "{text}");
4091 assert!(!text.contains("call"), "{text}");
4092
4093 let text = ir("unsigned long f(const char *p) { return __builtin_strlen(p); }\n");
4095 assert!(text.contains("call @strlen("), "{text}");
4096 }
4097
4098 #[test]
4105 fn a_sign_builtin_is_a_mask_over_the_bits_and_not_a_call() {
4106 let text = body("double f(double x) { return __builtin_fabs(x); }\n");
4107 assert!(text.contains("bitcast.i64 %0"), "{text}");
4108 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
4109 assert!(text.contains("and %1, %2"), "{text}");
4110 assert!(text.contains("bitcast.f64 %3"), "{text}");
4111 assert!(!text.contains("call"), "{text}");
4112
4113 let text = body("double f(double x, double y) { return __builtin_copysign(x, y); }\n");
4114 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
4115 assert!(text.contains("%8 = or %4, %7"), "{text}");
4116 assert!(!text.contains("call"), "{text}");
4117
4118 let text = body("long double f(long double x) { return __builtin_fabsl(x); }\n");
4121 assert!(text.contains("bitcast.i80 %0"), "{text}");
4122 assert!(text.contains("bitcast.f80"), "{text}");
4123
4124 let text = body("double f(float x) { return __builtin_fabs(x); }\n");
4127 assert!(text.contains("fpext.f64 %0"), "{text}");
4128 assert!(text.contains("bitcast.i64 %1"), "{text}");
4129 }
4130
4131 #[test]
4140 fn the_sign_builtins_answer_a_zero_and_a_nan_the_way_the_bits_say() {
4141 let text = ir(concat!(
4142 "double a = __builtin_fabs(-3.5);\n",
4143 "double b = __builtin_copysign(1.0, -0.0);\n",
4144 "double c = __builtin_copysign(0.0, -2.0);\n",
4145 "double d = __builtin_copysign(-__builtin_nan(\"\"), 1.0);\n",
4147 "double e = __builtin_fabs(-__builtin_nan(\"0x1\"));\n",
4148 "float g = __builtin_copysignf(-0.0f, 2.0f);\n",
4149 "long double h = __builtin_copysignl(1.0L, -1.0L);\n",
4150 "long double i = __builtin_fabsl(-__builtin_infl());\n",
4151 ));
4152 assert!(text.contains("global @a : f64 = 0x400c000000000000,"), "{text}");
4153 assert!(text.contains("global @b : f64 = 0xbff0000000000000,"), "{text}");
4154 assert!(text.contains("global @c : f64 = 0x8000000000000000,"), "{text}");
4155 assert!(text.contains("global @d : f64 = 0x7ff8000000000000,"), "{text}");
4156 assert!(text.contains("global @e : f64 = 0x7ff8000000000001,"), "{text}");
4157 assert!(text.contains("global @g : f32 = 0x0,"), "{text}");
4158 assert!(text.contains("f80 0xbfff8000000000000000"), "{text}");
4159 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
4160 }
4161
4162 #[test]
4169 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
4170 let text = ir(concat!(
4171 "constexpr int side = 4;\n",
4172 "constexpr int wider = side + 1;\n",
4173 "constexpr double half = 1.5;\n",
4174 "struct point { int x; int y; };\n",
4175 "constexpr struct point origin = { 5, 6 };\n",
4176 "int square[side * side];\n",
4177 "int rectangle[wider];\n",
4178 "int rounded[(int)half * 2];\n",
4179 "int across[origin.y];\n",
4180 "enum named { four = side };\n",
4181 "int e = four;\n",
4182 ));
4183 assert!(text.contains("global @square : bytes 64 ="), "{text}");
4184 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
4185 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
4186 assert!(text.contains("global @across : bytes 24 ="), "{text}");
4187 assert!(text.contains("global @e : i32 = 4,"), "{text}");
4188
4189 let mut opts = options();
4192 opts.emit = EmitKind::Ir;
4193 let konst = "const int n = 1;\nint a[n];\n";
4194 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
4195 assert_eq!(run(&opts, konst).messages, [message]);
4196
4197 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
4199 assert_eq!(run(&opts, subscript).messages, [message]);
4200
4201 let address = "constexpr int c = 3;\nint *p = &c;\n";
4203 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
4204 pointer target type [E0514]";
4205 assert_eq!(run(&opts, address).messages, [warning]);
4206 }
4207
4208 #[test]
4217 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
4218 let mut opts = options();
4221 opts.std = Std::C17;
4222 let source = concat!(
4223 "int add(a, b)\n",
4224 "int a;\n",
4225 "int b;\n",
4226 "{ return a + b; }\n",
4227 "int promoted(c)\n",
4228 "char c;\n",
4229 "{ return c; }\n",
4230 "int narrow(char);\n",
4231 "int narrow(c)\n",
4232 "char c;\n",
4233 "{ return c; }\n",
4234 "int first(a)\n",
4235 "int a[4];\n",
4236 "{ return a[0]; }\n",
4237 );
4238 let result = run(&opts, source);
4239 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
4240 let text = result.text();
4241 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
4242 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
4243 assert!(text.contains("c : char object automatic defined"), "{text}");
4245 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
4246 assert!(text.contains("first : int(int *) function external defined"), "{text}");
4248 }
4249
4250 #[test]
4257 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
4258 let mut opts = options();
4259 opts.std = Std::C17;
4260 for (source, message) in [
4261 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
4262 (
4263 "int f(a)\nint a;\nint b;\n{ return a; }\n",
4264 "3:5: error: declaration for parameter 'b' but no such parameter",
4265 ),
4266 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
4267 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
4268 (
4269 "int f(a)\nstatic int a;\n{ return a; }\n",
4270 "2:12: error: storage class specified for parameter 'a'",
4271 ),
4272 (
4273 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
4274 "2:7: error: argument 'a' doesn't match prototype",
4275 ),
4276 ] {
4277 let result = run(&opts, source);
4278 assert!(result.failed(), "expected this to fail:\n{source}");
4279 assert!(result.messages[0].contains(message), "{:?}", result.messages);
4280 }
4281
4282 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
4285 let mut older = options();
4286 older.std = Std::C89;
4287 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
4288 let result = run(&opts, implicit);
4289 assert!(
4290 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
4291 "{:?}",
4292 result.messages
4293 );
4294
4295 let mut newer = options();
4299 newer.std = Std::C23;
4300 let plain = "int f(a)\nint a;\n{ return a; }\n";
4301 let result = run(&newer, plain);
4302 assert!(!result.failed(), "{:?}", result.messages);
4303 assert_eq!(
4304 result.messages,
4305 ["/main.c:1:5: warning: old-style function definition [E0412]"]
4306 );
4307 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
4308 }
4309
4310 #[test]
4317 fn the_obsolete_designators_are_taken_and_are_pedantic_warnings() {
4318 let array = "int a[8] = { [3] 7 };\n";
4319 let member = "struct s { int x; } v = { x: 7 };\n";
4320 for source in [array, member] {
4321 let result = run(&options(), source);
4322 assert!(!result.failed(), "{:?}", result.messages);
4323 assert!(result.messages.is_empty(), "nothing to say: {:?}", result.messages);
4324 }
4325
4326 let mut asked = options();
4327 asked.pedantic = true;
4328 assert_eq!(
4329 run(&asked, array).messages,
4330 ["/main.c:1:18: warning: obsolete designator, write `[i] =` instead [E0415]"]
4331 );
4332 assert_eq!(
4333 run(&asked, member).messages,
4334 ["/main.c:1:27: warning: obsolete designator, write `.field =` instead [E0413]"]
4335 );
4336 }
4337
4338 #[test]
4345 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
4346 let text = ir(concat!(
4347 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
4348 "struct brim { char buf[9223372036854775807L]; };\n",
4349 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
4350 "unsigned long h = sizeof(struct huge_struct);\n",
4351 "unsigned long b = sizeof(struct brim);\n",
4352 "unsigned long y = sizeof(struct bitty);\n",
4353 ));
4354 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
4355 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
4356 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
4357
4358 let mut opts = options();
4359 opts.emit = EmitKind::Ir;
4360 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
4361 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
4362 assert_eq!(run(&opts, over).messages, [message]);
4363 let array = "struct wide { short buf[1L << 62]; };\n";
4364 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
4365 maximum object size '9223372036854775807' [E0537]";
4366 assert_eq!(run(&opts, array).messages[0], message);
4367 }
4368
4369 fn compile_bytes(source: &[u8]) -> Compiled {
4374 let mut opts = options();
4375 opts.emit = EmitKind::Ir;
4376 let mut fs = MemoryFileSystem::new();
4377 fs.insert("/main.c", source.to_vec());
4378 compile(&opts, "/main.c", &fs)
4379 }
4380
4381 #[test]
4388 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
4389 let mut source = b"char s[] = \"a".to_vec();
4390 source.push(0xff);
4391 source.extend_from_slice(b"b\";\nchar c = '");
4392 source.push(0xff);
4393 source.extend_from_slice(b"';\n");
4394 let result = compile_bytes(&source);
4395 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
4396 assert!(result.text().contains(r#"bytes "a\ffb\00""#), "{}", result.text());
4397 assert!(result.text().contains("global @c : i8 = -1,"), "{}", result.text());
4399
4400 let mut stray = b"int a".to_vec();
4401 stray.push(0xff);
4402 stray.extend_from_slice(b" = 1;\n");
4403 let result = compile_bytes(&stray);
4404 assert!(
4405 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
4406 "{:?}",
4407 result.messages
4408 );
4409 }
4410
4411 #[test]
4412 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
4413 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
4414 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
4415 let expected = "\
4416func @add(i32, i32) -> i32, linkage(external) {
4417block0(%0: i32, %1: i32):
4418 %2 = add.nsw %0, %1
4419 return %2
4420}
4421";
4422 assert!(text.contains(expected), "{text}");
4423 }
4424
4425 #[test]
4426 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
4427 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
4428 assert!(!text.contains("alloca"), "{text}");
4429 assert!(!text.contains("load"), "{text}");
4430 assert!(!text.contains("store"), "{text}");
4431 }
4432
4433 #[test]
4434 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
4435 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
4436 let expected = "\
4437block0:
4438 %0 = alloca, size 4, align 4
4439 %1 = iconst.i32 1
4440 store %1 -> %0, align 4
4441 %2 = call @g(%0) : (ptr) -> i32
4442 return %2
4443";
4444 assert_eq!(text, expected);
4445 }
4446
4447 #[test]
4448 fn a_loop_carries_what_it_changes_as_block_parameters() {
4449 let text = body(
4452 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
4453 return total;\n}\n",
4454 );
4455 assert!(!text.contains("alloca"), "{text}");
4456 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
4457 assert!(text.contains("jump block1("), "{text}");
4458 }
4459
4460 #[test]
4461 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
4462 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
4463 assert!(text.contains("icmp slt %0, %1"), "{text}");
4464 assert!(!text.contains("zext"), "{text}");
4465 }
4466
4467 #[test]
4468 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
4469 let text = body("int f(int a, int b) { return a && b; }\n");
4470 let expected = "\
4471block0(%0: i32, %1: i32):
4472 %2 = iconst.i32 0
4473 %3 = icmp ne %0, %2
4474 %4 = iconst.i1 0
4475 br_if %3, block1, block2(%4)
4476
4477block1:
4478 %5 = iconst.i32 0
4479 %6 = icmp ne %1, %5
4480 jump block2(%6)
4481
4482block2(%7: i1):
4483 %8 = zext.i32 %7
4484 return %8
4485";
4486 assert_eq!(text, expected);
4487 }
4488
4489 #[test]
4490 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
4491 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
4492 assert!(!text.contains("block3"), "{text}");
4495 assert!(!text.contains("iconst.i32 3"), "{text}");
4496 }
4497
4498 #[test]
4499 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
4500 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
4501 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
4502 assert!(body("int f(void) { }\n").contains("unreachable"));
4503 }
4504
4505 #[test]
4506 fn a_structure_is_copied_rather_than_held_in_a_value() {
4507 let text = body(
4508 "struct point { int x, y; };\n\
4509 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
4510 );
4511 assert!(text.contains("memcpy"), "{text}");
4512 }
4513
4514 #[test]
4515 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
4516 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
4517 assert!(text.contains("memset"), "{text}");
4518 }
4519
4520 #[test]
4521 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
4522 let text = body(
4523 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
4524 default: r = 4; } return r; }\n",
4525 );
4526 let expected = "\
4527block0(%0: i32):
4528 %1 = iconst.i32 0
4529 switch %0, block1, [1 => block2, 2 => block3(%1)]
4530
4531block1:
4532 %2 = iconst.i32 4
4533 jump block4(%2)
4534
4535block2:
4536 %3 = iconst.i32 1
4537 jump block3(%3)
4538
4539block3(%4: i32):
4540 %5 = iconst.i32 2
4541 %6 = add.nsw %4, %5
4542 jump block4(%6)
4543
4544block4(%7: i32):
4545 return %7
4546";
4547 assert_eq!(text, expected);
4548 }
4549
4550 #[test]
4551 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
4552 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
4555 assert!(text.contains("%2 = sub %0, %1"), "{text}");
4556 assert!(text.contains("icmp ule"), "{text}");
4557 assert!(!text.contains("switch"), "{text}");
4558 }
4559
4560 #[test]
4561 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
4562 let text = body(
4563 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
4564 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
4565 );
4566 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
4569 assert!(text.contains("block5:\n jump block7("), "{text}");
4570 assert!(text.contains("block6:\n jump block8("), "{text}");
4571 }
4572
4573 #[test]
4574 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
4575 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
4576 }
4577
4578 #[test]
4579 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
4580 let text = body(
4585 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
4586 return n; }\n",
4587 );
4588 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
4591 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
4592 assert!(text.contains("block4:\n jump block3("), "{text}");
4593 }
4594
4595 #[test]
4596 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
4597 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
4600 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
4601 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
4602 assert!(text.contains("br_if %6, block2, block3"), "{text}");
4603 }
4604
4605 #[test]
4606 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
4607 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
4608 assert!(!text.contains("alloca"), "{text}");
4612 assert!(text.contains("block2(%4: i32):\n return %4"), "{text}");
4613 assert_eq!(text.matches("jump block2(").count(), 2, "{text}");
4614 }
4615
4616 #[test]
4617 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
4618 let text =
4619 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
4620 assert!(!text.contains("alloca"), "{text}");
4621 assert!(text.contains("block1(%2: i32):"), "{text}");
4622 assert!(text.contains("jump block1(%5)"), "{text}");
4623 }
4624
4625 #[test]
4626 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
4627 assert_eq!(
4630 body("int f(int x) { return x; spare: return 0; }\n"),
4631 "block0(%0: i32):\n return %0\n"
4632 );
4633 }
4634
4635 #[test]
4636 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
4637 let text = body(
4638 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
4639 );
4640 assert_eq!(
4643 text,
4644 "\
4645block0(%0: ptr):
4646 %1 = load.i8 %0, align 1
4647 %2 = iconst.i8 3
4648 %3 = ashr %1, %2
4649 %4 = sext.i32 %3
4650 return %4
4651"
4652 );
4653 }
4654
4655 #[test]
4656 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
4657 let text =
4661 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
4662 assert_eq!(
4663 text,
4664 "\
4665block0(%0: ptr, %1: i32):
4666 %2 = iconst.i32 16777215
4667 %3 = and %1, %2
4668 %4 = trunc.i16 %3
4669 store %4 -> %0, align 2
4670 %5 = iconst.i32 16
4671 %6 = lshr %3, %5
4672 %7 = trunc.i8 %6
4673 %8 = iconst.i64 2
4674 %9 = ptr_add %0, %8
4675 store %7 -> %9, align 1
4676 return
4677"
4678 );
4679 }
4680
4681 #[test]
4682 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
4683 let text =
4684 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
4685 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
4688 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
4689 }
4690
4691 #[test]
4692 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
4693 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
4696 assert_eq!(text.matches("ashr").count(), 0, "{text}");
4697 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
4698 }
4699
4700 #[test]
4701 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
4702 let text = body(
4706 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
4707 );
4708 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
4709 }
4710
4711 #[test]
4712 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
4713 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
4716 assert!(
4717 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
4718 "{text}"
4719 );
4720 }
4721
4722 #[test]
4723 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
4724 let text = ir(concat!(
4729 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
4730 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
4731 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
4732 "char s[2] = \"hi\";\n",
4733 ));
4734 assert!(
4735 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
4736 "{text}"
4737 );
4738 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
4739 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
4740 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
4743 }
4744
4745 #[test]
4746 fn a_definition_takes_a_parameter_it_left_unnamed() {
4747 let text = ir("int f(int a, int) { return a; }\n");
4751 assert!(text.contains("func @f(i32, i32) -> i32"), "{text}");
4752 assert!(text.contains("block0(%0: i32, %1: i32):"), "{text}");
4753
4754 let text = ir("int g(int, int n) { return n; }\n");
4757 assert!(text.contains("block0(%0: i32, %1: i32):\n return %1\n"), "{text}");
4758 }
4759
4760 #[test]
4761 fn an_assignment_of_a_structure_is_the_object_it_wrote() {
4762 let text = body(concat!(
4767 "struct s { int f; int g; };\n",
4768 "void h(struct s *a, struct s *c, struct s *d, struct s *e)\n",
4769 "{ *d = *e = a[0] = *c; }\n",
4770 ));
4771 assert_eq!(text.matches("memcpy").count(), 3, "{text}");
4772 assert!(text.contains("memcpy %8, %1, size 8, align 4\n"), "{text}");
4773 assert!(text.contains("memcpy %3, %8, size 8, align 4\n"), "{text}");
4774 assert!(text.contains("memcpy %2, %3, size 8, align 4\n"), "{text}");
4775 }
4776
4777 #[test]
4778 fn a_string_literal_stops_at_the_end_of_the_array_it_is_filling() {
4779 let mut opts = options();
4784 opts.emit = EmitKind::Ir;
4785 let result = run(
4786 &opts,
4787 concat!(
4788 "const char a[2][3] = { \"1234\", \"xyz\" };\n",
4789 "static const char b[3][5] = { \"12345\", \"678\", \"9\" };\n",
4790 "union u { struct { char x[4]; char y[4]; }; struct { char z[8]; }; };\n",
4791 "const union u c = { { \"1234\", \"567\" } };\n",
4792 ),
4793 );
4794 let text = result.text();
4795 assert_eq!(
4796 result.messages,
4797 ["/main.c:1:24: warning: initializer-string for array of 'const char' is too long \
4798 (5 chars into 3 available) [E0637]"]
4799 );
4800 assert!(text.contains("global @a : bytes 6 = { bytes \"123\", bytes \"xyz\" }"), "{text}");
4801 assert!(
4802 text.contains(
4803 "global @b : bytes 15 = { bytes \"12345\", bytes \"678\\00\", zero 1, \
4804 bytes \"9\\00\", zero 3 }"
4805 ),
4806 "{text}"
4807 );
4808 assert!(
4811 text.contains("global @c : bytes 8 = { bytes \"1234\", bytes \"567\\00\" }"),
4812 "{text}"
4813 );
4814 }
4815
4816 #[test]
4817 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
4818 let text = body(concat!(
4822 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
4823 "void g(struct v *);\n",
4824 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
4825 ));
4826 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
4827 }
4828
4829 #[test]
4830 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
4831 let text = ir(concat!(
4836 "struct s { int x; };\n",
4837 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
4838 "int n = (int){ 7 };\n",
4839 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
4840 ));
4841 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
4842 assert!(text.contains("global @n : i32 = 7,"), "{text}");
4843 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
4846 }
4847
4848 #[test]
4849 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
4850 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
4854 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
4855 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
4856 }
4857
4858 #[test]
4859 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
4860 let text = ir("unsigned char foo[1][0];\n");
4864 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
4865 }
4866
4867 #[test]
4868 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
4869 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
4872 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
4873 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
4874 }
4875
4876 #[test]
4877 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
4878 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
4882 assert!(
4883 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
4884 "{text}"
4885 );
4886 }
4887
4888 #[test]
4889 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
4890 let text = body(
4895 "\
4896struct s { int a, b; };
4897struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
4898",
4899 );
4900 assert!(text.contains("block3(%7: ptr)"), "{text}");
4902 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
4903 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
4904 }
4905
4906 #[test]
4914 fn the_left_side_of_a_conditional_with_no_middle_is_evaluated_once() {
4915 let text = body("int f(int i) { return ++i ?: 10; }\n");
4916 assert!(text.contains("jump block3(%2)"), "the arm is the value that was tested: {text}");
4917 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
4918
4919 let text = body("long f(int i) { return ++i ?: 10L; }\n");
4922 assert!(text.contains("%5 = sext.i64 %2"), "the arm widens what was tested: {text}");
4923 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
4924
4925 let text = body("int g(void);\nint f(void) { return g() ?: 10; }\n");
4927 assert_eq!(text.matches("call @g").count(), 1, "called once: {text}");
4928
4929 let text = body("int f(int i) { return ++i ? ++i : 10; }\n");
4932 assert_eq!(text.matches("add.nsw").count(), 2, "incremented twice: {text}");
4933 }
4934
4935 #[test]
4936 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
4937 let text = ir("\
4941struct pair { int a, b; };
4942struct pair make(int a, int b);
4943struct pair twice(struct pair p) { return make(p.a, p.b); }
4944");
4945 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
4946 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
4947 }
4948
4949 #[test]
4950 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
4951 let text = ir("\
4955struct big { double v[8]; };
4956struct big grow(struct big b);
4957struct big twice(struct big b) { return grow(grow(b)); }
4958");
4959 assert!(
4960 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
4961 "{text}"
4962 );
4963 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
4964 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
4967 }
4968
4969 #[test]
4970 fn a_structure_passed_to_a_variadic_function_says_so_at_the_call() {
4971 let text = ir("\
4976struct big { double v[8]; };
4977struct pair { int a, b; };
4978int p(const char *, ...);
4979int f(struct big b, struct pair q) { return p(\"\", 1, b, q); }
4980");
4981 assert!(
4982 text.contains("call @p(%4, %5, %2 byval(64, align 8), %6) : (ptr, ...) -> i32"),
4983 "{text}"
4984 );
4985 }
4986
4987 #[test]
4988 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
4989 let body = body(
4992 "\
4993struct pair { int a, b; };
4994struct pair make(int a, int b);
4995int second(void) { return make(1, 2).b; }
4996",
4997 );
4998 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
4999 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
5000 }
5001
5002 #[test]
5003 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
5004 let source = "\
5008struct hfa { float x, y, z; };
5009int take(struct hfa h);
5010int give(struct hfa h) { return take(h); }
5011";
5012 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
5013 let mut opts = options();
5014 opts.emit = EmitKind::Ir;
5015 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
5016 let result = run(&opts, source);
5017 assert_eq!(result.messages, Vec::<String>::new());
5018 assert!(result.text().contains("func @take(f32, f32, f32) -> i32"), "{}", result.text());
5019 }
5020
5021 #[test]
5022 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
5023 let source = "\
5026int use(int *);
5027void f(int n) {
5028 {
5029 int a[n];
5030 use(a);
5031 }
5032 use(0);
5033}
5034";
5035 let body = body(source);
5036 assert!(body.contains("mul.nsw"), "{body}");
5037 assert!(body.contains("stacksave"), "{body}");
5038 assert!(body.contains("alloca %"), "{body}");
5039 assert!(body.contains("stackrestore"), "{body}");
5040 }
5041
5042 #[test]
5043 fn a_goto_out_of_the_scope_of_one_gives_its_stack_back_on_the_way() {
5044 let source = "\
5049int use(int *);
5050int f(int n) {
5051 {
5052 int a[n];
5053 if (use(a)) goto out;
5054 use(0);
5055 }
5056out:
5057 return 0;
5058}
5059";
5060 let body = body(source);
5061 assert_eq!(body.matches("stackrestore").count(), 2, "{body}");
5063 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5064 assert!(after.starts_with(" %4\n jump block"), "{body}");
5065 }
5066
5067 #[test]
5068 fn a_goto_to_a_label_the_array_is_still_alive_at_leaves_the_stack_alone() {
5069 let source = "\
5073int use(int *);
5074int f(int n) {
5075 int a[n];
5076again:
5077 if (use(a)) goto again;
5078 return 0;
5079}
5080";
5081 let body = body(source);
5082 assert!(body.contains("stacksave"), "{body}");
5083 assert!(!body.contains("stackrestore"), "{body}");
5084 }
5085
5086 #[test]
5087 fn a_goto_back_to_a_label_in_front_of_one_gives_it_back_every_time_round() {
5088 let source = "\
5093int use(int *);
5094int f(int n) {
5095again:
5096 {
5097 int a[n];
5098 if (use(a)) goto again;
5099 }
5100 return 0;
5101}
5102";
5103 let body = body(source);
5104 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
5105 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5106 assert!(after.starts_with(" %4\n jump block1\n"), "{body}");
5107 }
5108
5109 #[test]
5110 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
5111 let source = "\
5117int f(void);
5118void t(void) {
5119 int count = 10;
5120 for (; count--;) {
5121 int b[f()];
5122 int i;
5123 for (i = 0; i < f(); i++) {
5124 b[i] = count;
5125 }
5126 }
5127}
5128";
5129 let body = body(source);
5130 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
5134 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5135 let next = after.split("\n\n").next().expect("the block the restore is in");
5138 assert!(next.contains("jump block1("), "{body}");
5139 }
5140
5141 #[test]
5142 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
5143 let source = "\
5146unsigned long f(int n) {
5147 int a[n];
5148 n = 0;
5149 return sizeof a;
5150}
5151";
5152 let body = body(source);
5153 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
5155 }
5156
5157 #[test]
5158 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
5159 let source = "\
5162int use(int);
5163int f(int x) {
5164 return ({
5165 int t = use(x);
5166 t * t;
5167 });
5168}
5169";
5170 let expected = "\
5171block0(%0: i32):
5172 %1 = call @use(%0) : (i32) -> i32
5173 %2 = mul.nsw %1, %1
5174 return %2
5175";
5176 assert_eq!(body(source), expected);
5177 }
5178
5179 #[test]
5180 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
5181 let source = "int f(int x) { return ({ return x; 0; }); }\n";
5185 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
5186 }
5187
5188 #[test]
5189 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
5190 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
5194 let expected = "\
5195block0(%0: ptr):
5196 %1 = va_arg.f64 %0
5197 %2 = va_arg.f64 %0
5198 %3 = fadd %1, %2
5199 return %3
5200";
5201 assert_eq!(body(source), expected);
5202 }
5203
5204 #[test]
5205 fn one_that_reads_a_structure_answers_where_the_object_is() {
5206 let source = "\
5220struct s { int a; long b; };
5221long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.b; }
5222";
5223 let expected = "\
5224block0(%0: ptr):
5225 %1 = alloca, size 16, align 16
5226 %2 = va_object %0, size 16, align 8, in(int 8 at 0, int 8 at 8)
5227 memcpy %1, %2, size 16, align 8
5228 %3 = iconst.i64 8
5229 %4 = ptr_add %1, %3
5230 %5 = load.i64 %4, align 8
5231 return %5
5232";
5233 assert_eq!(body(source), expected);
5234 }
5235
5236 #[test]
5240 fn the_classification_says_which_registers_the_object_arrived_in() {
5241 let source = "\
5242struct s { double a; double b; };
5243double f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a; }
5244";
5245 assert!(
5246 body(source)
5247 .contains("va_object %0, size 16, align 8, in(float f64 at 0, float f64 at 8)"),
5248 "{}",
5249 body(source)
5250 );
5251
5252 let big = "\
5253struct s { long a[4]; };
5254long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a[0]; }
5255";
5256 assert!(body(big).contains("va_object %0, size 32, align 8\n"), "{}", body(big));
5257 }
5258
5259 #[test]
5260 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
5261 let source = "\
5265int f(int c) {
5266 void *p = c ? &&one : &&two;
5267 goto *p;
5268one:
5269 return 1;
5270two:
5271 return 2;
5272}
5273";
5274 let expected = "\
5275block0(%0: i32):
5276 %1 = iconst.i32 0
5277 %2 = icmp ne %0, %1
5278 br_if %2, block1, block2
5279
5280block1:
5281 %3 = block_addr block3
5282 jump block4(%3)
5283
5284block2:
5285 %4 = block_addr block5
5286 jump block4(%4)
5287
5288block3:
5289 %5 = iconst.i32 1
5290 return %5
5291
5292block4(%6: ptr):
5293 indirect_br %6, block3, block5
5294
5295block5:
5296 %7 = iconst.i32 2
5297 return %7
5298";
5299 assert_eq!(body(source), expected);
5300 }
5301
5302 #[test]
5303 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
5304 let source = "void **next(void);
5307void f(void) { goto *next(); }
5308";
5309 let expected = "\
5310block0:
5311 %0 = call @next() : () -> ptr
5312 unreachable
5313";
5314 assert_eq!(body(source), expected);
5315 }
5316
5317 #[test]
5318 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
5319 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
5322 let expected = "\
5323block0:
5324 inline_asm.volatile \"mfence\", \"\", \"memory\"()
5325 return
5326";
5327 assert_eq!(body(source), expected);
5328 }
5329
5330 #[test]
5331 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
5332 let source = "\
5335int f(int x, int y) {
5336 int r;
5337 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
5338 return r + y;
5339}
5340";
5341 let expected = "\
5342block0(%0: i32, %1: i32):
5343 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
5344 %4 = add.nsw %2, %3
5345 return %4
5346";
5347 assert_eq!(body(source), expected);
5348 }
5349
5350 #[test]
5351 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
5352 let source = "\
5357struct pair { int a, b; };
5358int f(int x) {
5359 int slot = x;
5360 struct pair p = { x, x };
5361 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
5362 return slot + p.a;
5363}
5364";
5365 let text = body(source);
5366 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
5367 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
5368 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
5369 }
5370
5371 #[test]
5372 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
5373 let source = "\
5378int f(int x) {
5379 int r = 7;
5380 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
5381 return r;
5382away:
5383 return r;
5384}
5385";
5386 let expected = "\
5387block0(%0: i32):
5388 %1 = iconst.i32 7
5389 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
5390
5391block1:
5392 return %2
5393
5394block2:
5395 return %1
5396";
5397 assert_eq!(body(source), expected);
5398 }
5399
5400 #[test]
5401 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
5402 let mut opts = options();
5406 opts.emit = EmitKind::Ir;
5407 for (source, expected) in [
5408 (
5409 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
5410 "output operand constraint lacks '='",
5411 ),
5412 (
5413 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
5414 "lvalue required in 'asm' statement",
5415 ),
5416 (
5417 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
5418 "read-only variable 'g' used as 'asm' output",
5419 ),
5420 (
5421 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
5422 "input operand constraint contains '='",
5423 ),
5424 (
5425 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
5426 "memory input 0 is not directly addressable",
5427 ),
5428 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
5429 (
5430 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
5431 "duplicate asm operand name 'a'",
5432 ),
5433 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
5434 ] {
5435 let result = run(&opts, source);
5436 assert!(result.failed(), "expected this to be reported:\n{source}");
5437 assert!(
5438 result.messages.iter().any(|m| m.contains(expected)),
5439 "{expected}\n{:?}",
5440 result.messages
5441 );
5442 }
5443 }
5444
5445 #[test]
5446 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
5447 let mut opts = options();
5448 opts.emit = EmitKind::Ir;
5449 for source in [
5450 "int f(int n) { void *p = &&out; if (n) goto *p; { int a[n]; out: return 1; } }\n",
5451 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
5452 ] {
5453 let result = run(&opts, source);
5454 assert!(result.failed(), "expected this to be reported:\n{source}");
5455 assert!(
5456 result.messages.iter().any(|m| m.contains("not supported yet")),
5457 "{:?}",
5458 result.messages
5459 );
5460 }
5461 }
5462
5463 fn round_trip(source: &str) -> (String, String) {
5465 let printed = ir(source);
5466 let mut opts = options();
5467 opts.emit = EmitKind::Ir;
5468 let mut fs = MemoryFileSystem::new();
5469 fs.insert("/main.ir", printed.clone().into_bytes());
5470 let result = compile_ir(&opts, "/main.ir", &fs);
5471 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
5472 (printed, result.text().to_owned())
5473 }
5474
5475 #[test]
5476 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
5477 let (printed, again) = round_trip(
5481 "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",
5482 );
5483 assert_eq!(printed, again);
5484 }
5485
5486 #[test]
5487 fn ir_that_is_not_ir_says_which_line_stopped_it() {
5488 let mut opts = options();
5489 opts.emit = EmitKind::Ir;
5490 let mut fs = MemoryFileSystem::new();
5491 let text = "\
5492; ModuleID = 'a.c'
5493; format 0
5494target triple = \"x86_64-unknown-linux-gnu\"
5495target datalayout = \"e-p:64:64-i64:64-S128\"
5496
5497func @f(), linkage(external) {
5498block0:
5499 frobnicate
5500}
5501";
5502 fs.insert("/main.ir", text.as_bytes().to_vec());
5503 let result = compile_ir(&opts, "/main.ir", &fs);
5504 assert!(result.failed());
5505 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
5506 }
5507
5508 #[test]
5509 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
5510 let mut opts = options();
5513 opts.emit = EmitKind::Ir;
5514 let mut fs = MemoryFileSystem::new();
5515 let text = "\
5516; ModuleID = 'a.c'
5517; format 0
5518target triple = \"x86_64-unknown-linux-gnu\"
5519target datalayout = \"e-p:64:64-i64:64-S128\"
5520
5521func @f(), linkage(external) {
5522block0:
5523 %0 = iconst.i32 1
5524 return %0
5525}
5526";
5527 fs.insert("/main.ir", text.as_bytes().to_vec());
5528 let result = compile_ir(&opts, "/main.ir", &fs);
5529 assert!(result.failed());
5530 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
5531 }
5532
5533 #[test]
5534 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
5535 let mut fs = MemoryFileSystem::new();
5537 fs.insert("/main.ir", Vec::new());
5538 let result = compile_ir(&options(), "/main.ir", &fs);
5539 assert!(result.failed());
5540 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
5541 }
5542
5543 #[test]
5544 fn the_printed_ir_reads_back_as_the_same_module() {
5545 let text = ir("\
5548struct point { int x, y; };
5549static const char greeting[] = \"hi\";
5550int table[4] = { 1, 2, 3 };
5551int puts(const char *);
5552double half(double x) { return x / 2.0; }
5553int f(int n) {
5554 int total = 0;
5555 for (int i = 0; i < n; i++) {
5556 if (i == 3) continue;
5557 total += table[i];
5558 }
5559 switch (n) {
5560 case 0: total = 1;
5561 case 1: total++; break;
5562 default: total = -total;
5563 }
5564 struct point p = { total, 1 };
5565 int *q = &p.y;
5566 puts(greeting);
5567 return p.x + *q;
5568}
5569int dispatch(int c) {
5570 void *p = c ? &&one : &&two;
5571 goto *p;
5572one:
5573 return 1;
5574two:
5575 return 2;
5576}
5577int assembly(int x, int *p) {
5578 int r;
5579 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
5580 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
5581 return r;
5582away:
5583 return 0;
5584}
5585");
5586 let mut names = Interner::new();
5587 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
5588 assert_eq!(rucc_ir::print(&module, &names), text);
5589 }
5590
5591 #[test]
5592 fn what_save_temps_keeps_is_the_text_that_was_compiled_and_the_assembly_that_was_assembled() {
5593 let mut opts = options();
5597 opts.emit = EmitKind::Object;
5598 opts.save_temps = rucc_session::SaveTemps::Object;
5599 let result = run(&opts, "#define N 2\nint a[N];\n");
5600 assert_eq!(result.messages, Vec::<String>::new());
5601 let text = result.temps.preprocessed.expect("the preprocessed text");
5602 assert!(text.contains("int a[2];"), "{text}");
5603 assert!(text.starts_with("# 1 \"/main.c\""), "{text}");
5604 let asm = result.temps.assembly.expect("the assembly");
5605 assert!(asm.contains("a:"), "{asm}");
5606 assert!(matches!(result.artifact, Artifact::Object(_)), "{:?}", result.artifact);
5607 }
5608
5609 #[test]
5610 fn nothing_is_kept_unless_the_flag_asked_for_it() {
5611 let mut opts = options();
5614 opts.emit = EmitKind::Object;
5615 assert_eq!(run(&opts, "int a;\n").temps, Temps::default());
5616 }
5617
5618 #[test]
5619 fn a_compilation_that_stops_before_the_back_end_keeps_the_text_and_no_assembly() {
5620 let mut opts = options();
5623 opts.emit = EmitKind::Ir;
5624 opts.save_temps = rucc_session::SaveTemps::Cwd;
5625 let result = run(&opts, "int a;\n");
5626 assert!(result.temps.preprocessed.is_some());
5627 assert_eq!(result.temps.assembly, None);
5628 }
5629}