1use std::path::Path;
14
15use rucc_base::Interner;
16use rucc_codegen::coverage::Fired;
17use rucc_codegen::elsewhere::Elsewhere;
18use rucc_codegen::lowering::Lowerings;
19use rucc_codegen::pipeline::{self, Machine, Recording};
20use rucc_codegen::pressure::Pressure;
21use rucc_diag::{Diagnostic, Severity, Span};
22use rucc_ir::{FpContract, Pic as IrPic, Visibility as IrVisibility};
23use rucc_lex::{Convert, Keywords, PpToken, convert};
24use rucc_lower::Protector as LowerProtector;
25use rucc_sema::{Checker, Context as CheckContext};
26use rucc_session::{
27 Contract, EmitKind, FileSystem, Options, Padding, Pic, Protector, Session, Visibility,
28};
29use rucc_target::TargetInfo;
30use rucc_tuple::{Arch, ObjectFormat};
31
32use crate::preprocess::render;
33
34#[derive(Debug, Clone, PartialEq, Eq, Default)]
41pub enum Artifact {
42 #[default]
45 Nothing,
46 Text(String),
48 Object {
55 bytes: Vec<u8>,
57 defines: Vec<String>,
61 },
62}
63
64impl Artifact {
65 #[must_use]
67 pub fn bytes(&self) -> &[u8] {
68 match self {
69 Artifact::Nothing => &[],
70 Artifact::Text(text) => text.as_bytes(),
71 Artifact::Object { bytes, .. } => bytes,
72 }
73 }
74}
75
76#[derive(Debug, Clone, PartialEq, Eq)]
78pub struct Compiled {
79 pub artifact: Artifact,
81 pub messages: Vec<String>,
83 pub errors: u32,
85 pub fired: Fired,
91 pub pressure: Pressure,
96 pub lowerings: Lowerings,
101 pub dumps: Vec<rucc_opt::Dump>,
107 pub remarks: String,
113 pub deps: Vec<rucc_pp::Dependency>,
118 pub temps: Temps,
125}
126
127#[derive(Debug, Clone, PartialEq, Eq, Default)]
134pub struct Temps {
135 pub preprocessed: Option<String>,
137 pub assembly: Option<String>,
139}
140
141impl Compiled {
142 #[must_use]
144 pub fn failed(&self) -> bool {
145 self.errors > 0
146 }
147
148 #[must_use]
153 pub fn text(&self) -> &str {
154 match &self.artifact {
155 Artifact::Text(text) => text,
156 _ => "",
157 }
158 }
159}
160
161#[must_use]
174pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
175 let mut sess = Session::new(opts.clone());
176 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
180 let mut diagnostics: Vec<Diagnostic> = Vec::new();
181 let mut fired = Fired::new();
183 let mut pressure = Pressure::new();
185 let mut lowerings = Lowerings::asked(opts.lowering_dump.is_some());
186 let mut dumps = Vec::new();
188 let mut remarks = String::new();
189 let mut temps = Temps::default();
191
192 let bytes = match fs.read(Path::new(name)) {
193 Ok(bytes) => bytes,
194 Err(e) => return failure(format!("{name}: {e}")),
195 };
196 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
197 return failure(format!("{name}: the source map has no room left for this file"));
198 };
199
200 let mut pp = rucc_pp::Preprocessor::with_prefix_map(opts.prefix_map.macros.clone());
204 let predef = rucc_pp::Predef::for_options(opts);
205 let expanded: Vec<PpToken> = {
206 let mut tokens = Vec::new();
207 {
212 let mut cx =
213 rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
214 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
215 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
216 return failure(format!(
217 "{name}: the source map has no room for the built in macros"
218 ));
219 }
220 if pp.preinclude(&opts.preincludes, &mut tokens, &mut cx).is_err() {
221 return failure(format!("{name}: the source map has no room for the command line"));
222 }
223 tokens.append(&mut pp.run(file, &mut cx));
224 }
225 if opts.save_temps.wanted() {
226 temps.preprocessed = Some(rucc_pp::print(
227 file,
228 &tokens,
229 pp.line_directives(),
230 &sess.sources,
231 &sess.interner,
232 rucc_pp::PrintOptions { line_markers: opts.line_markers },
233 ));
234 }
235 tokens.iter().map(|token| token.to_pp()).collect()
236 };
237 diagnostics.extend(pp.take_diagnostics());
238 let deps = pp.dependencies().to_vec();
241
242 let cx = Convert {
245 keywords: &keywords,
246 interner: &sess.interner,
247 target: &sess.target,
248 std: opts.std,
249 gnu: opts.gnu_extensions,
250 pedantic: opts.pedantic,
251 };
252 let (tokens, complaints) = convert(&expanded, &cx);
253 diagnostics.extend(complaints);
254
255 let parsed = rucc_parse::parse(
256 &tokens,
257 rucc_parse::Context {
258 interner: &sess.interner,
259 std: opts.std,
260 gnu: opts.gnu_extensions,
261 pedantic: opts.pedantic,
262 error_limit: opts.error_limit as usize,
263 },
264 );
265 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
266 diagnostics.extend(parsed.diagnostics);
267
268 let mut artifact = Artifact::Nothing;
269 let mut instrumented = Instrumented::default();
272 if !parse_failed {
273 let mut checker = Checker::new(
274 &parsed.ast,
275 CheckContext {
276 names: &sess.interner,
277 target: &sess.target,
278 std: opts.std,
279 gnu: opts.gnu_extensions,
280 pedantic: opts.pedantic,
281 permissive: opts.permissive,
282 gnu89_inline: opts.gnu89_inline,
283 error_limit: opts.error_limit as usize,
284 builtins: opts.builtins && opts.hosted,
287 no_builtin: &opts.no_builtin,
288 short_enums: opts.short_enums,
289 trapping_math: opts.trapping_math,
290 },
291 );
292 checker.check_unit();
293 let checked = checker.finish();
294 if !checked.failed() {
295 match opts.emit {
296 EmitKind::Tast => {
297 artifact = Artifact::Text(rucc_sema::print(
298 &checked.tast,
299 &checked.types,
300 &sess.interner,
301 ));
302 }
303 EmitKind::TypeGranules => {
307 artifact = Artifact::Text(rucc_types::granule_report(
308 &checked.types,
309 &sess.interner,
310 &sess.target,
311 ));
312 }
313 EmitKind::Ir
314 | EmitKind::MirFinal
315 | EmitKind::Asm
316 | EmitKind::Object
317 | EmitKind::Archive
318 | EmitKind::Executable
319 | EmitKind::SafetySummary => {
320 let mut read = |named: &str| {
325 fs.read(Path::new(named))
326 .map(|bytes| bytes.as_slice().to_vec())
327 .map_err(|why| why.to_string())
328 };
329 let mut lowered = rucc_lower::lower(
330 name,
331 rucc_lower::Context {
332 tast: &checked.tast,
333 types: &checked.types,
334 target: &sess.target,
335 names: &mut sess.interner,
336 visibility: match opts.visibility {
337 Visibility::Default => IrVisibility::Default,
338 Visibility::Hidden => IrVisibility::Hidden,
339 Visibility::Protected => IrVisibility::Protected,
340 },
341 protector: match opts.protector {
342 Protector::None => LowerProtector::None,
343 Protector::Buffers => LowerProtector::Buffers,
344 Protector::Strong => LowerProtector::Strong,
345 Protector::All => LowerProtector::All,
346 },
347 wrapping: rucc_lower::Wrapping {
348 signed: opts.wrapping.signed,
349 pointer: opts.wrapping.pointer,
350 trap: opts.wrapping.trap,
351 },
352 aliasing: opts.strict_aliasing,
353 padding: opts.padding == Padding::Ignored,
354 contract: match opts.fp_contract {
355 Contract::Off => FpContract::Off,
356 Contract::On => FpContract::On,
357 Contract::Fast => FpContract::Fast,
358 },
359 read: &mut read,
360 },
361 );
362 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
366 if !failed {
367 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
372 for error in errors {
373 diagnostics.push(internal(&format!("invalid IR, {error}")));
374 }
375 } else if let Err(complaints) =
376 instrument(&mut lowered.module, &mut sess.interner, opts)
377 .map(|done| instrumented = done)
378 {
379 diagnostics.extend(complaints);
380 } else if let Err(complaints) = optimize(
381 &mut lowered.module,
382 &sess.interner,
383 &sess.target,
384 opts,
385 name,
386 &mut dumps,
387 &mut remarks,
388 ) {
389 diagnostics.extend(complaints);
390 } else if opts.emit == EmitKind::SafetySummary {
391 artifact = Artifact::Text(
396 rucc_safety::summarize(
397 &lowered.module,
398 &sess.interner,
399 name,
400 opts.safety.as_str(),
401 instrumented.checks,
402 instrumented.interposed,
403 instrumented.crossings,
404 )
405 .render(),
406 );
407 } else if opts.emit == EmitKind::Ir {
408 artifact =
413 Artifact::Text(rucc_ir::print(&lowered.module, &sess.interner));
414 } else {
415 match generate(
418 &mut lowered.module,
419 &mut sess.interner,
420 &sess.target,
421 opts,
422 &mut Recording {
423 fired: &mut fired,
424 pressure: &mut pressure,
425 lowerings: &mut lowerings,
426 },
427 &mut temps.assembly,
428 ) {
429 Ok(made) => artifact = made,
430 Err(complaints) => diagnostics.extend(complaints),
431 }
432 }
433 }
434 diagnostics.extend(lowered.diagnostics);
435 }
436 _ => {}
437 }
438 }
439 diagnostics.extend(checked.diagnostics);
440 }
441
442 let mut messages = Vec::with_capacity(diagnostics.len());
443 let mut errors = 0;
444 for diag in &diagnostics {
445 if !opts.warnings && diag.severity == Severity::Warning {
449 continue;
450 }
451 if diag.severity.is_fatal()
452 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
453 {
454 errors += 1;
455 }
456 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
457 }
458 if errors > 0 {
459 artifact = Artifact::Nothing;
461 }
462 Compiled { artifact, messages, errors, fired, pressure, lowerings, dumps, remarks, deps, temps }
465}
466
467#[must_use]
477pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
478 let mut sess = Session::new(opts.clone());
479 if opts.emit != EmitKind::Ir {
480 return failure(format!(
481 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
482 the C in front of it became",
483 opts.emit.as_str()
484 ));
485 }
486 let bytes = match fs.read(Path::new(name)) {
487 Ok(bytes) => bytes,
488 Err(e) => return failure(format!("{name}: {e}")),
489 };
490 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
491 return failure(format!("{name}: this is not text, so it is not IR"));
492 };
493
494 let module = match rucc_ir::parse(text, &mut sess.interner) {
495 Ok(module) => module,
496 Err(error) => {
497 return failure(format!("{name}:{}: {}", error.line, error.message));
498 }
499 };
500 let mut diagnostics: Vec<Diagnostic> = Vec::new();
501 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
502 for error in errors {
503 diagnostics.push(invalid(&format!("invalid IR, {error}")));
504 }
505 }
506 let mut messages = Vec::with_capacity(diagnostics.len());
507 for diag in &diagnostics {
508 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
509 }
510 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
511 let artifact = if errors > 0 {
512 Artifact::Nothing
513 } else {
514 Artifact::Text(rucc_ir::print(&module, &sess.interner))
515 };
516 Compiled {
518 artifact,
519 messages,
520 errors,
521 fired: Fired::new(),
522 pressure: Pressure::new(),
523 lowerings: Lowerings::new(),
524 dumps: Vec::new(),
525 remarks: String::new(),
526 deps: Vec::new(),
527 temps: Temps::default(),
528 }
529}
530
531fn instrument(
554 module: &mut rucc_ir::Module,
555 names: &mut Interner,
556 opts: &Options,
557) -> Result<Instrumented, Vec<Diagnostic>> {
558 if !opts.safety.instruments() {
559 return Ok(Instrumented::default());
560 }
561 let checks = rucc_safety::run(module, opts.subobject, opts.promise, opts.races);
562 let interposed = rucc_safety::redirect(module, names);
567 let crossings = rucc_safety::witness(module, names);
570 match rucc_ir::verify(module, names) {
571 Ok(()) => Ok(Instrumented { checks, interposed, crossings }),
572 Err(errors) => Err(errors
573 .iter()
574 .map(|e| internal(&format!("invalid IR after check insertion, {e}")))
575 .collect()),
576 }
577}
578
579#[derive(Clone, Copy, Debug, Default)]
585struct Instrumented {
586 checks: rucc_safety::Counts,
588 interposed: usize,
590 crossings: rucc_safety::Sites,
592}
593
594fn optimize(
606 module: &mut rucc_ir::Module,
607 names: &Interner,
608 target: &TargetInfo,
609 opts: &Options,
610 file: &str,
611 dumps: &mut Vec<rucc_opt::Dump>,
612 remarks: &mut String,
613) -> Result<(), Vec<Diagnostic>> {
614 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
615 settings.interposition = match opts.interposition {
621 true => replaceable(target, opts),
622 false => IrPic::Executable,
623 };
624 settings.toggles.clone_from(&opts.passes);
625 settings.fuel = opts.pass_fuel.iter().cloned().collect();
626 settings.global_fuel = opts.pass_fuel_global;
627 settings.verify |= opts.verify_each;
628 for (on, spec) in &opts.pass_gates {
629 if let Err(why) = settings.gates.add(*on, spec) {
632 return Err(vec![internal(&why)]);
633 }
634 }
635 for spec in &opts.dump_ir {
636 if let Err(why) = settings.dumps.add(spec) {
639 return Err(vec![internal(&why)]);
640 }
641 }
642 let mut wants = rucc_opt::Wants::none();
643 for spec in &opts.opt_info {
644 if let Err(why) = wants.add(spec) {
647 return Err(vec![internal(&why)]);
648 }
649 }
650 let report = rucc_opt::run(module, names, &settings);
651 remarks.push_str(&rucc_opt::optinfo::render(file, &report, names, wants));
652 dumps.extend(report.dumps);
653 match report.broke.is_empty() {
654 true => Ok(()),
655 false => Err(report.broke.iter().map(|why| internal(why)).collect()),
656 }
657}
658
659fn replaceable(target: &TargetInfo, opts: &Options) -> IrPic {
693 match (target.tuple.os().object_format(), opts.pic) {
694 (Some(ObjectFormat::Elf), Pic::Library) => IrPic::Library,
695 _ => IrPic::Executable,
696 }
697}
698
699fn generate(
700 module: &mut rucc_ir::Module,
701 names: &mut Interner,
702 target: &TargetInfo,
703 opts: &Options,
704 recording: &mut Recording<'_>,
705 assembly: &mut Option<String>,
706) -> Result<Artifact, Vec<Diagnostic>> {
707 let Some(machine) = Machine::for_target(target) else {
708 return Err(vec![unsupported(&format!(
709 "there is no back end for {} in this compiler yet, so there is nothing to generate",
710 target.tuple
711 ))]);
712 };
713 if opts.protector != Protector::None && machine.conv.guard.is_none() {
718 return Err(vec![unsupported(&format!(
719 "{} is not supported for {} yet, because the stack protector on that target is not \
720 the one this compiler writes",
721 opts.protector, target.tuple
722 ))]);
723 }
724 if opts.control.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
730 return Err(vec![unsupported(&format!(
731 "-fcf-protection={} is not supported for {} yet, because what says a file was built \
732 for it there is not the note this compiler writes",
733 opts.control, target.tuple
734 ))]);
735 }
736 let profile = match machine.conv.trace {
742 Some(trace) => opts.profile.then(|| opts.hook.early(trace.fentry)),
743 None if opts.profile => {
744 return Err(vec![unsupported(&format!(
745 "-pg is not supported for {} yet, because the profiler's hook on that target is \
746 not the one this compiler calls",
747 target.tuple
748 ))]);
749 }
750 None => None,
751 };
752 if opts.patchable.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
757 return Err(vec![unsupported(&format!(
758 "-fpatchable-function-entry= is not supported for {} yet, because what records where \
759 the room is there is not the section this compiler writes",
760 target.tuple
761 ))]);
762 }
763 let flags = pipeline::Flags {
764 frame_pointer: opts.frame_pointer,
765 red_zone: opts.red_zone,
766 stack_clash: opts.stack_clash,
767 landing: opts.control.branch(),
768 profile: match profile {
769 None => pipeline::Profile::No,
770 Some(true) => pipeline::Profile::Early,
771 Some(false) => pipeline::Profile::Late,
772 },
773 patch: pipeline::Room { after: opts.patchable.after(), before: opts.patchable.before },
774 reorder: opts.reorder_blocks.unwrap_or_else(|| opts.opt_level.runs_optimizer()),
781 reuse: opts.stack_reuse.unwrap_or_else(|| opts.opt_level.runs_optimizer()),
786 schedule: opts.schedule_insns.unwrap_or_else(|| opts.opt_level.schedules()),
791 accurate: opts.cycle_accurate_model,
793 };
794
795 if opts.safety.instruments() {
804 rucc_opt::heap::annotate(module, names);
814 rucc_safety::lower(module, names);
815 if let Err(errors) = rucc_ir::verify(module, names) {
816 return Err(errors
817 .iter()
818 .map(|e| internal(&format!("invalid IR after check lowering, {e}")))
819 .collect());
820 }
821 }
822
823 let elsewhere = Elsewhere::of(module, replaceable(target, opts));
831
832 let mut funcs = Vec::new();
833 let mut complaints = Vec::new();
834 for id in module.funcs() {
835 if module[id].is_declaration() {
836 continue;
837 }
838 match pipeline::compile_recording(
839 &mut module[id],
840 names,
841 &machine,
842 &elsewhere,
843 flags,
844 recording,
845 ) {
846 Ok(func) => funcs.push(func),
847 Err(why) => {
848 let name = names.resolve(module[id].name).to_owned();
849 let span = why.inst().map_or(Span::DUMMY, |inst| module[id].span(inst));
852 let said = format!("cannot generate code for '{name}': {why}");
853 complaints.push(unsupported_at(&said, span));
854 }
855 }
856 }
857 if !complaints.is_empty() {
858 return Err(complaints);
859 }
860 let (globals, aliases) = match opts.emit {
866 EmitKind::Asm | EmitKind::Object | EmitKind::Archive | EmitKind::Executable => (
867 rucc_asm::globals(module, names, target.object_format).map_err(refused)?,
868 rucc_asm::aliases(module, names).map_err(refused)?,
869 ),
870 _ => (rucc_asm::Globals::default(), Vec::new()),
871 };
872 let unwind = opts.unwinds();
876 match opts.emit {
877 EmitKind::Asm => {
878 rucc_asm::print(&funcs, &globals, &aliases, names, target, unwind, output(opts, target))
879 .map(Artifact::Text)
880 .map_err(refused)
881 }
882 EmitKind::Object | EmitKind::Archive | EmitKind::Executable => {
886 if opts.save_temps.wanted() {
887 let listing = rucc_asm::print(
888 &funcs,
889 &globals,
890 &aliases,
891 names,
892 target,
893 unwind,
894 output(opts, target),
895 );
896 *assembly = Some(listing.map_err(refused)?);
897 }
898 let text = rucc_asm::assemble(&funcs, names, target, unwind).map_err(refused)?;
899 let data = globals.image();
900 let bytes = rucc_object::write(&text, &data, &aliases, target, output(opts, target))
903 .map_err(wrote)?;
904 let defines = rucc_object::defines(&text, &data, &aliases, target).map_err(wrote)?;
909 Ok(Artifact::Object { bytes, defines })
910 }
911 _ => Ok(Artifact::Text(rucc_mir::print(&funcs, names, target.regs))),
912 }
913}
914
915fn output(opts: &Options, target: &TargetInfo) -> rucc_object::Output {
927 let mut features = 0;
928 if target.tuple.arch() == Arch::X86_64 {
929 if opts.control.branch() {
930 features |= rucc_object::Property::IBT;
931 }
932 if opts.control.ret() {
933 features |= rucc_object::Property::SHSTK;
934 }
935 }
936 rucc_object::Output {
937 sections: rucc_object::Sections {
938 functions: opts.function_sections,
939 data: opts.data_sections,
940 },
941 property: rucc_object::Property { features },
942 }
943}
944
945fn wrote(why: rucc_object::Error) -> Vec<Diagnostic> {
951 match why {
952 rucc_object::Error::Format { .. } => vec![unsupported(&why.to_string())],
953 rucc_object::Error::Refused { .. } => vec![internal(&why.to_string())],
954 }
955}
956
957fn refused(why: rucc_asm::Error) -> Vec<Diagnostic> {
963 match why {
964 rucc_asm::Error::Thread { .. } | rucc_asm::Error::IFunc { .. } => {
965 vec![unsupported(&why.to_string())]
966 }
967 _ => vec![internal(&why.to_string())],
968 }
969}
970
971fn unsupported(message: &str) -> Diagnostic {
977 unsupported_at(message, Span::DUMMY)
978}
979
980fn unsupported_at(message: &str, span: Span) -> Diagnostic {
986 Diagnostic::error(message.to_owned(), span)
987 .with_code("E0653")
988 .note("this construct is not lowered yet, see https://github.com/tamnd/rucc/issues", span)
989}
990
991fn invalid(message: &str) -> Diagnostic {
993 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
994}
995
996fn internal(message: &str) -> Diagnostic {
998 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
999 .with_code("E0652")
1000 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
1001}
1002
1003fn failure(message: String) -> Compiled {
1006 Compiled {
1007 artifact: Artifact::Nothing,
1008 messages: vec![format!("rucc: error: {message}")],
1009 errors: 1,
1010 fired: Fired::new(),
1011 pressure: Pressure::new(),
1012 lowerings: Lowerings::new(),
1013 dumps: Vec::new(),
1014 remarks: String::new(),
1015 deps: Vec::new(),
1016 temps: Temps::default(),
1017 }
1018}
1019
1020#[cfg(test)]
1021mod tests {
1022 use rucc_session::{MemoryFileSystem, Std};
1023 use rucc_target::Triple;
1024
1025 use super::*;
1026
1027 fn options() -> Options {
1028 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
1029 opts.emit = EmitKind::Tast;
1030 opts
1031 }
1032
1033 fn run(opts: &Options, source: &str) -> Compiled {
1034 let mut fs = MemoryFileSystem::new();
1035 fs.insert("/main.c", source.to_owned().into_bytes());
1036 compile(opts, "/main.c", &fs)
1037 }
1038
1039 fn freestanding() -> Options {
1043 let mut opts = options();
1044 opts.hosted = false;
1045 opts.search.push_system(rucc_session::runtime::DIR);
1046 opts
1047 }
1048
1049 fn shipped(source: &str) -> String {
1051 let result = run(&freestanding(), source);
1052 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1053 result.text().to_owned()
1054 }
1055
1056 fn tast(source: &str) -> String {
1058 let result = run(&options(), source);
1059 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1060 result.text().to_owned()
1061 }
1062
1063 #[test]
1064 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
1065 let text = shipped(concat!(
1066 "#include <stdarg.h>\n",
1067 "int sum(int n, ...) {\n",
1068 " va_list ap, copy;\n",
1069 " va_start(ap, n);\n",
1070 " va_copy(copy, ap);\n",
1071 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
1072 " va_end(ap);\n",
1073 " va_end(copy);\n",
1074 " return total;\n",
1075 "}\n",
1076 ));
1077 assert!(text.contains("va-start"), "{text}");
1078 assert!(text.contains("va-copy"), "{text}");
1079 assert!(text.contains("va-arg"), "{text}");
1080 assert!(text.contains("va-end"), "{text}");
1081 }
1082
1083 #[test]
1087 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
1088 let text = shipped(concat!(
1089 "#define __need___va_list\n",
1090 "#include <stdarg.h>\n",
1091 "int vprint(const char *f, __gnuc_va_list ap);\n",
1092 "#ifdef va_start\n",
1093 "#error va_start should not be defined\n",
1094 "#endif\n",
1095 "#ifdef _VA_LIST_DEFINED\n",
1096 "#error va_list should not have been made\n",
1097 "#endif\n",
1098 ));
1099 assert!(text.contains("vprint"), "{text}");
1100 }
1101
1102 #[test]
1105 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
1106 let text = shipped(concat!(
1107 "#define __need_size_t\n",
1108 "#include <stddef.h>\n",
1109 "#ifdef offsetof\n",
1110 "#error offsetof should not be defined yet\n",
1111 "#endif\n",
1112 "#define __need_ptrdiff_t\n",
1113 "#include <stddef.h>\n",
1114 "#include <stddef.h>\n",
1115 "size_t a;\n",
1116 "ptrdiff_t b;\n",
1117 "wchar_t c;\n",
1118 "max_align_t d;\n",
1119 "void *e = NULL;\n",
1120 "struct P { int x; long y; };\n",
1121 "size_t f = offsetof(struct P, y);\n",
1122 ));
1123 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
1124 assert!(text.contains("decl #1 b : long"), "{text}");
1125 }
1126
1127 #[test]
1128 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
1129 let text = shipped(concat!(
1130 "#include <limits.h>\n",
1131 "#include <float.h>\n",
1132 "int bits = CHAR_BIT;\n",
1133 "long big = LONG_MAX;\n",
1134 "int low = INT_MIN;\n",
1135 "int radix = FLT_RADIX;\n",
1136 "int digits = DBL_MANT_DIG;\n",
1137 ));
1138 assert!(text.contains("const 8 : int"), "{text}");
1139 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
1140 assert!(text.contains("const 2 : int"), "{text}");
1141 assert!(text.contains("const 53 : int"), "{text}");
1142 }
1143
1144 #[test]
1148 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
1149 let text = shipped(concat!(
1150 "#include <stdint.h>\n",
1151 "int64_t a = INT64_C(1);\n",
1152 "uint_least16_t b;\n",
1153 "intptr_t c;\n",
1154 "uintmax_t d = UINTMAX_MAX;\n",
1155 "int wide = sizeof(int_fast64_t);\n",
1156 ));
1157 assert!(text.contains("decl #0 a : long"), "{text}");
1158 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
1159 assert!(text.contains("decl #2 c : long"), "{text}");
1160 }
1161
1162 #[test]
1173 fn the_shipped_mmintrin_defines_the_mmx_type_and_the_operations_over_it() {
1174 let text = shipped(concat!(
1175 "#include <mmintrin.h>\n",
1176 "__m64 add(__m64 a, __m64 b) { return _mm_add_pi16(a, b); }\n",
1177 "__m64 pack(__m64 a, __m64 b) { return _m_packsswb(a, b); }\n",
1178 "__m64 shift(__m64 a) { return _mm_srai_pi32(a, 3); }\n",
1179 "int low(__m64 a) { return _mm_cvtsi64_si32(a); }\n",
1180 "void done(void) { _mm_empty(); }\n",
1181 ));
1182 assert!(text.contains("add"), "{text}");
1183 assert!(text.contains("pack"), "{text}");
1184 assert!(text.contains("shift"), "{text}");
1185 }
1186
1187 #[test]
1192 fn the_shipped_mm_malloc_asks_for_aligned_memory_and_gives_it_back() {
1193 let text = shipped(concat!(
1194 "#include <mm_malloc.h>\n",
1195 "void *get(void) { return _mm_malloc(64, 16); }\n",
1196 "void put(void *p) { _mm_free(p); }\n",
1197 ));
1198 assert!(text.contains("get"), "{text}");
1199 assert!(text.contains("put"), "{text}");
1200 }
1201
1202 #[test]
1214 fn the_shipped_xmmintrin_defines_the_sse_type_and_the_operations_over_it() {
1215 let text = shipped(concat!(
1216 "#include <xmmintrin.h>\n",
1217 "__m128 add(__m128 a, __m128 b) { return _mm_add_ps(a, b); }\n",
1218 "__m128 one(__m128 a, __m128 b) { return _mm_max_ss(a, b); }\n",
1219 "__m128 mask(__m128 a, __m128 b) { return _mm_cmpnle_ps(a, b); }\n",
1220 "__m128 pick(__m128 a, __m128 b) { return _mm_shuffle_ps(a, b, _MM_SHUFFLE(0,1,2,3)); }\n",
1221 "int bits(__m128 a) { return _mm_movemask_ps(a); }\n",
1222 "int near(__m128 a) { return _mm_cvtss_si32(a); }\n",
1223 "__m128 wide(__m64 a) { return _mm_cvtpi16_ps(a); }\n",
1224 "void *room(void) { return _mm_malloc(64, 16); }\n",
1225 "void hint(const float *p) { _mm_prefetch(p, _MM_HINT_T0); _mm_sfence(); }\n",
1226 ));
1227 assert!(text.contains("add"), "{text}");
1228 assert!(text.contains("mask"), "{text}");
1229 assert!(text.contains("pick"), "{text}");
1230 assert!(text.contains("wide"), "{text}");
1231 }
1232
1233 #[test]
1240 fn the_shipped_xmmintrin_leaves_out_the_names_that_need_an_instruction() {
1241 let text = rucc_session::runtime::header("xmmintrin.h").expect("xmmintrin.h is shipped");
1242 for absent in [
1243 "_mm_sqrt_ps",
1244 "_mm_sqrt_ss",
1245 "_mm_rsqrt_ps",
1246 "_mm_rsqrt_ss",
1247 "_mm_getcsr",
1248 "_mm_setcsr",
1249 ] {
1250 let defined = text.contains(&format!("{absent}("));
1251 assert!(!defined, "{absent} is defined and the header says it is not");
1252 assert!(text.contains(absent), "{absent} is absent and unexplained");
1253 }
1254 }
1255
1256 #[test]
1257 fn the_shipped_emmintrin_defines_both_sse2_types_and_the_operations_over_them() {
1258 let text = shipped(concat!(
1259 "#include <emmintrin.h>\n",
1260 "__m128i add(__m128i a, __m128i b) { return _mm_add_epi64(a, b); }\n",
1261 "__m128i wide(__m128i a, __m128i b) { return _mm_mul_epu32(a, b); }\n",
1262 "__m128i pick(__m128i a) { return _mm_shuffle_epi32(a, _MM_SHUFFLE(0,1,2,3)); }\n",
1263 "__m128i up(__m128i a) { return _mm_slli_epi64(a, 13); }\n",
1264 "__m128i down(__m128i a) { return _mm_srli_si128(a, 3); }\n",
1265 "__m128i pack(__m128i a, __m128i b) { return _mm_packus_epi16(a, b); }\n",
1266 "int bits(__m128i a) { return _mm_movemask_epi8(a); }\n",
1267 "__m128d sum(__m128d a, __m128d b) { return _mm_add_sd(a, b); }\n",
1268 "__m128d mask(__m128d a, __m128d b) { return _mm_cmpunord_pd(a, b); }\n",
1269 "__m128i near(__m128d a) { return _mm_cvtpd_epi32(a); }\n",
1270 "__m128d over(__m128 a) { return _mm_cvtps_pd(a); }\n",
1271 "__m128i half(__m64 a) { return _mm_movpi64_epi64(a); }\n",
1272 "__m128i grab(void const *p) { return _mm_loadu_si128(p); }\n",
1273 "void wall(void) { _mm_lfence(); _mm_mfence(); }\n",
1274 ));
1275 assert!(text.contains("wide"), "{text}");
1276 assert!(text.contains("pack"), "{text}");
1277 assert!(text.contains("near"), "{text}");
1278 assert!(text.contains("half"), "{text}");
1279 }
1280
1281 #[test]
1285 fn the_shipped_immintrin_reaches_the_names_the_headers_under_it_define() {
1286 let text = shipped(concat!(
1287 "#include <immintrin.h>\n",
1288 "unsigned long long matching(unsigned char tag, unsigned char const *bucket) {\n",
1289 " __m128i const want = _mm_set1_epi8((char)tag);\n",
1290 " __m128i const chunk = _mm_loadu_si128((__m128i const *)(void const *)bucket);\n",
1291 " __m128i const same = _mm_cmpeq_epi8(chunk, want);\n",
1292 " return (unsigned long long)_mm_movemask_epi8(same);\n",
1293 "}\n",
1294 "__m64 narrow(__m64 a, __m64 b) { return _mm_add_pi32(a, b); }\n",
1295 "__m128 single(__m128 a, __m128 b) { return _mm_add_ps(a, b); }\n",
1296 ));
1297 assert!(text.contains("matching"), "{text}");
1298 assert!(text.contains("narrow"), "the MMX header is not reached: {text}");
1299 assert!(text.contains("single"), "the SSE header is not reached: {text}");
1300 }
1301
1302 #[test]
1306 fn the_umbrella_and_the_header_under_it_can_both_be_included() {
1307 let text = shipped(concat!(
1308 "#include <immintrin.h>\n",
1309 "#include <emmintrin.h>\n",
1310 "#include <immintrin.h>\n",
1311 "__m128i twice(__m128i a, __m128i b) { return _mm_add_epi32(a, b); }\n",
1312 ));
1313 assert!(text.contains("twice"), "{text}");
1314 }
1315
1316 #[test]
1320 fn the_shipped_emmintrin_leaves_out_the_two_square_roots() {
1321 let text = rucc_session::runtime::header("emmintrin.h").expect("emmintrin.h is shipped");
1322 for absent in ["_mm_sqrt_pd", "_mm_sqrt_sd"] {
1323 let defined = text.contains(&format!("{absent}("));
1324 assert!(!defined, "{absent} is defined and the header says it is not");
1325 assert!(text.contains(absent), "{absent} is absent and unexplained");
1326 }
1327 }
1328
1329 #[test]
1330 fn the_three_formality_headers_still_have_to_work() {
1331 let text = shipped(concat!(
1332 "#include <stdbool.h>\n",
1333 "#include <stdalign.h>\n",
1334 "#include <iso646.h>\n",
1335 "#include <stdnoreturn.h>\n",
1336 "int t = true and not false;\n",
1337 "_Alignas(16) char buf[16];\n",
1338 "int a = alignof(long);\n",
1339 ));
1340 assert!(text.contains("decl #0 t : int"), "{text}");
1341 assert!(text.contains("const 8 : unsigned long"), "{text}");
1342 }
1343
1344 #[test]
1352 fn every_shipped_header_can_be_included_twice() {
1353 let once: String = rucc_session::runtime::names()
1354 .iter()
1355 .map(|name| format!("#include <{name}>\n"))
1356 .collect();
1357 let twice = once.repeat(2);
1358 assert_eq!(shipped(&format!("{once}int x;\n")), shipped(&format!("{twice}int x;\n")));
1359 }
1360
1361 #[test]
1362 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
1363 let fs = MemoryFileSystem::new();
1364 let result = compile(&options(), "/nope.c", &fs);
1365 assert!(result.failed());
1366 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
1367 assert!(result.text().is_empty());
1368 }
1369
1370 #[test]
1371 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
1372 let text = tast("int x = 1;\n");
1373 let expected = "\
1374decl #0 x : int object external static defined
1375 init
1376 +0
1377 const 1 : int
1378";
1379 assert_eq!(text, expected);
1380 }
1381
1382 #[test]
1383 fn the_macros_are_expanded_before_anything_is_parsed() {
1384 let text = tast("#define N 2\nint a[N];\n");
1388 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
1389 }
1390
1391 #[test]
1397 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
1398 let text = tast(concat!(
1399 "#pragma pack(4)\n",
1400 "struct s { int a; };\n",
1401 "#pragma pack()\n",
1402 "int b;\n",
1403 "_Pragma(\"GCC visibility push(default)\") int c;\n",
1404 ));
1405 assert!(text.contains("decl #0 b : int"), "{text}");
1406 assert!(text.contains("decl #1 c : int"), "{text}");
1407 }
1408
1409 #[test]
1417 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
1418 tast(concat!(
1419 "struct A { char c; int i; } __attribute__((packed));\n",
1420 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1421 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1422 "struct B { char c; int i; } __attribute__((aligned));\n",
1425 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
1426 "struct C { char c; int i __attribute__((packed)); };\n",
1427 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
1428 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
1429 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
1430 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
1431 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
1432 "struct E { char c; _Alignas(8) int i; };\n",
1433 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
1434 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
1435 "struct F { char c; int i __attribute__((aligned(8))); };\n",
1436 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
1437 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
1440 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
1441 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
1442 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
1443 "struct I { [[gnu::packed]] char c; int i; };\n",
1446 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1447 "struct J { char c; [[gnu::packed]] int i; };\n",
1448 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
1449 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
1450 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
1451 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
1452 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
1453 "union L { char c; int i; } __attribute__((packed));\n",
1454 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
1455 "struct O { char c; int i; } __attribute__((__packed__));\n",
1459 "_Static_assert(sizeof(struct O) == 5 && _Alignof(struct O) == 1, \"O\");\n",
1460 "struct P { char c; int i; } __attribute__((__aligned__(8)));\n",
1461 "_Static_assert(sizeof(struct P) == 8 && _Alignof(struct P) == 8, \"P\");\n",
1462 ));
1463 }
1464
1465 #[test]
1478 fn a_transparent_union_takes_the_member_a_value_fits_and_is_declared_either_way() {
1479 let text = tast(concat!(
1480 "struct one { int x; };\n",
1481 "struct two { long y; };\n",
1482 "typedef union { struct one *a; struct two *b; void *any; }\n",
1483 " __attribute__((__transparent_union__)) arg;\n",
1484 "int takes(arg v);\n",
1485 "int f(struct one *p, struct two *q, char *c) {\n",
1486 " return takes(p) + takes(q) + takes(c) + takes(0);\n",
1487 "}\n",
1488 "int takes(struct one *p);\n",
1490 "int (*as_a_member)(struct one *) = takes;\n",
1491 "int (*as_the_union)(arg) = takes;\n",
1492 ));
1493 assert!(text.contains("compound-literal"), "{text}");
1494 }
1495
1496 #[test]
1502 fn the_attribute_on_the_declarator_of_a_typedef_is_the_one_glibc_writes() {
1503 let text = tast(concat!(
1504 "struct sockaddr { int family; };\n",
1505 "struct sockaddr_in { int family; int addr; };\n",
1506 "typedef union { struct sockaddr *plain; struct sockaddr_in *inet; }\n",
1507 " addr_arg __attribute__((__transparent_union__));\n",
1508 "int bind_to(int fd, addr_arg where);\n",
1509 "int f(struct sockaddr_in *where) { return bind_to(0, where); }\n",
1510 ));
1511 assert!(text.contains("compound-literal"), "{text}");
1512 }
1513
1514 #[test]
1522 fn a_transparent_union_that_cannot_keep_the_promise_is_dropped_with_a_word_about_it() {
1523 let result = run(
1524 &options(),
1525 concat!(
1526 "union wider { int small; double large; } __attribute__((transparent_union));\n",
1527 "struct plain { int x; } __attribute__((transparent_union));\n",
1528 ),
1529 );
1530 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
1531 assert!(!result.failed(), "{:?}", result.messages);
1532 for message in &result.messages {
1533 assert!(message.contains("'transparent_union' attribute ignored"), "{message}");
1534 }
1535 assert!(result.messages[0].contains("first member"), "{:?}", result.messages);
1536 assert!(result.messages[1].contains("only a union"), "{:?}", result.messages);
1537 }
1538
1539 #[test]
1548 fn an_access_to_a_packed_member_says_the_alignment_the_layout_left_it() {
1549 let packed = body(concat!(
1550 "struct P { char c; int v; } __attribute__((packed));\n",
1551 "int f(struct P *p) { return p->v; }\n",
1552 ));
1553 assert!(packed.contains("load.i32 %2, align 1,"), "{packed}");
1554 let plain = body(concat!(
1556 "struct P { char c; int v; };\n",
1557 "int f(struct P *p) { return p->v; }\n",
1558 ));
1559 assert!(plain.contains("load.i32 %2, align 4,"), "{plain}");
1560 }
1561
1562 #[test]
1569 fn what_is_inside_a_packed_member_is_no_more_aligned_than_the_member_is() {
1570 let stepped = body(concat!(
1571 "struct P { char c; int v[4]; } __attribute__((packed));\n",
1572 "int f(struct P *p, int i) { return p->v[i]; }\n",
1573 ));
1574 assert!(stepped.contains(", align 1,"), "{stepped}");
1575 assert!(!stepped.contains(", align 4,"), "{stepped}");
1576 let nested = body(concat!(
1577 "struct Inner { int v; };\n",
1578 "struct P { char c; struct Inner in; } __attribute__((packed));\n",
1579 "int f(struct P *p) { return p->in.v; }\n",
1580 ));
1581 assert!(nested.contains(", align 1,"), "{nested}");
1582 assert!(!nested.contains(", align 4,"), "{nested}");
1583 }
1584
1585 #[test]
1594 fn the_aligned_attribute_on_a_declaration_raises_what_that_one_object_is_aligned_to() {
1595 tast(concat!(
1596 "int v __attribute__((aligned(64)));\n",
1597 "_Static_assert(__alignof__(v) == 64, \"v\");\n",
1598 "__attribute__((aligned(32))) int w;\n",
1601 "_Static_assert(__alignof__(w) == 32, \"w\");\n",
1602 "[[gnu::aligned(16)]] int x;\n",
1603 "_Static_assert(__alignof__(x) == 16, \"x\");\n",
1604 "int y __attribute__((aligned(2)));\n",
1607 "_Static_assert(__alignof__(y) == 4, \"y\");\n",
1608 "void f(void) { int a __attribute__((aligned(128)));\n",
1610 "_Static_assert(__alignof__(a) == 128, \"a\"); (void)a; }\n",
1611 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1614 "void g(void) __attribute__((aligned(256)));\n",
1617 "void g(void) {}\n",
1618 "_Static_assert(__alignof__(g) == 256, \"g\");\n",
1619 ));
1620 }
1621
1622 #[test]
1626 fn what_a_declaration_asked_to_be_aligned_to_is_what_the_assembler_is_told() {
1627 let text = asm(concat!(
1628 "int v __attribute__((aligned(64)));\n",
1629 "void g(void) __attribute__((aligned(256)));\n",
1630 "void g(void) {}\n",
1631 "void plain(void) {}\n",
1632 ));
1633 assert!(text.contains("\t.p2align\t6\n\t.type\tv, @object\n"), "{text}");
1634 assert!(text.contains("\t.p2align\t8, 0x90\n\t.globl\tg\n"), "{text}");
1635 assert!(text.contains("\t.p2align\t4, 0x90\n\t.globl\tplain\n"), "{text}");
1636 }
1637
1638 #[test]
1647 fn an_aligned_typedef_says_what_an_object_of_it_is_aligned_to_and_may_lower_it() {
1648 tast(concat!(
1649 "typedef int L __attribute__((aligned(2)));\n",
1650 "_Static_assert(__alignof__(L) == 2, \"L\");\n",
1651 "_Static_assert(_Alignof(L) == 2, \"L alignof\");\n",
1652 "_Static_assert(sizeof(L) == 4, \"L size\");\n",
1654 "struct T { char c; L x; };\n",
1655 "_Static_assert(sizeof(struct T) == 6, \"T\");\n",
1656 "_Static_assert(__builtin_offsetof(struct T, x) == 2, \"T.x\");\n",
1657 "typedef int H __attribute__((aligned(16)));\n",
1659 "_Static_assert(__alignof__(H) == 16, \"H\");\n",
1660 "_Static_assert(sizeof(H) == 4, \"H size\");\n",
1661 "struct U { char c; H x; };\n",
1662 "_Static_assert(sizeof(struct U) == 32, \"U\");\n",
1663 "_Static_assert(__builtin_offsetof(struct U, x) == 16, \"U.x\");\n",
1664 "typedef L M __attribute__((aligned(8)));\n",
1667 "_Static_assert(__alignof__(M) == 8, \"M\");\n",
1668 "typedef L N;\n",
1671 "_Static_assert(__alignof__(N) == 2, \"N\");\n",
1672 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1674 ));
1675 let text = asm(concat!(
1676 "typedef int L __attribute__((aligned(2)));\n",
1677 "typedef int H __attribute__((aligned(16)));\n",
1678 "L low;\n",
1679 "H high;\n",
1680 ));
1681 assert!(text.contains("\t.p2align\t1\n\t.type\tlow, @object\n"), "{text}");
1682 assert!(text.contains("\t.p2align\t4\n\t.type\thigh, @object\n"), "{text}");
1683 }
1684
1685 #[test]
1693 fn the_vector_size_attribute_builds_a_type_of_lanes_and_measures_it_in_bytes() {
1694 tast(concat!(
1695 "typedef int __attribute__((vector_size(16))) v4si;\n",
1696 "_Static_assert(sizeof(v4si) == 16 && _Alignof(v4si) == 16, \"v4si\");\n",
1697 "typedef char __attribute__((vector_size(16))) v16qi;\n",
1698 "_Static_assert(sizeof(v16qi) == 16, \"v16qi\");\n",
1699 "typedef int __attribute__((vector_size(4))) v1si;\n",
1702 "_Static_assert(sizeof(v1si) == 4, \"v1si\");\n",
1703 "typedef float __attribute__((__vector_size__(8))) v2sf;\n",
1705 "_Static_assert(sizeof(v2sf) == 8, \"v2sf\");\n",
1706 "typedef short [[gnu::vector_size(8)]] v4hi;\n",
1707 "_Static_assert(sizeof(v4hi) == 8, \"v4hi\");\n",
1708 "v4si g;\n",
1711 "_Static_assert(sizeof(g[0]) == 4, \"lane\");\n",
1712 "_Static_assert(sizeof(g + g) == 16, \"whole\");\n",
1713 "_Static_assert(sizeof(g + 1) == 16, \"broadcast\");\n",
1716 "_Static_assert(sizeof(v4si[3]) == 48, \"array\");\n",
1718 ));
1719 }
1720
1721 #[test]
1731 fn a_vector_is_written_whole_into_an_array_of_them_and_named_by_a_type_name() {
1732 tast(concat!(
1733 "typedef int __attribute__((vector_size(8))) v2si;\n",
1734 "v2si table[] = { (v2si){ 1, 2 }, (v2si){ 3, 4 } };\n",
1735 "_Static_assert(sizeof(table) == 16, \"two of them and not eight lanes\");\n",
1736 "v2si written = (int __attribute__((vector_size(8)))){ 5, 6 };\n",
1738 "_Static_assert(sizeof((int __attribute__((vector_size(16)))){ 0 }) == 16, \"named\");\n",
1739 "v2si lanes[2] = { 1, 2, 3, 4 };\n",
1742 "_Static_assert(sizeof(lanes) == 16, \"still elided\");\n",
1743 ));
1744 }
1745
1746 #[test]
1754 fn a_lane_is_assignable_and_a_shift_takes_a_count_of_its_own_lane() {
1755 let result = run(
1756 &options(),
1757 concat!(
1758 "typedef int __attribute__((vector_size(16))) v4si;\n",
1759 "typedef unsigned __attribute__((vector_size(16))) v4ui;\n",
1760 "void write(v4si *out, v4ui a, v4si b, int n) {\n",
1761 " v4si v = { 1, 2, 3, 4 };\n",
1762 " v[0] = n;\n",
1763 " v[1] += n;\n",
1764 " v[2]++;\n",
1765 " *&v[3] = n;\n",
1766 " v4ui shifted = a >> b;\n",
1768 " shifted <<= b;\n",
1769 " *out = v + (v4si)shifted + (1 << b);\n",
1772 "}\n",
1773 "void refused(const v4si c) {\n",
1776 " c[0] = 1;\n",
1777 "}\n",
1778 ),
1779 );
1780 assert_eq!(result.messages.len(), 1, "{:?}", result.messages);
1781 assert!(result.messages[0].contains("assignment of read-only"), "{:?}", result.messages);
1782 }
1783
1784 #[test]
1791 fn a_record_that_asks_for_the_other_byte_order_is_refused_rather_than_laid_out_in_this_one() {
1792 let opts = options();
1793 let big = "struct s { int i; } __attribute__((scalar_storage_order(\"big-endian\")));\n";
1794 assert_eq!(
1795 run(&opts, big).messages,
1796 ["/main.c:1:36: error: 'scalar_storage_order' is not implemented yet [E0688]\n\
1797 /main.c:1:36: note: every scalar in this record would be read in the wrong byte \
1798 order"]
1799 );
1800
1801 let armoured =
1802 "struct s { int i; } __attribute__((__scalar_storage_order__(\"little-endian\")));\n";
1803 let messages = run(&opts, armoured).messages;
1804 assert!(messages[0].contains("[E0688]"), "{messages:?}");
1805
1806 let front = "struct __attribute__((scalar_storage_order(\"big-endian\"))) s { int i; };\n";
1809 assert!(run(&opts, front).messages[0].contains("[E0688]"), "{front}");
1810 let standard = "struct s { int i; } [[gnu::scalar_storage_order(\"big-endian\")]];\n";
1811 assert!(run(&opts, standard).messages[0].contains("[E0688]"), "{standard}");
1812 }
1813
1814 #[test]
1824 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
1825 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
1827 assert_eq!(
1828 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
1829 1
1830 );
1831 assert_eq!(
1832 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
1833 1
1834 );
1835 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
1836 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
1838 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
1839 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
1841 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
1842 }
1843
1844 fn bit_field_byte(record: &str) -> u64 {
1846 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
1847 let body = body(&source);
1848 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
1849 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
1850 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
1851 }
1852
1853 #[test]
1859 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
1860 tast(concat!(
1861 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
1862 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
1863 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
1864 "struct b { char c; __attribute__((packed)) int i; };\n",
1865 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
1866 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
1867 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
1868 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
1869 ));
1870 }
1871
1872 #[test]
1878 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
1879 tast(concat!(
1880 "#pragma pack(1)\n",
1881 "struct A { char c; int i; };\n",
1882 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1883 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1884 "#pragma pack()\n",
1885 "struct B { char c; int i; };\n",
1886 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
1887 "#pragma pack(2)\n",
1888 "struct C { char c; int i; double d; };\n",
1889 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
1890 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
1891 "struct K { char c; int i __attribute__((aligned(8))); };\n",
1893 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
1894 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
1895 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
1897 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
1898 "#pragma pack()\n",
1899 "#pragma pack(push, 1)\n",
1900 "struct D { char c; short s; };\n",
1901 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
1902 "#pragma pack(pop)\n",
1903 "struct E { char c; short s; };\n",
1904 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
1905 "struct H { char c;\n",
1907 "#pragma pack(1)\n",
1908 " int i; };\n",
1909 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
1910 "#pragma pack(1)\n",
1911 "struct I { char c;\n",
1912 "#pragma pack()\n",
1913 " int i; };\n",
1914 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1915 "#pragma pack()\n",
1916 "#pragma pack(push, 8)\n",
1918 "#pragma pack(push, 1)\n",
1919 "struct P { char c; int i; };\n",
1920 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
1921 "#pragma pack(pop)\n",
1922 "struct Q { char c; int i; };\n",
1923 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
1924 "#pragma pack(pop)\n",
1925 "#pragma pack(16)\n",
1927 "struct R { char c; int i; };\n",
1928 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
1929 "#pragma pack()\n",
1930 "#pragma pack(1)\n",
1931 "struct S { char c; int i : 5; int j : 20; };\n",
1932 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
1933 "union T { char c; int i; };\n",
1934 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
1935 "#pragma pack()\n",
1936 ));
1937 }
1938
1939 #[test]
1943 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
1944 let result = run(
1945 &options(),
1946 concat!(
1947 "#pragma pack 4\n",
1948 "#pragma pack(pop)\n",
1949 "#pragma pack(3)\n",
1950 "#pragma pack(1) junk\n",
1951 "#pragma pack(push, 1\n",
1952 "#pragma pack(x)\n",
1953 "#pragma pack(0)\n",
1956 "#pragma pack(push)\n",
1957 "struct s { char c; int i; };\n",
1958 "#pragma pack(pop)\n",
1959 "#pragma pack(pop, foo)\n",
1960 ),
1961 );
1962 let expected = [
1963 "missing `(` after `#pragma pack` - ignored",
1964 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
1965 "alignment must be a small power of two, not 3",
1966 "junk at end of `#pragma pack`",
1967 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
1968 "unknown action `x` for `#pragma pack` - ignored",
1969 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
1970 ];
1971 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
1972 for (message, want) in result.messages.iter().zip(expected) {
1973 assert!(message.contains(want), "expected {want:?} in {message:?}");
1974 }
1975 }
1976
1977 #[test]
1981 fn the_wide_integer_answers_to_all_three_of_its_names() {
1982 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
1983 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
1984 assert!(text.contains("decl #1 b : __int128"), "{text}");
1985 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
1986 }
1987
1988 #[test]
1989 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
1990 let text = tast("long f(int a, long b) { return a + b; }\n");
1994 assert!(text.contains("convert arithmetic"), "{text}");
1995 }
1996
1997 #[test]
1998 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
1999 for source in [
2000 "#error stop\n",
2001 "int f(void) { return 1 + ; }\n",
2002 "int f(void) { return undeclared; }\n",
2003 ] {
2004 let result = run(&options(), source);
2005 assert!(result.failed(), "expected this to fail:\n{source}");
2006 assert!(
2007 result.text().is_empty(),
2008 "a file that did not compile wrote a tree:\n{source}"
2009 );
2010 }
2011 }
2012
2013 #[test]
2014 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
2015 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
2019 assert_eq!(result.errors, 1, "{:?}", result.messages);
2020 }
2021
2022 #[test]
2023 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
2024 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
2028 assert_eq!(result.errors, 1, "{:?}", result.messages);
2029 }
2030
2031 #[test]
2032 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
2033 let source = "int f(void) { char c = 300; return c; }\n";
2034 let plain = run(&options(), source);
2035 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
2036 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
2037 assert!(!plain.text().is_empty(), "a warning is not a reason to write nothing");
2038
2039 let mut opts = options();
2040 opts.warnings_are_errors = true;
2041 let strict = run(&opts, source);
2042 assert!(strict.failed());
2043 assert!(strict.text().is_empty(), "and under -Werror it is a reason to write nothing");
2044 for message in &strict.messages {
2045 assert!(!message.contains("warning:"), "{message}");
2046 }
2047 }
2048
2049 #[test]
2050 fn w_drops_the_warning_before_werror_can_promote_it() {
2051 let source = "int f(void) { char c = 300; return c; }\n";
2052 let mut opts = options();
2053 opts.warnings = false;
2054 let quiet = run(&opts, source);
2055 assert_eq!(quiet.messages, Vec::<String>::new());
2056 assert_eq!(quiet.errors, 0);
2057 assert!(!quiet.text().is_empty(), "and the file still compiles");
2058
2059 opts.warnings_are_errors = true;
2062 let both = run(&opts, source);
2063 assert_eq!(both.messages, Vec::<String>::new());
2064 assert!(!both.failed(), "-w -Werror is not an error about a warning nobody saw");
2065 }
2066
2067 #[test]
2068 fn the_dialect_reaches_the_keywords_and_the_checking() {
2069 let source = "typeof(1) x;\n";
2072 let mut opts = options();
2073 opts.std = Std::C23;
2074 opts.gnu_extensions = false;
2075 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
2076
2077 opts.std = Std::C17;
2078 assert!(run(&opts, source).failed());
2079 }
2080
2081 #[test]
2082 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
2083 let mut opts = options();
2084 opts.emit = EmitKind::Object;
2085 let result = run(&opts, "int x = 1;\n");
2086 assert!(!result.failed(), "{:?}", result.messages);
2087 assert!(result.text().is_empty());
2088 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
2091 }
2092
2093 fn mir(source: &str) -> String {
2095 let mut opts = options();
2096 opts.emit = EmitKind::MirFinal;
2097 let result = run(&opts, source);
2098 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2099 result.text().to_owned()
2100 }
2101
2102 #[test]
2108 fn a_function_goes_from_c_to_instructions_with_real_registers_in_them() {
2109 let text = mir("int add(int a, int b) { return a + b; }\n");
2110 assert!(text.starts_with("mfunc @add {"), "{text}");
2111 assert!(text.contains("x64.add_rr_32"), "{text}");
2112 assert!(text.contains("x64.ret"), "{text}");
2113 assert!(!text.contains('%'), "{text}");
2116 }
2117
2118 #[test]
2120 fn a_function_with_no_body_produces_no_machine_function() {
2121 let text = mir("int g(int);\nint f(int a) { return g(a); }\n");
2122 assert_eq!(text.matches("mfunc @").count(), 1, "{text}");
2123 assert!(text.contains("mfunc @f {"), "{text}");
2124 assert!(text.contains("x64.call"), "{text}");
2125 }
2126
2127 #[test]
2129 fn every_definition_in_the_file_is_generated_and_they_keep_their_order() {
2130 let text = mir("int a(int x) { return x; }\nint b(int x) { return x; }\n");
2131 let first = text.find("mfunc @a").expect("the first function");
2132 let second = text.find("mfunc @b").expect("the second function");
2133 assert!(first < second, "{text}");
2134 }
2135
2136 #[test]
2138 fn the_target_decides_which_convention_the_generated_code_follows() {
2139 let mut opts = options();
2140 opts.emit = EmitKind::MirFinal;
2141 let linux = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
2142 assert!(linux.contains("$rdi"), "{linux}");
2143
2144 opts.target = "x86_64-pc-windows-msvc".parse::<Triple>().unwrap();
2145 let windows = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
2146 assert!(windows.contains("$rcx"), "{windows}");
2147 assert!(!windows.contains("$rdi"), "{windows}");
2148 }
2149
2150 #[test]
2152 fn a_target_this_has_no_back_end_for_is_reported_rather_than_generated() {
2153 let mut opts = options();
2154 opts.emit = EmitKind::MirFinal;
2155 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
2156 let result = run(&opts, "int f(int a) { return a; }\n");
2157 assert!(result.failed());
2158 assert!(result.messages[0].contains("no back end for aarch64"), "{:?}", result.messages);
2159 assert!(result.text().is_empty());
2160 }
2161
2162 #[test]
2169 fn a_construct_the_back_end_cannot_reach_yet_is_reported_against_its_function() {
2170 let mut opts = options();
2171 opts.emit = EmitKind::MirFinal;
2172 let source = "void a(int n) { int v[n] __attribute__((aligned(32))); v[0] = 1; }\n\
2173 void b(int n) { int v[n] __attribute__((aligned(32))); v[0] = 1; }\n";
2174 let result = run(&opts, source);
2175 assert!(result.failed());
2176 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
2177 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
2178 assert!(result.messages[0].contains("wants more alignment"), "{:?}", result);
2179 assert!(result.messages[1].contains("cannot generate code for 'b'"), "{:?}", result);
2180 assert!(result.text().is_empty());
2181 }
2182
2183 #[test]
2190 fn a_variable_length_array_is_refused_where_every_page_of_the_frame_is_to_be_touched() {
2191 let mut opts = options();
2192 opts.emit = EmitKind::MirFinal;
2193 let source = "void a(int n) { int v[n]; v[0] = 1; }\n";
2194 assert!(!run(&opts, source).failed(), "it compiles without the flag");
2195
2196 opts.stack_clash = true;
2197 let result = run(&opts, source);
2198 assert!(result.failed());
2199 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
2200 assert!(result.messages[0].contains("a page at a time"), "{:?}", result);
2201 }
2202
2203 #[test]
2217 fn an_opcode_with_no_name_in_the_rule_language_is_named_by_its_own_spelling() {
2218 let mut opts = options();
2219 opts.emit = EmitKind::MirFinal;
2220 let source =
2221 "long double f(int a) {\n __int128 wide = a;\n return (long double) wide;\n}\n";
2222 let result = run(&opts, source);
2223 assert!(result.failed());
2224 assert!(
2225 result.messages[0].contains("no rule lowers a `sext` producing a `i128`"),
2226 "{result:?}"
2227 );
2228 assert!(result.messages[0].contains(":2:"), "the line the widening is on: {result:?}");
2229 assert!(!result.messages[0].contains("this instruction"), "{result:?}");
2230 }
2231
2232 #[test]
2234 fn the_note_on_unfinished_work_points_at_the_issues_rather_than_at_the_plan() {
2235 let mut opts = options();
2236 opts.emit = EmitKind::MirFinal;
2237 let source = "long double f(int a) { __int128 wide = a; return (long double) wide; }\n";
2238 let result = run(&opts, source);
2239 assert!(result.failed());
2240 let note = result.messages.iter().find(|line| line.contains("note:")).expect("a note");
2241 assert!(note.contains("https://github.com/tamnd/rucc/issues"), "{note}");
2242 assert!(!note.contains("spec/17-milestones.md"), "{note}");
2243 }
2244
2245 #[test]
2247 fn the_frame_flags_on_the_command_line_reach_the_generated_frame() {
2248 let source = "int f(int a) { return a; }\n";
2249 assert!(!mir(source).contains("$rbp"), "a leaf needs no frame pointer by default");
2250
2251 let mut opts = options();
2252 opts.emit = EmitKind::MirFinal;
2253 opts.frame_pointer = true;
2254 let kept = run(&opts, source).text().to_owned();
2255 assert!(kept.contains("x64.push_64 $rbp"), "{kept}");
2256 }
2257
2258 fn asm(source: &str) -> String {
2260 let mut opts = options();
2261 opts.emit = EmitKind::Asm;
2262 let result = run(&opts, source);
2263 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2264 result.text().to_owned()
2265 }
2266
2267 #[test]
2274 fn a_function_goes_from_c_to_assembly_an_assembler_would_take() {
2275 let text = asm("int add(int a, int b) { return a + b; }\n");
2276 assert!(text.contains("\t.globl\tadd\n"), "{text}");
2277 assert!(text.contains("\t.type\tadd, @function\n"), "{text}");
2278 assert!(text.contains("\nadd:\n"), "{text}");
2279 assert!(text.contains("\taddl\t"), "{text}");
2280 assert!(text.contains("\tret\n"), "{text}");
2281 assert!(text.contains("\t.size\tadd, .-add\n"), "{text}");
2282 assert!(text.contains(".note.GNU-stack"), "{text}");
2285 }
2286
2287 #[test]
2293 fn a_call_through_a_function_pointer_goes_through_the_register_it_is_in() {
2294 let text = asm("int g(int);\nint f(int (*p)(int), int a) { return p(a) + g(a); }\n");
2295 assert!(text.contains("\tcall\t*%"), "{text}");
2296 assert!(text.contains("\tcall\tg\n"), "{text}");
2297 assert!(text.contains("%rdi"), "{text}");
2301 }
2302
2303 #[test]
2307 fn the_address_of_a_global_is_read_from_the_instruction_pointer() {
2308 let text = asm("extern int counter;\nint f(void) { return counter; }\n");
2309 assert!(text.contains("\tmovl\tcounter(%rip), %eax\n"), "{text}");
2310 }
2311
2312 #[test]
2321 fn a_branch_on_a_comparison_jumps_on_the_opposite_of_what_it_compared() {
2322 let arms = "return 1; return 2;";
2323 let signed = [("==", "jne"), ("!=", "je"), ("<", "jge"), ("<=", "jg"), (">", "jle")];
2324 for (operator, jump) in signed.into_iter().chain([(">=", "jl")]) {
2325 let text = asm(&format!("int f(int a, int b) {{ if (a {operator} b) {arms} }}\n"));
2326 assert!(
2327 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
2328 "{operator}: {text}"
2329 );
2330 assert!(!text.contains("\tset"), "{operator}: {text}");
2331 assert!(!text.contains("\ttest"), "{operator}: {text}");
2332 }
2333 let unsigned = [("<", "jae"), ("<=", "ja"), (">", "jbe"), (">=", "jb")];
2334 for (operator, jump) in unsigned {
2335 let source =
2336 format!("int f(unsigned a, unsigned b) {{ if (a {operator} b) {arms} }}\n");
2337 let text = asm(&source);
2338 assert!(
2339 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
2340 "{operator}: {text}"
2341 );
2342 }
2343
2344 let text = asm("int f(int a) { if (a < 7) return 1; return 2; }\n");
2347 assert!(text.contains("\tcmpl\t$7, %edi\n\tjge\t"), "{text}");
2348 }
2349
2350 #[test]
2356 fn a_comparison_whose_answer_the_program_wanted_still_writes_a_byte() {
2357 let text = asm("int f(int a, int b) { return a < b; }\n");
2358 assert!(text.contains("\tsetl\t"), "{text}");
2359 }
2360
2361 fn optimized(source: &str) -> String {
2363 let mut opts = options();
2364 opts.emit = EmitKind::Asm;
2365 opts.opt_level = rucc_session::OptLevel::O2;
2366 let result = run(&opts, source);
2367 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2368 result.text().to_owned()
2369 }
2370
2371 #[test]
2381 fn a_switch_whose_arms_are_a_function_of_the_label_is_a_range_check_and_arithmetic() {
2382 let arms: String =
2383 (0..16).map(|k| format!("case {k}: return {};", k + 1)).collect::<Vec<_>>().join(" ");
2384 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
2385 assert!(text.contains("\tcmpl\t$15, %edi\n\tja\t"), "{text}");
2386 assert!(text.contains("\taddl\t$1, %edi"), "{text}");
2387 assert_eq!(text.matches("\tcmp").count(), 1, "{text}");
2388 }
2389
2390 #[test]
2397 fn a_dense_switch_whose_arms_are_not_a_line_keeps_its_comparisons() {
2398 let arms: String = (0..16)
2399 .map(|k| format!("case {k}: return {};", if k == 9 { 100 } else { k + 1 }))
2400 .collect::<Vec<_>>()
2401 .join(" ");
2402 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
2403 assert!(text.matches("\tcmp").count() > 1, "{text}");
2404 }
2405
2406 #[test]
2408 fn a_cast_between_a_pointer_and_an_integer_leaves_the_value_where_it_is() {
2409 let text = asm("long f(void *p) { return (long)p; }\n");
2410 for line in text.lines().filter(|line| line.starts_with('\t') && !line.contains('.')) {
2415 let mnemonic = line.split_whitespace().next().unwrap_or("");
2416 assert!(matches!(mnemonic, "movq" | "ret"), "{line} in\n{text}");
2417 }
2418 }
2419
2420 #[test]
2424 fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
2425 let six = "long a, long b, long c, long d, long e, long f";
2426 let text = asm(&format!("long f({six}, long g, long h) {{ return g + h; }}\n"));
2427
2428 assert!(text.contains("\tmovq\t8(%rsp), "), "{text}");
2435 assert!(text.contains("\taddq\t16(%rsp), "), "{text}");
2436
2437 let narrow = asm(&format!("int f({six}, int g) {{ return g; }}\n"));
2441 assert!(narrow.contains("\tmovl\t8(%rsp), "), "{narrow}");
2442 let eight =
2443 "double a, double b, double c, double d, double e, double f, double g, double h";
2444 let float = asm(&format!("double f({eight}, double i) {{ return i; }}\n"));
2445 assert!(float.contains("\tmovsd\t8(%rsp), "), "{float}");
2446 }
2447
2448 #[test]
2451 fn a_call_writes_the_arguments_with_no_register_left_at_the_stack_pointer() {
2452 let six = "1, 2, 3, 4, 5, 6";
2453 let decl = "long g(long, long, long, long, long, long, long, long);\n";
2454 let text = asm(&format!("{decl}long f(void) {{ return g({six}, 7, 8); }}\n"));
2455
2456 assert!(text.contains("\tmovq\t%"), "{text}");
2457 assert!(text.contains(", (%rsp)\n"), "{text}");
2458 assert!(text.contains(", 8(%rsp)\n"), "{text}");
2459 assert!(text.contains("\tsubq\t$"), "{text}");
2461
2462 let narrow = "int g(int, int, int, int, int, int, int);\n";
2464 let text = asm(&format!("{narrow}int f(void) {{ return g({six}, 7); }}\n"));
2465 assert!(text.contains("\tmovl\t%"), "{text}");
2466 assert!(text.contains(", (%rsp)\n"), "{text}");
2467 }
2468
2469 #[test]
2472 fn a_variadic_call_counts_registers_and_not_arguments() {
2473 let nine = "1., 2., 3., 4., 5., 6., 7., 8., 9.";
2474 let decl = "int g(int, ...);\n";
2475 let text = asm(&format!("{decl}int f(void) {{ return g(0, {nine}); }}\n"));
2476
2477 assert!(text.contains("\tmovl\t$8, "), "eight registers, not nine: {text}");
2478 assert!(text.contains("\tmovsd\t%"), "{text}");
2479 assert!(text.contains(", (%rsp)\n"), "{text}");
2480 }
2481
2482 #[test]
2487 fn a_variadic_function_writes_the_argument_registers_it_was_handed_into_its_frame() {
2488 let body =
2489 "__builtin_va_list ap; __builtin_va_start(ap, n); __builtin_va_end(ap); return n;";
2490 let text = asm(&format!("int f(int n, ...) {{ {body} }}\n"));
2491
2492 let stores = |mnemonic: &str| text.matches(&format!("\t{mnemonic}\t%")).count();
2495 assert!(text.contains(", 8(%r"), "the second slot, not the first: {text}");
2496 assert!(!text.contains(", 0(%r"), "{text}");
2497 assert_eq!(stores("movaps"), 8, "every vector register: {text}");
2500 assert_eq!(stores("movsd"), 0, "and the whole of each one: {text}");
2501
2502 assert!(text.contains("\tsubq\t$"), "{text}");
2504 }
2505
2506 #[test]
2509 fn va_start_writes_the_four_fields_the_psabi_describes() {
2510 let start = "__builtin_va_list ap; __builtin_va_start(ap, d);";
2511 let params = "int a, int b, int c, double d";
2512 let text = asm(&format!("int f({params}, ...) {{ {start} return a; }}\n"));
2513
2514 assert!(text.contains(" movl $24, "), "{text}");
2518 assert!(text.contains(" movl $64, "), "{text}");
2519 assert!(text.contains(", 8(%r"), "{text}");
2523 assert!(text.contains(", 16(%r"), "{text}");
2524 let frame: u32 = text
2525 .lines()
2526 .find_map(|line| line.trim().strip_prefix("subq $")?.split(',').next()?.parse().ok())
2527 .expect("a variadic function takes a frame for the save area");
2528 let above = |line: &str| {
2529 let at: u32 = line.trim().strip_prefix("leaq ")?.split('(').next()?.parse().ok()?;
2530 Some(at > frame)
2531 };
2532 assert!(text.lines().filter_map(above).any(|it| it), "{frame}: {text}");
2533 }
2534
2535 #[test]
2538 fn va_arg_branches_on_whether_the_argument_is_still_in_the_save_area() {
2539 let read = "__builtin_va_list ap; __builtin_va_start(ap, n);";
2540 let ints = format!("int f(int n, ...) {{ {read} return __builtin_va_arg(ap, int); }}\n");
2541 let text = asm(&ints);
2542
2543 assert!(text.contains("$40, "), "{text}");
2546 assert!(text.contains(" cmpl "), "{text}");
2547 assert!(text.contains(" ja "), "{text}");
2551
2552 let arg = "__builtin_va_arg(ap, double)";
2553 let text = asm(&format!("double f(int n, ...) {{ {read} return {arg}; }}\n"));
2554 assert!(text.contains("$160, "), "the last vector slot: {text}");
2555 }
2556
2557 #[test]
2560 fn a_structure_assignment_is_a_move_for_each_word_of_it() {
2561 let decl = "struct pair { long a, b; };\n";
2562 let body = "struct pair p = *q; return p.a + p.b;";
2563 let text = asm(&format!("{decl}long f(struct pair *q) {{ {body} }}\n"));
2564
2565 assert!(!text.contains("memcpy"), "nothing calls the library: {text}");
2566 assert!(!text.contains("\tcall"), "{text}");
2567 assert!(text.matches("\tmovq\t").count() >= 4, "two words each way: {text}");
2569 }
2570
2571 #[test]
2574 fn how_wide_a_word_of_a_copy_is_follows_the_alignment() {
2575 let decl = "struct bytes { char a[8]; };\n";
2576 let body = "struct bytes p = *q; return p.a[0];";
2577 let text = asm(&format!("{decl}int f(struct bytes *q) {{ {body} }}\n"));
2578
2579 assert!(text.matches("\tmovb\t").count() >= 16, "a byte at a time: {text}");
2581 }
2582
2583 #[test]
2586 fn the_part_of_an_initialiser_that_names_nothing_is_stored_as_zero() {
2587 let decl = "struct wide { long a, b, c; };\n";
2588 let text = asm(&format!("{decl}long f(void) {{ struct wide w = {{ 7 }}; return w.c; }}\n"));
2589
2590 assert!(!text.contains("memset"), "nothing calls the library: {text}");
2591 assert!(text.contains("\tmovq\t$0, ") || text.contains("$0, %"), "the zero: {text}");
2592 }
2593
2594 #[test]
2597 fn a_copy_too_large_to_unroll_calls_the_runtime() {
2598 let decl = "struct huge { char a[4096]; };\n";
2599 let mut opts = options();
2600 opts.emit = EmitKind::Asm;
2601 let source = format!("{decl}void f(struct huge *p, struct huge *q) {{ *p = *q; }}\n");
2602 let result = run(&opts, &source);
2603 assert!(!result.failed(), "{:?}", result.messages);
2604 let text = result.text();
2605 assert!(text.contains("call") && text.contains("memcpy"), "{text}");
2606 assert!(text.contains("4096"), "the size travels: {text}");
2609 }
2610
2611 #[test]
2614 fn a_realigned_frame_reads_them_through_the_frame_pointer() {
2615 let six = "long a, long b, long c, long d, long e, long f";
2616 let body = "_Alignas(32) long wide[4]; wide[0] = g; return wide[0];";
2617 let text = asm(&format!("long f({six}, long g) {{ {body} }}\n"));
2618
2619 assert!(text.contains("\tandq\t$-32, %rsp"), "{text}");
2623 assert!(text.contains("\tmovq\t16(%rbp), "), "{text}");
2624 assert!(!text.contains("\tmovq\t16(%rsp), "), "{text}");
2625 }
2626
2627 #[test]
2629 fn the_target_decides_how_the_assembly_is_spelled() {
2630 let mut opts = options();
2631 opts.emit = EmitKind::Asm;
2632 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2633 let text = run(&opts, "int f(void) { return 0; }\n").text().to_owned();
2634 assert!(text.contains("__TEXT,__text"), "{text}");
2635 assert!(text.contains("\n_f:\n"), "{text}");
2636 assert!(!text.contains(".note.GNU-stack"), "{text}");
2637 }
2638
2639 fn obj(source: &str) -> Vec<u8> {
2641 let mut opts = options();
2642 opts.emit = EmitKind::Object;
2643 let result = run(&opts, source);
2644 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2645 match result.artifact {
2646 Artifact::Object { bytes, .. } => bytes,
2647 other => panic!("expected an object, got {other:?}"),
2648 }
2649 }
2650
2651 #[test]
2657 fn a_function_goes_from_c_to_an_object_a_linker_would_take() {
2658 let bytes = obj("int add(int a, int b) { return a + b; }\n");
2659 assert_eq!(&bytes[..4], b"\x7fELF", "an object file starts by saying it is one");
2660 let text = asm("int add(int a, int b) { return a + b; }\n");
2661 assert!(
2662 text.contains("\taddl\t"),
2663 "and the listing of it is the same instructions:\n{text}"
2664 );
2665 }
2666
2667 #[test]
2669 fn a_variable_goes_from_c_to_the_section_it_belongs_in() {
2670 let text = asm("int counter = 42;\nstatic int hidden;\nconst int fixed = 7;\n");
2671 assert!(text.contains("\t.data\n\t.globl\tcounter\n"), "{text}");
2672 assert!(text.contains("\ncounter:\n\t.long\t42\n"), "{text}");
2673 assert!(text.contains("\t.size\tcounter, .-counter\n"), "{text}");
2674 assert!(text.contains("\t.bss\n\t.p2align\t2\n"), "{text}");
2677 assert!(text.contains("\nhidden:\n\t.space\t4\n"), "{text}");
2678 assert!(!text.contains(".globl\thidden"), "{text}");
2679 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2682 }
2683
2684 #[test]
2691 fn a_bit_field_initializer_writes_every_byte_of_the_value_and_not_only_the_ones_that_are_set() {
2692 let text = asm("struct s { unsigned f : 20; } x = { 0x12300 };\n");
2693 assert!(text.contains("\t.data\n"), "there is something to write: {text}");
2694 assert!(text.contains("\nx:\n\t.ascii\t\"\\000#\\001\"\n"), "and it is the value: {text}");
2695
2696 let text = asm("struct s { unsigned a : 8; unsigned b : 8; } x = { 0, 3 };\n");
2699 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\003\"\n"), "{text}");
2700
2701 let text = asm("struct s { unsigned long long f : 40; } x = { 0x100000 };\n");
2704 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\000\\020\"\n\t.space\t5\n"), "{text}");
2705
2706 let text = asm("struct s { unsigned f : 20; } x = { 0 };\n");
2708 assert!(text.contains("\t.bss\n"), "an object of zeroes is zeroes: {text}");
2709 assert!(text.contains("\nx:\n\t.space\t4\n"), "{text}");
2710 }
2711
2712 #[test]
2714 fn a_string_literal_is_a_variable_with_a_name_no_program_could_write() {
2715 let text = asm("const char *f(void) { return \"hi\"; }\n");
2716 assert!(text.contains("\t.ascii\t\"hi\\000\"\n"), "{text}");
2717 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2718 let label = text
2719 .lines()
2720 .find(|line| line.starts_with(".Lstr"))
2721 .unwrap_or_else(|| panic!("a label for the literal in\n{text}"));
2722 assert!(!text.contains(&format!(".globl\t{}", label.trim_end_matches(':'))), "{text}");
2723 }
2724
2725 #[test]
2727 fn an_address_in_an_initializer_is_left_to_the_linker() {
2728 let source = "int counter;\nint *p = &counter;\n";
2729 let text = asm(source);
2730 assert!(text.contains("\np:\n\t.quad\tcounter\n"), "{text}");
2731 let bytes = obj(source);
2734 assert!(bytes.windows(8).any(|w| w == b"counter\0"), "the object has to name it");
2735 }
2736
2737 #[test]
2746 fn a_constant_holding_an_address_goes_in_the_section_the_loader_may_write_once() {
2747 let text = asm("static void a(void) {}\nstatic void b(void) {}\n\
2750 struct m { void (*x)(void); void (*y)(void); };\n\
2751 const struct m t = { a, b };\n");
2752 assert!(text.contains("\t.section\t.data.rel.ro.local,\"aw\",@progbits\n"), "{text}");
2753 assert!(text.contains("\nt:\n\t.quad\ta\n\t.quad\tb\n"), "{text}");
2754
2755 let text =
2758 asm("void a(void);\nstruct m { void (*x)(void); };\nconst struct m t = { a };\n");
2759 assert!(text.contains("\t.section\t.data.rel.ro,\"aw\",@progbits\n"), "{text}");
2760
2761 let text = asm("const int fixed = 7;\n");
2763 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2764 }
2765
2766 #[test]
2773 fn a_thread_local_variable_is_storage_a_thread_gets_a_copy_of_and_an_offset_into_it() {
2774 let text = asm("_Thread_local int x = 1;\nint read(void) { return x; }\n");
2775 assert!(text.contains("\t.section\t.tdata,\"awT\",@progbits\n"), "{text}");
2778 assert!(text.contains("\t.type\tx, @tls_object\n"), "{text}");
2779 assert!(text.contains("x@GOTTPOFF(%rip)"), "{text}");
2782 assert!(text.contains("%fs:0"), "{text}");
2783 }
2784
2785 #[test]
2791 fn the_address_of_this_thread_s_own_storage_is_read_out_of_the_segment_register() {
2792 let text = asm("void *here(void) { return __builtin_thread_pointer(); }\n");
2793 assert!(text.contains("movq\t%fs:0, "), "{text}");
2794 assert!(!text.contains("GOTTPOFF"), "{text}");
2796 }
2797
2798 #[test]
2809 fn a_prefetch_is_one_of_four_instructions_and_the_locality_is_what_picks() {
2810 for (locality, wanted) in
2811 [(0, "prefetchnta"), (1, "prefetcht2"), (2, "prefetcht1"), (3, "prefetcht0")]
2812 {
2813 let source =
2814 format!("void warm(void *p) {{ __builtin_prefetch(p, 0, {locality}); }}\n");
2815 let text = asm(&source);
2816 assert!(text.contains(&format!("\t{wanted}\t")), "locality {locality}: {text}");
2817 }
2818 let text = asm("void warm(void *p) { __builtin_prefetch(p); }\n");
2820 assert!(text.contains("\tprefetcht0\t"), "{text}");
2821 let text = asm("void warm(void *p) { __builtin_prefetch(p, 1); }\n");
2824 assert!(text.contains("\tprefetcht0\t"), "{text}");
2825 assert!(!text.contains("prefetchw"), "{text}");
2826 }
2827
2828 #[test]
2839 fn a_trap_is_the_instruction_the_machine_has_no_meaning_for() {
2840 let text = asm("void stop(void) { __builtin_trap(); }\n");
2841 assert!(text.contains("\tud2\n"), "{text}");
2842 assert!(!text.contains("\tcall"), "a stop is not a call to anything: {text}");
2843
2844 let text = asm("int stop(int a) { __builtin_trap(); return a + 1; }\n");
2845 assert!(text.contains("\tud2\n"), "{text}");
2846 assert!(text.contains("\taddl\t"), "the block goes on after a stop: {text}");
2847 }
2848
2849 #[test]
2861 fn assume_aligned_is_its_first_argument_and_keeps_the_rest() {
2862 let text = asm("void *aligned(char *p) { return __builtin_assume_aligned(p, 16); }\n");
2863 assert!(!text.contains("assume_aligned"), "{text}");
2864 assert!(!text.contains("\tcall"), "nothing is called for an alignment fact: {text}");
2865
2866 let source = "unsigned long width(void);\n\
2867 void *aligned(char *p) { return __builtin_assume_aligned(p, width()); }\n";
2868 let text = asm(source);
2869 assert!(!text.contains("assume_aligned"), "{text}");
2870 assert!(text.contains("width"), "the argument that is not the answer still runs: {text}");
2871 }
2872
2873 #[test]
2883 fn the_frame_address_is_the_frame_pointer_after_walking_that_many_links() {
2884 let text = asm("void *here(void) { return __builtin_frame_address(0); }\n");
2885 assert!(text.contains("pushq\t%rbp"), "a function that asks keeps a frame pointer: {text}");
2886 assert!(text.contains("movq\t%rbp, %rax"), "{text}");
2887 assert!(!text.contains("\tcall"), "a frame address is not a call to anything: {text}");
2888
2889 let walk = |depth: u32| {
2890 let source = format!("void *up(void) {{ return __builtin_frame_address({depth}); }}\n");
2891 asm(&source).matches("movq\t(%r").count()
2892 };
2893 assert_eq!(walk(1), 1, "one link is one load");
2894 assert_eq!(walk(3), 3, "three links are three loads");
2895 }
2896
2897 #[test]
2907 fn the_return_address_is_one_word_above_the_frame_the_walk_ended_at() {
2908 let text = asm("void *back(void) { return __builtin_return_address(0); }\n");
2909 assert!(text.contains("pushq\t%rbp"), "a function that asks keeps a frame pointer: {text}");
2910 assert!(text.contains("movq\t8(%rbp), %rax"), "{text}");
2911 assert!(!text.contains("\tcall"), "a return address is not a call to anything: {text}");
2912
2913 let text = asm("void *back(void) { return __builtin_return_address(2); }\n");
2914 assert_eq!(text.matches("movq\t(%r").count(), 2, "two links are two loads: {text}");
2915 assert!(text.contains("movq\t8(%r"), "and the answer is above the last of them: {text}");
2916 }
2917
2918 #[test]
2929 fn a_depth_that_is_not_a_small_constant_is_refused() {
2930 let mut opts = options();
2931 opts.emit = EmitKind::Ir;
2932 for source in [
2933 "void *up(int n) { return __builtin_return_address(n); }\n",
2934 "void *up(void) { return __builtin_frame_address(1000); }\n",
2935 ] {
2936 let messages = run(&opts, source).messages;
2937 let named = messages.iter().any(|m| m.contains("E0705"));
2938 assert!(named, "expected a refusal in {messages:?}");
2939 }
2940 }
2941
2942 #[test]
2954 fn an_alloca_takes_the_bytes_off_the_stack_pointer_and_answers_where_they_are() {
2955 let text =
2956 asm("void use(void *p); void f(unsigned long n) { use(__builtin_alloca(n)); }\n");
2957 assert!(text.contains("andq\t$-16"), "the size is rounded up to sixteen: {text}");
2958 assert!(text.contains("subq\t%rdi, %rsp"), "and taken off the stack pointer: {text}");
2959 assert_eq!(text.matches("\tcall").count(), 1, "the only call is the one written: {text}");
2960
2961 let plain = concat!(
2964 "extern void *alloca(__SIZE_TYPE__);\n",
2965 "void use(void *p);\n",
2966 "void f(unsigned long n) { use(alloca(n)); }\n",
2967 );
2968 let text = asm(plain);
2969 assert!(text.contains("subq\t%rdi, %rsp"), "the plain name is the same bytes: {text}");
2970 assert_eq!(text.matches("\tcall").count(), 1, "and is not a call either: {text}");
2971
2972 let own = concat!(
2975 "static void *alloca(unsigned long n) { return 0; }\n",
2976 "void *f(unsigned long n) { return alloca(n); }\n",
2977 );
2978 assert!(asm(own).contains("\tcall"), "a name the program took back is a call");
2979 }
2980
2981 #[test]
2991 fn the_bytes_an_alloca_took_are_still_there_at_the_end_of_the_block_that_took_them() {
2992 let inner = "{ use(__builtin_alloca(n)); }";
2993 for body in [inner.to_owned(), format!("int a[n]; {inner} use(a);")] {
2994 let source = format!("void use(void *p);\nvoid f(unsigned long n) {{ {body} }}\n");
2995 let text = asm(&source);
2996 for line in text.lines().filter(|line| line.trim_end().ends_with(", %rsp")) {
3000 let taking = line.contains("subq");
3001 let leaving = line.contains("%rbp");
3002 assert!(taking || leaving, "nothing puts the stack back: {line} in {text}");
3003 }
3004 }
3005 }
3006
3007 #[test]
3009 fn the_object_and_the_listing_are_two_spellings_of_one_compilation() {
3010 let source = "int callee(void); int g(void) { return callee(); }\n";
3014 let bytes = obj(source);
3015 assert!(
3016 bytes.windows(7).any(|w| w == b"callee\0"),
3017 "the object has to name the callee for the linker to find it"
3018 );
3019 let text = asm(source);
3020 assert!(text.contains("\tcall\tcallee\n"), "{text}");
3021 }
3022
3023 #[test]
3029 fn compiling_for_an_executable_produces_an_object_and_not_a_dump() {
3030 let mut opts = options();
3031 opts.emit = EmitKind::Executable;
3033 let result = run(&opts, "int main(void) { return 0; }\n");
3034 assert_eq!(result.messages, Vec::<String>::new());
3035 match result.artifact {
3036 Artifact::Object { bytes, .. } => assert_eq!(&bytes[..4], b"\x7fELF"),
3037 other => panic!("expected an object, got {other:?}"),
3038 }
3039 }
3040
3041 #[test]
3043 fn a_platform_with_no_object_writer_is_said_so_rather_than_written_as_elf() {
3044 let mut opts = options();
3045 opts.emit = EmitKind::Object;
3046 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
3047 let result = run(&opts, "int f(void) { return 0; }\n");
3048 assert!(result.failed(), "an object nobody can read is worse than a message");
3049 assert!(
3050 result.messages.iter().any(|m| m.contains("no object writer")),
3051 "{:?}",
3052 result.messages
3053 );
3054 }
3055
3056 fn ir(source: &str) -> String {
3058 let mut opts = options();
3059 opts.emit = EmitKind::Ir;
3060 let result = run(&opts, source);
3061 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3062 result.text().to_owned()
3063 }
3064
3065 fn errors(source: &str) -> Vec<String> {
3067 let mut opts = options();
3068 opts.emit = EmitKind::Ir;
3069 let result = run(&opts, source);
3070 assert!(result.failed(), "expected this to be refused:\n{source}");
3071 result.messages
3072 }
3073
3074 fn body(source: &str) -> String {
3076 let text = ir(source);
3077 let (_, rest) = text.split_once("{\n").expect("a function definition");
3078 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
3079 body.to_owned()
3080 }
3081
3082 #[test]
3090 fn gnu89_inline_is_what_decides_whether_a_bare_inline_definition_reaches_the_module() {
3091 let source = "inline int f(int x) { return x + 1; }\n";
3092 let with = |flag: bool| {
3093 let mut opts = options();
3094 opts.emit = EmitKind::Ir;
3095 opts.gnu89_inline = flag;
3096 let result = run(&opts, source);
3097 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
3098 result.text().to_owned()
3099 };
3100
3101 assert!(!with(false).contains("block0"), "no body: {}", with(false));
3104
3105 assert!(with(true).contains("block0"), "a body: {}", with(true));
3108 }
3109
3110 #[test]
3117 fn an_access_through_a_type_names_the_type_it_went_through() {
3118 let source = "\
3119struct s { int a; float b; };\n\
3120union u { int i; float f; };\n\
3121int scalar(int *p) { return *p; }\n\
3122float member(struct s *p) { p->a = 1; return p->b; }\n\
3123int element(int *a, long i) { return a[i]; }\n\
3124float through_a_union(union u *p) { p->i = 1; return p->f; }\n";
3125 let text = ir(source);
3126 assert!(text.contains(r#"!0 = tbaa "char""#), "the root: {text}");
3127 assert!(text.contains(r#"tbaa "int", parent !0"#), "int under it: {text}");
3128 assert!(text.contains(r#"tbaa "float", parent !0"#), "float under it: {text}");
3129 let named = text.lines().filter(|line| line.contains(", tbaa !")).count();
3132 assert_eq!(named, 6, "six accesses: {text}");
3133 }
3134
3135 #[test]
3142 fn turning_strict_aliasing_off_leaves_the_type_off_every_access() {
3143 let source = "int punned(float *f, int *i) { *i = 1; *f = 2.0f; return *i; }\n";
3144 let mut opts = options();
3145 opts.emit = EmitKind::Ir;
3146 opts.strict_aliasing = false;
3147 let result = run(&opts, source);
3148 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
3149 let text = result.text().to_owned();
3150 assert!(!text.contains("tbaa"), "not even the root: {text}");
3151 }
3152
3153 #[test]
3161 fn a_bare_return_from_a_function_that_promised_a_value_gives_back_a_zero() {
3162 let mut opts = options();
3163 opts.emit = EmitKind::Ir;
3164 opts.std = Std::C89;
3165 let compiled = |source: &str| {
3166 let result = run(&opts, source);
3167 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
3168 result.text().to_owned()
3169 };
3170
3171 let text = compiled("int f(int x) { if (x) return; return 3; }\n");
3172 assert!(text.contains("iconst.i32 0\n return"), "zero goes back: {text}");
3173 assert!(!text.contains("unreachable"), "the branch that reached it is kept: {text}");
3174
3175 let text = compiled("double f(int x) { if (x) return; return 1.0; }\n");
3177 assert!(text.contains("fconst.f64 0x0\n return"), "a float zero goes back: {text}");
3178 }
3179
3180 #[test]
3188 fn a_call_to_a_name_nothing_declared_declares_it_as_c89_said_to() {
3189 let mut opts = options();
3190 opts.emit = EmitKind::Ir;
3191 opts.std = Std::C89;
3192 let compiled = |source: &str| {
3193 let result = run(&opts, source);
3194 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
3195 result.text().to_owned()
3196 };
3197
3198 let text = compiled("int f(void) { return g(); }\n");
3200 assert!(text.contains("call @g"), "the call is to the name that was written: {text}");
3201 assert!(text.contains("i32"), "and it gives back an int: {text}");
3202
3203 let text = compiled("int f(char c) { return g(c); }\n");
3206 assert!(text.contains("sext.i32"), "the argument is promoted: {text}");
3207
3208 let mut opts = options();
3211 opts.std = Std::C89;
3212 let said = run(&opts, "int f(void) { return h; }\n").messages.join("\n");
3213 assert!(said.contains("'h' undeclared"), "not a call, so not declared: {said}");
3214 }
3215
3216 #[test]
3226 fn a_name_called_before_it_is_defined_still_gets_its_definition() {
3227 let mut opts = options();
3228 opts.emit = EmitKind::Ir;
3229 opts.std = Std::C89;
3230 let text = run(&opts, "int f(void) { return dummy(); }\ndummy () { return 7; }\n")
3231 .text()
3232 .to_owned();
3233 assert!(text.contains("func @f()"), "the caller is there: {text}");
3234 assert!(text.contains("func @dummy"), "and so is what it calls: {text}");
3235 assert!(text.contains("iconst.i32 7"), "with the body it was given: {text}");
3236 }
3237
3238 #[test]
3246 fn an_old_style_parameter_is_converted_from_what_the_call_promoted_it_to() {
3247 let mut opts = options();
3248 opts.emit = EmitKind::Ir;
3249 opts.std = Std::C89;
3250 let compiled = |source: &str| run(&opts, source).text().to_owned();
3251
3252 let text = compiled("f (c) unsigned char c; { return c; }\n");
3253 assert!(text.contains("func @f(i32"), "an int arrives: {text}");
3254 assert!(text.contains("trunc.i8"), "and is cut down to what was declared: {text}");
3255 assert!(text.contains("zext.i32"), "then read back unsigned: {text}");
3256
3257 let text = compiled("f (s) short s; { return s; }\n");
3259 assert!(text.contains("trunc.i16"), "cut down: {text}");
3260 assert!(text.contains("sext.i32"), "and read back signed: {text}");
3261
3262 let text = compiled("f (x) float x; { return x * 2; }\n");
3265 assert!(text.contains("func @f(f64"), "a double arrives: {text}");
3266 assert!(text.contains("fptrunc.f32"), "and is narrowed to the float: {text}");
3267
3268 let text = compiled("int f(unsigned char c) { return c; }\n");
3271 assert!(text.contains("func @f(i8)"), "the declared type arrives: {text}");
3272 assert!(!text.contains("trunc"), "so there is nothing to cut down: {text}");
3273 }
3274
3275 #[test]
3284 fn the_rules_gcc_promoted_are_decided_by_the_dialect_and_by_fpermissive() {
3285 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
3287 let cases = [
3288 ("static counted;\n", ["", "error", "warning", "error"]),
3289 ("int f(void) { return g(); }\n", ["", "error", "warning", "error"]),
3290 ("int f(x) { return x; }\n", ["", "error", "warning", "error"]),
3291 ("int *p;\nvoid h(void) { p = 1; }\n", ["warning", "error", "warning", "error"]),
3292 (
3293 "char *q;\nint *r;\nvoid k(void) { r = q; }\n",
3294 ["warning", "error", "warning", "error"],
3295 ),
3296 ("int f(void) { return; }\n", ["", "error", "warning", "error"]),
3297 ("void g(void) { return 1; }\n", ["warning", "error", "warning", "error"]),
3298 ];
3299
3300 for (source, wanted) in cases {
3301 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
3302 let mut opts = options();
3303 opts.std = std;
3304 opts.permissive = permissive;
3305 let said = run(&opts, source).messages.join("\n");
3306 let severity = if said.contains(": error: ") {
3307 "error"
3308 } else if said.contains(": warning: ") {
3309 "warning"
3310 } else {
3311 ""
3312 };
3313 let how = if permissive { " -fpermissive" } else { "" };
3314 assert_eq!(
3315 severity,
3316 wanted,
3317 "under -std={}{how}, {source} was answered with `{said}`",
3318 std.as_str()
3319 );
3320 if wanted.is_empty() {
3321 assert!(said.is_empty(), "nothing to say, but said `{said}`");
3322 }
3323 }
3324 }
3325 }
3326
3327 #[test]
3336 fn the_three_variadic_builtins_answer_a_bad_list_the_way_a_call_answers_a_bad_argument() {
3337 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
3338 let cases = [
3339 (
3340 "int f(int n, ...) { char *p; return __builtin_va_arg(p, int); }\n",
3341 "first argument to 'va_arg' not of type 'va_list'",
3342 ["error", "error", "error", "error"],
3343 ),
3344 (
3345 "void f(int n, ...) { char *p; __builtin_va_start(p, n); }\n",
3346 "passing argument 1 of '__builtin_va_start' from incompatible pointer type",
3347 ["warning", "error", "warning", "error"],
3348 ),
3349 (
3350 "void f(int n, ...) { int x; __builtin_va_end(x); }\n",
3351 "passing argument 1 of '__builtin_va_end' makes pointer from integer without a \
3352 cast",
3353 ["warning", "error", "warning", "error"],
3354 ),
3355 (
3356 "void f(int n, ...) { __builtin_va_list a; char *p; __builtin_va_copy(a, p); }\n",
3357 "passing argument 2 of '__builtin_va_copy' from incompatible pointer type",
3358 ["warning", "error", "warning", "error"],
3359 ),
3360 ];
3361
3362 for (source, message, wanted) in cases {
3363 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
3364 let mut opts = options();
3365 opts.std = std;
3366 opts.permissive = permissive;
3367 let said = run(&opts, source).messages.join("\n");
3368 let how = if permissive { " -fpermissive" } else { "" };
3369 assert!(
3370 said.contains(&format!(": {wanted}: {message}")),
3371 "under -std={}{how}, {source} was answered with `{said}`",
3372 std.as_str()
3373 );
3374 }
3375 }
3376 }
3377
3378 fn safe_ir(tier: rucc_session::Safety, source: &str) -> String {
3380 let mut opts = options();
3381 opts.emit = EmitKind::Ir;
3382 opts.safety = tier;
3383 let result = run(&opts, source);
3384 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3385 result.text().to_owned()
3386 }
3387
3388 const READS_THROUGH_A_POINTER: &str = "int read(int *p) { return p[1]; }\n";
3389
3390 fn padded_ir(padding: Padding, source: &str) -> String {
3392 let mut opts = options();
3393 opts.emit = EmitKind::Ir;
3394 opts.safety = rucc_session::Safety::Detect;
3395 opts.padding = padding;
3396 let result = run(&opts, source);
3397 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3398 result.text().to_owned()
3399 }
3400
3401 const FILLS_A_RECORD_A_MEMBER_AT_A_TIME: &str = "struct padded { char tag; int value; };\n\
3402 void fill(struct padded *p) { p->tag = 1; p->value = 2; }\n";
3403
3404 #[test]
3405 fn a_record_filled_a_member_at_a_time_comes_out_whole_when_padding_does_not_participate() {
3406 let text = padded_ir(Padding::Ignored, FILLS_A_RECORD_A_MEMBER_AT_A_TIME);
3410 assert_eq!(text.matches("owns 4").count(), 2, "{text}");
3411 }
3412
3413 #[test]
3414 fn a_store_says_only_what_it_wrote_when_padding_does_participate() {
3415 let text = padded_ir(Padding::Tracked, FILLS_A_RECORD_A_MEMBER_AT_A_TIME);
3418 assert!(!text.contains("owns"), "{text}");
3419 }
3420
3421 #[test]
3422 fn a_member_of_a_union_owns_nothing_after_it() {
3423 let text = padded_ir(
3427 Padding::Ignored,
3428 "union u { char tag; long wide; };\nvoid fill(union u *p) { p->tag = 1; }\n",
3429 );
3430 assert!(!text.contains("owns"), "{text}");
3431 }
3432
3433 #[test]
3434 fn an_inner_records_trailing_padding_reaches_the_outer_records() {
3435 let text = padded_ir(
3440 Padding::Ignored,
3441 "struct inner { char c; };\n\
3442 struct outer { struct inner in; int x; };\n\
3443 void fill(struct outer *p) { p->in.c = 1; p->x = 2; }\n",
3444 );
3445 assert_eq!(text.matches("owns 4").count(), 2, "{text}");
3446 }
3447
3448 #[test]
3449 fn a_build_that_did_not_ask_for_the_monitor_is_compiled_the_way_it_always_was() {
3450 let text = ir(READS_THROUGH_A_POINTER);
3454 assert!(!text.contains("check_"), "{text}");
3455 assert!(!text.contains("cap_of"), "{text}");
3456 }
3457
3458 #[test]
3459 fn asking_for_a_tier_puts_the_checks_in_before_the_optimizer_sees_them() {
3460 let text = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3461 assert!(text.contains("cap_of"), "{text}");
3462 assert!(text.contains("check_bounds"), "{text}");
3463 assert!(text.contains("check_live"), "{text}");
3464 assert!(text.contains("check_deriv"), "{text}");
3466 assert!(text.contains("check_type"), "{text}");
3468 }
3469
3470 #[test]
3471 fn the_three_tiers_that_are_not_off_all_check_the_same_accesses_so_far() {
3472 let detect = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3476 for tier in [rucc_session::Safety::Enforce, rucc_session::Safety::Kernel] {
3477 assert_eq!(safe_ir(tier, READS_THROUGH_A_POINTER), detect, "{tier}");
3478 }
3479 }
3480
3481 fn summary(tier: rucc_session::Safety, source: &str) -> String {
3483 let mut opts = options();
3484 opts.emit = EmitKind::SafetySummary;
3485 opts.safety = tier;
3486 let result = run(&opts, source);
3487 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3488 result.text().to_owned()
3489 }
3490
3491 #[test]
3492 fn the_summary_counts_the_checks_that_went_in_and_the_ones_still_standing() {
3493 let text = summary(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3494 assert!(text.contains("\"tier\": \"detect\""), "{text}");
3495 assert!(
3497 text.contains("\"bounds\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"),
3498 "{text}"
3499 );
3500 assert!(
3501 text.contains(
3502 "\"derivation\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"
3503 ),
3504 "{text}"
3505 );
3506 }
3507
3508 #[test]
3509 fn a_build_without_the_monitor_summarises_as_a_build_with_no_checks_in_it() {
3510 let text = summary(rucc_session::Safety::Off, READS_THROUGH_A_POINTER);
3514 assert!(text.contains("\"tier\": \"off\""), "{text}");
3515 assert!(
3516 text.contains("\"bounds\": { \"emitted\": 0, \"remaining\": 0, \"discharged\": 0 }"),
3517 "{text}"
3518 );
3519 }
3520
3521 #[test]
3522 fn a_call_the_boundary_models_is_counted_apart_from_one_it_does_not() {
3523 let text = summary(
3524 rucc_session::Safety::Detect,
3525 "void *memcpy(void *, const void *, unsigned long);\n\
3526 int puts(const char *);\n\
3527 void f(char *d, char *s) { memcpy(d, s, 4); puts(d); }\n",
3528 );
3529 assert!(text.contains("\"interposed\": 1"), "{text}");
3530 assert!(text.contains("\"puts\""), "{text}");
3531 assert!(!text.contains("__rucc_wrap_memcpy\""), "{text}");
3535 }
3536
3537 #[test]
3538 fn the_two_directions_a_pointer_crosses_the_boundary_are_counted_apart() {
3539 let text = summary(
3543 rucc_session::Safety::Detect,
3544 "void *notes_open(void);\n\
3545 char *f(char *p) { char *q = notes_open(); return q ? q : p; }\n",
3546 );
3547 assert!(text.contains("\"crossings\": { \"entered\": 1, \"returned\": 1 }"), "{text}");
3548 assert!(text.contains("\"notes_open\""), "{text}");
3549 }
3550
3551 #[test]
3552 fn a_static_function_nobody_takes_the_address_of_is_not_a_crossing() {
3553 let text = summary(
3556 rucc_session::Safety::Detect,
3557 "static int len(const char *p) { return p ? 1 : 0; }\n\
3558 int f(void) { return len(\"x\"); }\n",
3559 );
3560 assert!(text.contains("\"crossings\": { \"entered\": 0, \"returned\": 0 }"), "{text}");
3561 }
3562
3563 fn granules(source: &str) -> String {
3565 let mut opts = options();
3566 opts.emit = EmitKind::TypeGranules;
3567 let result = run(&opts, source);
3568 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3569 result.text().to_owned()
3570 }
3571
3572 #[test]
3573 fn the_granule_report_names_every_record_and_both_keyings() {
3574 let text = granules(
3575 "struct hot { char *p; int a; int b; };\n\
3576 int f(struct hot *h) { return h->a; }\n",
3577 );
3578 assert!(text.contains("struct hot"), "{text}");
3579 assert!(text.contains("every type distinct"), "{text}");
3582 assert!(text.contains("every pointer one type"), "{text}");
3583 assert!(text.contains("budget"), "{text}");
3584 }
3585
3586 #[test]
3587 fn a_record_nothing_uses_is_still_measured() {
3588 let text = granules("struct unused { long a; double b; };\nint f(void) { return 0; }\n");
3591 assert!(text.contains("struct unused"), "{text}");
3592 }
3593
3594 #[test]
3595 fn the_granule_report_stops_before_anything_is_lowered() {
3596 let text = granules(
3600 "struct wide { long double d; };\n\
3601 long double f(long double x) { return x * x; }\n",
3602 );
3603 assert!(text.contains("struct wide"), "{text}");
3604 }
3605
3606 #[test]
3607 fn a_witness_reaches_the_assembler_as_a_call_to_the_runtime() {
3608 let text = safe_asm(rucc_session::Safety::Detect, "char *f(char *p) { return p; }\n");
3611 assert!(text.contains("\tcall\t__rucc_cap_witness\n"), "{text}");
3612 }
3613
3614 #[test]
3615 fn a_pointer_turned_into_an_integer_is_on_the_trust_set() {
3616 let text = summary(
3617 rucc_session::Safety::Detect,
3618 "unsigned long f(int *p) { return (unsigned long) p; }\n",
3619 );
3620 assert!(text.contains("\"exposed\": 1"), "{text}");
3621 }
3622
3623 fn safe_asm(tier: rucc_session::Safety, source: &str) -> String {
3625 let mut opts = options();
3626 opts.emit = EmitKind::Asm;
3627 opts.safety = tier;
3628 let result = run(&opts, source);
3629 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3630 result.text().to_owned()
3631 }
3632
3633 #[test]
3634 fn a_check_reaches_the_assembler_as_a_call_to_the_runtime() {
3635 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3636 assert!(text.contains("\tcall\t__rucc_check_bounds\n"), "{text}");
3637 assert!(text.contains("\tcall\t__rucc_check_live\n"), "{text}");
3638 assert!(text.contains("\tcall\t__rucc_check_deriv\n"), "{text}");
3639 assert!(text.contains("\tcall\t__rucc_check_type\n"), "{text}");
3640 assert!(text.contains("\tcall\t__rucc_check_init\n"), "{text}");
3641 }
3642
3643 #[test]
3644 fn every_check_that_reached_the_assembler_has_a_row_describing_it() {
3645 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3649 let section = format!("\t.section\t{},", rucc_safety::SECTION);
3650 assert_eq!(text.matches(§ion).count(), 5, "{text}");
3651 for index in 0..5 {
3652 let name = format!("__rucc_safety_desc_{index}");
3653 assert!(text.contains(&format!("{name}:\n")), "{text}");
3656 assert!(text.contains(&format!("{name}(%rip)")), "{text}");
3657 }
3658 assert!(!text.contains("__rucc_safety_desc_5"), "{text}");
3659 }
3660
3661 #[test]
3669 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
3670 let text = ir(concat!(
3671 "int g;\n",
3672 "int a = __builtin_constant_p(1);\n",
3673 "int b = __builtin_constant_p(g);\n",
3674 "int c = __builtin_constant_p(\"abc\");\n",
3675 "int d = __builtin_constant_p(&g);\n",
3676 "int e = __builtin_constant_p(1.5);\n",
3677 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
3678 ));
3679 assert!(text.contains("global @a : i32 = 1,"), "{text}");
3680 assert!(text.contains("global @b : i32 = 0,"), "{text}");
3681 assert!(text.contains("global @c : i32 = 1,"), "{text}");
3682 assert!(text.contains("global @d : i32 = 0,"), "{text}");
3683 assert!(text.contains("global @e : i32 = 1,"), "{text}");
3684 assert!(text.contains("global @h : i32 = 11,"), "{text}");
3685 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
3686
3687 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
3691 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
3692 }
3693
3694 #[test]
3703 fn a_call_to_a_library_builtin_reaches_the_library_function() {
3704 let text = body("void f(void) { __builtin_abort(); }\n");
3705 assert_eq!(text, "block0:\n call @abort() : ()\n return\n");
3706
3707 let text = ir("int f(const char *s) { return __builtin_puts(s) + __builtin_strlen(s); }\n");
3710 assert!(text.contains("call @puts(%0) : (ptr) -> i32"), "{text}");
3711 assert!(text.contains("call @strlen(%0) : (ptr) -> i64"), "{text}");
3712 assert!(!text.contains("__builtin_"), "the prefix is not part of any name here:\n{text}");
3713 }
3714
3715 #[test]
3726 fn the_absolute_value_family_is_the_magnitude_and_not_a_call() {
3727 let text = body(concat!(
3728 "long long llabs(long long);\n",
3729 "long long f(long long x) { return llabs(x); }\n",
3730 ));
3731 assert!(text.contains("%1 = iconst.i64 63"), "{text}");
3732 assert!(text.contains("%2 = ashr %0, %1"), "{text}");
3733 assert!(text.contains("%3 = xor %0, %2"), "{text}");
3734 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3735 assert!(!text.contains("call"), "the call does not happen:\n{text}");
3736
3737 let text = body("int abs(int);\nint f(int x) { return abs(x); }\n");
3740 assert!(text.contains("iconst.i32 31"), "{text}");
3741 let text = body("long labs(long);\nlong f(long x) { return labs(x); }\n");
3742 assert!(text.contains("iconst.i64 63"), "{text}");
3743
3744 let text = body("long long f(long long x) { return __builtin_llabs(x); }\n");
3747 assert!(!text.contains("call"), "{text}");
3748
3749 let text = ir(concat!(
3751 "long long llabs(long long b);\n",
3752 "long long g(long long x) { return llabs(x); }\n",
3753 "long long llabs(long long b) { return 7; }\n",
3754 ));
3755 assert!(!text.contains("call @llabs"), "{text}");
3756 }
3757
3758 #[test]
3765 fn a_byte_swap_is_arithmetic_and_not_a_call() {
3766 let text = body("unsigned f(unsigned x) { return __builtin_bswap32(x); }\n");
3767 assert_eq!(text, "block0(%0: i32):\n %1 = bswap %0\n return %1\n");
3768
3769 let text = body("unsigned f(unsigned char c) { return __builtin_bswap32(c); }\n");
3772 assert!(text.contains("zext.i32 %0"), "widened first: {text}");
3773 assert!(text.contains("bswap %1"), "and swapped at four bytes: {text}");
3774 }
3775
3776 #[test]
3782 fn the_byte_swaps_reverse_at_the_width_their_name_says() {
3783 for (name, ty, width) in [
3784 ("__builtin_bswap16", "unsigned short", "i16"),
3785 ("__builtin_bswap32", "unsigned", "i32"),
3786 ("__builtin_bswap64", "unsigned long long", "i64"),
3787 ] {
3788 let source = format!("{ty} f({ty} x) {{ return {name}(x); }}\n");
3789 let text = body(&source);
3790 assert_eq!(
3791 text,
3792 format!("block0(%0: {width}):\n %1 = bswap %0\n return %1\n"),
3793 "{name}"
3794 );
3795 }
3796 }
3797
3798 #[test]
3805 fn the_bit_counts_are_instructions_and_not_calls() {
3806 let text = body("int f(unsigned x) { return __builtin_clz(x); }\n");
3807 assert_eq!(text, "block0(%0: i32):\n %1 = ctlz %0\n return %1\n");
3808
3809 let text = body("int f(unsigned x) { return __builtin_ctz(x); }\n");
3810 assert_eq!(text, "block0(%0: i32):\n %1 = cttz %0\n return %1\n");
3811
3812 let text = body("int f(unsigned x) { return __builtin_popcount(x); }\n");
3813 assert_eq!(text, "block0(%0: i32):\n %1 = ctpop %0\n return %1\n");
3814 }
3815
3816 #[test]
3825 fn the_bit_counts_ask_about_the_width_their_name_says() {
3826 let text = body("int f(unsigned long long x) { return __builtin_clzll(x); }\n");
3827 assert!(text.starts_with("block0(%0: i64):"), "counted at eight bytes: {text}");
3828 assert!(text.contains("%1 = ctlz %0"), "{text}");
3829 assert!(text.contains("trunc.i32 %1"), "and answered in an int: {text}");
3830
3831 let text = body("int f(unsigned long long x) { return __builtin_clz(x); }\n");
3834 assert!(text.contains("trunc.i32 %0"), "narrowed to what was asked about: {text}");
3835 assert!(text.contains("ctlz %1"), "and counted there: {text}");
3836
3837 let text = body("int f(unsigned long x) { return __builtin_popcountl(x); }\n");
3838 assert!(text.contains("%1 = ctpop %0"), "{text}");
3839 assert!(!text.contains("call"), "{text}");
3840 }
3841
3842 #[test]
3847 fn a_parity_is_the_low_bit_of_the_set_bit_count() {
3848 let text = body("int f(unsigned x) { return __builtin_parity(x); }\n");
3849 assert!(text.contains("%1 = ctpop %0"), "{text}");
3850 assert!(text.contains("iconst.i32 1"), "{text}");
3851 assert!(text.contains("and %1, %2"), "the low bit of it: {text}");
3852 }
3853
3854 #[test]
3860 fn the_first_set_bit_is_one_based_and_zero_for_a_zero() {
3861 let text = body("int f(int x) { return __builtin_ffs(x); }\n");
3862 assert!(text.contains("%1 = cttz %0"), "{text}");
3863 assert!(text.contains("%4 = add %1, %2"), "one more than the count: {text}");
3864 assert!(text.contains("%5 = icmp ne %0, %3"), "whether there was a bit at all: {text}");
3865 assert!(text.contains("%7 = sub %3, %6"), "spread to a mask: {text}");
3866 assert!(text.contains("%8 = and %4, %7"), "and kept only then: {text}");
3867 assert!(!text.contains("br_if"), "no branch: {text}");
3868 }
3869
3870 #[test]
3880 fn the_redundant_sign_bit_count_is_instructions_and_not_a_call() {
3881 let text = body("int f(int x) { return __builtin_clrsb(x); }\n");
3882 assert!(text.contains("%1 = iconst.i32 31"), "{text}");
3883 assert!(text.contains("%2 = ashr %0, %1"), "the sign over every bit: {text}");
3884 assert!(text.contains("%3 = xor %0, %2"), "folded onto it: {text}");
3885 assert!(text.contains("%5 = shl %3, %4"), "one less than the count: {text}");
3886 assert!(text.contains("%6 = or %5, %4"), "with something to count at zero: {text}");
3887 assert!(text.contains("%7 = ctlz %6"), "{text}");
3888 assert!(!text.contains("call"), "{text}");
3889 assert!(!text.contains("br_if"), "no branch: {text}");
3890 }
3891
3892 #[test]
3898 fn the_unsigned_absolute_value_family_answers_in_the_unsigned_type() {
3899 let text = body("unsigned f(int x) { return __builtin_uabs(x); }\n");
3900 assert!(text.contains("%1 = iconst.i32 31"), "{text}");
3901 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3902 assert!(!text.contains("call"), "nothing declares uabs, so a call would not link: {text}");
3903
3904 let text = body("unsigned long long f(long long x) { return __builtin_ullabs(x); }\n");
3905 assert!(text.contains("iconst.i64 63"), "at the width the name says: {text}");
3906
3907 let text = body("int f(int x) { return __builtin_uabs(x) > 2147483647u; }\n");
3910 assert!(text.contains("icmp ugt"), "compared unsigned: {text}");
3911 }
3912
3913 #[test]
3921 fn the_widest_absolute_value_is_whichever_type_the_target_makes_intmax_t() {
3922 let text = body("long f(long x) { return __builtin_imaxabs(x); }\n");
3923 assert!(text.contains("iconst.i64 63"), "{text}");
3924 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3925 assert!(!text.contains("call"), "{text}");
3926
3927 let text = body("unsigned long f(long x) { return __builtin_umaxabs(x); }\n");
3928 assert!(text.contains("iconst.i64 63"), "{text}");
3929 assert!(!text.contains("call"), "{text}");
3930 }
3931
3932 #[test]
3940 fn an_overflow_predicate_writes_nothing_and_answers_the_bit_the_check_would() {
3941 let text =
3942 body("int f(int a, int b) { return __builtin_add_overflow_p(a, b, (int) 0); }\n");
3943 assert!(text.contains("%2, %3 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
3944 assert!(!text.contains("store"), "nothing is written: {text}");
3945 assert!(!text.contains("call"), "{text}");
3946
3947 let text =
3950 body("int f(int a, int b) { return __builtin_mul_overflow_p(a, b, (long long) 0); }\n");
3951 assert!(text.contains("smul_overflow.(i64, i1)"), "{text}");
3952 assert!(!text.contains("store"), "{text}");
3953
3954 let text = body(concat!(
3957 "int g(void);\n",
3958 "int f(int a, int b) { return __builtin_sub_overflow_p(a, b, g()); }\n",
3959 ));
3960 assert!(!text.contains("call @g"), "the third argument is not evaluated: {text}");
3961 }
3962
3963 #[test]
3973 fn an_overflow_check_is_arithmetic_and_not_a_call() {
3974 let text =
3975 body("int f(int a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3976 assert!(text.contains("%3, %4 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
3977 assert!(text.contains("store %3 -> %2"), "{text}");
3978 assert!(!text.contains("call"), "{text}");
3979
3980 let text =
3981 body("int f(int a, int b, int *r) { return __builtin_sub_overflow(a, b, r); }\n");
3982 assert!(text.contains("ssub_overflow.(i32, i1) %0, %1"), "{text}");
3983
3984 let text =
3985 body("int f(int a, int b, int *r) { return __builtin_mul_overflow(a, b, r); }\n");
3986 assert!(text.contains("smul_overflow.(i32, i1) %0, %1"), "{text}");
3987
3988 let text = body(
3991 "int f(unsigned a, unsigned b, unsigned *r) { return __builtin_add_overflow(a, b, r); }\n",
3992 );
3993 assert!(text.contains("uadd_overflow.(i32, i1) %0, %1"), "{text}");
3994 }
3995
3996 #[test]
4004 fn an_overflow_check_is_done_at_a_type_that_holds_every_operand() {
4005 let text = body(
4006 "int f(unsigned a, int b, long long *r) { return __builtin_add_overflow(a, b, r); }\n",
4007 );
4008 assert!(text.contains("%3 = zext.i64 %0"), "the unsigned operand keeps its value: {text}");
4009 assert!(text.contains("%4 = sext.i64 %1"), "and so does the signed one: {text}");
4010 assert!(text.contains("sadd_overflow.(i64, i1) %3, %4"), "{text}");
4011
4012 let text = body(
4015 "int f(long long a, long long b, long long *r) { return __builtin_mul_overflow(a, b, r); }\n",
4016 );
4017 assert!(text.contains("smul_overflow.(i64, i1) %0, %1"), "{text}");
4018 assert!(!text.contains("sext."), "{text}");
4019 assert!(!text.contains("zext.i64"), "{text}");
4021 }
4022
4023 #[test]
4031 fn an_overflow_check_writes_the_wrapped_answer_whether_or_not_it_fit() {
4032 let text =
4033 body("int f(int a, int b, char *r) { return __builtin_sub_overflow(a, b, r); }\n");
4034 assert!(text.contains("%3, %4 = ssub_overflow.(i32, i1) %0, %1"), "{text}");
4035 assert!(text.contains("%5 = trunc.i8 %3"), "narrowed to where it goes: {text}");
4036 assert!(text.contains("%6 = sext.i32 %5"), "and back: {text}");
4037 assert!(text.contains("%7 = icmp ne %6, %3"), "which is whether it fit: {text}");
4038 assert!(text.contains("store %5 -> %2"), "the narrowed value is stored either way: {text}");
4039 assert!(text.contains("%8 = or %4, %7"), "and either bit is an overflow: {text}");
4040 }
4041
4042 #[test]
4049 fn a_call_needing_more_than_the_widest_type_still_compiles() {
4050 for name in ["add", "sub", "mul"] {
4051 let source = format!(
4052 "int f(unsigned __int128 a, long long b, __int128 *r) {{\n \
4053 return __builtin_{name}_overflow(a, b, r);\n}}\n"
4054 );
4055 let mut opts = options();
4056 opts.emit = EmitKind::MirFinal;
4057 assert!(!run(&opts, &source).failed(), "{name} was refused or stopped the back end");
4058 }
4059 }
4060
4061 #[test]
4064 fn an_overflow_check_over_something_that_is_not_an_integer_says_so() {
4065 let messages =
4066 errors("int f(double a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
4067 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
4068
4069 let messages =
4070 errors("int f(int a, int b, double *r) { return __builtin_add_overflow(a, b, r); }\n");
4071 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
4072 }
4073
4074 #[test]
4085 fn an_ordered_access_is_ordered_in_the_ir() {
4086 let text = body("int f(int *p) { return __atomic_load_n(p, 0); }\n");
4087 assert!(text.contains("atomic_load.i32 %0, align 4, relaxed"), "{text}");
4088
4089 let text = body("long f(long *p) { return __atomic_load_n(p, 2); }\n");
4090 assert!(text.contains("atomic_load.i64 %0, align 8, acquire"), "{text}");
4091
4092 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
4093 assert!(text.contains("atomic_store %1 -> %0, align 4, release"), "{text}");
4094
4095 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
4096 assert!(text.contains("atomic_store %1 -> %0, align 4, seq_cst"), "{text}");
4097
4098 let text = body("void f(char *p, int v) { __atomic_store_n(p, v, 0); }\n");
4101 assert!(text.contains("trunc.i8 %1"), "{text}");
4102 assert!(text.contains("atomic_store %2 -> %0, align 1, relaxed"), "{text}");
4103 }
4104
4105 #[test]
4114 fn an_ordered_access_is_the_plain_instruction_on_this_machine() {
4115 let text = asm("int f(int *p) { return __atomic_load_n(p, 5); }\n");
4116 assert!(text.contains("movl\t(%rdi), %eax"), "{text}");
4117 assert!(!text.contains("mfence"), "a load needs no barrier here: {text}");
4118
4119 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
4120 assert!(text.contains("movl\t%esi, (%rdi)"), "{text}");
4121 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
4122
4123 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
4124 let (before, after) = text.split_once("mfence").expect("a barrier: {text}");
4125 assert!(before.contains("movl\t%esi, (%rdi)"), "the store comes first: {text}");
4126 assert!(!after.contains("movl"), "and nothing else is between them: {text}");
4127 }
4128
4129 #[test]
4139 fn a_barrier_is_one_instruction_at_the_strongest_ordering_and_none_below_it() {
4140 assert!(asm("void f(void) { __atomic_thread_fence(5); }\n").contains("mfence"));
4141 assert!(asm("void f(void) { __sync_synchronize(); }\n").contains("mfence"));
4142
4143 for weaker in ["1", "2", "3", "4"] {
4144 let source = format!("void f(void) {{ __atomic_thread_fence({weaker}); }}\n");
4145 assert!(!asm(&source).contains("mfence"), "{weaker} costs nothing here");
4146 }
4147 }
4148
4149 #[test]
4155 fn a_compare_and_exchange_is_one_instruction_answering_two_things() {
4156 let text =
4159 body("int f(int *p, int e, int d) { return __sync_val_compare_and_swap(p, e, d); }\n");
4160 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
4161 assert!(text.contains("return %3"), "the value it found: {text}");
4162
4163 let text =
4164 body("int f(int *p, int e, int d) { return __sync_bool_compare_and_swap(p, e, d); }\n");
4165 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
4166 assert!(text.contains("zext.i32 %4"), "whether it happened: {text}");
4167
4168 let text = body(
4171 "int f(int *p, int *e, int d) { return __atomic_compare_exchange_n(p, e, d, 0, 4, 2); }\n",
4172 );
4173 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
4174 assert!(text.contains("%4, %5 = cmpxchg.(i32, i1) %0, %3, %2, align 4, acq_rel"), "{text}");
4175 assert!(text.contains("br_if %5, block2, block1"), "{text}");
4176 assert!(text.contains("store %4 -> %1, align 4"), "{text}");
4177
4178 let text = body(
4181 "int f(int *p, int *e, int *d) { return __atomic_compare_exchange(p, e, d, 0, 5, 5); }\n",
4182 );
4183 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
4184 assert!(text.contains("%4 = load.i32 %2, align 4"), "{text}");
4185 assert!(text.contains("%5, %6 = cmpxchg.(i32, i1) %0, %3, %4, align 4, seq_cst"), "{text}");
4186 }
4187
4188 #[test]
4195 fn a_compare_and_exchange_is_a_locked_instruction_at_the_width_of_the_object() {
4196 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
4197 for (ty, suffix, reg) in widths {
4198 let source = format!(
4199 "int f({ty} *p, {ty} e, {ty} d) {{ return __sync_bool_compare_and_swap(p, e, d); }}\n"
4200 );
4201 let text = asm(&source);
4202 assert!(text.contains("\tlock\n"), "{ty}: {text}");
4203 assert!(text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
4204 assert!(text.contains("sete\t"), "{ty}: {text}");
4205 }
4206 let source =
4207 "int f(long *p, long e, long d) { return __sync_bool_compare_and_swap(p, e, d); }\n";
4208 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
4209
4210 for order in ["0", "2", "3", "4", "5"] {
4214 let call = format!("__atomic_compare_exchange_n(p, e, d, 0, {order}, 0)");
4215 let source = format!("int f(int *p, int *e, int d) {{ return {call}; }}\n");
4216 let text = asm(&source);
4217 assert!(text.contains("cmpxchgl\t"), "{order}: {text}");
4218 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
4219 }
4220 }
4221
4222 #[test]
4234 fn a_read_modify_write_is_one_instruction_and_the_arithmetic_a_name_asks_for() {
4235 let text = body("int f(int *p, int v) { return __atomic_fetch_add(p, v, 5); }\n");
4236 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
4237 assert!(text.contains("return %2"), "the value that was there: {text}");
4238
4239 let text = body("int f(int *p, int v) { return __atomic_add_fetch(p, v, 5); }\n");
4240 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
4241 assert!(text.contains("%3 = add %2, %1"), "and the value afterwards: {text}");
4242
4243 let text = body("int f(int *p, int v) { return __atomic_sub_fetch(p, v, 5); }\n");
4244 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
4245 assert!(text.contains("%3 = sub %2, %1"), "{text}");
4246
4247 let text = body("int f(int *p, int v) { return __sync_fetch_and_sub(p, v); }\n");
4249 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
4250
4251 let text = body("int f(int *p, int v) { return __atomic_exchange_n(p, v, 5); }\n");
4254 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, seq_cst"), "{text}");
4255
4256 let text = body("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
4257 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, acquire"), "{text}");
4258
4259 let text = body("void f(int *p) { __sync_lock_release(p); }\n");
4262 assert!(text.contains("release"), "{text}");
4263 assert!(text.contains("%1 = iconst.i32 0"), "{text}");
4264
4265 let text = body("void f(int *p, int guard) { __sync_lock_release(p, guard); }\n");
4269 assert!(text.contains("%2 = iconst.i32 0"), "{text}");
4270 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
4271
4272 let text = body("int f(int *p, int v) { return __atomic_fetch_and(p, v, 5); }\n");
4275 assert!(text.contains("%2 = atomic_rmw.i32 and %0, %1, align 4, seq_cst"), "{text}");
4276
4277 let text = body("int f(int *p, int v) { return __sync_or_and_fetch(p, v); }\n");
4278 assert!(text.contains("%2 = atomic_rmw.i32 or %0, %1, align 4, seq_cst"), "{text}");
4279 assert!(text.contains("%3 = or %2, %1"), "and the value afterwards: {text}");
4280
4281 let text = body("int f(int *p, int v) { return __atomic_nand_fetch(p, v, 5); }\n");
4284 assert!(text.contains("%2 = atomic_rmw.i32 nand %0, %1, align 4, seq_cst"), "{text}");
4285 assert!(text.contains("%3 = and %2, %1"), "{text}");
4286 assert!(text.contains("%4 = iconst.i32 -1"), "{text}");
4287 assert!(text.contains("%5 = xor %3, %4"), "{text}");
4288 }
4289
4290 #[test]
4301 fn a_bitwise_read_modify_write_is_a_loop_around_the_compare_and_exchange() {
4302 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
4303 for (ty, suffix, reg) in widths {
4304 for (name, call, insn) in [
4305 ("and", "__atomic_fetch_and(p, v, 5)", "and"),
4306 ("or", "__sync_fetch_and_or(p, v)", "or"),
4307 ("xor", "__atomic_xor_fetch(p, v, 5)", "xor"),
4308 ] {
4309 let source = format!("{ty} f({ty} *p, {ty} v) {{ return {call}; }}\n");
4310 let text = asm(&source);
4311 assert!(text.contains("\tlock\n"), "{ty} {name}: {text}");
4312 assert!(
4313 text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")),
4314 "{ty} {name}: {text}"
4315 );
4316 assert!(text.contains(&format!("{insn}{suffix}\t")), "{ty} {name}: {text}");
4317 assert!(!text.contains("\txadd"), "{ty} {name} is not an add: {text}");
4319 assert!(!text.contains("\txchg"), "{ty} {name} is not an exchange: {text}");
4320 }
4321 }
4322 let source = "long f(long *p, long v) { return __atomic_fetch_or(p, v, 5); }\n";
4323 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
4324
4325 let text = asm("int f(int *p, int v) { return __sync_fetch_and_nand(p, v); }\n");
4329 assert!(text.contains("cmpxchgl\t"), "{text}");
4330 assert!(text.contains("andl\t"), "{text}");
4331 assert!(text.contains("notl\t"), "{text}");
4332 }
4333
4334 #[test]
4343 fn an_access_through_a_second_pointer_is_the_same_access_and_one_more() {
4344 let text = body("void f(int *p, int *r) { __atomic_load(p, r, 5); }\n");
4345 assert!(text.contains("%2 = atomic_load.i32 %0, align 4, seq_cst"), "{text}");
4346 assert!(text.contains("store %2 -> %1, align 4"), "and out through the place: {text}");
4347
4348 let text = body("void f(int *p, int *v) { __atomic_store(p, v, 3); }\n");
4349 assert!(text.contains("%2 = load.i32 %1, align 4"), "in through the place: {text}");
4350 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
4351
4352 let text = body("void f(int *p, int *v, int *r) { __atomic_exchange(p, v, r, 5); }\n");
4355 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
4356 assert!(text.contains("%4 = atomic_rmw.i32 xchg %0, %3, align 4, seq_cst"), "{text}");
4357 assert!(text.contains("store %4 -> %2, align 4"), "{text}");
4358 }
4359
4360 #[test]
4371 fn a_flag_is_an_exchange_of_one_byte_and_a_store_of_a_zero_over_the_same_byte() {
4372 for pointer in ["char", "int", "void"] {
4373 let source = format!("int f({pointer} *p) {{ return __atomic_test_and_set(p, 5); }}\n");
4374 let text = body(&source);
4375 assert!(text.contains("%1 = iconst.i8 1"), "{pointer}: {text}");
4376 assert!(
4377 text.contains("%2 = atomic_rmw.i8 xchg %0, %1, align 1, seq_cst"),
4378 "{pointer}: {text}"
4379 );
4380 assert!(text.contains("%4 = icmp ne %2, %3"), "{pointer}: {text}");
4381
4382 let source = format!("void f({pointer} *p) {{ __atomic_clear(p, 3); }}\n");
4383 let text = body(&source);
4384 assert!(text.contains("atomic_store %2 -> %0, align 1, release"), "{pointer}: {text}");
4385 }
4386
4387 let text = asm("int f(int *p) { return __atomic_test_and_set(p, 5); }\n");
4390 assert!(text.contains("xchgb\t%al, (%rdi)"), "{text}");
4391 assert!(text.contains("setne\t"), "{text}");
4392 }
4393
4394 #[test]
4402 fn a_read_modify_write_is_an_exchange_or_a_locked_add_at_the_width_of_the_object() {
4403 let widths = [("char", "b", "%sil"), ("short", "w", "%si"), ("int", "l", "%esi")];
4404 for (ty, suffix, reg) in widths {
4405 let source =
4406 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_fetch_add(p, v, 5); }}\n");
4407 let text = asm(&source);
4408 assert!(text.contains("\tlock\n"), "{ty}: {text}");
4409 assert!(text.contains(&format!("xadd{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
4410
4411 let source =
4412 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_exchange_n(p, v, 5); }}\n");
4413 let text = asm(&source);
4414 assert!(text.contains(&format!("xchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
4415 assert!(!text.contains("\tlock\n"), "an exchange is locked already: {ty}: {text}");
4416 }
4417 let source = "long f(long *p, long v) { return __atomic_fetch_add(p, v, 5); }\n";
4418 assert!(asm(source).contains("xaddq\t%rsi, (%rdi)"), "{}", asm(source));
4419
4420 let source = "int f(int *p, int v) { return __atomic_fetch_sub(p, v, 5); }\n";
4423 let text = asm(source);
4424 assert!(text.contains("negl\t"), "{text}");
4425 assert!(text.contains("xaddl\t"), "{text}");
4426
4427 for order in ["0", "2", "3", "4", "5"] {
4430 let source =
4431 format!("int f(int *p, int v) {{ return __atomic_fetch_add(p, v, {order}); }}\n");
4432 let text = asm(&source);
4433 assert!(text.contains("xaddl\t"), "{order}: {text}");
4434 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
4435 }
4436
4437 let text = asm("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
4441 assert!(text.contains("xchgl\t%esi, (%rdi)"), "{text}");
4442 let text = asm("void f(int *p) { __sync_lock_release(p); }\n");
4447 assert!(text.contains("movl\t$0, %eax"), "{text}");
4448 assert!(text.contains("movl\t%eax, (%rdi)"), "{text}");
4449 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
4450 }
4451
4452 #[test]
4464 fn the_lock_free_questions_are_answered_as_constants() {
4465 for size in ["1", "2", "4", "8"] {
4466 let source =
4467 format!("int f(void) {{ return __atomic_always_lock_free({size}, 0); }}\n");
4468 let text = asm(&source);
4469 assert!(text.contains("movb\t$1, %al"), "{size} bytes is lock free: {text}");
4470 assert!(!text.contains("call"), "and is not a call: {text}");
4471 }
4472 for size in ["3", "16", "sizeof(long double)"] {
4473 let source = format!("int f(void) {{ return __atomic_is_lock_free({size}, 0); }}\n");
4474 let text = asm(&source);
4475 assert!(text.contains("movb\t$0, %al"), "{size} bytes is not: {text}");
4476 assert!(!text.contains("call"), "and is not a call either: {text}");
4477 }
4478
4479 let text = asm("int f(int n) { return __atomic_is_lock_free(n, 0); }\n");
4483 assert!(text.contains("movb\t$0, %al"), "a size nobody knows is not lock free: {text}");
4484 let text = asm("int f(int *p) { return __atomic_always_lock_free(8, p); }\n");
4485 assert!(text.contains("movb\t$0, %al"), "eight bytes at four is not: {text}");
4486 let text = asm("int f(long *p) { return __atomic_always_lock_free(8, p); }\n");
4487 assert!(text.contains("movb\t$1, %al"), "and at eight it is: {text}");
4488 }
4489
4490 #[test]
4502 fn a_memory_order_an_operation_cannot_carry_is_read_as_the_strongest() {
4503 let mut opts = options();
4504 opts.emit = EmitKind::Ir;
4505
4506 let acquire_store = run(&opts, "void f(int *p, int v) { __atomic_store_n(p, v, 2); }\n");
4507 assert!(acquire_store.text().contains("seq_cst"), "{:?}", acquire_store.text());
4508 assert!(acquire_store.messages[0].contains("[W0333]"), "{:?}", acquire_store.messages);
4509
4510 let nonsense = run(&opts, "int f(int *p) { return __atomic_load_n(p, 99); }\n");
4511 assert!(nonsense.text().contains("seq_cst"), "{:?}", nonsense.text());
4512 assert!(nonsense.messages[0].contains("[W0333]"), "{:?}", nonsense.messages);
4513
4514 let computed = run(&opts, "int f(int *p, int n) { return __atomic_load_n(p, n); }\n");
4515 assert!(computed.text().contains("seq_cst"), "{:?}", computed.text());
4516 assert_eq!(computed.messages, Vec::<String>::new(), "a computed order is not a mistake");
4517 }
4518
4519 #[test]
4531 fn a_conversion_between_a_float_and_the_widest_unsigned_integer_is_written_without_a_branch() {
4532 let text = asm("double f(unsigned long long x) { return (double)x; }\n");
4533 assert!(text.contains("cvtsi2sdq"), "the signed conversion is what runs: {text}");
4534 assert!(text.contains("shrq"), "with the value halved first: {text}");
4535 assert!(text.contains("addsd"), "and doubled after: {text}");
4536 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
4537
4538 let text = asm("unsigned long long f(double d) { return (unsigned long long)d; }\n");
4539 assert!(text.contains("cvttsd2siq"), "the signed conversion is what runs: {text}");
4540 assert!(text.contains("subsd"), "with half the range taken off first: {text}");
4541 assert!(text.contains("shlq\t$63"), "and the top bit put back: {text}");
4542 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
4543 }
4544
4545 #[test]
4556 fn a_plain_name_the_program_took_is_the_programs_own_function() {
4557 let taken = concat!(
4558 "static long long llabs(long long b) { return 7; }\n",
4559 "long long f(long long x) { return llabs(x); }\n",
4560 );
4561 assert!(ir(taken).contains("call @llabs"), "a static definition is the program's own");
4562
4563 let retyped = concat!("int llabs(int b);\n", "int f(int x) { return llabs(x); }\n",);
4564 assert!(ir(retyped).contains("call @llabs"), "another type is another function");
4565
4566 let plain = concat!(
4567 "long long llabs(long long b);\n",
4568 "long long f(long long x) { return llabs(x); }\n",
4569 );
4570 let mut opts = options();
4571 opts.emit = EmitKind::Ir;
4572 assert!(!run(&opts, plain).text().contains("call @llabs"), "the library's by default");
4573
4574 opts.builtins = false;
4575 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin");
4576
4577 opts.builtins = true;
4578 opts.no_builtin = vec!["llabs".to_owned()];
4579 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin-llabs");
4580 let one = "long labs(long b);\nlong f(long x) { return labs(x); }\n";
4581 assert!(!run(&opts, one).text().contains("call @labs"), "one name and not the family");
4582
4583 opts.no_builtin = Vec::new();
4586 opts.builtins = false;
4587 let prefixed = "long long f(long long x) { return __builtin_llabs(x); }\n";
4588 assert!(!run(&opts, prefixed).text().contains("call @llabs"), "the prefix is a promise");
4589 }
4590
4591 #[test]
4604 fn the_hint_builtins_are_their_first_argument_and_the_hint_leaves_no_trace() {
4605 let text = ir(concat!(
4606 "long a = __builtin_expect(7, 1);\n",
4607 "long b = __builtin_expect_with_probability(9, 1, 0.9);\n",
4608 "unsigned long c = sizeof(__builtin_expect((char)1, 1));\n",
4609 ));
4610 assert!(text.contains("global @a : i64 = 7,"), "{text}");
4611 assert!(text.contains("global @b : i64 = 9,"), "{text}");
4612 assert!(text.contains("global @c : i64 = 8,"), "{text}");
4613 assert!(!text.contains("__builtin_expect"), "it is not a call to anything:\n{text}");
4614
4615 let text = body("long f(char c) { return __builtin_expect(c, 1); }\n");
4618 assert!(text.contains("sext"), "{text}");
4619
4620 let one = "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 1\n %2 = sext.i64 %1\n return %0\n";
4624 assert_eq!(body("int f(void) { int i = 0; __builtin_expect(1, i++); return i; }\n"), one);
4625 let source = "int g(void) { int i = 0; __builtin_expect_with_probability(1, i++, 0.5); return i; }\n";
4626 assert_eq!(body(source), one);
4627
4628 let kept = body("int f(int n) { int i = 0; __builtin_expect(n, i++); return i; }\n");
4633 assert!(kept.contains("add.nsw"), "the hint still runs: {kept}");
4634 assert!(kept.ends_with("return %3\n"), "and the answer is what it left behind: {kept}");
4635 let both = "int g(int n) { int i = 0; __builtin_expect_with_probability(n, i++, 0.5); return i; }\n";
4636 assert!(body(both).contains("add.nsw"), "and so does the one with three arguments");
4637 }
4638
4639 #[test]
4651 fn a_promise_that_control_does_not_arrive_writes_no_instruction() {
4652 let promised = "int f(int x) { if (x) return 1; __builtin_unreachable(); }\n";
4653 let text = ir(promised);
4654 assert!(text.contains(" unreachable_hint\n"), "{text}");
4655 assert!(!text.contains("call"), "it is not a call to anything:\n{text}");
4656
4657 let after = body("int g(int x) { __builtin_unreachable(); return x; }\n");
4661 assert!(after.contains("return"), "{after}");
4662
4663 let text = asm(promised);
4666 let mine = text.split_once("\nf:\n").expect("a definition").1;
4667 let mine = mine.split_once("\t.size").expect("a definition").0;
4668 let plain = asm("int f(int x) { if (x) return 1; }\n");
4669 let plain = plain.split_once("\nf:\n").expect("a definition").1;
4670 let plain = plain.split_once("\t.size").expect("a definition").0;
4671 assert_eq!(mine, plain);
4672 let last = mine.lines().rfind(|line| !line.trim_start().starts_with('.'));
4675 assert_eq!(last.map(str::trim), Some("ret"), "{mine}");
4676 assert!(!mine.contains("ud2"), "{mine}");
4677 }
4678
4679 #[test]
4686 fn a_library_builtin_is_diagnosed_under_the_name_the_program_wrote() {
4687 let mut opts = options();
4688 opts.emit = EmitKind::Ir;
4689 let messages = run(&opts, "void f(void) { __builtin_abort(1); }\n").messages;
4690 assert!(
4691 messages.iter().any(|m| m.contains("__builtin_abort")),
4692 "expected the written name in {messages:?}"
4693 );
4694 }
4695
4696 #[test]
4705 fn a_builtin_nothing_lowers_is_refused_by_name() {
4706 let mut opts = options();
4707 opts.emit = EmitKind::Ir;
4708 for (builtin, call) in [
4709 ("__builtin_object_size", "(int)__builtin_object_size(&counter, 0)"),
4710 ("__builtin_dynamic_object_size", "(int)__builtin_dynamic_object_size(&counter, 0)"),
4711 ("__atomic_signal_fence", "(__atomic_signal_fence(5), 0)"),
4712 ] {
4713 let source = format!("int counter;\nint f(void) {{ return {call}; }}\n");
4714 let messages = run(&opts, &source).messages;
4715 let named = messages.iter().any(|m| m.contains(builtin) && m.contains("E0686"));
4716 assert!(named, "expected {builtin} to be refused by name in {messages:?}");
4717 }
4718 }
4719
4720 #[test]
4728 fn what_is_refused_is_the_call_and_not_the_name() {
4729 let text = ir("unsigned long n = sizeof(__builtin_object_size(0, 0));\n");
4730 assert!(text.contains("global @n : i64 = 8,"), "{text}");
4731
4732 let text = ir(concat!(
4733 "unsigned long __builtin_object_size(const void *p, int kind) { return 0; }\n",
4734 "unsigned long f(void) { return __builtin_object_size(0, 0); }\n",
4735 ));
4736 assert!(text.contains("call @__builtin_object_size"), "{text}");
4737 }
4738
4739 #[test]
4744 fn a_static_function_nothing_refers_to_is_not_emitted() {
4745 let text = ir("static int dropped(void) { return 1; }\n\
4746 static int kept(void) { return 2; }\n\
4747 int main(void) { return kept(); }\n");
4748 assert!(text.contains("func @kept"), "{text}");
4749 assert!(!text.contains("dropped"), "{text}");
4750 }
4751
4752 #[test]
4758 fn two_static_functions_that_only_call_each_other_are_both_dropped() {
4759 let text = ir("static int ping(void);\n\
4760 static int pong(void) { return ping(); }\n\
4761 static int ping(void) { return pong(); }\n\
4762 int main(void) { return 0; }\n");
4763 assert!(!text.contains("ping"), "{text}");
4764 assert!(!text.contains("pong"), "{text}");
4765 }
4766
4767 #[test]
4773 fn naming_a_static_function_anywhere_keeps_it() {
4774 let text = ir("static int by_address(void) { return 1; }\n\
4775 static int in_an_image(void) { return 2; }\n\
4776 static int deeper(void) { return 3; }\n\
4777 static int reaches_deeper(void) { return deeper(); }\n\
4778 static int (*table[1])(void) = {in_an_image};\n\
4779 int main(void) {\n\
4780 int (*p)(void) = by_address;\n\
4781 return p() + table[0]() + reaches_deeper();\n\
4782 }\n");
4783 for kept in ["by_address", "in_an_image", "deeper", "reaches_deeper"] {
4784 assert!(text.contains(&format!("func @{kept}")), "expected {kept} in:\n{text}");
4785 }
4786 }
4787
4788 #[test]
4794 fn an_attribute_keeps_a_static_function_nothing_refers_to() {
4795 for attribute in ["used", "retain", "constructor", "destructor", "__used__"] {
4796 let source = format!(
4797 "__attribute__(({attribute})) static int kept(void) {{ return 1; }}\n\
4798 int main(void) {{ return 0; }}\n"
4799 );
4800 let text = ir(&source);
4801 assert!(text.contains("func @kept"), "for {attribute}:\n{text}");
4802 }
4803 }
4804
4805 #[test]
4808 fn a_function_anything_could_call_is_emitted_without_being_called() {
4809 let text =
4810 ir("int nobody_here_calls_it(void) { return 1; }\nint main(void) { return 0; }\n");
4811 assert!(text.contains("func @nobody_here_calls_it"), "{text}");
4812 }
4813
4814 #[test]
4821 fn a_classification_c_has_an_operator_for_is_that_operator() {
4822 for (builtin, operator) in [
4823 ("__builtin_isgreater", "binary >"),
4824 ("__builtin_isgreaterequal", "binary >="),
4825 ("__builtin_isless", "binary <"),
4826 ("__builtin_islessequal", "binary <="),
4827 ] {
4828 let source = format!("int f(double x, double y) {{ return {builtin}(x, y); }}\n");
4829 let text = tast(&source);
4830 assert!(text.contains(&format!("{operator} : int")), "for {builtin}:\n{text}");
4831 }
4832 }
4833
4834 #[test]
4843 fn the_classification_builtins_are_comparisons_and_not_calls() {
4844 let text = body("int f(double x, double y) { return __builtin_isunordered(x, y); }\n");
4845 assert_eq!(
4846 text,
4847 "block0(%0: f64, %1: f64):\n %2 = fcmp uno %0, %1\n %3 = zext.i32 \
4848 %2\n return %3\n"
4849 );
4850
4851 let text = body("int f(double x, double y) { return __builtin_islessgreater(x, y); }\n");
4853 assert!(text.contains("fcmp one %0, %1"), "{text}");
4854
4855 let text = body("int f(double x) { return __builtin_isnan(x); }\n");
4856 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4857
4858 let text = body("int f(double x) { return __builtin_isinf(x); }\n");
4859 assert!(text.contains("fconst.f64 0x7ff0000000000000"), "{text}");
4860 assert!(text.contains("fconst.f64 0xfff0000000000000"), "{text}");
4861 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4862 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4863 assert!(text.contains("%5 = or %3, %4"), "{text}");
4864
4865 let text = body("int f(double x) { return __builtin_isfinite(x); }\n");
4868 assert!(text.contains("%3 = fcmp olt %2, %0"), "{text}");
4869 assert!(text.contains("%4 = fcmp olt %0, %1"), "{text}");
4870 assert!(text.contains("%5 = and %3, %4"), "{text}");
4871
4872 let text = body("int f(double x) { return __builtin_signbit(x); }\n");
4873 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4874 assert!(text.contains("icmp slt %1, %2"), "{text}");
4875
4876 let text = body("int f(long double x) { return __builtin_signbitl(x); }\n");
4879 assert!(text.contains("%1 = bitcast.i80 %0"), "{text}");
4880
4881 let text = body("double g(void);\nint f(void) { return __builtin_isnan(g()); }\n");
4884 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4885 }
4886
4887 #[test]
4894 fn a_classification_spelling_that_names_a_width_converts_before_it_asks() {
4895 let text = ir(concat!(
4896 "int a = __builtin_isinff(1e300);\n",
4897 "int b = __builtin_isinf(1e300);\n",
4898 "int c = __builtin_isnan(0.0);\n",
4902 "int d = __builtin_signbit(-0.0);\n",
4903 "int e = __builtin_islessgreater(1.0, 2.0);\n",
4904 ));
4905 assert!(text.contains("global @a : i32 = 1,"), "{text}");
4906 assert!(text.contains("global @b : i32 = 0,"), "{text}");
4907 assert!(text.contains("global @c : i32 = 0,"), "{text}");
4908 assert!(text.contains("global @d : i32 = 1,"), "{text}");
4909 assert!(text.contains("global @e : i32 = 1,"), "{text}");
4910 }
4911
4912 #[test]
4914 fn a_classification_builtin_refuses_an_argument_that_is_not_floating_point() {
4915 let mut opts = options();
4916 opts.emit = EmitKind::Ir;
4917 let source = concat!(
4918 "int a(int x) { return __builtin_isnan(x); }\n",
4919 "int b(int x, int y) { return __builtin_isunordered(x, y); }\n",
4920 "int c(double x) { return __builtin_isnan(x, x); }\n",
4921 );
4922 let messages = run(&opts, source).messages;
4923 assert_eq!(
4924 messages,
4925 [
4926 "/main.c:1:23: error: non-floating-point argument in call to function \
4927 '__builtin_isnan' [E0685]",
4928 "/main.c:2:30: error: non-floating-point arguments in call to function \
4929 '__builtin_isunordered' [E0685]",
4930 "/main.c:3:26: error: too many arguments to function '__builtin_isnan' [E0511]",
4931 ]
4932 );
4933 }
4934
4935 #[test]
4944 fn the_last_three_classification_builtins_are_comparisons_and_not_calls() {
4945 let text = body("int f(double x) { return __builtin_isnormal(x); }\n");
4946 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4950 assert!(text.contains("%2 = iconst.i64 9223372036854775807"), "{text}");
4951 assert!(text.contains("%3 = and %1, %2"), "{text}");
4952 assert!(text.contains("%4 = iconst.i64 4503599627370496"), "{text}");
4953 assert!(text.contains("%5 = iconst.i64 9218868437227405312"), "{text}");
4954 assert!(text.contains("%6 = icmp uge %3, %4"), "{text}");
4955 assert!(text.contains("%7 = icmp ult %3, %5"), "{text}");
4956 assert!(text.contains("%8 = and %6, %7"), "{text}");
4957
4958 let text = body("int f(long double x) { return __builtin_isnormal(x); }\n");
4962 assert!(text.contains("%4 = iconst.i80 27670116110564327424"), "{text}");
4963 assert!(text.contains("%5 = iconst.i80 604453686435277732577280"), "{text}");
4964
4965 let text = body("int f(double x) { return __builtin_isinf_sign(x); }\n");
4966 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4967 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4968 assert!(text.contains("%7 = sub %5, %6"), "{text}");
4969
4970 let text = body("int f(double x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n");
4971 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4972 assert!(text.contains("fcmp oeq %0, %6"), "{text}");
4973 assert_eq!(text.matches(" = zext.i32 ").count(), 4, "{text}");
4977 assert_eq!(text.matches(" = xor ").count(), 4, "{text}");
4978 assert!(!text.contains("call"), "{text}");
4979
4980 let text = body(concat!(
4983 "double g(void);\n",
4984 "int f(void) { return __builtin_fpclassify(0, 1, 2, 3, 4, g()); }\n",
4985 ));
4986 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4987 }
4988
4989 #[test]
4996 fn the_last_three_classification_builtins_fold_where_their_operand_is_a_constant() {
4997 let text = ir(concat!(
4998 "int a = __builtin_isnormal(1.0);\n",
4999 "int b = __builtin_isnormal(0.0);\n",
5000 "int c = __builtin_isnormal(1.0 / 0.0);\n",
5001 "int d = __builtin_isinf_sign(-1.0 / 0.0);\n",
5002 "int e = __builtin_isinf_sign(1.0);\n",
5003 "int g = __builtin_fpclassify(0, 1, 2, 3, 4, 0.0);\n",
5004 "int h = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0);\n",
5005 "int i = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0 / 0.0);\n",
5006 ));
5007 assert!(text.contains("global @a : i32 = 1,"), "{text}");
5008 assert!(text.contains("global @b : i32 = 0,"), "{text}");
5009 assert!(text.contains("global @c : i32 = 0,"), "{text}");
5010 assert!(text.contains("global @d : i32 = -1,"), "{text}");
5011 assert!(text.contains("global @e : i32 = 0,"), "{text}");
5012 assert!(text.contains("global @g : i32 = 4,"), "{text}");
5013 assert!(text.contains("global @h : i32 = 2,"), "{text}");
5014 assert!(text.contains("global @i : i32 = 1,"), "{text}");
5015 }
5016
5017 #[test]
5023 fn fpclassify_refuses_an_answer_that_is_not_an_integer_constant() {
5024 let mut opts = options();
5025 opts.emit = EmitKind::Ir;
5026 let source = concat!(
5027 "int a(double x, int n) { return __builtin_fpclassify(0, 1, n, 3, 4, x); }\n",
5028 "int b(double x) { return __builtin_fpclassify(0, 1, 2, 3, x); }\n",
5029 "int c(int x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n",
5030 );
5031 let messages = run(&opts, source).messages;
5032 assert_eq!(
5033 messages,
5034 [
5035 "/main.c:1:60: error: non-const integer argument 3 in call to function \
5036 '__builtin_fpclassify' [E0687]",
5037 "/main.c:2:26: error: too few arguments to function '__builtin_fpclassify' \
5038 [E0511]",
5039 "/main.c:3:23: error: non-floating-point argument in call to function \
5040 '__builtin_fpclassify' [E0685]",
5041 ]
5042 );
5043 }
5044
5045 #[test]
5053 fn a_builtin_whose_answer_is_a_constant_is_one_and_not_a_call() {
5054 let text = ir(concat!(
5055 "double a = __builtin_inf();\n",
5056 "float b = __builtin_huge_valf();\n",
5057 "long double c = __builtin_infl();\n",
5058 "double d = __builtin_huge_val();\n",
5059 ));
5060 assert!(text.contains("global @a : f64 = 0x7ff0000000000000,"), "{text}");
5061 assert!(text.contains("global @b : f32 = 0x7f800000,"), "{text}");
5062 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
5063 assert!(text.contains("global @d : f64 = 0x7ff0000000000000,"), "{text}");
5064 assert!(!text.contains("call"), "{text}");
5065 }
5066
5067 #[test]
5076 fn a_nan_is_written_with_the_payload_the_program_asked_for() {
5077 let text = ir(concat!(
5078 "double a = __builtin_nan(\"\");\n",
5079 "double b = __builtin_nan(\"0x1\");\n",
5080 "double c = __builtin_nan(\"010\");\n",
5082 "double d = __builtin_nans(\"\");\n",
5083 "double e = __builtin_nans(\"0x1\");\n",
5084 "float f = __builtin_nanf(\"0x1\");\n",
5085 "float g = __builtin_nansf(\"\");\n",
5086 "long double h = __builtin_nansl(\"\");\n",
5087 ));
5088 assert!(text.contains("global @a : f64 = 0x7ff8000000000000,"), "{text}");
5089 assert!(text.contains("global @b : f64 = 0x7ff8000000000001,"), "{text}");
5090 assert!(text.contains("global @c : f64 = 0x7ff8000000000008,"), "{text}");
5091 assert!(text.contains("global @d : f64 = 0x7ff4000000000000,"), "{text}");
5092 assert!(text.contains("global @e : f64 = 0x7ff0000000000001,"), "{text}");
5093 assert!(text.contains("global @f : f32 = 0x7fc00001,"), "{text}");
5094 assert!(text.contains("global @g : f32 = 0x7fa00000,"), "{text}");
5095 assert!(text.contains("f80 0x7fffa000000000000000"), "{text}");
5096
5097 let text = ir(concat!(
5100 "double f(const char *p) { return __builtin_nan(p); }\n",
5101 "double g(void) { return __builtin_nans(\"1x\"); }\n",
5102 ));
5103 assert_eq!(text.matches("call @nan(").count(), 1, "{text}");
5104 assert_eq!(text.matches("call @nans(").count(), 1, "{text}");
5105 }
5106
5107 #[test]
5115 fn the_length_and_the_order_of_a_string_literal_are_known_here() {
5116 let text = ir(concat!(
5117 "unsigned long a = __builtin_strlen(\"hello\");\n",
5118 "unsigned long b = __builtin_strlen(\"a\\0bc\");\n",
5119 "int c = __builtin_strcmp(\"X\", \"X\\376\") < 0;\n",
5120 "int d = __builtin_strcmp(\"abc\", \"abc\");\n",
5121 "int e = __builtin_strcmp(\"abc\", \"ab\") > 0;\n",
5122 ));
5123 assert!(text.contains("global @a : i64 = 5,"), "{text}");
5124 assert!(text.contains("global @b : i64 = 1,"), "{text}");
5125 assert!(text.contains("global @c : i32 = 1,"), "{text}");
5126 assert!(text.contains("global @d : i32 = 0,"), "{text}");
5127 assert!(text.contains("global @e : i32 = 1,"), "{text}");
5128 assert!(!text.contains("call"), "{text}");
5129
5130 let text = ir("unsigned long f(const char *p) { return __builtin_strlen(p); }\n");
5132 assert!(text.contains("call @strlen("), "{text}");
5133 }
5134
5135 #[test]
5142 fn a_sign_builtin_is_a_mask_over_the_bits_and_not_a_call() {
5143 let text = body("double f(double x) { return __builtin_fabs(x); }\n");
5144 assert!(text.contains("bitcast.i64 %0"), "{text}");
5145 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
5146 assert!(text.contains("and %1, %2"), "{text}");
5147 assert!(text.contains("bitcast.f64 %3"), "{text}");
5148 assert!(!text.contains("call"), "{text}");
5149
5150 let text = body("double f(double x, double y) { return __builtin_copysign(x, y); }\n");
5151 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
5152 assert!(text.contains("%8 = or %4, %7"), "{text}");
5153 assert!(!text.contains("call"), "{text}");
5154
5155 let text = body("long double f(long double x) { return __builtin_fabsl(x); }\n");
5158 assert!(text.contains("bitcast.i80 %0"), "{text}");
5159 assert!(text.contains("bitcast.f80"), "{text}");
5160
5161 let text = body("double f(float x) { return __builtin_fabs(x); }\n");
5164 assert!(text.contains("fpext.f64 %0"), "{text}");
5165 assert!(text.contains("bitcast.i64 %1"), "{text}");
5166 }
5167
5168 #[test]
5177 fn the_plain_math_names_are_the_same_mask_and_not_a_call() {
5178 let text =
5179 body(concat!("double fabs(double x);\n", "double f(double x) { return fabs(x); }\n",));
5180 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
5181 assert!(!text.contains("call"), "{text}");
5182
5183 let text =
5184 body(concat!("float fabsf(float x);\n", "float f(float x) { return fabsf(x); }\n",));
5185 assert!(text.contains("bitcast.i32 %0"), "{text}");
5186 assert!(!text.contains("call"), "{text}");
5187
5188 let text = body(concat!(
5189 "double copysign(double x, double y);\n",
5190 "double f(double x, double y) { return copysign(x, y); }\n",
5191 ));
5192 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
5193 assert!(!text.contains("call"), "{text}");
5194
5195 let text = body(concat!(
5196 "float copysignf(float x, float y);\n",
5197 "float f(float x, float y) { return copysignf(x, y); }\n",
5198 ));
5199 assert!(!text.contains("call"), "{text}");
5200
5201 let text = ir(concat!(
5205 "long double fabsl(long double x);\n",
5206 "long double f(long double x) { return fabsl(x); }\n",
5207 ));
5208 assert!(text.contains("call @fabsl"), "{text}");
5209 }
5210
5211 #[test]
5219 fn a_plain_math_name_the_program_took_is_the_programs_own_function() {
5220 let taken = concat!(
5221 "static double fabs(double b) { return 7; }\n",
5222 "double f(double x) { return fabs(x); }\n",
5223 );
5224 assert!(ir(taken).contains("call @fabs"), "a static definition is the program's own");
5225
5226 let retyped = concat!("int fabs(int b);\n", "int f(int x) { return fabs(x); }\n");
5227 assert!(ir(retyped).contains("call @fabs"), "another type is another function");
5228
5229 let plain = concat!("double fabs(double b);\n", "double f(double x) { return fabs(x); }\n");
5230 let mut opts = options();
5231 opts.emit = EmitKind::Ir;
5232 assert!(!run(&opts, plain).text().contains("call @fabs"), "the library's by default");
5233
5234 opts.builtins = false;
5235 assert!(run(&opts, plain).text().contains("call @fabs"), "-fno-builtin");
5236
5237 opts.builtins = true;
5238 opts.no_builtin = vec!["fabs".to_owned()];
5239 assert!(run(&opts, plain).text().contains("call @fabs"), "-fno-builtin-fabs");
5240 let one = concat!(
5241 "double copysign(double a, double b);\n",
5242 "double f(double x) { return copysign(x, 1.0); }\n",
5243 );
5244 assert!(!run(&opts, one).text().contains("call @copysign"), "one name and not the family");
5245
5246 opts.no_builtin = Vec::new();
5248 opts.builtins = false;
5249 let prefixed = "double f(double x) { return __builtin_fabs(x); }\n";
5250 assert!(!run(&opts, prefixed).text().contains("call @fabs"), "the prefix is not a library");
5251 }
5252
5253 #[test]
5262 fn the_sign_builtins_answer_a_zero_and_a_nan_the_way_the_bits_say() {
5263 let text = ir(concat!(
5264 "double a = __builtin_fabs(-3.5);\n",
5265 "double b = __builtin_copysign(1.0, -0.0);\n",
5266 "double c = __builtin_copysign(0.0, -2.0);\n",
5267 "double d = __builtin_copysign(-__builtin_nan(\"\"), 1.0);\n",
5269 "double e = __builtin_fabs(-__builtin_nan(\"0x1\"));\n",
5270 "float g = __builtin_copysignf(-0.0f, 2.0f);\n",
5271 "long double h = __builtin_copysignl(1.0L, -1.0L);\n",
5272 "long double i = __builtin_fabsl(-__builtin_infl());\n",
5273 ));
5274 assert!(text.contains("global @a : f64 = 0x400c000000000000,"), "{text}");
5275 assert!(text.contains("global @b : f64 = 0xbff0000000000000,"), "{text}");
5276 assert!(text.contains("global @c : f64 = 0x8000000000000000,"), "{text}");
5277 assert!(text.contains("global @d : f64 = 0x7ff8000000000000,"), "{text}");
5278 assert!(text.contains("global @e : f64 = 0x7ff8000000000001,"), "{text}");
5279 assert!(text.contains("global @g : f32 = 0x0,"), "{text}");
5280 assert!(text.contains("f80 0xbfff8000000000000000"), "{text}");
5281 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
5282 }
5283
5284 #[test]
5292 fn the_complex_builtins_are_the_halves_of_the_value_and_not_a_call() {
5293 let text = body("double f(_Complex double z) { return __builtin_creal(z); }\n");
5294 assert!(!text.contains("call"), "{text}");
5295 let text = body("double f(_Complex double z) { return __builtin_cimag(z); }\n");
5296 assert!(!text.contains("call"), "{text}");
5297
5298 let text = body("_Complex double f(_Complex double z) { return __builtin_conj(z); }\n");
5301 assert_eq!(text.matches("fneg").count(), 1, "{text}");
5302 assert!(!text.contains("call"), "{text}");
5303 let negated = body("_Complex double f(_Complex double z) { return -z; }\n");
5304 assert_eq!(negated.matches("fneg").count(), 2, "{negated}");
5305
5306 let written = body("_Complex double f(_Complex double z) { return ~z; }\n");
5309 assert_eq!(written, text, "the name and the operator are the same thing");
5310
5311 let text = body(concat!(
5313 "double creal(_Complex double z);\n",
5314 "double f(_Complex double z) { return creal(z); }\n",
5315 ));
5316 assert!(!text.contains("call"), "{text}");
5317 let text = body(concat!(
5318 "_Complex float conjf(_Complex float z);\n",
5319 "_Complex float f(_Complex float z) { return conjf(z); }\n",
5320 ));
5321 assert_eq!(text.matches("fneg").count(), 1, "{text}");
5322 assert!(!text.contains("call"), "{text}");
5323
5324 let taken = concat!(
5327 "static double creal(_Complex double z) { return 7; }\n",
5328 "double f(_Complex double z) { return creal(z); }\n",
5329 );
5330 assert!(ir(taken).contains("call @creal"), "a static definition is the program's own");
5331 let retyped = concat!("int cimag(int z);\n", "int f(int z) { return cimag(z); }\n");
5332 assert!(ir(retyped).contains("call @cimag"), "another type is another function");
5333 let plain = concat!(
5334 "double cimag(_Complex double z);\n",
5335 "double f(_Complex double z) { return cimag(z); }\n",
5336 );
5337 let mut opts = options();
5338 opts.emit = EmitKind::Ir;
5339 opts.builtins = false;
5340 assert!(run(&opts, plain).text().contains("call @cimag"), "-fno-builtin");
5341 opts.builtins = true;
5342 opts.no_builtin = vec!["cimag".to_owned()];
5343 assert!(run(&opts, plain).text().contains("call @cimag"), "-fno-builtin-cimag");
5344
5345 let text = ir(concat!(
5347 "double a = __builtin_creal(1.5 + 2.5i);\n",
5348 "double b = __builtin_cimag(1.5 + 2.5i);\n",
5349 "_Complex double c = __builtin_conj(1.5 + 2.5i);\n",
5350 ));
5351 assert!(text.contains("global @a : f64 = 0x3ff8000000000000,"), "{text}");
5352 assert!(text.contains("global @b : f64 = 0x4004000000000000,"), "{text}");
5353 assert!(
5354 text.contains("{ f64 0x3ff8000000000000, f64 0xc004000000000000 }"),
5355 "the conjugate of a constant is the constant with the second half negated: {text}"
5356 );
5357 assert!(!text.contains("call"), "{text}");
5358 }
5359
5360 #[test]
5368 fn a_math_library_builtin_of_a_constant_is_the_answer_and_not_a_call() {
5369 let text = ir(concat!(
5370 "double a = __builtin_ceil(1.5);\n",
5371 "double b = __builtin_floor(1.5);\n",
5372 "double c = __builtin_trunc(-1.5);\n",
5373 "double d = __builtin_round(2.5);\n",
5376 "double e = __builtin_ceil(-0.5);\n",
5378 "double f = __builtin_fmax(1.0, 2.0);\n",
5379 "double g = __builtin_fmin(1.0, 2.0);\n",
5380 "float h = __builtin_ceilf(1.25f);\n",
5381 "double ceil(double x);\n",
5384 "double i = ceil(2.25);\n",
5385 ));
5386 assert!(text.contains("global @a : f64 = 0x4000000000000000,"), "{text}");
5387 assert!(text.contains("global @b : f64 = 0x3ff0000000000000,"), "{text}");
5388 assert!(text.contains("global @c : f64 = 0xbff0000000000000,"), "{text}");
5389 assert!(text.contains("global @d : f64 = 0x4008000000000000,"), "{text}");
5390 assert!(text.contains("global @e : f64 = 0x8000000000000000,"), "{text}");
5391 assert!(text.contains("global @f : f64 = 0x4000000000000000,"), "{text}");
5392 assert!(text.contains("global @g : f64 = 0x3ff0000000000000,"), "{text}");
5393 assert!(text.contains("global @h : f32 = 0x40000000,"), "{text}");
5394 assert!(text.contains("global @i : f64 = 0x4008000000000000,"), "{text}");
5395 assert!(!text.contains("call"), "{text}");
5396 }
5397
5398 #[test]
5406 fn a_math_library_builtin_of_anything_else_is_a_call_to_the_library() {
5407 let text = ir(concat!(
5408 "double f(double x) { return __builtin_ceil(x); }\n",
5409 "float g(float x) { return __builtin_floorf(x); }\n",
5410 "double h(double x, double y) { return __builtin_fmax(x, y); }\n",
5411 ));
5412 assert!(text.contains("call @ceil("), "{text}");
5413 assert!(text.contains("call @floorf("), "{text}");
5414 assert!(text.contains("call @fmax("), "{text}");
5415
5416 let text = ir(concat!(
5420 "double f(void) { return __builtin_rint(2.5); }\n",
5421 "double g(void) { return __builtin_nearbyint(2.5); }\n",
5422 ));
5423 assert!(text.contains("call @rint("), "{text}");
5424 assert!(text.contains("call @nearbyint("), "{text}");
5425
5426 let text = ir("double f(void) { return __builtin_fmin(__builtin_nan(\"\"), 1.0); }\n");
5429 assert!(text.contains("call @fmin("), "{text}");
5430
5431 let plain = concat!("double ceil(double x);\n", "double f(void) { return ceil(2.25); }\n");
5434 let mut opts = options();
5435 opts.emit = EmitKind::Ir;
5436 opts.no_builtin = vec!["ceil".to_owned()];
5437 assert!(run(&opts, plain).text().contains("call @ceil("), "-fno-builtin-ceil");
5438 }
5439
5440 #[test]
5447 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
5448 let text = ir(concat!(
5449 "constexpr int side = 4;\n",
5450 "constexpr int wider = side + 1;\n",
5451 "constexpr double half = 1.5;\n",
5452 "struct point { int x; int y; };\n",
5453 "constexpr struct point origin = { 5, 6 };\n",
5454 "int square[side * side];\n",
5455 "int rectangle[wider];\n",
5456 "int rounded[(int)half * 2];\n",
5457 "int across[origin.y];\n",
5458 "enum named { four = side };\n",
5459 "int e = four;\n",
5460 ));
5461 assert!(text.contains("global @square : bytes 64 ="), "{text}");
5462 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
5463 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
5464 assert!(text.contains("global @across : bytes 24 ="), "{text}");
5465 assert!(text.contains("global @e : i32 = 4,"), "{text}");
5466
5467 let mut opts = options();
5470 opts.emit = EmitKind::Ir;
5471 let konst = "const int n = 1;\nint a[n];\n";
5472 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
5473 assert_eq!(run(&opts, konst).messages, [message]);
5474
5475 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
5477 assert_eq!(run(&opts, subscript).messages, [message]);
5478
5479 let address = "constexpr int c = 3;\nint *p = &c;\n";
5481 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
5482 pointer target type [E0514]";
5483 assert_eq!(run(&opts, address).messages, [warning]);
5484 }
5485
5486 #[test]
5495 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
5496 let mut opts = options();
5499 opts.std = Std::C17;
5500 let source = concat!(
5501 "int add(a, b)\n",
5502 "int a;\n",
5503 "int b;\n",
5504 "{ return a + b; }\n",
5505 "int promoted(c)\n",
5506 "char c;\n",
5507 "{ return c; }\n",
5508 "int narrow(char);\n",
5509 "int narrow(c)\n",
5510 "char c;\n",
5511 "{ return c; }\n",
5512 "int first(a)\n",
5513 "int a[4];\n",
5514 "{ return a[0]; }\n",
5515 );
5516 let result = run(&opts, source);
5517 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
5518 let text = result.text();
5519 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
5520 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
5521 assert!(text.contains("c : char object automatic defined"), "{text}");
5523 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
5524 assert!(text.contains("first : int(int *) function external defined"), "{text}");
5526 }
5527
5528 #[test]
5535 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
5536 let mut opts = options();
5537 opts.std = Std::C17;
5538 for (source, message) in [
5539 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
5540 (
5541 "int f(a)\nint a;\nint b;\n{ return a; }\n",
5542 "3:5: error: declaration for parameter 'b' but no such parameter",
5543 ),
5544 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
5545 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
5546 (
5547 "int f(a)\nstatic int a;\n{ return a; }\n",
5548 "2:12: error: storage class specified for parameter 'a'",
5549 ),
5550 (
5551 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
5552 "2:7: error: argument 'a' doesn't match prototype",
5553 ),
5554 ] {
5555 let result = run(&opts, source);
5556 assert!(result.failed(), "expected this to fail:\n{source}");
5557 assert!(result.messages[0].contains(message), "{:?}", result.messages);
5558 }
5559
5560 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
5563 let mut older = options();
5564 older.std = Std::C89;
5565 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
5566 let result = run(&opts, implicit);
5567 assert!(
5568 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
5569 "{:?}",
5570 result.messages
5571 );
5572
5573 let mut newer = options();
5577 newer.std = Std::C23;
5578 let plain = "int f(a)\nint a;\n{ return a; }\n";
5579 let result = run(&newer, plain);
5580 assert!(!result.failed(), "{:?}", result.messages);
5581 assert_eq!(
5582 result.messages,
5583 ["/main.c:1:5: warning: old-style function definition [E0412]"]
5584 );
5585 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
5586 }
5587
5588 #[test]
5595 fn the_obsolete_designators_are_taken_and_are_pedantic_warnings() {
5596 let array = "int a[8] = { [3] 7 };\n";
5597 let member = "struct s { int x; } v = { x: 7 };\n";
5598 for source in [array, member] {
5599 let result = run(&options(), source);
5600 assert!(!result.failed(), "{:?}", result.messages);
5601 assert!(result.messages.is_empty(), "nothing to say: {:?}", result.messages);
5602 }
5603
5604 let mut asked = options();
5605 asked.pedantic = true;
5606 assert_eq!(
5607 run(&asked, array).messages,
5608 ["/main.c:1:18: warning: obsolete designator, write `[i] =` instead [E0415]"]
5609 );
5610 assert_eq!(
5611 run(&asked, member).messages,
5612 ["/main.c:1:27: warning: obsolete designator, write `.field =` instead [E0413]"]
5613 );
5614 }
5615
5616 #[test]
5623 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
5624 let text = ir(concat!(
5625 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
5626 "struct brim { char buf[9223372036854775807L]; };\n",
5627 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
5628 "unsigned long h = sizeof(struct huge_struct);\n",
5629 "unsigned long b = sizeof(struct brim);\n",
5630 "unsigned long y = sizeof(struct bitty);\n",
5631 ));
5632 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
5633 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
5634 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
5635
5636 let mut opts = options();
5637 opts.emit = EmitKind::Ir;
5638 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
5639 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
5640 assert_eq!(run(&opts, over).messages, [message]);
5641 let array = "struct wide { short buf[1L << 62]; };\n";
5642 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
5643 maximum object size '9223372036854775807' [E0537]";
5644 assert_eq!(run(&opts, array).messages[0], message);
5645 }
5646
5647 fn compile_bytes(source: &[u8]) -> Compiled {
5652 let mut opts = options();
5653 opts.emit = EmitKind::Ir;
5654 let mut fs = MemoryFileSystem::new();
5655 fs.insert("/main.c", source.to_vec());
5656 compile(&opts, "/main.c", &fs)
5657 }
5658
5659 #[test]
5666 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
5667 let mut source = b"char s[] = \"a".to_vec();
5668 source.push(0xff);
5669 source.extend_from_slice(b"b\";\nchar c = '");
5670 source.push(0xff);
5671 source.extend_from_slice(b"';\n");
5672 let result = compile_bytes(&source);
5673 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
5674 assert!(result.text().contains(r#"bytes "a\ffb\00""#), "{}", result.text());
5675 assert!(result.text().contains("global @c : i8 = -1,"), "{}", result.text());
5677
5678 let mut stray = b"int a".to_vec();
5679 stray.push(0xff);
5680 stray.extend_from_slice(b" = 1;\n");
5681 let result = compile_bytes(&stray);
5682 assert!(
5683 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
5684 "{:?}",
5685 result.messages
5686 );
5687 }
5688
5689 #[test]
5690 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
5691 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
5692 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
5693 let expected = "\
5694func @add(i32, i32) -> i32, linkage(external) {
5695block0(%0: i32, %1: i32):
5696 %2 = add.nsw %0, %1
5697 return %2
5698}
5699";
5700 assert!(text.contains(expected), "{text}");
5701 }
5702
5703 #[test]
5704 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
5705 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
5706 assert!(!text.contains("alloca"), "{text}");
5707 assert!(!text.contains("load"), "{text}");
5708 assert!(!text.contains("store"), "{text}");
5709 }
5710
5711 #[test]
5712 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
5713 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
5714 let expected = "\
5715block0:
5716 %0 = alloca, size 4, align 4
5717 %1 = iconst.i32 1
5718 store %1 -> %0, align 4, tbaa !1
5719 %2 = call @g(%0) : (ptr) -> i32
5720 return %2
5721";
5722 assert_eq!(text, expected);
5723 }
5724
5725 #[test]
5726 fn a_loop_carries_what_it_changes_as_block_parameters() {
5727 let text = body(
5730 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
5731 return total;\n}\n",
5732 );
5733 assert!(!text.contains("alloca"), "{text}");
5734 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
5735 assert!(text.contains("jump block1("), "{text}");
5736 }
5737
5738 #[test]
5739 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
5740 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
5741 assert!(text.contains("icmp slt %0, %1"), "{text}");
5742 assert!(!text.contains("zext"), "{text}");
5743 }
5744
5745 #[test]
5746 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
5747 let text = body("int f(int a, int b) { return a && b; }\n");
5748 let expected = "\
5749block0(%0: i32, %1: i32):
5750 %2 = iconst.i32 0
5751 %3 = icmp ne %0, %2
5752 %4 = iconst.i1 0
5753 br_if %3, block1, block2(%4)
5754
5755block1:
5756 %5 = iconst.i32 0
5757 %6 = icmp ne %1, %5
5758 jump block2(%6)
5759
5760block2(%7: i1):
5761 %8 = zext.i32 %7
5762 return %8
5763";
5764 assert_eq!(text, expected);
5765 }
5766
5767 #[test]
5768 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
5769 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
5770 assert!(!text.contains("block3"), "{text}");
5773 assert!(!text.contains("iconst.i32 3"), "{text}");
5774 }
5775
5776 #[test]
5777 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
5778 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
5779 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
5780 assert!(body("int f(void) { }\n").contains("unreachable"));
5781 }
5782
5783 #[test]
5784 fn a_structure_is_copied_rather_than_held_in_a_value() {
5785 let text = body(
5786 "struct point { int x, y; };\n\
5787 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
5788 );
5789 assert!(text.contains("memcpy"), "{text}");
5790 }
5791
5792 #[test]
5793 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
5794 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
5795 assert!(text.contains("memset"), "{text}");
5796 }
5797
5798 #[test]
5799 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
5800 let text = body(
5801 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
5802 default: r = 4; } return r; }\n",
5803 );
5804 let expected = "\
5805block0(%0: i32):
5806 %1 = iconst.i32 0
5807 switch %0, block1, [1 => block2, 2 => block3(%1)]
5808
5809block1:
5810 %2 = iconst.i32 4
5811 jump block4(%2)
5812
5813block2:
5814 %3 = iconst.i32 1
5815 jump block3(%3)
5816
5817block3(%4: i32):
5818 %5 = iconst.i32 2
5819 %6 = add.nsw %4, %5
5820 jump block4(%6)
5821
5822block4(%7: i32):
5823 return %7
5824";
5825 assert_eq!(text, expected);
5826 }
5827
5828 #[test]
5829 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
5830 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
5833 assert!(text.contains("%2 = sub %0, %1"), "{text}");
5834 assert!(text.contains("icmp ule"), "{text}");
5835 assert!(!text.contains("switch"), "{text}");
5836 }
5837
5838 #[test]
5839 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
5840 let text = body(
5841 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
5842 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
5843 );
5844 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
5847 assert!(text.contains("block5:\n jump block7("), "{text}");
5848 assert!(text.contains("block6:\n jump block8("), "{text}");
5849 }
5850
5851 #[test]
5852 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
5853 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
5854 }
5855
5856 #[test]
5857 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
5858 let text = body(
5863 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
5864 return n; }\n",
5865 );
5866 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
5869 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
5870 assert!(text.contains("block4:\n jump block3("), "{text}");
5871 }
5872
5873 #[test]
5874 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
5875 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
5878 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
5879 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
5880 assert!(text.contains("br_if %6, block2, block3"), "{text}");
5881 }
5882
5883 #[test]
5884 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
5885 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
5886 assert!(!text.contains("alloca"), "{text}");
5890 assert!(text.contains("block2(%4: i32):\n return %4"), "{text}");
5891 assert_eq!(text.matches("jump block2(").count(), 2, "{text}");
5892 }
5893
5894 #[test]
5895 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
5896 let text =
5897 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
5898 assert!(!text.contains("alloca"), "{text}");
5899 assert!(text.contains("block1(%2: i32):"), "{text}");
5900 assert!(text.contains("jump block1(%5)"), "{text}");
5901 }
5902
5903 #[test]
5904 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
5905 assert_eq!(
5908 body("int f(int x) { return x; spare: return 0; }\n"),
5909 "block0(%0: i32):\n return %0\n"
5910 );
5911 }
5912
5913 #[test]
5914 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
5915 let text = body(
5916 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
5917 );
5918 assert_eq!(
5921 text,
5922 "\
5923block0(%0: ptr):
5924 %1 = load.i8 %0, align 1
5925 %2 = iconst.i8 3
5926 %3 = ashr %1, %2
5927 %4 = sext.i32 %3
5928 return %4
5929"
5930 );
5931 }
5932
5933 #[test]
5934 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
5935 let text =
5939 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
5940 assert_eq!(
5941 text,
5942 "\
5943block0(%0: ptr, %1: i32):
5944 %2 = iconst.i32 16777215
5945 %3 = and %1, %2
5946 %4 = trunc.i16 %3
5947 store %4 -> %0, align 2
5948 %5 = iconst.i32 16
5949 %6 = lshr %3, %5
5950 %7 = trunc.i8 %6
5951 %8 = iconst.i64 2
5952 %9 = ptr_add %0, %8
5953 store %7 -> %9, align 1
5954 return
5955"
5956 );
5957 }
5958
5959 #[test]
5960 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
5961 let text =
5962 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
5963 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
5966 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
5967 }
5968
5969 #[test]
5970 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
5971 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
5974 assert_eq!(text.matches("ashr").count(), 0, "{text}");
5975 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
5976 }
5977
5978 #[test]
5979 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
5980 let text = body(
5984 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
5985 );
5986 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
5987 }
5988
5989 #[test]
5990 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
5991 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
5994 assert!(
5995 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
5996 "{text}"
5997 );
5998 }
5999
6000 #[test]
6001 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
6002 let text = ir(concat!(
6007 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
6008 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
6009 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
6010 "char s[2] = \"hi\";\n",
6011 ));
6012 assert!(
6013 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
6014 "{text}"
6015 );
6016 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
6017 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
6018 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
6021 }
6022
6023 #[test]
6024 fn a_definition_takes_a_parameter_it_left_unnamed() {
6025 let text = ir("int f(int a, int) { return a; }\n");
6029 assert!(text.contains("func @f(i32, i32) -> i32"), "{text}");
6030 assert!(text.contains("block0(%0: i32, %1: i32):"), "{text}");
6031
6032 let text = ir("int g(int, int n) { return n; }\n");
6035 assert!(text.contains("block0(%0: i32, %1: i32):\n return %1\n"), "{text}");
6036 }
6037
6038 #[test]
6039 fn an_assignment_of_a_structure_is_the_object_it_wrote() {
6040 let text = body(concat!(
6045 "struct s { int f; int g; };\n",
6046 "void h(struct s *a, struct s *c, struct s *d, struct s *e)\n",
6047 "{ *d = *e = a[0] = *c; }\n",
6048 ));
6049 assert_eq!(text.matches("memcpy").count(), 3, "{text}");
6050 assert!(text.contains("memcpy %8, %1, size 8, align 4\n"), "{text}");
6051 assert!(text.contains("memcpy %3, %8, size 8, align 4\n"), "{text}");
6052 assert!(text.contains("memcpy %2, %3, size 8, align 4\n"), "{text}");
6053 }
6054
6055 #[test]
6056 fn a_string_literal_stops_at_the_end_of_the_array_it_is_filling() {
6057 let mut opts = options();
6062 opts.emit = EmitKind::Ir;
6063 let result = run(
6064 &opts,
6065 concat!(
6066 "const char a[2][3] = { \"1234\", \"xyz\" };\n",
6067 "static const char b[3][5] = { \"12345\", \"678\", \"9\" };\n",
6068 "union u { struct { char x[4]; char y[4]; }; struct { char z[8]; }; };\n",
6069 "const union u c = { { \"1234\", \"567\" } };\n",
6070 ),
6071 );
6072 let text = result.text();
6073 assert_eq!(
6074 result.messages,
6075 ["/main.c:1:24: warning: initializer-string for array of 'const char' is too long \
6076 (5 chars into 3 available) [E0637]"]
6077 );
6078 assert!(text.contains("global @a : bytes 6 = { bytes \"123\", bytes \"xyz\" }"), "{text}");
6079 assert!(
6080 text.contains(
6081 "global @b : bytes 15 = { bytes \"12345\", bytes \"678\\00\", zero 1, \
6082 bytes \"9\\00\", zero 3 }"
6083 ),
6084 "{text}"
6085 );
6086 assert!(
6089 text.contains("global @c : bytes 8 = { bytes \"1234\", bytes \"567\\00\" }"),
6090 "{text}"
6091 );
6092 }
6093
6094 #[test]
6095 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
6096 let text = body(concat!(
6100 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
6101 "void g(struct v *);\n",
6102 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
6103 ));
6104 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
6105 }
6106
6107 #[test]
6108 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
6109 let text = ir(concat!(
6114 "struct s { int x; };\n",
6115 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
6116 "int n = (int){ 7 };\n",
6117 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
6118 ));
6119 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
6120 assert!(text.contains("global @n : i32 = 7,"), "{text}");
6121 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
6124 }
6125
6126 #[test]
6127 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
6128 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
6132 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
6133 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
6134 }
6135
6136 #[test]
6137 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
6138 let text = ir("unsigned char foo[1][0];\n");
6142 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
6143 }
6144
6145 #[test]
6146 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
6147 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
6150 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
6151 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
6152 }
6153
6154 #[test]
6155 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
6156 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
6160 assert!(
6161 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
6162 "{text}"
6163 );
6164 }
6165
6166 #[test]
6167 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
6168 let text = body(
6173 "\
6174struct s { int a, b; };
6175struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
6176",
6177 );
6178 assert!(text.contains("block3(%7: ptr)"), "{text}");
6180 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
6181 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
6182 }
6183
6184 #[test]
6192 fn the_left_side_of_a_conditional_with_no_middle_is_evaluated_once() {
6193 let text = body("int f(int i) { return ++i ?: 10; }\n");
6194 assert!(text.contains("jump block3(%2)"), "the arm is the value that was tested: {text}");
6195 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
6196
6197 let text = body("long f(int i) { return ++i ?: 10L; }\n");
6200 assert!(text.contains("%5 = sext.i64 %2"), "the arm widens what was tested: {text}");
6201 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
6202
6203 let text = body("int g(void);\nint f(void) { return g() ?: 10; }\n");
6205 assert_eq!(text.matches("call @g").count(), 1, "called once: {text}");
6206
6207 let text = body("int f(int i) { return ++i ? ++i : 10; }\n");
6210 assert_eq!(text.matches("add.nsw").count(), 2, "incremented twice: {text}");
6211 }
6212
6213 #[test]
6214 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
6215 let text = ir("\
6219struct pair { int a, b; };
6220struct pair make(int a, int b);
6221struct pair twice(struct pair p) { return make(p.a, p.b); }
6222");
6223 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
6224 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
6225 }
6226
6227 #[test]
6228 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
6229 let text = ir("\
6233struct big { double v[8]; };
6234struct big grow(struct big b);
6235struct big twice(struct big b) { return grow(grow(b)); }
6236");
6237 assert!(
6238 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
6239 "{text}"
6240 );
6241 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
6242 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
6245 }
6246
6247 #[test]
6248 fn a_structure_passed_to_a_variadic_function_says_so_at_the_call() {
6249 let text = ir("\
6254struct big { double v[8]; };
6255struct pair { int a, b; };
6256int p(const char *, ...);
6257int f(struct big b, struct pair q) { return p(\"\", 1, b, q); }
6258");
6259 assert!(
6260 text.contains("call @p(%4, %5, %2 byval(64, align 8), %6) : (ptr, ...) -> i32"),
6261 "{text}"
6262 );
6263 }
6264
6265 #[test]
6266 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
6267 let body = body(
6270 "\
6271struct pair { int a, b; };
6272struct pair make(int a, int b);
6273int second(void) { return make(1, 2).b; }
6274",
6275 );
6276 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
6277 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
6278 }
6279
6280 #[test]
6281 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
6282 let source = "\
6286struct hfa { float x, y, z; };
6287int take(struct hfa h);
6288int give(struct hfa h) { return take(h); }
6289";
6290 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
6291 let mut opts = options();
6292 opts.emit = EmitKind::Ir;
6293 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
6294 let result = run(&opts, source);
6295 assert_eq!(result.messages, Vec::<String>::new());
6296 assert!(result.text().contains("func @take(f32, f32, f32) -> i32"), "{}", result.text());
6297 }
6298
6299 #[test]
6300 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
6301 let source = "\
6304int use(int *);
6305void f(int n) {
6306 {
6307 int a[n];
6308 use(a);
6309 }
6310 use(0);
6311}
6312";
6313 let body = body(source);
6314 assert!(body.contains("mul.nsw"), "{body}");
6315 assert!(body.contains("stacksave"), "{body}");
6316 assert!(body.contains("alloca %"), "{body}");
6317 assert!(body.contains("stackrestore"), "{body}");
6318 }
6319
6320 #[test]
6321 fn a_goto_out_of_the_scope_of_one_gives_its_stack_back_on_the_way() {
6322 let source = "\
6327int use(int *);
6328int f(int n) {
6329 {
6330 int a[n];
6331 if (use(a)) goto out;
6332 use(0);
6333 }
6334out:
6335 return 0;
6336}
6337";
6338 let body = body(source);
6339 assert_eq!(body.matches("stackrestore").count(), 2, "{body}");
6341 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
6342 assert!(after.starts_with(" %4\n jump block"), "{body}");
6343 }
6344
6345 #[test]
6346 fn a_goto_to_a_label_the_array_is_still_alive_at_leaves_the_stack_alone() {
6347 let source = "\
6351int use(int *);
6352int f(int n) {
6353 int a[n];
6354again:
6355 if (use(a)) goto again;
6356 return 0;
6357}
6358";
6359 let body = body(source);
6360 assert!(body.contains("stacksave"), "{body}");
6361 assert!(!body.contains("stackrestore"), "{body}");
6362 }
6363
6364 #[test]
6365 fn a_goto_back_to_a_label_in_front_of_one_gives_it_back_every_time_round() {
6366 let source = "\
6371int use(int *);
6372int f(int n) {
6373again:
6374 {
6375 int a[n];
6376 if (use(a)) goto again;
6377 }
6378 return 0;
6379}
6380";
6381 let body = body(source);
6382 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
6383 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
6384 assert!(after.starts_with(" %4\n jump block1\n"), "{body}");
6385 }
6386
6387 #[test]
6388 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
6389 let source = "\
6395int f(void);
6396void t(void) {
6397 int count = 10;
6398 for (; count--;) {
6399 int b[f()];
6400 int i;
6401 for (i = 0; i < f(); i++) {
6402 b[i] = count;
6403 }
6404 }
6405}
6406";
6407 let body = body(source);
6408 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
6412 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
6413 let next = after.split("\n\n").next().expect("the block the restore is in");
6416 assert!(next.contains("jump block1("), "{body}");
6417 }
6418
6419 #[test]
6420 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
6421 let source = "\
6424unsigned long f(int n) {
6425 int a[n];
6426 n = 0;
6427 return sizeof a;
6428}
6429";
6430 let body = body(source);
6431 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
6433 }
6434
6435 #[test]
6436 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
6437 let source = "\
6440int use(int);
6441int f(int x) {
6442 return ({
6443 int t = use(x);
6444 t * t;
6445 });
6446}
6447";
6448 let expected = "\
6449block0(%0: i32):
6450 %1 = call @use(%0) : (i32) -> i32
6451 %2 = mul.nsw %1, %1
6452 return %2
6453";
6454 assert_eq!(body(source), expected);
6455 }
6456
6457 #[test]
6458 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
6459 let source = "int f(int x) { return ({ return x; 0; }); }\n";
6463 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
6464 }
6465
6466 #[test]
6467 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
6468 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
6472 let expected = "\
6473block0(%0: ptr):
6474 %1 = va_arg.f64 %0
6475 %2 = va_arg.f64 %0
6476 %3 = fadd %1, %2
6477 return %3
6478";
6479 assert_eq!(body(source), expected);
6480 }
6481
6482 #[test]
6483 fn one_that_reads_a_structure_answers_where_the_object_is() {
6484 let source = "\
6498struct s { int a; long b; };
6499long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.b; }
6500";
6501 let expected = "\
6502block0(%0: ptr):
6503 %1 = alloca, size 16, align 16
6504 %2 = va_object %0, size 16, align 8, in(int 8 at 0, int 8 at 8)
6505 memcpy %1, %2, size 16, align 8
6506 %3 = iconst.i64 8
6507 %4 = ptr_add %1, %3
6508 %5 = load.i64 %4, align 8, tbaa !1
6509 return %5
6510";
6511 assert_eq!(body(source), expected);
6512 }
6513
6514 #[test]
6518 fn the_classification_says_which_registers_the_object_arrived_in() {
6519 let source = "\
6520struct s { double a; double b; };
6521double f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a; }
6522";
6523 assert!(
6524 body(source)
6525 .contains("va_object %0, size 16, align 8, in(float f64 at 0, float f64 at 8)"),
6526 "{}",
6527 body(source)
6528 );
6529
6530 let big = "\
6531struct s { long a[4]; };
6532long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a[0]; }
6533";
6534 assert!(body(big).contains("va_object %0, size 32, align 8\n"), "{}", body(big));
6535 }
6536
6537 #[test]
6538 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
6539 let source = "\
6543int f(int c) {
6544 void *p = c ? &&one : &&two;
6545 goto *p;
6546one:
6547 return 1;
6548two:
6549 return 2;
6550}
6551";
6552 let expected = "\
6553block0(%0: i32):
6554 %1 = iconst.i32 0
6555 %2 = icmp ne %0, %1
6556 br_if %2, block1, block2
6557
6558block1:
6559 %3 = block_addr block3
6560 jump block4(%3)
6561
6562block2:
6563 %4 = block_addr block5
6564 jump block4(%4)
6565
6566block3:
6567 %5 = iconst.i32 1
6568 return %5
6569
6570block4(%6: ptr):
6571 indirect_br %6, block3, block5
6572
6573block5:
6574 %7 = iconst.i32 2
6575 return %7
6576";
6577 assert_eq!(body(source), expected);
6578 }
6579
6580 #[test]
6581 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
6582 let source = "void **next(void);
6585void f(void) { goto *next(); }
6586";
6587 let expected = "\
6588block0:
6589 %0 = call @next() : () -> ptr
6590 unreachable
6591";
6592 assert_eq!(body(source), expected);
6593 }
6594
6595 #[test]
6596 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
6597 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
6600 let expected = "\
6601block0:
6602 inline_asm.volatile \"mfence\", \"\", \"memory\"()
6603 return
6604";
6605 assert_eq!(body(source), expected);
6606 }
6607
6608 #[test]
6609 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
6610 let source = "\
6613int f(int x, int y) {
6614 int r;
6615 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
6616 return r + y;
6617}
6618";
6619 let expected = "\
6620block0(%0: i32, %1: i32):
6621 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
6622 %4 = add.nsw %2, %3
6623 return %4
6624";
6625 assert_eq!(body(source), expected);
6626 }
6627
6628 #[test]
6629 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
6630 let source = "\
6635struct pair { int a, b; };
6636int f(int x) {
6637 int slot = x;
6638 struct pair p = { x, x };
6639 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
6640 return slot + p.a;
6641}
6642";
6643 let text = body(source);
6644 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
6645 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
6646 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
6647 }
6648
6649 #[test]
6650 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
6651 let source = "\
6656int f(int x) {
6657 int r = 7;
6658 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
6659 return r;
6660away:
6661 return r;
6662}
6663";
6664 let expected = "\
6665block0(%0: i32):
6666 %1 = iconst.i32 7
6667 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
6668
6669block1:
6670 return %2
6671
6672block2:
6673 return %1
6674";
6675 assert_eq!(body(source), expected);
6676 }
6677
6678 #[test]
6679 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
6680 let mut opts = options();
6684 opts.emit = EmitKind::Ir;
6685 for (source, expected) in [
6686 (
6687 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
6688 "output operand constraint lacks '='",
6689 ),
6690 (
6691 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
6692 "lvalue required in 'asm' statement",
6693 ),
6694 (
6695 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
6696 "read-only variable 'g' used as 'asm' output",
6697 ),
6698 (
6699 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
6700 "input operand constraint contains '='",
6701 ),
6702 (
6703 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
6704 "memory input 0 is not directly addressable",
6705 ),
6706 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
6707 (
6708 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
6709 "duplicate asm operand name 'a'",
6710 ),
6711 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
6712 ] {
6713 let result = run(&opts, source);
6714 assert!(result.failed(), "expected this to be reported:\n{source}");
6715 assert!(
6716 result.messages.iter().any(|m| m.contains(expected)),
6717 "{expected}\n{:?}",
6718 result.messages
6719 );
6720 }
6721 }
6722
6723 #[test]
6728 fn an_asm_at_file_scope_that_is_directives_becomes_the_objects_it_defines() {
6729 let text = ir(concat!(
6730 "__asm__(\n",
6731 " \".section .rodata\\n\"\n",
6732 " \".globl first\\n\"\n",
6733 " \".balign 8\\n\"\n",
6734 " \"first:\\n\"\n",
6735 " \".long 1\\n\"\n",
6736 " \".long 2\\n\"\n",
6737 " \".globl last\\n\"\n",
6738 " \"last:\\n\"\n",
6739 " \".quad last - first\\n\");\n",
6740 "extern const int first[];\n",
6741 "extern const long last;\n",
6742 ));
6743 assert!(text.contains("global @first : bytes 8 = { i32 1, i32 2 }, align 8"), "{text}");
6744 assert!(text.contains("global @last : i64 = 8"), "{text}");
6745 }
6746
6747 #[test]
6751 fn a_name_an_asm_at_file_scope_defined_is_not_undone_by_a_declaration_of_it() {
6752 let text = ir(concat!(
6753 "__asm__(\".data\\n.globl counter\\ncounter:\\n.long 7\\n\");\n",
6754 "extern int counter;\n",
6755 "int read(void) { return counter; }\n",
6756 ));
6757 assert!(text.contains("global @counter : i32 = 7"), "{text}");
6758 }
6759
6760 #[test]
6763 fn an_incbin_at_file_scope_is_the_bytes_of_the_file_it_names() {
6764 let mut opts = options();
6765 opts.emit = EmitKind::Ir;
6766 let mut fs = MemoryFileSystem::new();
6767 fs.insert(
6768 "/main.c",
6769 b"__asm__(\".data\\n.globl blob\\nblob:\\n.incbin \\\"seed\\\"\\n\");\n".to_vec(),
6770 );
6771 fs.insert("seed", b"hi".to_vec());
6772 let result = compile(&opts, "/main.c", &fs);
6773 assert_eq!(result.messages, Vec::<String>::new());
6774 let text = result.text();
6775 assert!(text.contains("global @blob : bytes 2 = { bytes \"hi\" }"), "{text}");
6776 }
6777
6778 #[test]
6781 fn an_incbin_naming_a_file_that_is_not_there_says_which_file() {
6782 let messages = errors("__asm__(\".data\\nb:\\n.incbin \\\"nowhere\\\"\\n\");\n");
6783 assert!(
6784 messages
6785 .iter()
6786 .any(|m| m.contains("cannot open 'nowhere' for reading") && m.contains("E0702")),
6787 "{messages:?}"
6788 );
6789 }
6790
6791 #[test]
6794 fn an_instruction_in_an_asm_at_file_scope_is_refused_rather_than_ignored() {
6795 for source in [
6796 "__asm__(\".text\\n.globl f\\nf:\\n ret\\n\");\n",
6797 "__asm__(\".data\\n.set alias, 4\\n\");\n",
6798 ] {
6799 let messages = errors(source);
6800 assert!(
6801 messages
6802 .iter()
6803 .any(|m| m.contains("not supported yet")
6804 && m.contains("in an `asm` at file scope")),
6805 "{source}\n{messages:?}"
6806 );
6807 }
6808 }
6809
6810 #[test]
6811 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
6812 let mut opts = options();
6813 opts.emit = EmitKind::Ir;
6814 for source in [
6815 "int f(int n) { void *p = &&out; if (n) goto *p; { int a[n]; out: return 1; } }\n",
6816 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
6817 ] {
6818 let result = run(&opts, source);
6819 assert!(result.failed(), "expected this to be reported:\n{source}");
6820 assert!(
6821 result.messages.iter().any(|m| m.contains("not supported yet")),
6822 "{:?}",
6823 result.messages
6824 );
6825 }
6826 }
6827
6828 fn round_trip(source: &str) -> (String, String) {
6830 let printed = ir(source);
6831 let mut opts = options();
6832 opts.emit = EmitKind::Ir;
6833 let mut fs = MemoryFileSystem::new();
6834 fs.insert("/main.ir", printed.clone().into_bytes());
6835 let result = compile_ir(&opts, "/main.ir", &fs);
6836 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
6837 (printed, result.text().to_owned())
6838 }
6839
6840 #[test]
6841 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
6842 let (printed, again) = round_trip(
6846 "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",
6847 );
6848 assert_eq!(printed, again);
6849 }
6850
6851 #[test]
6852 fn ir_that_is_not_ir_says_which_line_stopped_it() {
6853 let mut opts = options();
6854 opts.emit = EmitKind::Ir;
6855 let mut fs = MemoryFileSystem::new();
6856 let text = "\
6857; ModuleID = 'a.c'
6858; format 0
6859target triple = \"x86_64-unknown-linux-gnu\"
6860target datalayout = \"e-p:64:64-i64:64-S128\"
6861
6862func @f(), linkage(external) {
6863block0:
6864 frobnicate
6865}
6866";
6867 fs.insert("/main.ir", text.as_bytes().to_vec());
6868 let result = compile_ir(&opts, "/main.ir", &fs);
6869 assert!(result.failed());
6870 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
6871 }
6872
6873 #[test]
6874 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
6875 let mut opts = options();
6878 opts.emit = EmitKind::Ir;
6879 let mut fs = MemoryFileSystem::new();
6880 let text = "\
6881; ModuleID = 'a.c'
6882; format 0
6883target triple = \"x86_64-unknown-linux-gnu\"
6884target datalayout = \"e-p:64:64-i64:64-S128\"
6885
6886func @f(), linkage(external) {
6887block0:
6888 %0 = iconst.i32 1
6889 return %0
6890}
6891";
6892 fs.insert("/main.ir", text.as_bytes().to_vec());
6893 let result = compile_ir(&opts, "/main.ir", &fs);
6894 assert!(result.failed());
6895 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
6896 }
6897
6898 #[test]
6899 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
6900 let mut fs = MemoryFileSystem::new();
6902 fs.insert("/main.ir", Vec::new());
6903 let result = compile_ir(&options(), "/main.ir", &fs);
6904 assert!(result.failed());
6905 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
6906 }
6907
6908 #[test]
6909 fn the_printed_ir_reads_back_as_the_same_module() {
6910 let text = ir("\
6913struct point { int x, y; };
6914static const char greeting[] = \"hi\";
6915int table[4] = { 1, 2, 3 };
6916int puts(const char *);
6917double half(double x) { return x / 2.0; }
6918int f(int n) {
6919 int total = 0;
6920 for (int i = 0; i < n; i++) {
6921 if (i == 3) continue;
6922 total += table[i];
6923 }
6924 switch (n) {
6925 case 0: total = 1;
6926 case 1: total++; break;
6927 default: total = -total;
6928 }
6929 struct point p = { total, 1 };
6930 int *q = &p.y;
6931 puts(greeting);
6932 return p.x + *q;
6933}
6934int dispatch(int c) {
6935 void *p = c ? &&one : &&two;
6936 goto *p;
6937one:
6938 return 1;
6939two:
6940 return 2;
6941}
6942int assembly(int x, int *p) {
6943 int r;
6944 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
6945 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
6946 return r;
6947away:
6948 return 0;
6949}
6950");
6951 let mut names = Interner::new();
6952 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
6953 assert_eq!(rucc_ir::print(&module, &names), text);
6954 }
6955
6956 #[test]
6957 fn what_save_temps_keeps_is_the_text_that_was_compiled_and_the_assembly_that_was_assembled() {
6958 let mut opts = options();
6962 opts.emit = EmitKind::Object;
6963 opts.save_temps = rucc_session::SaveTemps::Object;
6964 let result = run(&opts, "#define N 2\nint a[N];\n");
6965 assert_eq!(result.messages, Vec::<String>::new());
6966 let text = result.temps.preprocessed.expect("the preprocessed text");
6967 assert!(text.contains("int a[2];"), "{text}");
6968 assert!(text.starts_with("# 1 \"/main.c\""), "{text}");
6969 let asm = result.temps.assembly.expect("the assembly");
6970 assert!(asm.contains("a:"), "{asm}");
6971 assert!(matches!(result.artifact, Artifact::Object { .. }), "{:?}", result.artifact);
6972 }
6973
6974 #[test]
6975 fn nothing_is_kept_unless_the_flag_asked_for_it() {
6976 let mut opts = options();
6979 opts.emit = EmitKind::Object;
6980 assert_eq!(run(&opts, "int a;\n").temps, Temps::default());
6981 }
6982
6983 #[test]
6984 fn a_compilation_that_stops_before_the_back_end_keeps_the_text_and_no_assembly() {
6985 let mut opts = options();
6988 opts.emit = EmitKind::Ir;
6989 opts.save_temps = rucc_session::SaveTemps::Cwd;
6990 let result = run(&opts, "int a;\n");
6991 assert!(result.temps.preprocessed.is_some());
6992 assert_eq!(result.temps.assembly, None);
6993 }
6994}