1use std::path::Path;
14
15use rucc_base::Interner;
16use rucc_codegen::coverage::Fired;
17use rucc_codegen::elsewhere::Elsewhere;
18use rucc_codegen::pipeline::{self, Machine};
19use rucc_codegen::pressure::Pressure;
20use rucc_diag::{Diagnostic, Severity, Span};
21use rucc_ir::{FpContract, Pic as IrPic, Visibility as IrVisibility};
22use rucc_lex::{Convert, Keywords, PpToken, convert};
23use rucc_lower::Protector as LowerProtector;
24use rucc_sema::{Checker, Context as CheckContext};
25use rucc_session::{
26 Contract, EmitKind, FileSystem, Options, Padding, Pic, Protector, Session, Visibility,
27};
28use rucc_target::TargetInfo;
29use rucc_tuple::{Arch, ObjectFormat};
30
31use crate::preprocess::render;
32
33#[derive(Debug, Clone, PartialEq, Eq, Default)]
40pub enum Artifact {
41 #[default]
44 Nothing,
45 Text(String),
47 Object {
54 bytes: Vec<u8>,
56 defines: Vec<String>,
60 },
61}
62
63impl Artifact {
64 #[must_use]
66 pub fn bytes(&self) -> &[u8] {
67 match self {
68 Artifact::Nothing => &[],
69 Artifact::Text(text) => text.as_bytes(),
70 Artifact::Object { bytes, .. } => bytes,
71 }
72 }
73}
74
75#[derive(Debug, Clone, PartialEq, Eq)]
77pub struct Compiled {
78 pub artifact: Artifact,
80 pub messages: Vec<String>,
82 pub errors: u32,
84 pub fired: Fired,
90 pub pressure: Pressure,
95 pub dumps: Vec<rucc_opt::Dump>,
101 pub remarks: String,
107 pub deps: Vec<rucc_pp::Dependency>,
112 pub temps: Temps,
119}
120
121#[derive(Debug, Clone, PartialEq, Eq, Default)]
128pub struct Temps {
129 pub preprocessed: Option<String>,
131 pub assembly: Option<String>,
133}
134
135impl Compiled {
136 #[must_use]
138 pub fn failed(&self) -> bool {
139 self.errors > 0
140 }
141
142 #[must_use]
147 pub fn text(&self) -> &str {
148 match &self.artifact {
149 Artifact::Text(text) => text,
150 _ => "",
151 }
152 }
153}
154
155#[must_use]
168pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
169 let mut sess = Session::new(opts.clone());
170 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
174 let mut diagnostics: Vec<Diagnostic> = Vec::new();
175 let mut fired = Fired::new();
177 let mut pressure = Pressure::new();
179 let mut dumps = Vec::new();
181 let mut remarks = String::new();
182 let mut temps = Temps::default();
184
185 let bytes = match fs.read(Path::new(name)) {
186 Ok(bytes) => bytes,
187 Err(e) => return failure(format!("{name}: {e}")),
188 };
189 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
190 return failure(format!("{name}: the source map has no room left for this file"));
191 };
192
193 let mut pp = rucc_pp::Preprocessor::with_prefix_map(opts.prefix_map.macros.clone());
197 let predef = rucc_pp::Predef::for_options(opts);
198 let expanded: Vec<PpToken> = {
199 let mut tokens = Vec::new();
200 {
205 let mut cx =
206 rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
207 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
208 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
209 return failure(format!(
210 "{name}: the source map has no room for the built in macros"
211 ));
212 }
213 if pp.preinclude(&opts.preincludes, &mut tokens, &mut cx).is_err() {
214 return failure(format!("{name}: the source map has no room for the command line"));
215 }
216 tokens.append(&mut pp.run(file, &mut cx));
217 }
218 if opts.save_temps.wanted() {
219 temps.preprocessed = Some(rucc_pp::print(
220 file,
221 &tokens,
222 pp.line_directives(),
223 &sess.sources,
224 &sess.interner,
225 rucc_pp::PrintOptions { line_markers: opts.line_markers },
226 ));
227 }
228 tokens.iter().map(|token| token.to_pp()).collect()
229 };
230 diagnostics.extend(pp.take_diagnostics());
231 let deps = pp.dependencies().to_vec();
234
235 let cx = Convert {
238 keywords: &keywords,
239 interner: &sess.interner,
240 target: &sess.target,
241 std: opts.std,
242 gnu: opts.gnu_extensions,
243 pedantic: opts.pedantic,
244 };
245 let (tokens, complaints) = convert(&expanded, &cx);
246 diagnostics.extend(complaints);
247
248 let parsed = rucc_parse::parse(
249 &tokens,
250 rucc_parse::Context {
251 interner: &sess.interner,
252 std: opts.std,
253 gnu: opts.gnu_extensions,
254 pedantic: opts.pedantic,
255 error_limit: opts.error_limit as usize,
256 },
257 );
258 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
259 diagnostics.extend(parsed.diagnostics);
260
261 let mut artifact = Artifact::Nothing;
262 let mut instrumented = Instrumented::default();
265 if !parse_failed {
266 let mut checker = Checker::new(
267 &parsed.ast,
268 CheckContext {
269 names: &sess.interner,
270 target: &sess.target,
271 std: opts.std,
272 gnu: opts.gnu_extensions,
273 pedantic: opts.pedantic,
274 permissive: opts.permissive,
275 gnu89_inline: opts.gnu89_inline,
276 error_limit: opts.error_limit as usize,
277 builtins: opts.builtins && opts.hosted,
280 no_builtin: &opts.no_builtin,
281 short_enums: opts.short_enums,
282 trapping_math: opts.trapping_math,
283 },
284 );
285 checker.check_unit();
286 let checked = checker.finish();
287 if !checked.failed() {
288 match opts.emit {
289 EmitKind::Tast => {
290 artifact = Artifact::Text(rucc_sema::print(
291 &checked.tast,
292 &checked.types,
293 &sess.interner,
294 ));
295 }
296 EmitKind::TypeGranules => {
300 artifact = Artifact::Text(rucc_types::granule_report(
301 &checked.types,
302 &sess.interner,
303 &sess.target,
304 ));
305 }
306 EmitKind::Ir
307 | EmitKind::MirFinal
308 | EmitKind::Asm
309 | EmitKind::Object
310 | EmitKind::Archive
311 | EmitKind::Executable
312 | EmitKind::SafetySummary => {
313 let mut read = |named: &str| {
318 fs.read(Path::new(named))
319 .map(|bytes| bytes.as_slice().to_vec())
320 .map_err(|why| why.to_string())
321 };
322 let mut lowered = rucc_lower::lower(
323 name,
324 rucc_lower::Context {
325 tast: &checked.tast,
326 types: &checked.types,
327 target: &sess.target,
328 names: &mut sess.interner,
329 visibility: match opts.visibility {
330 Visibility::Default => IrVisibility::Default,
331 Visibility::Hidden => IrVisibility::Hidden,
332 Visibility::Protected => IrVisibility::Protected,
333 },
334 protector: match opts.protector {
335 Protector::None => LowerProtector::None,
336 Protector::Buffers => LowerProtector::Buffers,
337 Protector::Strong => LowerProtector::Strong,
338 Protector::All => LowerProtector::All,
339 },
340 wrapping: rucc_lower::Wrapping {
341 signed: opts.wrapping.signed,
342 pointer: opts.wrapping.pointer,
343 trap: opts.wrapping.trap,
344 },
345 aliasing: opts.strict_aliasing,
346 padding: opts.padding == Padding::Ignored,
347 contract: match opts.fp_contract {
348 Contract::Off => FpContract::Off,
349 Contract::On => FpContract::On,
350 Contract::Fast => FpContract::Fast,
351 },
352 read: &mut read,
353 },
354 );
355 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
359 if !failed {
360 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
365 for error in errors {
366 diagnostics.push(internal(&format!("invalid IR, {error}")));
367 }
368 } else if let Err(complaints) =
369 instrument(&mut lowered.module, &mut sess.interner, opts)
370 .map(|done| instrumented = done)
371 {
372 diagnostics.extend(complaints);
373 } else if let Err(complaints) = optimize(
374 &mut lowered.module,
375 &sess.interner,
376 &sess.target,
377 opts,
378 name,
379 &mut dumps,
380 &mut remarks,
381 ) {
382 diagnostics.extend(complaints);
383 } else if opts.emit == EmitKind::SafetySummary {
384 artifact = Artifact::Text(
389 rucc_safety::summarize(
390 &lowered.module,
391 &sess.interner,
392 name,
393 opts.safety.as_str(),
394 instrumented.checks,
395 instrumented.interposed,
396 instrumented.crossings,
397 )
398 .render(),
399 );
400 } else if opts.emit == EmitKind::Ir {
401 artifact =
406 Artifact::Text(rucc_ir::print(&lowered.module, &sess.interner));
407 } else {
408 match generate(
411 &mut lowered.module,
412 &mut sess.interner,
413 &sess.target,
414 opts,
415 &mut fired,
416 &mut pressure,
417 &mut temps.assembly,
418 ) {
419 Ok(made) => artifact = made,
420 Err(complaints) => diagnostics.extend(complaints),
421 }
422 }
423 }
424 diagnostics.extend(lowered.diagnostics);
425 }
426 _ => {}
427 }
428 }
429 diagnostics.extend(checked.diagnostics);
430 }
431
432 let mut messages = Vec::with_capacity(diagnostics.len());
433 let mut errors = 0;
434 for diag in &diagnostics {
435 if !opts.warnings && diag.severity == Severity::Warning {
439 continue;
440 }
441 if diag.severity.is_fatal()
442 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
443 {
444 errors += 1;
445 }
446 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
447 }
448 if errors > 0 {
449 artifact = Artifact::Nothing;
451 }
452 Compiled { artifact, messages, errors, fired, pressure, dumps, remarks, deps, temps }
455}
456
457#[must_use]
467pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
468 let mut sess = Session::new(opts.clone());
469 if opts.emit != EmitKind::Ir {
470 return failure(format!(
471 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
472 the C in front of it became",
473 opts.emit.as_str()
474 ));
475 }
476 let bytes = match fs.read(Path::new(name)) {
477 Ok(bytes) => bytes,
478 Err(e) => return failure(format!("{name}: {e}")),
479 };
480 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
481 return failure(format!("{name}: this is not text, so it is not IR"));
482 };
483
484 let module = match rucc_ir::parse(text, &mut sess.interner) {
485 Ok(module) => module,
486 Err(error) => {
487 return failure(format!("{name}:{}: {}", error.line, error.message));
488 }
489 };
490 let mut diagnostics: Vec<Diagnostic> = Vec::new();
491 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
492 for error in errors {
493 diagnostics.push(invalid(&format!("invalid IR, {error}")));
494 }
495 }
496 let mut messages = Vec::with_capacity(diagnostics.len());
497 for diag in &diagnostics {
498 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
499 }
500 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
501 let artifact = if errors > 0 {
502 Artifact::Nothing
503 } else {
504 Artifact::Text(rucc_ir::print(&module, &sess.interner))
505 };
506 Compiled {
508 artifact,
509 messages,
510 errors,
511 fired: Fired::new(),
512 pressure: Pressure::new(),
513 dumps: Vec::new(),
514 remarks: String::new(),
515 deps: Vec::new(),
516 temps: Temps::default(),
517 }
518}
519
520fn instrument(
543 module: &mut rucc_ir::Module,
544 names: &mut Interner,
545 opts: &Options,
546) -> Result<Instrumented, Vec<Diagnostic>> {
547 if !opts.safety.instruments() {
548 return Ok(Instrumented::default());
549 }
550 let checks = rucc_safety::run(module, opts.subobject, opts.promise, opts.races);
551 let interposed = rucc_safety::redirect(module, names);
556 let crossings = rucc_safety::witness(module, names);
559 match rucc_ir::verify(module, names) {
560 Ok(()) => Ok(Instrumented { checks, interposed, crossings }),
561 Err(errors) => Err(errors
562 .iter()
563 .map(|e| internal(&format!("invalid IR after check insertion, {e}")))
564 .collect()),
565 }
566}
567
568#[derive(Clone, Copy, Debug, Default)]
574struct Instrumented {
575 checks: rucc_safety::Counts,
577 interposed: usize,
579 crossings: rucc_safety::Sites,
581}
582
583fn optimize(
595 module: &mut rucc_ir::Module,
596 names: &Interner,
597 target: &TargetInfo,
598 opts: &Options,
599 file: &str,
600 dumps: &mut Vec<rucc_opt::Dump>,
601 remarks: &mut String,
602) -> Result<(), Vec<Diagnostic>> {
603 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
604 settings.interposition = match opts.interposition {
610 true => replaceable(target, opts),
611 false => IrPic::Executable,
612 };
613 settings.toggles.clone_from(&opts.passes);
614 settings.fuel = opts.pass_fuel.iter().cloned().collect();
615 settings.global_fuel = opts.pass_fuel_global;
616 settings.verify |= opts.verify_each;
617 for (on, spec) in &opts.pass_gates {
618 if let Err(why) = settings.gates.add(*on, spec) {
621 return Err(vec![internal(&why)]);
622 }
623 }
624 for spec in &opts.dump_ir {
625 if let Err(why) = settings.dumps.add(spec) {
628 return Err(vec![internal(&why)]);
629 }
630 }
631 let mut wants = rucc_opt::Wants::none();
632 for spec in &opts.opt_info {
633 if let Err(why) = wants.add(spec) {
636 return Err(vec![internal(&why)]);
637 }
638 }
639 let report = rucc_opt::run(module, names, &settings);
640 remarks.push_str(&rucc_opt::optinfo::render(file, &report, names, wants));
641 dumps.extend(report.dumps);
642 match report.broke.is_empty() {
643 true => Ok(()),
644 false => Err(report.broke.iter().map(|why| internal(why)).collect()),
645 }
646}
647
648fn replaceable(target: &TargetInfo, opts: &Options) -> IrPic {
682 match (target.tuple.os().object_format(), opts.pic) {
683 (Some(ObjectFormat::Elf), Pic::Library) => IrPic::Library,
684 _ => IrPic::Executable,
685 }
686}
687
688fn generate(
689 module: &mut rucc_ir::Module,
690 names: &mut Interner,
691 target: &TargetInfo,
692 opts: &Options,
693 fired: &mut Fired,
694 pressure: &mut Pressure,
695 assembly: &mut Option<String>,
696) -> Result<Artifact, Vec<Diagnostic>> {
697 let Some(machine) = Machine::for_target(target) else {
698 return Err(vec![unsupported(&format!(
699 "there is no back end for {} in this compiler yet, so there is nothing to generate",
700 target.tuple
701 ))]);
702 };
703 if opts.protector != Protector::None && machine.conv.guard.is_none() {
708 return Err(vec![unsupported(&format!(
709 "{} is not supported for {} yet, because the stack protector on that target is not \
710 the one this compiler writes",
711 opts.protector, target.tuple
712 ))]);
713 }
714 if opts.control.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
720 return Err(vec![unsupported(&format!(
721 "-fcf-protection={} is not supported for {} yet, because what says a file was built \
722 for it there is not the note this compiler writes",
723 opts.control, target.tuple
724 ))]);
725 }
726 let profile = match machine.conv.trace {
732 Some(trace) => opts.profile.then(|| opts.hook.early(trace.fentry)),
733 None if opts.profile => {
734 return Err(vec![unsupported(&format!(
735 "-pg is not supported for {} yet, because the profiler's hook on that target is \
736 not the one this compiler calls",
737 target.tuple
738 ))]);
739 }
740 None => None,
741 };
742 if opts.patchable.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
747 return Err(vec![unsupported(&format!(
748 "-fpatchable-function-entry= is not supported for {} yet, because what records where \
749 the room is there is not the section this compiler writes",
750 target.tuple
751 ))]);
752 }
753 let flags = pipeline::Flags {
754 frame_pointer: opts.frame_pointer,
755 red_zone: opts.red_zone,
756 stack_clash: opts.stack_clash,
757 landing: opts.control.branch(),
758 profile: match profile {
759 None => pipeline::Profile::No,
760 Some(true) => pipeline::Profile::Early,
761 Some(false) => pipeline::Profile::Late,
762 },
763 patch: pipeline::Room { after: opts.patchable.after(), before: opts.patchable.before },
764 reorder: opts.reorder_blocks.unwrap_or_else(|| opts.opt_level.runs_optimizer()),
771 reuse: opts.stack_reuse.unwrap_or_else(|| opts.opt_level.runs_optimizer()),
776 schedule: opts.schedule_insns.unwrap_or_else(|| opts.opt_level.schedules()),
781 accurate: opts.cycle_accurate_model,
783 };
784
785 if opts.safety.instruments() {
794 rucc_opt::heap::annotate(module, names);
804 rucc_safety::lower(module, names);
805 if let Err(errors) = rucc_ir::verify(module, names) {
806 return Err(errors
807 .iter()
808 .map(|e| internal(&format!("invalid IR after check lowering, {e}")))
809 .collect());
810 }
811 }
812
813 let elsewhere = Elsewhere::of(module, replaceable(target, opts));
821
822 let mut funcs = Vec::new();
823 let mut complaints = Vec::new();
824 for id in module.funcs() {
825 if module[id].is_declaration() {
826 continue;
827 }
828 match pipeline::compile_recording(
829 &mut module[id],
830 names,
831 &machine,
832 &elsewhere,
833 flags,
834 fired,
835 pressure,
836 ) {
837 Ok(func) => funcs.push(func),
838 Err(why) => {
839 let name = names.resolve(module[id].name).to_owned();
840 let span = why.inst().map_or(Span::DUMMY, |inst| module[id].span(inst));
843 let said = format!("cannot generate code for '{name}': {why}");
844 complaints.push(unsupported_at(&said, span));
845 }
846 }
847 }
848 if !complaints.is_empty() {
849 return Err(complaints);
850 }
851 let (globals, aliases) = match opts.emit {
857 EmitKind::Asm | EmitKind::Object | EmitKind::Archive | EmitKind::Executable => (
858 rucc_asm::globals(module, names, target.object_format).map_err(refused)?,
859 rucc_asm::aliases(module, names).map_err(refused)?,
860 ),
861 _ => (rucc_asm::Globals::default(), Vec::new()),
862 };
863 let unwind = opts.unwinds();
867 match opts.emit {
868 EmitKind::Asm => {
869 rucc_asm::print(&funcs, &globals, &aliases, names, target, unwind, output(opts, target))
870 .map(Artifact::Text)
871 .map_err(refused)
872 }
873 EmitKind::Object | EmitKind::Archive | EmitKind::Executable => {
877 if opts.save_temps.wanted() {
878 let listing = rucc_asm::print(
879 &funcs,
880 &globals,
881 &aliases,
882 names,
883 target,
884 unwind,
885 output(opts, target),
886 );
887 *assembly = Some(listing.map_err(refused)?);
888 }
889 let text = rucc_asm::assemble(&funcs, names, target, unwind).map_err(refused)?;
890 let data = globals.image();
891 let bytes = rucc_object::write(&text, &data, &aliases, target, output(opts, target))
894 .map_err(wrote)?;
895 let defines = rucc_object::defines(&text, &data, &aliases, target).map_err(wrote)?;
900 Ok(Artifact::Object { bytes, defines })
901 }
902 _ => Ok(Artifact::Text(rucc_mir::print(&funcs, names, target.regs))),
903 }
904}
905
906fn output(opts: &Options, target: &TargetInfo) -> rucc_object::Output {
918 let mut features = 0;
919 if target.tuple.arch() == Arch::X86_64 {
920 if opts.control.branch() {
921 features |= rucc_object::Property::IBT;
922 }
923 if opts.control.ret() {
924 features |= rucc_object::Property::SHSTK;
925 }
926 }
927 rucc_object::Output {
928 sections: rucc_object::Sections {
929 functions: opts.function_sections,
930 data: opts.data_sections,
931 },
932 property: rucc_object::Property { features },
933 }
934}
935
936fn wrote(why: rucc_object::Error) -> Vec<Diagnostic> {
942 match why {
943 rucc_object::Error::Format { .. } => vec![unsupported(&why.to_string())],
944 rucc_object::Error::Refused { .. } => vec![internal(&why.to_string())],
945 }
946}
947
948fn refused(why: rucc_asm::Error) -> Vec<Diagnostic> {
954 match why {
955 rucc_asm::Error::Thread { .. } | rucc_asm::Error::IFunc { .. } => {
956 vec![unsupported(&why.to_string())]
957 }
958 _ => vec![internal(&why.to_string())],
959 }
960}
961
962fn unsupported(message: &str) -> Diagnostic {
968 unsupported_at(message, Span::DUMMY)
969}
970
971fn unsupported_at(message: &str, span: Span) -> Diagnostic {
977 Diagnostic::error(message.to_owned(), span)
978 .with_code("E0653")
979 .note("this construct is not lowered yet, see https://github.com/tamnd/rucc/issues", span)
980}
981
982fn invalid(message: &str) -> Diagnostic {
984 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
985}
986
987fn internal(message: &str) -> Diagnostic {
989 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
990 .with_code("E0652")
991 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
992}
993
994fn failure(message: String) -> Compiled {
997 Compiled {
998 artifact: Artifact::Nothing,
999 messages: vec![format!("rucc: error: {message}")],
1000 errors: 1,
1001 fired: Fired::new(),
1002 pressure: Pressure::new(),
1003 dumps: Vec::new(),
1004 remarks: String::new(),
1005 deps: Vec::new(),
1006 temps: Temps::default(),
1007 }
1008}
1009
1010#[cfg(test)]
1011mod tests {
1012 use rucc_session::{MemoryFileSystem, Std};
1013 use rucc_target::Triple;
1014
1015 use super::*;
1016
1017 fn options() -> Options {
1018 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
1019 opts.emit = EmitKind::Tast;
1020 opts
1021 }
1022
1023 fn run(opts: &Options, source: &str) -> Compiled {
1024 let mut fs = MemoryFileSystem::new();
1025 fs.insert("/main.c", source.to_owned().into_bytes());
1026 compile(opts, "/main.c", &fs)
1027 }
1028
1029 fn freestanding() -> Options {
1033 let mut opts = options();
1034 opts.hosted = false;
1035 opts.search.push_system(rucc_session::runtime::DIR);
1036 opts
1037 }
1038
1039 fn shipped(source: &str) -> String {
1041 let result = run(&freestanding(), source);
1042 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1043 result.text().to_owned()
1044 }
1045
1046 fn tast(source: &str) -> String {
1048 let result = run(&options(), source);
1049 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1050 result.text().to_owned()
1051 }
1052
1053 #[test]
1054 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
1055 let text = shipped(concat!(
1056 "#include <stdarg.h>\n",
1057 "int sum(int n, ...) {\n",
1058 " va_list ap, copy;\n",
1059 " va_start(ap, n);\n",
1060 " va_copy(copy, ap);\n",
1061 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
1062 " va_end(ap);\n",
1063 " va_end(copy);\n",
1064 " return total;\n",
1065 "}\n",
1066 ));
1067 assert!(text.contains("va-start"), "{text}");
1068 assert!(text.contains("va-copy"), "{text}");
1069 assert!(text.contains("va-arg"), "{text}");
1070 assert!(text.contains("va-end"), "{text}");
1071 }
1072
1073 #[test]
1077 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
1078 let text = shipped(concat!(
1079 "#define __need___va_list\n",
1080 "#include <stdarg.h>\n",
1081 "int vprint(const char *f, __gnuc_va_list ap);\n",
1082 "#ifdef va_start\n",
1083 "#error va_start should not be defined\n",
1084 "#endif\n",
1085 "#ifdef _VA_LIST_DEFINED\n",
1086 "#error va_list should not have been made\n",
1087 "#endif\n",
1088 ));
1089 assert!(text.contains("vprint"), "{text}");
1090 }
1091
1092 #[test]
1095 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
1096 let text = shipped(concat!(
1097 "#define __need_size_t\n",
1098 "#include <stddef.h>\n",
1099 "#ifdef offsetof\n",
1100 "#error offsetof should not be defined yet\n",
1101 "#endif\n",
1102 "#define __need_ptrdiff_t\n",
1103 "#include <stddef.h>\n",
1104 "#include <stddef.h>\n",
1105 "size_t a;\n",
1106 "ptrdiff_t b;\n",
1107 "wchar_t c;\n",
1108 "max_align_t d;\n",
1109 "void *e = NULL;\n",
1110 "struct P { int x; long y; };\n",
1111 "size_t f = offsetof(struct P, y);\n",
1112 ));
1113 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
1114 assert!(text.contains("decl #1 b : long"), "{text}");
1115 }
1116
1117 #[test]
1118 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
1119 let text = shipped(concat!(
1120 "#include <limits.h>\n",
1121 "#include <float.h>\n",
1122 "int bits = CHAR_BIT;\n",
1123 "long big = LONG_MAX;\n",
1124 "int low = INT_MIN;\n",
1125 "int radix = FLT_RADIX;\n",
1126 "int digits = DBL_MANT_DIG;\n",
1127 ));
1128 assert!(text.contains("const 8 : int"), "{text}");
1129 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
1130 assert!(text.contains("const 2 : int"), "{text}");
1131 assert!(text.contains("const 53 : int"), "{text}");
1132 }
1133
1134 #[test]
1138 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
1139 let text = shipped(concat!(
1140 "#include <stdint.h>\n",
1141 "int64_t a = INT64_C(1);\n",
1142 "uint_least16_t b;\n",
1143 "intptr_t c;\n",
1144 "uintmax_t d = UINTMAX_MAX;\n",
1145 "int wide = sizeof(int_fast64_t);\n",
1146 ));
1147 assert!(text.contains("decl #0 a : long"), "{text}");
1148 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
1149 assert!(text.contains("decl #2 c : long"), "{text}");
1150 }
1151
1152 #[test]
1163 fn the_shipped_mmintrin_defines_the_mmx_type_and_the_operations_over_it() {
1164 let text = shipped(concat!(
1165 "#include <mmintrin.h>\n",
1166 "__m64 add(__m64 a, __m64 b) { return _mm_add_pi16(a, b); }\n",
1167 "__m64 pack(__m64 a, __m64 b) { return _m_packsswb(a, b); }\n",
1168 "__m64 shift(__m64 a) { return _mm_srai_pi32(a, 3); }\n",
1169 "int low(__m64 a) { return _mm_cvtsi64_si32(a); }\n",
1170 "void done(void) { _mm_empty(); }\n",
1171 ));
1172 assert!(text.contains("add"), "{text}");
1173 assert!(text.contains("pack"), "{text}");
1174 assert!(text.contains("shift"), "{text}");
1175 }
1176
1177 #[test]
1182 fn the_shipped_mm_malloc_asks_for_aligned_memory_and_gives_it_back() {
1183 let text = shipped(concat!(
1184 "#include <mm_malloc.h>\n",
1185 "void *get(void) { return _mm_malloc(64, 16); }\n",
1186 "void put(void *p) { _mm_free(p); }\n",
1187 ));
1188 assert!(text.contains("get"), "{text}");
1189 assert!(text.contains("put"), "{text}");
1190 }
1191
1192 #[test]
1204 fn the_shipped_xmmintrin_defines_the_sse_type_and_the_operations_over_it() {
1205 let text = shipped(concat!(
1206 "#include <xmmintrin.h>\n",
1207 "__m128 add(__m128 a, __m128 b) { return _mm_add_ps(a, b); }\n",
1208 "__m128 one(__m128 a, __m128 b) { return _mm_max_ss(a, b); }\n",
1209 "__m128 mask(__m128 a, __m128 b) { return _mm_cmpnle_ps(a, b); }\n",
1210 "__m128 pick(__m128 a, __m128 b) { return _mm_shuffle_ps(a, b, _MM_SHUFFLE(0,1,2,3)); }\n",
1211 "int bits(__m128 a) { return _mm_movemask_ps(a); }\n",
1212 "int near(__m128 a) { return _mm_cvtss_si32(a); }\n",
1213 "__m128 wide(__m64 a) { return _mm_cvtpi16_ps(a); }\n",
1214 "void *room(void) { return _mm_malloc(64, 16); }\n",
1215 "void hint(const float *p) { _mm_prefetch(p, _MM_HINT_T0); _mm_sfence(); }\n",
1216 ));
1217 assert!(text.contains("add"), "{text}");
1218 assert!(text.contains("mask"), "{text}");
1219 assert!(text.contains("pick"), "{text}");
1220 assert!(text.contains("wide"), "{text}");
1221 }
1222
1223 #[test]
1230 fn the_shipped_xmmintrin_leaves_out_the_names_that_need_an_instruction() {
1231 let text = rucc_session::runtime::header("xmmintrin.h").expect("xmmintrin.h is shipped");
1232 for absent in [
1233 "_mm_sqrt_ps",
1234 "_mm_sqrt_ss",
1235 "_mm_rsqrt_ps",
1236 "_mm_rsqrt_ss",
1237 "_mm_getcsr",
1238 "_mm_setcsr",
1239 ] {
1240 let defined = text.contains(&format!("{absent}("));
1241 assert!(!defined, "{absent} is defined and the header says it is not");
1242 assert!(text.contains(absent), "{absent} is absent and unexplained");
1243 }
1244 }
1245
1246 #[test]
1247 fn the_shipped_emmintrin_defines_both_sse2_types_and_the_operations_over_them() {
1248 let text = shipped(concat!(
1249 "#include <emmintrin.h>\n",
1250 "__m128i add(__m128i a, __m128i b) { return _mm_add_epi64(a, b); }\n",
1251 "__m128i wide(__m128i a, __m128i b) { return _mm_mul_epu32(a, b); }\n",
1252 "__m128i pick(__m128i a) { return _mm_shuffle_epi32(a, _MM_SHUFFLE(0,1,2,3)); }\n",
1253 "__m128i up(__m128i a) { return _mm_slli_epi64(a, 13); }\n",
1254 "__m128i down(__m128i a) { return _mm_srli_si128(a, 3); }\n",
1255 "__m128i pack(__m128i a, __m128i b) { return _mm_packus_epi16(a, b); }\n",
1256 "int bits(__m128i a) { return _mm_movemask_epi8(a); }\n",
1257 "__m128d sum(__m128d a, __m128d b) { return _mm_add_sd(a, b); }\n",
1258 "__m128d mask(__m128d a, __m128d b) { return _mm_cmpunord_pd(a, b); }\n",
1259 "__m128i near(__m128d a) { return _mm_cvtpd_epi32(a); }\n",
1260 "__m128d over(__m128 a) { return _mm_cvtps_pd(a); }\n",
1261 "__m128i half(__m64 a) { return _mm_movpi64_epi64(a); }\n",
1262 "__m128i grab(void const *p) { return _mm_loadu_si128(p); }\n",
1263 "void wall(void) { _mm_lfence(); _mm_mfence(); }\n",
1264 ));
1265 assert!(text.contains("wide"), "{text}");
1266 assert!(text.contains("pack"), "{text}");
1267 assert!(text.contains("near"), "{text}");
1268 assert!(text.contains("half"), "{text}");
1269 }
1270
1271 #[test]
1275 fn the_shipped_immintrin_reaches_the_names_the_headers_under_it_define() {
1276 let text = shipped(concat!(
1277 "#include <immintrin.h>\n",
1278 "unsigned long long matching(unsigned char tag, unsigned char const *bucket) {\n",
1279 " __m128i const want = _mm_set1_epi8((char)tag);\n",
1280 " __m128i const chunk = _mm_loadu_si128((__m128i const *)(void const *)bucket);\n",
1281 " __m128i const same = _mm_cmpeq_epi8(chunk, want);\n",
1282 " return (unsigned long long)_mm_movemask_epi8(same);\n",
1283 "}\n",
1284 "__m64 narrow(__m64 a, __m64 b) { return _mm_add_pi32(a, b); }\n",
1285 "__m128 single(__m128 a, __m128 b) { return _mm_add_ps(a, b); }\n",
1286 ));
1287 assert!(text.contains("matching"), "{text}");
1288 assert!(text.contains("narrow"), "the MMX header is not reached: {text}");
1289 assert!(text.contains("single"), "the SSE header is not reached: {text}");
1290 }
1291
1292 #[test]
1296 fn the_umbrella_and_the_header_under_it_can_both_be_included() {
1297 let text = shipped(concat!(
1298 "#include <immintrin.h>\n",
1299 "#include <emmintrin.h>\n",
1300 "#include <immintrin.h>\n",
1301 "__m128i twice(__m128i a, __m128i b) { return _mm_add_epi32(a, b); }\n",
1302 ));
1303 assert!(text.contains("twice"), "{text}");
1304 }
1305
1306 #[test]
1310 fn the_shipped_emmintrin_leaves_out_the_two_square_roots() {
1311 let text = rucc_session::runtime::header("emmintrin.h").expect("emmintrin.h is shipped");
1312 for absent in ["_mm_sqrt_pd", "_mm_sqrt_sd"] {
1313 let defined = text.contains(&format!("{absent}("));
1314 assert!(!defined, "{absent} is defined and the header says it is not");
1315 assert!(text.contains(absent), "{absent} is absent and unexplained");
1316 }
1317 }
1318
1319 #[test]
1320 fn the_three_formality_headers_still_have_to_work() {
1321 let text = shipped(concat!(
1322 "#include <stdbool.h>\n",
1323 "#include <stdalign.h>\n",
1324 "#include <iso646.h>\n",
1325 "#include <stdnoreturn.h>\n",
1326 "int t = true and not false;\n",
1327 "_Alignas(16) char buf[16];\n",
1328 "int a = alignof(long);\n",
1329 ));
1330 assert!(text.contains("decl #0 t : int"), "{text}");
1331 assert!(text.contains("const 8 : unsigned long"), "{text}");
1332 }
1333
1334 #[test]
1342 fn every_shipped_header_can_be_included_twice() {
1343 let once: String = rucc_session::runtime::names()
1344 .iter()
1345 .map(|name| format!("#include <{name}>\n"))
1346 .collect();
1347 let twice = once.repeat(2);
1348 assert_eq!(shipped(&format!("{once}int x;\n")), shipped(&format!("{twice}int x;\n")));
1349 }
1350
1351 #[test]
1352 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
1353 let fs = MemoryFileSystem::new();
1354 let result = compile(&options(), "/nope.c", &fs);
1355 assert!(result.failed());
1356 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
1357 assert!(result.text().is_empty());
1358 }
1359
1360 #[test]
1361 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
1362 let text = tast("int x = 1;\n");
1363 let expected = "\
1364decl #0 x : int object external static defined
1365 init
1366 +0
1367 const 1 : int
1368";
1369 assert_eq!(text, expected);
1370 }
1371
1372 #[test]
1373 fn the_macros_are_expanded_before_anything_is_parsed() {
1374 let text = tast("#define N 2\nint a[N];\n");
1378 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
1379 }
1380
1381 #[test]
1387 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
1388 let text = tast(concat!(
1389 "#pragma pack(4)\n",
1390 "struct s { int a; };\n",
1391 "#pragma pack()\n",
1392 "int b;\n",
1393 "_Pragma(\"GCC visibility push(default)\") int c;\n",
1394 ));
1395 assert!(text.contains("decl #0 b : int"), "{text}");
1396 assert!(text.contains("decl #1 c : int"), "{text}");
1397 }
1398
1399 #[test]
1407 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
1408 tast(concat!(
1409 "struct A { char c; int i; } __attribute__((packed));\n",
1410 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1411 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1412 "struct B { char c; int i; } __attribute__((aligned));\n",
1415 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
1416 "struct C { char c; int i __attribute__((packed)); };\n",
1417 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
1418 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
1419 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
1420 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
1421 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
1422 "struct E { char c; _Alignas(8) int i; };\n",
1423 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
1424 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
1425 "struct F { char c; int i __attribute__((aligned(8))); };\n",
1426 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
1427 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
1430 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
1431 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
1432 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
1433 "struct I { [[gnu::packed]] char c; int i; };\n",
1436 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1437 "struct J { char c; [[gnu::packed]] int i; };\n",
1438 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
1439 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
1440 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
1441 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
1442 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
1443 "union L { char c; int i; } __attribute__((packed));\n",
1444 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
1445 "struct O { char c; int i; } __attribute__((__packed__));\n",
1449 "_Static_assert(sizeof(struct O) == 5 && _Alignof(struct O) == 1, \"O\");\n",
1450 "struct P { char c; int i; } __attribute__((__aligned__(8)));\n",
1451 "_Static_assert(sizeof(struct P) == 8 && _Alignof(struct P) == 8, \"P\");\n",
1452 ));
1453 }
1454
1455 #[test]
1464 fn an_access_to_a_packed_member_says_the_alignment_the_layout_left_it() {
1465 let packed = body(concat!(
1466 "struct P { char c; int v; } __attribute__((packed));\n",
1467 "int f(struct P *p) { return p->v; }\n",
1468 ));
1469 assert!(packed.contains("load.i32 %2, align 1,"), "{packed}");
1470 let plain = body(concat!(
1472 "struct P { char c; int v; };\n",
1473 "int f(struct P *p) { return p->v; }\n",
1474 ));
1475 assert!(plain.contains("load.i32 %2, align 4,"), "{plain}");
1476 }
1477
1478 #[test]
1485 fn what_is_inside_a_packed_member_is_no_more_aligned_than_the_member_is() {
1486 let stepped = body(concat!(
1487 "struct P { char c; int v[4]; } __attribute__((packed));\n",
1488 "int f(struct P *p, int i) { return p->v[i]; }\n",
1489 ));
1490 assert!(stepped.contains(", align 1,"), "{stepped}");
1491 assert!(!stepped.contains(", align 4,"), "{stepped}");
1492 let nested = body(concat!(
1493 "struct Inner { int v; };\n",
1494 "struct P { char c; struct Inner in; } __attribute__((packed));\n",
1495 "int f(struct P *p) { return p->in.v; }\n",
1496 ));
1497 assert!(nested.contains(", align 1,"), "{nested}");
1498 assert!(!nested.contains(", align 4,"), "{nested}");
1499 }
1500
1501 #[test]
1510 fn the_aligned_attribute_on_a_declaration_raises_what_that_one_object_is_aligned_to() {
1511 tast(concat!(
1512 "int v __attribute__((aligned(64)));\n",
1513 "_Static_assert(__alignof__(v) == 64, \"v\");\n",
1514 "__attribute__((aligned(32))) int w;\n",
1517 "_Static_assert(__alignof__(w) == 32, \"w\");\n",
1518 "[[gnu::aligned(16)]] int x;\n",
1519 "_Static_assert(__alignof__(x) == 16, \"x\");\n",
1520 "int y __attribute__((aligned(2)));\n",
1523 "_Static_assert(__alignof__(y) == 4, \"y\");\n",
1524 "void f(void) { int a __attribute__((aligned(128)));\n",
1526 "_Static_assert(__alignof__(a) == 128, \"a\"); (void)a; }\n",
1527 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1530 "void g(void) __attribute__((aligned(256)));\n",
1533 "void g(void) {}\n",
1534 "_Static_assert(__alignof__(g) == 256, \"g\");\n",
1535 ));
1536 }
1537
1538 #[test]
1542 fn what_a_declaration_asked_to_be_aligned_to_is_what_the_assembler_is_told() {
1543 let text = asm(concat!(
1544 "int v __attribute__((aligned(64)));\n",
1545 "void g(void) __attribute__((aligned(256)));\n",
1546 "void g(void) {}\n",
1547 "void plain(void) {}\n",
1548 ));
1549 assert!(text.contains("\t.p2align\t6\n\t.type\tv, @object\n"), "{text}");
1550 assert!(text.contains("\t.p2align\t8, 0x90\n\t.globl\tg\n"), "{text}");
1551 assert!(text.contains("\t.p2align\t4, 0x90\n\t.globl\tplain\n"), "{text}");
1552 }
1553
1554 #[test]
1563 fn an_aligned_typedef_says_what_an_object_of_it_is_aligned_to_and_may_lower_it() {
1564 tast(concat!(
1565 "typedef int L __attribute__((aligned(2)));\n",
1566 "_Static_assert(__alignof__(L) == 2, \"L\");\n",
1567 "_Static_assert(_Alignof(L) == 2, \"L alignof\");\n",
1568 "_Static_assert(sizeof(L) == 4, \"L size\");\n",
1570 "struct T { char c; L x; };\n",
1571 "_Static_assert(sizeof(struct T) == 6, \"T\");\n",
1572 "_Static_assert(__builtin_offsetof(struct T, x) == 2, \"T.x\");\n",
1573 "typedef int H __attribute__((aligned(16)));\n",
1575 "_Static_assert(__alignof__(H) == 16, \"H\");\n",
1576 "_Static_assert(sizeof(H) == 4, \"H size\");\n",
1577 "struct U { char c; H x; };\n",
1578 "_Static_assert(sizeof(struct U) == 32, \"U\");\n",
1579 "_Static_assert(__builtin_offsetof(struct U, x) == 16, \"U.x\");\n",
1580 "typedef L M __attribute__((aligned(8)));\n",
1583 "_Static_assert(__alignof__(M) == 8, \"M\");\n",
1584 "typedef L N;\n",
1587 "_Static_assert(__alignof__(N) == 2, \"N\");\n",
1588 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1590 ));
1591 let text = asm(concat!(
1592 "typedef int L __attribute__((aligned(2)));\n",
1593 "typedef int H __attribute__((aligned(16)));\n",
1594 "L low;\n",
1595 "H high;\n",
1596 ));
1597 assert!(text.contains("\t.p2align\t1\n\t.type\tlow, @object\n"), "{text}");
1598 assert!(text.contains("\t.p2align\t4\n\t.type\thigh, @object\n"), "{text}");
1599 }
1600
1601 #[test]
1609 fn the_vector_size_attribute_builds_a_type_of_lanes_and_measures_it_in_bytes() {
1610 tast(concat!(
1611 "typedef int __attribute__((vector_size(16))) v4si;\n",
1612 "_Static_assert(sizeof(v4si) == 16 && _Alignof(v4si) == 16, \"v4si\");\n",
1613 "typedef char __attribute__((vector_size(16))) v16qi;\n",
1614 "_Static_assert(sizeof(v16qi) == 16, \"v16qi\");\n",
1615 "typedef int __attribute__((vector_size(4))) v1si;\n",
1618 "_Static_assert(sizeof(v1si) == 4, \"v1si\");\n",
1619 "typedef float __attribute__((__vector_size__(8))) v2sf;\n",
1621 "_Static_assert(sizeof(v2sf) == 8, \"v2sf\");\n",
1622 "typedef short [[gnu::vector_size(8)]] v4hi;\n",
1623 "_Static_assert(sizeof(v4hi) == 8, \"v4hi\");\n",
1624 "v4si g;\n",
1627 "_Static_assert(sizeof(g[0]) == 4, \"lane\");\n",
1628 "_Static_assert(sizeof(g + g) == 16, \"whole\");\n",
1629 "_Static_assert(sizeof(g + 1) == 16, \"broadcast\");\n",
1632 "_Static_assert(sizeof(v4si[3]) == 48, \"array\");\n",
1634 ));
1635 }
1636
1637 #[test]
1647 fn a_vector_is_written_whole_into_an_array_of_them_and_named_by_a_type_name() {
1648 tast(concat!(
1649 "typedef int __attribute__((vector_size(8))) v2si;\n",
1650 "v2si table[] = { (v2si){ 1, 2 }, (v2si){ 3, 4 } };\n",
1651 "_Static_assert(sizeof(table) == 16, \"two of them and not eight lanes\");\n",
1652 "v2si written = (int __attribute__((vector_size(8)))){ 5, 6 };\n",
1654 "_Static_assert(sizeof((int __attribute__((vector_size(16)))){ 0 }) == 16, \"named\");\n",
1655 "v2si lanes[2] = { 1, 2, 3, 4 };\n",
1658 "_Static_assert(sizeof(lanes) == 16, \"still elided\");\n",
1659 ));
1660 }
1661
1662 #[test]
1670 fn a_lane_is_assignable_and_a_shift_takes_a_count_of_its_own_lane() {
1671 let result = run(
1672 &options(),
1673 concat!(
1674 "typedef int __attribute__((vector_size(16))) v4si;\n",
1675 "typedef unsigned __attribute__((vector_size(16))) v4ui;\n",
1676 "void write(v4si *out, v4ui a, v4si b, int n) {\n",
1677 " v4si v = { 1, 2, 3, 4 };\n",
1678 " v[0] = n;\n",
1679 " v[1] += n;\n",
1680 " v[2]++;\n",
1681 " *&v[3] = n;\n",
1682 " v4ui shifted = a >> b;\n",
1684 " shifted <<= b;\n",
1685 " *out = v + (v4si)shifted + (1 << b);\n",
1688 "}\n",
1689 "void refused(const v4si c) {\n",
1692 " c[0] = 1;\n",
1693 "}\n",
1694 ),
1695 );
1696 assert_eq!(result.messages.len(), 1, "{:?}", result.messages);
1697 assert!(result.messages[0].contains("assignment of read-only"), "{:?}", result.messages);
1698 }
1699
1700 #[test]
1707 fn a_record_that_asks_for_the_other_byte_order_is_refused_rather_than_laid_out_in_this_one() {
1708 let opts = options();
1709 let big = "struct s { int i; } __attribute__((scalar_storage_order(\"big-endian\")));\n";
1710 assert_eq!(
1711 run(&opts, big).messages,
1712 ["/main.c:1:36: error: 'scalar_storage_order' is not implemented yet [E0688]\n\
1713 /main.c:1:36: note: every scalar in this record would be read in the wrong byte \
1714 order"]
1715 );
1716
1717 let armoured =
1718 "struct s { int i; } __attribute__((__scalar_storage_order__(\"little-endian\")));\n";
1719 let messages = run(&opts, armoured).messages;
1720 assert!(messages[0].contains("[E0688]"), "{messages:?}");
1721
1722 let front = "struct __attribute__((scalar_storage_order(\"big-endian\"))) s { int i; };\n";
1725 assert!(run(&opts, front).messages[0].contains("[E0688]"), "{front}");
1726 let standard = "struct s { int i; } [[gnu::scalar_storage_order(\"big-endian\")]];\n";
1727 assert!(run(&opts, standard).messages[0].contains("[E0688]"), "{standard}");
1728 }
1729
1730 #[test]
1740 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
1741 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
1743 assert_eq!(
1744 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
1745 1
1746 );
1747 assert_eq!(
1748 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
1749 1
1750 );
1751 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
1752 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
1754 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
1755 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
1757 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
1758 }
1759
1760 fn bit_field_byte(record: &str) -> u64 {
1762 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
1763 let body = body(&source);
1764 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
1765 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
1766 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
1767 }
1768
1769 #[test]
1775 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
1776 tast(concat!(
1777 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
1778 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
1779 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
1780 "struct b { char c; __attribute__((packed)) int i; };\n",
1781 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
1782 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
1783 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
1784 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
1785 ));
1786 }
1787
1788 #[test]
1794 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
1795 tast(concat!(
1796 "#pragma pack(1)\n",
1797 "struct A { char c; int i; };\n",
1798 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1799 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1800 "#pragma pack()\n",
1801 "struct B { char c; int i; };\n",
1802 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
1803 "#pragma pack(2)\n",
1804 "struct C { char c; int i; double d; };\n",
1805 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
1806 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
1807 "struct K { char c; int i __attribute__((aligned(8))); };\n",
1809 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
1810 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
1811 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
1813 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
1814 "#pragma pack()\n",
1815 "#pragma pack(push, 1)\n",
1816 "struct D { char c; short s; };\n",
1817 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
1818 "#pragma pack(pop)\n",
1819 "struct E { char c; short s; };\n",
1820 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
1821 "struct H { char c;\n",
1823 "#pragma pack(1)\n",
1824 " int i; };\n",
1825 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
1826 "#pragma pack(1)\n",
1827 "struct I { char c;\n",
1828 "#pragma pack()\n",
1829 " int i; };\n",
1830 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1831 "#pragma pack()\n",
1832 "#pragma pack(push, 8)\n",
1834 "#pragma pack(push, 1)\n",
1835 "struct P { char c; int i; };\n",
1836 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
1837 "#pragma pack(pop)\n",
1838 "struct Q { char c; int i; };\n",
1839 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
1840 "#pragma pack(pop)\n",
1841 "#pragma pack(16)\n",
1843 "struct R { char c; int i; };\n",
1844 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
1845 "#pragma pack()\n",
1846 "#pragma pack(1)\n",
1847 "struct S { char c; int i : 5; int j : 20; };\n",
1848 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
1849 "union T { char c; int i; };\n",
1850 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
1851 "#pragma pack()\n",
1852 ));
1853 }
1854
1855 #[test]
1859 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
1860 let result = run(
1861 &options(),
1862 concat!(
1863 "#pragma pack 4\n",
1864 "#pragma pack(pop)\n",
1865 "#pragma pack(3)\n",
1866 "#pragma pack(1) junk\n",
1867 "#pragma pack(push, 1\n",
1868 "#pragma pack(x)\n",
1869 "#pragma pack(0)\n",
1872 "#pragma pack(push)\n",
1873 "struct s { char c; int i; };\n",
1874 "#pragma pack(pop)\n",
1875 "#pragma pack(pop, foo)\n",
1876 ),
1877 );
1878 let expected = [
1879 "missing `(` after `#pragma pack` - ignored",
1880 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
1881 "alignment must be a small power of two, not 3",
1882 "junk at end of `#pragma pack`",
1883 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
1884 "unknown action `x` for `#pragma pack` - ignored",
1885 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
1886 ];
1887 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
1888 for (message, want) in result.messages.iter().zip(expected) {
1889 assert!(message.contains(want), "expected {want:?} in {message:?}");
1890 }
1891 }
1892
1893 #[test]
1897 fn the_wide_integer_answers_to_all_three_of_its_names() {
1898 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
1899 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
1900 assert!(text.contains("decl #1 b : __int128"), "{text}");
1901 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
1902 }
1903
1904 #[test]
1905 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
1906 let text = tast("long f(int a, long b) { return a + b; }\n");
1910 assert!(text.contains("convert arithmetic"), "{text}");
1911 }
1912
1913 #[test]
1914 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
1915 for source in [
1916 "#error stop\n",
1917 "int f(void) { return 1 + ; }\n",
1918 "int f(void) { return undeclared; }\n",
1919 ] {
1920 let result = run(&options(), source);
1921 assert!(result.failed(), "expected this to fail:\n{source}");
1922 assert!(
1923 result.text().is_empty(),
1924 "a file that did not compile wrote a tree:\n{source}"
1925 );
1926 }
1927 }
1928
1929 #[test]
1930 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
1931 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
1935 assert_eq!(result.errors, 1, "{:?}", result.messages);
1936 }
1937
1938 #[test]
1939 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
1940 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
1944 assert_eq!(result.errors, 1, "{:?}", result.messages);
1945 }
1946
1947 #[test]
1948 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
1949 let source = "int f(void) { char c = 300; return c; }\n";
1950 let plain = run(&options(), source);
1951 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
1952 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
1953 assert!(!plain.text().is_empty(), "a warning is not a reason to write nothing");
1954
1955 let mut opts = options();
1956 opts.warnings_are_errors = true;
1957 let strict = run(&opts, source);
1958 assert!(strict.failed());
1959 assert!(strict.text().is_empty(), "and under -Werror it is a reason to write nothing");
1960 for message in &strict.messages {
1961 assert!(!message.contains("warning:"), "{message}");
1962 }
1963 }
1964
1965 #[test]
1966 fn w_drops_the_warning_before_werror_can_promote_it() {
1967 let source = "int f(void) { char c = 300; return c; }\n";
1968 let mut opts = options();
1969 opts.warnings = false;
1970 let quiet = run(&opts, source);
1971 assert_eq!(quiet.messages, Vec::<String>::new());
1972 assert_eq!(quiet.errors, 0);
1973 assert!(!quiet.text().is_empty(), "and the file still compiles");
1974
1975 opts.warnings_are_errors = true;
1978 let both = run(&opts, source);
1979 assert_eq!(both.messages, Vec::<String>::new());
1980 assert!(!both.failed(), "-w -Werror is not an error about a warning nobody saw");
1981 }
1982
1983 #[test]
1984 fn the_dialect_reaches_the_keywords_and_the_checking() {
1985 let source = "typeof(1) x;\n";
1988 let mut opts = options();
1989 opts.std = Std::C23;
1990 opts.gnu_extensions = false;
1991 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
1992
1993 opts.std = Std::C17;
1994 assert!(run(&opts, source).failed());
1995 }
1996
1997 #[test]
1998 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
1999 let mut opts = options();
2000 opts.emit = EmitKind::Object;
2001 let result = run(&opts, "int x = 1;\n");
2002 assert!(!result.failed(), "{:?}", result.messages);
2003 assert!(result.text().is_empty());
2004 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
2007 }
2008
2009 fn mir(source: &str) -> String {
2011 let mut opts = options();
2012 opts.emit = EmitKind::MirFinal;
2013 let result = run(&opts, source);
2014 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2015 result.text().to_owned()
2016 }
2017
2018 #[test]
2024 fn a_function_goes_from_c_to_instructions_with_real_registers_in_them() {
2025 let text = mir("int add(int a, int b) { return a + b; }\n");
2026 assert!(text.starts_with("mfunc @add {"), "{text}");
2027 assert!(text.contains("x64.add_rr_32"), "{text}");
2028 assert!(text.contains("x64.ret"), "{text}");
2029 assert!(!text.contains('%'), "{text}");
2032 }
2033
2034 #[test]
2036 fn a_function_with_no_body_produces_no_machine_function() {
2037 let text = mir("int g(int);\nint f(int a) { return g(a); }\n");
2038 assert_eq!(text.matches("mfunc @").count(), 1, "{text}");
2039 assert!(text.contains("mfunc @f {"), "{text}");
2040 assert!(text.contains("x64.call"), "{text}");
2041 }
2042
2043 #[test]
2045 fn every_definition_in_the_file_is_generated_and_they_keep_their_order() {
2046 let text = mir("int a(int x) { return x; }\nint b(int x) { return x; }\n");
2047 let first = text.find("mfunc @a").expect("the first function");
2048 let second = text.find("mfunc @b").expect("the second function");
2049 assert!(first < second, "{text}");
2050 }
2051
2052 #[test]
2054 fn the_target_decides_which_convention_the_generated_code_follows() {
2055 let mut opts = options();
2056 opts.emit = EmitKind::MirFinal;
2057 let linux = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
2058 assert!(linux.contains("$rdi"), "{linux}");
2059
2060 opts.target = "x86_64-pc-windows-msvc".parse::<Triple>().unwrap();
2061 let windows = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
2062 assert!(windows.contains("$rcx"), "{windows}");
2063 assert!(!windows.contains("$rdi"), "{windows}");
2064 }
2065
2066 #[test]
2068 fn a_target_this_has_no_back_end_for_is_reported_rather_than_generated() {
2069 let mut opts = options();
2070 opts.emit = EmitKind::MirFinal;
2071 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
2072 let result = run(&opts, "int f(int a) { return a; }\n");
2073 assert!(result.failed());
2074 assert!(result.messages[0].contains("no back end for aarch64"), "{:?}", result.messages);
2075 assert!(result.text().is_empty());
2076 }
2077
2078 #[test]
2085 fn a_construct_the_back_end_cannot_reach_yet_is_reported_against_its_function() {
2086 let mut opts = options();
2087 opts.emit = EmitKind::MirFinal;
2088 let source = "void a(int n) { int v[n] __attribute__((aligned(32))); v[0] = 1; }\n\
2089 void b(int n) { int v[n] __attribute__((aligned(32))); v[0] = 1; }\n";
2090 let result = run(&opts, source);
2091 assert!(result.failed());
2092 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
2093 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
2094 assert!(result.messages[0].contains("wants more alignment"), "{:?}", result);
2095 assert!(result.messages[1].contains("cannot generate code for 'b'"), "{:?}", result);
2096 assert!(result.text().is_empty());
2097 }
2098
2099 #[test]
2106 fn a_variable_length_array_is_refused_where_every_page_of_the_frame_is_to_be_touched() {
2107 let mut opts = options();
2108 opts.emit = EmitKind::MirFinal;
2109 let source = "void a(int n) { int v[n]; v[0] = 1; }\n";
2110 assert!(!run(&opts, source).failed(), "it compiles without the flag");
2111
2112 opts.stack_clash = true;
2113 let result = run(&opts, source);
2114 assert!(result.failed());
2115 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
2116 assert!(result.messages[0].contains("a page at a time"), "{:?}", result);
2117 }
2118
2119 #[test]
2133 fn an_opcode_with_no_name_in_the_rule_language_is_named_by_its_own_spelling() {
2134 let mut opts = options();
2135 opts.emit = EmitKind::MirFinal;
2136 let source =
2137 "long double f(int a) {\n __int128 wide = a;\n return (long double) wide;\n}\n";
2138 let result = run(&opts, source);
2139 assert!(result.failed());
2140 assert!(
2141 result.messages[0].contains("no rule lowers a `sext` producing a `i128`"),
2142 "{result:?}"
2143 );
2144 assert!(result.messages[0].contains(":2:"), "the line the widening is on: {result:?}");
2145 assert!(!result.messages[0].contains("this instruction"), "{result:?}");
2146 }
2147
2148 #[test]
2150 fn the_note_on_unfinished_work_points_at_the_issues_rather_than_at_the_plan() {
2151 let mut opts = options();
2152 opts.emit = EmitKind::MirFinal;
2153 let source = "long double f(int a) { __int128 wide = a; return (long double) wide; }\n";
2154 let result = run(&opts, source);
2155 assert!(result.failed());
2156 let note = result.messages.iter().find(|line| line.contains("note:")).expect("a note");
2157 assert!(note.contains("https://github.com/tamnd/rucc/issues"), "{note}");
2158 assert!(!note.contains("spec/17-milestones.md"), "{note}");
2159 }
2160
2161 #[test]
2163 fn the_frame_flags_on_the_command_line_reach_the_generated_frame() {
2164 let source = "int f(int a) { return a; }\n";
2165 assert!(!mir(source).contains("$rbp"), "a leaf needs no frame pointer by default");
2166
2167 let mut opts = options();
2168 opts.emit = EmitKind::MirFinal;
2169 opts.frame_pointer = true;
2170 let kept = run(&opts, source).text().to_owned();
2171 assert!(kept.contains("x64.push_64 $rbp"), "{kept}");
2172 }
2173
2174 fn asm(source: &str) -> String {
2176 let mut opts = options();
2177 opts.emit = EmitKind::Asm;
2178 let result = run(&opts, source);
2179 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2180 result.text().to_owned()
2181 }
2182
2183 #[test]
2190 fn a_function_goes_from_c_to_assembly_an_assembler_would_take() {
2191 let text = asm("int add(int a, int b) { return a + b; }\n");
2192 assert!(text.contains("\t.globl\tadd\n"), "{text}");
2193 assert!(text.contains("\t.type\tadd, @function\n"), "{text}");
2194 assert!(text.contains("\nadd:\n"), "{text}");
2195 assert!(text.contains("\taddl\t"), "{text}");
2196 assert!(text.contains("\tret\n"), "{text}");
2197 assert!(text.contains("\t.size\tadd, .-add\n"), "{text}");
2198 assert!(text.contains(".note.GNU-stack"), "{text}");
2201 }
2202
2203 #[test]
2209 fn a_call_through_a_function_pointer_goes_through_the_register_it_is_in() {
2210 let text = asm("int g(int);\nint f(int (*p)(int), int a) { return p(a) + g(a); }\n");
2211 assert!(text.contains("\tcall\t*%"), "{text}");
2212 assert!(text.contains("\tcall\tg\n"), "{text}");
2213 assert!(text.contains("%rdi"), "{text}");
2217 }
2218
2219 #[test]
2223 fn the_address_of_a_global_is_read_from_the_instruction_pointer() {
2224 let text = asm("extern int counter;\nint f(void) { return counter; }\n");
2225 assert!(text.contains("\tmovl\tcounter(%rip), %eax\n"), "{text}");
2226 }
2227
2228 #[test]
2237 fn a_branch_on_a_comparison_jumps_on_the_opposite_of_what_it_compared() {
2238 let arms = "return 1; return 2;";
2239 let signed = [("==", "jne"), ("!=", "je"), ("<", "jge"), ("<=", "jg"), (">", "jle")];
2240 for (operator, jump) in signed.into_iter().chain([(">=", "jl")]) {
2241 let text = asm(&format!("int f(int a, int b) {{ if (a {operator} b) {arms} }}\n"));
2242 assert!(
2243 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
2244 "{operator}: {text}"
2245 );
2246 assert!(!text.contains("\tset"), "{operator}: {text}");
2247 assert!(!text.contains("\ttest"), "{operator}: {text}");
2248 }
2249 let unsigned = [("<", "jae"), ("<=", "ja"), (">", "jbe"), (">=", "jb")];
2250 for (operator, jump) in unsigned {
2251 let source =
2252 format!("int f(unsigned a, unsigned b) {{ if (a {operator} b) {arms} }}\n");
2253 let text = asm(&source);
2254 assert!(
2255 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
2256 "{operator}: {text}"
2257 );
2258 }
2259
2260 let text = asm("int f(int a) { if (a < 7) return 1; return 2; }\n");
2263 assert!(text.contains("\tcmpl\t$7, %edi\n\tjge\t"), "{text}");
2264 }
2265
2266 #[test]
2272 fn a_comparison_whose_answer_the_program_wanted_still_writes_a_byte() {
2273 let text = asm("int f(int a, int b) { return a < b; }\n");
2274 assert!(text.contains("\tsetl\t"), "{text}");
2275 }
2276
2277 fn optimized(source: &str) -> String {
2279 let mut opts = options();
2280 opts.emit = EmitKind::Asm;
2281 opts.opt_level = rucc_session::OptLevel::O2;
2282 let result = run(&opts, source);
2283 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2284 result.text().to_owned()
2285 }
2286
2287 #[test]
2297 fn a_switch_whose_arms_are_a_function_of_the_label_is_a_range_check_and_arithmetic() {
2298 let arms: String =
2299 (0..16).map(|k| format!("case {k}: return {};", k + 1)).collect::<Vec<_>>().join(" ");
2300 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
2301 assert!(text.contains("\tcmpl\t$15, %edi\n\tja\t"), "{text}");
2302 assert!(text.contains("\taddl\t$1, %edi"), "{text}");
2303 assert_eq!(text.matches("\tcmp").count(), 1, "{text}");
2304 }
2305
2306 #[test]
2313 fn a_dense_switch_whose_arms_are_not_a_line_keeps_its_comparisons() {
2314 let arms: String = (0..16)
2315 .map(|k| format!("case {k}: return {};", if k == 9 { 100 } else { k + 1 }))
2316 .collect::<Vec<_>>()
2317 .join(" ");
2318 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
2319 assert!(text.matches("\tcmp").count() > 1, "{text}");
2320 }
2321
2322 #[test]
2324 fn a_cast_between_a_pointer_and_an_integer_leaves_the_value_where_it_is() {
2325 let text = asm("long f(void *p) { return (long)p; }\n");
2326 for line in text.lines().filter(|line| line.starts_with('\t') && !line.contains('.')) {
2331 let mnemonic = line.split_whitespace().next().unwrap_or("");
2332 assert!(matches!(mnemonic, "movq" | "ret"), "{line} in\n{text}");
2333 }
2334 }
2335
2336 #[test]
2340 fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
2341 let six = "long a, long b, long c, long d, long e, long f";
2342 let text = asm(&format!("long f({six}, long g, long h) {{ return g + h; }}\n"));
2343
2344 assert!(text.contains("\tmovq\t8(%rsp), "), "{text}");
2351 assert!(text.contains("\taddq\t16(%rsp), "), "{text}");
2352
2353 let narrow = asm(&format!("int f({six}, int g) {{ return g; }}\n"));
2357 assert!(narrow.contains("\tmovl\t8(%rsp), "), "{narrow}");
2358 let eight =
2359 "double a, double b, double c, double d, double e, double f, double g, double h";
2360 let float = asm(&format!("double f({eight}, double i) {{ return i; }}\n"));
2361 assert!(float.contains("\tmovsd\t8(%rsp), "), "{float}");
2362 }
2363
2364 #[test]
2367 fn a_call_writes_the_arguments_with_no_register_left_at_the_stack_pointer() {
2368 let six = "1, 2, 3, 4, 5, 6";
2369 let decl = "long g(long, long, long, long, long, long, long, long);\n";
2370 let text = asm(&format!("{decl}long f(void) {{ return g({six}, 7, 8); }}\n"));
2371
2372 assert!(text.contains("\tmovq\t%"), "{text}");
2373 assert!(text.contains(", (%rsp)\n"), "{text}");
2374 assert!(text.contains(", 8(%rsp)\n"), "{text}");
2375 assert!(text.contains("\tsubq\t$"), "{text}");
2377
2378 let narrow = "int g(int, int, int, int, int, int, int);\n";
2380 let text = asm(&format!("{narrow}int f(void) {{ return g({six}, 7); }}\n"));
2381 assert!(text.contains("\tmovl\t%"), "{text}");
2382 assert!(text.contains(", (%rsp)\n"), "{text}");
2383 }
2384
2385 #[test]
2388 fn a_variadic_call_counts_registers_and_not_arguments() {
2389 let nine = "1., 2., 3., 4., 5., 6., 7., 8., 9.";
2390 let decl = "int g(int, ...);\n";
2391 let text = asm(&format!("{decl}int f(void) {{ return g(0, {nine}); }}\n"));
2392
2393 assert!(text.contains("\tmovl\t$8, "), "eight registers, not nine: {text}");
2394 assert!(text.contains("\tmovsd\t%"), "{text}");
2395 assert!(text.contains(", (%rsp)\n"), "{text}");
2396 }
2397
2398 #[test]
2403 fn a_variadic_function_writes_the_argument_registers_it_was_handed_into_its_frame() {
2404 let body =
2405 "__builtin_va_list ap; __builtin_va_start(ap, n); __builtin_va_end(ap); return n;";
2406 let text = asm(&format!("int f(int n, ...) {{ {body} }}\n"));
2407
2408 let stores = |mnemonic: &str| text.matches(&format!("\t{mnemonic}\t%")).count();
2411 assert!(text.contains(", 8(%r"), "the second slot, not the first: {text}");
2412 assert!(!text.contains(", 0(%r"), "{text}");
2413 assert_eq!(stores("movaps"), 8, "every vector register: {text}");
2416 assert_eq!(stores("movsd"), 0, "and the whole of each one: {text}");
2417
2418 assert!(text.contains("\tsubq\t$"), "{text}");
2420 }
2421
2422 #[test]
2425 fn va_start_writes_the_four_fields_the_psabi_describes() {
2426 let start = "__builtin_va_list ap; __builtin_va_start(ap, d);";
2427 let params = "int a, int b, int c, double d";
2428 let text = asm(&format!("int f({params}, ...) {{ {start} return a; }}\n"));
2429
2430 assert!(text.contains(" movl $24, "), "{text}");
2434 assert!(text.contains(" movl $64, "), "{text}");
2435 assert!(text.contains(", 8(%r"), "{text}");
2439 assert!(text.contains(", 16(%r"), "{text}");
2440 let frame: u32 = text
2441 .lines()
2442 .find_map(|line| line.trim().strip_prefix("subq $")?.split(',').next()?.parse().ok())
2443 .expect("a variadic function takes a frame for the save area");
2444 let above = |line: &str| {
2445 let at: u32 = line.trim().strip_prefix("leaq ")?.split('(').next()?.parse().ok()?;
2446 Some(at > frame)
2447 };
2448 assert!(text.lines().filter_map(above).any(|it| it), "{frame}: {text}");
2449 }
2450
2451 #[test]
2454 fn va_arg_branches_on_whether_the_argument_is_still_in_the_save_area() {
2455 let read = "__builtin_va_list ap; __builtin_va_start(ap, n);";
2456 let ints = format!("int f(int n, ...) {{ {read} return __builtin_va_arg(ap, int); }}\n");
2457 let text = asm(&ints);
2458
2459 assert!(text.contains("$40, "), "{text}");
2462 assert!(text.contains(" cmpl "), "{text}");
2463 assert!(text.contains(" ja "), "{text}");
2467
2468 let arg = "__builtin_va_arg(ap, double)";
2469 let text = asm(&format!("double f(int n, ...) {{ {read} return {arg}; }}\n"));
2470 assert!(text.contains("$160, "), "the last vector slot: {text}");
2471 }
2472
2473 #[test]
2476 fn a_structure_assignment_is_a_move_for_each_word_of_it() {
2477 let decl = "struct pair { long a, b; };\n";
2478 let body = "struct pair p = *q; return p.a + p.b;";
2479 let text = asm(&format!("{decl}long f(struct pair *q) {{ {body} }}\n"));
2480
2481 assert!(!text.contains("memcpy"), "nothing calls the library: {text}");
2482 assert!(!text.contains("\tcall"), "{text}");
2483 assert!(text.matches("\tmovq\t").count() >= 4, "two words each way: {text}");
2485 }
2486
2487 #[test]
2490 fn how_wide_a_word_of_a_copy_is_follows_the_alignment() {
2491 let decl = "struct bytes { char a[8]; };\n";
2492 let body = "struct bytes p = *q; return p.a[0];";
2493 let text = asm(&format!("{decl}int f(struct bytes *q) {{ {body} }}\n"));
2494
2495 assert!(text.matches("\tmovb\t").count() >= 16, "a byte at a time: {text}");
2497 }
2498
2499 #[test]
2502 fn the_part_of_an_initialiser_that_names_nothing_is_stored_as_zero() {
2503 let decl = "struct wide { long a, b, c; };\n";
2504 let text = asm(&format!("{decl}long f(void) {{ struct wide w = {{ 7 }}; return w.c; }}\n"));
2505
2506 assert!(!text.contains("memset"), "nothing calls the library: {text}");
2507 assert!(text.contains("\tmovq\t$0, ") || text.contains("$0, %"), "the zero: {text}");
2508 }
2509
2510 #[test]
2513 fn a_copy_too_large_to_unroll_calls_the_runtime() {
2514 let decl = "struct huge { char a[4096]; };\n";
2515 let mut opts = options();
2516 opts.emit = EmitKind::Asm;
2517 let source = format!("{decl}void f(struct huge *p, struct huge *q) {{ *p = *q; }}\n");
2518 let result = run(&opts, &source);
2519 assert!(!result.failed(), "{:?}", result.messages);
2520 let text = result.text();
2521 assert!(text.contains("call") && text.contains("memcpy"), "{text}");
2522 assert!(text.contains("4096"), "the size travels: {text}");
2525 }
2526
2527 #[test]
2530 fn a_realigned_frame_reads_them_through_the_frame_pointer() {
2531 let six = "long a, long b, long c, long d, long e, long f";
2532 let body = "_Alignas(32) long wide[4]; wide[0] = g; return wide[0];";
2533 let text = asm(&format!("long f({six}, long g) {{ {body} }}\n"));
2534
2535 assert!(text.contains("\tandq\t$-32, %rsp"), "{text}");
2539 assert!(text.contains("\tmovq\t16(%rbp), "), "{text}");
2540 assert!(!text.contains("\tmovq\t16(%rsp), "), "{text}");
2541 }
2542
2543 #[test]
2545 fn the_target_decides_how_the_assembly_is_spelled() {
2546 let mut opts = options();
2547 opts.emit = EmitKind::Asm;
2548 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2549 let text = run(&opts, "int f(void) { return 0; }\n").text().to_owned();
2550 assert!(text.contains("__TEXT,__text"), "{text}");
2551 assert!(text.contains("\n_f:\n"), "{text}");
2552 assert!(!text.contains(".note.GNU-stack"), "{text}");
2553 }
2554
2555 fn obj(source: &str) -> Vec<u8> {
2557 let mut opts = options();
2558 opts.emit = EmitKind::Object;
2559 let result = run(&opts, source);
2560 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2561 match result.artifact {
2562 Artifact::Object { bytes, .. } => bytes,
2563 other => panic!("expected an object, got {other:?}"),
2564 }
2565 }
2566
2567 #[test]
2573 fn a_function_goes_from_c_to_an_object_a_linker_would_take() {
2574 let bytes = obj("int add(int a, int b) { return a + b; }\n");
2575 assert_eq!(&bytes[..4], b"\x7fELF", "an object file starts by saying it is one");
2576 let text = asm("int add(int a, int b) { return a + b; }\n");
2577 assert!(
2578 text.contains("\taddl\t"),
2579 "and the listing of it is the same instructions:\n{text}"
2580 );
2581 }
2582
2583 #[test]
2585 fn a_variable_goes_from_c_to_the_section_it_belongs_in() {
2586 let text = asm("int counter = 42;\nstatic int hidden;\nconst int fixed = 7;\n");
2587 assert!(text.contains("\t.data\n\t.globl\tcounter\n"), "{text}");
2588 assert!(text.contains("\ncounter:\n\t.long\t42\n"), "{text}");
2589 assert!(text.contains("\t.size\tcounter, .-counter\n"), "{text}");
2590 assert!(text.contains("\t.bss\n\t.p2align\t2\n"), "{text}");
2593 assert!(text.contains("\nhidden:\n\t.space\t4\n"), "{text}");
2594 assert!(!text.contains(".globl\thidden"), "{text}");
2595 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2598 }
2599
2600 #[test]
2607 fn a_bit_field_initializer_writes_every_byte_of_the_value_and_not_only_the_ones_that_are_set() {
2608 let text = asm("struct s { unsigned f : 20; } x = { 0x12300 };\n");
2609 assert!(text.contains("\t.data\n"), "there is something to write: {text}");
2610 assert!(text.contains("\nx:\n\t.ascii\t\"\\000#\\001\"\n"), "and it is the value: {text}");
2611
2612 let text = asm("struct s { unsigned a : 8; unsigned b : 8; } x = { 0, 3 };\n");
2615 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\003\"\n"), "{text}");
2616
2617 let text = asm("struct s { unsigned long long f : 40; } x = { 0x100000 };\n");
2620 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\000\\020\"\n\t.space\t5\n"), "{text}");
2621
2622 let text = asm("struct s { unsigned f : 20; } x = { 0 };\n");
2624 assert!(text.contains("\t.bss\n"), "an object of zeroes is zeroes: {text}");
2625 assert!(text.contains("\nx:\n\t.space\t4\n"), "{text}");
2626 }
2627
2628 #[test]
2630 fn a_string_literal_is_a_variable_with_a_name_no_program_could_write() {
2631 let text = asm("const char *f(void) { return \"hi\"; }\n");
2632 assert!(text.contains("\t.ascii\t\"hi\\000\"\n"), "{text}");
2633 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2634 let label = text
2635 .lines()
2636 .find(|line| line.starts_with(".Lstr"))
2637 .unwrap_or_else(|| panic!("a label for the literal in\n{text}"));
2638 assert!(!text.contains(&format!(".globl\t{}", label.trim_end_matches(':'))), "{text}");
2639 }
2640
2641 #[test]
2643 fn an_address_in_an_initializer_is_left_to_the_linker() {
2644 let source = "int counter;\nint *p = &counter;\n";
2645 let text = asm(source);
2646 assert!(text.contains("\np:\n\t.quad\tcounter\n"), "{text}");
2647 let bytes = obj(source);
2650 assert!(bytes.windows(8).any(|w| w == b"counter\0"), "the object has to name it");
2651 }
2652
2653 #[test]
2662 fn a_constant_holding_an_address_goes_in_the_section_the_loader_may_write_once() {
2663 let text = asm("static void a(void) {}\nstatic void b(void) {}\n\
2666 struct m { void (*x)(void); void (*y)(void); };\n\
2667 const struct m t = { a, b };\n");
2668 assert!(text.contains("\t.section\t.data.rel.ro.local,\"aw\",@progbits\n"), "{text}");
2669 assert!(text.contains("\nt:\n\t.quad\ta\n\t.quad\tb\n"), "{text}");
2670
2671 let text =
2674 asm("void a(void);\nstruct m { void (*x)(void); };\nconst struct m t = { a };\n");
2675 assert!(text.contains("\t.section\t.data.rel.ro,\"aw\",@progbits\n"), "{text}");
2676
2677 let text = asm("const int fixed = 7;\n");
2679 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2680 }
2681
2682 #[test]
2689 fn a_thread_local_variable_is_storage_a_thread_gets_a_copy_of_and_an_offset_into_it() {
2690 let text = asm("_Thread_local int x = 1;\nint read(void) { return x; }\n");
2691 assert!(text.contains("\t.section\t.tdata,\"awT\",@progbits\n"), "{text}");
2694 assert!(text.contains("\t.type\tx, @tls_object\n"), "{text}");
2695 assert!(text.contains("x@GOTTPOFF(%rip)"), "{text}");
2698 assert!(text.contains("%fs:0"), "{text}");
2699 }
2700
2701 #[test]
2707 fn the_address_of_this_thread_s_own_storage_is_read_out_of_the_segment_register() {
2708 let text = asm("void *here(void) { return __builtin_thread_pointer(); }\n");
2709 assert!(text.contains("movq\t%fs:0, "), "{text}");
2710 assert!(!text.contains("GOTTPOFF"), "{text}");
2712 }
2713
2714 #[test]
2725 fn a_prefetch_is_one_of_four_instructions_and_the_locality_is_what_picks() {
2726 for (locality, wanted) in
2727 [(0, "prefetchnta"), (1, "prefetcht2"), (2, "prefetcht1"), (3, "prefetcht0")]
2728 {
2729 let source =
2730 format!("void warm(void *p) {{ __builtin_prefetch(p, 0, {locality}); }}\n");
2731 let text = asm(&source);
2732 assert!(text.contains(&format!("\t{wanted}\t")), "locality {locality}: {text}");
2733 }
2734 let text = asm("void warm(void *p) { __builtin_prefetch(p); }\n");
2736 assert!(text.contains("\tprefetcht0\t"), "{text}");
2737 let text = asm("void warm(void *p) { __builtin_prefetch(p, 1); }\n");
2740 assert!(text.contains("\tprefetcht0\t"), "{text}");
2741 assert!(!text.contains("prefetchw"), "{text}");
2742 }
2743
2744 #[test]
2755 fn a_trap_is_the_instruction_the_machine_has_no_meaning_for() {
2756 let text = asm("void stop(void) { __builtin_trap(); }\n");
2757 assert!(text.contains("\tud2\n"), "{text}");
2758 assert!(!text.contains("\tcall"), "a stop is not a call to anything: {text}");
2759
2760 let text = asm("int stop(int a) { __builtin_trap(); return a + 1; }\n");
2761 assert!(text.contains("\tud2\n"), "{text}");
2762 assert!(text.contains("\taddl\t"), "the block goes on after a stop: {text}");
2763 }
2764
2765 #[test]
2777 fn assume_aligned_is_its_first_argument_and_keeps_the_rest() {
2778 let text = asm("void *aligned(char *p) { return __builtin_assume_aligned(p, 16); }\n");
2779 assert!(!text.contains("assume_aligned"), "{text}");
2780 assert!(!text.contains("\tcall"), "nothing is called for an alignment fact: {text}");
2781
2782 let source = "unsigned long width(void);\n\
2783 void *aligned(char *p) { return __builtin_assume_aligned(p, width()); }\n";
2784 let text = asm(source);
2785 assert!(!text.contains("assume_aligned"), "{text}");
2786 assert!(text.contains("width"), "the argument that is not the answer still runs: {text}");
2787 }
2788
2789 #[test]
2799 fn the_frame_address_is_the_frame_pointer_after_walking_that_many_links() {
2800 let text = asm("void *here(void) { return __builtin_frame_address(0); }\n");
2801 assert!(text.contains("pushq\t%rbp"), "a function that asks keeps a frame pointer: {text}");
2802 assert!(text.contains("movq\t%rbp, %rax"), "{text}");
2803 assert!(!text.contains("\tcall"), "a frame address is not a call to anything: {text}");
2804
2805 let walk = |depth: u32| {
2806 let source = format!("void *up(void) {{ return __builtin_frame_address({depth}); }}\n");
2807 asm(&source).matches("movq\t(%r").count()
2808 };
2809 assert_eq!(walk(1), 1, "one link is one load");
2810 assert_eq!(walk(3), 3, "three links are three loads");
2811 }
2812
2813 #[test]
2823 fn the_return_address_is_one_word_above_the_frame_the_walk_ended_at() {
2824 let text = asm("void *back(void) { return __builtin_return_address(0); }\n");
2825 assert!(text.contains("pushq\t%rbp"), "a function that asks keeps a frame pointer: {text}");
2826 assert!(text.contains("movq\t8(%rbp), %rax"), "{text}");
2827 assert!(!text.contains("\tcall"), "a return address is not a call to anything: {text}");
2828
2829 let text = asm("void *back(void) { return __builtin_return_address(2); }\n");
2830 assert_eq!(text.matches("movq\t(%r").count(), 2, "two links are two loads: {text}");
2831 assert!(text.contains("movq\t8(%r"), "and the answer is above the last of them: {text}");
2832 }
2833
2834 #[test]
2845 fn a_depth_that_is_not_a_small_constant_is_refused() {
2846 let mut opts = options();
2847 opts.emit = EmitKind::Ir;
2848 for source in [
2849 "void *up(int n) { return __builtin_return_address(n); }\n",
2850 "void *up(void) { return __builtin_frame_address(1000); }\n",
2851 ] {
2852 let messages = run(&opts, source).messages;
2853 let named = messages.iter().any(|m| m.contains("E0705"));
2854 assert!(named, "expected a refusal in {messages:?}");
2855 }
2856 }
2857
2858 #[test]
2870 fn an_alloca_takes_the_bytes_off_the_stack_pointer_and_answers_where_they_are() {
2871 let text =
2872 asm("void use(void *p); void f(unsigned long n) { use(__builtin_alloca(n)); }\n");
2873 assert!(text.contains("andq\t$-16"), "the size is rounded up to sixteen: {text}");
2874 assert!(text.contains("subq\t%rdi, %rsp"), "and taken off the stack pointer: {text}");
2875 assert_eq!(text.matches("\tcall").count(), 1, "the only call is the one written: {text}");
2876
2877 let plain = concat!(
2880 "extern void *alloca(__SIZE_TYPE__);\n",
2881 "void use(void *p);\n",
2882 "void f(unsigned long n) { use(alloca(n)); }\n",
2883 );
2884 let text = asm(plain);
2885 assert!(text.contains("subq\t%rdi, %rsp"), "the plain name is the same bytes: {text}");
2886 assert_eq!(text.matches("\tcall").count(), 1, "and is not a call either: {text}");
2887
2888 let own = concat!(
2891 "static void *alloca(unsigned long n) { return 0; }\n",
2892 "void *f(unsigned long n) { return alloca(n); }\n",
2893 );
2894 assert!(asm(own).contains("\tcall"), "a name the program took back is a call");
2895 }
2896
2897 #[test]
2907 fn the_bytes_an_alloca_took_are_still_there_at_the_end_of_the_block_that_took_them() {
2908 let inner = "{ use(__builtin_alloca(n)); }";
2909 for body in [inner.to_owned(), format!("int a[n]; {inner} use(a);")] {
2910 let source = format!("void use(void *p);\nvoid f(unsigned long n) {{ {body} }}\n");
2911 let text = asm(&source);
2912 for line in text.lines().filter(|line| line.trim_end().ends_with(", %rsp")) {
2916 let taking = line.contains("subq");
2917 let leaving = line.contains("%rbp");
2918 assert!(taking || leaving, "nothing puts the stack back: {line} in {text}");
2919 }
2920 }
2921 }
2922
2923 #[test]
2925 fn the_object_and_the_listing_are_two_spellings_of_one_compilation() {
2926 let source = "int callee(void); int g(void) { return callee(); }\n";
2930 let bytes = obj(source);
2931 assert!(
2932 bytes.windows(7).any(|w| w == b"callee\0"),
2933 "the object has to name the callee for the linker to find it"
2934 );
2935 let text = asm(source);
2936 assert!(text.contains("\tcall\tcallee\n"), "{text}");
2937 }
2938
2939 #[test]
2945 fn compiling_for_an_executable_produces_an_object_and_not_a_dump() {
2946 let mut opts = options();
2947 opts.emit = EmitKind::Executable;
2949 let result = run(&opts, "int main(void) { return 0; }\n");
2950 assert_eq!(result.messages, Vec::<String>::new());
2951 match result.artifact {
2952 Artifact::Object { bytes, .. } => assert_eq!(&bytes[..4], b"\x7fELF"),
2953 other => panic!("expected an object, got {other:?}"),
2954 }
2955 }
2956
2957 #[test]
2959 fn a_platform_with_no_object_writer_is_said_so_rather_than_written_as_elf() {
2960 let mut opts = options();
2961 opts.emit = EmitKind::Object;
2962 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2963 let result = run(&opts, "int f(void) { return 0; }\n");
2964 assert!(result.failed(), "an object nobody can read is worse than a message");
2965 assert!(
2966 result.messages.iter().any(|m| m.contains("no object writer")),
2967 "{:?}",
2968 result.messages
2969 );
2970 }
2971
2972 fn ir(source: &str) -> String {
2974 let mut opts = options();
2975 opts.emit = EmitKind::Ir;
2976 let result = run(&opts, source);
2977 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2978 result.text().to_owned()
2979 }
2980
2981 fn errors(source: &str) -> Vec<String> {
2983 let mut opts = options();
2984 opts.emit = EmitKind::Ir;
2985 let result = run(&opts, source);
2986 assert!(result.failed(), "expected this to be refused:\n{source}");
2987 result.messages
2988 }
2989
2990 fn body(source: &str) -> String {
2992 let text = ir(source);
2993 let (_, rest) = text.split_once("{\n").expect("a function definition");
2994 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
2995 body.to_owned()
2996 }
2997
2998 #[test]
3006 fn gnu89_inline_is_what_decides_whether_a_bare_inline_definition_reaches_the_module() {
3007 let source = "inline int f(int x) { return x + 1; }\n";
3008 let with = |flag: bool| {
3009 let mut opts = options();
3010 opts.emit = EmitKind::Ir;
3011 opts.gnu89_inline = flag;
3012 let result = run(&opts, source);
3013 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
3014 result.text().to_owned()
3015 };
3016
3017 assert!(!with(false).contains("block0"), "no body: {}", with(false));
3020
3021 assert!(with(true).contains("block0"), "a body: {}", with(true));
3024 }
3025
3026 #[test]
3033 fn an_access_through_a_type_names_the_type_it_went_through() {
3034 let source = "\
3035struct s { int a; float b; };\n\
3036union u { int i; float f; };\n\
3037int scalar(int *p) { return *p; }\n\
3038float member(struct s *p) { p->a = 1; return p->b; }\n\
3039int element(int *a, long i) { return a[i]; }\n\
3040float through_a_union(union u *p) { p->i = 1; return p->f; }\n";
3041 let text = ir(source);
3042 assert!(text.contains(r#"!0 = tbaa "char""#), "the root: {text}");
3043 assert!(text.contains(r#"tbaa "int", parent !0"#), "int under it: {text}");
3044 assert!(text.contains(r#"tbaa "float", parent !0"#), "float under it: {text}");
3045 let named = text.lines().filter(|line| line.contains(", tbaa !")).count();
3048 assert_eq!(named, 6, "six accesses: {text}");
3049 }
3050
3051 #[test]
3058 fn turning_strict_aliasing_off_leaves_the_type_off_every_access() {
3059 let source = "int punned(float *f, int *i) { *i = 1; *f = 2.0f; return *i; }\n";
3060 let mut opts = options();
3061 opts.emit = EmitKind::Ir;
3062 opts.strict_aliasing = false;
3063 let result = run(&opts, source);
3064 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
3065 let text = result.text().to_owned();
3066 assert!(!text.contains("tbaa"), "not even the root: {text}");
3067 }
3068
3069 #[test]
3077 fn a_bare_return_from_a_function_that_promised_a_value_gives_back_a_zero() {
3078 let mut opts = options();
3079 opts.emit = EmitKind::Ir;
3080 opts.std = Std::C89;
3081 let compiled = |source: &str| {
3082 let result = run(&opts, source);
3083 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
3084 result.text().to_owned()
3085 };
3086
3087 let text = compiled("int f(int x) { if (x) return; return 3; }\n");
3088 assert!(text.contains("iconst.i32 0\n return"), "zero goes back: {text}");
3089 assert!(!text.contains("unreachable"), "the branch that reached it is kept: {text}");
3090
3091 let text = compiled("double f(int x) { if (x) return; return 1.0; }\n");
3093 assert!(text.contains("fconst.f64 0x0\n return"), "a float zero goes back: {text}");
3094 }
3095
3096 #[test]
3104 fn a_call_to_a_name_nothing_declared_declares_it_as_c89_said_to() {
3105 let mut opts = options();
3106 opts.emit = EmitKind::Ir;
3107 opts.std = Std::C89;
3108 let compiled = |source: &str| {
3109 let result = run(&opts, source);
3110 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
3111 result.text().to_owned()
3112 };
3113
3114 let text = compiled("int f(void) { return g(); }\n");
3116 assert!(text.contains("call @g"), "the call is to the name that was written: {text}");
3117 assert!(text.contains("i32"), "and it gives back an int: {text}");
3118
3119 let text = compiled("int f(char c) { return g(c); }\n");
3122 assert!(text.contains("sext.i32"), "the argument is promoted: {text}");
3123
3124 let mut opts = options();
3127 opts.std = Std::C89;
3128 let said = run(&opts, "int f(void) { return h; }\n").messages.join("\n");
3129 assert!(said.contains("'h' undeclared"), "not a call, so not declared: {said}");
3130 }
3131
3132 #[test]
3142 fn a_name_called_before_it_is_defined_still_gets_its_definition() {
3143 let mut opts = options();
3144 opts.emit = EmitKind::Ir;
3145 opts.std = Std::C89;
3146 let text = run(&opts, "int f(void) { return dummy(); }\ndummy () { return 7; }\n")
3147 .text()
3148 .to_owned();
3149 assert!(text.contains("func @f()"), "the caller is there: {text}");
3150 assert!(text.contains("func @dummy"), "and so is what it calls: {text}");
3151 assert!(text.contains("iconst.i32 7"), "with the body it was given: {text}");
3152 }
3153
3154 #[test]
3162 fn an_old_style_parameter_is_converted_from_what_the_call_promoted_it_to() {
3163 let mut opts = options();
3164 opts.emit = EmitKind::Ir;
3165 opts.std = Std::C89;
3166 let compiled = |source: &str| run(&opts, source).text().to_owned();
3167
3168 let text = compiled("f (c) unsigned char c; { return c; }\n");
3169 assert!(text.contains("func @f(i32"), "an int arrives: {text}");
3170 assert!(text.contains("trunc.i8"), "and is cut down to what was declared: {text}");
3171 assert!(text.contains("zext.i32"), "then read back unsigned: {text}");
3172
3173 let text = compiled("f (s) short s; { return s; }\n");
3175 assert!(text.contains("trunc.i16"), "cut down: {text}");
3176 assert!(text.contains("sext.i32"), "and read back signed: {text}");
3177
3178 let text = compiled("f (x) float x; { return x * 2; }\n");
3181 assert!(text.contains("func @f(f64"), "a double arrives: {text}");
3182 assert!(text.contains("fptrunc.f32"), "and is narrowed to the float: {text}");
3183
3184 let text = compiled("int f(unsigned char c) { return c; }\n");
3187 assert!(text.contains("func @f(i8)"), "the declared type arrives: {text}");
3188 assert!(!text.contains("trunc"), "so there is nothing to cut down: {text}");
3189 }
3190
3191 #[test]
3200 fn the_rules_gcc_promoted_are_decided_by_the_dialect_and_by_fpermissive() {
3201 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
3203 let cases = [
3204 ("static counted;\n", ["", "error", "warning", "error"]),
3205 ("int f(void) { return g(); }\n", ["", "error", "warning", "error"]),
3206 ("int f(x) { return x; }\n", ["", "error", "warning", "error"]),
3207 ("int *p;\nvoid h(void) { p = 1; }\n", ["warning", "error", "warning", "error"]),
3208 (
3209 "char *q;\nint *r;\nvoid k(void) { r = q; }\n",
3210 ["warning", "error", "warning", "error"],
3211 ),
3212 ("int f(void) { return; }\n", ["", "error", "warning", "error"]),
3213 ("void g(void) { return 1; }\n", ["warning", "error", "warning", "error"]),
3214 ];
3215
3216 for (source, wanted) in cases {
3217 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
3218 let mut opts = options();
3219 opts.std = std;
3220 opts.permissive = permissive;
3221 let said = run(&opts, source).messages.join("\n");
3222 let severity = if said.contains(": error: ") {
3223 "error"
3224 } else if said.contains(": warning: ") {
3225 "warning"
3226 } else {
3227 ""
3228 };
3229 let how = if permissive { " -fpermissive" } else { "" };
3230 assert_eq!(
3231 severity,
3232 wanted,
3233 "under -std={}{how}, {source} was answered with `{said}`",
3234 std.as_str()
3235 );
3236 if wanted.is_empty() {
3237 assert!(said.is_empty(), "nothing to say, but said `{said}`");
3238 }
3239 }
3240 }
3241 }
3242
3243 #[test]
3252 fn the_three_variadic_builtins_answer_a_bad_list_the_way_a_call_answers_a_bad_argument() {
3253 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
3254 let cases = [
3255 (
3256 "int f(int n, ...) { char *p; return __builtin_va_arg(p, int); }\n",
3257 "first argument to 'va_arg' not of type 'va_list'",
3258 ["error", "error", "error", "error"],
3259 ),
3260 (
3261 "void f(int n, ...) { char *p; __builtin_va_start(p, n); }\n",
3262 "passing argument 1 of '__builtin_va_start' from incompatible pointer type",
3263 ["warning", "error", "warning", "error"],
3264 ),
3265 (
3266 "void f(int n, ...) { int x; __builtin_va_end(x); }\n",
3267 "passing argument 1 of '__builtin_va_end' makes pointer from integer without a \
3268 cast",
3269 ["warning", "error", "warning", "error"],
3270 ),
3271 (
3272 "void f(int n, ...) { __builtin_va_list a; char *p; __builtin_va_copy(a, p); }\n",
3273 "passing argument 2 of '__builtin_va_copy' from incompatible pointer type",
3274 ["warning", "error", "warning", "error"],
3275 ),
3276 ];
3277
3278 for (source, message, wanted) in cases {
3279 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
3280 let mut opts = options();
3281 opts.std = std;
3282 opts.permissive = permissive;
3283 let said = run(&opts, source).messages.join("\n");
3284 let how = if permissive { " -fpermissive" } else { "" };
3285 assert!(
3286 said.contains(&format!(": {wanted}: {message}")),
3287 "under -std={}{how}, {source} was answered with `{said}`",
3288 std.as_str()
3289 );
3290 }
3291 }
3292 }
3293
3294 fn safe_ir(tier: rucc_session::Safety, source: &str) -> String {
3296 let mut opts = options();
3297 opts.emit = EmitKind::Ir;
3298 opts.safety = tier;
3299 let result = run(&opts, source);
3300 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3301 result.text().to_owned()
3302 }
3303
3304 const READS_THROUGH_A_POINTER: &str = "int read(int *p) { return p[1]; }\n";
3305
3306 fn padded_ir(padding: Padding, source: &str) -> String {
3308 let mut opts = options();
3309 opts.emit = EmitKind::Ir;
3310 opts.safety = rucc_session::Safety::Detect;
3311 opts.padding = padding;
3312 let result = run(&opts, source);
3313 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3314 result.text().to_owned()
3315 }
3316
3317 const FILLS_A_RECORD_A_MEMBER_AT_A_TIME: &str = "struct padded { char tag; int value; };\n\
3318 void fill(struct padded *p) { p->tag = 1; p->value = 2; }\n";
3319
3320 #[test]
3321 fn a_record_filled_a_member_at_a_time_comes_out_whole_when_padding_does_not_participate() {
3322 let text = padded_ir(Padding::Ignored, FILLS_A_RECORD_A_MEMBER_AT_A_TIME);
3326 assert_eq!(text.matches("owns 4").count(), 2, "{text}");
3327 }
3328
3329 #[test]
3330 fn a_store_says_only_what_it_wrote_when_padding_does_participate() {
3331 let text = padded_ir(Padding::Tracked, FILLS_A_RECORD_A_MEMBER_AT_A_TIME);
3334 assert!(!text.contains("owns"), "{text}");
3335 }
3336
3337 #[test]
3338 fn a_member_of_a_union_owns_nothing_after_it() {
3339 let text = padded_ir(
3343 Padding::Ignored,
3344 "union u { char tag; long wide; };\nvoid fill(union u *p) { p->tag = 1; }\n",
3345 );
3346 assert!(!text.contains("owns"), "{text}");
3347 }
3348
3349 #[test]
3350 fn an_inner_records_trailing_padding_reaches_the_outer_records() {
3351 let text = padded_ir(
3356 Padding::Ignored,
3357 "struct inner { char c; };\n\
3358 struct outer { struct inner in; int x; };\n\
3359 void fill(struct outer *p) { p->in.c = 1; p->x = 2; }\n",
3360 );
3361 assert_eq!(text.matches("owns 4").count(), 2, "{text}");
3362 }
3363
3364 #[test]
3365 fn a_build_that_did_not_ask_for_the_monitor_is_compiled_the_way_it_always_was() {
3366 let text = ir(READS_THROUGH_A_POINTER);
3370 assert!(!text.contains("check_"), "{text}");
3371 assert!(!text.contains("cap_of"), "{text}");
3372 }
3373
3374 #[test]
3375 fn asking_for_a_tier_puts_the_checks_in_before_the_optimizer_sees_them() {
3376 let text = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3377 assert!(text.contains("cap_of"), "{text}");
3378 assert!(text.contains("check_bounds"), "{text}");
3379 assert!(text.contains("check_live"), "{text}");
3380 assert!(text.contains("check_deriv"), "{text}");
3382 assert!(text.contains("check_type"), "{text}");
3384 }
3385
3386 #[test]
3387 fn the_three_tiers_that_are_not_off_all_check_the_same_accesses_so_far() {
3388 let detect = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3392 for tier in [rucc_session::Safety::Enforce, rucc_session::Safety::Kernel] {
3393 assert_eq!(safe_ir(tier, READS_THROUGH_A_POINTER), detect, "{tier}");
3394 }
3395 }
3396
3397 fn summary(tier: rucc_session::Safety, source: &str) -> String {
3399 let mut opts = options();
3400 opts.emit = EmitKind::SafetySummary;
3401 opts.safety = tier;
3402 let result = run(&opts, source);
3403 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3404 result.text().to_owned()
3405 }
3406
3407 #[test]
3408 fn the_summary_counts_the_checks_that_went_in_and_the_ones_still_standing() {
3409 let text = summary(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3410 assert!(text.contains("\"tier\": \"detect\""), "{text}");
3411 assert!(
3413 text.contains("\"bounds\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"),
3414 "{text}"
3415 );
3416 assert!(
3417 text.contains(
3418 "\"derivation\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"
3419 ),
3420 "{text}"
3421 );
3422 }
3423
3424 #[test]
3425 fn a_build_without_the_monitor_summarises_as_a_build_with_no_checks_in_it() {
3426 let text = summary(rucc_session::Safety::Off, READS_THROUGH_A_POINTER);
3430 assert!(text.contains("\"tier\": \"off\""), "{text}");
3431 assert!(
3432 text.contains("\"bounds\": { \"emitted\": 0, \"remaining\": 0, \"discharged\": 0 }"),
3433 "{text}"
3434 );
3435 }
3436
3437 #[test]
3438 fn a_call_the_boundary_models_is_counted_apart_from_one_it_does_not() {
3439 let text = summary(
3440 rucc_session::Safety::Detect,
3441 "void *memcpy(void *, const void *, unsigned long);\n\
3442 int puts(const char *);\n\
3443 void f(char *d, char *s) { memcpy(d, s, 4); puts(d); }\n",
3444 );
3445 assert!(text.contains("\"interposed\": 1"), "{text}");
3446 assert!(text.contains("\"puts\""), "{text}");
3447 assert!(!text.contains("__rucc_wrap_memcpy\""), "{text}");
3451 }
3452
3453 #[test]
3454 fn the_two_directions_a_pointer_crosses_the_boundary_are_counted_apart() {
3455 let text = summary(
3459 rucc_session::Safety::Detect,
3460 "void *notes_open(void);\n\
3461 char *f(char *p) { char *q = notes_open(); return q ? q : p; }\n",
3462 );
3463 assert!(text.contains("\"crossings\": { \"entered\": 1, \"returned\": 1 }"), "{text}");
3464 assert!(text.contains("\"notes_open\""), "{text}");
3465 }
3466
3467 #[test]
3468 fn a_static_function_nobody_takes_the_address_of_is_not_a_crossing() {
3469 let text = summary(
3472 rucc_session::Safety::Detect,
3473 "static int len(const char *p) { return p ? 1 : 0; }\n\
3474 int f(void) { return len(\"x\"); }\n",
3475 );
3476 assert!(text.contains("\"crossings\": { \"entered\": 0, \"returned\": 0 }"), "{text}");
3477 }
3478
3479 fn granules(source: &str) -> String {
3481 let mut opts = options();
3482 opts.emit = EmitKind::TypeGranules;
3483 let result = run(&opts, source);
3484 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3485 result.text().to_owned()
3486 }
3487
3488 #[test]
3489 fn the_granule_report_names_every_record_and_both_keyings() {
3490 let text = granules(
3491 "struct hot { char *p; int a; int b; };\n\
3492 int f(struct hot *h) { return h->a; }\n",
3493 );
3494 assert!(text.contains("struct hot"), "{text}");
3495 assert!(text.contains("every type distinct"), "{text}");
3498 assert!(text.contains("every pointer one type"), "{text}");
3499 assert!(text.contains("budget"), "{text}");
3500 }
3501
3502 #[test]
3503 fn a_record_nothing_uses_is_still_measured() {
3504 let text = granules("struct unused { long a; double b; };\nint f(void) { return 0; }\n");
3507 assert!(text.contains("struct unused"), "{text}");
3508 }
3509
3510 #[test]
3511 fn the_granule_report_stops_before_anything_is_lowered() {
3512 let text = granules(
3516 "struct wide { long double d; };\n\
3517 long double f(long double x) { return x * x; }\n",
3518 );
3519 assert!(text.contains("struct wide"), "{text}");
3520 }
3521
3522 #[test]
3523 fn a_witness_reaches_the_assembler_as_a_call_to_the_runtime() {
3524 let text = safe_asm(rucc_session::Safety::Detect, "char *f(char *p) { return p; }\n");
3527 assert!(text.contains("\tcall\t__rucc_cap_witness\n"), "{text}");
3528 }
3529
3530 #[test]
3531 fn a_pointer_turned_into_an_integer_is_on_the_trust_set() {
3532 let text = summary(
3533 rucc_session::Safety::Detect,
3534 "unsigned long f(int *p) { return (unsigned long) p; }\n",
3535 );
3536 assert!(text.contains("\"exposed\": 1"), "{text}");
3537 }
3538
3539 fn safe_asm(tier: rucc_session::Safety, source: &str) -> String {
3541 let mut opts = options();
3542 opts.emit = EmitKind::Asm;
3543 opts.safety = tier;
3544 let result = run(&opts, source);
3545 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3546 result.text().to_owned()
3547 }
3548
3549 #[test]
3550 fn a_check_reaches_the_assembler_as_a_call_to_the_runtime() {
3551 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3552 assert!(text.contains("\tcall\t__rucc_check_bounds\n"), "{text}");
3553 assert!(text.contains("\tcall\t__rucc_check_live\n"), "{text}");
3554 assert!(text.contains("\tcall\t__rucc_check_deriv\n"), "{text}");
3555 assert!(text.contains("\tcall\t__rucc_check_type\n"), "{text}");
3556 assert!(text.contains("\tcall\t__rucc_check_init\n"), "{text}");
3557 }
3558
3559 #[test]
3560 fn every_check_that_reached_the_assembler_has_a_row_describing_it() {
3561 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3565 let section = format!("\t.section\t{},", rucc_safety::SECTION);
3566 assert_eq!(text.matches(§ion).count(), 5, "{text}");
3567 for index in 0..5 {
3568 let name = format!("__rucc_safety_desc_{index}");
3569 assert!(text.contains(&format!("{name}:\n")), "{text}");
3572 assert!(text.contains(&format!("{name}(%rip)")), "{text}");
3573 }
3574 assert!(!text.contains("__rucc_safety_desc_5"), "{text}");
3575 }
3576
3577 #[test]
3585 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
3586 let text = ir(concat!(
3587 "int g;\n",
3588 "int a = __builtin_constant_p(1);\n",
3589 "int b = __builtin_constant_p(g);\n",
3590 "int c = __builtin_constant_p(\"abc\");\n",
3591 "int d = __builtin_constant_p(&g);\n",
3592 "int e = __builtin_constant_p(1.5);\n",
3593 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
3594 ));
3595 assert!(text.contains("global @a : i32 = 1,"), "{text}");
3596 assert!(text.contains("global @b : i32 = 0,"), "{text}");
3597 assert!(text.contains("global @c : i32 = 1,"), "{text}");
3598 assert!(text.contains("global @d : i32 = 0,"), "{text}");
3599 assert!(text.contains("global @e : i32 = 1,"), "{text}");
3600 assert!(text.contains("global @h : i32 = 11,"), "{text}");
3601 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
3602
3603 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
3607 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
3608 }
3609
3610 #[test]
3619 fn a_call_to_a_library_builtin_reaches_the_library_function() {
3620 let text = body("void f(void) { __builtin_abort(); }\n");
3621 assert_eq!(text, "block0:\n call @abort() : ()\n return\n");
3622
3623 let text = ir("int f(const char *s) { return __builtin_puts(s) + __builtin_strlen(s); }\n");
3626 assert!(text.contains("call @puts(%0) : (ptr) -> i32"), "{text}");
3627 assert!(text.contains("call @strlen(%0) : (ptr) -> i64"), "{text}");
3628 assert!(!text.contains("__builtin_"), "the prefix is not part of any name here:\n{text}");
3629 }
3630
3631 #[test]
3642 fn the_absolute_value_family_is_the_magnitude_and_not_a_call() {
3643 let text = body(concat!(
3644 "long long llabs(long long);\n",
3645 "long long f(long long x) { return llabs(x); }\n",
3646 ));
3647 assert!(text.contains("%1 = iconst.i64 63"), "{text}");
3648 assert!(text.contains("%2 = ashr %0, %1"), "{text}");
3649 assert!(text.contains("%3 = xor %0, %2"), "{text}");
3650 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3651 assert!(!text.contains("call"), "the call does not happen:\n{text}");
3652
3653 let text = body("int abs(int);\nint f(int x) { return abs(x); }\n");
3656 assert!(text.contains("iconst.i32 31"), "{text}");
3657 let text = body("long labs(long);\nlong f(long x) { return labs(x); }\n");
3658 assert!(text.contains("iconst.i64 63"), "{text}");
3659
3660 let text = body("long long f(long long x) { return __builtin_llabs(x); }\n");
3663 assert!(!text.contains("call"), "{text}");
3664
3665 let text = ir(concat!(
3667 "long long llabs(long long b);\n",
3668 "long long g(long long x) { return llabs(x); }\n",
3669 "long long llabs(long long b) { return 7; }\n",
3670 ));
3671 assert!(!text.contains("call @llabs"), "{text}");
3672 }
3673
3674 #[test]
3681 fn a_byte_swap_is_arithmetic_and_not_a_call() {
3682 let text = body("unsigned f(unsigned x) { return __builtin_bswap32(x); }\n");
3683 assert_eq!(text, "block0(%0: i32):\n %1 = bswap %0\n return %1\n");
3684
3685 let text = body("unsigned f(unsigned char c) { return __builtin_bswap32(c); }\n");
3688 assert!(text.contains("zext.i32 %0"), "widened first: {text}");
3689 assert!(text.contains("bswap %1"), "and swapped at four bytes: {text}");
3690 }
3691
3692 #[test]
3698 fn the_byte_swaps_reverse_at_the_width_their_name_says() {
3699 for (name, ty, width) in [
3700 ("__builtin_bswap16", "unsigned short", "i16"),
3701 ("__builtin_bswap32", "unsigned", "i32"),
3702 ("__builtin_bswap64", "unsigned long long", "i64"),
3703 ] {
3704 let source = format!("{ty} f({ty} x) {{ return {name}(x); }}\n");
3705 let text = body(&source);
3706 assert_eq!(
3707 text,
3708 format!("block0(%0: {width}):\n %1 = bswap %0\n return %1\n"),
3709 "{name}"
3710 );
3711 }
3712 }
3713
3714 #[test]
3721 fn the_bit_counts_are_instructions_and_not_calls() {
3722 let text = body("int f(unsigned x) { return __builtin_clz(x); }\n");
3723 assert_eq!(text, "block0(%0: i32):\n %1 = ctlz %0\n return %1\n");
3724
3725 let text = body("int f(unsigned x) { return __builtin_ctz(x); }\n");
3726 assert_eq!(text, "block0(%0: i32):\n %1 = cttz %0\n return %1\n");
3727
3728 let text = body("int f(unsigned x) { return __builtin_popcount(x); }\n");
3729 assert_eq!(text, "block0(%0: i32):\n %1 = ctpop %0\n return %1\n");
3730 }
3731
3732 #[test]
3741 fn the_bit_counts_ask_about_the_width_their_name_says() {
3742 let text = body("int f(unsigned long long x) { return __builtin_clzll(x); }\n");
3743 assert!(text.starts_with("block0(%0: i64):"), "counted at eight bytes: {text}");
3744 assert!(text.contains("%1 = ctlz %0"), "{text}");
3745 assert!(text.contains("trunc.i32 %1"), "and answered in an int: {text}");
3746
3747 let text = body("int f(unsigned long long x) { return __builtin_clz(x); }\n");
3750 assert!(text.contains("trunc.i32 %0"), "narrowed to what was asked about: {text}");
3751 assert!(text.contains("ctlz %1"), "and counted there: {text}");
3752
3753 let text = body("int f(unsigned long x) { return __builtin_popcountl(x); }\n");
3754 assert!(text.contains("%1 = ctpop %0"), "{text}");
3755 assert!(!text.contains("call"), "{text}");
3756 }
3757
3758 #[test]
3763 fn a_parity_is_the_low_bit_of_the_set_bit_count() {
3764 let text = body("int f(unsigned x) { return __builtin_parity(x); }\n");
3765 assert!(text.contains("%1 = ctpop %0"), "{text}");
3766 assert!(text.contains("iconst.i32 1"), "{text}");
3767 assert!(text.contains("and %1, %2"), "the low bit of it: {text}");
3768 }
3769
3770 #[test]
3776 fn the_first_set_bit_is_one_based_and_zero_for_a_zero() {
3777 let text = body("int f(int x) { return __builtin_ffs(x); }\n");
3778 assert!(text.contains("%1 = cttz %0"), "{text}");
3779 assert!(text.contains("%4 = add %1, %2"), "one more than the count: {text}");
3780 assert!(text.contains("%5 = icmp ne %0, %3"), "whether there was a bit at all: {text}");
3781 assert!(text.contains("%7 = sub %3, %6"), "spread to a mask: {text}");
3782 assert!(text.contains("%8 = and %4, %7"), "and kept only then: {text}");
3783 assert!(!text.contains("br_if"), "no branch: {text}");
3784 }
3785
3786 #[test]
3796 fn the_redundant_sign_bit_count_is_instructions_and_not_a_call() {
3797 let text = body("int f(int x) { return __builtin_clrsb(x); }\n");
3798 assert!(text.contains("%1 = iconst.i32 31"), "{text}");
3799 assert!(text.contains("%2 = ashr %0, %1"), "the sign over every bit: {text}");
3800 assert!(text.contains("%3 = xor %0, %2"), "folded onto it: {text}");
3801 assert!(text.contains("%5 = shl %3, %4"), "one less than the count: {text}");
3802 assert!(text.contains("%6 = or %5, %4"), "with something to count at zero: {text}");
3803 assert!(text.contains("%7 = ctlz %6"), "{text}");
3804 assert!(!text.contains("call"), "{text}");
3805 assert!(!text.contains("br_if"), "no branch: {text}");
3806 }
3807
3808 #[test]
3814 fn the_unsigned_absolute_value_family_answers_in_the_unsigned_type() {
3815 let text = body("unsigned f(int x) { return __builtin_uabs(x); }\n");
3816 assert!(text.contains("%1 = iconst.i32 31"), "{text}");
3817 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3818 assert!(!text.contains("call"), "nothing declares uabs, so a call would not link: {text}");
3819
3820 let text = body("unsigned long long f(long long x) { return __builtin_ullabs(x); }\n");
3821 assert!(text.contains("iconst.i64 63"), "at the width the name says: {text}");
3822
3823 let text = body("int f(int x) { return __builtin_uabs(x) > 2147483647u; }\n");
3826 assert!(text.contains("icmp ugt"), "compared unsigned: {text}");
3827 }
3828
3829 #[test]
3837 fn the_widest_absolute_value_is_whichever_type_the_target_makes_intmax_t() {
3838 let text = body("long f(long x) { return __builtin_imaxabs(x); }\n");
3839 assert!(text.contains("iconst.i64 63"), "{text}");
3840 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3841 assert!(!text.contains("call"), "{text}");
3842
3843 let text = body("unsigned long f(long x) { return __builtin_umaxabs(x); }\n");
3844 assert!(text.contains("iconst.i64 63"), "{text}");
3845 assert!(!text.contains("call"), "{text}");
3846 }
3847
3848 #[test]
3856 fn an_overflow_predicate_writes_nothing_and_answers_the_bit_the_check_would() {
3857 let text =
3858 body("int f(int a, int b) { return __builtin_add_overflow_p(a, b, (int) 0); }\n");
3859 assert!(text.contains("%2, %3 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
3860 assert!(!text.contains("store"), "nothing is written: {text}");
3861 assert!(!text.contains("call"), "{text}");
3862
3863 let text =
3866 body("int f(int a, int b) { return __builtin_mul_overflow_p(a, b, (long long) 0); }\n");
3867 assert!(text.contains("smul_overflow.(i64, i1)"), "{text}");
3868 assert!(!text.contains("store"), "{text}");
3869
3870 let text = body(concat!(
3873 "int g(void);\n",
3874 "int f(int a, int b) { return __builtin_sub_overflow_p(a, b, g()); }\n",
3875 ));
3876 assert!(!text.contains("call @g"), "the third argument is not evaluated: {text}");
3877 }
3878
3879 #[test]
3889 fn an_overflow_check_is_arithmetic_and_not_a_call() {
3890 let text =
3891 body("int f(int a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3892 assert!(text.contains("%3, %4 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
3893 assert!(text.contains("store %3 -> %2"), "{text}");
3894 assert!(!text.contains("call"), "{text}");
3895
3896 let text =
3897 body("int f(int a, int b, int *r) { return __builtin_sub_overflow(a, b, r); }\n");
3898 assert!(text.contains("ssub_overflow.(i32, i1) %0, %1"), "{text}");
3899
3900 let text =
3901 body("int f(int a, int b, int *r) { return __builtin_mul_overflow(a, b, r); }\n");
3902 assert!(text.contains("smul_overflow.(i32, i1) %0, %1"), "{text}");
3903
3904 let text = body(
3907 "int f(unsigned a, unsigned b, unsigned *r) { return __builtin_add_overflow(a, b, r); }\n",
3908 );
3909 assert!(text.contains("uadd_overflow.(i32, i1) %0, %1"), "{text}");
3910 }
3911
3912 #[test]
3920 fn an_overflow_check_is_done_at_a_type_that_holds_every_operand() {
3921 let text = body(
3922 "int f(unsigned a, int b, long long *r) { return __builtin_add_overflow(a, b, r); }\n",
3923 );
3924 assert!(text.contains("%3 = zext.i64 %0"), "the unsigned operand keeps its value: {text}");
3925 assert!(text.contains("%4 = sext.i64 %1"), "and so does the signed one: {text}");
3926 assert!(text.contains("sadd_overflow.(i64, i1) %3, %4"), "{text}");
3927
3928 let text = body(
3931 "int f(long long a, long long b, long long *r) { return __builtin_mul_overflow(a, b, r); }\n",
3932 );
3933 assert!(text.contains("smul_overflow.(i64, i1) %0, %1"), "{text}");
3934 assert!(!text.contains("sext."), "{text}");
3935 assert!(!text.contains("zext.i64"), "{text}");
3937 }
3938
3939 #[test]
3947 fn an_overflow_check_writes_the_wrapped_answer_whether_or_not_it_fit() {
3948 let text =
3949 body("int f(int a, int b, char *r) { return __builtin_sub_overflow(a, b, r); }\n");
3950 assert!(text.contains("%3, %4 = ssub_overflow.(i32, i1) %0, %1"), "{text}");
3951 assert!(text.contains("%5 = trunc.i8 %3"), "narrowed to where it goes: {text}");
3952 assert!(text.contains("%6 = sext.i32 %5"), "and back: {text}");
3953 assert!(text.contains("%7 = icmp ne %6, %3"), "which is whether it fit: {text}");
3954 assert!(text.contains("store %5 -> %2"), "the narrowed value is stored either way: {text}");
3955 assert!(text.contains("%8 = or %4, %7"), "and either bit is an overflow: {text}");
3956 }
3957
3958 #[test]
3965 fn a_call_needing_more_than_the_widest_type_still_compiles() {
3966 for name in ["add", "sub", "mul"] {
3967 let source = format!(
3968 "int f(unsigned __int128 a, long long b, __int128 *r) {{\n \
3969 return __builtin_{name}_overflow(a, b, r);\n}}\n"
3970 );
3971 let mut opts = options();
3972 opts.emit = EmitKind::MirFinal;
3973 assert!(!run(&opts, &source).failed(), "{name} was refused or stopped the back end");
3974 }
3975 }
3976
3977 #[test]
3980 fn an_overflow_check_over_something_that_is_not_an_integer_says_so() {
3981 let messages =
3982 errors("int f(double a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3983 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3984
3985 let messages =
3986 errors("int f(int a, int b, double *r) { return __builtin_add_overflow(a, b, r); }\n");
3987 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3988 }
3989
3990 #[test]
4001 fn an_ordered_access_is_ordered_in_the_ir() {
4002 let text = body("int f(int *p) { return __atomic_load_n(p, 0); }\n");
4003 assert!(text.contains("atomic_load.i32 %0, align 4, relaxed"), "{text}");
4004
4005 let text = body("long f(long *p) { return __atomic_load_n(p, 2); }\n");
4006 assert!(text.contains("atomic_load.i64 %0, align 8, acquire"), "{text}");
4007
4008 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
4009 assert!(text.contains("atomic_store %1 -> %0, align 4, release"), "{text}");
4010
4011 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
4012 assert!(text.contains("atomic_store %1 -> %0, align 4, seq_cst"), "{text}");
4013
4014 let text = body("void f(char *p, int v) { __atomic_store_n(p, v, 0); }\n");
4017 assert!(text.contains("trunc.i8 %1"), "{text}");
4018 assert!(text.contains("atomic_store %2 -> %0, align 1, relaxed"), "{text}");
4019 }
4020
4021 #[test]
4030 fn an_ordered_access_is_the_plain_instruction_on_this_machine() {
4031 let text = asm("int f(int *p) { return __atomic_load_n(p, 5); }\n");
4032 assert!(text.contains("movl\t(%rdi), %eax"), "{text}");
4033 assert!(!text.contains("mfence"), "a load needs no barrier here: {text}");
4034
4035 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
4036 assert!(text.contains("movl\t%esi, (%rdi)"), "{text}");
4037 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
4038
4039 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
4040 let (before, after) = text.split_once("mfence").expect("a barrier: {text}");
4041 assert!(before.contains("movl\t%esi, (%rdi)"), "the store comes first: {text}");
4042 assert!(!after.contains("movl"), "and nothing else is between them: {text}");
4043 }
4044
4045 #[test]
4055 fn a_barrier_is_one_instruction_at_the_strongest_ordering_and_none_below_it() {
4056 assert!(asm("void f(void) { __atomic_thread_fence(5); }\n").contains("mfence"));
4057 assert!(asm("void f(void) { __sync_synchronize(); }\n").contains("mfence"));
4058
4059 for weaker in ["1", "2", "3", "4"] {
4060 let source = format!("void f(void) {{ __atomic_thread_fence({weaker}); }}\n");
4061 assert!(!asm(&source).contains("mfence"), "{weaker} costs nothing here");
4062 }
4063 }
4064
4065 #[test]
4071 fn a_compare_and_exchange_is_one_instruction_answering_two_things() {
4072 let text =
4075 body("int f(int *p, int e, int d) { return __sync_val_compare_and_swap(p, e, d); }\n");
4076 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
4077 assert!(text.contains("return %3"), "the value it found: {text}");
4078
4079 let text =
4080 body("int f(int *p, int e, int d) { return __sync_bool_compare_and_swap(p, e, d); }\n");
4081 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
4082 assert!(text.contains("zext.i32 %4"), "whether it happened: {text}");
4083
4084 let text = body(
4087 "int f(int *p, int *e, int d) { return __atomic_compare_exchange_n(p, e, d, 0, 4, 2); }\n",
4088 );
4089 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
4090 assert!(text.contains("%4, %5 = cmpxchg.(i32, i1) %0, %3, %2, align 4, acq_rel"), "{text}");
4091 assert!(text.contains("br_if %5, block2, block1"), "{text}");
4092 assert!(text.contains("store %4 -> %1, align 4"), "{text}");
4093
4094 let text = body(
4097 "int f(int *p, int *e, int *d) { return __atomic_compare_exchange(p, e, d, 0, 5, 5); }\n",
4098 );
4099 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
4100 assert!(text.contains("%4 = load.i32 %2, align 4"), "{text}");
4101 assert!(text.contains("%5, %6 = cmpxchg.(i32, i1) %0, %3, %4, align 4, seq_cst"), "{text}");
4102 }
4103
4104 #[test]
4111 fn a_compare_and_exchange_is_a_locked_instruction_at_the_width_of_the_object() {
4112 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
4113 for (ty, suffix, reg) in widths {
4114 let source = format!(
4115 "int f({ty} *p, {ty} e, {ty} d) {{ return __sync_bool_compare_and_swap(p, e, d); }}\n"
4116 );
4117 let text = asm(&source);
4118 assert!(text.contains("\tlock\n"), "{ty}: {text}");
4119 assert!(text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
4120 assert!(text.contains("sete\t"), "{ty}: {text}");
4121 }
4122 let source =
4123 "int f(long *p, long e, long d) { return __sync_bool_compare_and_swap(p, e, d); }\n";
4124 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
4125
4126 for order in ["0", "2", "3", "4", "5"] {
4130 let call = format!("__atomic_compare_exchange_n(p, e, d, 0, {order}, 0)");
4131 let source = format!("int f(int *p, int *e, int d) {{ return {call}; }}\n");
4132 let text = asm(&source);
4133 assert!(text.contains("cmpxchgl\t"), "{order}: {text}");
4134 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
4135 }
4136 }
4137
4138 #[test]
4150 fn a_read_modify_write_is_one_instruction_and_the_arithmetic_a_name_asks_for() {
4151 let text = body("int f(int *p, int v) { return __atomic_fetch_add(p, v, 5); }\n");
4152 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
4153 assert!(text.contains("return %2"), "the value that was there: {text}");
4154
4155 let text = body("int f(int *p, int v) { return __atomic_add_fetch(p, v, 5); }\n");
4156 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
4157 assert!(text.contains("%3 = add %2, %1"), "and the value afterwards: {text}");
4158
4159 let text = body("int f(int *p, int v) { return __atomic_sub_fetch(p, v, 5); }\n");
4160 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
4161 assert!(text.contains("%3 = sub %2, %1"), "{text}");
4162
4163 let text = body("int f(int *p, int v) { return __sync_fetch_and_sub(p, v); }\n");
4165 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
4166
4167 let text = body("int f(int *p, int v) { return __atomic_exchange_n(p, v, 5); }\n");
4170 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, seq_cst"), "{text}");
4171
4172 let text = body("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
4173 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, acquire"), "{text}");
4174
4175 let text = body("void f(int *p) { __sync_lock_release(p); }\n");
4178 assert!(text.contains("release"), "{text}");
4179 assert!(text.contains("%1 = iconst.i32 0"), "{text}");
4180
4181 let text = body("void f(int *p, int guard) { __sync_lock_release(p, guard); }\n");
4185 assert!(text.contains("%2 = iconst.i32 0"), "{text}");
4186 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
4187
4188 let text = body("int f(int *p, int v) { return __atomic_fetch_and(p, v, 5); }\n");
4191 assert!(text.contains("%2 = atomic_rmw.i32 and %0, %1, align 4, seq_cst"), "{text}");
4192
4193 let text = body("int f(int *p, int v) { return __sync_or_and_fetch(p, v); }\n");
4194 assert!(text.contains("%2 = atomic_rmw.i32 or %0, %1, align 4, seq_cst"), "{text}");
4195 assert!(text.contains("%3 = or %2, %1"), "and the value afterwards: {text}");
4196
4197 let text = body("int f(int *p, int v) { return __atomic_nand_fetch(p, v, 5); }\n");
4200 assert!(text.contains("%2 = atomic_rmw.i32 nand %0, %1, align 4, seq_cst"), "{text}");
4201 assert!(text.contains("%3 = and %2, %1"), "{text}");
4202 assert!(text.contains("%4 = iconst.i32 -1"), "{text}");
4203 assert!(text.contains("%5 = xor %3, %4"), "{text}");
4204 }
4205
4206 #[test]
4217 fn a_bitwise_read_modify_write_is_a_loop_around_the_compare_and_exchange() {
4218 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
4219 for (ty, suffix, reg) in widths {
4220 for (name, call, insn) in [
4221 ("and", "__atomic_fetch_and(p, v, 5)", "and"),
4222 ("or", "__sync_fetch_and_or(p, v)", "or"),
4223 ("xor", "__atomic_xor_fetch(p, v, 5)", "xor"),
4224 ] {
4225 let source = format!("{ty} f({ty} *p, {ty} v) {{ return {call}; }}\n");
4226 let text = asm(&source);
4227 assert!(text.contains("\tlock\n"), "{ty} {name}: {text}");
4228 assert!(
4229 text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")),
4230 "{ty} {name}: {text}"
4231 );
4232 assert!(text.contains(&format!("{insn}{suffix}\t")), "{ty} {name}: {text}");
4233 assert!(!text.contains("\txadd"), "{ty} {name} is not an add: {text}");
4235 assert!(!text.contains("\txchg"), "{ty} {name} is not an exchange: {text}");
4236 }
4237 }
4238 let source = "long f(long *p, long v) { return __atomic_fetch_or(p, v, 5); }\n";
4239 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
4240
4241 let text = asm("int f(int *p, int v) { return __sync_fetch_and_nand(p, v); }\n");
4245 assert!(text.contains("cmpxchgl\t"), "{text}");
4246 assert!(text.contains("andl\t"), "{text}");
4247 assert!(text.contains("notl\t"), "{text}");
4248 }
4249
4250 #[test]
4259 fn an_access_through_a_second_pointer_is_the_same_access_and_one_more() {
4260 let text = body("void f(int *p, int *r) { __atomic_load(p, r, 5); }\n");
4261 assert!(text.contains("%2 = atomic_load.i32 %0, align 4, seq_cst"), "{text}");
4262 assert!(text.contains("store %2 -> %1, align 4"), "and out through the place: {text}");
4263
4264 let text = body("void f(int *p, int *v) { __atomic_store(p, v, 3); }\n");
4265 assert!(text.contains("%2 = load.i32 %1, align 4"), "in through the place: {text}");
4266 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
4267
4268 let text = body("void f(int *p, int *v, int *r) { __atomic_exchange(p, v, r, 5); }\n");
4271 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
4272 assert!(text.contains("%4 = atomic_rmw.i32 xchg %0, %3, align 4, seq_cst"), "{text}");
4273 assert!(text.contains("store %4 -> %2, align 4"), "{text}");
4274 }
4275
4276 #[test]
4287 fn a_flag_is_an_exchange_of_one_byte_and_a_store_of_a_zero_over_the_same_byte() {
4288 for pointer in ["char", "int", "void"] {
4289 let source = format!("int f({pointer} *p) {{ return __atomic_test_and_set(p, 5); }}\n");
4290 let text = body(&source);
4291 assert!(text.contains("%1 = iconst.i8 1"), "{pointer}: {text}");
4292 assert!(
4293 text.contains("%2 = atomic_rmw.i8 xchg %0, %1, align 1, seq_cst"),
4294 "{pointer}: {text}"
4295 );
4296 assert!(text.contains("%4 = icmp ne %2, %3"), "{pointer}: {text}");
4297
4298 let source = format!("void f({pointer} *p) {{ __atomic_clear(p, 3); }}\n");
4299 let text = body(&source);
4300 assert!(text.contains("atomic_store %2 -> %0, align 1, release"), "{pointer}: {text}");
4301 }
4302
4303 let text = asm("int f(int *p) { return __atomic_test_and_set(p, 5); }\n");
4306 assert!(text.contains("xchgb\t%al, (%rdi)"), "{text}");
4307 assert!(text.contains("setne\t"), "{text}");
4308 }
4309
4310 #[test]
4318 fn a_read_modify_write_is_an_exchange_or_a_locked_add_at_the_width_of_the_object() {
4319 let widths = [("char", "b", "%sil"), ("short", "w", "%si"), ("int", "l", "%esi")];
4320 for (ty, suffix, reg) in widths {
4321 let source =
4322 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_fetch_add(p, v, 5); }}\n");
4323 let text = asm(&source);
4324 assert!(text.contains("\tlock\n"), "{ty}: {text}");
4325 assert!(text.contains(&format!("xadd{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
4326
4327 let source =
4328 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_exchange_n(p, v, 5); }}\n");
4329 let text = asm(&source);
4330 assert!(text.contains(&format!("xchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
4331 assert!(!text.contains("\tlock\n"), "an exchange is locked already: {ty}: {text}");
4332 }
4333 let source = "long f(long *p, long v) { return __atomic_fetch_add(p, v, 5); }\n";
4334 assert!(asm(source).contains("xaddq\t%rsi, (%rdi)"), "{}", asm(source));
4335
4336 let source = "int f(int *p, int v) { return __atomic_fetch_sub(p, v, 5); }\n";
4339 let text = asm(source);
4340 assert!(text.contains("negl\t"), "{text}");
4341 assert!(text.contains("xaddl\t"), "{text}");
4342
4343 for order in ["0", "2", "3", "4", "5"] {
4346 let source =
4347 format!("int f(int *p, int v) {{ return __atomic_fetch_add(p, v, {order}); }}\n");
4348 let text = asm(&source);
4349 assert!(text.contains("xaddl\t"), "{order}: {text}");
4350 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
4351 }
4352
4353 let text = asm("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
4357 assert!(text.contains("xchgl\t%esi, (%rdi)"), "{text}");
4358 let text = asm("void f(int *p) { __sync_lock_release(p); }\n");
4363 assert!(text.contains("movl\t$0, %eax"), "{text}");
4364 assert!(text.contains("movl\t%eax, (%rdi)"), "{text}");
4365 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
4366 }
4367
4368 #[test]
4380 fn the_lock_free_questions_are_answered_as_constants() {
4381 for size in ["1", "2", "4", "8"] {
4382 let source =
4383 format!("int f(void) {{ return __atomic_always_lock_free({size}, 0); }}\n");
4384 let text = asm(&source);
4385 assert!(text.contains("movb\t$1, %al"), "{size} bytes is lock free: {text}");
4386 assert!(!text.contains("call"), "and is not a call: {text}");
4387 }
4388 for size in ["3", "16", "sizeof(long double)"] {
4389 let source = format!("int f(void) {{ return __atomic_is_lock_free({size}, 0); }}\n");
4390 let text = asm(&source);
4391 assert!(text.contains("movb\t$0, %al"), "{size} bytes is not: {text}");
4392 assert!(!text.contains("call"), "and is not a call either: {text}");
4393 }
4394
4395 let text = asm("int f(int n) { return __atomic_is_lock_free(n, 0); }\n");
4399 assert!(text.contains("movb\t$0, %al"), "a size nobody knows is not lock free: {text}");
4400 let text = asm("int f(int *p) { return __atomic_always_lock_free(8, p); }\n");
4401 assert!(text.contains("movb\t$0, %al"), "eight bytes at four is not: {text}");
4402 let text = asm("int f(long *p) { return __atomic_always_lock_free(8, p); }\n");
4403 assert!(text.contains("movb\t$1, %al"), "and at eight it is: {text}");
4404 }
4405
4406 #[test]
4418 fn a_memory_order_an_operation_cannot_carry_is_read_as_the_strongest() {
4419 let mut opts = options();
4420 opts.emit = EmitKind::Ir;
4421
4422 let acquire_store = run(&opts, "void f(int *p, int v) { __atomic_store_n(p, v, 2); }\n");
4423 assert!(acquire_store.text().contains("seq_cst"), "{:?}", acquire_store.text());
4424 assert!(acquire_store.messages[0].contains("[W0333]"), "{:?}", acquire_store.messages);
4425
4426 let nonsense = run(&opts, "int f(int *p) { return __atomic_load_n(p, 99); }\n");
4427 assert!(nonsense.text().contains("seq_cst"), "{:?}", nonsense.text());
4428 assert!(nonsense.messages[0].contains("[W0333]"), "{:?}", nonsense.messages);
4429
4430 let computed = run(&opts, "int f(int *p, int n) { return __atomic_load_n(p, n); }\n");
4431 assert!(computed.text().contains("seq_cst"), "{:?}", computed.text());
4432 assert_eq!(computed.messages, Vec::<String>::new(), "a computed order is not a mistake");
4433 }
4434
4435 #[test]
4447 fn a_conversion_between_a_float_and_the_widest_unsigned_integer_is_written_without_a_branch() {
4448 let text = asm("double f(unsigned long long x) { return (double)x; }\n");
4449 assert!(text.contains("cvtsi2sdq"), "the signed conversion is what runs: {text}");
4450 assert!(text.contains("shrq"), "with the value halved first: {text}");
4451 assert!(text.contains("addsd"), "and doubled after: {text}");
4452 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
4453
4454 let text = asm("unsigned long long f(double d) { return (unsigned long long)d; }\n");
4455 assert!(text.contains("cvttsd2siq"), "the signed conversion is what runs: {text}");
4456 assert!(text.contains("subsd"), "with half the range taken off first: {text}");
4457 assert!(text.contains("shlq\t$63"), "and the top bit put back: {text}");
4458 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
4459 }
4460
4461 #[test]
4472 fn a_plain_name_the_program_took_is_the_programs_own_function() {
4473 let taken = concat!(
4474 "static long long llabs(long long b) { return 7; }\n",
4475 "long long f(long long x) { return llabs(x); }\n",
4476 );
4477 assert!(ir(taken).contains("call @llabs"), "a static definition is the program's own");
4478
4479 let retyped = concat!("int llabs(int b);\n", "int f(int x) { return llabs(x); }\n",);
4480 assert!(ir(retyped).contains("call @llabs"), "another type is another function");
4481
4482 let plain = concat!(
4483 "long long llabs(long long b);\n",
4484 "long long f(long long x) { return llabs(x); }\n",
4485 );
4486 let mut opts = options();
4487 opts.emit = EmitKind::Ir;
4488 assert!(!run(&opts, plain).text().contains("call @llabs"), "the library's by default");
4489
4490 opts.builtins = false;
4491 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin");
4492
4493 opts.builtins = true;
4494 opts.no_builtin = vec!["llabs".to_owned()];
4495 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin-llabs");
4496 let one = "long labs(long b);\nlong f(long x) { return labs(x); }\n";
4497 assert!(!run(&opts, one).text().contains("call @labs"), "one name and not the family");
4498
4499 opts.no_builtin = Vec::new();
4502 opts.builtins = false;
4503 let prefixed = "long long f(long long x) { return __builtin_llabs(x); }\n";
4504 assert!(!run(&opts, prefixed).text().contains("call @llabs"), "the prefix is a promise");
4505 }
4506
4507 #[test]
4520 fn the_hint_builtins_are_their_first_argument_and_the_hint_leaves_no_trace() {
4521 let text = ir(concat!(
4522 "long a = __builtin_expect(7, 1);\n",
4523 "long b = __builtin_expect_with_probability(9, 1, 0.9);\n",
4524 "unsigned long c = sizeof(__builtin_expect((char)1, 1));\n",
4525 ));
4526 assert!(text.contains("global @a : i64 = 7,"), "{text}");
4527 assert!(text.contains("global @b : i64 = 9,"), "{text}");
4528 assert!(text.contains("global @c : i64 = 8,"), "{text}");
4529 assert!(!text.contains("__builtin_expect"), "it is not a call to anything:\n{text}");
4530
4531 let text = body("long f(char c) { return __builtin_expect(c, 1); }\n");
4534 assert!(text.contains("sext"), "{text}");
4535
4536 let one = "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 1\n %2 = sext.i64 %1\n return %0\n";
4540 assert_eq!(body("int f(void) { int i = 0; __builtin_expect(1, i++); return i; }\n"), one);
4541 let source = "int g(void) { int i = 0; __builtin_expect_with_probability(1, i++, 0.5); return i; }\n";
4542 assert_eq!(body(source), one);
4543
4544 let kept = body("int f(int n) { int i = 0; __builtin_expect(n, i++); return i; }\n");
4549 assert!(kept.contains("add.nsw"), "the hint still runs: {kept}");
4550 assert!(kept.ends_with("return %3\n"), "and the answer is what it left behind: {kept}");
4551 let both = "int g(int n) { int i = 0; __builtin_expect_with_probability(n, i++, 0.5); return i; }\n";
4552 assert!(body(both).contains("add.nsw"), "and so does the one with three arguments");
4553 }
4554
4555 #[test]
4567 fn a_promise_that_control_does_not_arrive_writes_no_instruction() {
4568 let promised = "int f(int x) { if (x) return 1; __builtin_unreachable(); }\n";
4569 let text = ir(promised);
4570 assert!(text.contains(" unreachable_hint\n"), "{text}");
4571 assert!(!text.contains("call"), "it is not a call to anything:\n{text}");
4572
4573 let after = body("int g(int x) { __builtin_unreachable(); return x; }\n");
4577 assert!(after.contains("return"), "{after}");
4578
4579 let text = asm(promised);
4582 let mine = text.split_once("\nf:\n").expect("a definition").1;
4583 let mine = mine.split_once("\t.size").expect("a definition").0;
4584 let plain = asm("int f(int x) { if (x) return 1; }\n");
4585 let plain = plain.split_once("\nf:\n").expect("a definition").1;
4586 let plain = plain.split_once("\t.size").expect("a definition").0;
4587 assert_eq!(mine, plain);
4588 let last = mine.lines().rfind(|line| !line.trim_start().starts_with('.'));
4591 assert_eq!(last.map(str::trim), Some("ret"), "{mine}");
4592 assert!(!mine.contains("ud2"), "{mine}");
4593 }
4594
4595 #[test]
4602 fn a_library_builtin_is_diagnosed_under_the_name_the_program_wrote() {
4603 let mut opts = options();
4604 opts.emit = EmitKind::Ir;
4605 let messages = run(&opts, "void f(void) { __builtin_abort(1); }\n").messages;
4606 assert!(
4607 messages.iter().any(|m| m.contains("__builtin_abort")),
4608 "expected the written name in {messages:?}"
4609 );
4610 }
4611
4612 #[test]
4621 fn a_builtin_nothing_lowers_is_refused_by_name() {
4622 let mut opts = options();
4623 opts.emit = EmitKind::Ir;
4624 for (builtin, call) in [
4625 ("__builtin_object_size", "(int)__builtin_object_size(&counter, 0)"),
4626 ("__builtin_dynamic_object_size", "(int)__builtin_dynamic_object_size(&counter, 0)"),
4627 ("__atomic_signal_fence", "(__atomic_signal_fence(5), 0)"),
4628 ] {
4629 let source = format!("int counter;\nint f(void) {{ return {call}; }}\n");
4630 let messages = run(&opts, &source).messages;
4631 let named = messages.iter().any(|m| m.contains(builtin) && m.contains("E0686"));
4632 assert!(named, "expected {builtin} to be refused by name in {messages:?}");
4633 }
4634 }
4635
4636 #[test]
4644 fn what_is_refused_is_the_call_and_not_the_name() {
4645 let text = ir("unsigned long n = sizeof(__builtin_object_size(0, 0));\n");
4646 assert!(text.contains("global @n : i64 = 8,"), "{text}");
4647
4648 let text = ir(concat!(
4649 "unsigned long __builtin_object_size(const void *p, int kind) { return 0; }\n",
4650 "unsigned long f(void) { return __builtin_object_size(0, 0); }\n",
4651 ));
4652 assert!(text.contains("call @__builtin_object_size"), "{text}");
4653 }
4654
4655 #[test]
4660 fn a_static_function_nothing_refers_to_is_not_emitted() {
4661 let text = ir("static int dropped(void) { return 1; }\n\
4662 static int kept(void) { return 2; }\n\
4663 int main(void) { return kept(); }\n");
4664 assert!(text.contains("func @kept"), "{text}");
4665 assert!(!text.contains("dropped"), "{text}");
4666 }
4667
4668 #[test]
4674 fn two_static_functions_that_only_call_each_other_are_both_dropped() {
4675 let text = ir("static int ping(void);\n\
4676 static int pong(void) { return ping(); }\n\
4677 static int ping(void) { return pong(); }\n\
4678 int main(void) { return 0; }\n");
4679 assert!(!text.contains("ping"), "{text}");
4680 assert!(!text.contains("pong"), "{text}");
4681 }
4682
4683 #[test]
4689 fn naming_a_static_function_anywhere_keeps_it() {
4690 let text = ir("static int by_address(void) { return 1; }\n\
4691 static int in_an_image(void) { return 2; }\n\
4692 static int deeper(void) { return 3; }\n\
4693 static int reaches_deeper(void) { return deeper(); }\n\
4694 static int (*table[1])(void) = {in_an_image};\n\
4695 int main(void) {\n\
4696 int (*p)(void) = by_address;\n\
4697 return p() + table[0]() + reaches_deeper();\n\
4698 }\n");
4699 for kept in ["by_address", "in_an_image", "deeper", "reaches_deeper"] {
4700 assert!(text.contains(&format!("func @{kept}")), "expected {kept} in:\n{text}");
4701 }
4702 }
4703
4704 #[test]
4710 fn an_attribute_keeps_a_static_function_nothing_refers_to() {
4711 for attribute in ["used", "retain", "constructor", "destructor", "__used__"] {
4712 let source = format!(
4713 "__attribute__(({attribute})) static int kept(void) {{ return 1; }}\n\
4714 int main(void) {{ return 0; }}\n"
4715 );
4716 let text = ir(&source);
4717 assert!(text.contains("func @kept"), "for {attribute}:\n{text}");
4718 }
4719 }
4720
4721 #[test]
4724 fn a_function_anything_could_call_is_emitted_without_being_called() {
4725 let text =
4726 ir("int nobody_here_calls_it(void) { return 1; }\nint main(void) { return 0; }\n");
4727 assert!(text.contains("func @nobody_here_calls_it"), "{text}");
4728 }
4729
4730 #[test]
4737 fn a_classification_c_has_an_operator_for_is_that_operator() {
4738 for (builtin, operator) in [
4739 ("__builtin_isgreater", "binary >"),
4740 ("__builtin_isgreaterequal", "binary >="),
4741 ("__builtin_isless", "binary <"),
4742 ("__builtin_islessequal", "binary <="),
4743 ] {
4744 let source = format!("int f(double x, double y) {{ return {builtin}(x, y); }}\n");
4745 let text = tast(&source);
4746 assert!(text.contains(&format!("{operator} : int")), "for {builtin}:\n{text}");
4747 }
4748 }
4749
4750 #[test]
4759 fn the_classification_builtins_are_comparisons_and_not_calls() {
4760 let text = body("int f(double x, double y) { return __builtin_isunordered(x, y); }\n");
4761 assert_eq!(
4762 text,
4763 "block0(%0: f64, %1: f64):\n %2 = fcmp uno %0, %1\n %3 = zext.i32 \
4764 %2\n return %3\n"
4765 );
4766
4767 let text = body("int f(double x, double y) { return __builtin_islessgreater(x, y); }\n");
4769 assert!(text.contains("fcmp one %0, %1"), "{text}");
4770
4771 let text = body("int f(double x) { return __builtin_isnan(x); }\n");
4772 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4773
4774 let text = body("int f(double x) { return __builtin_isinf(x); }\n");
4775 assert!(text.contains("fconst.f64 0x7ff0000000000000"), "{text}");
4776 assert!(text.contains("fconst.f64 0xfff0000000000000"), "{text}");
4777 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4778 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4779 assert!(text.contains("%5 = or %3, %4"), "{text}");
4780
4781 let text = body("int f(double x) { return __builtin_isfinite(x); }\n");
4784 assert!(text.contains("%3 = fcmp olt %2, %0"), "{text}");
4785 assert!(text.contains("%4 = fcmp olt %0, %1"), "{text}");
4786 assert!(text.contains("%5 = and %3, %4"), "{text}");
4787
4788 let text = body("int f(double x) { return __builtin_signbit(x); }\n");
4789 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4790 assert!(text.contains("icmp slt %1, %2"), "{text}");
4791
4792 let text = body("int f(long double x) { return __builtin_signbitl(x); }\n");
4795 assert!(text.contains("%1 = bitcast.i80 %0"), "{text}");
4796
4797 let text = body("double g(void);\nint f(void) { return __builtin_isnan(g()); }\n");
4800 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4801 }
4802
4803 #[test]
4810 fn a_classification_spelling_that_names_a_width_converts_before_it_asks() {
4811 let text = ir(concat!(
4812 "int a = __builtin_isinff(1e300);\n",
4813 "int b = __builtin_isinf(1e300);\n",
4814 "int c = __builtin_isnan(0.0);\n",
4818 "int d = __builtin_signbit(-0.0);\n",
4819 "int e = __builtin_islessgreater(1.0, 2.0);\n",
4820 ));
4821 assert!(text.contains("global @a : i32 = 1,"), "{text}");
4822 assert!(text.contains("global @b : i32 = 0,"), "{text}");
4823 assert!(text.contains("global @c : i32 = 0,"), "{text}");
4824 assert!(text.contains("global @d : i32 = 1,"), "{text}");
4825 assert!(text.contains("global @e : i32 = 1,"), "{text}");
4826 }
4827
4828 #[test]
4830 fn a_classification_builtin_refuses_an_argument_that_is_not_floating_point() {
4831 let mut opts = options();
4832 opts.emit = EmitKind::Ir;
4833 let source = concat!(
4834 "int a(int x) { return __builtin_isnan(x); }\n",
4835 "int b(int x, int y) { return __builtin_isunordered(x, y); }\n",
4836 "int c(double x) { return __builtin_isnan(x, x); }\n",
4837 );
4838 let messages = run(&opts, source).messages;
4839 assert_eq!(
4840 messages,
4841 [
4842 "/main.c:1:23: error: non-floating-point argument in call to function \
4843 '__builtin_isnan' [E0685]",
4844 "/main.c:2:30: error: non-floating-point arguments in call to function \
4845 '__builtin_isunordered' [E0685]",
4846 "/main.c:3:26: error: too many arguments to function '__builtin_isnan' [E0511]",
4847 ]
4848 );
4849 }
4850
4851 #[test]
4860 fn the_last_three_classification_builtins_are_comparisons_and_not_calls() {
4861 let text = body("int f(double x) { return __builtin_isnormal(x); }\n");
4862 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4866 assert!(text.contains("%2 = iconst.i64 9223372036854775807"), "{text}");
4867 assert!(text.contains("%3 = and %1, %2"), "{text}");
4868 assert!(text.contains("%4 = iconst.i64 4503599627370496"), "{text}");
4869 assert!(text.contains("%5 = iconst.i64 9218868437227405312"), "{text}");
4870 assert!(text.contains("%6 = icmp uge %3, %4"), "{text}");
4871 assert!(text.contains("%7 = icmp ult %3, %5"), "{text}");
4872 assert!(text.contains("%8 = and %6, %7"), "{text}");
4873
4874 let text = body("int f(long double x) { return __builtin_isnormal(x); }\n");
4878 assert!(text.contains("%4 = iconst.i80 27670116110564327424"), "{text}");
4879 assert!(text.contains("%5 = iconst.i80 604453686435277732577280"), "{text}");
4880
4881 let text = body("int f(double x) { return __builtin_isinf_sign(x); }\n");
4882 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4883 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4884 assert!(text.contains("%7 = sub %5, %6"), "{text}");
4885
4886 let text = body("int f(double x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n");
4887 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4888 assert!(text.contains("fcmp oeq %0, %6"), "{text}");
4889 assert_eq!(text.matches(" = zext.i32 ").count(), 4, "{text}");
4893 assert_eq!(text.matches(" = xor ").count(), 4, "{text}");
4894 assert!(!text.contains("call"), "{text}");
4895
4896 let text = body(concat!(
4899 "double g(void);\n",
4900 "int f(void) { return __builtin_fpclassify(0, 1, 2, 3, 4, g()); }\n",
4901 ));
4902 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4903 }
4904
4905 #[test]
4912 fn the_last_three_classification_builtins_fold_where_their_operand_is_a_constant() {
4913 let text = ir(concat!(
4914 "int a = __builtin_isnormal(1.0);\n",
4915 "int b = __builtin_isnormal(0.0);\n",
4916 "int c = __builtin_isnormal(1.0 / 0.0);\n",
4917 "int d = __builtin_isinf_sign(-1.0 / 0.0);\n",
4918 "int e = __builtin_isinf_sign(1.0);\n",
4919 "int g = __builtin_fpclassify(0, 1, 2, 3, 4, 0.0);\n",
4920 "int h = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0);\n",
4921 "int i = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0 / 0.0);\n",
4922 ));
4923 assert!(text.contains("global @a : i32 = 1,"), "{text}");
4924 assert!(text.contains("global @b : i32 = 0,"), "{text}");
4925 assert!(text.contains("global @c : i32 = 0,"), "{text}");
4926 assert!(text.contains("global @d : i32 = -1,"), "{text}");
4927 assert!(text.contains("global @e : i32 = 0,"), "{text}");
4928 assert!(text.contains("global @g : i32 = 4,"), "{text}");
4929 assert!(text.contains("global @h : i32 = 2,"), "{text}");
4930 assert!(text.contains("global @i : i32 = 1,"), "{text}");
4931 }
4932
4933 #[test]
4939 fn fpclassify_refuses_an_answer_that_is_not_an_integer_constant() {
4940 let mut opts = options();
4941 opts.emit = EmitKind::Ir;
4942 let source = concat!(
4943 "int a(double x, int n) { return __builtin_fpclassify(0, 1, n, 3, 4, x); }\n",
4944 "int b(double x) { return __builtin_fpclassify(0, 1, 2, 3, x); }\n",
4945 "int c(int x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n",
4946 );
4947 let messages = run(&opts, source).messages;
4948 assert_eq!(
4949 messages,
4950 [
4951 "/main.c:1:60: error: non-const integer argument 3 in call to function \
4952 '__builtin_fpclassify' [E0687]",
4953 "/main.c:2:26: error: too few arguments to function '__builtin_fpclassify' \
4954 [E0511]",
4955 "/main.c:3:23: error: non-floating-point argument in call to function \
4956 '__builtin_fpclassify' [E0685]",
4957 ]
4958 );
4959 }
4960
4961 #[test]
4969 fn a_builtin_whose_answer_is_a_constant_is_one_and_not_a_call() {
4970 let text = ir(concat!(
4971 "double a = __builtin_inf();\n",
4972 "float b = __builtin_huge_valf();\n",
4973 "long double c = __builtin_infl();\n",
4974 "double d = __builtin_huge_val();\n",
4975 ));
4976 assert!(text.contains("global @a : f64 = 0x7ff0000000000000,"), "{text}");
4977 assert!(text.contains("global @b : f32 = 0x7f800000,"), "{text}");
4978 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
4979 assert!(text.contains("global @d : f64 = 0x7ff0000000000000,"), "{text}");
4980 assert!(!text.contains("call"), "{text}");
4981 }
4982
4983 #[test]
4992 fn a_nan_is_written_with_the_payload_the_program_asked_for() {
4993 let text = ir(concat!(
4994 "double a = __builtin_nan(\"\");\n",
4995 "double b = __builtin_nan(\"0x1\");\n",
4996 "double c = __builtin_nan(\"010\");\n",
4998 "double d = __builtin_nans(\"\");\n",
4999 "double e = __builtin_nans(\"0x1\");\n",
5000 "float f = __builtin_nanf(\"0x1\");\n",
5001 "float g = __builtin_nansf(\"\");\n",
5002 "long double h = __builtin_nansl(\"\");\n",
5003 ));
5004 assert!(text.contains("global @a : f64 = 0x7ff8000000000000,"), "{text}");
5005 assert!(text.contains("global @b : f64 = 0x7ff8000000000001,"), "{text}");
5006 assert!(text.contains("global @c : f64 = 0x7ff8000000000008,"), "{text}");
5007 assert!(text.contains("global @d : f64 = 0x7ff4000000000000,"), "{text}");
5008 assert!(text.contains("global @e : f64 = 0x7ff0000000000001,"), "{text}");
5009 assert!(text.contains("global @f : f32 = 0x7fc00001,"), "{text}");
5010 assert!(text.contains("global @g : f32 = 0x7fa00000,"), "{text}");
5011 assert!(text.contains("f80 0x7fffa000000000000000"), "{text}");
5012
5013 let text = ir(concat!(
5016 "double f(const char *p) { return __builtin_nan(p); }\n",
5017 "double g(void) { return __builtin_nans(\"1x\"); }\n",
5018 ));
5019 assert_eq!(text.matches("call @nan(").count(), 1, "{text}");
5020 assert_eq!(text.matches("call @nans(").count(), 1, "{text}");
5021 }
5022
5023 #[test]
5031 fn the_length_and_the_order_of_a_string_literal_are_known_here() {
5032 let text = ir(concat!(
5033 "unsigned long a = __builtin_strlen(\"hello\");\n",
5034 "unsigned long b = __builtin_strlen(\"a\\0bc\");\n",
5035 "int c = __builtin_strcmp(\"X\", \"X\\376\") < 0;\n",
5036 "int d = __builtin_strcmp(\"abc\", \"abc\");\n",
5037 "int e = __builtin_strcmp(\"abc\", \"ab\") > 0;\n",
5038 ));
5039 assert!(text.contains("global @a : i64 = 5,"), "{text}");
5040 assert!(text.contains("global @b : i64 = 1,"), "{text}");
5041 assert!(text.contains("global @c : i32 = 1,"), "{text}");
5042 assert!(text.contains("global @d : i32 = 0,"), "{text}");
5043 assert!(text.contains("global @e : i32 = 1,"), "{text}");
5044 assert!(!text.contains("call"), "{text}");
5045
5046 let text = ir("unsigned long f(const char *p) { return __builtin_strlen(p); }\n");
5048 assert!(text.contains("call @strlen("), "{text}");
5049 }
5050
5051 #[test]
5058 fn a_sign_builtin_is_a_mask_over_the_bits_and_not_a_call() {
5059 let text = body("double f(double x) { return __builtin_fabs(x); }\n");
5060 assert!(text.contains("bitcast.i64 %0"), "{text}");
5061 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
5062 assert!(text.contains("and %1, %2"), "{text}");
5063 assert!(text.contains("bitcast.f64 %3"), "{text}");
5064 assert!(!text.contains("call"), "{text}");
5065
5066 let text = body("double f(double x, double y) { return __builtin_copysign(x, y); }\n");
5067 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
5068 assert!(text.contains("%8 = or %4, %7"), "{text}");
5069 assert!(!text.contains("call"), "{text}");
5070
5071 let text = body("long double f(long double x) { return __builtin_fabsl(x); }\n");
5074 assert!(text.contains("bitcast.i80 %0"), "{text}");
5075 assert!(text.contains("bitcast.f80"), "{text}");
5076
5077 let text = body("double f(float x) { return __builtin_fabs(x); }\n");
5080 assert!(text.contains("fpext.f64 %0"), "{text}");
5081 assert!(text.contains("bitcast.i64 %1"), "{text}");
5082 }
5083
5084 #[test]
5093 fn the_plain_math_names_are_the_same_mask_and_not_a_call() {
5094 let text =
5095 body(concat!("double fabs(double x);\n", "double f(double x) { return fabs(x); }\n",));
5096 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
5097 assert!(!text.contains("call"), "{text}");
5098
5099 let text =
5100 body(concat!("float fabsf(float x);\n", "float f(float x) { return fabsf(x); }\n",));
5101 assert!(text.contains("bitcast.i32 %0"), "{text}");
5102 assert!(!text.contains("call"), "{text}");
5103
5104 let text = body(concat!(
5105 "double copysign(double x, double y);\n",
5106 "double f(double x, double y) { return copysign(x, y); }\n",
5107 ));
5108 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
5109 assert!(!text.contains("call"), "{text}");
5110
5111 let text = body(concat!(
5112 "float copysignf(float x, float y);\n",
5113 "float f(float x, float y) { return copysignf(x, y); }\n",
5114 ));
5115 assert!(!text.contains("call"), "{text}");
5116
5117 let text = ir(concat!(
5121 "long double fabsl(long double x);\n",
5122 "long double f(long double x) { return fabsl(x); }\n",
5123 ));
5124 assert!(text.contains("call @fabsl"), "{text}");
5125 }
5126
5127 #[test]
5135 fn a_plain_math_name_the_program_took_is_the_programs_own_function() {
5136 let taken = concat!(
5137 "static double fabs(double b) { return 7; }\n",
5138 "double f(double x) { return fabs(x); }\n",
5139 );
5140 assert!(ir(taken).contains("call @fabs"), "a static definition is the program's own");
5141
5142 let retyped = concat!("int fabs(int b);\n", "int f(int x) { return fabs(x); }\n");
5143 assert!(ir(retyped).contains("call @fabs"), "another type is another function");
5144
5145 let plain = concat!("double fabs(double b);\n", "double f(double x) { return fabs(x); }\n");
5146 let mut opts = options();
5147 opts.emit = EmitKind::Ir;
5148 assert!(!run(&opts, plain).text().contains("call @fabs"), "the library's by default");
5149
5150 opts.builtins = false;
5151 assert!(run(&opts, plain).text().contains("call @fabs"), "-fno-builtin");
5152
5153 opts.builtins = true;
5154 opts.no_builtin = vec!["fabs".to_owned()];
5155 assert!(run(&opts, plain).text().contains("call @fabs"), "-fno-builtin-fabs");
5156 let one = concat!(
5157 "double copysign(double a, double b);\n",
5158 "double f(double x) { return copysign(x, 1.0); }\n",
5159 );
5160 assert!(!run(&opts, one).text().contains("call @copysign"), "one name and not the family");
5161
5162 opts.no_builtin = Vec::new();
5164 opts.builtins = false;
5165 let prefixed = "double f(double x) { return __builtin_fabs(x); }\n";
5166 assert!(!run(&opts, prefixed).text().contains("call @fabs"), "the prefix is not a library");
5167 }
5168
5169 #[test]
5178 fn the_sign_builtins_answer_a_zero_and_a_nan_the_way_the_bits_say() {
5179 let text = ir(concat!(
5180 "double a = __builtin_fabs(-3.5);\n",
5181 "double b = __builtin_copysign(1.0, -0.0);\n",
5182 "double c = __builtin_copysign(0.0, -2.0);\n",
5183 "double d = __builtin_copysign(-__builtin_nan(\"\"), 1.0);\n",
5185 "double e = __builtin_fabs(-__builtin_nan(\"0x1\"));\n",
5186 "float g = __builtin_copysignf(-0.0f, 2.0f);\n",
5187 "long double h = __builtin_copysignl(1.0L, -1.0L);\n",
5188 "long double i = __builtin_fabsl(-__builtin_infl());\n",
5189 ));
5190 assert!(text.contains("global @a : f64 = 0x400c000000000000,"), "{text}");
5191 assert!(text.contains("global @b : f64 = 0xbff0000000000000,"), "{text}");
5192 assert!(text.contains("global @c : f64 = 0x8000000000000000,"), "{text}");
5193 assert!(text.contains("global @d : f64 = 0x7ff8000000000000,"), "{text}");
5194 assert!(text.contains("global @e : f64 = 0x7ff8000000000001,"), "{text}");
5195 assert!(text.contains("global @g : f32 = 0x0,"), "{text}");
5196 assert!(text.contains("f80 0xbfff8000000000000000"), "{text}");
5197 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
5198 }
5199
5200 #[test]
5208 fn the_complex_builtins_are_the_halves_of_the_value_and_not_a_call() {
5209 let text = body("double f(_Complex double z) { return __builtin_creal(z); }\n");
5210 assert!(!text.contains("call"), "{text}");
5211 let text = body("double f(_Complex double z) { return __builtin_cimag(z); }\n");
5212 assert!(!text.contains("call"), "{text}");
5213
5214 let text = body("_Complex double f(_Complex double z) { return __builtin_conj(z); }\n");
5217 assert_eq!(text.matches("fneg").count(), 1, "{text}");
5218 assert!(!text.contains("call"), "{text}");
5219 let negated = body("_Complex double f(_Complex double z) { return -z; }\n");
5220 assert_eq!(negated.matches("fneg").count(), 2, "{negated}");
5221
5222 let written = body("_Complex double f(_Complex double z) { return ~z; }\n");
5225 assert_eq!(written, text, "the name and the operator are the same thing");
5226
5227 let text = body(concat!(
5229 "double creal(_Complex double z);\n",
5230 "double f(_Complex double z) { return creal(z); }\n",
5231 ));
5232 assert!(!text.contains("call"), "{text}");
5233 let text = body(concat!(
5234 "_Complex float conjf(_Complex float z);\n",
5235 "_Complex float f(_Complex float z) { return conjf(z); }\n",
5236 ));
5237 assert_eq!(text.matches("fneg").count(), 1, "{text}");
5238 assert!(!text.contains("call"), "{text}");
5239
5240 let taken = concat!(
5243 "static double creal(_Complex double z) { return 7; }\n",
5244 "double f(_Complex double z) { return creal(z); }\n",
5245 );
5246 assert!(ir(taken).contains("call @creal"), "a static definition is the program's own");
5247 let retyped = concat!("int cimag(int z);\n", "int f(int z) { return cimag(z); }\n");
5248 assert!(ir(retyped).contains("call @cimag"), "another type is another function");
5249 let plain = concat!(
5250 "double cimag(_Complex double z);\n",
5251 "double f(_Complex double z) { return cimag(z); }\n",
5252 );
5253 let mut opts = options();
5254 opts.emit = EmitKind::Ir;
5255 opts.builtins = false;
5256 assert!(run(&opts, plain).text().contains("call @cimag"), "-fno-builtin");
5257 opts.builtins = true;
5258 opts.no_builtin = vec!["cimag".to_owned()];
5259 assert!(run(&opts, plain).text().contains("call @cimag"), "-fno-builtin-cimag");
5260
5261 let text = ir(concat!(
5263 "double a = __builtin_creal(1.5 + 2.5i);\n",
5264 "double b = __builtin_cimag(1.5 + 2.5i);\n",
5265 "_Complex double c = __builtin_conj(1.5 + 2.5i);\n",
5266 ));
5267 assert!(text.contains("global @a : f64 = 0x3ff8000000000000,"), "{text}");
5268 assert!(text.contains("global @b : f64 = 0x4004000000000000,"), "{text}");
5269 assert!(
5270 text.contains("{ f64 0x3ff8000000000000, f64 0xc004000000000000 }"),
5271 "the conjugate of a constant is the constant with the second half negated: {text}"
5272 );
5273 assert!(!text.contains("call"), "{text}");
5274 }
5275
5276 #[test]
5284 fn a_math_library_builtin_of_a_constant_is_the_answer_and_not_a_call() {
5285 let text = ir(concat!(
5286 "double a = __builtin_ceil(1.5);\n",
5287 "double b = __builtin_floor(1.5);\n",
5288 "double c = __builtin_trunc(-1.5);\n",
5289 "double d = __builtin_round(2.5);\n",
5292 "double e = __builtin_ceil(-0.5);\n",
5294 "double f = __builtin_fmax(1.0, 2.0);\n",
5295 "double g = __builtin_fmin(1.0, 2.0);\n",
5296 "float h = __builtin_ceilf(1.25f);\n",
5297 "double ceil(double x);\n",
5300 "double i = ceil(2.25);\n",
5301 ));
5302 assert!(text.contains("global @a : f64 = 0x4000000000000000,"), "{text}");
5303 assert!(text.contains("global @b : f64 = 0x3ff0000000000000,"), "{text}");
5304 assert!(text.contains("global @c : f64 = 0xbff0000000000000,"), "{text}");
5305 assert!(text.contains("global @d : f64 = 0x4008000000000000,"), "{text}");
5306 assert!(text.contains("global @e : f64 = 0x8000000000000000,"), "{text}");
5307 assert!(text.contains("global @f : f64 = 0x4000000000000000,"), "{text}");
5308 assert!(text.contains("global @g : f64 = 0x3ff0000000000000,"), "{text}");
5309 assert!(text.contains("global @h : f32 = 0x40000000,"), "{text}");
5310 assert!(text.contains("global @i : f64 = 0x4008000000000000,"), "{text}");
5311 assert!(!text.contains("call"), "{text}");
5312 }
5313
5314 #[test]
5322 fn a_math_library_builtin_of_anything_else_is_a_call_to_the_library() {
5323 let text = ir(concat!(
5324 "double f(double x) { return __builtin_ceil(x); }\n",
5325 "float g(float x) { return __builtin_floorf(x); }\n",
5326 "double h(double x, double y) { return __builtin_fmax(x, y); }\n",
5327 ));
5328 assert!(text.contains("call @ceil("), "{text}");
5329 assert!(text.contains("call @floorf("), "{text}");
5330 assert!(text.contains("call @fmax("), "{text}");
5331
5332 let text = ir(concat!(
5336 "double f(void) { return __builtin_rint(2.5); }\n",
5337 "double g(void) { return __builtin_nearbyint(2.5); }\n",
5338 ));
5339 assert!(text.contains("call @rint("), "{text}");
5340 assert!(text.contains("call @nearbyint("), "{text}");
5341
5342 let text = ir("double f(void) { return __builtin_fmin(__builtin_nan(\"\"), 1.0); }\n");
5345 assert!(text.contains("call @fmin("), "{text}");
5346
5347 let plain = concat!("double ceil(double x);\n", "double f(void) { return ceil(2.25); }\n");
5350 let mut opts = options();
5351 opts.emit = EmitKind::Ir;
5352 opts.no_builtin = vec!["ceil".to_owned()];
5353 assert!(run(&opts, plain).text().contains("call @ceil("), "-fno-builtin-ceil");
5354 }
5355
5356 #[test]
5363 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
5364 let text = ir(concat!(
5365 "constexpr int side = 4;\n",
5366 "constexpr int wider = side + 1;\n",
5367 "constexpr double half = 1.5;\n",
5368 "struct point { int x; int y; };\n",
5369 "constexpr struct point origin = { 5, 6 };\n",
5370 "int square[side * side];\n",
5371 "int rectangle[wider];\n",
5372 "int rounded[(int)half * 2];\n",
5373 "int across[origin.y];\n",
5374 "enum named { four = side };\n",
5375 "int e = four;\n",
5376 ));
5377 assert!(text.contains("global @square : bytes 64 ="), "{text}");
5378 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
5379 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
5380 assert!(text.contains("global @across : bytes 24 ="), "{text}");
5381 assert!(text.contains("global @e : i32 = 4,"), "{text}");
5382
5383 let mut opts = options();
5386 opts.emit = EmitKind::Ir;
5387 let konst = "const int n = 1;\nint a[n];\n";
5388 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
5389 assert_eq!(run(&opts, konst).messages, [message]);
5390
5391 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
5393 assert_eq!(run(&opts, subscript).messages, [message]);
5394
5395 let address = "constexpr int c = 3;\nint *p = &c;\n";
5397 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
5398 pointer target type [E0514]";
5399 assert_eq!(run(&opts, address).messages, [warning]);
5400 }
5401
5402 #[test]
5411 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
5412 let mut opts = options();
5415 opts.std = Std::C17;
5416 let source = concat!(
5417 "int add(a, b)\n",
5418 "int a;\n",
5419 "int b;\n",
5420 "{ return a + b; }\n",
5421 "int promoted(c)\n",
5422 "char c;\n",
5423 "{ return c; }\n",
5424 "int narrow(char);\n",
5425 "int narrow(c)\n",
5426 "char c;\n",
5427 "{ return c; }\n",
5428 "int first(a)\n",
5429 "int a[4];\n",
5430 "{ return a[0]; }\n",
5431 );
5432 let result = run(&opts, source);
5433 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
5434 let text = result.text();
5435 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
5436 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
5437 assert!(text.contains("c : char object automatic defined"), "{text}");
5439 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
5440 assert!(text.contains("first : int(int *) function external defined"), "{text}");
5442 }
5443
5444 #[test]
5451 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
5452 let mut opts = options();
5453 opts.std = Std::C17;
5454 for (source, message) in [
5455 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
5456 (
5457 "int f(a)\nint a;\nint b;\n{ return a; }\n",
5458 "3:5: error: declaration for parameter 'b' but no such parameter",
5459 ),
5460 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
5461 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
5462 (
5463 "int f(a)\nstatic int a;\n{ return a; }\n",
5464 "2:12: error: storage class specified for parameter 'a'",
5465 ),
5466 (
5467 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
5468 "2:7: error: argument 'a' doesn't match prototype",
5469 ),
5470 ] {
5471 let result = run(&opts, source);
5472 assert!(result.failed(), "expected this to fail:\n{source}");
5473 assert!(result.messages[0].contains(message), "{:?}", result.messages);
5474 }
5475
5476 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
5479 let mut older = options();
5480 older.std = Std::C89;
5481 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
5482 let result = run(&opts, implicit);
5483 assert!(
5484 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
5485 "{:?}",
5486 result.messages
5487 );
5488
5489 let mut newer = options();
5493 newer.std = Std::C23;
5494 let plain = "int f(a)\nint a;\n{ return a; }\n";
5495 let result = run(&newer, plain);
5496 assert!(!result.failed(), "{:?}", result.messages);
5497 assert_eq!(
5498 result.messages,
5499 ["/main.c:1:5: warning: old-style function definition [E0412]"]
5500 );
5501 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
5502 }
5503
5504 #[test]
5511 fn the_obsolete_designators_are_taken_and_are_pedantic_warnings() {
5512 let array = "int a[8] = { [3] 7 };\n";
5513 let member = "struct s { int x; } v = { x: 7 };\n";
5514 for source in [array, member] {
5515 let result = run(&options(), source);
5516 assert!(!result.failed(), "{:?}", result.messages);
5517 assert!(result.messages.is_empty(), "nothing to say: {:?}", result.messages);
5518 }
5519
5520 let mut asked = options();
5521 asked.pedantic = true;
5522 assert_eq!(
5523 run(&asked, array).messages,
5524 ["/main.c:1:18: warning: obsolete designator, write `[i] =` instead [E0415]"]
5525 );
5526 assert_eq!(
5527 run(&asked, member).messages,
5528 ["/main.c:1:27: warning: obsolete designator, write `.field =` instead [E0413]"]
5529 );
5530 }
5531
5532 #[test]
5539 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
5540 let text = ir(concat!(
5541 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
5542 "struct brim { char buf[9223372036854775807L]; };\n",
5543 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
5544 "unsigned long h = sizeof(struct huge_struct);\n",
5545 "unsigned long b = sizeof(struct brim);\n",
5546 "unsigned long y = sizeof(struct bitty);\n",
5547 ));
5548 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
5549 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
5550 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
5551
5552 let mut opts = options();
5553 opts.emit = EmitKind::Ir;
5554 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
5555 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
5556 assert_eq!(run(&opts, over).messages, [message]);
5557 let array = "struct wide { short buf[1L << 62]; };\n";
5558 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
5559 maximum object size '9223372036854775807' [E0537]";
5560 assert_eq!(run(&opts, array).messages[0], message);
5561 }
5562
5563 fn compile_bytes(source: &[u8]) -> Compiled {
5568 let mut opts = options();
5569 opts.emit = EmitKind::Ir;
5570 let mut fs = MemoryFileSystem::new();
5571 fs.insert("/main.c", source.to_vec());
5572 compile(&opts, "/main.c", &fs)
5573 }
5574
5575 #[test]
5582 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
5583 let mut source = b"char s[] = \"a".to_vec();
5584 source.push(0xff);
5585 source.extend_from_slice(b"b\";\nchar c = '");
5586 source.push(0xff);
5587 source.extend_from_slice(b"';\n");
5588 let result = compile_bytes(&source);
5589 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
5590 assert!(result.text().contains(r#"bytes "a\ffb\00""#), "{}", result.text());
5591 assert!(result.text().contains("global @c : i8 = -1,"), "{}", result.text());
5593
5594 let mut stray = b"int a".to_vec();
5595 stray.push(0xff);
5596 stray.extend_from_slice(b" = 1;\n");
5597 let result = compile_bytes(&stray);
5598 assert!(
5599 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
5600 "{:?}",
5601 result.messages
5602 );
5603 }
5604
5605 #[test]
5606 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
5607 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
5608 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
5609 let expected = "\
5610func @add(i32, i32) -> i32, linkage(external) {
5611block0(%0: i32, %1: i32):
5612 %2 = add.nsw %0, %1
5613 return %2
5614}
5615";
5616 assert!(text.contains(expected), "{text}");
5617 }
5618
5619 #[test]
5620 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
5621 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
5622 assert!(!text.contains("alloca"), "{text}");
5623 assert!(!text.contains("load"), "{text}");
5624 assert!(!text.contains("store"), "{text}");
5625 }
5626
5627 #[test]
5628 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
5629 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
5630 let expected = "\
5631block0:
5632 %0 = alloca, size 4, align 4
5633 %1 = iconst.i32 1
5634 store %1 -> %0, align 4, tbaa !1
5635 %2 = call @g(%0) : (ptr) -> i32
5636 return %2
5637";
5638 assert_eq!(text, expected);
5639 }
5640
5641 #[test]
5642 fn a_loop_carries_what_it_changes_as_block_parameters() {
5643 let text = body(
5646 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
5647 return total;\n}\n",
5648 );
5649 assert!(!text.contains("alloca"), "{text}");
5650 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
5651 assert!(text.contains("jump block1("), "{text}");
5652 }
5653
5654 #[test]
5655 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
5656 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
5657 assert!(text.contains("icmp slt %0, %1"), "{text}");
5658 assert!(!text.contains("zext"), "{text}");
5659 }
5660
5661 #[test]
5662 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
5663 let text = body("int f(int a, int b) { return a && b; }\n");
5664 let expected = "\
5665block0(%0: i32, %1: i32):
5666 %2 = iconst.i32 0
5667 %3 = icmp ne %0, %2
5668 %4 = iconst.i1 0
5669 br_if %3, block1, block2(%4)
5670
5671block1:
5672 %5 = iconst.i32 0
5673 %6 = icmp ne %1, %5
5674 jump block2(%6)
5675
5676block2(%7: i1):
5677 %8 = zext.i32 %7
5678 return %8
5679";
5680 assert_eq!(text, expected);
5681 }
5682
5683 #[test]
5684 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
5685 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
5686 assert!(!text.contains("block3"), "{text}");
5689 assert!(!text.contains("iconst.i32 3"), "{text}");
5690 }
5691
5692 #[test]
5693 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
5694 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
5695 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
5696 assert!(body("int f(void) { }\n").contains("unreachable"));
5697 }
5698
5699 #[test]
5700 fn a_structure_is_copied_rather_than_held_in_a_value() {
5701 let text = body(
5702 "struct point { int x, y; };\n\
5703 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
5704 );
5705 assert!(text.contains("memcpy"), "{text}");
5706 }
5707
5708 #[test]
5709 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
5710 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
5711 assert!(text.contains("memset"), "{text}");
5712 }
5713
5714 #[test]
5715 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
5716 let text = body(
5717 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
5718 default: r = 4; } return r; }\n",
5719 );
5720 let expected = "\
5721block0(%0: i32):
5722 %1 = iconst.i32 0
5723 switch %0, block1, [1 => block2, 2 => block3(%1)]
5724
5725block1:
5726 %2 = iconst.i32 4
5727 jump block4(%2)
5728
5729block2:
5730 %3 = iconst.i32 1
5731 jump block3(%3)
5732
5733block3(%4: i32):
5734 %5 = iconst.i32 2
5735 %6 = add.nsw %4, %5
5736 jump block4(%6)
5737
5738block4(%7: i32):
5739 return %7
5740";
5741 assert_eq!(text, expected);
5742 }
5743
5744 #[test]
5745 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
5746 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
5749 assert!(text.contains("%2 = sub %0, %1"), "{text}");
5750 assert!(text.contains("icmp ule"), "{text}");
5751 assert!(!text.contains("switch"), "{text}");
5752 }
5753
5754 #[test]
5755 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
5756 let text = body(
5757 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
5758 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
5759 );
5760 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
5763 assert!(text.contains("block5:\n jump block7("), "{text}");
5764 assert!(text.contains("block6:\n jump block8("), "{text}");
5765 }
5766
5767 #[test]
5768 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
5769 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
5770 }
5771
5772 #[test]
5773 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
5774 let text = body(
5779 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
5780 return n; }\n",
5781 );
5782 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
5785 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
5786 assert!(text.contains("block4:\n jump block3("), "{text}");
5787 }
5788
5789 #[test]
5790 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
5791 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
5794 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
5795 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
5796 assert!(text.contains("br_if %6, block2, block3"), "{text}");
5797 }
5798
5799 #[test]
5800 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
5801 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
5802 assert!(!text.contains("alloca"), "{text}");
5806 assert!(text.contains("block2(%4: i32):\n return %4"), "{text}");
5807 assert_eq!(text.matches("jump block2(").count(), 2, "{text}");
5808 }
5809
5810 #[test]
5811 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
5812 let text =
5813 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
5814 assert!(!text.contains("alloca"), "{text}");
5815 assert!(text.contains("block1(%2: i32):"), "{text}");
5816 assert!(text.contains("jump block1(%5)"), "{text}");
5817 }
5818
5819 #[test]
5820 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
5821 assert_eq!(
5824 body("int f(int x) { return x; spare: return 0; }\n"),
5825 "block0(%0: i32):\n return %0\n"
5826 );
5827 }
5828
5829 #[test]
5830 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
5831 let text = body(
5832 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
5833 );
5834 assert_eq!(
5837 text,
5838 "\
5839block0(%0: ptr):
5840 %1 = load.i8 %0, align 1
5841 %2 = iconst.i8 3
5842 %3 = ashr %1, %2
5843 %4 = sext.i32 %3
5844 return %4
5845"
5846 );
5847 }
5848
5849 #[test]
5850 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
5851 let text =
5855 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
5856 assert_eq!(
5857 text,
5858 "\
5859block0(%0: ptr, %1: i32):
5860 %2 = iconst.i32 16777215
5861 %3 = and %1, %2
5862 %4 = trunc.i16 %3
5863 store %4 -> %0, align 2
5864 %5 = iconst.i32 16
5865 %6 = lshr %3, %5
5866 %7 = trunc.i8 %6
5867 %8 = iconst.i64 2
5868 %9 = ptr_add %0, %8
5869 store %7 -> %9, align 1
5870 return
5871"
5872 );
5873 }
5874
5875 #[test]
5876 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
5877 let text =
5878 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
5879 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
5882 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
5883 }
5884
5885 #[test]
5886 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
5887 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
5890 assert_eq!(text.matches("ashr").count(), 0, "{text}");
5891 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
5892 }
5893
5894 #[test]
5895 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
5896 let text = body(
5900 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
5901 );
5902 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
5903 }
5904
5905 #[test]
5906 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
5907 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
5910 assert!(
5911 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
5912 "{text}"
5913 );
5914 }
5915
5916 #[test]
5917 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
5918 let text = ir(concat!(
5923 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
5924 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
5925 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
5926 "char s[2] = \"hi\";\n",
5927 ));
5928 assert!(
5929 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
5930 "{text}"
5931 );
5932 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
5933 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
5934 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
5937 }
5938
5939 #[test]
5940 fn a_definition_takes_a_parameter_it_left_unnamed() {
5941 let text = ir("int f(int a, int) { return a; }\n");
5945 assert!(text.contains("func @f(i32, i32) -> i32"), "{text}");
5946 assert!(text.contains("block0(%0: i32, %1: i32):"), "{text}");
5947
5948 let text = ir("int g(int, int n) { return n; }\n");
5951 assert!(text.contains("block0(%0: i32, %1: i32):\n return %1\n"), "{text}");
5952 }
5953
5954 #[test]
5955 fn an_assignment_of_a_structure_is_the_object_it_wrote() {
5956 let text = body(concat!(
5961 "struct s { int f; int g; };\n",
5962 "void h(struct s *a, struct s *c, struct s *d, struct s *e)\n",
5963 "{ *d = *e = a[0] = *c; }\n",
5964 ));
5965 assert_eq!(text.matches("memcpy").count(), 3, "{text}");
5966 assert!(text.contains("memcpy %8, %1, size 8, align 4\n"), "{text}");
5967 assert!(text.contains("memcpy %3, %8, size 8, align 4\n"), "{text}");
5968 assert!(text.contains("memcpy %2, %3, size 8, align 4\n"), "{text}");
5969 }
5970
5971 #[test]
5972 fn a_string_literal_stops_at_the_end_of_the_array_it_is_filling() {
5973 let mut opts = options();
5978 opts.emit = EmitKind::Ir;
5979 let result = run(
5980 &opts,
5981 concat!(
5982 "const char a[2][3] = { \"1234\", \"xyz\" };\n",
5983 "static const char b[3][5] = { \"12345\", \"678\", \"9\" };\n",
5984 "union u { struct { char x[4]; char y[4]; }; struct { char z[8]; }; };\n",
5985 "const union u c = { { \"1234\", \"567\" } };\n",
5986 ),
5987 );
5988 let text = result.text();
5989 assert_eq!(
5990 result.messages,
5991 ["/main.c:1:24: warning: initializer-string for array of 'const char' is too long \
5992 (5 chars into 3 available) [E0637]"]
5993 );
5994 assert!(text.contains("global @a : bytes 6 = { bytes \"123\", bytes \"xyz\" }"), "{text}");
5995 assert!(
5996 text.contains(
5997 "global @b : bytes 15 = { bytes \"12345\", bytes \"678\\00\", zero 1, \
5998 bytes \"9\\00\", zero 3 }"
5999 ),
6000 "{text}"
6001 );
6002 assert!(
6005 text.contains("global @c : bytes 8 = { bytes \"1234\", bytes \"567\\00\" }"),
6006 "{text}"
6007 );
6008 }
6009
6010 #[test]
6011 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
6012 let text = body(concat!(
6016 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
6017 "void g(struct v *);\n",
6018 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
6019 ));
6020 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
6021 }
6022
6023 #[test]
6024 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
6025 let text = ir(concat!(
6030 "struct s { int x; };\n",
6031 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
6032 "int n = (int){ 7 };\n",
6033 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
6034 ));
6035 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
6036 assert!(text.contains("global @n : i32 = 7,"), "{text}");
6037 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
6040 }
6041
6042 #[test]
6043 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
6044 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
6048 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
6049 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
6050 }
6051
6052 #[test]
6053 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
6054 let text = ir("unsigned char foo[1][0];\n");
6058 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
6059 }
6060
6061 #[test]
6062 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
6063 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
6066 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
6067 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
6068 }
6069
6070 #[test]
6071 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
6072 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
6076 assert!(
6077 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
6078 "{text}"
6079 );
6080 }
6081
6082 #[test]
6083 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
6084 let text = body(
6089 "\
6090struct s { int a, b; };
6091struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
6092",
6093 );
6094 assert!(text.contains("block3(%7: ptr)"), "{text}");
6096 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
6097 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
6098 }
6099
6100 #[test]
6108 fn the_left_side_of_a_conditional_with_no_middle_is_evaluated_once() {
6109 let text = body("int f(int i) { return ++i ?: 10; }\n");
6110 assert!(text.contains("jump block3(%2)"), "the arm is the value that was tested: {text}");
6111 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
6112
6113 let text = body("long f(int i) { return ++i ?: 10L; }\n");
6116 assert!(text.contains("%5 = sext.i64 %2"), "the arm widens what was tested: {text}");
6117 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
6118
6119 let text = body("int g(void);\nint f(void) { return g() ?: 10; }\n");
6121 assert_eq!(text.matches("call @g").count(), 1, "called once: {text}");
6122
6123 let text = body("int f(int i) { return ++i ? ++i : 10; }\n");
6126 assert_eq!(text.matches("add.nsw").count(), 2, "incremented twice: {text}");
6127 }
6128
6129 #[test]
6130 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
6131 let text = ir("\
6135struct pair { int a, b; };
6136struct pair make(int a, int b);
6137struct pair twice(struct pair p) { return make(p.a, p.b); }
6138");
6139 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
6140 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
6141 }
6142
6143 #[test]
6144 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
6145 let text = ir("\
6149struct big { double v[8]; };
6150struct big grow(struct big b);
6151struct big twice(struct big b) { return grow(grow(b)); }
6152");
6153 assert!(
6154 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
6155 "{text}"
6156 );
6157 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
6158 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
6161 }
6162
6163 #[test]
6164 fn a_structure_passed_to_a_variadic_function_says_so_at_the_call() {
6165 let text = ir("\
6170struct big { double v[8]; };
6171struct pair { int a, b; };
6172int p(const char *, ...);
6173int f(struct big b, struct pair q) { return p(\"\", 1, b, q); }
6174");
6175 assert!(
6176 text.contains("call @p(%4, %5, %2 byval(64, align 8), %6) : (ptr, ...) -> i32"),
6177 "{text}"
6178 );
6179 }
6180
6181 #[test]
6182 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
6183 let body = body(
6186 "\
6187struct pair { int a, b; };
6188struct pair make(int a, int b);
6189int second(void) { return make(1, 2).b; }
6190",
6191 );
6192 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
6193 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
6194 }
6195
6196 #[test]
6197 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
6198 let source = "\
6202struct hfa { float x, y, z; };
6203int take(struct hfa h);
6204int give(struct hfa h) { return take(h); }
6205";
6206 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
6207 let mut opts = options();
6208 opts.emit = EmitKind::Ir;
6209 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
6210 let result = run(&opts, source);
6211 assert_eq!(result.messages, Vec::<String>::new());
6212 assert!(result.text().contains("func @take(f32, f32, f32) -> i32"), "{}", result.text());
6213 }
6214
6215 #[test]
6216 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
6217 let source = "\
6220int use(int *);
6221void f(int n) {
6222 {
6223 int a[n];
6224 use(a);
6225 }
6226 use(0);
6227}
6228";
6229 let body = body(source);
6230 assert!(body.contains("mul.nsw"), "{body}");
6231 assert!(body.contains("stacksave"), "{body}");
6232 assert!(body.contains("alloca %"), "{body}");
6233 assert!(body.contains("stackrestore"), "{body}");
6234 }
6235
6236 #[test]
6237 fn a_goto_out_of_the_scope_of_one_gives_its_stack_back_on_the_way() {
6238 let source = "\
6243int use(int *);
6244int f(int n) {
6245 {
6246 int a[n];
6247 if (use(a)) goto out;
6248 use(0);
6249 }
6250out:
6251 return 0;
6252}
6253";
6254 let body = body(source);
6255 assert_eq!(body.matches("stackrestore").count(), 2, "{body}");
6257 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
6258 assert!(after.starts_with(" %4\n jump block"), "{body}");
6259 }
6260
6261 #[test]
6262 fn a_goto_to_a_label_the_array_is_still_alive_at_leaves_the_stack_alone() {
6263 let source = "\
6267int use(int *);
6268int f(int n) {
6269 int a[n];
6270again:
6271 if (use(a)) goto again;
6272 return 0;
6273}
6274";
6275 let body = body(source);
6276 assert!(body.contains("stacksave"), "{body}");
6277 assert!(!body.contains("stackrestore"), "{body}");
6278 }
6279
6280 #[test]
6281 fn a_goto_back_to_a_label_in_front_of_one_gives_it_back_every_time_round() {
6282 let source = "\
6287int use(int *);
6288int f(int n) {
6289again:
6290 {
6291 int a[n];
6292 if (use(a)) goto again;
6293 }
6294 return 0;
6295}
6296";
6297 let body = body(source);
6298 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
6299 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
6300 assert!(after.starts_with(" %4\n jump block1\n"), "{body}");
6301 }
6302
6303 #[test]
6304 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
6305 let source = "\
6311int f(void);
6312void t(void) {
6313 int count = 10;
6314 for (; count--;) {
6315 int b[f()];
6316 int i;
6317 for (i = 0; i < f(); i++) {
6318 b[i] = count;
6319 }
6320 }
6321}
6322";
6323 let body = body(source);
6324 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
6328 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
6329 let next = after.split("\n\n").next().expect("the block the restore is in");
6332 assert!(next.contains("jump block1("), "{body}");
6333 }
6334
6335 #[test]
6336 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
6337 let source = "\
6340unsigned long f(int n) {
6341 int a[n];
6342 n = 0;
6343 return sizeof a;
6344}
6345";
6346 let body = body(source);
6347 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
6349 }
6350
6351 #[test]
6352 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
6353 let source = "\
6356int use(int);
6357int f(int x) {
6358 return ({
6359 int t = use(x);
6360 t * t;
6361 });
6362}
6363";
6364 let expected = "\
6365block0(%0: i32):
6366 %1 = call @use(%0) : (i32) -> i32
6367 %2 = mul.nsw %1, %1
6368 return %2
6369";
6370 assert_eq!(body(source), expected);
6371 }
6372
6373 #[test]
6374 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
6375 let source = "int f(int x) { return ({ return x; 0; }); }\n";
6379 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
6380 }
6381
6382 #[test]
6383 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
6384 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
6388 let expected = "\
6389block0(%0: ptr):
6390 %1 = va_arg.f64 %0
6391 %2 = va_arg.f64 %0
6392 %3 = fadd %1, %2
6393 return %3
6394";
6395 assert_eq!(body(source), expected);
6396 }
6397
6398 #[test]
6399 fn one_that_reads_a_structure_answers_where_the_object_is() {
6400 let source = "\
6414struct s { int a; long b; };
6415long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.b; }
6416";
6417 let expected = "\
6418block0(%0: ptr):
6419 %1 = alloca, size 16, align 16
6420 %2 = va_object %0, size 16, align 8, in(int 8 at 0, int 8 at 8)
6421 memcpy %1, %2, size 16, align 8
6422 %3 = iconst.i64 8
6423 %4 = ptr_add %1, %3
6424 %5 = load.i64 %4, align 8, tbaa !1
6425 return %5
6426";
6427 assert_eq!(body(source), expected);
6428 }
6429
6430 #[test]
6434 fn the_classification_says_which_registers_the_object_arrived_in() {
6435 let source = "\
6436struct s { double a; double b; };
6437double f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a; }
6438";
6439 assert!(
6440 body(source)
6441 .contains("va_object %0, size 16, align 8, in(float f64 at 0, float f64 at 8)"),
6442 "{}",
6443 body(source)
6444 );
6445
6446 let big = "\
6447struct s { long a[4]; };
6448long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a[0]; }
6449";
6450 assert!(body(big).contains("va_object %0, size 32, align 8\n"), "{}", body(big));
6451 }
6452
6453 #[test]
6454 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
6455 let source = "\
6459int f(int c) {
6460 void *p = c ? &&one : &&two;
6461 goto *p;
6462one:
6463 return 1;
6464two:
6465 return 2;
6466}
6467";
6468 let expected = "\
6469block0(%0: i32):
6470 %1 = iconst.i32 0
6471 %2 = icmp ne %0, %1
6472 br_if %2, block1, block2
6473
6474block1:
6475 %3 = block_addr block3
6476 jump block4(%3)
6477
6478block2:
6479 %4 = block_addr block5
6480 jump block4(%4)
6481
6482block3:
6483 %5 = iconst.i32 1
6484 return %5
6485
6486block4(%6: ptr):
6487 indirect_br %6, block3, block5
6488
6489block5:
6490 %7 = iconst.i32 2
6491 return %7
6492";
6493 assert_eq!(body(source), expected);
6494 }
6495
6496 #[test]
6497 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
6498 let source = "void **next(void);
6501void f(void) { goto *next(); }
6502";
6503 let expected = "\
6504block0:
6505 %0 = call @next() : () -> ptr
6506 unreachable
6507";
6508 assert_eq!(body(source), expected);
6509 }
6510
6511 #[test]
6512 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
6513 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
6516 let expected = "\
6517block0:
6518 inline_asm.volatile \"mfence\", \"\", \"memory\"()
6519 return
6520";
6521 assert_eq!(body(source), expected);
6522 }
6523
6524 #[test]
6525 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
6526 let source = "\
6529int f(int x, int y) {
6530 int r;
6531 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
6532 return r + y;
6533}
6534";
6535 let expected = "\
6536block0(%0: i32, %1: i32):
6537 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
6538 %4 = add.nsw %2, %3
6539 return %4
6540";
6541 assert_eq!(body(source), expected);
6542 }
6543
6544 #[test]
6545 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
6546 let source = "\
6551struct pair { int a, b; };
6552int f(int x) {
6553 int slot = x;
6554 struct pair p = { x, x };
6555 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
6556 return slot + p.a;
6557}
6558";
6559 let text = body(source);
6560 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
6561 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
6562 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
6563 }
6564
6565 #[test]
6566 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
6567 let source = "\
6572int f(int x) {
6573 int r = 7;
6574 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
6575 return r;
6576away:
6577 return r;
6578}
6579";
6580 let expected = "\
6581block0(%0: i32):
6582 %1 = iconst.i32 7
6583 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
6584
6585block1:
6586 return %2
6587
6588block2:
6589 return %1
6590";
6591 assert_eq!(body(source), expected);
6592 }
6593
6594 #[test]
6595 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
6596 let mut opts = options();
6600 opts.emit = EmitKind::Ir;
6601 for (source, expected) in [
6602 (
6603 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
6604 "output operand constraint lacks '='",
6605 ),
6606 (
6607 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
6608 "lvalue required in 'asm' statement",
6609 ),
6610 (
6611 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
6612 "read-only variable 'g' used as 'asm' output",
6613 ),
6614 (
6615 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
6616 "input operand constraint contains '='",
6617 ),
6618 (
6619 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
6620 "memory input 0 is not directly addressable",
6621 ),
6622 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
6623 (
6624 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
6625 "duplicate asm operand name 'a'",
6626 ),
6627 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
6628 ] {
6629 let result = run(&opts, source);
6630 assert!(result.failed(), "expected this to be reported:\n{source}");
6631 assert!(
6632 result.messages.iter().any(|m| m.contains(expected)),
6633 "{expected}\n{:?}",
6634 result.messages
6635 );
6636 }
6637 }
6638
6639 #[test]
6644 fn an_asm_at_file_scope_that_is_directives_becomes_the_objects_it_defines() {
6645 let text = ir(concat!(
6646 "__asm__(\n",
6647 " \".section .rodata\\n\"\n",
6648 " \".globl first\\n\"\n",
6649 " \".balign 8\\n\"\n",
6650 " \"first:\\n\"\n",
6651 " \".long 1\\n\"\n",
6652 " \".long 2\\n\"\n",
6653 " \".globl last\\n\"\n",
6654 " \"last:\\n\"\n",
6655 " \".quad last - first\\n\");\n",
6656 "extern const int first[];\n",
6657 "extern const long last;\n",
6658 ));
6659 assert!(text.contains("global @first : bytes 8 = { i32 1, i32 2 }, align 8"), "{text}");
6660 assert!(text.contains("global @last : i64 = 8"), "{text}");
6661 }
6662
6663 #[test]
6667 fn a_name_an_asm_at_file_scope_defined_is_not_undone_by_a_declaration_of_it() {
6668 let text = ir(concat!(
6669 "__asm__(\".data\\n.globl counter\\ncounter:\\n.long 7\\n\");\n",
6670 "extern int counter;\n",
6671 "int read(void) { return counter; }\n",
6672 ));
6673 assert!(text.contains("global @counter : i32 = 7"), "{text}");
6674 }
6675
6676 #[test]
6679 fn an_incbin_at_file_scope_is_the_bytes_of_the_file_it_names() {
6680 let mut opts = options();
6681 opts.emit = EmitKind::Ir;
6682 let mut fs = MemoryFileSystem::new();
6683 fs.insert(
6684 "/main.c",
6685 b"__asm__(\".data\\n.globl blob\\nblob:\\n.incbin \\\"seed\\\"\\n\");\n".to_vec(),
6686 );
6687 fs.insert("seed", b"hi".to_vec());
6688 let result = compile(&opts, "/main.c", &fs);
6689 assert_eq!(result.messages, Vec::<String>::new());
6690 let text = result.text();
6691 assert!(text.contains("global @blob : bytes 2 = { bytes \"hi\" }"), "{text}");
6692 }
6693
6694 #[test]
6697 fn an_incbin_naming_a_file_that_is_not_there_says_which_file() {
6698 let messages = errors("__asm__(\".data\\nb:\\n.incbin \\\"nowhere\\\"\\n\");\n");
6699 assert!(
6700 messages
6701 .iter()
6702 .any(|m| m.contains("cannot open 'nowhere' for reading") && m.contains("E0702")),
6703 "{messages:?}"
6704 );
6705 }
6706
6707 #[test]
6710 fn an_instruction_in_an_asm_at_file_scope_is_refused_rather_than_ignored() {
6711 for source in [
6712 "__asm__(\".text\\n.globl f\\nf:\\n ret\\n\");\n",
6713 "__asm__(\".data\\n.set alias, 4\\n\");\n",
6714 ] {
6715 let messages = errors(source);
6716 assert!(
6717 messages
6718 .iter()
6719 .any(|m| m.contains("not supported yet")
6720 && m.contains("in an `asm` at file scope")),
6721 "{source}\n{messages:?}"
6722 );
6723 }
6724 }
6725
6726 #[test]
6727 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
6728 let mut opts = options();
6729 opts.emit = EmitKind::Ir;
6730 for source in [
6731 "int f(int n) { void *p = &&out; if (n) goto *p; { int a[n]; out: return 1; } }\n",
6732 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
6733 ] {
6734 let result = run(&opts, source);
6735 assert!(result.failed(), "expected this to be reported:\n{source}");
6736 assert!(
6737 result.messages.iter().any(|m| m.contains("not supported yet")),
6738 "{:?}",
6739 result.messages
6740 );
6741 }
6742 }
6743
6744 fn round_trip(source: &str) -> (String, String) {
6746 let printed = ir(source);
6747 let mut opts = options();
6748 opts.emit = EmitKind::Ir;
6749 let mut fs = MemoryFileSystem::new();
6750 fs.insert("/main.ir", printed.clone().into_bytes());
6751 let result = compile_ir(&opts, "/main.ir", &fs);
6752 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
6753 (printed, result.text().to_owned())
6754 }
6755
6756 #[test]
6757 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
6758 let (printed, again) = round_trip(
6762 "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",
6763 );
6764 assert_eq!(printed, again);
6765 }
6766
6767 #[test]
6768 fn ir_that_is_not_ir_says_which_line_stopped_it() {
6769 let mut opts = options();
6770 opts.emit = EmitKind::Ir;
6771 let mut fs = MemoryFileSystem::new();
6772 let text = "\
6773; ModuleID = 'a.c'
6774; format 0
6775target triple = \"x86_64-unknown-linux-gnu\"
6776target datalayout = \"e-p:64:64-i64:64-S128\"
6777
6778func @f(), linkage(external) {
6779block0:
6780 frobnicate
6781}
6782";
6783 fs.insert("/main.ir", text.as_bytes().to_vec());
6784 let result = compile_ir(&opts, "/main.ir", &fs);
6785 assert!(result.failed());
6786 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
6787 }
6788
6789 #[test]
6790 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
6791 let mut opts = options();
6794 opts.emit = EmitKind::Ir;
6795 let mut fs = MemoryFileSystem::new();
6796 let text = "\
6797; ModuleID = 'a.c'
6798; format 0
6799target triple = \"x86_64-unknown-linux-gnu\"
6800target datalayout = \"e-p:64:64-i64:64-S128\"
6801
6802func @f(), linkage(external) {
6803block0:
6804 %0 = iconst.i32 1
6805 return %0
6806}
6807";
6808 fs.insert("/main.ir", text.as_bytes().to_vec());
6809 let result = compile_ir(&opts, "/main.ir", &fs);
6810 assert!(result.failed());
6811 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
6812 }
6813
6814 #[test]
6815 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
6816 let mut fs = MemoryFileSystem::new();
6818 fs.insert("/main.ir", Vec::new());
6819 let result = compile_ir(&options(), "/main.ir", &fs);
6820 assert!(result.failed());
6821 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
6822 }
6823
6824 #[test]
6825 fn the_printed_ir_reads_back_as_the_same_module() {
6826 let text = ir("\
6829struct point { int x, y; };
6830static const char greeting[] = \"hi\";
6831int table[4] = { 1, 2, 3 };
6832int puts(const char *);
6833double half(double x) { return x / 2.0; }
6834int f(int n) {
6835 int total = 0;
6836 for (int i = 0; i < n; i++) {
6837 if (i == 3) continue;
6838 total += table[i];
6839 }
6840 switch (n) {
6841 case 0: total = 1;
6842 case 1: total++; break;
6843 default: total = -total;
6844 }
6845 struct point p = { total, 1 };
6846 int *q = &p.y;
6847 puts(greeting);
6848 return p.x + *q;
6849}
6850int dispatch(int c) {
6851 void *p = c ? &&one : &&two;
6852 goto *p;
6853one:
6854 return 1;
6855two:
6856 return 2;
6857}
6858int assembly(int x, int *p) {
6859 int r;
6860 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
6861 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
6862 return r;
6863away:
6864 return 0;
6865}
6866");
6867 let mut names = Interner::new();
6868 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
6869 assert_eq!(rucc_ir::print(&module, &names), text);
6870 }
6871
6872 #[test]
6873 fn what_save_temps_keeps_is_the_text_that_was_compiled_and_the_assembly_that_was_assembled() {
6874 let mut opts = options();
6878 opts.emit = EmitKind::Object;
6879 opts.save_temps = rucc_session::SaveTemps::Object;
6880 let result = run(&opts, "#define N 2\nint a[N];\n");
6881 assert_eq!(result.messages, Vec::<String>::new());
6882 let text = result.temps.preprocessed.expect("the preprocessed text");
6883 assert!(text.contains("int a[2];"), "{text}");
6884 assert!(text.starts_with("# 1 \"/main.c\""), "{text}");
6885 let asm = result.temps.assembly.expect("the assembly");
6886 assert!(asm.contains("a:"), "{asm}");
6887 assert!(matches!(result.artifact, Artifact::Object { .. }), "{:?}", result.artifact);
6888 }
6889
6890 #[test]
6891 fn nothing_is_kept_unless_the_flag_asked_for_it() {
6892 let mut opts = options();
6895 opts.emit = EmitKind::Object;
6896 assert_eq!(run(&opts, "int a;\n").temps, Temps::default());
6897 }
6898
6899 #[test]
6900 fn a_compilation_that_stops_before_the_back_end_keeps_the_text_and_no_assembly() {
6901 let mut opts = options();
6904 opts.emit = EmitKind::Ir;
6905 opts.save_temps = rucc_session::SaveTemps::Cwd;
6906 let result = run(&opts, "int a;\n");
6907 assert!(result.temps.preprocessed.is_some());
6908 assert_eq!(result.temps.assembly, None);
6909 }
6910}