Skip to main content

rucc_codegen/
abi.rs

1//! Where a function's arguments already are when it starts running, and where a call puts its own.
2//!
3//! Design: `spec/12-abi-and-runtime.md`.
4//!
5//! This is the one part of the calling convention that is not a lowering rule, and it is worth
6//! saying why, because everything else in this crate is. A rule matches a term and rewrites it,
7//! and which register the third argument arrives in is not a fact about any term: it depends on
8//! the argument's position and on the classification of every argument before it. A pattern has
9//! nowhere to put that. So the arguments are built here, by hand, out of what the convention
10//! says, the same way [`crate::finish`] builds a prologue.
11//!
12//! The classification itself is not here either. `rucc-lower` has already run it by the time a
13//! function reaches this crate, which is why the parameters read here are nearly all plain
14//! scalars: an aggregate has been split into the pieces it travels in, and a return through memory
15//! is an ordinary pointer parameter in front of the rest. What is left for this is the step after
16//! classification, from how a value travels to which register it is actually in, which is
17//! [`rucc_target::Places`].
18//!
19//! The one parameter that is not a scalar is an aggregate the classification put in the argument
20//! area whole, which is [`rucc_ir::Abi::ByVal`]. The IR calls it a pointer, because a pointer is
21//! what an instruction reading it has to have, and the convention says the bytes travel and the
22//! pointer does not. So this is the one place that reads what the classification said rather than
23//! only the type, on both sides of the call, and the two sides are the two halves of one copy.
24//!
25//! # What it writes
26//!
27//! One `x64.arg_val_*` per parameter that arrived in a register, at the top of the entry block,
28//! each defining a fresh register constrained to the one the argument arrived in. They encode to
29//! nothing. The point of them is that a parameter has to be defined somewhere for the allocator to
30//! have anything to move, and the entry block cannot define it as a block parameter: there is no
31//! edge into the entry block for the move to go on, which is what `rucc_regalloc::rewrite` asserts.
32//!
33//! What the allocator does with them is the whole of the argument sequence. A parameter that is
34//! read where it arrived costs nothing, and one that is not gets a copy, which is the same
35//! bargain the return already makes and is decided by the same code.
36//!
37//! A parameter whose bytes travelled is the exception to all of that. Its bytes are already in
38//! this function, at a place in the caller's argument area the same walk gives, so nothing is
39//! brought in at all: what the parameter is is where they are, and that is one `lea`. It waits on
40//! the frame the way the loads below it do, and for the same reason.
41//!
42//! A parameter past the last register arrived in the caller's memory rather than in a register, so
43//! it is a load and not a pseudo, and it is a real instruction that encodes to real bytes. How far
44//! up the caller's argument area it is is a number [`rucc_target::Places`] answers here, but where
45//! that area is from inside this function is a distance into a frame, and no frame exists until
46//! after allocation. So the load is written with nothing in its displacement, which of the two
47//! registers it reads through is left to be settled too, and both are filled in by [`crate::finish`]
48//! out of [`crate::frame::Frame::incoming`]. That is the same bargain an `alloca` already makes,
49//! for the same reason and in the same two places.
50//!
51//! # A call
52//!
53//! The same reasoning the other way round, and one instruction rather than several. `x64.call`
54//! and `x64.call_reg` are the only opcodes in the description whose operand vector is empty
55//! there, because nothing about a call's operands is the same from one call to the next, so they
56//! are built here: one read per argument constrained to the register the convention passes it in,
57//! one definition for the value that comes back constrained to the register it comes back in, and
58//! one definition per register the convention does not preserve.
59//!
60//! A call through an address has one operand more, which is the address, and it is the one
61//! operand of a call that is a fact about the instruction rather than about the signature. It
62//! goes in front of the arguments, because the assembler has to find it and an index into a
63//! vector whose length depends on the convention is not a way of finding anything.
64//!
65//! Those last ones are the clobbers, and they are the whole of what the allocator has to know
66//! about a call besides where the values go. Each is a definition of the physical register itself
67//! rather than of a value, since there is no value: it says the register is written here, which
68//! is exactly what stops the allocator from leaving something in one across the call. A register
69//! an argument or the result already names is not repeated, because naming it once already blocks
70//! it for the length of the instruction, which is all a clobber does.
71//!
72//! An argument past the last register the convention has for it is a store into the outgoing area
73//! rather than an operand of the call, written in front of the call in the same block. Where that
74//! area is does not have to wait for the frame the way the incoming one does, because the outgoing
75//! area is at the bottom of the frame and the bottom of the frame is where the stack pointer is:
76//! that is the whole reason the frame puts it there, since it is where the callee will look. So the
77//! offset [`rucc_target::Places`] gives back is the offset the store is written with.
78//!
79//! An object passed by value in memory is the same thing again and a copy rather than a store. The
80//! caller owes the callee a copy it is free to write to, which is what makes a C call by value
81//! different from passing a pointer the callee must not keep, and the argument area is where the
82//! convention says that copy goes. So the bytes are read out of the object and written into the
83//! area a word at a time, in front of the call, with the words chosen by the same function that
84//! chooses them for a `memcpy`. An object with more words than that unrolls to is turned down,
85//! because the copy it wants is a call to the runtime and one call cannot be built inside another.
86//!
87//! The call still reports how many bytes it needed, because the frame reserves as many as the
88//! widest call in the function asked for and cannot know that until every call has been seen.
89
90use rucc_base::{Interner, Symbol};
91use rucc_ir::{Abi, Param, Type};
92use rucc_mir as mir;
93use rucc_target::x86_64;
94use rucc_target::{CallRegs, Constraint, PhysReg, Places, RegClass, Where};
95
96use crate::varargs::Area;
97
98/// Why a parameter could not be brought in.
99#[derive(Debug, Clone, Copy, PartialEq, Eq)]
100pub enum Missing {
101    /// It travels on the x87 stack, which is a `long double` and nothing else. That stack is a
102    /// third register file, it is not one the allocator has, and no instruction in the
103    /// description touches it.
104    OnX87,
105    /// It is a float passed to a callee that takes arguments beyond the ones its signature names,
106    /// on a convention that puts such a float in a vector register and in the general purpose
107    /// register at the same position at once. Which arguments are the ones beyond the signature is
108    /// what decides whether the second copy is needed, and a call does not carry that yet.
109    InBothFiles,
110    /// It is a width no pseudo covers, which is anything a machine register does not hold.
111    Width,
112    /// It is a value that comes back in more registers than the convention returns in. A structure
113    /// of at most sixteen bytes comes back in up to two, which is as many as SysV has, and a
114    /// convention with fewer of them returns such a structure through a hidden pointer instead. So
115    /// this is what a signature the classification did not produce would get.
116    NoRoom,
117    /// It is an object whose bytes travel in the argument area and there are more of them than a
118    /// copy a word at a time is worth. Such a copy belongs in a call to the runtime, and the place
119    /// this is decided is in the middle of building a call, where another one cannot go.
120    TooBig,
121}
122
123impl Missing {
124    /// What it says when a function could not be compiled because of it.
125    ///
126    /// Worded so that it reads the same about a value arriving and a value being passed, since
127    /// the two are the same fact seen from the two ends of one call.
128    #[must_use]
129    pub fn why(self) -> &'static str {
130        match self {
131            Missing::OnX87 => "is on the x87 stack",
132            Missing::InBothFiles => "is a float passed to a variadic callee on this convention",
133            Missing::Width => "is a width no argument register holds",
134            Missing::NoRoom => "takes more registers than this convention has for it",
135            Missing::TooBig => "is more bytes than a copy into the argument area unrolls to",
136        }
137    }
138}
139
140/// Which register file a value of that type travels in.
141///
142/// The whole of what the two files mean to this module. A float is in the vector one and
143/// everything else is in the general purpose one, which is what both of this machine's conventions
144/// say. A `long double` is in neither and what its register holds is an address, which is a general
145/// purpose value like every other address, so it answers with the other file rather than with the
146/// one its type would suggest.
147fn class_of(ty: Type, conv: &CallRegs) -> RegClass {
148    if ty.is_float() && !on_the_stack(ty) { conv.sse_class } else { conv.int_class }
149}
150
151/// How many bytes of the argument area a value in the vector file takes.
152///
153/// Its own width, lanes included, which is what [`Places::float`] wants and is a number that only
154/// matters to a value the registers ran out before. Everything the machine computes in is a word
155/// or narrower and takes a word either way. The `_Float128` is the one that is not: two words, and
156/// aligned to two words, which is the difference between the argument behind it being placed after
157/// it and being placed on top of half of it.
158pub(crate) fn float_bytes(ty: Type) -> u32 {
159    ty.bits().div_ceil(8).saturating_mul(ty.lanes())
160}
161
162/// Whether a value of that type travels as bytes in the argument area because of what it is.
163///
164/// One type does, and it is the `long double`. SysV classifies it X87 and X87UP, which is the
165/// classification that means memory, so it goes where a structure the classification put in memory
166/// goes and what the two ends pass is the address of the bytes. That is not a decision about
167/// registers running out: a `long double` travels in the argument area when it is the only argument
168/// there is.
169///
170/// Sixteen bytes aligned to sixteen, which is what the psABI says the type takes and is the same
171/// number [`crate::lower`] gives one in the frame, so a value being passed and a value being worked
172/// on are the same shape of object in two places.
173#[must_use]
174pub fn on_the_stack(ty: Type) -> bool {
175    ty.is_float() && ty.is_scalar() && ty.bits() == 80
176}
177
178/// How much room one takes in the argument area, as a size and an alignment.
179pub(crate) const X87_AREA: (u32, u32) = (16, 16);
180
181/// Why a value of that type cannot travel at all, or nothing if it can.
182///
183/// The width question and the file question in one place, so that the two ends of a call give the
184/// same answer about the same type, and so that a `return` this cannot make says the same thing
185/// about a type as the call that would have received it.
186///
187/// A `long double` is not one of them any more when it is an argument, since an argument of that
188/// type travels as bytes and [`on_the_stack`] is what says so before this is asked. What is left
189/// here is the value that comes back, because coming back is the one direction where it is not
190/// bytes: it arrives in `st(0)`, which is a register file this cannot name.
191#[must_use]
192pub fn refuses(ty: Type) -> Option<Missing> {
193    if head_of(ty).is_some() {
194        return None;
195    }
196    // A `long double` is the one type here that is in neither of the two files. Saying so is worth
197    // more than calling it a width, because eighty bits is a width this machine computes in and
198    // the file it computes in is what actually stands in the way.
199    if on_the_stack(ty) {
200        return Some(Missing::OnX87);
201    }
202    Some(Missing::Width)
203}
204
205/// What a function's parameters came to.
206#[derive(Debug, Default, Clone, PartialEq, Eq)]
207pub struct Arrived {
208    /// The register each parameter is in, in the order the parameters were given, so the caller can
209    /// bind each IR parameter to the one at its position.
210    pub regs: Vec<mir::Reg>,
211    /// The loads that read a parameter out of the caller's argument area, and how far up that area
212    /// each of them reads.
213    ///
214    /// Empty for almost every function, because almost every function has few enough parameters to
215    /// have been handed all of them in registers. The distance is from the bottom of the caller's
216    /// argument area, which is somewhere [`crate::finish`] works out and this cannot.
217    pub stack: Vec<(mir::Inst, u32)>,
218    /// How many general purpose argument registers the parameters took, and how many vector ones.
219    ///
220    /// Nothing about an ordinary function needs this. A variadic one does: the first argument its
221    /// signature does not name is the one after the last it does, so where each of the two walks
222    /// stopped is where `va_start` has to say the next argument begins.
223    pub took: (usize, usize),
224    /// How many bytes of the caller's argument area the parameters took, which is where the first
225    /// argument the signature does not name begins for the same reason.
226    pub used: u32,
227    /// The argument registers left over for the arguments the signature does not name, as the
228    /// register each was bound into and how far up the save area its slot is.
229    ///
230    /// Empty unless a save area was asked for. The ones a named parameter took are not here,
231    /// because their slots are behind where `va_start` sets the two offsets and nothing ever reads
232    /// them, so writing them would be fourteen stores where six are wanted.
233    pub spare: Vec<(mir::Reg, RegClass, u32)>,
234}
235
236/// Binds a function's parameters to where the convention says they arrive.
237///
238/// # Errors
239///
240/// The first parameter this cannot bring in, and why. A function with one is reported rather
241/// than compiled, because the alternative is a function that reads an argument from wherever the
242/// last one happened to leave a register.
243pub fn entry(
244    out: &mut mir::Func,
245    block: mir::Block,
246    params: &[Param],
247    conv: &CallRegs,
248    names: &mut Interner,
249    save: Option<Area>,
250) -> Result<Arrived, (usize, Missing)> {
251    // Where everything is, worked out before anything is written, both so that a parameter this
252    // cannot bring in stops the function before half of one is built and so that the two loops
253    // below can be two loops. Asking for the place of a parameter that cannot be brought in is
254    // still done, because every place after it depends on it and a reader stepping through this
255    // should see the same numbers a working version would.
256    let mut places = Places::new(conv);
257    let mut where_from = Vec::with_capacity(params.len());
258    for (index, &Param { ty, abi }) in params.iter().enumerate() {
259        // A structure the classification put in the argument area arrived as bytes, and the
260        // parameter the IR sees is a pointer to them. So there is nothing to bring in: the bytes
261        // are already in this function's frame, and what the pointer holds is where they are.
262        if let Abi::ByVal { size, align } = abi {
263            let size = u32::try_from(size).map_err(|_| (index, Missing::TooBig))?;
264            where_from.push((ty, places.on_stack(size, align), abi));
265            continue;
266        }
267        // And an eighty bit float, which arrives the same way for the same reason and is told
268        // apart only by the classification having said nothing about it: the front end passes it
269        // as a value of its own type, and it is this that knows the type is one that travels as
270        // bytes. What arrives is the address of those bytes, which is what an object in the
271        // argument area always hands over.
272        if on_the_stack(ty) {
273            let (size, align) = X87_AREA;
274            where_from.push((
275                ty,
276                places.on_stack(size, align),
277                Abi::ByVal { size: size.into(), align },
278            ));
279            continue;
280        }
281        let at = if ty.is_float() { places.float(float_bytes(ty)) } else { places.integer() };
282        if let Some(missing) = refuses(ty) {
283            return Err((index, missing));
284        }
285        where_from.push((ty, at, abi));
286    }
287    let mut arrived = Arrived {
288        regs: Vec::with_capacity(params.len()),
289        took: (places.integers(), places.floats()),
290        used: places.size(),
291        ..Arrived::default()
292    };
293
294    // Every pseudo first and everything else after, which is not a preference. A pseudo says a
295    // register holds an argument and defines nothing before it, so as far as the allocator can see
296    // the register was dead until then and is free to be used as a scratch. That is true of a
297    // register no pseudo has named yet, and it stops being true the moment one does. Anything that
298    // needs a scratch has to come after all of them, and a load out of the caller's stack needs one
299    // for the value it loads.
300    for (index, &(ty, at, _)) in where_from.iter().enumerate() {
301        let class = class_of(ty, conv);
302        let reg = out.new_vreg(class);
303        arrived.regs.push(reg);
304        let Where::Reg(arrived_in) = at else { continue };
305        let head = head_of(ty).ok_or((index, Missing::Width))?;
306        let opcode = mir::Opcode::new(names.intern(head));
307        let operand = mir::Operand::write(reg, class).with(Constraint::Fixed(arrived_in));
308        out.build(block, opcode).operand(operand).finish();
309    }
310    if let Some(area) = save {
311        arrived.spare = spare(out, block, conv, names, area, arrived.took);
312    }
313
314    let lea = format!("{}{}", crate::lower::PREFIX, x86_64::FRAME.lea);
315    // The stack pointer is written down as the register to read through because it is the one that
316    // reaches the caller's stack in almost every function, and a realigned frame is the exception
317    // that [`crate::finish`] rewrites. Putting something here rather than nothing keeps the
318    // instruction printable and verifiable in between.
319    for (index, &(ty, at, abi)) in where_from.iter().enumerate() {
320        let Where::Stack(up) = at else { continue };
321        let class = class_of(ty, conv);
322        // The bytes of an object that travelled as bytes are read by whatever reads the parameter,
323        // and what the parameter is is their address, so this takes the address rather than a
324        // value out of it. Everything else about it is the load's, including the two fields
325        // [`crate::finish`] fills in, because where the caller's argument area is is the same
326        // question for both.
327        let name = match abi {
328            Abi::ByVal { .. } => lea.as_str(),
329            _ => load_of(ty).ok_or((index, Missing::Width))?,
330        };
331        let opcode = mir::Opcode::new(names.intern(name));
332        let sp = mir::Operand::read(mir::Reg::physical(conv.stack_pointer), conv.int_class);
333        let made =
334            out.build(block, opcode).def(arrived.regs[index], class).mem(mir::Mem::at(sp)).finish();
335        arrived.stack.push((made, up));
336    }
337    Ok(arrived)
338}
339
340/// Binds the argument registers no parameter the signature names took, which are the ones the
341/// arguments it does not name arrived in.
342///
343/// One pseudo each and nothing else, for the reason the loop above them gives: what these do is say
344/// the register holds something, and the stores that put it in the save area are written by
345/// [`crate::lower`] once it has an address to store to, which is after every pseudo in the block.
346fn spare(
347    out: &mut mir::Func,
348    block: mir::Block,
349    conv: &CallRegs,
350    names: &mut Interner,
351    area: Area,
352    took: (usize, usize),
353) -> Vec<(mir::Reg, RegClass, u32)> {
354    let word = Type::int(64);
355    let double = Type::float(rucc_ir::Float::F64);
356    let files = [(conv.int_args, took.0, word, false), (conv.sse_args, took.1, double, true)];
357    let mut spare = Vec::new();
358    for (regs, taken, ty, float) in files {
359        let Some(head) = head_of(ty) else { continue };
360        let class = class_of(ty, conv);
361        for (index, &arrived_in) in regs.iter().enumerate().skip(taken) {
362            let reg = out.new_vreg(class);
363            let opcode = mir::Opcode::new(names.intern(head));
364            let operand = mir::Operand::write(reg, class).with(Constraint::Fixed(arrived_in));
365            out.build(block, opcode).operand(operand).finish();
366            let at = area.starts_at(float) + area.stride(float) * u32::try_from(index).unwrap_or(0);
367            spare.push((reg, class, at));
368        }
369    }
370    spare
371}
372
373/// What the instruction that calls a name is called.
374///
375/// Here rather than in a rule for the same reason the arguments are: a rule pattern sees one term
376/// and a call's operands are whatever the signature made them, so no pattern could name them.
377pub const CALL: &str = "x64.call";
378
379/// What the instruction that calls an address in a register is called.
380///
381/// A different instruction rather than the same one with a different operand, which is what the
382/// machine says too: one carries the distance to somewhere in the program and takes a relocation,
383/// and the other carries the register the address is in and takes none. Sharing an opcode would
384/// mean an instruction whose bytes depend on whether a field beside it happens to be set.
385pub const CALL_REG: &str = "x64.call_reg";
386
387/// One value a call passes.
388#[derive(Debug, Clone, Copy, PartialEq, Eq)]
389pub struct Passing {
390    /// The type it travels as, which for an object travelling as bytes is the pointer's rather
391    /// than the object's, because the pointer is what the machine IR has.
392    pub ty: Type,
393    /// The register holding it, or holding its address when the bytes are what travel.
394    pub reg: mir::Reg,
395    /// What the classification asked of it. The one thing read here is whether the object behind
396    /// the pointer is the argument, since everything else it can say is about a value that is
397    /// already in a register in the form it travels in.
398    pub abi: Abi,
399}
400
401/// What one call came to.
402#[derive(Debug, Clone, PartialEq, Eq)]
403pub struct Made {
404    /// The registers the value came back in, in the order the signature returns them, which is
405    /// empty for a call that gives nothing back and holds two for a structure that comes back in a
406    /// pair. Which register each of them is is the classification's answer and is worked out here
407    /// rather than in a table, for the reason the second half of [`Calling::returns`] gives.
408    pub results: Vec<mir::Reg>,
409    /// How many bytes below the stack pointer this call needs for the arguments it passes there.
410    ///
411    /// Not always zero for a call that passes everything in registers: a Windows caller reserves
412    /// thirty two bytes for the callee to spill its register arguments into whether it uses them
413    /// or not, and that reservation is this.
414    pub outgoing: u32,
415}
416
417/// Which of a call's values could not be passed, and why.
418#[derive(Debug, Clone, Copy, PartialEq, Eq)]
419pub struct Refused {
420    /// Its position among the arguments, or `None` for the value that comes back.
421    pub argument: Option<usize>,
422    /// What is wrong with where it travels.
423    pub missing: Missing,
424}
425
426/// What a call goes to.
427///
428/// The whole of the difference between the two calls. Everything else about them, which is what
429/// they pass and what comes back and which registers they destroy, is the signature's answer and
430/// is the same answer either way.
431#[derive(Debug, Clone, Copy, PartialEq, Eq)]
432pub enum Callee {
433    /// A name, which the linker resolves.
434    Named(Symbol),
435    /// An address in a register, which nothing resolves because there is nothing to resolve: the
436    /// value is not known until the program runs.
437    ///
438    /// The register is unconstrained, and it has to be, because every register the convention
439    /// does not preserve is one this instruction writes and every register an argument travels in
440    /// is spoken for. What is left is the registers the callee has to put back, which is where
441    /// the allocator will put the address, and it is the right answer for the same reason it is
442    /// the only one.
443    Through(mir::Reg),
444}
445
446/// One call, as everything about it that is not the function it is being built into.
447#[derive(Debug, Clone, Copy)]
448pub struct Calling<'a> {
449    /// What it calls.
450    pub callee: Callee,
451    /// What it passes, in the order the signature holds them, which is the order the convention
452    /// places them in.
453    pub args: &'a [Passing],
454    /// What comes back, which is empty for a call that gives nothing back, one type for a value,
455    /// and two for a structure small enough to come back in a pair of registers.
456    ///
457    /// A pair is placed here rather than named by a rule for the reason the arguments are: which
458    /// register each half goes in depends on the halves before it, since the two files are walked
459    /// separately, and a pattern over a term cannot see them.
460    pub returns: &'a [Type],
461    /// Whether the callee takes arguments beyond the ones its signature names, which is what says
462    /// whether it reads the count of vector registers the call passed arguments in.
463    pub variadic: bool,
464}
465
466/// Builds one call: what it passes, what comes back, and what it destroys.
467///
468/// # Errors
469///
470/// The first value this cannot pass, and why, before anything is written. A call with one is
471/// reported rather than compiled, because the alternative is a call that leaves an argument
472/// wherever the last one happened to put a register.
473///
474/// # Panics
475///
476/// If a call passes two gigabytes of arguments on the stack, which is a distance no offset in a
477/// frame can hold and a call no program makes.
478pub fn call(
479    out: &mut mir::Func,
480    block: mir::Block,
481    made: &Calling<'_>,
482    conv: &CallRegs,
483    names: &mut Interner,
484) -> Result<Made, Refused> {
485    let &Calling { callee, args, returns, variadic } = made;
486    // Where everything goes, worked out before anything is built, so that a call this cannot make
487    // leaves no half of one behind.
488    let mut places = Places::new(conv);
489    let mut passed = Vec::with_capacity(args.len());
490    // The ones with no register left for them, as the store each of them becomes and how far up
491    // the outgoing area it writes. Almost always empty.
492    let mut on_stack = Vec::new();
493    // How many of them went in vector registers, which is what a SysV variadic callee is told.
494    let mut vectors = 0u32;
495    // The ones whose bytes travel rather than their address, as the register that address is in,
496    // how far up the outgoing area they go and which words the copy is made of. Almost always
497    // empty too, and never at the same time as a register: an object in the argument area is in
498    // the argument area whatever is left of the register files.
499    let mut as_bytes = Vec::new();
500    for (index, &Passing { ty, reg, abi }) in args.iter().enumerate() {
501        let refused = |missing| Refused { argument: Some(index), missing };
502        // An eighty bit float is bytes in the argument area whatever the classification said, for
503        // the reason [`on_the_stack`] gives, and the register holding it holds their address. So it
504        // joins the objects below rather than being a case of its own, and the copy it becomes is
505        // the copy any other sixteen byte object gets.
506        let abi = match abi {
507            _ if on_the_stack(ty) => {
508                let (size, align) = X87_AREA;
509                Abi::ByVal { size: size.into(), align }
510            }
511            abi => abi,
512        };
513        if let Abi::ByVal { size, align } = abi {
514            let size = u32::try_from(size).map_err(|_| refused(Missing::TooBig))?;
515            let Where::Stack(up) = places.on_stack(size, align) else {
516                unreachable!("an object in the argument area is in the argument area")
517            };
518            let plan = crate::expand::plan(u64::from(size), align, conv.word)
519                .ok_or(refused(Missing::TooBig))?;
520            as_bytes.push((reg, up, plan));
521            continue;
522        }
523        let at = if ty.is_float() { places.float(float_bytes(ty)) } else { places.integer() };
524        if let Some(missing) = refuses(ty) {
525            return Err(refused(missing));
526        }
527        // Windows passes a float to a variadic callee in the vector register and in the general
528        // purpose register at the same position, both at once, because the callee has no
529        // prototype to tell it which file to look in. Doing that needs to know which arguments are
530        // the ones the signature does not name, and a call carries whether the callee is variadic
531        // rather than how many arguments it names, so this is turned down rather than passed in
532        // one file and read from the other.
533        if ty.is_float() && variadic && conv.shared_positions {
534            return Err(refused(Missing::InBothFiles));
535        }
536        let class = class_of(ty, conv);
537        match at {
538            Where::Reg(at) => {
539                // Only a register counts, because the count is of registers. An argument that went
540                // to memory is one the callee reads from memory whatever this says.
541                if class == conv.sse_class {
542                    vectors += 1;
543                }
544                passed.push((reg, at, class));
545            }
546            Where::Stack(up) => {
547                let store = store_of(ty).ok_or(refused(Missing::Width))?;
548                on_stack.push((reg, class, names.intern(store), up));
549            }
550        }
551    }
552    // A `long double` comes back in `st(0)`, which is not a register in either file and not one
553    // this call can be said to write. So nothing is placed for it and nothing is constrained, and
554    // the call gives back no register at all: what takes the value off that stack is the `fstp`
555    // [`crate::lower`] writes straight after the call, which is the same shape every other use of
556    // the x87 stack is written in. Only on its own, because a value that comes back beside another
557    // one comes back in a pair of registers and there is no pair with that stack in it.
558    let comes_back = if matches!(returns, [ty] if on_the_stack(*ty)) {
559        Vec::new()
560    } else {
561        places_back(returns, conv)?
562    };
563
564    // A variadic callee on SysV reads how many vector registers the call passed arguments in and
565    // skips saving them when the answer is none, which is what makes `printf` with no floating
566    // point argument cheap. It is an obligation rather than an optimization: leaving whatever was
567    // in the register there makes the callee save a register file it was not given, and a count
568    // that is too low makes it read an argument out of a register nothing put one in.
569    let counted = if variadic { conv.vector_count } else { None };
570
571    // The arguments that go to memory go there now, in front of the call and after everything this
572    // could have refused, so that a call it cannot make leaves no store behind either. The offset
573    // is written straight in rather than left for [`crate::finish`]: the outgoing area is at the
574    // bottom of the frame because that is where the callee looks for it, and the bottom of the
575    // frame is where the stack pointer already is.
576    for (reg, class, store, up) in on_stack {
577        let sp = mir::Operand::read(mir::Reg::physical(conv.stack_pointer), conv.int_class);
578        let up = i32::try_from(up).expect("an argument area under two gigabytes");
579        let build = out.build(block, mir::Opcode::new(store));
580        build.uses(reg, class).mem(mir::Mem::at(sp).plus(up)).finish();
581    }
582
583    // And the objects whose bytes go there, as a load and a store for each word of each of them.
584    // This is the copy the caller owes a callee that takes a structure by value: the callee is
585    // free to write to what it was handed, so what it was handed cannot be the caller's own copy,
586    // and the argument area is where the convention says the caller's copy goes. The words are the
587    // same words `crate::expand` would have chosen for a `memcpy` of the same block, because they
588    // are chosen by the same function.
589    for (from, up, plan) in as_bytes {
590        let up = i32::try_from(up).expect("an argument area under two gigabytes");
591        for (at, width) in plan {
592            let ty = Type::int(width * 8);
593            let at = i32::try_from(at).expect("an object under two gigabytes");
594            let word = out.new_vreg(conv.int_class);
595            let load = names
596                .intern(load_of(ty).ok_or(Refused { argument: None, missing: Missing::Width })?);
597            let there = mir::Operand::read(from, conv.int_class);
598            let build = out.build(block, mir::Opcode::new(load));
599            build.def(word, conv.int_class).mem(mir::Mem::at(there).plus(at)).finish();
600            let store = names
601                .intern(store_of(ty).ok_or(Refused { argument: None, missing: Missing::Width })?);
602            let sp = mir::Operand::read(mir::Reg::physical(conv.stack_pointer), conv.int_class);
603            let build = out.build(block, mir::Opcode::new(store));
604            build.uses(word, conv.int_class).mem(mir::Mem::at(sp).plus(up + at)).finish();
605        }
606    }
607
608    // The definitions first and the reads after, which is the order every operand vector in the
609    // machine IR is in and the order `rucc_mir::defs` counts.
610    let mut operands = Vec::with_capacity(args.len() + conv.int_order.len() + 2);
611    let results: Vec<mir::Reg> = comes_back
612        .iter()
613        .map(|&(at, class)| {
614            let reg = out.new_vreg(class);
615            operands.push(mir::Operand::write(reg, class).with(Constraint::Fixed(at)));
616            reg
617        })
618        .collect();
619    // One list per file, because a physical register is a number and the class is what says which
620    // file it is a number in. One list would have `xmm0` blocking `rax`.
621    let spoken_for = |class: RegClass| -> Vec<PhysReg> {
622        comes_back
623            .iter()
624            .filter(|&&(_, at)| at == class)
625            .map(|&(reg, _)| reg)
626            .chain(counted.filter(|_| class == conv.int_class))
627            .chain(passed.iter().filter(|&&(_, _, at)| at == class).map(|&(_, reg, _)| reg))
628            .collect()
629    };
630    let named = spoken_for(conv.int_class);
631    for &reg in conv.int_order {
632        if !conv.preserves_int(reg) && !named.contains(&reg) {
633            operands.push(mir::Operand::write(mir::Reg::physical(reg), conv.int_class));
634        }
635    }
636    let named = spoken_for(conv.sse_class);
637    for &reg in conv.sse_order {
638        if !conv.preserves_sse(reg) && !named.contains(&reg) {
639            operands.push(mir::Operand::write(mir::Reg::physical(reg), conv.sse_class));
640        }
641    }
642    // The address in front of the arguments, because a call through one is written with the
643    // register it goes through and nothing in the operand vector is at a place a table could name.
644    // First read is a place that does not depend on the signature, which is what
645    // [`rucc_target::x86_64::Arg::Through`] is written against.
646    if let Callee::Through(reg) = callee {
647        operands.push(mir::Operand::read(reg, conv.int_class));
648    }
649    for (reg, at, class) in passed {
650        operands.push(mir::Operand::read(reg, class).with(Constraint::Fixed(at)));
651    }
652    if let Some(at) = counted {
653        let count = out.new_vreg(conv.int_class);
654        let zero = mir::Opcode::new(names.intern("x64.mov_ri_32"));
655        out.build(block, zero).def(count, conv.int_class).imm(i64::from(vectors)).finish();
656        operands.push(mir::Operand::read(count, conv.int_class).with(Constraint::Fixed(at)));
657    }
658
659    let opcode = mir::Opcode::new(names.intern(match callee {
660        Callee::Named(_) => CALL,
661        Callee::Through(_) => CALL_REG,
662    }));
663    let mut build = out.build(block, opcode);
664    if let Callee::Named(symbol) = callee {
665        build = build.symbol(symbol);
666    }
667    for operand in operands {
668        build = build.operand(operand);
669    }
670    build.finish();
671    Ok(Made { results, outgoing: places.size() })
672}
673
674/// Which register each value comes back in, walked the way the arguments are.
675///
676/// The two files are counted separately, because a structure of a `double` and a `long` comes back
677/// with the `double` in the first vector register and the `long` in the first integer one, and a
678/// single count would put the second half one place further along a list it is not on.
679///
680/// # Errors
681///
682/// The first value that cannot come back at all, and why, so that a call this cannot make leaves
683/// nothing behind. Nothing here reports which value it was, because the caller has one answer for
684/// all of them: the value that comes back is not an argument and has no position among them.
685fn places_back(returns: &[Type], conv: &CallRegs) -> Result<Vec<(PhysReg, RegClass)>, Refused> {
686    let refused = |missing| Refused { argument: None, missing };
687    let mut back = Vec::with_capacity(returns.len());
688    let (mut ints, mut sses) = (0usize, 0usize);
689    for &ty in returns {
690        if let Some(missing) = refuses(ty) {
691            return Err(refused(missing));
692        }
693        let class = class_of(ty, conv);
694        let (file, at) = if class == conv.sse_class {
695            (conv.sse_returns, &mut sses)
696        } else {
697            (conv.int_returns, &mut ints)
698        };
699        let reg = *file.get(*at).ok_or_else(|| refused(Missing::NoRoom))?;
700        *at += 1;
701        back.push((reg, class));
702    }
703    Ok(back)
704}
705
706/// Which of the four widths a value travels at, where a truth value travels as the byte it lives
707/// in.
708///
709/// [`crate::term::slot`] is the question the rule set asks and it answers nothing for one bit,
710/// because there is no register of that width and so no instruction written at it. A convention
711/// asks a different question. It has nothing narrower than a byte to put an argument in either,
712/// and what it says about the one type that is a bit is that the byte holding it is the argument,
713/// with the seven bits above unspecified. So the two lists differ by exactly this entry.
714///
715/// It is written here and not in [`crate::term::slot`] because moving it there would tell the rule
716/// set that one bit is a byte, and then every byte rule in the file would match a term that is not
717/// one. What the convention needs is narrower: a name for the register an argument arrives in, and
718/// that name says a width because a listing is easier to read when it does.
719fn place(ty: Type) -> Option<usize> {
720    if crate::term::is_bit(ty) { Some(0) } else { crate::term::slot(ty) }
721}
722
723/// What the pseudo for an argument of that type is called.
724///
725/// The width is in the name for the same reason it is in every other opcode here: it is what the
726/// instruction is about. Nothing encodes it, so nothing depends on it being right, but a listing
727/// that says an argument arrived and does not say how much of it did is a listing worth less.
728///
729/// Which widths there are is `place` above, and not a list of its own, because it has to be the
730/// same list the three below use. An argument brought in at a width nothing downstream has a name
731/// for is a register nothing could then read, and a width the others cover that this refuses is a
732/// function turned away for no reason. Asking one question in one place is what keeps the four
733/// answers from drifting, and an address is what they used to disagree about.
734#[must_use]
735pub fn head_of(ty: Type) -> Option<&'static str> {
736    // The format that fills a whole vector register, which travels in one of them: the psABI
737    // classifies it SSE and SSEUP, and those two eightbytes are the one register the pair names
738    // rather than two registers.
739    if crate::term::is_quad(ty) {
740        return Some("x64.arg_val_f128");
741    }
742    if let Some(at) = crate::term::float_slot(ty) {
743        return Some(["x64.arg_val_f32", "x64.arg_val_f64"][at]);
744    }
745    let names = ["x64.arg_val_8", "x64.arg_val_16", "x64.arg_val_32", "x64.arg_val_64"];
746    Some(names[place(ty)?])
747}
748
749/// What the instruction that reads an argument of that type out of memory is called.
750///
751/// Keyed off the same two questions [`head_of`] asks and answering for the same set of types, so
752/// that a parameter this compiler can bring in from a register is one it can bring in from the
753/// caller's stack as well. A width one of them covered and the other did not would be a function
754/// turned away for where its sixth argument happened to land.
755///
756/// Reading a narrow argument at its own width and not at a word is deliberate. The caller wrote a
757/// whole word, but what it put in the part above the value is not something the convention says, so
758/// the bits this reads are exactly the bits that mean anything. That is the same thing an argument
759/// arriving in a register gets: `x64.arg_val_8` says the low byte of that register is the argument
760/// and says nothing at all about the rest of it.
761#[must_use]
762pub fn load_of(ty: Type) -> Option<&'static str> {
763    // Sixteen bytes, which is the whole register and is also the whole value, so the instruction
764    // a spill uses and the instruction an argument uses are the same one here. They are two
765    // different instructions at the two narrower formats because there the value is part of the
766    // register, and at this format there is no part of it to leave behind.
767    if crate::term::is_quad(ty) {
768        return Some("x64.movaps_rm");
769    }
770    if let Some(at) = crate::term::float_slot(ty) {
771        return Some(["x64.movss_rm", "x64.movsd_rm"][at]);
772    }
773    let names = ["x64.mov_rm_8", "x64.mov_rm_16", "x64.mov_rm_32", "x64.mov_rm_64"];
774    Some(names[place(ty)?])
775}
776
777/// What the instruction that writes an argument of that type into memory is called.
778///
779/// The mirror of [`load_of`], keyed off the same two questions and answering for the same set of
780/// types, so that the two ends of one call agree about what travels. A type a callee can read out
781/// of the argument area and a caller cannot write into it would be a call turned away for a reason
782/// the function it calls does not have.
783///
784/// Writing a narrow argument at its own width leaves whatever was already in the rest of the word.
785/// That is allowed, and it is what [`load_of`] is written against: the convention does not say what
786/// is above the value, so the callee reads only the bits that mean anything and neither end has to
787/// agree about the rest.
788#[must_use]
789pub fn store_of(ty: Type) -> Option<&'static str> {
790    if crate::term::is_quad(ty) {
791        return Some("x64.movaps_mr");
792    }
793    if let Some(at) = crate::term::float_slot(ty) {
794        return Some(["x64.movss_mr", "x64.movsd_mr"][at]);
795    }
796    let names = ["x64.mov_mr_8", "x64.mov_mr_16", "x64.mov_mr_32", "x64.mov_mr_64"];
797    Some(names[place(ty)?])
798}
799
800/// What the instruction that leaves a returned value in its register is called, for the value at
801/// that place in its own register file.
802///
803/// Keyed off the same two questions [`head_of`] asks, so a type this can give back is a type it can
804/// take in. The place is the one a call counted to when it laid the return out, which is per file
805/// rather than over the whole list: a structure of a `double` and a `long` gives both of them back
806/// at place zero.
807///
808/// The register itself is not here. It is in the operand table in `rucc_target::x86_64`, which is
809/// where the first one has always been, and the two names below are how a value says which of the
810/// two it is. A convention with more than two registers to come back in would need more names, and
811/// there is none, which is what the `None` at the end is about.
812#[must_use]
813pub fn ret_of(ty: Type, at: usize) -> Option<&'static str> {
814    if crate::term::is_quad(ty) {
815        return Some(*["x64.ret_val_f128", "x64.ret_val2_f128"].get(at)?);
816    }
817    if let Some(width) = crate::term::float_slot(ty) {
818        let names =
819            [["x64.ret_val_f32", "x64.ret_val_f64"], ["x64.ret_val2_f32", "x64.ret_val2_f64"]];
820        return Some(names.get(at)?[width]);
821    }
822    let names = [
823        ["x64.ret_val_8", "x64.ret_val_16", "x64.ret_val_32", "x64.ret_val_64"],
824        ["x64.ret_val2_8", "x64.ret_val2_16", "x64.ret_val2_32", "x64.ret_val2_64"],
825    ];
826    Some(names.get(at)?[place(ty)?])
827}
828
829#[cfg(test)]
830mod tests {
831    use rucc_target::x86_64::{REGS, SYSV, WIN64};
832
833    use super::*;
834
835    /// Those types as parameters that travel as the values they are, which is every one of them
836    /// that is not a structure the classification put in the argument area.
837    fn plain(params: &[Type]) -> Vec<Param> {
838        params.iter().copied().map(Param::new).collect()
839    }
840
841    /// The parameters of a function under a convention, as machine IR text.
842    fn bind(params: &[Type], conv: &CallRegs) -> String {
843        let mut names = Interner::new();
844        let mut out = mir::Func::new(names.intern("f"));
845        let block = out.create_block();
846        entry(&mut out, block, &plain(params), conv, &mut names, None)
847            .expect("every parameter arrives");
848        mir::print_func(&out, &names, &REGS)
849    }
850
851    #[test]
852    fn the_first_arguments_arrive_where_the_convention_puts_them() {
853        let i32 = Type::int(32);
854        assert_eq!(
855            bind(&[i32, i32, Type::int(64)], &SYSV),
856            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
857             %1:gpr($rsi) = x64.arg_val_32\n    %2:gpr($rdx) = x64.arg_val_64\n}\n"
858        );
859    }
860
861    #[test]
862    fn the_other_convention_puts_the_same_arguments_somewhere_else() {
863        // The first argument is in `rcx` here and in `rdi` above, which is the difference that
864        // makes a SysV binary calling a Windows one read the wrong value rather than fail.
865        let i64 = Type::int(64);
866        assert_eq!(
867            bind(&[i64, i64], &WIN64),
868            "mfunc @f {\nblock0:\n    %0:gpr($rcx) = x64.arg_val_64\n    \
869             %1:gpr($rdx) = x64.arg_val_64\n}\n"
870        );
871    }
872
873    /// The parameters of a function under a convention, and what each of the ones that arrived in
874    /// memory is waiting on.
875    fn arrive(params: &[Type], conv: &CallRegs) -> (String, Vec<u32>) {
876        let mut names = Interner::new();
877        let mut out = mir::Func::new(names.intern("f"));
878        let block = out.create_block();
879        let arrived = entry(&mut out, block, &plain(params), conv, &mut names, None)
880            .expect("every parameter");
881        let up = arrived.stack.iter().map(|&(_, up)| up).collect();
882        (mir::print_func(&out, &names, &REGS), up)
883    }
884
885    #[test]
886    fn an_argument_past_the_last_register_is_read_out_of_the_caller_s_stack() {
887        let (text, up) = arrive(&[Type::int(64); 7], &SYSV);
888
889        // Six of them got registers and the seventh did not, so the seventh is a load rather than
890        // a pseudo. It reads through the stack pointer with nothing in its displacement, because
891        // where the caller's argument area is from in here is a distance into a frame that does
892        // not exist yet, and it is at the bottom of that area because it is the first one in it.
893        assert_eq!(up, [0]);
894        assert!(text.contains("%6:gpr = x64.mov_rm_64 [$rsp]"), "{text}");
895        assert_eq!(text.matches("x64.arg_val_64").count(), 6, "{text}");
896    }
897
898    #[test]
899    fn the_other_convention_runs_out_of_registers_three_arguments_earlier() {
900        let (text, up) = arrive(&[Type::int(64); 7], &WIN64);
901
902        // Windows passes four integers in registers and reserves thirty two bytes below the call
903        // whether they are used or not, so the fifth argument is not at the bottom of the argument
904        // area but above the shadow space, and the three after it follow it a word at a time.
905        assert_eq!(up, [32, 40, 48]);
906        assert_eq!(text.matches("x64.arg_val_64").count(), 4, "{text}");
907        assert!(text.contains("%4:gpr = x64.mov_rm_64 [$rsp]"), "{text}");
908    }
909
910    /// The parameters of a function under a convention, with one of them a structure whose bytes
911    /// travel, and what each of the ones that arrived in memory is waiting on.
912    fn arrive_with(params: &[Param], conv: &CallRegs) -> (String, Vec<u32>) {
913        let mut names = Interner::new();
914        let mut out = mir::Func::new(names.intern("f"));
915        let block = out.create_block();
916        let arrived =
917            entry(&mut out, block, params, conv, &mut names, None).expect("every parameter");
918        let up = arrived.stack.iter().map(|&(_, up)| up).collect();
919        (mir::print_func(&out, &names, &REGS), up)
920    }
921
922    #[test]
923    fn a_structure_that_arrived_as_bytes_is_an_address_and_not_a_load() {
924        let byval = Param::with_abi(Type::PTR, Abi::ByVal { size: 32, align: 8 });
925        let (text, up) =
926            arrive_with(&[Param::new(Type::int(32)), byval, Param::new(Type::int(32))], &SYSV);
927
928        // `int f(int a, struct Big b, int c)`. The bytes of `b` are already in this function, at
929        // the bottom of the caller's argument area, so nothing is read out of them here: what the
930        // parameter is is where they are, which is one address. The two integers still travel in
931        // registers, because an object in the argument area takes no register and the arguments
932        // behind it do not shift along.
933        assert_eq!(up, [0]);
934        assert_eq!(text.matches("x64.arg_val_32").count(), 2, "{text}");
935        assert!(text.contains("%2:gpr = x64.lea_64 [$rsp]"), "{text}");
936        assert!(!text.contains("mov_rm"), "nothing is read out of the bytes: {text}");
937    }
938
939    #[test]
940    fn the_argument_behind_a_structure_that_travelled_as_bytes_is_above_all_of_them() {
941        let byval = Param::with_abi(Type::PTR, Abi::ByVal { size: 24, align: 16 });
942        let params: Vec<Param> = (0..7).map(|_| Param::new(Type::int(64))).collect();
943        let (_, up) = arrive_with(&[&params[..], &[byval], &params[..1]].concat(), &SYSV);
944
945        // Six integers take the six registers, the seventh is at the bottom of the argument area,
946        // and the structure is above it at the alignment its type asks for rather than at a word.
947        // The one behind the structure is above all twenty four of its bytes, rounded up to a
948        // whole number of words, because the area is a run of words.
949        assert_eq!(up, [0, 16, 40]);
950    }
951
952    /// A parameter narrower than a word is read at its own width rather than at a word, and one in
953    /// the other register file is read with the other file's instruction. Both are the same list
954    /// [`head_of`] answers from, which is what stops a function being turned away for the width of
955    /// its seventh argument alone.
956    #[test]
957    fn what_a_stack_argument_is_read_with_is_its_own_width_and_its_own_file() {
958        let f32 = Type::float(rucc_ir::Float::F32);
959        let params = [Type::int(64), Type::int(64), Type::int(64), Type::int(64), Type::int(8)];
960        let (text, up) = arrive(&params, &WIN64);
961        assert_eq!(up, [32]);
962        assert!(text.contains("x64.mov_rm_8 [$rsp]"), "{text}");
963
964        let floats = [f32; 5];
965        let (text, up) = arrive(&floats, &WIN64);
966        assert_eq!(up, [32]);
967        assert!(text.contains("%4:xmm = x64.movss_rm [$rsp]"), "{text}");
968    }
969
970    /// Every type a parameter can arrive in a register at is one it can be read from memory at.
971    /// The two lists are keyed off the same two questions so that they cannot drift, and this is
972    /// what says so: a width one covered and the other did not would be a function turned away for
973    /// where its arguments happened to land rather than for anything about it.
974    #[test]
975    fn the_two_lists_of_widths_answer_for_the_same_types() {
976        let types = [
977            Type::int(1),
978            Type::int(8),
979            Type::int(16),
980            Type::int(32),
981            Type::int(64),
982            Type::int(128),
983            Type::PTR,
984            Type::float(rucc_ir::Float::F32),
985            Type::float(rucc_ir::Float::F64),
986            Type::float(rucc_ir::Float::F80),
987            Type::float(rucc_ir::Float::F128),
988        ];
989        for ty in types {
990            assert_eq!(head_of(ty).is_some(), load_of(ty).is_some(), "{ty:?}");
991            assert_eq!(head_of(ty).is_some(), store_of(ty).is_some(), "{ty:?}");
992            assert_eq!(head_of(ty).is_some(), ret_of(ty, 0).is_some(), "{ty:?}");
993        }
994    }
995
996    /// A hundred and twenty eight bit float arrives in a vector register like the two narrower
997    /// formats, and it takes one of them rather than two: the psABI classifies it SSE and SSEUP,
998    /// and what that pair names is the one register both eightbytes are in.
999    ///
1000    /// The second float here is what says so. If the quad had taken two vector registers the
1001    /// `double` after it would be in `xmm2`.
1002    #[test]
1003    fn a_quad_float_arrives_in_one_vector_register_and_not_in_two() {
1004        let quad = Type::float(rucc_ir::Float::F128);
1005        let f64 = Type::float(rucc_ir::Float::F64);
1006        assert_eq!(
1007            bind(&[quad, f64], &SYSV),
1008            "mfunc @f {\nblock0:\n    %0:xmm($xmm0) = x64.arg_val_f128\n    \
1009             %1:xmm($xmm1) = x64.arg_val_f64\n}\n"
1010        );
1011    }
1012
1013    /// And it is not the width that was refused before there was an instruction to move it with,
1014    /// which is the one thing about this type that used to turn a whole function away.
1015    #[test]
1016    fn a_quad_float_is_no_longer_a_width_nothing_can_carry() {
1017        assert_eq!(refuses(Type::float(rucc_ir::Float::F128)), None);
1018        assert_eq!(refuses(Type::float(rucc_ir::Float::F80)), Some(Missing::OnX87));
1019        assert_eq!(refuses(Type::int(128)), Some(Missing::Width));
1020    }
1021
1022    /// A float arrives in the other file, and the two files are counted apart on SysV: the
1023    /// integer here is the first integer argument and the float is the first float one, so they
1024    /// are in `rdi` and `xmm0` rather than in the first and second of anything.
1025    #[test]
1026    fn a_float_arrives_in_a_vector_register_and_is_counted_apart_from_the_integers() {
1027        let f32 = Type::float(rucc_ir::Float::F32);
1028        let f64 = Type::float(rucc_ir::Float::F64);
1029        assert_eq!(
1030            bind(&[Type::int(32), f64, f32], &SYSV),
1031            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
1032             %1:xmm($xmm0) = x64.arg_val_f64\n    %2:xmm($xmm1) = x64.arg_val_f32\n}\n"
1033        );
1034    }
1035
1036    /// Windows counts the two files together, so the same three arguments land in different
1037    /// registers: the float is the second argument and takes the second vector register rather
1038    /// than the first, which is the difference that makes a mismatched call read the wrong value.
1039    #[test]
1040    fn the_other_convention_counts_the_two_files_as_one_run_of_positions() {
1041        let f64 = Type::float(rucc_ir::Float::F64);
1042        assert_eq!(
1043            bind(&[Type::int(32), f64, Type::int(64)], &WIN64),
1044            "mfunc @f {\nblock0:\n    %0:gpr($rcx) = x64.arg_val_32\n    \
1045             %1:xmm($xmm1) = x64.arg_val_f64\n    %2:gpr($r8) = x64.arg_val_64\n}\n"
1046        );
1047    }
1048
1049    /// A `long double` is in neither file and travels in the argument area, which is what SysV's
1050    /// X87 classification comes to. So it arrives the way a structure the classification put in
1051    /// memory arrives, as the address of its bytes in a general purpose register, and it does that
1052    /// while the vector file is untouched: this one is in the argument area because of what it is
1053    /// rather than because the registers ran out.
1054    #[test]
1055    fn a_long_double_arrives_as_the_address_of_its_bytes_in_the_argument_area() {
1056        let params = [Type::int(32), Type::float(rucc_ir::Float::F80)];
1057        assert_eq!(
1058            bind(&params, &SYSV),
1059            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_32\n    \
1060             %1:gpr = x64.lea_64 [$rsp]\n}\n"
1061        );
1062    }
1063
1064    /// It still cannot come back beside another value, and what it is turned away for says which
1065    /// file is in the way rather than calling eighty bits a width no register holds. A pair comes
1066    /// back in a pair of registers and there is no pair with the x87 stack in it.
1067    #[test]
1068    fn a_long_double_in_a_pair_is_reported_as_the_x87_stack_it_travels_on() {
1069        let returns = [Type::float(rucc_ir::Float::F80), Type::int(64)];
1070        assert_eq!(
1071            make(&[], &returns, false, &SYSV).2,
1072            Err(Refused { argument: None, missing: Missing::OnX87 })
1073        );
1074    }
1075
1076    /// One call to `g`, with a register for each argument arriving in the block that makes it.
1077    fn make(
1078        args: &[Type],
1079        returns: &[Type],
1080        variadic: bool,
1081        conv: &CallRegs,
1082    ) -> (Interner, mir::Func, Result<Made, Refused>) {
1083        let mut names = Interner::new();
1084        let mut out = mir::Func::new(names.intern("f"));
1085        let block = out.create_block();
1086        let passed: Vec<Passing> = args
1087            .iter()
1088            .map(|&ty| Passing {
1089                ty,
1090                reg: out.append_param(block, class_of(ty, conv)),
1091                abi: Abi::Plain,
1092            })
1093            .collect();
1094        let callee = Callee::Named(names.intern("g"));
1095        let what = Calling { callee, args: &passed, returns, variadic };
1096        let made = call(&mut out, block, &what, conv, &mut names);
1097        (names, out, made)
1098    }
1099
1100    /// What the call in that function reads and writes, by register name, in the order the
1101    /// operands are in.
1102    fn operands(func: &mir::Func) -> (Vec<String>, Vec<String>) {
1103        let block = func.entry().expect("a function with a block in it");
1104        let call = func.terminator(block).expect("the call is the last thing in the block");
1105        let name = |operand: &mir::Operand| match (operand.reg.phys(), operand.constraint) {
1106            (Some(reg), _) | (None, Constraint::Fixed(reg)) => {
1107                REGS.name(operand.class, reg).expect("a register the file describes").to_string()
1108            }
1109            _ => format!("{:?}", operand.reg),
1110        };
1111        let mut written = Vec::new();
1112        let mut read = Vec::new();
1113        for operand in &func[func[call].operands] {
1114            let into = if operand.role == mir::Role::Use { &mut read } else { &mut written };
1115            into.push(name(operand));
1116        }
1117        (written, read)
1118    }
1119
1120    #[test]
1121    fn a_call_passes_its_arguments_where_the_convention_puts_them() {
1122        let i32 = Type::int(32);
1123        let (_, func, made) = make(&[i32, i32, i32], &[], false, &SYSV);
1124        assert_eq!(made.expect("three integers all fit in registers").results, []);
1125        assert_eq!(operands(&func).1, ["rdi", "rsi", "rdx"]);
1126    }
1127
1128    #[test]
1129    fn the_other_convention_passes_the_same_arguments_somewhere_else() {
1130        let i64 = Type::int(64);
1131        let (_, func, made) = make(&[i64, i64], &[], false, &WIN64);
1132        // Thirty two bytes of stack for a call that passes nothing on the stack, which is what
1133        // Windows asks a caller to leave the callee whether the callee uses it or not.
1134        assert_eq!(made.expect("two integers fit in registers").outgoing, 32);
1135        assert_eq!(operands(&func).1, ["rcx", "rdx"]);
1136    }
1137
1138    #[test]
1139    fn what_a_call_gives_back_comes_out_of_the_register_the_convention_returns_in() {
1140        let (names, func, made) = make(&[], &[Type::int(32)], false, &SYSV);
1141        let made = made.expect("an integer comes back");
1142        let [result] = made.results[..] else { panic!("one register") };
1143        // The first thing written is the result, and it is the only thing written that is a value
1144        // rather than a register the callee destroyed.
1145        assert_eq!(operands(&func).0.first().map(String::as_str), Some("rax"));
1146        assert_eq!(func.class_of(result), Some(SYSV.int_class));
1147        assert!(mir::print_func(&func, &names, &REGS).contains("x64.call"));
1148    }
1149
1150    #[test]
1151    fn every_register_the_callee_may_destroy_is_written_by_the_call() {
1152        let (_, func, _) = make(&[Type::int(64)], &[Type::int(64)], false, &SYSV);
1153        let (written, read) = operands(&func);
1154        // The callee saved registers are not here, because a value in one of those survives a
1155        // call and that is the whole difference between the two halves of the convention.
1156        for saved in ["rbx", "rbp", "r12", "r13", "r14", "r15"] {
1157            assert!(!written.contains(&saved.to_string()), "{saved} survives a call");
1158        }
1159        // Every other integer register is, once. The two named ones are named by the result and
1160        // by the argument instead, and naming one twice would be blocking it twice.
1161        for destroyed in ["rcx", "rdx", "rsi", "r8", "r9", "r10", "r11"] {
1162            let count = written.iter().filter(|name| *name == destroyed).count();
1163            assert_eq!(count, 1, "{destroyed} is destroyed by a call and is written {count} times");
1164        }
1165        assert_eq!(written.iter().filter(|name| *name == "rax").count(), 1);
1166        assert_eq!(read, ["rdi"]);
1167        // The vector registers are all destroyed on SysV, and they are in the other class.
1168        assert!(written.contains(&"xmm0".to_string()));
1169    }
1170
1171    #[test]
1172    fn a_variadic_call_says_how_many_vector_registers_it_passed_arguments_in() {
1173        let (names, func, made) = make(&[Type::int(64)], &[], true, &SYSV);
1174        made.expect("an integer argument to a variadic callee");
1175        let (_, read) = operands(&func);
1176        // Zero of them here, and `al` is where a SysV callee looks for it. Leaving whatever was in
1177        // the register there would make a callee that saves its vector registers save ones it was
1178        // never given.
1179        assert_eq!(read, ["rdi", "rax"]);
1180        assert_eq!(
1181            mir::print_func(&func, &names, &REGS).lines().nth(2),
1182            Some("    %1:gpr = x64.mov_ri_32 0")
1183        );
1184
1185        // Two of them here, which is the number that decides how much of the register save area a
1186        // callee like `printf` fills in. A count of zero with a float in `xmm0` would be a callee
1187        // reading its first `%f` out of a register nothing wrote.
1188        let f64 = Type::float(rucc_ir::Float::F64);
1189        let (names, func, made) = make(&[Type::int(64), f64, f64], &[], true, &SYSV);
1190        made.expect("one integer and two floats all fit in registers");
1191        assert_eq!(operands(&func).1, ["rdi", "xmm0", "xmm1", "rax"]);
1192        assert!(mir::print_func(&func, &names, &REGS).contains("x64.mov_ri_32 2"));
1193    }
1194
1195    /// Windows passes a float to a variadic callee in both files at once, and which arguments are
1196    /// the ones the signature does not name is not something a call carries, so it is turned down
1197    /// rather than passed in one file and read from the other.
1198    #[test]
1199    fn a_float_passed_to_a_variadic_callee_on_windows_is_reported() {
1200        let f64 = Type::float(rucc_ir::Float::F64);
1201        assert_eq!(
1202            make(&[Type::int(32), f64], &[], true, &WIN64).2,
1203            Err(Refused { argument: Some(1), missing: Missing::InBothFiles })
1204        );
1205        // The same call to a callee whose signature names both arguments is fine, because there is
1206        // no second copy to make.
1207        assert!(make(&[Type::int(32), f64], &[], false, &WIN64).2.is_ok());
1208    }
1209
1210    #[test]
1211    fn a_call_through_an_address_reads_it_in_front_of_the_arguments() {
1212        let i32 = Type::int(32);
1213        let mut names = Interner::new();
1214        let mut out = mir::Func::new(names.intern("f"));
1215        let block = out.create_block();
1216        let address = out.append_param(block, SYSV.int_class);
1217        let reg = out.append_param(block, SYSV.int_class);
1218        let passed = vec![Passing { ty: i32, reg, abi: Abi::Plain }];
1219        let what = Calling {
1220            callee: Callee::Through(address),
1221            args: &passed,
1222            returns: &[i32],
1223            variadic: false,
1224        };
1225        call(&mut out, block, &what, &SYSV, &mut names).expect("one integer fits in a register");
1226
1227        // The address is the first thing read and the arguments follow it, which is the order the
1228        // assembler counts on, and it is in no particular register because every register a call
1229        // could insist on is one the call has already spoken for.
1230        let text = mir::print_func(&out, &names, &REGS);
1231        assert!(text.contains("= x64.call_reg %0, %1($rdi)\n"), "{text}");
1232        assert!(!text.contains("@g"), "a call through an address names nobody: {text}");
1233    }
1234
1235    #[test]
1236    fn a_call_with_no_register_left_writes_the_argument_into_the_outgoing_area() {
1237        let i64 = Type::int(64);
1238        let (names, func, made) = make(&[i64; 7], &[], false, &SYSV);
1239        let made = made.expect("the seventh goes to memory");
1240
1241        // At the stack pointer, because the outgoing area is at the bottom of the frame, and in
1242        // front of the call rather than as an operand of it.
1243        let text = mir::print_func(&func, &names, &REGS);
1244        assert!(text.contains("x64.mov_mr_64 %6, [$rsp]\n"), "{text}");
1245        let store = text.find("x64.mov_mr_64").expect("the store");
1246        assert!(store < text.find("x64.call").expect("the call"), "{text}");
1247        // One word of it, which is what the frame has to reserve for this call.
1248        assert_eq!(made.outgoing, 8);
1249    }
1250
1251    /// One call to `g`, passing that many words, then an object of that size and alignment by
1252    /// value, then one more integer, which is `int g(long.., struct Big, int)` after the
1253    /// classification.
1254    fn pass_bytes(
1255        before: usize,
1256        size: u64,
1257        align: u32,
1258        conv: &CallRegs,
1259    ) -> (Interner, mir::Func, Result<Made, Refused>) {
1260        let mut names = Interner::new();
1261        let mut out = mir::Func::new(names.intern("f"));
1262        let block = out.create_block();
1263        let mut args: Vec<Passing> = (0..before)
1264            .map(|_| Passing {
1265                ty: Type::int(64),
1266                reg: out.append_param(block, conv.int_class),
1267                abi: Abi::Plain,
1268            })
1269            .collect();
1270        args.push(Passing {
1271            ty: Type::PTR,
1272            reg: out.append_param(block, conv.int_class),
1273            abi: Abi::ByVal { size, align },
1274        });
1275        args.push(Passing {
1276            ty: Type::int(32),
1277            reg: out.append_param(block, conv.int_class),
1278            abi: Abi::Plain,
1279        });
1280        let callee = Callee::Named(names.intern("g"));
1281        let what = Calling { callee, args: &args, returns: &[], variadic: false };
1282        let made = call(&mut out, block, &what, conv, &mut names);
1283        (names, out, made)
1284    }
1285
1286    #[test]
1287    fn a_structure_passed_by_value_in_memory_is_copied_into_the_outgoing_area() {
1288        let (names, func, made) = pass_bytes(1, 24, 8, &SYSV);
1289        let made = made.expect("an object of three words is copied a word at a time");
1290
1291        // The bytes travel and the address does not, so the copy is a load and a store for each
1292        // word of it, in front of the call, and the callee's copy is at the bottom of the outgoing
1293        // area. The caller owes it this copy: the callee is free to write to what it was handed,
1294        // so what it was handed cannot be the object itself.
1295        let text = mir::print_func(&func, &names, &REGS);
1296        assert!(text.contains("x64.mov_mr_64 %3, [$rsp]\n"), "{text}");
1297        assert!(text.contains("x64.mov_mr_64 %4, [$rsp + 8]\n"), "{text}");
1298        assert!(text.contains("x64.mov_mr_64 %5, [$rsp + 16]\n"), "{text}");
1299        assert_eq!(text.matches("x64.mov_rm_64").count(), 3, "{text}");
1300        assert!(text.find("x64.mov_mr_64") < text.find("x64.call"), "{text}");
1301        assert_eq!(made.outgoing, 24);
1302    }
1303
1304    #[test]
1305    fn the_integers_beside_it_still_travel_in_registers() {
1306        let (_, func, _) = pass_bytes(1, 24, 8, &SYSV);
1307
1308        // An object in the argument area takes no argument register, so the integer behind it is
1309        // in the second one and not the third. Counting it as a register is the mistake that would
1310        // shift every argument after it along by one.
1311        let (clobbered, read) = operands(&func);
1312        assert_eq!(read, ["rdi", "rsi"]);
1313        assert!(clobbered.contains(&"rdx".to_owned()), "the third is free: {clobbered:?}");
1314    }
1315
1316    #[test]
1317    fn an_object_wanting_more_alignment_than_a_word_gets_it() {
1318        let (names, func, made) = pass_bytes(7, 24, 16, &SYSV);
1319        let made = made.expect("an object of three words");
1320
1321        // Six of the integers took the registers and the seventh is at the bottom of the area, so
1322        // the object cannot start where it left off: sixteen byte alignment moves it up to the
1323        // next multiple of sixteen and leaves a word of nothing behind it. The integer after it is
1324        // above all three of its words.
1325        let text = mir::print_func(&func, &names, &REGS);
1326        assert!(text.contains("x64.mov_mr_64 %9, [$rsp + 16]\n"), "{text}");
1327        assert!(text.contains("x64.mov_mr_32 %8, [$rsp + 40]\n"), "{text}");
1328        assert_eq!(made.outgoing, 48);
1329    }
1330
1331    #[test]
1332    fn an_object_too_large_to_copy_a_word_at_a_time_is_reported_rather_than_passed() {
1333        let (_, _, made) = pass_bytes(1, 4096, 8, &SYSV);
1334
1335        // Five hundred and twelve words is past what unrolling is worth, and the copy that size
1336        // wants is a call to the runtime, which cannot be built in the middle of building a call.
1337        // Saying so is the point: the alternative is a call that passes the address of the object
1338        // where the callee is going to read the object.
1339        assert_eq!(made, Err(Refused { argument: Some(1), missing: Missing::TooBig }));
1340        assert_eq!(
1341            Missing::TooBig.why(),
1342            "is more bytes than a copy into the argument area unrolls to"
1343        );
1344    }
1345
1346    /// The other convention runs out three arguments earlier and starts its argument area above the
1347    /// shadow space it also has to reserve, and both of those are what `Places` already said.
1348    #[test]
1349    fn where_the_outgoing_area_starts_is_the_convention_s_answer() {
1350        let i64 = Type::int(64);
1351        let (names, func, made) = make(&[i64; 7], &[], false, &WIN64);
1352        assert_eq!(made.expect("the last three go to memory").outgoing, 56);
1353
1354        // Thirty two bytes of shadow space first, which the caller writes nothing into and the
1355        // callee owns, and the fifth argument above it.
1356        let text = mir::print_func(&func, &names, &REGS);
1357        assert!(text.contains("x64.mov_mr_64 %4, [$rsp + 32]\n"), "{text}");
1358        assert!(text.contains("x64.mov_mr_64 %5, [$rsp + 40]\n"), "{text}");
1359        assert!(text.contains("x64.mov_mr_64 %6, [$rsp + 48]\n"), "{text}");
1360    }
1361
1362    /// What a stack argument is written with is its own width and its own register file, matching
1363    /// what the callee reads it back with.
1364    #[test]
1365    fn a_narrow_or_floating_argument_keeps_its_own_store() {
1366        let i64 = Type::int(64);
1367        let narrow = [i64, i64, i64, i64, i64, i64, Type::int(8)];
1368        let (names, func, made) = make(&narrow, &[], false, &SYSV);
1369        made.expect("the seventh goes to memory");
1370        let text = mir::print_func(&func, &names, &REGS);
1371        assert!(text.contains("x64.mov_mr_8 %6, [$rsp]\n"), "{text}");
1372
1373        let f32 = Type::float(rucc_ir::Float::F32);
1374        let (names, func, made) = make(&[f32; 9], &[], false, &SYSV);
1375        made.expect("the ninth goes to memory");
1376        let text = mir::print_func(&func, &names, &REGS);
1377        assert!(text.contains("x64.movss_mr %8, [$rsp]\n"), "{text}");
1378    }
1379
1380    /// The count a SysV variadic callee reads is a count of registers, so an argument that went to
1381    /// memory instead is not in it.
1382    #[test]
1383    fn an_argument_in_memory_is_not_counted_as_a_vector_register() {
1384        let f64 = Type::float(rucc_ir::Float::F64);
1385        let (names, func, made) = make(&[f64; 9], &[], true, &SYSV);
1386        made.expect("the ninth goes to memory");
1387        let text = mir::print_func(&func, &names, &REGS);
1388        assert!(text.contains("x64.mov_ri_32 8\n"), "eight registers, not nine: {text}");
1389    }
1390
1391    /// The two lists of widths answer for the same set of types, so that a value the callee can
1392    /// read out of the argument area is one the caller can write into it.
1393    #[test]
1394    fn what_can_be_read_can_be_written() {
1395        let types = [
1396            Type::int(1),
1397            Type::int(8),
1398            Type::int(16),
1399            Type::int(32),
1400            Type::int(64),
1401            Type::int(128),
1402            Type::PTR,
1403            Type::float(rucc_ir::Float::F32),
1404            Type::float(rucc_ir::Float::F64),
1405            Type::float(rucc_ir::Float::F80),
1406        ];
1407        for ty in types {
1408            assert_eq!(load_of(ty).is_some(), store_of(ty).is_some(), "{ty:?}");
1409        }
1410    }
1411
1412    /// A float travels in the other file at both ends of a call, and the register it comes back in
1413    /// is the first of that file rather than the first of the other one.
1414    #[test]
1415    fn a_call_passes_and_returns_a_float_in_a_vector_register() {
1416        let f64 = Type::float(rucc_ir::Float::F64);
1417        let (_, func, made) = make(&[Type::int(32), f64], &[f64], false, &SYSV);
1418        let result = made.expect("an integer and a float both fit in registers");
1419        let (written, read) = operands(&func);
1420        assert_eq!(read, ["rdi", "xmm0"]);
1421        assert_eq!(written.first().map(String::as_str), Some("xmm0"));
1422        assert_eq!(func.class_of(result.results[0]), Some(SYSV.sse_class));
1423        // Written once, because the register the result comes back in is already blocked by being
1424        // named and a clobber that repeated it would be blocking it twice. `rax` is a clobber here
1425        // rather than the result, which is the same register number in the other file and is the
1426        // whole reason the two lists are counted apart.
1427        assert_eq!(written.iter().filter(|name| *name == "xmm0").count(), 1);
1428        assert!(written.contains(&"rax".to_string()));
1429    }
1430
1431    #[test]
1432    fn a_call_at_a_width_no_register_holds_is_reported_on_either_side() {
1433        let i128 = Type::int(128);
1434        assert_eq!(
1435            make(&[i128], &[], false, &SYSV).2,
1436            Err(Refused { argument: Some(0), missing: Missing::Width })
1437        );
1438        assert_eq!(
1439            make(&[], &[i128], false, &SYSV).2,
1440            Err(Refused { argument: None, missing: Missing::Width })
1441        );
1442    }
1443
1444    #[test]
1445    fn an_argument_wider_than_a_register_has_no_name() {
1446        assert_eq!(head_of(Type::int(128)), None);
1447        assert_eq!(head_of(Type::int(8)), Some("x64.arg_val_8"));
1448        assert_eq!(head_of(Type::int(64)), Some("x64.arg_val_64"));
1449    }
1450
1451    /// An address arrives in a general purpose register like any other integer of its width, and
1452    /// used to be turned away here as a width no register holds, which is what issue 274 is.
1453    /// `int g(char *s)` is the smallest program that was.
1454    #[test]
1455    fn an_address_arrives_in_a_register_like_the_integer_it_is() {
1456        assert_eq!(head_of(Type::PTR), Some("x64.arg_val_64"));
1457        assert_eq!(
1458            bind(&[Type::PTR], &SYSV),
1459            "mfunc @f {\nblock0:\n    %0:gpr($rdi) = x64.arg_val_64\n}\n"
1460        );
1461        // And it travels the same way at a call, on both sides of one.
1462        assert!(make(&[Type::PTR], &[Type::PTR], false, &SYSV).2.is_ok());
1463    }
1464}