1use std::path::Path;
14
15use rucc_diag::{Diagnostic, Severity, Span};
16use rucc_lex::{Convert, Keywords, PpToken, convert};
17use rucc_sema::{Checker, Context as CheckContext};
18use rucc_session::{EmitKind, FileSystem, Options, Session};
19
20use crate::preprocess::render;
21
22#[derive(Debug, Clone, PartialEq, Eq)]
24pub struct Compiled {
25 pub text: String,
27 pub messages: Vec<String>,
29 pub errors: u32,
31}
32
33impl Compiled {
34 #[must_use]
36 pub fn failed(&self) -> bool {
37 self.errors > 0
38 }
39}
40
41#[must_use]
54pub fn compile(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
55 let mut sess = Session::new(opts.clone());
56 let keywords = Keywords::new(&mut sess.interner, opts.std, opts.gnu_extensions);
60 let mut diagnostics: Vec<Diagnostic> = Vec::new();
61
62 let bytes = match fs.read(Path::new(name)) {
63 Ok(bytes) => bytes,
64 Err(e) => return failure(format!("{name}: {e}")),
65 };
66 let Ok(file) = sess.sources.add_shared(name, bytes, None) else {
67 return failure(format!("{name}: the source map has no room left for this file"));
68 };
69
70 let mut pp = rucc_pp::Preprocessor::new();
74 let predef = rucc_pp::Predef::for_options(opts);
75 let expanded: Vec<PpToken> = {
76 let mut cx = rucc_pp::Context::new(&mut sess.interner, &mut sess.sources, fs, &opts.search);
77 cx.lex = rucc_lex::Options::for_dialect(opts.std, opts.gnu_extensions);
78 if pp.predefine(&sess.target, &predef, &mut cx).is_err() {
79 return failure(format!("{name}: the source map has no room for the built in macros"));
80 }
81 pp.run(file, &mut cx).iter().map(|token| token.to_pp()).collect()
82 };
83 diagnostics.extend(pp.take_diagnostics());
84
85 let cx = Convert {
88 keywords: &keywords,
89 interner: &sess.interner,
90 target: &sess.target,
91 std: opts.std,
92 gnu: opts.gnu_extensions,
93 pedantic: opts.pedantic,
94 };
95 let (tokens, complaints) = convert(&expanded, &cx);
96 diagnostics.extend(complaints);
97
98 let parsed = rucc_parse::parse(
99 &tokens,
100 rucc_parse::Context {
101 interner: &sess.interner,
102 std: opts.std,
103 gnu: opts.gnu_extensions,
104 pedantic: opts.pedantic,
105 error_limit: opts.error_limit as usize,
106 },
107 );
108 let parse_failed = parsed.diagnostics.iter().any(|d| d.severity.is_fatal());
109 diagnostics.extend(parsed.diagnostics);
110
111 let mut text = String::new();
112 if !parse_failed {
113 let mut checker = Checker::new(
114 &parsed.ast,
115 CheckContext {
116 names: &sess.interner,
117 target: &sess.target,
118 std: opts.std,
119 gnu: opts.gnu_extensions,
120 pedantic: opts.pedantic,
121 error_limit: opts.error_limit as usize,
122 },
123 );
124 checker.check_unit();
125 let checked = checker.finish();
126 if !checked.failed() {
127 match opts.emit {
128 EmitKind::Tast => {
129 text = rucc_sema::print(&checked.tast, &checked.types, &sess.interner);
130 }
131 EmitKind::Ir => {
132 let lowered = rucc_lower::lower(
133 name,
134 rucc_lower::Context {
135 tast: &checked.tast,
136 types: &checked.types,
137 target: &sess.target,
138 names: &mut sess.interner,
139 },
140 );
141 let failed = lowered.diagnostics.iter().any(|d| d.severity.is_fatal());
145 if !failed {
146 if let Err(errors) = rucc_ir::verify(&lowered.module, &sess.interner) {
151 for error in errors {
152 diagnostics.push(internal(&format!("invalid IR, {error}")));
153 }
154 } else {
155 text = rucc_ir::print(&lowered.module, &sess.interner);
156 }
157 }
158 diagnostics.extend(lowered.diagnostics);
159 }
160 _ => {}
161 }
162 }
163 diagnostics.extend(checked.diagnostics);
164 }
165
166 let mut messages = Vec::with_capacity(diagnostics.len());
167 let mut errors = 0;
168 for diag in &diagnostics {
169 if diag.severity.is_fatal()
170 || (diag.severity == Severity::Warning && opts.warnings_are_errors)
171 {
172 errors += 1;
173 }
174 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
175 }
176 if errors > 0 {
177 text.clear();
179 }
180 Compiled { text, messages, errors }
181}
182
183#[must_use]
193pub fn compile_ir(opts: &Options, name: &str, fs: &dyn FileSystem) -> Compiled {
194 let mut sess = Session::new(opts.clone());
195 if opts.emit != EmitKind::Ir {
196 return failure(format!(
197 "{name}: an input of IR can only be emitted as IR, and `--emit={}` asks for what \
198 the C in front of it became",
199 opts.emit.as_str()
200 ));
201 }
202 let bytes = match fs.read(Path::new(name)) {
203 Ok(bytes) => bytes,
204 Err(e) => return failure(format!("{name}: {e}")),
205 };
206 let Ok(text) = std::str::from_utf8(bytes.as_slice()) else {
207 return failure(format!("{name}: this is not text, so it is not IR"));
208 };
209
210 let module = match rucc_ir::parse(text, &mut sess.interner) {
211 Ok(module) => module,
212 Err(error) => {
213 return failure(format!("{name}:{}: {}", error.line, error.message));
214 }
215 };
216 let mut diagnostics: Vec<Diagnostic> = Vec::new();
217 if let Err(errors) = rucc_ir::verify(&module, &sess.interner) {
218 for error in errors {
219 diagnostics.push(invalid(&format!("invalid IR, {error}")));
220 }
221 }
222 let mut messages = Vec::with_capacity(diagnostics.len());
223 for diag in &diagnostics {
224 messages.push(render(diag, &sess.sources, opts.warnings_are_errors));
225 }
226 let errors = u32::try_from(messages.len()).unwrap_or(u32::MAX);
227 let text = if errors > 0 { String::new() } else { rucc_ir::print(&module, &sess.interner) };
228 Compiled { text, messages, errors }
229}
230
231fn invalid(message: &str) -> Diagnostic {
233 Diagnostic::error(message.to_owned(), Span::DUMMY).with_code("E0661")
234}
235
236fn internal(message: &str) -> Diagnostic {
238 Diagnostic::error(format!("internal error: {message}"), Span::DUMMY)
239 .with_code("E0652")
240 .note("this is a bug in rucc rather than in the program, please report it", Span::DUMMY)
241}
242
243fn failure(message: String) -> Compiled {
246 Compiled { text: String::new(), messages: vec![format!("rucc: error: {message}")], errors: 1 }
247}
248
249#[cfg(test)]
250mod tests {
251 use rucc_session::{MemoryFileSystem, Std};
252 use rucc_target::Triple;
253
254 use super::*;
255
256 fn options() -> Options {
257 let mut opts = Options::new("x86_64-unknown-linux-gnu".parse::<Triple>().unwrap());
258 opts.emit = EmitKind::Tast;
259 opts
260 }
261
262 fn run(opts: &Options, source: &str) -> Compiled {
263 let mut fs = MemoryFileSystem::new();
264 fs.insert("/main.c", source.to_owned().into_bytes());
265 compile(opts, "/main.c", &fs)
266 }
267
268 fn freestanding() -> Options {
272 let mut opts = options();
273 opts.hosted = false;
274 opts.search.push_system(rucc_session::runtime::DIR);
275 opts
276 }
277
278 fn shipped(source: &str) -> String {
280 let result = run(&freestanding(), source);
281 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
282 result.text
283 }
284
285 fn tast(source: &str) -> String {
287 let result = run(&options(), source);
288 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
289 result.text
290 }
291
292 #[test]
293 fn the_shipped_stdarg_declares_a_list_and_the_four_operators() {
294 let text = shipped(concat!(
295 "#include <stdarg.h>\n",
296 "int sum(int n, ...) {\n",
297 " va_list ap, copy;\n",
298 " va_start(ap, n);\n",
299 " va_copy(copy, ap);\n",
300 " int total = va_arg(ap, int) + va_arg(copy, int);\n",
301 " va_end(ap);\n",
302 " va_end(copy);\n",
303 " return total;\n",
304 "}\n",
305 ));
306 assert!(text.contains("va-start"), "{text}");
307 assert!(text.contains("va-copy"), "{text}");
308 assert!(text.contains("va-arg"), "{text}");
309 assert!(text.contains("va-end"), "{text}");
310 }
311
312 #[test]
316 fn stdarg_hands_out_the_type_alone_when_that_is_all_that_was_asked_for() {
317 let text = shipped(concat!(
318 "#define __need___va_list\n",
319 "#include <stdarg.h>\n",
320 "int vprint(const char *f, __gnuc_va_list ap);\n",
321 "#ifdef va_start\n",
322 "#error va_start should not be defined\n",
323 "#endif\n",
324 "#ifdef _VA_LIST_DEFINED\n",
325 "#error va_list should not have been made\n",
326 "#endif\n",
327 ));
328 assert!(text.contains("vprint"), "{text}");
329 }
330
331 #[test]
334 fn stddef_answers_one_piece_at_a_time_and_the_next_request_still_gets_through() {
335 let text = shipped(concat!(
336 "#define __need_size_t\n",
337 "#include <stddef.h>\n",
338 "#ifdef offsetof\n",
339 "#error offsetof should not be defined yet\n",
340 "#endif\n",
341 "#define __need_ptrdiff_t\n",
342 "#include <stddef.h>\n",
343 "#include <stddef.h>\n",
344 "size_t a;\n",
345 "ptrdiff_t b;\n",
346 "wchar_t c;\n",
347 "max_align_t d;\n",
348 "void *e = NULL;\n",
349 "struct P { int x; long y; };\n",
350 "size_t f = offsetof(struct P, y);\n",
351 ));
352 assert!(text.contains("decl #0 a : unsigned long"), "{text}");
353 assert!(text.contains("decl #1 b : long"), "{text}");
354 }
355
356 #[test]
357 fn the_shipped_limits_and_float_are_the_targets_own_answers() {
358 let text = shipped(concat!(
359 "#include <limits.h>\n",
360 "#include <float.h>\n",
361 "int bits = CHAR_BIT;\n",
362 "long big = LONG_MAX;\n",
363 "int low = INT_MIN;\n",
364 "int radix = FLT_RADIX;\n",
365 "int digits = DBL_MANT_DIG;\n",
366 ));
367 assert!(text.contains("const 8 : int"), "{text}");
368 assert!(text.contains("const 9223372036854775807 : long"), "{text}");
369 assert!(text.contains("const 2 : int"), "{text}");
370 assert!(text.contains("const 53 : int"), "{text}");
371 }
372
373 #[test]
377 fn the_shipped_stdint_writes_the_whole_set_when_there_is_no_library_to_defer_to() {
378 let text = shipped(concat!(
379 "#include <stdint.h>\n",
380 "int64_t a = INT64_C(1);\n",
381 "uint_least16_t b;\n",
382 "intptr_t c;\n",
383 "uintmax_t d = UINTMAX_MAX;\n",
384 "int wide = sizeof(int_fast64_t);\n",
385 ));
386 assert!(text.contains("decl #0 a : long"), "{text}");
387 assert!(text.contains("decl #1 b : unsigned short"), "{text}");
388 assert!(text.contains("decl #2 c : long"), "{text}");
389 }
390
391 #[test]
392 fn the_three_formality_headers_still_have_to_work() {
393 let text = shipped(concat!(
394 "#include <stdbool.h>\n",
395 "#include <stdalign.h>\n",
396 "#include <iso646.h>\n",
397 "#include <stdnoreturn.h>\n",
398 "int t = true and not false;\n",
399 "_Alignas(16) char buf[16];\n",
400 "int a = alignof(long);\n",
401 ));
402 assert!(text.contains("decl #0 t : int"), "{text}");
403 assert!(text.contains("const 8 : unsigned long"), "{text}");
404 }
405
406 #[test]
409 fn every_shipped_header_can_be_included_twice() {
410 let mut source = String::new();
411 for _ in 0..2 {
412 for name in rucc_session::runtime::names() {
413 source.push_str(&format!("#include <{name}>\n"));
414 }
415 }
416 source.push_str("int x;\n");
417 let text = shipped(&source);
418 assert!(text.starts_with("decl #0 x : int"), "{text}");
419 }
420
421 #[test]
422 fn a_file_that_is_not_there_says_so_and_produces_nothing() {
423 let fs = MemoryFileSystem::new();
424 let result = compile(&options(), "/nope.c", &fs);
425 assert!(result.failed());
426 assert!(result.messages[0].contains("/nope.c"), "{:?}", result.messages);
427 assert!(result.text.is_empty());
428 }
429
430 #[test]
431 fn an_object_comes_out_with_its_type_its_linkage_and_how_much_of_a_definition_it_is() {
432 let text = tast("int x = 1;\n");
433 let expected = "\
434decl #0 x : int object external static defined
435 init
436 +0
437 const 1 : int
438";
439 assert_eq!(text, expected);
440 }
441
442 #[test]
443 fn the_macros_are_expanded_before_anything_is_parsed() {
444 let text = tast("#define N 2\nint a[N];\n");
448 assert!(text.starts_with("decl #0 a : int[2] object external static tentative"), "{text}");
449 }
450
451 #[test]
457 fn a_pragma_is_not_a_declaration_and_the_parse_walks_past_the_ones_it_does_not_read() {
458 let text = tast(concat!(
459 "#pragma pack(4)\n",
460 "struct s { int a; };\n",
461 "#pragma pack()\n",
462 "int b;\n",
463 "_Pragma(\"GCC visibility push(default)\") int c;\n",
464 ));
465 assert!(text.contains("decl #0 b : int"), "{text}");
466 assert!(text.contains("decl #1 c : int"), "{text}");
467 }
468
469 #[test]
477 fn the_layout_attributes_move_the_members_and_the_record_the_way_gcc_lays_them_out() {
478 tast(concat!(
479 "struct A { char c; int i; } __attribute__((packed));\n",
480 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
481 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
482 "struct B { char c; int i; } __attribute__((aligned));\n",
485 "_Static_assert(sizeof(struct B) == 16 && _Alignof(struct B) == 16, \"B\");\n",
486 "struct C { char c; int i __attribute__((packed)); };\n",
487 "_Static_assert(sizeof(struct C) == 5 && _Alignof(struct C) == 1, \"C\");\n",
488 "_Static_assert(__builtin_offsetof(struct C, i) == 1, \"C.i\");\n",
489 "struct D { char c; int i; } __attribute__((packed, aligned(4)));\n",
490 "_Static_assert(sizeof(struct D) == 8 && _Alignof(struct D) == 4, \"D\");\n",
491 "_Static_assert(__builtin_offsetof(struct D, i) == 1, \"D.i\");\n",
492 "struct E { char c; _Alignas(8) int i; };\n",
493 "_Static_assert(sizeof(struct E) == 16 && _Alignof(struct E) == 8, \"E\");\n",
494 "_Static_assert(__builtin_offsetof(struct E, i) == 8, \"E.i\");\n",
495 "struct F { char c; int i __attribute__((aligned(8))); };\n",
496 "_Static_assert(sizeof(struct F) == 16 && _Alignof(struct F) == 8, \"F\");\n",
497 "struct G { char c; short s; } __attribute__((aligned(2)));\n",
500 "_Static_assert(sizeof(struct G) == 4 && _Alignof(struct G) == 2, \"G\");\n",
501 "struct H { char c; int i; } __attribute__((aligned(2)));\n",
502 "_Static_assert(sizeof(struct H) == 8 && _Alignof(struct H) == 4, \"H\");\n",
503 "struct I { [[gnu::packed]] char c; int i; };\n",
506 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
507 "struct J { char c; [[gnu::packed]] int i; };\n",
508 "_Static_assert(sizeof(struct J) == 5 && _Alignof(struct J) == 1, \"J\");\n",
509 "struct M { char c; int i : 5; int j : 20; } __attribute__((packed));\n",
510 "_Static_assert(sizeof(struct M) == 5 && _Alignof(struct M) == 1, \"M\");\n",
511 "struct N { char c; long long l; } __attribute__((aligned(32)));\n",
512 "_Static_assert(sizeof(struct N) == 32 && _Alignof(struct N) == 32, \"N\");\n",
513 "union L { char c; int i; } __attribute__((packed));\n",
514 "_Static_assert(sizeof(union L) == 4 && _Alignof(union L) == 1, \"L\");\n",
515 ));
516 }
517
518 #[test]
528 fn packing_is_what_decides_whether_a_bit_field_may_straddle_its_own_storage() {
529 assert_eq!(bit_field_byte("struct s { int x : 12; char y : 6; };"), 2);
531 assert_eq!(
532 bit_field_byte("struct s { int x : 12; char y : 6; } __attribute__((packed));"),
533 1
534 );
535 assert_eq!(
536 bit_field_byte("struct s { int x : 12; __attribute__((packed)) char y : 6; };"),
537 1
538 );
539 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { int x : 12; char y : 6; };"), 1);
540 assert_eq!(bit_field_byte("struct s { char x; int y : 30; };"), 4);
542 assert_eq!(bit_field_byte("struct s { char x; int y : 30; } __attribute__((packed));"), 1);
543 assert_eq!(bit_field_byte("#pragma pack(4)\nstruct s { char x; int y : 30; };"), 1);
545 assert_eq!(bit_field_byte("#pragma pack(2)\nstruct s { char x; int y : 30; };"), 1);
546 }
547
548 fn bit_field_byte(record: &str) -> u64 {
550 let source = format!("{record}\nint f(struct s *p) {{ return p->y; }}\n");
551 let body = body(&source);
552 let Some((before, _)) = body.split_once("ptr_add") else { return 0 };
553 let (_, constant) = before.rsplit_once("iconst.i64 ").expect("an offset constant");
554 constant.lines().next().expect("a line").trim().parse().expect("a byte offset")
555 }
556
557 #[test]
563 fn an_attribute_among_the_specifiers_is_kept_beside_the_ones_written_in_front() {
564 tast(concat!(
565 "struct a { char c; __attribute__((aligned(8))) int i; };\n",
566 "_Static_assert(sizeof(struct a) == 16 && _Alignof(struct a) == 8, \"a\");\n",
567 "_Static_assert(__builtin_offsetof(struct a, i) == 8, \"a.i\");\n",
568 "struct b { char c; __attribute__((packed)) int i; };\n",
569 "_Static_assert(sizeof(struct b) == 5 && _Alignof(struct b) == 1, \"b\");\n",
570 "_Static_assert(__builtin_offsetof(struct b, i) == 1, \"b.i\");\n",
571 "typedef struct { char c; int i; } __attribute__((packed)) c;\n",
572 "_Static_assert(sizeof(c) == 5 && _Alignof(c) == 1, \"c\");\n",
573 ));
574 }
575
576 #[test]
582 fn pragma_pack_caps_every_member_and_is_read_where_the_body_closes() {
583 tast(concat!(
584 "#pragma pack(1)\n",
585 "struct A { char c; int i; };\n",
586 "_Static_assert(sizeof(struct A) == 5 && _Alignof(struct A) == 1, \"A\");\n",
587 "_Static_assert(__builtin_offsetof(struct A, i) == 1, \"A.i\");\n",
588 "#pragma pack()\n",
589 "struct B { char c; int i; };\n",
590 "_Static_assert(sizeof(struct B) == 8 && _Alignof(struct B) == 4, \"B\");\n",
591 "#pragma pack(2)\n",
592 "struct C { char c; int i; double d; };\n",
593 "_Static_assert(sizeof(struct C) == 14 && _Alignof(struct C) == 2, \"C\");\n",
594 "_Static_assert(__builtin_offsetof(struct C, d) == 6, \"C.d\");\n",
595 "struct K { char c; int i __attribute__((aligned(8))); };\n",
597 "_Static_assert(sizeof(struct K) == 6 && _Alignof(struct K) == 2, \"K\");\n",
598 "_Static_assert(__builtin_offsetof(struct K, i) == 2, \"K.i\");\n",
599 "struct J { char c; int i; } __attribute__((aligned(8)));\n",
601 "_Static_assert(sizeof(struct J) == 8 && _Alignof(struct J) == 8, \"J\");\n",
602 "#pragma pack()\n",
603 "#pragma pack(push, 1)\n",
604 "struct D { char c; short s; };\n",
605 "_Static_assert(sizeof(struct D) == 3 && _Alignof(struct D) == 1, \"D\");\n",
606 "#pragma pack(pop)\n",
607 "struct E { char c; short s; };\n",
608 "_Static_assert(sizeof(struct E) == 4 && _Alignof(struct E) == 2, \"E\");\n",
609 "struct H { char c;\n",
611 "#pragma pack(1)\n",
612 " int i; };\n",
613 "_Static_assert(sizeof(struct H) == 5 && _Alignof(struct H) == 1, \"H\");\n",
614 "#pragma pack(1)\n",
615 "struct I { char c;\n",
616 "#pragma pack()\n",
617 " int i; };\n",
618 "_Static_assert(sizeof(struct I) == 8 && _Alignof(struct I) == 4, \"I\");\n",
619 "#pragma pack()\n",
620 "#pragma pack(push, 8)\n",
622 "#pragma pack(push, 1)\n",
623 "struct P { char c; int i; };\n",
624 "_Static_assert(sizeof(struct P) == 5 && _Alignof(struct P) == 1, \"P\");\n",
625 "#pragma pack(pop)\n",
626 "struct Q { char c; int i; };\n",
627 "_Static_assert(sizeof(struct Q) == 8 && _Alignof(struct Q) == 4, \"Q\");\n",
628 "#pragma pack(pop)\n",
629 "#pragma pack(16)\n",
631 "struct R { char c; int i; };\n",
632 "_Static_assert(sizeof(struct R) == 8 && _Alignof(struct R) == 4, \"R\");\n",
633 "#pragma pack()\n",
634 "#pragma pack(1)\n",
635 "struct S { char c; int i : 5; int j : 20; };\n",
636 "_Static_assert(sizeof(struct S) == 5 && _Alignof(struct S) == 1, \"S\");\n",
637 "union T { char c; int i; };\n",
638 "_Static_assert(sizeof(union T) == 4 && _Alignof(union T) == 1, \"T\");\n",
639 "#pragma pack()\n",
640 ));
641 }
642
643 #[test]
647 fn a_pack_line_that_is_not_one_is_reported_in_the_words_gcc_uses() {
648 let result = run(
649 &options(),
650 concat!(
651 "#pragma pack 4\n",
652 "#pragma pack(pop)\n",
653 "#pragma pack(3)\n",
654 "#pragma pack(1) junk\n",
655 "#pragma pack(push, 1\n",
656 "#pragma pack(x)\n",
657 "#pragma pack(0)\n",
660 "#pragma pack(push)\n",
661 "struct s { char c; int i; };\n",
662 "#pragma pack(pop)\n",
663 "#pragma pack(pop, foo)\n",
664 ),
665 );
666 let expected = [
667 "missing `(` after `#pragma pack` - ignored",
668 "`#pragma pack (pop)` encountered without matching `#pragma pack (push)`",
669 "alignment must be a small power of two, not 3",
670 "junk at end of `#pragma pack`",
671 "malformed `#pragma pack(push[, id][, <n>])` - ignored",
672 "unknown action `x` for `#pragma pack` - ignored",
673 "`#pragma pack(pop, foo)` encountered without matching `#pragma pack(push, foo)`",
674 ];
675 assert_eq!(result.messages.len(), expected.len(), "{:?}", result.messages);
676 for (message, want) in result.messages.iter().zip(expected) {
677 assert!(message.contains(want), "expected {want:?} in {message:?}");
678 }
679 }
680
681 #[test]
685 fn the_wide_integer_answers_to_all_three_of_its_names() {
686 let text = tast("__uint128_t a; __int128_t b; unsigned __int128 c;\n");
687 assert!(text.contains("decl #0 a : unsigned __int128"), "{text}");
688 assert!(text.contains("decl #1 b : __int128"), "{text}");
689 assert!(text.contains("decl #2 c : unsigned __int128"), "{text}");
690 }
691
692 #[test]
693 fn every_conversion_the_language_performs_is_a_node_in_the_output() {
694 let text = tast("long f(int a, long b) { return a + b; }\n");
698 assert!(text.contains("convert arithmetic"), "{text}");
699 }
700
701 #[test]
702 fn a_mistake_in_each_phase_reaches_the_caller_and_writes_no_tree() {
703 for source in [
704 "#error stop\n",
705 "int f(void) { return 1 + ; }\n",
706 "int f(void) { return undeclared; }\n",
707 ] {
708 let result = run(&options(), source);
709 assert!(result.failed(), "expected this to fail:\n{source}");
710 assert!(result.text.is_empty(), "a file that did not compile wrote a tree:\n{source}");
711 }
712 }
713
714 #[test]
715 fn one_undeclared_name_is_one_message_and_not_one_per_use() {
716 let result = run(&options(), "int f(void) { return nope + nope * nope; }\n");
720 assert_eq!(result.errors, 1, "{:?}", result.messages);
721 }
722
723 #[test]
724 fn a_declaration_the_parser_skipped_does_not_become_an_undeclared_name_as_well() {
725 let result = run(&options(), "int x = ;\nint f(void) { return x; }\n");
729 assert_eq!(result.errors, 1, "{:?}", result.messages);
730 }
731
732 #[test]
733 fn werror_turns_a_warning_into_an_error_in_the_count_and_in_the_word() {
734 let source = "int f(void) { char c = 300; return c; }\n";
735 let plain = run(&options(), source);
736 assert_eq!(plain.errors, 0, "{:?}", plain.messages);
737 assert_eq!(plain.messages.len(), 1, "expected a warning about the narrowed constant");
738 assert!(!plain.text.is_empty(), "a warning is not a reason to write nothing");
739
740 let mut opts = options();
741 opts.warnings_are_errors = true;
742 let strict = run(&opts, source);
743 assert!(strict.failed());
744 assert!(strict.text.is_empty(), "and under -Werror it is a reason to write nothing");
745 for message in &strict.messages {
746 assert!(!message.contains("warning:"), "{message}");
747 }
748 }
749
750 #[test]
751 fn the_dialect_reaches_the_keywords_and_the_checking() {
752 let source = "typeof(1) x;\n";
755 let mut opts = options();
756 opts.std = Std::C23;
757 opts.gnu_extensions = false;
758 assert!(!run(&opts, source).failed(), "{:?}", run(&opts, source).messages);
759
760 opts.std = Std::C17;
761 assert!(run(&opts, source).failed());
762 }
763
764 #[test]
765 fn asking_for_a_kind_that_is_not_written_yet_runs_the_front_end_and_writes_nothing() {
766 let mut opts = options();
767 opts.emit = EmitKind::MirFinal;
768 let result = run(&opts, "int x = 1;\n");
769 assert!(!result.failed(), "{:?}", result.messages);
770 assert!(result.text.is_empty());
771 assert!(run(&opts, "int f(void) { return undeclared; }\n").failed());
774 }
775
776 fn ir(source: &str) -> String {
778 let mut opts = options();
779 opts.emit = EmitKind::Ir;
780 let result = run(&opts, source);
781 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
782 result.text
783 }
784
785 fn body(source: &str) -> String {
787 let text = ir(source);
788 let (_, rest) = text.split_once("{\n").expect("a function definition");
789 let (body, _) = rest.rsplit_once("}\n").expect("a function definition");
790 body.to_owned()
791 }
792
793 #[test]
801 fn builtin_constant_p_is_folded_where_it_is_written_rather_than_called() {
802 let text = ir(concat!(
803 "int g;\n",
804 "int a = __builtin_constant_p(1);\n",
805 "int b = __builtin_constant_p(g);\n",
806 "int c = __builtin_constant_p(\"abc\");\n",
807 "int d = __builtin_constant_p(&g);\n",
808 "int e = __builtin_constant_p(1.5);\n",
809 "int h = __builtin_choose_expr(__builtin_constant_p(3), 11, 22);\n",
810 ));
811 assert!(text.contains("global @a : i32 = 1,"), "{text}");
812 assert!(text.contains("global @b : i32 = 0,"), "{text}");
813 assert!(text.contains("global @c : i32 = 1,"), "{text}");
814 assert!(text.contains("global @d : i32 = 0,"), "{text}");
815 assert!(text.contains("global @e : i32 = 1,"), "{text}");
816 assert!(text.contains("global @h : i32 = 11,"), "{text}");
817 assert!(!text.contains("__builtin_constant_p"), "it is not a call to anything:\n{text}");
818
819 let text = body("int f(void) { int i = 0; __builtin_constant_p(i++); return i; }\n");
823 assert_eq!(text, "block0:\n %0 = iconst.i32 0\n %1 = iconst.i32 0\n return %0\n");
824 }
825
826 #[test]
833 fn a_constexpr_object_is_a_constant_wherever_one_is_required() {
834 let text = ir(concat!(
835 "constexpr int side = 4;\n",
836 "constexpr int wider = side + 1;\n",
837 "constexpr double half = 1.5;\n",
838 "struct point { int x; int y; };\n",
839 "constexpr struct point origin = { 5, 6 };\n",
840 "int square[side * side];\n",
841 "int rectangle[wider];\n",
842 "int rounded[(int)half * 2];\n",
843 "int across[origin.y];\n",
844 "enum named { four = side };\n",
845 "int e = four;\n",
846 ));
847 assert!(text.contains("global @square : bytes 64 ="), "{text}");
848 assert!(text.contains("global @rectangle : bytes 20 ="), "{text}");
849 assert!(text.contains("global @rounded : bytes 8 ="), "{text}");
850 assert!(text.contains("global @across : bytes 24 ="), "{text}");
851 assert!(text.contains("global @e : i32 = 4,"), "{text}");
852
853 let mut opts = options();
856 opts.emit = EmitKind::Ir;
857 let konst = "const int n = 1;\nint a[n];\n";
858 let message = "/main.c:2:5: error: variably modified 'a' at file scope [E0538]";
859 assert_eq!(run(&opts, konst).messages, [message]);
860
861 let subscript = "constexpr int t[3] = { 1, 2, 3 };\nint a[t[1]];\n";
863 assert_eq!(run(&opts, subscript).messages, [message]);
864
865 let address = "constexpr int c = 3;\nint *p = &c;\n";
867 let warning = "/main.c:2:6: warning: initialization discards 'const' qualifier from \
868 pointer target type [E0514]";
869 assert_eq!(run(&opts, address).messages, [warning]);
870 }
871
872 #[test]
881 fn an_old_style_definition_takes_its_types_from_the_declarations_under_its_list() {
882 let mut opts = options();
885 opts.std = Std::C17;
886 let source = concat!(
887 "int add(a, b)\n",
888 "int a;\n",
889 "int b;\n",
890 "{ return a + b; }\n",
891 "int promoted(c)\n",
892 "char c;\n",
893 "{ return c; }\n",
894 "int narrow(char);\n",
895 "int narrow(c)\n",
896 "char c;\n",
897 "{ return c; }\n",
898 "int first(a)\n",
899 "int a[4];\n",
900 "{ return a[0]; }\n",
901 );
902 let result = run(&opts, source);
903 assert_eq!(result.messages, Vec::<String>::new(), "expected this to compile:\n{source}");
904 let text = result.text;
905 assert!(text.contains("add : int(int, int) function external defined"), "{text}");
906 assert!(text.contains("promoted : int(int) function external defined"), "{text}");
907 assert!(text.contains("c : char object automatic defined"), "{text}");
909 assert!(text.contains("narrow : int(char) function external defined"), "{text}");
910 assert!(text.contains("first : int(int *) function external defined"), "{text}");
912 }
913
914 #[test]
921 fn the_two_halves_of_an_old_style_parameter_list_have_to_agree() {
922 let mut opts = options();
923 opts.std = Std::C17;
924 for (source, message) in [
925 ("int f(a, a)\nint a;\n{ return a; }\n", "1:10: error: multiple parameters named 'a'"),
926 (
927 "int f(a)\nint a;\nint b;\n{ return a; }\n",
928 "3:5: error: declaration for parameter 'b' but no such parameter",
929 ),
930 ("int f(a)\nint a;\nint a;\n{ return a; }\n", "3:5: error: redefinition of parameter"),
931 ("int f(a)\nint a = 1;\n{ return a; }\n", "2:5: error: parameter 'a' is initialized"),
932 (
933 "int f(a)\nstatic int a;\n{ return a; }\n",
934 "2:12: error: storage class specified for parameter 'a'",
935 ),
936 (
937 "int f(char);\nint f(a)\nshort a;\n{ return a; }\n",
938 "2:7: error: argument 'a' doesn't match prototype",
939 ),
940 ] {
941 let result = run(&opts, source);
942 assert!(result.failed(), "expected this to fail:\n{source}");
943 assert!(result.messages[0].contains(message), "{:?}", result.messages);
944 }
945
946 let implicit = "int f(a, b)\nint a;\n{ return a + b; }\n";
949 let mut older = options();
950 older.std = Std::C89;
951 assert!(!run(&older, implicit).failed(), "{:?}", run(&older, implicit).messages);
952 let result = run(&opts, implicit);
953 assert!(
954 result.messages[0].contains("1:10: error: type of 'b' defaults to 'int'"),
955 "{:?}",
956 result.messages
957 );
958
959 let mut newer = options();
963 newer.std = Std::C23;
964 let plain = "int f(a)\nint a;\n{ return a; }\n";
965 let result = run(&newer, plain);
966 assert!(!result.failed(), "{:?}", result.messages);
967 assert_eq!(
968 result.messages,
969 ["/main.c:1:5: warning: old-style function definition [E0412]"]
970 );
971 assert!(run(&opts, plain).messages.is_empty(), "and nothing to say in the dialects before");
972 }
973
974 #[test]
981 fn a_type_is_refused_when_it_passes_the_largest_object_and_not_before() {
982 let text = ir(concat!(
983 "struct huge_struct { short buf[(1L << 62) - 256]; int a, b, c, d; };\n",
984 "struct brim { char buf[9223372036854775807L]; };\n",
985 "struct bitty { char buf[9223372036854775800L]; int x : 1; };\n",
986 "unsigned long h = sizeof(struct huge_struct);\n",
987 "unsigned long b = sizeof(struct brim);\n",
988 "unsigned long y = sizeof(struct bitty);\n",
989 ));
990 assert!(text.contains("global @h : i64 = 9223372036854775312,"), "{text}");
991 assert!(text.contains("global @b : i64 = 9223372036854775807,"), "{text}");
992 assert!(text.contains("global @y : i64 = 9223372036854775804,"), "{text}");
993
994 let mut opts = options();
995 opts.emit = EmitKind::Ir;
996 let over = "struct over { char buf[9223372036854775800L]; char x[8]; };\n";
997 let message = "/main.c:1:1: error: type 'struct over' is too large [E0560]";
998 assert_eq!(run(&opts, over).messages, [message]);
999 let array = "struct wide { short buf[1L << 62]; };\n";
1000 let message = "/main.c:1:25: error: size of array 'buf' exceeds \
1001 maximum object size '9223372036854775807' [E0537]";
1002 assert_eq!(run(&opts, array).messages[0], message);
1003 }
1004
1005 fn compile_bytes(source: &[u8]) -> Compiled {
1010 let mut opts = options();
1011 opts.emit = EmitKind::Ir;
1012 let mut fs = MemoryFileSystem::new();
1013 fs.insert("/main.c", source.to_vec());
1014 compile(&opts, "/main.c", &fs)
1015 }
1016
1017 #[test]
1024 fn a_byte_that_is_not_a_character_is_kept_in_a_literal_and_refused_outside_one() {
1025 let mut source = b"char s[] = \"a".to_vec();
1026 source.push(0xff);
1027 source.extend_from_slice(b"b\";\nchar c = '");
1028 source.push(0xff);
1029 source.extend_from_slice(b"';\n");
1030 let result = compile_bytes(&source);
1031 assert_eq!(result.messages, Vec::<String>::new(), "a raw byte in a literal is that byte");
1032 assert!(result.text.contains(r#"bytes "a\ffb\00""#), "{}", result.text);
1033 assert!(result.text.contains("global @c : i8 = -1,"), "{}", result.text);
1035
1036 let mut stray = b"int a".to_vec();
1037 stray.push(0xff);
1038 stray.extend_from_slice(b" = 1;\n");
1039 let result = compile_bytes(&stray);
1040 assert!(
1041 result.messages.iter().any(|m| m.contains("source is not valid UTF-8 here")),
1042 "{:?}",
1043 result.messages
1044 );
1045 }
1046
1047 #[test]
1048 fn an_object_becomes_a_global_with_an_image_and_a_function_becomes_a_func() {
1049 let text = ir("int x = 7;\nint add(int a, int b) { return a + b; }\n");
1050 assert!(text.contains("global @x : i32 = 7, align 4, linkage(external)\n"), "{text}");
1051 let expected = "\
1052func @add(i32, i32) -> i32, linkage(external) {
1053block0(%0: i32, %1: i32):
1054 %2 = add.nsw %0, %1
1055 return %2
1056}
1057";
1058 assert!(text.contains(expected), "{text}");
1059 }
1060
1061 #[test]
1062 fn a_local_nothing_takes_the_address_of_is_a_value_and_never_a_stack_slot() {
1063 let text = body("int f(int n) { int a = n + 1; int b = a * 2; return a + b; }\n");
1064 assert!(!text.contains("alloca"), "{text}");
1065 assert!(!text.contains("load"), "{text}");
1066 assert!(!text.contains("store"), "{text}");
1067 }
1068
1069 #[test]
1070 fn a_local_whose_address_is_taken_gets_a_slot_in_the_entry_block() {
1071 let text = body("int g(int *);\nint f(void) { int a = 1; return g(&a); }\n");
1072 let expected = "\
1073block0:
1074 %0 = alloca, size 4, align 4
1075 %1 = iconst.i32 1
1076 store %1 -> %0, align 4
1077 %2 = call @g(%0) : (ptr) -> i32
1078 return %2
1079";
1080 assert_eq!(text, expected);
1081 }
1082
1083 #[test]
1084 fn a_loop_carries_what_it_changes_as_block_parameters() {
1085 let text = body(
1088 "int f(int n) {\n int total = 0;\n for (int i = 0; i < n; i++) total += i;\n \
1089 return total;\n}\n",
1090 );
1091 assert!(!text.contains("alloca"), "{text}");
1092 assert!(text.contains("block1(%3: i32, %4: i32):"), "{text}");
1093 assert!(text.contains("jump block1("), "{text}");
1094 }
1095
1096 #[test]
1097 fn a_comparison_used_as_a_condition_is_not_widened_and_narrowed_again() {
1098 let text = body("int f(int a, int b) { if (a < b) return 1; return 0; }\n");
1099 assert!(text.contains("icmp slt %0, %1"), "{text}");
1100 assert!(!text.contains("zext"), "{text}");
1101 }
1102
1103 #[test]
1104 fn the_right_side_of_a_short_circuit_is_in_a_block_of_its_own() {
1105 let text = body("int f(int a, int b) { return a && b; }\n");
1106 let expected = "\
1107block0(%0: i32, %1: i32):
1108 %2 = iconst.i32 0
1109 %3 = icmp ne %0, %2
1110 %4 = iconst.i1 0
1111 br_if %3, block1, block2(%4)
1112
1113block1:
1114 %5 = iconst.i32 0
1115 %6 = icmp ne %1, %5
1116 jump block2(%6)
1117
1118block2(%7: i1):
1119 %8 = zext.i32 %7
1120 return %8
1121";
1122 assert_eq!(text, expected);
1123 }
1124
1125 #[test]
1126 fn code_after_a_return_is_not_built_and_does_not_leave_an_empty_block_behind() {
1127 let text = body("int f(int a) { if (a) return 1; else return 2; return 3; }\n");
1128 assert!(!text.contains("block3"), "{text}");
1131 assert!(!text.contains("iconst.i32 3"), "{text}");
1132 }
1133
1134 #[test]
1135 fn falling_off_the_end_returns_zero_from_main_and_nothing_from_a_void_function() {
1136 assert!(body("int main(void) { }\n").contains("iconst.i32 0\n return"));
1137 assert_eq!(body("void f(void) { }\n"), "block0:\n return\n");
1138 assert!(body("int f(void) { }\n").contains("unreachable"));
1139 }
1140
1141 #[test]
1142 fn a_structure_is_copied_rather_than_held_in_a_value() {
1143 let text = body(
1144 "struct point { int x, y; };\n\
1145 int f(void) { struct point p = { 1, 2 }; struct point q = p; return q.x; }\n",
1146 );
1147 assert!(text.contains("memcpy"), "{text}");
1148 }
1149
1150 #[test]
1151 fn an_initializer_that_leaves_part_of_an_object_unwritten_zeroes_it_first() {
1152 let text = body("int f(void) { int a[4] = { 1 }; return a[3]; }\n");
1153 assert!(text.contains("memset"), "{text}");
1154 }
1155
1156 #[test]
1157 fn a_switch_is_one_branch_and_a_case_that_falls_through_carries_what_it_wrote() {
1158 let text = body(
1159 "int f(int x) { int r = 0; switch (x) { case 1: r = 1; case 2: r += 2; break; \
1160 default: r = 4; } return r; }\n",
1161 );
1162 let expected = "\
1163block0(%0: i32):
1164 %1 = iconst.i32 0
1165 switch %0, block1, [1 => block2, 2 => block3(%1)]
1166
1167block1:
1168 %2 = iconst.i32 4
1169 jump block4(%2)
1170
1171block2:
1172 %3 = iconst.i32 1
1173 jump block3(%3)
1174
1175block3(%4: i32):
1176 %5 = iconst.i32 2
1177 %6 = add.nsw %4, %5
1178 jump block4(%6)
1179
1180block4(%7: i32):
1181 return %7
1182";
1183 assert_eq!(text, expected);
1184 }
1185
1186 #[test]
1187 fn a_case_range_is_tested_for_rather_than_put_in_the_table() {
1188 let text = body("int f(int x) { switch (x) { case 1 ... 9: return 1; } return 0; }\n");
1191 assert!(text.contains("%2 = sub %0, %1"), "{text}");
1192 assert!(text.contains("icmp ule"), "{text}");
1193 assert!(!text.contains("switch"), "{text}");
1194 }
1195
1196 #[test]
1197 fn break_leaves_the_switch_and_continue_leaves_the_loop_around_it() {
1198 let text = body(
1199 "int f(int n) { int t = 0; for (int i = 0; i < n; i++) { switch (i) { \
1200 case 0: continue; case 1: break; default: t += i; } t++; } return t; }\n",
1201 );
1202 assert!(text.contains("switch %3, block4, [0 => block5, 1 => block6]"), "{text}");
1205 assert!(text.contains("block5:\n jump block7("), "{text}");
1206 assert!(text.contains("block6:\n jump block8("), "{text}");
1207 }
1208
1209 #[test]
1210 fn a_switch_with_nothing_to_branch_on_still_runs_what_comes_after_it() {
1211 assert_eq!(body("void f(int x) { switch (x) { } }\n"), "block0(%0: i32):\n return\n");
1212 }
1213
1214 #[test]
1215 fn a_label_a_loop_is_only_entered_through_builds_the_loop_around_it() {
1216 let text = body(
1221 "int f(int x, int n) { switch (x) { case 1: break; while (n) { case 2: n--; } } \
1222 return n; }\n",
1223 );
1224 assert!(text.contains("switch %0, block1(%1), [1 => block2, 2 => block3(%1)]"), "{text}");
1227 assert!(text.contains("block3(%3: i32):\n %4 = iconst.i32 1"), "{text}");
1228 assert!(text.contains("block5:\n jump block3("), "{text}");
1229 }
1230
1231 #[test]
1232 fn a_goto_into_a_loop_body_enters_it_without_the_test() {
1233 let text = body("int f(int x, int n) { goto in; while (n) { in: n--; } return n; }\n");
1236 assert!(text.starts_with("block0(%0: i32, %1: i32):\n jump block1(%1)"), "{text}");
1237 assert!(text.contains("block1(%2: i32):\n %3 = iconst.i32 1"), "{text}");
1238 assert!(text.contains("br_if %7, block3, block4"), "{text}");
1239 }
1240
1241 #[test]
1242 fn a_goto_is_a_jump_to_the_block_the_label_starts() {
1243 let text = body("int f(int x) { int r = 0; if (x) goto out; r = 1; out: return r; }\n");
1244 assert!(!text.contains("alloca"), "{text}");
1246 assert!(text.contains("block3(%4: i32):\n return %4"), "{text}");
1247 assert_eq!(text.matches("jump block3(").count(), 2, "{text}");
1248 }
1249
1250 #[test]
1251 fn a_backward_goto_is_a_loop_and_carries_what_it_changes() {
1252 let text =
1253 body("int f(int n) { int i = 0; again: if (i < n) { i++; goto again; } return i; }\n");
1254 assert!(!text.contains("alloca"), "{text}");
1255 assert!(text.contains("block1(%2: i32):"), "{text}");
1256 assert!(text.contains("jump block1(%5)"), "{text}");
1257 }
1258
1259 #[test]
1260 fn a_label_nothing_reaches_is_taken_out_rather_than_left_for_the_verifier() {
1261 assert_eq!(
1264 body("int f(int x) { return x; spare: return 0; }\n"),
1265 "block0(%0: i32):\n return %0\n"
1266 );
1267 }
1268
1269 #[test]
1270 fn a_bit_field_is_read_by_loading_the_bytes_it_lies_in_and_shifting() {
1271 let text = body(
1272 "struct s { unsigned a : 3; signed b : 5; };\nint f(struct s *p) { return p->b; }\n",
1273 );
1274 assert_eq!(
1277 text,
1278 "\
1279block0(%0: ptr):
1280 %1 = load.i8 %0, align 1
1281 %2 = iconst.i8 3
1282 %3 = ashr %1, %2
1283 %4 = sext.i32 %3
1284 return %4
1285"
1286 );
1287 }
1288
1289 #[test]
1290 fn a_store_to_a_bit_field_does_not_write_a_byte_it_has_no_bit_in() {
1291 let text =
1295 body("struct s { int a : 24; char c; };\nvoid f(struct s *p, int v) { p->a = v; }\n");
1296 assert_eq!(
1297 text,
1298 "\
1299block0(%0: ptr, %1: i32):
1300 %2 = iconst.i32 16777215
1301 %3 = and %1, %2
1302 %4 = trunc.i16 %3
1303 store %4 -> %0, align 2
1304 %5 = iconst.i32 16
1305 %6 = lshr %3, %5
1306 %7 = trunc.i8 %6
1307 %8 = iconst.i64 2
1308 %9 = ptr_add %0, %8
1309 store %7 -> %9, align 1
1310 return
1311"
1312 );
1313 }
1314
1315 #[test]
1316 fn what_an_assignment_to_a_bit_field_is_worth_is_what_fits_in_it() {
1317 let text =
1318 body("struct s { unsigned b : 5; };\nunsigned f(struct s *p) { return p->b = 33; }\n");
1319 assert!(text.contains("%3 = iconst.i8 31\n %4 = and %2, %3"), "{text}");
1322 assert!(text.ends_with("%9 = zext.i32 %4\n return %9\n"), "{text}");
1323 }
1324
1325 #[test]
1326 fn an_assignment_a_statement_throws_away_builds_none_of_what_it_is_worth() {
1327 let text = body("struct s { signed b : 5; };\nvoid f(struct s *p) { p->b = 3; }\n");
1330 assert_eq!(text.matches("ashr").count(), 0, "{text}");
1331 assert!(text.ends_with("store %8 -> %0, align 1\n return\n"), "{text}");
1332 }
1333
1334 #[test]
1335 fn a_bit_field_in_an_initializer_goes_in_over_bytes_that_were_zeroed_first() {
1336 let text = body(
1340 "struct s { int a : 3; int b; };\nint f(void) { struct s v = { 1 }; return v.b; }\n",
1341 );
1342 assert!(text.contains("memset %0, %1, size 8, align 4"), "{text}");
1343 }
1344
1345 #[test]
1346 fn the_image_of_a_static_bit_field_is_the_bytes_the_fields_share() {
1347 let text = ir("struct s { unsigned a : 3; unsigned b : 5; } g = { 1, 2 };\n");
1350 assert!(
1351 text.contains("global @g : bytes 4 = { bytes \"\\11\", zero 3 }, align 4"),
1352 "{text}"
1353 );
1354 }
1355
1356 #[test]
1357 fn an_initialized_flexible_array_member_makes_the_object_larger_than_its_type() {
1358 let text = ir(concat!(
1363 "struct a { int i; int j[]; } x = { 1, { 2, 0, 2, 3 } };\n",
1364 "struct b { char c; char p[]; } y = { 'o', \"wx\" };\n",
1365 "struct c { char c; char p[]; } z = { '9', { 'e', 'b' } };\n",
1366 "char s[2] = \"hi\";\n",
1367 ));
1368 assert!(
1369 text.contains("global @x : bytes 20 = { i32 1, i32 2, i32 0, i32 2, i32 3 }"),
1370 "{text}"
1371 );
1372 assert!(text.contains("global @y : bytes 4 = { i8 111, bytes \"wx\\00\" }"), "{text}");
1373 assert!(text.contains("global @z : bytes 3 = { i8 57, i8 101, i8 98 }"), "{text}");
1374 assert!(text.contains("global @s : bytes 2 = { bytes \"hi\" }"), "{text}");
1377 }
1378
1379 #[test]
1380 fn a_cast_of_a_record_to_its_own_type_is_the_object_that_was_cast() {
1381 let text = body(concat!(
1385 "struct s { int a, b; };\nstruct v { struct s s; int t; };\n",
1386 "void g(struct v *);\n",
1387 "void f(struct s *p) { struct v w = { (struct s)*p, 5 }; g(&w); }\n",
1388 ));
1389 assert_eq!(text.matches("memcpy").count(), 1, "{text}");
1390 }
1391
1392 #[test]
1393 fn a_compound_literal_read_in_a_static_initializer_lays_its_bytes_into_the_image() {
1394 let text = ir(concat!(
1399 "struct s { int x; };\n",
1400 "struct t { struct s s; int o; } a = { (struct s){ 2 }, 3 };\n",
1401 "int n = (int){ 7 };\n",
1402 "struct u { struct s p; struct s q; } b = { (struct s){ 1 }, (struct s){ } };\n",
1403 ));
1404 assert!(text.contains("global @a : bytes 8 = { i32 2, i32 3 }"), "{text}");
1405 assert!(text.contains("global @n : i32 = 7,"), "{text}");
1406 assert!(text.contains("global @b : bytes 8 = { i32 1, zero 4 }"), "{text}");
1409 }
1410
1411 #[test]
1412 fn the_address_of_a_compound_literal_asks_for_the_object_it_points_at() {
1413 let text = ir("struct s { int x; };\nstruct s *q = &(struct s){ 9 };\n");
1417 assert!(text.contains("global @.Lanon.0 : i32 = 9, align 4, linkage(internal)"), "{text}");
1418 assert!(text.contains("global @q : bytes 8 = { addr.8 @.Lanon.0 }"), "{text}");
1419 }
1420
1421 #[test]
1422 fn an_object_of_no_size_at_all_has_an_image_with_nothing_in_it() {
1423 let text = ir("unsigned char foo[1][0];\n");
1427 assert!(text.contains("global @foo : bytes 0 = {}, align 1"), "{text}");
1428 }
1429
1430 #[test]
1431 fn a_null_pointer_in_an_image_is_the_bits_an_address_has_room_for() {
1432 let text = ir("void *p = 0;\nchar *q = (char *) 4096;\n");
1435 assert!(text.contains("global @p : i64 = 0, align 8"), "{text}");
1436 assert!(text.contains("global @q : i64 = 4096, align 8"), "{text}");
1437 }
1438
1439 #[test]
1440 fn an_object_another_module_defines_may_be_one_that_cannot_be_written_through() {
1441 let text = ir("extern const int limit;\nint f(void) { return limit; }\n");
1445 assert!(
1446 text.contains("global @limit : bytes 4, align 4, linkage(external), constant"),
1447 "{text}"
1448 );
1449 }
1450
1451 #[test]
1452 fn a_conditional_whose_value_is_an_object_answers_where_the_object_is() {
1453 let text = body(
1458 "\
1459struct s { int a, b; };
1460struct s pick(int c, struct s x, struct s y) { return c ? x : y; }
1461",
1462 );
1463 assert!(text.contains("block3(%7: ptr)"), "{text}");
1465 assert!(text.contains("jump block3(%3)") && text.contains("jump block3(%4)"), "{text}");
1466 assert!(!text.contains("memcpy"), "the arms are joined rather than copied: {text}");
1467 }
1468
1469 #[test]
1470 fn a_structure_that_fits_in_registers_travels_as_the_registers_it_fits_in() {
1471 let text = ir("\
1475struct pair { int a, b; };
1476struct pair make(int a, int b);
1477struct pair twice(struct pair p) { return make(p.a, p.b); }
1478");
1479 assert!(text.contains("func @make(i32, i32) -> i64"), "{text}");
1480 assert!(text.contains("func @twice(i64) -> i64"), "{text}");
1481 }
1482
1483 #[test]
1484 fn a_structure_too_large_for_the_registers_travels_as_where_its_bytes_are() {
1485 let text = ir("\
1489struct big { double v[8]; };
1490struct big grow(struct big b);
1491struct big twice(struct big b) { return grow(grow(b)); }
1492");
1493 assert!(
1494 text.contains("func @grow(ptr sret(64, align 8), ptr byval(64, align 8))"),
1495 "{text}"
1496 );
1497 assert!(text.contains("block0(%0: ptr, %1: ptr):"), "{text}");
1498 assert_eq!(text.matches("call @grow").count(), 2, "{text}");
1501 }
1502
1503 #[test]
1504 fn what_a_call_produced_is_somewhere_before_anything_is_read_out_of_it() {
1505 let body = body(
1508 "\
1509struct pair { int a, b; };
1510struct pair make(int a, int b);
1511int second(void) { return make(1, 2).b; }
1512",
1513 );
1514 assert!(body.starts_with("block0:\n %0 = alloca, size 8, align 4\n"), "{body}");
1515 assert!(body.contains("store %3 -> %0, align 4\n"), "{body}");
1516 }
1517
1518 #[test]
1519 fn a_structure_of_floats_travels_in_floating_point_registers_on_aarch64() {
1520 let source = "\
1524struct hfa { float x, y, z; };
1525int take(struct hfa h);
1526int give(struct hfa h) { return take(h); }
1527";
1528 assert!(ir(source).contains("func @take(f64, f32) -> i32"), "{}", ir(source));
1529 let mut opts = options();
1530 opts.emit = EmitKind::Ir;
1531 opts.target = "aarch64-unknown-linux-gnu".parse::<Triple>().unwrap();
1532 let result = run(&opts, source);
1533 assert_eq!(result.messages, Vec::<String>::new());
1534 assert!(result.text.contains("func @take(f32, f32, f32) -> i32"), "{}", result.text);
1535 }
1536
1537 #[test]
1538 fn an_array_whose_length_is_not_a_constant_is_a_slot_made_where_its_declaration_is() {
1539 let source = "\
1542int use(int *);
1543void f(int n) {
1544 {
1545 int a[n];
1546 use(a);
1547 }
1548 use(0);
1549}
1550";
1551 let body = body(source);
1552 assert!(body.contains("mul.nsw"), "{body}");
1553 assert!(body.contains("stacksave"), "{body}");
1554 assert!(body.contains("alloca %"), "{body}");
1555 assert!(body.contains("stackrestore"), "{body}");
1556 }
1557
1558 #[test]
1559 fn the_head_of_a_for_loop_is_a_scope_that_closes_where_the_loop_is_left() {
1560 let source = "\
1566int f(void);
1567void t(void) {
1568 int count = 10;
1569 for (; count--;) {
1570 int b[f()];
1571 int i;
1572 for (i = 0; i < f(); i++) {
1573 b[i] = count;
1574 }
1575 }
1576}
1577";
1578 let body = body(source);
1579 assert_eq!(body.matches("stacksave").count(), 1, "{body}");
1583 let (_, after) = body.split_once("stackrestore").expect("the stack is given back");
1584 let (next, _) = after.split_once("\n\n").expect("a block after the restore");
1585 assert!(next.contains("jump block1("), "{body}");
1586 }
1587
1588 #[test]
1589 fn how_long_one_of_those_is_was_decided_where_it_was_declared_and_not_where_it_is_asked() {
1590 let source = "\
1593unsigned long f(int n) {
1594 int a[n];
1595 n = 0;
1596 return sizeof a;
1597}
1598";
1599 let body = body(source);
1600 assert_eq!(body.matches("sext.i64 %0").count(), 2, "{body}");
1602 }
1603
1604 #[test]
1605 fn a_block_in_the_middle_of_an_expression_is_walked_where_the_expression_is() {
1606 let source = "\
1609int use(int);
1610int f(int x) {
1611 return ({
1612 int t = use(x);
1613 t * t;
1614 });
1615}
1616";
1617 let expected = "\
1618block0(%0: i32):
1619 %1 = call @use(%0) : (i32) -> i32
1620 %2 = mul.nsw %1, %1
1621 return %2
1622";
1623 assert_eq!(body(source), expected);
1624 }
1625
1626 #[test]
1627 fn one_of_those_that_control_never_leaves_is_lowered_and_what_follows_it_is_dropped() {
1628 let source = "int f(int x) { return ({ return x; 0; }); }\n";
1632 assert_eq!(body(source), "block0(%0: i32):\n return %0\n");
1633 }
1634
1635 #[test]
1636 fn one_argument_off_a_variable_argument_list_stays_an_intrinsic() {
1637 let source = "double f(__builtin_va_list ap) { return __builtin_va_arg(ap, double) + __builtin_va_arg(ap, double); }\n";
1641 let expected = "\
1642block0(%0: ptr):
1643 %1 = va_arg.f64 %0
1644 %2 = va_arg.f64 %0
1645 %3 = fadd %1, %2
1646 return %3
1647";
1648 assert_eq!(body(source), expected);
1649 }
1650
1651 #[test]
1652 fn a_jump_to_an_address_branches_to_every_label_the_function_takes_the_address_of() {
1653 let source = "\
1657int f(int c) {
1658 void *p = c ? &&one : &&two;
1659 goto *p;
1660one:
1661 return 1;
1662two:
1663 return 2;
1664}
1665";
1666 let expected = "\
1667block0(%0: i32):
1668 %1 = iconst.i32 0
1669 %2 = icmp ne %0, %1
1670 br_if %2, block1, block2
1671
1672block1:
1673 %3 = block_addr block3
1674 jump block4(%3)
1675
1676block2:
1677 %4 = block_addr block5
1678 jump block4(%4)
1679
1680block3:
1681 %5 = iconst.i32 1
1682 return %5
1683
1684block4(%6: ptr):
1685 indirect_br %6, block3, block5
1686
1687block5:
1688 %7 = iconst.i32 2
1689 return %7
1690";
1691 assert_eq!(body(source), expected);
1692 }
1693
1694 #[test]
1695 fn a_jump_to_an_address_no_label_in_the_function_has_arrives_nowhere() {
1696 let source = "void **next(void);
1699void f(void) { goto *next(); }
1700";
1701 let expected = "\
1702block0:
1703 %0 = call @next() : () -> ptr
1704 unreachable
1705";
1706 assert_eq!(body(source), expected);
1707 }
1708
1709 #[test]
1710 fn an_asm_with_no_operands_is_volatile_and_the_clobbers_are_the_whole_of_what_it_says() {
1711 let source = "void f(void) { __asm__(\"mfence\" ::: \"memory\"); }\n";
1714 let expected = "\
1715block0:
1716 inline_asm.volatile \"mfence\", \"\", \"memory\"()
1717 return
1718";
1719 assert_eq!(body(source), expected);
1720 }
1721
1722 #[test]
1723 fn the_constraints_are_one_list_in_the_order_the_template_counts_the_operands() {
1724 let source = "\
1727int f(int x, int y) {
1728 int r;
1729 __asm__(\"addl %2, %0\" : \"=r\"(r), \"+r\"(y) : \"r\"(x));
1730 return r + y;
1731}
1732";
1733 let expected = "\
1734block0(%0: i32, %1: i32):
1735 %2, %3 = inline_asm.(i32, i32) \"addl %2, %0\", \"=r,+r,r\", \"\"(%1, %0)
1736 %4 = add.nsw %2, %3
1737 return %4
1738";
1739 assert_eq!(body(source), expected);
1740 }
1741
1742 #[test]
1743 fn a_memory_operand_travels_as_the_address_of_an_object_that_is_given_a_slot() {
1744 let source = "\
1749struct pair { int a, b; };
1750int f(int x) {
1751 int slot = x;
1752 struct pair p = { x, x };
1753 __asm__(\"incl %0\" : \"+m\"(slot), \"=m\"(p));
1754 return slot + p.a;
1755}
1756";
1757 let text = body(source);
1758 assert!(text.contains("inline_asm \"incl %0\", \"+m,=m\", \"\"(%1, %2)\n"), "{text}");
1759 assert!(text.contains("%1 = alloca, size 4, align 4\n"), "{text}");
1760 assert!(text.contains("%2 = alloca, size 8, align 4\n"), "{text}");
1761 }
1762
1763 #[test]
1764 fn an_asm_goto_falls_through_to_its_first_target_and_writes_its_outputs_there() {
1765 let source = "\
1770int f(int x) {
1771 int r = 7;
1772 __asm__ goto(\"cbnz %0, %l1\" : \"=r\"(r) : \"r\"(x) :: away);
1773 return r;
1774away:
1775 return r;
1776}
1777";
1778 let expected = "\
1779block0(%0: i32):
1780 %1 = iconst.i32 7
1781 %2 = inline_asm.volatile \"cbnz %0, %l1\", \"=r,r\", \"\"(%0), labels [block1, block2]
1782
1783block1:
1784 return %2
1785
1786block2:
1787 return %1
1788";
1789 assert_eq!(body(source), expected);
1790 }
1791
1792 #[test]
1793 fn an_asm_statement_that_is_not_well_formed_is_reported_in_the_words_gcc_uses() {
1794 let mut opts = options();
1798 opts.emit = EmitKind::Ir;
1799 for (source, expected) in [
1800 (
1801 "void f(int x) { __asm__(\"\" : \"r\"(x)); }\n",
1802 "output operand constraint lacks '='",
1803 ),
1804 (
1805 "void f(int x) { __asm__(\"\" : \"=r\"(x + 1)); }\n",
1806 "lvalue required in 'asm' statement",
1807 ),
1808 (
1809 "const int g = 1;\nvoid f(void) { __asm__(\"\" : \"=r\"(g)); }\n",
1810 "read-only variable 'g' used as 'asm' output",
1811 ),
1812 (
1813 "void f(int x) { __asm__(\"\" : : \"=r\"(x)); }\n",
1814 "input operand constraint contains '='",
1815 ),
1816 (
1817 "void f(void) { __asm__(\"\" : : \"m\"(1)); }\n",
1818 "memory input 0 is not directly addressable",
1819 ),
1820 ("void f(void) { __asm__(L\"\"); }\n", "wide string literal in 'asm'"),
1821 (
1822 "void f(int x, int y) { __asm__(\"\" : [a] \"=r\"(x) : [a] \"r\"(y)); }\n",
1823 "duplicate asm operand name 'a'",
1824 ),
1825 ("void f(int x) { __asm__(\"%[in]\" : \"=r\"(x)); }\n", "undefined named operand 'in'"),
1826 ] {
1827 let result = run(&opts, source);
1828 assert!(result.failed(), "expected this to be reported:\n{source}");
1829 assert!(
1830 result.messages.iter().any(|m| m.contains(expected)),
1831 "{expected}\n{:?}",
1832 result.messages
1833 );
1834 }
1835 }
1836
1837 #[test]
1838 fn what_the_walk_cannot_build_yet_is_reported_rather_than_mislowered() {
1839 let mut opts = options();
1840 opts.emit = EmitKind::Ir;
1841 for source in [
1842 "int f(int n) { int a[n]; goto out; out: return a[0]; }\n",
1843 "int f(int n) { int a[n]; void *p = &&out; goto *p; out: return a[0]; }\n",
1844 "struct s { double a[8]; };\nint p(const char *, ...);\nint g(struct s v) { return p(\"\", v); }\n",
1845 "struct s { int a; };\nstruct s f(__builtin_va_list ap) { return __builtin_va_arg(ap, struct s); }\n",
1846 "int f(int n) { int a[n]; __asm__ goto(\"\" ::::out); out: return a[0]; }\n",
1847 ] {
1848 let result = run(&opts, source);
1849 assert!(result.failed(), "expected this to be reported:\n{source}");
1850 assert!(
1851 result.messages.iter().any(|m| m.contains("not supported yet")),
1852 "{:?}",
1853 result.messages
1854 );
1855 }
1856 }
1857
1858 fn round_trip(source: &str) -> (String, String) {
1860 let printed = ir(source);
1861 let mut opts = options();
1862 opts.emit = EmitKind::Ir;
1863 let mut fs = MemoryFileSystem::new();
1864 fs.insert("/main.ir", printed.clone().into_bytes());
1865 let result = compile_ir(&opts, "/main.ir", &fs);
1866 assert_eq!(result.messages, Vec::<String>::new(), "expected this to read back:\n{printed}");
1867 (printed, result.text)
1868 }
1869
1870 #[test]
1871 fn ir_that_arrives_as_an_input_is_read_back_and_written_out_the_same() {
1872 let (printed, again) = round_trip(
1876 "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",
1877 );
1878 assert_eq!(printed, again);
1879 }
1880
1881 #[test]
1882 fn ir_that_is_not_ir_says_which_line_stopped_it() {
1883 let mut opts = options();
1884 opts.emit = EmitKind::Ir;
1885 let mut fs = MemoryFileSystem::new();
1886 let text = "\
1887; ModuleID = 'a.c'
1888; format 0
1889target triple = \"x86_64-unknown-linux-gnu\"
1890target datalayout = \"e-p:64:64-i64:64-S128\"
1891
1892func @f(), linkage(external) {
1893block0:
1894 frobnicate
1895}
1896";
1897 fs.insert("/main.ir", text.as_bytes().to_vec());
1898 let result = compile_ir(&opts, "/main.ir", &fs);
1899 assert!(result.failed());
1900 assert!(result.messages[0].contains("/main.ir:8"), "{:?}", result.messages);
1901 }
1902
1903 #[test]
1904 fn ir_that_reads_but_does_not_hold_together_is_reported_by_the_verifier() {
1905 let mut opts = options();
1908 opts.emit = EmitKind::Ir;
1909 let mut fs = MemoryFileSystem::new();
1910 let text = "\
1911; ModuleID = 'a.c'
1912; format 0
1913target triple = \"x86_64-unknown-linux-gnu\"
1914target datalayout = \"e-p:64:64-i64:64-S128\"
1915
1916func @f(), linkage(external) {
1917block0:
1918 %0 = iconst.i32 1
1919 return %0
1920}
1921";
1922 fs.insert("/main.ir", text.as_bytes().to_vec());
1923 let result = compile_ir(&opts, "/main.ir", &fs);
1924 assert!(result.failed());
1925 assert!(result.messages[0].contains("invalid IR"), "{:?}", result.messages);
1926 }
1927
1928 #[test]
1929 fn a_typed_tree_is_not_something_an_input_of_ir_can_produce() {
1930 let mut fs = MemoryFileSystem::new();
1932 fs.insert("/main.ir", Vec::new());
1933 let result = compile_ir(&options(), "/main.ir", &fs);
1934 assert!(result.failed());
1935 assert!(result.messages[0].contains("can only be emitted as IR"), "{:?}", result.messages);
1936 }
1937
1938 #[test]
1939 fn the_printed_ir_reads_back_as_the_same_module() {
1940 let text = ir("\
1943struct point { int x, y; };
1944static const char greeting[] = \"hi\";
1945int table[4] = { 1, 2, 3 };
1946int puts(const char *);
1947double half(double x) { return x / 2.0; }
1948int f(int n) {
1949 int total = 0;
1950 for (int i = 0; i < n; i++) {
1951 if (i == 3) continue;
1952 total += table[i];
1953 }
1954 switch (n) {
1955 case 0: total = 1;
1956 case 1: total++; break;
1957 default: total = -total;
1958 }
1959 struct point p = { total, 1 };
1960 int *q = &p.y;
1961 puts(greeting);
1962 return p.x + *q;
1963}
1964int dispatch(int c) {
1965 void *p = c ? &&one : &&two;
1966 goto *p;
1967one:
1968 return 1;
1969two:
1970 return 2;
1971}
1972int assembly(int x, int *p) {
1973 int r;
1974 __asm__ volatile(\"xadd %0, %2\" : \"=r\"(r), \"+m\"(*p) : \"0\"(x) : \"cc\");
1975 __asm__ goto(\"cbnz %0, %l1\" : : \"r\"(r) : : away);
1976 return r;
1977away:
1978 return 0;
1979}
1980");
1981 let mut names = rucc_base::Interner::new();
1982 let module = rucc_ir::parse(&text, &mut names).expect("the printer writes what it reads");
1983 assert_eq!(rucc_ir::print(&module, &names), text);
1984 }
1985}