Skip to main content

rucc_driver/
lib.rs

1//! The driver: command line parsing, the phase graph, job scheduling and the linker
2//! invocation.
3//!
4//! Design: `spec/04-driver-and-cli.md`. Layer rank 12, see `spec/18-package-layout.md`.
5//!
6//! This is the only crate that is allowed to know the process exists. It reads the command
7//! line, touches the file system, spawns the linker and writes to the terminal, and it hands
8//! everything below it a [`Session`]. The binary crate is a `main` that calls
9//! [`run`] and nothing else, so that the whole driver is reachable from a test.
10//!
11//! # Status
12//!
13//! `--help`, `--version` and `--print-config` are real, which is the `M0` exit criterion in
14//! `spec/17-milestones.md`. The phase graph is real and `-###` prints it, and the scheduler
15//! that will run it is real and tested.
16//!
17//! Two phases run. `-E` reads the file, runs phase 4 over it and writes the result, to `-o` or
18//! to standard output. `--emit=tast` carries on through phase 7, the parse and the checking,
19//! and writes the typed tree. The flags those two read are real with them, which is `-D`, `-U`,
20//! `-I`, `-iquote`, `-isystem`, `-idirafter`, `--sysroot=`, `-isysroot`, `-P`, `-std=`,
21//! `-fgnuc-version=`, `-ansi`, `-ffreestanding`, `-fno-builtin`, `-fno-builtin-<name>`,
22//! `-fgnu89-inline`, `-pedantic` and `-Werror`.
23//! The phases after them still say they are not implemented.
24//!
25//! This crate is tier 3 in `spec/18-package-layout.md` section 18.5: its Rust API is
26//! explicitly unstable and will change without a major version bump.
27
28#![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
53/// The compiler's version, taken from the workspace manifest.
54pub const VERSION: &str = env!("CARGO_PKG_VERSION");
55
56/// What the command line asked for.
57#[derive(Debug, Clone, PartialEq, Eq)]
58pub enum Action {
59    /// Print usage and exit successfully.
60    Help,
61    /// Print the version and exit successfully.
62    Version,
63    /// Print one line and exit successfully, which is what the `-dump` and `-print` family do.
64    ///
65    /// A build system asks these before it compiles anything, and what it does with the answer
66    /// is paste it into a path or into another command line, so each one is a single line with
67    /// no decoration around it.
68    Print(String),
69    /// Print the resolved configuration and exit successfully.
70    PrintConfig(Box<Options>),
71    /// Print the passes the level will run and exit successfully.
72    PrintPipeline(Box<Options>),
73    /// Print the phase plan and the link line and exit successfully, which is `-###`.
74    PrintPlan {
75        /// The resolved options, which is what says what the link line is for.
76        opts: Box<Options>,
77        /// What to do to each input, and in what order.
78        plan: Box<Plan>,
79        /// What the command line said about linking.
80        link: Box<LinkOptions>,
81    },
82    /// Compile the given inputs.
83    Compile {
84        /// The resolved options.
85        opts: Box<Options>,
86        /// What to do to each input, and in what order.
87        plan: Box<Plan>,
88        /// What the command line said about linking.
89        link: Box<LinkOptions>,
90        /// How many translation units to compile at once.
91        jobs: Jobs,
92        /// Whether `-v` asked for the plan to be printed while it runs.
93        verbose: bool,
94    },
95}
96
97/// Why a command line was rejected.
98#[derive(Debug, Clone, PartialEq, Eq)]
99pub struct CliError {
100    /// The message, lowercase and without a trailing period, in the same shape as any other
101    /// diagnostic.
102    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
117/// A question the command line asked instead of asking for a compilation.
118///
119/// These are answered after the loop rather than where they are read, because every one of them
120/// is about the target or about the library search and the last word on both is the end of the
121/// command line.
122enum Query {
123    /// `-dumpmachine`, the triple.
124    Machine,
125    /// `-dumpversion` and `-dumpfullversion`, which are the same three numbers here.
126    Version,
127    /// `-print-multiarch`, the directory name a distribution files this target under.
128    Multiarch,
129    /// `-print-search-dirs`, in the three lines GCC prints.
130    SearchDirs,
131    /// `-print-file-name=<name>`, the full path of a library file.
132    FileName(String),
133    /// `-print-prog-name=<name>`, the full path of a program.
134    ProgName(String),
135    /// `-print-libgcc-file-name`, which is `-print-file-name=libgcc.a` under another spelling.
136    Libgcc,
137}
138
139/// Usage text.
140///
141/// Deliberately short. `spec/04-driver-and-cli.md` puts the full flag reference in the
142/// manual page, because a `--help` nobody can read in one screen is a `--help` nobody reads.
143pub 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
187/// The argument of a flag that may be joined to it or may be the next word.
188///
189/// `-DFOO` and `-D FOO` are the same thing, and `at` is where the flag's own letters end.
190fn 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
204/// Parses a command line, without the program name.
205///
206/// # Errors
207///
208/// Returns the message to print when the arguments do not name a compilation this compiler
209/// can attempt.
210pub 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    // `-x` applies to inputs that come after it and stays in effect until the next one, which
227    // is why it is tracked across the loop rather than attached to a single argument.
228    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            // GCC's own levels of how much debug information to write. Zero is none and every
246            // other number is some, and this compiler has one amount, so the numbers above zero
247            // all mean the same thing here. `-ggdb` is the same flag asking for whatever the
248            // debugger on the machine prefers, which is what we emit anyway.
249            "-g0" => opts.debug_info = false,
250            "-g1" | "-g2" | "-g3" | "-ggdb" | "-ggdb1" | "-ggdb2" | "-ggdb3" => {
251                opts.debug_info = true;
252            }
253            // The version of DWARF to write. We write DWARF 5 and nothing else, so a build that
254            // asks for another version is told rather than handed a file it cannot read.
255            "-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            // Nothing that is not fatal is said at all. Read at the one place a diagnostic goes
264            // through rather than here, so that a warning `-w` dropped is not counted either.
265            "-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            // The questions a build system asks before it compiles anything. Answered after the
272            // loop, because each one is about the target or the library search and the command
273            // line has not finished saying what those are.
274            "-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            // A program built to run in more than one thread. On every platform this compiler
286            // targets that is a macro the library's headers read and one more library on the
287            // link line, and the library is added after the loop so that it lands after the
288            // objects that refer to it.
289            "-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            // `-Wpedantic` is the same flag under the name the `-W` family gives it, which is
298            // the spelling a build system that groups its warning flags tends to write.
299            "-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            // The C89 dialects are under GNU's reading whatever this says, so turning it off
305            // there is turning off something the dialect asked for, which is accepted and does
306            // nothing. gcc refuses that command line, and there is nothing it could have meant.
307            "-fgnu89-inline" => opts.gnu89_inline = true,
308            "-fno-gnu89-inline" => opts.gnu89_inline = false,
309            // Both directions of each, because a build system that wants one of these usually
310            // writes it beside the flag that turns it back off for one directory.
311            "-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            // GCC drops its own include directory along with the system ones, because its
316            // headers are half of a pair with the library's and half a pair is worse than
317            // none. A build that passes this is supplying the whole set itself.
318            "-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            // The flags that take a directory only in the separated form. GCC spells them
324            // this way and nothing writes `-iquotedir`, so accepting the joined form would
325            // mean guessing at a path that starts with the flag's own letters.
326            // Apple's spelling of `--sysroot`, and the one its own build systems pass. The
327            // two mean the same thing here: the configured directories are under there rather
328            // than under the root.
329            "-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            // Not a GCC flag. spec/03-architecture.md section 3.5 compiles several
353            // translation units in one process rather than making the build system fork, and
354            // section 3.8's determinism check compares `-j1` against `-j16`, so the knob has
355            // to exist and has to be spelled the way `make` spells it.
356            // `-DFOO`, `-D FOO` and the same for `-U` and `-I`. Both forms are in wide use
357            // and a build system may produce either, so both are read here rather than
358            // being normalised by whatever generated the command line.
359            _ 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            // Section 4.5. The claim decides which half of glibc's `sys/cdefs.h` we are
379            // handed, so a differential run that does not set it is comparing two compilers
380            // that believe they are different compilers.
381            // GCC packs these into one flag, so `-dDI` is two of them. Letters in the family
382            // that we have not written yet are accepted and ignored, because a dump is a
383            // debugging aid and a build that asks for one should still compile. A letter
384            // outside the family falls through to the unknown option error, which is what
385            // keeps `-dumpversion` from being read as a dump of nothing.
386            _ if Dumps::is_family(arg) => {
387                opts.dumps.add(&arg[2..]);
388            }
389            // One name at a time, which is what a build that means its own `memcpy` and the
390            // library's everything else writes. The name is not checked against a list, because
391            // the flag is about what the program means by a name and a program is allowed to mean
392            // something by a name this compiler has never heard of.
393            _ 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            // spec/13-gnu-compat.md section 13.3 promises this flag an error that says why rather
401            // than the unknown option one, because a build reaching for it is asking for a feature
402            // and deserves to be told it is not coming rather than told the spelling is wrong.
403            // The negative form is what this compiler does anyway, so it is taken and dropped.
404            "-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            // The link flags. None of them changes the compilation, which is why they are
413            // collected apart from `opts` and why `-lm` on a `-c` line is a note rather than an
414            // error: it is a thing said to a linker that is not going to run.
415            "-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                // Commas separate arguments rather than being part of one, which is what makes
433                // `-Wl,-rpath,/opt/lib` two words to the linker and one word here.
434                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            // A bare `-O` is `-O1`, which is what GCC has and what a hand written makefile tends
470            // to write. `-Og` is GCC's level for a build somebody is going to step through, and
471            // it is `-O1` with the transformations that move code around left out; this compiler
472            // has no such level yet, so it is the nearest one and `--print-pipeline` says what
473            // that came to rather than the flag pretending otherwise.
474            "-O" | "-Og" => opts.opt_level = rucc_session::OptLevel::O1,
475            // The union of `-O3` and `-ffast-math`, and the second half of that changes what
476            // floating point arithmetic means. Refused rather than taken as `-O3`, because a
477            // build that asks for fast math and is quietly given ordinary arithmetic gets a
478            // slower program than it asked for and a build that is given fast math it did not
479            // ask for gets a wrong one.
480            "-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            // The memory safety monitor, from section 15.4 of
492            // `spec/safe-memory/15-integration.md`. Before the optimizer's `-f` family below,
493            // because a pass that took the name `safety=detect` would otherwise be handed the
494            // flag, and the tier is not a pass.
495            _ 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            // The optimizer's own flags, from section 9.10 of `spec/09-optimizer.md`. These come
504            // after every `-f` the rest of the compiler answers to, so a pass can never take a
505            // name that already means something else on the command line.
506            _ 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            // Everything from `-fopt-info` to the end of the argument, which is optional
528            // keywords joined by hyphens and an optional `=<file>`. Checked here rather than
529            // where the remarks are printed, because by then the compilation somebody wanted
530            // to hear about is over.
531            _ 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                // Checked here rather than where the dumps are taken, because the compilation
552                // that would have been dumped is over by then.
553                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            // Before the bare `-f<pass>` below, because a pass called `enable-something` would
558            // otherwise take the flag away from the gate. Checked here rather than where the
559            // pipeline reads it, for the reason that applies to all of these: a misspelled pass
560            // name that quietly gated nothing looks exactly like a pass that is not the guilty
561            // one, and a bisection would carry on past the thing it was looking for.
562            _ 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            // The unstable options, spelled the way rustc spells them and carrying the same
575            // promise, which is none: one of these may change or go away in any release. They are
576            // measurements and debugging aids rather than things a build asks for, which is why
577            // none of them is in the usage text and all of them are in section 4.11 of
578            // `spec/04-driver-and-cli.md`.
579            "-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            // The word size, which is a statement about the target and is taken as one. A build
594            // that says the size the target already has is saying nothing, and one that says the
595            // other size is asking for a target this compiler does not have, which it is told
596            // rather than being given the wrong one.
597            "-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            // Which processor in the family to generate for. This compiler emits the base
612            // instruction set of the architecture and nothing above it, so a program built with
613            // any of these runs on the machine that was named; it is a program that could have
614            // been faster rather than a program that is wrong, which is what makes these safe to
615            // take and ignore where a flag that changed the meaning of the code would not be.
616            _ if arg.starts_with("-march=")
617                || arg.starts_with("-mtune=")
618                || arg.starts_with("-mcpu=") => {}
619            // The calling convention, which is not safe to ignore. Taken when it names the one
620            // the target already uses and refused otherwise.
621            _ 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            // How far apart the pieces of the program may be. The small model is what we emit and
636            // it is every hosted program's default; the kernel model is a different one and a
637            // build that asks for it and does not get it links and then does not run.
638            "-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            // GCC's own scripting language for how the driver builds a command line.
646            // `spec/04-driver-and-cli.md` section 4.4 settles that we will not have it, so a
647            // build reaching for it is told which flags do the same job.
648            _ 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            // Arguments meant for a separate assembler or preprocessor, which this compiler does
656            // not have: both are inside it and neither reads a command line. Refused rather than
657            // dropped, because every one of these says something about the output and a build
658            // that asked for `-Wa,--noexecstack` and was silently given an executable stack got
659            // the opposite of what it asked for.
660            _ 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            // Everything else in the `-W` family. `spec/04-driver-and-cli.md` section 4.1 has
673            // this one as a rule about build systems rather than about warnings: autoconf finds
674            // out whether a warning flag exists by passing it and looking at the exit status, so
675            // a compiler that refuses one it has not heard of fails a configure script written
676            // for a GCC newer than itself. The names are not checked against a list because this
677            // compiler has no warning groups for a list to be of, which #485 is about.
678            _ if arg.starts_with("-W") => {}
679            // Flags that name something this compiler does not do and would not do differently
680            // if it did. `-fno-ident` is about a comment in the output that we do not write
681            // either way, and the others are about a way of ordering the compilation that has
682            // been GCC's only way for twenty years. Section 4.1 asks for the list to be short
683            // and for adding to it to be deliberate, which is why it is written out here.
684            "-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                // Silently ignoring an unknown flag is how a build ends up not doing what
692                // its author asked. spec/13-gnu-compat.md section 13.4 makes this an error
693                // for the flags that change code generation, and the safe default until the
694                // flag table is populated is to reject everything we do not know.
695                return Err(err(format!("unknown option `{arg}`")));
696            }
697            _ => inputs.push(Input { path: arg.to_owned(), forced, library: false }),
698        }
699    }
700
701    // Last, so that it lands after every `-isystem` the command line gave. That is GCC's
702    // order: a directory the user names outranks the compiler's own, and the compiler's own
703    // outranks the library's. It is pushed after the loop rather than before it because
704    // `SearchPath` appends within a group and the position is what the order is.
705    // The same directory the headers were looked for under, because a sysroot is a statement
706    // about a whole installation and not about half of one.
707    link.sysroot = sysroot.clone();
708    // After the loop rather than where `-pthread` was read, so that it lands after the objects
709    // that refer to it. A static link takes the definitions it needs from a library when it
710    // reaches it and not afterwards, so a library before the objects is a library that answers
711    // nothing.
712    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        // And the library's after ours, which is the other half of the same order. They go on
721        // here rather than at the point `--target=` or `--sysroot=` was read because either
722        // one changes the answer and the last word on both is the end of the loop.
723        for dir in library::system_dirs(opts.target, sysroot.as_deref()) {
724            opts.search.push_system(dir);
725        }
726    }
727    // Once, here, rather than as each directory is pushed. A `-I` that names a system
728    // directory has to lose to the system entry and the system entry is added last, so the
729    // question cannot be answered until the whole path is known.
730    opts.search.remove_duplicates();
731
732    // The target has to be resolved before the configuration is printed, so this check comes
733    // after the loop rather than at the point `--print-config` was seen.
734    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
757/// What one of the `-dump` and `-print` flags prints.
758///
759/// GCC prints the name back unchanged when it cannot find the file a `-print` flag asked about,
760/// which is what makes the answer safe to paste into a link line whether or not the file is
761/// there, and this does the same.
762fn 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        // The three lines GCC prints, in its order and with its punctuation, because what reads
772        // them is a script written against that shape. There is no installation directory to
773        // report: this compiler is one binary that works wherever it is copied, and the headers
774        // it ships are inside it, so `install` is where the binary is and nothing is under it.
775        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        // The name GCC gives the library of routines a compiler's output calls that the C
793        // library does not have. Ours is built in and there is no file, so the answer is the
794        // name itself, which is what GCC prints when it cannot find one either.
795        Query::Libgcc => found("libgcc.a"),
796        // A program rather than a library: the linker and the archiver are the ones a build asks
797        // about, and this compiler finds them on the path or under `-B` rather than shipping
798        // them, so the name back is the honest answer unless a `-B` prefix holds one.
799        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/// Renders the passes this level will run, in order, with what each one does.
809///
810/// The level is the whole of the answer unless a `-f` flag edited it, which is section 9.1 of
811/// `spec/09-optimizer.md`: a level is a list somebody wrote down rather than something that
812/// emerges from which flags happen to be set, and this is how that list is read.
813#[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        // Every spelling was checked while the arguments were parsed, so there is nothing here
820        // this can refuse, and a listing is not the place to report it if there were.
821        let _ = settings.gates.add(*on, spec);
822    }
823    rucc_opt::pipeline::print(&settings)
824}
825
826/// Renders the resolved configuration.
827///
828/// One `key: value` per line, sorted by nothing in particular but fixed in order, because
829/// this output is diffed across hosts in CI and a reordering would read as a change.
830#[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    // The register file as a count per class, which is enough to tell a target whose registers
848    // are described from one whose are not without printing sixteen names nobody asked for.
849    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    // Last because it is the one key with more than one line under it, and the only one
866    // whose value is a property of the machine rather than of the command line.
867    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
874/// Runs phase 4 over every input that has one, and writes what came out.
875///
876/// One input that fails does not stop the others. A build that reports every file it could
877/// not preprocess in one run is worth more than one that stops at the first, and the exit
878/// status is still a failure either way.
879fn 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            // An input that is already preprocessed, or an object file. GCC passes these
886            // through untouched, and the plan has already said so in its notes.
887            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
905/// Runs the front end over every input that has a compile phase, and writes what came out.
906///
907/// The same rule as [`preprocess_all`]: one input that fails does not stop the others, and the
908/// exit status is a failure either way. An input that is already assembly or an object has no
909/// compile phase and is passed over here, which the plan has already said in its notes.
910fn 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        // An input of IR is read back rather than compiled, since the C it came from is not
922        // here any more. Everything after this is the same, so the two paths meet again at the
923        // messages and the file the result is written to.
924        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
948/// A directory for the object files only the link step ever sees, removed when it goes away.
949///
950/// `-c` writes its object where the user can see it and linking does not, which is the whole of
951/// the difference: a `rucc a.c b.c` leaves an executable behind and nothing else, the same as
952/// every other compiler. Removing them on drop rather than at the end of a function is so that a
953/// link that failed leaves nothing behind either.
954struct Scratch {
955    /// Where the objects go.
956    dir: PathBuf,
957}
958
959impl Scratch {
960    /// Makes one, under whatever the platform calls its temporary directory.
961    ///
962    /// The name carries the process id so that two compilers running at once do not share a
963    /// directory, which they would otherwise do the moment two of them compiled a file of the
964    /// same name.
965    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
978/// The link line the plan describes, for `-###`.
979///
980/// The names in it are the hints the plan carries rather than the temporaries a real compilation
981/// would choose, because `-###` prints the line without having compiled anything and so has
982/// nothing to point at. That also makes the printed line readable rather than naming a directory
983/// that only exists while a compilation is running.
984fn 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
990/// Compiles everything, then links it.
991///
992/// The objects go in a directory that is removed afterwards, which is why this is not
993/// [`compile_all`] followed by a link: the plan says an object feeding the linker is temporary
994/// and does not say where, because where is a question that only has an answer once something is
995/// running.
996fn link_all(opts: &Options, plan: &Plan, link: &LinkOptions, verbose: bool) -> i32 {
997    let Some(job) = &plan.link else {
998        // Every path into here comes from a plan whose last phase is the link, and such a plan
999        // has a link job. Saying so is cheaper than an unwrap that would have to be explained.
1000        let mut stderr = std::io::stderr().lock();
1001        let _ = writeln!(stderr, "rucc: error: there is nothing to link");
1002        return 1;
1003    };
1004    // Before anything is compiled, because a linker that is not on the machine is worth knowing
1005    // about in the second it takes to look rather than after the compilation.
1006    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    // One per job, in job order, which is what lets the link line below be rebuilt with the real
1019    // paths in it: every job contributes exactly one file to the line and does so in this order.
1020    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                    // The index because two inputs in different directories can have the same
1030                    // name, and the two objects of `rucc a/x.c b/x.c` must not be one file.
1031                    scratch.dir.join(format!("{at}-{hint}")).display().to_string()
1032                }
1033                Output::File(path) => path.clone(),
1034                // A job feeding the linker never writes to standard output, since the plan gives
1035                // it a temporary. This is here so that the match is total rather than a panic.
1036                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                // Worth saying rather than writing whatever it is and letting the linker read it.
1059                // An empty file is a valid empty linker script, so a link handed one gets as far
1060                // as reporting every symbol of this file undefined, which is a page of messages
1061                // about something that went wrong here.
1062                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        // Nothing is linked from a compilation that did not finish. A linker run over the objects
1079        // that did compile would report every function of the file that did not as undefined,
1080        // which is a page of messages about a mistake already reported once.
1081        return 1;
1082    }
1083
1084    // The items in command line order with the temporaries filled in. A library contributes no
1085    // job and passes through, and every file item takes the next job's real output, which is
1086    // what keeps a library that was written between two objects between them here.
1087    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        // The linker has already said what was wrong on its own error output, and repeating that
1110        // linking failed would only push its message further up the screen.
1111        Err(link::Error::Refused { .. }) => 1,
1112        Err(why) => complain(why),
1113    }
1114}
1115
1116/// Prints one driver level message and gives back the exit status that goes with it.
1117fn 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
1123/// Writes what `-Zrule-coverage=FILE` asked for, and says whether it could.
1124///
1125/// Once for the whole command line rather than once per input, because the question is which
1126/// lowering rules this run of the compiler reached and a file per input would leave the reader
1127/// unioning files to find out something one process already knew.
1128///
1129/// A file that could not be written is a failure and not a warning. What asks for this is a
1130/// measurement run, and a measurement that quietly did not happen is worse than one that stopped.
1131fn 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
1151/// Where the `-fopt-info` remarks go, and how much of the run has already gone there.
1152///
1153/// Standard error by default, and one file for the whole run when `-fopt-info=<file>` named one.
1154/// A file rather than the diagnostic stream is what a harness wants: the corpus in
1155/// `tamnd/rucc-corpus` matches a rejection against what the compiler said on standard error, and
1156/// a few thousand remarks mixed into that would bury it.
1157struct Remarks {
1158    /// The file, if there is one.
1159    file: Option<String>,
1160    /// Whether anything has been written to it yet, which decides between truncating and
1161    /// appending. One file holds the whole run rather than the last input in it.
1162    started: bool,
1163}
1164
1165impl Remarks {
1166    /// Prepares the destination, emptying the file if there is one.
1167    ///
1168    /// Emptied here rather than at the first remark, because a run where no pass had anything to
1169    /// say should leave an empty file and not yesterday's. An absent file and an empty one are
1170    /// different facts and something reading this will act on the difference.
1171    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    /// Writes one input's remarks, and says whether that worked.
1183    ///
1184    /// A file that cannot be written is a failure and not a warning, for the reason
1185    /// [`write_dumps`] gives: remarks that quietly did not arrive look exactly like a compilation
1186    /// where nothing happened.
1187    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
1212/// Writes what `-fdump-ir=` asked to see, one file per dump.
1213///
1214/// The name is the input file with the dump's own name and `.ir` after it, so a directory listing
1215/// after a run is the passes in the order they ran, per input. They go in the working directory
1216/// rather than beside the output, because a dump is something a person asked for at a prompt and
1217/// the working directory is where that person is.
1218///
1219/// A file that could not be written is a failure and not a warning, for the reason
1220/// [`write_coverage`] gives: what asked for this is somebody debugging a pass, and a dump that
1221/// quietly did not happen looks exactly like a pass that did not run.
1222fn 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
1237/// Writes one job's result where the plan said it goes.
1238///
1239/// # Errors
1240///
1241/// Returns the message to print, which names the file when there is one, because "permission
1242/// denied" on its own does not say which file was refused.
1243fn 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
1255/// Runs the driver and returns the process exit code.
1256///
1257/// `args` excludes the program name. Output goes to `stdout` and errors to `stderr`, which
1258/// is the one place in the compiler that is true.
1259pub 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            // The line as it would be typed, which is the half of `-###` that section 4.3 says
1284            // arrives with the link. It is printed even when the linker is not on this machine,
1285            // because what a build wants from `-###` is what the compiler would do.
1286            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    /// The unstable options, which are spelled apart from everything else on purpose: what is
1364    /// under `-Z` promises nothing, and a build that reaches for one should have had to say so.
1365    #[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        // RISC-V has a register file and this compiler has not written it down yet, and the
1456        // dump says which of those two it is rather than leaving the line out.
1457        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        // The last one wins, the way every other repeated flag on this command line does.
1488        let (opts, _) = compile(&["-fsafety=enforce", "-fsafety=off", "a.c"]);
1489        assert_eq!(opts.safety, rucc_session::Safety::Off);
1490
1491        // A misspelled tier is refused rather than ignored. Silently compiling without the
1492        // monitor a build asked for is the one failure mode this feature cannot have.
1493        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        // One pass runs at `-O0` and it is the one that removes code nothing reaches, which is
1509        // not an optimization. See issue 359.
1510        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        // And with that one turned off there is nothing left, which the dump says rather than
1515        // printing an empty list.
1516        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        // The one that was named is gone and the rest of the level is not, which is the whole
1525        // of what a toggle promises.
1526        assert!(!text.contains("fold"), "{text}");
1527        assert!(text.contains("dce"), "{text}");
1528
1529        // Every pass the compiler has, named off. Built from the registry rather than written
1530        // out, so a pass added later is turned off here too and this keeps testing the thing it
1531        // is about, which is that the toggles can empty a level.
1532        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        // Because the listing is the answer to what this compilation will do, and a run that
1550        // stops after four rewrites is not doing what the level says it does.
1551        assert!(text.contains("global fuel: 4"), "{text}");
1552    }
1553
1554    /// A pass is turned on and off by its own name, and the order the flags were given in is
1555    /// kept, because the last spelling of a name is the one that decides.
1556    #[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        // And it is not the per pass flag with a longer name, so neither spelling swallows the
1589        // other.
1590        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    /// The spelling is checked while the arguments are read, because a dump that names a pass
1622    /// this compiler does not have is a typo, and a typo found after the compilation has run is
1623    /// found too late to be any use.
1624    #[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    /// Every spelling `-fopt-info` takes, and the one it does not.
1635    ///
1636    /// The keywords are checked here for the same reason a dump's pass name is: a person who
1637    /// misspelled one gets no output, and no output is also what a compilation where nothing
1638    /// happened looks like. Telling those two apart is the entire reason to reach for this flag.
1639    #[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        // Two flags add up rather than the second replacing the first, and the file is the last
1649        // one that named a file, which is how GCC treats both.
1650        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        // A sysroot with nothing under it, so that the library's own directories are the
1685        // same on every machine this test runs on, which is none of them.
1686        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        // The compiler's own headers sit after every `-isystem` and before `-idirafter`,
1699        // which is where GCC puts its own: a directory the user named outranks ours.
1700        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        // Which machine this runs on decides what is on the path, so the test is about the
1708        // order rather than about the names: ours is on it, the library's follow it, and
1709        // `-nostdinc` is the one flag that takes both halves of the pair off at once.
1710        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        // Packed, the way GCC takes them, and a letter in the family we have not written yet
1757        // is accepted and does nothing rather than failing a build.
1758        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        // `-dumpversion` is a different flag that happens to start the same way, and it is read
1767        // as itself rather than as a dump of nothing.
1768        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        // A missing component is zero. `gcc -dumpversion` says `15` on a release with no
1784        // patchlevel and a harness that pastes that back has to be understood.
1785        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        // The `-W` family's name for it, which is what a build that groups its warning flags
1805        // tends to write.
1806        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    /// The two ways a build says it means its own function by a name the C library also has.
1822    ///
1823    /// `-fno-builtin` is all of them and `-fno-builtin-<name>` is one, and the second is what a
1824    /// build writes when it means its own `memcpy` and the library's everything else. The name is
1825    /// kept as it was written and not checked against anything, because a program is allowed to
1826    /// mean something by a name this compiler has never heard of.
1827    #[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    /// `-fgnu89-inline`, which is off by default and is not implied by anything on the command
1845    /// line, since the dialect asks for GNU's reading further in rather than through this.
1846    #[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        // The C89 dialects are under GNU's reading whether this was written or not, so the flag
1858        // stays off there and the dialect is what the checker and the macro set both ask. That is
1859        // also why `-std=c89 -fno-gnu89-inline` needs no diagnostic: it asks for the reading the
1860        // dialect already has. gcc refuses that command line, which is measured in the issue.
1861        let (opts, _) = compile(&["-c", "-std=c89", "a.c"]);
1862        assert!(!opts.gnu89_inline);
1863    }
1864
1865    /// Both spellings of both frame flags, since a build that wants one usually writes the
1866    /// other beside it for the one file that has to be compiled the ordinary way.
1867    #[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        // Link order is semantic: `-lm` written between two files resolves for the one before
1920        // it and not for the one after, so a library cannot be collected into a list of its own.
1921        // The target is named because the suffix of an object is the target's and this asserts
1922        // on the names: the same command line on a Windows host plans two `.obj` files.
1923        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        // And it is not a job, because there is nothing to compile in a library.
1934        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        // The rule in section 4.1, and the reason for it is autoconf: a configure script finds
1964        // out whether a warning flag exists by passing it and looking at the exit status, so a
1965        // compiler that refuses one it does not know fails a script written for a newer GCC.
1966        let (opts, _) = compile(&["-Wall", "-Wextra", "-Wno-format-truncation", "-c", "a.c"]);
1967        assert!(!opts.warnings_are_errors);
1968        assert!(opts.warnings);
1969        // The two spellings that do mean something are still read.
1970        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        // Every one of these says something about the output, so the wrong answer is silence.
1981        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        // The word size the target does not have, which is a target this compiler was not asked
1988        // for rather than a flag it does not know.
1989        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        // After the input, because a static link takes what it needs from a library when it
2015        // reaches it and not afterwards.
2016        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        // A name nothing holds comes back unchanged, which is GCC's rule and is what makes the
2028        // answer safe to paste into a link line whether or not the file is there.
2029        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        // Not a style preference. A help text that scrolls is one nobody reads, and this is
2039        // the cheapest way to keep it honest as flags accumulate. The number goes up only when
2040        // a family of flags arrives that has nowhere to share a line, which the two pass gates
2041        // were and which the two fuel flags and `-fsafety=` now are, and it goes up by exactly
2042        // the lines that family took. The four it went up by last are the flags a build system
2043        // passes without being asked to: how much to say, what machine to generate for, threads,
2044        // and the questions `configure` asks before it compiles anything. The one it went up by
2045        // last is the second line of `--emit`, whose kinds are a family that has now outgrown
2046        // one line and has nowhere else to go.
2047        assert!(USAGE.lines().count() < 42, "usage text has grown past one screen");
2048    }
2049}