Skip to main content

rucc_codegen/
lower.rs

1//! The selector: an IR function becomes a machine IR function.
2//!
3//! Design: `spec/10-backend.md` sections 10.2 and 10.3.
4//!
5//! What the matcher in [`crate::select`] does is answer one question about one term. What this
6//! does is ask it: walk a function, decide which terms are worth asking about, and build machine
7//! instructions out of what comes back. Nothing here decides what an IR term lowers to. That is
8//! in `rules/x86-64.rules` and it is proved before it is used, which is the whole point of the
9//! arrangement and the reason this file is short.
10//!
11//! # What it does with an instruction
12//!
13//! It tries the ways the instruction can be shown to the matcher, in order, and takes the first
14//! that a rule fires on. [`crate::term`] is what a way of showing one is, and the order is the
15//! most specific first: an operand that is a constant is offered as a constant before it is
16//! offered as a register, and an operand computed by an instruction of its own is offered as
17//! that instruction before it is offered as a register. A rule that wants an immediate too wide
18//! for the machine has a guard that turns it down, and the search carries on to the way of
19//! showing it that puts the constant in a register, which is the right answer and is one nobody
20//! had to write down.
21//!
22//! A constant is not lowered where it is written. It is materialized where a register for it is
23//! first wanted, which is what keeps a constant that every use folded into an immediate from
24//! leaving a dead instruction behind, and it also gives the value the shortest live range it
25//! could have. The instruction that materializes it comes from the rule set like everything else.
26//!
27//! # What it does not do yet
28//!
29//! Everything is in the general purpose registers, because every rule in the set is about an
30//! integer, so a call that passes a `double` and a function that returns one are both reported
31//! rather than lowered. So is an argument that travels on the stack, on either side of a call,
32//! and so is a call through an address rather than to a name.
33//!
34//! # A call
35//!
36//! Not a rule, because a rule pattern sees one term and what a call's operands are is whatever
37//! the signature made them. [`crate::abi`] builds one instead, out of the same description of the
38//! convention the arguments come from: the values it passes are reads constrained to the
39//! registers the convention places them in, what comes back is a write constrained to the
40//! register it comes back in, and every other register the callee is free to destroy is a write
41//! of that register and nothing else, which is all the allocator needs to keep a value out of it.
42//!
43//! What that costs the frame is an argument area, and nothing after selection could work out how
44//! big, so the size of the widest call is given back with the function. A function that makes no
45//! call at all is a leaf, and a leaf is the function that may use the red zone.
46//!
47//! # Where a block goes
48//!
49//! On the block, which is what machine IR does with an edge and is why the branches need no more
50//! rule language than the arithmetic did. A rule never names a block, so an unconditional jump
51//! has no rule at all and a conditional branch has one that is about its condition and nothing
52//! else. The arms are copied across after the block is filled, arguments and all, because an
53//! argument that is a constant is materialized where a register for it is first wanted and the
54//! end of the block is where an edge wants it.
55//!
56//! What this leaves behind is a function whose blocks are in the order the IR held them and whose
57//! branches are still branches on a register. Turning one into a `test` and a `jcc` is the block
58//! layout's, since which of the two arms falls through is the layout's answer, and [`crate::split`]
59//! has to run before allocation so that every edge carrying a value has somewhere to put it.
60//!
61//! A store and a return are the two things here that write no register. A store is emitted like
62//! everything else and the only difference is that there is no result to put anywhere, so the
63//! operands the target describes are all reads. A return is the same, and what it is for is its
64//! one operand: the target constrains it to the register the caller reads the value out of, and
65//! the allocator is what gets it there. The instruction that leaves is not chosen here at all,
66//! because the epilogue has to give the frame back first and [`crate::finish`] writes that after
67//! allocation, so a return of nothing is lowered to nothing.
68//!
69//! The entry block is the one block whose parameters are not block parameters here. They are the
70//! function's arguments, they are already somewhere when it starts, and [`crate::abi`] is what
71//! says where. An argument that arrives on the stack is reported rather than read, because where
72//! the stack put it is a distance into a frame and no frame exists until after allocation.
73//!
74//! Blocks are walked in the order the function holds them and a value is expected to be defined
75//! before it is used, which is true of the IR this is given because every pass before it keeps
76//! definitions ahead of uses.
77
78use std::fmt;
79
80use rucc_base::Interner;
81use rucc_ir::{Abi, Block, Def, Extra, Func, Inst, Opcode, Param, Type, Value};
82use rucc_mir as mir;
83use rucc_target::x86_64;
84use rucc_target::{CallRegs, RegClass};
85
86use crate::abi::{self, Missing, Refused};
87use crate::frame::{Layout, Local};
88use crate::select::{Match, Piece, Rule, Table};
89use crate::term::{MAX_ARGS, PLAIN, Plan, Shown, Term, Terms};
90use crate::varargs;
91
92/// The prefix a rule file puts in front of a machine term, which says which target it belongs
93/// to and is not part of the opcode.
94pub(crate) const PREFIX: &str = "x64.";
95
96/// How wide an address is on this target, which is the width a cast between a pointer and an
97/// integer has to be at for the cast to be nothing.
98const ADDRESS_BITS: u32 = 64;
99
100/// Why a function could not be lowered.
101///
102/// One reason and then nothing. A function with no rule for something in it is a function this
103/// cannot finish, and the second thing it could not lower is not news.
104#[derive(Debug, Clone, PartialEq, Eq)]
105pub enum Unsupported {
106    /// An instruction no rule fires on.
107    Inst {
108        /// The instruction that stopped it.
109        inst: Inst,
110        /// What the rule file would call it, or nothing if the rule language has no name for it
111        /// at all, which is what an instruction at a width nothing is written about looks like.
112        term: Option<&'static str>,
113        /// The opcode, which is what gets named when the rule language has no word for it.
114        ///
115        /// An opcode the rule language has no word for is exactly the opcode no rule lowers, so
116        /// without this the message would be empty in every case where somebody needs it.
117        opcode: Opcode,
118        /// What it produces, or nothing for an instruction that is only an effect.
119        ty: Option<Type>,
120    },
121    /// A parameter that does not arrive somewhere this can bring it in from.
122    ///
123    /// Not an instruction, which is why it is a separate arm: it is a fact about the signature
124    /// and there is nothing in the body of the function to point at.
125    Argument {
126        /// Its position in the signature.
127        index: usize,
128        /// What is wrong with where it arrives.
129        missing: Missing,
130    },
131    /// A call that passes or gives back a value this cannot put where the convention wants it.
132    Call {
133        /// The call.
134        inst: Inst,
135        /// Which value, and what is wrong with where it travels.
136        refused: Refused,
137    },
138    /// A `return` this cannot put where the convention wants it.
139    ///
140    /// A separate arm from [`Unsupported::Inst`] because it is not an instruction no rule fires
141    /// on. A return of more than one value is built from the convention rather than matched, the
142    /// same way a call is, so what goes wrong with one is what goes wrong with a call and not the
143    /// absence of a rule.
144    Returned {
145        /// The `return`.
146        inst: Inst,
147        /// What is wrong with where one of the values travels.
148        missing: Missing,
149    },
150    /// A stack slot whose size is not known until the function runs, which is what a variable
151    /// length array is.
152    ///
153    /// Not an instruction no rule covers. Growing the stack where the declaration stands is
154    /// arithmetic on the stack pointer, and everything else in the frame then has to be reached
155    /// through a frame pointer instead, and neither of those is a term a rule could be written
156    /// about or a thing the frame here knows how to lay out.
157    Dynamic {
158        /// The `alloca`.
159        inst: Inst,
160    },
161}
162
163impl Unsupported {
164    /// The instruction it is about, or nothing for the one arm that is about a signature.
165    ///
166    /// What a caller wants this for is the span. The function knows where every instruction in
167    /// it came from, so a caller holding both can point a message at the line somebody wrote
168    /// rather than at the file as a whole, and nothing here has to carry a span of its own.
169    pub fn inst(&self) -> Option<Inst> {
170        match *self {
171            Unsupported::Inst { inst, .. }
172            | Unsupported::Call { inst, .. }
173            | Unsupported::Returned { inst, .. }
174            | Unsupported::Dynamic { inst, .. } => Some(inst),
175            Unsupported::Argument { .. } => None,
176        }
177    }
178}
179
180impl fmt::Display for Unsupported {
181    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
182        match *self {
183            Unsupported::Inst { term: Some(term), .. } => write!(f, "no rule lowers `{term}`"),
184            Unsupported::Inst { term: None, opcode, ty: Some(ty), .. } => {
185                write!(f, "no rule lowers a `{opcode}` producing a `{ty}`")
186            }
187            Unsupported::Inst { term: None, opcode, ty: None, .. } => {
188                write!(f, "no rule lowers a `{opcode}`")
189            }
190            Unsupported::Argument { index, missing } => {
191                write!(f, "parameter {index} {}", missing.why())
192            }
193            Unsupported::Call { refused: Refused { argument: Some(index), missing }, .. } => {
194                write!(f, "argument {index} of this call {}", missing.why())
195            }
196            Unsupported::Call { refused: Refused { argument: None, missing }, .. } => {
197                write!(f, "what this call gives back {}", missing.why())
198            }
199            Unsupported::Returned { missing, .. } => {
200                write!(f, "what this function gives back {}", missing.why())
201            }
202            Unsupported::Dynamic { .. } => {
203                f.write_str("nothing here grows the stack for a variable length array")
204            }
205        }
206    }
207}
208
209impl std::error::Error for Unsupported {}
210
211/// A lowered function, and what the frame needs that the machine IR does not hold.
212#[derive(Debug)]
213pub struct Lowered {
214    /// The function, in machine instructions.
215    pub func: mir::Func,
216    /// What it wants its stack to look like, which is separate from the function so that the two
217    /// can be read and written at the same time.
218    pub stack: Stack,
219}
220
221/// What a function's stack has to hold, as far as selection is able to say.
222///
223/// All of it is answered here because selection is where a call is built and where an `alloca`
224/// is read, and nothing after it could tell what either of them needed.
225#[derive(Debug, Default)]
226pub struct Stack {
227    /// How many bytes the widest call in the function needs below the stack pointer for the
228    /// arguments it passes there, or `None` for a function that makes no call at all.
229    ///
230    /// `None` is a leaf, which is the function that may use the red zone and the one whose stack
231    /// pointer does not have to be left aligned for anybody.
232    pub calls: Option<u32>,
233    /// The memory the function asked for itself, one entry for every `alloca` in it, in the order
234    /// the walk reached them.
235    pub locals: Vec<Local>,
236    /// Which instruction computes the address of which of those locals.
237    ///
238    /// An address in the frame is a distance from the stack pointer, and there is no frame until
239    /// after allocation, so the instruction is written here with nothing in its displacement and
240    /// [`crate::finish`] writes the number in once [`crate::frame::Frame`] knows it.
241    pub addresses: Vec<(mir::Inst, usize)>,
242    /// Which instruction reads which of the arguments the caller passed on the stack, as how far up
243    /// the caller's argument area it reads.
244    ///
245    /// Waiting on [`crate::finish`] for the same reason the addresses above are, and on one thing
246    /// more: where the caller's argument area is from inside this function depends on whether the
247    /// prologue had to force the stack pointer's alignment, so which register the load reads
248    /// through is not settled here either.
249    pub arguments: Vec<(mir::Inst, u32)>,
250}
251
252impl Stack {
253    /// The layout given, with the three fields only the lowering knows the answer to filled in.
254    ///
255    /// Everything else in a layout comes from the flags the function is compiled under or from the
256    /// allocation, so this takes one and returns it rather than building one.
257    #[must_use]
258    pub fn layout<'a>(&'a self, base: Layout<'a>) -> Layout<'a> {
259        Layout {
260            leaf: self.calls.is_none(),
261            outgoing: self.calls.unwrap_or(0),
262            locals: &self.locals,
263            ..base
264        }
265    }
266}
267
268/// The x86-64 machine IR for that function.
269///
270/// # Errors
271///
272/// The first instruction no rule fires on, which today is anything at a width the rule set is not
273/// written at, a parameter that does not arrive in a register this can read, or a call that
274/// passes something this cannot put where the convention wants it.
275pub fn func(
276    source: &Func,
277    names: &mut Interner,
278    conv: &'static CallRegs,
279) -> Result<Lowered, Unsupported> {
280    Lowering::new(source, names, conv).run()
281}
282
283/// One function being lowered.
284struct Lowering<'a> {
285    source: &'a Func,
286    names: &'a mut Interner,
287    out: mir::Func,
288    /// The machine register each IR value is in, once it has one.
289    regs: Vec<Option<mir::Reg>>,
290    /// For a constant that has been written into a register, the block it was written into,
291    /// which is the only block that register is any good in.
292    written: Vec<Option<mir::Block>>,
293    /// How many times each IR value is read, which is what says whether an instruction may be
294    /// folded into the one that reads it.
295    uses: Vec<u32>,
296    /// The block being filled.
297    at: Option<mir::Block>,
298    /// The machine IR block each IR block became.
299    blocks: Vec<Option<mir::Block>>,
300    /// The class an address is in, which is the general purpose one and is not a question: every
301    /// register an addressing mode names holds part of an address, and there is no machine here
302    /// that computes an address anywhere but in this file. Which class a *value* is in is
303    /// [`Lowering::class_of`], and it is a question, because a float is in the other one.
304    gpr: RegClass,
305    /// Where the convention this function is compiled for puts things, which is read for the
306    /// arguments and for the calls.
307    conv: &'static CallRegs,
308    /// What the function wants its stack to look like, filled in as the walk finds out.
309    stack: Stack,
310    /// What a `va_start` in this function has to write, or nothing for a function that takes no
311    /// arguments its signature does not name.
312    ///
313    /// Worked out once, when the entry block binds the parameters, because every number in it is
314    /// about where those parameters left the walk over the argument registers and there is nowhere
315    /// else that knows.
316    varargs: Option<Varargs>,
317}
318
319/// What a `va_start` in a variadic function writes into the list it is given.
320///
321/// Three of the four are settled here and the fourth is not a number at all yet: where the save
322/// area is and where the caller's argument area is are both distances into a frame that does not
323/// exist until after allocation, so both are `lea` instructions [`crate::finish`] fills in.
324#[derive(Debug, Clone, Copy, PartialEq, Eq)]
325struct Varargs {
326    /// Which of the function's stack objects is the register save area.
327    save: usize,
328    /// How far up the caller's argument area the first argument the signature does not name is,
329    /// which is the whole of that area the named ones did not take.
330    incoming: u32,
331    /// What `gp_offset` starts at, which is past the general purpose registers the named arguments
332    /// took.
333    integers: u32,
334    /// What `fp_offset` starts at, which is past the vector ones.
335    floats: u32,
336}
337
338impl<'a> Lowering<'a> {
339    fn new(source: &'a Func, names: &'a mut Interner, conv: &'static CallRegs) -> Self {
340        let counts = source.counts();
341        let name = source.name;
342        let mut uses = vec![0; counts.values];
343        for block in source.blocks() {
344            for inst in source.insts(block) {
345                for &arg in &source[source[inst].args] {
346                    uses[arg.index()] += 1;
347                }
348                for call in source.successors(inst) {
349                    for &arg in &source[call.args] {
350                        uses[arg.index()] += 1;
351                    }
352                }
353            }
354        }
355        Self {
356            source,
357            names,
358            out: mir::Func::new(name),
359            regs: vec![None; counts.values],
360            written: vec![None; counts.values],
361            blocks: vec![None; counts.blocks],
362            uses,
363            at: None,
364            gpr: x86_64::GPR,
365            conv,
366            stack: Stack::default(),
367            varargs: None,
368        }
369    }
370
371    fn run(mut self) -> Result<Lowered, Unsupported> {
372        // Every block before any of them is filled, because a block that jumps forward has to
373        // name the block it jumps to and a machine IR block is named by a handle rather than by
374        // the IR block it came from.
375        for block in self.source.blocks() {
376            let out = self.out.create_block();
377            self.blocks[block.index()] = Some(out);
378        }
379        for block in self.source.blocks() {
380            self.block(block)?;
381        }
382        Ok(Lowered { func: self.out, stack: self.stack })
383    }
384
385    /// One block: its parameters, then every instruction in it that is not folded into another.
386    fn block(&mut self, block: Block) -> Result<(), Unsupported> {
387        let out = self.out_block(block);
388        self.at = Some(out);
389        if self.source.entry() == Some(block) {
390            self.arrive(block, out)?;
391        } else {
392            for &param in self.source[block].params.iter() {
393                let reg = self.out.append_param(out, self.class_of(self.source[param].ty));
394                self.regs[param.index()] = Some(reg);
395            }
396        }
397
398        // What each instruction matched, and which instructions were folded into another. The
399        // instruction that is folded comes before the one that folds it, so the decision has to
400        // be made for the whole block before any of it is written, and it is made backwards: an
401        // instruction that has been folded into a later one does not get to fold anything into
402        // itself, because the rule that took it only reached one level down.
403        let insts: Vec<Inst> = self.source.insts(block).collect();
404        let mut found: Vec<Option<Match<Term>>> = (0..insts.len()).map(|_| None).collect();
405        let mut folded: Vec<Inst> = Vec::new();
406        for (index, &inst) in insts.iter().enumerate().rev() {
407            if folded.contains(&inst) {
408                continue;
409            }
410            if let Some((plan, matched)) = self.select(inst) {
411                folded.extend(self.folds(inst, plan));
412                found[index] = Some(matched);
413            }
414        }
415
416        for (&inst, matched) in insts.iter().zip(found) {
417            if folded.contains(&inst) || self.writes_nothing(inst) {
418                continue;
419            }
420            // A call is built from the convention rather than matched, which is why it is the one
421            // opcode looked at by name here. Through an address it is a different instruction and
422            // the same convention, so the two arrive at the same place and differ in one line of
423            // it.
424            match self.source[inst].opcode {
425                Opcode::Call | Opcode::CallIndirect => {
426                    self.called(inst)?;
427                    continue;
428                }
429                // Built from the frame rather than matched, for the same shape of reason a call
430                // is built from the convention: what a rule replaces a term with is instructions,
431                // and what an `alloca` needs first is bytes, which the rule language has no way
432                // to ask for.
433                Opcode::Alloca => {
434                    self.reserve(inst)?;
435                    continue;
436                }
437                // The address of a name, built here for the same reason an `alloca` is: what a
438                // rule replaces a term with is instructions over values, and the operand of this
439                // one is a symbol, which is a thing the rule language has no way to bind and the
440                // solver has no way to say anything about. There is nothing in `lea sym(%rip)` a
441                // proof over bitvectors could discharge, because what makes it the right answer
442                // is the relocation and what the linker does with it.
443                Opcode::GlobalAddr => {
444                    self.address_of(inst)?;
445                    continue;
446                }
447                // Built from the frame for the reason an `alloca` is, and from the convention for
448                // the reason a call is: three of the four fields it writes are distances that do
449                // not exist until the frame does, and the fourth is where the walk over the
450                // argument registers stopped. A function that is not variadic has no such walk to
451                // report, so it has nothing here and is refused below, which is the right answer
452                // for a `va_start` in one.
453                Opcode::VaStart if self.varargs.is_some() => {
454                    self.va_start(inst)?;
455                    continue;
456                }
457                // A return of more than one value, which is a structure small enough to come
458                // back in a pair of registers. Built from the convention for the reason a call
459                // is: which register each half goes in depends on the halves in front of it,
460                // because the two register files are walked separately, and a pattern over a term
461                // cannot see them. A return of one value is a term with a name and a rule, and it
462                // stays one.
463                //
464                // A return of none in a function whose answer went through memory is here too,
465                // and for a different reason: what it gives back is not written in the IR at all.
466                // The convention says the address the caller handed over comes back, and only the
467                // signature says this function was handed one.
468                Opcode::Return
469                    if self.source[self.source[inst].args].len() > 1 || self.sret().is_some() =>
470                {
471                    self.returned(inst)?;
472                    continue;
473                }
474                // A cast between a pointer and an integer of the same width, which on this
475                // machine is every one the front end writes. No instruction at all, so no rule
476                // could name one.
477                Opcode::PtrToInt | Opcode::IntToPtr => {
478                    self.rename(inst)?;
479                    continue;
480                }
481                _ => {}
482            }
483            let matched = matched.ok_or_else(|| self.unsupported(inst))?;
484            self.emit(inst, &matched)?;
485        }
486        self.edges(block, out)
487    }
488
489    /// One call, which is built from the convention rather than matched against the table for the
490    /// same reason the arguments of the function itself are.
491    ///
492    /// The arguments are read before the call is built, which is what materializes a constant
493    /// argument into a register, since no call passes an immediate.
494    ///
495    /// A call to a name and a call through an address are both here, and what tells them apart is
496    /// the opcode rather than whether a callee was recorded, which is the same thing the verifier
497    /// reads. Through an address the first operand is the address and the arguments are the ones
498    /// behind it, and everything after that is the same: where each argument goes, where the value
499    /// comes back and which registers are gone across it are the convention's answers and the
500    /// convention does not ask what is being called.
501    fn called(&mut self, inst: Inst) -> Result<(), Unsupported> {
502        let data = &self.source[inst];
503        let Extra::Call(info) = data.extra else { return Err(self.unsupported(inst)) };
504        let info = self.source[info];
505        let indirect = data.opcode == Opcode::CallIndirect;
506
507        let values: Vec<Value> = self.source[data.args].to_vec();
508        let callee = if indirect {
509            let &address = values.first().ok_or_else(|| self.unsupported(inst))?;
510            abi::Callee::Through(self.reg_of(address)?)
511        } else {
512            abi::Callee::Named(info.callee.ok_or_else(|| self.unsupported(inst))?)
513        };
514
515        // What the ABI asks of each argument, read out before any of them is, because reading one
516        // borrows the function this is a table in. The ones the signature names are the signature's
517        // answer and the ones behind them are the call's, which is where a structure passed to a
518        // variadic callee by value says that its bytes travel: there is no parameter to say it on.
519        let signature = &self.source[info.signature];
520        let variadic = signature.variadic;
521        let named: Vec<Abi> = signature.params.iter().map(|param| param.abi).collect();
522        let beyond: Vec<Abi> = self.source[info.varargs].to_vec();
523        // Every value that comes back and not only the first. A structure small enough to travel
524        // in registers comes back in up to two of them, and which register each half is in is the
525        // convention's answer, which is why the whole list goes to the same place the arguments do
526        // rather than to a rule.
527        let returns: Vec<Type> = signature.return_types().collect();
528
529        let mut args = Vec::with_capacity(values.len());
530        for (index, value) in values.into_iter().skip(usize::from(indirect)).enumerate() {
531            let abi = named.get(index).or_else(|| beyond.get(index - named.len()));
532            let abi = abi.copied().unwrap_or_default();
533            args.push(abi::Passing { ty: self.source[value].ty, reg: self.reg_of(value)?, abi });
534        }
535        let block = self.at.expect("a block is being filled");
536        let what = abi::Calling { callee, args: &args, returns: &returns, variadic };
537        let made = abi::call(&mut self.out, block, &what, self.conv, self.names)
538            .map_err(|refused| Unsupported::Call { inst, refused })?;
539        let calls = &mut self.stack.calls;
540        *calls = Some(calls.unwrap_or(0).max(made.outgoing));
541        for (result, &reg) in self.source[inst].results().zip(&made.results) {
542            self.regs[result.index()] = Some(reg);
543        }
544        Ok(())
545    }
546
547    /// The pointer a function returning through memory was handed, or nothing in a function that
548    /// was not.
549    ///
550    /// It is the first parameter and the signature is what says so, since in the IR it is an
551    /// ordinary pointer and reads like one everywhere in the body. A function with a signature
552    /// like that and no entry block has nothing to give back and no body to give it back from.
553    fn sret(&self) -> Option<Value> {
554        let first = self.source.signature().params.first()?;
555        if !matches!(first.abi, Abi::Sret { .. }) {
556            return None;
557        }
558        self.source[self.source.entry()?].params.first().copied()
559    }
560
561    /// One `return` the convention has to write, as the place each value has to be in by the end.
562    ///
563    /// One pseudo per value, each a read constrained to a return register, which is what a return
564    /// of one value already is and is the whole of what either does. The `ret` itself comes from
565    /// the epilogue for both, long after this, because the frame has to be given back first.
566    ///
567    /// The two register files are counted separately, so a structure of a `double` and a `long`
568    /// leaves the `double` in the first vector register and the `long` in the first integer one
569    /// rather than in the second of either. That is the same walk `rucc_codegen::abi` makes on
570    /// the other side of the call, which is what makes the two ends agree.
571    ///
572    /// A function whose answer went through memory gives back the address it was handed, in front
573    /// of nothing else, because a signature that returns that way returns nothing else. That the
574    /// caller already knows the address is not enough: it is allowed to read the register instead,
575    /// and a caller that does gets whatever the allocator last left there. In a leaf function that
576    /// is usually the right answer by accident, and one call in the body is enough to make it a
577    /// wild pointer, which is why this is written rather than left to luck.
578    ///
579    /// Where everything goes is worked out before anything is written, so a return this cannot
580    /// make leaves no half of one behind.
581    fn returned(&mut self, inst: Inst) -> Result<(), Unsupported> {
582        let values: Vec<Value> = self.source[self.source[inst].args].to_vec();
583        let (mut ints, mut floats) = (0usize, 0usize);
584        let mut parts = Vec::with_capacity(values.len() + 1);
585        for value in self.sret().into_iter().chain(values) {
586            let ty = self.source[value].ty;
587            let at = if crate::term::float_slot(ty).is_some() { &mut floats } else { &mut ints };
588            // Why it cannot come back, and not only that it cannot. A type that travels nowhere
589            // says so itself, and a type that travels perfectly well ran out of registers.
590            let missing = abi::refuses(ty).unwrap_or(Missing::NoRoom);
591            let name = abi::ret_of(ty, *at).ok_or(Unsupported::Returned { inst, missing })?;
592            *at += 1;
593            // The register is the target's answer and not one worked out here, the same as it is
594            // for a return of one value, so that both halves of a pair and every rule that writes
595            // half of one are reading the same table.
596            let opcode = name.strip_prefix(PREFIX).expect("a machine instruction of this target");
597            let form = x86_64::form(opcode).ok_or_else(|| self.unsupported(inst))?;
598            let [desc] = form.operands() else { return Err(self.unsupported(inst)) };
599            parts.push((self.names.intern(name), self.reg_of(value)?, *desc));
600        }
601
602        let block = self.at.expect("a block is being filled");
603        let span = self.source.span(inst);
604        for (opcode, reg, desc) in parts {
605            let operand = mir::Operand {
606                reg,
607                class: desc.class,
608                role: desc.role,
609                constraint: desc.constraint,
610            };
611            self.out.build(block, mir::Opcode::new(opcode)).at(span).operand(operand).finish();
612        }
613        Ok(())
614    }
615
616    /// One `alloca`: the bytes it asks for go on the list the frame is laid out from, and the
617    /// address of them is one instruction.
618    ///
619    /// The instruction is a `lea` off the stack pointer, which is the one register that reaches
620    /// the frame in every function, and its displacement is left at nothing because there is no
621    /// frame yet. Which instruction is waiting for which local is remembered, and
622    /// [`crate::finish`] fills the numbers in after [`crate::frame::Frame`] has placed them.
623    ///
624    /// There is deliberately no rule for `alloca` and no name for one in [`crate::term`], and
625    /// that is what stops it being folded into something else. An operand shown as the
626    /// instruction that computed it is offered to the matcher by its name, so an `alloca` with no
627    /// name is one no pattern can reach past, and the address it computes is always in a register
628    /// by the time anything reads it.
629    fn reserve(&mut self, inst: Inst) -> Result<(), Unsupported> {
630        let data = &self.source[inst];
631        // A variable length array carries the size it wants as an operand rather than in the
632        // instruction, which is the whole of what tells the two apart here.
633        if !self.source[data.args].is_empty() {
634            return Err(Unsupported::Dynamic { inst });
635        }
636        let Extra::Mem(mem) = data.extra else { return Err(self.unsupported(inst)) };
637        let info = self.source[mem];
638        let size = u32::try_from(info.size).map_err(|_| Unsupported::Dynamic { inst })?;
639        let result = data.first_result.ok_or_else(|| self.unsupported(inst))?;
640
641        // At least one, because the frame divides by the alignment and an object with no
642        // alignment at all is one the front end had nothing to say about rather than one that may
643        // go anywhere.
644        let index = self.stack.locals.len();
645        self.stack.locals.push(Local { size, align: info.align.max(1) });
646
647        let block = self.at.expect("a block is being filled");
648        let reg = self.new_reg(result);
649        let span = self.source.span(inst);
650        let lea = mir::Opcode::new(self.names.intern(&format!("{PREFIX}{}", x86_64::FRAME.lea)));
651        let sp = mir::Operand::read(mir::Reg::physical(self.conv.stack_pointer), self.gpr);
652        let made =
653            self.out.build(block, lea).at(span).def(reg, self.gpr).mem(mir::Mem::at(sp)).finish();
654        self.stack.addresses.push((made, index));
655        Ok(())
656    }
657
658    /// One `va_start`, as the four fields of the list it was handed.
659    ///
660    /// Two of them are numbers this already knows, and each costs an instruction to put in a
661    /// register before it can be stored, because the machine here has no store of an immediate to
662    /// memory. The other two are addresses in the frame, and each is a `lea` [`crate::finish`]
663    /// finishes: the save area is one of the function's own stack objects, and the caller's
664    /// argument area is where the parameters that had no register came from, which is the same
665    /// place and the same fixup a parameter past the sixth already uses.
666    ///
667    /// What is written is exactly the four fields [`crate::varargs`] describes, in the order they
668    /// are laid out, so that reading this beside that table is the whole of the check.
669    fn va_start(&mut self, inst: Inst) -> Result<(), Unsupported> {
670        let Some(&list) = self.source[self.source[inst].args].first() else {
671            return Err(self.unsupported(inst));
672        };
673        let started = self.varargs.ok_or_else(|| self.unsupported(inst))?;
674        let list = self.reg_of(list)?;
675        let block = self.at.expect("a block is being filled");
676        let span = self.source.span(inst);
677
678        for (at, count) in
679            [(varargs::GP_OFFSET, started.integers), (varargs::FP_OFFSET, started.floats)]
680        {
681            let held = self.out.new_vreg(self.gpr);
682            let load = mir::Opcode::new(self.names.intern("x64.mov_ri_32"));
683            self.out.build(block, load).at(span).def(held, self.gpr).imm(i64::from(count)).finish();
684
685            let store = mir::Opcode::new(self.names.intern("x64.mov_mr_32"));
686            let mem = self.field(list, at);
687            self.out.build(block, store).at(span).uses(held, self.gpr).mem(mem).finish();
688        }
689
690        // The first argument the signature did not name, which is as far up the caller's argument
691        // area as the ones it did name reached. Nothing here knows where that area is, so the
692        // distance is recorded the way a parameter read out of it is and finished with it.
693        let overflow = self.out.new_vreg(self.gpr);
694        let lea = mir::Opcode::new(self.names.intern(&format!("{PREFIX}{}", x86_64::FRAME.lea)));
695        let sp = mir::Operand::read(mir::Reg::physical(self.conv.stack_pointer), self.gpr);
696        let made = self
697            .out
698            .build(block, lea)
699            .at(span)
700            .def(overflow, self.gpr)
701            .mem(mir::Mem::at(sp))
702            .finish();
703        self.stack.arguments.push((made, started.incoming));
704
705        let save = self.frame_address(block, started.save);
706        for (at, held) in [(varargs::OVERFLOW, overflow), (varargs::SAVE_AREA, save)] {
707            let store = mir::Opcode::new(self.names.intern("x64.mov_mr_64"));
708            let mem = self.field(list, at);
709            self.out.build(block, store).at(span).uses(held, self.gpr).mem(mem).finish();
710        }
711        Ok(())
712    }
713
714    /// One field of a list, as the addressing mode that reaches it.
715    fn field(&self, list: mir::Reg, at: i64) -> mir::Mem {
716        let base = mir::Operand::read(list, self.gpr);
717        mir::Mem::at(base).plus(i32::try_from(at).expect("a field of a list is a small offset"))
718    }
719
720    /// The address of a name: one `lea` off the instruction pointer, with the name on it.
721    ///
722    /// The same instruction an `alloca` gets and for a related reason. An address that is not in
723    /// the program is a `lea` of an addressing mode that names no register, and the mode carries
724    /// the symbol so that [`rucc_asm`] can write it relative to `%rip` and leave the relocation
725    /// for the assembler. Both halves of that already existed: the printer writes `sym(%rip)` and
726    /// the encoder emits the relocation, because a call to a name the file does not define needed
727    /// them first.
728    ///
729    /// There is deliberately no name for this in [`crate::term`], which is what stops the address
730    /// being folded into the instruction that reads it. Folding it is the right thing to do and
731    /// is what turns a load of a global from two instructions into one, but it is a separate
732    /// question about addressing modes and issue #282 is it. Until then the address is in a
733    /// register before anything uses it, which is correct and one instruction longer.
734    ///
735    /// What this does not do is give the name anything to refer to. A module carries its globals
736    /// and nothing writes them out, so a file that defines the variable it reads compiles to a
737    /// reference the linker cannot resolve. Issue #293 is the other half.
738    fn address_of(&mut self, inst: Inst) -> Result<(), Unsupported> {
739        let data = &self.source[inst];
740        let Extra::Symbol(symbol) = data.extra else { return Err(self.unsupported(inst)) };
741        let result = data.first_result.ok_or_else(|| self.unsupported(inst))?;
742
743        let block = self.at.expect("a block is being filled");
744        let reg = self.new_reg(result);
745        let span = self.source.span(inst);
746        let lea = mir::Opcode::new(self.names.intern(&format!("{PREFIX}{}", x86_64::FRAME.lea)));
747        self.out.build(block, lea).at(span).def(reg, self.gpr).mem(mir::Mem::of(symbol)).finish();
748        Ok(())
749    }
750
751    /// A conversion that converts nothing: the result is the operand under another type.
752    ///
753    /// `ptrtoint` and `inttoptr` at one width are the whole of this. An address on this machine is
754    /// an integer as wide as the machine addresses, so a cast between the two changes what the
755    /// type system calls the value and changes nothing about the value, and the register holding
756    /// it is the register that already held it. The front end never writes either of them at any
757    /// other width, because it widens or narrows around the cast rather than through it, so the
758    /// two widths disagreeing here means the IR came from somewhere else and is refused rather
759    /// than guessed at.
760    ///
761    /// Reading the operand first is what materializes it when it is a constant, which is the case
762    /// that matters: a null pointer is an `inttoptr` of zero, and that zero has to reach a
763    /// register before anything can call it an address.
764    fn rename(&mut self, inst: Inst) -> Result<(), Unsupported> {
765        let data = &self.source[inst];
766        let [arg] = self.source[data.args] else { return Err(self.unsupported(inst)) };
767        let result = data.first_result.ok_or_else(|| self.unsupported(inst))?;
768        if !self.is_address_width(self.source[arg].ty)
769            || !self.is_address_width(self.source[result].ty)
770        {
771            return Err(self.unsupported(inst));
772        }
773        let reg = self.reg_of(arg)?;
774        self.regs[result.index()] = Some(reg);
775        Ok(())
776    }
777
778    /// Whether a type is the width an address is, which is what makes a cast to or from one free.
779    fn is_address_width(&self, ty: Type) -> bool {
780        ty.is_ptr() || (ty.is_int() && ty.bits() == ADDRESS_BITS)
781    }
782
783    /// Where a block goes, which in machine IR is on the block rather than on its terminator.
784    ///
785    /// That is why no rule ever names a block: a branch is selected for what it reads and the
786    /// edges are copied across here, arguments and all. The arguments are read last, after every
787    /// instruction of the block is written, because an argument that is a constant is
788    /// materialized where it is first wanted and the end of the block is where an edge wants it.
789    ///
790    /// Which is not quite the end. A block that leaves two ways has the branch as its last
791    /// instruction, and anything appended after a branch is something the branch has already
792    /// jumped past, so a constant materialized here would be a register the block below reads and
793    /// nothing ever writes. The branch is put back on the end when that happened, which is the
794    /// only reordering anything in this crate does and is why the branch is remembered before a
795    /// single argument is read.
796    fn edges(&mut self, block: Block, out: mir::Block) -> Result<(), Unsupported> {
797        let Some(term) = self.source.terminator(block) else { return Ok(()) };
798        let branch =
799            if self.source[term].opcode == Opcode::BrIf { self.out.terminator(out) } else { None };
800
801        let calls: Vec<rucc_ir::BlockCall> = self.source.successors(term).collect();
802        let mut succs = Vec::with_capacity(calls.len());
803        for call in calls {
804            let args: Vec<Value> = self.source[call.args].to_vec();
805            let mut regs = Vec::with_capacity(args.len());
806            for value in args {
807                regs.push(self.reg_of(value)?);
808            }
809            succs.push(mir::BlockCall { block: self.out_block(call.block), args: regs });
810        }
811        if let Some(branch) = branch {
812            if self.out.terminator(out) != Some(branch) {
813                self.out.remove_inst(branch);
814                self.out.append_inst(out, branch);
815            }
816        }
817        *self.out.succs_mut(out) = succs;
818        Ok(())
819    }
820
821    /// The machine IR block an IR block became.
822    fn out_block(&self, block: Block) -> mir::Block {
823        self.blocks[block.index()].expect("every block was created before any was filled")
824    }
825
826    /// The parameters of the entry block, which are the function's arguments.
827    ///
828    /// They are not block parameters in the machine IR and they cannot be. A block parameter is
829    /// given its value by a move on the edge into the block, and there is no edge into an entry
830    /// block, so what arrives in a function is the convention's to say. [`crate::abi`] is what
831    /// says it.
832    ///
833    /// The ones past the last register arrived in the caller's memory and are read out of it, and
834    /// the loads that read them come back here so that the frame can finish them the way it
835    /// finishes an `alloca`.
836    fn arrive(&mut self, block: Block, out: mir::Block) -> Result<(), Unsupported> {
837        let params = self.source[block].params.clone();
838        // The type of each is the block's answer and what the ABI asks of it is the signature's,
839        // and the two lists are the same list: a parameter the classification turned into a
840        // pointer is a pointer in the block too. A block with more parameters than the signature
841        // names is not one the front end writes, and each of those is taken as a plain value.
842        let asked: Vec<Abi> = self.source.signature().params.iter().map(|it| it.abi).collect();
843        let types: Vec<Param> = params
844            .iter()
845            .enumerate()
846            .map(|(index, &value)| {
847                let abi = asked.get(index).copied().unwrap_or_default();
848                Param { ty: self.source[value].ty, abi }
849            })
850            .collect();
851        // A save area for a function that takes arguments its signature does not name, on a
852        // convention whose list is the four field one. Windows is the other kind and has no area at
853        // all, so a `va_start` in one is refused rather than built wrong.
854        let variadic = self.source.signature().variadic && !self.conv.shared_positions;
855        let area = variadic.then(|| varargs::Area::of(self.conv));
856        let arrived = abi::entry(&mut self.out, out, &types, self.conv, self.names, area)
857            .map_err(|(index, missing)| Unsupported::Argument { index, missing })?;
858        for (&param, reg) in params.iter().zip(&arrived.regs) {
859            self.regs[param.index()] = Some(*reg);
860        }
861        if let Some(area) = area {
862            self.save_area(out, &arrived, area);
863        }
864        self.stack.arguments.extend(arrived.stack);
865        Ok(())
866    }
867
868    /// The prologue of a variadic function, which is every argument register it was handed written
869    /// into the frame.
870    ///
871    /// Every one the signature did not name, that is. Which of those hold anything is a thing only
872    /// the caller knew and there is nothing here to ask, so all of them are written, and the ones a
873    /// named parameter took are not, because `va_start` sets the two offsets past them and nothing
874    /// ever reads their slots.
875    ///
876    /// What that costs is up to fourteen stores in the prologue of a function that may read none of
877    /// them, and the convention's answer to that is the count of vector registers in `%al`, which
878    /// lets a callee skip the eight vector stores when the call passed no floats. Skipping them is a
879    /// branch in a prologue, and a prologue is written long after this by [`crate::finish`], which
880    /// has no blocks to branch between. So they are all written every time, which is correct and is
881    /// what `-O0` costs. Issue #323 is the branch.
882    ///
883    /// A vector register is written eight bytes at a time and not sixteen, for the reason
884    /// [`crate::varargs`] gives: the upper half of a slot is not something any reader of a list
885    /// looks at.
886    ///
887    /// The address is computed once into a register rather than written as a displacement off the
888    /// stack pointer, because a displacement into a frame is not known until after allocation and
889    /// one `lea` costs less than a fixup list for a dozen stores. It is the same `lea` an `alloca`
890    /// gets and [`crate::finish`] fills it in the same way.
891    fn save_area(&mut self, out: mir::Block, arrived: &abi::Arrived, area: varargs::Area) {
892        let save = self.stack.locals.len();
893        self.stack.locals.push(Local { size: area.size, align: varargs::VECTOR_SLOT });
894        self.varargs = Some(Varargs {
895            save,
896            incoming: arrived.used,
897            integers: u32::try_from(arrived.took.0).unwrap_or(0) * area.stride(false),
898            floats: area.starts_at(true)
899                + u32::try_from(arrived.took.1).unwrap_or(0) * area.stride(true),
900        });
901
902        let base = self.frame_address(out, save);
903        for &(reg, class, at) in &arrived.spare {
904            let name = if class == self.gpr { "x64.mov_mr_64" } else { "x64.movsd_mr" };
905            let store = mir::Opcode::new(self.names.intern(name));
906            let up = i32::try_from(at).expect("a register save area under two gigabytes");
907            let mem = mir::Mem::at(mir::Operand::read(base, self.gpr)).plus(up);
908            self.out.build(out, store).uses(reg, class).mem(mem).finish();
909        }
910    }
911
912    /// The address of one of the function's stack objects, in a fresh register.
913    ///
914    /// Written with nothing in its displacement, because where an object is in a frame is not known
915    /// until after allocation, and given to [`crate::finish`] to fill in the way an `alloca` is.
916    fn frame_address(&mut self, out: mir::Block, local: usize) -> mir::Reg {
917        let reg = self.out.new_vreg(self.gpr);
918        let lea = mir::Opcode::new(self.names.intern(&format!("{PREFIX}{}", x86_64::FRAME.lea)));
919        let sp = mir::Operand::read(mir::Reg::physical(self.conv.stack_pointer), self.gpr);
920        let made = self.out.build(out, lea).def(reg, self.gpr).mem(mir::Mem::at(sp)).finish();
921        self.stack.addresses.push((made, local));
922        reg
923    }
924
925    /// Whether an instruction is one no machine instruction is written for where it stands.
926    ///
927    /// Four of them, and none is a lowering decision, which is why none is a rule. A constant is
928    /// written where a register for it is first wanted rather than where the IR put it, and every
929    /// reader of one may have folded it into an immediate, in which case nowhere is the right
930    /// place. A return of nothing has nothing to put anywhere: the epilogue gives the frame back
931    /// and leaves, and it is appended to every block with no successors long after this has
932    /// finished, so a return with a value is one instruction here and a return without one is
933    /// none. Unless the value went back through memory, in which case there is something to put
934    /// somewhere after all and the IR does not carry it: the address the caller handed over has
935    /// to be in `rax` on the way out, and [`Lowering::returned`] is what writes that.
936    ///
937    /// An unconditional jump is the third, and there is even less of it: the edge is on the
938    /// block, and whether the block it goes to is the next one and needs no jump at all is the
939    /// block layout's answer rather than this one's.
940    ///
941    /// The fourth is a point control does not arrive at, in both of the forms the IR has for it:
942    /// the `unreachable` terminator the front end puts at the end of a function whose body can run
943    /// off the bottom, and the `unreachable_hint` a call to `__builtin_unreachable` becomes. What
944    /// to write for a place nothing reaches is a question with no wrong answer, and nothing is the
945    /// smallest one and the one gcc 16.2.0 gives at `-O0`. The terminator leaves the block with no
946    /// successors, so the epilogue lands at the end of it the way it does on any other block that
947    /// goes nowhere, and the function cannot fall out of its own last instruction into whatever
948    /// the assembler puts next.
949    fn writes_nothing(&self, inst: Inst) -> bool {
950        let data = &self.source[inst];
951        match data.opcode {
952            Opcode::IConst | Opcode::Jump | Opcode::Unreachable | Opcode::UnreachableHint => true,
953            Opcode::Return => self.source[data.args].is_empty() && self.sret().is_none(),
954            _ => false,
955        }
956    }
957
958    /// The rule that fires on an instruction, and what it bound.
959    ///
960    /// The plans are tried in order and the first that matches wins, which is the maximal munch
961    /// `spec/10-backend.md` asks for: a plan that offers more to the matcher is tried before one
962    /// that offers less.
963    fn select(&self, inst: Inst) -> Option<(Plan, Match<Term>)> {
964        for plan in self.plans(inst) {
965            let terms = Terms::new(self.source, inst, plan);
966            if let Some(matched) = TABLE.find(&terms, Term::Root) {
967                return Some((plan, matched));
968            }
969        }
970        None
971    }
972
973    /// Every way this instruction can be shown to the matcher, most offered first.
974    fn plans(&self, inst: Inst) -> Vec<Plan> {
975        let args = &self.source[self.source[inst].args];
976        let mut plans = vec![PLAIN];
977        for (index, &arg) in args.iter().enumerate().take(MAX_ARGS) {
978            let mut ways = Vec::new();
979            if self.foldable(inst, arg) {
980                ways.push(Shown::Expand);
981            }
982            if Terms::new(self.source, inst, PLAIN).constant(arg).is_some() {
983                ways.push(Shown::Const);
984            }
985            ways.push(Shown::Reg);
986            plans = plans
987                .into_iter()
988                .flat_map(|plan| {
989                    ways.iter().map(move |&way| {
990                        let mut next = plan;
991                        next[index] = way;
992                        next
993                    })
994                })
995                .collect();
996        }
997        plans
998    }
999
1000    /// Whether an operand may be shown as the instruction that computed it.
1001    ///
1002    /// It has to be in the same block, because a rule that folds one instruction into another
1003    /// moves the work to where the second one is. It has to be read only by this instruction,
1004    /// because folding it does not delete it for anybody else and doing the work twice is not a
1005    /// saving. And it has to be something rather than a block parameter, and not a constant,
1006    /// which is shown as a constant instead.
1007    fn foldable(&self, into: Inst, value: Value) -> bool {
1008        let Def::Result { inst, .. } = self.source[value].def else { return false };
1009        if self.source[inst].opcode == Opcode::IConst || self.uses[value.index()] != 1 {
1010            return false;
1011        }
1012        self.source.block_of(inst).is_some()
1013            && self.source.block_of(inst) == self.source.block_of(into)
1014    }
1015
1016    /// The instructions a match folded into the one it matched.
1017    ///
1018    /// The plan is what says this, not the bindings: a binding is a register or a number either
1019    /// way, and an operand shown as the instruction that computed it is one no rule could have
1020    /// matched without taking that instruction, because the plan offered the matcher nothing
1021    /// else to call it.
1022    fn folds(&self, inst: Inst, plan: Plan) -> Vec<Inst> {
1023        let args = &self.source[self.source[inst].args];
1024        args.iter()
1025            .take(MAX_ARGS)
1026            .enumerate()
1027            .filter(|&(index, _)| plan[index] == Shown::Expand)
1028            .filter_map(|(_, &arg)| match self.source[arg].def {
1029                Def::Result { inst, .. } => Some(inst),
1030                Def::Param { .. } => None,
1031            })
1032            .collect()
1033    }
1034
1035    /// Build the machine instruction a match calls for.
1036    fn emit(&mut self, inst: Inst, matched: &Match<Term>) -> Result<(), Unsupported> {
1037        let rule: &Rule = TABLE.rule(matched);
1038        let pieces = rule.replacement;
1039        let Some(Piece::App { head, arity }) = pieces.first() else {
1040            return Err(self.unsupported(inst));
1041        };
1042        let opcode = head.strip_prefix(PREFIX).ok_or_else(|| self.unsupported(inst))?;
1043        let form = x86_64::form(opcode).ok_or_else(|| self.unsupported(inst))?;
1044
1045        let mut read = Read::default();
1046        let mut at = 1;
1047        for _ in 0..*arity {
1048            at = self.read(inst, pieces, at, &matched.bindings, &mut read)?;
1049        }
1050
1051        let descs = form.operands();
1052        let writes = descs.iter().take_while(|desc| desc.role.is_def()).count();
1053        if descs.len() - writes != read.regs.len() {
1054            return Err(self.unsupported(inst));
1055        }
1056
1057        // The first thing the instruction writes is what it computes, and any others are
1058        // registers the machine destroys on the way, which are fresh because nothing else is in
1059        // them and nothing reads them. An instruction that writes nothing at all is one whose
1060        // whole purpose is its effect, which is what a store is, and there is no result to put
1061        // anywhere.
1062        let mut regs = Vec::new();
1063        if writes > 0 {
1064            let result = self.source[inst].first_result.ok_or_else(|| self.unsupported(inst))?;
1065            regs.push(self.new_reg(result));
1066            // The rest are the registers the machine destroys on the way, and the class each is in
1067            // is the one the instruction's description gives it rather than a guess, so that an
1068            // instruction that wrecks a register in the other file says so.
1069            regs.extend(descs[1..writes].iter().map(|desc| self.out.new_vreg(desc.class)));
1070        } else if self.source[inst].first_result.is_some() {
1071            // A rule that throws away a value the IR gave a name to would leave every reader of
1072            // that name with nothing to read, so it is a rule this and the target disagree about.
1073            return Err(self.unsupported(inst));
1074        }
1075        regs.extend(read.regs.iter().copied());
1076
1077        let block = self.at.expect("a block is being filled");
1078        let opcode = mir::Opcode::new(self.names.intern(head));
1079        let mut build = self.out.build(block, opcode).at(self.source.span(inst));
1080        for (desc, reg) in descs.iter().zip(regs) {
1081            let operand = mir::Operand {
1082                reg,
1083                class: desc.class,
1084                role: desc.role,
1085                constraint: desc.constraint,
1086            };
1087            build = build.operand(operand);
1088        }
1089        if let Some(mem) = read.mem {
1090            build = build.mem(mem);
1091        }
1092        if let Some(imm) = read.imm {
1093            build = build.imm(imm);
1094        }
1095        build.finish();
1096        Ok(())
1097    }
1098
1099    /// Read one argument of a replacement, which is a register, a number or an address.
1100    ///
1101    /// Gives back the position after it, because a replacement is flat and an address takes
1102    /// arguments of its own.
1103    fn read(
1104        &mut self,
1105        inst: Inst,
1106        pieces: &'static [Piece],
1107        at: usize,
1108        bindings: &[Term],
1109        out: &mut Read,
1110    ) -> Result<usize, Unsupported> {
1111        match pieces.get(at) {
1112            Some(Piece::Int(value)) => {
1113                out.imm = i64::try_from(*value).ok();
1114                Ok(at + 1)
1115            }
1116            Some(Piece::Var { index, .. }) => {
1117                match bindings.get(*index) {
1118                    Some(&Term::Reg(value)) => {
1119                        let reg = self.reg_of(value)?;
1120                        out.regs.push(reg);
1121                    }
1122                    Some(&Term::Num(value)) => out.imm = i64::try_from(value).ok(),
1123                    // A pattern binds a register or a number and nothing else, so this is a
1124                    // rule the matcher and this file disagree about.
1125                    _ => return Err(self.unsupported(inst)),
1126                }
1127                Ok(at + 1)
1128            }
1129            Some(Piece::App { head, arity }) => {
1130                let kind = x86_64::address(head).ok_or_else(|| self.unsupported(inst))?;
1131                let mut inner = Read::default();
1132                let mut next = at + 1;
1133                for _ in 0..*arity {
1134                    next = self.read(inst, pieces, next, bindings, &mut inner)?;
1135                }
1136                let mem = address(kind, &inner, self.gpr).ok_or_else(|| self.unsupported(inst))?;
1137                out.mem = Some(mem);
1138                Ok(next)
1139            }
1140            None => Err(self.unsupported(inst)),
1141        }
1142    }
1143
1144    /// The register a value is in, materializing it if it is a constant that has not been put in
1145    /// one yet.
1146    ///
1147    /// A constant is written where it is wanted rather than where the IR defined it, and where it
1148    /// is wanted is a block that need not be the one the IR defined it in. So the register holding
1149    /// one is only good inside the block it was written into, and a second block that wants the
1150    /// same constant gets its own. Anything else is a register read where nothing wrote it: the
1151    /// IR guarantees a definition dominates its uses, and this moved the definition.
1152    ///
1153    /// Writing the number again is also the right answer and not merely the safe one. It is one
1154    /// instruction that reads nothing, which is cheaper than holding a register live across a
1155    /// branch for it, and it is what a rematerializing allocator would do with the value anyway.
1156    fn reg_of(&mut self, value: Value) -> Result<mir::Reg, Unsupported> {
1157        let constant = match self.source[value].def {
1158            Def::Result { inst, .. } => {
1159                (self.source[inst].opcode == Opcode::IConst).then_some(inst)
1160            }
1161            Def::Param { .. } => None,
1162        };
1163        let here = self.at.expect("a block is being filled");
1164        if let Some(reg) = self.regs[value.index()] {
1165            if constant.is_none() || self.written[value.index()] == Some(here) {
1166                return Ok(reg);
1167            }
1168        }
1169        if let Some(inst) = constant {
1170            // Cleared so that the register the constant is written into is a new one rather than
1171            // the one the block above wrote, which is still being read up there.
1172            self.regs[value.index()] = None;
1173            let matched = self
1174                .select(inst)
1175                .map(|(_, matched)| matched)
1176                .ok_or_else(|| self.unsupported(inst))?;
1177            self.emit(inst, &matched)?;
1178            self.written[value.index()] = Some(here);
1179            return Ok(self.regs[value.index()].expect("a constant is written into a register"));
1180        }
1181        Ok(self.new_reg(value))
1182    }
1183
1184    /// Which register file a value of that type lives in.
1185    ///
1186    /// The vector one for the two float widths the machine has scalar instructions for, and the
1187    /// general purpose one for everything else. A `long double` is in neither, and it is here
1188    /// rather than in the vector class on purpose: it would be put in a register that cannot hold
1189    /// it, and there is no rule that names one, so the instruction computing it is reported. The
1190    /// wrong class would make that a wrong program instead of a refused one.
1191    fn class_of(&self, ty: Type) -> RegClass {
1192        match crate::term::float_slot(ty) {
1193            Some(_) => self.conv.sse_class,
1194            None => self.gpr,
1195        }
1196    }
1197
1198    /// A fresh register for a value, which is what the instruction computing it writes.
1199    fn new_reg(&mut self, value: Value) -> mir::Reg {
1200        if let Some(reg) = self.regs[value.index()] {
1201            return reg;
1202        }
1203        let reg = self.out.new_vreg(self.class_of(self.source[value].ty));
1204        self.regs[value.index()] = Some(reg);
1205        reg
1206    }
1207
1208    fn unsupported(&self, inst: Inst) -> Unsupported {
1209        let data = &self.source[inst];
1210        Unsupported::Inst {
1211            inst,
1212            term: Terms::new(self.source, inst, PLAIN).name(inst),
1213            opcode: data.opcode,
1214            ty: data.first_result.map(|result| self.source[result].ty),
1215        }
1216    }
1217}
1218
1219/// What the arguments of one replacement came to.
1220#[derive(Debug, Default)]
1221struct Read {
1222    regs: Vec<mir::Reg>,
1223    imm: Option<i64>,
1224    mem: Option<mir::Mem>,
1225}
1226
1227/// The addressing mode an address constructor's arguments make.
1228///
1229/// One arm per constructor rather than a question asked of the kind, because what the arguments
1230/// mean is the whole of what tells the four apart: the same register is a base in one and an
1231/// index in another, and the same constant is a scale in one and a displacement in another.
1232fn address(kind: x86_64::Address, read: &Read, gpr: RegClass) -> Option<mir::Mem> {
1233    let mut regs = read.regs.iter().copied().map(|reg| mir::Operand::read(reg, gpr));
1234    match kind {
1235        x86_64::Address::BaseIndexScale => {
1236            let base = regs.next()?;
1237            let index = regs.next()?;
1238            Some(mir::Mem::at(base).indexed(index, u8::try_from(read.imm?).ok()?))
1239        }
1240        x86_64::Address::IndexScale => Some(mir::Mem {
1241            base: None,
1242            index: Some(regs.next()?),
1243            scale: u8::try_from(read.imm?).ok()?,
1244            disp: 0,
1245            symbol: None,
1246        }),
1247        x86_64::Address::Base => Some(mir::Mem::at(regs.next()?)),
1248        // The rule that writes this has a guard saying the constant fits, so a displacement that
1249        // does not is a rule and a target that disagree rather than a program this cannot compile.
1250        x86_64::Address::BaseOffset => {
1251            Some(mir::Mem { disp: i32::try_from(read.imm?).ok()?, ..mir::Mem::at(regs.next()?) })
1252        }
1253    }
1254}
1255
1256/// The table this selector matches with.
1257///
1258/// One target for now, because one target has a rule file. Which table to use becomes a question
1259/// the moment a second one does, and the answer will be the target the session was given rather
1260/// than a constant here.
1261static TABLE: &Table = &crate::select::x86_64::TABLE;
1262
1263#[cfg(test)]
1264mod tests {
1265    use rucc_ir::{Builder, CallInfo, Flags, InstData, MemInfo, MemOrder, Signature, Type};
1266    use rucc_regalloc::assign::Env;
1267    use rucc_target::x86_64::{FRAME, REGS, SYSV};
1268
1269    use super::*;
1270    use crate::finish::finish;
1271    use crate::frame::{Frame, Incoming, Layout};
1272
1273    /// A function of as many 64 bit parameters as the test wants, and the block they are in.
1274    fn blank(params: &[Type]) -> (Interner, Func, Block, Vec<Value>) {
1275        let mut names = Interner::new();
1276        let mut func = Func::new(names.intern("f"), Signature::new());
1277        let block = func.create_block();
1278        let values = params.iter().map(|&ty| func.append_param(block, ty)).collect();
1279        (names, func, block, values)
1280    }
1281
1282    /// An ordinary access: not atomic, and aligned enough that nothing here has an opinion.
1283    /// Neither field reaches selection, which is the point of saying it once here.
1284    fn plain() -> MemInfo {
1285        MemInfo { size: 0, align: 1, order: MemOrder::NotAtomic, tbaa: None }
1286    }
1287
1288    /// What the allocator is given: every integer register the convention offers except two, held
1289    /// back so that a move on an edge has somewhere to break a cycle and a spilled value has
1290    /// somewhere to be read into. Which two does not matter, and holding back the last two the
1291    /// convention would reach for leaves every expectation below unchanged.
1292    fn env() -> Env {
1293        const SCRATCH: [rucc_target::PhysReg; 2] = [x86_64::R10, x86_64::R11];
1294        let order: Vec<rucc_target::PhysReg> =
1295            SYSV.int_order.iter().copied().filter(|reg| !SCRATCH.contains(reg)).collect();
1296        Env::new().with(x86_64::GPR, &order, &SCRATCH)
1297    }
1298
1299    /// The machine IR text a function lowers to.
1300    fn lower(names: &mut Interner, source: &Func) -> String {
1301        let out = func(source, names, &SYSV).expect("every instruction has a rule");
1302        mir::print_func(&out.func, names, &REGS)
1303    }
1304
1305    #[test]
1306    fn an_addition_of_two_registers_is_one_instruction() {
1307        let i32 = Type::int(32);
1308        let (mut names, mut func, block, args) = blank(&[i32, i32]);
1309        let mut build = Builder::new(&mut func, block);
1310        build.binary(Opcode::Add, args[0], args[1], Flags::default());
1311
1312        assert_eq!(
1313            lower(&mut names, &func),
1314            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
1315             %1:gpr($rsi) = x64.arg_val_32\n    %2:gpr(reuse 1) = x64.add_rr_32 %0, %1\n}\n"
1316        );
1317    }
1318
1319    #[test]
1320    fn a_constant_operand_becomes_an_immediate() {
1321        let i32 = Type::int(32);
1322        let (mut names, mut func, block, args) = blank(&[i32]);
1323        let mut build = Builder::new(&mut func, block);
1324        let seven = build.iconst(i32, 7);
1325        build.binary(Opcode::Add, args[0], seven, Flags::default());
1326
1327        // The constant is in the instruction and nothing was written to hold it, which is what
1328        // materializing one where a register for it is wanted buys.
1329        assert_eq!(
1330            lower(&mut names, &func),
1331            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
1332             %1:gpr(reuse 1) = x64.add_ri_32 %0, 7\n}\n"
1333        );
1334    }
1335
1336    #[test]
1337    fn a_constant_too_wide_for_an_immediate_goes_into_a_register() {
1338        let i64 = Type::int(64);
1339        let (mut names, mut func, block, args) = blank(&[i64]);
1340        let mut build = Builder::new(&mut func, block);
1341        let big = build.iconst(i64, i128::from(i32::MAX) + 1);
1342        build.binary(Opcode::Add, args[0], big, Flags::default());
1343
1344        // Nobody wrote this fallback down. The rule that takes an immediate has a guard that
1345        // turns a number this wide down, so it does not fire, and the next way of showing the
1346        // operand puts it in a register.
1347        assert_eq!(
1348            lower(&mut names, &func),
1349            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
1350             %1:gpr = x64.mov_ri_64 2147483648\n    %2:gpr(reuse 1) = x64.add_rr_64 %0, %1\n}\n"
1351        );
1352    }
1353
1354    #[test]
1355    fn an_index_calculation_folds_into_an_address() {
1356        let i64 = Type::int(64);
1357        let (mut names, mut func, block, args) = blank(&[i64, i64]);
1358        let mut build = Builder::new(&mut func, block);
1359        let four = build.iconst(i64, 4);
1360        let scaled = build.binary(Opcode::Mul, args[1], four, Flags::default());
1361        build.binary(Opcode::Add, args[0], scaled, Flags::default());
1362
1363        // Three IR instructions and one machine instruction. The multiply is gone because the
1364        // rule that matched reached down and took it.
1365        assert_eq!(
1366            lower(&mut names, &func),
1367            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
1368             %1:gpr($rsi) = x64.arg_val_64\n    %2:gpr = x64.lea_64 [%0 + %1*4]\n}\n"
1369        );
1370    }
1371
1372    #[test]
1373    fn an_instruction_read_twice_is_not_folded_into_either_reader() {
1374        let i64 = Type::int(64);
1375        let (mut names, mut func, block, args) = blank(&[i64, i64]);
1376        let mut build = Builder::new(&mut func, block);
1377        let four = build.iconst(i64, 4);
1378        let scaled = build.binary(Opcode::Mul, args[1], four, Flags::default());
1379        let first = build.binary(Opcode::Add, args[0], scaled, Flags::default());
1380        build.binary(Opcode::Add, first, scaled, Flags::default());
1381
1382        // Folding it into both would compute it twice, which is not a saving, so it stays where
1383        // it is and both readers read the register it wrote.
1384        let text = lower(&mut names, &func);
1385        assert!(text.contains("x64.lea_64 [%1*4]"), "{text}");
1386        assert_eq!(text.matches("x64.add_rr_64").count(), 2, "{text}");
1387    }
1388
1389    #[test]
1390    fn a_shift_by_a_register_asks_for_it_in_cl() {
1391        let i32 = Type::int(32);
1392        let (mut names, mut func, block, args) = blank(&[i32, i32]);
1393        let mut build = Builder::new(&mut func, block);
1394        build.binary(Opcode::Shl, args[0], args[1], Flags::default());
1395
1396        // The fixed register is not in the rule. It is what the target says the instruction does
1397        // with its operands, and the allocator is what will act on it.
1398        let text = lower(&mut names, &func);
1399        assert!(text.contains("x64.shl_rcl_32 %0, %1($rcx)"), "{text}");
1400    }
1401
1402    #[test]
1403    fn a_division_names_the_registers_and_the_register_it_destroys() {
1404        let i32 = Type::int(32);
1405        let (mut names, mut func, block, args) = blank(&[i32, i32]);
1406        let mut build = Builder::new(&mut func, block);
1407        build.binary(Opcode::SDiv, args[0], args[1], Flags::default());
1408
1409        // Two definitions, because a division writes the remainder whether anybody wanted it or
1410        // not, and the second one is early because it is destroyed before the operands are read.
1411        let text = lower(&mut names, &func);
1412        assert!(
1413            text.contains("%2:gpr($rax), early %3:gpr($rdx) = x64.idiv_quo_32 %0($rax), %1"),
1414            "{text}"
1415        );
1416    }
1417
1418    #[test]
1419    fn a_load_reads_through_the_register_the_address_is_in() {
1420        let i64 = Type::int(64);
1421        let (mut names, mut func, block, args) = blank(&[i64]);
1422        let mut build = Builder::new(&mut func, block);
1423        build.load(Type::int(32), args[0], plain(), Flags::default());
1424
1425        assert_eq!(
1426            lower(&mut names, &func),
1427            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
1428             %1:gpr = x64.mov_rm_32 [%0]\n}\n"
1429        );
1430    }
1431
1432    #[test]
1433    fn a_store_writes_no_register_and_the_value_it_writes_is_the_one_the_ir_gave_it() {
1434        let (mut names, mut func, block, args) = blank(&[Type::int(32), Type::int(64)]);
1435        let mut build = Builder::new(&mut func, block);
1436        build.store(args[0], args[1], plain(), Flags::default());
1437
1438        // The value is the first parameter and the address is the second, and the instruction
1439        // takes them the other way round. Getting that backwards would compile to a store of the
1440        // address into the value, which is a program that runs and does the wrong thing.
1441        assert_eq!(
1442            lower(&mut names, &func),
1443            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
1444             %1:gpr($rsi) = x64.arg_val_64\n    x64.mov_mr_32 %0, [%1]\n}\n"
1445        );
1446    }
1447
1448    #[test]
1449    fn an_address_with_a_constant_added_folds_into_the_access() {
1450        let i64 = Type::int(64);
1451        let (mut names, mut func, block, args) = blank(&[i64]);
1452        let mut build = Builder::new(&mut func, block);
1453        let twelve = build.iconst(i64, 12);
1454        let field = build.binary(Opcode::Add, args[0], twelve, Flags::default());
1455        build.load(Type::int(64), field, plain(), Flags::default());
1456
1457        // Two IR instructions and one machine instruction, which is what every read of a field
1458        // of a structure comes to.
1459        assert_eq!(
1460            lower(&mut names, &func),
1461            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
1462             %1:gpr = x64.mov_rm_64 [%0 + 12]\n}\n"
1463        );
1464    }
1465
1466    #[test]
1467    fn a_displacement_too_wide_to_encode_leaves_the_addition_where_it_is() {
1468        let i64 = Type::int(64);
1469        let (mut names, mut func, block, args) = blank(&[i64]);
1470        let mut build = Builder::new(&mut func, block);
1471        let big = build.iconst(i64, i128::from(i32::MAX) + 1);
1472        let far = build.binary(Opcode::Add, args[0], big, Flags::default());
1473        build.load(Type::int(32), far, plain(), Flags::default());
1474
1475        // A displacement is signed and 32 bits. The rule that folds one has a guard that turns
1476        // this down, so the addition stays and the load reads through what it produced. Nobody
1477        // wrote that fallback: it is the next way of showing the operand.
1478        let text = lower(&mut names, &func);
1479        assert!(text.contains("x64.mov_rm_32 [%2]"), "{text}");
1480        assert!(text.contains("x64.add_rr_64"), "{text}");
1481    }
1482
1483    #[test]
1484    fn a_store_of_a_value_that_was_loaded_is_two_instructions_and_no_arithmetic() {
1485        let i64 = Type::int(64);
1486        let (mut names, mut func, block, args) = blank(&[i64, i64]);
1487        let mut build = Builder::new(&mut func, block);
1488        let got = build.load(Type::int(8), args[0], plain(), Flags::default());
1489        build.store(got, args[1], plain(), Flags::default());
1490
1491        // A load feeding a store is the one place folding would be wrong: an x86-64 `mov` has at
1492        // most one memory operand, and there is no rule that takes two, so the load is left where
1493        // it is and the store reads the register it wrote.
1494        assert_eq!(
1495            lower(&mut names, &func),
1496            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
1497             %1:gpr($rsi) = x64.arg_val_64\n    %2:gpr = x64.mov_rm_8 [%0]\n    \
1498             x64.mov_mr_8 %2, [%1]\n}\n"
1499        );
1500    }
1501
1502    #[test]
1503    fn an_access_at_a_width_no_rule_is_written_at_is_reported() {
1504        let i64 = Type::int(64);
1505        let (mut names, mut source, block, args) = blank(&[i64]);
1506        let mut build = Builder::new(&mut source, block);
1507        build.load(Type::int(128), args[0], plain(), Flags::default());
1508
1509        // The width is the whole of what is wrong here, so the width is in the message: `load`
1510        // on its own is written about at every other width and would send a reader looking in
1511        // the wrong place.
1512        let failed = func(&source, &mut names, &SYSV).expect_err("nothing loads 128 bits");
1513        assert_eq!(failed.to_string(), "no rule lowers a `load` producing a `i128`");
1514    }
1515
1516    #[test]
1517    fn a_return_asks_for_the_value_in_the_register_the_caller_reads() {
1518        let (mut names, mut func, block, args) = blank(&[Type::int(32)]);
1519        let mut build = Builder::new(&mut func, block);
1520        build.ret(&[args[0]]);
1521
1522        // The register is not in the rule, the same way `cl` is not in the rule for a shift. It
1523        // is what the target says the instruction does with its operand, and the allocator is
1524        // what will act on it. There is no `ret` here, because giving the frame back has to
1525        // happen between this and leaving and the frame is not worked out yet.
1526        assert_eq!(
1527            lower(&mut names, &func),
1528            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
1529             x64.ret_val_32 %0($rax)\n}\n"
1530        );
1531    }
1532
1533    #[test]
1534    fn a_return_of_two_values_asks_for_the_second_register_as_well() {
1535        let i64 = Type::int(64);
1536        let (mut names, mut func, block, args) = blank(&[i64, i64]);
1537        let mut build = Builder::new(&mut func, block);
1538        build.ret(&[args[0], args[1]]);
1539
1540        // `struct { long a, b; } f(long a, long b)`, after the front end has classified it. Both
1541        // halves are integers, so the second is in the second integer return register, and both
1542        // pseudos say so the same way the one for a single value does.
1543        assert_eq!(
1544            lower(&mut names, &func),
1545            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
1546             %1:gpr($rsi) = x64.arg_val_64\n    x64.ret_val_64 %0($rax)\n    \
1547             x64.ret_val2_64 %1($rdx)\n}\n"
1548        );
1549    }
1550
1551    #[test]
1552    fn two_values_back_in_different_files_are_both_the_first_of_their_own() {
1553        let f64 = Type::float(rucc_ir::Float::F64);
1554        let (mut names, mut func, block, args) = blank(&[f64, Type::int(64)]);
1555        let mut build = Builder::new(&mut func, block);
1556        build.ret(&[args[0], args[1]]);
1557
1558        // `struct { double a; long b; } f(double a, long b)`. The two files are counted apart, so
1559        // neither half is the second of anything and the `double` is in `xmm0` rather than in the
1560        // register a second `double` would have been in. Getting this wrong is not a crash: the
1561        // caller reads a register nobody wrote, and this is where that is ruled out.
1562        assert_eq!(
1563            lower(&mut names, &func),
1564            "mfunc @f {\nblock0:\n    %0:xmm($xmm0) = x64.arg_val_f64\n    \
1565             %1:gpr($rdi) = x64.arg_val_64\n    x64.ret_val_f64 %0($xmm0)\n    \
1566             x64.ret_val_64 %1($rax)\n}\n"
1567        );
1568    }
1569
1570    #[test]
1571    fn two_of_the_same_file_back_take_the_first_two_of_it() {
1572        let f64 = Type::float(rucc_ir::Float::F64);
1573        let (mut names, mut func, block, args) = blank(&[f64, f64]);
1574        let mut build = Builder::new(&mut func, block);
1575        build.ret(&[args[0], args[1]]);
1576
1577        // `struct { double x, y; } f(double x, double y)`, which is the vector half of the pair
1578        // above and counts in its own file the same way.
1579        assert_eq!(
1580            lower(&mut names, &func),
1581            "mfunc @f {\nblock0:\n    %0:xmm($xmm0) = x64.arg_val_f64\n    \
1582             %1:xmm($xmm1) = x64.arg_val_f64\n    x64.ret_val_f64 %0($xmm0)\n    \
1583             x64.ret_val2_f64 %1($xmm1)\n}\n"
1584        );
1585    }
1586
1587    /// A function whose answer goes back through memory, with the pointer to the space for it in
1588    /// front of whatever else it takes. Only the signature says it is one.
1589    fn returning_through_memory(params: &[Type]) -> (Interner, Func, Block, Vec<Value>) {
1590        let mut names = Interner::new();
1591        let sret = Abi::Sret { size: 32, align: 8 };
1592        let mut signature = Signature::new().and_param(Param::with_abi(Type::PTR, sret));
1593        signature.params.extend(params.iter().copied().map(Param::new));
1594        let mut func = Func::new(names.intern("f"), signature);
1595        let block = func.create_block();
1596        let space = func.append_param(block, Type::PTR);
1597        let values = std::iter::once(space)
1598            .chain(params.iter().map(|&ty| func.append_param(block, ty)))
1599            .collect();
1600        (names, func, block, values)
1601    }
1602
1603    #[test]
1604    fn the_space_a_return_through_memory_was_given_goes_back_in_the_first_return_register() {
1605        let (mut names, mut func, block, _) = returning_through_memory(&[]);
1606        Builder::new(&mut func, block).ret(&[]);
1607
1608        // `struct big f(void)`, where `big` is too large to come back in registers. The `return`
1609        // carries nothing, because the value went into the space the caller handed over, and the
1610        // document still says that address comes back in `rax`. Nothing in the IR says it, so the
1611        // convention says it, and the pseudo is the one any other pointer return would use.
1612        assert_eq!(
1613            lower(&mut names, &func),
1614            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
1615             x64.ret_val_64 %0($rax)\n}\n"
1616        );
1617    }
1618
1619    #[test]
1620    fn what_the_function_did_in_between_does_not_take_the_register_off_it() {
1621        let (mut names, mut func, block, args) = returning_through_memory(&[Type::int(32)]);
1622        let mut build = Builder::new(&mut func, block);
1623        build.store(args[1], args[0], plain(), Flags::default());
1624        build.ret(&[]);
1625
1626        // The register is a read at the end and not a move at the start, so it is live across
1627        // everything between the two and the allocator has to keep it somewhere. In a function
1628        // with a call in it that somewhere is a callee saved register, and the address comes back
1629        // into `rax` here rather than whatever the last instruction happened to leave there. That
1630        // is issue #333, and a store is enough to show the value outlives the entry block.
1631        let text = lower(&mut names, &func);
1632        assert!(text.contains("x64.mov_mr_32 %1, [%0]"), "{text}");
1633        assert!(text.ends_with("    x64.ret_val_64 %0($rax)\n}\n"), "{text}");
1634    }
1635
1636    #[test]
1637    fn a_pointer_that_is_only_a_pointer_is_not_given_back() {
1638        let (mut names, mut func, block, args) = blank(&[Type::PTR]);
1639        let mut build = Builder::new(&mut func, block);
1640        build.store(args[0], args[0], plain(), Flags::default());
1641        build.ret(&[]);
1642
1643        // `void f(void **p)`. It takes a pointer first and returns nothing, which is the shape of
1644        // the one above and none of its meaning, and what tells them apart is the signature. A
1645        // `void` function leaves `rax` alone.
1646        assert!(!lower(&mut names, &func).contains("ret_val"));
1647    }
1648
1649    #[test]
1650    fn a_return_of_a_constant_puts_it_in_a_register_first() {
1651        let (mut names, mut func, block, _) = blank(&[]);
1652        let mut build = Builder::new(&mut func, block);
1653        let zero = build.iconst(Type::int(32), 0);
1654        build.ret(&[zero]);
1655
1656        // No rule returns an immediate, so the plan that offers one is turned down and the next
1657        // one materializes it. That is `int main(void) { return 0; }` in full, once the epilogue
1658        // is appended to it.
1659        assert_eq!(
1660            lower(&mut names, &func),
1661            "mfunc @f {\nblock0:\n    %0:gpr = x64.mov_ri_32 0\n    x64.ret_val_32 %0($rax)\n}\n"
1662        );
1663    }
1664
1665    #[test]
1666    fn a_return_of_nothing_is_no_instruction_at_all() {
1667        let (mut names, mut func, block, _) = blank(&[]);
1668        let mut build = Builder::new(&mut func, block);
1669        build.ret(&[]);
1670
1671        // Every part of leaving a function that returns nothing is the epilogue's, and the
1672        // epilogue goes in after allocation. A block with nothing in it is the right answer here
1673        // rather than a function that could not be lowered.
1674        assert_eq!(lower(&mut names, &func), "mfunc @f {\nblock0:\n}\n");
1675    }
1676
1677    #[test]
1678    fn the_allocator_is_what_moves_the_answer_into_the_return_register() {
1679        let (mut names, mut source, block, _) = blank(&[]);
1680        let mut build = Builder::new(&mut source, block);
1681        let zero = build.iconst(Type::int(32), 0);
1682        build.ret(&[zero]);
1683
1684        let mut out = func(&source, &mut names, &SYSV).expect("every instruction has a rule").func;
1685        let env = env();
1686        let allocation = rucc_regalloc::run(&mut out, &env);
1687        let frame = Frame::of(&out, &allocation, &Layout::new(&SYSV, REGS));
1688        finish(&mut out, &allocation, &frame, &Stack::default(), &SYSV, &FRAME, &mut names);
1689
1690        // `int main(void) { return 0; }` end to end. Nothing here asked for `rax`: the rule said
1691        // the value goes back, the target said where, and the allocator is what made it true. The
1692        // epilogue is what leaves, and this function needs no frame, so it is the return alone.
1693        //
1694        // The copy is a register allocator that takes no hints. It hands `%0` a register at the
1695        // instruction that writes it, where it does not yet know that a later use insists on
1696        // `rax`, and `rax` is not free to hand out because that later use is holding it. So the
1697        // value goes somewhere else and is copied in. Every division and every shift by a
1698        // register already pays the same thing, and paying it once per return is what makes it
1699        // worth fixing rather than a new problem.
1700        assert_eq!(
1701            mir::print_func(&out, &names, &REGS),
1702            "mfunc @f {\nblock0:\n    $rcx = x64.mov_ri_32 0\n    $rax = x64.mov_rr_64 $rcx\n    \
1703             x64.ret_val_32 $rax($rax)\n    x64.ret\n}\n"
1704        );
1705    }
1706
1707    #[test]
1708    fn a_function_of_two_arguments_is_a_whole_function_now() {
1709        let i32 = Type::int(32);
1710        let (mut names, mut source, block, args) = blank(&[i32, i32]);
1711        let mut build = Builder::new(&mut source, block);
1712        let sum = build.binary(Opcode::Add, args[0], args[1], Flags::default());
1713        build.ret(&[sum]);
1714
1715        let mut out = func(&source, &mut names, &SYSV).expect("every instruction has a rule").func;
1716        let env = env();
1717        let allocation = rucc_regalloc::run(&mut out, &env);
1718        let frame = Frame::of(&out, &allocation, &Layout::new(&SYSV, REGS));
1719        finish(&mut out, &allocation, &frame, &Stack::default(), &SYSV, &FRAME, &mut names);
1720
1721        // `int f(int a, int b) { return a + b; }` end to end, and this is the test the argument
1722        // side exists for. Before it there was no way to write one: the allocator refuses a
1723        // function whose entry block takes parameters, because there is no edge into an entry
1724        // block for the moves that give a block parameter its value to go on.
1725        //
1726        // Four moves that a good allocator writes none of, and it is the same allocator that
1727        // takes no hints as in the return above rather than anything new. It hands each argument
1728        // a register at the pseudo that defines it, without looking at the fixed register that
1729        // pseudo insists on, so every argument is copied straight back out of where it already
1730        // was. Issue #255 is this, and this function is the shortest program that shows what it
1731        // costs: one hint per argument and one per return would leave nothing here but the
1732        // addition. What the test is for meanwhile is that the answer is right, and it is: the
1733        // copy in front of a two address instruction is what makes its destination one of the
1734        // registers it reads, and the source operand keeps its own name because the destination
1735        // is what the encoder writes.
1736        assert_eq!(
1737            mir::print_func(&out, &names, &REGS),
1738            "mfunc @f {\nblock0:\n    $rdi($rdi) = x64.arg_val_32\n    \
1739             $rax = x64.mov_rr_64 $rdi\n    $rsi($rsi) = x64.arg_val_32\n    \
1740             $rcx = x64.mov_rr_64 $rsi\n    $rdx = x64.mov_rr_64 $rax\n    \
1741             $rdx(reuse 1) = x64.add_rr_32 $rax, $rcx\n    $rax = x64.mov_rr_64 $rdx\n    \
1742             x64.ret_val_32 $rax($rax)\n    x64.ret\n}\n"
1743        );
1744    }
1745
1746    #[test]
1747    fn an_argument_with_no_register_left_for_it_is_read_out_of_the_caller_s_stack() {
1748        let i64 = Type::int(64);
1749        let (mut names, mut source, block, args) = blank(&[i64; 7]);
1750        let mut build = Builder::new(&mut source, block);
1751        build.ret(&[args[6]]);
1752
1753        let lowered = func(&source, &mut names, &SYSV).expect("the seventh is read from memory");
1754
1755        // SysV passes six integers in registers and the seventh in the caller's memory, so six of
1756        // these are pseudos that encode to nothing and the seventh is a load that encodes to real
1757        // bytes. Its displacement is nothing here for the reason a local's is: there is no frame
1758        // yet. What the walk hands on is which instruction is waiting, and for how far up the
1759        // caller's argument area, which is the bottom of it because it is the first one there.
1760        assert_eq!(lowered.stack.arguments.len(), 1);
1761        assert_eq!(lowered.stack.arguments[0].1, 0);
1762        let text = mir::print_func(&lowered.func, &names, &REGS);
1763        assert!(text.contains("%6:gpr = x64.mov_rm_64 [$rsp]"), "{text}");
1764        assert_eq!(text.matches("x64.arg_val_64").count(), 6, "{text}");
1765    }
1766
1767    #[test]
1768    fn the_frame_is_what_says_how_far_up_the_caller_s_stack_an_argument_is() {
1769        let i64 = Type::int(64);
1770        let (mut names, mut source, block, args) = blank(&[i64; 8]);
1771        let mut build = Builder::new(&mut source, block);
1772        let sum = build.binary(Opcode::Add, args[6], args[7], Flags::default());
1773        build.ret(&[sum]);
1774
1775        let lowered = func(&source, &mut names, &SYSV).expect("both are read from memory");
1776        let stack = lowered.stack;
1777        let mut out = lowered.func;
1778        let env = env();
1779        let allocation = rucc_regalloc::run(&mut out, &env);
1780        let layout = stack.layout(Layout::new(&SYSV, REGS));
1781        let frame = Frame::of(&out, &allocation, &layout);
1782        finish(&mut out, &allocation, &frame, &stack, &SYSV, &FRAME, &mut names);
1783
1784        // A leaf that takes no frame, so the stack pointer never moves and the only thing between
1785        // it and the caller's arguments is the return address the call pushed. The seventh
1786        // parameter is at the bottom of the caller's argument area and the eighth is one word
1787        // further up, which is the eight bytes between the two offsets.
1788        let text = mir::print_func(&out, &names, &REGS);
1789        assert_eq!(frame.size(), 0);
1790        assert_eq!(frame.incoming(), Incoming::from_stack(8));
1791        assert!(text.contains("x64.mov_rm_64 [$rsp + 8]"), "{text}");
1792        assert!(text.contains("x64.mov_rm_64 [$rsp + 16]"), "{text}");
1793    }
1794
1795    #[test]
1796    fn a_realigned_frame_reaches_the_caller_s_arguments_through_the_frame_pointer() {
1797        let i64 = Type::int(64);
1798        let (mut names, mut source, block, args) = blank(&[i64; 7]);
1799        let wide = slot(&mut source, block, 64, 32);
1800        let mut build = Builder::new(&mut source, block);
1801        build.store(args[6], wide, plain(), Flags::default());
1802        build.ret(&[args[6]]);
1803
1804        let lowered = func(&source, &mut names, &SYSV).expect("every instruction has a rule");
1805        let stack = lowered.stack;
1806        let mut out = lowered.func;
1807        let env = env();
1808        let allocation = rucc_regalloc::run(&mut out, &env);
1809        let layout = stack.layout(Layout::new(&SYSV, REGS));
1810        let frame = Frame::of(&out, &allocation, &layout);
1811        finish(&mut out, &allocation, &frame, &stack, &SYSV, &FRAME, &mut names);
1812
1813        // A local wanting thirty two byte alignment makes the prologue force the stack pointer,
1814        // which throws away how far the caller's stack was. So the load the lowering wrote off the
1815        // stack pointer is rewritten to read through the frame pointer, at the one distance that
1816        // survives: the word the prologue pushed the frame pointer into, and the return address
1817        // above it.
1818        let text = mir::print_func(&out, &names, &REGS);
1819        assert_eq!(frame.realign(), Some(32));
1820        assert_eq!(frame.incoming(), Incoming::from_frame(16));
1821        assert!(text.contains("x64.mov_rm_64 [$rbp + 16]"), "{text}");
1822        assert!(!text.contains("x64.mov_rm_64 [$rsp"), "{text}");
1823    }
1824
1825    #[test]
1826    fn a_jump_is_the_edge_and_nothing_else() {
1827        let i32 = Type::int(32);
1828        let (mut names, mut source, entry, args) = blank(&[i32]);
1829        let next = source.create_block();
1830        let got = source.append_param(next, i32);
1831        Builder::new(&mut source, entry).jump(next, &[args[0]]);
1832        Builder::new(&mut source, next).ret(&[got]);
1833
1834        // Two blocks and two instructions, and the jump is neither of them. What it was is the
1835        // arm on the first block, and what the arm carries is the argument it was called with.
1836        assert_eq!(
1837            lower(&mut names, &source),
1838            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32 block1(%0)\n\n\
1839             block1(%1:gpr):\n    x64.ret_val_32 %1($rax)\n}\n"
1840        );
1841    }
1842
1843    /// A constant is written where it is wanted rather than where the IR defined it, and two
1844    /// blocks wanting the same one is two places. Writing it once and reading it in both is a
1845    /// register read where nothing wrote it, unless the block it was written in happens to
1846    /// dominate the other, which nothing here checks and which the second arm of a branch never
1847    /// does. Each block gets its own copy of the number instead.
1848    #[test]
1849    fn a_constant_two_blocks_want_is_written_in_both_of_them() {
1850        let i32 = Type::int(32);
1851        let (mut names, mut source, entry, args) = blank(&[i32, i32]);
1852        let then = source.create_block();
1853        let other = source.create_block();
1854        let join = source.create_block();
1855        let got = source.append_param(join, i32);
1856
1857        let mut build = Builder::new(&mut source, entry);
1858        let seven = build.iconst(i32, 7);
1859        let cond = build.icmp(rucc_ir::IntPred::Slt, args[0], args[1]);
1860        build.br_if(cond, then, &[], other, &[]);
1861        // Both arms want the seven in a register, because a block argument is never an immediate,
1862        // and neither arm dominates the other.
1863        Builder::new(&mut source, then).jump(join, &[seven]);
1864        Builder::new(&mut source, other).jump(join, &[seven]);
1865        Builder::new(&mut source, join).ret(&[got]);
1866
1867        let text = lower(&mut names, &source);
1868        assert_eq!(text.matches("x64.mov_ri_32 7").count(), 2, "one seven per block: {text}");
1869    }
1870
1871    /// An argument on an edge out of a block that leaves two ways is read after every instruction
1872    /// of the block is written, and reading one can write an instruction, which would land after
1873    /// the branch that has already jumped past it. The branch goes back on the end.
1874    #[test]
1875    fn a_constant_an_edge_wants_is_written_before_the_branch_and_not_after_it() {
1876        let i32 = Type::int(32);
1877        let (mut names, mut source, entry, args) = blank(&[i32, i32]);
1878        let then = source.create_block();
1879        let join = source.create_block();
1880        let got = source.append_param(join, i32);
1881
1882        let mut build = Builder::new(&mut source, entry);
1883        let nine = build.iconst(i32, 9);
1884        let cond = build.icmp(rucc_ir::IntPred::Slt, args[0], args[1]);
1885        build.br_if(cond, then, &[], join, &[nine]);
1886        Builder::new(&mut source, then).jump(join, &[args[0]]);
1887        Builder::new(&mut source, join).ret(&[got]);
1888
1889        let out = func(&source, &mut names, &SYSV).expect("every instruction has a rule").func;
1890        let entry = out.entry().expect("an entry block");
1891        let last = out.terminator(entry).expect("a block that leaves two ways has a branch");
1892        let branch = names.intern("x64.br_cond_8");
1893        assert_eq!(
1894            out[last].opcode,
1895            mir::Opcode::new(branch),
1896            "the branch is last: {}",
1897            mir::print_func(&out, &names, &REGS)
1898        );
1899    }
1900
1901    #[test]
1902    fn a_conditional_branch_is_lowered_to_the_condition_and_nothing_about_where_it_goes() {
1903        let i32 = Type::int(32);
1904        let (mut names, mut source, entry, args) = blank(&[i32, i32]);
1905        let then = source.create_block();
1906        let other = source.create_block();
1907        let mut build = Builder::new(&mut source, entry);
1908        let cond = build.icmp(rucc_ir::IntPred::Slt, args[0], args[1]);
1909        build.br_if(cond, then, &[], other, &[]);
1910        Builder::new(&mut source, then).ret(&[args[0]]);
1911        Builder::new(&mut source, other).ret(&[args[1]]);
1912
1913        // The comparison writes a byte and the branch reads it, and neither says a block. Both
1914        // arms are on the entry block, in the order the branch took them, so the arm that runs
1915        // when the condition holds is the first.
1916        assert_eq!(
1917            lower(&mut names, &source),
1918            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
1919             %1:gpr($rsi) = x64.arg_val_32\n    %2:gpr = x64.cmp_set_l_32 %0, %1\n    \
1920             x64.br_cond_8 %2, block1, block2\n\n\
1921             block1:\n    x64.ret_val_32 %0($rax)\n\n\
1922             block2:\n    x64.ret_val_32 %1($rax)\n}\n"
1923        );
1924    }
1925
1926    #[test]
1927    fn a_branch_over_a_block_is_a_whole_function_now() {
1928        let i32 = Type::int(32);
1929        let (mut names, mut source, entry, args) = blank(&[i32, i32]);
1930        let then = source.create_block();
1931        let other = source.create_block();
1932        let join = source.create_block();
1933        let got = source.append_param(join, i32);
1934        let mut build = Builder::new(&mut source, entry);
1935        let cond = build.icmp(rucc_ir::IntPred::Slt, args[0], args[1]);
1936        build.br_if(cond, then, &[], other, &[]);
1937        let mut build = Builder::new(&mut source, then);
1938        let sum = build.binary(Opcode::Add, args[0], args[1], Flags::default());
1939        build.jump(join, &[sum]);
1940        Builder::new(&mut source, other).jump(join, &[args[1]]);
1941        Builder::new(&mut source, join).ret(&[got]);
1942
1943        // `int f(int a, int b) { if (a < b) return a + b; else return b; }` end to end, written
1944        // the way a front end writes it: both arms of the branch are blocks of their own and the
1945        // return is the block they meet at. No edge here is critical, because the two arms out of
1946        // the entry carry nothing and the two arms into the join each leave a block that goes
1947        // nowhere else, so each has its own end to put its move at.
1948        let mut out = func(&source, &mut names, &SYSV).expect("every instruction has a rule").func;
1949        assert_eq!(crate::split::critical(&mut out), 0, "no edge here is critical");
1950        let env = env();
1951        let allocation = rucc_regalloc::run(&mut out, &env);
1952        let frame = Frame::of(&out, &allocation, &Layout::new(&SYSV, REGS));
1953        finish(&mut out, &allocation, &frame, &Stack::default(), &SYSV, &FRAME, &mut names);
1954
1955        // One epilogue, on the join, which is the one block the function leaves from, and the
1956        // moves that give the join its parameter are at the end of each arm. Every register is
1957        // physical and the branch is still a branch on a register, because turning it into a
1958        // `test` and a `jcc` is the block layout's and there is no block layout yet.
1959        let text = mir::print_func(&out, &names, &REGS);
1960        assert_eq!(text.matches("x64.ret\n").count(), 1, "{text}");
1961        assert!(text.contains("x64.br_cond_8"), "{text}");
1962        assert!(text.contains("x64.add_rr_32"), "{text}");
1963        assert!(!text.contains('%'), "{text}");
1964    }
1965
1966    #[test]
1967    fn a_critical_edge_is_split_before_the_allocator_ever_sees_it() {
1968        let i32 = Type::int(32);
1969        let (mut names, mut source, entry, args) = blank(&[i32, i32]);
1970        let then = source.create_block();
1971        let join = source.create_block();
1972        let got = source.append_param(join, i32);
1973        let mut build = Builder::new(&mut source, entry);
1974        let cond = build.icmp(rucc_ir::IntPred::Slt, args[0], args[1]);
1975        build.br_if(cond, then, &[], join, &[args[1]]);
1976        Builder::new(&mut source, then).jump(join, &[args[0]]);
1977        let mut build = Builder::new(&mut source, join);
1978        let twice = build.binary(Opcode::Add, got, got, Flags::default());
1979        build.ret(&[twice]);
1980
1981        // The else arm is critical: the entry block leaves two ways and the join is arrived at
1982        // two ways, and the arm carries a value. Without splitting it the allocator asserts,
1983        // because the move that gives the join its parameter would have to run at the end of a
1984        // block that also goes to the other arm.
1985        let mut out = func(&source, &mut names, &SYSV).expect("every instruction has a rule").func;
1986        assert_eq!(crate::split::critical(&mut out), 1);
1987        let env = env();
1988        let allocation = rucc_regalloc::run(&mut out, &env);
1989        let frame = Frame::of(&out, &allocation, &Layout::new(&SYSV, REGS));
1990        finish(&mut out, &allocation, &frame, &Stack::default(), &SYSV, &FRAME, &mut names);
1991
1992        // The block the split added is where the move went, and it is the whole of that block.
1993        let text = mir::print_func(&out, &names, &REGS);
1994        assert_eq!(out.block_count(), 4, "{text}");
1995        assert_eq!(text.matches("x64.ret\n").count(), 1, "{text}");
1996    }
1997
1998    #[test]
1999    fn a_call_passes_what_the_convention_says_and_takes_back_what_it_says() {
2000        let i32 = Type::int(32);
2001        let (mut names, mut source, block, args) = blank(&[i32, i32]);
2002        let sig =
2003            source.add_signature(Signature::new().with_params(&[i32, i32]).with_returns(&[i32]));
2004        let callee = names.intern("g");
2005        let call = Builder::new(&mut source, block).call(callee, sig, &[args[0], args[1]]);
2006        let got = source[call].first_result.expect("an integer comes back");
2007        Builder::new(&mut source, block).ret(&[got]);
2008
2009        // `int f(int a, int b) { return g(a, b); }`. The arguments arrived where the call wants
2010        // them, so what the call reads is what arrived, and the whole of the convention is in the
2011        // constraints rather than in a move.
2012        let text = lower(&mut names, &source);
2013        assert!(text.contains("= x64.call %0($rdi), %1($rsi), @g"), "{text}");
2014        assert!(text.contains("x64.ret_val_32 %2($rax)"), "{text}");
2015        // What the call writes is the value that comes back and then every register the callee is
2016        // free to destroy, in both classes, which is the whole of what stops the allocator from
2017        // leaving something in one of them.
2018        assert!(text.contains("%2:gpr($rax), $rcx, $rdx, $r8, $r9, $r10, $r11, $xmm0,"), "{text}");
2019        assert!(text.contains("$xmm15 = x64.call"), "{text}");
2020    }
2021
2022    #[test]
2023    fn what_the_frame_owes_a_call_comes_back_with_the_function() {
2024        let i32 = Type::int(32);
2025        let sig = |source: &mut Func| source.add_signature(Signature::new().with_params(&[i32]));
2026
2027        let (mut names, mut source, block, args) = blank(&[i32]);
2028        let sig = sig(&mut source);
2029        let callee = names.intern("g");
2030        Builder::new(&mut source, block).call(callee, sig, &[args[0]]);
2031        let out = func(&source, &mut names, &SYSV).expect("every instruction has a rule");
2032
2033        // Nothing on the stack, so nothing owed, but not a leaf either: a function that calls
2034        // owes the callee an aligned stack pointer and may not use the red zone.
2035        assert_eq!(out.stack.calls, Some(0));
2036        let layout = out.stack.layout(Layout::new(&SYSV, REGS));
2037        assert!(!layout.leaf);
2038        assert_eq!(layout.outgoing, 0);
2039
2040        // The same call under the other convention owes thirty two bytes for the callee to spill
2041        // its register arguments into, which is a fact about the convention and not about the call.
2042        let out = func(&source, &mut names, &x86_64::WIN64).expect("every instruction has a rule");
2043        assert_eq!(out.stack.calls, Some(32));
2044
2045        // And a function that calls nothing is a leaf, which is what says it may use the red zone.
2046        let (mut names, mut source, block, args) = blank(&[i32]);
2047        Builder::new(&mut source, block).ret(&[args[0]]);
2048        let out = func(&source, &mut names, &SYSV).expect("every instruction has a rule");
2049        assert_eq!(out.stack.calls, None);
2050        assert!(out.stack.layout(Layout::new(&SYSV, REGS)).leaf);
2051    }
2052
2053    #[test]
2054    fn a_value_that_outlives_a_call_is_not_left_where_the_call_destroys_it() {
2055        let i32 = Type::int(32);
2056        let (mut names, mut source, block, args) = blank(&[i32]);
2057        let sig = source.add_signature(Signature::new().with_params(&[i32]).with_returns(&[i32]));
2058        let callee = names.intern("g");
2059        let call = Builder::new(&mut source, block).call(callee, sig, &[args[0]]);
2060        let got = source[call].first_result.expect("an integer comes back");
2061        let mut build = Builder::new(&mut source, block);
2062        let sum = build.binary(Opcode::Add, got, args[0], Flags::default());
2063        build.ret(&[sum]);
2064
2065        // `int f(int a) { return g(a) + a; }`, which is the smallest program that asks the
2066        // question: `a` is read after the call and `rdi` is a register the call destroys.
2067        let lowered = func(&source, &mut names, &SYSV).expect("every instruction has a rule");
2068        let layout = lowered.stack.layout(Layout::new(&SYSV, REGS));
2069        let mut out = lowered.func;
2070        let env = env();
2071        let allocation = rucc_regalloc::run(&mut out, &env);
2072        let frame = Frame::of(&out, &allocation, &layout);
2073        finish(&mut out, &allocation, &frame, &Stack::default(), &SYSV, &FRAME, &mut names);
2074
2075        // It went to a register the callee has to put back, and the prologue and epilogue are what
2076        // put it back, which is the whole bargain the two halves of a convention make.
2077        let text = mir::print_func(&out, &names, &REGS);
2078        assert!(text.contains("$rbx"), "{text}");
2079        assert!(!text.contains('%'), "{text}");
2080        assert_eq!(text.matches("x64.call").count(), 1, "{text}");
2081    }
2082
2083    #[test]
2084    fn a_call_with_more_arguments_than_registers_writes_the_rest_into_the_outgoing_area() {
2085        let i64 = Type::int(64);
2086        let (mut names, mut source, block, args) = blank(&[i64]);
2087        let seven = vec![i64; 7];
2088        let sig = source.add_signature(Signature::new().with_params(&seven));
2089        let callee = names.intern("g");
2090        let passed = vec![args[0]; 7];
2091        Builder::new(&mut source, block).call(callee, sig, &passed);
2092
2093        let lowered = func(&source, &mut names, &SYSV).expect("the seventh goes to memory");
2094        // The bytes the call needs are on the layout the frame is worked out from, so that the
2095        // frame reserves as many as the widest call in the function asked for.
2096        assert_eq!(lowered.stack.calls, Some(8));
2097        let text = mir::print_func(&lowered.func, &names, &REGS);
2098        assert!(text.contains("x64.mov_mr_64 %0, [$rsp]\n"), "{text}");
2099    }
2100
2101    #[test]
2102    fn a_call_this_cannot_make_is_reported_rather_than_made() {
2103        let (mut names, mut source, block, _) = blank(&[]);
2104        let sig = source
2105            .add_signature(Signature::new().with_returns(&[Type::float(rucc_ir::Float::F80)]));
2106        let callee = names.intern("g");
2107        Builder::new(&mut source, block).call(callee, sig, &[]);
2108        let failed = func(&source, &mut names, &SYSV).expect_err("a long double is on the x87");
2109        assert_eq!(failed.to_string(), "what this call gives back is on the x87 stack");
2110    }
2111
2112    #[test]
2113    fn a_call_through_an_address_goes_through_the_register_the_address_is_in() {
2114        let i32 = Type::int(32);
2115        let (mut names, mut source, block, args) = blank(&[Type::PTR, i32]);
2116        let sig = source.add_signature(Signature::new().with_params(&[i32]).with_returns(&[i32]));
2117        let varargs = source.push_abis(&[]);
2118        let info = source.add_call(CallInfo { callee: None, signature: sig, varargs });
2119        let mut build = Builder::new(&mut source, block);
2120        let inst = InstData {
2121            args: build.func().push_values(&[args[0], args[1]]),
2122            extra: Extra::Call(info),
2123            ..InstData::new(Opcode::CallIndirect)
2124        };
2125        let called = build.inst(inst, &[i32]);
2126        let got = source[called].first_result.expect("an integer comes back");
2127        Builder::new(&mut source, block).ret(&[got]);
2128
2129        // `int f(int (*g)(int), int a) { return g(a); }`. The first operand is the address and
2130        // the arguments are the ones behind it, and everything else about the call is what a call
2131        // to a name would have been.
2132        let text = lower(&mut names, &source);
2133        assert!(text.contains("= x64.call_reg %0, %1($rdi)"), "{text}");
2134        assert!(text.contains("x64.ret_val_32 %2($rax)"), "{text}");
2135        assert!(!text.contains("@g"), "a call through an address names nobody: {text}");
2136    }
2137
2138    #[test]
2139    fn an_instruction_no_rule_covers_is_reported() {
2140        let (mut names, mut source, block, _) = blank(&[]);
2141        let mut build = Builder::new(&mut source, block);
2142        let order = Extra::Order(MemOrder::SeqCst);
2143        build.inst(InstData { extra: order, ..InstData::new(Opcode::Fence) }, &[]);
2144
2145        // A barrier on its own, which the rules do not write yet. Nothing about it is a width or
2146        // a register, so there is nothing for the message to add beyond the name.
2147        let failed = func(&source, &mut names, &SYSV).expect_err("no rule writes a barrier");
2148        assert_eq!(failed.to_string(), "no rule lowers a `fence`");
2149
2150        // A `fence` produces nothing, so there is no type in the message and nothing invents
2151        // one, and the instruction comes back so a caller can ask the function where it was.
2152        let inst = failed.inst().expect("the instruction it is about");
2153        assert_eq!(source[inst].opcode, Opcode::Fence);
2154    }
2155
2156    #[test]
2157    fn more_values_back_than_the_convention_has_registers_for_is_reported() {
2158        let i64 = Type::int(64);
2159        let (mut names, mut source, block, args) = blank(&[i64, i64, i64]);
2160        let mut build = Builder::new(&mut source, block);
2161        build.ret(&[args[0], args[1], args[2]]);
2162
2163        // Two integers come back in `rax` and `rdx` and a third has nowhere to go, which is not a
2164        // gap in the rules but the convention saying no. The front end classifies before it gets
2165        // here, so this is the shape that would mean the classification went wrong.
2166        let failed = func(&source, &mut names, &SYSV).expect_err("only two come back");
2167        assert_eq!(
2168            failed.to_string(),
2169            "what this function gives back takes more registers than this convention has for it"
2170        );
2171
2172        let inst = failed.inst().expect("the instruction it is about");
2173        assert_eq!(source[inst].opcode, Opcode::Return);
2174    }
2175
2176    /// A refusal about a signature has no instruction, which is what makes it the one arm apart.
2177    ///
2178    /// Everything else is about something written somewhere in the body and hands it back so a
2179    /// caller can ask the function where it came from. A parameter arrives before the first
2180    /// instruction runs, so there is nothing in the body to point at and the message is about
2181    /// the function.
2182    #[test]
2183    fn a_refusal_about_a_parameter_has_no_instruction_to_point_at() {
2184        let missing = Unsupported::Argument { index: 0, missing: Missing::OnX87 };
2185        assert_eq!(missing.inst(), None);
2186    }
2187
2188    /// An `alloca` of a fixed size, which is what every local whose address is taken becomes.
2189    fn slot(source: &mut Func, block: Block, size: u64, align: u32) -> Value {
2190        let info = MemInfo { size, align, ..plain() };
2191        let mut build = Builder::new(source, block);
2192        let mem = build.func().add_mem(info);
2193        build.value(InstData { extra: Extra::Mem(mem), ..InstData::new(Opcode::Alloca) }, Type::PTR)
2194    }
2195
2196    #[test]
2197    fn a_local_is_memory_in_the_frame_and_one_instruction_that_says_where() {
2198        let (mut names, mut source, block, _) = blank(&[]);
2199        let slot = slot(&mut source, block, 4, 4);
2200        let mut build = Builder::new(&mut source, block);
2201        let nine = build.iconst(Type::int(32), 9);
2202        build.store(nine, slot, plain(), Flags::default());
2203        let loaded = build.load(Type::int(32), slot, plain(), Flags::default());
2204        build.ret(&[loaded]);
2205
2206        let lowered = func(&source, &mut names, &SYSV).expect("every instruction has a rule");
2207
2208        // Four bytes on the list the frame is laid out from, and the one instruction that reads
2209        // where they went. Its displacement is nothing here because there is no frame yet, and
2210        // which instruction is waiting for which local is what `finish` is handed.
2211        assert_eq!(lowered.stack.locals, vec![Local { size: 4, align: 4 }]);
2212        assert_eq!(lowered.stack.addresses.len(), 1);
2213        assert_eq!(lowered.stack.addresses[0].1, 0);
2214        assert_eq!(
2215            mir::print_func(&lowered.func, &names, &REGS),
2216            "mfunc @f {\nblock0:\n    %0:gpr = x64.lea_64 [$rsp]\n    \
2217             %1:gpr = x64.mov_ri_32 9\n    x64.mov_mr_32 %1, [%0]\n    \
2218             %2:gpr = x64.mov_rm_32 [%0]\n    x64.ret_val_32 %2($rax)\n}\n"
2219        );
2220    }
2221
2222    #[test]
2223    fn the_frame_is_what_fills_the_address_of_a_local_in() {
2224        let (mut names, mut source, block, _) = blank(&[]);
2225        let slot = slot(&mut source, block, 4, 4);
2226        let mut build = Builder::new(&mut source, block);
2227        let nine = build.iconst(Type::int(32), 9);
2228        build.store(nine, slot, plain(), Flags::default());
2229        let loaded = build.load(Type::int(32), slot, plain(), Flags::default());
2230        build.ret(&[loaded]);
2231
2232        let lowered = func(&source, &mut names, &SYSV).expect("every instruction has a rule");
2233        let stack = lowered.stack;
2234        let mut out = lowered.func;
2235        let env = env();
2236        let allocation = rucc_regalloc::run(&mut out, &env);
2237        let layout = stack.layout(Layout::new(&SYSV, REGS));
2238        let frame = Frame::of(&out, &allocation, &layout);
2239        finish(&mut out, &allocation, &frame, &stack, &SYSV, &FRAME, &mut names);
2240
2241        // `int f(void) { int x; x = 9; return x; }` with the address of `x` taken, end to end.
2242        // A leaf small enough to live in the red zone takes no frame at all, so the stack pointer
2243        // never moves and the four bytes are below it, which is what the negative offset is. The
2244        // instruction the lowering left with nothing in its displacement now has the answer in it.
2245        let text = mir::print_func(&out, &names, &REGS);
2246        assert!(text.contains("$rax = x64.lea_64 [$rsp - 8]"), "{text}");
2247        assert!(!text.contains("x64.sub_ri_64"), "{text}");
2248        assert_eq!(frame.size(), 0);
2249        assert_eq!(frame.local(0), Some(-8));
2250    }
2251
2252    #[test]
2253    fn a_stack_slot_whose_size_is_not_known_until_it_runs_is_reported() {
2254        let i64 = Type::int(64);
2255        let (mut names, mut source, block, args) = blank(&[i64]);
2256        let info = MemInfo { size: 0, align: 16, ..plain() };
2257        let mut build = Builder::new(&mut source, block);
2258        let mem = build.func().add_mem(info);
2259        let size = build.func().push_values(&[args[0]]);
2260        let slot = build.value(
2261            InstData { args: size, extra: Extra::Mem(mem), ..InstData::new(Opcode::Alloca) },
2262            Type::PTR,
2263        );
2264        Builder::new(&mut source, block).ret(&[slot]);
2265
2266        // A variable length array. Growing the stack where the declaration stands means moving the
2267        // stack pointer in the middle of the function and reaching everything else through a
2268        // frame pointer afterwards, and the frame here lays out neither.
2269        let failed = func(&source, &mut names, &SYSV).expect_err("nothing grows the stack");
2270        assert_eq!(failed.to_string(), "nothing here grows the stack for a variable length array");
2271    }
2272
2273    #[test]
2274    fn an_address_is_read_written_and_added_to_like_the_integer_it_is() {
2275        let (mut names, mut source, block, args) = blank(&[Type::PTR, Type::int(64)]);
2276        let mut build = Builder::new(&mut source, block);
2277        let stepped = build.func().push_values(&[args[0], args[1]]);
2278        let next =
2279            build.value(InstData { args: stepped, ..InstData::new(Opcode::PtrAdd) }, Type::PTR);
2280        let loaded = build.load(Type::int(32), next, plain(), Flags::default());
2281        build.ret(&[loaded]);
2282
2283        // `int f(int *p, long i) { return *(int *)((char *)p + i); }`. Nothing about this is new
2284        // in the rule set, which is the point: the two addresses arrive in registers because an
2285        // address is an integer as wide as one, and the arithmetic on them is the add it always
2286        // was, so every rule written about an add reaches it.
2287        //
2288        // The add stays its own instruction rather than folding into the address the load reads
2289        // from. Two registers with no scale on either is the one addressing mode the rules have no
2290        // load through, because the folds that exist are the displacement one and the scaled ones,
2291        // and this is neither. That is a peephole worth having and not a thing this changes.
2292        assert_eq!(
2293            lower(&mut names, &source),
2294            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
2295             %1:gpr($rsi) = x64.arg_val_64\n    %2:gpr(reuse 1) = x64.add_rr_64 %0, %1\n    \
2296             %3:gpr = x64.mov_rm_32 [%2]\n    x64.ret_val_32 %3($rax)\n}\n"
2297        );
2298    }
2299
2300    /// The address of a file scope name, which is what every use of a global and every string
2301    /// literal starts from.
2302    fn address_of(source: &mut Func, block: Block, names: &mut Interner, name: &str) -> Value {
2303        let symbol = names.intern(name);
2304        let mut build = Builder::new(source, block);
2305        build.value(
2306            InstData { extra: Extra::Symbol(symbol), ..InstData::new(Opcode::GlobalAddr) },
2307            Type::PTR,
2308        )
2309    }
2310
2311    #[test]
2312    fn the_address_of_a_name_is_one_instruction_carrying_the_name() {
2313        let (mut names, mut source, block, _) = blank(&[]);
2314        let counter = address_of(&mut source, block, &mut names, "counter");
2315        let mut build = Builder::new(&mut source, block);
2316        let loaded = build.load(Type::int(32), counter, plain(), Flags::default());
2317        build.ret(&[loaded]);
2318
2319        // `extern int counter; int f(void) { return counter; }`. The address is an addressing mode
2320        // that names no register and carries the symbol, which is what the assembler writes
2321        // relative to `%rip` and what the object writer leaves a relocation for.
2322        assert_eq!(
2323            lower(&mut names, &source),
2324            "mfunc @f {\nblock0:\n    %0:gpr = x64.lea_64 [@counter]\n    \
2325             %1:gpr = x64.mov_rm_32 [%0]\n    x64.ret_val_32 %1($rax)\n}\n"
2326        );
2327    }
2328
2329    /// A cast between a pointer and an integer, at whatever width the result is asked for.
2330    fn cast(source: &mut Func, block: Block, opcode: Opcode, from: Value, to: Type) -> Value {
2331        let mut build = Builder::new(source, block);
2332        let args = build.func().push_values(&[from]);
2333        build.value(InstData { args, ..InstData::new(opcode) }, to)
2334    }
2335
2336    #[test]
2337    fn a_cast_between_a_pointer_and_an_integer_as_wide_is_no_instruction_at_all() {
2338        let (mut names, mut source, block, args) = blank(&[Type::PTR]);
2339        let number = cast(&mut source, block, Opcode::PtrToInt, args[0], Type::int(64));
2340        Builder::new(&mut source, block).ret(&[number]);
2341
2342        // `long f(void *p) { return (long)p; }`. An address on this machine is an integer as wide
2343        // as the machine addresses, so the cast changes what the type system calls the value and
2344        // changes nothing about the value, and the register holding it is the one that held it.
2345        assert_eq!(
2346            lower(&mut names, &source),
2347            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n    \
2348             x64.ret_val_64 %0($rax)\n}\n"
2349        );
2350    }
2351
2352    #[test]
2353    fn a_null_pointer_is_a_constant_that_reaches_a_register_before_anything_reads_it() {
2354        let (mut names, mut source, block, _) = blank(&[]);
2355        let mut build = Builder::new(&mut source, block);
2356        let zero = build.iconst(Type::int(64), 0);
2357        let null = cast(&mut source, block, Opcode::IntToPtr, zero, Type::PTR);
2358        Builder::new(&mut source, block).ret(&[null]);
2359
2360        // `void *f(void) { return 0; }`. The cast is nothing, and reading its operand is what
2361        // writes the zero down: a constant is materialized where it is wanted rather than where
2362        // the IR defined it, and without the read there would be no instruction at all.
2363        assert_eq!(
2364            lower(&mut names, &source),
2365            "mfunc @f {\nblock0:\n    %0:gpr = x64.mov_ri_64 0\n    x64.ret_val_64 %0($rax)\n}\n"
2366        );
2367    }
2368
2369    #[test]
2370    fn a_cast_between_a_pointer_and_a_narrower_integer_is_reported() {
2371        let (mut names, mut source, block, args) = blank(&[Type::PTR]);
2372        let number = cast(&mut source, block, Opcode::PtrToInt, args[0], Type::int(32));
2373        Builder::new(&mut source, block).ret(&[number]);
2374
2375        // The front end never writes one: it casts at the address width and truncates or extends
2376        // around it, so both of those are the rules they always were. IR from somewhere else that
2377        // does write one is refused rather than compiled to a move that keeps the high half.
2378        let failed = func(&source, &mut names, &SYSV).expect_err("no rule narrows an address");
2379        assert_eq!(failed.to_string(), "no rule lowers a `ptrtoint` producing a `i32`");
2380    }
2381}