1#![doc(html_root_url = "https://docs.rs/rucc-driver/0.10.2")]
30
31pub mod compile;
32pub mod deps;
33pub mod library;
34pub mod link;
35mod map;
36pub mod phase;
37pub mod preprocess;
38pub mod schedule;
39
40use std::fmt::Write as _;
41use std::io::Write as _;
42use std::path::PathBuf;
43
44use rucc_codegen::coverage::{self, Fired};
45use rucc_pp::Dependency;
46use rucc_session::{Dumps, EmitKind, Options, Preinclude, Session, Std, runtime};
47use rucc_target::Triple;
48
49use crate::link::LinkOptions;
50
51pub use crate::compile::{Artifact, Compiled, compile, compile_ir};
52pub use crate::phase::{Input, InputKind, Job, LinkJob, Output, Phase, Plan};
53pub use crate::preprocess::{OsFileSystem, Preprocessed, preprocess};
54pub use crate::schedule::Jobs;
55
56pub const VERSION: &str = env!("CARGO_PKG_VERSION");
58
59#[derive(Debug, Clone, PartialEq, Eq)]
61pub enum Action {
62 Help,
64 Version,
66 Print(String),
72 PrintConfig(Box<Options>),
74 PrintPipeline(Box<Options>),
76 PrintPlan {
78 opts: Box<Options>,
80 plan: Box<Plan>,
82 link: Box<LinkOptions>,
84 },
85 Compile {
87 opts: Box<Options>,
89 plan: Box<Plan>,
91 link: Box<LinkOptions>,
93 jobs: Jobs,
95 verbose: bool,
97 },
98}
99
100#[derive(Debug, Clone, PartialEq, Eq)]
102pub struct CliError {
103 pub message: String,
106}
107
108impl std::fmt::Display for CliError {
109 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
110 f.write_str(&self.message)
111 }
112}
113
114impl std::error::Error for CliError {}
115
116fn err(message: impl Into<String>) -> CliError {
117 CliError { message: message.into() }
118}
119
120enum Query {
126 Machine,
128 Version,
130 Multiarch,
132 SearchDirs,
134 FileName(String),
136 ProgName(String),
138 Libgcc,
140}
141
142pub const USAGE: &str = "\
147rucc, an optimizing C compiler
148
149usage: rucc [options] file...
150
151options:
152 -c compile and assemble, do not link
153 -S compile only, emit assembly
154 -E preprocess only
155 -o <file> write output to <file>, or to standard output for -
156 -D <name>[=<value>], -U <name> define a macro, or undefine one after every -D
157 -I <dir> add <dir> to the include search path
158 -iquote -isystem -idirafter <dir> the other chains, -nostdinc drops ours
159 -I-, -iprefix <p>, -iwithprefix[before] <dir> the older spellings of those
160 -include <file>, -imacros <file> read <file> first, the second for its macros only
161 --sysroot=<dir> look for the library's headers under <dir>, -isysroot too
162 -P, -dM with -E: leave out the markers, or dump the macros
163 -M -MM -MD -MMD write a make rule for the source, the last two compile as well
164 -MF <file> -MT <t> -MQ <t> -MP where the rule goes, what it builds, targets with no recipe
165 -std=<dialect> c89 through c23, and the gnu spellings
166 -fgnuc-version=<v> the GCC release to claim, default 7.0.0
167 -x <lang> treat later inputs as <lang>, or none to stop
168 -O<level> optimize: 0, 1, 2, 3, s, z
169 -fsafety=<tier> check memory safety: off, detect, enforce, kernel
170 -f<pass> -fno-<pass> -fdump-ir=<what> -fopt-info[-<kind>][=FILE]
171 -fpass-fuel=<pass>=<n>, -fpass-fuel-global=<n> stop a pass, or all of them, after n
172 -fdisable-<pass>[=<funcs>], -fenable-<pass>[=<funcs>] run a pass on some functions only
173 -g -g0 -gdwarf-5, -fno-omit-frame-pointer, -mno-red-zone debug info, frame pointer, red zone
174 -l<name>, -L <dir>, -B <dir> link a library, where to look for one, where our own tools are
175 -fPIC -fpic -fPIE -fpie what this compiler does anyway, so they ask for nothing
176 -static -shared -pie -no-pie -nostdlib -nostartfiles -nodefaultlibs -rdynamic -s how to link
177 -Wl,<arg>, -Xlinker <arg>, -fuse-ld=<name> hand an argument to the linker, or pick one
178 -Werror -pedantic -pedantic-errors -w how much to say, and whether it is fatal
179 -m64 -march= -mtune= -mcpu= -mabi= -mcmodel= what machine to generate for
180 -pthread build for more than one thread, and link the library for it
181 -dumpmachine -dumpversion -print-multiarch -print-search-dirs what this compiler is
182 -print-file-name=<name> -print-prog-name=<name> where a file or a program is
183 -j[n] compile n translation units at once, default all
184 -v, -### print each phase as it runs, or without running any
185 --target=<triple> generate code for <triple>
186 --emit=<kind> exe, obj, asm, preprocessed, tast, ir, mir-final,
187 safety-summary, type-granules
188 --print-config, --print-pipeline print the configuration or the pipeline, and exit
189 --version print the version and exit
190 -h, --help print this message and exit
191
192See spec/04-driver-and-cli.md for the full flag reference.
193";
194
195fn joined_or_next(
199 arg: &str,
200 at: usize,
201 args: &[String],
202 i: &mut usize,
203) -> Result<String, CliError> {
204 if arg.len() > at {
205 return Ok(arg[at..].to_owned());
206 }
207 let next = args.get(*i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
208 *i += 1;
209 Ok(next.clone())
210}
211
212pub fn parse_args(args: &[String]) -> Result<Action, CliError> {
219 let host = Triple::host()
220 .ok_or_else(|| err("this host is not a supported target and no --target was given"))?;
221 let mut opts = Options::new(host);
222 let mut inputs: Vec<Input> = Vec::new();
223 let mut print_config = false;
224 let mut print_pipeline = false;
225 let mut print_plan = false;
226 let mut verbose = false;
227 let mut jobs = Jobs::default();
228 let mut nostdinc = false;
229 let mut sysroot: Option<PathBuf> = None;
230 let mut output = None;
231 let mut link = LinkOptions::default();
232 let mut query: Option<Query> = None;
233 let mut threads = false;
234 let mut forced: Option<InputKind> = None;
237 let mut iprefix = String::new();
244
245 let mut i = 0;
246 while i < args.len() {
247 let arg = args[i].as_str();
248 i += 1;
249 match arg {
250 "-h" | "--help" => return Ok(Action::Help),
251 "--version" => return Ok(Action::Version),
252 "--print-config" => print_config = true,
253 "--print-pipeline" => print_pipeline = true,
254 "-###" => print_plan = true,
255 "-v" => verbose = true,
256 "-c" => opts.emit = EmitKind::Object,
257 "-S" => opts.emit = EmitKind::Asm,
258 "-E" => opts.emit = EmitKind::Preprocessed,
259 "-g" => opts.debug_info = true,
260 "-g0" => opts.debug_info = false,
265 "-g1" | "-g2" | "-g3" | "-ggdb" | "-ggdb1" | "-ggdb2" | "-ggdb3" => {
266 opts.debug_info = true;
267 }
268 "-gdwarf" | "-gdwarf-5" => opts.debug_info = true,
271 _ if arg.starts_with("-gdwarf-") => {
272 return Err(err(format!(
273 "{arg}: this compiler writes DWARF 5 and no other version, see \
274 spec/11-debug-info.md"
275 )));
276 }
277 "-Werror" => opts.warnings_are_errors = true,
278 "-w" => opts.warnings = false,
281 "-pedantic-errors" => {
282 opts.pedantic = true;
283 opts.warnings_are_errors = true;
284 }
285 "-P" => opts.line_markers = false,
286 "-M" => {
293 opts.deps.emit = true;
294 opts.deps.instead_of_compiling = true;
295 }
296 "-MM" => {
297 opts.deps.emit = true;
298 opts.deps.instead_of_compiling = true;
299 opts.deps.system_headers = false;
300 }
301 "-MD" => opts.deps.emit = true,
302 "-MMD" => {
303 opts.deps.emit = true;
304 opts.deps.system_headers = false;
305 }
306 "-MP" => opts.deps.phony = true,
307 "-MF" | "-MT" | "-MQ" => {
310 let value =
311 args.get(i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
312 i += 1;
313 match arg {
314 "-MF" => opts.deps.file = Some(value.clone()),
315 "-MT" => opts.deps.targets.push(value.clone()),
319 _ => opts.deps.targets.push(deps::escaped(value)),
320 }
321 }
322 "-dumpmachine" => query = Some(Query::Machine),
326 "-dumpversion" | "-dumpfullversion" => query = Some(Query::Version),
327 "-print-multiarch" => query = Some(Query::Multiarch),
328 "-print-search-dirs" => query = Some(Query::SearchDirs),
329 "-print-libgcc-file-name" => query = Some(Query::Libgcc),
330 _ if arg.starts_with("-print-file-name=") => {
331 query = Some(Query::FileName(arg["-print-file-name=".len()..].to_owned()));
332 }
333 _ if arg.starts_with("-print-prog-name=") => {
334 query = Some(Query::ProgName(arg["-print-prog-name=".len()..].to_owned()));
335 }
336 "-pthread" | "-pthreads" => {
341 opts.defines.push("_REENTRANT".to_owned());
342 threads = true;
343 }
344 "-ansi" => {
345 opts.std = Std::C89;
346 opts.gnu_extensions = false;
347 }
348 "-pedantic" | "-Wpedantic" => opts.pedantic = true,
351 "-fpermissive" => opts.permissive = true,
354 "-fno-permissive" => opts.permissive = false,
355 "-ffreestanding" => opts.hosted = false,
356 "-fhosted" => opts.hosted = true,
357 "-fno-builtin" => opts.builtins = false,
358 "-fbuiltin" => opts.builtins = true,
359 "-fgnu89-inline" => opts.gnu89_inline = true,
363 "-fno-gnu89-inline" => opts.gnu89_inline = false,
364 "-fno-omit-frame-pointer" => opts.frame_pointer = true,
367 "-fomit-frame-pointer" => opts.frame_pointer = false,
368 "-mno-red-zone" => opts.red_zone = false,
369 "-mred-zone" => opts.red_zone = true,
370 "-nostdinc" => nostdinc = true,
374 "-o" => {
375 output = Some(args.get(i).ok_or_else(|| err("-o requires an argument"))?.clone());
376 i += 1;
377 }
378 "-isysroot" => {
385 let dir = args.get(i).ok_or_else(|| err("-isysroot requires an argument"))?;
386 i += 1;
387 sysroot = Some(PathBuf::from(dir));
388 }
389 "-iquote" | "-isystem" | "-idirafter" => {
390 let dir = args.get(i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
391 i += 1;
392 match arg {
393 "-iquote" => opts.search.push_quote(dir.clone()),
394 "-isystem" => opts.search.push_system(dir.clone()),
395 _ => opts.search.push_after(dir.clone()),
396 }
397 }
398 "-iprefix" => {
399 iprefix = args.get(i).ok_or_else(|| err("-iprefix requires an argument"))?.clone();
400 i += 1;
401 }
402 "-iwithprefix" | "-iwithprefixbefore" => {
408 let dir = args.get(i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
409 i += 1;
410 let dir = format!("{iprefix}{dir}");
411 if arg == "-iwithprefix" {
412 opts.search.push_system(dir);
413 } else {
414 opts.search.push_bracket(dir);
415 }
416 }
417 "-include" | "-imacros" => {
418 let name = args.get(i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
419 i += 1;
420 opts.preincludes
421 .push(Preinclude { name: name.clone(), macros_only: arg == "-imacros" });
422 }
423 "-I-" => opts.search.split_quote_chain(),
428 "-x" => {
429 let lang = args.get(i).ok_or_else(|| err("-x requires an argument"))?;
430 i += 1;
431 forced = if lang == "none" {
432 None
433 } else {
434 Some(InputKind::from_x_arg(lang).map_err(|e| err(format!("{e}")))?)
435 };
436 }
437 _ if arg.starts_with("-D") => {
445 let value = joined_or_next(arg, 2, args, &mut i)?;
446 opts.defines.push(value);
447 }
448 _ if arg.starts_with("-U") => {
449 let value = joined_or_next(arg, 2, args, &mut i)?;
450 opts.undefines.push(value);
451 }
452 _ if arg.starts_with("-I") => {
453 let dir = joined_or_next(arg, 2, args, &mut i)?;
454 opts.search.push_bracket(dir);
455 }
456 _ if arg.starts_with("-std=") => {
457 let name = &arg["-std=".len()..];
458 let (std, gnu) = Std::from_flag(name)
459 .ok_or_else(|| err(format!("unknown dialect `{name}`, see --help")))?;
460 opts.std = std;
461 opts.gnu_extensions = gnu;
462 }
463 _ if Dumps::is_family(arg) => {
472 opts.dumps.add(&arg[2..]);
473 }
474 _ if arg.starts_with("-fno-builtin-") => {
479 opts.no_builtin.push(arg["-fno-builtin-".len()..].to_owned());
480 }
481 _ if arg.starts_with("-fgnuc-version=") => {
482 let v = &arg["-fgnuc-version=".len()..];
483 opts.gnuc = v.parse().map_err(err)?;
484 }
485 "-fnested-functions" => {
490 return Err(err(
491 "nested functions are not supported: a call to one goes through a trampoline \
492 written on the stack, which no target that enforces an unexecutable stack \
493 allows",
494 ));
495 }
496 "-fno-nested-functions" => {}
497 "-fPIC" | "-fpic" | "-fPIE" | "-fpie" => {}
508 "-fno-pic" | "-fno-pie" => {
515 return Err(err(
516 "position dependent code is not supported: an address that may be in another \
517 object is loaded out of the global offset table, and nothing here emits the \
518 absolute form this asks for. Use -no-pie if what you meant was how to link",
519 ));
520 }
521 "-static" => link.is_static = true,
525 "-shared" => link.shared = true,
526 "-pie" => link.pie = Some(true),
527 "-no-pie" | "-nopie" => link.pie = Some(false),
528 "-nostdlib" => link.no_stdlib = true,
529 "-nostartfiles" => link.no_startfiles = true,
530 "-nodefaultlibs" => link.no_defaultlibs = true,
531 "-fno-builtins-lib" => link.no_builtins_lib = true,
532 "-fbuiltins-lib" => link.no_builtins_lib = false,
533 "-rdynamic" | "-export-dynamic" => link.export_dynamic = true,
534 "-s" => link.strip = true,
535 "-Xlinker" => {
536 let next = args.get(i).ok_or_else(|| err("-Xlinker requires an argument"))?;
537 i += 1;
538 link.passthrough.push(next.clone());
539 }
540 _ if arg.starts_with("-Wl,") => {
541 link.passthrough.extend(arg["-Wl,".len()..].split(',').map(str::to_owned));
544 }
545 _ if arg.starts_with("-fuse-ld=") => {
546 link.use_ld = Some(arg["-fuse-ld=".len()..].to_owned());
547 }
548 _ if arg.starts_with("-l") && arg.len() > 2 => {
549 inputs.push(Input::library(&arg[2..]));
550 }
551 "-l" => {
552 let next = args.get(i).ok_or_else(|| err("-l requires an argument"))?;
553 i += 1;
554 inputs.push(Input::library(next));
555 }
556 _ if arg.starts_with("-L") => {
557 link.search.push(PathBuf::from(joined_or_next(arg, 2, args, &mut i)?));
558 }
559 _ if arg.starts_with("-B") => {
560 link.prefixes.push(PathBuf::from(joined_or_next(arg, 2, args, &mut i)?));
561 }
562 _ if arg.starts_with("-j") => {
563 jobs = Jobs::parse(&arg[2..]).map_err(err)?;
564 }
565 _ if arg.starts_with("--sysroot=") => {
566 sysroot = Some(PathBuf::from(&arg["--sysroot=".len()..]));
567 }
568 _ if arg.starts_with("--target=") => {
569 let t = &arg["--target=".len()..];
570 opts.target = t.parse().map_err(|e| err(format!("{e}")))?;
571 }
572 _ if arg.starts_with("--emit=") => {
573 let k = &arg["--emit=".len()..];
574 opts.emit = k
575 .parse()
576 .map_err(|()| err(format!("unknown --emit kind `{k}`, see --help")))?;
577 }
578 "-O" | "-Og" => opts.opt_level = rucc_session::OptLevel::O1,
584 "-Ofast" => {
590 return Err(err(
591 "-Ofast is -O3 with fast math, and fast math is not implemented, see \
592 spec/04-driver-and-cli.md section 4.6",
593 ));
594 }
595 _ if arg.starts_with("-O") => {
596 opts.opt_level = arg[2..]
597 .parse()
598 .map_err(|()| err(format!("unknown optimization level `{arg}`")))?;
599 }
600 _ if arg.starts_with("-fsafety=") => {
605 let tier = &arg["-fsafety=".len()..];
606 opts.safety = tier.parse().map_err(|()| {
607 err(format!(
608 "`{tier}` is not a safety tier, which is off, detect, enforce or kernel"
609 ))
610 })?;
611 }
612 _ if arg.starts_with("-fpass-fuel=") => {
616 let (name, count) = arg["-fpass-fuel=".len()..]
617 .split_once('=')
618 .ok_or_else(|| err("-fpass-fuel= is spelled <pass>=<count>"))?;
619 if rucc_opt::pass::find(name).is_none() {
620 return Err(err(format!(
621 "`{name}` is not a pass this compiler has, see --print-pipeline"
622 )));
623 }
624 let count: u32 = count
625 .parse()
626 .map_err(|_| err(format!("`{count}` is not a number of transformations")))?;
627 opts.pass_fuel.push((name.to_owned(), count));
628 }
629 _ if arg.starts_with("-fpass-fuel-global=") => {
630 let count = &arg["-fpass-fuel-global=".len()..];
631 let count: u32 = count
632 .parse()
633 .map_err(|_| err(format!("`{count}` is not a number of transformations")))?;
634 opts.pass_fuel_global = Some(count);
635 }
636 _ if arg == "-fopt-info"
641 || arg.starts_with("-fopt-info=")
642 || arg.starts_with("-fopt-info-") =>
643 {
644 let rest = &arg["-fopt-info".len()..];
645 let (kinds, file) = match rest.split_once('=') {
646 Some((kinds, file)) => (kinds, Some(file)),
647 None => (rest, None),
648 };
649 let kinds = kinds.strip_prefix('-').unwrap_or(kinds);
650 rucc_opt::Wants::none().add(kinds).map_err(err)?;
651 opts.opt_info.push(kinds.to_owned());
652 if let Some(file) = file {
653 if file.is_empty() {
654 return Err(err("-fopt-info= was given no file to write to"));
655 }
656 opts.opt_info_file = Some(file.to_owned());
657 }
658 }
659 _ if arg.starts_with("-fdump-ir=") => {
660 let spec = &arg["-fdump-ir=".len()..];
663 rucc_opt::Dumps::default().add(spec).map_err(err)?;
664 opts.dump_ir.push(spec.to_owned());
665 }
666 _ if arg.starts_with("-fdisable-") || arg.starts_with("-fenable-") => {
672 let on = arg.starts_with("-fenable-");
673 let spec = &arg[if on { "-fenable-".len() } else { "-fdisable-".len() }..];
674 rucc_opt::Gates::default().add(on, spec).map_err(err)?;
675 opts.pass_gates.push((on, spec.to_owned()));
676 }
677 _ if arg.strip_prefix("-fno-").is_some_and(|n| rucc_opt::pass::find(n).is_some()) => {
678 opts.passes.push((arg["-fno-".len()..].to_owned(), false));
679 }
680 _ if arg.strip_prefix("-f").is_some_and(|n| rucc_opt::pass::find(n).is_some()) => {
681 opts.passes.push((arg["-f".len()..].to_owned(), true));
682 }
683 "-Zverify-each" => opts.verify_each = true,
689 _ if arg.starts_with("-Zrule-coverage=") => {
690 let file = &arg["-Zrule-coverage=".len()..];
691 if file.is_empty() {
692 return Err(err("-Zrule-coverage= needs a file to write to"));
693 }
694 opts.rule_coverage = Some(file.to_owned());
695 }
696 _ if arg.starts_with("-Z") => {
697 return Err(err(format!(
698 "`{arg}` is not an unstable option this compiler has, see \
699 spec/04-driver-and-cli.md section 4.11 for the ones it does"
700 )));
701 }
702 "-m64" | "-m32" | "-mx32" => {
707 let want: u32 = match arg {
708 "-m64" => 64,
709 _ => 32,
710 };
711 let have = rucc_target::TargetInfo::new(opts.target).pointer_width;
712 if have != want {
713 return Err(err(format!(
714 "{arg} asks for a {want} bit target and {} is {have} bit, use \
715 --target= to name the one you mean",
716 opts.target
717 )));
718 }
719 }
720 _ if arg.starts_with("-march=")
726 || arg.starts_with("-mtune=")
727 || arg.starts_with("-mcpu=") => {}
728 _ if arg.starts_with("-mabi=") => {
731 let want = &arg["-mabi=".len()..];
732 let have = match opts.target.arch {
733 rucc_target::Arch::X86_64 => "sysv",
734 rucc_target::Arch::Aarch64 => "lp64",
735 rucc_target::Arch::Riscv64 => "lp64d",
736 };
737 if want != have {
738 return Err(err(format!(
739 "{arg}: {} uses the {have} convention and this compiler has no other",
740 opts.target
741 )));
742 }
743 }
744 "-mcmodel=small" => {}
748 _ if arg.starts_with("-mcmodel=") => {
749 return Err(err(format!(
750 "{arg}: this compiler emits the small code model and no other, see \
751 spec/12-targets.md"
752 )));
753 }
754 _ if arg.starts_with("-specs=") => {
758 return Err(err(
759 "-specs= is not supported: the parts of it builds rely on are -B, -L, \
760 -nostdlib, -nostartfiles and -Wl,, see spec/04-driver-and-cli.md \
761 section 4.4",
762 ));
763 }
764 _ if arg.starts_with("-Wa,") || arg.starts_with("-Wp,") => {
770 return Err(err(format!(
771 "`{arg}` is an argument for a separate assembler or preprocessor, and both \
772 are inside this compiler rather than programs it runs"
773 )));
774 }
775 "-Xassembler" | "-Xpreprocessor" => {
776 return Err(err(format!(
777 "{arg} hands an argument to a separate assembler or preprocessor, and both \
778 are inside this compiler rather than programs it runs"
779 )));
780 }
781 _ if arg.starts_with("-W") => {}
788 "-fno-ident"
794 | "-fident"
795 | "-funit-at-a-time"
796 | "-fno-unit-at-a-time"
797 | "-shared-libgcc"
798 | "-static-libgcc" => {}
799 _ if arg.starts_with('-') && arg.len() > 1 => {
800 return Err(err(format!("unknown option `{arg}`")));
805 }
806 _ => inputs.push(Input { path: arg.to_owned(), forced, library: false }),
807 }
808 }
809
810 link.sysroot = sysroot.clone();
817 if threads {
822 inputs.push(Input::library("pthread"));
823 }
824 if let Some(query) = query {
825 return Ok(Action::Print(answer(&query, &opts, &link)));
826 }
827 if opts.deps.instead_of_compiling {
833 opts.emit = EmitKind::Preprocessed;
834 }
835 if !nostdinc {
836 opts.search.push_system(runtime::DIR);
837 for dir in library::system_dirs(opts.target, sysroot.as_deref()) {
841 opts.search.push_system(dir);
842 }
843 }
844 opts.search.remove_duplicates();
848
849 if print_config {
852 return Ok(Action::PrintConfig(Box::new(opts)));
853 }
854 if print_pipeline {
855 return Ok(Action::PrintPipeline(Box::new(opts)));
856 }
857 let plan = Plan::new(&opts, &inputs, output.as_deref()).map_err(|e| err(e.message))?;
858 if print_plan {
859 return Ok(Action::PrintPlan {
860 opts: Box::new(opts),
861 plan: Box::new(plan),
862 link: Box::new(link),
863 });
864 }
865 Ok(Action::Compile {
866 opts: Box::new(opts),
867 plan: Box::new(plan),
868 link: Box::new(link),
869 jobs,
870 verbose,
871 })
872}
873
874fn answer(query: &Query, opts: &Options, link: &LinkOptions) -> String {
880 let found = |name: &str| {
881 link::find_in_search(link, opts.target, name)
882 .map_or_else(|| name.to_owned(), |path| path.display().to_string())
883 };
884 match query {
885 Query::Machine => opts.target.to_string(),
886 Query::Version => VERSION.to_owned(),
887 Query::Multiarch => link::multiarch(opts.target),
888 Query::SearchDirs => {
893 let here = std::env::current_exe()
894 .ok()
895 .and_then(|p| p.parent().map(std::path::Path::to_path_buf))
896 .unwrap_or_default();
897 let list = |dirs: &[PathBuf]| {
898 dirs.iter().map(|d| d.display().to_string()).collect::<Vec<_>>().join(":")
899 };
900 let libraries = link::search_dirs(link, opts.target);
901 format!(
902 "install: {}\nprograms: ={}\nlibraries: ={}",
903 here.display(),
904 list(&link.prefixes),
905 list(&libraries)
906 )
907 }
908 Query::FileName(name) => found(name),
909 Query::Libgcc => found("libgcc.a"),
913 Query::ProgName(name) => link
917 .prefixes
918 .iter()
919 .map(|dir| dir.join(name))
920 .find(|path| path.is_file())
921 .map_or_else(|| name.clone(), |path| path.display().to_string()),
922 }
923}
924
925#[must_use]
931pub fn print_pipeline(opts: &Options) -> String {
932 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
933 settings.toggles.clone_from(&opts.passes);
934 settings.global_fuel = opts.pass_fuel_global;
935 for (on, spec) in &opts.pass_gates {
936 let _ = settings.gates.add(*on, spec);
939 }
940 rucc_opt::pipeline::print(&settings)
941}
942
943#[must_use]
948pub fn print_config(opts: &Options) -> String {
949 let sess = Session::new(opts.clone());
950 let t = &sess.target;
951 let mut out = String::new();
952 let _ = writeln!(out, "version: {VERSION}");
953 let _ = writeln!(out, "target: {}", opts.target);
957 let _ = writeln!(out, "arch: {}", opts.target.arch.as_str());
958 let _ = writeln!(out, "os: {}", opts.target.os.as_str());
959 let _ = writeln!(out, "env: {}", opts.target.env.as_str());
960 let _ = writeln!(out, "object-format: {}", t.object_format.as_str());
961 let _ = writeln!(out, "pointer-width: {}", t.pointer_width);
962 let _ = writeln!(out, "long-width: {}", t.long_width);
963 let _ = writeln!(out, "long-double-width: {}", t.long_double_width);
964 let _ = writeln!(out, "endian: {}", if t.little_endian { "little" } else { "big" });
965 let _ = writeln!(out, "char-signed: {}", t.char_is_signed);
966 let _ = writeln!(out, "va-list: {}", t.va_list.map_or("none", |list| list.as_str()));
967 let regs: Vec<String> = t
970 .regs
971 .classes()
972 .map(|(class, info)| format!("{} {}", info.name, t.regs.len(class)))
973 .collect();
974 let _ = writeln!(
975 out,
976 "registers: {}",
977 if regs.is_empty() { "none".to_string() } else { regs.join(", ") }
978 );
979 let _ = writeln!(out, "opt-level: {}", sess.opts.opt_level);
980 let _ = writeln!(out, "safety: {}", sess.opts.safety);
981 let _ = writeln!(out, "emit: {}", sess.opts.emit.as_str());
982 let _ = writeln!(out, "debug-info: {}", sess.opts.debug_info);
983 let _ = writeln!(out, "frame-pointer: {}", sess.opts.frame_pointer);
984 let _ = writeln!(out, "red-zone: {}", sess.opts.red_zone);
985 for dir in sess.opts.search.dirs() {
988 let system = if dir.is_system { " (system)" } else { "" };
989 let _ = writeln!(out, "include: {}{system}", dir.path.display());
990 }
991 out
992}
993
994fn deps_target_output<'a>(opts: &Options, plan: &'a Plan) -> Option<&'a str> {
1002 if opts.emit == EmitKind::Preprocessed { None } else { plan.output.as_deref() }
1003}
1004
1005fn write_named(path: &str, bytes: &[u8]) -> Result<(), String> {
1008 if path == "-" {
1009 return write_out(&Output::Stdout, bytes);
1010 }
1011 write_out(&Output::File(path.to_owned()), bytes)
1012}
1013
1014fn write_deps(
1020 opts: &Options,
1021 plan: &Plan,
1022 job: &Job,
1023 found: &[Dependency],
1024 stderr: &mut impl std::io::Write,
1025) -> bool {
1026 let targets = if opts.deps.targets.is_empty() {
1027 vec![deps::default_target(&job.input, deps_target_output(opts, plan))]
1028 } else {
1029 opts.deps.targets.clone()
1030 };
1031 let rule = deps::rule(&opts.deps, &targets, &job.input, found);
1032 let wrote = match deps::default_file(&opts.deps, &job.input, plan.output.as_deref()) {
1035 Some(path) => write_named(&path, rule.as_bytes()).and_then(|()| {
1039 if opts.deps.instead_of_compiling { write_out(&job.output, b"") } else { Ok(()) }
1040 }),
1041 None => write_out(&job.output, rule.as_bytes()),
1042 };
1043 if let Err(e) = wrote {
1044 let _ = writeln!(stderr, "rucc: error: {e}");
1045 return false;
1046 }
1047 true
1048}
1049
1050fn preprocess_all(opts: &Options, plan: &Plan) -> i32 {
1056 let fs = OsFileSystem::new();
1057 let mut stderr = std::io::stderr().lock();
1058 let mut failed = false;
1059 for job in &plan.jobs {
1060 if !job.phases.first().is_some_and(|p| *p == Phase::Preprocess) {
1061 continue;
1064 }
1065 let result = preprocess(opts, &job.input, &fs);
1066 for message in &result.messages {
1067 let _ = writeln!(stderr, "{message}");
1068 }
1069 if result.failed() {
1070 failed = true;
1071 continue;
1072 }
1073 if opts.deps.emit {
1074 failed |= !write_deps(opts, plan, job, &result.deps, &mut stderr);
1075 if opts.deps.instead_of_compiling {
1078 continue;
1079 }
1080 }
1081 if let Err(e) = write_out(&job.output, result.text.as_bytes()) {
1082 let _ = writeln!(stderr, "rucc: error: {e}");
1083 failed = true;
1084 }
1085 }
1086 i32::from(failed)
1087}
1088
1089fn compile_all(opts: &Options, plan: &Plan) -> i32 {
1095 let fs = OsFileSystem::new();
1096 let mut stderr = std::io::stderr().lock();
1097 let mut failed = false;
1098 let (mut remarks, ok) = Remarks::new(opts.opt_info_file.as_ref(), &mut stderr);
1099 failed |= !ok;
1100 let mut fired = Fired::new();
1101 for job in &plan.jobs {
1102 if !job.phases.contains(&Phase::Compile) {
1103 continue;
1104 }
1105 let result = if job.kind == InputKind::Ir {
1109 compile_ir(opts, &job.input, &fs)
1110 } else {
1111 compile(opts, &job.input, &fs)
1112 };
1113 fired.merge(&result.fired);
1114 failed |= !write_dumps(&job.input, &result.dumps, &mut stderr);
1115 failed |= !remarks.write(&result.remarks, &mut stderr);
1116 for message in &result.messages {
1117 let _ = writeln!(stderr, "{message}");
1118 }
1119 if result.failed() {
1120 failed = true;
1121 continue;
1122 }
1123 if opts.deps.emit {
1128 failed |= !write_deps(opts, plan, job, &result.deps, &mut stderr);
1129 }
1130 if let Err(e) = write_out(&job.output, result.artifact.bytes()) {
1131 let _ = writeln!(stderr, "rucc: error: {e}");
1132 failed = true;
1133 }
1134 }
1135 failed |= !write_coverage(opts, &fired, &mut stderr);
1136 i32::from(failed)
1137}
1138
1139struct Scratch {
1146 dir: PathBuf,
1148}
1149
1150impl Scratch {
1151 fn new() -> Result<Scratch, String> {
1157 let dir = std::env::temp_dir().join(format!("rucc-{}", std::process::id()));
1158 std::fs::create_dir_all(&dir).map_err(|e| format!("{}: {e}", dir.display()))?;
1159 Ok(Scratch { dir })
1160 }
1161}
1162
1163impl Drop for Scratch {
1164 fn drop(&mut self) {
1165 let _ = std::fs::remove_dir_all(&self.dir);
1166 }
1167}
1168
1169fn link_line(opts: &Options, link: &LinkOptions, job: &LinkJob) -> Result<String, link::Error> {
1176 let linker = link::find(opts.target, link)?;
1177 let args = link::line(opts.target, link, &job.inputs, &job.output)?;
1178 Ok(link::render(&linker, &args))
1179}
1180
1181fn link_all(opts: &Options, plan: &Plan, link: &LinkOptions, verbose: bool) -> i32 {
1188 let Some(job) = &plan.link else {
1189 let mut stderr = std::io::stderr().lock();
1192 let _ = writeln!(stderr, "rucc: error: there is nothing to link");
1193 return 1;
1194 };
1195 let linker = match link::find(opts.target, link) {
1198 Ok(linker) => linker,
1199 Err(why) => return complain(why),
1200 };
1201
1202 let scratch = match Scratch::new() {
1203 Ok(scratch) => scratch,
1204 Err(why) => return complain(format!("could not make a place for the object files: {why}")),
1205 };
1206
1207 let fs = OsFileSystem::new();
1208 let mut failed = false;
1209 let mut produced: Vec<String> = Vec::with_capacity(plan.jobs.len());
1212 let mut fired = Fired::new();
1213 {
1214 let mut stderr = std::io::stderr().lock();
1215 let (mut remarks, ok) = Remarks::new(opts.opt_info_file.as_ref(), &mut stderr);
1216 failed |= !ok;
1217 for (at, job) in plan.jobs.iter().enumerate() {
1218 let out = match &job.output {
1219 Output::Temporary(hint) => {
1220 scratch.dir.join(format!("{at}-{hint}")).display().to_string()
1223 }
1224 Output::File(path) => path.clone(),
1225 Output::Stdout => continue,
1228 };
1229 produced.push(out.clone());
1230 if !job.phases.contains(&Phase::Compile) {
1231 continue;
1232 }
1233 let result = if job.kind == InputKind::Ir {
1234 compile_ir(opts, &job.input, &fs)
1235 } else {
1236 compile(opts, &job.input, &fs)
1237 };
1238 fired.merge(&result.fired);
1239 failed |= !write_dumps(&job.input, &result.dumps, &mut stderr);
1240 failed |= !remarks.write(&result.remarks, &mut stderr);
1241 for message in &result.messages {
1242 let _ = writeln!(stderr, "{message}");
1243 }
1244 if result.failed() {
1245 failed = true;
1246 continue;
1247 }
1248 if opts.deps.emit {
1253 failed |= !write_deps(opts, plan, job, &result.deps, &mut stderr);
1254 }
1255 if !matches!(result.artifact, Artifact::Object(_)) {
1256 let _ = writeln!(
1261 stderr,
1262 "rucc: internal error: {}: no object file was produced for the link",
1263 job.input
1264 );
1265 failed = true;
1266 continue;
1267 }
1268 if let Err(e) = std::fs::write(&out, result.artifact.bytes()) {
1269 let _ = writeln!(stderr, "rucc: error: {out}: {e}");
1270 failed = true;
1271 }
1272 }
1273 failed |= !write_coverage(opts, &fired, &mut stderr);
1274 }
1275 if failed {
1276 return 1;
1280 }
1281
1282 let mut outputs = produced.into_iter();
1286 let mut items = Vec::with_capacity(job.inputs.len());
1287 for item in &job.inputs {
1288 match item {
1289 link::Item::Library(name) => items.push(link::Item::Library(name.clone())),
1290 link::Item::File(_) => match outputs.next() {
1291 Some(path) => items.push(link::Item::File(path)),
1292 None => return complain("the plan asks the linker for a file nothing produced"),
1293 },
1294 }
1295 }
1296
1297 let args = match link::line(opts.target, link, &items, &job.output) {
1298 Ok(args) => args,
1299 Err(why) => return complain(why),
1300 };
1301 if verbose {
1302 let mut stderr = std::io::stderr().lock();
1303 let _ = writeln!(stderr, "{}", link::render(&linker, &args));
1304 }
1305 match link::run(&linker, &args) {
1306 Ok(()) => 0,
1307 Err(link::Error::Refused { .. }) => 1,
1310 Err(why) => complain(why),
1311 }
1312}
1313
1314fn complain(why: impl std::fmt::Display) -> i32 {
1316 let mut stderr = std::io::stderr().lock();
1317 let _ = writeln!(stderr, "rucc: error: {why}");
1318 1
1319}
1320
1321fn write_coverage(opts: &Options, fired: &Fired, stderr: &mut impl std::io::Write) -> bool {
1330 let Some(path) = &opts.rule_coverage else { return true };
1331 let Some(table) = coverage::table(opts.target.arch) else {
1332 let _ = writeln!(
1333 stderr,
1334 "rucc: error: there are no lowering rules for {} yet, so there is no coverage of them \
1335 to report",
1336 opts.target
1337 );
1338 return false;
1339 };
1340 match std::fs::write(path, fired.listing(table)) {
1341 Ok(()) => true,
1342 Err(e) => {
1343 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1344 false
1345 }
1346 }
1347}
1348
1349struct Remarks {
1356 file: Option<String>,
1358 started: bool,
1361}
1362
1363impl Remarks {
1364 fn new(file: Option<&String>, stderr: &mut impl std::io::Write) -> (Self, bool) {
1370 let mut ok = true;
1371 if let Some(path) = file {
1372 if let Err(e) = std::fs::write(path, "") {
1373 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1374 ok = false;
1375 }
1376 }
1377 (Self { file: file.cloned(), started: false }, ok)
1378 }
1379
1380 fn write(&mut self, text: &str, stderr: &mut impl std::io::Write) -> bool {
1386 if text.is_empty() {
1387 return true;
1388 }
1389 let Some(path) = &self.file else {
1390 let _ = write!(stderr, "{text}");
1391 return true;
1392 };
1393 let opened = std::fs::OpenOptions::new()
1394 .write(true)
1395 .append(self.started)
1396 .truncate(!self.started)
1397 .create(true)
1398 .open(path);
1399 self.started = true;
1400 let result =
1401 opened.and_then(|mut file| std::io::Write::write_all(&mut file, text.as_bytes()));
1402 if let Err(e) = result {
1403 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1404 return false;
1405 }
1406 true
1407 }
1408}
1409
1410fn write_dumps(input: &str, dumps: &[rucc_opt::Dump], stderr: &mut impl std::io::Write) -> bool {
1421 let stem = std::path::Path::new(input)
1422 .file_name()
1423 .map_or_else(|| input.to_owned(), |name| name.to_string_lossy().into_owned());
1424 let mut ok = true;
1425 for dump in dumps {
1426 let path = format!("{stem}.{}.ir", dump.name);
1427 if let Err(e) = std::fs::write(&path, &dump.text) {
1428 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1429 ok = false;
1430 }
1431 }
1432 ok
1433}
1434
1435fn write_out(output: &Output, bytes: &[u8]) -> Result<(), String> {
1442 match output {
1443 Output::Stdout => {
1444 let mut stdout = std::io::stdout().lock();
1445 stdout.write_all(bytes).map_err(|e| format!("writing to standard output: {e}"))
1446 }
1447 Output::File(path) | Output::Temporary(path) => {
1448 std::fs::write(path, bytes).map_err(|e| format!("{path}: {e}"))
1449 }
1450 }
1451}
1452
1453pub fn run(args: &[String]) -> i32 {
1458 match parse_args(args) {
1459 Ok(Action::Help) => {
1460 print!("{USAGE}");
1461 0
1462 }
1463 Ok(Action::Version) => {
1464 println!("rucc {VERSION}");
1465 0
1466 }
1467 Ok(Action::Print(line)) => {
1468 println!("{line}");
1469 0
1470 }
1471 Ok(Action::PrintConfig(opts)) => {
1472 print!("{}", print_config(&opts));
1473 0
1474 }
1475 Ok(Action::PrintPipeline(opts)) => {
1476 print!("{}", print_pipeline(&opts));
1477 0
1478 }
1479 Ok(Action::PrintPlan { opts, plan, link }) => {
1480 print!("{}", plan.render());
1481 if let Some(job) = &plan.link {
1485 match link_line(&opts, &link, job) {
1486 Ok(line) => println!("{line}"),
1487 Err(why) => {
1488 let mut stderr = std::io::stderr().lock();
1489 let _ = writeln!(stderr, "rucc: error: {why}");
1490 return 1;
1491 }
1492 }
1493 }
1494 0
1495 }
1496 Ok(Action::Compile { opts, plan, link, jobs, verbose }) => {
1497 {
1498 let mut stderr = std::io::stderr().lock();
1499 if verbose {
1500 let _ = write!(stderr, "{}", plan.render());
1501 let _ = writeln!(stderr, "workers: {}", jobs.count());
1502 }
1503 }
1504 if opts.emit == EmitKind::Preprocessed {
1505 return preprocess_all(&opts, &plan);
1506 }
1507 if opts.emit != EmitKind::Executable {
1508 return compile_all(&opts, &plan);
1509 }
1510 link_all(&opts, &plan, &link, verbose)
1511 }
1512 Err(e) => {
1513 let mut stderr = std::io::stderr().lock();
1514 let _ = writeln!(stderr, "rucc: error: {e}");
1515 let _ = writeln!(stderr, "rucc: note: run `rucc --help` for usage");
1516 1
1517 }
1518 }
1519}
1520
1521#[cfg(test)]
1522mod tests {
1523 use rucc_session::{GnucVersion, IncludeForm, OptLevel};
1524
1525 use super::*;
1526
1527 fn args(s: &[&str]) -> Vec<String> {
1528 s.iter().map(|x| (*x).to_owned()).collect()
1529 }
1530
1531 #[test]
1532 fn help_and_version_win_over_everything_else() {
1533 assert_eq!(parse_args(&args(&["-c", "--help", "x.c"])).unwrap(), Action::Help);
1534 assert_eq!(parse_args(&args(&["--version"])).unwrap(), Action::Version);
1535 }
1536
1537 fn compile(s: &[&str]) -> (Box<Options>, Box<Plan>) {
1538 match parse_args(&args(s)).expect("expected a compilation") {
1539 Action::Compile { opts, plan, .. } => (opts, plan),
1540 other => panic!("expected a compilation, got {other:?}"),
1541 }
1542 }
1543
1544 fn linking(s: &[&str]) -> (Box<LinkOptions>, Box<Plan>) {
1545 match parse_args(&args(s)).expect("expected a compilation") {
1546 Action::Compile { link, plan, .. } => (link, plan),
1547 other => panic!("expected a compilation, got {other:?}"),
1548 }
1549 }
1550
1551 #[test]
1552 fn collects_inputs_and_flags() {
1553 let (opts, plan) = compile(&["-c", "-O2", "-g", "a.c", "b.c"]);
1554 let paths: Vec<&str> = plan.jobs.iter().map(|j| j.input.as_str()).collect();
1555 assert_eq!(paths, vec!["a.c", "b.c"]);
1556 assert_eq!(opts.opt_level, OptLevel::O2);
1557 assert_eq!(opts.emit, EmitKind::Object);
1558 assert!(opts.debug_info);
1559 }
1560
1561 #[test]
1564 fn an_unstable_option_is_taken_and_one_that_does_not_exist_is_refused() {
1565 let (opts, _) = compile(&["-c", "-Zrule-coverage=/tmp/rules.cov", "a.c"]);
1566 assert_eq!(opts.rule_coverage.as_deref(), Some("/tmp/rules.cov"));
1567
1568 let (plain, _) = compile(&["-c", "a.c"]);
1569 assert_eq!(plain.rule_coverage, None, "nothing is measured unless it was asked for");
1570
1571 assert!(parse_args(&args(&["-Zrule-coverage=", "a.c"])).is_err(), "a file with no name");
1572 let unknown = parse_args(&args(&["-Zwhat", "a.c"])).expect_err("there is no such option");
1573 assert!(unknown.message.contains("4.11"), "{}", unknown.message);
1574 }
1575
1576 #[test]
1577 fn a_bare_dash_o_means_o1_the_way_gcc_reads_it() {
1578 let (opts, _) = compile(&["-O", "a.c"]);
1579 assert_eq!(opts.opt_level, OptLevel::O1);
1580 }
1581
1582 #[test]
1583 fn dash_x_applies_to_later_inputs_only_and_none_stops_it() {
1584 let (_, plan) = compile(&["a.o", "-x", "c", "b.txt", "-x", "none", "c.o"]);
1585 assert_eq!(plan.jobs[0].kind, InputKind::LinkerInput);
1586 assert_eq!(plan.jobs[1].kind, InputKind::C);
1587 assert_eq!(plan.jobs[2].kind, InputKind::LinkerInput);
1588 }
1589
1590 #[test]
1591 fn dash_j_reaches_the_scheduler_and_defaults_to_the_machine() {
1592 let (_, _, jobs) = match parse_args(&args(&["-j4", "a.c"])).unwrap() {
1593 Action::Compile { opts, plan, jobs, .. } => (opts, plan, jobs),
1594 other => panic!("expected a compilation, got {other:?}"),
1595 };
1596 assert_eq!(jobs.count(), 4);
1597
1598 let default = match parse_args(&args(&["a.c"])).unwrap() {
1599 Action::Compile { jobs, .. } => jobs,
1600 other => panic!("expected a compilation, got {other:?}"),
1601 };
1602 assert_eq!(default, Jobs::available());
1603 assert!(parse_args(&args(&["-j0", "a.c"])).is_err());
1604 }
1605
1606 #[test]
1607 fn triple_hash_prints_the_plan_and_runs_nothing() {
1608 let a = parse_args(&args(&["-###", "-c", "a.c"])).unwrap();
1609 let Action::PrintPlan { plan, .. } = a else { panic!("expected a plan dump") };
1610 assert!(plan.render().contains("a.c: preprocess, compile, assemble -> a.o"));
1611 }
1612
1613 #[test]
1614 fn dash_x_names_what_it_accepts_when_it_does_not_know_a_language() {
1615 let e = parse_args(&args(&["-x", "fortran", "a.c"])).unwrap_err();
1616 assert!(e.message.contains("assembler-with-cpp"), "{}", e.message);
1617 }
1618
1619 #[test]
1620 fn an_unknown_flag_is_an_error_rather_than_a_shrug() {
1621 let e = parse_args(&args(&["-fno-such-thing", "a.c"])).unwrap_err();
1622 assert!(e.message.contains("unknown option"), "{}", e.message);
1623 }
1624
1625 #[test]
1628 fn permissive_reads_in_both_directions_and_the_last_one_wins() {
1629 let (opts, _) = compile(&["-c", "a.c"]);
1630 assert!(!opts.permissive, "off unless it is asked for");
1631
1632 let (opts, _) = compile(&["-c", "-fpermissive", "a.c"]);
1633 assert!(opts.permissive);
1634
1635 let (opts, _) = compile(&["-c", "-fpermissive", "-fno-permissive", "a.c"]);
1636 assert!(!opts.permissive);
1637 }
1638
1639 #[test]
1640 fn asking_for_nested_functions_is_told_why_it_is_not_coming() {
1641 let e = parse_args(&args(&["-fnested-functions", "a.c"])).unwrap_err();
1642 assert!(e.message.contains("trampoline"), "{}", e.message);
1643 assert!(parse_args(&args(&["-fno-nested-functions", "a.c"])).is_ok());
1644 }
1645
1646 #[test]
1647 fn the_flag_every_configure_script_writes_is_taken() {
1648 for flag in ["-fPIC", "-fpic", "-fPIE", "-fpie"] {
1651 let (opts, _) = compile(&["-c", flag, "a.c"]);
1652 assert_eq!(opts.emit, EmitKind::Object, "{flag}");
1653 }
1654 }
1655
1656 #[test]
1657 fn asking_for_position_dependent_code_is_told_why_it_is_not_coming() {
1658 for flag in ["-fno-pic", "-fno-pie"] {
1659 let e = parse_args(&args(&[flag, "a.c"])).unwrap_err();
1660 assert!(e.message.contains("global offset table"), "{flag}: {}", e.message);
1661 assert!(e.message.contains("-no-pie"), "{flag}: {}", e.message);
1664 }
1665 }
1666
1667 #[test]
1668 fn an_unsupported_target_names_itself() {
1669 let e = parse_args(&args(&["--target=sparc64-linux-gnu", "a.c"])).unwrap_err();
1670 assert!(e.message.contains("sparc64"), "{}", e.message);
1671 }
1672
1673 #[test]
1674 fn no_inputs_is_an_error_but_print_config_needs_none() {
1675 assert!(parse_args(&args(&[])).is_err());
1676 assert!(matches!(parse_args(&args(&["--print-config"])), Ok(Action::PrintConfig(_))));
1677 }
1678
1679 #[test]
1680 fn print_config_reports_the_target_it_was_given_not_the_host() {
1681 let a = parse_args(&args(&["--print-config", "--target=riscv64-linux-musl"])).unwrap();
1682 let Action::PrintConfig(opts) = a else { panic!("expected a configuration dump") };
1683 let text = print_config(&opts);
1684 assert!(text.contains("target: riscv64-unknown-linux-musl"), "{text}");
1685 assert!(text.contains("char-signed: false"), "{text}");
1686 assert!(text.contains("object-format: elf"), "{text}");
1687 assert!(text.contains("va-list: void-pointer"), "{text}");
1688 assert!(text.contains("registers: none"), "{text}");
1691 }
1692
1693 #[test]
1694 fn print_config_has_one_key_per_line_and_a_fixed_order() {
1695 let opts = Options::new("x86_64-unknown-linux-gnu".parse().unwrap());
1696 let text = print_config(&opts);
1697 let keys: Vec<&str> =
1698 text.lines().map(|l| l.split(':').next().unwrap_or_default()).collect();
1699 assert_eq!(keys[0], "version");
1700 assert_eq!(keys[1], "target");
1701 assert_eq!(keys.len(), 19);
1702 assert!(text.ends_with('\n'));
1703 }
1704
1705 #[test]
1706 fn the_safety_tier_is_read_off_the_command_line_and_a_wrong_one_is_refused() {
1707 let (opts, _) = compile(&["a.c"]);
1708 assert_eq!(opts.safety, rucc_session::Safety::Off);
1709
1710 for (flag, tier) in [
1711 ("-fsafety=detect", rucc_session::Safety::Detect),
1712 ("-fsafety=enforce", rucc_session::Safety::Enforce),
1713 ("-fsafety=kernel", rucc_session::Safety::Kernel),
1714 ("-fsafety=off", rucc_session::Safety::Off),
1715 ] {
1716 let (opts, _) = compile(&[flag, "a.c"]);
1717 assert_eq!(opts.safety, tier, "{flag}");
1718 }
1719
1720 let (opts, _) = compile(&["-fsafety=enforce", "-fsafety=off", "a.c"]);
1722 assert_eq!(opts.safety, rucc_session::Safety::Off);
1723
1724 let e = parse_args(&args(&["-fsafety=on", "a.c"])).unwrap_err();
1727 assert!(e.message.contains("is not a safety tier"), "{}", e.message);
1728 assert!(parse_args(&args(&["-fsafety", "a.c"])).is_err());
1729 }
1730
1731 #[test]
1732 fn print_pipeline_answers_with_the_passes_the_level_asked_for() {
1733 let a = parse_args(&args(&["--print-pipeline", "-O2"])).unwrap();
1734 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1735 let text = print_pipeline(&opts);
1736 assert!(text.starts_with("level: -O2\n"), "{text}");
1737 assert!(text.contains("fold"), "{text}");
1738
1739 let a = parse_args(&args(&["--print-pipeline"])).unwrap();
1740 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1741 assert!(print_pipeline(&opts).contains("1: simplify-cfg,"), "{}", print_pipeline(&opts));
1744
1745 let a = parse_args(&args(&["--print-pipeline", "-fno-simplify-cfg"])).unwrap();
1746 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1747 assert!(print_pipeline(&opts).contains("no passes"), "{}", print_pipeline(&opts));
1750 }
1751
1752 #[test]
1753 fn print_pipeline_takes_the_toggles_into_account() {
1754 let a = parse_args(&args(&["--print-pipeline", "-O2", "-fno-fold"])).unwrap();
1755 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1756 let text = print_pipeline(&opts);
1757 assert!(!text.contains("fold"), "{text}");
1760 assert!(text.contains("dce"), "{text}");
1761
1762 let mut off = vec!["--print-pipeline".to_owned(), "-O2".to_owned()];
1766 off.extend(rucc_opt::PASSES.iter().map(|p| format!("-fno-{}", p.name())));
1767 let spelled: Vec<&str> = off.iter().map(String::as_str).collect();
1768 let a = parse_args(&args(&spelled)).unwrap();
1769 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1770 assert!(print_pipeline(&opts).contains("no passes"), "{}", print_pipeline(&opts));
1771 }
1772
1773 #[test]
1774 fn print_pipeline_says_when_a_budget_will_stop_the_run_short() {
1775 let a = parse_args(&args(&["--print-pipeline", "-O2"])).unwrap();
1776 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1777 assert!(!print_pipeline(&opts).contains("global fuel"));
1778
1779 let a = parse_args(&args(&["--print-pipeline", "-O2", "-fpass-fuel-global=4"])).unwrap();
1780 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1781 let text = print_pipeline(&opts);
1782 assert!(text.contains("global fuel: 4"), "{text}");
1785 }
1786
1787 #[test]
1790 fn a_pass_is_named_by_dash_f_and_unnamed_by_dash_f_no() {
1791 let (opts, _) = compile(&["-c", "-O0", "-ffold", "-fno-fold", "-ffold", "a.c"]);
1792 assert_eq!(
1793 opts.passes,
1794 [("fold".to_owned(), true), ("fold".to_owned(), false), ("fold".to_owned(), true)]
1795 );
1796
1797 let e = parse_args(&args(&["-fno-such-pass", "a.c"])).unwrap_err();
1798 assert!(e.message.contains("unknown option"), "{}", e.message);
1799 }
1800
1801 #[test]
1802 fn pass_fuel_names_a_pass_and_a_count_and_refuses_anything_else() {
1803 let (opts, _) = compile(&["-c", "-O2", "-fpass-fuel=fold=3", "a.c"]);
1804 assert_eq!(opts.pass_fuel, [("fold".to_owned(), 3)]);
1805
1806 let e = parse_args(&args(&["-fpass-fuel=fold", "a.c"])).unwrap_err();
1807 assert!(e.message.contains("<pass>=<count>"), "{}", e.message);
1808 let e = parse_args(&args(&["-fpass-fuel=nosuch=3", "a.c"])).unwrap_err();
1809 assert!(e.message.contains("--print-pipeline"), "{}", e.message);
1810 let e = parse_args(&args(&["-fpass-fuel=fold=lots", "a.c"])).unwrap_err();
1811 assert!(e.message.contains("not a number"), "{}", e.message);
1812 }
1813
1814 #[test]
1815 fn global_pass_fuel_is_a_count_on_its_own_and_defaults_to_no_limit() {
1816 let (opts, _) = compile(&["-c", "-O2", "a.c"]);
1817 assert_eq!(opts.pass_fuel_global, None);
1818
1819 let (opts, _) = compile(&["-c", "-O2", "-fpass-fuel-global=12", "a.c"]);
1820 assert_eq!(opts.pass_fuel_global, Some(12));
1821 assert!(opts.pass_fuel.is_empty());
1824
1825 let e = parse_args(&args(&["-fpass-fuel-global=lots", "a.c"])).unwrap_err();
1826 assert!(e.message.contains("not a number"), "{}", e.message);
1827 }
1828
1829 #[test]
1830 fn a_gate_names_a_pass_and_optionally_the_functions_it_covers() {
1831 let (opts, _) = compile(&["-c", "-O2", "-fdisable-fold", "-fenable-fold=2-4,main", "a.c"]);
1832 assert_eq!(
1833 opts.pass_gates,
1834 [(false, "fold".to_owned()), (true, "fold=2-4,main".to_owned())],
1835 "the order is what decides, so it has to survive the parse"
1836 );
1837
1838 let e = parse_args(&args(&["-fdisable-nosuch", "a.c"])).unwrap_err();
1839 assert!(e.message.contains("--print-pipeline"), "{}", e.message);
1840 let e = parse_args(&args(&["-fenable-fold=9-2", "a.c"])).unwrap_err();
1841 assert!(e.message.contains("ends before it starts"), "{}", e.message);
1842 let e = parse_args(&args(&["-fdisable-fold=", "a.c"])).unwrap_err();
1843 assert!(e.message.contains("is empty"), "{}", e.message);
1844 }
1845
1846 #[test]
1847 fn the_pipeline_listing_says_which_passes_a_gate_touched() {
1848 let (opts, _) = compile(&["-c", "-O2", "-fdisable-fold=main", "a.c"]);
1849 let text = print_pipeline(&opts);
1850 assert!(text.contains("fold, "), "{text}");
1851 assert!(text.contains("[off for main]"), "{text}");
1852 }
1853
1854 #[test]
1858 fn a_dump_is_checked_when_it_is_asked_for_rather_than_when_it_is_taken() {
1859 let (opts, _) = compile(&["-c", "-O2", "-fdump-ir=all", "-fdump-ir=after-fold", "a.c"]);
1860 assert_eq!(opts.dump_ir, ["all", "after-fold"]);
1861
1862 let e = parse_args(&args(&["-fdump-ir=after-nosuch", "a.c"])).unwrap_err();
1863 assert!(e.message.contains("nosuch"), "{}", e.message);
1864 assert!(parse_args(&args(&["-fdump-ir=sideways-fold", "a.c"])).is_err());
1865 }
1866
1867 #[test]
1873 fn opt_info_takes_kinds_and_a_file_and_refuses_a_kind_it_does_not_have() {
1874 let (opts, _) = compile(&["-c", "-O2", "-fopt-info", "a.c"]);
1875 assert_eq!(opts.opt_info, [""], "a bare flag asks for the rewrites");
1876 assert_eq!(opts.opt_info_file, None, "and goes to standard error");
1877
1878 let (opts, _) = compile(&["-c", "-O2", "-fopt-info-missed-note", "a.c"]);
1879 assert_eq!(opts.opt_info, ["missed-note"]);
1880
1881 let (opts, _) =
1884 compile(&["-c", "-O2", "-fopt-info-missed=one.txt", "-fopt-info-all=two.txt", "a.c"]);
1885 assert_eq!(opts.opt_info, ["missed", "all"]);
1886 assert_eq!(opts.opt_info_file.as_deref(), Some("two.txt"));
1887
1888 let e = parse_args(&args(&["-fopt-info-vectorized", "a.c"])).unwrap_err();
1889 assert!(e.message.contains("vectorized"), "{}", e.message);
1890 assert!(e.message.contains("`missed`"), "{}", e.message);
1891 let e = parse_args(&args(&["-fopt-info-missed=", "a.c"])).unwrap_err();
1892 assert!(e.message.contains("no file"), "{}", e.message);
1893 }
1894
1895 #[test]
1896 fn verify_each_is_unstable_and_off_unless_it_was_asked_for() {
1897 let (opts, _) = compile(&["-c", "-Zverify-each", "a.c"]);
1898 assert!(opts.verify_each);
1899 assert!(!USAGE.contains("verify-each"), "an unstable option stays out of the usage text");
1900 }
1901
1902 #[test]
1903 fn dash_o_needs_an_argument() {
1904 let e = parse_args(&args(&["a.c", "-o"])).unwrap_err();
1905 assert_eq!(e.message, "-o requires an argument");
1906 }
1907
1908 #[test]
1909 fn dash_d_and_dash_u_are_read_joined_or_separated_and_keep_their_order() {
1910 let (opts, _) = compile(&["-DFOO=1", "-D", "BAR", "-UBAZ", "-U", "QUX", "a.c"]);
1911 assert_eq!(opts.defines, ["FOO=1", "BAR"]);
1912 assert_eq!(opts.undefines, ["BAZ", "QUX"]);
1913 }
1914
1915 #[test]
1916 fn the_include_flags_land_on_the_chain_each_one_names() {
1917 let (opts, _) = compile(&[
1920 "-Ii",
1921 "-iquote",
1922 "q",
1923 "-isystem",
1924 "sys",
1925 "-idirafter",
1926 "after",
1927 "--sysroot=/nowhere-at-all",
1928 "a.c",
1929 ]);
1930 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1931 assert_eq!(dirs, ["q", "i", "sys", runtime::DIR, "after"]);
1934 assert!(!opts.search.dirs()[1].is_system);
1935 assert!(opts.search.dirs()[2].is_system);
1936 }
1937
1938 #[test]
1939 fn the_librarys_headers_come_after_the_compilers_own_and_go_away_with_them() {
1940 let (opts, _) = compile(&["a.c"]);
1944 let dirs = opts.search.dirs();
1945 let ours = dirs.iter().position(|d| d.path.to_str() == Some(runtime::DIR));
1946 assert_eq!(ours, Some(0), "{dirs:?}");
1947 assert!(dirs[1..].iter().all(|d| d.is_system), "{dirs:?}");
1948 let (bare, _) = compile(&["-nostdinc", "a.c"]);
1949 assert!(bare.search.dirs().is_empty(), "{:?}", bare.search.dirs());
1950 }
1951
1952 #[test]
1953 fn a_sysroot_moves_the_librarys_directories_and_nothing_else() {
1954 let (opts, _) = compile(&["-isystem", "sys", "--sysroot=/nowhere-at-all", "a.c"]);
1955 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1956 assert_eq!(dirs, ["sys", runtime::DIR]);
1957 }
1958
1959 #[test]
1960 fn dash_i_dash_moves_the_bracket_directories_into_the_quoted_chain() {
1961 let (opts, _) =
1962 compile(&["-Iinc1", "-iquote", "inc2", "-I-", "-Iinc3", "-nostdinc", "a.c"]);
1963 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1964 assert_eq!(dirs, ["inc1", "inc2", "inc3"]);
1965 assert_eq!(opts.search.start(IncludeForm::Angled), 2);
1967 assert!(!opts.search.searches_current_dir());
1968 }
1969
1970 #[test]
1971 fn the_prefix_flags_stick_what_iprefix_said_on_the_front_of_what_follows_it() {
1972 let (opts, _) = compile(&[
1973 "-iprefix",
1974 "/tools/",
1975 "-iwithprefix",
1976 "late",
1977 "-iwithprefixbefore",
1978 "early",
1979 "-iprefix",
1980 "/other/",
1981 "-iwithprefix",
1982 "last",
1983 "-nostdinc",
1984 "a.c",
1985 ]);
1986 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1987 assert_eq!(dirs, ["/tools/early", "/tools/late", "/other/last"]);
1990 assert!(!opts.search.dirs()[0].is_system);
1991 assert!(opts.search.dirs()[1].is_system);
1992 }
1993
1994 #[test]
1995 fn the_files_named_on_the_command_line_keep_their_order_and_which_flag_named_them() {
1996 let (opts, _) =
1997 compile(&["-include", "one.h", "-imacros", "two.h", "-include", "3.h", "a.c"]);
1998 let names: Vec<&str> = opts.preincludes.iter().map(|p| p.name.as_str()).collect();
1999 assert_eq!(names, ["one.h", "two.h", "3.h"]);
2000 assert_eq!(opts.preincludes.iter().filter(|p| p.macros_only).count(), 1);
2001 }
2002
2003 #[test]
2004 fn nostdinc_takes_the_compilers_own_headers_off_the_path() {
2005 let (opts, _) = compile(&["-Ii", "-nostdinc", "a.c"]);
2006 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
2007 assert_eq!(dirs, ["i"]);
2008 }
2009
2010 #[test]
2011 fn the_dialect_flags_set_the_language_and_the_extensions_separately() {
2012 let (opts, _) = compile(&["-std=gnu11", "a.c"]);
2013 assert_eq!(opts.std, Std::C11);
2014 assert!(opts.gnu_extensions);
2015
2016 let (opts, _) = compile(&["-std=iso9899:1999", "a.c"]);
2017 assert_eq!(opts.std, Std::C99);
2018 assert!(!opts.gnu_extensions);
2019
2020 let (opts, _) = compile(&["-ansi", "a.c"]);
2021 assert_eq!(opts.std, Std::C89);
2022 assert!(!opts.gnu_extensions);
2023
2024 let e = parse_args(&args(&["-std=c94jr", "a.c"])).unwrap_err();
2025 assert!(e.message.contains("unknown dialect"), "{}", e.message);
2026 }
2027
2028 #[test]
2029 fn the_dump_letters_are_a_family_and_everything_else_beginning_with_d_is_not() {
2030 let (opts, _) = compile(&["-dM", "a.c"]);
2031 assert!(opts.dumps.macros);
2032
2033 let (opts, _) = compile(&["-dDM", "a.c"]);
2036 assert!(opts.dumps.macros);
2037 let (opts, _) = compile(&["-dD", "a.c"]);
2038 assert!(!opts.dumps.macros);
2039
2040 let (opts, _) = compile(&["a.c"]);
2041 assert!(!opts.dumps.any());
2042
2043 assert_eq!(printed(&["-dumpversion", "a.c"]), VERSION);
2046 }
2047
2048 #[test]
2049 fn the_gcc_version_claimed_is_a_flag_and_the_short_spellings_are_the_ones_people_write() {
2050 let (opts, _) = compile(&["a.c"]);
2051 assert_eq!(
2052 opts.gnuc,
2053 GnucVersion { major: 7, minor: 0, patch: 0 },
2054 "the lowest claim a modern glibc gives its own declarations to"
2055 );
2056
2057 let (opts, _) = compile(&["-fgnuc-version=15.1.0", "a.c"]);
2058 assert_eq!(opts.gnuc, GnucVersion { major: 15, minor: 1, patch: 0 });
2059
2060 let (opts, _) = compile(&["-fgnuc-version=15", "a.c"]);
2063 assert_eq!(opts.gnuc, GnucVersion { major: 15, minor: 0, patch: 0 });
2064
2065 let (opts, _) = compile(&["-fgnuc-version=13.2", "a.c"]);
2066 assert_eq!(opts.gnuc, GnucVersion { major: 13, minor: 2, patch: 0 });
2067
2068 let e = parse_args(&args(&["-fgnuc-version=15.x", "a.c"])).unwrap_err();
2069 assert!(e.message.contains("minor that is not a number"), "{}", e.message);
2070
2071 let e = parse_args(&args(&["-fgnuc-version=1.2.3.4", "a.c"])).unwrap_err();
2072 assert!(e.message.contains("more than three"), "{}", e.message);
2073 }
2074
2075 #[test]
2076 fn pedantic_has_two_spellings_and_is_not_the_same_knob_as_the_dialect() {
2077 let (opts, _) = compile(&["-std=c17", "-pedantic", "a.c"]);
2078 assert!(opts.pedantic);
2079 assert_eq!(opts.std, Std::C17);
2080
2081 let (opts, _) = compile(&["-Wpedantic", "a.c"]);
2084 assert!(opts.pedantic);
2085
2086 let (opts, _) = compile(&["-std=c17", "a.c"]);
2087 assert!(!opts.pedantic, "a dialect on its own does not diagnose an extension");
2088 }
2089
2090 #[test]
2091 fn dash_p_and_dash_ffreestanding_reach_the_options() {
2092 let (opts, _) = compile(&["-E", "-P", "-ffreestanding", "a.c"]);
2093 assert!(!opts.line_markers);
2094 assert!(!opts.hosted);
2095 assert_eq!(opts.emit, EmitKind::Preprocessed);
2096 }
2097
2098 #[test]
2105 fn the_builtin_flags_are_read_in_both_directions_and_one_name_at_a_time() {
2106 let (opts, _) = compile(&["-c", "a.c"]);
2107 assert!(opts.builtins, "a library name means the library function by default");
2108 assert!(opts.no_builtin.is_empty());
2109
2110 let (opts, _) = compile(&["-c", "-fno-builtin", "a.c"]);
2111 assert!(!opts.builtins);
2112
2113 let (opts, _) = compile(&["-c", "-fno-builtin", "-fbuiltin", "a.c"]);
2114 assert!(opts.builtins, "the last mention decides");
2115
2116 let (opts, _) = compile(&["-c", "-fno-builtin-memcpy", "-fno-builtin-nonesuch", "a.c"]);
2117 assert!(opts.builtins, "one name is not the family");
2118 assert_eq!(opts.no_builtin, vec!["memcpy".to_owned(), "nonesuch".to_owned()]);
2119 }
2120
2121 #[test]
2124 fn gnu89_inline_is_off_until_it_is_asked_for_and_the_last_mention_decides() {
2125 let (opts, _) = compile(&["-c", "a.c"]);
2126 assert!(!opts.gnu89_inline, "C's reading of inline by default");
2127
2128 let (opts, _) = compile(&["-c", "-fgnu89-inline", "a.c"]);
2129 assert!(opts.gnu89_inline);
2130
2131 let (opts, _) = compile(&["-c", "-fgnu89-inline", "-fno-gnu89-inline", "a.c"]);
2132 assert!(!opts.gnu89_inline, "the last mention decides");
2133
2134 let (opts, _) = compile(&["-c", "-std=c89", "a.c"]);
2139 assert!(!opts.gnu89_inline);
2140 }
2141
2142 #[test]
2145 fn the_two_frame_flags_are_read_in_both_directions() {
2146 let (opts, _) = compile(&["-c", "a.c"]);
2147 assert!(!opts.frame_pointer, "gcc omits it above -O0 and so does this");
2148 assert!(opts.red_zone, "the psABI has one and nothing said not to use it");
2149
2150 let (opts, _) = compile(&["-c", "-fno-omit-frame-pointer", "-mno-red-zone", "a.c"]);
2151 assert!(opts.frame_pointer);
2152 assert!(!opts.red_zone);
2153
2154 let (opts, _) = compile(&[
2155 "-c",
2156 "-fno-omit-frame-pointer",
2157 "-fomit-frame-pointer",
2158 "-mno-red-zone",
2159 "-mred-zone",
2160 "a.c",
2161 ]);
2162 assert!(!opts.frame_pointer, "the last one wins, as it does in gcc");
2163 assert!(opts.red_zone);
2164 }
2165
2166 #[test]
2167 fn the_link_flags_are_collected_apart_from_the_compilation() {
2168 let (link, _) = linking(&[
2169 "-static",
2170 "-nostartfiles",
2171 "-rdynamic",
2172 "-s",
2173 "-fuse-ld=mold",
2174 "-L/opt/lib",
2175 "-B",
2176 "/opt/tools",
2177 "a.c",
2178 ]);
2179 assert!(link.is_static);
2180 assert!(link.no_startfiles);
2181 assert!(link.export_dynamic);
2182 assert!(link.strip);
2183 assert_eq!(link.use_ld.as_deref(), Some("mold"));
2184 assert_eq!(link.search, vec![PathBuf::from("/opt/lib")]);
2185 assert_eq!(link.prefixes, vec![PathBuf::from("/opt/tools")]);
2186 }
2187
2188 #[test]
2189 fn a_comma_in_dash_wl_separates_two_arguments() {
2190 let (link, _) = linking(&["-Wl,-rpath,/opt/lib", "-Xlinker", "--as-needed", "a.c"]);
2191 assert_eq!(link.passthrough, vec!["-rpath", "/opt/lib", "--as-needed"]);
2192 }
2193
2194 #[test]
2195 fn a_library_keeps_its_place_between_the_objects() {
2196 let (_, plan) = linking(&["--target=x86_64-unknown-linux-gnu", "a.c", "-lm", "b.c"]);
2201 let link = plan.link.expect("expected a link step");
2202 assert_eq!(
2203 link.inputs,
2204 vec![
2205 link::Item::File("a.o".into()),
2206 link::Item::Library("m".into()),
2207 link::Item::File("b.o".into()),
2208 ]
2209 );
2210 assert_eq!(plan.jobs.len(), 2);
2212 }
2213
2214 #[test]
2215 fn a_library_on_a_dash_c_line_is_a_note_rather_than_an_error() {
2216 let (_, plan) = linking(&["-c", "-lm", "a.c"]);
2217 assert!(plan.link.is_none());
2218 assert!(plan.notes.iter().any(|n| n.contains("-lm")), "{:?}", plan.notes);
2219 }
2220
2221 #[test]
2222 fn the_sysroot_reaches_the_linker_as_well_as_the_headers() {
2223 let (link, _) = linking(&["--sysroot=/opt/root", "a.c"]);
2224 assert_eq!(link.sysroot, Some(PathBuf::from("/opt/root")));
2225 }
2226
2227 fn printed(s: &[&str]) -> String {
2228 match parse_args(&args(s)).expect("expected an answer") {
2229 Action::Print(line) => line,
2230 other => panic!("expected an answer, got {other:?}"),
2231 }
2232 }
2233
2234 fn refused(s: &[&str]) -> String {
2235 parse_args(&args(s)).expect_err("expected a refusal").message
2236 }
2237
2238 #[test]
2239 fn a_warning_flag_this_compiler_has_not_heard_of_is_taken_rather_than_refused() {
2240 let (opts, _) = compile(&["-Wall", "-Wextra", "-Wno-format-truncation", "-c", "a.c"]);
2244 assert!(!opts.warnings_are_errors);
2245 assert!(opts.warnings);
2246 let (opts, _) = compile(&["-Werror", "-c", "a.c"]);
2248 assert!(opts.warnings_are_errors);
2249 let (opts, _) = compile(&["-w", "-c", "a.c"]);
2250 assert!(!opts.warnings);
2251 let (opts, _) = compile(&["-pedantic-errors", "-c", "a.c"]);
2252 assert!(opts.pedantic && opts.warnings_are_errors);
2253 }
2254
2255 #[test]
2256 fn an_argument_for_a_separate_tool_is_refused_rather_than_dropped() {
2257 assert!(refused(&["-Wa,--noexecstack", "-c", "a.c"]).contains("separate assembler"));
2259 assert!(refused(&["-Wp,-DX", "-c", "a.c"]).contains("separate assembler"));
2260 assert!(refused(&["-specs=/x", "a.c"]).contains("-specs= is not supported"));
2261 assert!(refused(&["-mcmodel=kernel", "-c", "a.c"]).contains("small code model"));
2262 assert!(refused(&["-gdwarf-4", "-c", "a.c"]).contains("DWARF 5"));
2263 assert!(refused(&["-Ofast", "-c", "a.c"]).contains("fast math"));
2264 let no32 = refused(&["--target=x86_64-unknown-linux-gnu", "-m32", "-c", "a.c"]);
2267 assert!(no32.contains("32 bit target"), "{no32}");
2268 }
2269
2270 #[test]
2271 fn the_levels_gcc_spells_differently_are_the_levels_they_mean() {
2272 assert_eq!(compile(&["-O", "-c", "a.c"]).0.opt_level, OptLevel::O1);
2273 assert_eq!(compile(&["-Og", "-c", "a.c"]).0.opt_level, OptLevel::O1);
2274 assert_eq!(compile(&["-O2", "-c", "a.c"]).0.opt_level, OptLevel::O2);
2275 }
2276
2277 #[test]
2278 fn the_machine_flags_that_name_what_we_already_do_are_taken_and_the_rest_are_not() {
2279 let line = ["--target=x86_64-unknown-linux-gnu", "-m64", "-march=x86-64-v3"];
2280 let (opts, _) =
2281 compile(&[&line[..], &["-mtune=native", "-mabi=sysv", "-c", "a.c"]].concat());
2282 assert_eq!(opts.target.to_string(), "x86_64-unknown-linux-gnu");
2283 let wrong = refused(&["--target=x86_64-unknown-linux-gnu", "-mabi=ms", "-c", "a.c"]);
2284 assert!(wrong.contains("sysv convention"), "{wrong}");
2285 }
2286
2287 #[test]
2288 fn the_thread_flag_is_a_macro_and_a_library_and_the_library_goes_last() {
2289 let (opts, plan) = compile(&["-pthread", "-c", "a.c"]);
2290 assert!(opts.defines.iter().any(|d| d == "_REENTRANT"));
2291 let names: Vec<&str> = plan.jobs.iter().map(|j| j.input.as_str()).collect();
2294 assert_eq!(names, vec!["a.c"]);
2295 }
2296
2297 #[test]
2298 fn the_questions_a_build_system_asks_before_it_compiles_anything() {
2299 let target = "--target=x86_64-unknown-linux-gnu";
2300 assert_eq!(printed(&[target, "-dumpmachine"]), "x86_64-unknown-linux-gnu");
2301 assert_eq!(printed(&[target, "-dumpversion"]), VERSION);
2302 assert_eq!(printed(&[target, "-dumpfullversion"]), VERSION);
2303 assert_eq!(printed(&[target, "-print-multiarch"]), "x86_64-linux-gnu");
2304 assert_eq!(printed(&[target, "-print-file-name=no-such-library.a"]), "no-such-library.a");
2307 assert_eq!(printed(&[target, "-print-prog-name=ld"]), "ld");
2308 let dirs = printed(&[target, "-print-search-dirs"]);
2309 assert!(dirs.starts_with("install: "), "{dirs}");
2310 assert!(dirs.contains("\nlibraries: ="), "{dirs}");
2311 }
2312
2313 #[test]
2314 fn the_two_dependency_flags_that_stop_after_the_rule_stop_after_the_rule() {
2315 let (opts, _) = compile(&["-M", "a.c"]);
2316 assert!(opts.deps.emit && opts.deps.instead_of_compiling);
2317 assert!(opts.deps.system_headers, "plain -M lists them");
2318 assert_eq!(opts.emit, EmitKind::Preprocessed);
2319
2320 let (opts, _) = compile(&["-M", "-c", "a.c"]);
2323 assert_eq!(opts.emit, EmitKind::Preprocessed);
2324
2325 let (opts, _) = compile(&["-MM", "a.c"]);
2326 assert!(!opts.deps.system_headers);
2327 }
2328
2329 #[test]
2330 fn the_two_that_end_in_d_leave_the_compilation_alone() {
2331 let (opts, _) = compile(&["-MD", "-c", "a.c"]);
2332 assert!(opts.deps.emit && !opts.deps.instead_of_compiling);
2333 assert!(opts.deps.system_headers);
2334 assert_eq!(opts.emit, EmitKind::Object);
2335
2336 let (opts, _) = compile(&["-MMD", "-c", "a.c"]);
2337 assert!(opts.deps.emit && !opts.deps.instead_of_compiling);
2338 assert!(!opts.deps.system_headers);
2339 }
2340
2341 #[test]
2342 fn nothing_puts_the_system_headers_back_once_a_flag_has_taken_them_out() {
2343 let (opts, _) = compile(&["-MM", "-M", "a.c"]);
2346 assert!(!opts.deps.system_headers);
2347 let (opts, _) = compile(&["-MD", "-MMD", "-c", "a.c"]);
2348 assert!(!opts.deps.system_headers);
2349 let (opts, _) = compile(&["-MMD", "-MD", "-c", "a.c"]);
2350 assert!(!opts.deps.system_headers);
2351 }
2352
2353 #[test]
2354 fn a_target_arrives_escaped_from_one_flag_and_untouched_from_the_other() {
2355 let (opts, _) = compile(&["-MM", "-MT", "a b.o", "-MQ", "a b.o", "a.c"]);
2356 assert_eq!(opts.deps.targets, vec!["a b.o".to_owned(), "a\\ b.o".to_owned()]);
2357 }
2358
2359 #[test]
2360 fn the_rest_of_the_family_is_a_file_and_a_switch() {
2361 let (opts, _) = compile(&["-MM", "-MF", "dep.d", "-MP", "a.c"]);
2362 assert_eq!(opts.deps.file.as_deref(), Some("dep.d"));
2363 assert!(opts.deps.phony);
2364
2365 for flag in ["-MF", "-MT", "-MQ"] {
2366 let e = parse_args(&args(&[flag])).unwrap_err();
2367 assert!(e.message.contains("requires an argument"), "{}", e.message);
2368 }
2369 }
2370
2371 struct TempTree(PathBuf);
2373
2374 impl Drop for TempTree {
2375 fn drop(&mut self) {
2376 let _ = std::fs::remove_dir_all(&self.0);
2377 }
2378 }
2379
2380 impl TempTree {
2381 fn new(name: &str, files: &[(&str, &str)]) -> TempTree {
2382 let dir = std::env::temp_dir().join(format!("rucc-deps-{}-{name}", std::process::id()));
2383 let _ = std::fs::remove_dir_all(&dir);
2384 std::fs::create_dir_all(&dir).expect("temporary directory should be writable");
2385 for (path, text) in files {
2386 let at = dir.join(path);
2387 if let Some(parent) = at.parent() {
2388 std::fs::create_dir_all(parent).expect("creating a subdirectory should work");
2389 }
2390 std::fs::write(&at, text).expect("writing a temporary file should work");
2391 }
2392 TempTree(dir)
2393 }
2394
2395 fn path(&self, name: &str) -> String {
2396 self.0.join(name).to_string_lossy().into_owned()
2397 }
2398 }
2399
2400 #[test]
2401 fn the_rule_names_what_the_includes_found_and_names_each_of_them_once() {
2402 let tree = TempTree::new(
2406 "found",
2407 &[
2408 ("a.c", "#include \"one.h\"\n#include \"two.h\"\nint main(void) { return X; }\n"),
2409 ("one.h", "#define X 0\n"),
2410 ("two.h", "#include \"one.h\"\n"),
2411 ],
2412 );
2413 let out = tree.path("dep.d");
2414 let code = run(&args(&["-MM", "-MF", &out, "-o", &tree.path("a.i"), &tree.path("a.c")]));
2415 assert_eq!(code, 0);
2416
2417 let text = std::fs::read_to_string(&out).expect("the rule should have been written");
2418 let names: Vec<&str> = text.split_whitespace().collect();
2419 assert_eq!(names.first(), Some(&"a.o:"), "{text}");
2421 assert_eq!(names.iter().filter(|n| n.ends_with("one.h")).count(), 1, "{text}");
2422 assert_eq!(names.iter().filter(|n| n.ends_with("two.h")).count(), 1, "{text}");
2423 assert_eq!(std::fs::read(tree.path("a.i")).expect("the output should exist"), b"");
2426 }
2427
2428 #[test]
2429 fn a_header_that_is_only_reached_under_a_guard_is_still_a_dependency() {
2430 let tree = TempTree::new(
2433 "guarded",
2434 &[
2435 ("a.c", "#include \"g.h\"\n#include \"g.h\"\nint main(void) { return 0; }\n"),
2436 ("g.h", "#ifndef G\n#define G\n#endif\n"),
2437 ],
2438 );
2439 let out = tree.path("dep.d");
2440 let code = run(&args(&["-MM", "-MF", &out, "-o", &tree.path("a.i"), &tree.path("a.c")]));
2441 assert_eq!(code, 0);
2442 let text = std::fs::read_to_string(&out).expect("the rule should have been written");
2443 assert_eq!(text.split_whitespace().filter(|n| n.ends_with("g.h")).count(), 1, "{text}");
2444 }
2445
2446 #[test]
2447 fn every_imacros_file_is_read_before_every_include_file_whatever_order_they_were_written() {
2448 let tree = TempTree::new(
2453 "preinclude",
2454 &[
2455 ("a.c", "int main(void) { return 0; }\n"),
2456 ("i.h", "#ifdef FROM_MACROS\nint saw_it;\n#else\nint missed_it;\n#endif\n"),
2457 ("m.h", "#define FROM_MACROS 1\nint macros_text;\n"),
2458 ],
2459 );
2460 let out = tree.path("a.i");
2461 let code = run(&args(&[
2462 "-E",
2463 "-include",
2464 &tree.path("i.h"),
2465 "-imacros",
2466 &tree.path("m.h"),
2467 "-o",
2468 &out,
2469 &tree.path("a.c"),
2470 ]));
2471 assert_eq!(code, 0);
2472 let text = std::fs::read_to_string(&out).expect("the output should have been written");
2473 assert!(text.contains("saw_it"), "{text}");
2474 assert!(!text.contains("macros_text"), "{text}");
2477 }
2478
2479 #[test]
2480 fn a_file_the_command_line_named_is_a_prerequisite_the_same_as_one_a_directive_named() {
2481 let tree = TempTree::new(
2482 "preinclude-deps",
2483 &[
2484 ("a.c", "int main(void) { return 0; }\n"),
2485 ("i.h", "int from_include;\n"),
2486 ("m.h", "#define M 1\n"),
2487 ],
2488 );
2489 let out = tree.path("dep.d");
2490 let code = run(&args(&[
2491 "-MM",
2492 "-MF",
2493 &out,
2494 "-include",
2495 &tree.path("i.h"),
2496 "-imacros",
2497 &tree.path("m.h"),
2498 "-o",
2499 &tree.path("a.i"),
2500 &tree.path("a.c"),
2501 ]));
2502 assert_eq!(code, 0);
2503 let text = std::fs::read_to_string(&out).expect("the rule should have been written");
2504 assert!(text.contains("i.h"), "{text}");
2505 assert!(text.contains("m.h"), "{text}");
2506 }
2507
2508 #[test]
2509 fn a_command_line_include_that_is_nowhere_on_the_path_is_an_error_and_not_a_warning() {
2510 let tree = TempTree::new(
2514 "preinclude-missing",
2515 &[("sub/a.c", "int main(void) { return 0; }\n"), ("sub/beside.h", "int x;\n")],
2516 );
2517 let code = run(&args(&["-E", "-include", "beside.h", "-o", "-", &tree.path("sub/a.c")]));
2518 assert_eq!(code, 1);
2519 }
2520
2521 #[test]
2522 fn a_command_line_that_links_names_the_executable_and_not_the_object_it_went_through() {
2523 let (opts, plan) = compile(&["-MD", "sub/a.c", "-o", "prog"]);
2527 assert_eq!(plan.output.as_deref(), Some("prog"));
2528 assert_eq!(deps::default_target("sub/a.c", deps_target_output(&opts, &plan)), "prog");
2529 assert_eq!(
2530 deps::default_file(&opts.deps, "sub/a.c", plan.output.as_deref()).as_deref(),
2531 Some("prog.d")
2532 );
2533 }
2534
2535 #[test]
2536 fn the_plan_keeps_the_output_name_because_the_rule_is_written_from_it() {
2537 let (_, plan) = compile(&["-MMD", "-c", "sub/a.c", "-o", "obj/x.o"]);
2538 assert_eq!(plan.output.as_deref(), Some("obj/x.o"));
2539 let (_, plan) = compile(&["-MMD", "-c", "sub/a.c"]);
2540 assert_eq!(plan.output, None);
2541 }
2542
2543 #[test]
2544 fn usage_fits_on_a_screen() {
2545 assert!(USAGE.lines().count() < 47, "usage text has grown past one screen");
2561 }
2562}