Skip to main content

rucc_codegen/
finish.rs

1//! The prologue, the epilogue, and the moves the allocator asked for.
2//!
3//! Design: `spec/10-backend.md` sections 10.4 and 10.7.
4//!
5//! [`crate::frame`] works out what a function's stack looks like and writes nothing. This is what
6//! writes it. Three things are still missing from a function the allocator has finished with, and
7//! all three of them are instructions no lowering rule chose:
8//!
9//! ```text
10//!   the prologue     takes the frame the layout worked out, and puts away the registers a call
11//!                    leaves alone that this function writes anyway
12//!   the moves        every spill, every reload and every copy the allocator handed back as an
13//!                    edit, in the place it said and in the order it said
14//!   the epilogue     gives the frame back and puts the registers back, at the end of every block
15//!                    the function returns from
16//! ```
17//!
18//! There is a fourth thing and it is not an instruction but a number. The lowering wrote an
19//! instruction for every `alloca` that computes the address of the memory it asked for, and could
20//! not write how far into the frame that memory is, because when it ran there was no frame. So
21//! the displacement of each of those is filled in here, out of the same [`Frame`] everything else
22//! here reads, and off the same stack pointer every other offset in it is from.
23//!
24//! There is a fifth thing on a command line that asked for the stack to be touched a page at a
25//! time, and it is the only one of them that is written into the middle of a block rather than at
26//! one end of the function. A variable length array moves the stack pointer by a number that is not
27//! known until the declaration runs, so walking it a page at a time is a loop written around the one
28//! instruction the lowering left, and that turns the block the declaration was in into four.
29//!
30//! The loads that read the arguments the caller passed on the stack are waiting on the same number
31//! and on one more. Those bytes are the caller's rather than this function's, and a frame that had
32//! to force its own alignment cannot say how far away the caller's stack pointer was, so it reaches
33//! back through the frame pointer instead. Which register a load reads through is therefore settled
34//! here too, and it is the only base register in a finished function that was not settled by
35//! whoever wrote the instruction.
36//!
37//! After this the function is one an encoder can read: every register is physical, every offset
38//! into the frame is a constant, and the stack pointer is where the convention says it should be
39//! at every instruction that could look.
40//!
41//! # Why the moves go in first
42//!
43//! Every offset the frame reports is from the stack pointer as it stands in the body of the
44//! function. A spill written before the prologue exists would be written in front of the
45//! instruction it belongs to and behind nothing, which is where the prologue then goes, so the
46//! prologue ends up in front of it and the offsets stay true. Writing them the other way round
47//! would put the first reload above the instruction that takes the frame, and it would read from
48//! an address that is one frame out.
49//!
50//! # Where a return is
51//!
52//! A block that goes nowhere is a block the function leaves from. Mostly that is a return, and
53//! the other kind is a block ending in `unreachable`, which is a point the front end says control
54//! does not arrive at and which the lowering writes no instruction for. Both want the same thing
55//! here. A return wants the epilogue because that is what a return is once the frame is known,
56//! and an unreachable block wants it because the alternative is a function whose last instruction
57//! falls into whatever the assembler put after it, which is worse than an epilogue nothing runs.
58//! So the epilogue goes at the end of every block with an empty successor list, and there may be
59//! several, because nothing here insists a function has one exit.
60//!
61//! # What is target-specific here
62//!
63//! The names, and only the names. Which instruction pushes a register and which one moves the
64//! stack pointer is [`rucc_target::FrameInsts`], which the target says and this reads, so what
65//! is written below is the shape of a prologue rather than any particular machine's. That is
66//! `spec/10-backend.md` section 10.8 as it applies to the one pass that would otherwise be full
67//! of `x64.` by hand.
68
69use std::collections::HashMap;
70
71use rucc_base::Interner;
72use rucc_diag::Span;
73use rucc_mir::{Block, BlockCall, CfiOp, Func, Inst, Mem, Opcode, Operand, Patch, Reg};
74use rucc_regalloc::Allocation;
75use rucc_regalloc::assign::Place;
76use rucc_regalloc::rewrite::{At, Edit};
77use rucc_target::{BranchInsts, CallRegs, Chkstk, FrameInsts, Guard, PhysReg, Probe, RegClass};
78
79use crate::frame::Frame;
80use crate::lower::Stack;
81
82/// What the stack protector's check needs beyond the frame, in a function that has one.
83///
84/// Three things that come from three places, which is why they arrive together rather than being
85/// looked up here. Where the word the canary is copied from lives is a fact about the runtime the
86/// code is linked against. What a branch on a register is is a fact about the machine. And the two
87/// registers are neither: they are the ones the allocator was told to hold back, which is a
88/// decision about the allocator, and they are free at a return for exactly that reason.
89#[derive(Debug, Clone, Copy)]
90pub struct Protect<'a> {
91    /// Where the word the canary is a copy of lives, and what to call when the copy has changed.
92    pub guard: &'a Guard,
93    /// What a branch on a register is, which is what the check ends its block with.
94    pub branch: &'a BranchInsts,
95    /// The two registers the check may use, which are two the allocator never handed out.
96    pub scratch: [PhysReg; 2],
97}
98
99/// What a function that takes its stack a page at a time needs beyond the frame.
100///
101/// What `-fstack-clash-protection` asks for, and the same three kinds of thing [`Protect`] is:
102/// one fact about the platform, one about the machine, and two registers that are neither. See
103/// [`rucc_target::Probe`] for what the sequence is defending against.
104///
105/// Read in two places, because a function has two ways of moving its stack pointer and the flag is
106/// about both of them. The prologue takes the frame the layout worked out, and a variable length
107/// array takes however many bytes its declaration asked for while the function runs. The same three
108/// things answer both.
109#[derive(Debug, Clone, Copy)]
110pub struct Probing<'a> {
111    /// What touches a page and how far apart the pages are.
112    pub probe: &'a Probe,
113    /// What a branch on a register is, which is what the loop under a large frame ends with.
114    pub branch: &'a BranchInsts,
115    /// The two registers the sequence may use, which are two the allocator never handed out.
116    pub scratch: [PhysReg; 2],
117}
118
119/// What a profiler's hook at the top of a function is, in a function that has one.
120///
121/// What `-pg` asks for. See [`rucc_target::Trace`] for why there are two of these and what each of
122/// them lets the hook see. Only the name survives to here, because by this point the flag has been
123/// read against the target and a prologue that has the name has everything it needs.
124#[derive(Debug, Clone, Copy)]
125pub struct Tracing {
126    /// What is called, which is a routine the runtime provides and not one the program wrote.
127    pub name: &'static str,
128    /// Whether the call goes in front of the prologue rather than once the frame is taken.
129    pub early: bool,
130}
131
132/// The room at the top of a function for something to be written over later, in a function that
133/// was promised any.
134///
135/// What `-fpatchable-function-entry=` asks for. The room is a run of the shortest instruction the
136/// machine has that does nothing, and what makes it worth reserving is that it is never run for
137/// long: a tracer or a live patcher writes a jump or a call over it once the program is up, and
138/// what it needs from the compiler is a known address and a known number of bytes.
139///
140/// Two counts because the room can be on either side of the function's own label. Only the half
141/// after it is written here, since the stream starts at the label and there is nowhere in it to put
142/// the other half; the half in front is carried through so that whatever lays the function down can
143/// lay that many bytes ahead of the symbol.
144#[derive(Debug, Clone, Copy)]
145pub struct Padding {
146    /// What the instruction that does nothing is called on this target.
147    pub name: &'static str,
148    /// How many of them go in front of the function's own label.
149    pub before: u32,
150    /// How many go after it.
151    pub after: u32,
152}
153
154/// What the convention this function is compiled for says a frame is.
155///
156/// Seven answers to the one question, which is why they travel together: where it puts things,
157/// which instructions build one, whether this function's carries a protector, whether it is taken a
158/// page at a time, whether the function opens with a landing pad, whether it calls a profiler on
159/// the way in, and how much room it opens with for a patcher. The last five are the only ones about
160/// this function rather than about every function on the target, and they are here because what
161/// they need is the other two and nothing else.
162#[derive(Debug, Clone, Copy)]
163pub struct Convention<'a> {
164    /// Where the convention puts things.
165    pub regs: &'a CallRegs,
166    /// The instructions a prologue, an epilogue, a spill and a reload are made of on it.
167    pub insts: &'a FrameInsts,
168    /// What this function's stack protector needs, or `None` in a function with none.
169    pub protect: Option<Protect<'a>>,
170    /// What this function's probing prologue needs, or `None` when the frame is taken in one
171    /// subtraction, which is what a command line that did not ask asks for.
172    pub probe: Option<Probing<'a>>,
173    /// What says an indirect branch may arrive at the top of this function, or `None` when the
174    /// command line did not ask for one and on a target that has no such instruction.
175    ///
176    /// See [`rucc_target::FrameInsts::landing`]. A name rather than a flag because the flag has
177    /// already been read against the target by the time this is built, and because a prologue that
178    /// has the name has everything it needs.
179    pub landing: Option<&'static str>,
180    /// What this function's call to a profiler is, or `None` in one that makes none, which is every
181    /// function on a command line that did not ask.
182    pub trace: Option<Tracing>,
183    /// What room this function opens with for a patcher, or `None` in one that was promised none,
184    /// which is every function on a command line that did not ask.
185    pub pad: Option<Padding>,
186}
187
188impl<'a> Convention<'a> {
189    /// That convention, for a function with no stack protector, no probing, no landing pad, no
190    /// call to a profiler and no room for a patcher, which is most of them.
191    #[must_use]
192    pub fn new(regs: &'a CallRegs, insts: &'a FrameInsts) -> Self {
193        Self { regs, insts, protect: None, probe: None, landing: None, trace: None, pad: None }
194    }
195}
196
197/// Which instruction each of the allocator's moves became.
198///
199/// A spill and a copy are both a `mov` once they are written, and so is an instruction the lowering
200/// wrote that happens to move the same register to the same address. Telling them apart afterwards
201/// by looking at them is guesswork, and a pass that guesses wrong about a store to a volatile
202/// variable deletes a read the program insisted on. So what the allocator asked for is recorded as
203/// it is written, and a later pass that is only allowed to touch the allocator's own moves has the
204/// list rather than a heuristic. See [`crate::copies`], which is the one pass that reads this.
205#[derive(Debug, Default)]
206pub struct Moves(HashMap<Inst, Edit>);
207
208impl Moves {
209    /// What the allocator asked for at this instruction, or `None` at an instruction that is not
210    /// one of its moves.
211    #[must_use]
212    pub fn at(&self, inst: Inst) -> Option<Edit> {
213        self.0.get(&inst).copied()
214    }
215
216    /// Records that this instruction is what that move came to.
217    pub fn record(&mut self, inst: Inst, edit: Edit) {
218        self.0.insert(inst, edit);
219    }
220}
221
222/// Writes the moves, the prologue and the epilogue into a function the allocator has finished
223/// with.
224///
225/// Hands back which instruction each of the allocator's moves became, for the one pass that is
226/// allowed to take one of them out again.
227///
228/// # Panics
229///
230/// Panics on a function with no blocks in it, on a frame whose slots or locals the allocation and
231/// the lowering do not match, and on a move of a class the target did not say how to move. All of
232/// them are the caller handing it a frame and a function that were not worked out from each other.
233pub fn finish(
234    func: &mut Func,
235    allocation: &Allocation,
236    frame: &Frame,
237    stack: &Stack,
238    convention: Convention<'_>,
239    names: &mut Interner,
240) -> Moves {
241    let Convention { regs: conv, insts, protect, probe, landing, trace, pad } = convention;
242    let entry = func.entry().expect("a function with a block in it");
243
244    // Before anything is written, because these are instructions the lowering already put in the
245    // function and every one of them is somewhere the prologue is about to go in front of, which
246    // is what makes an offset from the stack pointer the right thing to write into them. In a
247    // frame that grows it is an offset from the frame pointer instead, so the base register is
248    // rewritten the way an incoming argument's is, and for a version of the same reason.
249    //
250    // Added rather than assigned. The instruction named here is the `lea` the lowering wrote, or
251    // whatever [`crate::fold`] folded that `lea` into, and a reader that took it brought a
252    // displacement of its own: the address of a local is where the object starts and reading a
253    // field of it is some way past that. Assigning would throw the field offset away and read the
254    // front of the object every time.
255    for &(inst, local) in &stack.addresses {
256        let at = frame.local(local).expect("a local the frame was worked out from");
257        let mem = func[inst].mem.expect("the address of a local is an address");
258        func[mem].disp += at;
259        if frame.grows() {
260            rebase(func, inst, conv.frame_pointer);
261        }
262    }
263
264    // The bytes a variable length array takes are already off the stack pointer by the time one of
265    // these runs, so what is left to write is how far above the new stack pointer the array starts,
266    // which is however much of the bottom of the frame belongs to the arguments of a call. That
267    // area stays at the bottom wherever the bottom has moved to. Added rather than assigned for the
268    // reason the loop above is: one of these folds into its readers like any other address, and a
269    // reader that took it brought a displacement of its own.
270    for &inst in &stack.dynamic {
271        let mem = func[inst].mem.expect("the address of a growable local is an address");
272        func[mem].disp += offset(frame.below());
273    }
274
275    // The same, one area further up, and through the frame pointer when that is what reaches it.
276    // These are in the entry block ahead of everything, so the prologue still goes in front of
277    // them, which is what makes both registers hold what these offsets are counted from.
278    let incoming = frame.incoming();
279    for &(inst, up) in &stack.arguments {
280        let mem = func[inst].mem.expect("an argument read out of memory is read from an address");
281        func[mem].disp += incoming.at + offset(up);
282        if incoming.through_frame_pointer {
283            rebase(func, inst, conv.frame_pointer);
284        }
285    }
286
287    // Every offset the frame reports is from this one register, which is the stack pointer in an
288    // ordinary frame and the frame pointer in one that moves the stack pointer while it runs.
289    let base = if frame.grows() { conv.frame_pointer } else { conv.stack_pointer };
290    let mut writer = Writer { func, conv, insts, names, base, ahead: None };
291
292    let mut cursors: HashMap<At, Inst> = HashMap::new();
293    let mut moves = Moves::default();
294    for edit in &allocation.edits {
295        let inst = writer.mov(edit, frame);
296        writer.put(&mut cursors, edit.at, inst);
297        moves.record(inst, *edit);
298    }
299
300    // Before the epilogues, because this is what turns one block into four and the last of the four
301    // is the one the function goes on to return from. A block that went nowhere before a variable
302    // length array was walked in the middle of it is not the block that goes nowhere afterwards, and
303    // an epilogue written into the wrong one of them gives the frame back before the body has run.
304    if let Some(probing) = probe {
305        for &took in &stack.grown {
306            writer.walk(took, probing);
307        }
308    }
309
310    // Almost nothing of either in a naked function, which is the whole of what the attribute asks
311    // for and the only place in this file that knows the word. The frame is empty by the time one
312    // gets here, since [`crate::pipeline`] refuses a naked function that wanted bytes, so what is
313    // left to leave out is the part of the prologue that is written on somebody's instruction
314    // rather than on the frame's: the room a patcher was promised and the profiler's hook, both of
315    // which are a call or a run of bytes in front of a body that said it is the whole of the
316    // function. The protector and the probe are already off, the first because the pipeline turned
317    // it off and the second because it only fires on a frame there is none of.
318    //
319    // The landing pad stays. It is not the compiler adding something to the function, it is the
320    // address of the function being made one an indirect branch may arrive at, and gcc writes it
321    // in a naked function too.
322    //
323    // The epilogue is the one that matters. It is what writes the `ret`, and a naked function ends
324    // where its own text ends, which for micropython's `nlr_push` is a jump somewhere else and for
325    // everything else is the `ud2` the lowering put there.
326    let bare = frame.naked();
327    let prologue = writer.prologue(
328        frame,
329        protect,
330        probe.filter(|_| !bare),
331        landing,
332        trace.filter(|_| !bare),
333        pad.filter(|_| !bare),
334    );
335    for &inst in prologue.iter().rev() {
336        writer.func.prepend_inst(entry, inst);
337    }
338    let returns: Vec<Block> = if bare {
339        Vec::new()
340    } else {
341        writer.func.blocks().filter(|&block| writer.func[block].succs.is_empty()).collect()
342    };
343    for block in returns {
344        // The check goes in front of the epilogue and takes the return with it. What is left in
345        // the block the function used to return from is the check, and the block the epilogue then
346        // goes in is the arm the canary was unchanged on.
347        let block = match protect {
348            Some(protect) => writer.check(block, frame, protect),
349            None => block,
350        };
351        let epilogue = writer.epilogue(frame);
352        for inst in epilogue {
353            writer.func.append_inst(block, inst);
354        }
355    }
356
357    // Last of everything, because the blocks a probing prologue made have to come in front of the
358    // block the function used to begin with and the ones the protector's check makes are made
359    // after that. Nothing has been laid out yet: `crate::layout` runs after this and puts every
360    // block in its own order, and all this decides is which block the function is entered at.
361    if let Some(ahead) = writer.ahead {
362        let rest: Vec<Block> =
363            writer.func.blocks().filter(|block| !ahead.contains(block)).collect();
364        let order: Vec<Block> = ahead.into_iter().chain(rest).collect();
365        writer.func.set_block_order(&order);
366    }
367    moves
368}
369
370/// How many pages a probing prologue touches one after another before it writes a loop instead.
371///
372/// Three, which is what gcc unrolls to. The loop is four instructions however many pages it walks
373/// and a page written out is two, so three is the last size at which the straight line is no
374/// longer than the loop, and the straight line has no branch in it and needs no register.
375const UNROLLED: u32 = 3;
376
377/// One function having its frame written into it.
378/// Points an address the lowering left counted from the stack pointer at another register.
379///
380/// The base register is an operand of the instruction and the addressing mode holds where in the
381/// operand vector it is, so the register is changed there and not in the mode.
382fn rebase(func: &mut Func, inst: Inst, to: PhysReg) {
383    let mem = func[inst].mem.expect("an address");
384    let at = func[mem].base.expect("an address the lowering wrote a base register into");
385    let operands = func[inst].operands;
386    func[operands][usize::from(at)].reg = Reg::physical(to);
387}
388
389struct Writer<'a> {
390    func: &'a mut Func,
391    conv: &'a CallRegs,
392    insts: &'a FrameInsts,
393    names: &'a mut Interner,
394    /// Which register every offset into the frame is counted from, which is the stack pointer
395    /// unless the function moves it while it runs. See `Growing` in [`crate::frame`].
396    base: PhysReg,
397    /// The blocks a probing prologue made, which go in front of the one the function began with.
398    ///
399    /// Empty in every function whose frame is taken in one subtraction, which is every function
400    /// on a command line that did not ask for the stack to be touched a page at a time and most
401    /// of them on one that did. See [`Writer::pages`].
402    ahead: Option<[Block; 2]>,
403}
404
405impl Writer<'_> {
406    /// The instructions the prologue is, in the order they run.
407    ///
408    /// The order is the one the epilogue undoes and it is not free. The frame pointer is saved
409    /// before anything else, so that it points at a fixed place whatever else happens. The
410    /// registers are pushed before the alignment is forced, so that the epilogue can find them
411    /// again from the frame pointer, since after the alignment is forced nothing else can. And the
412    /// vector registers are stored last, because until the frame has been taken there is nowhere
413    /// to store them.
414    ///
415    /// Where the pointer is pointed at the frame is the one part of that order the platform gets a
416    /// say in. Windows wants it after the frame has been taken rather than before, because the
417    /// unwind record it reads has no way to describe the other order, so on that target the move
418    /// goes between the frame and the vector stores instead. See `Late` in [`crate::frame`].
419    ///
420    /// The landing pad is in front of all of it, because the address it makes reachable is the
421    /// address of the function and the address of the function is where the first instruction is.
422    /// It has to be written here rather than after the fact, since a probing prologue moves the
423    /// instructions written so far into a block of its own and the pad has to move with them.
424    ///
425    /// The room a patcher was promised goes after the pad, because a patcher wants somewhere it can
426    /// write a call that happens before anything else, and the pad is the one instruction that has
427    /// to come first for a reason of its own.
428    ///
429    /// A profiler's hook goes next, or at the end when it is the kind that reads the frame pointer.
430    /// The early one is in front of everything the frame does for a reason of its own: what makes
431    /// it worth replacing while the program runs is that the stack at that instruction is exactly
432    /// what a call leaves, and a prologue that had already run would have changed it.
433    fn prologue(
434        &mut self,
435        frame: &Frame,
436        protect: Option<Protect<'_>>,
437        probe: Option<Probing<'_>>,
438        landing: Option<&'static str>,
439        trace: Option<Tracing>,
440        pad: Option<Padding>,
441    ) -> Vec<Inst> {
442        let sp = self.conv.stack_pointer;
443        let fp = self.conv.frame_pointer;
444        let int = self.conv.int_class;
445        let sse = self.conv.sse_class;
446        let word = offset(self.conv.word);
447        let mut out = Vec::new();
448        // What the prologue wrote before it had described anything, which is what decides whether
449        // there is a rule to remember at the end of it. Neither of these moves a register or takes
450        // a frame, so a function whose whole prologue is one of them has no rows and must not be
451        // given a pair of them that cancel out.
452        let mut quiet = Vec::new();
453        if let Some(name) = landing {
454            let opcode = self.opcode(name);
455            let inst = self.func.build_loose(opcode).finish();
456            out.push(inst);
457            quiet.push(inst);
458        }
459        // After the pad and in front of everything else, which is where gcc puts it. The pad is the
460        // function's first instruction because the address an indirect branch may arrive at is the
461        // address of the function, and the room comes next because what gets written over it is a
462        // call and the point of that call is that it happens before the function has done anything.
463        //
464        // Nothing is described for any of it. A byte that does nothing does not move the stack
465        // pointer, and what a patcher writes over it later is its own problem rather than this
466        // function's: the rules here say what this function did, and it did nothing.
467        if let Some(pad) = pad {
468            let opcode = self.opcode(pad.name);
469            let mut first = None;
470            for _ in 0..pad.after {
471                let inst = self.func.build_loose(opcode).finish();
472                out.push(inst);
473                quiet.push(inst);
474                first.get_or_insert(inst);
475            }
476            self.func.patch = Some(Patch { before: pad.before, pad: opcode, after: first });
477        }
478        // Nothing is described for it and nothing needs to be: the call pushes a return address and
479        // the hook pops it, so the frame is the same on both sides, and the hook preserves every
480        // register because it is written in assembly for exactly this. That is also why the
481        // allocator, which ran before any of this, never saw the call and did not have to.
482        if let Some(trace) = trace.filter(|trace| trace.early) {
483            let inst = self.hook(trace);
484            out.push(inst);
485            quiet.push(inst);
486        }
487        // How far the stack pointer is below the canonical frame address, and whether the address
488        // is still counted from the stack pointer at all. It starts at the return address the
489        // call itself pushed, which is the rule the CIE already states, so the first row here is
490        // the first thing this function does on top of that.
491        let mut below = offset(self.conv.return_address);
492        let mut from_sp = true;
493        if frame.frame_pointer() {
494            let inst = self.push(fp);
495            out.push(inst);
496            below += word;
497            self.row(inst, CfiOp::DefCfaOffset(below));
498            self.saved(inst, int, fp, -below);
499            // Straight away unless the platform wants it after the frame, where the same two
500            // instructions go at the bottom of this function instead. See `Late` in
501            // [`crate::frame`].
502            if !frame.late() {
503                let mov = self.opcode(self.insts.moves(int).expect("a move").mov);
504                let inst = self.two(mov, fp, sp);
505                out.push(inst);
506                let number = self.dwarf(int, fp);
507                self.row(inst, CfiOp::DefCfaRegister(number));
508                from_sp = false;
509            }
510        }
511        for &reg in frame.saved_int() {
512            let inst = self.push(reg);
513            out.push(inst);
514            below += word;
515            if from_sp {
516                self.row(inst, CfiOp::DefCfaOffset(below));
517            }
518            self.saved(inst, int, reg, -below);
519        }
520        if let Some(to) = frame.realign() {
521            // Nothing is written for this and nothing can be. After it the stack pointer is a
522            // rounded-down version of where it was rather than a fixed distance from it, which is
523            // exactly what a rule cannot say. It is also why a frame that realigns is a frame
524            // with a frame pointer: by here the address is already counted from that instead.
525            assert!(!from_sp, "a frame that forces its own alignment has a frame pointer");
526            let and = self.opcode(self.insts.align);
527            out.push(self.arith(and, -i64::from(to)));
528        }
529        if frame.size() > 0 {
530            self.take(&mut out, frame.size(), &mut below, from_sp, probe);
531        }
532        // The other half of the pair above, for the platform whose record counts everything from
533        // where the stack pointer ends the prologue. Here it is that register the pointer is a copy
534        // of, so what the row says is the whole distance rather than that nothing has changed, and
535        // the frame is described before it rather than after, which is the whole of what the record
536        // could not say about the early order.
537        if frame.late() && frame.frame_pointer() {
538            let mov = self.opcode(self.insts.moves(int).expect("a move").mov);
539            let inst = self.two(mov, fp, sp);
540            out.push(inst);
541            let number = self.dwarf(int, fp);
542            self.row(inst, CfiOp::DefCfa { reg: number, offset: below });
543        }
544        for save in frame.saved_sse() {
545            let inst = self.store(sse, save.reg, save.at);
546            out.push(inst);
547            // Where it went is an offset from whichever register the frame counts from, and the
548            // address is a constant above that register, so the two make one constant. In an
549            // ordinary frame that register is the stack pointer and the constant is `below`. In one
550            // that grows it is the frame pointer, which the address has been counted from since the
551            // prologue pointed it at where it saved the caller's copy, so the constant is the two
552            // words above it and nothing the prologue did afterwards changes it. Unless the pointer
553            // went up late, where it holds what the stack pointer holds and the constant is `below`
554            // again, which is why the question is about both. A realigned frame has no such constant
555            // at all and the rule is left out rather than guessed. On SysV that costs nothing, since
556            // it preserves no vector register for a prologue to save. On Windows it would be a save
557            // with no row, and what stops that reaching an object file is that a realigned frame
558            // there keeps the early order and is refused whole by the unwind writer, which is
559            // `tamnd/rucc#1422`.
560            if frame.realign().is_none() {
561                let above = if frame.grows() && !frame.late() {
562                    word + offset(self.conv.return_address)
563                } else {
564                    below
565                };
566                self.saved(inst, sse, save.reg, save.at - above);
567            }
568        }
569        // Before the canary and after the frame, which is where gcc puts it. The hook reads the
570        // frame pointer to find out who called this function, so it has to run once there is one,
571        // and it is a call, so it has to run before anything the function is keeping in the frame
572        // could be read back.
573        if let Some(trace) = trace.filter(|trace| !trace.early) {
574            let inst = self.hook(trace);
575            out.push(inst);
576        }
577        // Last of everything, because it writes into the frame and there is no frame to write into
578        // until the stack pointer has moved. Nothing is described for either instruction: they
579        // write a slot rather than save a register, and no unwinder wants to put a canary back.
580        if let Some(protect) = protect {
581            let at = frame.canary().expect("a protected function has a slot for its canary");
582            let [into, _] = protect.scratch;
583            out.push(self.read_guard(into, protect.guard));
584            out.push(self.store(self.conv.int_class, into, at));
585        }
586        // The rules the body runs under, kept so that each epilogue can put them back rather than
587        // leaving the next block reading whatever the last one ended on. See `epilogue`.
588        //
589        // Nothing is kept in a function whose whole prologue is the pieces that describe nothing.
590        // See `quiet` above.
591        if let Some(&last) = out.last() {
592            if !quiet.contains(&last) {
593                self.row(last, CfiOp::RememberState);
594            }
595        }
596        out
597    }
598
599    /// The call to a profiler's hook.
600    ///
601    /// No arguments and no result. Which function is being entered is not passed, because the hook
602    /// reads its own return address to find out, and that is the whole reason the call is written
603    /// rather than something cheaper.
604    fn hook(&mut self, trace: Tracing) -> Inst {
605        let call = self.opcode(self.insts.call);
606        let symbol = self.names.intern(trace.name);
607        self.func.build_loose(call).symbol(symbol).finish()
608    }
609
610    /// Takes the frame, which is one subtraction unless the command line asked for the stack to be
611    /// touched a page at a time.
612    ///
613    /// `below` is how far the canonical frame address is above the stack pointer, and it comes
614    /// back as what it is once the frame has been taken.
615    fn take(
616        &mut self,
617        out: &mut Vec<Inst>,
618        size: u32,
619        below: &mut i32,
620        from_sp: bool,
621        probe: Option<Probing<'_>>,
622    ) {
623        // In front of everything else, because a platform with a routine for this has it for every
624        // frame rather than for the ones a flag was passed about, and because the routine does the
625        // whole of what the walk below would have done. See [`rucc_target::Chkstk`].
626        let page = self.insts.probe.map_or(u32::MAX, |probe| probe.interval);
627        if let Some(chkstk) = self.conv.chkstk.filter(|_| size > page) {
628            let inst = self.reach(out, chkstk, size);
629            *below += offset(size);
630            if from_sp {
631                self.row(inst, CfiOp::DefCfaOffset(*below));
632            }
633            return;
634        }
635        let Some(probing) = probe.filter(|probing| size > probing.probe.interval) else {
636            let inst = self.sub(size);
637            out.push(inst);
638            *below += offset(size);
639            if from_sp {
640                self.row(inst, CfiOp::DefCfaOffset(*below));
641            }
642            return;
643        };
644        // Every step but the last is a whole page and is followed by a touch, and the last is
645        // whatever is left over, which is between one byte and one whole page. So the stack
646        // pointer never moves further than a page without something being written where it landed,
647        // and the unmapped page an operating system leaves below a stack cannot be stepped over.
648        //
649        // That is why the count is worked out from one less than the size. A frame that is an
650        // exact number of pages gets one fewer touch than it has pages, and the step left over is
651        // a whole page, which is a step that lands on the next page boundary rather than past it.
652        // gcc touches that last page as well, so this is one instruction shorter on a frame whose
653        // size is a multiple of the page and the same everywhere else.
654        let interval = probing.probe.interval;
655        let pages = (size - 1) / interval;
656        let rest = size - pages * interval;
657        let mut walked = false;
658        if pages <= UNROLLED {
659            for _ in 0..pages {
660                let inst = self.sub(interval);
661                out.push(inst);
662                *below += offset(interval);
663                if from_sp {
664                    self.row(inst, CfiOp::DefCfaOffset(*below));
665                }
666                let touch = self.touch(probing.probe);
667                out.push(touch);
668            }
669        } else {
670            self.pages(out, pages, below, from_sp, probing);
671            walked = from_sp;
672        }
673        let inst = self.sub(rest);
674        out.push(inst);
675        *below += offset(rest);
676        if from_sp {
677            // A loop leaves the address counted from the register the stack pointer was compared
678            // against, since that is the one thing in it that holds still. This is where it goes
679            // back to being counted from the stack pointer, and it is written behind this
680            // instruction rather than behind the branch because a row is written behind an
681            // instruction and the branch is not one that survives [`crate::layout`].
682            let op = if walked {
683                let number = self.dwarf(self.conv.int_class, self.conv.stack_pointer);
684                CfiOp::DefCfa { reg: number, offset: *below }
685            } else {
686                CfiOp::DefCfaOffset(*below)
687            };
688            self.row(inst, op);
689        }
690    }
691
692    /// Reaches the pages of a frame by calling the routine this platform has for it, and then takes
693    /// the frame.
694    ///
695    /// Three instructions, and the third is the one that moves anything:
696    ///
697    /// ```text
698    ///   the size into a register    which register is the platform's answer rather than ours
699    ///   the call                    touches every page from here down to that many bytes below
700    ///   the subtraction             takes the frame, of the register the size is still in
701    /// ```
702    ///
703    /// The routine comes back having moved nothing, which is what makes the third instruction
704    /// necessary and is also what makes it a subtraction of a register rather than of the constant
705    /// written again. Writing the constant twice would be the same number of bytes of code and one
706    /// more place for the two to disagree.
707    ///
708    /// Nothing is described to the unwinder for the first two. The call pushes a return address and
709    /// the routine pops it, so the frame is the same on both sides of it, which is the same argument
710    /// the profiler's hook makes a few lines above. The row goes behind the subtraction, where the
711    /// stack pointer has actually moved.
712    ///
713    /// No register here has to be asked about. The size goes in one the convention passes no
714    /// argument in, which is what lets the platform name it at all, and the routine destroys two
715    /// that are exactly the two the allocator was told to hold back. A prologue is also the one
716    /// place in a function where the only live values are the ones that arrived in the convention's
717    /// own registers.
718    fn reach(&mut self, out: &mut Vec<Inst>, chkstk: Chkstk, size: u32) -> Inst {
719        let class = self.conv.int_class;
720        let sp = Reg::physical(self.conv.stack_pointer);
721        let count = Reg::physical(chkstk.size);
722
723        let imm = self.opcode(self.insts.imm);
724        let inst = self.func.build_loose(imm).def(count, class).imm(i64::from(size)).finish();
725        out.push(inst);
726        let call = self.opcode(self.insts.call);
727        let symbol = self.names.intern(chkstk.name);
728        let inst = self.func.build_loose(call).symbol(symbol).finish();
729        out.push(inst);
730        let grow = self.opcode(self.insts.grow);
731        let inst =
732            self.func.build_loose(grow).def(sp, class).uses(sp, class).uses(count, class).finish();
733        out.push(inst);
734        inst
735    }
736
737    /// The loop that takes a frame too large for the touches to be written one after another.
738    ///
739    /// Three blocks, and the first two are new and go in front of the one the function began with:
740    ///
741    /// ```text
742    ///   what the function is entered at   everything the prologue did before this, and then the
743    ///                                     address the stack pointer is walking down to
744    ///   the loop                          one page, the touch, and the question of whether the
745    ///                                     stack pointer has got there yet
746    ///   what the function began with      the rest of the prologue, and then the body
747    /// ```
748    ///
749    /// The instructions the prologue has written so far move into the first of them, because a
750    /// block is entered at the top and they have to run before the loop does. Nothing is laid out
751    /// here: which block comes first in memory is [`crate::layout`]'s answer, and all this decides
752    /// is which one the function is entered at.
753    fn pages(
754        &mut self,
755        out: &mut Vec<Inst>,
756        pages: u32,
757        below: &mut i32,
758        from_sp: bool,
759        probing: Probing<'_>,
760    ) {
761        let class = self.conv.int_class;
762        let sp = self.conv.stack_pointer;
763        let all = offset(pages * probing.probe.interval);
764        let [limit, byte] = probing.scratch;
765
766        let head = self.func.create_block();
767        for &inst in out.iter() {
768            self.func.append_inst(head, inst);
769        }
770        out.clear();
771        // Where the stack pointer is walking down to, worked out before it starts moving. A loop
772        // that counted down instead would need somewhere to keep the count, and this is somewhere
773        // to keep it that the comparison can read without arithmetic.
774        let lea = self.opcode(self.insts.lea);
775        let inst = self.address(lea, limit, sp, -all);
776        self.func.append_inst(head, inst);
777        if from_sp {
778            // The address is counted from that register for as long as the loop runs, and it has
779            // to be: the stack pointer moves once an iteration, so no fixed distance from it is
780            // true twice, and this register was written so that one distance is.
781            let number = self.dwarf(class, limit);
782            self.row(inst, CfiOp::DefCfa { reg: number, offset: *below + all });
783        }
784
785        let body = self.func.create_block();
786        *self.func.succs_mut(head) = vec![BlockCall::to(body)];
787        let inst = self.sub(probing.probe.interval);
788        self.func.append_inst(body, inst);
789        let touch = self.touch(probing.probe);
790        self.func.append_inst(body, touch);
791        let differ = self.opcode(self.insts.differ);
792        let inst = self
793            .func
794            .build_loose(differ)
795            .def(Reg::physical(byte), class)
796            .uses(Reg::physical(sp), class)
797            .uses(Reg::physical(limit), class)
798            .finish();
799        self.func.append_inst(body, inst);
800        let cond = Opcode::new(
801            self.names.intern(&format!("{}{}", probing.branch.prefix, probing.branch.cond)),
802        );
803        let inst = self.func.build_loose(cond).uses(Reg::physical(byte), class).finish();
804        self.func.append_inst(body, inst);
805        // The first arm is the one taken when the condition held, and the condition is that the
806        // stack pointer and the address it is walking down to still differ, so the first arm is
807        // another page.
808        let began = self.func.entry().expect("a function with a block in it");
809        *self.func.succs_mut(body) = vec![BlockCall::to(body), BlockCall::to(began)];
810        *below += all;
811        self.ahead = Some([head, body]);
812    }
813
814    /// Walks the pages a variable length array takes, at the declaration that takes them.
815    ///
816    /// The prologue's own pages are counted when it is written, so it can step down to an address
817    /// it worked out in advance and stop when it gets there. A declaration in the body cannot: how
818    /// many bytes it asked for arrives in a register, so where it is going is arithmetic rather than
819    /// a constant, and how many pages that is is a number nothing has. What is written instead is a
820    /// loop that steps a page and asks whether it has arrived yet, which is the same walk with the
821    /// count taken out of it.
822    ///
823    /// The one instruction the lowering wrote becomes four blocks:
824    ///
825    /// ```text
826    ///   what the block was          everything it did before the declaration, and then where the
827    ///                               stack pointer is going, worked out before it starts moving
828    ///   the step                    one page, and whether the stack pointer is still above there
829    ///   the page it stepped onto    the touch, and round again
830    ///   the rest of the block       the stack pointer put where it was going, and then the body
831    /// ```
832    ///
833    /// The touch is behind the question rather than in front of it, so the only page ever written
834    /// is one the array reaches. The last step down is a whole page whatever is left, which puts the
835    /// stack pointer at or past the end of the array, and the block that follows puts it back on the
836    /// end. Nothing is touched there and nothing has to be: that is a move of less than a page from
837    /// a page this loop has already been to, which is the whole of what a guard page asks.
838    ///
839    /// Nothing is described to the unwinder for any of it. A function with a variable length array
840    /// in it keeps a frame pointer, because its own stack pointer is not a fixed distance from
841    /// anything, and by here the frame is already counted from that register rather than from the
842    /// stack pointer. So the rule that was true before the walk is still true after it.
843    fn walk(&mut self, took: Inst, probing: Probing<'_>) {
844        let class = self.conv.int_class;
845        let sp = self.conv.stack_pointer;
846        let span = self.func.span(took);
847        let block = self.func.block_of(took).expect("an instruction the lowering put in a block");
848
849        // Which register the bytes arrived in, and which two the walk may use. The bytes may be in
850        // one of the two, because a reload the rewriter wrote is written into one of them, and a
851        // value that arrived that way is read by the one instruction it was written in front of and
852        // is dead after it. So the limit goes in whichever of the pair the bytes are not in, and the
853        // other one is free from the moment the limit has been worked out.
854        let operands = self.func[took].operands;
855        let bytes = self.func[operands][2].reg.phys().expect("a register the allocator settled");
856        let [first, second] = probing.scratch;
857        let (limit, flag) = if bytes == first { (second, first) } else { (first, second) };
858
859        let tail: Vec<Inst> = {
860            let mut rest = self.func.insts(block).skip_while(|&inst| inst != took);
861            rest.next();
862            rest.collect()
863        };
864        let step = self.func.create_block();
865        let onto = self.func.create_block();
866        let done = self.func.create_block();
867
868        let mov = self.opcode(self.insts.moves(class).expect("a class the target can move").mov);
869        let inst = self.two(mov, sp, limit);
870        self.func.append_inst(done, inst);
871        for inst in tail {
872            self.func.remove_inst(inst);
873            self.func.append_inst(done, inst);
874        }
875        let succs = std::mem::take(self.func.succs_mut(block));
876        *self.func.succs_mut(done) = succs;
877
878        // The subtraction the lowering wrote is what the loop is instead of, so it goes. What is
879        // left in the block it was in is where the stack pointer is walking down to.
880        self.func.remove_inst(took);
881        let inst = self.two(mov, limit, sp);
882        self.func.append_inst(block, inst);
883        let grow = self.opcode(self.insts.grow);
884        let inst = self
885            .func
886            .build_loose(grow)
887            .at(span)
888            .def(Reg::physical(limit), class)
889            .uses(Reg::physical(limit), class)
890            .uses(Reg::physical(bytes), class)
891            .finish();
892        self.func.append_inst(block, inst);
893        *self.func.succs_mut(block) = vec![BlockCall::to(step)];
894
895        let inst = self.sub(probing.probe.interval);
896        self.func.append_inst(step, inst);
897        let above = self.opcode(self.insts.above);
898        let inst = self
899            .func
900            .build_loose(above)
901            .def(Reg::physical(flag), class)
902            .uses(Reg::physical(sp), class)
903            .uses(Reg::physical(limit), class)
904            .finish();
905        self.func.append_inst(step, inst);
906        let cond = Opcode::new(
907            self.names.intern(&format!("{}{}", probing.branch.prefix, probing.branch.cond)),
908        );
909        let inst = self.func.build_loose(cond).uses(Reg::physical(flag), class).finish();
910        self.func.append_inst(step, inst);
911        // The first arm is the one taken when the condition held, and the condition is that the
912        // stack pointer is still above where the array ends, so the first arm is the page it has
913        // just stepped onto being written and another time round.
914        *self.func.succs_mut(step) = vec![BlockCall::to(onto), BlockCall::to(done)];
915
916        let touch = self.touch(probing.probe);
917        self.func.append_inst(onto, touch);
918        *self.func.succs_mut(onto) = vec![BlockCall::to(step)];
919    }
920
921    /// Writes the page the stack pointer is on without changing what is there.
922    fn touch(&mut self, probe: &Probe) -> Inst {
923        let opcode = self.opcode(probe.inst);
924        let base = Operand::read(Reg::physical(self.conv.stack_pointer), self.conv.int_class);
925        self.func.build_loose(opcode).imm(0).mem(Mem::at(base)).finish()
926    }
927
928    /// Takes that many bytes off the stack pointer.
929    fn sub(&mut self, bytes: u32) -> Inst {
930        let sub = self.opcode(self.insts.sub);
931        self.arith(sub, i64::from(bytes))
932    }
933
934    /// The stack protector's check, written at the end of a block the function returns from.
935    ///
936    /// Gives back the block the epilogue goes in, which is a new one: the check has to be the last
937    /// thing the old block does, and what follows it is one of two arms rather than the return.
938    ///
939    /// ```text
940    ///   block that returned      reload the slot, read the word again, compare, branch
941    ///   the arm it changed on    call the function that does not come back, and nothing after
942    ///   the arm it did not       the epilogue, which the caller writes into what this gives back
943    /// ```
944    ///
945    /// The two registers are the ones the allocator was told to hold back, so nothing here has to
946    /// ask what is live: a scratch register holds nothing at the end of a block, because the only
947    /// thing that writes one is a move the rewriter put in and every one of those is read by the
948    /// instruction it was put in front of.
949    fn check(&mut self, block: Block, frame: &Frame, protect: Protect<'_>) -> Block {
950        let class = self.conv.int_class;
951        let at = frame.canary().expect("a protected function has a slot for its canary");
952        let [ours, theirs] = protect.scratch;
953
954        let inst = self.load(class, ours, at);
955        self.func.append_inst(block, inst);
956        let inst = self.read_guard(theirs, protect.guard);
957        self.func.append_inst(block, inst);
958        let differ = self.opcode(self.insts.differ);
959        let inst = self
960            .func
961            .build_loose(differ)
962            .def(Reg::physical(theirs), class)
963            .uses(Reg::physical(ours), class)
964            .uses(Reg::physical(theirs), class)
965            .finish();
966        self.func.append_inst(block, inst);
967
968        let failed = self.func.create_block();
969        let ok = self.func.create_block();
970        let cond = Opcode::new(
971            self.names.intern(&format!("{}{}", protect.branch.prefix, protect.branch.cond)),
972        );
973        let inst = self.func.build_loose(cond).uses(Reg::physical(theirs), class).finish();
974        self.func.append_inst(block, inst);
975        // The first arm is the one taken when the condition held, and the condition is that the
976        // two words differ, so the first arm is the one the canary was overwritten on.
977        *self.func.succs_mut(block) = vec![BlockCall::to(failed), BlockCall::to(ok)];
978
979        let call = self.opcode(self.insts.call);
980        let symbol = self.names.intern(protect.guard.fail);
981        self.func.build(failed, call).symbol(symbol).finish();
982        ok
983    }
984
985    /// Reads the word the canary is a copy of into a register.
986    ///
987    /// The address is a constant and names no register at all, because where the block a thread
988    /// has to itself begins is something only the machine knows and the segment register is what
989    /// holds it.
990    fn read_guard(&mut self, into: PhysReg, guard: &Guard) -> Inst {
991        let class = self.conv.int_class;
992        let load = self.opcode(self.insts.moves(class).expect("a class to load").load);
993        self.func
994            .build_loose(load)
995            .def(Reg::physical(into), class)
996            .mem(Mem::in_segment(guard.segment, guard.at))
997            .finish()
998    }
999
1000    /// The instructions the epilogue is, in the order they run.
1001    ///
1002    /// The vector registers are read back while the stack pointer is still where the body left it,
1003    /// because that is what their offsets are from. Then the stack pointer goes back to the last
1004    /// register the prologue pushed, which is arithmetic when the prologue knew how far it had
1005    /// moved and a read of the frame pointer when it did not.
1006    fn epilogue(&mut self, frame: &Frame) -> Vec<Inst> {
1007        let sp = self.conv.stack_pointer;
1008        let fp = self.conv.frame_pointer;
1009        let int = self.conv.int_class;
1010        let sse = self.conv.sse_class;
1011        let word = self.conv.word;
1012        let described = !self.func.cfi.is_empty();
1013        let mut out = Vec::new();
1014        // Where the body left things, which is where every epilogue starts from.
1015        let mut below = offset(self.conv.return_address)
1016            + offset(word) * self.pushes(frame)
1017            + offset(frame.size());
1018        let from_sp = !frame.frame_pointer();
1019        for save in frame.saved_sse() {
1020            let inst = self.load(sse, save.reg, save.at);
1021            out.push(inst);
1022            if frame.realign().is_none() {
1023                self.restored(inst, sse, save.reg);
1024            }
1025        }
1026        let pushed = u32::try_from(frame.saved_int().len()).expect("a frame");
1027        if frame.frame_pointer() {
1028            // No row for either of these. The address is counted from the frame pointer here and
1029            // this is what moves the stack pointer rather than the frame pointer, so the rule that
1030            // was true before it is still true after it.
1031            //
1032            // Where the pointer is decides how far back this has to go. The early order left it one
1033            // word above the first push, so the pops start that many words below it. The late one
1034            // left it where the body's stack pointer was, so they start the whole frame above it,
1035            // and in both cases the distance is a constant even in a frame that grew while it ran,
1036            // which is why this is written rather than an addition to the stack pointer.
1037            let back = if frame.late() { offset(frame.size()) } else { -offset(word * pushed) };
1038            if back == 0 {
1039                let mov = self.opcode(self.insts.moves(int).expect("a move").mov);
1040                out.push(self.two(mov, sp, fp));
1041            } else {
1042                let lea = self.opcode(self.insts.lea);
1043                out.push(self.address(lea, sp, fp, back));
1044            }
1045        } else if frame.size() > 0 {
1046            let add = self.opcode(self.insts.add);
1047            let inst = self.arith(add, i64::from(frame.size()));
1048            out.push(inst);
1049            below -= offset(frame.size());
1050            self.row(inst, CfiOp::DefCfaOffset(below));
1051        }
1052        for &reg in frame.saved_int().iter().rev() {
1053            let inst = self.pop(reg);
1054            out.push(inst);
1055            self.restored(inst, int, reg);
1056            below -= offset(word);
1057            if from_sp {
1058                self.row(inst, CfiOp::DefCfaOffset(below));
1059            }
1060        }
1061        if frame.frame_pointer() {
1062            let inst = self.pop(fp);
1063            out.push(inst);
1064            self.restored(inst, int, fp);
1065            // The frame pointer holds the caller's value again, so the address goes back to being
1066            // counted from the stack pointer, which by now is at the return address.
1067            let number = self.dwarf(int, sp);
1068            self.row(inst, CfiOp::DefCfa { reg: number, offset: offset(self.conv.return_address) });
1069        }
1070        let ret = self.opcode(self.insts.ret);
1071        let inst = self.func.build_loose(ret).finish();
1072        out.push(inst);
1073        // These take effect at the address just past the return, which is where the next block
1074        // begins, and the next block is body again. Popping the body's rules and pushing them
1075        // straight back leaves the stack one deep however many blocks the function returns from,
1076        // which is what makes one remembering in the prologue enough for all of them.
1077        if described {
1078            self.row(inst, CfiOp::RestoreState);
1079            self.row(inst, CfiOp::RememberState);
1080        }
1081        // And where all of it came from, which is the closing brace. Nothing here has a span of its
1082        // own: an epilogue is the frame going back the way it came and no expression in the source
1083        // asked for any of it, so without this the bytes are covered by whatever the last statement
1084        // of the body was. That is the hole the prologue used to have, at the other end, and gcc
1085        // fills it the same way it fills the other one, with the brace. A function whose body this
1086        // does not know is left alone and keeps covering those bytes with the last row before them.
1087        let closing = ending(self.func.declared);
1088        if !closing.is_dummy() {
1089            for &inst in &out {
1090                self.func.set_span(inst, closing);
1091            }
1092        }
1093        out
1094    }
1095
1096    /// How many general purpose registers the prologue put on the stack, the frame pointer
1097    /// included.
1098    fn pushes(&self, frame: &Frame) -> i32 {
1099        let saved = i32::try_from(frame.saved_int().len()).expect("a frame");
1100        saved + i32::from(frame.frame_pointer())
1101    }
1102
1103    /// One row of the unwind table, taking effect after that instruction.
1104    fn row(&mut self, inst: Inst, op: CfiOp) {
1105        self.func.cfi.push((inst, op));
1106    }
1107
1108    /// A row saying the caller's copy of that register is that far from the canonical frame
1109    /// address, which is below it and so is negative.
1110    fn saved(&mut self, inst: Inst, class: RegClass, reg: PhysReg, from_cfa: i32) {
1111        let number = self.dwarf(class, reg);
1112        self.row(inst, CfiOp::Offset { reg: number, offset: from_cfa });
1113    }
1114
1115    /// A row saying that register holds what the caller left in it again.
1116    fn restored(&mut self, inst: Inst, class: RegClass, reg: PhysReg) {
1117        let number = self.dwarf(class, reg);
1118        self.row(inst, CfiOp::Restore(number));
1119    }
1120
1121    /// What an unwind table calls that register.
1122    fn dwarf(&self, class: RegClass, reg: PhysReg) -> u16 {
1123        self.conv.dwarf(class, reg).expect("a register a frame saves is one the table can name")
1124    }
1125
1126    /// One edit as the instruction that makes it true.
1127    fn mov(&mut self, edit: &Edit, frame: &Frame) -> Inst {
1128        let moves = self.insts.moves(edit.class).expect("a class the target says how to move");
1129        match (edit.mov.to, edit.mov.from) {
1130            (Place::Reg(to), Place::Reg(from)) => {
1131                let mov = self.opcode(moves.mov);
1132                self.func
1133                    .build_loose(mov)
1134                    .def(Reg::physical(to), edit.class)
1135                    .uses(Reg::physical(from), edit.class)
1136                    .finish()
1137            }
1138            (Place::Reg(to), Place::Slot(slot)) => {
1139                let at = self.slot(frame, slot);
1140                self.load(edit.class, to, at)
1141            }
1142            (Place::Slot(slot), Place::Reg(from)) => {
1143                let at = self.slot(frame, slot);
1144                self.store(edit.class, from, at)
1145            }
1146            // The allocator expands this into two moves through a register of its own, because a
1147            // machine that could do it in one is not a machine any of this is written for.
1148            (Place::Slot(_), Place::Slot(_)) => {
1149                unreachable!("a move from one stack slot straight into another")
1150            }
1151        }
1152    }
1153
1154    /// Puts an instruction where an edit says it goes, after whatever earlier edits went there.
1155    ///
1156    /// The edits at one place are in the order they have to be made in, so each one goes behind
1157    /// the last, and the first of them is what the place itself means.
1158    fn put(&mut self, cursors: &mut HashMap<At, Inst>, at: At, inst: Inst) {
1159        if let Some(cursor) = cursors.get_mut(&at) {
1160            self.func.insert_after(*cursor, inst);
1161            *cursor = inst;
1162            return;
1163        }
1164        match at {
1165            At::Before(before) => self.func.insert_before(before, inst),
1166            At::After(after) => self.func.insert_after(after, inst),
1167            At::StartOf(block) => self.func.prepend_inst(block, inst),
1168            // Behind everything in the block. A block the allocator puts an edge's moves at the
1169            // end of is one with a single edge out of it, and an edge like that is not an
1170            // instruction here: [`crate::layout`] writes the jump it becomes after this has run.
1171            // So the last instruction is an ordinary one, which may still be waiting on moves of
1172            // its own that have to be made before the edge's are.
1173            At::EndOf(block) => self.func.append_inst(block, inst),
1174        }
1175        cursors.insert(at, inst);
1176    }
1177
1178    /// Where a spill slot is, from the stack pointer in the body of the function.
1179    fn slot(&self, frame: &Frame, slot: u32) -> i32 {
1180        frame.slot(slot).expect("a slot the frame was worked out from")
1181    }
1182
1183    /// Reads a register out of the frame.
1184    fn load(&mut self, class: RegClass, reg: PhysReg, at: i32) -> Inst {
1185        let load = self.opcode(self.insts.moves(class).expect("a class to load").load);
1186        let base = Operand::read(Reg::physical(self.base), self.conv.int_class);
1187        self.func
1188            .build_loose(load)
1189            .def(Reg::physical(reg), class)
1190            .mem(Mem::at(base).plus(at))
1191            .finish()
1192    }
1193
1194    /// Writes a register into the frame.
1195    fn store(&mut self, class: RegClass, reg: PhysReg, at: i32) -> Inst {
1196        let store = self.opcode(self.insts.moves(class).expect("a class to store").store);
1197        let base = Operand::read(Reg::physical(self.base), self.conv.int_class);
1198        self.func
1199            .build_loose(store)
1200            .uses(Reg::physical(reg), class)
1201            .mem(Mem::at(base).plus(at))
1202            .finish()
1203    }
1204
1205    /// Puts a general purpose register on the stack.
1206    fn push(&mut self, reg: PhysReg) -> Inst {
1207        let push = self.opcode(self.insts.push);
1208        self.func.build_loose(push).uses(Reg::physical(reg), self.conv.int_class).finish()
1209    }
1210
1211    /// Takes a general purpose register back off the stack.
1212    fn pop(&mut self, reg: PhysReg) -> Inst {
1213        let pop = self.opcode(self.insts.pop);
1214        self.func.build_loose(pop).def(Reg::physical(reg), self.conv.int_class).finish()
1215    }
1216
1217    /// One general purpose register written with another.
1218    fn two(&mut self, opcode: Opcode, to: PhysReg, from: PhysReg) -> Inst {
1219        let class = self.conv.int_class;
1220        self.func
1221            .build_loose(opcode)
1222            .def(Reg::physical(to), class)
1223            .uses(Reg::physical(from), class)
1224            .finish()
1225    }
1226
1227    /// Two-address arithmetic on the stack pointer, which reads it and writes it back.
1228    fn arith(&mut self, opcode: Opcode, value: i64) -> Inst {
1229        let class = self.conv.int_class;
1230        let sp = Reg::physical(self.conv.stack_pointer);
1231        self.func.build_loose(opcode).def(sp, class).uses(sp, class).imm(value).finish()
1232    }
1233
1234    /// One register written with an address rather than with what is at it.
1235    fn address(&mut self, opcode: Opcode, to: PhysReg, base: PhysReg, disp: i32) -> Inst {
1236        let class = self.conv.int_class;
1237        let base = Operand::read(Reg::physical(base), class);
1238        self.func
1239            .build_loose(opcode)
1240            .def(Reg::physical(to), class)
1241            .mem(Mem::at(base).plus(disp))
1242            .finish()
1243    }
1244
1245    /// The opcode of that name, in the machine IR's spelling, which is the target's prefix and
1246    /// then the name the target gave.
1247    fn opcode(&mut self, name: &str) -> Opcode {
1248        Opcode::new(self.names.intern(&format!("{}{name}", self.insts.prefix)))
1249    }
1250}
1251
1252/// A distance in a frame, as the signed number every offset is.
1253fn offset(bytes: u32) -> i32 {
1254    i32::try_from(bytes).expect("a frame under two gigabytes")
1255}
1256
1257/// The last character of a span, which for the span of a function body is its closing brace.
1258///
1259/// A span runs from the first byte to one past the last, so the brace is the byte before the end
1260/// rather than the end. [`Span::DUMMY`] for a function that came from no C source, which is what
1261/// the tests and the IR parser build, and for the empty span that cannot have a last character.
1262fn ending(body: Span) -> Span {
1263    if body.is_dummy() || body.hi <= body.lo {
1264        return Span::DUMMY;
1265    }
1266    Span::new(body.hi - 1, body.hi)
1267}
1268
1269#[cfg(test)]
1270mod tests {
1271    use rucc_base::Interner;
1272    use rucc_mir::{BlockCall, print_func};
1273    use rucc_regalloc::assign::Env;
1274    use rucc_target::x86_64::{
1275        BRANCH, FRAME, GPR, PROBE, R10, R11, RAX, REGS, SYSV, WIN64, XMM, xmm,
1276    };
1277
1278    use super::*;
1279    use crate::frame::{Layout, Local};
1280
1281    /// The closing brace of a body is the last character of its span and not the end of it, since a
1282    /// span runs to one past what it covers. A function that came from no source has no brace and
1283    /// asks for no row, which is what keeps the epilogue of one the IR parser built covered by the
1284    /// row before it rather than by a position in a file that is not there.
1285    #[test]
1286    fn the_end_of_a_body_is_its_closing_brace_and_not_one_past_it() {
1287        assert_eq!(ending(Span::new(10, 40)), Span::new(39, 40));
1288        assert_eq!(ending(Span::DUMMY), Span::DUMMY);
1289        assert_eq!(ending(Span::new(7, 7)), Span::DUMMY);
1290    }
1291
1292    /// An environment offering that many of the convention's registers, with everything after
1293    /// them held back as scratch.
1294    fn env(conv: &CallRegs, count: usize) -> Env {
1295        Env::new().with(GPR, &conv.int_order[..count], &conv.int_order[count..])
1296    }
1297
1298    /// A function of that many values, every one written before any is read, allocated with that
1299    /// many registers to hand out. The same shape the frame layout's own tests are written
1300    /// against, so that a frame here is one that has already been checked there.
1301    fn pressure(conv: &CallRegs, values: usize, count: usize) -> (Func, Allocation, Interner) {
1302        let mut names = Interner::new();
1303        let mut func = Func::new(names.intern("f"));
1304        let opcode = Opcode::new(names.intern("x64.nop"));
1305        let block = func.create_block();
1306        let regs: Vec<Reg> = (0..values).map(|_| func.new_vreg(GPR)).collect();
1307        for &reg in &regs {
1308            func.build(block, opcode).def(reg, GPR).finish();
1309        }
1310        for &reg in &regs {
1311            func.build(block, opcode).uses(reg, GPR).finish();
1312        }
1313        let allocation = rucc_regalloc::run(&mut func, &env(conv, count), "test", true);
1314        (func, allocation, names)
1315    }
1316
1317    /// The function with its frame written into it, as the lines a dump would show.
1318    fn written(
1319        func: &mut Func,
1320        allocation: &Allocation,
1321        layout: &Layout<'_>,
1322        names: &mut Interner,
1323    ) -> Vec<String> {
1324        with_protector(func, allocation, layout, None, names)
1325    }
1326
1327    /// The same, for a function the caller has decided is protected or is not.
1328    fn with_protector(
1329        func: &mut Func,
1330        allocation: &Allocation,
1331        layout: &Layout<'_>,
1332        protect: Option<Protect<'_>>,
1333        names: &mut Interner,
1334    ) -> Vec<String> {
1335        let convention = Convention { protect, ..Convention::new(layout.conv, &FRAME) };
1336        under(func, allocation, layout, &Stack::default(), convention, names)
1337    }
1338
1339    /// The same, for a function whose frame the caller has decided is taken a page at a time.
1340    fn with_probing(
1341        func: &mut Func,
1342        allocation: &Allocation,
1343        layout: &Layout<'_>,
1344        probe: Option<Probing<'_>>,
1345        names: &mut Interner,
1346    ) -> Vec<String> {
1347        let convention = Convention { probe, ..Convention::new(layout.conv, &FRAME) };
1348        under(func, allocation, layout, &Stack::default(), convention, names)
1349    }
1350
1351    /// A function whose one block takes a run of bytes off the stack pointer, which is what the
1352    /// lowering writes for a variable length array, with the count already in the register given.
1353    fn growing(count: PhysReg) -> (Func, Allocation, Interner, Stack) {
1354        let mut names = Interner::new();
1355        let mut func = Func::new(names.intern("f"));
1356        let block = func.create_block();
1357        let sp = Reg::physical(SYSV.stack_pointer);
1358        let grow = Opcode::new(names.intern("x64.sub_rr_64"));
1359        let took = func
1360            .build(block, grow)
1361            .def(sp, GPR)
1362            .uses(sp, GPR)
1363            .uses(Reg::physical(count), GPR)
1364            .finish();
1365        let nop = Opcode::new(names.intern("x64.nop"));
1366        func.build(block, nop).finish();
1367        let allocation = rucc_regalloc::run(&mut func, &env(&SYSV, 4), "test", true);
1368        (func, allocation, names, Stack { grown: vec![took], ..Stack::default() })
1369    }
1370
1371    /// The function with its frame written into it under that convention.
1372    fn under(
1373        func: &mut Func,
1374        allocation: &Allocation,
1375        layout: &Layout<'_>,
1376        stack: &Stack,
1377        convention: Convention<'_>,
1378        names: &mut Interner,
1379    ) -> Vec<String> {
1380        let frame = Frame::of(func, allocation, layout);
1381        finish(func, allocation, &frame, stack, convention, names);
1382        print_func(func, names, &REGS)
1383            .lines()
1384            .filter(|line| !line.is_empty())
1385            .map(|line| line.trim().to_string())
1386            .collect()
1387    }
1388
1389    /// Just the lines the frame put in, which is every line that is not the function it was
1390    /// given and not the shape of the dump around it.
1391    fn added(lines: &[String]) -> Vec<&str> {
1392        lines
1393            .iter()
1394            .map(String::as_str)
1395            .filter(|line| !line.contains("x64.nop"))
1396            .filter(|line| !line.starts_with("mfunc") && !line.starts_with("block") && *line != "}")
1397            .collect()
1398    }
1399
1400    #[test]
1401    fn a_function_that_needs_no_frame_is_given_a_return_and_nothing_else() {
1402        let (mut func, allocation, mut names) = pressure(&SYSV, 2, 4);
1403        let lines = written(&mut func, &allocation, &Layout::new(&SYSV, REGS), &mut names);
1404
1405        // Two values and four registers, so nothing is spilled, nothing is saved and the stack
1406        // pointer never moves. A prologue of nothing is the right prologue for that.
1407        assert_eq!(added(&lines), ["x64.ret"]);
1408    }
1409
1410    /// The bytes that give a frame back are filed under the closing brace, which is where a
1411    /// debugger says a function ends and which nothing in an epilogue could say for itself.
1412    #[test]
1413    fn an_epilogue_is_filed_under_the_closing_brace_of_the_body() {
1414        let (mut func, allocation, mut names) = pressure(&SYSV, 4, 2);
1415        func.declared = Span::new(100, 140);
1416        let base = Layout::new(&SYSV, REGS);
1417        let layout = Layout { red_zone: false, ..base };
1418        written(&mut func, &allocation, &layout, &mut names);
1419
1420        // Everything from the first instruction of the epilogue to the return, and nothing above
1421        // it: the body's own instructions keep the spans they arrived with, which here is none.
1422        let ends: Vec<Span> = func
1423            .blocks()
1424            .flat_map(|block| func.insts(block).collect::<Vec<_>>())
1425            .map(|inst| func.span(inst))
1426            .filter(|span| !span.is_dummy())
1427            .collect();
1428        assert!(!ends.is_empty(), "an epilogue was written");
1429        assert!(ends.iter().all(|&span| span == Span::new(139, 140)), "{ends:?}");
1430    }
1431
1432    #[test]
1433    fn a_spill_is_a_store_and_a_reload_is_a_load() {
1434        let (mut func, allocation, mut names) = pressure(&SYSV, 4, 2);
1435        let lines = written(&mut func, &allocation, &Layout::new(&SYSV, REGS), &mut names);
1436
1437        // Two registers for four values, so two of them go to the stack. The store goes behind the
1438        // instruction that wrote the value and the load in front of the one that wants it, both at
1439        // the offsets the frame gave, which are below the stack pointer because a small leaf
1440        // function is entitled to the red zone.
1441        assert_eq!(
1442            lines,
1443            [
1444                "mfunc @f {",
1445                "block0:",
1446                "$rax = x64.nop",
1447                "$rcx = x64.nop",
1448                "$rdx = x64.nop",
1449                "x64.mov_mr_64 $rdx, [$rsp - 16]",
1450                "$rdx = x64.nop",
1451                "x64.mov_mr_64 $rdx, [$rsp - 8]",
1452                "x64.nop $rax",
1453                "x64.nop $rcx",
1454                "$rdx = x64.mov_rm_64 [$rsp - 16]",
1455                "x64.nop $rdx",
1456                "$rdx = x64.mov_rm_64 [$rsp - 8]",
1457                "x64.nop $rdx",
1458                "x64.ret",
1459                "}",
1460            ]
1461        );
1462    }
1463
1464    #[test]
1465    fn the_frame_the_prologue_takes_is_the_frame_the_epilogue_gives_back() {
1466        let (mut func, allocation, mut names) = pressure(&SYSV, 4, 2);
1467        let base = Layout::new(&SYSV, REGS);
1468        let layout = Layout { red_zone: false, ..base };
1469        let lines = written(&mut func, &allocation, &layout, &mut names);
1470
1471        // The same function told it may not use the red zone takes sixteen bytes instead, and
1472        // every offset moves above the stack pointer to match.
1473        assert_eq!(
1474            added(&lines),
1475            [
1476                "$rsp = x64.sub_ri_64 $rsp, 16",
1477                "x64.mov_mr_64 $rdx, [$rsp]",
1478                "x64.mov_mr_64 $rdx, [$rsp + 8]",
1479                "$rdx = x64.mov_rm_64 [$rsp]",
1480                "$rdx = x64.mov_rm_64 [$rsp + 8]",
1481                "$rsp = x64.add_ri_64 $rsp, 16",
1482                "x64.ret",
1483            ]
1484        );
1485    }
1486
1487    #[test]
1488    fn the_registers_the_prologue_pushes_come_back_in_the_opposite_order() {
1489        let (mut func, allocation, mut names) = pressure(&SYSV, 13, 13);
1490        let lines = written(&mut func, &allocation, &Layout::new(&SYSV, REGS), &mut names);
1491
1492        // Four registers a call leaves alone, pushed in the convention's order and popped in the
1493        // other one, which is the only order that gets each of them its own value back.
1494        assert_eq!(
1495            added(&lines),
1496            [
1497                "x64.push_64 $rbx",
1498                "x64.push_64 $r12",
1499                "x64.push_64 $r13",
1500                "x64.push_64 $r14",
1501                "$r14 = x64.pop_64",
1502                "$r13 = x64.pop_64",
1503                "$r12 = x64.pop_64",
1504                "$rbx = x64.pop_64",
1505                "x64.ret",
1506            ]
1507        );
1508    }
1509
1510    #[test]
1511    fn a_function_that_keeps_a_frame_pointer_sets_it_up_and_leaves_by_it() {
1512        let (mut func, allocation, mut names) = pressure(&SYSV, 4, 2);
1513        let base = Layout::new(&SYSV, REGS);
1514        let layout = Layout { frame_pointer: true, red_zone: false, ..base };
1515        let lines = written(&mut func, &allocation, &layout, &mut names);
1516
1517        // The frame pointer is saved before anything else and points at where it was saved, so the
1518        // epilogue reaches the stack pointer through it rather than by counting the frame back.
1519        assert_eq!(
1520            added(&lines),
1521            [
1522                "x64.push_64 $rbp",
1523                "$rbp = x64.mov_rr_64 $rsp",
1524                "$rsp = x64.sub_ri_64 $rsp, 16",
1525                "x64.mov_mr_64 $rdx, [$rsp]",
1526                "x64.mov_mr_64 $rdx, [$rsp + 8]",
1527                "$rdx = x64.mov_rm_64 [$rsp]",
1528                "$rdx = x64.mov_rm_64 [$rsp + 8]",
1529                "$rsp = x64.mov_rr_64 $rbp",
1530                "$rbp = x64.pop_64",
1531                "x64.ret",
1532            ]
1533        );
1534    }
1535
1536    #[test]
1537    fn a_realigned_frame_forces_the_alignment_after_it_has_pushed_what_it_saves() {
1538        let (mut func, allocation, mut names) = pressure(&SYSV, 13, 13);
1539        let locals = [Local { size: 64, align: 32 }];
1540        let base = Layout::new(&SYSV, REGS);
1541        let layout = Layout { locals: &locals, ..base };
1542        let lines = written(&mut func, &allocation, &layout, &mut names);
1543
1544        // Forcing the alignment throws away how far the stack pointer had moved, so the registers
1545        // are pushed before it happens and the epilogue counts back from the frame pointer to find
1546        // them. The frame pointer is required here whatever the flags said.
1547        assert_eq!(
1548            added(&lines),
1549            [
1550                "x64.push_64 $rbp",
1551                "$rbp = x64.mov_rr_64 $rsp",
1552                "x64.push_64 $rbx",
1553                "x64.push_64 $r12",
1554                "x64.push_64 $r13",
1555                "x64.push_64 $r14",
1556                "$rsp = x64.and_ri_64 $rsp, -32",
1557                "$rsp = x64.sub_ri_64 $rsp, 64",
1558                "$rsp = x64.lea_64 [$rbp - 32]",
1559                "$r14 = x64.pop_64",
1560                "$r13 = x64.pop_64",
1561                "$r12 = x64.pop_64",
1562                "$rbx = x64.pop_64",
1563                "$rbp = x64.pop_64",
1564                "x64.ret",
1565            ]
1566        );
1567    }
1568
1569    #[test]
1570    fn every_block_the_function_returns_from_gets_an_epilogue() {
1571        let mut names = Interner::new();
1572        let mut func = Func::new(names.intern("f"));
1573        let opcode = Opcode::new(names.intern("x64.nop"));
1574        let head = func.create_block();
1575        let left = func.create_block();
1576        let right = func.create_block();
1577        func.build(head, opcode).finish();
1578        *func.succs_mut(head) = vec![BlockCall::to(left), BlockCall::to(right)];
1579        func.build(left, opcode).finish();
1580        func.build(right, opcode).finish();
1581        let allocation = rucc_regalloc::run(&mut func, &env(&SYSV, 4), "test", true);
1582        let base = Layout::new(&SYSV, REGS);
1583        let layout = Layout { leaf: false, ..base };
1584        let lines = written(&mut func, &allocation, &layout, &mut names);
1585
1586        // Both ways out get the frame given back, and the block that goes somewhere gets nothing,
1587        // because a block with an edge out of it is not a block anything returns from.
1588        assert_eq!(
1589            lines,
1590            [
1591                "mfunc @f {",
1592                "block0:",
1593                "$rsp = x64.sub_ri_64 $rsp, 8",
1594                "x64.nop block1, block2",
1595                "block1:",
1596                "x64.nop",
1597                "$rsp = x64.add_ri_64 $rsp, 8",
1598                "x64.ret",
1599                "block2:",
1600                "x64.nop",
1601                "$rsp = x64.add_ri_64 $rsp, 8",
1602                "x64.ret",
1603                "}",
1604            ]
1605        );
1606    }
1607
1608    #[test]
1609    fn a_protected_function_writes_the_canary_last_and_checks_it_before_it_returns() {
1610        let (mut func, allocation, mut names) = pressure(&SYSV, 4, 2);
1611        let base = Layout::new(&SYSV, REGS);
1612        let layout = Layout { leaf: false, protect: true, ..base };
1613        let guard = SYSV.guard.as_ref().expect("this convention has somewhere to keep the word");
1614        // The two the real pipeline holds back, which are held back in the environment above too:
1615        // it hands out the first two of the convention's order and keeps everything after them.
1616        let protect = Protect { guard, branch: &BRANCH, scratch: [R10, R11] };
1617        let lines = with_protector(&mut func, &allocation, &layout, Some(protect), &mut names);
1618
1619        // The read of the word and the store into the slot come after the stack pointer has moved,
1620        // because there is no slot to store into until it has. The check is the last thing the
1621        // block that returned does and the epilogue is on the arm the canary was unchanged on, so
1622        // a function whose canary changed never gives its frame back and never returns.
1623        assert_eq!(
1624            added(&lines),
1625            [
1626                "$rsp = x64.sub_ri_64 $rsp, 24",
1627                "$r10 = x64.mov_rm_64 [fs:40]",
1628                "x64.mov_mr_64 $r10, [$rsp + 16]",
1629                "x64.mov_mr_64 $rdx, [$rsp]",
1630                "x64.mov_mr_64 $rdx, [$rsp + 8]",
1631                "$rdx = x64.mov_rm_64 [$rsp]",
1632                "$rdx = x64.mov_rm_64 [$rsp + 8]",
1633                "$r10 = x64.mov_rm_64 [$rsp + 16]",
1634                "$r11 = x64.mov_rm_64 [fs:40]",
1635                "$r11 = x64.cmp_set_ne_64 $r10, $r11",
1636                "x64.br_cond_8 $r11, block1, block2",
1637                "x64.call @__stack_chk_fail",
1638                "$rsp = x64.add_ri_64 $rsp, 24",
1639                "x64.ret",
1640            ]
1641        );
1642    }
1643
1644    #[test]
1645    fn a_frame_that_fits_in_one_page_is_taken_in_one_subtraction_even_when_pages_are_touched() {
1646        let (mut func, allocation, mut names) = pressure(&SYSV, 2, 4);
1647        let locals = [Local { size: 4088, align: 16 }];
1648        let base = Layout::new(&SYSV, REGS);
1649        let layout = Layout { leaf: false, locals: &locals, ..base };
1650        let probing = Probing { probe: &PROBE, branch: &BRANCH, scratch: [R10, R11] };
1651        let lines = with_probing(&mut func, &allocation, &layout, Some(probing), &mut names);
1652
1653        // A frame of one page cannot step over the page below it, because the far end of it is the
1654        // near end of that page and anything written there is written to a page that is there. So
1655        // the flag costs such a function nothing, which is most functions.
1656        assert_eq!(
1657            added(&lines),
1658            ["$rsp = x64.sub_ri_64 $rsp, 4088", "$rsp = x64.add_ri_64 $rsp, 4088", "x64.ret",]
1659        );
1660    }
1661
1662    #[test]
1663    fn a_probing_prologue_touches_every_page_of_a_frame_a_few_pages_deep() {
1664        let (mut func, allocation, mut names) = pressure(&SYSV, 2, 4);
1665        let locals = [Local { size: 9000, align: 16 }];
1666        let base = Layout::new(&SYSV, REGS);
1667        let layout = Layout { leaf: false, locals: &locals, ..base };
1668        let probing = Probing { probe: &PROBE, branch: &BRANCH, scratch: [R10, R11] };
1669        let lines = with_probing(&mut func, &allocation, &layout, Some(probing), &mut names);
1670
1671        // A page of the stack pointer's own, then the touch that says the page is there, and only
1672        // then the next one, which is the whole of the defence: nothing here ever moves the stack
1673        // pointer further than one page without writing where it landed. The last subtraction is
1674        // the remainder and is smaller than a page, so it needs no touch of its own, and it exists
1675        // in every frame because the count of pages is taken off one less than the size.
1676        assert_eq!(
1677            added(&lines),
1678            [
1679                "$rsp = x64.sub_ri_64 $rsp, 4096",
1680                "x64.or_mi_8 [$rsp], 0",
1681                "$rsp = x64.sub_ri_64 $rsp, 4096",
1682                "x64.or_mi_8 [$rsp], 0",
1683                "$rsp = x64.sub_ri_64 $rsp, 808",
1684                "$rsp = x64.add_ri_64 $rsp, 9000",
1685                "x64.ret",
1686            ]
1687        );
1688    }
1689
1690    #[test]
1691    fn a_variable_length_array_walks_its_pages_where_the_declaration_stands() {
1692        let (mut func, allocation, mut names, stack) = growing(RAX);
1693        let base = Layout::new(&SYSV, REGS);
1694        let layout = Layout { leaf: false, grows: true, ..base };
1695        let probing = Probing { probe: &PROBE, branch: &BRANCH, scratch: [R10, R11] };
1696        let convention = Convention { probe: Some(probing), ..Convention::new(&SYSV, &FRAME) };
1697        let lines = under(&mut func, &allocation, &layout, &stack, convention, &mut names);
1698
1699        // The whole listing, because what the walk is cannot be read off the instructions alone.
1700        // The one subtraction the lowering wrote is gone and four blocks stand where its block was:
1701        // where the stack pointer is going, the step, the page the step landed on, and the rest of
1702        // what the block was doing with the stack pointer put back where it was going.
1703        assert_eq!(
1704            lines,
1705            [
1706                "mfunc @f {",
1707                "block0:",
1708                "x64.push_64 $rbp",
1709                "$rbp = x64.mov_rr_64 $rsp",
1710                "$r10 = x64.mov_rr_64 $rsp",
1711                "$r10 = x64.sub_rr_64 $r10, $rax, block1",
1712                "block1:",
1713                "$rsp = x64.sub_ri_64 $rsp, 4096",
1714                "$r11 = x64.cmp_set_a_64 $rsp, $r10",
1715                "x64.br_cond_8 $r11, block2, block3",
1716                "block2:",
1717                "x64.or_mi_8 [$rsp], 0, block1",
1718                "block3:",
1719                "$rsp = x64.mov_rr_64 $r10",
1720                "x64.nop",
1721                "$rsp = x64.mov_rr_64 $rbp",
1722                "$rbp = x64.pop_64",
1723                "x64.ret",
1724                "}",
1725            ]
1726        );
1727    }
1728
1729    #[test]
1730    fn the_walk_keeps_the_register_the_count_arrived_in() {
1731        let (mut func, allocation, mut names, stack) = growing(R10);
1732        let base = Layout::new(&SYSV, REGS);
1733        let layout = Layout { leaf: false, grows: true, ..base };
1734        let probing = Probing { probe: &PROBE, branch: &BRANCH, scratch: [R10, R11] };
1735        let convention = Convention { probe: Some(probing), ..Convention::new(&SYSV, &FRAME) };
1736        let lines = under(&mut func, &allocation, &layout, &stack, convention, &mut names);
1737
1738        // The count is in the first of the two registers the walk was given, which is where a
1739        // reload the rewriter wrote would have put it, so the limit goes in the other one and the
1740        // comparison writes the first one back only once the count has been read for the last time.
1741        let added = added(&lines);
1742        assert!(added.contains(&"$r11 = x64.mov_rr_64 $rsp"), "{added:?}");
1743        assert!(added.contains(&"$r11 = x64.sub_rr_64 $r11, $r10, block1"), "{added:?}");
1744        assert!(added.contains(&"$r10 = x64.cmp_set_a_64 $rsp, $r11"), "{added:?}");
1745    }
1746
1747    #[test]
1748    fn a_variable_length_array_takes_its_bytes_in_one_subtraction_when_nothing_asked() {
1749        let (mut func, allocation, mut names, stack) = growing(RAX);
1750        let base = Layout::new(&SYSV, REGS);
1751        let layout = Layout { leaf: false, grows: true, ..base };
1752        let convention = Convention::new(&SYSV, &FRAME);
1753        let lines = under(&mut func, &allocation, &layout, &stack, convention, &mut names);
1754
1755        // The instruction the lowering wrote, where it wrote it, and one block still.
1756        assert!(lines.contains(&"$rsp = x64.sub_rr_64 $rsp, $rax".to_owned()), "{lines:?}");
1757        assert_eq!(lines.iter().filter(|line| line.starts_with("block")).count(), 1, "{lines:?}");
1758    }
1759
1760    #[test]
1761    fn a_probing_prologue_deeper_than_that_walks_the_pages_in_a_loop() {
1762        let (mut func, allocation, mut names) = pressure(&SYSV, 2, 4);
1763        let locals = [Local { size: 100_000, align: 16 }];
1764        let base = Layout::new(&SYSV, REGS);
1765        let layout = Layout { leaf: false, locals: &locals, ..base };
1766        let probing = Probing { probe: &PROBE, branch: &BRANCH, scratch: [R10, R11] };
1767        let lines = with_probing(&mut func, &allocation, &layout, Some(probing), &mut names);
1768
1769        // Twenty-four pages, which is more than a straight line is worth, so the prologue works out
1770        // where it is going first and then walks there. The whole listing rather than the added
1771        // lines, because what matters as much as the instructions is that the two blocks the walk
1772        // is made of come in front of the block the function began with: the body the allocator
1773        // filled is block2 here and it was block0 before this ran.
1774        assert_eq!(
1775            lines,
1776            [
1777                "mfunc @f {",
1778                "block0:",
1779                "$r10 = x64.lea_64 [$rsp - 98304], block1",
1780                "block1:",
1781                "$rsp = x64.sub_ri_64 $rsp, 4096",
1782                "x64.or_mi_8 [$rsp], 0",
1783                "$r11 = x64.cmp_set_ne_64 $rsp, $r10",
1784                "x64.br_cond_8 $r11, block1, block2",
1785                "block2:",
1786                "$rsp = x64.sub_ri_64 $rsp, 1704",
1787                "$rax = x64.nop",
1788                "$rcx = x64.nop",
1789                "x64.nop $rax",
1790                "x64.nop $rcx",
1791                "$rsp = x64.add_ri_64 $rsp, 100008",
1792                "x64.ret",
1793                "}",
1794            ]
1795        );
1796    }
1797
1798    #[test]
1799    fn a_large_frame_on_a_platform_with_a_routine_for_its_pages_calls_the_routine() {
1800        let (mut func, allocation, mut names) = pressure(&WIN64, 2, 4);
1801        let locals = [Local { size: 100_000, align: 16 }];
1802        let base = Layout::new(&WIN64, REGS);
1803        let layout = Layout { leaf: false, locals: &locals, ..base };
1804        let lines = written(&mut func, &allocation, &layout, &mut names);
1805
1806        // Nothing asked for this on the command line, which is the point: Windows commits a stack
1807        // by having the pages touched in order, so a frame this size has to reach them whatever the
1808        // flags said. The size goes in the register the platform names, the routine touches every
1809        // page down to there, and the frame is taken afterwards, because the routine comes back
1810        // having moved nothing. What the epilogue gives back is what the register was given, which
1811        // is the one thing worth tying together here.
1812        let added = added(&lines);
1813        assert_eq!(added.len(), 5, "{added:?}");
1814        let size = added[0].strip_prefix("$rax = x64.mov_ri_64 ").expect("a size in a register");
1815        assert_eq!(added[1], "x64.call @__chkstk");
1816        assert_eq!(added[2], "$rsp = x64.sub_rr_64 $rsp, $rax");
1817        assert_eq!(added[3], format!("$rsp = x64.add_ri_64 $rsp, {size}"));
1818        assert_eq!(added[4], "x64.ret");
1819    }
1820
1821    #[test]
1822    fn a_frame_of_one_page_calls_nothing_on_that_platform_either() {
1823        let (mut func, allocation, mut names) = pressure(&WIN64, 2, 4);
1824        let locals = [Local { size: 4000, align: 16 }];
1825        let base = Layout::new(&WIN64, REGS);
1826        let layout = Layout { leaf: false, locals: &locals, ..base };
1827        let lines = written(&mut func, &allocation, &layout, &mut names);
1828
1829        // The same reason a frame of one page is taken in one subtraction under the flag. The far
1830        // end of such a frame is inside the page below the stack pointer, and touching that page is
1831        // what the function does on its way to using the frame at all, so there is nothing for a
1832        // routine to do and a call to it would be a call in every function that declares an array.
1833        let added = added(&lines);
1834        assert!(added.iter().all(|line| !line.contains("chkstk")), "{added:?}");
1835        assert_eq!(added.len(), 3, "{added:?}");
1836    }
1837
1838    #[test]
1839    fn a_windows_prologue_points_its_frame_pointer_at_the_frame_once_the_frame_is_whole() {
1840        let (mut func, allocation, mut names) = pressure(&WIN64, 4, 2);
1841        let base = Layout::new(&WIN64, REGS);
1842        let layout = Layout { frame_pointer: true, ..base };
1843        let lines = written(&mut func, &allocation, &layout, &mut names);
1844
1845        // The other order, which is what every other platform here writes, has no unwind record on
1846        // this one: the record counts its slots from where the stack pointer ends the prologue and
1847        // gets there by taking a constant off the frame pointer, so a register pushed after the
1848        // pointer was established sits below the place the record counts from. Pushing first and
1849        // pointing last is the order that has a record, and it leaves the pointer holding a copy of
1850        // the stack pointer, so the spills stay where they were and the epilogue counts the frame
1851        // back off the pointer rather than moving the pointer into the stack pointer.
1852        assert_eq!(
1853            added(&lines),
1854            [
1855                "x64.push_64 $rbp",
1856                "$rsp = x64.sub_ri_64 $rsp, 16",
1857                "$rbp = x64.mov_rr_64 $rsp",
1858                "x64.mov_mr_64 $rdx, [$rsp]",
1859                "x64.mov_mr_64 $rdx, [$rsp + 8]",
1860                "$rdx = x64.mov_rm_64 [$rsp]",
1861                "$rdx = x64.mov_rm_64 [$rsp + 8]",
1862                "$rsp = x64.lea_64 [$rbp + 16]",
1863                "$rbp = x64.pop_64",
1864                "x64.ret",
1865            ]
1866        );
1867    }
1868
1869    #[test]
1870    fn a_windows_prologue_that_saves_registers_too_pushes_all_of_them_before_the_frame() {
1871        let (mut func, allocation, mut names) = pressure(&WIN64, 9, 8);
1872        let base = Layout::new(&WIN64, REGS);
1873        let layout = Layout { leaf: false, frame_pointer: true, ..base };
1874        let lines = written(&mut func, &allocation, &layout, &mut names);
1875
1876        // The shape that made the order necessary. All three pushes are above the frame, so every
1877        // one of them has a row the record can write, and the pointer is the last thing the
1878        // prologue does. Forty eight bytes is the thirty two every Windows caller reserves below a
1879        // call, eight for the one value that did not fit in a register, and eight that put the
1880        // stack pointer back where a call wants it given three pushes and the return address.
1881        assert_eq!(
1882            added(&lines),
1883            [
1884                "x64.push_64 $rbp",
1885                "x64.push_64 $rbx",
1886                "x64.push_64 $rsi",
1887                "$rsp = x64.sub_ri_64 $rsp, 48",
1888                "$rbp = x64.mov_rr_64 $rsp",
1889                "x64.mov_mr_64 $rsi, [$rsp + 32]",
1890                "$rsi = x64.mov_rm_64 [$rsp + 32]",
1891                "$rsp = x64.lea_64 [$rbp + 48]",
1892                "$rsi = x64.pop_64",
1893                "$rbx = x64.pop_64",
1894                "$rbp = x64.pop_64",
1895                "x64.ret",
1896            ]
1897        );
1898    }
1899
1900    #[test]
1901    fn a_vector_register_a_windows_call_preserves_is_stored_and_read_back() {
1902        let mut names = Interner::new();
1903        let mut func = Func::new(names.intern("f"));
1904        let opcode = Opcode::new(names.intern("x64.nop"));
1905        let block = func.create_block();
1906        // An instruction that writes one of the vector registers Windows preserves, which is what
1907        // a rule for something that has to use it produces.
1908        func.build(block, opcode).operand(Operand::write(Reg::physical(xmm(6)), XMM)).finish();
1909        let allocation = rucc_regalloc::run(&mut func, &env(&WIN64, 4), "test", true);
1910        let lines = written(&mut func, &allocation, &Layout::new(&WIN64, REGS), &mut names);
1911
1912        // No machine here pushes a vector register, so it is stored into the frame rather than
1913        // pushed, and the frame has to be taken before there is anywhere to put it.
1914        assert_eq!(
1915            added(&lines),
1916            [
1917                "$rsp = x64.sub_ri_64 $rsp, 24",
1918                "x64.movaps_mr $xmm6, [$rsp]",
1919                "$xmm6 = x64.movaps_rm [$rsp]",
1920                "$rsp = x64.add_ri_64 $rsp, 24",
1921                "x64.ret",
1922            ]
1923        );
1924    }
1925}