1#![doc(html_root_url = "https://docs.rs/rucc-driver/0.8.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 "-ffreestanding" => opts.hosted = false,
301 "-fhosted" => opts.hosted = true,
302 "-fno-builtin" => opts.builtins = false,
303 "-fbuiltin" => opts.builtins = true,
304 "-fgnu89-inline" => opts.gnu89_inline = true,
308 "-fno-gnu89-inline" => opts.gnu89_inline = false,
309 "-fno-omit-frame-pointer" => opts.frame_pointer = true,
312 "-fomit-frame-pointer" => opts.frame_pointer = false,
313 "-mno-red-zone" => opts.red_zone = false,
314 "-mred-zone" => opts.red_zone = true,
315 "-nostdinc" => nostdinc = true,
319 "-o" => {
320 output = Some(args.get(i).ok_or_else(|| err("-o requires an argument"))?.clone());
321 i += 1;
322 }
323 "-isysroot" => {
330 let dir = args.get(i).ok_or_else(|| err("-isysroot requires an argument"))?;
331 i += 1;
332 sysroot = Some(PathBuf::from(dir));
333 }
334 "-iquote" | "-isystem" | "-idirafter" => {
335 let dir = args.get(i).ok_or_else(|| err(format!("{arg} requires an argument")))?;
336 i += 1;
337 match arg {
338 "-iquote" => opts.search.push_quote(dir.clone()),
339 "-isystem" => opts.search.push_system(dir.clone()),
340 _ => opts.search.push_after(dir.clone()),
341 }
342 }
343 "-x" => {
344 let lang = args.get(i).ok_or_else(|| err("-x requires an argument"))?;
345 i += 1;
346 forced = if lang == "none" {
347 None
348 } else {
349 Some(InputKind::from_x_arg(lang).map_err(|e| err(format!("{e}")))?)
350 };
351 }
352 _ if arg.starts_with("-D") => {
360 let value = joined_or_next(arg, 2, args, &mut i)?;
361 opts.defines.push(value);
362 }
363 _ if arg.starts_with("-U") => {
364 let value = joined_or_next(arg, 2, args, &mut i)?;
365 opts.undefines.push(value);
366 }
367 _ if arg.starts_with("-I") => {
368 let dir = joined_or_next(arg, 2, args, &mut i)?;
369 opts.search.push_bracket(dir);
370 }
371 _ if arg.starts_with("-std=") => {
372 let name = &arg["-std=".len()..];
373 let (std, gnu) = Std::from_flag(name)
374 .ok_or_else(|| err(format!("unknown dialect `{name}`, see --help")))?;
375 opts.std = std;
376 opts.gnu_extensions = gnu;
377 }
378 _ if Dumps::is_family(arg) => {
387 opts.dumps.add(&arg[2..]);
388 }
389 _ if arg.starts_with("-fno-builtin-") => {
394 opts.no_builtin.push(arg["-fno-builtin-".len()..].to_owned());
395 }
396 _ if arg.starts_with("-fgnuc-version=") => {
397 let v = &arg["-fgnuc-version=".len()..];
398 opts.gnuc = v.parse().map_err(err)?;
399 }
400 "-fnested-functions" => {
405 return Err(err(
406 "nested functions are not supported: a call to one goes through a trampoline \
407 written on the stack, which no target that enforces an unexecutable stack \
408 allows",
409 ));
410 }
411 "-fno-nested-functions" => {}
412 "-static" => link.is_static = true,
416 "-shared" => link.shared = true,
417 "-pie" => link.pie = Some(true),
418 "-no-pie" | "-nopie" => link.pie = Some(false),
419 "-nostdlib" => link.no_stdlib = true,
420 "-nostartfiles" => link.no_startfiles = true,
421 "-nodefaultlibs" => link.no_defaultlibs = true,
422 "-fno-builtins-lib" => link.no_builtins_lib = true,
423 "-fbuiltins-lib" => link.no_builtins_lib = false,
424 "-rdynamic" | "-export-dynamic" => link.export_dynamic = true,
425 "-s" => link.strip = true,
426 "-Xlinker" => {
427 let next = args.get(i).ok_or_else(|| err("-Xlinker requires an argument"))?;
428 i += 1;
429 link.passthrough.push(next.clone());
430 }
431 _ if arg.starts_with("-Wl,") => {
432 link.passthrough.extend(arg["-Wl,".len()..].split(',').map(str::to_owned));
435 }
436 _ if arg.starts_with("-fuse-ld=") => {
437 link.use_ld = Some(arg["-fuse-ld=".len()..].to_owned());
438 }
439 _ if arg.starts_with("-l") && arg.len() > 2 => {
440 inputs.push(Input::library(&arg[2..]));
441 }
442 "-l" => {
443 let next = args.get(i).ok_or_else(|| err("-l requires an argument"))?;
444 i += 1;
445 inputs.push(Input::library(next));
446 }
447 _ if arg.starts_with("-L") => {
448 link.search.push(PathBuf::from(joined_or_next(arg, 2, args, &mut i)?));
449 }
450 _ if arg.starts_with("-B") => {
451 link.prefixes.push(PathBuf::from(joined_or_next(arg, 2, args, &mut i)?));
452 }
453 _ if arg.starts_with("-j") => {
454 jobs = Jobs::parse(&arg[2..]).map_err(err)?;
455 }
456 _ if arg.starts_with("--sysroot=") => {
457 sysroot = Some(PathBuf::from(&arg["--sysroot=".len()..]));
458 }
459 _ if arg.starts_with("--target=") => {
460 let t = &arg["--target=".len()..];
461 opts.target = t.parse().map_err(|e| err(format!("{e}")))?;
462 }
463 _ if arg.starts_with("--emit=") => {
464 let k = &arg["--emit=".len()..];
465 opts.emit = k
466 .parse()
467 .map_err(|()| err(format!("unknown --emit kind `{k}`, see --help")))?;
468 }
469 "-O" | "-Og" => opts.opt_level = rucc_session::OptLevel::O1,
475 "-Ofast" => {
481 return Err(err(
482 "-Ofast is -O3 with fast math, and fast math is not implemented, see \
483 spec/04-driver-and-cli.md section 4.6",
484 ));
485 }
486 _ if arg.starts_with("-O") => {
487 opts.opt_level = arg[2..]
488 .parse()
489 .map_err(|()| err(format!("unknown optimization level `{arg}`")))?;
490 }
491 _ if arg.starts_with("-fsafety=") => {
496 let tier = &arg["-fsafety=".len()..];
497 opts.safety = tier.parse().map_err(|()| {
498 err(format!(
499 "`{tier}` is not a safety tier, which is off, detect, enforce or kernel"
500 ))
501 })?;
502 }
503 _ if arg.starts_with("-fpass-fuel=") => {
507 let (name, count) = arg["-fpass-fuel=".len()..]
508 .split_once('=')
509 .ok_or_else(|| err("-fpass-fuel= is spelled <pass>=<count>"))?;
510 if rucc_opt::pass::find(name).is_none() {
511 return Err(err(format!(
512 "`{name}` is not a pass this compiler has, see --print-pipeline"
513 )));
514 }
515 let count: u32 = count
516 .parse()
517 .map_err(|_| err(format!("`{count}` is not a number of transformations")))?;
518 opts.pass_fuel.push((name.to_owned(), count));
519 }
520 _ if arg.starts_with("-fpass-fuel-global=") => {
521 let count = &arg["-fpass-fuel-global=".len()..];
522 let count: u32 = count
523 .parse()
524 .map_err(|_| err(format!("`{count}` is not a number of transformations")))?;
525 opts.pass_fuel_global = Some(count);
526 }
527 _ if arg == "-fopt-info"
532 || arg.starts_with("-fopt-info=")
533 || arg.starts_with("-fopt-info-") =>
534 {
535 let rest = &arg["-fopt-info".len()..];
536 let (kinds, file) = match rest.split_once('=') {
537 Some((kinds, file)) => (kinds, Some(file)),
538 None => (rest, None),
539 };
540 let kinds = kinds.strip_prefix('-').unwrap_or(kinds);
541 rucc_opt::Wants::none().add(kinds).map_err(err)?;
542 opts.opt_info.push(kinds.to_owned());
543 if let Some(file) = file {
544 if file.is_empty() {
545 return Err(err("-fopt-info= was given no file to write to"));
546 }
547 opts.opt_info_file = Some(file.to_owned());
548 }
549 }
550 _ if arg.starts_with("-fdump-ir=") => {
551 let spec = &arg["-fdump-ir=".len()..];
554 rucc_opt::Dumps::default().add(spec).map_err(err)?;
555 opts.dump_ir.push(spec.to_owned());
556 }
557 _ if arg.starts_with("-fdisable-") || arg.starts_with("-fenable-") => {
563 let on = arg.starts_with("-fenable-");
564 let spec = &arg[if on { "-fenable-".len() } else { "-fdisable-".len() }..];
565 rucc_opt::Gates::default().add(on, spec).map_err(err)?;
566 opts.pass_gates.push((on, spec.to_owned()));
567 }
568 _ if arg.strip_prefix("-fno-").is_some_and(|n| rucc_opt::pass::find(n).is_some()) => {
569 opts.passes.push((arg["-fno-".len()..].to_owned(), false));
570 }
571 _ if arg.strip_prefix("-f").is_some_and(|n| rucc_opt::pass::find(n).is_some()) => {
572 opts.passes.push((arg["-f".len()..].to_owned(), true));
573 }
574 "-Zverify-each" => opts.verify_each = true,
580 _ if arg.starts_with("-Zrule-coverage=") => {
581 let file = &arg["-Zrule-coverage=".len()..];
582 if file.is_empty() {
583 return Err(err("-Zrule-coverage= needs a file to write to"));
584 }
585 opts.rule_coverage = Some(file.to_owned());
586 }
587 _ if arg.starts_with("-Z") => {
588 return Err(err(format!(
589 "`{arg}` is not an unstable option this compiler has, see \
590 spec/04-driver-and-cli.md section 4.11 for the ones it does"
591 )));
592 }
593 "-m64" | "-m32" | "-mx32" => {
598 let want: u32 = match arg {
599 "-m64" => 64,
600 _ => 32,
601 };
602 let have = rucc_target::TargetInfo::new(opts.target).pointer_width;
603 if have != want {
604 return Err(err(format!(
605 "{arg} asks for a {want} bit target and {} is {have} bit, use \
606 --target= to name the one you mean",
607 opts.target
608 )));
609 }
610 }
611 _ if arg.starts_with("-march=")
617 || arg.starts_with("-mtune=")
618 || arg.starts_with("-mcpu=") => {}
619 _ if arg.starts_with("-mabi=") => {
622 let want = &arg["-mabi=".len()..];
623 let have = match opts.target.arch {
624 rucc_target::Arch::X86_64 => "sysv",
625 rucc_target::Arch::Aarch64 => "lp64",
626 rucc_target::Arch::Riscv64 => "lp64d",
627 };
628 if want != have {
629 return Err(err(format!(
630 "{arg}: {} uses the {have} convention and this compiler has no other",
631 opts.target
632 )));
633 }
634 }
635 "-mcmodel=small" => {}
639 _ if arg.starts_with("-mcmodel=") => {
640 return Err(err(format!(
641 "{arg}: this compiler emits the small code model and no other, see \
642 spec/12-targets.md"
643 )));
644 }
645 _ if arg.starts_with("-specs=") => {
649 return Err(err(
650 "-specs= is not supported: the parts of it builds rely on are -B, -L, \
651 -nostdlib, -nostartfiles and -Wl,, see spec/04-driver-and-cli.md \
652 section 4.4",
653 ));
654 }
655 _ if arg.starts_with("-Wa,") || arg.starts_with("-Wp,") => {
661 return Err(err(format!(
662 "`{arg}` is an argument for a separate assembler or preprocessor, and both \
663 are inside this compiler rather than programs it runs"
664 )));
665 }
666 "-Xassembler" | "-Xpreprocessor" => {
667 return Err(err(format!(
668 "{arg} hands an argument to a separate assembler or preprocessor, and both \
669 are inside this compiler rather than programs it runs"
670 )));
671 }
672 _ if arg.starts_with("-W") => {}
679 "-fno-ident"
685 | "-fident"
686 | "-funit-at-a-time"
687 | "-fno-unit-at-a-time"
688 | "-shared-libgcc"
689 | "-static-libgcc" => {}
690 _ if arg.starts_with('-') && arg.len() > 1 => {
691 return Err(err(format!("unknown option `{arg}`")));
696 }
697 _ => inputs.push(Input { path: arg.to_owned(), forced, library: false }),
698 }
699 }
700
701 link.sysroot = sysroot.clone();
708 if threads {
713 inputs.push(Input::library("pthread"));
714 }
715 if let Some(query) = query {
716 return Ok(Action::Print(answer(&query, &opts, &link)));
717 }
718 if !nostdinc {
719 opts.search.push_system(runtime::DIR);
720 for dir in library::system_dirs(opts.target, sysroot.as_deref()) {
724 opts.search.push_system(dir);
725 }
726 }
727 opts.search.remove_duplicates();
731
732 if print_config {
735 return Ok(Action::PrintConfig(Box::new(opts)));
736 }
737 if print_pipeline {
738 return Ok(Action::PrintPipeline(Box::new(opts)));
739 }
740 let plan = Plan::new(&opts, &inputs, output.as_deref()).map_err(|e| err(e.message))?;
741 if print_plan {
742 return Ok(Action::PrintPlan {
743 opts: Box::new(opts),
744 plan: Box::new(plan),
745 link: Box::new(link),
746 });
747 }
748 Ok(Action::Compile {
749 opts: Box::new(opts),
750 plan: Box::new(plan),
751 link: Box::new(link),
752 jobs,
753 verbose,
754 })
755}
756
757fn answer(query: &Query, opts: &Options, link: &LinkOptions) -> String {
763 let found = |name: &str| {
764 link::find_in_search(link, opts.target, name)
765 .map_or_else(|| name.to_owned(), |path| path.display().to_string())
766 };
767 match query {
768 Query::Machine => opts.target.to_string(),
769 Query::Version => VERSION.to_owned(),
770 Query::Multiarch => link::multiarch(opts.target),
771 Query::SearchDirs => {
776 let here = std::env::current_exe()
777 .ok()
778 .and_then(|p| p.parent().map(std::path::Path::to_path_buf))
779 .unwrap_or_default();
780 let list = |dirs: &[PathBuf]| {
781 dirs.iter().map(|d| d.display().to_string()).collect::<Vec<_>>().join(":")
782 };
783 let libraries = link::search_dirs(link, opts.target);
784 format!(
785 "install: {}\nprograms: ={}\nlibraries: ={}",
786 here.display(),
787 list(&link.prefixes),
788 list(&libraries)
789 )
790 }
791 Query::FileName(name) => found(name),
792 Query::Libgcc => found("libgcc.a"),
796 Query::ProgName(name) => link
800 .prefixes
801 .iter()
802 .map(|dir| dir.join(name))
803 .find(|path| path.is_file())
804 .map_or_else(|| name.clone(), |path| path.display().to_string()),
805 }
806}
807
808#[must_use]
814pub fn print_pipeline(opts: &Options) -> String {
815 let mut settings = rucc_opt::Options::for_level(opts.opt_level);
816 settings.toggles.clone_from(&opts.passes);
817 settings.global_fuel = opts.pass_fuel_global;
818 for (on, spec) in &opts.pass_gates {
819 let _ = settings.gates.add(*on, spec);
822 }
823 rucc_opt::pipeline::print(&settings)
824}
825
826#[must_use]
831pub fn print_config(opts: &Options) -> String {
832 let sess = Session::new(opts.clone());
833 let t = &sess.target;
834 let mut out = String::new();
835 let _ = writeln!(out, "version: {VERSION}");
836 let _ = writeln!(out, "target: {}", t.triple);
837 let _ = writeln!(out, "arch: {}", t.triple.arch.as_str());
838 let _ = writeln!(out, "os: {}", t.triple.os.as_str());
839 let _ = writeln!(out, "env: {}", t.triple.env.as_str());
840 let _ = writeln!(out, "object-format: {}", t.object_format.as_str());
841 let _ = writeln!(out, "pointer-width: {}", t.pointer_width);
842 let _ = writeln!(out, "long-width: {}", t.long_width);
843 let _ = writeln!(out, "long-double-width: {}", t.long_double_width);
844 let _ = writeln!(out, "endian: {}", if t.little_endian { "little" } else { "big" });
845 let _ = writeln!(out, "char-signed: {}", t.char_is_signed);
846 let _ = writeln!(out, "va-list: {}", t.va_list.as_str());
847 let regs: Vec<String> = t
850 .regs
851 .classes()
852 .map(|(class, info)| format!("{} {}", info.name, t.regs.len(class)))
853 .collect();
854 let _ = writeln!(
855 out,
856 "registers: {}",
857 if regs.is_empty() { "none".to_string() } else { regs.join(", ") }
858 );
859 let _ = writeln!(out, "opt-level: {}", sess.opts.opt_level);
860 let _ = writeln!(out, "safety: {}", sess.opts.safety);
861 let _ = writeln!(out, "emit: {}", sess.opts.emit.as_str());
862 let _ = writeln!(out, "debug-info: {}", sess.opts.debug_info);
863 let _ = writeln!(out, "frame-pointer: {}", sess.opts.frame_pointer);
864 let _ = writeln!(out, "red-zone: {}", sess.opts.red_zone);
865 for dir in sess.opts.search.dirs() {
868 let system = if dir.is_system { " (system)" } else { "" };
869 let _ = writeln!(out, "include: {}{system}", dir.path.display());
870 }
871 out
872}
873
874fn preprocess_all(opts: &Options, plan: &Plan) -> i32 {
880 let fs = OsFileSystem::new();
881 let mut stderr = std::io::stderr().lock();
882 let mut failed = false;
883 for job in &plan.jobs {
884 if !job.phases.first().is_some_and(|p| *p == Phase::Preprocess) {
885 continue;
888 }
889 let result = preprocess(opts, &job.input, &fs);
890 for message in &result.messages {
891 let _ = writeln!(stderr, "{message}");
892 }
893 if result.failed() {
894 failed = true;
895 continue;
896 }
897 if let Err(e) = write_out(&job.output, result.text.as_bytes()) {
898 let _ = writeln!(stderr, "rucc: error: {e}");
899 failed = true;
900 }
901 }
902 i32::from(failed)
903}
904
905fn compile_all(opts: &Options, plan: &Plan) -> i32 {
911 let fs = OsFileSystem::new();
912 let mut stderr = std::io::stderr().lock();
913 let mut failed = false;
914 let (mut remarks, ok) = Remarks::new(opts.opt_info_file.as_ref(), &mut stderr);
915 failed |= !ok;
916 let mut fired = Fired::new();
917 for job in &plan.jobs {
918 if !job.phases.contains(&Phase::Compile) {
919 continue;
920 }
921 let result = if job.kind == InputKind::Ir {
925 compile_ir(opts, &job.input, &fs)
926 } else {
927 compile(opts, &job.input, &fs)
928 };
929 fired.merge(&result.fired);
930 failed |= !write_dumps(&job.input, &result.dumps, &mut stderr);
931 failed |= !remarks.write(&result.remarks, &mut stderr);
932 for message in &result.messages {
933 let _ = writeln!(stderr, "{message}");
934 }
935 if result.failed() {
936 failed = true;
937 continue;
938 }
939 if let Err(e) = write_out(&job.output, result.artifact.bytes()) {
940 let _ = writeln!(stderr, "rucc: error: {e}");
941 failed = true;
942 }
943 }
944 failed |= !write_coverage(opts, &fired, &mut stderr);
945 i32::from(failed)
946}
947
948struct Scratch {
955 dir: PathBuf,
957}
958
959impl Scratch {
960 fn new() -> Result<Scratch, String> {
966 let dir = std::env::temp_dir().join(format!("rucc-{}", std::process::id()));
967 std::fs::create_dir_all(&dir).map_err(|e| format!("{}: {e}", dir.display()))?;
968 Ok(Scratch { dir })
969 }
970}
971
972impl Drop for Scratch {
973 fn drop(&mut self) {
974 let _ = std::fs::remove_dir_all(&self.dir);
975 }
976}
977
978fn link_line(opts: &Options, link: &LinkOptions, job: &LinkJob) -> Result<String, link::Error> {
985 let linker = link::find(opts.target, link)?;
986 let args = link::line(opts.target, link, &job.inputs, &job.output)?;
987 Ok(link::render(&linker, &args))
988}
989
990fn link_all(opts: &Options, plan: &Plan, link: &LinkOptions, verbose: bool) -> i32 {
997 let Some(job) = &plan.link else {
998 let mut stderr = std::io::stderr().lock();
1001 let _ = writeln!(stderr, "rucc: error: there is nothing to link");
1002 return 1;
1003 };
1004 let linker = match link::find(opts.target, link) {
1007 Ok(linker) => linker,
1008 Err(why) => return complain(why),
1009 };
1010
1011 let scratch = match Scratch::new() {
1012 Ok(scratch) => scratch,
1013 Err(why) => return complain(format!("could not make a place for the object files: {why}")),
1014 };
1015
1016 let fs = OsFileSystem::new();
1017 let mut failed = false;
1018 let mut produced: Vec<String> = Vec::with_capacity(plan.jobs.len());
1021 let mut fired = Fired::new();
1022 {
1023 let mut stderr = std::io::stderr().lock();
1024 let (mut remarks, ok) = Remarks::new(opts.opt_info_file.as_ref(), &mut stderr);
1025 failed |= !ok;
1026 for (at, job) in plan.jobs.iter().enumerate() {
1027 let out = match &job.output {
1028 Output::Temporary(hint) => {
1029 scratch.dir.join(format!("{at}-{hint}")).display().to_string()
1032 }
1033 Output::File(path) => path.clone(),
1034 Output::Stdout => continue,
1037 };
1038 produced.push(out.clone());
1039 if !job.phases.contains(&Phase::Compile) {
1040 continue;
1041 }
1042 let result = if job.kind == InputKind::Ir {
1043 compile_ir(opts, &job.input, &fs)
1044 } else {
1045 compile(opts, &job.input, &fs)
1046 };
1047 fired.merge(&result.fired);
1048 failed |= !write_dumps(&job.input, &result.dumps, &mut stderr);
1049 failed |= !remarks.write(&result.remarks, &mut stderr);
1050 for message in &result.messages {
1051 let _ = writeln!(stderr, "{message}");
1052 }
1053 if result.failed() {
1054 failed = true;
1055 continue;
1056 }
1057 if !matches!(result.artifact, Artifact::Object(_)) {
1058 let _ = writeln!(
1063 stderr,
1064 "rucc: internal error: {}: no object file was produced for the link",
1065 job.input
1066 );
1067 failed = true;
1068 continue;
1069 }
1070 if let Err(e) = std::fs::write(&out, result.artifact.bytes()) {
1071 let _ = writeln!(stderr, "rucc: error: {out}: {e}");
1072 failed = true;
1073 }
1074 }
1075 failed |= !write_coverage(opts, &fired, &mut stderr);
1076 }
1077 if failed {
1078 return 1;
1082 }
1083
1084 let mut outputs = produced.into_iter();
1088 let mut items = Vec::with_capacity(job.inputs.len());
1089 for item in &job.inputs {
1090 match item {
1091 link::Item::Library(name) => items.push(link::Item::Library(name.clone())),
1092 link::Item::File(_) => match outputs.next() {
1093 Some(path) => items.push(link::Item::File(path)),
1094 None => return complain("the plan asks the linker for a file nothing produced"),
1095 },
1096 }
1097 }
1098
1099 let args = match link::line(opts.target, link, &items, &job.output) {
1100 Ok(args) => args,
1101 Err(why) => return complain(why),
1102 };
1103 if verbose {
1104 let mut stderr = std::io::stderr().lock();
1105 let _ = writeln!(stderr, "{}", link::render(&linker, &args));
1106 }
1107 match link::run(&linker, &args) {
1108 Ok(()) => 0,
1109 Err(link::Error::Refused { .. }) => 1,
1112 Err(why) => complain(why),
1113 }
1114}
1115
1116fn complain(why: impl std::fmt::Display) -> i32 {
1118 let mut stderr = std::io::stderr().lock();
1119 let _ = writeln!(stderr, "rucc: error: {why}");
1120 1
1121}
1122
1123fn write_coverage(opts: &Options, fired: &Fired, stderr: &mut impl std::io::Write) -> bool {
1132 let Some(path) = &opts.rule_coverage else { return true };
1133 let Some(table) = coverage::table(opts.target.arch) else {
1134 let _ = writeln!(
1135 stderr,
1136 "rucc: error: there are no lowering rules for {} yet, so there is no coverage of them \
1137 to report",
1138 opts.target
1139 );
1140 return false;
1141 };
1142 match std::fs::write(path, fired.listing(table)) {
1143 Ok(()) => true,
1144 Err(e) => {
1145 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1146 false
1147 }
1148 }
1149}
1150
1151struct Remarks {
1158 file: Option<String>,
1160 started: bool,
1163}
1164
1165impl Remarks {
1166 fn new(file: Option<&String>, stderr: &mut impl std::io::Write) -> (Self, bool) {
1172 let mut ok = true;
1173 if let Some(path) = file {
1174 if let Err(e) = std::fs::write(path, "") {
1175 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1176 ok = false;
1177 }
1178 }
1179 (Self { file: file.cloned(), started: false }, ok)
1180 }
1181
1182 fn write(&mut self, text: &str, stderr: &mut impl std::io::Write) -> bool {
1188 if text.is_empty() {
1189 return true;
1190 }
1191 let Some(path) = &self.file else {
1192 let _ = write!(stderr, "{text}");
1193 return true;
1194 };
1195 let opened = std::fs::OpenOptions::new()
1196 .write(true)
1197 .append(self.started)
1198 .truncate(!self.started)
1199 .create(true)
1200 .open(path);
1201 self.started = true;
1202 let result =
1203 opened.and_then(|mut file| std::io::Write::write_all(&mut file, text.as_bytes()));
1204 if let Err(e) = result {
1205 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1206 return false;
1207 }
1208 true
1209 }
1210}
1211
1212fn write_dumps(input: &str, dumps: &[rucc_opt::Dump], stderr: &mut impl std::io::Write) -> bool {
1223 let stem = std::path::Path::new(input)
1224 .file_name()
1225 .map_or_else(|| input.to_owned(), |name| name.to_string_lossy().into_owned());
1226 let mut ok = true;
1227 for dump in dumps {
1228 let path = format!("{stem}.{}.ir", dump.name);
1229 if let Err(e) = std::fs::write(&path, &dump.text) {
1230 let _ = writeln!(stderr, "rucc: error: {path}: {e}");
1231 ok = false;
1232 }
1233 }
1234 ok
1235}
1236
1237fn write_out(output: &Output, bytes: &[u8]) -> Result<(), String> {
1244 match output {
1245 Output::Stdout => {
1246 let mut stdout = std::io::stdout().lock();
1247 stdout.write_all(bytes).map_err(|e| format!("writing to standard output: {e}"))
1248 }
1249 Output::File(path) | Output::Temporary(path) => {
1250 std::fs::write(path, bytes).map_err(|e| format!("{path}: {e}"))
1251 }
1252 }
1253}
1254
1255pub fn run(args: &[String]) -> i32 {
1260 match parse_args(args) {
1261 Ok(Action::Help) => {
1262 print!("{USAGE}");
1263 0
1264 }
1265 Ok(Action::Version) => {
1266 println!("rucc {VERSION}");
1267 0
1268 }
1269 Ok(Action::Print(line)) => {
1270 println!("{line}");
1271 0
1272 }
1273 Ok(Action::PrintConfig(opts)) => {
1274 print!("{}", print_config(&opts));
1275 0
1276 }
1277 Ok(Action::PrintPipeline(opts)) => {
1278 print!("{}", print_pipeline(&opts));
1279 0
1280 }
1281 Ok(Action::PrintPlan { opts, plan, link }) => {
1282 print!("{}", plan.render());
1283 if let Some(job) = &plan.link {
1287 match link_line(&opts, &link, job) {
1288 Ok(line) => println!("{line}"),
1289 Err(why) => {
1290 let mut stderr = std::io::stderr().lock();
1291 let _ = writeln!(stderr, "rucc: error: {why}");
1292 return 1;
1293 }
1294 }
1295 }
1296 0
1297 }
1298 Ok(Action::Compile { opts, plan, link, jobs, verbose }) => {
1299 {
1300 let mut stderr = std::io::stderr().lock();
1301 if verbose {
1302 let _ = write!(stderr, "{}", plan.render());
1303 let _ = writeln!(stderr, "workers: {}", jobs.count());
1304 }
1305 }
1306 if opts.emit == EmitKind::Preprocessed {
1307 return preprocess_all(&opts, &plan);
1308 }
1309 if opts.emit != EmitKind::Executable {
1310 return compile_all(&opts, &plan);
1311 }
1312 link_all(&opts, &plan, &link, verbose)
1313 }
1314 Err(e) => {
1315 let mut stderr = std::io::stderr().lock();
1316 let _ = writeln!(stderr, "rucc: error: {e}");
1317 let _ = writeln!(stderr, "rucc: note: run `rucc --help` for usage");
1318 1
1319 }
1320 }
1321}
1322
1323#[cfg(test)]
1324mod tests {
1325 use rucc_session::{GnucVersion, OptLevel};
1326
1327 use super::*;
1328
1329 fn args(s: &[&str]) -> Vec<String> {
1330 s.iter().map(|x| (*x).to_owned()).collect()
1331 }
1332
1333 #[test]
1334 fn help_and_version_win_over_everything_else() {
1335 assert_eq!(parse_args(&args(&["-c", "--help", "x.c"])).unwrap(), Action::Help);
1336 assert_eq!(parse_args(&args(&["--version"])).unwrap(), Action::Version);
1337 }
1338
1339 fn compile(s: &[&str]) -> (Box<Options>, Box<Plan>) {
1340 match parse_args(&args(s)).expect("expected a compilation") {
1341 Action::Compile { opts, plan, .. } => (opts, plan),
1342 other => panic!("expected a compilation, got {other:?}"),
1343 }
1344 }
1345
1346 fn linking(s: &[&str]) -> (Box<LinkOptions>, Box<Plan>) {
1347 match parse_args(&args(s)).expect("expected a compilation") {
1348 Action::Compile { link, plan, .. } => (link, plan),
1349 other => panic!("expected a compilation, got {other:?}"),
1350 }
1351 }
1352
1353 #[test]
1354 fn collects_inputs_and_flags() {
1355 let (opts, plan) = compile(&["-c", "-O2", "-g", "a.c", "b.c"]);
1356 let paths: Vec<&str> = plan.jobs.iter().map(|j| j.input.as_str()).collect();
1357 assert_eq!(paths, vec!["a.c", "b.c"]);
1358 assert_eq!(opts.opt_level, OptLevel::O2);
1359 assert_eq!(opts.emit, EmitKind::Object);
1360 assert!(opts.debug_info);
1361 }
1362
1363 #[test]
1366 fn an_unstable_option_is_taken_and_one_that_does_not_exist_is_refused() {
1367 let (opts, _) = compile(&["-c", "-Zrule-coverage=/tmp/rules.cov", "a.c"]);
1368 assert_eq!(opts.rule_coverage.as_deref(), Some("/tmp/rules.cov"));
1369
1370 let (plain, _) = compile(&["-c", "a.c"]);
1371 assert_eq!(plain.rule_coverage, None, "nothing is measured unless it was asked for");
1372
1373 assert!(parse_args(&args(&["-Zrule-coverage=", "a.c"])).is_err(), "a file with no name");
1374 let unknown = parse_args(&args(&["-Zwhat", "a.c"])).expect_err("there is no such option");
1375 assert!(unknown.message.contains("4.11"), "{}", unknown.message);
1376 }
1377
1378 #[test]
1379 fn a_bare_dash_o_means_o1_the_way_gcc_reads_it() {
1380 let (opts, _) = compile(&["-O", "a.c"]);
1381 assert_eq!(opts.opt_level, OptLevel::O1);
1382 }
1383
1384 #[test]
1385 fn dash_x_applies_to_later_inputs_only_and_none_stops_it() {
1386 let (_, plan) = compile(&["a.o", "-x", "c", "b.txt", "-x", "none", "c.o"]);
1387 assert_eq!(plan.jobs[0].kind, InputKind::LinkerInput);
1388 assert_eq!(plan.jobs[1].kind, InputKind::C);
1389 assert_eq!(plan.jobs[2].kind, InputKind::LinkerInput);
1390 }
1391
1392 #[test]
1393 fn dash_j_reaches_the_scheduler_and_defaults_to_the_machine() {
1394 let (_, _, jobs) = match parse_args(&args(&["-j4", "a.c"])).unwrap() {
1395 Action::Compile { opts, plan, jobs, .. } => (opts, plan, jobs),
1396 other => panic!("expected a compilation, got {other:?}"),
1397 };
1398 assert_eq!(jobs.count(), 4);
1399
1400 let default = match parse_args(&args(&["a.c"])).unwrap() {
1401 Action::Compile { jobs, .. } => jobs,
1402 other => panic!("expected a compilation, got {other:?}"),
1403 };
1404 assert_eq!(default, Jobs::available());
1405 assert!(parse_args(&args(&["-j0", "a.c"])).is_err());
1406 }
1407
1408 #[test]
1409 fn triple_hash_prints_the_plan_and_runs_nothing() {
1410 let a = parse_args(&args(&["-###", "-c", "a.c"])).unwrap();
1411 let Action::PrintPlan { plan, .. } = a else { panic!("expected a plan dump") };
1412 assert!(plan.render().contains("a.c: preprocess, compile, assemble -> a.o"));
1413 }
1414
1415 #[test]
1416 fn dash_x_names_what_it_accepts_when_it_does_not_know_a_language() {
1417 let e = parse_args(&args(&["-x", "fortran", "a.c"])).unwrap_err();
1418 assert!(e.message.contains("assembler-with-cpp"), "{}", e.message);
1419 }
1420
1421 #[test]
1422 fn an_unknown_flag_is_an_error_rather_than_a_shrug() {
1423 let e = parse_args(&args(&["-fno-such-thing", "a.c"])).unwrap_err();
1424 assert!(e.message.contains("unknown option"), "{}", e.message);
1425 }
1426
1427 #[test]
1428 fn asking_for_nested_functions_is_told_why_it_is_not_coming() {
1429 let e = parse_args(&args(&["-fnested-functions", "a.c"])).unwrap_err();
1430 assert!(e.message.contains("trampoline"), "{}", e.message);
1431 assert!(parse_args(&args(&["-fno-nested-functions", "a.c"])).is_ok());
1432 }
1433
1434 #[test]
1435 fn an_unsupported_target_names_itself() {
1436 let e = parse_args(&args(&["--target=sparc64-linux-gnu", "a.c"])).unwrap_err();
1437 assert!(e.message.contains("sparc64"), "{}", e.message);
1438 }
1439
1440 #[test]
1441 fn no_inputs_is_an_error_but_print_config_needs_none() {
1442 assert!(parse_args(&args(&[])).is_err());
1443 assert!(matches!(parse_args(&args(&["--print-config"])), Ok(Action::PrintConfig(_))));
1444 }
1445
1446 #[test]
1447 fn print_config_reports_the_target_it_was_given_not_the_host() {
1448 let a = parse_args(&args(&["--print-config", "--target=riscv64-linux-musl"])).unwrap();
1449 let Action::PrintConfig(opts) = a else { panic!("expected a configuration dump") };
1450 let text = print_config(&opts);
1451 assert!(text.contains("target: riscv64-unknown-linux-musl"), "{text}");
1452 assert!(text.contains("char-signed: false"), "{text}");
1453 assert!(text.contains("object-format: elf"), "{text}");
1454 assert!(text.contains("va-list: void-pointer"), "{text}");
1455 assert!(text.contains("registers: none"), "{text}");
1458 }
1459
1460 #[test]
1461 fn print_config_has_one_key_per_line_and_a_fixed_order() {
1462 let opts = Options::new("x86_64-unknown-linux-gnu".parse().unwrap());
1463 let text = print_config(&opts);
1464 let keys: Vec<&str> =
1465 text.lines().map(|l| l.split(':').next().unwrap_or_default()).collect();
1466 assert_eq!(keys[0], "version");
1467 assert_eq!(keys[1], "target");
1468 assert_eq!(keys.len(), 19);
1469 assert!(text.ends_with('\n'));
1470 }
1471
1472 #[test]
1473 fn the_safety_tier_is_read_off_the_command_line_and_a_wrong_one_is_refused() {
1474 let (opts, _) = compile(&["a.c"]);
1475 assert_eq!(opts.safety, rucc_session::Safety::Off);
1476
1477 for (flag, tier) in [
1478 ("-fsafety=detect", rucc_session::Safety::Detect),
1479 ("-fsafety=enforce", rucc_session::Safety::Enforce),
1480 ("-fsafety=kernel", rucc_session::Safety::Kernel),
1481 ("-fsafety=off", rucc_session::Safety::Off),
1482 ] {
1483 let (opts, _) = compile(&[flag, "a.c"]);
1484 assert_eq!(opts.safety, tier, "{flag}");
1485 }
1486
1487 let (opts, _) = compile(&["-fsafety=enforce", "-fsafety=off", "a.c"]);
1489 assert_eq!(opts.safety, rucc_session::Safety::Off);
1490
1491 let e = parse_args(&args(&["-fsafety=on", "a.c"])).unwrap_err();
1494 assert!(e.message.contains("is not a safety tier"), "{}", e.message);
1495 assert!(parse_args(&args(&["-fsafety", "a.c"])).is_err());
1496 }
1497
1498 #[test]
1499 fn print_pipeline_answers_with_the_passes_the_level_asked_for() {
1500 let a = parse_args(&args(&["--print-pipeline", "-O2"])).unwrap();
1501 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1502 let text = print_pipeline(&opts);
1503 assert!(text.starts_with("level: -O2\n"), "{text}");
1504 assert!(text.contains("fold"), "{text}");
1505
1506 let a = parse_args(&args(&["--print-pipeline"])).unwrap();
1507 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1508 assert!(print_pipeline(&opts).contains("1: simplify-cfg,"), "{}", print_pipeline(&opts));
1511
1512 let a = parse_args(&args(&["--print-pipeline", "-fno-simplify-cfg"])).unwrap();
1513 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1514 assert!(print_pipeline(&opts).contains("no passes"), "{}", print_pipeline(&opts));
1517 }
1518
1519 #[test]
1520 fn print_pipeline_takes_the_toggles_into_account() {
1521 let a = parse_args(&args(&["--print-pipeline", "-O2", "-fno-fold"])).unwrap();
1522 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1523 let text = print_pipeline(&opts);
1524 assert!(!text.contains("fold"), "{text}");
1527 assert!(text.contains("dce"), "{text}");
1528
1529 let mut off = vec!["--print-pipeline".to_owned(), "-O2".to_owned()];
1533 off.extend(rucc_opt::PASSES.iter().map(|p| format!("-fno-{}", p.name())));
1534 let spelled: Vec<&str> = off.iter().map(String::as_str).collect();
1535 let a = parse_args(&args(&spelled)).unwrap();
1536 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1537 assert!(print_pipeline(&opts).contains("no passes"), "{}", print_pipeline(&opts));
1538 }
1539
1540 #[test]
1541 fn print_pipeline_says_when_a_budget_will_stop_the_run_short() {
1542 let a = parse_args(&args(&["--print-pipeline", "-O2"])).unwrap();
1543 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1544 assert!(!print_pipeline(&opts).contains("global fuel"));
1545
1546 let a = parse_args(&args(&["--print-pipeline", "-O2", "-fpass-fuel-global=4"])).unwrap();
1547 let Action::PrintPipeline(opts) = a else { panic!("expected a pipeline dump") };
1548 let text = print_pipeline(&opts);
1549 assert!(text.contains("global fuel: 4"), "{text}");
1552 }
1553
1554 #[test]
1557 fn a_pass_is_named_by_dash_f_and_unnamed_by_dash_f_no() {
1558 let (opts, _) = compile(&["-c", "-O0", "-ffold", "-fno-fold", "-ffold", "a.c"]);
1559 assert_eq!(
1560 opts.passes,
1561 [("fold".to_owned(), true), ("fold".to_owned(), false), ("fold".to_owned(), true)]
1562 );
1563
1564 let e = parse_args(&args(&["-fno-such-pass", "a.c"])).unwrap_err();
1565 assert!(e.message.contains("unknown option"), "{}", e.message);
1566 }
1567
1568 #[test]
1569 fn pass_fuel_names_a_pass_and_a_count_and_refuses_anything_else() {
1570 let (opts, _) = compile(&["-c", "-O2", "-fpass-fuel=fold=3", "a.c"]);
1571 assert_eq!(opts.pass_fuel, [("fold".to_owned(), 3)]);
1572
1573 let e = parse_args(&args(&["-fpass-fuel=fold", "a.c"])).unwrap_err();
1574 assert!(e.message.contains("<pass>=<count>"), "{}", e.message);
1575 let e = parse_args(&args(&["-fpass-fuel=nosuch=3", "a.c"])).unwrap_err();
1576 assert!(e.message.contains("--print-pipeline"), "{}", e.message);
1577 let e = parse_args(&args(&["-fpass-fuel=fold=lots", "a.c"])).unwrap_err();
1578 assert!(e.message.contains("not a number"), "{}", e.message);
1579 }
1580
1581 #[test]
1582 fn global_pass_fuel_is_a_count_on_its_own_and_defaults_to_no_limit() {
1583 let (opts, _) = compile(&["-c", "-O2", "a.c"]);
1584 assert_eq!(opts.pass_fuel_global, None);
1585
1586 let (opts, _) = compile(&["-c", "-O2", "-fpass-fuel-global=12", "a.c"]);
1587 assert_eq!(opts.pass_fuel_global, Some(12));
1588 assert!(opts.pass_fuel.is_empty());
1591
1592 let e = parse_args(&args(&["-fpass-fuel-global=lots", "a.c"])).unwrap_err();
1593 assert!(e.message.contains("not a number"), "{}", e.message);
1594 }
1595
1596 #[test]
1597 fn a_gate_names_a_pass_and_optionally_the_functions_it_covers() {
1598 let (opts, _) = compile(&["-c", "-O2", "-fdisable-fold", "-fenable-fold=2-4,main", "a.c"]);
1599 assert_eq!(
1600 opts.pass_gates,
1601 [(false, "fold".to_owned()), (true, "fold=2-4,main".to_owned())],
1602 "the order is what decides, so it has to survive the parse"
1603 );
1604
1605 let e = parse_args(&args(&["-fdisable-nosuch", "a.c"])).unwrap_err();
1606 assert!(e.message.contains("--print-pipeline"), "{}", e.message);
1607 let e = parse_args(&args(&["-fenable-fold=9-2", "a.c"])).unwrap_err();
1608 assert!(e.message.contains("ends before it starts"), "{}", e.message);
1609 let e = parse_args(&args(&["-fdisable-fold=", "a.c"])).unwrap_err();
1610 assert!(e.message.contains("is empty"), "{}", e.message);
1611 }
1612
1613 #[test]
1614 fn the_pipeline_listing_says_which_passes_a_gate_touched() {
1615 let (opts, _) = compile(&["-c", "-O2", "-fdisable-fold=main", "a.c"]);
1616 let text = print_pipeline(&opts);
1617 assert!(text.contains("fold, "), "{text}");
1618 assert!(text.contains("[off for main]"), "{text}");
1619 }
1620
1621 #[test]
1625 fn a_dump_is_checked_when_it_is_asked_for_rather_than_when_it_is_taken() {
1626 let (opts, _) = compile(&["-c", "-O2", "-fdump-ir=all", "-fdump-ir=after-fold", "a.c"]);
1627 assert_eq!(opts.dump_ir, ["all", "after-fold"]);
1628
1629 let e = parse_args(&args(&["-fdump-ir=after-nosuch", "a.c"])).unwrap_err();
1630 assert!(e.message.contains("nosuch"), "{}", e.message);
1631 assert!(parse_args(&args(&["-fdump-ir=sideways-fold", "a.c"])).is_err());
1632 }
1633
1634 #[test]
1640 fn opt_info_takes_kinds_and_a_file_and_refuses_a_kind_it_does_not_have() {
1641 let (opts, _) = compile(&["-c", "-O2", "-fopt-info", "a.c"]);
1642 assert_eq!(opts.opt_info, [""], "a bare flag asks for the rewrites");
1643 assert_eq!(opts.opt_info_file, None, "and goes to standard error");
1644
1645 let (opts, _) = compile(&["-c", "-O2", "-fopt-info-missed-note", "a.c"]);
1646 assert_eq!(opts.opt_info, ["missed-note"]);
1647
1648 let (opts, _) =
1651 compile(&["-c", "-O2", "-fopt-info-missed=one.txt", "-fopt-info-all=two.txt", "a.c"]);
1652 assert_eq!(opts.opt_info, ["missed", "all"]);
1653 assert_eq!(opts.opt_info_file.as_deref(), Some("two.txt"));
1654
1655 let e = parse_args(&args(&["-fopt-info-vectorized", "a.c"])).unwrap_err();
1656 assert!(e.message.contains("vectorized"), "{}", e.message);
1657 assert!(e.message.contains("`missed`"), "{}", e.message);
1658 let e = parse_args(&args(&["-fopt-info-missed=", "a.c"])).unwrap_err();
1659 assert!(e.message.contains("no file"), "{}", e.message);
1660 }
1661
1662 #[test]
1663 fn verify_each_is_unstable_and_off_unless_it_was_asked_for() {
1664 let (opts, _) = compile(&["-c", "-Zverify-each", "a.c"]);
1665 assert!(opts.verify_each);
1666 assert!(!USAGE.contains("verify-each"), "an unstable option stays out of the usage text");
1667 }
1668
1669 #[test]
1670 fn dash_o_needs_an_argument() {
1671 let e = parse_args(&args(&["a.c", "-o"])).unwrap_err();
1672 assert_eq!(e.message, "-o requires an argument");
1673 }
1674
1675 #[test]
1676 fn dash_d_and_dash_u_are_read_joined_or_separated_and_keep_their_order() {
1677 let (opts, _) = compile(&["-DFOO=1", "-D", "BAR", "-UBAZ", "-U", "QUX", "a.c"]);
1678 assert_eq!(opts.defines, ["FOO=1", "BAR"]);
1679 assert_eq!(opts.undefines, ["BAZ", "QUX"]);
1680 }
1681
1682 #[test]
1683 fn the_include_flags_land_on_the_chain_each_one_names() {
1684 let (opts, _) = compile(&[
1687 "-Ii",
1688 "-iquote",
1689 "q",
1690 "-isystem",
1691 "sys",
1692 "-idirafter",
1693 "after",
1694 "--sysroot=/nowhere-at-all",
1695 "a.c",
1696 ]);
1697 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1698 assert_eq!(dirs, ["q", "i", "sys", runtime::DIR, "after"]);
1701 assert!(!opts.search.dirs()[1].is_system);
1702 assert!(opts.search.dirs()[2].is_system);
1703 }
1704
1705 #[test]
1706 fn the_librarys_headers_come_after_the_compilers_own_and_go_away_with_them() {
1707 let (opts, _) = compile(&["a.c"]);
1711 let dirs = opts.search.dirs();
1712 let ours = dirs.iter().position(|d| d.path.to_str() == Some(runtime::DIR));
1713 assert_eq!(ours, Some(0), "{dirs:?}");
1714 assert!(dirs[1..].iter().all(|d| d.is_system), "{dirs:?}");
1715 let (bare, _) = compile(&["-nostdinc", "a.c"]);
1716 assert!(bare.search.dirs().is_empty(), "{:?}", bare.search.dirs());
1717 }
1718
1719 #[test]
1720 fn a_sysroot_moves_the_librarys_directories_and_nothing_else() {
1721 let (opts, _) = compile(&["-isystem", "sys", "--sysroot=/nowhere-at-all", "a.c"]);
1722 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1723 assert_eq!(dirs, ["sys", runtime::DIR]);
1724 }
1725
1726 #[test]
1727 fn nostdinc_takes_the_compilers_own_headers_off_the_path() {
1728 let (opts, _) = compile(&["-Ii", "-nostdinc", "a.c"]);
1729 let dirs: Vec<&str> = opts.search.dirs().iter().filter_map(|d| d.path.to_str()).collect();
1730 assert_eq!(dirs, ["i"]);
1731 }
1732
1733 #[test]
1734 fn the_dialect_flags_set_the_language_and_the_extensions_separately() {
1735 let (opts, _) = compile(&["-std=gnu11", "a.c"]);
1736 assert_eq!(opts.std, Std::C11);
1737 assert!(opts.gnu_extensions);
1738
1739 let (opts, _) = compile(&["-std=iso9899:1999", "a.c"]);
1740 assert_eq!(opts.std, Std::C99);
1741 assert!(!opts.gnu_extensions);
1742
1743 let (opts, _) = compile(&["-ansi", "a.c"]);
1744 assert_eq!(opts.std, Std::C89);
1745 assert!(!opts.gnu_extensions);
1746
1747 let e = parse_args(&args(&["-std=c94jr", "a.c"])).unwrap_err();
1748 assert!(e.message.contains("unknown dialect"), "{}", e.message);
1749 }
1750
1751 #[test]
1752 fn the_dump_letters_are_a_family_and_everything_else_beginning_with_d_is_not() {
1753 let (opts, _) = compile(&["-dM", "a.c"]);
1754 assert!(opts.dumps.macros);
1755
1756 let (opts, _) = compile(&["-dDM", "a.c"]);
1759 assert!(opts.dumps.macros);
1760 let (opts, _) = compile(&["-dD", "a.c"]);
1761 assert!(!opts.dumps.macros);
1762
1763 let (opts, _) = compile(&["a.c"]);
1764 assert!(!opts.dumps.any());
1765
1766 assert_eq!(printed(&["-dumpversion", "a.c"]), VERSION);
1769 }
1770
1771 #[test]
1772 fn the_gcc_version_claimed_is_a_flag_and_the_short_spellings_are_the_ones_people_write() {
1773 let (opts, _) = compile(&["a.c"]);
1774 assert_eq!(
1775 opts.gnuc,
1776 GnucVersion { major: 7, minor: 0, patch: 0 },
1777 "the lowest claim a modern glibc gives its own declarations to"
1778 );
1779
1780 let (opts, _) = compile(&["-fgnuc-version=15.1.0", "a.c"]);
1781 assert_eq!(opts.gnuc, GnucVersion { major: 15, minor: 1, patch: 0 });
1782
1783 let (opts, _) = compile(&["-fgnuc-version=15", "a.c"]);
1786 assert_eq!(opts.gnuc, GnucVersion { major: 15, minor: 0, patch: 0 });
1787
1788 let (opts, _) = compile(&["-fgnuc-version=13.2", "a.c"]);
1789 assert_eq!(opts.gnuc, GnucVersion { major: 13, minor: 2, patch: 0 });
1790
1791 let e = parse_args(&args(&["-fgnuc-version=15.x", "a.c"])).unwrap_err();
1792 assert!(e.message.contains("minor that is not a number"), "{}", e.message);
1793
1794 let e = parse_args(&args(&["-fgnuc-version=1.2.3.4", "a.c"])).unwrap_err();
1795 assert!(e.message.contains("more than three"), "{}", e.message);
1796 }
1797
1798 #[test]
1799 fn pedantic_has_two_spellings_and_is_not_the_same_knob_as_the_dialect() {
1800 let (opts, _) = compile(&["-std=c17", "-pedantic", "a.c"]);
1801 assert!(opts.pedantic);
1802 assert_eq!(opts.std, Std::C17);
1803
1804 let (opts, _) = compile(&["-Wpedantic", "a.c"]);
1807 assert!(opts.pedantic);
1808
1809 let (opts, _) = compile(&["-std=c17", "a.c"]);
1810 assert!(!opts.pedantic, "a dialect on its own does not diagnose an extension");
1811 }
1812
1813 #[test]
1814 fn dash_p_and_dash_ffreestanding_reach_the_options() {
1815 let (opts, _) = compile(&["-E", "-P", "-ffreestanding", "a.c"]);
1816 assert!(!opts.line_markers);
1817 assert!(!opts.hosted);
1818 assert_eq!(opts.emit, EmitKind::Preprocessed);
1819 }
1820
1821 #[test]
1828 fn the_builtin_flags_are_read_in_both_directions_and_one_name_at_a_time() {
1829 let (opts, _) = compile(&["-c", "a.c"]);
1830 assert!(opts.builtins, "a library name means the library function by default");
1831 assert!(opts.no_builtin.is_empty());
1832
1833 let (opts, _) = compile(&["-c", "-fno-builtin", "a.c"]);
1834 assert!(!opts.builtins);
1835
1836 let (opts, _) = compile(&["-c", "-fno-builtin", "-fbuiltin", "a.c"]);
1837 assert!(opts.builtins, "the last mention decides");
1838
1839 let (opts, _) = compile(&["-c", "-fno-builtin-memcpy", "-fno-builtin-nonesuch", "a.c"]);
1840 assert!(opts.builtins, "one name is not the family");
1841 assert_eq!(opts.no_builtin, vec!["memcpy".to_owned(), "nonesuch".to_owned()]);
1842 }
1843
1844 #[test]
1847 fn gnu89_inline_is_off_until_it_is_asked_for_and_the_last_mention_decides() {
1848 let (opts, _) = compile(&["-c", "a.c"]);
1849 assert!(!opts.gnu89_inline, "C's reading of inline by default");
1850
1851 let (opts, _) = compile(&["-c", "-fgnu89-inline", "a.c"]);
1852 assert!(opts.gnu89_inline);
1853
1854 let (opts, _) = compile(&["-c", "-fgnu89-inline", "-fno-gnu89-inline", "a.c"]);
1855 assert!(!opts.gnu89_inline, "the last mention decides");
1856
1857 let (opts, _) = compile(&["-c", "-std=c89", "a.c"]);
1862 assert!(!opts.gnu89_inline);
1863 }
1864
1865 #[test]
1868 fn the_two_frame_flags_are_read_in_both_directions() {
1869 let (opts, _) = compile(&["-c", "a.c"]);
1870 assert!(!opts.frame_pointer, "gcc omits it above -O0 and so does this");
1871 assert!(opts.red_zone, "the psABI has one and nothing said not to use it");
1872
1873 let (opts, _) = compile(&["-c", "-fno-omit-frame-pointer", "-mno-red-zone", "a.c"]);
1874 assert!(opts.frame_pointer);
1875 assert!(!opts.red_zone);
1876
1877 let (opts, _) = compile(&[
1878 "-c",
1879 "-fno-omit-frame-pointer",
1880 "-fomit-frame-pointer",
1881 "-mno-red-zone",
1882 "-mred-zone",
1883 "a.c",
1884 ]);
1885 assert!(!opts.frame_pointer, "the last one wins, as it does in gcc");
1886 assert!(opts.red_zone);
1887 }
1888
1889 #[test]
1890 fn the_link_flags_are_collected_apart_from_the_compilation() {
1891 let (link, _) = linking(&[
1892 "-static",
1893 "-nostartfiles",
1894 "-rdynamic",
1895 "-s",
1896 "-fuse-ld=mold",
1897 "-L/opt/lib",
1898 "-B",
1899 "/opt/tools",
1900 "a.c",
1901 ]);
1902 assert!(link.is_static);
1903 assert!(link.no_startfiles);
1904 assert!(link.export_dynamic);
1905 assert!(link.strip);
1906 assert_eq!(link.use_ld.as_deref(), Some("mold"));
1907 assert_eq!(link.search, vec![PathBuf::from("/opt/lib")]);
1908 assert_eq!(link.prefixes, vec![PathBuf::from("/opt/tools")]);
1909 }
1910
1911 #[test]
1912 fn a_comma_in_dash_wl_separates_two_arguments() {
1913 let (link, _) = linking(&["-Wl,-rpath,/opt/lib", "-Xlinker", "--as-needed", "a.c"]);
1914 assert_eq!(link.passthrough, vec!["-rpath", "/opt/lib", "--as-needed"]);
1915 }
1916
1917 #[test]
1918 fn a_library_keeps_its_place_between_the_objects() {
1919 let (_, plan) = linking(&["--target=x86_64-unknown-linux-gnu", "a.c", "-lm", "b.c"]);
1924 let link = plan.link.expect("expected a link step");
1925 assert_eq!(
1926 link.inputs,
1927 vec![
1928 link::Item::File("a.o".into()),
1929 link::Item::Library("m".into()),
1930 link::Item::File("b.o".into()),
1931 ]
1932 );
1933 assert_eq!(plan.jobs.len(), 2);
1935 }
1936
1937 #[test]
1938 fn a_library_on_a_dash_c_line_is_a_note_rather_than_an_error() {
1939 let (_, plan) = linking(&["-c", "-lm", "a.c"]);
1940 assert!(plan.link.is_none());
1941 assert!(plan.notes.iter().any(|n| n.contains("-lm")), "{:?}", plan.notes);
1942 }
1943
1944 #[test]
1945 fn the_sysroot_reaches_the_linker_as_well_as_the_headers() {
1946 let (link, _) = linking(&["--sysroot=/opt/root", "a.c"]);
1947 assert_eq!(link.sysroot, Some(PathBuf::from("/opt/root")));
1948 }
1949
1950 fn printed(s: &[&str]) -> String {
1951 match parse_args(&args(s)).expect("expected an answer") {
1952 Action::Print(line) => line,
1953 other => panic!("expected an answer, got {other:?}"),
1954 }
1955 }
1956
1957 fn refused(s: &[&str]) -> String {
1958 parse_args(&args(s)).expect_err("expected a refusal").message
1959 }
1960
1961 #[test]
1962 fn a_warning_flag_this_compiler_has_not_heard_of_is_taken_rather_than_refused() {
1963 let (opts, _) = compile(&["-Wall", "-Wextra", "-Wno-format-truncation", "-c", "a.c"]);
1967 assert!(!opts.warnings_are_errors);
1968 assert!(opts.warnings);
1969 let (opts, _) = compile(&["-Werror", "-c", "a.c"]);
1971 assert!(opts.warnings_are_errors);
1972 let (opts, _) = compile(&["-w", "-c", "a.c"]);
1973 assert!(!opts.warnings);
1974 let (opts, _) = compile(&["-pedantic-errors", "-c", "a.c"]);
1975 assert!(opts.pedantic && opts.warnings_are_errors);
1976 }
1977
1978 #[test]
1979 fn an_argument_for_a_separate_tool_is_refused_rather_than_dropped() {
1980 assert!(refused(&["-Wa,--noexecstack", "-c", "a.c"]).contains("separate assembler"));
1982 assert!(refused(&["-Wp,-DX", "-c", "a.c"]).contains("separate assembler"));
1983 assert!(refused(&["-specs=/x", "a.c"]).contains("-specs= is not supported"));
1984 assert!(refused(&["-mcmodel=kernel", "-c", "a.c"]).contains("small code model"));
1985 assert!(refused(&["-gdwarf-4", "-c", "a.c"]).contains("DWARF 5"));
1986 assert!(refused(&["-Ofast", "-c", "a.c"]).contains("fast math"));
1987 let no32 = refused(&["--target=x86_64-unknown-linux-gnu", "-m32", "-c", "a.c"]);
1990 assert!(no32.contains("32 bit target"), "{no32}");
1991 }
1992
1993 #[test]
1994 fn the_levels_gcc_spells_differently_are_the_levels_they_mean() {
1995 assert_eq!(compile(&["-O", "-c", "a.c"]).0.opt_level, OptLevel::O1);
1996 assert_eq!(compile(&["-Og", "-c", "a.c"]).0.opt_level, OptLevel::O1);
1997 assert_eq!(compile(&["-O2", "-c", "a.c"]).0.opt_level, OptLevel::O2);
1998 }
1999
2000 #[test]
2001 fn the_machine_flags_that_name_what_we_already_do_are_taken_and_the_rest_are_not() {
2002 let line = ["--target=x86_64-unknown-linux-gnu", "-m64", "-march=x86-64-v3"];
2003 let (opts, _) =
2004 compile(&[&line[..], &["-mtune=native", "-mabi=sysv", "-c", "a.c"]].concat());
2005 assert_eq!(opts.target.to_string(), "x86_64-unknown-linux-gnu");
2006 let wrong = refused(&["--target=x86_64-unknown-linux-gnu", "-mabi=ms", "-c", "a.c"]);
2007 assert!(wrong.contains("sysv convention"), "{wrong}");
2008 }
2009
2010 #[test]
2011 fn the_thread_flag_is_a_macro_and_a_library_and_the_library_goes_last() {
2012 let (opts, plan) = compile(&["-pthread", "-c", "a.c"]);
2013 assert!(opts.defines.iter().any(|d| d == "_REENTRANT"));
2014 let names: Vec<&str> = plan.jobs.iter().map(|j| j.input.as_str()).collect();
2017 assert_eq!(names, vec!["a.c"]);
2018 }
2019
2020 #[test]
2021 fn the_questions_a_build_system_asks_before_it_compiles_anything() {
2022 let target = "--target=x86_64-unknown-linux-gnu";
2023 assert_eq!(printed(&[target, "-dumpmachine"]), "x86_64-unknown-linux-gnu");
2024 assert_eq!(printed(&[target, "-dumpversion"]), VERSION);
2025 assert_eq!(printed(&[target, "-dumpfullversion"]), VERSION);
2026 assert_eq!(printed(&[target, "-print-multiarch"]), "x86_64-linux-gnu");
2027 assert_eq!(printed(&[target, "-print-file-name=no-such-library.a"]), "no-such-library.a");
2030 assert_eq!(printed(&[target, "-print-prog-name=ld"]), "ld");
2031 let dirs = printed(&[target, "-print-search-dirs"]);
2032 assert!(dirs.starts_with("install: "), "{dirs}");
2033 assert!(dirs.contains("\nlibraries: ="), "{dirs}");
2034 }
2035
2036 #[test]
2037 fn usage_fits_on_a_screen() {
2038 assert!(USAGE.lines().count() < 42, "usage text has grown past one screen");
2048 }
2049}