1#![doc(html_root_url = "https://docs.rs/rucc-driver/0.9.0")]
29
30pub mod compile;
31pub mod library;
32pub mod link;
33mod map;
34pub mod phase;
35pub mod preprocess;
36pub mod schedule;
37
38use std::fmt::Write as _;
39use std::io::Write as _;
40use std::path::PathBuf;
41
42use rucc_codegen::coverage::{self, Fired};
43use rucc_session::{Dumps, EmitKind, Options, Session, Std, runtime};
44use rucc_target::Triple;
45
46use crate::link::LinkOptions;
47
48pub use crate::compile::{Artifact, Compiled, compile, compile_ir};
49pub use crate::phase::{Input, InputKind, Job, LinkJob, Output, Phase, Plan};
50pub use crate::preprocess::{OsFileSystem, Preprocessed, preprocess};
51pub use crate::schedule::Jobs;
52
53pub const VERSION: &str = env!("CARGO_PKG_VERSION");
55
56#[derive(Debug, Clone, PartialEq, Eq)]
58pub enum Action {
59 Help,
61 Version,
63 Print(String),
69 PrintConfig(Box<Options>),
71 PrintPipeline(Box<Options>),
73 PrintPlan {
75 opts: Box<Options>,
77 plan: Box<Plan>,
79 link: Box<LinkOptions>,
81 },
82 Compile {
84 opts: Box<Options>,
86 plan: Box<Plan>,
88 link: Box<LinkOptions>,
90 jobs: Jobs,
92 verbose: bool,
94 },
95}
96
97#[derive(Debug, Clone, PartialEq, Eq)]
99pub struct CliError {
100 pub message: String,
103}
104
105impl std::fmt::Display for CliError {
106 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
107 f.write_str(&self.message)
108 }
109}
110
111impl std::error::Error for CliError {}
112
113fn err(message: impl Into<String>) -> CliError {
114 CliError { message: message.into() }
115}
116
117enum Query {
123 Machine,
125 Version,
127 Multiarch,
129 SearchDirs,
131 FileName(String),
133 ProgName(String),
135 Libgcc,
137}
138
139pub const USAGE: &str = "\
144rucc, an optimizing C compiler
145
146usage: rucc [options] file...
147
148options:
149 -c compile and assemble, do not link
150 -S compile only, emit assembly
151 -E preprocess only
152 -o <file> write output to <file>, or to standard output for -
153 -D <name>[=<value>], -U <name> define a macro, or undefine one after every -D
154 -I <dir> add <dir> to the include search path
155 -iquote -isystem -idirafter <dir> the other chains, -nostdinc drops ours
156 --sysroot=<dir> look for the library's headers under <dir>, -isysroot too
157 -P, -dM with -E: leave out the markers, or dump the macros
158 -std=<dialect> c89 through c23, and the gnu spellings
159 -fgnuc-version=<v> the GCC release to claim, default 7.0.0
160 -x <lang> treat later inputs as <lang>, or none to stop
161 -O<level> optimize: 0, 1, 2, 3, s, z
162 -fsafety=<tier> check memory safety: off, detect, enforce, kernel
163 -f<pass> -fno-<pass> -fdump-ir=<what> -fopt-info[-<kind>][=FILE]
164 -fpass-fuel=<pass>=<n>, -fpass-fuel-global=<n> stop a pass, or all of them, after n
165 -fdisable-<pass>[=<funcs>], -fenable-<pass>[=<funcs>] run a pass on some functions only
166 -g -g0 -gdwarf-5, -fno-omit-frame-pointer, -mno-red-zone debug info, frame pointer, red zone
167 -l<name>, -L <dir>, -B <dir> link a library, where to look for one, where our own tools are
168 -static -shared -pie -no-pie -nostdlib -nostartfiles -nodefaultlibs -rdynamic -s how to link
169 -Wl,<arg>, -Xlinker <arg>, -fuse-ld=<name> hand an argument to the linker, or pick one
170 -Werror -pedantic -pedantic-errors -w how much to say, and whether it is fatal
171 -m64 -march= -mtune= -mcpu= -mabi= -mcmodel= what machine to generate for
172 -pthread build for more than one thread, and link the library for it
173 -dumpmachine -dumpversion -print-multiarch -print-search-dirs what this compiler is
174 -print-file-name=<name> -print-prog-name=<name> where a file or a program is
175 -j[n] compile n translation units at once, default all
176 -v, -### print each phase as it runs, or without running any
177 --target=<triple> generate code for <triple>
178 --emit=<kind> exe, obj, asm, preprocessed, tast, ir, mir-final,
179 safety-summary, type-granules
180 --print-config, --print-pipeline print the configuration or the pipeline, and exit
181 --version print the version and exit
182 -h, --help print this message and exit
183
184See spec/04-driver-and-cli.md for the full flag reference.
185";
186
187fn joined_or_next(
191 arg: &str,
192 at: usize,
193 args: &[String],
194 i: &mut usize,
195) -> Result<String, CliError> {
196 if arg.len() > at {
197 return Ok(arg[at..].to_owned());
198 }
199 let next = args.get(*i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
200 *i += 1;
201 Ok(next.clone())
202}
203
204pub fn parse_args(args: &[String]) -> Result<Action, CliError> {
211 let host = Triple::host()
212 .ok_or_else(|| err("this host is not a supported target and no --target was given"))?;
213 let mut opts = Options::new(host);
214 let mut inputs: Vec<Input> = Vec::new();
215 let mut print_config = false;
216 let mut print_pipeline = false;
217 let mut print_plan = false;
218 let mut verbose = false;
219 let mut jobs = Jobs::default();
220 let mut nostdinc = false;
221 let mut sysroot: Option<PathBuf> = None;
222 let mut output = None;
223 let mut link = LinkOptions::default();
224 let mut query: Option<Query> = None;
225 let mut threads = false;
226 let mut forced: Option<InputKind> = None;
229
230 let mut i = 0;
231 while i < args.len() {
232 let arg = args[i].as_str();
233 i += 1;
234 match arg {
235 "-h" | "--help" => return Ok(Action::Help),
236 "--version" => return Ok(Action::Version),
237 "--print-config" => print_config = true,
238 "--print-pipeline" => print_pipeline = true,
239 "-###" => print_plan = true,
240 "-v" => verbose = true,
241 "-c" => opts.emit = EmitKind::Object,
242 "-S" => opts.emit = EmitKind::Asm,
243 "-E" => opts.emit = EmitKind::Preprocessed,
244 "-g" => opts.debug_info = true,
245 "-g0" => opts.debug_info = false,
250 "-g1" | "-g2" | "-g3" | "-ggdb" | "-ggdb1" | "-ggdb2" | "-ggdb3" => {
251 opts.debug_info = true;
252 }
253 "-gdwarf" | "-gdwarf-5" => opts.debug_info = true,
256 _ if arg.starts_with("-gdwarf-") => {
257 return Err(err(format!(
258 "{arg}: this compiler writes DWARF 5 and no other version, see \
259 spec/11-debug-info.md"
260 )));
261 }
262 "-Werror" => opts.warnings_are_errors = true,
263 "-w" => opts.warnings = false,
266 "-pedantic-errors" => {
267 opts.pedantic = true;
268 opts.warnings_are_errors = true;
269 }
270 "-P" => opts.line_markers = false,
271 "-dumpmachine" => query = Some(Query::Machine),
275 "-dumpversion" | "-dumpfullversion" => query = Some(Query::Version),
276 "-print-multiarch" => query = Some(Query::Multiarch),
277 "-print-search-dirs" => query = Some(Query::SearchDirs),
278 "-print-libgcc-file-name" => query = Some(Query::Libgcc),
279 _ if arg.starts_with("-print-file-name=") => {
280 query = Some(Query::FileName(arg["-print-file-name=".len()..].to_owned()));
281 }
282 _ if arg.starts_with("-print-prog-name=") => {
283 query = Some(Query::ProgName(arg["-print-prog-name=".len()..].to_owned()));
284 }
285 "-pthread" | "-pthreads" => {
290 opts.defines.push("_REENTRANT".to_owned());
291 threads = true;
292 }
293 "-ansi" => {
294 opts.std = Std::C89;
295 opts.gnu_extensions = false;
296 }
297 "-pedantic" | "-Wpedantic" => opts.pedantic = true,
300 "-fpermissive" => opts.permissive = true,
303 "-fno-permissive" => opts.permissive = false,
304 "-ffreestanding" => opts.hosted = false,
305 "-fhosted" => opts.hosted = true,
306 "-fno-builtin" => opts.builtins = false,
307 "-fbuiltin" => opts.builtins = true,
308 "-fgnu89-inline" => opts.gnu89_inline = true,
312 "-fno-gnu89-inline" => opts.gnu89_inline = false,
313 "-fno-omit-frame-pointer" => opts.frame_pointer = true,
316 "-fomit-frame-pointer" => opts.frame_pointer = false,
317 "-mno-red-zone" => opts.red_zone = false,
318 "-mred-zone" => opts.red_zone = true,
319 "-nostdinc" => nostdinc = true,
323 "-o" => {
324 output = Some(args.get(i).ok_or_else(|| err("-o requires an argument"))?.clone());
325 i += 1;
326 }
327 "-isysroot" => {
334 let dir = args.get(i).ok_or_else(|| err("-isysroot requires an argument"))?;
335 i += 1;
336 sysroot = Some(PathBuf::from(dir));
337 }
338 "-iquote" | "-isystem" | "-idirafter" => {
339 let dir = args.get(i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
340 i += 1;
341 match arg {
342 "-iquote" => opts.search.push_quote(dir.clone()),
343 "-isystem" => opts.search.push_system(dir.clone()),
344 _ => opts.search.push_after(dir.clone()),
345 }
346 }
347 "-x" => {
348 let lang = args.get(i).ok_or_else(|| err("-x requires an argument"))?;
349 i += 1;
350 forced = if lang == "none" {
351 None
352 } else {
353 Some(InputKind::from_x_arg(lang).map_err(|e| err(format!("{e}")))?)
354 };
355 }
356 _ if arg.starts_with("-D") => {
364 let value = joined_or_next(arg, 2, args, &mut i)?;
365 opts.defines.push(value);
366 }
367 _ if arg.starts_with("-U") => {
368 let value = joined_or_next(arg, 2, args, &mut i)?;
369 opts.undefines.push(value);
370 }
371 _ if arg.starts_with("-I") => {
372 let dir = joined_or_next(arg, 2, args, &mut i)?;
373 opts.search.push_bracket(dir);
374 }
375 _ if arg.starts_with("-std=") => {
376 let name = &arg["-std=".len()..];
377 let (std, gnu) = Std::from_flag(name)
378 .ok_or_else(|| err(format!("unknown dialect `{name}`, see --help")))?;
379 opts.std = std;
380 opts.gnu_extensions = gnu;
381 }
382 _ if Dumps::is_family(arg) => {
391 opts.dumps.add(&arg[2..]);
392 }
393 _ if arg.starts_with("-fno-builtin-") => {
398 opts.no_builtin.push(arg["-fno-builtin-".len()..].to_owned());
399 }
400 _ if arg.starts_with("-fgnuc-version=") => {
401 let v = &arg["-fgnuc-version=".len()..];
402 opts.gnuc = v.parse().map_err(err)?;
403 }
404 "-fnested-functions" => {
409 return Err(err(
410 "nested functions are not supported: a call to one goes through a trampoline \
411 written on the stack, which no target that enforces an unexecutable stack \
412 allows",
413 ));
414 }
415 "-fno-nested-functions" => {}
416 "-static" => link.is_static = true,
420 "-shared" => link.shared = true,
421 "-pie" => link.pie = Some(true),
422 "-no-pie" | "-nopie" => link.pie = Some(false),
423 "-nostdlib" => link.no_stdlib = true,
424 "-nostartfiles" => link.no_startfiles = true,
425 "-nodefaultlibs" => link.no_defaultlibs = true,
426 "-fno-builtins-lib" => link.no_builtins_lib = true,
427 "-fbuiltins-lib" => link.no_builtins_lib = false,
428 "-rdynamic" | "-export-dynamic" => link.export_dynamic = true,
429 "-s" => link.strip = true,
430 "-Xlinker" => {
431 let next = args.get(i).ok_or_else(|| err("-Xlinker requires an argument"))?;
432 i += 1;
433 link.passthrough.push(next.clone());
434 }
435 _ if arg.starts_with("-Wl,") => {
436 link.passthrough.extend(arg["-Wl,".len()..].split(',').map(str::to_owned));
439 }
440 _ if arg.starts_with("-fuse-ld=") => {
441 link.use_ld = Some(arg["-fuse-ld=".len()..].to_owned());
442 }
443 _ if arg.starts_with("-l") && arg.len() > 2 => {
444 inputs.push(Input::library(&arg[2..]));
445 }
446 "-l" => {
447 let next = args.get(i).ok_or_else(|| err("-l requires an argument"))?;
448 i += 1;
449 inputs.push(Input::library(next));
450 }
451 _ if arg.starts_with("-L") => {
452 link.search.push(PathBuf::from(joined_or_next(arg, 2, args, &mut i)?));
453 }
454 _ if arg.starts_with("-B") => {
455 link.prefixes.push(PathBuf::from(joined_or_next(arg, 2, args, &mut i)?));
456 }
457 _ if arg.starts_with("-j") => {
458 jobs = Jobs::parse(&arg[2..]).map_err(err)?;
459 }
460 _ if arg.starts_with("--sysroot=") => {
461 sysroot = Some(PathBuf::from(&arg["--sysroot=".len()..]));
462 }
463 _ if arg.starts_with("--target=") => {
464 let t = &arg["--target=".len()..];
465 opts.target = t.parse().map_err(|e| err(format!("{e}")))?;
466 }
467 _ if arg.starts_with("--emit=") => {
468 let k = &arg["--emit=".len()..];
469 opts.emit = k
470 .parse()
471 .map_err(|()| err(format!("unknown --emit kind `{k}`, see --help")))?;
472 }
473 "-O" | "-Og" => opts.opt_level = rucc_session::OptLevel::O1,
479 "-Ofast" => {
485 return Err(err(
486 "-Ofast is -O3 with fast math, and fast math is not implemented, see \
487 spec/04-driver-and-cli.md section 4.6",
488 ));
489 }
490 _ if arg.starts_with("-O") => {
491 opts.opt_level = arg[2..]
492 .parse()
493 .map_err(|()| err(format!("unknown optimization level `{arg}`")))?;
494 }
495 _ if arg.starts_with("-fsafety=") => {
500 let tier = &arg["-fsafety=".len()..];
501 opts.safety = tier.parse().map_err(|()| {
502 err(format!(
503 "`{tier}` is not a safety tier, which is off, detect, enforce or kernel"
504 ))
505 })?;
506 }
507 _ if arg.starts_with("-fpass-fuel=") => {
511 let (name, count) = arg["-fpass-fuel=".len()..]
512 .split_once('=')
513 .ok_or_else(|| err("-fpass-fuel= is spelled <pass>=<count>"))?;
514 if rucc_opt::pass::find(name).is_none() {
515 return Err(err(format!(
516 "`{name}` is not a pass this compiler has, see --print-pipeline"
517 )));
518 }
519 let count: u32 = count
520 .parse()
521 .map_err(|_| err(format!("`{count}` is not a number of transformations")))?;
522 opts.pass_fuel.push((name.to_owned(), count));
523 }
524 _ if arg.starts_with("-fpass-fuel-global=") => {
525 let count = &arg["-fpass-fuel-global=".len()..];
526 let count: u32 = count
527 .parse()
528 .map_err(|_| err(format!("`{count}` is not a number of transformations")))?;
529 opts.pass_fuel_global = Some(count);
530 }
531 _ if arg == "-fopt-info"
536 || arg.starts_with("-fopt-info=")
537 || arg.starts_with("-fopt-info-") =>
538 {
539 let rest = &arg["-fopt-info".len()..];
540 let (kinds, file) = match rest.split_once('=') {
541 Some((kinds, file)) => (kinds, Some(file)),
542 None => (rest, None),
543 };
544 let kinds = kinds.strip_prefix('-').unwrap_or(kinds);
545 rucc_opt::Wants::none().add(kinds).map_err(err)?;
546 opts.opt_info.push(kinds.to_owned());
547 if let Some(file) = file {
548 if file.is_empty() {
549 return Err(err("-fopt-info= was given no file to write to"));
550 }
551 opts.opt_info_file = Some(file.to_owned());
552 }
553 }
554 _ if arg.starts_with("-fdump-ir=") => {
555 let spec = &arg["-fdump-ir=".len()..];
558 rucc_opt::Dumps::default().add(spec).map_err(err)?;
559 opts.dump_ir.push(spec.to_owned());
560 }
561 _ if arg.starts_with("-fdisable-") || arg.starts_with("-fenable-") => {
567 let on = arg.starts_with("-fenable-");
568 let spec = &arg[if on { "-fenable-".len() } else { "-fdisable-".len() }..];
569 rucc_opt::Gates::default().add(on, spec).map_err(err)?;
570 opts.pass_gates.push((on, spec.to_owned()));
571 }
572 _ if arg.strip_prefix("-fno-").is_some_and(|n| rucc_opt::pass::find(n).is_some()) => {
573 opts.passes.push((arg["-fno-".len()..].to_owned(), false));
574 }
575 _ if arg.strip_prefix("-f").is_some_and(|n| rucc_opt::pass::find(n).is_some()) => {
576 opts.passes.push((arg["-f".len()..].to_owned(), true));
577 }
578 "-Zverify-each" => opts.verify_each = true,
584 _ if arg.starts_with("-Zrule-coverage=") => {
585 let file = &arg["-Zrule-coverage=".len()..];
586 if file.is_empty() {
587 return Err(err("-Zrule-coverage= needs a file to write to"));
588 }
589 opts.rule_coverage = Some(file.to_owned());
590 }
591 _ if arg.starts_with("-Z") => {
592 return Err(err(format!(
593 "`{arg}` is not an unstable option this compiler has, see \
594 spec/04-driver-and-cli.md section 4.11 for the ones it does"
595 )));
596 }
597 "-m64" | "-m32" | "-mx32" => {
602 let want: u32 = match arg {
603 "-m64" => 64,
604 _ => 32,
605 };
606 let have = rucc_target::TargetInfo::new(opts.target).pointer_width;
607 if have != want {
608 return Err(err(format!(
609 "{arg} asks for a {want} bit target and {} is {have} bit, use \
610 --target= to name the one you mean",
611 opts.target
612 )));
613 }
614 }
615 _ if arg.starts_with("-march=")
621 || arg.starts_with("-mtune=")
622 || arg.starts_with("-mcpu=") => {}
623 _ if arg.starts_with("-mabi=") => {
626 let want = &arg["-mabi=".len()..];
627 let have = match opts.target.arch {
628 rucc_target::Arch::X86_64 => "sysv",
629 rucc_target::Arch::Aarch64 => "lp64",
630 rucc_target::Arch::Riscv64 => "lp64d",
631 };
632 if want != have {
633 return Err(err(format!(
634 "{arg}: {} uses the {have} convention and this compiler has no other",
635 opts.target
636 )));
637 }
638 }
639 "-mcmodel=small" => {}
643 _ if arg.starts_with("-mcmodel=") => {
644 return Err(err(format!(
645 "{arg}: this compiler emits the small code model and no other, see \
646 spec/12-targets.md"
647 )));
648 }
649 _ if arg.starts_with("-specs=") => {
653 return Err(err(
654 "-specs= is not supported: the parts of it builds rely on are -B, -L, \
655 -nostdlib, -nostartfiles and -Wl,, see spec/04-driver-and-cli.md \
656 section 4.4",
657 ));
658 }
659 _ if arg.starts_with("-Wa,") || arg.starts_with("-Wp,") => {
665 return Err(err(format!(
666 "`{arg}` is an argument for a separate assembler or preprocessor, and both \
667 are inside this compiler rather than programs it runs"
668 )));
669 }
670 "-Xassembler" | "-Xpreprocessor" => {
671 return Err(err(format!(
672 "{arg} hands an argument to a separate assembler or preprocessor, and both \
673 are inside this compiler rather than programs it runs"
674 )));
675 }
676 _ if arg.starts_with("-W") => {}
683 "-fno-ident"
689 | "-fident"
690 | "-funit-at-a-time"
691 | "-fno-unit-at-a-time"
692 | "-shared-libgcc"
693 | "-static-libgcc" => {}
694 _ if arg.starts_with('-') && arg.len() > 1 => {
695 return Err(err(format!("unknown option `{arg}`")));
700 }
701 _ => inputs.push(Input { path: arg.to_owned(), forced, library: false }),
702 }
703 }
704
705 link.sysroot = sysroot.clone();
712 if threads {
717 inputs.push(Input::library("pthread"));
718 }
719 if let Some(query) = query {
720 return Ok(Action::Print(answer(&query, &opts, &link)));
721 }
722 if !nostdinc {
723 opts.search.push_system(runtime::DIR);
724 for dir in library::system_dirs(opts.target, sysroot.as_deref()) {
728 opts.search.push_system(dir);
729 }
730 }
731 opts.search.remove_duplicates();
735
736 if print_config {
739 return Ok(Action::PrintConfig(Box::new(opts)));
740 }
741 if print_pipeline {
742 return Ok(Action::PrintPipeline(Box::new(opts)));
743 }
744 let plan = Plan::new(&opts, &inputs, output.as_deref()).map_err(|e| err(e.message))?;
745 if print_plan {
746 return Ok(Action::PrintPlan {
747 opts: Box::new(opts),
748 plan: Box::new(plan),
749 link: Box::new(link),
750 });
751 }
752 Ok(Action::Compile {
753 opts: Box::new(opts),
754 plan: Box::new(plan),
755 link: Box::new(link),
756 jobs,
757 verbose,
758 })
759}
760
761fn answer(query: &Query, opts: &Options, link: &LinkOptions) -> String {
767 let found = |name: &str| {
768 link::find_in_search(link, opts.target, name)
769 .map_or_else(|| name.to_owned(), |path| path.display().to_string())
770 };
771 match query {
772 Query::Machine => opts.target.to_string(),
773 Query::Version => VERSION.to_owned(),
774 Query::Multiarch => link::multiarch(opts.target),
775 Query::SearchDirs => {
780 let here = std::env::current_exe()
781 .ok()
782 .and_then(|p| p.parent().map(std::path::Path::to_path_buf))
783 .unwrap_or_default();
784 let list = |dirs: &[PathBuf]| {
785 dirs.iter().map(|d| d.display().to_string()).collect::<Vec<_>>().join(":")
786 };
787 let libraries = link::search_dirs(link, opts.target);
788 format!(
789 "install: {}\nprograms: ={}\nlibraries: ={}",
790 here.display(),
791 list(&link.prefixes),
792 list(&libraries)
793 )
794 }
795 Query::FileName(name) => found(name),
796 Query::Libgcc => found("libgcc.a"),
800 Query::ProgName(name) => link
804 .prefixes
805 .iter()
806 .map(|dir| dir.join(name))
807 .find(|path| path.is_file())
808 .map_or_else(|| name.clone(), |path| path.display().to_string()),
809 }
810}
811
812#[must_use]
818pub fn print_pipeline(opts: &Options) -> String {
819 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
820 settings.toggles.clone_from(&opts.passes);
821 settings.global_fuel = opts.pass_fuel_global;
822 for (on, spec) in &opts.pass_gates {
823 let _ = settings.gates.add(*on, spec);
826 }
827 rucc_opt::pipeline::print(&settings)
828}
829
830#[must_use]
835pub fn print_config(opts: &Options) -> String {
836 let sess = Session::new(opts.clone());
837 let t = &sess.target;
838 let mut out = String::new();
839 let _ = writeln!(out, "version: {VERSION}");
840 let _ = writeln!(out, "target: {}", t.triple);
841 let _ = writeln!(out, "arch: {}", t.triple.arch.as_str());
842 let _ = writeln!(out, "os: {}", t.triple.os.as_str());
843 let _ = writeln!(out, "env: {}", t.triple.env.as_str());
844 let _ = writeln!(out, "object-format: {}", t.object_format.as_str());
845 let _ = writeln!(out, "pointer-width: {}", t.pointer_width);
846 let _ = writeln!(out, "long-width: {}", t.long_width);
847 let _ = writeln!(out, "long-double-width: {}", t.long_double_width);
848 let _ = writeln!(out, "endian: {}", if t.little_endian { "little" } else { "big" });
849 let _ = writeln!(out, "char-signed: {}", t.char_is_signed);
850 let _ = writeln!(out, "va-list: {}", t.va_list.as_str());
851 let regs: Vec<String> = t
854 .regs
855 .classes()
856 .map(|(class, info)| format!("{} {}", info.name, t.regs.len(class)))
857 .collect();
858 let _ = writeln!(
859 out,
860 "registers: {}",
861 if regs.is_empty() { "none".to_string() } else { regs.join(", ") }
862 );
863 let _ = writeln!(out, "opt-level: {}", sess.opts.opt_level);
864 let _ = writeln!(out, "safety: {}", sess.opts.safety);
865 let _ = writeln!(out, "emit: {}", sess.opts.emit.as_str());
866 let _ = writeln!(out, "debug-info: {}", sess.opts.debug_info);
867 let _ = writeln!(out, "frame-pointer: {}", sess.opts.frame_pointer);
868 let _ = writeln!(out, "red-zone: {}", sess.opts.red_zone);
869 for dir in sess.opts.search.dirs() {
872 let system = if dir.is_system { " (system)" } else { "" };
873 let _ = writeln!(out, "include: {}{system}", dir.path.display());
874 }
875 out
876}
877
878fn preprocess_all(opts: &Options, plan: &Plan) -> i32 {
884 let fs = OsFileSystem::new();
885 let mut stderr = std::io::stderr().lock();
886 let mut failed = false;
887 for job in &plan.jobs {
888 if !job.phases.first().is_some_and(|p| *p == Phase::Preprocess) {
889 continue;
892 }
893 let result = preprocess(opts, &job.input, &fs);
894 for message in &result.messages {
895 let _ = writeln!(stderr, "{message}");
896 }
897 if result.failed() {
898 failed = true;
899 continue;
900 }
901 if let Err(e) = write_out(&job.output, result.text.as_bytes()) {
902 let _ = writeln!(stderr, "rucc: error: {e}");
903 failed = true;
904 }
905 }
906 i32::from(failed)
907}
908
909fn compile_all(opts: &Options, plan: &Plan) -> i32 {
915 let fs = OsFileSystem::new();
916 let mut stderr = std::io::stderr().lock();
917 let mut failed = false;
918 let (mut remarks, ok) = Remarks::new(opts.opt_info_file.as_ref(), &mut stderr);
919 failed |= !ok;
920 let mut fired = Fired::new();
921 for job in &plan.jobs {
922 if !job.phases.contains(&Phase::Compile) {
923 continue;
924 }
925 let result = if job.kind == InputKind::Ir {
929 compile_ir(opts, &job.input, &fs)
930 } else {
931 compile(opts, &job.input, &fs)
932 };
933 fired.merge(&result.fired);
934 failed |= !write_dumps(&job.input, &result.dumps, &mut stderr);
935 failed |= !remarks.write(&result.remarks, &mut stderr);
936 for message in &result.messages {
937 let _ = writeln!(stderr, "{message}");
938 }
939 if result.failed() {
940 failed = true;
941 continue;
942 }
943 if let Err(e) = write_out(&job.output, result.artifact.bytes()) {
944 let _ = writeln!(stderr, "rucc: error: {e}");
945 failed = true;
946 }
947 }
948 failed |= !write_coverage(opts, &fired, &mut stderr);
949 i32::from(failed)
950}
951
952struct Scratch {
959 dir: PathBuf,
961}
962
963impl Scratch {
964 fn new() -> Result<Scratch, String> {
970 let dir = std::env::temp_dir().join(format!("rucc-{}", std::process::id()));
971 std::fs::create_dir_all(&dir).map_err(|e| format!("{}: {e}", dir.display()))?;
972 Ok(Scratch { dir })
973 }
974}
975
976impl Drop for Scratch {
977 fn drop(&mut self) {
978 let _ = std::fs::remove_dir_all(&self.dir);
979 }
980}
981
982fn link_line(opts: &Options, link: &LinkOptions, job: &LinkJob) -> Result<String, link::Error> {
989 let linker = link::find(opts.target, link)?;
990 let args = link::line(opts.target, link, &job.inputs, &job.output)?;
991 Ok(link::render(&linker, &args))
992}
993
994fn link_all(opts: &Options, plan: &Plan, link: &LinkOptions, verbose: bool) -> i32 {
1001 let Some(job) = &plan.link else {
1002 let mut stderr = std::io::stderr().lock();
1005 let _ = writeln!(stderr, "rucc: error: there is nothing to link");
1006 return 1;
1007 };
1008 let linker = match link::find(opts.target, link) {
1011 Ok(linker) => linker,
1012 Err(why) => return complain(why),
1013 };
1014
1015 let scratch = match Scratch::new() {
1016 Ok(scratch) => scratch,
1017 Err(why) => return complain(format!("could not make a place for the object files: {why}")),
1018 };
1019
1020 let fs = OsFileSystem::new();
1021 let mut failed = false;
1022 let mut produced: Vec<String> = Vec::with_capacity(plan.jobs.len());
1025 let mut fired = Fired::new();
1026 {
1027 let mut stderr = std::io::stderr().lock();
1028 let (mut remarks, ok) = Remarks::new(opts.opt_info_file.as_ref(), &mut stderr);
1029 failed |= !ok;
1030 for (at, job) in plan.jobs.iter().enumerate() {
1031 let out = match &job.output {
1032 Output::Temporary(hint) => {
1033 scratch.dir.join(format!("{at}-{hint}")).display().to_string()
1036 }
1037 Output::File(path) => path.clone(),
1038 Output::Stdout => continue,
1041 };
1042 produced.push(out.clone());
1043 if !job.phases.contains(&Phase::Compile) {
1044 continue;
1045 }
1046 let result = if job.kind == InputKind::Ir {
1047 compile_ir(opts, &job.input, &fs)
1048 } else {
1049 compile(opts, &job.input, &fs)
1050 };
1051 fired.merge(&result.fired);
1052 failed |= !write_dumps(&job.input, &result.dumps, &mut stderr);
1053 failed |= !remarks.write(&result.remarks, &mut stderr);
1054 for message in &result.messages {
1055 let _ = writeln!(stderr, "{message}");
1056 }
1057 if result.failed() {
1058 failed = true;
1059 continue;
1060 }
1061 if !matches!(result.artifact, Artifact::Object(_)) {
1062 let _ = writeln!(
1067 stderr,
1068 "rucc: internal error: {}: no object file was produced for the link",
1069 job.input
1070 );
1071 failed = true;
1072 continue;
1073 }
1074 if let Err(e) = std::fs::write(&out, result.artifact.bytes()) {
1075 let _ = writeln!(stderr, "rucc: error: {out}: {e}");
1076 failed = true;
1077 }
1078 }
1079 failed |= !write_coverage(opts, &fired, &mut stderr);
1080 }
1081 if failed {
1082 return 1;
1086 }
1087
1088 let mut outputs = produced.into_iter();
1092 let mut items = Vec::with_capacity(job.inputs.len());
1093 for item in &job.inputs {
1094 match item {
1095 link::Item::Library(name) => items.push(link::Item::Library(name.clone())),
1096 link::Item::File(_) => match outputs.next() {
1097 Some(path) => items.push(link::Item::File(path)),
1098 None => return complain("the plan asks the linker for a file nothing produced"),
1099 },
1100 }
1101 }
1102
1103 let args = match link::line(opts.target, link, &items, &job.output) {
1104 Ok(args) => args,
1105 Err(why) => return complain(why),
1106 };
1107 if verbose {
1108 let mut stderr = std::io::stderr().lock();
1109 let _ = writeln!(stderr, "{}", link::render(&linker, &args));
1110 }
1111 match link::run(&linker, &args) {
1112 Ok(()) => 0,
1113 Err(link::Error::Refused { .. }) => 1,
1116 Err(why) => complain(why),
1117 }
1118}
1119
1120fn complain(why: impl std::fmt::Display) -> i32 {
1122 let mut stderr = std::io::stderr().lock();
1123 let _ = writeln!(stderr, "rucc: error: {why}");
1124 1
1125}
1126
1127fn write_coverage(opts: &Options, fired: &Fired, stderr: &mut impl std::io::Write) -> bool {
1136 let Some(path) = &opts.rule_coverage else { return true };
1137 let Some(table) = coverage::table(opts.target.arch) else {
1138 let _ = writeln!(
1139 stderr,
1140 "rucc: error: there are no lowering rules for {} yet, so there is no coverage of them \
1141 to report",
1142 opts.target
1143 );
1144 return false;
1145 };
1146 match std::fs::write(path, fired.listing(table)) {
1147 Ok(()) => true,
1148 Err(e) => {
1149 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1150 false
1151 }
1152 }
1153}
1154
1155struct Remarks {
1162 file: Option<String>,
1164 started: bool,
1167}
1168
1169impl Remarks {
1170 fn new(file: Option<&String>, stderr: &mut impl std::io::Write) -> (Self, bool) {
1176 let mut ok = true;
1177 if let Some(path) = file {
1178 if let Err(e) = std::fs::write(path, "") {
1179 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1180 ok = false;
1181 }
1182 }
1183 (Self { file: file.cloned(), started: false }, ok)
1184 }
1185
1186 fn write(&mut self, text: &str, stderr: &mut impl std::io::Write) -> bool {
1192 if text.is_empty() {
1193 return true;
1194 }
1195 let Some(path) = &self.file else {
1196 let _ = write!(stderr, "{text}");
1197 return true;
1198 };
1199 let opened = std::fs::OpenOptions::new()
1200 .write(true)
1201 .append(self.started)
1202 .truncate(!self.started)
1203 .create(true)
1204 .open(path);
1205 self.started = true;
1206 let result =
1207 opened.and_then(|mut file| std::io::Write::write_all(&mut file, text.as_bytes()));
1208 if let Err(e) = result {
1209 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1210 return false;
1211 }
1212 true
1213 }
1214}
1215
1216fn write_dumps(input: &str, dumps: &[rucc_opt::Dump], stderr: &mut impl std::io::Write) -> bool {
1227 let stem = std::path::Path::new(input)
1228 .file_name()
1229 .map_or_else(|| input.to_owned(), |name| name.to_string_lossy().into_owned());
1230 let mut ok = true;
1231 for dump in dumps {
1232 let path = format!("{stem}.{}.ir", dump.name);
1233 if let Err(e) = std::fs::write(&path, &dump.text) {
1234 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1235 ok = false;
1236 }
1237 }
1238 ok
1239}
1240
1241fn write_out(output: &Output, bytes: &[u8]) -> Result<(), String> {
1248 match output {
1249 Output::Stdout => {
1250 let mut stdout = std::io::stdout().lock();
1251 stdout.write_all(bytes).map_err(|e| format!("writing to standard output: {e}"))
1252 }
1253 Output::File(path) | Output::Temporary(path) => {
1254 std::fs::write(path, bytes).map_err(|e| format!("{path}: {e}"))
1255 }
1256 }
1257}
1258
1259pub fn run(args: &[String]) -> i32 {
1264 match parse_args(args) {
1265 Ok(Action::Help) => {
1266 print!("{USAGE}");
1267 0
1268 }
1269 Ok(Action::Version) => {
1270 println!("rucc {VERSION}");
1271 0
1272 }
1273 Ok(Action::Print(line)) => {
1274 println!("{line}");
1275 0
1276 }
1277 Ok(Action::PrintConfig(opts)) => {
1278 print!("{}", print_config(&opts));
1279 0
1280 }
1281 Ok(Action::PrintPipeline(opts)) => {
1282 print!("{}", print_pipeline(&opts));
1283 0
1284 }
1285 Ok(Action::PrintPlan { opts, plan, link }) => {
1286 print!("{}", plan.render());
1287 if let Some(job) = &plan.link {
1291 match link_line(&opts, &link, job) {
1292 Ok(line) => println!("{line}"),
1293 Err(why) => {
1294 let mut stderr = std::io::stderr().lock();
1295 let _ = writeln!(stderr, "rucc: error: {why}");
1296 return 1;
1297 }
1298 }
1299 }
1300 0
1301 }
1302 Ok(Action::Compile { opts, plan, link, jobs, verbose }) => {
1303 {
1304 let mut stderr = std::io::stderr().lock();
1305 if verbose {
1306 let _ = write!(stderr, "{}", plan.render());
1307 let _ = writeln!(stderr, "workers: {}", jobs.count());
1308 }
1309 }
1310 if opts.emit == EmitKind::Preprocessed {
1311 return preprocess_all(&opts, &plan);
1312 }
1313 if opts.emit != EmitKind::Executable {
1314 return compile_all(&opts, &plan);
1315 }
1316 link_all(&opts, &plan, &link, verbose)
1317 }
1318 Err(e) => {
1319 let mut stderr = std::io::stderr().lock();
1320 let _ = writeln!(stderr, "rucc: error: {e}");
1321 let _ = writeln!(stderr, "rucc: note: run `rucc --help` for usage");
1322 1
1323 }
1324 }
1325}
1326
1327#[cfg(test)]
1328mod tests {
1329 use rucc_session::{GnucVersion, OptLevel};
1330
1331 use super::*;
1332
1333 fn args(s: &[&str]) -> Vec<String> {
1334 s.iter().map(|x| (*x).to_owned()).collect()
1335 }
1336
1337 #[test]
1338 fn help_and_version_win_over_everything_else() {
1339 assert_eq!(parse_args(&args(&["-c", "--help", "x.c"])).unwrap(), Action::Help);
1340 assert_eq!(parse_args(&args(&["--version"])).unwrap(), Action::Version);
1341 }
1342
1343 fn compile(s: &[&str]) -> (Box<Options>, Box<Plan>) {
1344 match parse_args(&args(s)).expect("expected a compilation") {
1345 Action::Compile { opts, plan, .. } => (opts, plan),
1346 other => panic!("expected a compilation, got {other:?}"),
1347 }
1348 }
1349
1350 fn linking(s: &[&str]) -> (Box<LinkOptions>, Box<Plan>) {
1351 match parse_args(&args(s)).expect("expected a compilation") {
1352 Action::Compile { link, plan, .. } => (link, plan),
1353 other => panic!("expected a compilation, got {other:?}"),
1354 }
1355 }
1356
1357 #[test]
1358 fn collects_inputs_and_flags() {
1359 let (opts, plan) = compile(&["-c", "-O2", "-g", "a.c", "b.c"]);
1360 let paths: Vec<&str> = plan.jobs.iter().map(|j| j.input.as_str()).collect();
1361 assert_eq!(paths, vec!["a.c", "b.c"]);
1362 assert_eq!(opts.opt_level, OptLevel::O2);
1363 assert_eq!(opts.emit, EmitKind::Object);
1364 assert!(opts.debug_info);
1365 }
1366
1367 #[test]
1370 fn an_unstable_option_is_taken_and_one_that_does_not_exist_is_refused() {
1371 let (opts, _) = compile(&["-c", "-Zrule-coverage=/tmp/rules.cov", "a.c"]);
1372 assert_eq!(opts.rule_coverage.as_deref(), Some("/tmp/rules.cov"));
1373
1374 let (plain, _) = compile(&["-c", "a.c"]);
1375 assert_eq!(plain.rule_coverage, None, "nothing is measured unless it was asked for");
1376
1377 assert!(parse_args(&args(&["-Zrule-coverage=", "a.c"])).is_err(), "a file with no name");
1378 let unknown = parse_args(&args(&["-Zwhat", "a.c"])).expect_err("there is no such option");
1379 assert!(unknown.message.contains("4.11"), "{}", unknown.message);
1380 }
1381
1382 #[test]
1383 fn a_bare_dash_o_means_o1_the_way_gcc_reads_it() {
1384 let (opts, _) = compile(&["-O", "a.c"]);
1385 assert_eq!(opts.opt_level, OptLevel::O1);
1386 }
1387
1388 #[test]
1389 fn dash_x_applies_to_later_inputs_only_and_none_stops_it() {
1390 let (_, plan) = compile(&["a.o", "-x", "c", "b.txt", "-x", "none", "c.o"]);
1391 assert_eq!(plan.jobs[0].kind, InputKind::LinkerInput);
1392 assert_eq!(plan.jobs[1].kind, InputKind::C);
1393 assert_eq!(plan.jobs[2].kind, InputKind::LinkerInput);
1394 }
1395
1396 #[test]
1397 fn dash_j_reaches_the_scheduler_and_defaults_to_the_machine() {
1398 let (_, _, jobs) = match parse_args(&args(&["-j4", "a.c"])).unwrap() {
1399 Action::Compile { opts, plan, jobs, .. } => (opts, plan, jobs),
1400 other => panic!("expected a compilation, got {other:?}"),
1401 };
1402 assert_eq!(jobs.count(), 4);
1403
1404 let default = match parse_args(&args(&["a.c"])).unwrap() {
1405 Action::Compile { jobs, .. } => jobs,
1406 other => panic!("expected a compilation, got {other:?}"),
1407 };
1408 assert_eq!(default, Jobs::available());
1409 assert!(parse_args(&args(&["-j0", "a.c"])).is_err());
1410 }
1411
1412 #[test]
1413 fn triple_hash_prints_the_plan_and_runs_nothing() {
1414 let a = parse_args(&args(&["-###", "-c", "a.c"])).unwrap();
1415 let Action::PrintPlan { plan, .. } = a else { panic!("expected a plan dump") };
1416 assert!(plan.render().contains("a.c: preprocess, compile, assemble -> a.o"));
1417 }
1418
1419 #[test]
1420 fn dash_x_names_what_it_accepts_when_it_does_not_know_a_language() {
1421 let e = parse_args(&args(&["-x", "fortran", "a.c"])).unwrap_err();
1422 assert!(e.message.contains("assembler-with-cpp"), "{}", e.message);
1423 }
1424
1425 #[test]
1426 fn an_unknown_flag_is_an_error_rather_than_a_shrug() {
1427 let e = parse_args(&args(&["-fno-such-thing", "a.c"])).unwrap_err();
1428 assert!(e.message.contains("unknown option"), "{}", e.message);
1429 }
1430
1431 #[test]
1434 fn permissive_reads_in_both_directions_and_the_last_one_wins() {
1435 let (opts, _) = compile(&["-c", "a.c"]);
1436 assert!(!opts.permissive, "off unless it is asked for");
1437
1438 let (opts, _) = compile(&["-c", "-fpermissive", "a.c"]);
1439 assert!(opts.permissive);
1440
1441 let (opts, _) = compile(&["-c", "-fpermissive", "-fno-permissive", "a.c"]);
1442 assert!(!opts.permissive);
1443 }
1444
1445 #[test]
1446 fn asking_for_nested_functions_is_told_why_it_is_not_coming() {
1447 let e = parse_args(&args(&["-fnested-functions", "a.c"])).unwrap_err();
1448 assert!(e.message.contains("trampoline"), "{}", e.message);
1449 assert!(parse_args(&args(&["-fno-nested-functions", "a.c"])).is_ok());
1450 }
1451
1452 #[test]
1453 fn an_unsupported_target_names_itself() {
1454 let e = parse_args(&args(&["--target=sparc64-linux-gnu", "a.c"])).unwrap_err();
1455 assert!(e.message.contains("sparc64"), "{}", e.message);
1456 }
1457
1458 #[test]
1459 fn no_inputs_is_an_error_but_print_config_needs_none() {
1460 assert!(parse_args(&args(&[])).is_err());
1461 assert!(matches!(parse_args(&args(&["--print-config"])), Ok(Action::PrintConfig(_))));
1462 }
1463
1464 #[test]
1465 fn print_config_reports_the_target_it_was_given_not_the_host() {
1466 let a = parse_args(&args(&["--print-config", "--target=riscv64-linux-musl"])).unwrap();
1467 let Action::PrintConfig(opts) = a else { panic!("expected a configuration dump") };
1468 let text = print_config(&opts);
1469 assert!(text.contains("target: riscv64-unknown-linux-musl"), "{text}");
1470 assert!(text.contains("char-signed: false"), "{text}");
1471 assert!(text.contains("object-format: elf"), "{text}");
1472 assert!(text.contains("va-list: void-pointer"), "{text}");
1473 assert!(text.contains("registers: none"), "{text}");
1476 }
1477
1478 #[test]
1479 fn print_config_has_one_key_per_line_and_a_fixed_order() {
1480 let opts = Options::new("x86_64-unknown-linux-gnu".parse().unwrap());
1481 let text = print_config(&opts);
1482 let keys: Vec<&str> =
1483 text.lines().map(|l| l.split(':').next().unwrap_or_default()).collect();
1484 assert_eq!(keys[0], "version");
1485 assert_eq!(keys[1], "target");
1486 assert_eq!(keys.len(), 19);
1487 assert!(text.ends_with('\n'));
1488 }
1489
1490 #[test]
1491 fn the_safety_tier_is_read_off_the_command_line_and_a_wrong_one_is_refused() {
1492 let (opts, _) = compile(&["a.c"]);
1493 assert_eq!(opts.safety, rucc_session::Safety::Off);
1494
1495 for (flag, tier) in [
1496 ("-fsafety=detect", rucc_session::Safety::Detect),
1497 ("-fsafety=enforce", rucc_session::Safety::Enforce),
1498 ("-fsafety=kernel", rucc_session::Safety::Kernel),
1499 ("-fsafety=off", rucc_session::Safety::Off),
1500 ] {
1501 let (opts, _) = compile(&[flag, "a.c"]);
1502 assert_eq!(opts.safety, tier, "{flag}");
1503 }
1504
1505 let (opts, _) = compile(&["-fsafety=enforce", "-fsafety=off", "a.c"]);
1507 assert_eq!(opts.safety, rucc_session::Safety::Off);
1508
1509 let e = parse_args(&args(&["-fsafety=on", "a.c"])).unwrap_err();
1512 assert!(e.message.contains("is not a safety tier"), "{}", e.message);
1513 assert!(parse_args(&args(&["-fsafety", "a.c"])).is_err());
1514 }
1515
1516 #[test]
1517 fn print_pipeline_answers_with_the_passes_the_level_asked_for() {
1518 let a = parse_args(&args(&["--print-pipeline", "-O2"])).unwrap();
1519 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1520 let text = print_pipeline(&opts);
1521 assert!(text.starts_with("level: -O2\n"), "{text}");
1522 assert!(text.contains("fold"), "{text}");
1523
1524 let a = parse_args(&args(&["--print-pipeline"])).unwrap();
1525 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1526 assert!(print_pipeline(&opts).contains("1: simplify-cfg,"), "{}", print_pipeline(&opts));
1529
1530 let a = parse_args(&args(&["--print-pipeline", "-fno-simplify-cfg"])).unwrap();
1531 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1532 assert!(print_pipeline(&opts).contains("no passes"), "{}", print_pipeline(&opts));
1535 }
1536
1537 #[test]
1538 fn print_pipeline_takes_the_toggles_into_account() {
1539 let a = parse_args(&args(&["--print-pipeline", "-O2", "-fno-fold"])).unwrap();
1540 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1541 let text = print_pipeline(&opts);
1542 assert!(!text.contains("fold"), "{text}");
1545 assert!(text.contains("dce"), "{text}");
1546
1547 let mut off = vec!["--print-pipeline".to_owned(), "-O2".to_owned()];
1551 off.extend(rucc_opt::PASSES.iter().map(|p| format!("-fno-{}", p.name())));
1552 let spelled: Vec<&str> = off.iter().map(String::as_str).collect();
1553 let a = parse_args(&args(&spelled)).unwrap();
1554 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1555 assert!(print_pipeline(&opts).contains("no passes"), "{}", print_pipeline(&opts));
1556 }
1557
1558 #[test]
1559 fn print_pipeline_says_when_a_budget_will_stop_the_run_short() {
1560 let a = parse_args(&args(&["--print-pipeline", "-O2"])).unwrap();
1561 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1562 assert!(!print_pipeline(&opts).contains("global fuel"));
1563
1564 let a = parse_args(&args(&["--print-pipeline", "-O2", "-fpass-fuel-global=4"])).unwrap();
1565 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1566 let text = print_pipeline(&opts);
1567 assert!(text.contains("global fuel: 4"), "{text}");
1570 }
1571
1572 #[test]
1575 fn a_pass_is_named_by_dash_f_and_unnamed_by_dash_f_no() {
1576 let (opts, _) = compile(&["-c", "-O0", "-ffold", "-fno-fold", "-ffold", "a.c"]);
1577 assert_eq!(
1578 opts.passes,
1579 [("fold".to_owned(), true), ("fold".to_owned(), false), ("fold".to_owned(), true)]
1580 );
1581
1582 let e = parse_args(&args(&["-fno-such-pass", "a.c"])).unwrap_err();
1583 assert!(e.message.contains("unknown option"), "{}", e.message);
1584 }
1585
1586 #[test]
1587 fn pass_fuel_names_a_pass_and_a_count_and_refuses_anything_else() {
1588 let (opts, _) = compile(&["-c", "-O2", "-fpass-fuel=fold=3", "a.c"]);
1589 assert_eq!(opts.pass_fuel, [("fold".to_owned(), 3)]);
1590
1591 let e = parse_args(&args(&["-fpass-fuel=fold", "a.c"])).unwrap_err();
1592 assert!(e.message.contains("<pass>=<count>"), "{}", e.message);
1593 let e = parse_args(&args(&["-fpass-fuel=nosuch=3", "a.c"])).unwrap_err();
1594 assert!(e.message.contains("--print-pipeline"), "{}", e.message);
1595 let e = parse_args(&args(&["-fpass-fuel=fold=lots", "a.c"])).unwrap_err();
1596 assert!(e.message.contains("not a number"), "{}", e.message);
1597 }
1598
1599 #[test]
1600 fn global_pass_fuel_is_a_count_on_its_own_and_defaults_to_no_limit() {
1601 let (opts, _) = compile(&["-c", "-O2", "a.c"]);
1602 assert_eq!(opts.pass_fuel_global, None);
1603
1604 let (opts, _) = compile(&["-c", "-O2", "-fpass-fuel-global=12", "a.c"]);
1605 assert_eq!(opts.pass_fuel_global, Some(12));
1606 assert!(opts.pass_fuel.is_empty());
1609
1610 let e = parse_args(&args(&["-fpass-fuel-global=lots", "a.c"])).unwrap_err();
1611 assert!(e.message.contains("not a number"), "{}", e.message);
1612 }
1613
1614 #[test]
1615 fn a_gate_names_a_pass_and_optionally_the_functions_it_covers() {
1616 let (opts, _) = compile(&["-c", "-O2", "-fdisable-fold", "-fenable-fold=2-4,main", "a.c"]);
1617 assert_eq!(
1618 opts.pass_gates,
1619 [(false, "fold".to_owned()), (true, "fold=2-4,main".to_owned())],
1620 "the order is what decides, so it has to survive the parse"
1621 );
1622
1623 let e = parse_args(&args(&["-fdisable-nosuch", "a.c"])).unwrap_err();
1624 assert!(e.message.contains("--print-pipeline"), "{}", e.message);
1625 let e = parse_args(&args(&["-fenable-fold=9-2", "a.c"])).unwrap_err();
1626 assert!(e.message.contains("ends before it starts"), "{}", e.message);
1627 let e = parse_args(&args(&["-fdisable-fold=", "a.c"])).unwrap_err();
1628 assert!(e.message.contains("is empty"), "{}", e.message);
1629 }
1630
1631 #[test]
1632 fn the_pipeline_listing_says_which_passes_a_gate_touched() {
1633 let (opts, _) = compile(&["-c", "-O2", "-fdisable-fold=main", "a.c"]);
1634 let text = print_pipeline(&opts);
1635 assert!(text.contains("fold, "), "{text}");
1636 assert!(text.contains("[off for main]"), "{text}");
1637 }
1638
1639 #[test]
1643 fn a_dump_is_checked_when_it_is_asked_for_rather_than_when_it_is_taken() {
1644 let (opts, _) = compile(&["-c", "-O2", "-fdump-ir=all", "-fdump-ir=after-fold", "a.c"]);
1645 assert_eq!(opts.dump_ir, ["all", "after-fold"]);
1646
1647 let e = parse_args(&args(&["-fdump-ir=after-nosuch", "a.c"])).unwrap_err();
1648 assert!(e.message.contains("nosuch"), "{}", e.message);
1649 assert!(parse_args(&args(&["-fdump-ir=sideways-fold", "a.c"])).is_err());
1650 }
1651
1652 #[test]
1658 fn opt_info_takes_kinds_and_a_file_and_refuses_a_kind_it_does_not_have() {
1659 let (opts, _) = compile(&["-c", "-O2", "-fopt-info", "a.c"]);
1660 assert_eq!(opts.opt_info, [""], "a bare flag asks for the rewrites");
1661 assert_eq!(opts.opt_info_file, None, "and goes to standard error");
1662
1663 let (opts, _) = compile(&["-c", "-O2", "-fopt-info-missed-note", "a.c"]);
1664 assert_eq!(opts.opt_info, ["missed-note"]);
1665
1666 let (opts, _) =
1669 compile(&["-c", "-O2", "-fopt-info-missed=one.txt", "-fopt-info-all=two.txt", "a.c"]);
1670 assert_eq!(opts.opt_info, ["missed", "all"]);
1671 assert_eq!(opts.opt_info_file.as_deref(), Some("two.txt"));
1672
1673 let e = parse_args(&args(&["-fopt-info-vectorized", "a.c"])).unwrap_err();
1674 assert!(e.message.contains("vectorized"), "{}", e.message);
1675 assert!(e.message.contains("`missed`"), "{}", e.message);
1676 let e = parse_args(&args(&["-fopt-info-missed=", "a.c"])).unwrap_err();
1677 assert!(e.message.contains("no file"), "{}", e.message);
1678 }
1679
1680 #[test]
1681 fn verify_each_is_unstable_and_off_unless_it_was_asked_for() {
1682 let (opts, _) = compile(&["-c", "-Zverify-each", "a.c"]);
1683 assert!(opts.verify_each);
1684 assert!(!USAGE.contains("verify-each"), "an unstable option stays out of the usage text");
1685 }
1686
1687 #[test]
1688 fn dash_o_needs_an_argument() {
1689 let e = parse_args(&args(&["a.c", "-o"])).unwrap_err();
1690 assert_eq!(e.message, "-o requires an argument");
1691 }
1692
1693 #[test]
1694 fn dash_d_and_dash_u_are_read_joined_or_separated_and_keep_their_order() {
1695 let (opts, _) = compile(&["-DFOO=1", "-D", "BAR", "-UBAZ", "-U", "QUX", "a.c"]);
1696 assert_eq!(opts.defines, ["FOO=1", "BAR"]);
1697 assert_eq!(opts.undefines, ["BAZ", "QUX"]);
1698 }
1699
1700 #[test]
1701 fn the_include_flags_land_on_the_chain_each_one_names() {
1702 let (opts, _) = compile(&[
1705 "-Ii",
1706 "-iquote",
1707 "q",
1708 "-isystem",
1709 "sys",
1710 "-idirafter",
1711 "after",
1712 "--sysroot=/nowhere-at-all",
1713 "a.c",
1714 ]);
1715 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1716 assert_eq!(dirs, ["q", "i", "sys", runtime::DIR, "after"]);
1719 assert!(!opts.search.dirs()[1].is_system);
1720 assert!(opts.search.dirs()[2].is_system);
1721 }
1722
1723 #[test]
1724 fn the_librarys_headers_come_after_the_compilers_own_and_go_away_with_them() {
1725 let (opts, _) = compile(&["a.c"]);
1729 let dirs = opts.search.dirs();
1730 let ours = dirs.iter().position(|d| d.path.to_str() == Some(runtime::DIR));
1731 assert_eq!(ours, Some(0), "{dirs:?}");
1732 assert!(dirs[1..].iter().all(|d| d.is_system), "{dirs:?}");
1733 let (bare, _) = compile(&["-nostdinc", "a.c"]);
1734 assert!(bare.search.dirs().is_empty(), "{:?}", bare.search.dirs());
1735 }
1736
1737 #[test]
1738 fn a_sysroot_moves_the_librarys_directories_and_nothing_else() {
1739 let (opts, _) = compile(&["-isystem", "sys", "--sysroot=/nowhere-at-all", "a.c"]);
1740 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1741 assert_eq!(dirs, ["sys", runtime::DIR]);
1742 }
1743
1744 #[test]
1745 fn nostdinc_takes_the_compilers_own_headers_off_the_path() {
1746 let (opts, _) = compile(&["-Ii", "-nostdinc", "a.c"]);
1747 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1748 assert_eq!(dirs, ["i"]);
1749 }
1750
1751 #[test]
1752 fn the_dialect_flags_set_the_language_and_the_extensions_separately() {
1753 let (opts, _) = compile(&["-std=gnu11", "a.c"]);
1754 assert_eq!(opts.std, Std::C11);
1755 assert!(opts.gnu_extensions);
1756
1757 let (opts, _) = compile(&["-std=iso9899:1999", "a.c"]);
1758 assert_eq!(opts.std, Std::C99);
1759 assert!(!opts.gnu_extensions);
1760
1761 let (opts, _) = compile(&["-ansi", "a.c"]);
1762 assert_eq!(opts.std, Std::C89);
1763 assert!(!opts.gnu_extensions);
1764
1765 let e = parse_args(&args(&["-std=c94jr", "a.c"])).unwrap_err();
1766 assert!(e.message.contains("unknown dialect"), "{}", e.message);
1767 }
1768
1769 #[test]
1770 fn the_dump_letters_are_a_family_and_everything_else_beginning_with_d_is_not() {
1771 let (opts, _) = compile(&["-dM", "a.c"]);
1772 assert!(opts.dumps.macros);
1773
1774 let (opts, _) = compile(&["-dDM", "a.c"]);
1777 assert!(opts.dumps.macros);
1778 let (opts, _) = compile(&["-dD", "a.c"]);
1779 assert!(!opts.dumps.macros);
1780
1781 let (opts, _) = compile(&["a.c"]);
1782 assert!(!opts.dumps.any());
1783
1784 assert_eq!(printed(&["-dumpversion", "a.c"]), VERSION);
1787 }
1788
1789 #[test]
1790 fn the_gcc_version_claimed_is_a_flag_and_the_short_spellings_are_the_ones_people_write() {
1791 let (opts, _) = compile(&["a.c"]);
1792 assert_eq!(
1793 opts.gnuc,
1794 GnucVersion { major: 7, minor: 0, patch: 0 },
1795 "the lowest claim a modern glibc gives its own declarations to"
1796 );
1797
1798 let (opts, _) = compile(&["-fgnuc-version=15.1.0", "a.c"]);
1799 assert_eq!(opts.gnuc, GnucVersion { major: 15, minor: 1, patch: 0 });
1800
1801 let (opts, _) = compile(&["-fgnuc-version=15", "a.c"]);
1804 assert_eq!(opts.gnuc, GnucVersion { major: 15, minor: 0, patch: 0 });
1805
1806 let (opts, _) = compile(&["-fgnuc-version=13.2", "a.c"]);
1807 assert_eq!(opts.gnuc, GnucVersion { major: 13, minor: 2, patch: 0 });
1808
1809 let e = parse_args(&args(&["-fgnuc-version=15.x", "a.c"])).unwrap_err();
1810 assert!(e.message.contains("minor that is not a number"), "{}", e.message);
1811
1812 let e = parse_args(&args(&["-fgnuc-version=1.2.3.4", "a.c"])).unwrap_err();
1813 assert!(e.message.contains("more than three"), "{}", e.message);
1814 }
1815
1816 #[test]
1817 fn pedantic_has_two_spellings_and_is_not_the_same_knob_as_the_dialect() {
1818 let (opts, _) = compile(&["-std=c17", "-pedantic", "a.c"]);
1819 assert!(opts.pedantic);
1820 assert_eq!(opts.std, Std::C17);
1821
1822 let (opts, _) = compile(&["-Wpedantic", "a.c"]);
1825 assert!(opts.pedantic);
1826
1827 let (opts, _) = compile(&["-std=c17", "a.c"]);
1828 assert!(!opts.pedantic, "a dialect on its own does not diagnose an extension");
1829 }
1830
1831 #[test]
1832 fn dash_p_and_dash_ffreestanding_reach_the_options() {
1833 let (opts, _) = compile(&["-E", "-P", "-ffreestanding", "a.c"]);
1834 assert!(!opts.line_markers);
1835 assert!(!opts.hosted);
1836 assert_eq!(opts.emit, EmitKind::Preprocessed);
1837 }
1838
1839 #[test]
1846 fn the_builtin_flags_are_read_in_both_directions_and_one_name_at_a_time() {
1847 let (opts, _) = compile(&["-c", "a.c"]);
1848 assert!(opts.builtins, "a library name means the library function by default");
1849 assert!(opts.no_builtin.is_empty());
1850
1851 let (opts, _) = compile(&["-c", "-fno-builtin", "a.c"]);
1852 assert!(!opts.builtins);
1853
1854 let (opts, _) = compile(&["-c", "-fno-builtin", "-fbuiltin", "a.c"]);
1855 assert!(opts.builtins, "the last mention decides");
1856
1857 let (opts, _) = compile(&["-c", "-fno-builtin-memcpy", "-fno-builtin-nonesuch", "a.c"]);
1858 assert!(opts.builtins, "one name is not the family");
1859 assert_eq!(opts.no_builtin, vec!["memcpy".to_owned(), "nonesuch".to_owned()]);
1860 }
1861
1862 #[test]
1865 fn gnu89_inline_is_off_until_it_is_asked_for_and_the_last_mention_decides() {
1866 let (opts, _) = compile(&["-c", "a.c"]);
1867 assert!(!opts.gnu89_inline, "C's reading of inline by default");
1868
1869 let (opts, _) = compile(&["-c", "-fgnu89-inline", "a.c"]);
1870 assert!(opts.gnu89_inline);
1871
1872 let (opts, _) = compile(&["-c", "-fgnu89-inline", "-fno-gnu89-inline", "a.c"]);
1873 assert!(!opts.gnu89_inline, "the last mention decides");
1874
1875 let (opts, _) = compile(&["-c", "-std=c89", "a.c"]);
1880 assert!(!opts.gnu89_inline);
1881 }
1882
1883 #[test]
1886 fn the_two_frame_flags_are_read_in_both_directions() {
1887 let (opts, _) = compile(&["-c", "a.c"]);
1888 assert!(!opts.frame_pointer, "gcc omits it above -O0 and so does this");
1889 assert!(opts.red_zone, "the psABI has one and nothing said not to use it");
1890
1891 let (opts, _) = compile(&["-c", "-fno-omit-frame-pointer", "-mno-red-zone", "a.c"]);
1892 assert!(opts.frame_pointer);
1893 assert!(!opts.red_zone);
1894
1895 let (opts, _) = compile(&[
1896 "-c",
1897 "-fno-omit-frame-pointer",
1898 "-fomit-frame-pointer",
1899 "-mno-red-zone",
1900 "-mred-zone",
1901 "a.c",
1902 ]);
1903 assert!(!opts.frame_pointer, "the last one wins, as it does in gcc");
1904 assert!(opts.red_zone);
1905 }
1906
1907 #[test]
1908 fn the_link_flags_are_collected_apart_from_the_compilation() {
1909 let (link, _) = linking(&[
1910 "-static",
1911 "-nostartfiles",
1912 "-rdynamic",
1913 "-s",
1914 "-fuse-ld=mold",
1915 "-L/opt/lib",
1916 "-B",
1917 "/opt/tools",
1918 "a.c",
1919 ]);
1920 assert!(link.is_static);
1921 assert!(link.no_startfiles);
1922 assert!(link.export_dynamic);
1923 assert!(link.strip);
1924 assert_eq!(link.use_ld.as_deref(), Some("mold"));
1925 assert_eq!(link.search, vec![PathBuf::from("/opt/lib")]);
1926 assert_eq!(link.prefixes, vec![PathBuf::from("/opt/tools")]);
1927 }
1928
1929 #[test]
1930 fn a_comma_in_dash_wl_separates_two_arguments() {
1931 let (link, _) = linking(&["-Wl,-rpath,/opt/lib", "-Xlinker", "--as-needed", "a.c"]);
1932 assert_eq!(link.passthrough, vec!["-rpath", "/opt/lib", "--as-needed"]);
1933 }
1934
1935 #[test]
1936 fn a_library_keeps_its_place_between_the_objects() {
1937 let (_, plan) = linking(&["--target=x86_64-unknown-linux-gnu", "a.c", "-lm", "b.c"]);
1942 let link = plan.link.expect("expected a link step");
1943 assert_eq!(
1944 link.inputs,
1945 vec![
1946 link::Item::File("a.o".into()),
1947 link::Item::Library("m".into()),
1948 link::Item::File("b.o".into()),
1949 ]
1950 );
1951 assert_eq!(plan.jobs.len(), 2);
1953 }
1954
1955 #[test]
1956 fn a_library_on_a_dash_c_line_is_a_note_rather_than_an_error() {
1957 let (_, plan) = linking(&["-c", "-lm", "a.c"]);
1958 assert!(plan.link.is_none());
1959 assert!(plan.notes.iter().any(|n| n.contains("-lm")), "{:?}", plan.notes);
1960 }
1961
1962 #[test]
1963 fn the_sysroot_reaches_the_linker_as_well_as_the_headers() {
1964 let (link, _) = linking(&["--sysroot=/opt/root", "a.c"]);
1965 assert_eq!(link.sysroot, Some(PathBuf::from("/opt/root")));
1966 }
1967
1968 fn printed(s: &[&str]) -> String {
1969 match parse_args(&args(s)).expect("expected an answer") {
1970 Action::Print(line) => line,
1971 other => panic!("expected an answer, got {other:?}"),
1972 }
1973 }
1974
1975 fn refused(s: &[&str]) -> String {
1976 parse_args(&args(s)).expect_err("expected a refusal").message
1977 }
1978
1979 #[test]
1980 fn a_warning_flag_this_compiler_has_not_heard_of_is_taken_rather_than_refused() {
1981 let (opts, _) = compile(&["-Wall", "-Wextra", "-Wno-format-truncation", "-c", "a.c"]);
1985 assert!(!opts.warnings_are_errors);
1986 assert!(opts.warnings);
1987 let (opts, _) = compile(&["-Werror", "-c", "a.c"]);
1989 assert!(opts.warnings_are_errors);
1990 let (opts, _) = compile(&["-w", "-c", "a.c"]);
1991 assert!(!opts.warnings);
1992 let (opts, _) = compile(&["-pedantic-errors", "-c", "a.c"]);
1993 assert!(opts.pedantic && opts.warnings_are_errors);
1994 }
1995
1996 #[test]
1997 fn an_argument_for_a_separate_tool_is_refused_rather_than_dropped() {
1998 assert!(refused(&["-Wa,--noexecstack", "-c", "a.c"]).contains("separate assembler"));
2000 assert!(refused(&["-Wp,-DX", "-c", "a.c"]).contains("separate assembler"));
2001 assert!(refused(&["-specs=/x", "a.c"]).contains("-specs= is not supported"));
2002 assert!(refused(&["-mcmodel=kernel", "-c", "a.c"]).contains("small code model"));
2003 assert!(refused(&["-gdwarf-4", "-c", "a.c"]).contains("DWARF 5"));
2004 assert!(refused(&["-Ofast", "-c", "a.c"]).contains("fast math"));
2005 let no32 = refused(&["--target=x86_64-unknown-linux-gnu", "-m32", "-c", "a.c"]);
2008 assert!(no32.contains("32 bit target"), "{no32}");
2009 }
2010
2011 #[test]
2012 fn the_levels_gcc_spells_differently_are_the_levels_they_mean() {
2013 assert_eq!(compile(&["-O", "-c", "a.c"]).0.opt_level, OptLevel::O1);
2014 assert_eq!(compile(&["-Og", "-c", "a.c"]).0.opt_level, OptLevel::O1);
2015 assert_eq!(compile(&["-O2", "-c", "a.c"]).0.opt_level, OptLevel::O2);
2016 }
2017
2018 #[test]
2019 fn the_machine_flags_that_name_what_we_already_do_are_taken_and_the_rest_are_not() {
2020 let line = ["--target=x86_64-unknown-linux-gnu", "-m64", "-march=x86-64-v3"];
2021 let (opts, _) =
2022 compile(&[&line[..], &["-mtune=native", "-mabi=sysv", "-c", "a.c"]].concat());
2023 assert_eq!(opts.target.to_string(), "x86_64-unknown-linux-gnu");
2024 let wrong = refused(&["--target=x86_64-unknown-linux-gnu", "-mabi=ms", "-c", "a.c"]);
2025 assert!(wrong.contains("sysv convention"), "{wrong}");
2026 }
2027
2028 #[test]
2029 fn the_thread_flag_is_a_macro_and_a_library_and_the_library_goes_last() {
2030 let (opts, plan) = compile(&["-pthread", "-c", "a.c"]);
2031 assert!(opts.defines.iter().any(|d| d == "_REENTRANT"));
2032 let names: Vec<&str> = plan.jobs.iter().map(|j| j.input.as_str()).collect();
2035 assert_eq!(names, vec!["a.c"]);
2036 }
2037
2038 #[test]
2039 fn the_questions_a_build_system_asks_before_it_compiles_anything() {
2040 let target = "--target=x86_64-unknown-linux-gnu";
2041 assert_eq!(printed(&[target, "-dumpmachine"]), "x86_64-unknown-linux-gnu");
2042 assert_eq!(printed(&[target, "-dumpversion"]), VERSION);
2043 assert_eq!(printed(&[target, "-dumpfullversion"]), VERSION);
2044 assert_eq!(printed(&[target, "-print-multiarch"]), "x86_64-linux-gnu");
2045 assert_eq!(printed(&[target, "-print-file-name=no-such-library.a"]), "no-such-library.a");
2048 assert_eq!(printed(&[target, "-print-prog-name=ld"]), "ld");
2049 let dirs = printed(&[target, "-print-search-dirs"]);
2050 assert!(dirs.starts_with("install: "), "{dirs}");
2051 assert!(dirs.contains("\nlibraries: ="), "{dirs}");
2052 }
2053
2054 #[test]
2055 fn usage_fits_on_a_screen() {
2056 assert!(USAGE.lines().count() < 42, "usage text has grown past one screen");
2066 }
2067}