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(Vec<u8>),
49}
50
51impl Artifact {
52 #[must_use]
54 pub fn bytes(&self) -> &[u8] {
55 match self {
56 Artifact::Nothing => &[],
57 Artifact::Text(text) => text.as_bytes(),
58 Artifact::Object(bytes) => bytes,
59 }
60 }
61}
62
63#[derive(Debug, Clone, PartialEq, Eq)]
65pub struct Compiled {
66 pub artifact: Artifact,
68 pub messages: Vec<String>,
70 pub errors: u32,
72 pub fired: Fired,
78 pub pressure: Pressure,
83 pub dumps: Vec<rucc_opt::Dump>,
89 pub remarks: String,
95 pub deps: Vec<rucc_pp::Dependency>,
100 pub temps: Temps,
107}
108
109#[derive(Debug, Clone, PartialEq, Eq, Default)]
116pub struct Temps {
117 pub preprocessed: Option<String>,
119 pub assembly: Option<String>,
121}
122
123impl Compiled {
124 #[must_use]
126 pub fn failed(&self) -> bool {
127 self.errors > 0
128 }
129
130 #[must_use]
135 pub fn text(&self) -> &str {
136 match &self.artifact {
137 Artifact::Text(text) => text,
138 _ => "",
139 }
140 }
141}
142
143#[must_use]
156pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
157 let mut sess = Session::new(opts.clone());
158 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
162 let mut diagnostics: Vec<Diagnostic> = Vec::new();
163 let mut fired = Fired::new();
165 let mut pressure = Pressure::new();
167 let mut dumps = Vec::new();
169 let mut remarks = String::new();
170 let mut temps = Temps::default();
172
173 let bytes = match fs.read(Path::new(name)) {
174 Ok(bytes) => bytes,
175 Err(e) => return failure(format!("{name}: {e}")),
176 };
177 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
178 return failure(format!("{name}: the source map has no room left for this file"));
179 };
180
181 let mut pp = rucc_pp::Preprocessor::with_prefix_map(opts.prefix_map.macros.clone());
185 let predef = rucc_pp::Predef::for_options(opts);
186 let expanded: Vec<PpToken> = {
187 let mut tokens = Vec::new();
188 {
193 let mut cx =
194 rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
195 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
196 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
197 return failure(format!(
198 "{name}: the source map has no room for the built in macros"
199 ));
200 }
201 if pp.preinclude(&opts.preincludes, &mut tokens, &mut cx).is_err() {
202 return failure(format!("{name}: the source map has no room for the command line"));
203 }
204 tokens.append(&mut pp.run(file, &mut cx));
205 }
206 if opts.save_temps.wanted() {
207 temps.preprocessed = Some(rucc_pp::print(
208 file,
209 &tokens,
210 pp.line_directives(),
211 &sess.sources,
212 &sess.interner,
213 rucc_pp::PrintOptions { line_markers: opts.line_markers },
214 ));
215 }
216 tokens.iter().map(|token| token.to_pp()).collect()
217 };
218 diagnostics.extend(pp.take_diagnostics());
219 let deps = pp.dependencies().to_vec();
222
223 let cx = Convert {
226 keywords: &keywords,
227 interner: &sess.interner,
228 target: &sess.target,
229 std: opts.std,
230 gnu: opts.gnu_extensions,
231 pedantic: opts.pedantic,
232 };
233 let (tokens, complaints) = convert(&expanded, &cx);
234 diagnostics.extend(complaints);
235
236 let parsed = rucc_parse::parse(
237 &tokens,
238 rucc_parse::Context {
239 interner: &sess.interner,
240 std: opts.std,
241 gnu: opts.gnu_extensions,
242 pedantic: opts.pedantic,
243 error_limit: opts.error_limit as usize,
244 },
245 );
246 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
247 diagnostics.extend(parsed.diagnostics);
248
249 let mut artifact = Artifact::Nothing;
250 let mut instrumented = Instrumented::default();
253 if !parse_failed {
254 let mut checker = Checker::new(
255 &parsed.ast,
256 CheckContext {
257 names: &sess.interner,
258 target: &sess.target,
259 std: opts.std,
260 gnu: opts.gnu_extensions,
261 pedantic: opts.pedantic,
262 permissive: opts.permissive,
263 gnu89_inline: opts.gnu89_inline,
264 error_limit: opts.error_limit as usize,
265 builtins: opts.builtins && opts.hosted,
268 no_builtin: &opts.no_builtin,
269 short_enums: opts.short_enums,
270 },
271 );
272 checker.check_unit();
273 let checked = checker.finish();
274 if !checked.failed() {
275 match opts.emit {
276 EmitKind::Tast => {
277 artifact = Artifact::Text(rucc_sema::print(
278 &checked.tast,
279 &checked.types,
280 &sess.interner,
281 ));
282 }
283 EmitKind::TypeGranules => {
287 artifact = Artifact::Text(rucc_types::granule_report(
288 &checked.types,
289 &sess.interner,
290 &sess.target,
291 ));
292 }
293 EmitKind::Ir
294 | EmitKind::MirFinal
295 | EmitKind::Asm
296 | EmitKind::Object
297 | EmitKind::Executable
298 | EmitKind::SafetySummary => {
299 let mut read = |named: &str| {
304 fs.read(Path::new(named))
305 .map(|bytes| bytes.as_slice().to_vec())
306 .map_err(|why| why.to_string())
307 };
308 let mut lowered = rucc_lower::lower(
309 name,
310 rucc_lower::Context {
311 tast: &checked.tast,
312 types: &checked.types,
313 target: &sess.target,
314 names: &mut sess.interner,
315 visibility: match opts.visibility {
316 Visibility::Default => IrVisibility::Default,
317 Visibility::Hidden => IrVisibility::Hidden,
318 Visibility::Protected => IrVisibility::Protected,
319 },
320 protector: match opts.protector {
321 Protector::None => LowerProtector::None,
322 Protector::Buffers => LowerProtector::Buffers,
323 Protector::Strong => LowerProtector::Strong,
324 Protector::All => LowerProtector::All,
325 },
326 wrapping: rucc_lower::Wrapping {
327 signed: opts.wrapping.signed,
328 pointer: opts.wrapping.pointer,
329 trap: opts.wrapping.trap,
330 },
331 aliasing: opts.strict_aliasing,
332 padding: opts.padding == Padding::Ignored,
333 contract: match opts.fp_contract {
334 Contract::Off => FpContract::Off,
335 Contract::On => FpContract::On,
336 Contract::Fast => FpContract::Fast,
337 },
338 read: &mut read,
339 },
340 );
341 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
345 if !failed {
346 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
351 for error in errors {
352 diagnostics.push(internal(&format!("invalid IR, {error}")));
353 }
354 } else if let Err(complaints) =
355 instrument(&mut lowered.module, &mut sess.interner, opts)
356 .map(|done| instrumented = done)
357 {
358 diagnostics.extend(complaints);
359 } else if let Err(complaints) = optimize(
360 &mut lowered.module,
361 &sess.interner,
362 &sess.target,
363 opts,
364 name,
365 &mut dumps,
366 &mut remarks,
367 ) {
368 diagnostics.extend(complaints);
369 } else if opts.emit == EmitKind::SafetySummary {
370 artifact = Artifact::Text(
375 rucc_safety::summarize(
376 &lowered.module,
377 &sess.interner,
378 name,
379 opts.safety.as_str(),
380 instrumented.checks,
381 instrumented.interposed,
382 instrumented.crossings,
383 )
384 .render(),
385 );
386 } else if opts.emit == EmitKind::Ir {
387 artifact =
392 Artifact::Text(rucc_ir::print(&lowered.module, &sess.interner));
393 } else {
394 match generate(
397 &mut lowered.module,
398 &mut sess.interner,
399 &sess.target,
400 opts,
401 &mut fired,
402 &mut pressure,
403 &mut temps.assembly,
404 ) {
405 Ok(made) => artifact = made,
406 Err(complaints) => diagnostics.extend(complaints),
407 }
408 }
409 }
410 diagnostics.extend(lowered.diagnostics);
411 }
412 _ => {}
413 }
414 }
415 diagnostics.extend(checked.diagnostics);
416 }
417
418 let mut messages = Vec::with_capacity(diagnostics.len());
419 let mut errors = 0;
420 for diag in &diagnostics {
421 if !opts.warnings && diag.severity == Severity::Warning {
425 continue;
426 }
427 if diag.severity.is_fatal()
428 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
429 {
430 errors += 1;
431 }
432 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
433 }
434 if errors > 0 {
435 artifact = Artifact::Nothing;
437 }
438 Compiled { artifact, messages, errors, fired, pressure, dumps, remarks, deps, temps }
441}
442
443#[must_use]
453pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
454 let mut sess = Session::new(opts.clone());
455 if opts.emit != EmitKind::Ir {
456 return failure(format!(
457 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
458 the C in front of it became",
459 opts.emit.as_str()
460 ));
461 }
462 let bytes = match fs.read(Path::new(name)) {
463 Ok(bytes) => bytes,
464 Err(e) => return failure(format!("{name}: {e}")),
465 };
466 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
467 return failure(format!("{name}: this is not text, so it is not IR"));
468 };
469
470 let module = match rucc_ir::parse(text, &mut sess.interner) {
471 Ok(module) => module,
472 Err(error) => {
473 return failure(format!("{name}:{}: {}", error.line, error.message));
474 }
475 };
476 let mut diagnostics: Vec<Diagnostic> = Vec::new();
477 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
478 for error in errors {
479 diagnostics.push(invalid(&format!("invalid IR, {error}")));
480 }
481 }
482 let mut messages = Vec::with_capacity(diagnostics.len());
483 for diag in &diagnostics {
484 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
485 }
486 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
487 let artifact = if errors > 0 {
488 Artifact::Nothing
489 } else {
490 Artifact::Text(rucc_ir::print(&module, &sess.interner))
491 };
492 Compiled {
494 artifact,
495 messages,
496 errors,
497 fired: Fired::new(),
498 pressure: Pressure::new(),
499 dumps: Vec::new(),
500 remarks: String::new(),
501 deps: Vec::new(),
502 temps: Temps::default(),
503 }
504}
505
506fn instrument(
529 module: &mut rucc_ir::Module,
530 names: &mut Interner,
531 opts: &Options,
532) -> Result<Instrumented, Vec<Diagnostic>> {
533 if !opts.safety.instruments() {
534 return Ok(Instrumented::default());
535 }
536 let checks = rucc_safety::run(module, opts.subobject, opts.promise);
537 let interposed = rucc_safety::redirect(module, names);
542 let crossings = rucc_safety::witness(module, names);
545 match rucc_ir::verify(module, names) {
546 Ok(()) => Ok(Instrumented { checks, interposed, crossings }),
547 Err(errors) => Err(errors
548 .iter()
549 .map(|e| internal(&format!("invalid IR after check insertion, {e}")))
550 .collect()),
551 }
552}
553
554#[derive(Clone, Copy, Debug, Default)]
560struct Instrumented {
561 checks: rucc_safety::Counts,
563 interposed: usize,
565 crossings: rucc_safety::Sites,
567}
568
569fn optimize(
581 module: &mut rucc_ir::Module,
582 names: &Interner,
583 target: &TargetInfo,
584 opts: &Options,
585 file: &str,
586 dumps: &mut Vec<rucc_opt::Dump>,
587 remarks: &mut String,
588) -> Result<(), Vec<Diagnostic>> {
589 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
590 settings.interposition = match opts.interposition {
596 true => replaceable(target, opts),
597 false => IrPic::Executable,
598 };
599 settings.toggles.clone_from(&opts.passes);
600 settings.fuel = opts.pass_fuel.iter().cloned().collect();
601 settings.global_fuel = opts.pass_fuel_global;
602 settings.verify |= opts.verify_each;
603 for (on, spec) in &opts.pass_gates {
604 if let Err(why) = settings.gates.add(*on, spec) {
607 return Err(vec![internal(&why)]);
608 }
609 }
610 for spec in &opts.dump_ir {
611 if let Err(why) = settings.dumps.add(spec) {
614 return Err(vec![internal(&why)]);
615 }
616 }
617 let mut wants = rucc_opt::Wants::none();
618 for spec in &opts.opt_info {
619 if let Err(why) = wants.add(spec) {
622 return Err(vec![internal(&why)]);
623 }
624 }
625 let report = rucc_opt::run(module, names, &settings);
626 remarks.push_str(&rucc_opt::optinfo::render(file, &report, names, wants));
627 dumps.extend(report.dumps);
628 match report.broke.is_empty() {
629 true => Ok(()),
630 false => Err(report.broke.iter().map(|why| internal(why)).collect()),
631 }
632}
633
634fn replaceable(target: &TargetInfo, opts: &Options) -> IrPic {
668 match (target.tuple.os().object_format(), opts.pic) {
669 (Some(ObjectFormat::Elf), Pic::Library) => IrPic::Library,
670 _ => IrPic::Executable,
671 }
672}
673
674fn generate(
675 module: &mut rucc_ir::Module,
676 names: &mut Interner,
677 target: &TargetInfo,
678 opts: &Options,
679 fired: &mut Fired,
680 pressure: &mut Pressure,
681 assembly: &mut Option<String>,
682) -> Result<Artifact, Vec<Diagnostic>> {
683 let Some(machine) = Machine::for_target(target) else {
684 return Err(vec![unsupported(&format!(
685 "there is no back end for {} in this compiler yet, so there is nothing to generate",
686 target.tuple
687 ))]);
688 };
689 if opts.protector != Protector::None && machine.conv.guard.is_none() {
694 return Err(vec![unsupported(&format!(
695 "{} is not supported for {} yet, because the stack protector on that target is not \
696 the one this compiler writes",
697 opts.protector, target.tuple
698 ))]);
699 }
700 if opts.control.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
706 return Err(vec![unsupported(&format!(
707 "-fcf-protection={} is not supported for {} yet, because what says a file was built \
708 for it there is not the note this compiler writes",
709 opts.control, target.tuple
710 ))]);
711 }
712 let profile = match machine.conv.trace {
718 Some(trace) => opts.profile.then(|| opts.hook.early(trace.fentry)),
719 None if opts.profile => {
720 return Err(vec![unsupported(&format!(
721 "-pg is not supported for {} yet, because the profiler's hook on that target is \
722 not the one this compiler calls",
723 target.tuple
724 ))]);
725 }
726 None => None,
727 };
728 if opts.patchable.any() && target.tuple.os().object_format() != Some(ObjectFormat::Elf) {
733 return Err(vec![unsupported(&format!(
734 "-fpatchable-function-entry= is not supported for {} yet, because what records where \
735 the room is there is not the section this compiler writes",
736 target.tuple
737 ))]);
738 }
739 let flags = pipeline::Flags {
740 frame_pointer: opts.frame_pointer,
741 red_zone: opts.red_zone,
742 stack_clash: opts.stack_clash,
743 landing: opts.control.branch(),
744 profile: match profile {
745 None => pipeline::Profile::No,
746 Some(true) => pipeline::Profile::Early,
747 Some(false) => pipeline::Profile::Late,
748 },
749 patch: pipeline::Room { after: opts.patchable.after(), before: opts.patchable.before },
750 };
751
752 if opts.safety.instruments() {
761 rucc_safety::lower(module, names);
762 if let Err(errors) = rucc_ir::verify(module, names) {
763 return Err(errors
764 .iter()
765 .map(|e| internal(&format!("invalid IR after check lowering, {e}")))
766 .collect());
767 }
768 }
769
770 let elsewhere = Elsewhere::of(module, replaceable(target, opts));
778
779 let mut funcs = Vec::new();
780 let mut complaints = Vec::new();
781 for id in module.funcs() {
782 if module[id].is_declaration() {
783 continue;
784 }
785 match pipeline::compile_recording(
786 &mut module[id],
787 names,
788 &machine,
789 &elsewhere,
790 flags,
791 fired,
792 pressure,
793 ) {
794 Ok(func) => funcs.push(func),
795 Err(why) => {
796 let name = names.resolve(module[id].name).to_owned();
797 let span = why.inst().map_or(Span::DUMMY, |inst| module[id].span(inst));
800 let said = format!("cannot generate code for '{name}': {why}");
801 complaints.push(unsupported_at(&said, span));
802 }
803 }
804 }
805 if !complaints.is_empty() {
806 return Err(complaints);
807 }
808 let (globals, aliases) = match opts.emit {
814 EmitKind::Asm | EmitKind::Object | EmitKind::Executable => (
815 rucc_asm::globals(module, names).map_err(refused)?,
816 rucc_asm::aliases(module, names).map_err(refused)?,
817 ),
818 _ => (rucc_asm::Globals::default(), Vec::new()),
819 };
820 let unwind = opts.unwinds();
824 match opts.emit {
825 EmitKind::Asm => {
826 rucc_asm::print(&funcs, &globals, &aliases, names, target, unwind, output(opts, target))
827 .map(Artifact::Text)
828 .map_err(refused)
829 }
830 EmitKind::Object | EmitKind::Executable => {
833 if opts.save_temps.wanted() {
834 let listing = rucc_asm::print(
835 &funcs,
836 &globals,
837 &aliases,
838 names,
839 target,
840 unwind,
841 output(opts, target),
842 );
843 *assembly = Some(listing.map_err(refused)?);
844 }
845 let text = rucc_asm::assemble(&funcs, names, target, unwind).map_err(refused)?;
846 let data = globals.image();
847 rucc_object::write(&text, &data, &aliases, target, output(opts, target))
850 .map(Artifact::Object)
851 .map_err(|why| match why {
852 rucc_object::Error::Format { .. } => vec![unsupported(&why.to_string())],
853 rucc_object::Error::Refused { .. } => vec![internal(&why.to_string())],
854 })
855 }
856 _ => Ok(Artifact::Text(rucc_mir::print(&funcs, names, target.regs))),
857 }
858}
859
860fn output(opts: &Options, target: &TargetInfo) -> rucc_object::Output {
872 let mut features = 0;
873 if target.tuple.arch() == Arch::X86_64 {
874 if opts.control.branch() {
875 features |= rucc_object::Property::IBT;
876 }
877 if opts.control.ret() {
878 features |= rucc_object::Property::SHSTK;
879 }
880 }
881 rucc_object::Output {
882 sections: rucc_object::Sections {
883 functions: opts.function_sections,
884 data: opts.data_sections,
885 },
886 property: rucc_object::Property { features },
887 }
888}
889
890fn refused(why: rucc_asm::Error) -> Vec<Diagnostic> {
896 match why {
897 rucc_asm::Error::Thread { .. } | rucc_asm::Error::IFunc { .. } => {
898 vec![unsupported(&why.to_string())]
899 }
900 _ => vec![internal(&why.to_string())],
901 }
902}
903
904fn unsupported(message: &str) -> Diagnostic {
910 unsupported_at(message, Span::DUMMY)
911}
912
913fn unsupported_at(message: &str, span: Span) -> Diagnostic {
919 Diagnostic::error(message.to_owned(), span)
920 .with_code("E0653")
921 .note("this construct is not lowered yet, see https://github.com/tamnd/rucc/issues", span)
922}
923
924fn invalid(message: &str) -> Diagnostic {
926 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
927}
928
929fn internal(message: &str) -> Diagnostic {
931 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
932 .with_code("E0652")
933 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
934}
935
936fn failure(message: String) -> Compiled {
939 Compiled {
940 artifact: Artifact::Nothing,
941 messages: vec![format!("rucc: error: {message}")],
942 errors: 1,
943 fired: Fired::new(),
944 pressure: Pressure::new(),
945 dumps: Vec::new(),
946 remarks: String::new(),
947 deps: Vec::new(),
948 temps: Temps::default(),
949 }
950}
951
952#[cfg(test)]
953mod tests {
954 use rucc_session::{MemoryFileSystem, Std};
955 use rucc_target::Triple;
956
957 use super::*;
958
959 fn options() -> Options {
960 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
961 opts.emit = EmitKind::Tast;
962 opts
963 }
964
965 fn run(opts: &Options, source: &str) -> Compiled {
966 let mut fs = MemoryFileSystem::new();
967 fs.insert("/main.c", source.to_owned().into_bytes());
968 compile(opts, "/main.c", &fs)
969 }
970
971 fn freestanding() -> Options {
975 let mut opts = options();
976 opts.hosted = false;
977 opts.search.push_system(rucc_session::runtime::DIR);
978 opts
979 }
980
981 fn shipped(source: &str) -> String {
983 let result = run(&freestanding(), source);
984 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
985 result.text().to_owned()
986 }
987
988 fn tast(source: &str) -> String {
990 let result = run(&options(), source);
991 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
992 result.text().to_owned()
993 }
994
995 #[test]
996 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
997 let text = shipped(concat!(
998 "#include <stdarg.h>\n",
999 "int sum(int n, ...) {\n",
1000 " va_list ap, copy;\n",
1001 " va_start(ap, n);\n",
1002 " va_copy(copy, ap);\n",
1003 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
1004 " va_end(ap);\n",
1005 " va_end(copy);\n",
1006 " return total;\n",
1007 "}\n",
1008 ));
1009 assert!(text.contains("va-start"), "{text}");
1010 assert!(text.contains("va-copy"), "{text}");
1011 assert!(text.contains("va-arg"), "{text}");
1012 assert!(text.contains("va-end"), "{text}");
1013 }
1014
1015 #[test]
1019 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
1020 let text = shipped(concat!(
1021 "#define __need___va_list\n",
1022 "#include <stdarg.h>\n",
1023 "int vprint(const char *f, __gnuc_va_list ap);\n",
1024 "#ifdef va_start\n",
1025 "#error va_start should not be defined\n",
1026 "#endif\n",
1027 "#ifdef _VA_LIST_DEFINED\n",
1028 "#error va_list should not have been made\n",
1029 "#endif\n",
1030 ));
1031 assert!(text.contains("vprint"), "{text}");
1032 }
1033
1034 #[test]
1037 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
1038 let text = shipped(concat!(
1039 "#define __need_size_t\n",
1040 "#include <stddef.h>\n",
1041 "#ifdef offsetof\n",
1042 "#error offsetof should not be defined yet\n",
1043 "#endif\n",
1044 "#define __need_ptrdiff_t\n",
1045 "#include <stddef.h>\n",
1046 "#include <stddef.h>\n",
1047 "size_t a;\n",
1048 "ptrdiff_t b;\n",
1049 "wchar_t c;\n",
1050 "max_align_t d;\n",
1051 "void *e = NULL;\n",
1052 "struct P { int x; long y; };\n",
1053 "size_t f = offsetof(struct P, y);\n",
1054 ));
1055 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
1056 assert!(text.contains("decl #1 b : long"), "{text}");
1057 }
1058
1059 #[test]
1060 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
1061 let text = shipped(concat!(
1062 "#include <limits.h>\n",
1063 "#include <float.h>\n",
1064 "int bits = CHAR_BIT;\n",
1065 "long big = LONG_MAX;\n",
1066 "int low = INT_MIN;\n",
1067 "int radix = FLT_RADIX;\n",
1068 "int digits = DBL_MANT_DIG;\n",
1069 ));
1070 assert!(text.contains("const 8 : int"), "{text}");
1071 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
1072 assert!(text.contains("const 2 : int"), "{text}");
1073 assert!(text.contains("const 53 : int"), "{text}");
1074 }
1075
1076 #[test]
1080 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
1081 let text = shipped(concat!(
1082 "#include <stdint.h>\n",
1083 "int64_t a = INT64_C(1);\n",
1084 "uint_least16_t b;\n",
1085 "intptr_t c;\n",
1086 "uintmax_t d = UINTMAX_MAX;\n",
1087 "int wide = sizeof(int_fast64_t);\n",
1088 ));
1089 assert!(text.contains("decl #0 a : long"), "{text}");
1090 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
1091 assert!(text.contains("decl #2 c : long"), "{text}");
1092 }
1093
1094 #[test]
1095 fn the_three_formality_headers_still_have_to_work() {
1096 let text = shipped(concat!(
1097 "#include <stdbool.h>\n",
1098 "#include <stdalign.h>\n",
1099 "#include <iso646.h>\n",
1100 "#include <stdnoreturn.h>\n",
1101 "int t = true and not false;\n",
1102 "_Alignas(16) char buf[16];\n",
1103 "int a = alignof(long);\n",
1104 ));
1105 assert!(text.contains("decl #0 t : int"), "{text}");
1106 assert!(text.contains("const 8 : unsigned long"), "{text}");
1107 }
1108
1109 #[test]
1112 fn every_shipped_header_can_be_included_twice() {
1113 let mut source = String::new();
1114 for _ in 0..2 {
1115 for name in rucc_session::runtime::names() {
1116 source.push_str(&format!("#include <{name}>\n"));
1117 }
1118 }
1119 source.push_str("int x;\n");
1120 let text = shipped(&source);
1121 assert!(text.starts_with("decl #0 x : int"), "{text}");
1122 }
1123
1124 #[test]
1125 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
1126 let fs = MemoryFileSystem::new();
1127 let result = compile(&options(), "/nope.c", &fs);
1128 assert!(result.failed());
1129 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
1130 assert!(result.text().is_empty());
1131 }
1132
1133 #[test]
1134 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
1135 let text = tast("int x = 1;\n");
1136 let expected = "\
1137decl #0 x : int object external static defined
1138 init
1139 +0
1140 const 1 : int
1141";
1142 assert_eq!(text, expected);
1143 }
1144
1145 #[test]
1146 fn the_macros_are_expanded_before_anything_is_parsed() {
1147 let text = tast("#define N 2\nint a[N];\n");
1151 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
1152 }
1153
1154 #[test]
1160 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
1161 let text = tast(concat!(
1162 "#pragma pack(4)\n",
1163 "struct s { int a; };\n",
1164 "#pragma pack()\n",
1165 "int b;\n",
1166 "_Pragma(\"GCC visibility push(default)\") int c;\n",
1167 ));
1168 assert!(text.contains("decl #0 b : int"), "{text}");
1169 assert!(text.contains("decl #1 c : int"), "{text}");
1170 }
1171
1172 #[test]
1180 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
1181 tast(concat!(
1182 "struct A { char c; int i; } __attribute__((packed));\n",
1183 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1184 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1185 "struct B { char c; int i; } __attribute__((aligned));\n",
1188 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
1189 "struct C { char c; int i __attribute__((packed)); };\n",
1190 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
1191 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
1192 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
1193 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
1194 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
1195 "struct E { char c; _Alignas(8) int i; };\n",
1196 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
1197 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
1198 "struct F { char c; int i __attribute__((aligned(8))); };\n",
1199 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
1200 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
1203 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
1204 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
1205 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
1206 "struct I { [[gnu::packed]] char c; int i; };\n",
1209 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1210 "struct J { char c; [[gnu::packed]] int i; };\n",
1211 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
1212 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
1213 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
1214 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
1215 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
1216 "union L { char c; int i; } __attribute__((packed));\n",
1217 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
1218 "struct O { char c; int i; } __attribute__((__packed__));\n",
1222 "_Static_assert(sizeof(struct O) == 5 && _Alignof(struct O) == 1, \"O\");\n",
1223 "struct P { char c; int i; } __attribute__((__aligned__(8)));\n",
1224 "_Static_assert(sizeof(struct P) == 8 && _Alignof(struct P) == 8, \"P\");\n",
1225 ));
1226 }
1227
1228 #[test]
1237 fn an_access_to_a_packed_member_says_the_alignment_the_layout_left_it() {
1238 let packed = body(concat!(
1239 "struct P { char c; int v; } __attribute__((packed));\n",
1240 "int f(struct P *p) { return p->v; }\n",
1241 ));
1242 assert!(packed.contains("load.i32 %2, align 1,"), "{packed}");
1243 let plain = body(concat!(
1245 "struct P { char c; int v; };\n",
1246 "int f(struct P *p) { return p->v; }\n",
1247 ));
1248 assert!(plain.contains("load.i32 %2, align 4,"), "{plain}");
1249 }
1250
1251 #[test]
1258 fn what_is_inside_a_packed_member_is_no_more_aligned_than_the_member_is() {
1259 let stepped = body(concat!(
1260 "struct P { char c; int v[4]; } __attribute__((packed));\n",
1261 "int f(struct P *p, int i) { return p->v[i]; }\n",
1262 ));
1263 assert!(stepped.contains(", align 1,"), "{stepped}");
1264 assert!(!stepped.contains(", align 4,"), "{stepped}");
1265 let nested = body(concat!(
1266 "struct Inner { int v; };\n",
1267 "struct P { char c; struct Inner in; } __attribute__((packed));\n",
1268 "int f(struct P *p) { return p->in.v; }\n",
1269 ));
1270 assert!(nested.contains(", align 1,"), "{nested}");
1271 assert!(!nested.contains(", align 4,"), "{nested}");
1272 }
1273
1274 #[test]
1283 fn the_aligned_attribute_on_a_declaration_raises_what_that_one_object_is_aligned_to() {
1284 tast(concat!(
1285 "int v __attribute__((aligned(64)));\n",
1286 "_Static_assert(__alignof__(v) == 64, \"v\");\n",
1287 "__attribute__((aligned(32))) int w;\n",
1290 "_Static_assert(__alignof__(w) == 32, \"w\");\n",
1291 "[[gnu::aligned(16)]] int x;\n",
1292 "_Static_assert(__alignof__(x) == 16, \"x\");\n",
1293 "int y __attribute__((aligned(2)));\n",
1296 "_Static_assert(__alignof__(y) == 4, \"y\");\n",
1297 "void f(void) { int a __attribute__((aligned(128)));\n",
1299 "_Static_assert(__alignof__(a) == 128, \"a\"); (void)a; }\n",
1300 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1303 "void g(void) __attribute__((aligned(256)));\n",
1306 "void g(void) {}\n",
1307 "_Static_assert(__alignof__(g) == 256, \"g\");\n",
1308 ));
1309 }
1310
1311 #[test]
1315 fn what_a_declaration_asked_to_be_aligned_to_is_what_the_assembler_is_told() {
1316 let text = asm(concat!(
1317 "int v __attribute__((aligned(64)));\n",
1318 "void g(void) __attribute__((aligned(256)));\n",
1319 "void g(void) {}\n",
1320 "void plain(void) {}\n",
1321 ));
1322 assert!(text.contains("\t.p2align\t6\n\t.type\tv, @object\n"), "{text}");
1323 assert!(text.contains("\t.p2align\t8, 0x90\n\t.globl\tg\n"), "{text}");
1324 assert!(text.contains("\t.p2align\t4, 0x90\n\t.globl\tplain\n"), "{text}");
1325 }
1326
1327 #[test]
1336 fn an_aligned_typedef_says_what_an_object_of_it_is_aligned_to_and_may_lower_it() {
1337 tast(concat!(
1338 "typedef int L __attribute__((aligned(2)));\n",
1339 "_Static_assert(__alignof__(L) == 2, \"L\");\n",
1340 "_Static_assert(_Alignof(L) == 2, \"L alignof\");\n",
1341 "_Static_assert(sizeof(L) == 4, \"L size\");\n",
1343 "struct T { char c; L x; };\n",
1344 "_Static_assert(sizeof(struct T) == 6, \"T\");\n",
1345 "_Static_assert(__builtin_offsetof(struct T, x) == 2, \"T.x\");\n",
1346 "typedef int H __attribute__((aligned(16)));\n",
1348 "_Static_assert(__alignof__(H) == 16, \"H\");\n",
1349 "_Static_assert(sizeof(H) == 4, \"H size\");\n",
1350 "struct U { char c; H x; };\n",
1351 "_Static_assert(sizeof(struct U) == 32, \"U\");\n",
1352 "_Static_assert(__builtin_offsetof(struct U, x) == 16, \"U.x\");\n",
1353 "typedef L M __attribute__((aligned(8)));\n",
1356 "_Static_assert(__alignof__(M) == 8, \"M\");\n",
1357 "typedef L N;\n",
1360 "_Static_assert(__alignof__(N) == 2, \"N\");\n",
1361 "_Static_assert(__alignof__(int) == 4, \"int\");\n",
1363 ));
1364 let text = asm(concat!(
1365 "typedef int L __attribute__((aligned(2)));\n",
1366 "typedef int H __attribute__((aligned(16)));\n",
1367 "L low;\n",
1368 "H high;\n",
1369 ));
1370 assert!(text.contains("\t.p2align\t1\n\t.type\tlow, @object\n"), "{text}");
1371 assert!(text.contains("\t.p2align\t4\n\t.type\thigh, @object\n"), "{text}");
1372 }
1373
1374 #[test]
1382 fn the_vector_size_attribute_builds_a_type_of_lanes_and_measures_it_in_bytes() {
1383 tast(concat!(
1384 "typedef int __attribute__((vector_size(16))) v4si;\n",
1385 "_Static_assert(sizeof(v4si) == 16 && _Alignof(v4si) == 16, \"v4si\");\n",
1386 "typedef char __attribute__((vector_size(16))) v16qi;\n",
1387 "_Static_assert(sizeof(v16qi) == 16, \"v16qi\");\n",
1388 "typedef int __attribute__((vector_size(4))) v1si;\n",
1391 "_Static_assert(sizeof(v1si) == 4, \"v1si\");\n",
1392 "typedef float __attribute__((__vector_size__(8))) v2sf;\n",
1394 "_Static_assert(sizeof(v2sf) == 8, \"v2sf\");\n",
1395 "typedef short [[gnu::vector_size(8)]] v4hi;\n",
1396 "_Static_assert(sizeof(v4hi) == 8, \"v4hi\");\n",
1397 "v4si g;\n",
1400 "_Static_assert(sizeof(g[0]) == 4, \"lane\");\n",
1401 "_Static_assert(sizeof(g + g) == 16, \"whole\");\n",
1402 "_Static_assert(sizeof(g + 1) == 16, \"broadcast\");\n",
1405 "_Static_assert(sizeof(v4si[3]) == 48, \"array\");\n",
1407 ));
1408 }
1409
1410 #[test]
1420 fn a_vector_is_written_whole_into_an_array_of_them_and_named_by_a_type_name() {
1421 tast(concat!(
1422 "typedef int __attribute__((vector_size(8))) v2si;\n",
1423 "v2si table[] = { (v2si){ 1, 2 }, (v2si){ 3, 4 } };\n",
1424 "_Static_assert(sizeof(table) == 16, \"two of them and not eight lanes\");\n",
1425 "v2si written = (int __attribute__((vector_size(8)))){ 5, 6 };\n",
1427 "_Static_assert(sizeof((int __attribute__((vector_size(16)))){ 0 }) == 16, \"named\");\n",
1428 "v2si lanes[2] = { 1, 2, 3, 4 };\n",
1431 "_Static_assert(sizeof(lanes) == 16, \"still elided\");\n",
1432 ));
1433 }
1434
1435 #[test]
1443 fn a_lane_is_assignable_and_a_shift_takes_a_count_of_its_own_lane() {
1444 let result = run(
1445 &options(),
1446 concat!(
1447 "typedef int __attribute__((vector_size(16))) v4si;\n",
1448 "typedef unsigned __attribute__((vector_size(16))) v4ui;\n",
1449 "void write(v4si *out, v4ui a, v4si b, int n) {\n",
1450 " v4si v = { 1, 2, 3, 4 };\n",
1451 " v[0] = n;\n",
1452 " v[1] += n;\n",
1453 " v[2]++;\n",
1454 " *&v[3] = n;\n",
1455 " v4ui shifted = a >> b;\n",
1457 " shifted <<= b;\n",
1458 " *out = v + (v4si)shifted + (1 << b);\n",
1461 "}\n",
1462 "void refused(const v4si c) {\n",
1465 " c[0] = 1;\n",
1466 "}\n",
1467 ),
1468 );
1469 assert_eq!(result.messages.len(), 1, "{:?}", result.messages);
1470 assert!(result.messages[0].contains("assignment of read-only"), "{:?}", result.messages);
1471 }
1472
1473 #[test]
1480 fn a_record_that_asks_for_the_other_byte_order_is_refused_rather_than_laid_out_in_this_one() {
1481 let opts = options();
1482 let big = "struct s { int i; } __attribute__((scalar_storage_order(\"big-endian\")));\n";
1483 assert_eq!(
1484 run(&opts, big).messages,
1485 ["/main.c:1:36: error: 'scalar_storage_order' is not implemented yet [E0688]\n\
1486 /main.c:1:36: note: every scalar in this record would be read in the wrong byte \
1487 order"]
1488 );
1489
1490 let armoured =
1491 "struct s { int i; } __attribute__((__scalar_storage_order__(\"little-endian\")));\n";
1492 let messages = run(&opts, armoured).messages;
1493 assert!(messages[0].contains("[E0688]"), "{messages:?}");
1494
1495 let front = "struct __attribute__((scalar_storage_order(\"big-endian\"))) s { int i; };\n";
1498 assert!(run(&opts, front).messages[0].contains("[E0688]"), "{front}");
1499 let standard = "struct s { int i; } [[gnu::scalar_storage_order(\"big-endian\")]];\n";
1500 assert!(run(&opts, standard).messages[0].contains("[E0688]"), "{standard}");
1501 }
1502
1503 #[test]
1513 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
1514 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
1516 assert_eq!(
1517 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
1518 1
1519 );
1520 assert_eq!(
1521 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
1522 1
1523 );
1524 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
1525 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
1527 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
1528 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
1530 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
1531 }
1532
1533 fn bit_field_byte(record: &str) -> u64 {
1535 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
1536 let body = body(&source);
1537 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
1538 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
1539 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
1540 }
1541
1542 #[test]
1548 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
1549 tast(concat!(
1550 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
1551 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
1552 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
1553 "struct b { char c; __attribute__((packed)) int i; };\n",
1554 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
1555 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
1556 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
1557 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
1558 ));
1559 }
1560
1561 #[test]
1567 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
1568 tast(concat!(
1569 "#pragma pack(1)\n",
1570 "struct A { char c; int i; };\n",
1571 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
1572 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
1573 "#pragma pack()\n",
1574 "struct B { char c; int i; };\n",
1575 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
1576 "#pragma pack(2)\n",
1577 "struct C { char c; int i; double d; };\n",
1578 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
1579 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
1580 "struct K { char c; int i __attribute__((aligned(8))); };\n",
1582 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
1583 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
1584 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
1586 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
1587 "#pragma pack()\n",
1588 "#pragma pack(push, 1)\n",
1589 "struct D { char c; short s; };\n",
1590 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
1591 "#pragma pack(pop)\n",
1592 "struct E { char c; short s; };\n",
1593 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
1594 "struct H { char c;\n",
1596 "#pragma pack(1)\n",
1597 " int i; };\n",
1598 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
1599 "#pragma pack(1)\n",
1600 "struct I { char c;\n",
1601 "#pragma pack()\n",
1602 " int i; };\n",
1603 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
1604 "#pragma pack()\n",
1605 "#pragma pack(push, 8)\n",
1607 "#pragma pack(push, 1)\n",
1608 "struct P { char c; int i; };\n",
1609 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
1610 "#pragma pack(pop)\n",
1611 "struct Q { char c; int i; };\n",
1612 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
1613 "#pragma pack(pop)\n",
1614 "#pragma pack(16)\n",
1616 "struct R { char c; int i; };\n",
1617 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
1618 "#pragma pack()\n",
1619 "#pragma pack(1)\n",
1620 "struct S { char c; int i : 5; int j : 20; };\n",
1621 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
1622 "union T { char c; int i; };\n",
1623 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
1624 "#pragma pack()\n",
1625 ));
1626 }
1627
1628 #[test]
1632 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
1633 let result = run(
1634 &options(),
1635 concat!(
1636 "#pragma pack 4\n",
1637 "#pragma pack(pop)\n",
1638 "#pragma pack(3)\n",
1639 "#pragma pack(1) junk\n",
1640 "#pragma pack(push, 1\n",
1641 "#pragma pack(x)\n",
1642 "#pragma pack(0)\n",
1645 "#pragma pack(push)\n",
1646 "struct s { char c; int i; };\n",
1647 "#pragma pack(pop)\n",
1648 "#pragma pack(pop, foo)\n",
1649 ),
1650 );
1651 let expected = [
1652 "missing `(` after `#pragma pack` - ignored",
1653 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
1654 "alignment must be a small power of two, not 3",
1655 "junk at end of `#pragma pack`",
1656 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
1657 "unknown action `x` for `#pragma pack` - ignored",
1658 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
1659 ];
1660 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
1661 for (message, want) in result.messages.iter().zip(expected) {
1662 assert!(message.contains(want), "expected {want:?} in {message:?}");
1663 }
1664 }
1665
1666 #[test]
1670 fn the_wide_integer_answers_to_all_three_of_its_names() {
1671 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
1672 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
1673 assert!(text.contains("decl #1 b : __int128"), "{text}");
1674 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
1675 }
1676
1677 #[test]
1678 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
1679 let text = tast("long f(int a, long b) { return a + b; }\n");
1683 assert!(text.contains("convert arithmetic"), "{text}");
1684 }
1685
1686 #[test]
1687 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
1688 for source in [
1689 "#error stop\n",
1690 "int f(void) { return 1 + ; }\n",
1691 "int f(void) { return undeclared; }\n",
1692 ] {
1693 let result = run(&options(), source);
1694 assert!(result.failed(), "expected this to fail:\n{source}");
1695 assert!(
1696 result.text().is_empty(),
1697 "a file that did not compile wrote a tree:\n{source}"
1698 );
1699 }
1700 }
1701
1702 #[test]
1703 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
1704 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
1708 assert_eq!(result.errors, 1, "{:?}", result.messages);
1709 }
1710
1711 #[test]
1712 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
1713 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
1717 assert_eq!(result.errors, 1, "{:?}", result.messages);
1718 }
1719
1720 #[test]
1721 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
1722 let source = "int f(void) { char c = 300; return c; }\n";
1723 let plain = run(&options(), source);
1724 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
1725 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
1726 assert!(!plain.text().is_empty(), "a warning is not a reason to write nothing");
1727
1728 let mut opts = options();
1729 opts.warnings_are_errors = true;
1730 let strict = run(&opts, source);
1731 assert!(strict.failed());
1732 assert!(strict.text().is_empty(), "and under -Werror it is a reason to write nothing");
1733 for message in &strict.messages {
1734 assert!(!message.contains("warning:"), "{message}");
1735 }
1736 }
1737
1738 #[test]
1739 fn w_drops_the_warning_before_werror_can_promote_it() {
1740 let source = "int f(void) { char c = 300; return c; }\n";
1741 let mut opts = options();
1742 opts.warnings = false;
1743 let quiet = run(&opts, source);
1744 assert_eq!(quiet.messages, Vec::<String>::new());
1745 assert_eq!(quiet.errors, 0);
1746 assert!(!quiet.text().is_empty(), "and the file still compiles");
1747
1748 opts.warnings_are_errors = true;
1751 let both = run(&opts, source);
1752 assert_eq!(both.messages, Vec::<String>::new());
1753 assert!(!both.failed(), "-w -Werror is not an error about a warning nobody saw");
1754 }
1755
1756 #[test]
1757 fn the_dialect_reaches_the_keywords_and_the_checking() {
1758 let source = "typeof(1) x;\n";
1761 let mut opts = options();
1762 opts.std = Std::C23;
1763 opts.gnu_extensions = false;
1764 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
1765
1766 opts.std = Std::C17;
1767 assert!(run(&opts, source).failed());
1768 }
1769
1770 #[test]
1771 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
1772 let mut opts = options();
1773 opts.emit = EmitKind::Object;
1774 let result = run(&opts, "int x = 1;\n");
1775 assert!(!result.failed(), "{:?}", result.messages);
1776 assert!(result.text().is_empty());
1777 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
1780 }
1781
1782 fn mir(source: &str) -> String {
1784 let mut opts = options();
1785 opts.emit = EmitKind::MirFinal;
1786 let result = run(&opts, source);
1787 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1788 result.text().to_owned()
1789 }
1790
1791 #[test]
1797 fn a_function_goes_from_c_to_instructions_with_real_registers_in_them() {
1798 let text = mir("int add(int a, int b) { return a + b; }\n");
1799 assert!(text.starts_with("mfunc @add {"), "{text}");
1800 assert!(text.contains("x64.add_rr_32"), "{text}");
1801 assert!(text.contains("x64.ret"), "{text}");
1802 assert!(!text.contains('%'), "{text}");
1805 }
1806
1807 #[test]
1809 fn a_function_with_no_body_produces_no_machine_function() {
1810 let text = mir("int g(int);\nint f(int a) { return g(a); }\n");
1811 assert_eq!(text.matches("mfunc @").count(), 1, "{text}");
1812 assert!(text.contains("mfunc @f {"), "{text}");
1813 assert!(text.contains("x64.call"), "{text}");
1814 }
1815
1816 #[test]
1818 fn every_definition_in_the_file_is_generated_and_they_keep_their_order() {
1819 let text = mir("int a(int x) { return x; }\nint b(int x) { return x; }\n");
1820 let first = text.find("mfunc @a").expect("the first function");
1821 let second = text.find("mfunc @b").expect("the second function");
1822 assert!(first < second, "{text}");
1823 }
1824
1825 #[test]
1827 fn the_target_decides_which_convention_the_generated_code_follows() {
1828 let mut opts = options();
1829 opts.emit = EmitKind::MirFinal;
1830 let linux = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1831 assert!(linux.contains("$rdi"), "{linux}");
1832
1833 opts.target = "x86_64-pc-windows-msvc".parse::<Triple>().unwrap();
1834 let windows = run(&opts, "int f(int a) { return a; }\n").text().to_owned();
1835 assert!(windows.contains("$rcx"), "{windows}");
1836 assert!(!windows.contains("$rdi"), "{windows}");
1837 }
1838
1839 #[test]
1841 fn a_target_this_has_no_back_end_for_is_reported_rather_than_generated() {
1842 let mut opts = options();
1843 opts.emit = EmitKind::MirFinal;
1844 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
1845 let result = run(&opts, "int f(int a) { return a; }\n");
1846 assert!(result.failed());
1847 assert!(result.messages[0].contains("no back end for aarch64"), "{:?}", result.messages);
1848 assert!(result.text().is_empty());
1849 }
1850
1851 #[test]
1858 fn a_construct_the_back_end_cannot_reach_yet_is_reported_against_its_function() {
1859 let mut opts = options();
1860 opts.emit = EmitKind::MirFinal;
1861 let source = "void a(int n) { int v[n] __attribute__((aligned(32))); v[0] = 1; }\n\
1862 void b(int n) { int v[n] __attribute__((aligned(32))); v[0] = 1; }\n";
1863 let result = run(&opts, source);
1864 assert!(result.failed());
1865 assert_eq!(result.messages.len(), 2, "{:?}", result.messages);
1866 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
1867 assert!(result.messages[0].contains("wants more alignment"), "{:?}", result);
1868 assert!(result.messages[1].contains("cannot generate code for 'b'"), "{:?}", result);
1869 assert!(result.text().is_empty());
1870 }
1871
1872 #[test]
1879 fn a_variable_length_array_is_refused_where_every_page_of_the_frame_is_to_be_touched() {
1880 let mut opts = options();
1881 opts.emit = EmitKind::MirFinal;
1882 let source = "void a(int n) { int v[n]; v[0] = 1; }\n";
1883 assert!(!run(&opts, source).failed(), "it compiles without the flag");
1884
1885 opts.stack_clash = true;
1886 let result = run(&opts, source);
1887 assert!(result.failed());
1888 assert!(result.messages[0].contains("cannot generate code for 'a'"), "{:?}", result);
1889 assert!(result.messages[0].contains("a page at a time"), "{:?}", result);
1890 }
1891
1892 #[test]
1904 fn an_opcode_with_no_name_in_the_rule_language_is_named_by_its_own_spelling() {
1905 let mut opts = options();
1906 opts.emit = EmitKind::MirFinal;
1907 let source = "int f(int a) {\n __int128 wide = a;\n return (int) (wide / wide);\n}\n";
1908 let result = run(&opts, source);
1909 assert!(result.failed());
1910 assert!(
1911 result.messages[0].contains("no rule lowers a `sext` producing a `i128`"),
1912 "{result:?}"
1913 );
1914 assert!(result.messages[0].contains(":2:"), "the line the widening is on: {result:?}");
1915 assert!(!result.messages[0].contains("this instruction"), "{result:?}");
1916 }
1917
1918 #[test]
1920 fn the_note_on_unfinished_work_points_at_the_issues_rather_than_at_the_plan() {
1921 let mut opts = options();
1922 opts.emit = EmitKind::MirFinal;
1923 let source = "int f(int a) { __int128 wide = a; return (int) (wide / wide); }\n";
1924 let result = run(&opts, source);
1925 assert!(result.failed());
1926 let note = result.messages.iter().find(|line| line.contains("note:")).expect("a note");
1927 assert!(note.contains("https://github.com/tamnd/rucc/issues"), "{note}");
1928 assert!(!note.contains("spec/17-milestones.md"), "{note}");
1929 }
1930
1931 #[test]
1933 fn the_frame_flags_on_the_command_line_reach_the_generated_frame() {
1934 let source = "int f(int a) { return a; }\n";
1935 assert!(!mir(source).contains("$rbp"), "a leaf needs no frame pointer by default");
1936
1937 let mut opts = options();
1938 opts.emit = EmitKind::MirFinal;
1939 opts.frame_pointer = true;
1940 let kept = run(&opts, source).text().to_owned();
1941 assert!(kept.contains("x64.push_64 $rbp"), "{kept}");
1942 }
1943
1944 fn asm(source: &str) -> String {
1946 let mut opts = options();
1947 opts.emit = EmitKind::Asm;
1948 let result = run(&opts, source);
1949 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
1950 result.text().to_owned()
1951 }
1952
1953 #[test]
1960 fn a_function_goes_from_c_to_assembly_an_assembler_would_take() {
1961 let text = asm("int add(int a, int b) { return a + b; }\n");
1962 assert!(text.contains("\t.globl\tadd\n"), "{text}");
1963 assert!(text.contains("\t.type\tadd, @function\n"), "{text}");
1964 assert!(text.contains("\nadd:\n"), "{text}");
1965 assert!(text.contains("\taddl\t"), "{text}");
1966 assert!(text.contains("\tret\n"), "{text}");
1967 assert!(text.contains("\t.size\tadd, .-add\n"), "{text}");
1968 assert!(text.contains(".note.GNU-stack"), "{text}");
1971 }
1972
1973 #[test]
1979 fn a_call_through_a_function_pointer_goes_through_the_register_it_is_in() {
1980 let text = asm("int g(int);\nint f(int (*p)(int), int a) { return p(a) + g(a); }\n");
1981 assert!(text.contains("\tcall\t*%"), "{text}");
1982 assert!(text.contains("\tcall\tg\n"), "{text}");
1983 assert!(text.contains("%rdi"), "{text}");
1987 }
1988
1989 #[test]
1993 fn the_address_of_a_global_is_read_from_the_instruction_pointer() {
1994 let text = asm("extern int counter;\nint f(void) { return counter; }\n");
1995 assert!(text.contains("\tmovl\tcounter(%rip), %eax\n"), "{text}");
1996 }
1997
1998 #[test]
2007 fn a_branch_on_a_comparison_jumps_on_the_opposite_of_what_it_compared() {
2008 let arms = "return 1; return 2;";
2009 let signed = [("==", "jne"), ("!=", "je"), ("<", "jge"), ("<=", "jg"), (">", "jle")];
2010 for (operator, jump) in signed.into_iter().chain([(">=", "jl")]) {
2011 let text = asm(&format!("int f(int a, int b) {{ if (a {operator} b) {arms} }}\n"));
2012 assert!(
2013 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
2014 "{operator}: {text}"
2015 );
2016 assert!(!text.contains("\tset"), "{operator}: {text}");
2017 assert!(!text.contains("\ttest"), "{operator}: {text}");
2018 }
2019 let unsigned = [("<", "jae"), ("<=", "ja"), (">", "jbe"), (">=", "jb")];
2020 for (operator, jump) in unsigned {
2021 let source =
2022 format!("int f(unsigned a, unsigned b) {{ if (a {operator} b) {arms} }}\n");
2023 let text = asm(&source);
2024 assert!(
2025 text.contains(&format!("\tcmpl\t%esi, %edi\n\t{jump}\t")),
2026 "{operator}: {text}"
2027 );
2028 }
2029
2030 let text = asm("int f(int a) { if (a < 7) return 1; return 2; }\n");
2033 assert!(text.contains("\tcmpl\t$7, %edi\n\tjge\t"), "{text}");
2034 }
2035
2036 #[test]
2042 fn a_comparison_whose_answer_the_program_wanted_still_writes_a_byte() {
2043 let text = asm("int f(int a, int b) { return a < b; }\n");
2044 assert!(text.contains("\tsetl\t"), "{text}");
2045 }
2046
2047 fn optimized(source: &str) -> String {
2049 let mut opts = options();
2050 opts.emit = EmitKind::Asm;
2051 opts.opt_level = rucc_session::OptLevel::O2;
2052 let result = run(&opts, source);
2053 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2054 result.text().to_owned()
2055 }
2056
2057 #[test]
2067 fn a_switch_whose_arms_are_a_function_of_the_label_is_a_range_check_and_arithmetic() {
2068 let arms: String =
2069 (0..16).map(|k| format!("case {k}: return {};", k + 1)).collect::<Vec<_>>().join(" ");
2070 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
2071 assert!(text.contains("\tcmpl\t$15, %edi\n\tja\t"), "{text}");
2072 assert!(text.contains("\taddl\t$1, %edi"), "{text}");
2073 assert_eq!(text.matches("\tcmp").count(), 1, "{text}");
2074 }
2075
2076 #[test]
2083 fn a_dense_switch_whose_arms_are_not_a_line_keeps_its_comparisons() {
2084 let arms: String = (0..16)
2085 .map(|k| format!("case {k}: return {};", if k == 9 { 100 } else { k + 1 }))
2086 .collect::<Vec<_>>()
2087 .join(" ");
2088 let text = optimized(&format!("int f(int x) {{ switch (x) {{ {arms} }} return 0; }}\n"));
2089 assert!(text.matches("\tcmp").count() > 1, "{text}");
2090 }
2091
2092 #[test]
2094 fn a_cast_between_a_pointer_and_an_integer_leaves_the_value_where_it_is() {
2095 let text = asm("long f(void *p) { return (long)p; }\n");
2096 for line in text.lines().filter(|line| line.starts_with('\t') && !line.contains('.')) {
2101 let mnemonic = line.split_whitespace().next().unwrap_or("");
2102 assert!(matches!(mnemonic, "movq" | "ret"), "{line} in\n{text}");
2103 }
2104 }
2105
2106 #[test]
2110 fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
2111 let six = "long a, long b, long c, long d, long e, long f";
2112 let text = asm(&format!("long f({six}, long g, long h) {{ return g + h; }}\n"));
2113
2114 assert!(text.contains("\tmovq\t8(%rsp), "), "{text}");
2118 assert!(text.contains("\tmovq\t16(%rsp), "), "{text}");
2119
2120 let narrow = asm(&format!("int f({six}, int g) {{ return g; }}\n"));
2124 assert!(narrow.contains("\tmovl\t8(%rsp), "), "{narrow}");
2125 let eight =
2126 "double a, double b, double c, double d, double e, double f, double g, double h";
2127 let float = asm(&format!("double f({eight}, double i) {{ return i; }}\n"));
2128 assert!(float.contains("\tmovsd\t8(%rsp), "), "{float}");
2129 }
2130
2131 #[test]
2134 fn a_call_writes_the_arguments_with_no_register_left_at_the_stack_pointer() {
2135 let six = "1, 2, 3, 4, 5, 6";
2136 let decl = "long g(long, long, long, long, long, long, long, long);\n";
2137 let text = asm(&format!("{decl}long f(void) {{ return g({six}, 7, 8); }}\n"));
2138
2139 assert!(text.contains("\tmovq\t%"), "{text}");
2140 assert!(text.contains(", (%rsp)\n"), "{text}");
2141 assert!(text.contains(", 8(%rsp)\n"), "{text}");
2142 assert!(text.contains("\tsubq\t$"), "{text}");
2144
2145 let narrow = "int g(int, int, int, int, int, int, int);\n";
2147 let text = asm(&format!("{narrow}int f(void) {{ return g({six}, 7); }}\n"));
2148 assert!(text.contains("\tmovl\t%"), "{text}");
2149 assert!(text.contains(", (%rsp)\n"), "{text}");
2150 }
2151
2152 #[test]
2155 fn a_variadic_call_counts_registers_and_not_arguments() {
2156 let nine = "1., 2., 3., 4., 5., 6., 7., 8., 9.";
2157 let decl = "int g(int, ...);\n";
2158 let text = asm(&format!("{decl}int f(void) {{ return g(0, {nine}); }}\n"));
2159
2160 assert!(text.contains("\tmovl\t$8, "), "eight registers, not nine: {text}");
2161 assert!(text.contains("\tmovsd\t%"), "{text}");
2162 assert!(text.contains(", (%rsp)\n"), "{text}");
2163 }
2164
2165 #[test]
2170 fn a_variadic_function_writes_the_argument_registers_it_was_handed_into_its_frame() {
2171 let body =
2172 "__builtin_va_list ap; __builtin_va_start(ap, n); __builtin_va_end(ap); return n;";
2173 let text = asm(&format!("int f(int n, ...) {{ {body} }}\n"));
2174
2175 let stores = |mnemonic: &str| text.matches(&format!("\t{mnemonic}\t%")).count();
2178 assert!(text.contains(", 8(%r"), "the second slot, not the first: {text}");
2179 assert!(!text.contains(", 0(%r"), "{text}");
2180 assert_eq!(stores("movsd"), 8, "every vector register: {text}");
2181
2182 assert!(text.contains("\tsubq\t$"), "{text}");
2184 }
2185
2186 #[test]
2189 fn va_start_writes_the_four_fields_the_psabi_describes() {
2190 let start = "__builtin_va_list ap; __builtin_va_start(ap, d);";
2191 let params = "int a, int b, int c, double d";
2192 let text = asm(&format!("int f({params}, ...) {{ {start} return a; }}\n"));
2193
2194 assert!(text.contains(" movl $24, "), "{text}");
2198 assert!(text.contains(" movl $64, "), "{text}");
2199 assert!(text.contains(", 8(%r"), "{text}");
2203 assert!(text.contains(", 16(%r"), "{text}");
2204 let frame: u32 = text
2205 .lines()
2206 .find_map(|line| line.trim().strip_prefix("subq $")?.split(',').next()?.parse().ok())
2207 .expect("a variadic function takes a frame for the save area");
2208 let above = |line: &str| {
2209 let at: u32 = line.trim().strip_prefix("leaq ")?.split('(').next()?.parse().ok()?;
2210 Some(at > frame)
2211 };
2212 assert!(text.lines().filter_map(above).any(|it| it), "{frame}: {text}");
2213 }
2214
2215 #[test]
2218 fn va_arg_branches_on_whether_the_argument_is_still_in_the_save_area() {
2219 let read = "__builtin_va_list ap; __builtin_va_start(ap, n);";
2220 let ints = format!("int f(int n, ...) {{ {read} return __builtin_va_arg(ap, int); }}\n");
2221 let text = asm(&ints);
2222
2223 assert!(text.contains("$40, "), "{text}");
2226 assert!(text.contains(" cmpl "), "{text}");
2227 assert!(text.contains(" ja "), "{text}");
2231
2232 let arg = "__builtin_va_arg(ap, double)";
2233 let text = asm(&format!("double f(int n, ...) {{ {read} return {arg}; }}\n"));
2234 assert!(text.contains("$160, "), "the last vector slot: {text}");
2235 }
2236
2237 #[test]
2240 fn a_structure_assignment_is_a_move_for_each_word_of_it() {
2241 let decl = "struct pair { long a, b; };\n";
2242 let body = "struct pair p = *q; return p.a + p.b;";
2243 let text = asm(&format!("{decl}long f(struct pair *q) {{ {body} }}\n"));
2244
2245 assert!(!text.contains("memcpy"), "nothing calls the library: {text}");
2246 assert!(!text.contains("\tcall"), "{text}");
2247 assert!(text.matches("\tmovq\t").count() >= 4, "two words each way: {text}");
2249 }
2250
2251 #[test]
2254 fn how_wide_a_word_of_a_copy_is_follows_the_alignment() {
2255 let decl = "struct bytes { char a[8]; };\n";
2256 let body = "struct bytes p = *q; return p.a[0];";
2257 let text = asm(&format!("{decl}int f(struct bytes *q) {{ {body} }}\n"));
2258
2259 assert!(text.matches("\tmovb\t").count() >= 16, "a byte at a time: {text}");
2261 }
2262
2263 #[test]
2266 fn the_part_of_an_initialiser_that_names_nothing_is_stored_as_zero() {
2267 let decl = "struct wide { long a, b, c; };\n";
2268 let text = asm(&format!("{decl}long f(void) {{ struct wide w = {{ 7 }}; return w.c; }}\n"));
2269
2270 assert!(!text.contains("memset"), "nothing calls the library: {text}");
2271 assert!(text.contains("\tmovq\t$0, ") || text.contains("$0, %"), "the zero: {text}");
2272 }
2273
2274 #[test]
2277 fn a_copy_too_large_to_unroll_calls_the_runtime() {
2278 let decl = "struct huge { char a[4096]; };\n";
2279 let mut opts = options();
2280 opts.emit = EmitKind::Asm;
2281 let source = format!("{decl}void f(struct huge *p, struct huge *q) {{ *p = *q; }}\n");
2282 let result = run(&opts, &source);
2283 assert!(!result.failed(), "{:?}", result.messages);
2284 let text = result.text();
2285 assert!(text.contains("call") && text.contains("memcpy"), "{text}");
2286 assert!(text.contains("4096"), "the size travels: {text}");
2289 }
2290
2291 #[test]
2294 fn a_realigned_frame_reads_them_through_the_frame_pointer() {
2295 let six = "long a, long b, long c, long d, long e, long f";
2296 let body = "_Alignas(32) long wide[4]; wide[0] = g; return wide[0];";
2297 let text = asm(&format!("long f({six}, long g) {{ {body} }}\n"));
2298
2299 assert!(text.contains("\tandq\t$-32, %rsp"), "{text}");
2303 assert!(text.contains("\tmovq\t16(%rbp), "), "{text}");
2304 assert!(!text.contains("\tmovq\t16(%rsp), "), "{text}");
2305 }
2306
2307 #[test]
2309 fn the_target_decides_how_the_assembly_is_spelled() {
2310 let mut opts = options();
2311 opts.emit = EmitKind::Asm;
2312 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2313 let text = run(&opts, "int f(void) { return 0; }\n").text().to_owned();
2314 assert!(text.contains("__TEXT,__text"), "{text}");
2315 assert!(text.contains("\n_f:\n"), "{text}");
2316 assert!(!text.contains(".note.GNU-stack"), "{text}");
2317 }
2318
2319 fn obj(source: &str) -> Vec<u8> {
2321 let mut opts = options();
2322 opts.emit = EmitKind::Object;
2323 let result = run(&opts, source);
2324 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2325 match result.artifact {
2326 Artifact::Object(bytes) => bytes,
2327 other => panic!("expected an object, got {other:?}"),
2328 }
2329 }
2330
2331 #[test]
2337 fn a_function_goes_from_c_to_an_object_a_linker_would_take() {
2338 let bytes = obj("int add(int a, int b) { return a + b; }\n");
2339 assert_eq!(&bytes[..4], b"\x7fELF", "an object file starts by saying it is one");
2340 let text = asm("int add(int a, int b) { return a + b; }\n");
2341 assert!(
2342 text.contains("\taddl\t"),
2343 "and the listing of it is the same instructions:\n{text}"
2344 );
2345 }
2346
2347 #[test]
2349 fn a_variable_goes_from_c_to_the_section_it_belongs_in() {
2350 let text = asm("int counter = 42;\nstatic int hidden;\nconst int fixed = 7;\n");
2351 assert!(text.contains("\t.data\n\t.globl\tcounter\n"), "{text}");
2352 assert!(text.contains("\ncounter:\n\t.long\t42\n"), "{text}");
2353 assert!(text.contains("\t.size\tcounter, .-counter\n"), "{text}");
2354 assert!(text.contains("\t.bss\n\t.p2align\t2\n"), "{text}");
2357 assert!(text.contains("\nhidden:\n\t.space\t4\n"), "{text}");
2358 assert!(!text.contains(".globl\thidden"), "{text}");
2359 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2362 }
2363
2364 #[test]
2371 fn a_bit_field_initializer_writes_every_byte_of_the_value_and_not_only_the_ones_that_are_set() {
2372 let text = asm("struct s { unsigned f : 20; } x = { 0x12300 };\n");
2373 assert!(text.contains("\t.data\n"), "there is something to write: {text}");
2374 assert!(text.contains("\nx:\n\t.ascii\t\"\\000#\\001\"\n"), "and it is the value: {text}");
2375
2376 let text = asm("struct s { unsigned a : 8; unsigned b : 8; } x = { 0, 3 };\n");
2379 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\003\"\n"), "{text}");
2380
2381 let text = asm("struct s { unsigned long long f : 40; } x = { 0x100000 };\n");
2384 assert!(text.contains("\nx:\n\t.ascii\t\"\\000\\000\\020\"\n\t.space\t5\n"), "{text}");
2385
2386 let text = asm("struct s { unsigned f : 20; } x = { 0 };\n");
2388 assert!(text.contains("\t.bss\n"), "an object of zeroes is zeroes: {text}");
2389 assert!(text.contains("\nx:\n\t.space\t4\n"), "{text}");
2390 }
2391
2392 #[test]
2394 fn a_string_literal_is_a_variable_with_a_name_no_program_could_write() {
2395 let text = asm("const char *f(void) { return \"hi\"; }\n");
2396 assert!(text.contains("\t.ascii\t\"hi\\000\"\n"), "{text}");
2397 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2398 let label = text
2399 .lines()
2400 .find(|line| line.starts_with(".Lstr"))
2401 .unwrap_or_else(|| panic!("a label for the literal in\n{text}"));
2402 assert!(!text.contains(&format!(".globl\t{}", label.trim_end_matches(':'))), "{text}");
2403 }
2404
2405 #[test]
2407 fn an_address_in_an_initializer_is_left_to_the_linker() {
2408 let source = "int counter;\nint *p = &counter;\n";
2409 let text = asm(source);
2410 assert!(text.contains("\np:\n\t.quad\tcounter\n"), "{text}");
2411 let bytes = obj(source);
2414 assert!(bytes.windows(8).any(|w| w == b"counter\0"), "the object has to name it");
2415 }
2416
2417 #[test]
2426 fn a_constant_holding_an_address_goes_in_the_section_the_loader_may_write_once() {
2427 let text = asm("static void a(void) {}\nstatic void b(void) {}\n\
2430 struct m { void (*x)(void); void (*y)(void); };\n\
2431 const struct m t = { a, b };\n");
2432 assert!(text.contains("\t.section\t.data.rel.ro.local,\"aw\",@progbits\n"), "{text}");
2433 assert!(text.contains("\nt:\n\t.quad\ta\n\t.quad\tb\n"), "{text}");
2434
2435 let text =
2438 asm("void a(void);\nstruct m { void (*x)(void); };\nconst struct m t = { a };\n");
2439 assert!(text.contains("\t.section\t.data.rel.ro,\"aw\",@progbits\n"), "{text}");
2440
2441 let text = asm("const int fixed = 7;\n");
2443 assert!(text.contains("\t.section\t.rodata\n"), "{text}");
2444 }
2445
2446 #[test]
2448 fn a_thread_local_variable_is_reported_as_work_that_is_not_done() {
2449 let mut opts = options();
2450 opts.emit = EmitKind::Asm;
2451 let result = run(&opts, "_Thread_local int x = 1;\n");
2452 assert!(result.failed(), "every thread sharing one variable is worse than a message");
2453 assert!(result.messages.iter().any(|m| m.contains("thread-local")), "{:?}", result);
2454 assert!(!result.messages.iter().any(|m| m.contains("internal")), "{:?}", result);
2456 }
2457
2458 #[test]
2460 fn the_object_and_the_listing_are_two_spellings_of_one_compilation() {
2461 let source = "int callee(void); int g(void) { return callee(); }\n";
2465 let bytes = obj(source);
2466 assert!(
2467 bytes.windows(7).any(|w| w == b"callee\0"),
2468 "the object has to name the callee for the linker to find it"
2469 );
2470 let text = asm(source);
2471 assert!(text.contains("\tcall\tcallee\n"), "{text}");
2472 }
2473
2474 #[test]
2480 fn compiling_for_an_executable_produces_an_object_and_not_a_dump() {
2481 let mut opts = options();
2482 opts.emit = EmitKind::Executable;
2484 let result = run(&opts, "int main(void) { return 0; }\n");
2485 assert_eq!(result.messages, Vec::<String>::new());
2486 match result.artifact {
2487 Artifact::Object(bytes) => assert_eq!(&bytes[..4], b"\x7fELF"),
2488 other => panic!("expected an object, got {other:?}"),
2489 }
2490 }
2491
2492 #[test]
2494 fn a_platform_with_no_object_writer_is_said_so_rather_than_written_as_elf() {
2495 let mut opts = options();
2496 opts.emit = EmitKind::Object;
2497 opts.target = "x86_64-apple-darwin".parse::<Triple>().unwrap();
2498 let result = run(&opts, "int f(void) { return 0; }\n");
2499 assert!(result.failed(), "an object nobody can read is worse than a message");
2500 assert!(
2501 result.messages.iter().any(|m| m.contains("no object writer")),
2502 "{:?}",
2503 result.messages
2504 );
2505 }
2506
2507 fn ir(source: &str) -> String {
2509 let mut opts = options();
2510 opts.emit = EmitKind::Ir;
2511 let result = run(&opts, source);
2512 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2513 result.text().to_owned()
2514 }
2515
2516 fn errors(source: &str) -> Vec<String> {
2518 let mut opts = options();
2519 opts.emit = EmitKind::Ir;
2520 let result = run(&opts, source);
2521 assert!(result.failed(), "expected this to be refused:\n{source}");
2522 result.messages
2523 }
2524
2525 fn body(source: &str) -> String {
2527 let text = ir(source);
2528 let (_, rest) = text.split_once("{\n").expect("a function definition");
2529 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
2530 body.to_owned()
2531 }
2532
2533 #[test]
2541 fn gnu89_inline_is_what_decides_whether_a_bare_inline_definition_reaches_the_module() {
2542 let source = "inline int f(int x) { return x + 1; }\n";
2543 let with = |flag: bool| {
2544 let mut opts = options();
2545 opts.emit = EmitKind::Ir;
2546 opts.gnu89_inline = flag;
2547 let result = run(&opts, source);
2548 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
2549 result.text().to_owned()
2550 };
2551
2552 assert!(!with(false).contains("block0"), "no body: {}", with(false));
2555
2556 assert!(with(true).contains("block0"), "a body: {}", with(true));
2559 }
2560
2561 #[test]
2568 fn an_access_through_a_type_names_the_type_it_went_through() {
2569 let source = "\
2570struct s { int a; float b; };\n\
2571union u { int i; float f; };\n\
2572int scalar(int *p) { return *p; }\n\
2573float member(struct s *p) { p->a = 1; return p->b; }\n\
2574int element(int *a, long i) { return a[i]; }\n\
2575float through_a_union(union u *p) { p->i = 1; return p->f; }\n";
2576 let text = ir(source);
2577 assert!(text.contains(r#"!0 = tbaa "char""#), "the root: {text}");
2578 assert!(text.contains(r#"tbaa "int", parent !0"#), "int under it: {text}");
2579 assert!(text.contains(r#"tbaa "float", parent !0"#), "float under it: {text}");
2580 let named = text.lines().filter(|line| line.contains(", tbaa !")).count();
2583 assert_eq!(named, 6, "six accesses: {text}");
2584 }
2585
2586 #[test]
2593 fn turning_strict_aliasing_off_leaves_the_type_off_every_access() {
2594 let source = "int punned(float *f, int *i) { *i = 1; *f = 2.0f; return *i; }\n";
2595 let mut opts = options();
2596 opts.emit = EmitKind::Ir;
2597 opts.strict_aliasing = false;
2598 let result = run(&opts, source);
2599 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile");
2600 let text = result.text().to_owned();
2601 assert!(!text.contains("tbaa"), "not even the root: {text}");
2602 }
2603
2604 #[test]
2612 fn a_bare_return_from_a_function_that_promised_a_value_gives_back_a_zero() {
2613 let mut opts = options();
2614 opts.emit = EmitKind::Ir;
2615 opts.std = Std::C89;
2616 let compiled = |source: &str| {
2617 let result = run(&opts, source);
2618 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
2619 result.text().to_owned()
2620 };
2621
2622 let text = compiled("int f(int x) { if (x) return; return 3; }\n");
2623 assert!(text.contains("iconst.i32 0\n return"), "zero goes back: {text}");
2624 assert!(!text.contains("unreachable"), "the branch that reached it is kept: {text}");
2625
2626 let text = compiled("double f(int x) { if (x) return; return 1.0; }\n");
2628 assert!(text.contains("fconst.f64 0x0\n return"), "a float zero goes back: {text}");
2629 }
2630
2631 #[test]
2639 fn a_call_to_a_name_nothing_declared_declares_it_as_c89_said_to() {
2640 let mut opts = options();
2641 opts.emit = EmitKind::Ir;
2642 opts.std = Std::C89;
2643 let compiled = |source: &str| {
2644 let result = run(&opts, source);
2645 assert_eq!(result.messages, Vec::<String>::new(), "C89 has nothing to say about this");
2646 result.text().to_owned()
2647 };
2648
2649 let text = compiled("int f(void) { return g(); }\n");
2651 assert!(text.contains("call @g"), "the call is to the name that was written: {text}");
2652 assert!(text.contains("i32"), "and it gives back an int: {text}");
2653
2654 let text = compiled("int f(char c) { return g(c); }\n");
2657 assert!(text.contains("sext.i32"), "the argument is promoted: {text}");
2658
2659 let mut opts = options();
2662 opts.std = Std::C89;
2663 let said = run(&opts, "int f(void) { return h; }\n").messages.join("\n");
2664 assert!(said.contains("'h' undeclared"), "not a call, so not declared: {said}");
2665 }
2666
2667 #[test]
2677 fn a_name_called_before_it_is_defined_still_gets_its_definition() {
2678 let mut opts = options();
2679 opts.emit = EmitKind::Ir;
2680 opts.std = Std::C89;
2681 let text = run(&opts, "int f(void) { return dummy(); }\ndummy () { return 7; }\n")
2682 .text()
2683 .to_owned();
2684 assert!(text.contains("func @f()"), "the caller is there: {text}");
2685 assert!(text.contains("func @dummy"), "and so is what it calls: {text}");
2686 assert!(text.contains("iconst.i32 7"), "with the body it was given: {text}");
2687 }
2688
2689 #[test]
2697 fn an_old_style_parameter_is_converted_from_what_the_call_promoted_it_to() {
2698 let mut opts = options();
2699 opts.emit = EmitKind::Ir;
2700 opts.std = Std::C89;
2701 let compiled = |source: &str| run(&opts, source).text().to_owned();
2702
2703 let text = compiled("f (c) unsigned char c; { return c; }\n");
2704 assert!(text.contains("func @f(i32"), "an int arrives: {text}");
2705 assert!(text.contains("trunc.i8"), "and is cut down to what was declared: {text}");
2706 assert!(text.contains("zext.i32"), "then read back unsigned: {text}");
2707
2708 let text = compiled("f (s) short s; { return s; }\n");
2710 assert!(text.contains("trunc.i16"), "cut down: {text}");
2711 assert!(text.contains("sext.i32"), "and read back signed: {text}");
2712
2713 let text = compiled("f (x) float x; { return x * 2; }\n");
2716 assert!(text.contains("func @f(f64"), "a double arrives: {text}");
2717 assert!(text.contains("fptrunc.f32"), "and is narrowed to the float: {text}");
2718
2719 let text = compiled("int f(unsigned char c) { return c; }\n");
2722 assert!(text.contains("func @f(i8)"), "the declared type arrives: {text}");
2723 assert!(!text.contains("trunc"), "so there is nothing to cut down: {text}");
2724 }
2725
2726 #[test]
2735 fn the_rules_gcc_promoted_are_decided_by_the_dialect_and_by_fpermissive() {
2736 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
2738 let cases = [
2739 ("static counted;\n", ["", "error", "warning", "error"]),
2740 ("int f(void) { return g(); }\n", ["", "error", "warning", "error"]),
2741 ("int f(x) { return x; }\n", ["", "error", "warning", "error"]),
2742 ("int *p;\nvoid h(void) { p = 1; }\n", ["warning", "error", "warning", "error"]),
2743 (
2744 "char *q;\nint *r;\nvoid k(void) { r = q; }\n",
2745 ["warning", "error", "warning", "error"],
2746 ),
2747 ("int f(void) { return; }\n", ["", "error", "warning", "error"]),
2748 ("void g(void) { return 1; }\n", ["warning", "error", "warning", "error"]),
2749 ];
2750
2751 for (source, wanted) in cases {
2752 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
2753 let mut opts = options();
2754 opts.std = std;
2755 opts.permissive = permissive;
2756 let said = run(&opts, source).messages.join("\n");
2757 let severity = if said.contains(": error: ") {
2758 "error"
2759 } else if said.contains(": warning: ") {
2760 "warning"
2761 } else {
2762 ""
2763 };
2764 let how = if permissive { " -fpermissive" } else { "" };
2765 assert_eq!(
2766 severity,
2767 wanted,
2768 "under -std={}{how}, {source} was answered with `{said}`",
2769 std.as_str()
2770 );
2771 if wanted.is_empty() {
2772 assert!(said.is_empty(), "nothing to say, but said `{said}`");
2773 }
2774 }
2775 }
2776 }
2777
2778 #[test]
2787 fn the_three_variadic_builtins_answer_a_bad_list_the_way_a_call_answers_a_bad_argument() {
2788 let modes = [(Std::C89, false), (Std::C17, false), (Std::C17, true), (Std::C23, false)];
2789 let cases = [
2790 (
2791 "int f(int n, ...) { char *p; return __builtin_va_arg(p, int); }\n",
2792 "first argument to 'va_arg' not of type 'va_list'",
2793 ["error", "error", "error", "error"],
2794 ),
2795 (
2796 "void f(int n, ...) { char *p; __builtin_va_start(p, n); }\n",
2797 "passing argument 1 of '__builtin_va_start' from incompatible pointer type",
2798 ["warning", "error", "warning", "error"],
2799 ),
2800 (
2801 "void f(int n, ...) { int x; __builtin_va_end(x); }\n",
2802 "passing argument 1 of '__builtin_va_end' makes pointer from integer without a \
2803 cast",
2804 ["warning", "error", "warning", "error"],
2805 ),
2806 (
2807 "void f(int n, ...) { __builtin_va_list a; char *p; __builtin_va_copy(a, p); }\n",
2808 "passing argument 2 of '__builtin_va_copy' from incompatible pointer type",
2809 ["warning", "error", "warning", "error"],
2810 ),
2811 ];
2812
2813 for (source, message, wanted) in cases {
2814 for (&(std, permissive), wanted) in modes.iter().zip(wanted) {
2815 let mut opts = options();
2816 opts.std = std;
2817 opts.permissive = permissive;
2818 let said = run(&opts, source).messages.join("\n");
2819 let how = if permissive { " -fpermissive" } else { "" };
2820 assert!(
2821 said.contains(&format!(": {wanted}: {message}")),
2822 "under -std={}{how}, {source} was answered with `{said}`",
2823 std.as_str()
2824 );
2825 }
2826 }
2827 }
2828
2829 fn safe_ir(tier: rucc_session::Safety, source: &str) -> String {
2831 let mut opts = options();
2832 opts.emit = EmitKind::Ir;
2833 opts.safety = tier;
2834 let result = run(&opts, source);
2835 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2836 result.text().to_owned()
2837 }
2838
2839 const READS_THROUGH_A_POINTER: &str = "int read(int *p) { return p[1]; }\n";
2840
2841 fn padded_ir(padding: Padding, source: &str) -> String {
2843 let mut opts = options();
2844 opts.emit = EmitKind::Ir;
2845 opts.safety = rucc_session::Safety::Detect;
2846 opts.padding = padding;
2847 let result = run(&opts, source);
2848 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2849 result.text().to_owned()
2850 }
2851
2852 const FILLS_A_RECORD_A_MEMBER_AT_A_TIME: &str = "struct padded { char tag; int value; };\n\
2853 void fill(struct padded *p) { p->tag = 1; p->value = 2; }\n";
2854
2855 #[test]
2856 fn a_record_filled_a_member_at_a_time_comes_out_whole_when_padding_does_not_participate() {
2857 let text = padded_ir(Padding::Ignored, FILLS_A_RECORD_A_MEMBER_AT_A_TIME);
2861 assert_eq!(text.matches("owns 4").count(), 2, "{text}");
2862 }
2863
2864 #[test]
2865 fn a_store_says_only_what_it_wrote_when_padding_does_participate() {
2866 let text = padded_ir(Padding::Tracked, FILLS_A_RECORD_A_MEMBER_AT_A_TIME);
2869 assert!(!text.contains("owns"), "{text}");
2870 }
2871
2872 #[test]
2873 fn a_member_of_a_union_owns_nothing_after_it() {
2874 let text = padded_ir(
2878 Padding::Ignored,
2879 "union u { char tag; long wide; };\nvoid fill(union u *p) { p->tag = 1; }\n",
2880 );
2881 assert!(!text.contains("owns"), "{text}");
2882 }
2883
2884 #[test]
2885 fn an_inner_records_trailing_padding_reaches_the_outer_records() {
2886 let text = padded_ir(
2891 Padding::Ignored,
2892 "struct inner { char c; };\n\
2893 struct outer { struct inner in; int x; };\n\
2894 void fill(struct outer *p) { p->in.c = 1; p->x = 2; }\n",
2895 );
2896 assert_eq!(text.matches("owns 4").count(), 2, "{text}");
2897 }
2898
2899 #[test]
2900 fn a_build_that_did_not_ask_for_the_monitor_is_compiled_the_way_it_always_was() {
2901 let text = ir(READS_THROUGH_A_POINTER);
2905 assert!(!text.contains("check_"), "{text}");
2906 assert!(!text.contains("cap_of"), "{text}");
2907 }
2908
2909 #[test]
2910 fn asking_for_a_tier_puts_the_checks_in_before_the_optimizer_sees_them() {
2911 let text = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2912 assert!(text.contains("cap_of"), "{text}");
2913 assert!(text.contains("check_bounds"), "{text}");
2914 assert!(text.contains("check_live"), "{text}");
2915 assert!(text.contains("check_deriv"), "{text}");
2917 assert!(text.contains("check_type"), "{text}");
2919 }
2920
2921 #[test]
2922 fn the_three_tiers_that_are_not_off_all_check_the_same_accesses_so_far() {
2923 let detect = safe_ir(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2927 for tier in [rucc_session::Safety::Enforce, rucc_session::Safety::Kernel] {
2928 assert_eq!(safe_ir(tier, READS_THROUGH_A_POINTER), detect, "{tier}");
2929 }
2930 }
2931
2932 fn summary(tier: rucc_session::Safety, source: &str) -> String {
2934 let mut opts = options();
2935 opts.emit = EmitKind::SafetySummary;
2936 opts.safety = tier;
2937 let result = run(&opts, source);
2938 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
2939 result.text().to_owned()
2940 }
2941
2942 #[test]
2943 fn the_summary_counts_the_checks_that_went_in_and_the_ones_still_standing() {
2944 let text = summary(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
2945 assert!(text.contains("\"tier\": \"detect\""), "{text}");
2946 assert!(
2948 text.contains("\"bounds\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"),
2949 "{text}"
2950 );
2951 assert!(
2952 text.contains(
2953 "\"derivation\": { \"emitted\": 1, \"remaining\": 1, \"discharged\": 0 }"
2954 ),
2955 "{text}"
2956 );
2957 }
2958
2959 #[test]
2960 fn a_build_without_the_monitor_summarises_as_a_build_with_no_checks_in_it() {
2961 let text = summary(rucc_session::Safety::Off, READS_THROUGH_A_POINTER);
2965 assert!(text.contains("\"tier\": \"off\""), "{text}");
2966 assert!(
2967 text.contains("\"bounds\": { \"emitted\": 0, \"remaining\": 0, \"discharged\": 0 }"),
2968 "{text}"
2969 );
2970 }
2971
2972 #[test]
2973 fn a_call_the_boundary_models_is_counted_apart_from_one_it_does_not() {
2974 let text = summary(
2975 rucc_session::Safety::Detect,
2976 "void *memcpy(void *, const void *, unsigned long);\n\
2977 int puts(const char *);\n\
2978 void f(char *d, char *s) { memcpy(d, s, 4); puts(d); }\n",
2979 );
2980 assert!(text.contains("\"interposed\": 1"), "{text}");
2981 assert!(text.contains("\"puts\""), "{text}");
2982 assert!(!text.contains("__rucc_wrap_memcpy\""), "{text}");
2986 }
2987
2988 #[test]
2989 fn the_two_directions_a_pointer_crosses_the_boundary_are_counted_apart() {
2990 let text = summary(
2994 rucc_session::Safety::Detect,
2995 "void *notes_open(void);\n\
2996 char *f(char *p) { char *q = notes_open(); return q ? q : p; }\n",
2997 );
2998 assert!(text.contains("\"crossings\": { \"entered\": 1, \"returned\": 1 }"), "{text}");
2999 assert!(text.contains("\"notes_open\""), "{text}");
3000 }
3001
3002 #[test]
3003 fn a_static_function_nobody_takes_the_address_of_is_not_a_crossing() {
3004 let text = summary(
3007 rucc_session::Safety::Detect,
3008 "static int len(const char *p) { return p ? 1 : 0; }\n\
3009 int f(void) { return len(\"x\"); }\n",
3010 );
3011 assert!(text.contains("\"crossings\": { \"entered\": 0, \"returned\": 0 }"), "{text}");
3012 }
3013
3014 fn granules(source: &str) -> String {
3016 let mut opts = options();
3017 opts.emit = EmitKind::TypeGranules;
3018 let result = run(&opts, source);
3019 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3020 result.text().to_owned()
3021 }
3022
3023 #[test]
3024 fn the_granule_report_names_every_record_and_both_keyings() {
3025 let text = granules(
3026 "struct hot { char *p; int a; int b; };\n\
3027 int f(struct hot *h) { return h->a; }\n",
3028 );
3029 assert!(text.contains("struct hot"), "{text}");
3030 assert!(text.contains("every type distinct"), "{text}");
3033 assert!(text.contains("every pointer one type"), "{text}");
3034 assert!(text.contains("budget"), "{text}");
3035 }
3036
3037 #[test]
3038 fn a_record_nothing_uses_is_still_measured() {
3039 let text = granules("struct unused { long a; double b; };\nint f(void) { return 0; }\n");
3042 assert!(text.contains("struct unused"), "{text}");
3043 }
3044
3045 #[test]
3046 fn the_granule_report_stops_before_anything_is_lowered() {
3047 let text = granules(
3051 "struct wide { long double d; };\n\
3052 long double f(long double x) { return x * x; }\n",
3053 );
3054 assert!(text.contains("struct wide"), "{text}");
3055 }
3056
3057 #[test]
3058 fn a_witness_reaches_the_assembler_as_a_call_to_the_runtime() {
3059 let text = safe_asm(rucc_session::Safety::Detect, "char *f(char *p) { return p; }\n");
3062 assert!(text.contains("\tcall\t__rucc_cap_witness\n"), "{text}");
3063 }
3064
3065 #[test]
3066 fn a_pointer_turned_into_an_integer_is_on_the_trust_set() {
3067 let text = summary(
3068 rucc_session::Safety::Detect,
3069 "unsigned long f(int *p) { return (unsigned long) p; }\n",
3070 );
3071 assert!(text.contains("\"exposed\": 1"), "{text}");
3072 }
3073
3074 fn safe_asm(tier: rucc_session::Safety, source: &str) -> String {
3076 let mut opts = options();
3077 opts.emit = EmitKind::Asm;
3078 opts.safety = tier;
3079 let result = run(&opts, source);
3080 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
3081 result.text().to_owned()
3082 }
3083
3084 #[test]
3085 fn a_check_reaches_the_assembler_as_a_call_to_the_runtime() {
3086 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3087 assert!(text.contains("\tcall\t__rucc_check_bounds\n"), "{text}");
3088 assert!(text.contains("\tcall\t__rucc_check_live\n"), "{text}");
3089 assert!(text.contains("\tcall\t__rucc_check_deriv\n"), "{text}");
3090 assert!(text.contains("\tcall\t__rucc_check_type\n"), "{text}");
3091 assert!(text.contains("\tcall\t__rucc_check_init\n"), "{text}");
3092 }
3093
3094 #[test]
3095 fn every_check_that_reached_the_assembler_has_a_row_describing_it() {
3096 let text = safe_asm(rucc_session::Safety::Detect, READS_THROUGH_A_POINTER);
3100 let section = format!("\t.section\t{},", rucc_safety::SECTION);
3101 assert_eq!(text.matches(§ion).count(), 5, "{text}");
3102 for index in 0..5 {
3103 let name = format!("__rucc_safety_desc_{index}");
3104 assert!(text.contains(&format!("{name}:\n")), "{text}");
3107 assert!(text.contains(&format!("{name}(%rip)")), "{text}");
3108 }
3109 assert!(!text.contains("__rucc_safety_desc_5"), "{text}");
3110 }
3111
3112 #[test]
3120 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
3121 let text = ir(concat!(
3122 "int g;\n",
3123 "int a = __builtin_constant_p(1);\n",
3124 "int b = __builtin_constant_p(g);\n",
3125 "int c = __builtin_constant_p(\"abc\");\n",
3126 "int d = __builtin_constant_p(&g);\n",
3127 "int e = __builtin_constant_p(1.5);\n",
3128 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
3129 ));
3130 assert!(text.contains("global @a : i32 = 1,"), "{text}");
3131 assert!(text.contains("global @b : i32 = 0,"), "{text}");
3132 assert!(text.contains("global @c : i32 = 1,"), "{text}");
3133 assert!(text.contains("global @d : i32 = 0,"), "{text}");
3134 assert!(text.contains("global @e : i32 = 1,"), "{text}");
3135 assert!(text.contains("global @h : i32 = 11,"), "{text}");
3136 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
3137
3138 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
3142 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
3143 }
3144
3145 #[test]
3154 fn a_call_to_a_library_builtin_reaches_the_library_function() {
3155 let text = body("void f(void) { __builtin_abort(); }\n");
3156 assert_eq!(text, "block0:\n call @abort() : ()\n return\n");
3157
3158 let text = ir("int f(const char *s) { return __builtin_puts(s) + __builtin_strlen(s); }\n");
3161 assert!(text.contains("call @puts(%0) : (ptr) -> i32"), "{text}");
3162 assert!(text.contains("call @strlen(%0) : (ptr) -> i64"), "{text}");
3163 assert!(!text.contains("__builtin_"), "the prefix is not part of any name here:\n{text}");
3164 }
3165
3166 #[test]
3177 fn the_absolute_value_family_is_the_magnitude_and_not_a_call() {
3178 let text = body(concat!(
3179 "long long llabs(long long);\n",
3180 "long long f(long long x) { return llabs(x); }\n",
3181 ));
3182 assert!(text.contains("%1 = iconst.i64 63"), "{text}");
3183 assert!(text.contains("%2 = ashr %0, %1"), "{text}");
3184 assert!(text.contains("%3 = xor %0, %2"), "{text}");
3185 assert!(text.contains("%4 = sub %3, %2"), "{text}");
3186 assert!(!text.contains("call"), "the call does not happen:\n{text}");
3187
3188 let text = body("int abs(int);\nint f(int x) { return abs(x); }\n");
3191 assert!(text.contains("iconst.i32 31"), "{text}");
3192 let text = body("long labs(long);\nlong f(long x) { return labs(x); }\n");
3193 assert!(text.contains("iconst.i64 63"), "{text}");
3194
3195 let text = body("long long f(long long x) { return __builtin_llabs(x); }\n");
3198 assert!(!text.contains("call"), "{text}");
3199
3200 let text = ir(concat!(
3202 "long long llabs(long long b);\n",
3203 "long long g(long long x) { return llabs(x); }\n",
3204 "long long llabs(long long b) { return 7; }\n",
3205 ));
3206 assert!(!text.contains("call @llabs"), "{text}");
3207 }
3208
3209 #[test]
3216 fn a_byte_swap_is_arithmetic_and_not_a_call() {
3217 let text = body("unsigned f(unsigned x) { return __builtin_bswap32(x); }\n");
3218 assert_eq!(text, "block0(%0: i32):\n %1 = bswap %0\n return %1\n");
3219
3220 let text = body("unsigned f(unsigned char c) { return __builtin_bswap32(c); }\n");
3223 assert!(text.contains("zext.i32 %0"), "widened first: {text}");
3224 assert!(text.contains("bswap %1"), "and swapped at four bytes: {text}");
3225 }
3226
3227 #[test]
3233 fn the_byte_swaps_reverse_at_the_width_their_name_says() {
3234 for (name, ty, width) in [
3235 ("__builtin_bswap16", "unsigned short", "i16"),
3236 ("__builtin_bswap32", "unsigned", "i32"),
3237 ("__builtin_bswap64", "unsigned long long", "i64"),
3238 ] {
3239 let source = format!("{ty} f({ty} x) {{ return {name}(x); }}\n");
3240 let text = body(&source);
3241 assert_eq!(
3242 text,
3243 format!("block0(%0: {width}):\n %1 = bswap %0\n return %1\n"),
3244 "{name}"
3245 );
3246 }
3247 }
3248
3249 #[test]
3256 fn the_bit_counts_are_instructions_and_not_calls() {
3257 let text = body("int f(unsigned x) { return __builtin_clz(x); }\n");
3258 assert_eq!(text, "block0(%0: i32):\n %1 = ctlz %0\n return %1\n");
3259
3260 let text = body("int f(unsigned x) { return __builtin_ctz(x); }\n");
3261 assert_eq!(text, "block0(%0: i32):\n %1 = cttz %0\n return %1\n");
3262
3263 let text = body("int f(unsigned x) { return __builtin_popcount(x); }\n");
3264 assert_eq!(text, "block0(%0: i32):\n %1 = ctpop %0\n return %1\n");
3265 }
3266
3267 #[test]
3276 fn the_bit_counts_ask_about_the_width_their_name_says() {
3277 let text = body("int f(unsigned long long x) { return __builtin_clzll(x); }\n");
3278 assert!(text.starts_with("block0(%0: i64):"), "counted at eight bytes: {text}");
3279 assert!(text.contains("%1 = ctlz %0"), "{text}");
3280 assert!(text.contains("trunc.i32 %1"), "and answered in an int: {text}");
3281
3282 let text = body("int f(unsigned long long x) { return __builtin_clz(x); }\n");
3285 assert!(text.contains("trunc.i32 %0"), "narrowed to what was asked about: {text}");
3286 assert!(text.contains("ctlz %1"), "and counted there: {text}");
3287
3288 let text = body("int f(unsigned long x) { return __builtin_popcountl(x); }\n");
3289 assert!(text.contains("%1 = ctpop %0"), "{text}");
3290 assert!(!text.contains("call"), "{text}");
3291 }
3292
3293 #[test]
3298 fn a_parity_is_the_low_bit_of_the_set_bit_count() {
3299 let text = body("int f(unsigned x) { return __builtin_parity(x); }\n");
3300 assert!(text.contains("%1 = ctpop %0"), "{text}");
3301 assert!(text.contains("iconst.i32 1"), "{text}");
3302 assert!(text.contains("and %1, %2"), "the low bit of it: {text}");
3303 }
3304
3305 #[test]
3311 fn the_first_set_bit_is_one_based_and_zero_for_a_zero() {
3312 let text = body("int f(int x) { return __builtin_ffs(x); }\n");
3313 assert!(text.contains("%1 = cttz %0"), "{text}");
3314 assert!(text.contains("%4 = add %1, %2"), "one more than the count: {text}");
3315 assert!(text.contains("%5 = icmp ne %0, %3"), "whether there was a bit at all: {text}");
3316 assert!(text.contains("%7 = sub %3, %6"), "spread to a mask: {text}");
3317 assert!(text.contains("%8 = and %4, %7"), "and kept only then: {text}");
3318 assert!(!text.contains("br_if"), "no branch: {text}");
3319 }
3320
3321 #[test]
3331 fn an_overflow_check_is_arithmetic_and_not_a_call() {
3332 let text =
3333 body("int f(int a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3334 assert!(text.contains("%3, %4 = sadd_overflow.(i32, i1) %0, %1"), "{text}");
3335 assert!(text.contains("store %3 -> %2"), "{text}");
3336 assert!(!text.contains("call"), "{text}");
3337
3338 let text =
3339 body("int f(int a, int b, int *r) { return __builtin_sub_overflow(a, b, r); }\n");
3340 assert!(text.contains("ssub_overflow.(i32, i1) %0, %1"), "{text}");
3341
3342 let text =
3343 body("int f(int a, int b, int *r) { return __builtin_mul_overflow(a, b, r); }\n");
3344 assert!(text.contains("smul_overflow.(i32, i1) %0, %1"), "{text}");
3345
3346 let text = body(
3349 "int f(unsigned a, unsigned b, unsigned *r) { return __builtin_add_overflow(a, b, r); }\n",
3350 );
3351 assert!(text.contains("uadd_overflow.(i32, i1) %0, %1"), "{text}");
3352 }
3353
3354 #[test]
3362 fn an_overflow_check_is_done_at_a_type_that_holds_every_operand() {
3363 let text = body(
3364 "int f(unsigned a, int b, long long *r) { return __builtin_add_overflow(a, b, r); }\n",
3365 );
3366 assert!(text.contains("%3 = zext.i64 %0"), "the unsigned operand keeps its value: {text}");
3367 assert!(text.contains("%4 = sext.i64 %1"), "and so does the signed one: {text}");
3368 assert!(text.contains("sadd_overflow.(i64, i1) %3, %4"), "{text}");
3369
3370 let text = body(
3373 "int f(long long a, long long b, long long *r) { return __builtin_mul_overflow(a, b, r); }\n",
3374 );
3375 assert!(text.contains("smul_overflow.(i64, i1) %0, %1"), "{text}");
3376 assert!(!text.contains("sext."), "{text}");
3377 assert!(!text.contains("zext.i64"), "{text}");
3379 }
3380
3381 #[test]
3389 fn an_overflow_check_writes_the_wrapped_answer_whether_or_not_it_fit() {
3390 let text =
3391 body("int f(int a, int b, char *r) { return __builtin_sub_overflow(a, b, r); }\n");
3392 assert!(text.contains("%3, %4 = ssub_overflow.(i32, i1) %0, %1"), "{text}");
3393 assert!(text.contains("%5 = trunc.i8 %3"), "narrowed to where it goes: {text}");
3394 assert!(text.contains("%6 = sext.i32 %5"), "and back: {text}");
3395 assert!(text.contains("%7 = icmp ne %6, %3"), "which is whether it fit: {text}");
3396 assert!(text.contains("store %5 -> %2"), "the narrowed value is stored either way: {text}");
3397 assert!(text.contains("%8 = or %4, %7"), "and either bit is an overflow: {text}");
3398 }
3399
3400 #[test]
3410 fn a_call_needing_more_than_the_widest_type_says_so() {
3411 let refused = concat!(
3412 "int f(unsigned __int128 a, long long b, __int128 *r) {\n",
3413 " return __builtin_add_overflow(a, b, r);\n",
3414 "}\n",
3415 );
3416 let messages = errors(refused);
3417 assert_eq!(messages.len(), 1, "{messages:?}");
3418 assert!(messages[0].contains("E0694"), "{messages:?}");
3419 assert!(messages[0].contains("wider than 128 bits"), "{messages:?}");
3420
3421 let taken = concat!(
3422 "int f(unsigned long long a, long long b, long long *r) {\n",
3423 " return __builtin_add_overflow(a, b, r);\n",
3424 "}\n",
3425 );
3426 let mut opts = options();
3427 opts.emit = EmitKind::MirFinal;
3428 assert!(!run(&opts, taken).failed(), "the width above it is where that one goes now");
3429 }
3430
3431 #[test]
3434 fn an_overflow_check_over_something_that_is_not_an_integer_says_so() {
3435 let messages =
3436 errors("int f(double a, int b, int *r) { return __builtin_add_overflow(a, b, r); }\n");
3437 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3438
3439 let messages =
3440 errors("int f(int a, int b, double *r) { return __builtin_add_overflow(a, b, r); }\n");
3441 assert!(messages.iter().any(|line| line.contains("E0671")), "{messages:?}");
3442 }
3443
3444 #[test]
3455 fn an_ordered_access_is_ordered_in_the_ir() {
3456 let text = body("int f(int *p) { return __atomic_load_n(p, 0); }\n");
3457 assert!(text.contains("atomic_load.i32 %0, align 4, relaxed"), "{text}");
3458
3459 let text = body("long f(long *p) { return __atomic_load_n(p, 2); }\n");
3460 assert!(text.contains("atomic_load.i64 %0, align 8, acquire"), "{text}");
3461
3462 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
3463 assert!(text.contains("atomic_store %1 -> %0, align 4, release"), "{text}");
3464
3465 let text = body("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
3466 assert!(text.contains("atomic_store %1 -> %0, align 4, seq_cst"), "{text}");
3467
3468 let text = body("void f(char *p, int v) { __atomic_store_n(p, v, 0); }\n");
3471 assert!(text.contains("trunc.i8 %1"), "{text}");
3472 assert!(text.contains("atomic_store %2 -> %0, align 1, relaxed"), "{text}");
3473 }
3474
3475 #[test]
3484 fn an_ordered_access_is_the_plain_instruction_on_this_machine() {
3485 let text = asm("int f(int *p) { return __atomic_load_n(p, 5); }\n");
3486 assert!(text.contains("movl\t(%rdi), %eax"), "{text}");
3487 assert!(!text.contains("mfence"), "a load needs no barrier here: {text}");
3488
3489 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 3); }\n");
3490 assert!(text.contains("movl\t%esi, (%rdi)"), "{text}");
3491 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
3492
3493 let text = asm("void f(int *p, int v) { __atomic_store_n(p, v, 5); }\n");
3494 let (before, after) = text.split_once("mfence").expect("a barrier: {text}");
3495 assert!(before.contains("movl\t%esi, (%rdi)"), "the store comes first: {text}");
3496 assert!(!after.contains("movl"), "and nothing else is between them: {text}");
3497 }
3498
3499 #[test]
3509 fn a_barrier_is_one_instruction_at_the_strongest_ordering_and_none_below_it() {
3510 assert!(asm("void f(void) { __atomic_thread_fence(5); }\n").contains("mfence"));
3511 assert!(asm("void f(void) { __sync_synchronize(); }\n").contains("mfence"));
3512
3513 for weaker in ["1", "2", "3", "4"] {
3514 let source = format!("void f(void) {{ __atomic_thread_fence({weaker}); }}\n");
3515 assert!(!asm(&source).contains("mfence"), "{weaker} costs nothing here");
3516 }
3517 }
3518
3519 #[test]
3525 fn a_compare_and_exchange_is_one_instruction_answering_two_things() {
3526 let text =
3529 body("int f(int *p, int e, int d) { return __sync_val_compare_and_swap(p, e, d); }\n");
3530 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
3531 assert!(text.contains("return %3"), "the value it found: {text}");
3532
3533 let text =
3534 body("int f(int *p, int e, int d) { return __sync_bool_compare_and_swap(p, e, d); }\n");
3535 assert!(text.contains("%3, %4 = cmpxchg.(i32, i1) %0, %1, %2, align 4, seq_cst"), "{text}");
3536 assert!(text.contains("zext.i32 %4"), "whether it happened: {text}");
3537
3538 let text = body(
3541 "int f(int *p, int *e, int d) { return __atomic_compare_exchange_n(p, e, d, 0, 4, 2); }\n",
3542 );
3543 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3544 assert!(text.contains("%4, %5 = cmpxchg.(i32, i1) %0, %3, %2, align 4, acq_rel"), "{text}");
3545 assert!(text.contains("br_if %5, block2, block1"), "{text}");
3546 assert!(text.contains("store %4 -> %1, align 4"), "{text}");
3547
3548 let text = body(
3551 "int f(int *p, int *e, int *d) { return __atomic_compare_exchange(p, e, d, 0, 5, 5); }\n",
3552 );
3553 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3554 assert!(text.contains("%4 = load.i32 %2, align 4"), "{text}");
3555 assert!(text.contains("%5, %6 = cmpxchg.(i32, i1) %0, %3, %4, align 4, seq_cst"), "{text}");
3556 }
3557
3558 #[test]
3565 fn a_compare_and_exchange_is_a_locked_instruction_at_the_width_of_the_object() {
3566 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
3567 for (ty, suffix, reg) in widths {
3568 let source = format!(
3569 "int f({ty} *p, {ty} e, {ty} d) {{ return __sync_bool_compare_and_swap(p, e, d); }}\n"
3570 );
3571 let text = asm(&source);
3572 assert!(text.contains("\tlock\n"), "{ty}: {text}");
3573 assert!(text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3574 assert!(text.contains("sete\t"), "{ty}: {text}");
3575 }
3576 let source =
3577 "int f(long *p, long e, long d) { return __sync_bool_compare_and_swap(p, e, d); }\n";
3578 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
3579
3580 for order in ["0", "2", "3", "4", "5"] {
3584 let call = format!("__atomic_compare_exchange_n(p, e, d, 0, {order}, 0)");
3585 let source = format!("int f(int *p, int *e, int d) {{ return {call}; }}\n");
3586 let text = asm(&source);
3587 assert!(text.contains("cmpxchgl\t"), "{order}: {text}");
3588 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
3589 }
3590 }
3591
3592 #[test]
3604 fn a_read_modify_write_is_one_instruction_and_the_arithmetic_a_name_asks_for() {
3605 let text = body("int f(int *p, int v) { return __atomic_fetch_add(p, v, 5); }\n");
3606 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
3607 assert!(text.contains("return %2"), "the value that was there: {text}");
3608
3609 let text = body("int f(int *p, int v) { return __atomic_add_fetch(p, v, 5); }\n");
3610 assert!(text.contains("%2 = atomic_rmw.i32 add %0, %1, align 4, seq_cst"), "{text}");
3611 assert!(text.contains("%3 = add %2, %1"), "and the value afterwards: {text}");
3612
3613 let text = body("int f(int *p, int v) { return __atomic_sub_fetch(p, v, 5); }\n");
3614 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
3615 assert!(text.contains("%3 = sub %2, %1"), "{text}");
3616
3617 let text = body("int f(int *p, int v) { return __sync_fetch_and_sub(p, v); }\n");
3619 assert!(text.contains("%2 = atomic_rmw.i32 sub %0, %1, align 4, seq_cst"), "{text}");
3620
3621 let text = body("int f(int *p, int v) { return __atomic_exchange_n(p, v, 5); }\n");
3624 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, seq_cst"), "{text}");
3625
3626 let text = body("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
3627 assert!(text.contains("%2 = atomic_rmw.i32 xchg %0, %1, align 4, acquire"), "{text}");
3628
3629 let text = body("void f(int *p) { __sync_lock_release(p); }\n");
3632 assert!(text.contains("release"), "{text}");
3633 assert!(text.contains("%1 = iconst.i32 0"), "{text}");
3634
3635 let text = body("void f(int *p, int guard) { __sync_lock_release(p, guard); }\n");
3639 assert!(text.contains("%2 = iconst.i32 0"), "{text}");
3640 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
3641
3642 let text = body("int f(int *p, int v) { return __atomic_fetch_and(p, v, 5); }\n");
3645 assert!(text.contains("%2 = atomic_rmw.i32 and %0, %1, align 4, seq_cst"), "{text}");
3646
3647 let text = body("int f(int *p, int v) { return __sync_or_and_fetch(p, v); }\n");
3648 assert!(text.contains("%2 = atomic_rmw.i32 or %0, %1, align 4, seq_cst"), "{text}");
3649 assert!(text.contains("%3 = or %2, %1"), "and the value afterwards: {text}");
3650
3651 let text = body("int f(int *p, int v) { return __atomic_nand_fetch(p, v, 5); }\n");
3654 assert!(text.contains("%2 = atomic_rmw.i32 nand %0, %1, align 4, seq_cst"), "{text}");
3655 assert!(text.contains("%3 = and %2, %1"), "{text}");
3656 assert!(text.contains("%4 = iconst.i32 -1"), "{text}");
3657 assert!(text.contains("%5 = xor %3, %4"), "{text}");
3658 }
3659
3660 #[test]
3671 fn a_bitwise_read_modify_write_is_a_loop_around_the_compare_and_exchange() {
3672 let widths = [("char", "b", "%dl"), ("short", "w", "%dx"), ("int", "l", "%edx")];
3673 for (ty, suffix, reg) in widths {
3674 for (name, call, insn) in [
3675 ("and", "__atomic_fetch_and(p, v, 5)", "and"),
3676 ("or", "__sync_fetch_and_or(p, v)", "or"),
3677 ("xor", "__atomic_xor_fetch(p, v, 5)", "xor"),
3678 ] {
3679 let source = format!("{ty} f({ty} *p, {ty} v) {{ return {call}; }}\n");
3680 let text = asm(&source);
3681 assert!(text.contains("\tlock\n"), "{ty} {name}: {text}");
3682 assert!(
3683 text.contains(&format!("cmpxchg{suffix}\t{reg}, (%rdi)")),
3684 "{ty} {name}: {text}"
3685 );
3686 assert!(text.contains(&format!("{insn}{suffix}\t")), "{ty} {name}: {text}");
3687 assert!(!text.contains("\txadd"), "{ty} {name} is not an add: {text}");
3689 assert!(!text.contains("\txchg"), "{ty} {name} is not an exchange: {text}");
3690 }
3691 }
3692 let source = "long f(long *p, long v) { return __atomic_fetch_or(p, v, 5); }\n";
3693 assert!(asm(source).contains("cmpxchgq\t%rdx, (%rdi)"), "{}", asm(source));
3694
3695 let text = asm("int f(int *p, int v) { return __sync_fetch_and_nand(p, v); }\n");
3699 assert!(text.contains("cmpxchgl\t"), "{text}");
3700 assert!(text.contains("andl\t"), "{text}");
3701 assert!(text.contains("notl\t"), "{text}");
3702 }
3703
3704 #[test]
3713 fn an_access_through_a_second_pointer_is_the_same_access_and_one_more() {
3714 let text = body("void f(int *p, int *r) { __atomic_load(p, r, 5); }\n");
3715 assert!(text.contains("%2 = atomic_load.i32 %0, align 4, seq_cst"), "{text}");
3716 assert!(text.contains("store %2 -> %1, align 4"), "and out through the place: {text}");
3717
3718 let text = body("void f(int *p, int *v) { __atomic_store(p, v, 3); }\n");
3719 assert!(text.contains("%2 = load.i32 %1, align 4"), "in through the place: {text}");
3720 assert!(text.contains("atomic_store %2 -> %0, align 4, release"), "{text}");
3721
3722 let text = body("void f(int *p, int *v, int *r) { __atomic_exchange(p, v, r, 5); }\n");
3725 assert!(text.contains("%3 = load.i32 %1, align 4"), "{text}");
3726 assert!(text.contains("%4 = atomic_rmw.i32 xchg %0, %3, align 4, seq_cst"), "{text}");
3727 assert!(text.contains("store %4 -> %2, align 4"), "{text}");
3728 }
3729
3730 #[test]
3741 fn a_flag_is_an_exchange_of_one_byte_and_a_store_of_a_zero_over_the_same_byte() {
3742 for pointer in ["char", "int", "void"] {
3743 let source = format!("int f({pointer} *p) {{ return __atomic_test_and_set(p, 5); }}\n");
3744 let text = body(&source);
3745 assert!(text.contains("%1 = iconst.i8 1"), "{pointer}: {text}");
3746 assert!(
3747 text.contains("%2 = atomic_rmw.i8 xchg %0, %1, align 1, seq_cst"),
3748 "{pointer}: {text}"
3749 );
3750 assert!(text.contains("%4 = icmp ne %2, %3"), "{pointer}: {text}");
3751
3752 let source = format!("void f({pointer} *p) {{ __atomic_clear(p, 3); }}\n");
3753 let text = body(&source);
3754 assert!(text.contains("atomic_store %2 -> %0, align 1, release"), "{pointer}: {text}");
3755 }
3756
3757 let text = asm("int f(int *p) { return __atomic_test_and_set(p, 5); }\n");
3760 assert!(text.contains("xchgb\t%al, (%rdi)"), "{text}");
3761 assert!(text.contains("setne\t"), "{text}");
3762 }
3763
3764 #[test]
3772 fn a_read_modify_write_is_an_exchange_or_a_locked_add_at_the_width_of_the_object() {
3773 let widths = [("char", "b", "%sil"), ("short", "w", "%si"), ("int", "l", "%esi")];
3774 for (ty, suffix, reg) in widths {
3775 let source =
3776 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_fetch_add(p, v, 5); }}\n");
3777 let text = asm(&source);
3778 assert!(text.contains("\tlock\n"), "{ty}: {text}");
3779 assert!(text.contains(&format!("xadd{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3780
3781 let source =
3782 format!("{ty} f({ty} *p, {ty} v) {{ return __atomic_exchange_n(p, v, 5); }}\n");
3783 let text = asm(&source);
3784 assert!(text.contains(&format!("xchg{suffix}\t{reg}, (%rdi)")), "{ty}: {text}");
3785 assert!(!text.contains("\tlock\n"), "an exchange is locked already: {ty}: {text}");
3786 }
3787 let source = "long f(long *p, long v) { return __atomic_fetch_add(p, v, 5); }\n";
3788 assert!(asm(source).contains("xaddq\t%rsi, (%rdi)"), "{}", asm(source));
3789
3790 let source = "int f(int *p, int v) { return __atomic_fetch_sub(p, v, 5); }\n";
3793 let text = asm(source);
3794 assert!(text.contains("negl\t"), "{text}");
3795 assert!(text.contains("xaddl\t"), "{text}");
3796
3797 for order in ["0", "2", "3", "4", "5"] {
3800 let source =
3801 format!("int f(int *p, int v) {{ return __atomic_fetch_add(p, v, {order}); }}\n");
3802 let text = asm(&source);
3803 assert!(text.contains("xaddl\t"), "{order}: {text}");
3804 assert!(!text.contains("mfence"), "{order} needs no barrier here: {text}");
3805 }
3806
3807 let text = asm("int f(int *p, int v) { return __sync_lock_test_and_set(p, v); }\n");
3811 assert!(text.contains("xchgl\t%esi, (%rdi)"), "{text}");
3812 let text = asm("void f(int *p) { __sync_lock_release(p); }\n");
3817 assert!(text.contains("movl\t$0, %eax"), "{text}");
3818 assert!(text.contains("movl\t%eax, (%rdi)"), "{text}");
3819 assert!(!text.contains("mfence"), "a release store needs no barrier here: {text}");
3820 }
3821
3822 #[test]
3834 fn the_lock_free_questions_are_answered_as_constants() {
3835 for size in ["1", "2", "4", "8"] {
3836 let source =
3837 format!("int f(void) {{ return __atomic_always_lock_free({size}, 0); }}\n");
3838 let text = asm(&source);
3839 assert!(text.contains("movb\t$1, %al"), "{size} bytes is lock free: {text}");
3840 assert!(!text.contains("call"), "and is not a call: {text}");
3841 }
3842 for size in ["3", "16", "sizeof(long double)"] {
3843 let source = format!("int f(void) {{ return __atomic_is_lock_free({size}, 0); }}\n");
3844 let text = asm(&source);
3845 assert!(text.contains("movb\t$0, %al"), "{size} bytes is not: {text}");
3846 assert!(!text.contains("call"), "and is not a call either: {text}");
3847 }
3848
3849 let text = asm("int f(int n) { return __atomic_is_lock_free(n, 0); }\n");
3853 assert!(text.contains("movb\t$0, %al"), "a size nobody knows is not lock free: {text}");
3854 let text = asm("int f(int *p) { return __atomic_always_lock_free(8, p); }\n");
3855 assert!(text.contains("movb\t$0, %al"), "eight bytes at four is not: {text}");
3856 let text = asm("int f(long *p) { return __atomic_always_lock_free(8, p); }\n");
3857 assert!(text.contains("movb\t$1, %al"), "and at eight it is: {text}");
3858 }
3859
3860 #[test]
3872 fn a_memory_order_an_operation_cannot_carry_is_read_as_the_strongest() {
3873 let mut opts = options();
3874 opts.emit = EmitKind::Ir;
3875
3876 let acquire_store = run(&opts, "void f(int *p, int v) { __atomic_store_n(p, v, 2); }\n");
3877 assert!(acquire_store.text().contains("seq_cst"), "{:?}", acquire_store.text());
3878 assert!(acquire_store.messages[0].contains("[W0333]"), "{:?}", acquire_store.messages);
3879
3880 let nonsense = run(&opts, "int f(int *p) { return __atomic_load_n(p, 99); }\n");
3881 assert!(nonsense.text().contains("seq_cst"), "{:?}", nonsense.text());
3882 assert!(nonsense.messages[0].contains("[W0333]"), "{:?}", nonsense.messages);
3883
3884 let computed = run(&opts, "int f(int *p, int n) { return __atomic_load_n(p, n); }\n");
3885 assert!(computed.text().contains("seq_cst"), "{:?}", computed.text());
3886 assert_eq!(computed.messages, Vec::<String>::new(), "a computed order is not a mistake");
3887 }
3888
3889 #[test]
3901 fn a_conversion_between_a_float_and_the_widest_unsigned_integer_is_written_without_a_branch() {
3902 let text = asm("double f(unsigned long long x) { return (double)x; }\n");
3903 assert!(text.contains("cvtsi2sdq"), "the signed conversion is what runs: {text}");
3904 assert!(text.contains("shrq"), "with the value halved first: {text}");
3905 assert!(text.contains("addsd"), "and doubled after: {text}");
3906 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
3907
3908 let text = asm("unsigned long long f(double d) { return (unsigned long long)d; }\n");
3909 assert!(text.contains("cvttsd2siq"), "the signed conversion is what runs: {text}");
3910 assert!(text.contains("subsd"), "with half the range taken off first: {text}");
3911 assert!(text.contains("shlq\t$63"), "and the top bit put back: {text}");
3912 assert!(!text.contains("\tj"), "and no branch anywhere: {text}");
3913 }
3914
3915 #[test]
3926 fn a_plain_name_the_program_took_is_the_programs_own_function() {
3927 let taken = concat!(
3928 "static long long llabs(long long b) { return 7; }\n",
3929 "long long f(long long x) { return llabs(x); }\n",
3930 );
3931 assert!(ir(taken).contains("call @llabs"), "a static definition is the program's own");
3932
3933 let retyped = concat!("int llabs(int b);\n", "int f(int x) { return llabs(x); }\n",);
3934 assert!(ir(retyped).contains("call @llabs"), "another type is another function");
3935
3936 let plain = concat!(
3937 "long long llabs(long long b);\n",
3938 "long long f(long long x) { return llabs(x); }\n",
3939 );
3940 let mut opts = options();
3941 opts.emit = EmitKind::Ir;
3942 assert!(!run(&opts, plain).text().contains("call @llabs"), "the library's by default");
3943
3944 opts.builtins = false;
3945 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin");
3946
3947 opts.builtins = true;
3948 opts.no_builtin = vec!["llabs".to_owned()];
3949 assert!(run(&opts, plain).text().contains("call @llabs"), "-fno-builtin-llabs");
3950 let one = "long labs(long b);\nlong f(long x) { return labs(x); }\n";
3951 assert!(!run(&opts, one).text().contains("call @labs"), "one name and not the family");
3952
3953 opts.no_builtin = Vec::new();
3956 opts.builtins = false;
3957 let prefixed = "long long f(long long x) { return __builtin_llabs(x); }\n";
3958 assert!(!run(&opts, prefixed).text().contains("call @llabs"), "the prefix is a promise");
3959 }
3960
3961 #[test]
3974 fn the_hint_builtins_are_their_first_argument_and_the_hint_leaves_no_trace() {
3975 let text = ir(concat!(
3976 "long a = __builtin_expect(7, 1);\n",
3977 "long b = __builtin_expect_with_probability(9, 1, 0.9);\n",
3978 "unsigned long c = sizeof(__builtin_expect((char)1, 1));\n",
3979 ));
3980 assert!(text.contains("global @a : i64 = 7,"), "{text}");
3981 assert!(text.contains("global @b : i64 = 9,"), "{text}");
3982 assert!(text.contains("global @c : i64 = 8,"), "{text}");
3983 assert!(!text.contains("__builtin_expect"), "it is not a call to anything:\n{text}");
3984
3985 let text = body("long f(char c) { return __builtin_expect(c, 1); }\n");
3988 assert!(text.contains("sext"), "{text}");
3989
3990 let one = "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 1\n %2 = sext.i64 %1\n return %0\n";
3994 assert_eq!(body("int f(void) { int i = 0; __builtin_expect(1, i++); return i; }\n"), one);
3995 let source = "int g(void) { int i = 0; __builtin_expect_with_probability(1, i++, 0.5); return i; }\n";
3996 assert_eq!(body(source), one);
3997
3998 let kept = body("int f(int n) { int i = 0; __builtin_expect(n, i++); return i; }\n");
4003 assert!(kept.contains("add.nsw"), "the hint still runs: {kept}");
4004 assert!(kept.ends_with("return %3\n"), "and the answer is what it left behind: {kept}");
4005 let both = "int g(int n) { int i = 0; __builtin_expect_with_probability(n, i++, 0.5); return i; }\n";
4006 assert!(body(both).contains("add.nsw"), "and so does the one with three arguments");
4007 }
4008
4009 #[test]
4021 fn a_promise_that_control_does_not_arrive_writes_no_instruction() {
4022 let promised = "int f(int x) { if (x) return 1; __builtin_unreachable(); }\n";
4023 let text = ir(promised);
4024 assert!(text.contains(" unreachable_hint\n"), "{text}");
4025 assert!(!text.contains("call"), "it is not a call to anything:\n{text}");
4026
4027 let after = body("int g(int x) { __builtin_unreachable(); return x; }\n");
4031 assert!(after.contains("return"), "{after}");
4032
4033 let text = asm(promised);
4036 let mine = text.split_once("\nf:\n").expect("a definition").1;
4037 let mine = mine.split_once("\t.size").expect("a definition").0;
4038 let plain = asm("int f(int x) { if (x) return 1; }\n");
4039 let plain = plain.split_once("\nf:\n").expect("a definition").1;
4040 let plain = plain.split_once("\t.size").expect("a definition").0;
4041 assert_eq!(mine, plain);
4042 let last = mine.lines().rfind(|line| !line.trim_start().starts_with('.'));
4045 assert_eq!(last.map(str::trim), Some("ret"), "{mine}");
4046 assert!(!mine.contains("ud2"), "{mine}");
4047 }
4048
4049 #[test]
4056 fn a_library_builtin_is_diagnosed_under_the_name_the_program_wrote() {
4057 let mut opts = options();
4058 opts.emit = EmitKind::Ir;
4059 let messages = run(&opts, "void f(void) { __builtin_abort(1); }\n").messages;
4060 assert!(
4061 messages.iter().any(|m| m.contains("__builtin_abort")),
4062 "expected the written name in {messages:?}"
4063 );
4064 }
4065
4066 #[test]
4074 fn a_builtin_nothing_lowers_is_refused_by_name() {
4075 let mut opts = options();
4076 opts.emit = EmitKind::Ir;
4077 for (builtin, call) in [
4078 ("__builtin_return_address", "(int)(long)__builtin_return_address(0)"),
4079 ("__builtin_alloca", "(int)(long)__builtin_alloca(8)"),
4080 ("__atomic_signal_fence", "(__atomic_signal_fence(5), 0)"),
4081 ] {
4082 let source = format!("int counter;\nint f(void) {{ return {call}; }}\n");
4083 let messages = run(&opts, &source).messages;
4084 let named = messages.iter().any(|m| m.contains(builtin) && m.contains("E0686"));
4085 assert!(named, "expected {builtin} to be refused by name in {messages:?}");
4086 }
4087 }
4088
4089 #[test]
4097 fn what_is_refused_is_the_call_and_not_the_name() {
4098 let text = ir("unsigned long n = sizeof(__builtin_return_address(0));\n");
4099 assert!(text.contains("global @n : i64 = 8,"), "{text}");
4100
4101 let text = ir(concat!(
4102 "void *__builtin_return_address(unsigned x) { return 0; }\n",
4103 "void *f(void) { return __builtin_return_address(0); }\n",
4104 ));
4105 assert!(text.contains("call @__builtin_return_address"), "{text}");
4106 }
4107
4108 #[test]
4113 fn a_static_function_nothing_refers_to_is_not_emitted() {
4114 let text = ir("static int dropped(void) { return 1; }\n\
4115 static int kept(void) { return 2; }\n\
4116 int main(void) { return kept(); }\n");
4117 assert!(text.contains("func @kept"), "{text}");
4118 assert!(!text.contains("dropped"), "{text}");
4119 }
4120
4121 #[test]
4127 fn two_static_functions_that_only_call_each_other_are_both_dropped() {
4128 let text = ir("static int ping(void);\n\
4129 static int pong(void) { return ping(); }\n\
4130 static int ping(void) { return pong(); }\n\
4131 int main(void) { return 0; }\n");
4132 assert!(!text.contains("ping"), "{text}");
4133 assert!(!text.contains("pong"), "{text}");
4134 }
4135
4136 #[test]
4142 fn naming_a_static_function_anywhere_keeps_it() {
4143 let text = ir("static int by_address(void) { return 1; }\n\
4144 static int in_an_image(void) { return 2; }\n\
4145 static int deeper(void) { return 3; }\n\
4146 static int reaches_deeper(void) { return deeper(); }\n\
4147 static int (*table[1])(void) = {in_an_image};\n\
4148 int main(void) {\n\
4149 int (*p)(void) = by_address;\n\
4150 return p() + table[0]() + reaches_deeper();\n\
4151 }\n");
4152 for kept in ["by_address", "in_an_image", "deeper", "reaches_deeper"] {
4153 assert!(text.contains(&format!("func @{kept}")), "expected {kept} in:\n{text}");
4154 }
4155 }
4156
4157 #[test]
4163 fn an_attribute_keeps_a_static_function_nothing_refers_to() {
4164 for attribute in ["used", "retain", "constructor", "destructor", "__used__"] {
4165 let source = format!(
4166 "__attribute__(({attribute})) static int kept(void) {{ return 1; }}\n\
4167 int main(void) {{ return 0; }}\n"
4168 );
4169 let text = ir(&source);
4170 assert!(text.contains("func @kept"), "for {attribute}:\n{text}");
4171 }
4172 }
4173
4174 #[test]
4177 fn a_function_anything_could_call_is_emitted_without_being_called() {
4178 let text =
4179 ir("int nobody_here_calls_it(void) { return 1; }\nint main(void) { return 0; }\n");
4180 assert!(text.contains("func @nobody_here_calls_it"), "{text}");
4181 }
4182
4183 #[test]
4190 fn a_classification_c_has_an_operator_for_is_that_operator() {
4191 for (builtin, operator) in [
4192 ("__builtin_isgreater", "binary >"),
4193 ("__builtin_isgreaterequal", "binary >="),
4194 ("__builtin_isless", "binary <"),
4195 ("__builtin_islessequal", "binary <="),
4196 ] {
4197 let source = format!("int f(double x, double y) {{ return {builtin}(x, y); }}\n");
4198 let text = tast(&source);
4199 assert!(text.contains(&format!("{operator} : int")), "for {builtin}:\n{text}");
4200 }
4201 }
4202
4203 #[test]
4212 fn the_classification_builtins_are_comparisons_and_not_calls() {
4213 let text = body("int f(double x, double y) { return __builtin_isunordered(x, y); }\n");
4214 assert_eq!(
4215 text,
4216 "block0(%0: f64, %1: f64):\n %2 = fcmp uno %0, %1\n %3 = zext.i32 \
4217 %2\n return %3\n"
4218 );
4219
4220 let text = body("int f(double x, double y) { return __builtin_islessgreater(x, y); }\n");
4222 assert!(text.contains("fcmp one %0, %1"), "{text}");
4223
4224 let text = body("int f(double x) { return __builtin_isnan(x); }\n");
4225 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4226
4227 let text = body("int f(double x) { return __builtin_isinf(x); }\n");
4228 assert!(text.contains("fconst.f64 0x7ff0000000000000"), "{text}");
4229 assert!(text.contains("fconst.f64 0xfff0000000000000"), "{text}");
4230 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4231 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4232 assert!(text.contains("%5 = or %3, %4"), "{text}");
4233
4234 let text = body("int f(double x) { return __builtin_isfinite(x); }\n");
4237 assert!(text.contains("%3 = fcmp olt %2, %0"), "{text}");
4238 assert!(text.contains("%4 = fcmp olt %0, %1"), "{text}");
4239 assert!(text.contains("%5 = and %3, %4"), "{text}");
4240
4241 let text = body("int f(double x) { return __builtin_signbit(x); }\n");
4242 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4243 assert!(text.contains("icmp slt %1, %2"), "{text}");
4244
4245 let text = body("int f(long double x) { return __builtin_signbitl(x); }\n");
4248 assert!(text.contains("%1 = bitcast.i80 %0"), "{text}");
4249
4250 let text = body("double g(void);\nint f(void) { return __builtin_isnan(g()); }\n");
4253 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4254 }
4255
4256 #[test]
4263 fn a_classification_spelling_that_names_a_width_converts_before_it_asks() {
4264 let text = ir(concat!(
4265 "int a = __builtin_isinff(1e300);\n",
4266 "int b = __builtin_isinf(1e300);\n",
4267 "int c = __builtin_isnan(0.0);\n",
4271 "int d = __builtin_signbit(-0.0);\n",
4272 "int e = __builtin_islessgreater(1.0, 2.0);\n",
4273 ));
4274 assert!(text.contains("global @a : i32 = 1,"), "{text}");
4275 assert!(text.contains("global @b : i32 = 0,"), "{text}");
4276 assert!(text.contains("global @c : i32 = 0,"), "{text}");
4277 assert!(text.contains("global @d : i32 = 1,"), "{text}");
4278 assert!(text.contains("global @e : i32 = 1,"), "{text}");
4279 }
4280
4281 #[test]
4283 fn a_classification_builtin_refuses_an_argument_that_is_not_floating_point() {
4284 let mut opts = options();
4285 opts.emit = EmitKind::Ir;
4286 let source = concat!(
4287 "int a(int x) { return __builtin_isnan(x); }\n",
4288 "int b(int x, int y) { return __builtin_isunordered(x, y); }\n",
4289 "int c(double x) { return __builtin_isnan(x, x); }\n",
4290 );
4291 let messages = run(&opts, source).messages;
4292 assert_eq!(
4293 messages,
4294 [
4295 "/main.c:1:23: error: non-floating-point argument in call to function \
4296 '__builtin_isnan' [E0685]",
4297 "/main.c:2:30: error: non-floating-point arguments in call to function \
4298 '__builtin_isunordered' [E0685]",
4299 "/main.c:3:26: error: too many arguments to function '__builtin_isnan' [E0511]",
4300 ]
4301 );
4302 }
4303
4304 #[test]
4313 fn the_last_three_classification_builtins_are_comparisons_and_not_calls() {
4314 let text = body("int f(double x) { return __builtin_isnormal(x); }\n");
4315 assert!(text.contains("%1 = bitcast.i64 %0"), "{text}");
4319 assert!(text.contains("%2 = iconst.i64 9223372036854775807"), "{text}");
4320 assert!(text.contains("%3 = and %1, %2"), "{text}");
4321 assert!(text.contains("%4 = iconst.i64 4503599627370496"), "{text}");
4322 assert!(text.contains("%5 = iconst.i64 9218868437227405312"), "{text}");
4323 assert!(text.contains("%6 = icmp uge %3, %4"), "{text}");
4324 assert!(text.contains("%7 = icmp ult %3, %5"), "{text}");
4325 assert!(text.contains("%8 = and %6, %7"), "{text}");
4326
4327 let text = body("int f(long double x) { return __builtin_isnormal(x); }\n");
4331 assert!(text.contains("%4 = iconst.i80 27670116110564327424"), "{text}");
4332 assert!(text.contains("%5 = iconst.i80 604453686435277732577280"), "{text}");
4333
4334 let text = body("int f(double x) { return __builtin_isinf_sign(x); }\n");
4335 assert!(text.contains("%3 = fcmp oeq %0, %1"), "{text}");
4336 assert!(text.contains("%4 = fcmp oeq %0, %2"), "{text}");
4337 assert!(text.contains("%7 = sub %5, %6"), "{text}");
4338
4339 let text = body("int f(double x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n");
4340 assert!(text.contains("fcmp uno %0, %0"), "{text}");
4341 assert!(text.contains("fcmp oeq %0, %6"), "{text}");
4342 assert_eq!(text.matches(" = zext.i32 ").count(), 4, "{text}");
4346 assert_eq!(text.matches(" = xor ").count(), 4, "{text}");
4347 assert!(!text.contains("call"), "{text}");
4348
4349 let text = body(concat!(
4352 "double g(void);\n",
4353 "int f(void) { return __builtin_fpclassify(0, 1, 2, 3, 4, g()); }\n",
4354 ));
4355 assert_eq!(text.matches("call @g()").count(), 1, "{text}");
4356 }
4357
4358 #[test]
4365 fn the_last_three_classification_builtins_fold_where_their_operand_is_a_constant() {
4366 let text = ir(concat!(
4367 "int a = __builtin_isnormal(1.0);\n",
4368 "int b = __builtin_isnormal(0.0);\n",
4369 "int c = __builtin_isnormal(1.0 / 0.0);\n",
4370 "int d = __builtin_isinf_sign(-1.0 / 0.0);\n",
4371 "int e = __builtin_isinf_sign(1.0);\n",
4372 "int g = __builtin_fpclassify(0, 1, 2, 3, 4, 0.0);\n",
4373 "int h = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0);\n",
4374 "int i = __builtin_fpclassify(0, 1, 2, 3, 4, 1.0 / 0.0);\n",
4375 ));
4376 assert!(text.contains("global @a : i32 = 1,"), "{text}");
4377 assert!(text.contains("global @b : i32 = 0,"), "{text}");
4378 assert!(text.contains("global @c : i32 = 0,"), "{text}");
4379 assert!(text.contains("global @d : i32 = -1,"), "{text}");
4380 assert!(text.contains("global @e : i32 = 0,"), "{text}");
4381 assert!(text.contains("global @g : i32 = 4,"), "{text}");
4382 assert!(text.contains("global @h : i32 = 2,"), "{text}");
4383 assert!(text.contains("global @i : i32 = 1,"), "{text}");
4384 }
4385
4386 #[test]
4392 fn fpclassify_refuses_an_answer_that_is_not_an_integer_constant() {
4393 let mut opts = options();
4394 opts.emit = EmitKind::Ir;
4395 let source = concat!(
4396 "int a(double x, int n) { return __builtin_fpclassify(0, 1, n, 3, 4, x); }\n",
4397 "int b(double x) { return __builtin_fpclassify(0, 1, 2, 3, x); }\n",
4398 "int c(int x) { return __builtin_fpclassify(0, 1, 2, 3, 4, x); }\n",
4399 );
4400 let messages = run(&opts, source).messages;
4401 assert_eq!(
4402 messages,
4403 [
4404 "/main.c:1:60: error: non-const integer argument 3 in call to function \
4405 '__builtin_fpclassify' [E0687]",
4406 "/main.c:2:26: error: too few arguments to function '__builtin_fpclassify' \
4407 [E0511]",
4408 "/main.c:3:23: error: non-floating-point argument in call to function \
4409 '__builtin_fpclassify' [E0685]",
4410 ]
4411 );
4412 }
4413
4414 #[test]
4422 fn a_builtin_whose_answer_is_a_constant_is_one_and_not_a_call() {
4423 let text = ir(concat!(
4424 "double a = __builtin_inf();\n",
4425 "float b = __builtin_huge_valf();\n",
4426 "long double c = __builtin_infl();\n",
4427 "double d = __builtin_huge_val();\n",
4428 ));
4429 assert!(text.contains("global @a : f64 = 0x7ff0000000000000,"), "{text}");
4430 assert!(text.contains("global @b : f32 = 0x7f800000,"), "{text}");
4431 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
4432 assert!(text.contains("global @d : f64 = 0x7ff0000000000000,"), "{text}");
4433 assert!(!text.contains("call"), "{text}");
4434 }
4435
4436 #[test]
4445 fn a_nan_is_written_with_the_payload_the_program_asked_for() {
4446 let text = ir(concat!(
4447 "double a = __builtin_nan(\"\");\n",
4448 "double b = __builtin_nan(\"0x1\");\n",
4449 "double c = __builtin_nan(\"010\");\n",
4451 "double d = __builtin_nans(\"\");\n",
4452 "double e = __builtin_nans(\"0x1\");\n",
4453 "float f = __builtin_nanf(\"0x1\");\n",
4454 "float g = __builtin_nansf(\"\");\n",
4455 "long double h = __builtin_nansl(\"\");\n",
4456 ));
4457 assert!(text.contains("global @a : f64 = 0x7ff8000000000000,"), "{text}");
4458 assert!(text.contains("global @b : f64 = 0x7ff8000000000001,"), "{text}");
4459 assert!(text.contains("global @c : f64 = 0x7ff8000000000008,"), "{text}");
4460 assert!(text.contains("global @d : f64 = 0x7ff4000000000000,"), "{text}");
4461 assert!(text.contains("global @e : f64 = 0x7ff0000000000001,"), "{text}");
4462 assert!(text.contains("global @f : f32 = 0x7fc00001,"), "{text}");
4463 assert!(text.contains("global @g : f32 = 0x7fa00000,"), "{text}");
4464 assert!(text.contains("f80 0x7fffa000000000000000"), "{text}");
4465
4466 let text = ir(concat!(
4469 "double f(const char *p) { return __builtin_nan(p); }\n",
4470 "double g(void) { return __builtin_nans(\"1x\"); }\n",
4471 ));
4472 assert_eq!(text.matches("call @nan(").count(), 1, "{text}");
4473 assert_eq!(text.matches("call @nans(").count(), 1, "{text}");
4474 }
4475
4476 #[test]
4484 fn the_length_and_the_order_of_a_string_literal_are_known_here() {
4485 let text = ir(concat!(
4486 "unsigned long a = __builtin_strlen(\"hello\");\n",
4487 "unsigned long b = __builtin_strlen(\"a\\0bc\");\n",
4488 "int c = __builtin_strcmp(\"X\", \"X\\376\") < 0;\n",
4489 "int d = __builtin_strcmp(\"abc\", \"abc\");\n",
4490 "int e = __builtin_strcmp(\"abc\", \"ab\") > 0;\n",
4491 ));
4492 assert!(text.contains("global @a : i64 = 5,"), "{text}");
4493 assert!(text.contains("global @b : i64 = 1,"), "{text}");
4494 assert!(text.contains("global @c : i32 = 1,"), "{text}");
4495 assert!(text.contains("global @d : i32 = 0,"), "{text}");
4496 assert!(text.contains("global @e : i32 = 1,"), "{text}");
4497 assert!(!text.contains("call"), "{text}");
4498
4499 let text = ir("unsigned long f(const char *p) { return __builtin_strlen(p); }\n");
4501 assert!(text.contains("call @strlen("), "{text}");
4502 }
4503
4504 #[test]
4511 fn a_sign_builtin_is_a_mask_over_the_bits_and_not_a_call() {
4512 let text = body("double f(double x) { return __builtin_fabs(x); }\n");
4513 assert!(text.contains("bitcast.i64 %0"), "{text}");
4514 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
4515 assert!(text.contains("and %1, %2"), "{text}");
4516 assert!(text.contains("bitcast.f64 %3"), "{text}");
4517 assert!(!text.contains("call"), "{text}");
4518
4519 let text = body("double f(double x, double y) { return __builtin_copysign(x, y); }\n");
4520 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
4521 assert!(text.contains("%8 = or %4, %7"), "{text}");
4522 assert!(!text.contains("call"), "{text}");
4523
4524 let text = body("long double f(long double x) { return __builtin_fabsl(x); }\n");
4527 assert!(text.contains("bitcast.i80 %0"), "{text}");
4528 assert!(text.contains("bitcast.f80"), "{text}");
4529
4530 let text = body("double f(float x) { return __builtin_fabs(x); }\n");
4533 assert!(text.contains("fpext.f64 %0"), "{text}");
4534 assert!(text.contains("bitcast.i64 %1"), "{text}");
4535 }
4536
4537 #[test]
4546 fn the_plain_math_names_are_the_same_mask_and_not_a_call() {
4547 let text =
4548 body(concat!("double fabs(double x);\n", "double f(double x) { return fabs(x); }\n",));
4549 assert!(text.contains("iconst.i64 9223372036854775807"), "{text}");
4550 assert!(!text.contains("call"), "{text}");
4551
4552 let text =
4553 body(concat!("float fabsf(float x);\n", "float f(float x) { return fabsf(x); }\n",));
4554 assert!(text.contains("bitcast.i32 %0"), "{text}");
4555 assert!(!text.contains("call"), "{text}");
4556
4557 let text = body(concat!(
4558 "double copysign(double x, double y);\n",
4559 "double f(double x, double y) { return copysign(x, y); }\n",
4560 ));
4561 assert!(text.contains("iconst.i64 -9223372036854775808"), "{text}");
4562 assert!(!text.contains("call"), "{text}");
4563
4564 let text = body(concat!(
4565 "float copysignf(float x, float y);\n",
4566 "float f(float x, float y) { return copysignf(x, y); }\n",
4567 ));
4568 assert!(!text.contains("call"), "{text}");
4569
4570 let text = ir(concat!(
4574 "long double fabsl(long double x);\n",
4575 "long double f(long double x) { return fabsl(x); }\n",
4576 ));
4577 assert!(text.contains("call @fabsl"), "{text}");
4578 }
4579
4580 #[test]
4588 fn a_plain_math_name_the_program_took_is_the_programs_own_function() {
4589 let taken = concat!(
4590 "static double fabs(double b) { return 7; }\n",
4591 "double f(double x) { return fabs(x); }\n",
4592 );
4593 assert!(ir(taken).contains("call @fabs"), "a static definition is the program's own");
4594
4595 let retyped = concat!("int fabs(int b);\n", "int f(int x) { return fabs(x); }\n");
4596 assert!(ir(retyped).contains("call @fabs"), "another type is another function");
4597
4598 let plain = concat!("double fabs(double b);\n", "double f(double x) { return fabs(x); }\n");
4599 let mut opts = options();
4600 opts.emit = EmitKind::Ir;
4601 assert!(!run(&opts, plain).text().contains("call @fabs"), "the library's by default");
4602
4603 opts.builtins = false;
4604 assert!(run(&opts, plain).text().contains("call @fabs"), "-fno-builtin");
4605
4606 opts.builtins = true;
4607 opts.no_builtin = vec!["fabs".to_owned()];
4608 assert!(run(&opts, plain).text().contains("call @fabs"), "-fno-builtin-fabs");
4609 let one = concat!(
4610 "double copysign(double a, double b);\n",
4611 "double f(double x) { return copysign(x, 1.0); }\n",
4612 );
4613 assert!(!run(&opts, one).text().contains("call @copysign"), "one name and not the family");
4614
4615 opts.no_builtin = Vec::new();
4617 opts.builtins = false;
4618 let prefixed = "double f(double x) { return __builtin_fabs(x); }\n";
4619 assert!(!run(&opts, prefixed).text().contains("call @fabs"), "the prefix is not a library");
4620 }
4621
4622 #[test]
4631 fn the_sign_builtins_answer_a_zero_and_a_nan_the_way_the_bits_say() {
4632 let text = ir(concat!(
4633 "double a = __builtin_fabs(-3.5);\n",
4634 "double b = __builtin_copysign(1.0, -0.0);\n",
4635 "double c = __builtin_copysign(0.0, -2.0);\n",
4636 "double d = __builtin_copysign(-__builtin_nan(\"\"), 1.0);\n",
4638 "double e = __builtin_fabs(-__builtin_nan(\"0x1\"));\n",
4639 "float g = __builtin_copysignf(-0.0f, 2.0f);\n",
4640 "long double h = __builtin_copysignl(1.0L, -1.0L);\n",
4641 "long double i = __builtin_fabsl(-__builtin_infl());\n",
4642 ));
4643 assert!(text.contains("global @a : f64 = 0x400c000000000000,"), "{text}");
4644 assert!(text.contains("global @b : f64 = 0xbff0000000000000,"), "{text}");
4645 assert!(text.contains("global @c : f64 = 0x8000000000000000,"), "{text}");
4646 assert!(text.contains("global @d : f64 = 0x7ff8000000000000,"), "{text}");
4647 assert!(text.contains("global @e : f64 = 0x7ff8000000000001,"), "{text}");
4648 assert!(text.contains("global @g : f32 = 0x0,"), "{text}");
4649 assert!(text.contains("f80 0xbfff8000000000000000"), "{text}");
4650 assert!(text.contains("f80 0x7fff8000000000000000"), "{text}");
4651 }
4652
4653 #[test]
4660 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
4661 let text = ir(concat!(
4662 "constexpr int side = 4;\n",
4663 "constexpr int wider = side + 1;\n",
4664 "constexpr double half = 1.5;\n",
4665 "struct point { int x; int y; };\n",
4666 "constexpr struct point origin = { 5, 6 };\n",
4667 "int square[side * side];\n",
4668 "int rectangle[wider];\n",
4669 "int rounded[(int)half * 2];\n",
4670 "int across[origin.y];\n",
4671 "enum named { four = side };\n",
4672 "int e = four;\n",
4673 ));
4674 assert!(text.contains("global @square : bytes 64 ="), "{text}");
4675 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
4676 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
4677 assert!(text.contains("global @across : bytes 24 ="), "{text}");
4678 assert!(text.contains("global @e : i32 = 4,"), "{text}");
4679
4680 let mut opts = options();
4683 opts.emit = EmitKind::Ir;
4684 let konst = "const int n = 1;\nint a[n];\n";
4685 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
4686 assert_eq!(run(&opts, konst).messages, [message]);
4687
4688 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
4690 assert_eq!(run(&opts, subscript).messages, [message]);
4691
4692 let address = "constexpr int c = 3;\nint *p = &c;\n";
4694 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
4695 pointer target type [E0514]";
4696 assert_eq!(run(&opts, address).messages, [warning]);
4697 }
4698
4699 #[test]
4708 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
4709 let mut opts = options();
4712 opts.std = Std::C17;
4713 let source = concat!(
4714 "int add(a, b)\n",
4715 "int a;\n",
4716 "int b;\n",
4717 "{ return a + b; }\n",
4718 "int promoted(c)\n",
4719 "char c;\n",
4720 "{ return c; }\n",
4721 "int narrow(char);\n",
4722 "int narrow(c)\n",
4723 "char c;\n",
4724 "{ return c; }\n",
4725 "int first(a)\n",
4726 "int a[4];\n",
4727 "{ return a[0]; }\n",
4728 );
4729 let result = run(&opts, source);
4730 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
4731 let text = result.text();
4732 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
4733 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
4734 assert!(text.contains("c : char object automatic defined"), "{text}");
4736 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
4737 assert!(text.contains("first : int(int *) function external defined"), "{text}");
4739 }
4740
4741 #[test]
4748 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
4749 let mut opts = options();
4750 opts.std = Std::C17;
4751 for (source, message) in [
4752 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
4753 (
4754 "int f(a)\nint a;\nint b;\n{ return a; }\n",
4755 "3:5: error: declaration for parameter 'b' but no such parameter",
4756 ),
4757 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
4758 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
4759 (
4760 "int f(a)\nstatic int a;\n{ return a; }\n",
4761 "2:12: error: storage class specified for parameter 'a'",
4762 ),
4763 (
4764 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
4765 "2:7: error: argument 'a' doesn't match prototype",
4766 ),
4767 ] {
4768 let result = run(&opts, source);
4769 assert!(result.failed(), "expected this to fail:\n{source}");
4770 assert!(result.messages[0].contains(message), "{:?}", result.messages);
4771 }
4772
4773 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
4776 let mut older = options();
4777 older.std = Std::C89;
4778 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
4779 let result = run(&opts, implicit);
4780 assert!(
4781 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
4782 "{:?}",
4783 result.messages
4784 );
4785
4786 let mut newer = options();
4790 newer.std = Std::C23;
4791 let plain = "int f(a)\nint a;\n{ return a; }\n";
4792 let result = run(&newer, plain);
4793 assert!(!result.failed(), "{:?}", result.messages);
4794 assert_eq!(
4795 result.messages,
4796 ["/main.c:1:5: warning: old-style function definition [E0412]"]
4797 );
4798 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
4799 }
4800
4801 #[test]
4808 fn the_obsolete_designators_are_taken_and_are_pedantic_warnings() {
4809 let array = "int a[8] = { [3] 7 };\n";
4810 let member = "struct s { int x; } v = { x: 7 };\n";
4811 for source in [array, member] {
4812 let result = run(&options(), source);
4813 assert!(!result.failed(), "{:?}", result.messages);
4814 assert!(result.messages.is_empty(), "nothing to say: {:?}", result.messages);
4815 }
4816
4817 let mut asked = options();
4818 asked.pedantic = true;
4819 assert_eq!(
4820 run(&asked, array).messages,
4821 ["/main.c:1:18: warning: obsolete designator, write `[i] =` instead [E0415]"]
4822 );
4823 assert_eq!(
4824 run(&asked, member).messages,
4825 ["/main.c:1:27: warning: obsolete designator, write `.field =` instead [E0413]"]
4826 );
4827 }
4828
4829 #[test]
4836 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
4837 let text = ir(concat!(
4838 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
4839 "struct brim { char buf[9223372036854775807L]; };\n",
4840 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
4841 "unsigned long h = sizeof(struct huge_struct);\n",
4842 "unsigned long b = sizeof(struct brim);\n",
4843 "unsigned long y = sizeof(struct bitty);\n",
4844 ));
4845 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
4846 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
4847 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
4848
4849 let mut opts = options();
4850 opts.emit = EmitKind::Ir;
4851 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
4852 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
4853 assert_eq!(run(&opts, over).messages, [message]);
4854 let array = "struct wide { short buf[1L << 62]; };\n";
4855 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
4856 maximum object size '9223372036854775807' [E0537]";
4857 assert_eq!(run(&opts, array).messages[0], message);
4858 }
4859
4860 fn compile_bytes(source: &[u8]) -> Compiled {
4865 let mut opts = options();
4866 opts.emit = EmitKind::Ir;
4867 let mut fs = MemoryFileSystem::new();
4868 fs.insert("/main.c", source.to_vec());
4869 compile(&opts, "/main.c", &fs)
4870 }
4871
4872 #[test]
4879 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
4880 let mut source = b"char s[] = \"a".to_vec();
4881 source.push(0xff);
4882 source.extend_from_slice(b"b\";\nchar c = '");
4883 source.push(0xff);
4884 source.extend_from_slice(b"';\n");
4885 let result = compile_bytes(&source);
4886 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
4887 assert!(result.text().contains(r#"bytes "a\ffb\00""#), "{}", result.text());
4888 assert!(result.text().contains("global @c : i8 = -1,"), "{}", result.text());
4890
4891 let mut stray = b"int a".to_vec();
4892 stray.push(0xff);
4893 stray.extend_from_slice(b" = 1;\n");
4894 let result = compile_bytes(&stray);
4895 assert!(
4896 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
4897 "{:?}",
4898 result.messages
4899 );
4900 }
4901
4902 #[test]
4903 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
4904 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
4905 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
4906 let expected = "\
4907func @add(i32, i32) -> i32, linkage(external) {
4908block0(%0: i32, %1: i32):
4909 %2 = add.nsw %0, %1
4910 return %2
4911}
4912";
4913 assert!(text.contains(expected), "{text}");
4914 }
4915
4916 #[test]
4917 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
4918 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
4919 assert!(!text.contains("alloca"), "{text}");
4920 assert!(!text.contains("load"), "{text}");
4921 assert!(!text.contains("store"), "{text}");
4922 }
4923
4924 #[test]
4925 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
4926 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
4927 let expected = "\
4928block0:
4929 %0 = alloca, size 4, align 4
4930 %1 = iconst.i32 1
4931 store %1 -> %0, align 4, tbaa !1
4932 %2 = call @g(%0) : (ptr) -> i32
4933 return %2
4934";
4935 assert_eq!(text, expected);
4936 }
4937
4938 #[test]
4939 fn a_loop_carries_what_it_changes_as_block_parameters() {
4940 let text = body(
4943 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
4944 return total;\n}\n",
4945 );
4946 assert!(!text.contains("alloca"), "{text}");
4947 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
4948 assert!(text.contains("jump block1("), "{text}");
4949 }
4950
4951 #[test]
4952 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
4953 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
4954 assert!(text.contains("icmp slt %0, %1"), "{text}");
4955 assert!(!text.contains("zext"), "{text}");
4956 }
4957
4958 #[test]
4959 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
4960 let text = body("int f(int a, int b) { return a && b; }\n");
4961 let expected = "\
4962block0(%0: i32, %1: i32):
4963 %2 = iconst.i32 0
4964 %3 = icmp ne %0, %2
4965 %4 = iconst.i1 0
4966 br_if %3, block1, block2(%4)
4967
4968block1:
4969 %5 = iconst.i32 0
4970 %6 = icmp ne %1, %5
4971 jump block2(%6)
4972
4973block2(%7: i1):
4974 %8 = zext.i32 %7
4975 return %8
4976";
4977 assert_eq!(text, expected);
4978 }
4979
4980 #[test]
4981 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
4982 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
4983 assert!(!text.contains("block3"), "{text}");
4986 assert!(!text.contains("iconst.i32 3"), "{text}");
4987 }
4988
4989 #[test]
4990 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
4991 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
4992 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
4993 assert!(body("int f(void) { }\n").contains("unreachable"));
4994 }
4995
4996 #[test]
4997 fn a_structure_is_copied_rather_than_held_in_a_value() {
4998 let text = body(
4999 "struct point { int x, y; };\n\
5000 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
5001 );
5002 assert!(text.contains("memcpy"), "{text}");
5003 }
5004
5005 #[test]
5006 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
5007 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
5008 assert!(text.contains("memset"), "{text}");
5009 }
5010
5011 #[test]
5012 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
5013 let text = body(
5014 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
5015 default: r = 4; } return r; }\n",
5016 );
5017 let expected = "\
5018block0(%0: i32):
5019 %1 = iconst.i32 0
5020 switch %0, block1, [1 => block2, 2 => block3(%1)]
5021
5022block1:
5023 %2 = iconst.i32 4
5024 jump block4(%2)
5025
5026block2:
5027 %3 = iconst.i32 1
5028 jump block3(%3)
5029
5030block3(%4: i32):
5031 %5 = iconst.i32 2
5032 %6 = add.nsw %4, %5
5033 jump block4(%6)
5034
5035block4(%7: i32):
5036 return %7
5037";
5038 assert_eq!(text, expected);
5039 }
5040
5041 #[test]
5042 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
5043 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
5046 assert!(text.contains("%2 = sub %0, %1"), "{text}");
5047 assert!(text.contains("icmp ule"), "{text}");
5048 assert!(!text.contains("switch"), "{text}");
5049 }
5050
5051 #[test]
5052 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
5053 let text = body(
5054 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
5055 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
5056 );
5057 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
5060 assert!(text.contains("block5:\n jump block7("), "{text}");
5061 assert!(text.contains("block6:\n jump block8("), "{text}");
5062 }
5063
5064 #[test]
5065 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
5066 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
5067 }
5068
5069 #[test]
5070 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
5071 let text = body(
5076 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
5077 return n; }\n",
5078 );
5079 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
5082 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
5083 assert!(text.contains("block4:\n jump block3("), "{text}");
5084 }
5085
5086 #[test]
5087 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
5088 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
5091 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
5092 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
5093 assert!(text.contains("br_if %6, block2, block3"), "{text}");
5094 }
5095
5096 #[test]
5097 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
5098 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
5099 assert!(!text.contains("alloca"), "{text}");
5103 assert!(text.contains("block2(%4: i32):\n return %4"), "{text}");
5104 assert_eq!(text.matches("jump block2(").count(), 2, "{text}");
5105 }
5106
5107 #[test]
5108 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
5109 let text =
5110 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
5111 assert!(!text.contains("alloca"), "{text}");
5112 assert!(text.contains("block1(%2: i32):"), "{text}");
5113 assert!(text.contains("jump block1(%5)"), "{text}");
5114 }
5115
5116 #[test]
5117 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
5118 assert_eq!(
5121 body("int f(int x) { return x; spare: return 0; }\n"),
5122 "block0(%0: i32):\n return %0\n"
5123 );
5124 }
5125
5126 #[test]
5127 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
5128 let text = body(
5129 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
5130 );
5131 assert_eq!(
5134 text,
5135 "\
5136block0(%0: ptr):
5137 %1 = load.i8 %0, align 1
5138 %2 = iconst.i8 3
5139 %3 = ashr %1, %2
5140 %4 = sext.i32 %3
5141 return %4
5142"
5143 );
5144 }
5145
5146 #[test]
5147 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
5148 let text =
5152 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
5153 assert_eq!(
5154 text,
5155 "\
5156block0(%0: ptr, %1: i32):
5157 %2 = iconst.i32 16777215
5158 %3 = and %1, %2
5159 %4 = trunc.i16 %3
5160 store %4 -> %0, align 2
5161 %5 = iconst.i32 16
5162 %6 = lshr %3, %5
5163 %7 = trunc.i8 %6
5164 %8 = iconst.i64 2
5165 %9 = ptr_add %0, %8
5166 store %7 -> %9, align 1
5167 return
5168"
5169 );
5170 }
5171
5172 #[test]
5173 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
5174 let text =
5175 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
5176 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
5179 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
5180 }
5181
5182 #[test]
5183 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
5184 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
5187 assert_eq!(text.matches("ashr").count(), 0, "{text}");
5188 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
5189 }
5190
5191 #[test]
5192 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
5193 let text = body(
5197 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
5198 );
5199 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
5200 }
5201
5202 #[test]
5203 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
5204 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
5207 assert!(
5208 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
5209 "{text}"
5210 );
5211 }
5212
5213 #[test]
5214 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
5215 let text = ir(concat!(
5220 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
5221 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
5222 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
5223 "char s[2] = \"hi\";\n",
5224 ));
5225 assert!(
5226 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
5227 "{text}"
5228 );
5229 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
5230 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
5231 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
5234 }
5235
5236 #[test]
5237 fn a_definition_takes_a_parameter_it_left_unnamed() {
5238 let text = ir("int f(int a, int) { return a; }\n");
5242 assert!(text.contains("func @f(i32, i32) -> i32"), "{text}");
5243 assert!(text.contains("block0(%0: i32, %1: i32):"), "{text}");
5244
5245 let text = ir("int g(int, int n) { return n; }\n");
5248 assert!(text.contains("block0(%0: i32, %1: i32):\n return %1\n"), "{text}");
5249 }
5250
5251 #[test]
5252 fn an_assignment_of_a_structure_is_the_object_it_wrote() {
5253 let text = body(concat!(
5258 "struct s { int f; int g; };\n",
5259 "void h(struct s *a, struct s *c, struct s *d, struct s *e)\n",
5260 "{ *d = *e = a[0] = *c; }\n",
5261 ));
5262 assert_eq!(text.matches("memcpy").count(), 3, "{text}");
5263 assert!(text.contains("memcpy %8, %1, size 8, align 4\n"), "{text}");
5264 assert!(text.contains("memcpy %3, %8, size 8, align 4\n"), "{text}");
5265 assert!(text.contains("memcpy %2, %3, size 8, align 4\n"), "{text}");
5266 }
5267
5268 #[test]
5269 fn a_string_literal_stops_at_the_end_of_the_array_it_is_filling() {
5270 let mut opts = options();
5275 opts.emit = EmitKind::Ir;
5276 let result = run(
5277 &opts,
5278 concat!(
5279 "const char a[2][3] = { \"1234\", \"xyz\" };\n",
5280 "static const char b[3][5] = { \"12345\", \"678\", \"9\" };\n",
5281 "union u { struct { char x[4]; char y[4]; }; struct { char z[8]; }; };\n",
5282 "const union u c = { { \"1234\", \"567\" } };\n",
5283 ),
5284 );
5285 let text = result.text();
5286 assert_eq!(
5287 result.messages,
5288 ["/main.c:1:24: warning: initializer-string for array of 'const char' is too long \
5289 (5 chars into 3 available) [E0637]"]
5290 );
5291 assert!(text.contains("global @a : bytes 6 = { bytes \"123\", bytes \"xyz\" }"), "{text}");
5292 assert!(
5293 text.contains(
5294 "global @b : bytes 15 = { bytes \"12345\", bytes \"678\\00\", zero 1, \
5295 bytes \"9\\00\", zero 3 }"
5296 ),
5297 "{text}"
5298 );
5299 assert!(
5302 text.contains("global @c : bytes 8 = { bytes \"1234\", bytes \"567\\00\" }"),
5303 "{text}"
5304 );
5305 }
5306
5307 #[test]
5308 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
5309 let text = body(concat!(
5313 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
5314 "void g(struct v *);\n",
5315 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
5316 ));
5317 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
5318 }
5319
5320 #[test]
5321 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
5322 let text = ir(concat!(
5327 "struct s { int x; };\n",
5328 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
5329 "int n = (int){ 7 };\n",
5330 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
5331 ));
5332 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
5333 assert!(text.contains("global @n : i32 = 7,"), "{text}");
5334 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
5337 }
5338
5339 #[test]
5340 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
5341 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
5345 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
5346 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
5347 }
5348
5349 #[test]
5350 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
5351 let text = ir("unsigned char foo[1][0];\n");
5355 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
5356 }
5357
5358 #[test]
5359 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
5360 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
5363 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
5364 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
5365 }
5366
5367 #[test]
5368 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
5369 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
5373 assert!(
5374 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
5375 "{text}"
5376 );
5377 }
5378
5379 #[test]
5380 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
5381 let text = body(
5386 "\
5387struct s { int a, b; };
5388struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
5389",
5390 );
5391 assert!(text.contains("block3(%7: ptr)"), "{text}");
5393 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
5394 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
5395 }
5396
5397 #[test]
5405 fn the_left_side_of_a_conditional_with_no_middle_is_evaluated_once() {
5406 let text = body("int f(int i) { return ++i ?: 10; }\n");
5407 assert!(text.contains("jump block3(%2)"), "the arm is the value that was tested: {text}");
5408 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
5409
5410 let text = body("long f(int i) { return ++i ?: 10L; }\n");
5413 assert!(text.contains("%5 = sext.i64 %2"), "the arm widens what was tested: {text}");
5414 assert_eq!(text.matches("add.nsw").count(), 1, "incremented once: {text}");
5415
5416 let text = body("int g(void);\nint f(void) { return g() ?: 10; }\n");
5418 assert_eq!(text.matches("call @g").count(), 1, "called once: {text}");
5419
5420 let text = body("int f(int i) { return ++i ? ++i : 10; }\n");
5423 assert_eq!(text.matches("add.nsw").count(), 2, "incremented twice: {text}");
5424 }
5425
5426 #[test]
5427 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
5428 let text = ir("\
5432struct pair { int a, b; };
5433struct pair make(int a, int b);
5434struct pair twice(struct pair p) { return make(p.a, p.b); }
5435");
5436 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
5437 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
5438 }
5439
5440 #[test]
5441 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
5442 let text = ir("\
5446struct big { double v[8]; };
5447struct big grow(struct big b);
5448struct big twice(struct big b) { return grow(grow(b)); }
5449");
5450 assert!(
5451 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
5452 "{text}"
5453 );
5454 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
5455 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
5458 }
5459
5460 #[test]
5461 fn a_structure_passed_to_a_variadic_function_says_so_at_the_call() {
5462 let text = ir("\
5467struct big { double v[8]; };
5468struct pair { int a, b; };
5469int p(const char *, ...);
5470int f(struct big b, struct pair q) { return p(\"\", 1, b, q); }
5471");
5472 assert!(
5473 text.contains("call @p(%4, %5, %2 byval(64, align 8), %6) : (ptr, ...) -> i32"),
5474 "{text}"
5475 );
5476 }
5477
5478 #[test]
5479 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
5480 let body = body(
5483 "\
5484struct pair { int a, b; };
5485struct pair make(int a, int b);
5486int second(void) { return make(1, 2).b; }
5487",
5488 );
5489 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
5490 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
5491 }
5492
5493 #[test]
5494 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
5495 let source = "\
5499struct hfa { float x, y, z; };
5500int take(struct hfa h);
5501int give(struct hfa h) { return take(h); }
5502";
5503 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
5504 let mut opts = options();
5505 opts.emit = EmitKind::Ir;
5506 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
5507 let result = run(&opts, source);
5508 assert_eq!(result.messages, Vec::<String>::new());
5509 assert!(result.text().contains("func @take(f32, f32, f32) -> i32"), "{}", result.text());
5510 }
5511
5512 #[test]
5513 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
5514 let source = "\
5517int use(int *);
5518void f(int n) {
5519 {
5520 int a[n];
5521 use(a);
5522 }
5523 use(0);
5524}
5525";
5526 let body = body(source);
5527 assert!(body.contains("mul.nsw"), "{body}");
5528 assert!(body.contains("stacksave"), "{body}");
5529 assert!(body.contains("alloca %"), "{body}");
5530 assert!(body.contains("stackrestore"), "{body}");
5531 }
5532
5533 #[test]
5534 fn a_goto_out_of_the_scope_of_one_gives_its_stack_back_on_the_way() {
5535 let source = "\
5540int use(int *);
5541int f(int n) {
5542 {
5543 int a[n];
5544 if (use(a)) goto out;
5545 use(0);
5546 }
5547out:
5548 return 0;
5549}
5550";
5551 let body = body(source);
5552 assert_eq!(body.matches("stackrestore").count(), 2, "{body}");
5554 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5555 assert!(after.starts_with(" %4\n jump block"), "{body}");
5556 }
5557
5558 #[test]
5559 fn a_goto_to_a_label_the_array_is_still_alive_at_leaves_the_stack_alone() {
5560 let source = "\
5564int use(int *);
5565int f(int n) {
5566 int a[n];
5567again:
5568 if (use(a)) goto again;
5569 return 0;
5570}
5571";
5572 let body = body(source);
5573 assert!(body.contains("stacksave"), "{body}");
5574 assert!(!body.contains("stackrestore"), "{body}");
5575 }
5576
5577 #[test]
5578 fn a_goto_back_to_a_label_in_front_of_one_gives_it_back_every_time_round() {
5579 let source = "\
5584int use(int *);
5585int f(int n) {
5586again:
5587 {
5588 int a[n];
5589 if (use(a)) goto again;
5590 }
5591 return 0;
5592}
5593";
5594 let body = body(source);
5595 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
5596 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5597 assert!(after.starts_with(" %4\n jump block1\n"), "{body}");
5598 }
5599
5600 #[test]
5601 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
5602 let source = "\
5608int f(void);
5609void t(void) {
5610 int count = 10;
5611 for (; count--;) {
5612 int b[f()];
5613 int i;
5614 for (i = 0; i < f(); i++) {
5615 b[i] = count;
5616 }
5617 }
5618}
5619";
5620 let body = body(source);
5621 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
5625 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
5626 let next = after.split("\n\n").next().expect("the block the restore is in");
5629 assert!(next.contains("jump block1("), "{body}");
5630 }
5631
5632 #[test]
5633 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
5634 let source = "\
5637unsigned long f(int n) {
5638 int a[n];
5639 n = 0;
5640 return sizeof a;
5641}
5642";
5643 let body = body(source);
5644 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
5646 }
5647
5648 #[test]
5649 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
5650 let source = "\
5653int use(int);
5654int f(int x) {
5655 return ({
5656 int t = use(x);
5657 t * t;
5658 });
5659}
5660";
5661 let expected = "\
5662block0(%0: i32):
5663 %1 = call @use(%0) : (i32) -> i32
5664 %2 = mul.nsw %1, %1
5665 return %2
5666";
5667 assert_eq!(body(source), expected);
5668 }
5669
5670 #[test]
5671 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
5672 let source = "int f(int x) { return ({ return x; 0; }); }\n";
5676 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
5677 }
5678
5679 #[test]
5680 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
5681 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
5685 let expected = "\
5686block0(%0: ptr):
5687 %1 = va_arg.f64 %0
5688 %2 = va_arg.f64 %0
5689 %3 = fadd %1, %2
5690 return %3
5691";
5692 assert_eq!(body(source), expected);
5693 }
5694
5695 #[test]
5696 fn one_that_reads_a_structure_answers_where_the_object_is() {
5697 let source = "\
5711struct s { int a; long b; };
5712long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.b; }
5713";
5714 let expected = "\
5715block0(%0: ptr):
5716 %1 = alloca, size 16, align 16
5717 %2 = va_object %0, size 16, align 8, in(int 8 at 0, int 8 at 8)
5718 memcpy %1, %2, size 16, align 8
5719 %3 = iconst.i64 8
5720 %4 = ptr_add %1, %3
5721 %5 = load.i64 %4, align 8, tbaa !1
5722 return %5
5723";
5724 assert_eq!(body(source), expected);
5725 }
5726
5727 #[test]
5731 fn the_classification_says_which_registers_the_object_arrived_in() {
5732 let source = "\
5733struct s { double a; double b; };
5734double f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a; }
5735";
5736 assert!(
5737 body(source)
5738 .contains("va_object %0, size 16, align 8, in(float f64 at 0, float f64 at 8)"),
5739 "{}",
5740 body(source)
5741 );
5742
5743 let big = "\
5744struct s { long a[4]; };
5745long f(__builtin_va_list ap) { struct s v = __builtin_va_arg(ap, struct s); return v.a[0]; }
5746";
5747 assert!(body(big).contains("va_object %0, size 32, align 8\n"), "{}", body(big));
5748 }
5749
5750 #[test]
5751 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
5752 let source = "\
5756int f(int c) {
5757 void *p = c ? &&one : &&two;
5758 goto *p;
5759one:
5760 return 1;
5761two:
5762 return 2;
5763}
5764";
5765 let expected = "\
5766block0(%0: i32):
5767 %1 = iconst.i32 0
5768 %2 = icmp ne %0, %1
5769 br_if %2, block1, block2
5770
5771block1:
5772 %3 = block_addr block3
5773 jump block4(%3)
5774
5775block2:
5776 %4 = block_addr block5
5777 jump block4(%4)
5778
5779block3:
5780 %5 = iconst.i32 1
5781 return %5
5782
5783block4(%6: ptr):
5784 indirect_br %6, block3, block5
5785
5786block5:
5787 %7 = iconst.i32 2
5788 return %7
5789";
5790 assert_eq!(body(source), expected);
5791 }
5792
5793 #[test]
5794 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
5795 let source = "void **next(void);
5798void f(void) { goto *next(); }
5799";
5800 let expected = "\
5801block0:
5802 %0 = call @next() : () -> ptr
5803 unreachable
5804";
5805 assert_eq!(body(source), expected);
5806 }
5807
5808 #[test]
5809 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
5810 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
5813 let expected = "\
5814block0:
5815 inline_asm.volatile \"mfence\", \"\", \"memory\"()
5816 return
5817";
5818 assert_eq!(body(source), expected);
5819 }
5820
5821 #[test]
5822 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
5823 let source = "\
5826int f(int x, int y) {
5827 int r;
5828 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
5829 return r + y;
5830}
5831";
5832 let expected = "\
5833block0(%0: i32, %1: i32):
5834 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
5835 %4 = add.nsw %2, %3
5836 return %4
5837";
5838 assert_eq!(body(source), expected);
5839 }
5840
5841 #[test]
5842 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
5843 let source = "\
5848struct pair { int a, b; };
5849int f(int x) {
5850 int slot = x;
5851 struct pair p = { x, x };
5852 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
5853 return slot + p.a;
5854}
5855";
5856 let text = body(source);
5857 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
5858 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
5859 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
5860 }
5861
5862 #[test]
5863 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
5864 let source = "\
5869int f(int x) {
5870 int r = 7;
5871 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
5872 return r;
5873away:
5874 return r;
5875}
5876";
5877 let expected = "\
5878block0(%0: i32):
5879 %1 = iconst.i32 7
5880 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
5881
5882block1:
5883 return %2
5884
5885block2:
5886 return %1
5887";
5888 assert_eq!(body(source), expected);
5889 }
5890
5891 #[test]
5892 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
5893 let mut opts = options();
5897 opts.emit = EmitKind::Ir;
5898 for (source, expected) in [
5899 (
5900 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
5901 "output operand constraint lacks '='",
5902 ),
5903 (
5904 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
5905 "lvalue required in 'asm' statement",
5906 ),
5907 (
5908 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
5909 "read-only variable 'g' used as 'asm' output",
5910 ),
5911 (
5912 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
5913 "input operand constraint contains '='",
5914 ),
5915 (
5916 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
5917 "memory input 0 is not directly addressable",
5918 ),
5919 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
5920 (
5921 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
5922 "duplicate asm operand name 'a'",
5923 ),
5924 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
5925 ] {
5926 let result = run(&opts, source);
5927 assert!(result.failed(), "expected this to be reported:\n{source}");
5928 assert!(
5929 result.messages.iter().any(|m| m.contains(expected)),
5930 "{expected}\n{:?}",
5931 result.messages
5932 );
5933 }
5934 }
5935
5936 #[test]
5941 fn an_asm_at_file_scope_that_is_directives_becomes_the_objects_it_defines() {
5942 let text = ir(concat!(
5943 "__asm__(\n",
5944 " \".section .rodata\\n\"\n",
5945 " \".globl first\\n\"\n",
5946 " \".balign 8\\n\"\n",
5947 " \"first:\\n\"\n",
5948 " \".long 1\\n\"\n",
5949 " \".long 2\\n\"\n",
5950 " \".globl last\\n\"\n",
5951 " \"last:\\n\"\n",
5952 " \".quad last - first\\n\");\n",
5953 "extern const int first[];\n",
5954 "extern const long last;\n",
5955 ));
5956 assert!(text.contains("global @first : bytes 8 = { i32 1, i32 2 }, align 8"), "{text}");
5957 assert!(text.contains("global @last : i64 = 8"), "{text}");
5958 }
5959
5960 #[test]
5964 fn a_name_an_asm_at_file_scope_defined_is_not_undone_by_a_declaration_of_it() {
5965 let text = ir(concat!(
5966 "__asm__(\".data\\n.globl counter\\ncounter:\\n.long 7\\n\");\n",
5967 "extern int counter;\n",
5968 "int read(void) { return counter; }\n",
5969 ));
5970 assert!(text.contains("global @counter : i32 = 7"), "{text}");
5971 }
5972
5973 #[test]
5976 fn an_incbin_at_file_scope_is_the_bytes_of_the_file_it_names() {
5977 let mut opts = options();
5978 opts.emit = EmitKind::Ir;
5979 let mut fs = MemoryFileSystem::new();
5980 fs.insert(
5981 "/main.c",
5982 b"__asm__(\".data\\n.globl blob\\nblob:\\n.incbin \\\"seed\\\"\\n\");\n".to_vec(),
5983 );
5984 fs.insert("seed", b"hi".to_vec());
5985 let result = compile(&opts, "/main.c", &fs);
5986 assert_eq!(result.messages, Vec::<String>::new());
5987 let text = result.text();
5988 assert!(text.contains("global @blob : bytes 2 = { bytes \"hi\" }"), "{text}");
5989 }
5990
5991 #[test]
5994 fn an_incbin_naming_a_file_that_is_not_there_says_which_file() {
5995 let messages = errors("__asm__(\".data\\nb:\\n.incbin \\\"nowhere\\\"\\n\");\n");
5996 assert!(
5997 messages
5998 .iter()
5999 .any(|m| m.contains("cannot open 'nowhere' for reading") && m.contains("E0702")),
6000 "{messages:?}"
6001 );
6002 }
6003
6004 #[test]
6007 fn an_instruction_in_an_asm_at_file_scope_is_refused_rather_than_ignored() {
6008 for source in [
6009 "__asm__(\".text\\n.globl f\\nf:\\n ret\\n\");\n",
6010 "__asm__(\".data\\n.set alias, 4\\n\");\n",
6011 ] {
6012 let messages = errors(source);
6013 assert!(
6014 messages
6015 .iter()
6016 .any(|m| m.contains("not supported yet")
6017 && m.contains("in an `asm` at file scope")),
6018 "{source}\n{messages:?}"
6019 );
6020 }
6021 }
6022
6023 #[test]
6024 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
6025 let mut opts = options();
6026 opts.emit = EmitKind::Ir;
6027 for source in [
6028 "int f(int n) { void *p = &&out; if (n) goto *p; { int a[n]; out: return 1; } }\n",
6029 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
6030 ] {
6031 let result = run(&opts, source);
6032 assert!(result.failed(), "expected this to be reported:\n{source}");
6033 assert!(
6034 result.messages.iter().any(|m| m.contains("not supported yet")),
6035 "{:?}",
6036 result.messages
6037 );
6038 }
6039 }
6040
6041 fn round_trip(source: &str) -> (String, String) {
6043 let printed = ir(source);
6044 let mut opts = options();
6045 opts.emit = EmitKind::Ir;
6046 let mut fs = MemoryFileSystem::new();
6047 fs.insert("/main.ir", printed.clone().into_bytes());
6048 let result = compile_ir(&opts, "/main.ir", &fs);
6049 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
6050 (printed, result.text().to_owned())
6051 }
6052
6053 #[test]
6054 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
6055 let (printed, again) = round_trip(
6059 "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",
6060 );
6061 assert_eq!(printed, again);
6062 }
6063
6064 #[test]
6065 fn ir_that_is_not_ir_says_which_line_stopped_it() {
6066 let mut opts = options();
6067 opts.emit = EmitKind::Ir;
6068 let mut fs = MemoryFileSystem::new();
6069 let text = "\
6070; ModuleID = 'a.c'
6071; format 0
6072target triple = \"x86_64-unknown-linux-gnu\"
6073target datalayout = \"e-p:64:64-i64:64-S128\"
6074
6075func @f(), linkage(external) {
6076block0:
6077 frobnicate
6078}
6079";
6080 fs.insert("/main.ir", text.as_bytes().to_vec());
6081 let result = compile_ir(&opts, "/main.ir", &fs);
6082 assert!(result.failed());
6083 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
6084 }
6085
6086 #[test]
6087 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
6088 let mut opts = options();
6091 opts.emit = EmitKind::Ir;
6092 let mut fs = MemoryFileSystem::new();
6093 let text = "\
6094; ModuleID = 'a.c'
6095; format 0
6096target triple = \"x86_64-unknown-linux-gnu\"
6097target datalayout = \"e-p:64:64-i64:64-S128\"
6098
6099func @f(), linkage(external) {
6100block0:
6101 %0 = iconst.i32 1
6102 return %0
6103}
6104";
6105 fs.insert("/main.ir", text.as_bytes().to_vec());
6106 let result = compile_ir(&opts, "/main.ir", &fs);
6107 assert!(result.failed());
6108 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
6109 }
6110
6111 #[test]
6112 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
6113 let mut fs = MemoryFileSystem::new();
6115 fs.insert("/main.ir", Vec::new());
6116 let result = compile_ir(&options(), "/main.ir", &fs);
6117 assert!(result.failed());
6118 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
6119 }
6120
6121 #[test]
6122 fn the_printed_ir_reads_back_as_the_same_module() {
6123 let text = ir("\
6126struct point { int x, y; };
6127static const char greeting[] = \"hi\";
6128int table[4] = { 1, 2, 3 };
6129int puts(const char *);
6130double half(double x) { return x / 2.0; }
6131int f(int n) {
6132 int total = 0;
6133 for (int i = 0; i < n; i++) {
6134 if (i == 3) continue;
6135 total += table[i];
6136 }
6137 switch (n) {
6138 case 0: total = 1;
6139 case 1: total++; break;
6140 default: total = -total;
6141 }
6142 struct point p = { total, 1 };
6143 int *q = &p.y;
6144 puts(greeting);
6145 return p.x + *q;
6146}
6147int dispatch(int c) {
6148 void *p = c ? &&one : &&two;
6149 goto *p;
6150one:
6151 return 1;
6152two:
6153 return 2;
6154}
6155int assembly(int x, int *p) {
6156 int r;
6157 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
6158 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
6159 return r;
6160away:
6161 return 0;
6162}
6163");
6164 let mut names = Interner::new();
6165 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
6166 assert_eq!(rucc_ir::print(&module, &names), text);
6167 }
6168
6169 #[test]
6170 fn what_save_temps_keeps_is_the_text_that_was_compiled_and_the_assembly_that_was_assembled() {
6171 let mut opts = options();
6175 opts.emit = EmitKind::Object;
6176 opts.save_temps = rucc_session::SaveTemps::Object;
6177 let result = run(&opts, "#define N 2\nint a[N];\n");
6178 assert_eq!(result.messages, Vec::<String>::new());
6179 let text = result.temps.preprocessed.expect("the preprocessed text");
6180 assert!(text.contains("int a[2];"), "{text}");
6181 assert!(text.starts_with("# 1 \"/main.c\""), "{text}");
6182 let asm = result.temps.assembly.expect("the assembly");
6183 assert!(asm.contains("a:"), "{asm}");
6184 assert!(matches!(result.artifact, Artifact::Object(_)), "{:?}", result.artifact);
6185 }
6186
6187 #[test]
6188 fn nothing_is_kept_unless_the_flag_asked_for_it() {
6189 let mut opts = options();
6192 opts.emit = EmitKind::Object;
6193 assert_eq!(run(&opts, "int a;\n").temps, Temps::default());
6194 }
6195
6196 #[test]
6197 fn a_compilation_that_stops_before_the_back_end_keeps_the_text_and_no_assembly() {
6198 let mut opts = options();
6201 opts.emit = EmitKind::Ir;
6202 opts.save_temps = rucc_session::SaveTemps::Cwd;
6203 let result = run(&opts, "int a;\n");
6204 assert!(result.temps.preprocessed.is_some());
6205 assert_eq!(result.temps.assembly, None);
6206 }
6207}