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_codegen::pressure::Pressure;
20use rucc_diag::{Diagnostic, Severity, Span};
21use rucc_ir::{Pic as IrPic, Visibility as IrVisibility};
22use rucc_lex::{Convert, Keywords, PpToken, convert};
23use rucc_lower::Protector as LowerProtector;
24use rucc_sema::{Checker, Context as CheckContext};
25use rucc_session::{EmitKind, FileSystem, Options, Pic, Protector, Session, Visibility};
26use rucc_target::TargetInfo;
27use rucc_tuple::{Arch, ObjectFormat};
28
29use crate::preprocess::render;
30
31#[derive(Debug, Clone, PartialEq, Eq, Default)]
38pub enum Artifact {
39 #[default]
42 Nothing,
43 Text(String),
45 Object(Vec<u8>),
47}
48
49impl Artifact {
50 #[must_use]
52 pub fn bytes(&self) -> &[u8] {
53 match self {
54 Artifact::Nothing => &[],
55 Artifact::Text(text) => text.as_bytes(),
56 Artifact::Object(bytes) => bytes,
57 }
58 }
59}
60
61#[derive(Debug, Clone, PartialEq, Eq)]
63pub struct Compiled {
64 pub artifact: Artifact,
66 pub messages: Vec<String>,
68 pub errors: u32,
70 pub fired: Fired,
76 pub pressure: Pressure,
81 pub dumps: Vec<rucc_opt::Dump>,
87 pub remarks: String,
93 pub deps: Vec<rucc_pp::Dependency>,
98 pub temps: Temps,
105}
106
107#[derive(Debug, Clone, PartialEq, Eq, Default)]
114pub struct Temps {
115 pub preprocessed: Option<String>,
117 pub assembly: Option<String>,
119}
120
121impl Compiled {
122 #[must_use]
124 pub fn failed(&self) -> bool {
125 self.errors > 0
126 }
127
128 #[must_use]
133 pub fn text(&self) -> &str {
134 match &self.artifact {
135 Artifact::Text(text) => text,
136 _ => "",
137 }
138 }
139}
140
141#[must_use]
154pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
155 let mut sess = Session::new(opts.clone());
156 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
160 let mut diagnostics: Vec<Diagnostic> = Vec::new();
161 let mut fired = Fired::new();
163 let mut pressure = Pressure::new();
165 let mut dumps = Vec::new();
167 let mut remarks = String::new();
168 let mut temps = Temps::default();
170
171 let bytes = match fs.read(Path::new(name)) {
172 Ok(bytes) => bytes,
173 Err(e) => return failure(format!("{name}: {e}")),
174 };
175 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
176 return failure(format!("{name}: the source map has no room left for this file"));
177 };
178
179 let mut pp = rucc_pp::Preprocessor::new();
183 let predef = rucc_pp::Predef::for_options(opts);
184 let expanded: Vec<PpToken> = {
185 let mut tokens = Vec::new();
186 {
191 let mut cx =
192 rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
193 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
194 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
195 return failure(format!(
196 "{name}: the source map has no room for the built in macros"
197 ));
198 }
199 if pp.preinclude(&opts.preincludes, &mut tokens, &mut cx).is_err() {
200 return failure(format!("{name}: the source map has no room for the command line"));
201 }
202 tokens.append(&mut pp.run(file, &mut cx));
203 }
204 if opts.save_temps.wanted() {
205 temps.preprocessed = Some(rucc_pp::print(
206 file,
207 &tokens,
208 pp.line_directives(),
209 &sess.sources,
210 &sess.interner,
211 rucc_pp::PrintOptions { line_markers: opts.line_markers },
212 ));
213 }
214 tokens.iter().map(|token| token.to_pp()).collect()
215 };
216 diagnostics.extend(pp.take_diagnostics());
217 let deps = pp.dependencies().to_vec();
220
221 let cx = Convert {
224 keywords: &keywords,
225 interner: &sess.interner,
226 target: &sess.target,
227 std: opts.std,
228 gnu: opts.gnu_extensions,
229 pedantic: opts.pedantic,
230 };
231 let (tokens, complaints) = convert(&expanded, &cx);
232 diagnostics.extend(complaints);
233
234 let parsed = rucc_parse::parse(
235 &tokens,
236 rucc_parse::Context {
237 interner: &sess.interner,
238 std: opts.std,
239 gnu: opts.gnu_extensions,
240 pedantic: opts.pedantic,
241 error_limit: opts.error_limit as usize,
242 },
243 );
244 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
245 diagnostics.extend(parsed.diagnostics);
246
247 let mut artifact = Artifact::Nothing;
248 let mut instrumented = Instrumented::default();
251 if !parse_failed {
252 let mut checker = Checker::new(
253 &parsed.ast,
254 CheckContext {
255 names: &sess.interner,
256 target: &sess.target,
257 std: opts.std,
258 gnu: opts.gnu_extensions,
259 pedantic: opts.pedantic,
260 permissive: opts.permissive,
261 gnu89_inline: opts.gnu89_inline,
262 error_limit: opts.error_limit as usize,
263 builtins: opts.builtins && opts.hosted,
266 no_builtin: &opts.no_builtin,
267 },
268 );
269 checker.check_unit();
270 let checked = checker.finish();
271 if !checked.failed() {
272 match opts.emit {
273 EmitKind::Tast => {
274 artifact = Artifact::Text(rucc_sema::print(
275 &checked.tast,
276 &checked.types,
277 &sess.interner,
278 ));
279 }
280 EmitKind::TypeGranules => {
284 artifact = Artifact::Text(rucc_types::granule_report(
285 &checked.types,
286 &sess.interner,
287 &sess.target,
288 ));
289 }
290 EmitKind::Ir
291 | EmitKind::MirFinal
292 | EmitKind::Asm
293 | EmitKind::Object
294 | EmitKind::Executable
295 | EmitKind::SafetySummary => {
296 let mut lowered = rucc_lower::lower(
297 name,
298 rucc_lower::Context {
299 tast: &checked.tast,
300 types: &checked.types,
301 target: &sess.target,
302 names: &mut sess.interner,
303 visibility: match opts.visibility {
304 Visibility::Default => IrVisibility::Default,
305 Visibility::Hidden => IrVisibility::Hidden,
306 Visibility::Protected => IrVisibility::Protected,
307 },
308 protector: match opts.protector {
309 Protector::None => LowerProtector::None,
310 Protector::Buffers => LowerProtector::Buffers,
311 Protector::Strong => LowerProtector::Strong,
312 Protector::All => LowerProtector::All,
313 },
314 wrapping: rucc_lower::Wrapping {
315 signed: opts.wrapping.signed,
316 pointer: opts.wrapping.pointer,
317 trap: opts.wrapping.trap,
318 },
319 },
320 );
321 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
325 if !failed {
326 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
331 for error in errors {
332 diagnostics.push(internal(&format!("invalid IR, {error}")));
333 }
334 } else if let Err(complaints) =
335 instrument(&mut lowered.module, &mut sess.interner, opts)
336 .map(|done| instrumented = done)
337 {
338 diagnostics.extend(complaints);
339 } else if let Err(complaints) = optimize(
340 &mut lowered.module,
341 &sess.interner,
342 &sess.target,
343 opts,
344 name,
345 &mut dumps,
346 &mut remarks,
347 ) {
348 diagnostics.extend(complaints);
349 } else if opts.emit == EmitKind::SafetySummary {
350 artifact = Artifact::Text(
355 rucc_safety::summarize(
356 &lowered.module,
357 &sess.interner,
358 name,
359 opts.safety.as_str(),
360 instrumented.checks,
361 instrumented.interposed,
362 instrumented.crossings,
363 )
364 .render(),
365 );
366 } else if opts.emit == EmitKind::Ir {
367 artifact =
372 Artifact::Text(rucc_ir::print(&lowered.module, &sess.interner));
373 } else {
374 match generate(
377 &mut lowered.module,
378 &mut sess.interner,
379 &sess.target,
380 opts,
381 &mut fired,
382 &mut pressure,
383 &mut temps.assembly,
384 ) {
385 Ok(made) => artifact = made,
386 Err(complaints) => diagnostics.extend(complaints),
387 }
388 }
389 }
390 diagnostics.extend(lowered.diagnostics);
391 }
392 _ => {}
393 }
394 }
395 diagnostics.extend(checked.diagnostics);
396 }
397
398 let mut messages = Vec::with_capacity(diagnostics.len());
399 let mut errors = 0;
400 for diag in &diagnostics {
401 if !opts.warnings && diag.severity == Severity::Warning {
405 continue;
406 }
407 if diag.severity.is_fatal()
408 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
409 {
410 errors += 1;
411 }
412 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
413 }
414 if errors > 0 {
415 artifact = Artifact::Nothing;
417 }
418 Compiled { artifact, messages, errors, fired, pressure, dumps, remarks, deps, temps }
421}
422
423#[must_use]
433pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
434 let mut sess = Session::new(opts.clone());
435 if opts.emit != EmitKind::Ir {
436 return failure(format!(
437 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
438 the C in front of it became",
439 opts.emit.as_str()
440 ));
441 }
442 let bytes = match fs.read(Path::new(name)) {
443 Ok(bytes) => bytes,
444 Err(e) => return failure(format!("{name}: {e}")),
445 };
446 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
447 return failure(format!("{name}: this is not text, so it is not IR"));
448 };
449
450 let module = match rucc_ir::parse(text, &mut sess.interner) {
451 Ok(module) => module,
452 Err(error) => {
453 return failure(format!("{name}:{}: {}", error.line, error.message));
454 }
455 };
456 let mut diagnostics: Vec<Diagnostic> = Vec::new();
457 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
458 for error in errors {
459 diagnostics.push(invalid(&format!("invalid IR, {error}")));
460 }
461 }
462 let mut messages = Vec::with_capacity(diagnostics.len());
463 for diag in &diagnostics {
464 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
465 }
466 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
467 let artifact = if errors > 0 {
468 Artifact::Nothing
469 } else {
470 Artifact::Text(rucc_ir::print(&module, &sess.interner))
471 };
472 Compiled {
474 artifact,
475 messages,
476 errors,
477 fired: Fired::new(),
478 pressure: Pressure::new(),
479 dumps: Vec::new(),
480 remarks: String::new(),
481 deps: Vec::new(),
482 temps: Temps::default(),
483 }
484}
485
486fn instrument(
509 module: &mut rucc_ir::Module,
510 names: &mut Interner,
511 opts: &Options,
512) -> Result<Instrumented, Vec<Diagnostic>> {
513 if !opts.safety.instruments() {
514 return Ok(Instrumented::default());
515 }
516 let checks = rucc_safety::run(module);
517 let interposed = rucc_safety::redirect(module, names);
522 let crossings = rucc_safety::witness(module, names);
525 match rucc_ir::verify(module, names) {
526 Ok(()) => Ok(Instrumented { checks, interposed, crossings }),
527 Err(errors) => Err(errors
528 .iter()
529 .map(|e| internal(&format!("invalid IR after check insertion, {e}")))
530 .collect()),
531 }
532}
533
534#[derive(Clone, Copy, Debug, Default)]
540struct Instrumented {
541 checks: rucc_safety::Counts,
543 interposed: usize,
545 crossings: rucc_safety::Sites,
547}
548
549fn optimize(
561 module: &mut rucc_ir::Module,
562 names: &Interner,
563 target: &TargetInfo,
564 opts: &Options,
565 file: &str,
566 dumps: &mut Vec<rucc_opt::Dump>,
567 remarks: &mut String,
568) -> Result<(), Vec<Diagnostic>> {
569 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
570 settings.interposition = match opts.interposition {
576 true => replaceable(target, opts),
577 false => IrPic::Executable,
578 };
579 settings.toggles.clone_from(&opts.passes);
580 settings.fuel = opts.pass_fuel.iter().cloned().collect();
581 settings.global_fuel = opts.pass_fuel_global;
582 settings.verify |= opts.verify_each;
583 for (on, spec) in &opts.pass_gates {
584 if let Err(why) = settings.gates.add(*on, spec) {
587 return Err(vec![internal(&why)]);
588 }
589 }
590 for spec in &opts.dump_ir {
591 if let Err(why) = settings.dumps.add(spec) {
594 return Err(vec![internal(&why)]);
595 }
596 }
597 let mut wants = rucc_opt::Wants::none();
598 for spec in &opts.opt_info {
599 if let Err(why) = wants.add(spec) {
602 return Err(vec![internal(&why)]);
603 }
604 }
605 let report = rucc_opt::run(module, names, &settings);
606 remarks.push_str(&rucc_opt::optinfo::render(file, &report, names, wants));
607 dumps.extend(report.dumps);
608 match report.broke.is_empty() {
609 true => Ok(()),
610 false => Err(report.broke.iter().map(|why| internal(why)).collect()),
611 }
612}
613
614fn replaceable(target: &TargetInfo, opts: &Options) -> IrPic {
648 match (target.tuple.os().object_format(), opts.pic) {
649 (Some(ObjectFormat::Elf), Pic::Library) => IrPic::Library,
650 _ => IrPic::Executable,
651 }
652}
653
654fn generate(
655 module: &mut rucc_ir::Module,
656 names: &mut Interner,
657 target: &TargetInfo,
658 opts: &Options,
659 fired: &mut Fired,
660 pressure: &mut Pressure,
661 assembly: &mut Option<String>,
662) -> Result<Artifact, Vec<Diagnostic>> {
663 let Some(machine) = Machine::for_target(target) else {
664 return Err(vec![unsupported(&format!(
665 "there is no back end for {} in this compiler yet, so there is nothing to generate",
666 target.tuple
667 ))]);
668 };
669 if opts.protector != Protector::None && machine.conv.guard.is_none() {
674 return Err(vec![unsupported(&format!(
675 "{} is not supported for {} yet, because the stack protector on that target is not \
676 the one this compiler writes",
677 opts.protector, target.tuple
678 ))]);
679 }
680 if opts.control.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
686 return Err(vec![unsupported(&format!(
687 "-fcf-protection={} is not supported for {} yet, because what says a file was built \
688 for it there is not the note this compiler writes",
689 opts.control, target.tuple
690 ))]);
691 }
692 let profile = match machine.conv.trace {
698 Some(trace) => opts.profile.then(|| opts.hook.early(trace.fentry)),
699 None if opts.profile => {
700 return Err(vec![unsupported(&format!(
701 "-pg is not supported for {} yet, because the profiler's hook on that target is \
702 not the one this compiler calls",
703 target.tuple
704 ))]);
705 }
706 None => None,
707 };
708 if opts.patchable.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
713 return Err(vec![unsupported(&format!(
714 "-fpatchable-function-entry= is not supported for {} yet, because what records where \
715 the room is there is not the section this compiler writes",
716 target.tuple
717 ))]);
718 }
719 let flags = pipeline::Flags {
720 frame_pointer: opts.frame_pointer,
721 red_zone: opts.red_zone,
722 stack_clash: opts.stack_clash,
723 landing: opts.control.branch(),
724 profile: match profile {
725 None => pipeline::Profile::No,
726 Some(true) => pipeline::Profile::Early,
727 Some(false) => pipeline::Profile::Late,
728 },
729 patch: pipeline::Room { after: opts.patchable.after(), before: opts.patchable.before },
730 };
731
732 if opts.safety.instruments() {
741 rucc_safety::lower(module, names);
742 if let Err(errors) = rucc_ir::verify(module, names) {
743 return Err(errors
744 .iter()
745 .map(|e| internal(&format!("invalid IR after check lowering, {e}")))
746 .collect());
747 }
748 }
749
750 let elsewhere = Elsewhere::of(module, replaceable(target, opts));
758
759 let mut funcs = Vec::new();
760 let mut complaints = Vec::new();
761 for id in module.funcs() {
762 if module[id].is_declaration() {
763 continue;
764 }
765 match pipeline::compile_recording(
766 &mut module[id],
767 names,
768 &machine,
769 &elsewhere,
770 flags,
771 fired,
772 pressure,
773 ) {
774 Ok(func) => funcs.push(func),
775 Err(why) => {
776 let name = names.resolve(module[id].name).to_owned();
777 let span = why.inst().map_or(Span::DUMMY, |inst| module[id].span(inst));
780 let said = format!("cannot generate code for '{name}': {why}");
781 complaints.push(unsupported_at(&said, span));
782 }
783 }
784 }
785 if !complaints.is_empty() {
786 return Err(complaints);
787 }
788 let (globals, aliases) = match opts.emit {
794 EmitKind::Asm | EmitKind::Object | EmitKind::Executable => (
795 rucc_asm::globals(module, names).map_err(refused)?,
796 rucc_asm::aliases(module, names).map_err(refused)?,
797 ),
798 _ => (rucc_asm::Globals::default(), Vec::new()),
799 };
800 let unwind = opts.unwinds();
804 match opts.emit {
805 EmitKind::Asm => {
806 rucc_asm::print(&funcs, &globals, &aliases, names, target, unwind, output(opts, target))
807 .map(Artifact::Text)
808 .map_err(refused)
809 }
810 EmitKind::Object | EmitKind::Executable => {
813 if opts.save_temps.wanted() {
814 let listing = rucc_asm::print(
815 &funcs,
816 &globals,
817 &aliases,
818 names,
819 target,
820 unwind,
821 output(opts, target),
822 );
823 *assembly = Some(listing.map_err(refused)?);
824 }
825 let text = rucc_asm::assemble(&funcs, names, target, unwind).map_err(refused)?;
826 let data = globals.image();
827 rucc_object::write(&text, &data, &aliases, target, output(opts, target))
830 .map(Artifact::Object)
831 .map_err(|why| match why {
832 rucc_object::Error::Format { .. } => vec![unsupported(&why.to_string())],
833 rucc_object::Error::Refused { .. } => vec![internal(&why.to_string())],
834 })
835 }
836 _ => Ok(Artifact::Text(rucc_mir::print(&funcs, names, target.regs))),
837 }
838}
839
840fn output(opts: &Options, target: &TargetInfo) -> rucc_object::Output {
852 let mut features = 0;
853 if target.tuple.arch() == Arch::X86_64 {
854 if opts.control.branch() {
855 features |= rucc_object::Property::IBT;
856 }
857 if opts.control.ret() {
858 features |= rucc_object::Property::SHSTK;
859 }
860 }
861 rucc_object::Output {
862 sections: rucc_object::Sections {
863 functions: opts.function_sections,
864 data: opts.data_sections,
865 },
866 property: rucc_object::Property { features },
867 }
868}
869
870fn refused(why: rucc_asm::Error) -> Vec<Diagnostic> {
876 match why {
877 rucc_asm::Error::Thread { .. } | rucc_asm::Error::IFunc { .. } => {
878 vec![unsupported(&why.to_string())]
879 }
880 _ => vec![internal(&why.to_string())],
881 }
882}
883
884fn unsupported(message: &str) -> Diagnostic {
890 unsupported_at(message, Span::DUMMY)
891}
892
893fn unsupported_at(message: &str, span: Span) -> Diagnostic {
899 Diagnostic::error(message.to_owned(), span)
900 .with_code("E0653")
901 .note("this construct is not lowered yet, see https://github.com/tamnd/rucc/issues", span)
902}
903
904fn invalid(message: &str) -> Diagnostic {
906 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
907}
908
909fn internal(message: &str) -> Diagnostic {
911 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
912 .with_code("E0652")
913 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
914}
915
916fn failure(message: String) -> Compiled {
919 Compiled {
920 artifact: Artifact::Nothing,
921 messages: vec![format!("rucc: error: {message}")],
922 errors: 1,
923 fired: Fired::new(),
924 pressure: Pressure::new(),
925 dumps: Vec::new(),
926 remarks: String::new(),
927 deps: Vec::new(),
928 temps: Temps::default(),
929 }
930}
931
932#[cfg(test)]
933mod tests {
934 use rucc_session::{MemoryFileSystem, Std};
935 use rucc_target::Triple;
936
937 use super::*;
938
939 fn options() -> Options {
940 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
941 opts.emit = EmitKind::Tast;
942 opts
943 }
944
945 fn run(opts: &Options, source: &str) -> Compiled {
946 let mut fs = MemoryFileSystem::new();
947 fs.insert("/main.c", source.to_owned().into_bytes());
948 compile(opts, "/main.c", &fs)
949 }
950
951 fn freestanding() -> Options {
955 let mut opts = options();
956 opts.hosted = false;
957 opts.search.push_system(rucc_session::runtime::DIR);
958 opts
959 }
960
961 fn shipped(source: &str) -> String {
963 let result = run(&freestanding(), source);
964 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
965 result.text().to_owned()
966 }
967
968 fn tast(source: &str) -> String {
970 let result = run(&options(), source);
971 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
972 result.text().to_owned()
973 }
974
975 #[test]
976 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
977 let text = shipped(concat!(
978 "#include <stdarg.h>\n",
979 "int sum(int n, ...) {\n",
980 " va_list ap, copy;\n",
981 " va_start(ap, n);\n",
982 " va_copy(copy, ap);\n",
983 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
984 " va_end(ap);\n",
985 " va_end(copy);\n",
986 " return total;\n",
987 "}\n",
988 ));
989 assert!(text.contains("va-start"), "{text}");
990 assert!(text.contains("va-copy"), "{text}");
991 assert!(text.contains("va-arg"), "{text}");
992 assert!(text.contains("va-end"), "{text}");
993 }
994
995 #[test]
999 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
1000 let text = shipped(concat!(
1001 "#define __need___va_list\n",
1002 "#include <stdarg.h>\n",
1003 "int vprint(const char *f, __gnuc_va_list ap);\n",
1004 "#ifdef va_start\n",
1005 "#error va_start should not be defined\n",
1006 "#endif\n",
1007 "#ifdef _VA_LIST_DEFINED\n",
1008 "#error va_list should not have been made\n",
1009 "#endif\n",
1010 ));
1011 assert!(text.contains("vprint"), "{text}");
1012 }
1013
1014 #[test]
1017 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
1018 let text = shipped(concat!(
1019 "#define __need_size_t\n",
1020 "#include <stddef.h>\n",
1021 "#ifdef offsetof\n",
1022 "#error offsetof should not be defined yet\n",
1023 "#endif\n",
1024 "#define __need_ptrdiff_t\n",
1025 "#include <stddef.h>\n",
1026 "#include <stddef.h>\n",
1027 "size_t a;\n",
1028 "ptrdiff_t b;\n",
1029 "wchar_t c;\n",
1030 "max_align_t d;\n",
1031 "void *e = NULL;\n",
1032 "struct P { int x; long y; };\n",
1033 "size_t f = offsetof(struct P, y);\n",
1034 ));
1035 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
1036 assert!(text.contains("decl #1 b : long"), "{text}");
1037 }
1038
1039 #[test]
1040 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
1041 let text = shipped(concat!(
1042 "#include <limits.h>\n",
1043 "#include <float.h>\n",
1044 "int bits = CHAR_BIT;\n",
1045 "long big = LONG_MAX;\n",
1046 "int low = INT_MIN;\n",
1047 "int radix = FLT_RADIX;\n",
1048 "int digits = DBL_MANT_DIG;\n",
1049 ));
1050 assert!(text.contains("const 8 : int"), "{text}");
1051 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
1052 assert!(text.contains("const 2 : int"), "{text}");
1053 assert!(text.contains("const 53 : int"), "{text}");
1054 }
1055
1056 #[test]
1060 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
1061 let text = shipped(concat!(
1062 "#include <stdint.h>\n",
1063 "int64_t a = INT64_C(1);\n",
1064 "uint_least16_t b;\n",
1065 "intptr_t c;\n",
1066 "uintmax_t d = UINTMAX_MAX;\n",
1067 "int wide = sizeof(int_fast64_t);\n",
1068 ));
1069 assert!(text.contains("decl #0 a : long"), "{text}");
1070 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
1071 assert!(text.contains("decl #2 c : long"), "{text}");
1072 }
1073
1074 #[test]
1075 fn the_three_formality_headers_still_have_to_work() {
1076 let text = shipped(concat!(
1077 "#include <stdbool.h>\n",
1078 "#include <stdalign.h>\n",
1079 "#include <iso646.h>\n",
1080 "#include <stdnoreturn.h>\n",
1081 "int t = true and not false;\n",
1082 "_Alignas(16) char buf[16];\n",
1083 "int a = alignof(long);\n",
1084 ));
1085 assert!(text.contains("decl #0 t : int"), "{text}");
1086 assert!(text.contains("const 8 : unsigned long"), "{text}");
1087 }
1088
1089 #[test]
1092 fn every_shipped_header_can_be_included_twice() {
1093 let mut source = String::new();
1094 for _ in 0..2 {
1095 for name in rucc_session::runtime::names() {
1096 source.push_str(&format!("#include <{name}>\n"));
1097 }
1098 }
1099 source.push_str("int x;\n");
1100 let text = shipped(&source);
1101 assert!(text.starts_with("decl #0 x : int"), "{text}");
1102 }
1103
1104 #[test]
1105 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
1106 let fs = MemoryFileSystem::new();
1107 let result = compile(&options(), "/nope.c", &fs);
1108 assert!(result.failed());
1109 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
1110 assert!(result.text().is_empty());
1111 }
1112
1113 #[test]
1114 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
1115 let text = tast("int x = 1;\n");
1116 let expected = "\
1117decl #0 x : int object external static defined
1118 init
1119 +0
1120 const 1 : int
1121";
1122 assert_eq!(text, expected);
1123 }
1124
1125 #[test]
1126 fn the_macros_are_expanded_before_anything_is_parsed() {
1127 let text = tast("#define N 2\nint a[N];\n");
1131 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
1132 }
1133
1134 #[test]
1140 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
1141 let text = tast(concat!(
1142 "#pragma pack(4)\n",
1143 "struct s { int a; };\n",
1144 "#pragma pack()\n",
1145 "int b;\n",
1146 "_Pragma(\"GCC visibility push(default)\") int c;\n",
1147 ));
1148 assert!(text.contains("decl #0 b : int"), "{text}");
1149 assert!(text.contains("decl #1 c : int"), "{text}");
1150 }
1151
1152 #[test]
1160 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
1161 tast(concat!(
1162 "struct A { char c; int i; } __attribute__((packed));\n",
1163 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1164 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1165 "struct B { char c; int i; } __attribute__((aligned));\n",
1168 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
1169 "struct C { char c; int i __attribute__((packed)); };\n",
1170 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
1171 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
1172 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
1173 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
1174 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
1175 "struct E { char c; _Alignas(8) int i; };\n",
1176 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
1177 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
1178 "struct F { char c; int i __attribute__((aligned(8))); };\n",
1179 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
1180 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
1183 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
1184 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
1185 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
1186 "struct I { [[gnu::packed]] char c; int i; };\n",
1189 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1190 "struct J { char c; [[gnu::packed]] int i; };\n",
1191 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
1192 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
1193 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
1194 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
1195 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
1196 "union L { char c; int i; } __attribute__((packed));\n",
1197 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
1198 "struct O { char c; int i; } __attribute__((__packed__));\n",
1202 "_Static_assert(sizeof(struct O) == 5 && _Alignof(struct O) == 1, \"O\");\n",
1203 "struct P { char c; int i; } __attribute__((__aligned__(8)));\n",
1204 "_Static_assert(sizeof(struct P) == 8 && _Alignof(struct P) == 8, \"P\");\n",
1205 ));
1206 }
1207
1208 #[test]
1217 fn the_aligned_attribute_on_a_declaration_raises_what_that_one_object_is_aligned_to() {
1218 tast(concat!(
1219 "int v __attribute__((aligned(64)));\n",
1220 "_Static_assert(__alignof__(v) == 64, \"v\");\n",
1221 "__attribute__((aligned(32))) int w;\n",
1224 "_Static_assert(__alignof__(w) == 32, \"w\");\n",
1225 "[[gnu::aligned(16)]] int x;\n",
1226 "_Static_assert(__alignof__(x) == 16, \"x\");\n",
1227 "int y __attribute__((aligned(2)));\n",
1230 "_Static_assert(__alignof__(y) == 4, \"y\");\n",
1231 "void f(void) { int a __attribute__((aligned(128)));\n",
1233 "_Static_assert(__alignof__(a) == 128, \"a\"); (void)a; }\n",
1234 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1237 "void g(void) __attribute__((aligned(256)));\n",
1240 "void g(void) {}\n",
1241 "_Static_assert(__alignof__(g) == 256, \"g\");\n",
1242 ));
1243 }
1244
1245 #[test]
1249 fn what_a_declaration_asked_to_be_aligned_to_is_what_the_assembler_is_told() {
1250 let text = asm(concat!(
1251 "int v __attribute__((aligned(64)));\n",
1252 "void g(void) __attribute__((aligned(256)));\n",
1253 "void g(void) {}\n",
1254 "void plain(void) {}\n",
1255 ));
1256 assert!(text.contains("\t.p2align\t6\n\t.type\tv, @object\n"), "{text}");
1257 assert!(text.contains("\t.p2align\t8, 0x90\n\t.globl\tg\n"), "{text}");
1258 assert!(text.contains("\t.p2align\t4, 0x90\n\t.globl\tplain\n"), "{text}");
1259 }
1260
1261 #[test]
1270 fn an_aligned_typedef_says_what_an_object_of_it_is_aligned_to_and_may_lower_it() {
1271 tast(concat!(
1272 "typedef int L __attribute__((aligned(2)));\n",
1273 "_Static_assert(__alignof__(L) == 2, \"L\");\n",
1274 "_Static_assert(_Alignof(L) == 2, \"L alignof\");\n",
1275 "_Static_assert(sizeof(L) == 4, \"L size\");\n",
1277 "struct T { char c; L x; };\n",
1278 "_Static_assert(sizeof(struct T) == 6, \"T\");\n",
1279 "_Static_assert(__builtin_offsetof(struct T, x) == 2, \"T.x\");\n",
1280 "typedef int H __attribute__((aligned(16)));\n",
1282 "_Static_assert(__alignof__(H) == 16, \"H\");\n",
1283 "_Static_assert(sizeof(H) == 4, \"H size\");\n",
1284 "struct U { char c; H x; };\n",
1285 "_Static_assert(sizeof(struct U) == 32, \"U\");\n",
1286 "_Static_assert(__builtin_offsetof(struct U, x) == 16, \"U.x\");\n",
1287 "typedef L M __attribute__((aligned(8)));\n",
1290 "_Static_assert(__alignof__(M) == 8, \"M\");\n",
1291 "typedef L N;\n",
1294 "_Static_assert(__alignof__(N) == 2, \"N\");\n",
1295 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1297 ));
1298 let text = asm(concat!(
1299 "typedef int L __attribute__((aligned(2)));\n",
1300 "typedef int H __attribute__((aligned(16)));\n",
1301 "L low;\n",
1302 "H high;\n",
1303 ));
1304 assert!(text.contains("\t.p2align\t1\n\t.type\tlow, @object\n"), "{text}");
1305 assert!(text.contains("\t.p2align\t4\n\t.type\thigh, @object\n"), "{text}");
1306 }
1307
1308 #[test]
1316 fn the_vector_size_attribute_builds_a_type_of_lanes_and_measures_it_in_bytes() {
1317 tast(concat!(
1318 "typedef int __attribute__((vector_size(16))) v4si;\n",
1319 "_Static_assert(sizeof(v4si) == 16 && _Alignof(v4si) == 16, \"v4si\");\n",
1320 "typedef char __attribute__((vector_size(16))) v16qi;\n",
1321 "_Static_assert(sizeof(v16qi) == 16, \"v16qi\");\n",
1322 "typedef int __attribute__((vector_size(4))) v1si;\n",
1325 "_Static_assert(sizeof(v1si) == 4, \"v1si\");\n",
1326 "typedef float __attribute__((__vector_size__(8))) v2sf;\n",
1328 "_Static_assert(sizeof(v2sf) == 8, \"v2sf\");\n",
1329 "typedef short [[gnu::vector_size(8)]] v4hi;\n",
1330 "_Static_assert(sizeof(v4hi) == 8, \"v4hi\");\n",
1331 "v4si g;\n",
1334 "_Static_assert(sizeof(g[0]) == 4, \"lane\");\n",
1335 "_Static_assert(sizeof(g + g) == 16, \"whole\");\n",
1336 "_Static_assert(sizeof(g + 1) == 16, \"broadcast\");\n",
1339 "_Static_assert(sizeof(v4si[3]) == 48, \"array\");\n",
1341 ));
1342 }
1343
1344 #[test]
1354 fn a_vector_is_written_whole_into_an_array_of_them_and_named_by_a_type_name() {
1355 tast(concat!(
1356 "typedef int __attribute__((vector_size(8))) v2si;\n",
1357 "v2si table[] = { (v2si){ 1, 2 }, (v2si){ 3, 4 } };\n",
1358 "_Static_assert(sizeof(table) == 16, \"two of them and not eight lanes\");\n",
1359 "v2si written = (int __attribute__((vector_size(8)))){ 5, 6 };\n",
1361 "_Static_assert(sizeof((int __attribute__((vector_size(16)))){ 0 }) == 16, \"named\");\n",
1362 "v2si lanes[2] = { 1, 2, 3, 4 };\n",
1365 "_Static_assert(sizeof(lanes) == 16, \"still elided\");\n",
1366 ));
1367 }
1368
1369 #[test]
1377 fn a_lane_is_assignable_and_a_shift_takes_a_count_of_its_own_lane() {
1378 let result = run(
1379 &options(),
1380 concat!(
1381 "typedef int __attribute__((vector_size(16))) v4si;\n",
1382 "typedef unsigned __attribute__((vector_size(16))) v4ui;\n",
1383 "void write(v4si *out, v4ui a, v4si b, int n) {\n",
1384 " v4si v = { 1, 2, 3, 4 };\n",
1385 " v[0] = n;\n",
1386 " v[1] += n;\n",
1387 " v[2]++;\n",
1388 " *&v[3] = n;\n",
1389 " v4ui shifted = a >> b;\n",
1391 " shifted <<= b;\n",
1392 " *out = v + (v4si)shifted + (1 << b);\n",
1395 "}\n",
1396 "void refused(const v4si c) {\n",
1399 " c[0] = 1;\n",
1400 "}\n",
1401 ),
1402 );
1403 assert_eq!(result.messages.len(), 1, "{:?}", result.messages);
1404 assert!(result.messages[0].contains("assignment of read-only"), "{:?}", result.messages);
1405 }
1406
1407 #[test]
1414 fn a_record_that_asks_for_the_other_byte_order_is_refused_rather_than_laid_out_in_this_one() {
1415 let opts = options();
1416 let big = "struct s { int i; } __attribute__((scalar_storage_order(\"big-endian\")));\n";
1417 assert_eq!(
1418 run(&opts, big).messages,
1419 ["/main.c:1:36: error: 'scalar_storage_order' is not implemented yet [E0688]\n\
1420 /main.c:1:36: note: every scalar in this record would be read in the wrong byte \
1421 order"]
1422 );
1423
1424 let armoured =
1425 "struct s { int i; } __attribute__((__scalar_storage_order__(\"little-endian\")));\n";
1426 let messages = run(&opts, armoured).messages;
1427 assert!(messages[0].contains("[E0688]"), "{messages:?}");
1428
1429 let front = "struct __attribute__((scalar_storage_order(\"big-endian\"))) s { int i; };\n";
1432 assert!(run(&opts, front).messages[0].contains("[E0688]"), "{front}");
1433 let standard = "struct s { int i; } [[gnu::scalar_storage_order(\"big-endian\")]];\n";
1434 assert!(run(&opts, standard).messages[0].contains("[E0688]"), "{standard}");
1435 }
1436
1437 #[test]
1447 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
1448 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
1450 assert_eq!(
1451 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
1452 1
1453 );
1454 assert_eq!(
1455 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
1456 1
1457 );
1458 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
1459 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
1461 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
1462 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
1464 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
1465 }
1466
1467 fn bit_field_byte(record: &str) -> u64 {
1469 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
1470 let body = body(&source);
1471 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
1472 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
1473 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
1474 }
1475
1476 #[test]
1482 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
1483 tast(concat!(
1484 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
1485 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
1486 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
1487 "struct b { char c; __attribute__((packed)) int i; };\n",
1488 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
1489 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
1490 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
1491 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
1492 ));
1493 }
1494
1495 #[test]
1501 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
1502 tast(concat!(
1503 "#pragma pack(1)\n",
1504 "struct A { char c; int i; };\n",
1505 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1506 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1507 "#pragma pack()\n",
1508 "struct B { char c; int i; };\n",
1509 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
1510 "#pragma pack(2)\n",
1511 "struct C { char c; int i; double d; };\n",
1512 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
1513 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
1514 "struct K { char c; int i __attribute__((aligned(8))); };\n",
1516 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
1517 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
1518 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
1520 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
1521 "#pragma pack()\n",
1522 "#pragma pack(push, 1)\n",
1523 "struct D { char c; short s; };\n",
1524 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
1525 "#pragma pack(pop)\n",
1526 "struct E { char c; short s; };\n",
1527 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
1528 "struct H { char c;\n",
1530 "#pragma pack(1)\n",
1531 " int i; };\n",
1532 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
1533 "#pragma pack(1)\n",
1534 "struct I { char c;\n",
1535 "#pragma pack()\n",
1536 " int i; };\n",
1537 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1538 "#pragma pack()\n",
1539 "#pragma pack(push, 8)\n",
1541 "#pragma pack(push, 1)\n",
1542 "struct P { char c; int i; };\n",
1543 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
1544 "#pragma pack(pop)\n",
1545 "struct Q { char c; int i; };\n",
1546 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
1547 "#pragma pack(pop)\n",
1548 "#pragma pack(16)\n",
1550 "struct R { char c; int i; };\n",
1551 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
1552 "#pragma pack()\n",
1553 "#pragma pack(1)\n",
1554 "struct S { char c; int i : 5; int j : 20; };\n",
1555 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
1556 "union T { char c; int i; };\n",
1557 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
1558 "#pragma pack()\n",
1559 ));
1560 }
1561
1562 #[test]
1566 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
1567 let result = run(
1568 &options(),
1569 concat!(
1570 "#pragma pack 4\n",
1571 "#pragma pack(pop)\n",
1572 "#pragma pack(3)\n",
1573 "#pragma pack(1) junk\n",
1574 "#pragma pack(push, 1\n",
1575 "#pragma pack(x)\n",
1576 "#pragma pack(0)\n",
1579 "#pragma pack(push)\n",
1580 "struct s { char c; int i; };\n",
1581 "#pragma pack(pop)\n",
1582 "#pragma pack(pop, foo)\n",
1583 ),
1584 );
1585 let expected = [
1586 "missing `(` after `#pragma pack` - ignored",
1587 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
1588 "alignment must be a small power of two, not 3",
1589 "junk at end of `#pragma pack`",
1590 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
1591 "unknown action `x` for `#pragma pack` - ignored",
1592 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
1593 ];
1594 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
1595 for (message, want) in result.messages.iter().zip(expected) {
1596 assert!(message.contains(want), "expected {want:?} in {message:?}");
1597 }
1598 }
1599
1600 #[test]
1604 fn the_wide_integer_answers_to_all_three_of_its_names() {
1605 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
1606 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
1607 assert!(text.contains("decl #1 b : __int128"), "{text}");
1608 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
1609 }
1610
1611 #[test]
1612 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
1613 let text = tast("long f(int a, long b) { return a + b; }\n");
1617 assert!(text.contains("convert arithmetic"), "{text}");
1618 }
1619
1620 #[test]
1621 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
1622 for source in [
1623 "#error stop\n",
1624 "int f(void) { return 1 + ; }\n",
1625 "int f(void) { return undeclared; }\n",
1626 ] {
1627 let result = run(&options(), source);
1628 assert!(result.failed(), "expected this to fail:\n{source}");
1629 assert!(
1630 result.text().is_empty(),
1631 "a file that did not compile wrote a tree:\n{source}"
1632 );
1633 }
1634 }
1635
1636 #[test]
1637 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
1638 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
1642 assert_eq!(result.errors, 1, "{:?}", result.messages);
1643 }
1644
1645 #[test]
1646 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
1647 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
1651 assert_eq!(result.errors, 1, "{:?}", result.messages);
1652 }
1653
1654 #[test]
1655 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
1656 let source = "int f(void) { char c = 300; return c; }\n";
1657 let plain = run(&options(), source);
1658 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
1659 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
1660 assert!(!plain.text().is_empty(), "a warning is not a reason to write nothing");
1661
1662 let mut opts = options();
1663 opts.warnings_are_errors = true;
1664 let strict = run(&opts, source);
1665 assert!(strict.failed());
1666 assert!(strict.text().is_empty(), "and under -Werror it is a reason to write nothing");
1667 for message in &strict.messages {
1668 assert!(!message.contains("warning:"), "{message}");
1669 }
1670 }
1671
1672 #[test]
1673 fn w_drops_the_warning_before_werror_can_promote_it() {
1674 let source = "int f(void) { char c = 300; return c; }\n";
1675 let mut opts = options();
1676 opts.warnings = false;
1677 let quiet = run(&opts, source);
1678 assert_eq!(quiet.messages, Vec::<String>::new());
1679 assert_eq!(quiet.errors, 0);
1680 assert!(!quiet.text().is_empty(), "and the file still compiles");
1681
1682 opts.warnings_are_errors = true;
1685 let both = run(&opts, source);
1686 assert_eq!(both.messages, Vec::<String>::new());
1687 assert!(!both.failed(), "-w -Werror is not an error about a warning nobody saw");
1688 }
1689
1690 #[test]
1691 fn the_dialect_reaches_the_keywords_and_the_checking() {
1692 let source = "typeof(1) x;\n";
1695 let mut opts = options();
1696 opts.std = Std::C23;
1697 opts.gnu_extensions = false;
1698 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
1699
1700 opts.std = Std::C17;
1701 assert!(run(&opts, source).failed());
1702 }
1703
1704 #[test]
1705 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
1706 let mut opts = options();
1707 opts.emit = EmitKind::Object;
1708 let result = run(&opts, "int x = 1;\n");
1709 assert!(!result.failed(), "{:?}", result.messages);
1710 assert!(result.text().is_empty());
1711 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
1714 }
1715
1716 fn mir(source: &str) -> String {
1718 let mut opts = options();
1719 opts.emit = EmitKind::MirFinal;
1720 let result = run(&opts, source);
1721 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1722 result.text().to_owned()
1723 }
1724
1725 #[test]
1731 fn a_function_goes_from_c_to_instructions_with_real_registers_in_them() {
1732 let text = mir("int add(int a, int b) { return a + b; }\n");
1733 assert!(text.starts_with("mfunc @add {"), "{text}");
1734 assert!(text.contains("x64.add_rr_32"), "{text}");
1735 assert!(text.contains("x64.ret"), "{text}");
1736 assert!(!text.contains('%'), "{text}");
1739 }
1740
1741 #[test]
1743 fn a_function_with_no_body_produces_no_machine_function() {
1744 let text = mir("int g(int);\nint f(int a) { return g(a); }\n");
1745 assert_eq!(text.matches("mfunc @").count(), 1, "{text}");
1746 assert!(text.contains("mfunc @f {"), "{text}");
1747 assert!(text.contains("x64.call"), "{text}");
1748 }
1749
1750 #[test]
1752 fn every_definition_in_the_file_is_generated_and_they_keep_their_order() {
1753 let text = mir("int a(int x) { return x; }\nint b(int x) { return x; }\n");
1754 let first = text.find("mfunc @a").expect("the first function");
1755 let second = text.find("mfunc @b").expect("the second function");
1756 assert!(first < second, "{text}");
1757 }
1758
1759 #[test]
1761 fn the_target_decides_which_convention_the_generated_code_follows() {
1762 let mut opts = options();
1763 opts.emit = EmitKind::MirFinal;
1764 let linux = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1765 assert!(linux.contains("$rdi"), "{linux}");
1766
1767 opts.target = "x86_64-pc-windows-msvc".parse::<Triple>().unwrap();
1768 let windows = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1769 assert!(windows.contains("$rcx"), "{windows}");
1770 assert!(!windows.contains("$rdi"), "{windows}");
1771 }
1772
1773 #[test]
1775 fn a_target_this_has_no_back_end_for_is_reported_rather_than_generated() {
1776 let mut opts = options();
1777 opts.emit = EmitKind::MirFinal;
1778 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
1779 let result = run(&opts, "int f(int a) { return a; }\n");
1780 assert!(result.failed());
1781 assert!(result.messages[0].contains("no back end for aarch64"), "{:?}", result.messages);
1782 assert!(result.text().is_empty());
1783 }
1784
1785 #[test]
1792 fn a_construct_the_back_end_cannot_reach_yet_is_reported_against_its_function() {
1793 let mut opts = options();
1794 opts.emit = EmitKind::MirFinal;
1795 let source = "void a(int n) { int v[n]; v[0] = 1; }\n\
1796 void b(int n) { int v[n]; v[0] = 1; }\n";
1797 let result = run(&opts, source);
1798 assert!(result.failed());
1799 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
1800 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
1801 assert!(result.messages[0].contains("no rule lowers a `stacksave`"), "{:?}", result);
1802 assert!(result.messages[1].contains("cannot generate code for 'b'"), "{:?}", result);
1803 assert!(result.text().is_empty());
1804 }
1805
1806 #[test]
1813 fn an_opcode_with_no_name_in_the_rule_language_is_named_by_its_own_spelling() {
1814 let mut opts = options();
1815 opts.emit = EmitKind::MirFinal;
1816 let result = run(&opts, "int f(int a) {\n __int128 wide = a;\n return (int) wide;\n}\n");
1817 assert!(result.failed());
1818 assert!(
1819 result.messages[0].contains("no rule lowers a `sext` producing a `i128`"),
1820 "{result:?}"
1821 );
1822 assert!(result.messages[0].contains(":2:"), "the line the widening is on: {result:?}");
1823 assert!(!result.messages[0].contains("this instruction"), "{result:?}");
1824 }
1825
1826 #[test]
1828 fn the_note_on_unfinished_work_points_at_the_issues_rather_than_at_the_plan() {
1829 let mut opts = options();
1830 opts.emit = EmitKind::MirFinal;
1831 let result = run(&opts, "int f(int a) { __int128 wide = a; return (int) wide; }\n");
1832 assert!(result.failed());
1833 let note = result.messages.iter().find(|line| line.contains("note:")).expect("a note");
1834 assert!(note.contains("https://github.com/tamnd/rucc/issues"), "{note}");
1835 assert!(!note.contains("spec/17-milestones.md"), "{note}");
1836 }
1837
1838 #[test]
1840 fn the_frame_flags_on_the_command_line_reach_the_generated_frame() {
1841 let source = "int f(int a) { return a; }\n";
1842 assert!(!mir(source).contains("$rbp"), "a leaf needs no frame pointer by default");
1843
1844 let mut opts = options();
1845 opts.emit = EmitKind::MirFinal;
1846 opts.frame_pointer = true;
1847 let kept = run(&opts, source).text().to_owned();
1848 assert!(kept.contains("x64.push_64 $rbp"), "{kept}");
1849 }
1850
1851 fn asm(source: &str) -> String {
1853 let mut opts = options();
1854 opts.emit = EmitKind::Asm;
1855 let result = run(&opts, source);
1856 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1857 result.text().to_owned()
1858 }
1859
1860 #[test]
1867 fn a_function_goes_from_c_to_assembly_an_assembler_would_take() {
1868 let text = asm("int add(int a, int b) { return a + b; }\n");
1869 assert!(text.contains("\t.globl\tadd\n"), "{text}");
1870 assert!(text.contains("\t.type\tadd, @function\n"), "{text}");
1871 assert!(text.contains("\nadd:\n"), "{text}");
1872 assert!(text.contains("\taddl\t"), "{text}");
1873 assert!(text.contains("\tret\n"), "{text}");
1874 assert!(text.contains("\t.size\tadd, .-add\n"), "{text}");
1875 assert!(text.contains(".note.GNU-stack"), "{text}");
1878 }
1879
1880 #[test]
1886 fn a_call_through_a_function_pointer_goes_through_the_register_it_is_in() {
1887 let text = asm("int g(int);\nint f(int (*p)(int), int a) { return p(a) + g(a); }\n");
1888 assert!(text.contains("\tcall\t*%"), "{text}");
1889 assert!(text.contains("\tcall\tg\n"), "{text}");
1890 assert!(text.contains("%rdi"), "{text}");
1894 }
1895
1896 #[test]
1900 fn the_address_of_a_global_is_read_from_the_instruction_pointer() {
1901 let text = asm("extern int counter;\nint f(void) { return counter; }\n");
1902 assert!(text.contains("\tmovl\tcounter(%rip), %eax\n"), "{text}");
1903 }
1904
1905 #[test]
1914 fn a_branch_on_a_comparison_jumps_on_the_opposite_of_what_it_compared() {
1915 let arms = "return 1; return 2;";
1916 let signed = [("==", "jne"), ("!=", "je"), ("<", "jge"), ("<=", "jg"), (">", "jle")];
1917 for (operator, jump) in signed.into_iter().chain([(">=", "jl")]) {
1918 let text = asm(&format!("int f(int a, int b) {{ if (a {operator} b) {arms} }}\n"));
1919 assert!(
1920 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
1921 "{operator}: {text}"
1922 );
1923 assert!(!text.contains("\tset"), "{operator}: {text}");
1924 assert!(!text.contains("\ttest"), "{operator}: {text}");
1925 }
1926 let unsigned = [("<", "jae"), ("<=", "ja"), (">", "jbe"), (">=", "jb")];
1927 for (operator, jump) in unsigned {
1928 let source =
1929 format!("int f(unsigned a, unsigned b) {{ if (a {operator} b) {arms} }}\n");
1930 let text = asm(&source);
1931 assert!(
1932 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
1933 "{operator}: {text}"
1934 );
1935 }
1936
1937 let text = asm("int f(int a) { if (a < 7) return 1; return 2; }\n");
1940 assert!(text.contains("\tcmpl\t$7, %edi\n\tjge\t"), "{text}");
1941 }
1942
1943 #[test]
1949 fn a_comparison_whose_answer_the_program_wanted_still_writes_a_byte() {
1950 let text = asm("int f(int a, int b) { return a < b; }\n");
1951 assert!(text.contains("\tsetl\t"), "{text}");
1952 }
1953
1954 fn optimized(source: &str) -> String {
1956 let mut opts = options();
1957 opts.emit = EmitKind::Asm;
1958 opts.opt_level = rucc_session::OptLevel::O2;
1959 let result = run(&opts, source);
1960 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1961 result.text().to_owned()
1962 }
1963
1964 #[test]
1974 fn a_switch_whose_arms_are_a_function_of_the_label_is_a_range_check_and_arithmetic() {
1975 let arms: String =
1976 (0..16).map(|k| format!("case {k}: return {};", k + 1)).collect::<Vec<_>>().join(" ");
1977 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
1978 assert!(text.contains("\tcmpl\t$15, %edi\n\tja\t"), "{text}");
1979 assert!(text.contains("\taddl\t$1, %edi"), "{text}");
1980 assert_eq!(text.matches("\tcmp").count(), 1, "{text}");
1981 }
1982
1983 #[test]
1990 fn a_dense_switch_whose_arms_are_not_a_line_keeps_its_comparisons() {
1991 let arms: String = (0..16)
1992 .map(|k| format!("case {k}: return {};", if k == 9 { 100 } else { k + 1 }))
1993 .collect::<Vec<_>>()
1994 .join(" ");
1995 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
1996 assert!(text.matches("\tcmp").count() > 1, "{text}");
1997 }
1998
1999 #[test]
2001 fn a_cast_between_a_pointer_and_an_integer_leaves_the_value_where_it_is() {
2002 let text = asm("long f(void *p) { return (long)p; }\n");
2003 for line in text.lines().filter(|line| line.starts_with('\t') && !line.contains('.')) {
2008 let mnemonic = line.split_whitespace().next().unwrap_or("");
2009 assert!(matches!(mnemonic, "movq" | "ret"), "{line} in\n{text}");
2010 }
2011 }
2012
2013 #[test]
2017 fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
2018 let six = "long a, long b, long c, long d, long e, long f";
2019 let text = asm(&format!("long f({six}, long g, long h) {{ return g + h; }}\n"));
2020
2021 assert!(text.contains("\tmovq\t8(%rsp), "), "{text}");
2025 assert!(text.contains("\tmovq\t16(%rsp), "), "{text}");
2026
2027 let narrow = asm(&format!("int f({six}, int g) {{ return g; }}\n"));
2031 assert!(narrow.contains("\tmovl\t8(%rsp), "), "{narrow}");
2032 let eight =
2033 "double a, double b, double c, double d, double e, double f, double g, double h";
2034 let float = asm(&format!("double f({eight}, double i) {{ return i; }}\n"));
2035 assert!(float.contains("\tmovsd\t8(%rsp), "), "{float}");
2036 }
2037
2038 #[test]
2041 fn a_call_writes_the_arguments_with_no_register_left_at_the_stack_pointer() {
2042 let six = "1, 2, 3, 4, 5, 6";
2043 let decl = "long g(long, long, long, long, long, long, long, long);\n";
2044 let text = asm(&format!("{decl}long f(void) {{ return g({six}, 7, 8); }}\n"));
2045
2046 assert!(text.contains("\tmovq\t%"), "{text}");
2047 assert!(text.contains(", (%rsp)\n"), "{text}");
2048 assert!(text.contains(", 8(%rsp)\n"), "{text}");
2049 assert!(text.contains("\tsubq\t$"), "{text}");
2051
2052 let narrow = "int g(int, int, int, int, int, int, int);\n";
2054 let text = asm(&format!("{narrow}int f(void) {{ return g({six}, 7); }}\n"));
2055 assert!(text.contains("\tmovl\t%"), "{text}");
2056 assert!(text.contains(", (%rsp)\n"), "{text}");
2057 }
2058
2059 #[test]
2062 fn a_variadic_call_counts_registers_and_not_arguments() {
2063 let nine = "1., 2., 3., 4., 5., 6., 7., 8., 9.";
2064 let decl = "int g(int, ...);\n";
2065 let text = asm(&format!("{decl}int f(void) {{ return g(0, {nine}); }}\n"));
2066
2067 assert!(text.contains("\tmovl\t$8, "), "eight registers, not nine: {text}");
2068 assert!(text.contains("\tmovsd\t%"), "{text}");
2069 assert!(text.contains(", (%rsp)\n"), "{text}");
2070 }
2071
2072 #[test]
2077 fn a_variadic_function_writes_the_argument_registers_it_was_handed_into_its_frame() {
2078 let body =
2079 "__builtin_va_list ap; __builtin_va_start(ap, n); __builtin_va_end(ap); return n;";
2080 let text = asm(&format!("int f(int n, ...) {{ {body} }}\n"));
2081
2082 let stores = |mnemonic: &str| text.matches(&format!("\t{mnemonic}\t%")).count();
2085 assert!(text.contains(", 8(%r"), "the second slot, not the first: {text}");
2086 assert!(!text.contains(", 0(%r"), "{text}");
2087 assert_eq!(stores("movsd"), 8, "every vector register: {text}");
2088
2089 assert!(text.contains("\tsubq\t$"), "{text}");
2091 }
2092
2093 #[test]
2096 fn va_start_writes_the_four_fields_the_psabi_describes() {
2097 let start = "__builtin_va_list ap; __builtin_va_start(ap, d);";
2098 let params = "int a, int b, int c, double d";
2099 let text = asm(&format!("int f({params}, ...) {{ {start} return a; }}\n"));
2100
2101 assert!(text.contains(" movl $24, "), "{text}");
2105 assert!(text.contains(" movl $64, "), "{text}");
2106 assert!(text.contains(", 8(%r"), "{text}");
2110 assert!(text.contains(", 16(%r"), "{text}");
2111 let frame: u32 = text
2112 .lines()
2113 .find_map(|line| line.trim().strip_prefix("subq $")?.split(',').next()?.parse().ok())
2114 .expect("a variadic function takes a frame for the save area");
2115 let above = |line: &str| {
2116 let at: u32 = line.trim().strip_prefix("leaq ")?.split('(').next()?.parse().ok()?;
2117 Some(at > frame)
2118 };
2119 assert!(text.lines().filter_map(above).any(|it| it), "{frame}: {text}");
2120 }
2121
2122 #[test]
2125 fn va_arg_branches_on_whether_the_argument_is_still_in_the_save_area() {
2126 let read = "__builtin_va_list ap; __builtin_va_start(ap, n);";
2127 let ints = format!("int f(int n, ...) {{ {read} return __builtin_va_arg(ap, int); }}\n");
2128 let text = asm(&ints);
2129
2130 assert!(text.contains("$40, "), "{text}");
2133 assert!(text.contains(" cmpl "), "{text}");
2134 assert!(text.contains(" ja "), "{text}");
2138
2139 let arg = "__builtin_va_arg(ap, double)";
2140 let text = asm(&format!("double f(int n, ...) {{ {read} return {arg}; }}\n"));
2141 assert!(text.contains("$160, "), "the last vector slot: {text}");
2142 }
2143
2144 #[test]
2147 fn a_structure_assignment_is_a_move_for_each_word_of_it() {
2148 let decl = "struct pair { long a, b; };\n";
2149 let body = "struct pair p = *q; return p.a + p.b;";
2150 let text = asm(&format!("{decl}long f(struct pair *q) {{ {body} }}\n"));
2151
2152 assert!(!text.contains("memcpy"), "nothing calls the library: {text}");
2153 assert!(!text.contains("\tcall"), "{text}");
2154 assert!(text.matches("\tmovq\t").count() >= 4, "two words each way: {text}");
2156 }
2157
2158 #[test]
2161 fn how_wide_a_word_of_a_copy_is_follows_the_alignment() {
2162 let decl = "struct bytes { char a[8]; };\n";
2163 let body = "struct bytes p = *q; return p.a[0];";
2164 let text = asm(&format!("{decl}int f(struct bytes *q) {{ {body} }}\n"));
2165
2166 assert!(text.matches("\tmovb\t").count() >= 16, "a byte at a time: {text}");
2168 }
2169
2170 #[test]
2173 fn the_part_of_an_initialiser_that_names_nothing_is_stored_as_zero() {
2174 let decl = "struct wide { long a, b, c; };\n";
2175 let text = asm(&format!("{decl}long f(void) {{ struct wide w = {{ 7 }}; return w.c; }}\n"));
2176
2177 assert!(!text.contains("memset"), "nothing calls the library: {text}");
2178 assert!(text.contains("\tmovq\t$0, ") || text.contains("$0, %"), "the zero: {text}");
2179 }
2180
2181 #[test]
2184 fn a_copy_too_large_to_unroll_calls_the_runtime() {
2185 let decl = "struct huge { char a[4096]; };\n";
2186 let mut opts = options();
2187 opts.emit = EmitKind::Asm;
2188 let source = format!("{decl}void f(struct huge *p, struct huge *q) {{ *p = *q; }}\n");
2189 let result = run(&opts, &source);
2190 assert!(!result.failed(), "{:?}", result.messages);
2191 let text = result.text();
2192 assert!(text.contains("call") && text.contains("memcpy"), "{text}");
2193 assert!(text.contains("4096"), "the size travels: {text}");
2196 }
2197
2198 #[test]
2201 fn a_realigned_frame_reads_them_through_the_frame_pointer() {
2202 let six = "long a, long b, long c, long d, long e, long f";
2203 let body = "_Alignas(32) long wide[4]; wide[0] = g; return wide[0];";
2204 let text = asm(&format!("long f({six}, long g) {{ {body} }}\n"));
2205
2206 assert!(text.contains("\tandq\t$-32, %rsp"), "{text}");
2210 assert!(text.contains("\tmovq\t16(%rbp), "), "{text}");
2211 assert!(!text.contains("\tmovq\t16(%rsp), "), "{text}");
2212 }
2213
2214 #[test]
2216 fn the_target_decides_how_the_assembly_is_spelled() {
2217 let mut opts = options();
2218 opts.emit = EmitKind::Asm;
2219 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2220 let text = run(&opts, "int f(void) { return 0; }\n").text().to_owned();
2221 assert!(text.contains("__TEXT,__text"), "{text}");
2222 assert!(text.contains("\n_f:\n"), "{text}");
2223 assert!(!text.contains(".note.GNU-stack"), "{text}");
2224 }
2225
2226 fn obj(source: &str) -> Vec<u8> {
2228 let mut opts = options();
2229 opts.emit = EmitKind::Object;
2230 let result = run(&opts, source);
2231 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2232 match result.artifact {
2233 Artifact::Object(bytes) => bytes,
2234 other => panic!("expected an object, got {other:?}"),
2235 }
2236 }
2237
2238 #[test]
2244 fn a_function_goes_from_c_to_an_object_a_linker_would_take() {
2245 let bytes = obj("int add(int a, int b) { return a + b; }\n");
2246 assert_eq!(&bytes[..4], b"\x7fELF", "an object file starts by saying it is one");
2247 let text = asm("int add(int a, int b) { return a + b; }\n");
2248 assert!(
2249 text.contains("\taddl\t"),
2250 "and the listing of it is the same instructions:\n{text}"
2251 );
2252 }
2253
2254 #[test]
2256 fn a_variable_goes_from_c_to_the_section_it_belongs_in() {
2257 let text = asm("int counter = 42;\nstatic int hidden;\nconst int fixed = 7;\n");
2258 assert!(text.contains("\t.data\n\t.globl\tcounter\n"), "{text}");
2259 assert!(text.contains("\ncounter:\n\t.long\t42\n"), "{text}");
2260 assert!(text.contains("\t.size\tcounter, .-counter\n"), "{text}");
2261 assert!(text.contains("\t.bss\n\t.p2align\t2\n"), "{text}");
2264 assert!(text.contains("\nhidden:\n\t.space\t4\n"), "{text}");
2265 assert!(!text.contains(".globl\thidden"), "{text}");
2266 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2269 }
2270
2271 #[test]
2278 fn a_bit_field_initializer_writes_every_byte_of_the_value_and_not_only_the_ones_that_are_set() {
2279 let text = asm("struct s { unsigned f : 20; } x = { 0x12300 };\n");
2280 assert!(text.contains("\t.data\n"), "there is something to write: {text}");
2281 assert!(text.contains("\nx:\n\t.ascii\t\"\\000#\\001\"\n"), "and it is the value: {text}");
2282
2283 let text = asm("struct s { unsigned a : 8; unsigned b : 8; } x = { 0, 3 };\n");
2286 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\003\"\n"), "{text}");
2287
2288 let text = asm("struct s { unsigned long long f : 40; } x = { 0x100000 };\n");
2291 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\000\\020\"\n\t.space\t5\n"), "{text}");
2292
2293 let text = asm("struct s { unsigned f : 20; } x = { 0 };\n");
2295 assert!(text.contains("\t.bss\n"), "an object of zeroes is zeroes: {text}");
2296 assert!(text.contains("\nx:\n\t.space\t4\n"), "{text}");
2297 }
2298
2299 #[test]
2301 fn a_string_literal_is_a_variable_with_a_name_no_program_could_write() {
2302 let text = asm("const char *f(void) { return \"hi\"; }\n");
2303 assert!(text.contains("\t.ascii\t\"hi\\000\"\n"), "{text}");
2304 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2305 let label = text
2306 .lines()
2307 .find(|line| line.starts_with(".Lstr"))
2308 .unwrap_or_else(|| panic!("a label for the literal in\n{text}"));
2309 assert!(!text.contains(&format!(".globl\t{}", label.trim_end_matches(':'))), "{text}");
2310 }
2311
2312 #[test]
2314 fn an_address_in_an_initializer_is_left_to_the_linker() {
2315 let source = "int counter;\nint *p = &counter;\n";
2316 let text = asm(source);
2317 assert!(text.contains("\np:\n\t.quad\tcounter\n"), "{text}");
2318 let bytes = obj(source);
2321 assert!(bytes.windows(8).any(|w| w == b"counter\0"), "the object has to name it");
2322 }
2323
2324 #[test]
2333 fn a_constant_holding_an_address_goes_in_the_section_the_loader_may_write_once() {
2334 let text = asm("static void a(void) {}\nstatic void b(void) {}\n\
2337 struct m { void (*x)(void); void (*y)(void); };\n\
2338 const struct m t = { a, b };\n");
2339 assert!(text.contains("\t.section\t.data.rel.ro.local,\"aw\",@progbits\n"), "{text}");
2340 assert!(text.contains("\nt:\n\t.quad\ta\n\t.quad\tb\n"), "{text}");
2341
2342 let text =
2345 asm("void a(void);\nstruct m { void (*x)(void); };\nconst struct m t = { a };\n");
2346 assert!(text.contains("\t.section\t.data.rel.ro,\"aw\",@progbits\n"), "{text}");
2347
2348 let text = asm("const int fixed = 7;\n");
2350 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2351 }
2352
2353 #[test]
2355 fn a_thread_local_variable_is_reported_as_work_that_is_not_done() {
2356 let mut opts = options();
2357 opts.emit = EmitKind::Asm;
2358 let result = run(&opts, "_Thread_local int x = 1;\n");
2359 assert!(result.failed(), "every thread sharing one variable is worse than a message");
2360 assert!(result.messages.iter().any(|m| m.contains("thread-local")), "{:?}", result);
2361 assert!(!result.messages.iter().any(|m| m.contains("internal")), "{:?}", result);
2363 }
2364
2365 #[test]
2367 fn the_object_and_the_listing_are_two_spellings_of_one_compilation() {
2368 let source = "int callee(void); int g(void) { return callee(); }\n";
2372 let bytes = obj(source);
2373 assert!(
2374 bytes.windows(7).any(|w| w == b"callee\0"),
2375 "the object has to name the callee for the linker to find it"
2376 );
2377 let text = asm(source);
2378 assert!(text.contains("\tcall\tcallee\n"), "{text}");
2379 }
2380
2381 #[test]
2387 fn compiling_for_an_executable_produces_an_object_and_not_a_dump() {
2388 let mut opts = options();
2389 opts.emit = EmitKind::Executable;
2391 let result = run(&opts, "int main(void) { return 0; }\n");
2392 assert_eq!(result.messages, Vec::<String>::new());
2393 match result.artifact {
2394 Artifact::Object(bytes) => assert_eq!(&bytes[..4], b"\x7fELF"),
2395 other => panic!("expected an object, got {other:?}"),
2396 }
2397 }
2398
2399 #[test]
2401 fn a_platform_with_no_object_writer_is_said_so_rather_than_written_as_elf() {
2402 let mut opts = options();
2403 opts.emit = EmitKind::Object;
2404 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2405 let result = run(&opts, "int f(void) { return 0; }\n");
2406 assert!(result.failed(), "an object nobody can read is worse than a message");
2407 assert!(
2408 result.messages.iter().any(|m| m.contains("no object writer")),
2409 "{:?}",
2410 result.messages
2411 );
2412 }
2413
2414 fn ir(source: &str) -> String {
2416 let mut opts = options();
2417 opts.emit = EmitKind::Ir;
2418 let result = run(&opts, source);
2419 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2420 result.text().to_owned()
2421 }
2422
2423 fn errors(source: &str) -> Vec<String> {
2425 let mut opts = options();
2426 opts.emit = EmitKind::Ir;
2427 let result = run(&opts, source);
2428 assert!(result.failed(), "expected this to be refused:\n{source}");
2429 result.messages
2430 }
2431
2432 fn body(source: &str) -> String {
2434 let text = ir(source);
2435 let (_, rest) = text.split_once("{\n").expect("a function definition");
2436 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
2437 body.to_owned()
2438 }
2439
2440 #[test]
2448 fn gnu89_inline_is_what_decides_whether_a_bare_inline_definition_reaches_the_module() {
2449 let source = "inline int f(int x) { return x + 1; }\n";
2450 let with = |flag: bool| {
2451 let mut opts = options();
2452 opts.emit = EmitKind::Ir;
2453 opts.gnu89_inline = flag;
2454 let result = run(&opts, source);
2455 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
2456 result.text().to_owned()
2457 };
2458
2459 assert!(!with(false).contains("block0"), "no body: {}", with(false));
2462
2463 assert!(with(true).contains("block0"), "a body: {}", with(true));
2466 }
2467
2468 #[test]
2482 fn lowering_says_nothing_about_the_type_an_access_went_through() {
2483 let source = "\
2486struct s { int a; float b; };\n\
2487union u { int i; float f; };\n\
2488int scalar(int *p) { return *p; }\n\
2489float member(struct s *p) { p->a = 1; return p->b; }\n\
2490int element(int *a, long i) { return a[i]; }\n\
2491float through_a_union(union u *p) { p->i = 1; return p->f; }\n";
2492 assert!(!ir(source).contains("tbaa"), "{}", ir(source));
2493 }
2494
2495 #[test]
2503 fn a_bare_return_from_a_function_that_promised_a_value_gives_back_a_zero() {
2504 let mut opts = options();
2505 opts.emit = EmitKind::Ir;
2506 opts.std = Std::C89;
2507 let compiled = |source: &str| {
2508 let result = run(&opts, source);
2509 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
2510 result.text().to_owned()
2511 };
2512
2513 let text = compiled("int f(int x) { if (x) return; return 3; }\n");
2514 assert!(text.contains("iconst.i32 0\n return"), "zero goes back: {text}");
2515 assert!(!text.contains("unreachable"), "the branch that reached it is kept: {text}");
2516
2517 let text = compiled("double f(int x) { if (x) return; return 1.0; }\n");
2519 assert!(text.contains("fconst.f64 0x0\n return"), "a float zero goes back: {text}");
2520 }
2521
2522 #[test]
2530 fn a_call_to_a_name_nothing_declared_declares_it_as_c89_said_to() {
2531 let mut opts = options();
2532 opts.emit = EmitKind::Ir;
2533 opts.std = Std::C89;
2534 let compiled = |source: &str| {
2535 let result = run(&opts, source);
2536 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
2537 result.text().to_owned()
2538 };
2539
2540 let text = compiled("int f(void) { return g(); }\n");
2542 assert!(text.contains("call @g"), "the call is to the name that was written: {text}");
2543 assert!(text.contains("i32"), "and it gives back an int: {text}");
2544
2545 let text = compiled("int f(char c) { return g(c); }\n");
2548 assert!(text.contains("sext.i32"), "the argument is promoted: {text}");
2549
2550 let mut opts = options();
2553 opts.std = Std::C89;
2554 let said = run(&opts, "int f(void) { return h; }\n").messages.join("\n");
2555 assert!(said.contains("'h' undeclared"), "not a call, so not declared: {said}");
2556 }
2557
2558 #[test]
2568 fn a_name_called_before_it_is_defined_still_gets_its_definition() {
2569 let mut opts = options();
2570 opts.emit = EmitKind::Ir;
2571 opts.std = Std::C89;
2572 let text = run(&opts, "int f(void) { return dummy(); }\ndummy () { return 7; }\n")
2573 .text()
2574 .to_owned();
2575 assert!(text.contains("func @f()"), "the caller is there: {text}");
2576 assert!(text.contains("func @dummy"), "and so is what it calls: {text}");
2577 assert!(text.contains("iconst.i32 7"), "with the body it was given: {text}");
2578 }
2579
2580 #[test]
2588 fn an_old_style_parameter_is_converted_from_what_the_call_promoted_it_to() {
2589 let mut opts = options();
2590 opts.emit = EmitKind::Ir;
2591 opts.std = Std::C89;
2592 let compiled = |source: &str| run(&opts, source).text().to_owned();
2593
2594 let text = compiled("f (c) unsigned char c; { return c; }\n");
2595 assert!(text.contains("func @f(i32"), "an int arrives: {text}");
2596 assert!(text.contains("trunc.i8"), "and is cut down to what was declared: {text}");
2597 assert!(text.contains("zext.i32"), "then read back unsigned: {text}");
2598
2599 let text = compiled("f (s) short s; { return s; }\n");
2601 assert!(text.contains("trunc.i16"), "cut down: {text}");
2602 assert!(text.contains("sext.i32"), "and read back signed: {text}");
2603
2604 let text = compiled("f (x) float x; { return x * 2; }\n");
2607 assert!(text.contains("func @f(f64"), "a double arrives: {text}");
2608 assert!(text.contains("fptrunc.f32"), "and is narrowed to the float: {text}");
2609
2610 let text = compiled("int f(unsigned char c) { return c; }\n");
2613 assert!(text.contains("func @f(i8)"), "the declared type arrives: {text}");
2614 assert!(!text.contains("trunc"), "so there is nothing to cut down: {text}");
2615 }
2616
2617 #[test]
2626 fn the_rules_gcc_promoted_are_decided_by_the_dialect_and_by_fpermissive() {
2627 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
2629 let cases = [
2630 ("static counted;\n", ["", "error", "warning", "error"]),
2631 ("int f(void) { return g(); }\n", ["", "error", "warning", "error"]),
2632 ("int f(x) { return x; }\n", ["", "error", "warning", "error"]),
2633 ("int *p;\nvoid h(void) { p = 1; }\n", ["warning", "error", "warning", "error"]),
2634 (
2635 "char *q;\nint *r;\nvoid k(void) { r = q; }\n",
2636 ["warning", "error", "warning", "error"],
2637 ),
2638 ("int f(void) { return; }\n", ["", "error", "warning", "error"]),
2639 ("void g(void) { return 1; }\n", ["warning", "error", "warning", "error"]),
2640 ];
2641
2642 for (source, wanted) in cases {
2643 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
2644 let mut opts = options();
2645 opts.std = std;
2646 opts.permissive = permissive;
2647 let said = run(&opts, source).messages.join("\n");
2648 let severity = if said.contains(": error: ") {
2649 "error"
2650 } else if said.contains(": warning: ") {
2651 "warning"
2652 } else {
2653 ""
2654 };
2655 let how = if permissive { " -fpermissive" } else { "" };
2656 assert_eq!(
2657 severity,
2658 wanted,
2659 "under -std={}{how}, {source} was answered with `{said}`",
2660 std.as_str()
2661 );
2662 if wanted.is_empty() {
2663 assert!(said.is_empty(), "nothing to say, but said `{said}`");
2664 }
2665 }
2666 }
2667 }
2668
2669 #[test]
2678 fn the_three_variadic_builtins_answer_a_bad_list_the_way_a_call_answers_a_bad_argument() {
2679 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
2680 let cases = [
2681 (
2682 "int f(int n, ...) { char *p; return __builtin_va_arg(p, int); }\n",
2683 "first argument to 'va_arg' not of type 'va_list'",
2684 ["error", "error", "error", "error"],
2685 ),
2686 (
2687 "void f(int n, ...) { char *p; __builtin_va_start(p, n); }\n",
2688 "passing argument 1 of '__builtin_va_start' from incompatible pointer type",
2689 ["warning", "error", "warning", "error"],
2690 ),
2691 (
2692 "void f(int n, ...) { int x; __builtin_va_end(x); }\n",
2693 "passing argument 1 of '__builtin_va_end' makes pointer from integer without a \
2694 cast",
2695 ["warning", "error", "warning", "error"],
2696 ),
2697 (
2698 "void f(int n, ...) { __builtin_va_list a; char *p; __builtin_va_copy(a, p); }\n",
2699 "passing argument 2 of '__builtin_va_copy' from incompatible pointer type",
2700 ["warning", "error", "warning", "error"],
2701 ),
2702 ];
2703
2704 for (source, message, wanted) in cases {
2705 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
2706 let mut opts = options();
2707 opts.std = std;
2708 opts.permissive = permissive;
2709 let said = run(&opts, source).messages.join("\n");
2710 let how = if permissive { " -fpermissive" } else { "" };
2711 assert!(
2712 said.contains(&format!(": {wanted}: {message}")),
2713 "under -std={}{how}, {source} was answered with `{said}`",
2714 std.as_str()
2715 );
2716 }
2717 }
2718 }
2719
2720 fn safe_ir(tier: rucc_session::Safety, source: &str) -> String {
2722 let mut opts = options();
2723 opts.emit = EmitKind::Ir;
2724 opts.safety = tier;
2725 let result = run(&opts, source);
2726 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2727 result.text().to_owned()
2728 }
2729
2730 const READS_THROUGH_A_POINTER: &str = "int read(int *p) { return p[1]; }\n";
2731
2732 #[test]
2733 fn a_build_that_did_not_ask_for_the_monitor_is_compiled_the_way_it_always_was() {
2734 let text = ir(READS_THROUGH_A_POINTER);
2738 assert!(!text.contains("check_"), "{text}");
2739 assert!(!text.contains("cap_of"), "{text}");
2740 }
2741
2742 #[test]
2743 fn asking_for_a_tier_puts_the_checks_in_before_the_optimizer_sees_them() {
2744 let text = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2745 assert!(text.contains("cap_of"), "{text}");
2746 assert!(text.contains("check_bounds"), "{text}");
2747 assert!(text.contains("check_live"), "{text}");
2748 assert!(text.contains("check_deriv"), "{text}");
2750 }
2751
2752 #[test]
2753 fn the_three_tiers_that_are_not_off_all_check_the_same_accesses_so_far() {
2754 let detect = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2758 for tier in [rucc_session::Safety::Enforce, rucc_session::Safety::Kernel] {
2759 assert_eq!(safe_ir(tier, READS_THROUGH_A_POINTER), detect, "{tier}");
2760 }
2761 }
2762
2763 fn summary(tier: rucc_session::Safety, source: &str) -> String {
2765 let mut opts = options();
2766 opts.emit = EmitKind::SafetySummary;
2767 opts.safety = tier;
2768 let result = run(&opts, source);
2769 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2770 result.text().to_owned()
2771 }
2772
2773 #[test]
2774 fn the_summary_counts_the_checks_that_went_in_and_the_ones_still_standing() {
2775 let text = summary(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2776 assert!(text.contains("\"tier\": \"detect\""), "{text}");
2777 assert!(
2779 text.contains("\"bounds\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"),
2780 "{text}"
2781 );
2782 assert!(
2783 text.contains(
2784 "\"derivation\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"
2785 ),
2786 "{text}"
2787 );
2788 }
2789
2790 #[test]
2791 fn a_build_without_the_monitor_summarises_as_a_build_with_no_checks_in_it() {
2792 let text = summary(rucc_session::Safety::Off, READS_THROUGH_A_POINTER);
2796 assert!(text.contains("\"tier\": \"off\""), "{text}");
2797 assert!(
2798 text.contains("\"bounds\": { \"emitted\": 0, \"remaining\": 0, \"discharged\": 0 }"),
2799 "{text}"
2800 );
2801 }
2802
2803 #[test]
2804 fn a_call_the_boundary_models_is_counted_apart_from_one_it_does_not() {
2805 let text = summary(
2806 rucc_session::Safety::Detect,
2807 "void *memcpy(void *, const void *, unsigned long);\n\
2808 int puts(const char *);\n\
2809 void f(char *d, char *s) { memcpy(d, s, 4); puts(d); }\n",
2810 );
2811 assert!(text.contains("\"interposed\": 1"), "{text}");
2812 assert!(text.contains("\"puts\""), "{text}");
2813 assert!(!text.contains("__rucc_wrap_memcpy\""), "{text}");
2817 }
2818
2819 #[test]
2820 fn the_two_directions_a_pointer_crosses_the_boundary_are_counted_apart() {
2821 let text = summary(
2825 rucc_session::Safety::Detect,
2826 "void *notes_open(void);\n\
2827 char *f(char *p) { char *q = notes_open(); return q ? q : p; }\n",
2828 );
2829 assert!(text.contains("\"crossings\": { \"entered\": 1, \"returned\": 1 }"), "{text}");
2830 assert!(text.contains("\"notes_open\""), "{text}");
2831 }
2832
2833 #[test]
2834 fn a_static_function_nobody_takes_the_address_of_is_not_a_crossing() {
2835 let text = summary(
2838 rucc_session::Safety::Detect,
2839 "static int len(const char *p) { return p ? 1 : 0; }\n\
2840 int f(void) { return len(\"x\"); }\n",
2841 );
2842 assert!(text.contains("\"crossings\": { \"entered\": 0, \"returned\": 0 }"), "{text}");
2843 }
2844
2845 fn granules(source: &str) -> String {
2847 let mut opts = options();
2848 opts.emit = EmitKind::TypeGranules;
2849 let result = run(&opts, source);
2850 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2851 result.text().to_owned()
2852 }
2853
2854 #[test]
2855 fn the_granule_report_names_every_record_and_both_keyings() {
2856 let text = granules(
2857 "struct hot { char *p; int a; int b; };\n\
2858 int f(struct hot *h) { return h->a; }\n",
2859 );
2860 assert!(text.contains("struct hot"), "{text}");
2861 assert!(text.contains("every type distinct"), "{text}");
2864 assert!(text.contains("every pointer one type"), "{text}");
2865 assert!(text.contains("budget"), "{text}");
2866 }
2867
2868 #[test]
2869 fn a_record_nothing_uses_is_still_measured() {
2870 let text = granules("struct unused { long a; double b; };\nint f(void) { return 0; }\n");
2873 assert!(text.contains("struct unused"), "{text}");
2874 }
2875
2876 #[test]
2877 fn the_granule_report_stops_before_anything_is_lowered() {
2878 let text = granules(
2882 "struct wide { long double d; };\n\
2883 long double f(long double x) { return x * x; }\n",
2884 );
2885 assert!(text.contains("struct wide"), "{text}");
2886 }
2887
2888 #[test]
2889 fn a_witness_reaches_the_assembler_as_a_call_to_the_runtime() {
2890 let text = safe_asm(rucc_session::Safety::Detect, "char *f(char *p) { return p; }\n");
2893 assert!(text.contains("\tcall\t__rucc_cap_witness\n"), "{text}");
2894 }
2895
2896 #[test]
2897 fn a_pointer_turned_into_an_integer_is_on_the_trust_set() {
2898 let text = summary(
2899 rucc_session::Safety::Detect,
2900 "unsigned long f(int *p) { return (unsigned long) p; }\n",
2901 );
2902 assert!(text.contains("\"exposed\": 1"), "{text}");
2903 }
2904
2905 fn safe_asm(tier: rucc_session::Safety, source: &str) -> String {
2907 let mut opts = options();
2908 opts.emit = EmitKind::Asm;
2909 opts.safety = tier;
2910 let result = run(&opts, source);
2911 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2912 result.text().to_owned()
2913 }
2914
2915 #[test]
2916 fn a_check_reaches_the_assembler_as_a_call_to_the_runtime() {
2917 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2918 assert!(text.contains("\tcall\t__rucc_check_bounds\n"), "{text}");
2919 assert!(text.contains("\tcall\t__rucc_check_live\n"), "{text}");
2920 assert!(text.contains("\tcall\t__rucc_check_deriv\n"), "{text}");
2921 }
2922
2923 #[test]
2924 fn every_check_that_reached_the_assembler_has_a_row_describing_it() {
2925 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2929 let section = format!("\t.section\t{},", rucc_safety::SECTION);
2930 assert_eq!(text.matches(§ion).count(), 3, "{text}");
2931 for index in 0..3 {
2932 let name = format!("__rucc_safety_desc_{index}");
2933 assert!(text.contains(&format!("{name}:\n")), "{text}");
2936 assert!(text.contains(&format!("{name}(%rip)")), "{text}");
2937 }
2938 assert!(!text.contains("__rucc_safety_desc_3"), "{text}");
2939 }
2940
2941 #[test]
2949 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
2950 let text = ir(concat!(
2951 "int g;\n",
2952 "int a = __builtin_constant_p(1);\n",
2953 "int b = __builtin_constant_p(g);\n",
2954 "int c = __builtin_constant_p(\"abc\");\n",
2955 "int d = __builtin_constant_p(&g);\n",
2956 "int e = __builtin_constant_p(1.5);\n",
2957 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
2958 ));
2959 assert!(text.contains("global @a : i32 = 1,"), "{text}");
2960 assert!(text.contains("global @b : i32 = 0,"), "{text}");
2961 assert!(text.contains("global @c : i32 = 1,"), "{text}");
2962 assert!(text.contains("global @d : i32 = 0,"), "{text}");
2963 assert!(text.contains("global @e : i32 = 1,"), "{text}");
2964 assert!(text.contains("global @h : i32 = 11,"), "{text}");
2965 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
2966
2967 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
2971 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
2972 }
2973
2974 #[test]
2983 fn a_call_to_a_library_builtin_reaches_the_library_function() {
2984 let text = body("void f(void) { __builtin_abort(); }\n");
2985 assert_eq!(text, "block0:\n call @abort() : ()\n return\n");
2986
2987 let text = ir("int f(const char *s) { return __builtin_puts(s) + __builtin_strlen(s); }\n");
2990 assert!(text.contains("call @puts(%0) : (ptr) -> i32"), "{text}");
2991 assert!(text.contains("call @strlen(%0) : (ptr) -> i64"), "{text}");
2992 assert!(!text.contains("__builtin_"), "the prefix is not part of any name here:\n{text}");
2993 }
2994
2995 #[test]
3006 fn the_absolute_value_family_is_the_magnitude_and_not_a_call() {
3007 let text = body(concat!(
3008 "long long llabs(long long);\n",
3009 "long long f(long long x) { return llabs(x); }\n",
3010 ));
3011 assert!(text.contains("%1 = iconst.i64 63"), "{text}");
3012 assert!(text.contains("%2 = ashr %0, %1"), "{text}");
3013 assert!(text.contains("%3 = xor %0, %2"), "{text}");
3014 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3015 assert!(!text.contains("call"), "the call does not happen:\n{text}");
3016
3017 let text = body("int abs(int);\nint f(int x) { return abs(x); }\n");
3020 assert!(text.contains("iconst.i32 31"), "{text}");
3021 let text = body("long labs(long);\nlong f(long x) { return labs(x); }\n");
3022 assert!(text.contains("iconst.i64 63"), "{text}");
3023
3024 let text = body("long long f(long long x) { return __builtin_llabs(x); }\n");
3027 assert!(!text.contains("call"), "{text}");
3028
3029 let text = ir(concat!(
3031 "long long llabs(long long b);\n",
3032 "long long g(long long x) { return llabs(x); }\n",
3033 "long long llabs(long long b) { return 7; }\n",
3034 ));
3035 assert!(!text.contains("call @llabs"), "{text}");
3036 }
3037
3038 #[test]
3045 fn a_byte_swap_is_arithmetic_and_not_a_call() {
3046 let text = body("unsigned f(unsigned x) { return __builtin_bswap32(x); }\n");
3047 assert_eq!(text, "block0(%0: i32):\n %1 = bswap %0\n return %1\n");
3048
3049 let text = body("unsigned f(unsigned char c) { return __builtin_bswap32(c); }\n");
3052 assert!(text.contains("zext.i32 %0"), "widened first: {text}");
3053 assert!(text.contains("bswap %1"), "and swapped at four bytes: {text}");
3054 }
3055
3056 #[test]
3062 fn the_byte_swaps_reverse_at_the_width_their_name_says() {
3063 for (name, ty, width) in [
3064 ("__builtin_bswap16", "unsigned short", "i16"),
3065 ("__builtin_bswap32", "unsigned", "i32"),
3066 ("__builtin_bswap64", "unsigned long long", "i64"),
3067 ] {
3068 let source = format!("{ty} f({ty} x) {{ return {name}(x); }}\n");
3069 let text = body(&source);
3070 assert_eq!(
3071 text,
3072 format!("block0(%0: {width}):\n %1 = bswap %0\n return %1\n"),
3073 "{name}"
3074 );
3075 }
3076 }
3077
3078 #[test]
3085 fn the_bit_counts_are_instructions_and_not_calls() {
3086 let text = body("int f(unsigned x) { return __builtin_clz(x); }\n");
3087 assert_eq!(text, "block0(%0: i32):\n %1 = ctlz %0\n return %1\n");
3088
3089 let text = body("int f(unsigned x) { return __builtin_ctz(x); }\n");
3090 assert_eq!(text, "block0(%0: i32):\n %1 = cttz %0\n return %1\n");
3091
3092 let text = body("int f(unsigned x) { return __builtin_popcount(x); }\n");
3093 assert_eq!(text, "block0(%0: i32):\n %1 = ctpop %0\n return %1\n");
3094 }
3095
3096 #[test]
3105 fn the_bit_counts_ask_about_the_width_their_name_says() {
3106 let text = body("int f(unsigned long long x) { return __builtin_clzll(x); }\n");
3107 assert!(text.starts_with("block0(%0: i64):"), "counted at eight bytes: {text}");
3108 assert!(text.contains("%1 = ctlz %0"), "{text}");
3109 assert!(text.contains("trunc.i32 %1"), "and answered in an int: {text}");
3110
3111 let text = body("int f(unsigned long long x) { return __builtin_clz(x); }\n");
3114 assert!(text.contains("trunc.i32 %0"), "narrowed to what was asked about: {text}");
3115 assert!(text.contains("ctlz %1"), "and counted there: {text}");
3116
3117 let text = body("int f(unsigned long x) { return __builtin_popcountl(x); }\n");
3118 assert!(text.contains("%1 = ctpop %0"), "{text}");
3119 assert!(!text.contains("call"), "{text}");
3120 }
3121
3122 #[test]
3127 fn a_parity_is_the_low_bit_of_the_set_bit_count() {
3128 let text = body("int f(unsigned x) { return __builtin_parity(x); }\n");
3129 assert!(text.contains("%1 = ctpop %0"), "{text}");
3130 assert!(text.contains("iconst.i32 1"), "{text}");
3131 assert!(text.contains("and %1, %2"), "the low bit of it: {text}");
3132 }
3133
3134 #[test]
3140 fn the_first_set_bit_is_one_based_and_zero_for_a_zero() {
3141 let text = body("int f(int x) { return __builtin_ffs(x); }\n");
3142 assert!(text.contains("%1 = cttz %0"), "{text}");
3143 assert!(text.contains("%4 = add %1, %2"), "one more than the count: {text}");
3144 assert!(text.contains("%5 = icmp ne %0, %3"), "whether there was a bit at all: {text}");
3145 assert!(text.contains("%7 = sub %3, %6"), "spread to a mask: {text}");
3146 assert!(text.contains("%8 = and %4, %7"), "and kept only then: {text}");
3147 assert!(!text.contains("br_if"), "no branch: {text}");
3148 }
3149
3150 #[test]
3160 fn an_overflow_check_is_arithmetic_and_not_a_call() {
3161 let text =
3162 body("int f(int a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3163 assert!(text.contains("%3, %4 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
3164 assert!(text.contains("store %3 -> %2"), "{text}");
3165 assert!(!text.contains("call"), "{text}");
3166
3167 let text =
3168 body("int f(int a, int b, int *r) { return __builtin_sub_overflow(a, b, r); }\n");
3169 assert!(text.contains("ssub_overflow.(i32, i1) %0, %1"), "{text}");
3170
3171 let text =
3172 body("int f(int a, int b, int *r) { return __builtin_mul_overflow(a, b, r); }\n");
3173 assert!(text.contains("smul_overflow.(i32, i1) %0, %1"), "{text}");
3174
3175 let text = body(
3178 "int f(unsigned a, unsigned b, unsigned *r) { return __builtin_add_overflow(a, b, r); }\n",
3179 );
3180 assert!(text.contains("uadd_overflow.(i32, i1) %0, %1"), "{text}");
3181 }
3182
3183 #[test]
3191 fn an_overflow_check_is_done_at_a_type_that_holds_every_operand() {
3192 let text = body(
3193 "int f(unsigned a, int b, long long *r) { return __builtin_add_overflow(a, b, r); }\n",
3194 );
3195 assert!(text.contains("%3 = zext.i64 %0"), "the unsigned operand keeps its value: {text}");
3196 assert!(text.contains("%4 = sext.i64 %1"), "and so does the signed one: {text}");
3197 assert!(text.contains("sadd_overflow.(i64, i1) %3, %4"), "{text}");
3198
3199 let text = body(
3202 "int f(long long a, long long b, long long *r) { return __builtin_mul_overflow(a, b, r); }\n",
3203 );
3204 assert!(text.contains("smul_overflow.(i64, i1) %0, %1"), "{text}");
3205 assert!(!text.contains("sext."), "{text}");
3206 assert!(!text.contains("zext.i64"), "{text}");
3208 }
3209
3210 #[test]
3218 fn an_overflow_check_writes_the_wrapped_answer_whether_or_not_it_fit() {
3219 let text =
3220 body("int f(int a, int b, char *r) { return __builtin_sub_overflow(a, b, r); }\n");
3221 assert!(text.contains("%3, %4 = ssub_overflow.(i32, i1) %0, %1"), "{text}");
3222 assert!(text.contains("%5 = trunc.i8 %3"), "narrowed to where it goes: {text}");
3223 assert!(text.contains("%6 = sext.i32 %5"), "and back: {text}");
3224 assert!(text.contains("%7 = icmp ne %6, %3"), "which is whether it fit: {text}");
3225 assert!(text.contains("store %5 -> %2"), "the narrowed value is stored either way: {text}");
3226 assert!(text.contains("%8 = or %4, %7"), "and either bit is an overflow: {text}");
3227 }
3228
3229 #[test]
3236 fn a_call_needing_more_than_sixty_four_bits_says_so() {
3237 let refused = concat!(
3238 "int f(unsigned long long a, long long b, long long *r) {\n",
3239 " return __builtin_add_overflow(a, b, r);\n",
3240 "}\n",
3241 );
3242 let messages = errors(refused);
3243 assert_eq!(messages.len(), 1, "{messages:?}");
3244 assert!(messages[0].contains("E0694"), "{messages:?}");
3245 assert!(messages[0].contains("wider than 64 bits"), "{messages:?}");
3246 }
3247
3248 #[test]
3251 fn an_overflow_check_over_something_that_is_not_an_integer_says_so() {
3252 let messages =
3253 errors("int f(double a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3254 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3255
3256 let messages =
3257 errors("int f(int a, int b, double *r) { return __builtin_add_overflow(a, b, r); }\n");
3258 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3259 }
3260
3261 #[test]
3272 fn an_ordered_access_is_ordered_in_the_ir() {
3273 let text = body("int f(int *p) { return __atomic_load_n(p, 0); }\n");
3274 assert!(text.contains("atomic_load.i32 %0, align 4, relaxed"), "{text}");
3275
3276 let text = body("long f(long *p) { return __atomic_load_n(p, 2); }\n");
3277 assert!(text.contains("atomic_load.i64 %0, align 8, acquire"), "{text}");
3278
3279 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
3280 assert!(text.contains("atomic_store %1 -> %0, align 4, release"), "{text}");
3281
3282 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
3283 assert!(text.contains("atomic_store %1 -> %0, align 4, seq_cst"), "{text}");
3284
3285 let text = body("void f(char *p, int v) { __atomic_store_n(p, v, 0); }\n");
3288 assert!(text.contains("trunc.i8 %1"), "{text}");
3289 assert!(text.contains("atomic_store %2 -> %0, align 1, relaxed"), "{text}");
3290 }
3291
3292 #[test]
3301 fn an_ordered_access_is_the_plain_instruction_on_this_machine() {
3302 let text = asm("int f(int *p) { return __atomic_load_n(p, 5); }\n");
3303 assert!(text.contains("movl\t(%rdi), %eax"), "{text}");
3304 assert!(!text.contains("mfence"), "a load needs no barrier here: {text}");
3305
3306 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
3307 assert!(text.contains("movl\t%esi, (%rdi)"), "{text}");
3308 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
3309
3310 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
3311 let (before, after) = text.split_once("mfence").expect("a barrier: {text}");
3312 assert!(before.contains("movl\t%esi, (%rdi)"), "the store comes first: {text}");
3313 assert!(!after.contains("movl"), "and nothing else is between them: {text}");
3314 }
3315
3316 #[test]
3326 fn a_barrier_is_one_instruction_at_the_strongest_ordering_and_none_below_it() {
3327 assert!(asm("void f(void) { __atomic_thread_fence(5); }\n").contains("mfence"));
3328 assert!(asm("void f(void) { __sync_synchronize(); }\n").contains("mfence"));
3329
3330 for weaker in ["1", "2", "3", "4"] {
3331 let source = format!("void f(void) {{ __atomic_thread_fence({weaker}); }}\n");
3332 assert!(!asm(&source).contains("mfence"), "{weaker} costs nothing here");
3333 }
3334 }
3335
3336 #[test]
3342 fn a_compare_and_exchange_is_one_instruction_answering_two_things() {
3343 let text =
3346 body("int f(int *p, int e, int d) { return __sync_val_compare_and_swap(p, e, d); }\n");
3347 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
3348 assert!(text.contains("return %3"), "the value it found: {text}");
3349
3350 let text =
3351 body("int f(int *p, int e, int d) { return __sync_bool_compare_and_swap(p, e, d); }\n");
3352 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
3353 assert!(text.contains("zext.i32 %4"), "whether it happened: {text}");
3354
3355 let text = body(
3358 "int f(int *p, int *e, int d) { return __atomic_compare_exchange_n(p, e, d, 0, 4, 2); }\n",
3359 );
3360 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3361 assert!(text.contains("%4, %5 = cmpxchg.(i32, i1) %0, %3, %2, align 4, acq_rel"), "{text}");
3362 assert!(text.contains("br_if %5, block2, block1"), "{text}");
3363 assert!(text.contains("store %4 -> %1, align 4"), "{text}");
3364
3365 let text = body(
3368 "int f(int *p, int *e, int *d) { return __atomic_compare_exchange(p, e, d, 0, 5, 5); }\n",
3369 );
3370 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3371 assert!(text.contains("%4 = load.i32 %2, align 4"), "{text}");
3372 assert!(text.contains("%5, %6 = cmpxchg.(i32, i1) %0, %3, %4, align 4, seq_cst"), "{text}");
3373 }
3374
3375 #[test]
3382 fn a_compare_and_exchange_is_a_locked_instruction_at_the_width_of_the_object() {
3383 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
3384 for (ty, suffix, reg) in widths {
3385 let source = format!(
3386 "int f({ty} *p, {ty} e, {ty} d) {{ return __sync_bool_compare_and_swap(p, e, d); }}\n"
3387 );
3388 let text = asm(&source);
3389 assert!(text.contains("\tlock\n"), "{ty}: {text}");
3390 assert!(text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3391 assert!(text.contains("sete\t"), "{ty}: {text}");
3392 }
3393 let source =
3394 "int f(long *p, long e, long d) { return __sync_bool_compare_and_swap(p, e, d); }\n";
3395 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
3396
3397 for order in ["0", "2", "3", "4", "5"] {
3401 let call = format!("__atomic_compare_exchange_n(p, e, d, 0, {order}, 0)");
3402 let source = format!("int f(int *p, int *e, int d) {{ return {call}; }}\n");
3403 let text = asm(&source);
3404 assert!(text.contains("cmpxchgl\t"), "{order}: {text}");
3405 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
3406 }
3407 }
3408
3409 #[test]
3421 fn a_read_modify_write_is_one_instruction_and_the_arithmetic_a_name_asks_for() {
3422 let text = body("int f(int *p, int v) { return __atomic_fetch_add(p, v, 5); }\n");
3423 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
3424 assert!(text.contains("return %2"), "the value that was there: {text}");
3425
3426 let text = body("int f(int *p, int v) { return __atomic_add_fetch(p, v, 5); }\n");
3427 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
3428 assert!(text.contains("%3 = add %2, %1"), "and the value afterwards: {text}");
3429
3430 let text = body("int f(int *p, int v) { return __atomic_sub_fetch(p, v, 5); }\n");
3431 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
3432 assert!(text.contains("%3 = sub %2, %1"), "{text}");
3433
3434 let text = body("int f(int *p, int v) { return __sync_fetch_and_sub(p, v); }\n");
3436 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
3437
3438 let text = body("int f(int *p, int v) { return __atomic_exchange_n(p, v, 5); }\n");
3441 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, seq_cst"), "{text}");
3442
3443 let text = body("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
3444 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, acquire"), "{text}");
3445
3446 let text = body("void f(int *p) { __sync_lock_release(p); }\n");
3449 assert!(text.contains("release"), "{text}");
3450 assert!(text.contains("%1 = iconst.i32 0"), "{text}");
3451
3452 let text = body("void f(int *p, int guard) { __sync_lock_release(p, guard); }\n");
3456 assert!(text.contains("%2 = iconst.i32 0"), "{text}");
3457 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
3458
3459 let text = body("int f(int *p, int v) { return __atomic_fetch_and(p, v, 5); }\n");
3462 assert!(text.contains("%2 = atomic_rmw.i32 and %0, %1, align 4, seq_cst"), "{text}");
3463
3464 let text = body("int f(int *p, int v) { return __sync_or_and_fetch(p, v); }\n");
3465 assert!(text.contains("%2 = atomic_rmw.i32 or %0, %1, align 4, seq_cst"), "{text}");
3466 assert!(text.contains("%3 = or %2, %1"), "and the value afterwards: {text}");
3467
3468 let text = body("int f(int *p, int v) { return __atomic_nand_fetch(p, v, 5); }\n");
3471 assert!(text.contains("%2 = atomic_rmw.i32 nand %0, %1, align 4, seq_cst"), "{text}");
3472 assert!(text.contains("%3 = and %2, %1"), "{text}");
3473 assert!(text.contains("%4 = iconst.i32 -1"), "{text}");
3474 assert!(text.contains("%5 = xor %3, %4"), "{text}");
3475 }
3476
3477 #[test]
3488 fn a_bitwise_read_modify_write_is_a_loop_around_the_compare_and_exchange() {
3489 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
3490 for (ty, suffix, reg) in widths {
3491 for (name, call, insn) in [
3492 ("and", "__atomic_fetch_and(p, v, 5)", "and"),
3493 ("or", "__sync_fetch_and_or(p, v)", "or"),
3494 ("xor", "__atomic_xor_fetch(p, v, 5)", "xor"),
3495 ] {
3496 let source = format!("{ty} f({ty} *p, {ty} v) {{ return {call}; }}\n");
3497 let text = asm(&source);
3498 assert!(text.contains("\tlock\n"), "{ty} {name}: {text}");
3499 assert!(
3500 text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")),
3501 "{ty} {name}: {text}"
3502 );
3503 assert!(text.contains(&format!("{insn}{suffix}\t")), "{ty} {name}: {text}");
3504 assert!(!text.contains("\txadd"), "{ty} {name} is not an add: {text}");
3506 assert!(!text.contains("\txchg"), "{ty} {name} is not an exchange: {text}");
3507 }
3508 }
3509 let source = "long f(long *p, long v) { return __atomic_fetch_or(p, v, 5); }\n";
3510 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
3511
3512 let text = asm("int f(int *p, int v) { return __sync_fetch_and_nand(p, v); }\n");
3516 assert!(text.contains("cmpxchgl\t"), "{text}");
3517 assert!(text.contains("andl\t"), "{text}");
3518 assert!(text.contains("notl\t"), "{text}");
3519 }
3520
3521 #[test]
3530 fn an_access_through_a_second_pointer_is_the_same_access_and_one_more() {
3531 let text = body("void f(int *p, int *r) { __atomic_load(p, r, 5); }\n");
3532 assert!(text.contains("%2 = atomic_load.i32 %0, align 4, seq_cst"), "{text}");
3533 assert!(text.contains("store %2 -> %1, align 4"), "and out through the place: {text}");
3534
3535 let text = body("void f(int *p, int *v) { __atomic_store(p, v, 3); }\n");
3536 assert!(text.contains("%2 = load.i32 %1, align 4"), "in through the place: {text}");
3537 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
3538
3539 let text = body("void f(int *p, int *v, int *r) { __atomic_exchange(p, v, r, 5); }\n");
3542 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3543 assert!(text.contains("%4 = atomic_rmw.i32 xchg %0, %3, align 4, seq_cst"), "{text}");
3544 assert!(text.contains("store %4 -> %2, align 4"), "{text}");
3545 }
3546
3547 #[test]
3558 fn a_flag_is_an_exchange_of_one_byte_and_a_store_of_a_zero_over_the_same_byte() {
3559 for pointer in ["char", "int", "void"] {
3560 let source = format!("int f({pointer} *p) {{ return __atomic_test_and_set(p, 5); }}\n");
3561 let text = body(&source);
3562 assert!(text.contains("%1 = iconst.i8 1"), "{pointer}: {text}");
3563 assert!(
3564 text.contains("%2 = atomic_rmw.i8 xchg %0, %1, align 1, seq_cst"),
3565 "{pointer}: {text}"
3566 );
3567 assert!(text.contains("%4 = icmp ne %2, %3"), "{pointer}: {text}");
3568
3569 let source = format!("void f({pointer} *p) {{ __atomic_clear(p, 3); }}\n");
3570 let text = body(&source);
3571 assert!(text.contains("atomic_store %2 -> %0, align 1, release"), "{pointer}: {text}");
3572 }
3573
3574 let text = asm("int f(int *p) { return __atomic_test_and_set(p, 5); }\n");
3577 assert!(text.contains("xchgb\t%al, (%rdi)"), "{text}");
3578 assert!(text.contains("setne\t"), "{text}");
3579 }
3580
3581 #[test]
3589 fn a_read_modify_write_is_an_exchange_or_a_locked_add_at_the_width_of_the_object() {
3590 let widths = [("char", "b", "%sil"), ("short", "w", "%si"), ("int", "l", "%esi")];
3591 for (ty, suffix, reg) in widths {
3592 let source =
3593 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_fetch_add(p, v, 5); }}\n");
3594 let text = asm(&source);
3595 assert!(text.contains("\tlock\n"), "{ty}: {text}");
3596 assert!(text.contains(&format!("xadd{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3597
3598 let source =
3599 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_exchange_n(p, v, 5); }}\n");
3600 let text = asm(&source);
3601 assert!(text.contains(&format!("xchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3602 assert!(!text.contains("\tlock\n"), "an exchange is locked already: {ty}: {text}");
3603 }
3604 let source = "long f(long *p, long v) { return __atomic_fetch_add(p, v, 5); }\n";
3605 assert!(asm(source).contains("xaddq\t%rsi, (%rdi)"), "{}", asm(source));
3606
3607 let source = "int f(int *p, int v) { return __atomic_fetch_sub(p, v, 5); }\n";
3610 let text = asm(source);
3611 assert!(text.contains("negl\t"), "{text}");
3612 assert!(text.contains("xaddl\t"), "{text}");
3613
3614 for order in ["0", "2", "3", "4", "5"] {
3617 let source =
3618 format!("int f(int *p, int v) {{ return __atomic_fetch_add(p, v, {order}); }}\n");
3619 let text = asm(&source);
3620 assert!(text.contains("xaddl\t"), "{order}: {text}");
3621 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
3622 }
3623
3624 let text = asm("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
3628 assert!(text.contains("xchgl\t%esi, (%rdi)"), "{text}");
3629 let text = asm("void f(int *p) { __sync_lock_release(p); }\n");
3634 assert!(text.contains("movl\t$0, %eax"), "{text}");
3635 assert!(text.contains("movl\t%eax, (%rdi)"), "{text}");
3636 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
3637 }
3638
3639 #[test]
3651 fn the_lock_free_questions_are_answered_as_constants() {
3652 for size in ["1", "2", "4", "8"] {
3653 let source =
3654 format!("int f(void) {{ return __atomic_always_lock_free({size}, 0); }}\n");
3655 let text = asm(&source);
3656 assert!(text.contains("movb\t$1, %al"), "{size} bytes is lock free: {text}");
3657 assert!(!text.contains("call"), "and is not a call: {text}");
3658 }
3659 for size in ["3", "16", "sizeof(long double)"] {
3660 let source = format!("int f(void) {{ return __atomic_is_lock_free({size}, 0); }}\n");
3661 let text = asm(&source);
3662 assert!(text.contains("movb\t$0, %al"), "{size} bytes is not: {text}");
3663 assert!(!text.contains("call"), "and is not a call either: {text}");
3664 }
3665
3666 let text = asm("int f(int n) { return __atomic_is_lock_free(n, 0); }\n");
3670 assert!(text.contains("movb\t$0, %al"), "a size nobody knows is not lock free: {text}");
3671 let text = asm("int f(int *p) { return __atomic_always_lock_free(8, p); }\n");
3672 assert!(text.contains("movb\t$0, %al"), "eight bytes at four is not: {text}");
3673 let text = asm("int f(long *p) { return __atomic_always_lock_free(8, p); }\n");
3674 assert!(text.contains("movb\t$1, %al"), "and at eight it is: {text}");
3675 }
3676
3677 #[test]
3689 fn a_memory_order_an_operation_cannot_carry_is_read_as_the_strongest() {
3690 let mut opts = options();
3691 opts.emit = EmitKind::Ir;
3692
3693 let acquire_store = run(&opts, "void f(int *p, int v) { __atomic_store_n(p, v, 2); }\n");
3694 assert!(acquire_store.text().contains("seq_cst"), "{:?}", acquire_store.text());
3695 assert!(acquire_store.messages[0].contains("[W0333]"), "{:?}", acquire_store.messages);
3696
3697 let nonsense = run(&opts, "int f(int *p) { return __atomic_load_n(p, 99); }\n");
3698 assert!(nonsense.text().contains("seq_cst"), "{:?}", nonsense.text());
3699 assert!(nonsense.messages[0].contains("[W0333]"), "{:?}", nonsense.messages);
3700
3701 let computed = run(&opts, "int f(int *p, int n) { return __atomic_load_n(p, n); }\n");
3702 assert!(computed.text().contains("seq_cst"), "{:?}", computed.text());
3703 assert_eq!(computed.messages, Vec::<String>::new(), "a computed order is not a mistake");
3704 }
3705
3706 #[test]
3718 fn a_conversion_between_a_float_and_the_widest_unsigned_integer_is_written_without_a_branch() {
3719 let text = asm("double f(unsigned long long x) { return (double)x; }\n");
3720 assert!(text.contains("cvtsi2sdq"), "the signed conversion is what runs: {text}");
3721 assert!(text.contains("shrq"), "with the value halved first: {text}");
3722 assert!(text.contains("addsd"), "and doubled after: {text}");
3723 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
3724
3725 let text = asm("unsigned long long f(double d) { return (unsigned long long)d; }\n");
3726 assert!(text.contains("cvttsd2siq"), "the signed conversion is what runs: {text}");
3727 assert!(text.contains("subsd"), "with half the range taken off first: {text}");
3728 assert!(text.contains("shlq\t$63"), "and the top bit put back: {text}");
3729 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
3730 }
3731
3732 #[test]
3743 fn a_plain_name_the_program_took_is_the_programs_own_function() {
3744 let taken = concat!(
3745 "static long long llabs(long long b) { return 7; }\n",
3746 "long long f(long long x) { return llabs(x); }\n",
3747 );
3748 assert!(ir(taken).contains("call @llabs"), "a static definition is the program's own");
3749
3750 let retyped = concat!("int llabs(int b);\n", "int f(int x) { return llabs(x); }\n",);
3751 assert!(ir(retyped).contains("call @llabs"), "another type is another function");
3752
3753 let plain = concat!(
3754 "long long llabs(long long b);\n",
3755 "long long f(long long x) { return llabs(x); }\n",
3756 );
3757 let mut opts = options();
3758 opts.emit = EmitKind::Ir;
3759 assert!(!run(&opts, plain).text().contains("call @llabs"), "the library's by default");
3760
3761 opts.builtins = false;
3762 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin");
3763
3764 opts.builtins = true;
3765 opts.no_builtin = vec!["llabs".to_owned()];
3766 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin-llabs");
3767 let one = "long labs(long b);\nlong f(long x) { return labs(x); }\n";
3768 assert!(!run(&opts, one).text().contains("call @labs"), "one name and not the family");
3769
3770 opts.no_builtin = Vec::new();
3773 opts.builtins = false;
3774 let prefixed = "long long f(long long x) { return __builtin_llabs(x); }\n";
3775 assert!(!run(&opts, prefixed).text().contains("call @llabs"), "the prefix is a promise");
3776 }
3777
3778 #[test]
3791 fn the_hint_builtins_are_their_first_argument_and_the_hint_leaves_no_trace() {
3792 let text = ir(concat!(
3793 "long a = __builtin_expect(7, 1);\n",
3794 "long b = __builtin_expect_with_probability(9, 1, 0.9);\n",
3795 "unsigned long c = sizeof(__builtin_expect((char)1, 1));\n",
3796 ));
3797 assert!(text.contains("global @a : i64 = 7,"), "{text}");
3798 assert!(text.contains("global @b : i64 = 9,"), "{text}");
3799 assert!(text.contains("global @c : i64 = 8,"), "{text}");
3800 assert!(!text.contains("__builtin_expect"), "it is not a call to anything:\n{text}");
3801
3802 let text = body("long f(char c) { return __builtin_expect(c, 1); }\n");
3805 assert!(text.contains("sext"), "{text}");
3806
3807 let one = "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 1\n %2 = sext.i64 %1\n return %0\n";
3811 assert_eq!(body("int f(void) { int i = 0; __builtin_expect(1, i++); return i; }\n"), one);
3812 let source = "int g(void) { int i = 0; __builtin_expect_with_probability(1, i++, 0.5); return i; }\n";
3813 assert_eq!(body(source), one);
3814
3815 let kept = body("int f(int n) { int i = 0; __builtin_expect(n, i++); return i; }\n");
3820 assert!(kept.contains("add.nsw"), "the hint still runs: {kept}");
3821 assert!(kept.ends_with("return %3\n"), "and the answer is what it left behind: {kept}");
3822 let both = "int g(int n) { int i = 0; __builtin_expect_with_probability(n, i++, 0.5); return i; }\n";
3823 assert!(body(both).contains("add.nsw"), "and so does the one with three arguments");
3824 }
3825
3826 #[test]
3838 fn a_promise_that_control_does_not_arrive_writes_no_instruction() {
3839 let promised = "int f(int x) { if (x) return 1; __builtin_unreachable(); }\n";
3840 let text = ir(promised);
3841 assert!(text.contains(" unreachable_hint\n"), "{text}");
3842 assert!(!text.contains("call"), "it is not a call to anything:\n{text}");
3843
3844 let after = body("int g(int x) { __builtin_unreachable(); return x; }\n");
3848 assert!(after.contains("return"), "{after}");
3849
3850 let text = asm(promised);
3853 let mine = text.split_once("\nf:\n").expect("a definition").1;
3854 let mine = mine.split_once("\t.size").expect("a definition").0;
3855 let plain = asm("int f(int x) { if (x) return 1; }\n");
3856 let plain = plain.split_once("\nf:\n").expect("a definition").1;
3857 let plain = plain.split_once("\t.size").expect("a definition").0;
3858 assert_eq!(mine, plain);
3859 let last = mine.lines().rfind(|line| !line.trim_start().starts_with('.'));
3862 assert_eq!(last.map(str::trim), Some("ret"), "{mine}");
3863 assert!(!mine.contains("ud2"), "{mine}");
3864 }
3865
3866 #[test]
3873 fn a_library_builtin_is_diagnosed_under_the_name_the_program_wrote() {
3874 let mut opts = options();
3875 opts.emit = EmitKind::Ir;
3876 let messages = run(&opts, "void f(void) { __builtin_abort(1); }\n").messages;
3877 assert!(
3878 messages.iter().any(|m| m.contains("__builtin_abort")),
3879 "expected the written name in {messages:?}"
3880 );
3881 }
3882
3883 #[test]
3891 fn a_builtin_nothing_lowers_is_refused_by_name() {
3892 let mut opts = options();
3893 opts.emit = EmitKind::Ir;
3894 for (builtin, call) in [
3895 ("__builtin_return_address", "(int)(long)__builtin_return_address(0)"),
3896 ("__builtin_alloca", "(int)(long)__builtin_alloca(8)"),
3897 ("__atomic_signal_fence", "(__atomic_signal_fence(5), 0)"),
3898 ] {
3899 let source = format!("int counter;\nint f(void) {{ return {call}; }}\n");
3900 let messages = run(&opts, &source).messages;
3901 let named = messages.iter().any(|m| m.contains(builtin) && m.contains("E0686"));
3902 assert!(named, "expected {builtin} to be refused by name in {messages:?}");
3903 }
3904 }
3905
3906 #[test]
3914 fn what_is_refused_is_the_call_and_not_the_name() {
3915 let text = ir("unsigned long n = sizeof(__builtin_return_address(0));\n");
3916 assert!(text.contains("global @n : i64 = 8,"), "{text}");
3917
3918 let text = ir(concat!(
3919 "void *__builtin_return_address(unsigned x) { return 0; }\n",
3920 "void *f(void) { return __builtin_return_address(0); }\n",
3921 ));
3922 assert!(text.contains("call @__builtin_return_address"), "{text}");
3923 }
3924
3925 #[test]
3930 fn a_static_function_nothing_refers_to_is_not_emitted() {
3931 let text = ir("static int dropped(void) { return 1; }\n\
3932 static int kept(void) { return 2; }\n\
3933 int main(void) { return kept(); }\n");
3934 assert!(text.contains("func @kept"), "{text}");
3935 assert!(!text.contains("dropped"), "{text}");
3936 }
3937
3938 #[test]
3944 fn two_static_functions_that_only_call_each_other_are_both_dropped() {
3945 let text = ir("static int ping(void);\n\
3946 static int pong(void) { return ping(); }\n\
3947 static int ping(void) { return pong(); }\n\
3948 int main(void) { return 0; }\n");
3949 assert!(!text.contains("ping"), "{text}");
3950 assert!(!text.contains("pong"), "{text}");
3951 }
3952
3953 #[test]
3959 fn naming_a_static_function_anywhere_keeps_it() {
3960 let text = ir("static int by_address(void) { return 1; }\n\
3961 static int in_an_image(void) { return 2; }\n\
3962 static int deeper(void) { return 3; }\n\
3963 static int reaches_deeper(void) { return deeper(); }\n\
3964 static int (*table[1])(void) = {in_an_image};\n\
3965 int main(void) {\n\
3966 int (*p)(void) = by_address;\n\
3967 return p() + table[0]() + reaches_deeper();\n\
3968 }\n");
3969 for kept in ["by_address", "in_an_image", "deeper", "reaches_deeper"] {
3970 assert!(text.contains(&format!("func @{kept}")), "expected {kept} in:\n{text}");
3971 }
3972 }
3973
3974 #[test]
3980 fn an_attribute_keeps_a_static_function_nothing_refers_to() {
3981 for attribute in ["used", "retain", "constructor", "destructor", "__used__"] {
3982 let source = format!(
3983 "__attribute__(({attribute})) static int kept(void) {{ return 1; }}\n\
3984 int main(void) {{ return 0; }}\n"
3985 );
3986 let text = ir(&source);
3987 assert!(text.contains("func @kept"), "for {attribute}:\n{text}");
3988 }
3989 }
3990
3991 #[test]
3994 fn a_function_anything_could_call_is_emitted_without_being_called() {
3995 let text =
3996 ir("int nobody_here_calls_it(void) { return 1; }\nint main(void) { return 0; }\n");
3997 assert!(text.contains("func @nobody_here_calls_it"), "{text}");
3998 }
3999
4000 #[test]
4007 fn a_classification_c_has_an_operator_for_is_that_operator() {
4008 for (builtin, operator) in [
4009 ("__builtin_isgreater", "binary >"),
4010 ("__builtin_isgreaterequal", "binary >="),
4011 ("__builtin_isless", "binary <"),
4012 ("__builtin_islessequal", "binary <="),
4013 ] {
4014 let source = format!("int f(double x, double y) {{ return {builtin}(x, y); }}\n");
4015 let text = tast(&source);
4016 assert!(text.contains(&format!("{operator} : int")), "for {builtin}:\n{text}");
4017 }
4018 }
4019
4020 #[test]
4029 fn the_classification_builtins_are_comparisons_and_not_calls() {
4030 let text = body("int f(double x, double y) { return __builtin_isunordered(x, y); }\n");
4031 assert_eq!(
4032 text,
4033 "block0(%0: f64, %1: f64):\n %2 = fcmp uno %0, %1\n %3 = zext.i32 \
4034 %2\n return %3\n"
4035 );
4036
4037 let text = body("int f(double x, double y) { return __builtin_islessgreater(x, y); }\n");
4039 assert!(text.contains("fcmp one %0, %1"), "{text}");
4040
4041 let text = body("int f(double x) { return __builtin_isnan(x); }\n");
4042 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4043
4044 let text = body("int f(double x) { return __builtin_isinf(x); }\n");
4045 assert!(text.contains("fconst.f64 0x7ff0000000000000"), "{text}");
4046 assert!(text.contains("fconst.f64 0xfff0000000000000"), "{text}");
4047 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4048 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4049 assert!(text.contains("%5 = or %3, %4"), "{text}");
4050
4051 let text = body("int f(double x) { return __builtin_isfinite(x); }\n");
4054 assert!(text.contains("%3 = fcmp olt %2, %0"), "{text}");
4055 assert!(text.contains("%4 = fcmp olt %0, %1"), "{text}");
4056 assert!(text.contains("%5 = and %3, %4"), "{text}");
4057
4058 let text = body("int f(double x) { return __builtin_signbit(x); }\n");
4059 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4060 assert!(text.contains("icmp slt %1, %2"), "{text}");
4061
4062 let text = body("int f(long double x) { return __builtin_signbitl(x); }\n");
4065 assert!(text.contains("%1 = bitcast.i80 %0"), "{text}");
4066
4067 let text = body("double g(void);\nint f(void) { return __builtin_isnan(g()); }\n");
4070 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4071 }
4072
4073 #[test]
4080 fn a_classification_spelling_that_names_a_width_converts_before_it_asks() {
4081 let text = ir(concat!(
4082 "int a = __builtin_isinff(1e300);\n",
4083 "int b = __builtin_isinf(1e300);\n",
4084 "int c = __builtin_isnan(0.0);\n",
4088 "int d = __builtin_signbit(-0.0);\n",
4089 "int e = __builtin_islessgreater(1.0, 2.0);\n",
4090 ));
4091 assert!(text.contains("global @a : i32 = 1,"), "{text}");
4092 assert!(text.contains("global @b : i32 = 0,"), "{text}");
4093 assert!(text.contains("global @c : i32 = 0,"), "{text}");
4094 assert!(text.contains("global @d : i32 = 1,"), "{text}");
4095 assert!(text.contains("global @e : i32 = 1,"), "{text}");
4096 }
4097
4098 #[test]
4100 fn a_classification_builtin_refuses_an_argument_that_is_not_floating_point() {
4101 let mut opts = options();
4102 opts.emit = EmitKind::Ir;
4103 let source = concat!(
4104 "int a(int x) { return __builtin_isnan(x); }\n",
4105 "int b(int x, int y) { return __builtin_isunordered(x, y); }\n",
4106 "int c(double x) { return __builtin_isnan(x, x); }\n",
4107 );
4108 let messages = run(&opts, source).messages;
4109 assert_eq!(
4110 messages,
4111 [
4112 "/main.c:1:23: error: non-floating-point argument in call to function \
4113 '__builtin_isnan' [E0685]",
4114 "/main.c:2:30: error: non-floating-point arguments in call to function \
4115 '__builtin_isunordered' [E0685]",
4116 "/main.c:3:26: error: too many arguments to function '__builtin_isnan' [E0511]",
4117 ]
4118 );
4119 }
4120
4121 #[test]
4130 fn the_last_three_classification_builtins_are_comparisons_and_not_calls() {
4131 let text = body("int f(double x) { return __builtin_isnormal(x); }\n");
4132 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4136 assert!(text.contains("%2 = iconst.i64 9223372036854775807"), "{text}");
4137 assert!(text.contains("%3 = and %1, %2"), "{text}");
4138 assert!(text.contains("%4 = iconst.i64 4503599627370496"), "{text}");
4139 assert!(text.contains("%5 = iconst.i64 9218868437227405312"), "{text}");
4140 assert!(text.contains("%6 = icmp uge %3, %4"), "{text}");
4141 assert!(text.contains("%7 = icmp ult %3, %5"), "{text}");
4142 assert!(text.contains("%8 = and %6, %7"), "{text}");
4143
4144 let text = body("int f(long double x) { return __builtin_isnormal(x); }\n");
4148 assert!(text.contains("%4 = iconst.i80 27670116110564327424"), "{text}");
4149 assert!(text.contains("%5 = iconst.i80 604453686435277732577280"), "{text}");
4150
4151 let text = body("int f(double x) { return __builtin_isinf_sign(x); }\n");
4152 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4153 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4154 assert!(text.contains("%7 = sub %5, %6"), "{text}");
4155
4156 let text = body("int f(double x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n");
4157 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4158 assert!(text.contains("fcmp oeq %0, %6"), "{text}");
4159 assert_eq!(text.matches(" = zext.i32 ").count(), 4, "{text}");
4163 assert_eq!(text.matches(" = xor ").count(), 4, "{text}");
4164 assert!(!text.contains("call"), "{text}");
4165
4166 let text = body(concat!(
4169 "double g(void);\n",
4170 "int f(void) { return __builtin_fpclassify(0, 1, 2, 3, 4, g()); }\n",
4171 ));
4172 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4173 }
4174
4175 #[test]
4182 fn the_last_three_classification_builtins_fold_where_their_operand_is_a_constant() {
4183 let text = ir(concat!(
4184 "int a = __builtin_isnormal(1.0);\n",
4185 "int b = __builtin_isnormal(0.0);\n",
4186 "int c = __builtin_isnormal(1.0 / 0.0);\n",
4187 "int d = __builtin_isinf_sign(-1.0 / 0.0);\n",
4188 "int e = __builtin_isinf_sign(1.0);\n",
4189 "int g = __builtin_fpclassify(0, 1, 2, 3, 4, 0.0);\n",
4190 "int h = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0);\n",
4191 "int i = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0 / 0.0);\n",
4192 ));
4193 assert!(text.contains("global @a : i32 = 1,"), "{text}");
4194 assert!(text.contains("global @b : i32 = 0,"), "{text}");
4195 assert!(text.contains("global @c : i32 = 0,"), "{text}");
4196 assert!(text.contains("global @d : i32 = -1,"), "{text}");
4197 assert!(text.contains("global @e : i32 = 0,"), "{text}");
4198 assert!(text.contains("global @g : i32 = 4,"), "{text}");
4199 assert!(text.contains("global @h : i32 = 2,"), "{text}");
4200 assert!(text.contains("global @i : i32 = 1,"), "{text}");
4201 }
4202
4203 #[test]
4209 fn fpclassify_refuses_an_answer_that_is_not_an_integer_constant() {
4210 let mut opts = options();
4211 opts.emit = EmitKind::Ir;
4212 let source = concat!(
4213 "int a(double x, int n) { return __builtin_fpclassify(0, 1, n, 3, 4, x); }\n",
4214 "int b(double x) { return __builtin_fpclassify(0, 1, 2, 3, x); }\n",
4215 "int c(int x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n",
4216 );
4217 let messages = run(&opts, source).messages;
4218 assert_eq!(
4219 messages,
4220 [
4221 "/main.c:1:60: error: non-const integer argument 3 in call to function \
4222 '__builtin_fpclassify' [E0687]",
4223 "/main.c:2:26: error: too few arguments to function '__builtin_fpclassify' \
4224 [E0511]",
4225 "/main.c:3:23: error: non-floating-point argument in call to function \
4226 '__builtin_fpclassify' [E0685]",
4227 ]
4228 );
4229 }
4230
4231 #[test]
4239 fn a_builtin_whose_answer_is_a_constant_is_one_and_not_a_call() {
4240 let text = ir(concat!(
4241 "double a = __builtin_inf();\n",
4242 "float b = __builtin_huge_valf();\n",
4243 "long double c = __builtin_infl();\n",
4244 "double d = __builtin_huge_val();\n",
4245 ));
4246 assert!(text.contains("global @a : f64 = 0x7ff0000000000000,"), "{text}");
4247 assert!(text.contains("global @b : f32 = 0x7f800000,"), "{text}");
4248 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
4249 assert!(text.contains("global @d : f64 = 0x7ff0000000000000,"), "{text}");
4250 assert!(!text.contains("call"), "{text}");
4251 }
4252
4253 #[test]
4262 fn a_nan_is_written_with_the_payload_the_program_asked_for() {
4263 let text = ir(concat!(
4264 "double a = __builtin_nan(\"\");\n",
4265 "double b = __builtin_nan(\"0x1\");\n",
4266 "double c = __builtin_nan(\"010\");\n",
4268 "double d = __builtin_nans(\"\");\n",
4269 "double e = __builtin_nans(\"0x1\");\n",
4270 "float f = __builtin_nanf(\"0x1\");\n",
4271 "float g = __builtin_nansf(\"\");\n",
4272 "long double h = __builtin_nansl(\"\");\n",
4273 ));
4274 assert!(text.contains("global @a : f64 = 0x7ff8000000000000,"), "{text}");
4275 assert!(text.contains("global @b : f64 = 0x7ff8000000000001,"), "{text}");
4276 assert!(text.contains("global @c : f64 = 0x7ff8000000000008,"), "{text}");
4277 assert!(text.contains("global @d : f64 = 0x7ff4000000000000,"), "{text}");
4278 assert!(text.contains("global @e : f64 = 0x7ff0000000000001,"), "{text}");
4279 assert!(text.contains("global @f : f32 = 0x7fc00001,"), "{text}");
4280 assert!(text.contains("global @g : f32 = 0x7fa00000,"), "{text}");
4281 assert!(text.contains("f80 0x7fffa000000000000000"), "{text}");
4282
4283 let text = ir(concat!(
4286 "double f(const char *p) { return __builtin_nan(p); }\n",
4287 "double g(void) { return __builtin_nans(\"1x\"); }\n",
4288 ));
4289 assert_eq!(text.matches("call @nan(").count(), 1, "{text}");
4290 assert_eq!(text.matches("call @nans(").count(), 1, "{text}");
4291 }
4292
4293 #[test]
4301 fn the_length_and_the_order_of_a_string_literal_are_known_here() {
4302 let text = ir(concat!(
4303 "unsigned long a = __builtin_strlen(\"hello\");\n",
4304 "unsigned long b = __builtin_strlen(\"a\\0bc\");\n",
4305 "int c = __builtin_strcmp(\"X\", \"X\\376\") < 0;\n",
4306 "int d = __builtin_strcmp(\"abc\", \"abc\");\n",
4307 "int e = __builtin_strcmp(\"abc\", \"ab\") > 0;\n",
4308 ));
4309 assert!(text.contains("global @a : i64 = 5,"), "{text}");
4310 assert!(text.contains("global @b : i64 = 1,"), "{text}");
4311 assert!(text.contains("global @c : i32 = 1,"), "{text}");
4312 assert!(text.contains("global @d : i32 = 0,"), "{text}");
4313 assert!(text.contains("global @e : i32 = 1,"), "{text}");
4314 assert!(!text.contains("call"), "{text}");
4315
4316 let text = ir("unsigned long f(const char *p) { return __builtin_strlen(p); }\n");
4318 assert!(text.contains("call @strlen("), "{text}");
4319 }
4320
4321 #[test]
4328 fn a_sign_builtin_is_a_mask_over_the_bits_and_not_a_call() {
4329 let text = body("double f(double x) { return __builtin_fabs(x); }\n");
4330 assert!(text.contains("bitcast.i64 %0"), "{text}");
4331 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
4332 assert!(text.contains("and %1, %2"), "{text}");
4333 assert!(text.contains("bitcast.f64 %3"), "{text}");
4334 assert!(!text.contains("call"), "{text}");
4335
4336 let text = body("double f(double x, double y) { return __builtin_copysign(x, y); }\n");
4337 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
4338 assert!(text.contains("%8 = or %4, %7"), "{text}");
4339 assert!(!text.contains("call"), "{text}");
4340
4341 let text = body("long double f(long double x) { return __builtin_fabsl(x); }\n");
4344 assert!(text.contains("bitcast.i80 %0"), "{text}");
4345 assert!(text.contains("bitcast.f80"), "{text}");
4346
4347 let text = body("double f(float x) { return __builtin_fabs(x); }\n");
4350 assert!(text.contains("fpext.f64 %0"), "{text}");
4351 assert!(text.contains("bitcast.i64 %1"), "{text}");
4352 }
4353
4354 #[test]
4363 fn the_sign_builtins_answer_a_zero_and_a_nan_the_way_the_bits_say() {
4364 let text = ir(concat!(
4365 "double a = __builtin_fabs(-3.5);\n",
4366 "double b = __builtin_copysign(1.0, -0.0);\n",
4367 "double c = __builtin_copysign(0.0, -2.0);\n",
4368 "double d = __builtin_copysign(-__builtin_nan(\"\"), 1.0);\n",
4370 "double e = __builtin_fabs(-__builtin_nan(\"0x1\"));\n",
4371 "float g = __builtin_copysignf(-0.0f, 2.0f);\n",
4372 "long double h = __builtin_copysignl(1.0L, -1.0L);\n",
4373 "long double i = __builtin_fabsl(-__builtin_infl());\n",
4374 ));
4375 assert!(text.contains("global @a : f64 = 0x400c000000000000,"), "{text}");
4376 assert!(text.contains("global @b : f64 = 0xbff0000000000000,"), "{text}");
4377 assert!(text.contains("global @c : f64 = 0x8000000000000000,"), "{text}");
4378 assert!(text.contains("global @d : f64 = 0x7ff8000000000000,"), "{text}");
4379 assert!(text.contains("global @e : f64 = 0x7ff8000000000001,"), "{text}");
4380 assert!(text.contains("global @g : f32 = 0x0,"), "{text}");
4381 assert!(text.contains("f80 0xbfff8000000000000000"), "{text}");
4382 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
4383 }
4384
4385 #[test]
4392 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
4393 let text = ir(concat!(
4394 "constexpr int side = 4;\n",
4395 "constexpr int wider = side + 1;\n",
4396 "constexpr double half = 1.5;\n",
4397 "struct point { int x; int y; };\n",
4398 "constexpr struct point origin = { 5, 6 };\n",
4399 "int square[side * side];\n",
4400 "int rectangle[wider];\n",
4401 "int rounded[(int)half * 2];\n",
4402 "int across[origin.y];\n",
4403 "enum named { four = side };\n",
4404 "int e = four;\n",
4405 ));
4406 assert!(text.contains("global @square : bytes 64 ="), "{text}");
4407 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
4408 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
4409 assert!(text.contains("global @across : bytes 24 ="), "{text}");
4410 assert!(text.contains("global @e : i32 = 4,"), "{text}");
4411
4412 let mut opts = options();
4415 opts.emit = EmitKind::Ir;
4416 let konst = "const int n = 1;\nint a[n];\n";
4417 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
4418 assert_eq!(run(&opts, konst).messages, [message]);
4419
4420 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
4422 assert_eq!(run(&opts, subscript).messages, [message]);
4423
4424 let address = "constexpr int c = 3;\nint *p = &c;\n";
4426 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
4427 pointer target type [E0514]";
4428 assert_eq!(run(&opts, address).messages, [warning]);
4429 }
4430
4431 #[test]
4440 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
4441 let mut opts = options();
4444 opts.std = Std::C17;
4445 let source = concat!(
4446 "int add(a, b)\n",
4447 "int a;\n",
4448 "int b;\n",
4449 "{ return a + b; }\n",
4450 "int promoted(c)\n",
4451 "char c;\n",
4452 "{ return c; }\n",
4453 "int narrow(char);\n",
4454 "int narrow(c)\n",
4455 "char c;\n",
4456 "{ return c; }\n",
4457 "int first(a)\n",
4458 "int a[4];\n",
4459 "{ return a[0]; }\n",
4460 );
4461 let result = run(&opts, source);
4462 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
4463 let text = result.text();
4464 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
4465 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
4466 assert!(text.contains("c : char object automatic defined"), "{text}");
4468 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
4469 assert!(text.contains("first : int(int *) function external defined"), "{text}");
4471 }
4472
4473 #[test]
4480 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
4481 let mut opts = options();
4482 opts.std = Std::C17;
4483 for (source, message) in [
4484 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
4485 (
4486 "int f(a)\nint a;\nint b;\n{ return a; }\n",
4487 "3:5: error: declaration for parameter 'b' but no such parameter",
4488 ),
4489 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
4490 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
4491 (
4492 "int f(a)\nstatic int a;\n{ return a; }\n",
4493 "2:12: error: storage class specified for parameter 'a'",
4494 ),
4495 (
4496 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
4497 "2:7: error: argument 'a' doesn't match prototype",
4498 ),
4499 ] {
4500 let result = run(&opts, source);
4501 assert!(result.failed(), "expected this to fail:\n{source}");
4502 assert!(result.messages[0].contains(message), "{:?}", result.messages);
4503 }
4504
4505 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
4508 let mut older = options();
4509 older.std = Std::C89;
4510 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
4511 let result = run(&opts, implicit);
4512 assert!(
4513 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
4514 "{:?}",
4515 result.messages
4516 );
4517
4518 let mut newer = options();
4522 newer.std = Std::C23;
4523 let plain = "int f(a)\nint a;\n{ return a; }\n";
4524 let result = run(&newer, plain);
4525 assert!(!result.failed(), "{:?}", result.messages);
4526 assert_eq!(
4527 result.messages,
4528 ["/main.c:1:5: warning: old-style function definition [E0412]"]
4529 );
4530 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
4531 }
4532
4533 #[test]
4540 fn the_obsolete_designators_are_taken_and_are_pedantic_warnings() {
4541 let array = "int a[8] = { [3] 7 };\n";
4542 let member = "struct s { int x; } v = { x: 7 };\n";
4543 for source in [array, member] {
4544 let result = run(&options(), source);
4545 assert!(!result.failed(), "{:?}", result.messages);
4546 assert!(result.messages.is_empty(), "nothing to say: {:?}", result.messages);
4547 }
4548
4549 let mut asked = options();
4550 asked.pedantic = true;
4551 assert_eq!(
4552 run(&asked, array).messages,
4553 ["/main.c:1:18: warning: obsolete designator, write `[i] =` instead [E0415]"]
4554 );
4555 assert_eq!(
4556 run(&asked, member).messages,
4557 ["/main.c:1:27: warning: obsolete designator, write `.field =` instead [E0413]"]
4558 );
4559 }
4560
4561 #[test]
4568 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
4569 let text = ir(concat!(
4570 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
4571 "struct brim { char buf[9223372036854775807L]; };\n",
4572 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
4573 "unsigned long h = sizeof(struct huge_struct);\n",
4574 "unsigned long b = sizeof(struct brim);\n",
4575 "unsigned long y = sizeof(struct bitty);\n",
4576 ));
4577 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
4578 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
4579 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
4580
4581 let mut opts = options();
4582 opts.emit = EmitKind::Ir;
4583 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
4584 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
4585 assert_eq!(run(&opts, over).messages, [message]);
4586 let array = "struct wide { short buf[1L << 62]; };\n";
4587 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
4588 maximum object size '9223372036854775807' [E0537]";
4589 assert_eq!(run(&opts, array).messages[0], message);
4590 }
4591
4592 fn compile_bytes(source: &[u8]) -> Compiled {
4597 let mut opts = options();
4598 opts.emit = EmitKind::Ir;
4599 let mut fs = MemoryFileSystem::new();
4600 fs.insert("/main.c", source.to_vec());
4601 compile(&opts, "/main.c", &fs)
4602 }
4603
4604 #[test]
4611 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
4612 let mut source = b"char s[] = \"a".to_vec();
4613 source.push(0xff);
4614 source.extend_from_slice(b"b\";\nchar c = '");
4615 source.push(0xff);
4616 source.extend_from_slice(b"';\n");
4617 let result = compile_bytes(&source);
4618 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
4619 assert!(result.text().contains(r#"bytes "a\ffb\00""#), "{}", result.text());
4620 assert!(result.text().contains("global @c : i8 = -1,"), "{}", result.text());
4622
4623 let mut stray = b"int a".to_vec();
4624 stray.push(0xff);
4625 stray.extend_from_slice(b" = 1;\n");
4626 let result = compile_bytes(&stray);
4627 assert!(
4628 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
4629 "{:?}",
4630 result.messages
4631 );
4632 }
4633
4634 #[test]
4635 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
4636 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
4637 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
4638 let expected = "\
4639func @add(i32, i32) -> i32, linkage(external) {
4640block0(%0: i32, %1: i32):
4641 %2 = add.nsw %0, %1
4642 return %2
4643}
4644";
4645 assert!(text.contains(expected), "{text}");
4646 }
4647
4648 #[test]
4649 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
4650 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
4651 assert!(!text.contains("alloca"), "{text}");
4652 assert!(!text.contains("load"), "{text}");
4653 assert!(!text.contains("store"), "{text}");
4654 }
4655
4656 #[test]
4657 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
4658 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
4659 let expected = "\
4660block0:
4661 %0 = alloca, size 4, align 4
4662 %1 = iconst.i32 1
4663 store %1 -> %0, align 4
4664 %2 = call @g(%0) : (ptr) -> i32
4665 return %2
4666";
4667 assert_eq!(text, expected);
4668 }
4669
4670 #[test]
4671 fn a_loop_carries_what_it_changes_as_block_parameters() {
4672 let text = body(
4675 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
4676 return total;\n}\n",
4677 );
4678 assert!(!text.contains("alloca"), "{text}");
4679 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
4680 assert!(text.contains("jump block1("), "{text}");
4681 }
4682
4683 #[test]
4684 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
4685 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
4686 assert!(text.contains("icmp slt %0, %1"), "{text}");
4687 assert!(!text.contains("zext"), "{text}");
4688 }
4689
4690 #[test]
4691 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
4692 let text = body("int f(int a, int b) { return a && b; }\n");
4693 let expected = "\
4694block0(%0: i32, %1: i32):
4695 %2 = iconst.i32 0
4696 %3 = icmp ne %0, %2
4697 %4 = iconst.i1 0
4698 br_if %3, block1, block2(%4)
4699
4700block1:
4701 %5 = iconst.i32 0
4702 %6 = icmp ne %1, %5
4703 jump block2(%6)
4704
4705block2(%7: i1):
4706 %8 = zext.i32 %7
4707 return %8
4708";
4709 assert_eq!(text, expected);
4710 }
4711
4712 #[test]
4713 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
4714 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
4715 assert!(!text.contains("block3"), "{text}");
4718 assert!(!text.contains("iconst.i32 3"), "{text}");
4719 }
4720
4721 #[test]
4722 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
4723 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
4724 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
4725 assert!(body("int f(void) { }\n").contains("unreachable"));
4726 }
4727
4728 #[test]
4729 fn a_structure_is_copied_rather_than_held_in_a_value() {
4730 let text = body(
4731 "struct point { int x, y; };\n\
4732 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
4733 );
4734 assert!(text.contains("memcpy"), "{text}");
4735 }
4736
4737 #[test]
4738 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
4739 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
4740 assert!(text.contains("memset"), "{text}");
4741 }
4742
4743 #[test]
4744 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
4745 let text = body(
4746 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
4747 default: r = 4; } return r; }\n",
4748 );
4749 let expected = "\
4750block0(%0: i32):
4751 %1 = iconst.i32 0
4752 switch %0, block1, [1 => block2, 2 => block3(%1)]
4753
4754block1:
4755 %2 = iconst.i32 4
4756 jump block4(%2)
4757
4758block2:
4759 %3 = iconst.i32 1
4760 jump block3(%3)
4761
4762block3(%4: i32):
4763 %5 = iconst.i32 2
4764 %6 = add.nsw %4, %5
4765 jump block4(%6)
4766
4767block4(%7: i32):
4768 return %7
4769";
4770 assert_eq!(text, expected);
4771 }
4772
4773 #[test]
4774 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
4775 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
4778 assert!(text.contains("%2 = sub %0, %1"), "{text}");
4779 assert!(text.contains("icmp ule"), "{text}");
4780 assert!(!text.contains("switch"), "{text}");
4781 }
4782
4783 #[test]
4784 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
4785 let text = body(
4786 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
4787 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
4788 );
4789 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
4792 assert!(text.contains("block5:\n jump block7("), "{text}");
4793 assert!(text.contains("block6:\n jump block8("), "{text}");
4794 }
4795
4796 #[test]
4797 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
4798 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
4799 }
4800
4801 #[test]
4802 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
4803 let text = body(
4808 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
4809 return n; }\n",
4810 );
4811 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
4814 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
4815 assert!(text.contains("block4:\n jump block3("), "{text}");
4816 }
4817
4818 #[test]
4819 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
4820 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
4823 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
4824 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
4825 assert!(text.contains("br_if %6, block2, block3"), "{text}");
4826 }
4827
4828 #[test]
4829 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
4830 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
4831 assert!(!text.contains("alloca"), "{text}");
4835 assert!(text.contains("block2(%4: i32):\n return %4"), "{text}");
4836 assert_eq!(text.matches("jump block2(").count(), 2, "{text}");
4837 }
4838
4839 #[test]
4840 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
4841 let text =
4842 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
4843 assert!(!text.contains("alloca"), "{text}");
4844 assert!(text.contains("block1(%2: i32):"), "{text}");
4845 assert!(text.contains("jump block1(%5)"), "{text}");
4846 }
4847
4848 #[test]
4849 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
4850 assert_eq!(
4853 body("int f(int x) { return x; spare: return 0; }\n"),
4854 "block0(%0: i32):\n return %0\n"
4855 );
4856 }
4857
4858 #[test]
4859 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
4860 let text = body(
4861 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
4862 );
4863 assert_eq!(
4866 text,
4867 "\
4868block0(%0: ptr):
4869 %1 = load.i8 %0, align 1
4870 %2 = iconst.i8 3
4871 %3 = ashr %1, %2
4872 %4 = sext.i32 %3
4873 return %4
4874"
4875 );
4876 }
4877
4878 #[test]
4879 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
4880 let text =
4884 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
4885 assert_eq!(
4886 text,
4887 "\
4888block0(%0: ptr, %1: i32):
4889 %2 = iconst.i32 16777215
4890 %3 = and %1, %2
4891 %4 = trunc.i16 %3
4892 store %4 -> %0, align 2
4893 %5 = iconst.i32 16
4894 %6 = lshr %3, %5
4895 %7 = trunc.i8 %6
4896 %8 = iconst.i64 2
4897 %9 = ptr_add %0, %8
4898 store %7 -> %9, align 1
4899 return
4900"
4901 );
4902 }
4903
4904 #[test]
4905 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
4906 let text =
4907 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
4908 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
4911 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
4912 }
4913
4914 #[test]
4915 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
4916 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
4919 assert_eq!(text.matches("ashr").count(), 0, "{text}");
4920 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
4921 }
4922
4923 #[test]
4924 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
4925 let text = body(
4929 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
4930 );
4931 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
4932 }
4933
4934 #[test]
4935 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
4936 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
4939 assert!(
4940 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
4941 "{text}"
4942 );
4943 }
4944
4945 #[test]
4946 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
4947 let text = ir(concat!(
4952 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
4953 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
4954 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
4955 "char s[2] = \"hi\";\n",
4956 ));
4957 assert!(
4958 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
4959 "{text}"
4960 );
4961 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
4962 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
4963 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
4966 }
4967
4968 #[test]
4969 fn a_definition_takes_a_parameter_it_left_unnamed() {
4970 let text = ir("int f(int a, int) { return a; }\n");
4974 assert!(text.contains("func @f(i32, i32) -> i32"), "{text}");
4975 assert!(text.contains("block0(%0: i32, %1: i32):"), "{text}");
4976
4977 let text = ir("int g(int, int n) { return n; }\n");
4980 assert!(text.contains("block0(%0: i32, %1: i32):\n return %1\n"), "{text}");
4981 }
4982
4983 #[test]
4984 fn an_assignment_of_a_structure_is_the_object_it_wrote() {
4985 let text = body(concat!(
4990 "struct s { int f; int g; };\n",
4991 "void h(struct s *a, struct s *c, struct s *d, struct s *e)\n",
4992 "{ *d = *e = a[0] = *c; }\n",
4993 ));
4994 assert_eq!(text.matches("memcpy").count(), 3, "{text}");
4995 assert!(text.contains("memcpy %8, %1, size 8, align 4\n"), "{text}");
4996 assert!(text.contains("memcpy %3, %8, size 8, align 4\n"), "{text}");
4997 assert!(text.contains("memcpy %2, %3, size 8, align 4\n"), "{text}");
4998 }
4999
5000 #[test]
5001 fn a_string_literal_stops_at_the_end_of_the_array_it_is_filling() {
5002 let mut opts = options();
5007 opts.emit = EmitKind::Ir;
5008 let result = run(
5009 &opts,
5010 concat!(
5011 "const char a[2][3] = { \"1234\", \"xyz\" };\n",
5012 "static const char b[3][5] = { \"12345\", \"678\", \"9\" };\n",
5013 "union u { struct { char x[4]; char y[4]; }; struct { char z[8]; }; };\n",
5014 "const union u c = { { \"1234\", \"567\" } };\n",
5015 ),
5016 );
5017 let text = result.text();
5018 assert_eq!(
5019 result.messages,
5020 ["/main.c:1:24: warning: initializer-string for array of 'const char' is too long \
5021 (5 chars into 3 available) [E0637]"]
5022 );
5023 assert!(text.contains("global @a : bytes 6 = { bytes \"123\", bytes \"xyz\" }"), "{text}");
5024 assert!(
5025 text.contains(
5026 "global @b : bytes 15 = { bytes \"12345\", bytes \"678\\00\", zero 1, \
5027 bytes \"9\\00\", zero 3 }"
5028 ),
5029 "{text}"
5030 );
5031 assert!(
5034 text.contains("global @c : bytes 8 = { bytes \"1234\", bytes \"567\\00\" }"),
5035 "{text}"
5036 );
5037 }
5038
5039 #[test]
5040 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
5041 let text = body(concat!(
5045 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
5046 "void g(struct v *);\n",
5047 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
5048 ));
5049 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
5050 }
5051
5052 #[test]
5053 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
5054 let text = ir(concat!(
5059 "struct s { int x; };\n",
5060 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
5061 "int n = (int){ 7 };\n",
5062 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
5063 ));
5064 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
5065 assert!(text.contains("global @n : i32 = 7,"), "{text}");
5066 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
5069 }
5070
5071 #[test]
5072 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
5073 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
5077 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
5078 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
5079 }
5080
5081 #[test]
5082 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
5083 let text = ir("unsigned char foo[1][0];\n");
5087 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
5088 }
5089
5090 #[test]
5091 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
5092 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
5095 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
5096 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
5097 }
5098
5099 #[test]
5100 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
5101 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
5105 assert!(
5106 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
5107 "{text}"
5108 );
5109 }
5110
5111 #[test]
5112 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
5113 let text = body(
5118 "\
5119struct s { int a, b; };
5120struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
5121",
5122 );
5123 assert!(text.contains("block3(%7: ptr)"), "{text}");
5125 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
5126 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
5127 }
5128
5129 #[test]
5137 fn the_left_side_of_a_conditional_with_no_middle_is_evaluated_once() {
5138 let text = body("int f(int i) { return ++i ?: 10; }\n");
5139 assert!(text.contains("jump block3(%2)"), "the arm is the value that was tested: {text}");
5140 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
5141
5142 let text = body("long f(int i) { return ++i ?: 10L; }\n");
5145 assert!(text.contains("%5 = sext.i64 %2"), "the arm widens what was tested: {text}");
5146 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
5147
5148 let text = body("int g(void);\nint f(void) { return g() ?: 10; }\n");
5150 assert_eq!(text.matches("call @g").count(), 1, "called once: {text}");
5151
5152 let text = body("int f(int i) { return ++i ? ++i : 10; }\n");
5155 assert_eq!(text.matches("add.nsw").count(), 2, "incremented twice: {text}");
5156 }
5157
5158 #[test]
5159 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
5160 let text = ir("\
5164struct pair { int a, b; };
5165struct pair make(int a, int b);
5166struct pair twice(struct pair p) { return make(p.a, p.b); }
5167");
5168 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
5169 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
5170 }
5171
5172 #[test]
5173 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
5174 let text = ir("\
5178struct big { double v[8]; };
5179struct big grow(struct big b);
5180struct big twice(struct big b) { return grow(grow(b)); }
5181");
5182 assert!(
5183 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
5184 "{text}"
5185 );
5186 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
5187 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
5190 }
5191
5192 #[test]
5193 fn a_structure_passed_to_a_variadic_function_says_so_at_the_call() {
5194 let text = ir("\
5199struct big { double v[8]; };
5200struct pair { int a, b; };
5201int p(const char *, ...);
5202int f(struct big b, struct pair q) { return p(\"\", 1, b, q); }
5203");
5204 assert!(
5205 text.contains("call @p(%4, %5, %2 byval(64, align 8), %6) : (ptr, ...) -> i32"),
5206 "{text}"
5207 );
5208 }
5209
5210 #[test]
5211 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
5212 let body = body(
5215 "\
5216struct pair { int a, b; };
5217struct pair make(int a, int b);
5218int second(void) { return make(1, 2).b; }
5219",
5220 );
5221 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
5222 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
5223 }
5224
5225 #[test]
5226 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
5227 let source = "\
5231struct hfa { float x, y, z; };
5232int take(struct hfa h);
5233int give(struct hfa h) { return take(h); }
5234";
5235 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
5236 let mut opts = options();
5237 opts.emit = EmitKind::Ir;
5238 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
5239 let result = run(&opts, source);
5240 assert_eq!(result.messages, Vec::<String>::new());
5241 assert!(result.text().contains("func @take(f32, f32, f32) -> i32"), "{}", result.text());
5242 }
5243
5244 #[test]
5245 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
5246 let source = "\
5249int use(int *);
5250void f(int n) {
5251 {
5252 int a[n];
5253 use(a);
5254 }
5255 use(0);
5256}
5257";
5258 let body = body(source);
5259 assert!(body.contains("mul.nsw"), "{body}");
5260 assert!(body.contains("stacksave"), "{body}");
5261 assert!(body.contains("alloca %"), "{body}");
5262 assert!(body.contains("stackrestore"), "{body}");
5263 }
5264
5265 #[test]
5266 fn a_goto_out_of_the_scope_of_one_gives_its_stack_back_on_the_way() {
5267 let source = "\
5272int use(int *);
5273int f(int n) {
5274 {
5275 int a[n];
5276 if (use(a)) goto out;
5277 use(0);
5278 }
5279out:
5280 return 0;
5281}
5282";
5283 let body = body(source);
5284 assert_eq!(body.matches("stackrestore").count(), 2, "{body}");
5286 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5287 assert!(after.starts_with(" %4\n jump block"), "{body}");
5288 }
5289
5290 #[test]
5291 fn a_goto_to_a_label_the_array_is_still_alive_at_leaves_the_stack_alone() {
5292 let source = "\
5296int use(int *);
5297int f(int n) {
5298 int a[n];
5299again:
5300 if (use(a)) goto again;
5301 return 0;
5302}
5303";
5304 let body = body(source);
5305 assert!(body.contains("stacksave"), "{body}");
5306 assert!(!body.contains("stackrestore"), "{body}");
5307 }
5308
5309 #[test]
5310 fn a_goto_back_to_a_label_in_front_of_one_gives_it_back_every_time_round() {
5311 let source = "\
5316int use(int *);
5317int f(int n) {
5318again:
5319 {
5320 int a[n];
5321 if (use(a)) goto again;
5322 }
5323 return 0;
5324}
5325";
5326 let body = body(source);
5327 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
5328 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5329 assert!(after.starts_with(" %4\n jump block1\n"), "{body}");
5330 }
5331
5332 #[test]
5333 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
5334 let source = "\
5340int f(void);
5341void t(void) {
5342 int count = 10;
5343 for (; count--;) {
5344 int b[f()];
5345 int i;
5346 for (i = 0; i < f(); i++) {
5347 b[i] = count;
5348 }
5349 }
5350}
5351";
5352 let body = body(source);
5353 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
5357 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5358 let next = after.split("\n\n").next().expect("the block the restore is in");
5361 assert!(next.contains("jump block1("), "{body}");
5362 }
5363
5364 #[test]
5365 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
5366 let source = "\
5369unsigned long f(int n) {
5370 int a[n];
5371 n = 0;
5372 return sizeof a;
5373}
5374";
5375 let body = body(source);
5376 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
5378 }
5379
5380 #[test]
5381 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
5382 let source = "\
5385int use(int);
5386int f(int x) {
5387 return ({
5388 int t = use(x);
5389 t * t;
5390 });
5391}
5392";
5393 let expected = "\
5394block0(%0: i32):
5395 %1 = call @use(%0) : (i32) -> i32
5396 %2 = mul.nsw %1, %1
5397 return %2
5398";
5399 assert_eq!(body(source), expected);
5400 }
5401
5402 #[test]
5403 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
5404 let source = "int f(int x) { return ({ return x; 0; }); }\n";
5408 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
5409 }
5410
5411 #[test]
5412 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
5413 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
5417 let expected = "\
5418block0(%0: ptr):
5419 %1 = va_arg.f64 %0
5420 %2 = va_arg.f64 %0
5421 %3 = fadd %1, %2
5422 return %3
5423";
5424 assert_eq!(body(source), expected);
5425 }
5426
5427 #[test]
5428 fn one_that_reads_a_structure_answers_where_the_object_is() {
5429 let source = "\
5443struct s { int a; long b; };
5444long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.b; }
5445";
5446 let expected = "\
5447block0(%0: ptr):
5448 %1 = alloca, size 16, align 16
5449 %2 = va_object %0, size 16, align 8, in(int 8 at 0, int 8 at 8)
5450 memcpy %1, %2, size 16, align 8
5451 %3 = iconst.i64 8
5452 %4 = ptr_add %1, %3
5453 %5 = load.i64 %4, align 8
5454 return %5
5455";
5456 assert_eq!(body(source), expected);
5457 }
5458
5459 #[test]
5463 fn the_classification_says_which_registers_the_object_arrived_in() {
5464 let source = "\
5465struct s { double a; double b; };
5466double f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a; }
5467";
5468 assert!(
5469 body(source)
5470 .contains("va_object %0, size 16, align 8, in(float f64 at 0, float f64 at 8)"),
5471 "{}",
5472 body(source)
5473 );
5474
5475 let big = "\
5476struct s { long a[4]; };
5477long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a[0]; }
5478";
5479 assert!(body(big).contains("va_object %0, size 32, align 8\n"), "{}", body(big));
5480 }
5481
5482 #[test]
5483 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
5484 let source = "\
5488int f(int c) {
5489 void *p = c ? &&one : &&two;
5490 goto *p;
5491one:
5492 return 1;
5493two:
5494 return 2;
5495}
5496";
5497 let expected = "\
5498block0(%0: i32):
5499 %1 = iconst.i32 0
5500 %2 = icmp ne %0, %1
5501 br_if %2, block1, block2
5502
5503block1:
5504 %3 = block_addr block3
5505 jump block4(%3)
5506
5507block2:
5508 %4 = block_addr block5
5509 jump block4(%4)
5510
5511block3:
5512 %5 = iconst.i32 1
5513 return %5
5514
5515block4(%6: ptr):
5516 indirect_br %6, block3, block5
5517
5518block5:
5519 %7 = iconst.i32 2
5520 return %7
5521";
5522 assert_eq!(body(source), expected);
5523 }
5524
5525 #[test]
5526 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
5527 let source = "void **next(void);
5530void f(void) { goto *next(); }
5531";
5532 let expected = "\
5533block0:
5534 %0 = call @next() : () -> ptr
5535 unreachable
5536";
5537 assert_eq!(body(source), expected);
5538 }
5539
5540 #[test]
5541 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
5542 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
5545 let expected = "\
5546block0:
5547 inline_asm.volatile \"mfence\", \"\", \"memory\"()
5548 return
5549";
5550 assert_eq!(body(source), expected);
5551 }
5552
5553 #[test]
5554 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
5555 let source = "\
5558int f(int x, int y) {
5559 int r;
5560 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
5561 return r + y;
5562}
5563";
5564 let expected = "\
5565block0(%0: i32, %1: i32):
5566 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
5567 %4 = add.nsw %2, %3
5568 return %4
5569";
5570 assert_eq!(body(source), expected);
5571 }
5572
5573 #[test]
5574 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
5575 let source = "\
5580struct pair { int a, b; };
5581int f(int x) {
5582 int slot = x;
5583 struct pair p = { x, x };
5584 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
5585 return slot + p.a;
5586}
5587";
5588 let text = body(source);
5589 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
5590 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
5591 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
5592 }
5593
5594 #[test]
5595 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
5596 let source = "\
5601int f(int x) {
5602 int r = 7;
5603 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
5604 return r;
5605away:
5606 return r;
5607}
5608";
5609 let expected = "\
5610block0(%0: i32):
5611 %1 = iconst.i32 7
5612 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
5613
5614block1:
5615 return %2
5616
5617block2:
5618 return %1
5619";
5620 assert_eq!(body(source), expected);
5621 }
5622
5623 #[test]
5624 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
5625 let mut opts = options();
5629 opts.emit = EmitKind::Ir;
5630 for (source, expected) in [
5631 (
5632 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
5633 "output operand constraint lacks '='",
5634 ),
5635 (
5636 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
5637 "lvalue required in 'asm' statement",
5638 ),
5639 (
5640 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
5641 "read-only variable 'g' used as 'asm' output",
5642 ),
5643 (
5644 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
5645 "input operand constraint contains '='",
5646 ),
5647 (
5648 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
5649 "memory input 0 is not directly addressable",
5650 ),
5651 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
5652 (
5653 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
5654 "duplicate asm operand name 'a'",
5655 ),
5656 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
5657 ] {
5658 let result = run(&opts, source);
5659 assert!(result.failed(), "expected this to be reported:\n{source}");
5660 assert!(
5661 result.messages.iter().any(|m| m.contains(expected)),
5662 "{expected}\n{:?}",
5663 result.messages
5664 );
5665 }
5666 }
5667
5668 #[test]
5669 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
5670 let mut opts = options();
5671 opts.emit = EmitKind::Ir;
5672 for source in [
5673 "int f(int n) { void *p = &&out; if (n) goto *p; { int a[n]; out: return 1; } }\n",
5674 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
5675 ] {
5676 let result = run(&opts, source);
5677 assert!(result.failed(), "expected this to be reported:\n{source}");
5678 assert!(
5679 result.messages.iter().any(|m| m.contains("not supported yet")),
5680 "{:?}",
5681 result.messages
5682 );
5683 }
5684 }
5685
5686 fn round_trip(source: &str) -> (String, String) {
5688 let printed = ir(source);
5689 let mut opts = options();
5690 opts.emit = EmitKind::Ir;
5691 let mut fs = MemoryFileSystem::new();
5692 fs.insert("/main.ir", printed.clone().into_bytes());
5693 let result = compile_ir(&opts, "/main.ir", &fs);
5694 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
5695 (printed, result.text().to_owned())
5696 }
5697
5698 #[test]
5699 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
5700 let (printed, again) = round_trip(
5704 "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",
5705 );
5706 assert_eq!(printed, again);
5707 }
5708
5709 #[test]
5710 fn ir_that_is_not_ir_says_which_line_stopped_it() {
5711 let mut opts = options();
5712 opts.emit = EmitKind::Ir;
5713 let mut fs = MemoryFileSystem::new();
5714 let text = "\
5715; ModuleID = 'a.c'
5716; format 0
5717target triple = \"x86_64-unknown-linux-gnu\"
5718target datalayout = \"e-p:64:64-i64:64-S128\"
5719
5720func @f(), linkage(external) {
5721block0:
5722 frobnicate
5723}
5724";
5725 fs.insert("/main.ir", text.as_bytes().to_vec());
5726 let result = compile_ir(&opts, "/main.ir", &fs);
5727 assert!(result.failed());
5728 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
5729 }
5730
5731 #[test]
5732 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
5733 let mut opts = options();
5736 opts.emit = EmitKind::Ir;
5737 let mut fs = MemoryFileSystem::new();
5738 let text = "\
5739; ModuleID = 'a.c'
5740; format 0
5741target triple = \"x86_64-unknown-linux-gnu\"
5742target datalayout = \"e-p:64:64-i64:64-S128\"
5743
5744func @f(), linkage(external) {
5745block0:
5746 %0 = iconst.i32 1
5747 return %0
5748}
5749";
5750 fs.insert("/main.ir", text.as_bytes().to_vec());
5751 let result = compile_ir(&opts, "/main.ir", &fs);
5752 assert!(result.failed());
5753 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
5754 }
5755
5756 #[test]
5757 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
5758 let mut fs = MemoryFileSystem::new();
5760 fs.insert("/main.ir", Vec::new());
5761 let result = compile_ir(&options(), "/main.ir", &fs);
5762 assert!(result.failed());
5763 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
5764 }
5765
5766 #[test]
5767 fn the_printed_ir_reads_back_as_the_same_module() {
5768 let text = ir("\
5771struct point { int x, y; };
5772static const char greeting[] = \"hi\";
5773int table[4] = { 1, 2, 3 };
5774int puts(const char *);
5775double half(double x) { return x / 2.0; }
5776int f(int n) {
5777 int total = 0;
5778 for (int i = 0; i < n; i++) {
5779 if (i == 3) continue;
5780 total += table[i];
5781 }
5782 switch (n) {
5783 case 0: total = 1;
5784 case 1: total++; break;
5785 default: total = -total;
5786 }
5787 struct point p = { total, 1 };
5788 int *q = &p.y;
5789 puts(greeting);
5790 return p.x + *q;
5791}
5792int dispatch(int c) {
5793 void *p = c ? &&one : &&two;
5794 goto *p;
5795one:
5796 return 1;
5797two:
5798 return 2;
5799}
5800int assembly(int x, int *p) {
5801 int r;
5802 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
5803 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
5804 return r;
5805away:
5806 return 0;
5807}
5808");
5809 let mut names = Interner::new();
5810 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
5811 assert_eq!(rucc_ir::print(&module, &names), text);
5812 }
5813
5814 #[test]
5815 fn what_save_temps_keeps_is_the_text_that_was_compiled_and_the_assembly_that_was_assembled() {
5816 let mut opts = options();
5820 opts.emit = EmitKind::Object;
5821 opts.save_temps = rucc_session::SaveTemps::Object;
5822 let result = run(&opts, "#define N 2\nint a[N];\n");
5823 assert_eq!(result.messages, Vec::<String>::new());
5824 let text = result.temps.preprocessed.expect("the preprocessed text");
5825 assert!(text.contains("int a[2];"), "{text}");
5826 assert!(text.starts_with("# 1 \"/main.c\""), "{text}");
5827 let asm = result.temps.assembly.expect("the assembly");
5828 assert!(asm.contains("a:"), "{asm}");
5829 assert!(matches!(result.artifact, Artifact::Object(_)), "{:?}", result.artifact);
5830 }
5831
5832 #[test]
5833 fn nothing_is_kept_unless_the_flag_asked_for_it() {
5834 let mut opts = options();
5837 opts.emit = EmitKind::Object;
5838 assert_eq!(run(&opts, "int a;\n").temps, Temps::default());
5839 }
5840
5841 #[test]
5842 fn a_compilation_that_stops_before_the_back_end_keeps_the_text_and_no_assembly() {
5843 let mut opts = options();
5846 opts.emit = EmitKind::Ir;
5847 opts.save_temps = rucc_session::SaveTemps::Cwd;
5848 let result = run(&opts, "int a;\n");
5849 assert!(result.temps.preprocessed.is_some());
5850 assert_eq!(result.temps.assembly, None);
5851 }
5852}