rucc_target/regs.rs
1//! The register file: what registers a target has, and what classes they fall into.
2//!
3//! Design: `spec/10-backend.md` section 10.8.
4//!
5//! A register file is data rather than code, which is the same claim the rest of this crate
6//! makes and the one `M10` puts a number on. A class is a set of registers that an operand of
7//! that class may be assigned to, and a physical register is its number inside its class, so
8//! the allocator works in dense small integers and only the printer and the parser ever deal in
9//! names.
10//!
11//! The file lives here rather than in `rucc-mir` because more than one thing reads it. The
12//! machine IR needs it to print, the allocator needs the set it may assign from, and the ABI
13//! description needs to name the registers arguments arrive in. All three are above this crate,
14//! and the alternative is the register file living in whichever of them happens to be lowest,
15//! which is how a layering ends up describing itself as historical.
16//!
17//! Names are unique across the whole file, not merely inside a class. That is what lets a
18//! register be written `$rax` in a dump rather than `$gpr.0`, and it is a real constraint on a
19//! target that gives one register two classes: it has to say which class it is in, or use two
20//! names. [`RegFile::duplicate`] is what a target's own test asks to find out.
21
22use std::fmt;
23
24use rucc_abi::AbiDescription;
25
26/// One class of registers, and the registers in it.
27#[derive(Debug, Clone, Copy, PartialEq, Eq)]
28pub struct ClassInfo {
29 /// What the class is called in a dump, such as `gpr`.
30 pub name: &'static str,
31 /// How wide one of its registers is, in bits.
32 pub bits: u32,
33 /// The registers, in the order their numbers run, without the sigil a dump writes.
34 pub regs: &'static [&'static str],
35 /// Whether the allocator may put a value in one of these.
36 ///
37 /// True for every class a target means the allocator to use, which is nearly all of them.
38 /// False says the registers exist and are named and are not somewhere a value may be told to
39 /// live, so a virtual register of this class is a mistake at the point it was made rather than
40 /// a value the allocator has nowhere to put.
41 ///
42 /// The x87 stack is the case this exists for, and it is worth the sentence because it is not
43 /// the usual reason a register is unavailable. `rsp` is unavailable because it has a job;
44 /// `st0` is unavailable because the machine addresses it as a stack, so which register a name
45 /// means depends on how many values are on the stack at the time, and an allocator that hands
46 /// out a name has no way to say that. So nothing allocates from it, an eighty bit value lives
47 /// in a stack slot between one operation and the next, and the stack is empty on both sides of
48 /// every group of instructions that uses it. See `spec/10-backend.md` section 10.8, which says
49 /// what a group is and why nothing the allocator inserts can get into the middle of one, and
50 /// tamnd/rucc#540.
51 ///
52 /// A register in such a class can still be named, which is the whole reason the class is
53 /// described at all: a `long double` comes back from a call in `st0` and the convention has to
54 /// be able to say so.
55 pub allocatable: bool,
56}
57
58/// Which class a register or an operand belongs to.
59///
60/// A number into the file's classes rather than a name, because it is on every operand of every
61/// instruction and it is compared far more often than it is printed.
62#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
63pub struct RegClass(u8);
64
65impl RegClass {
66 /// The class with that number.
67 #[must_use]
68 pub const fn new(number: u8) -> Self {
69 Self(number)
70 }
71
72 /// Its number, which is what indexes the file.
73 #[must_use]
74 pub const fn number(self) -> u8 {
75 self.0
76 }
77}
78
79/// One physical register, as its number inside its class.
80///
81/// The class is not in here. An operand carries its class already, and a fixed-register
82/// constraint is a constraint on an operand, so repeating the class would be a second copy of
83/// something that can disagree with the first.
84#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
85pub struct PhysReg(u8);
86
87impl PhysReg {
88 /// The register with that number in its class.
89 #[must_use]
90 pub const fn new(number: u8) -> Self {
91 Self(number)
92 }
93
94 /// Its number inside its class.
95 #[must_use]
96 pub const fn number(self) -> u8 {
97 self.0
98 }
99}
100
101/// Every register a target has.
102#[derive(Debug, Clone, Copy, PartialEq, Eq)]
103pub struct RegFile {
104 classes: &'static [ClassInfo],
105}
106
107impl RegFile {
108 /// The file of a target whose registers nothing has described yet.
109 ///
110 /// A target reaches 1.0 with a real one. Until it has one, the honest answer to what
111 /// registers it has is that nobody has written them down, and that is a file with no
112 /// classes in it rather than a panic or a plausible guess.
113 pub const EMPTY: Self = Self::new(&[]);
114
115 /// A file made of those classes, numbered in the order they are given.
116 #[must_use]
117 pub const fn new(classes: &'static [ClassInfo]) -> Self {
118 Self { classes }
119 }
120
121 /// Its classes, each with the number it is known by.
122 pub fn classes(&self) -> impl Iterator<Item = (RegClass, &'static ClassInfo)> + use<> {
123 self.classes.iter().enumerate().map(|(number, info)| (RegClass::new(number as u8), info))
124 }
125
126 /// What is in one class.
127 #[must_use]
128 pub fn class(&self, class: RegClass) -> Option<&'static ClassInfo> {
129 self.classes.get(usize::from(class.number()))
130 }
131
132 /// The class of that name, such as `gpr`.
133 #[must_use]
134 pub fn class_named(&self, name: &str) -> Option<RegClass> {
135 self.classes().find(|(_, info)| info.name == name).map(|(class, _)| class)
136 }
137
138 /// Whether the allocator may put a value in that class, which is [`ClassInfo::allocatable`].
139 ///
140 /// A class the file does not have is not one either, which is the same answer as a class
141 /// nothing allocates from and is the one that keeps a caller from having to say what it means
142 /// by a class number the target never gave out.
143 #[must_use]
144 pub fn allocatable(&self, class: RegClass) -> bool {
145 self.class(class).is_some_and(|info| info.allocatable)
146 }
147
148 /// How many registers are in a class, which is one past the largest number in it.
149 #[must_use]
150 pub fn len(&self, class: RegClass) -> usize {
151 self.class(class).map_or(0, |info| info.regs.len())
152 }
153
154 /// Whether the file has no classes at all, which is a target that has not described one.
155 #[must_use]
156 pub fn is_empty(&self) -> bool {
157 self.classes.is_empty()
158 }
159
160 /// What one register is called.
161 #[must_use]
162 pub fn name(&self, class: RegClass, reg: PhysReg) -> Option<&'static str> {
163 self.class(class)?.regs.get(usize::from(reg.number())).copied()
164 }
165
166 /// The register of that name, and the class it is in.
167 ///
168 /// The name is written without the sigil, so `rax` rather than `$rax`.
169 #[must_use]
170 pub fn reg_named(&self, name: &str) -> Option<(RegClass, PhysReg)> {
171 for (class, info) in self.classes() {
172 if let Some(number) = info.regs.iter().position(|®| reg == name) {
173 return Some((class, PhysReg::new(number as u8)));
174 }
175 }
176 None
177 }
178
179 /// A name this file gives to two registers, if it gives one to two.
180 ///
181 /// Reading a dump back needs every name to say which register it means, and a target that
182 /// breaks that produces text that cannot be parsed rather than an error at the point of the
183 /// mistake. So every target's own test asks this, which is why it is here and public.
184 #[must_use]
185 pub fn duplicate(&self) -> Option<&'static str> {
186 let mut seen: Vec<&'static str> = Vec::new();
187 for (_, info) in self.classes() {
188 for ® in info.regs {
189 if seen.contains(®) {
190 return Some(reg);
191 }
192 seen.push(reg);
193 }
194 }
195 None
196 }
197}
198
199/// The storage an address is counted from, on a machine that has more than one.
200///
201/// x86 keeps a thread's own block of words at a fixed place reached through a segment register,
202/// and that block is the only thing anything here uses one for. The stack protector's canary lives
203/// in it, which is why `%fs:40` is an address a compiler writes and `%fs` is not a register any
204/// program names. Every other address this compiler writes is in the flat segment and says nothing
205/// at all, which is what `None` is.
206#[derive(Debug, Clone, Copy, PartialEq, Eq)]
207pub enum Segment {
208 /// `%fs`, which is where a thread's own block is on x86-64 under System V.
209 Fs,
210 /// `%gs`, which is where it is on x86-64 under Windows and inside a kernel.
211 Gs,
212}
213
214/// Where a target keeps the word a stack protector's canary is a copy of.
215///
216/// Not a register and not a symbol either, on the conventions here. The word is in the block a
217/// thread has to itself, which is reached through a segment register and no other way, so the only
218/// way to name it is a distance into that block. That is why `%fs:40` appears in every protected
219/// function glibc has ever linked and why no object file carries a relocation for it.
220///
221/// A convention that answers `None` is one this compiler has no protector for, and a command line
222/// that asks for one on such a target is told so rather than quietly given an unprotected frame.
223#[derive(Debug, Clone, Copy, PartialEq, Eq)]
224pub struct Guard {
225 /// The storage the word is in.
226 pub segment: Segment,
227 /// How far into it the word is.
228 pub at: i32,
229 /// The function called when the copy in the frame no longer matches it, which does not come
230 /// back.
231 pub fail: &'static str,
232}
233
234/// What a profiler's hook at the top of every function is called on this platform.
235///
236/// A profiler wants to know which function called which and how often, and the only place a
237/// compiler can tell it that is the moment a function is entered. So `-pg` puts a call there, and
238/// what it calls is a routine the runtime provides rather than anything the program wrote.
239///
240/// Two of them, because there are two conventions for the same job and they disagree about where
241/// the call goes as well as what it is called. The older one runs once the frame is taken, so the
242/// hook can walk back through the frame pointer, which is why it needs one. The newer one runs
243/// before the prologue has done anything at all, which is what makes the return address the top
244/// thing on the stack and the arguments still in the registers they arrived in, and that is what
245/// lets a tracer replace the call with something else while the program runs. Linux's ftrace is
246/// built on exactly that, and it is why every kernel is built with the newer one.
247///
248/// A convention that answers `None` is one this compiler has no hook for, and a command line that
249/// asks for one on such a target is told so rather than quietly given an unprofiled program.
250#[derive(Debug, Clone, Copy, PartialEq, Eq)]
251pub struct Trace {
252 /// What is called in front of the prologue, which is what `-mfentry` asks for.
253 pub early: &'static str,
254 /// What is called once the frame is taken, which is what `-mno-fentry` asks for.
255 pub late: &'static str,
256 /// Which of the two a command line that named neither gets.
257 pub fentry: bool,
258}
259
260/// What a platform calls the routine that reaches the pages of a frame before the frame is taken.
261///
262/// Windows is the platform this is about. A thread there is given a stack of which only a little is
263/// committed, and one page below what is committed is a guard page whose whole job is to be touched:
264/// the fault it raises is what tells the kernel to commit another page and move the guard down. So a
265/// frame larger than a page has to be reached a page at a time or the guard is stepped over, and the
266/// program gets an access violation on an address that was never going to be mapped. That is the
267/// calling convention rather than a hardening flag, which is what makes this a field here and not
268/// something `-fstack-clash-protection` turns on.
269///
270/// Every Windows toolchain calls a routine for it rather than writing the walk out, and the routine
271/// is in the C runtime, so the name is the platform's. It reads the size in one register, touches
272/// each page down to there, and comes back having moved nothing, which is why the caller still has
273/// to take the frame afterwards.
274///
275/// A convention that answers `None` is one where reaching the pages is not the convention's
276/// business. That is every System V target, where a stack grows by faulting anywhere below it and
277/// the pages between are filled in by the kernel without being asked in order.
278#[derive(Debug, Clone, Copy, PartialEq, Eq)]
279pub struct Chkstk {
280 /// What the routine is called.
281 pub name: &'static str,
282 /// The register the size goes in, which is also the register it comes back in.
283 pub size: PhysReg,
284}
285
286/// Which registers a calling convention gives which job.
287///
288/// This is the second half of a target description and it is separate from [`RegFile`] because
289/// the two do not vary together. x86-64 has one register file and two conventions over it, and
290/// they disagree about nearly everything below: `rdi` is where the first argument arrives on
291/// SysV and a register a callee has to preserve on Windows, and a Windows caller reserves
292/// thirty two bytes below the call that a SysV caller does not.
293///
294/// The allocation order is here rather than on a class because it is a consequence of what a
295/// call clobbers. A value that does not live across a call belongs in a register the callee is
296/// free to destroy, because putting it in a preserved one costs a push and a pop in the
297/// prologue of whichever function ends up owning it.
298///
299/// Every register named here is a register of the file the same target describes, and each list
300/// is in the order the convention uses them, so the fourth integer argument is `int_args[3]` and
301/// nothing has to count.
302#[derive(Debug, Clone, Copy, PartialEq, Eq)]
303pub struct CallRegs {
304 /// The ABI these registers are the register half of.
305 ///
306 /// The two halves of one convention: this structure says which register a value goes in and
307 /// the description says what form it travels in on the way there, and a target that has one
308 /// has the other. It is a link rather than a copy of the fact anybody wants, because a back end
309 /// pass that writes a call to a runtime routine has to ask the same classification question a
310 /// call in the program is asked and there is one place that answers it. A target's registers and
311 /// a target's ABI are still chosen by two separate matches, one in [`crate::TargetInfo`] and one
312 /// in `rucc_abi::abis::for_target`, and the crate's test that walks every triple asking both is
313 /// what holds them together.
314 pub abi: &'static AbiDescription,
315 /// The class the general purpose registers named here are in.
316 ///
317 /// A register is a number inside its class, so a list of them says nothing about which
318 /// registers they are without this. Everything else could get the class from the operand it
319 /// came off, and a frame cannot, because a saved register is not an operand of anything.
320 pub int_class: RegClass,
321 /// The class the vector registers named here are in.
322 pub sse_class: RegClass,
323 /// The general purpose registers integer arguments arrive in, in order.
324 pub int_args: &'static [PhysReg],
325 /// The vector registers floating point arguments arrive in, in order.
326 ///
327 /// Whether an argument's position counts against both lists or only against its own is
328 /// [`CallRegs::shared_positions`].
329 pub sse_args: &'static [PhysReg],
330 /// Whether an argument's position counts against both argument lists or only against its own.
331 ///
332 /// False on SysV, which counts each separately, so a `double` after six integers is still in
333 /// `xmm0`. True on Windows, which counts one position for both, so a `double` in the third
334 /// position is in `xmm2` and `r8` is skipped.
335 pub shared_positions: bool,
336 /// The general purpose registers an integer return value comes back in.
337 pub int_returns: &'static [PhysReg],
338 /// The vector registers a floating point return value comes back in.
339 pub sse_returns: &'static [PhysReg],
340 /// The x87 registers a `long double` comes back in, which is empty on a target whose
341 /// `long double` is a `double`.
342 pub x87_returns: &'static [PhysReg],
343 /// The general purpose registers a call leaves alone, so a value in one survives it.
344 pub int_saved: &'static [PhysReg],
345 /// The vector registers a call leaves alone, which is none of them on SysV.
346 pub sse_saved: &'static [PhysReg],
347 /// The general purpose registers the allocator may hand out, in the order it prefers them.
348 ///
349 /// The stack pointer is never in this list, and neither is the frame pointer, which a
350 /// target could allocate when nothing needs a frame and which nothing here does yet.
351 pub int_order: &'static [PhysReg],
352 /// The vector registers the allocator may hand out, in the order it prefers them.
353 pub sse_order: &'static [PhysReg],
354 /// The stack pointer.
355 pub stack_pointer: PhysReg,
356 /// The frame pointer, which is the register a prologue puts the old stack pointer in.
357 pub frame_pointer: PhysReg,
358 /// Whether the prologue points the frame pointer at the frame after taking it rather than
359 /// before taking it.
360 ///
361 /// False everywhere but Windows, and there it is the unwind table asking rather than a
362 /// preference. The record a function carries on that platform counts every slot in it from
363 /// where the stack pointer ends the prologue, and it finds that place by taking a constant off
364 /// the frame pointer, so a register pushed after the pointer was established is below the place
365 /// the record counts from and has no row the format can write. The order that does work is the
366 /// pushes, then the frame, and only then the pointer, which is what Microsoft's compiler emits
367 /// and what gcc emits for this target in every function that saves anything besides the pointer
368 /// itself.
369 ///
370 /// What it costs is the chain: the frame pointer holds a copy of the body's stack pointer
371 /// rather than the address of the caller's copy of itself, so the words that used to lead from
372 /// one frame to the next no longer do. Nothing on this platform walks that chain. Unwinding
373 /// reads the table, there is no profiler's hook that reads the pointer, and gcc gives the chain
374 /// up on the same functions for the same reason.
375 pub late_frame_pointer: bool,
376 /// Where a variadic call says how many vector registers it passed arguments in, when the
377 /// convention makes it say.
378 ///
379 /// SysV puts the count in `al` and a variadic callee reads it to decide whether to save the
380 /// vector argument registers at all, which is what makes a call to `printf` with no
381 /// floating point argument cheap.
382 pub vector_count: Option<PhysReg>,
383 /// How many bytes below the stack pointer a leaf function may use without moving it.
384 ///
385 /// A hundred and twenty eight on SysV and nothing on Windows. It is nothing in kernel code
386 /// on either, because an interrupt handler runs on the interrupted stack and writes over
387 /// exactly this, which is what `-mno-red-zone` is for.
388 pub red_zone: u32,
389 /// How many bytes a caller reserves below the call for the callee to spill its register
390 /// arguments into, which is thirty two on Windows and nothing on SysV.
391 pub shadow: u32,
392 /// What the stack pointer has to be a multiple of at the instruction that makes a call.
393 ///
394 /// Sixteen on every convention here, and it is a real obligation rather than a preference,
395 /// because a callee is entitled to use an aligned vector store on its own frame and gets a
396 /// fault rather than a wrong answer when a caller got this wrong.
397 pub stack_align: u32,
398 /// How many bytes the call instruction itself pushes before the callee starts running.
399 ///
400 /// Eight on x86-64, where the return address is on the stack, and nothing on a machine that
401 /// leaves it in a register. It is what makes the stack pointer misaligned on entry by
402 /// exactly one word, which every frame layout has to undo.
403 pub return_address: u32,
404 /// How many bytes one general purpose register takes when it is saved on the stack.
405 pub word: u32,
406 /// What DWARF calls each register, one list per class in the order the file numbers the
407 /// classes, and inside a list in the order the class numbers its registers.
408 ///
409 /// The two numberings are a real difference and not a formality. On x86-64 the machine puts
410 /// `rcx` at one and DWARF puts `rdx` there, so a table written with the machine's numbers is
411 /// well formed and describes the wrong registers, which is a backtrace with plausible
412 /// nonsense in it rather than an error. Shorter than the file when the classes at the end are
413 /// ones DWARF has no column for, and empty on a target nobody has written this down for yet.
414 pub dwarf: &'static [&'static [u16]],
415 /// The column an unwind table files the return address under.
416 ///
417 /// Not a register on x86-64, where it is sixteen and `rip` is not a register anything can
418 /// name, and a real one on a machine that returns through a link register.
419 pub dwarf_return_address: u16,
420 /// Where the word a stack protector's canary is copied from lives, on a convention that has
421 /// somewhere to put one.
422 ///
423 /// Here rather than beside the frame instructions because it is a fact about the runtime the
424 /// code is linked against rather than about the machine. The two x86-64 conventions share
425 /// every instruction the check is made of and disagree about this.
426 pub guard: Option<Guard>,
427 /// What a profiler's hook is called on this platform, on one that has one.
428 ///
429 /// Here for the same reason [`CallRegs::guard`] is: the names are the runtime's rather than the
430 /// machine's, and the two x86-64 conventions write the same call instruction and disagree about
431 /// what goes in it.
432 pub trace: Option<Trace>,
433 /// What this platform calls the routine a prologue reaches the pages of a large frame with, on
434 /// one where reaching them is the convention.
435 ///
436 /// Here for the same reason the two above are, and it is the clearest case of the three: the
437 /// walk itself is written out of the machine's own instructions and both x86-64 conventions
438 /// have them, and what the two disagree about is whether a frame may be taken in one step at
439 /// all. A prologue that leaves this out on Windows writes a function that faults on its own
440 /// locals.
441 pub chkstk: Option<Chkstk>,
442}
443
444impl CallRegs {
445 /// The number DWARF gives that register, or `None` for one it has no column for.
446 ///
447 /// The x87 stack is the case that answers `None` on x86-64, and it is not an omission: a
448 /// register whose name means whichever one is on top of the stack is not one a table can have
449 /// a column for. Nothing saves one across a call either, so nothing ever asks.
450 #[must_use]
451 pub fn dwarf(&self, class: RegClass, reg: PhysReg) -> Option<u16> {
452 self.dwarf.get(usize::from(class.number()))?.get(usize::from(reg.number())).copied()
453 }
454
455 /// Which register of that class DWARF gave that number to, which is [`Self::dwarf`] the other
456 /// way round, and `None` for a number no register of the class holds.
457 ///
458 /// What asks is a table written in the machine's own numbering rather than DWARF's, which is
459 /// what Windows unwinds from. A row a prologue produces carries DWARF's number, because that is
460 /// what the format two of the three platforms want is written in, and the machine's number is
461 /// this register's place in its class. The two disagree over the first eight general purpose
462 /// registers on x86-64 and nowhere else, which is the worst shape a disagreement can have: every
463 /// number is a register either way, so the table comes out well formed and about the wrong
464 /// registers.
465 #[must_use]
466 pub fn machine(&self, class: RegClass, dwarf: u16) -> Option<PhysReg> {
467 let numbers = self.dwarf.get(usize::from(class.number()))?;
468 let at = numbers.iter().position(|&number| number == dwarf)?;
469 Some(PhysReg::new(u8::try_from(at).ok()?))
470 }
471
472 /// Whether a call preserves that general purpose register.
473 #[must_use]
474 pub fn preserves_int(&self, reg: PhysReg) -> bool {
475 self.int_saved.contains(®)
476 }
477
478 /// Whether a call preserves that vector register.
479 #[must_use]
480 pub fn preserves_sse(&self, reg: PhysReg) -> bool {
481 self.sse_saved.contains(®)
482 }
483}
484
485/// Where one of the values a call passes is.
486#[derive(Debug, Clone, Copy, PartialEq, Eq)]
487pub enum Where {
488 /// In that register.
489 Reg(PhysReg),
490 /// That many bytes up the argument area, which is where the stack pointer points at the
491 /// instruction that makes the call and is one word above the return address in the callee.
492 Stack(u32),
493}
494
495/// Where the values a call passes are, worked out one after another.
496///
497/// [`crate::abi::Call`] answers a different question: whether a value travels in registers at all
498/// and in how many, which is what decides the shape of a signature and is settled before the IR
499/// for a function exists. This answers the question after it. Given values in the order the
500/// signature holds them, it says which register each one is in and how far up the argument area
501/// the ones that got no register are. Both count registers, and they agree about how many fit
502/// because they read the same lists, but they run at opposite ends of the compiler and neither
503/// can be the other.
504///
505/// Ask about each value in the order the signature holds them. Asking out of order answers about
506/// a different signature, because where a value is depends on every value before it.
507#[derive(Debug, Clone)]
508pub struct Places<'a> {
509 regs: &'a CallRegs,
510 int: usize,
511 sse: usize,
512 stack: u32,
513}
514
515impl<'a> Places<'a> {
516 /// Where the first value is, for a call under that convention.
517 #[must_use]
518 pub fn new(regs: &'a CallRegs) -> Self {
519 Self { regs, int: 0, sse: 0, stack: regs.shadow }
520 }
521
522 /// Where the next value is, when it travels in a general purpose register.
523 pub fn integer(&mut self) -> Where {
524 match self.regs.int_args.get(self.position(false)) {
525 Some(®) => {
526 self.int += 1;
527 Where::Reg(reg)
528 }
529 None => self.on_stack(self.regs.word, self.regs.word),
530 }
531 }
532
533 /// Where the next value is, when it travels in a vector register.
534 ///
535 /// `bytes` is how wide the value is, and it is asked for because of what happens once the
536 /// vector registers have run out. A register holds whatever is put in it, so up to that point
537 /// the width changes nothing, and the argument area is a run of words, so a `float` or a
538 /// `double` that got no register takes one word wherever it starts. A `_Float128` is neither:
539 /// it is sixteen bytes aligned to sixteen, so its slot is two words and begins on an even one.
540 /// A slot of one word would leave the argument behind it sitting on the top half of the value,
541 /// and the load that reads it back is a `movaps`, which faults on an address that is not a
542 /// multiple of sixteen rather than being slow.
543 pub fn float(&mut self, bytes: u32) -> Where {
544 match self.regs.sse_args.get(self.position(true)) {
545 Some(®) => {
546 self.sse += 1;
547 Where::Reg(reg)
548 }
549 // The alignment is the width, which [`Places::on_stack`] raises to a word for anything
550 // narrower, so this is the natural alignment of the value and not a rule of its own.
551 None => self.on_stack(bytes, bytes),
552 }
553 }
554
555 /// Where the next value is, when it travels in memory whatever is left.
556 ///
557 /// Every argument area is a run of whole words, so a value narrower than one still takes one
558 /// and a value that is not a whole number of them is rounded up. An alignment wider than a
559 /// word is respected, which is what a sixteen byte aligned structure passed by value needs.
560 pub fn on_stack(&mut self, size: u32, align: u32) -> Where {
561 let word = self.regs.word;
562 let at = self.stack.next_multiple_of(align.max(word));
563 self.stack = at.saturating_add(size.max(word).next_multiple_of(word));
564 Where::Stack(at)
565 }
566
567 /// How many bytes of argument area the values so far need, shadow space included.
568 #[must_use]
569 pub fn size(&self) -> u32 {
570 self.stack
571 }
572
573 /// How many general purpose argument registers the values so far took.
574 ///
575 /// What a variadic callee needs and nothing else does. `va_start` has to record how far into
576 /// each of the two register sequences the arguments the signature names got, because the first
577 /// argument it does not name is the one after them, and asking here is the only way to know
578 /// that is the same count the caller worked from.
579 #[must_use]
580 pub fn integers(&self) -> usize {
581 self.int
582 }
583
584 /// How many vector argument registers the values so far took.
585 #[must_use]
586 pub fn floats(&self) -> usize {
587 self.sse
588 }
589
590 /// The position the next value of a kind is at.
591 fn position(&self, sse: bool) -> usize {
592 if self.regs.shared_positions {
593 self.int + self.sse
594 } else if sse {
595 self.sse
596 } else {
597 self.int
598 }
599 }
600}
601
602impl fmt::Display for RegFile {
603 /// The file as a dump reads it, one class to a line.
604 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
605 for (_, info) in self.classes() {
606 writeln!(f, "class {} : i{} = {}", info.name, info.bits, info.regs.join(", "))?;
607 }
608 Ok(())
609 }
610}
611
612#[cfg(test)]
613mod tests {
614 use super::*;
615
616 static GPR: [&str; 3] = ["rax", "rcx", "rdx"];
617 static XMM: [&str; 2] = ["xmm0", "xmm1"];
618 static CLASSES: [ClassInfo; 2] = [
619 ClassInfo { name: "gpr", bits: 64, regs: &GPR, allocatable: true },
620 ClassInfo { name: "xmm", bits: 128, regs: &XMM, allocatable: true },
621 ];
622 static FILE: RegFile = RegFile::new(&CLASSES);
623
624 #[test]
625 fn a_class_is_found_by_its_name() {
626 let gpr = FILE.class_named("gpr").expect("the file has a gpr class");
627 assert_eq!(FILE.len(gpr), 3);
628 assert_eq!(FILE.class(gpr).map(|info| info.bits), Some(64));
629 assert_eq!(FILE.class_named("vec"), None);
630 }
631
632 #[test]
633 fn a_register_is_found_by_its_name_and_names_itself_back() {
634 let (class, reg) = FILE.reg_named("xmm1").expect("the file has xmm1");
635 assert_eq!(FILE.class(class).map(|info| info.name), Some("xmm"));
636 assert_eq!(reg.number(), 1);
637 assert_eq!(FILE.name(class, reg), Some("xmm1"));
638 assert_eq!(FILE.reg_named("r15"), None);
639 }
640
641 #[test]
642 fn a_number_past_the_end_of_a_class_has_no_name() {
643 let gpr = FILE.class_named("gpr").expect("the file has a gpr class");
644 assert_eq!(FILE.name(gpr, PhysReg::new(3)), None);
645 assert_eq!(FILE.name(RegClass::new(7), PhysReg::new(0)), None);
646 }
647
648 #[test]
649 fn a_file_that_names_two_registers_alike_says_so() {
650 assert_eq!(FILE.duplicate(), None);
651 static BOTH: [ClassInfo; 2] = [
652 ClassInfo { name: "gpr", bits: 64, regs: &GPR, allocatable: true },
653 ClassInfo { name: "shadow", bits: 64, regs: &GPR, allocatable: true },
654 ];
655 assert_eq!(RegFile::new(&BOTH).duplicate(), Some("rax"));
656 }
657
658 #[test]
659 fn a_class_nothing_allocates_from_is_still_a_class_in_every_other_way() {
660 static WITH_STACK: [ClassInfo; 2] = [
661 ClassInfo { name: "gpr", bits: 64, regs: &GPR, allocatable: true },
662 ClassInfo { name: "x87", bits: 80, regs: &XMM, allocatable: false },
663 ];
664 let file = RegFile::new(&WITH_STACK);
665 let stack = file.class_named("x87").expect("the file has an x87 class");
666
667 assert!(!file.allocatable(stack));
668 assert!(file.allocatable(file.class_named("gpr").expect("the file has a gpr class")));
669
670 // Everything else about it works, which is the point of describing a class the allocator
671 // will not touch: the registers are counted, are named, and name themselves back.
672 assert_eq!(file.len(stack), 2);
673 assert_eq!(file.name(stack, PhysReg::new(1)), Some("xmm1"));
674 assert_eq!(file.reg_named("xmm1"), Some((stack, PhysReg::new(1))));
675 }
676
677 #[test]
678 fn a_class_the_file_does_not_have_is_not_one_to_allocate_from_either() {
679 assert!(!FILE.allocatable(RegClass::new(7)));
680 }
681
682 #[test]
683 fn the_file_prints_one_class_to_a_line() {
684 assert_eq!(
685 FILE.to_string(),
686 "class gpr : i64 = rax, rcx, rdx\nclass xmm : i128 = xmm0, xmm1\n"
687 );
688 }
689
690 /// Two integer registers, two vector registers and nothing else, so running out of them takes
691 /// three arguments rather than seven and the interesting case is the one being tested.
692 fn convention(shared: bool, shadow: u32) -> CallRegs {
693 static INT: [PhysReg; 2] = [PhysReg::new(0), PhysReg::new(1)];
694 static SSE: [PhysReg; 2] = [PhysReg::new(10), PhysReg::new(11)];
695 static NONE: [PhysReg; 0] = [];
696 // The registers are invented and the description is not, so it is picked by the one flag
697 // the two real conventions disagree about, which is the flag these tests vary.
698 let abi: &'static AbiDescription =
699 if shared { &rucc_abi::abis::WIN64 } else { &rucc_abi::abis::SYSV_AMD64 };
700 CallRegs {
701 abi,
702 int_class: RegClass::new(0),
703 sse_class: RegClass::new(1),
704 int_args: &INT,
705 sse_args: &SSE,
706 shared_positions: shared,
707 int_returns: &INT,
708 sse_returns: &SSE,
709 x87_returns: &NONE,
710 int_saved: &NONE,
711 sse_saved: &NONE,
712 int_order: &INT,
713 sse_order: &SSE,
714 stack_pointer: PhysReg::new(4),
715 frame_pointer: PhysReg::new(5),
716 late_frame_pointer: false,
717 vector_count: None,
718 red_zone: 0,
719 shadow,
720 stack_align: 16,
721 return_address: 8,
722 word: 8,
723 // Empty, which is all a convention made up for a test of argument placement needs to
724 // say about a question it never asks.
725 dwarf: &[],
726 dwarf_return_address: 16,
727 guard: None,
728 trace: None,
729 chkstk: None,
730 }
731 }
732
733 #[test]
734 fn counting_each_kind_separately_leaves_the_first_vector_register_to_the_first_float() {
735 let regs = convention(false, 0);
736 let mut places = Places::new(®s);
737 assert_eq!(places.integer(), Where::Reg(PhysReg::new(0)));
738 assert_eq!(places.integer(), Where::Reg(PhysReg::new(1)));
739 // Two integers went past, and a convention that counts separately has not spent a vector
740 // register on either of them.
741 assert_eq!(places.float(8), Where::Reg(PhysReg::new(10)));
742 assert_eq!(places.size(), 0);
743 }
744
745 #[test]
746 fn counting_one_position_for_both_skips_the_register_the_other_kind_would_have_used() {
747 let regs = convention(true, 0);
748 let mut places = Places::new(®s);
749 assert_eq!(places.integer(), Where::Reg(PhysReg::new(0)));
750 // The second position, so the second vector register, and the second integer register is
751 // spent whether anything is in it or not.
752 assert_eq!(places.float(8), Where::Reg(PhysReg::new(11)));
753 assert_eq!(places.integer(), Where::Stack(0));
754 }
755
756 #[test]
757 fn running_out_of_one_kind_of_register_does_not_touch_the_other() {
758 let regs = convention(false, 0);
759 let mut places = Places::new(®s);
760 assert_eq!(places.integer(), Where::Reg(PhysReg::new(0)));
761 assert_eq!(places.integer(), Where::Reg(PhysReg::new(1)));
762 assert_eq!(places.integer(), Where::Stack(0));
763 assert_eq!(places.float(8), Where::Reg(PhysReg::new(10)));
764 assert_eq!(places.size(), 8);
765 }
766
767 #[test]
768 fn the_argument_area_starts_above_the_shadow_space_and_keeps_every_value_aligned() {
769 let regs = convention(false, 32);
770 let mut places = Places::new(®s);
771 // A Windows caller reserves this whether it passes anything on the stack or not, which is
772 // why an empty area is thirty two bytes rather than none.
773 assert_eq!(places.size(), 32);
774 assert_eq!(places.on_stack(4, 4), Where::Stack(32));
775 // Sixteen byte alignment skips the word at 40, which is what a vector or an over-aligned
776 // structure passed by value asks for. The four byte value before it still took a whole
777 // word, which is why the skipped word is there to skip.
778 assert_eq!(places.on_stack(16, 16), Where::Stack(48));
779 assert_eq!(places.on_stack(8, 8), Where::Stack(64));
780 assert_eq!(places.size(), 72);
781 }
782
783 #[test]
784 fn a_float_wider_than_a_word_takes_two_of_them_once_the_registers_are_gone() {
785 let regs = convention(false, 0);
786 let mut places = Places::new(®s);
787 assert_eq!(places.float(16), Where::Reg(PhysReg::new(10)));
788 assert_eq!(places.float(16), Where::Reg(PhysReg::new(11)));
789 // The vector registers are gone, and so are the two general purpose ones, so the rest of
790 // this is in the area. The word the integer took is not where the first quad starts,
791 // because sixteen bytes aligned to sixteen skips the odd word above it, and the second
792 // quad is sixteen bytes above the first rather than eight.
793 assert_eq!(places.integer(), Where::Reg(PhysReg::new(0)));
794 assert_eq!(places.integer(), Where::Reg(PhysReg::new(1)));
795 assert_eq!(places.integer(), Where::Stack(0));
796 assert_eq!(places.float(16), Where::Stack(16));
797 assert_eq!(places.float(16), Where::Stack(32));
798 assert_eq!(places.size(), 48);
799 // A float narrower than a word still takes one, which is what it took before any of this.
800 assert_eq!(places.float(4), Where::Stack(48));
801 assert_eq!(places.size(), 56);
802 }
803}