Skip to main content

rucc_ir/
func.rs

1//! The function: its blocks, its instructions, its values, and the tables they live in.
2//!
3//! Design: `spec/08-ir.md` sections 8.1 and 8.6.
4//!
5//! One [`Func`] owns everything in it. Nothing is boxed and nothing is individually freed: the
6//! instructions are a flat vector, a reference to one is a four-byte index, and the whole
7//! function is dropped in one go. The same shape as the AST, for the same reasons.
8//!
9//! Two things are not flat, and both for the same reason, which is that SSA construction
10//! finishes a loop header long after it has built the blocks inside the loop.
11//!
12//! The instructions in a block are a doubly linked list rather than a run, because the
13//! optimizer inserts and removes instructions constantly and a run would move every
14//! instruction after the edit, invalidating every [`Inst`] anybody was holding.
15//!
16//! A block's parameters are a `Vec` rather than a run in a pool, because a run in a pool
17//! cannot grow once something else has been put after it, and adding a parameter to a loop
18//! header is exactly the operation that has to grow one.
19//!
20//! # CFG invariants
21//!
22//! The entry block has no predecessors and its parameters are the function's arguments in
23//! their C-level form, before the ABI has been applied. Every other block ends in exactly one
24//! terminator and contains no terminator anywhere else. These are checked by the verifier
25//! rather than by the builder, because a function under construction breaks all of them and
26//! the useful question is whether it still does when the pass that was building it says it has
27//! finished.
28
29use std::ops::{Index, IndexMut};
30
31use rucc_base::{Idx, Symbol};
32use rucc_diag::Span;
33use rucc_target::Slot;
34
35use crate::inst::{
36    Abi, AbiList, AsmInfo, Block, BlockCall, BlockCallList, BlockData, Bulk, CallInfo, Def, Extra,
37    Imm, ImmList, Inst, InstData, InstLayout, MemInfo, Sig, Signature, SlotList, SwitchInfo,
38    VaInfo, Value, ValueData, ValueList,
39};
40use crate::module::{Linkage, Visibility};
41use crate::{Attrs, Facts, Flags, FloatPred, IntPred, MemOrder, Opcode, PrefetchHint, RmwOp, Type};
42
43/// One function.
44#[derive(Debug)]
45pub struct Func {
46    /// The name it is called by, which is what a direct call to it names.
47    pub name: Symbol,
48    /// How the linker sees it. `Internal` for a `static` function.
49    pub linkage: Linkage,
50    /// How the dynamic linker sees it.
51    pub visibility: Visibility,
52    /// The section to put it in, from `__attribute__((section(...)))`, or `None` to let the
53    /// object writer choose.
54    pub section: Option<Symbol>,
55    /// What its first instruction has to be aligned to, from `__attribute__((aligned(...)))`, or
56    /// `None` for the alignment the target gives every function anyway.
57    ///
58    /// A raise and never a lower, the way the attribute is everywhere: a function asked to be at
59    /// a multiple of two hundred and fifty six is at one, and one asked for less than the target's
60    /// own alignment keeps the target's.
61    pub align: Option<u32>,
62    /// What is true of the whole function, which is what a caller reads when it wants to know
63    /// what a call to it does without looking inside.
64    pub attrs: Attrs,
65
66    values: Vec<ValueData>,
67    insts: Vec<InstData>,
68    inst_layout: Vec<InstLayout>,
69    inst_spans: Vec<Span>,
70    blocks: Vec<BlockData>,
71
72    value_pool: Vec<Value>,
73    block_calls: Vec<BlockCall>,
74    imms: Vec<Imm>,
75    mem: Vec<MemInfo>,
76    calls: Vec<CallInfo>,
77    abis: Vec<Abi>,
78    switches: Vec<SwitchInfo>,
79    asms: Vec<AsmInfo>,
80    slots: Vec<Slot>,
81    va_objects: Vec<VaInfo>,
82    signatures: Vec<Signature>,
83    facts: Vec<(Value, Facts)>,
84    labels: Vec<(Block, Symbol)>,
85
86    first_block: Option<Block>,
87    last_block: Option<Block>,
88}
89
90impl Func {
91    /// A function with that name and that signature, and nothing in it.
92    ///
93    /// The signature becomes signature zero, which is what [`Func::signature`] gives back. The
94    /// entry block is not created here, because the caller is about to create it and give it
95    /// the parameters, and a half-built entry block is worse than no entry block. So a
96    /// function fresh from here is a declaration, and stops being one when it gets a block.
97    #[must_use]
98    pub fn new(name: Symbol, signature: Signature) -> Self {
99        Self {
100            name,
101            linkage: Linkage::External,
102            visibility: Visibility::Default,
103            section: None,
104            align: None,
105            attrs: Attrs::NONE,
106            values: Vec::new(),
107            insts: Vec::new(),
108            inst_layout: Vec::new(),
109            inst_spans: Vec::new(),
110            blocks: Vec::new(),
111            value_pool: Vec::new(),
112            block_calls: Vec::new(),
113            imms: Vec::new(),
114            mem: Vec::new(),
115            calls: Vec::new(),
116            abis: Vec::new(),
117            switches: Vec::new(),
118            asms: Vec::new(),
119            slots: Vec::new(),
120            va_objects: Vec::new(),
121            signatures: vec![signature],
122            facts: Vec::new(),
123            labels: Vec::new(),
124            first_block: None,
125            last_block: None,
126        }
127    }
128
129    /// Its own signature.
130    #[must_use]
131    pub fn signature(&self) -> &Signature {
132        &self.signatures[0]
133    }
134
135    /// Gives the function a different signature of its own.
136    ///
137    /// There is one caller and it is the back end pass that puts an integer the machine has no
138    /// register for into the pair of registers it travels in. A parameter that becomes two is a
139    /// parameter list that is not the one the function was created with, and the entry block's
140    /// parameters have to say the same thing, which is why this is next to
141    /// [`Func::retain_params`] in what a pass has to keep straight rather than something the
142    /// middle end reaches for. Nothing else changes a function's own signature, because a
143    /// signature is what its callers were compiled against.
144    pub fn set_signature(&mut self, signature: Signature) {
145        self.signatures[0] = signature;
146    }
147
148    /// Every signature the function holds, its own first and then the ones its calls name.
149    pub fn signatures(&self) -> impl Iterator<Item = &Signature> {
150        self.signatures.iter()
151    }
152
153    /// Records a signature a `call_indirect` is made with, and gives back its index.
154    pub fn add_signature(&mut self, signature: Signature) -> Sig {
155        self.signatures.push(signature);
156        Idx::from_usize(self.signatures.len() - 1)
157    }
158
159    /// The entry block, which is the first one in layout order.
160    ///
161    /// `None` only before one has been created. The verifier is what insists a finished
162    /// function has one.
163    #[must_use]
164    pub fn entry(&self) -> Option<Block> {
165        self.first_block
166    }
167
168    /// Whether this only says the function exists somewhere, which is a function with no
169    /// blocks in it.
170    ///
171    /// `extern int puts(const char *);` and every other declaration of something defined in
172    /// another object is one of these, and it is here rather than left out of the module
173    /// because a call needs its signature and its linkage.
174    #[must_use]
175    pub fn is_declaration(&self) -> bool {
176        self.first_block.is_none()
177    }
178
179    // Blocks.
180
181    /// Creates a block with no parameters and no instructions, at the end of the layout.
182    pub fn create_block(&mut self) -> Block {
183        let block = Idx::from_usize(self.blocks.len());
184        self.blocks.push(BlockData { prev: self.last_block, ..BlockData::default() });
185        match self.last_block {
186            Some(last) => self.blocks[last.index()].next = Some(block),
187            None => self.first_block = Some(block),
188        }
189        self.last_block = Some(block);
190        block
191    }
192
193    /// Takes a block out of the layout, along with everything in it.
194    ///
195    /// The block keeps its number, the way a removed instruction keeps its own, because
196    /// renumbering would move every block after it and invalidate every index anybody was
197    /// holding. What it stops being is a block of this function: nothing walks it, nothing
198    /// prints it, and the values defined in it are as gone as the instructions that defined
199    /// them. Deleting one whose branches something still reaches is how a function ends up
200    /// branching to nowhere, so the caller is the one that has to know nothing reaches it.
201    ///
202    /// # Panics
203    ///
204    /// Panics if the block is the entry block, which is the one block a function has to have.
205    pub fn remove_block(&mut self, block: Block) {
206        assert!(self.first_block != Some(block), "the entry block is not removable");
207        let (prev, next) = (self.blocks[block.index()].prev, self.blocks[block.index()].next);
208        match prev {
209            Some(prev) => self.blocks[prev.index()].next = next,
210            None => self.first_block = next,
211        }
212        match next {
213            Some(next) => self.blocks[next.index()].prev = prev,
214            None => self.last_block = prev,
215        }
216        // The instructions say they are in no block now, which is what a removed instruction
217        // says, so that asking one where it is gives an answer rather than a block nothing
218        // walks.
219        let insts: Vec<Inst> = self.insts(block).collect();
220        for inst in insts {
221            self.inst_layout[inst.index()] = InstLayout::default();
222        }
223        self.blocks[block.index()] = BlockData::default();
224    }
225
226    /// Adds a parameter of that type to a block, and gives back the value it arrives as.
227    ///
228    /// Every predecessor's branch has to grow an argument to match, which is
229    /// [`Func::append_arg`], and the verifier is what notices if one of them did not.
230    ///
231    /// # Panics
232    ///
233    /// Panics if the block already has four billion parameters, which no block does.
234    pub fn append_param(&mut self, block: Block, ty: Type) -> Value {
235        let index = u32::try_from(self.blocks[block.index()].params.len())
236            .expect("a block with four billion parameters");
237        let value = self.add_value(ValueData { ty, def: Def::Param { block, index } });
238        self.blocks[block.index()].params.push(value);
239        value
240    }
241
242    /// Drops the parameters of a block that a predicate turns down, and renumbers the rest.
243    ///
244    /// The predicate is asked about each parameter in the order the block takes them. A
245    /// parameter that goes has to take the argument in the same position out of every branch
246    /// to the block, which is the caller's work rather than this method's, because only the
247    /// caller knows which branches there are. This is what removing a redundant block
248    /// parameter is, and SSA construction is the thing that makes them.
249    ///
250    /// # Panics
251    ///
252    /// Panics if the block has four billion parameters, which no block does.
253    pub fn retain_params(&mut self, block: Block, mut keep: impl FnMut(Value) -> bool) {
254        let mut params = std::mem::take(&mut self.blocks[block.index()].params);
255        params.retain(|&value| keep(value));
256        for (index, &value) in params.iter().enumerate() {
257            let index = u32::try_from(index).expect("a block with four billion parameters");
258            self.values[value.index()].def = Def::Param { block, index };
259        }
260        self.blocks[block.index()].params = params;
261    }
262
263    /// Gives a value a different type, leaving where it comes from alone.
264    ///
265    /// There is one caller and it is the back end pass that puts an integer of a width the
266    /// machine has no register for into the width it does have one for. Nothing in the middle
267    /// end changes a value's type, because a value's type is what the instruction that made it
268    /// produces and changing one without changing the other is how an IR stops meaning
269    /// anything. That pass changes both, which is why this is a method and not a field.
270    ///
271    /// # Panics
272    ///
273    /// Panics if the value is not one of this function's.
274    pub fn retype(&mut self, value: Value, ty: Type) {
275        self.values[value.index()].ty = ty;
276    }
277
278    /// Every value the function has, including ones whose defining instruction has gone.
279    ///
280    /// In the order they were created, which is the order a pass that walks all of them wants:
281    /// a value is defined before it is used, so a walk in this order sees a definition first.
282    pub fn values(&self) -> impl Iterator<Item = Value> + use<'_> {
283        (0..self.values.len()).map(Idx::from_usize)
284    }
285
286    /// Every block, in layout order.
287    pub fn blocks(&self) -> impl Iterator<Item = Block> + use<'_> {
288        std::iter::successors(self.first_block, move |&block| self.blocks[block.index()].next)
289    }
290
291    /// Every instruction in a block, in order.
292    pub fn insts(&self, block: Block) -> impl Iterator<Item = Inst> + use<'_> {
293        std::iter::successors(self.blocks[block.index()].first, move |&inst| {
294            self.inst_layout[inst.index()].next
295        })
296    }
297
298    /// Every instruction in a block, last first.
299    ///
300    /// Which is the order a liveness walk needs, and it is here rather than at the caller because
301    /// the layout links are private and collecting the block into a vector to reverse it is an
302    /// allocation per block per round of a fixpoint.
303    pub fn insts_backwards(&self, block: Block) -> impl Iterator<Item = Inst> + use<'_> {
304        std::iter::successors(self.blocks[block.index()].last, move |&inst| {
305            self.inst_layout[inst.index()].prev
306        })
307    }
308
309    /// The last instruction of a block, which is its terminator once it is finished.
310    #[must_use]
311    pub fn terminator(&self, block: Block) -> Option<Inst> {
312        self.blocks[block.index()].last.filter(|&inst| self.is_terminator(inst))
313    }
314
315    /// Whether control leaves the block at this instruction.
316    ///
317    /// A question for the function rather than for the instruction, because inline assembly is
318    /// the one case where the opcode is not enough: `asm goto` has labels and everything else
319    /// does not, and the labels are in the function's table rather than on the instruction.
320    #[must_use]
321    pub fn is_terminator(&self, inst: Inst) -> bool {
322        let data = &self[inst];
323        match data.extra {
324            Extra::Asm(info) => {
325                data.opcode.is_terminator() || !self.asms[info.index()].targets.is_empty()
326            }
327            _ => data.opcode.is_terminator(),
328        }
329    }
330
331    // Instructions.
332
333    /// Creates an instruction and its result values, without putting it in a block.
334    ///
335    /// The results are allocated here and are contiguous, which is what lets an instruction
336    /// hold the first of them and a count rather than a list.
337    ///
338    /// # Panics
339    ///
340    /// Panics if `results` has more than 255 types, which no instruction in the set does.
341    pub fn create_inst(&mut self, mut data: InstData, results: &[Type], span: Span) -> Inst {
342        let inst = Idx::from_usize(self.insts.len());
343        data.results = u8::try_from(results.len()).expect("an instruction with too many results");
344        data.first_result = results.first().map(|_| Idx::from_usize(self.values.len()));
345        for (index, &ty) in results.iter().enumerate() {
346            let index = u8::try_from(index).expect("checked just above");
347            self.add_value(ValueData { ty, def: Def::Result { inst, index } });
348        }
349        self.insts.push(data);
350        self.inst_layout.push(InstLayout::default());
351        self.inst_spans.push(span);
352        inst
353    }
354
355    /// Puts an instruction at the end of a block.
356    ///
357    /// # Panics
358    ///
359    /// Panics if the instruction is already in a block. Moving one is removing it and
360    /// appending it, and doing it by accident is how a linked list ends up in two pieces.
361    pub fn append_inst(&mut self, block: Block, inst: Inst) {
362        assert!(self.inst_layout[inst.index()].block.is_none(), "the instruction is in a block");
363        let last = self.blocks[block.index()].last;
364        self.inst_layout[inst.index()] = InstLayout { block: Some(block), prev: last, next: None };
365        match last {
366            Some(last) => self.inst_layout[last.index()].next = Some(inst),
367            None => self.blocks[block.index()].first = Some(inst),
368        }
369        self.blocks[block.index()].last = Some(inst);
370    }
371
372    /// Puts an instruction immediately before another one, in the block that one is in.
373    ///
374    /// # Panics
375    ///
376    /// Panics if `inst` is already in a block, or if `before` is not in one.
377    pub fn insert_before(&mut self, inst: Inst, before: Inst) {
378        assert!(self.inst_layout[inst.index()].block.is_none(), "the instruction is in a block");
379        let at = self.inst_layout[before.index()];
380        let block = at.block.expect("the instruction to insert before is not in a block");
381        self.inst_layout[inst.index()] =
382            InstLayout { block: Some(block), prev: at.prev, next: Some(before) };
383        self.inst_layout[before.index()].prev = Some(inst);
384        match at.prev {
385            Some(prev) => self.inst_layout[prev.index()].next = Some(inst),
386            None => self.blocks[block.index()].first = Some(inst),
387        }
388    }
389
390    /// Puts an instruction immediately after another one, in the block that one is in.
391    ///
392    /// The mirror of [`Func::insert_before`], and it exists because a pass that has to talk about
393    /// a value an instruction produced has nowhere else to put what it is adding. Check insertion
394    /// is the caller: `check_deriv` is handed the pointer the derivation produced, so it goes
395    /// after the derivation and no amount of rearranging moves it earlier.
396    ///
397    /// # Panics
398    ///
399    /// Panics if `inst` is already in a block, if `after` is not in one, or if `after` is the
400    /// block's terminator, since nothing may come between a terminator and the branch it is.
401    pub fn insert_after(&mut self, inst: Inst, after: Inst) {
402        assert!(self.inst_layout[inst.index()].block.is_none(), "the instruction is in a block");
403        let at = self.inst_layout[after.index()];
404        let block = at.block.expect("the instruction to insert after is not in a block");
405        assert!(at.next.is_some(), "nothing goes after a terminator");
406        self.inst_layout[inst.index()] =
407            InstLayout { block: Some(block), prev: Some(after), next: at.next };
408        self.inst_layout[after.index()].next = Some(inst);
409        if let Some(next) = at.next {
410            self.inst_layout[next.index()].prev = Some(inst);
411        }
412    }
413
414    /// Takes an instruction out of its block, leaving it and its results in the tables.
415    ///
416    /// The instruction is not deleted, because deleting it would move every instruction after
417    /// it. A removed instruction is unreachable from any block and is dropped when the whole
418    /// function is.
419    ///
420    /// # Panics
421    ///
422    /// Panics if the instruction is not in a block.
423    pub fn remove_inst(&mut self, inst: Inst) {
424        let at = self.inst_layout[inst.index()];
425        let block = at.block.expect("the instruction is not in a block");
426        match at.prev {
427            Some(prev) => self.inst_layout[prev.index()].next = at.next,
428            None => self.blocks[block.index()].first = at.next,
429        }
430        match at.next {
431            Some(next) => self.inst_layout[next.index()].prev = at.prev,
432            None => self.blocks[block.index()].last = at.prev,
433        }
434        self.inst_layout[inst.index()] = InstLayout::default();
435    }
436
437    /// The block an instruction is in, or `None` if it has been removed from one.
438    #[must_use]
439    pub fn block_of(&self, inst: Inst) -> Option<Block> {
440        self.inst_layout[inst.index()].block
441    }
442
443    /// The version of memory an instruction reads, when the function carries memory SSA.
444    ///
445    /// Document 09 of `spec/optimizer`. Memory is a value of type `mem`, it is the last operand
446    /// of every instruction that touches memory, and it is absent in a function that does not
447    /// carry it, which is what `-O0` and `-O1` produce. Absent means unordered with respect to
448    /// everything, so a reader that gets `None` asks the alias analysis directly.
449    ///
450    /// The operand is last rather than first on purpose. Every other operand keeps the position
451    /// it had, so a pass that reads the address of a load as `args[0]` goes on working whether
452    /// or not memory has been threaded, and the only code that has to know about the extra
453    /// operand is this accessor and the verifier.
454    #[must_use]
455    pub fn mem_in(&self, inst: Inst) -> Option<Value> {
456        let args = &self[self[inst].args];
457        args.last().copied().filter(|&arg| self[arg].ty.is_mem())
458    }
459
460    /// The version of memory an instruction produces, when it writes memory and the function
461    /// carries memory SSA.
462    ///
463    /// Last among the results, for the reason [`Func::mem_in`] is last among the operands. A
464    /// `load` never has one, because it reads memory without changing it.
465    ///
466    /// Nothing reads the last version in a function, and that means nothing. A store whose
467    /// memory result has no reader is not dead, and what decides whether it is dead is dead
468    /// store elimination, which is document 17's.
469    #[must_use]
470    pub fn mem_out(&self, inst: Inst) -> Option<Value> {
471        self[inst].results().last().filter(|&result| self[result].ty.is_mem())
472    }
473
474    /// A bulk copy or fill taken apart, or nothing where the instruction is not one.
475    ///
476    /// The length is an operand on a bulk operation over an object whose length the program works
477    /// out and is [`MemInfo::size`] on every other one. Both shapes are here so that a pass which
478    /// asks this cannot read the payload's number on the one where it is not the count: what it
479    /// gets is the operand or nothing, and nothing is the only answer that means the payload.
480    ///
481    /// Memory is the last operand where the function carries it, which is why the length is found
482    /// by position from the front rather than from the back.
483    #[must_use]
484    pub fn bulk(&self, inst: Inst) -> Option<Bulk> {
485        if !matches!(self[inst].opcode, Opcode::Memcpy | Opcode::Memmove | Opcode::Memset) {
486            return None;
487        }
488        let all = &self[self[inst].args];
489        let args = &all[..all.len() - usize::from(self.mem_in(inst).is_some())];
490        let [to, with, rest @ ..] = args else { return None };
491        Some(Bulk { to: *to, with: *with, length: rest.first().copied() })
492    }
493
494    /// Whether an instruction has been threaded onto the memory chain.
495    #[must_use]
496    pub fn carries_mem(&self, inst: Inst) -> bool {
497        self.mem_in(inst).is_some() || self.mem_out(inst).is_some()
498    }
499
500    /// The same instruction with a version of memory threaded through it.
501    ///
502    /// A result cannot be added to an instruction that already exists, because the results of one
503    /// are values next to each other and there is no room after them. So threading memory makes a
504    /// new instruction and the caller puts it where the old one was, forwards the old results to
505    /// the new ones, which are at the same positions, and deletes the old one. That is what memory
506    /// SSA construction does in one pass over the function.
507    ///
508    /// The new instruction is not in any block. Its results are what the old one produced, in the
509    /// same order, and then the new version of memory where the opcode writes memory.
510    ///
511    /// # Panics
512    ///
513    /// Panics if `incoming` is not memory, if the instruction does not touch memory, or if it is
514    /// already on the chain. All three are a construction bug rather than bad input.
515    pub fn with_mem(&mut self, inst: Inst, incoming: Value) -> Inst {
516        assert!(self[incoming].ty.is_mem(), "the incoming version of memory is not memory");
517        assert!(self[inst].opcode.touches_memory(), "this does not touch memory");
518        assert!(self.mem_in(inst).is_none(), "this is already on the memory chain");
519        let data = self[inst];
520        let mut args = self[data.args].to_vec();
521        args.push(incoming);
522        let mut results: Vec<Type> = data.results().map(|result| self[result].ty).collect();
523        if data.opcode.writes_memory() {
524            results.push(Type::MEM);
525        }
526        let span = self.span(inst);
527        let args = self.push_values(&args);
528        self.create_inst(InstData { args, ..data }, &results, span)
529    }
530
531    /// The same instruction with the version of memory taken back off.
532    ///
533    /// The inverse of [`Func::with_mem`] and the same shape for the same reason: a result cannot be
534    /// taken off an instruction that already exists, so this makes a new one and the caller puts it
535    /// where the old one was, forwards the results it kept, which are at the same positions, and
536    /// deletes the old one. The memory result has no forwarding to do, because taking the chain off
537    /// is only ever done when nothing reads it any more.
538    ///
539    /// The new instruction is not in any block. Its results are what the old one produced without
540    /// the version of memory at the end of them.
541    ///
542    /// # Panics
543    ///
544    /// Panics if the instruction is not on the chain, which is a caller that did not look first.
545    pub fn without_mem(&mut self, inst: Inst) -> Inst {
546        assert!(self.carries_mem(inst), "this is not on the memory chain");
547        let data = self[inst];
548        let mut args = self[data.args].to_vec();
549        if self.mem_in(inst).is_some() {
550            args.pop();
551        }
552        let results: Vec<Type> =
553            data.results().map(|result| self[result].ty).filter(|ty| !ty.is_mem()).collect();
554        let span = self.span(inst);
555        let args = self.push_values(&args);
556        self.create_inst(InstData { args, ..data }, &results, span)
557    }
558
559    /// Where an instruction came from in the source.
560    #[must_use]
561    pub fn span(&self, inst: Inst) -> Span {
562        self.inst_spans[inst.index()]
563    }
564
565    /// Where an instruction branches to, which is empty when it does not branch.
566    ///
567    /// This is the one place that knows a `switch` keeps its targets in a side table and
568    /// `asm goto` in another one, so nothing walking the CFG has to.
569    pub fn successors(&self, inst: Inst) -> impl Iterator<Item = BlockCall> + use<'_> {
570        self.block_calls[self.target_list(inst).as_usize_range()].iter().copied()
571    }
572
573    /// Where a terminator keeps its targets, for something that edits them rather than reads
574    /// them.
575    ///
576    /// [`Func::successors`] is what walking the CFG wants. This is what recording an edge
577    /// wants, because an edge that will grow an argument later has to be named by its place in
578    /// the table rather than by the block it went to.
579    #[must_use]
580    pub fn target_list(&self, inst: Inst) -> BlockCallList {
581        match self[inst].extra {
582            Extra::Targets(targets) => targets,
583            Extra::Switch(info) => self.switches[info.index()].targets,
584            Extra::Asm(info) => self.asms[info.index()].targets,
585            _ => BlockCallList::EMPTY,
586        }
587    }
588
589    // The pools.
590
591    /// Records a run of value operands.
592    pub fn push_values(&mut self, values: &[Value]) -> ValueList {
593        let start = Idx::from_usize(self.value_pool.len());
594        self.value_pool.extend_from_slice(values);
595        ValueList::new(start, Idx::from_usize(self.value_pool.len()))
596    }
597
598    /// Adds one value to the end of a run, giving back the run it became.
599    ///
600    /// The run grows in place when nothing has been put after it, which is the case while a
601    /// list is being built. Otherwise it is copied to the end and the old space is left
602    /// behind, which is what makes adding a parameter to a loop header possible at all. That
603    /// happens once per value carried around a loop, so the copying is not what costs.
604    pub fn append_arg(&mut self, list: ValueList, value: Value) -> ValueList {
605        let range = list.as_usize_range();
606        if range.end == self.value_pool.len() {
607            self.value_pool.push(value);
608            return ValueList::new(Idx::from_usize(range.start), Idx::from_usize(range.end + 1));
609        }
610        let start = self.value_pool.len();
611        self.value_pool.extend_from_within(range);
612        self.value_pool.push(value);
613        ValueList::new(Idx::from_usize(start), Idx::from_usize(self.value_pool.len()))
614    }
615
616    /// Replaces the values in a run, which is what substituting one definition for another is.
617    ///
618    /// A run is a run whether it is an instruction's operands or a branch's arguments, so this
619    /// is the whole of the rewriting a substitution has to do.
620    pub fn rewrite(&mut self, list: ValueList, mut with: impl FnMut(Value) -> Value) {
621        for value in &mut self.value_pool[list.as_usize_range()] {
622            *value = with(*value);
623        }
624    }
625
626    /// Records a run of branch targets.
627    pub fn push_block_calls(&mut self, calls: &[BlockCall]) -> BlockCallList {
628        let start = Idx::from_usize(self.block_calls.len());
629        self.block_calls.extend_from_slice(calls);
630        BlockCallList::new(start, Idx::from_usize(self.block_calls.len()))
631    }
632
633    /// Replaces one branch target, which is what redirecting an edge is.
634    pub fn set_block_call(&mut self, at: Idx<BlockCall>, call: BlockCall) {
635        self.block_calls[at.index()] = call;
636    }
637
638    /// Records a run of case values.
639    pub fn push_imms(&mut self, imms: &[Imm]) -> ImmList {
640        let start = Idx::from_usize(self.imms.len());
641        self.imms.extend_from_slice(imms);
642        ImmList::new(start, Idx::from_usize(self.imms.len()))
643    }
644
645    /// Records a constant.
646    pub fn add_imm(&mut self, imm: Imm) -> Idx<Imm> {
647        self.imms.push(imm);
648        Idx::from_usize(self.imms.len() - 1)
649    }
650
651    /// Records where each eightbyte of an object travelled.
652    pub fn push_slots(&mut self, slots: &[Slot]) -> SlotList {
653        let start = Idx::from_usize(self.slots.len());
654        self.slots.extend_from_slice(slots);
655        SlotList::new(start, Idx::from_usize(self.slots.len()))
656    }
657
658    /// Records an object read off a variable argument list.
659    pub fn add_va_object(&mut self, info: VaInfo) -> Idx<VaInfo> {
660        self.va_objects.push(info);
661        Idx::from_usize(self.va_objects.len() - 1)
662    }
663
664    /// Records what an access does.
665    pub fn add_mem(&mut self, info: MemInfo) -> Idx<MemInfo> {
666        self.mem.push(info);
667        Idx::from_usize(self.mem.len() - 1)
668    }
669
670    /// Records what the ABI asks of the arguments a call's signature does not name.
671    pub fn push_abis(&mut self, abis: &[Abi]) -> AbiList {
672        let start = Idx::from_usize(self.abis.len());
673        self.abis.extend_from_slice(abis);
674        AbiList::new(start, Idx::from_usize(self.abis.len()))
675    }
676
677    /// Records a call's callee and signature.
678    pub fn add_call(&mut self, info: CallInfo) -> Idx<CallInfo> {
679        self.calls.push(info);
680        Idx::from_usize(self.calls.len() - 1)
681    }
682
683    /// Records a `switch`'s targets and case values.
684    pub fn add_switch(&mut self, info: SwitchInfo) -> Idx<SwitchInfo> {
685        self.switches.push(info);
686        Idx::from_usize(self.switches.len() - 1)
687    }
688
689    /// Records an inline assembly instruction's template and constraints.
690    pub fn add_asm(&mut self, info: AsmInfo) -> Idx<AsmInfo> {
691        self.asms.push(info);
692        Idx::from_usize(self.asms.len() - 1)
693    }
694
695    /// How many values, instructions and blocks there are, for a reader that wants to size
696    /// something by them.
697    #[must_use]
698    pub fn counts(&self) -> Counts {
699        Counts { values: self.values.len(), insts: self.insts.len(), blocks: self.blocks.len() }
700    }
701
702    /// What is known about a value, which is nothing at all unless somebody said otherwise.
703    ///
704    /// Section 6.2.3 of `spec/safe-memory/06-instrumentation.md`. Facts are in a side table and
705    /// not in the value, so a function nobody has said anything about carries no facts and is
706    /// the same size it was before facts existed.
707    #[must_use]
708    pub fn facts(&self, value: Value) -> Facts {
709        match self.facts.binary_search_by_key(&value.raw(), |&(at, _)| at.raw()) {
710            Ok(at) => self.facts[at].1,
711            Err(_) => Facts::NONE,
712        }
713    }
714
715    /// Says what is known about a value, replacing whatever was known before.
716    ///
717    /// Setting [`Facts::NONE`] takes the value back out of the table, which is what keeps the
718    /// table empty in a function that has had facts put on and then taken off again.
719    pub fn set_facts(&mut self, value: Value, facts: Facts) {
720        let found = self.facts.binary_search_by_key(&value.raw(), |&(at, _)| at.raw());
721        match (found, facts.is_empty()) {
722            (Ok(at), true) => drop(self.facts.remove(at)),
723            (Ok(at), false) => self.facts[at].1 = facts,
724            (Err(_), true) => {}
725            (Err(at), false) => self.facts.insert(at, (value, facts)),
726        }
727    }
728
729    /// Every value something is known about, in value order.
730    pub fn known(&self) -> impl Iterator<Item = (Value, Facts)> + '_ {
731        self.facts.iter().copied()
732    }
733
734    /// Gives a block a name of its own, which is how an image written before the program runs
735    /// says it holds the address of a place inside this function.
736    ///
737    /// What asks for this is GNU's address of a label in the initializer of an object with static
738    /// storage duration, which is how every threaded interpreter builds its dispatch table. The
739    /// address a `lea` produces needs none of this, because both ends of that distance are in the
740    /// same section and the object writer works it out for itself. An image is the other case: it
741    /// is in another section, so what it holds is a relocation, and a relocation names a symbol.
742    ///
743    /// One name per block. Two labels on the same statement are two labels and one block, and the
744    /// image asks for a name rather than for a particular one, so the second ask keeps the first
745    /// answer. Nothing outside this table ever sees the name, which is why it may be anything the
746    /// object format lets a local symbol be called.
747    pub fn name_block(&mut self, block: Block, name: Symbol) {
748        let found = self.labels.binary_search_by_key(&block.raw(), |&(at, _)| at.raw());
749        if let Err(at) = found {
750            self.labels.insert(at, (block, name));
751        }
752    }
753
754    /// The name a block was given, or `None` for a block nothing took the address of.
755    #[must_use]
756    pub fn block_name(&self, block: Block) -> Option<Symbol> {
757        match self.labels.binary_search_by_key(&block.raw(), |&(at, _)| at.raw()) {
758            Ok(at) => Some(self.labels[at].1),
759            Err(_) => None,
760        }
761    }
762
763    /// Every block that has a name, in block order.
764    pub fn named_blocks(&self) -> impl Iterator<Item = (Block, Symbol)> + '_ {
765        self.labels.iter().copied()
766    }
767
768    fn add_value(&mut self, data: ValueData) -> Value {
769        self.values.push(data);
770        Idx::from_usize(self.values.len() - 1)
771    }
772}
773
774/// How many of each thing a function holds.
775#[derive(Clone, Copy, Debug, PartialEq, Eq)]
776pub struct Counts {
777    /// Values, including the ones whose defining instruction has been removed.
778    pub values: usize,
779    /// Instructions, including the ones that have been removed from their block.
780    pub insts: usize,
781    /// Blocks.
782    pub blocks: usize,
783}
784
785// Reading is indexing. There is one of these for each handle, so `func[inst]` and `func[value]`
786// and `&func[args]` all work and none of them needs a method whose name says which table.
787impl Index<Value> for Func {
788    type Output = ValueData;
789
790    fn index(&self, value: Value) -> &ValueData {
791        &self.values[value.index()]
792    }
793}
794
795impl Index<Inst> for Func {
796    type Output = InstData;
797
798    fn index(&self, inst: Inst) -> &InstData {
799        &self.insts[inst.index()]
800    }
801}
802
803impl IndexMut<Inst> for Func {
804    fn index_mut(&mut self, inst: Inst) -> &mut InstData {
805        &mut self.insts[inst.index()]
806    }
807}
808
809impl Index<Block> for Func {
810    type Output = BlockData;
811
812    fn index(&self, block: Block) -> &BlockData {
813        &self.blocks[block.index()]
814    }
815}
816
817impl Index<Sig> for Func {
818    type Output = Signature;
819
820    fn index(&self, sig: Sig) -> &Signature {
821        &self.signatures[sig.index()]
822    }
823}
824
825impl Index<ValueList> for Func {
826    type Output = [Value];
827
828    fn index(&self, list: ValueList) -> &[Value] {
829        &self.value_pool[list.as_usize_range()]
830    }
831}
832
833impl Index<BlockCallList> for Func {
834    type Output = [BlockCall];
835
836    fn index(&self, list: BlockCallList) -> &[BlockCall] {
837        &self.block_calls[list.as_usize_range()]
838    }
839}
840
841impl Index<Idx<BlockCall>> for Func {
842    type Output = BlockCall;
843
844    fn index(&self, at: Idx<BlockCall>) -> &BlockCall {
845        &self.block_calls[at.index()]
846    }
847}
848
849impl Index<ImmList> for Func {
850    type Output = [Imm];
851
852    fn index(&self, list: ImmList) -> &[Imm] {
853        &self.imms[list.as_usize_range()]
854    }
855}
856
857impl Index<Idx<Imm>> for Func {
858    type Output = Imm;
859
860    fn index(&self, at: Idx<Imm>) -> &Imm {
861        &self.imms[at.index()]
862    }
863}
864
865impl Index<Idx<MemInfo>> for Func {
866    type Output = MemInfo;
867
868    fn index(&self, at: Idx<MemInfo>) -> &MemInfo {
869        &self.mem[at.index()]
870    }
871}
872
873impl Index<AbiList> for Func {
874    type Output = [Abi];
875
876    fn index(&self, list: AbiList) -> &[Abi] {
877        &self.abis[list.as_usize_range()]
878    }
879}
880
881impl Index<SlotList> for Func {
882    type Output = [Slot];
883
884    fn index(&self, list: SlotList) -> &[Slot] {
885        &self.slots[list.as_usize_range()]
886    }
887}
888
889impl Index<Idx<VaInfo>> for Func {
890    type Output = VaInfo;
891
892    fn index(&self, at: Idx<VaInfo>) -> &VaInfo {
893        &self.va_objects[at.index()]
894    }
895}
896
897impl Index<Idx<CallInfo>> for Func {
898    type Output = CallInfo;
899
900    fn index(&self, at: Idx<CallInfo>) -> &CallInfo {
901        &self.calls[at.index()]
902    }
903}
904
905impl Index<Idx<SwitchInfo>> for Func {
906    type Output = SwitchInfo;
907
908    fn index(&self, at: Idx<SwitchInfo>) -> &SwitchInfo {
909        &self.switches[at.index()]
910    }
911}
912
913impl Index<Idx<AsmInfo>> for Func {
914    type Output = AsmInfo;
915
916    fn index(&self, at: Idx<AsmInfo>) -> &AsmInfo {
917        &self.asms[at.index()]
918    }
919}
920
921/// A cursor that appends to the end of one block.
922///
923/// This is the shape lowering wants: it works on one block at a time, it appends, and it wants
924/// the value back so it can use it in the next instruction. Everything here is a thin wrapper
925/// over [`Func::create_inst`] and [`Func::append_inst`], and anything the wrappers do not
926/// cover is done with those two directly.
927#[derive(Debug)]
928pub struct Builder<'a> {
929    func: &'a mut Func,
930    block: Block,
931    span: Span,
932}
933
934impl<'a> Builder<'a> {
935    /// A cursor appending to that block, with every instruction taking that source location.
936    pub fn new(func: &'a mut Func, block: Block) -> Self {
937        Self { func, block, span: Span::DUMMY }
938    }
939
940    /// The same cursor, with a source location for the instructions after this.
941    #[must_use]
942    pub fn at(mut self, span: Span) -> Self {
943        self.span = span;
944        self
945    }
946
947    /// Sets the source location for the instructions after this.
948    pub fn set_span(&mut self, span: Span) {
949        self.span = span;
950    }
951
952    /// The function being built.
953    pub fn func(&mut self) -> &mut Func {
954        self.func
955    }
956
957    /// The block being appended to.
958    #[must_use]
959    pub fn block(&self) -> Block {
960        self.block
961    }
962
963    /// Appends an instruction as it is, and gives back its results.
964    pub fn inst(&mut self, data: InstData, results: &[Type]) -> Inst {
965        let inst = self.func.create_inst(data, results, self.span);
966        self.func.append_inst(self.block, inst);
967        inst
968    }
969
970    /// The one value an instruction produces.
971    ///
972    /// # Panics
973    ///
974    /// Panics if it did not produce exactly one.
975    pub fn value(&mut self, data: InstData, ty: Type) -> Value {
976        let inst = self.inst(data, &[ty]);
977        self.func[inst].first_result.expect("one result was asked for")
978    }
979
980    /// An integer constant.
981    ///
982    /// # Panics
983    ///
984    /// Panics if `ty` is not an integer type.
985    pub fn iconst(&mut self, ty: Type, value: i128) -> Value {
986        let imm = self.func.add_imm(Imm::int(value, ty.lane()));
987        self.value(InstData { extra: Extra::Imm(imm), ..InstData::new(Opcode::IConst) }, ty)
988    }
989
990    /// A floating point constant, given as the bits of its format.
991    pub fn fconst(&mut self, ty: Type, bits: u128) -> Value {
992        let imm = self.func.add_imm(Imm::from_bits(bits));
993        self.value(InstData { extra: Extra::Imm(imm), ..InstData::new(Opcode::FConst) }, ty)
994    }
995
996    /// A two-operand instruction whose result has the type of its operands.
997    pub fn binary(&mut self, opcode: Opcode, lhs: Value, rhs: Value, flags: Flags) -> Value {
998        let ty = self.func[lhs].ty;
999        let args = self.func.push_values(&[lhs, rhs]);
1000        self.value(InstData { args, flags, ..InstData::new(opcode) }, ty)
1001    }
1002
1003    /// Arithmetic that answers with both the wrapped result and whether it wrapped.
1004    ///
1005    /// The one shape in the IR whose result is two things, which is why it has a builder of its
1006    /// own rather than going through [`Builder::value`]. The first result is the answer in the
1007    /// type of the operands, the same as the ordinary form of the same arithmetic would give, and
1008    /// the second is one `i1` per lane saying whether the exact answer needed more bits than that
1009    /// type has.
1010    ///
1011    /// # Panics
1012    ///
1013    /// Panics if the instruction did not produce exactly the two results it was created with,
1014    /// which is the same promise [`Builder::value`] makes about its one.
1015    pub fn checked(&mut self, opcode: Opcode, lhs: Value, rhs: Value) -> (Value, Value) {
1016        let ty = self.func[lhs].ty;
1017        let args = self.func.push_values(&[lhs, rhs]);
1018        let results = [ty, ty.with_lane(Type::I1)];
1019        let inst = self.inst(InstData { args, ..InstData::new(opcode) }, &results);
1020        let mut answers = self.func[inst].results();
1021        let value = answers.next().expect("two results were asked for");
1022        let wrapped = answers.next().expect("two results were asked for");
1023        (value, wrapped)
1024    }
1025
1026    /// A one-operand instruction whose result has the type given.
1027    pub fn unary(&mut self, opcode: Opcode, arg: Value, ty: Type) -> Value {
1028        let args = self.func.push_values(&[arg]);
1029        self.value(InstData { args, ..InstData::new(opcode) }, ty)
1030    }
1031
1032    /// An integer comparison, which produces one `i1` per lane.
1033    pub fn icmp(&mut self, pred: IntPred, lhs: Value, rhs: Value) -> Value {
1034        let ty = self.func[lhs].ty.with_lane(Type::I1);
1035        let args = self.func.push_values(&[lhs, rhs]);
1036        self.value(
1037            InstData { args, extra: Extra::IntPred(pred), ..InstData::new(Opcode::ICmp) },
1038            ty,
1039        )
1040    }
1041
1042    /// One of two values, chosen by a bit, which is what a diamond becomes when it stops being one.
1043    ///
1044    /// The type comes from the arms rather than from the bit, and the two arms have to agree, which
1045    /// the verifier checks. Both are evaluated, so the caller owes the argument that evaluating the
1046    /// one that is not chosen is harmless.
1047    pub fn select(&mut self, cond: Value, then: Value, other: Value) -> Value {
1048        let ty = self.func[then].ty;
1049        let args = self.func.push_values(&[cond, then, other]);
1050        self.value(InstData { args, ..InstData::new(Opcode::Select) }, ty)
1051    }
1052
1053    /// A floating point comparison, which produces one `i1` per lane.
1054    pub fn fcmp(&mut self, pred: FloatPred, lhs: Value, rhs: Value, flags: Flags) -> Value {
1055        let ty = self.func[lhs].ty.with_lane(Type::I1);
1056        let args = self.func.push_values(&[lhs, rhs]);
1057        self.value(
1058            InstData { args, flags, extra: Extra::FloatPred(pred), ..InstData::new(Opcode::FCmp) },
1059            ty,
1060        )
1061    }
1062
1063    /// Memory as the function found it, which is where a memory SSA chain starts.
1064    ///
1065    /// It belongs at the top of the entry block and there is one of them in a function.
1066    pub fn mem_entry(&mut self) -> Value {
1067        self.value(InstData::new(Opcode::MemEntry), Type::MEM)
1068    }
1069
1070    /// A read of that type from that address.
1071    pub fn load(&mut self, ty: Type, addr: Value, info: MemInfo, flags: Flags) -> Value {
1072        let mem = self.func.add_mem(info);
1073        let args = self.func.push_values(&[addr]);
1074        self.value(
1075            InstData { args, flags, extra: Extra::Mem(mem), ..InstData::new(Opcode::Load) },
1076            ty,
1077        )
1078    }
1079
1080    /// A write of a value to an address.
1081    pub fn store(&mut self, value: Value, addr: Value, info: MemInfo, flags: Flags) -> Inst {
1082        let mem = self.func.add_mem(info);
1083        let args = self.func.push_values(&[value, addr]);
1084        self.inst(
1085            InstData { args, flags, extra: Extra::Mem(mem), ..InstData::new(Opcode::Store) },
1086            &[],
1087        )
1088    }
1089
1090    /// The same read, ordered.
1091    ///
1092    /// A separate opcode rather than an ordering on [`Builder::load`], because the two are not the
1093    /// same thing to anything that moves code: a plain load may be moved, duplicated and dropped,
1094    /// and this one may not. The IR verifier is what keeps the pair honest, since it refuses an
1095    /// ordering on a plain access and refuses an unordered one here, so no pass has to remember to
1096    /// check the payload before deciding a load is free.
1097    pub fn atomic_load(&mut self, ty: Type, addr: Value, info: MemInfo, flags: Flags) -> Value {
1098        let mem = self.func.add_mem(info);
1099        let args = self.func.push_values(&[addr]);
1100        self.value(
1101            InstData { args, flags, extra: Extra::Mem(mem), ..InstData::new(Opcode::AtomicLoad) },
1102            ty,
1103        )
1104    }
1105
1106    /// The same write, ordered.
1107    pub fn atomic_store(&mut self, value: Value, addr: Value, info: MemInfo, flags: Flags) -> Inst {
1108        let mem = self.func.add_mem(info);
1109        let args = self.func.push_values(&[value, addr]);
1110        self.inst(
1111            InstData { args, flags, extra: Extra::Mem(mem), ..InstData::new(Opcode::AtomicStore) },
1112            &[],
1113        )
1114    }
1115
1116    /// A compare and exchange, which answers what it found and whether that was what was expected.
1117    ///
1118    /// Two values out of one instruction, in that order, because a caller that had to ask twice
1119    /// would be asking about two different moments. The type of the first is the type of the value
1120    /// expected, which is what says how wide the access is, and the type of the second is
1121    /// [`Type::I1`] whatever the width was.
1122    pub fn cmpxchg(
1123        &mut self,
1124        addr: Value,
1125        expected: Value,
1126        desired: Value,
1127        info: MemInfo,
1128        flags: Flags,
1129    ) -> (Value, Value) {
1130        let ty = self.func[expected].ty;
1131        let mem = self.func.add_mem(info);
1132        let args = self.func.push_values(&[addr, expected, desired]);
1133        let inst = self.inst(
1134            InstData { args, flags, extra: Extra::Mem(mem), ..InstData::new(Opcode::Cmpxchg) },
1135            &[ty, Type::I1],
1136        );
1137        let results: Vec<Value> = self.func[inst].results().collect();
1138        let [old, exchanged] = results[..] else { unreachable!("two results were asked for") };
1139        (old, exchanged)
1140    }
1141
1142    /// A read, an operation on what was read, and a write back, with nothing able to get between
1143    /// them.
1144    ///
1145    /// The value it answers is the one that was there before, which is the convention every machine
1146    /// and every language in this area uses, and a caller that wanted the value afterwards works it
1147    /// out from the two it already has rather than asking for a second flavour of the instruction.
1148    /// The type of that value is the type of the operand, which is what says how wide the access is.
1149    pub fn atomic_rmw(
1150        &mut self,
1151        op: RmwOp,
1152        addr: Value,
1153        operand: Value,
1154        info: MemInfo,
1155        flags: Flags,
1156    ) -> Value {
1157        let ty = self.func[operand].ty;
1158        let mem = self.func.add_mem(info);
1159        let args = self.func.push_values(&[addr, operand]);
1160        self.value(
1161            InstData {
1162                args,
1163                flags,
1164                extra: Extra::Rmw(op, mem),
1165                ..InstData::new(Opcode::AtomicRmw)
1166            },
1167            ty,
1168        )
1169    }
1170
1171    /// A barrier, which touches no address and is its ordering and nothing else.
1172    pub fn fence(&mut self, order: MemOrder) -> Inst {
1173        self.inst(InstData { extra: Extra::Order(order), ..InstData::new(Opcode::Fence) }, &[])
1174    }
1175
1176    /// A hint that an address is about to be read or written, which produces nothing.
1177    ///
1178    /// It reads the address rather than the memory at it, in the sense that nothing after this
1179    /// sees anything it did not see before. What it is allowed to do is take time, so it is on the
1180    /// memory chain anyway: a prefetch of an address a store is about to write to has to stay on
1181    /// the side of that store it was written on, or it is a hint about the wrong thing.
1182    pub fn prefetch(&mut self, address: Value, hint: PrefetchHint) -> Inst {
1183        let args = self.func.push_values(&[address]);
1184        self.inst(
1185            InstData { args, extra: Extra::Prefetch(hint), ..InstData::new(Opcode::Prefetch) },
1186            &[],
1187        )
1188    }
1189
1190    /// An unconditional branch.
1191    pub fn jump(&mut self, target: Block, args: &[Value]) -> Inst {
1192        let call = self.block_call(target, args);
1193        let targets = self.func.push_block_calls(&[call]);
1194        self.inst(InstData { extra: Extra::Targets(targets), ..InstData::new(Opcode::Jump) }, &[])
1195    }
1196
1197    /// The address of a block, which is a value a later `indirect_br` can branch to.
1198    ///
1199    /// The block is a target here in the same sense a branch's is, so everything that asks an
1200    /// instruction which blocks it names finds this one, and a block whose address is taken is
1201    /// not mistaken for a block nothing mentions.
1202    pub fn block_addr(&mut self, target: Block) -> Value {
1203        let call = self.block_call(target, &[]);
1204        let targets = self.func.push_block_calls(&[call]);
1205        self.value(
1206            InstData { extra: Extra::Targets(targets), ..InstData::new(Opcode::BlockAddr) },
1207            Type::PTR,
1208        )
1209    }
1210
1211    /// A branch to an address, which arrives at one of the blocks listed.
1212    ///
1213    /// Every block the address can hold has to be there. The list is what the rest of the
1214    /// compiler reads, so a block left out of it is a block the branch is saying it never
1215    /// reaches, and none of it is checked against the addresses anybody took.
1216    pub fn indirect_br(&mut self, addr: Value, targets: &[Block]) -> Inst {
1217        let calls: Vec<BlockCall> =
1218            targets.iter().map(|&target| self.block_call(target, &[])).collect();
1219        let targets = self.func.push_block_calls(&calls);
1220        let args = self.func.push_values(&[addr]);
1221        self.inst(
1222            InstData { args, extra: Extra::Targets(targets), ..InstData::new(Opcode::IndirectBr) },
1223            &[],
1224        )
1225    }
1226
1227    /// A two-way branch, taking the first target when the condition is one.
1228    pub fn br_if(
1229        &mut self,
1230        cond: Value,
1231        then_block: Block,
1232        then_args: &[Value],
1233        else_block: Block,
1234        else_args: &[Value],
1235    ) -> Inst {
1236        let then_call = self.block_call(then_block, then_args);
1237        let else_call = self.block_call(else_block, else_args);
1238        let targets = self.func.push_block_calls(&[then_call, else_call]);
1239        let args = self.func.push_values(&[cond]);
1240        self.inst(
1241            InstData { args, extra: Extra::Targets(targets), ..InstData::new(Opcode::BrIf) },
1242            &[],
1243        )
1244    }
1245
1246    /// A branch on an integer, taking the target its value selects and the default when it
1247    /// selects none.
1248    ///
1249    /// The cases are values and blocks rather than a table with the default in it, because the
1250    /// order the side table wants, which is the default first, is not an order anybody building
1251    /// a `switch` has their cases in.
1252    pub fn switch(&mut self, value: Value, default: Block, cases: &[(i128, Block)]) -> Inst {
1253        let ty = self.func[value].ty.lane();
1254        let mut calls = vec![self.block_call(default, &[])];
1255        let mut values = Vec::with_capacity(cases.len());
1256        for &(value, block) in cases {
1257            calls.push(self.block_call(block, &[]));
1258            values.push(Imm::int(value, ty));
1259        }
1260        let targets = self.func.push_block_calls(&calls);
1261        let cases = self.func.push_imms(&values);
1262        let info = self.func.add_switch(SwitchInfo { targets, cases });
1263        let args = self.func.push_values(&[value]);
1264        self.inst(
1265            InstData { args, extra: Extra::Switch(info), ..InstData::new(Opcode::Switch) },
1266            &[],
1267        )
1268    }
1269
1270    /// A return of the values the signature says.
1271    pub fn ret(&mut self, values: &[Value]) -> Inst {
1272        let args = self.func.push_values(values);
1273        self.inst(InstData { args, ..InstData::new(Opcode::Return) }, &[])
1274    }
1275
1276    /// A place control does not reach.
1277    pub fn unreachable(&mut self) -> Inst {
1278        self.inst(InstData::new(Opcode::Unreachable), &[])
1279    }
1280
1281    /// A direct call, with the results its signature says it produces.
1282    pub fn call(&mut self, callee: Symbol, signature: Sig, args: &[Value]) -> Inst {
1283        self.call_varargs(callee, signature, args, &[])
1284    }
1285
1286    /// The same, saying how the arguments the signature does not name travel.
1287    ///
1288    /// Empty says they all travel as the values in hand, which is what [`Builder::call`] passes
1289    /// and is the usual case. Anything else has one entry for each argument past the ones the
1290    /// signature names.
1291    pub fn call_varargs(
1292        &mut self,
1293        callee: Symbol,
1294        signature: Sig,
1295        args: &[Value],
1296        varargs: &[Abi],
1297    ) -> Inst {
1298        let varargs = self.func.push_abis(varargs);
1299        let info = self.func.add_call(CallInfo { callee: Some(callee), signature, varargs });
1300        let returns: Vec<Type> = self.func[signature].return_types().collect();
1301        let args = self.func.push_values(args);
1302        self.inst(
1303            InstData { args, extra: Extra::Call(info), ..InstData::new(Opcode::Call) },
1304            &returns,
1305        )
1306    }
1307
1308    /// Inline assembly, which is a terminator when the info carries targets.
1309    ///
1310    /// The targets are built by the caller, because the frontend is the only thing that knows
1311    /// which block is the one control reaches when the assembly does not jump, and that block
1312    /// has to come first.
1313    pub fn inline_asm(
1314        &mut self,
1315        info: AsmInfo,
1316        args: &[Value],
1317        results: &[Type],
1318        flags: Flags,
1319    ) -> Inst {
1320        let info = self.func.add_asm(info);
1321        let args = self.func.push_values(args);
1322        self.inst(
1323            InstData { args, flags, extra: Extra::Asm(info), ..InstData::new(Opcode::InlineAsm) },
1324            results,
1325        )
1326    }
1327
1328    fn block_call(&mut self, block: Block, args: &[Value]) -> BlockCall {
1329        BlockCall::new(block, self.func.push_values(args))
1330    }
1331}
1332
1333#[cfg(test)]
1334mod tests {
1335    use rucc_base::Interner;
1336
1337    use super::*;
1338    use crate::inst::BlockCallList;
1339    use crate::{MemOrder, Restrict};
1340
1341    /// The example from the spec, near enough: a loop that sums one to n and stores it.
1342    fn sum() -> (Func, Block, Block, Block) {
1343        let mut names = Interner::new();
1344        let i32_ = Type::int(32);
1345        let mut func = Func::new(
1346            names.intern("sum"),
1347            Signature::new().with_params(&[i32_]).with_returns(&[i32_]),
1348        );
1349
1350        let entry = func.create_block();
1351        let n = func.append_param(entry, i32_);
1352        let header = func.create_block();
1353        let acc = func.append_param(header, i32_);
1354        let i = func.append_param(header, i32_);
1355        let exit = func.create_block();
1356        let result = func.append_param(exit, i32_);
1357
1358        let mut b = Builder::new(&mut func, entry);
1359        let zero = b.iconst(i32_, 0);
1360        let cmp = b.icmp(IntPred::Sle, n, zero);
1361        b.br_if(cmp, exit, &[zero], header, &[zero, zero]);
1362
1363        let mut b = Builder::new(&mut func, header);
1364        let one = b.iconst(i32_, 1);
1365        let next = b.binary(Opcode::Add, i, one, Flags::NSW);
1366        let total = b.binary(Opcode::Add, acc, next, Flags::NSW);
1367        let done = b.icmp(IntPred::Sge, next, n);
1368        b.br_if(done, exit, &[total], header, &[total, next]);
1369
1370        let mut b = Builder::new(&mut func, exit);
1371        b.ret(&[result]);
1372
1373        (func, entry, header, exit)
1374    }
1375
1376    #[test]
1377    fn the_blocks_come_back_in_the_order_they_were_made() {
1378        let (func, entry, header, exit) = sum();
1379        assert_eq!(func.blocks().collect::<Vec<_>>(), [entry, header, exit]);
1380        assert_eq!(func.entry(), Some(entry));
1381    }
1382
1383    #[test]
1384    fn a_removed_block_is_gone_from_the_layout_and_so_is_what_was_in_it() {
1385        let (mut func, entry, header, exit) = sum();
1386        let inside: Vec<Inst> = func.insts(header).collect();
1387        func.remove_block(header);
1388        assert_eq!(func.blocks().collect::<Vec<_>>(), [entry, exit]);
1389        assert_eq!(func.entry(), Some(entry));
1390        assert_eq!(func[entry].next, Some(exit));
1391        assert_eq!(func[exit].prev, Some(entry));
1392        // The instructions say they are in no block, the way a removed one does.
1393        assert!(inside.iter().all(|&inst| func.block_of(inst).is_none()));
1394        assert!(func.insts(header).next().is_none());
1395    }
1396
1397    #[test]
1398    fn each_block_holds_what_was_appended_to_it() {
1399        let (func, entry, header, exit) = sum();
1400        let opcodes =
1401            |block| func.insts(block).map(|inst| func[inst].opcode.name()).collect::<Vec<_>>();
1402        assert_eq!(opcodes(entry), ["iconst", "icmp", "br_if"]);
1403        assert_eq!(opcodes(header), ["iconst", "add", "add", "icmp", "br_if"]);
1404        assert_eq!(opcodes(exit), ["return"]);
1405    }
1406
1407    #[test]
1408    fn asm_ends_a_block_when_it_has_labels_and_not_otherwise() {
1409        // The labels are in the function's table, so the instruction on its own cannot answer
1410        // and anything asking it rather than the function would walk off the end of the block.
1411        let mut func = Func::new(Symbol::from_raw(0), Signature::new());
1412        let block = func.create_block();
1413        let plain = func.add_asm(AsmInfo {
1414            template: Symbol::from_raw(0),
1415            constraints: Symbol::from_raw(0),
1416            clobbers: Symbol::from_raw(0),
1417            targets: BlockCallList::EMPTY,
1418        });
1419        let call = BlockCall::to(block);
1420        let targets = func.push_block_calls(&[call]);
1421        let labelled = func.add_asm(AsmInfo {
1422            template: Symbol::from_raw(0),
1423            constraints: Symbol::from_raw(0),
1424            clobbers: Symbol::from_raw(0),
1425            targets,
1426        });
1427
1428        let mut make = |extra| {
1429            let data = InstData { extra, ..InstData::new(Opcode::InlineAsm) };
1430            func.create_inst(data, &[], Span::DUMMY)
1431        };
1432        let plain = make(Extra::Asm(plain));
1433        let labelled = make(Extra::Asm(labelled));
1434        assert!(!func.is_terminator(plain));
1435        assert!(func.is_terminator(labelled));
1436    }
1437
1438    #[test]
1439    fn every_block_ends_in_its_terminator() {
1440        let (func, entry, header, exit) = sum();
1441        for block in [entry, header, exit] {
1442            let last = func.terminator(block).expect("a terminator");
1443            assert_eq!(Some(last), func.insts(block).last());
1444        }
1445    }
1446
1447    #[test]
1448    fn a_branch_carries_the_arguments_the_block_takes() {
1449        let (func, entry, header, _) = sum();
1450        let br = func.terminator(entry).expect("a terminator");
1451        let calls: Vec<BlockCall> = func.successors(br).collect();
1452        assert_eq!(calls.len(), 2);
1453        // The loop header takes two parameters, so the branch to it passes two.
1454        assert_eq!(calls[1].block, header);
1455        assert_eq!(func[calls[1].args].len(), 2);
1456        assert_eq!(func[header].params.len(), 2);
1457        assert_eq!(func[calls[0].args].len(), 1);
1458    }
1459
1460    #[test]
1461    fn a_value_knows_what_defined_it() {
1462        let (func, entry, _, _) = sum();
1463        let first = func.insts(entry).next().expect("an instruction");
1464        let value = func[first].first_result.expect("a result");
1465        assert_eq!(func[value].def, Def::Result { inst: first, index: 0 });
1466        assert_eq!(func[value].ty, Type::int(32));
1467
1468        let param = func[entry].params[0];
1469        assert_eq!(func[param].def, Def::Param { block: entry, index: 0 });
1470    }
1471
1472    #[test]
1473    fn a_comparison_produces_one_bit() {
1474        let (func, entry, _, _) = sum();
1475        let cmp = func.insts(entry).nth(1).expect("the comparison");
1476        let value = func[cmp].first_result.expect("a result");
1477        assert_eq!(func[value].ty, Type::I1);
1478        assert_eq!(func[cmp].extra, Extra::IntPred(IntPred::Sle));
1479    }
1480
1481    #[test]
1482    fn flags_ride_along_on_the_instruction_that_was_given_them() {
1483        let (func, _, header, _) = sum();
1484        let add = func.insts(header).nth(1).expect("the addition");
1485        assert_eq!(func[add].flags, Flags::NSW);
1486        let cmp = func.insts(header).nth(3).expect("the comparison");
1487        assert_eq!(func[cmp].flags, Flags::NONE);
1488    }
1489
1490    #[test]
1491    fn removing_an_instruction_takes_it_out_of_the_middle() {
1492        let (mut func, _, header, _) = sum();
1493        let add = func.insts(header).nth(1).expect("the addition");
1494        func.remove_inst(add);
1495        let opcodes: Vec<&str> = func.insts(header).map(|inst| func[inst].opcode.name()).collect();
1496        assert_eq!(opcodes, ["iconst", "add", "icmp", "br_if"]);
1497        assert_eq!(func.block_of(add), None);
1498    }
1499
1500    #[test]
1501    fn removing_the_first_and_the_last_keeps_the_ends_right() {
1502        let (mut func, entry, _, _) = sum();
1503        let first = func.insts(entry).next().expect("an instruction");
1504        let last = func.terminator(entry).expect("a terminator");
1505        func.remove_inst(first);
1506        func.remove_inst(last);
1507        let opcodes: Vec<&str> = func.insts(entry).map(|inst| func[inst].opcode.name()).collect();
1508        assert_eq!(opcodes, ["icmp"]);
1509        assert_eq!(func[entry].first, func[entry].last);
1510    }
1511
1512    #[test]
1513    fn removing_the_only_instruction_empties_the_block() {
1514        let (mut func, _, _, exit) = sum();
1515        let only = func.insts(exit).next().expect("an instruction");
1516        func.remove_inst(only);
1517        assert_eq!(func.insts(exit).count(), 0);
1518        assert_eq!(func[exit].first, None);
1519        assert_eq!(func[exit].last, None);
1520    }
1521
1522    #[test]
1523    fn inserting_before_puts_it_in_the_right_place() {
1524        let (mut func, entry, _, _) = sum();
1525        let cmp = func.insts(entry).nth(1).expect("the comparison");
1526        let made = func.create_inst(InstData::new(Opcode::Unreachable), &[], Span::DUMMY);
1527        func.insert_before(made, cmp);
1528        let opcodes: Vec<&str> = func.insts(entry).map(|inst| func[inst].opcode.name()).collect();
1529        assert_eq!(opcodes, ["iconst", "unreachable", "icmp", "br_if"]);
1530    }
1531
1532    #[test]
1533    fn inserting_before_the_first_makes_it_the_first() {
1534        let (mut func, entry, _, _) = sum();
1535        let first = func.insts(entry).next().expect("an instruction");
1536        let made = func.create_inst(InstData::new(Opcode::Unreachable), &[], Span::DUMMY);
1537        func.insert_before(made, first);
1538        assert_eq!(func.insts(entry).next(), Some(made));
1539        assert_eq!(func[entry].first, Some(made));
1540    }
1541
1542    #[test]
1543    fn inserting_after_puts_it_in_the_right_place() {
1544        let (mut func, entry, _, _) = sum();
1545        let first = func.insts(entry).next().expect("an instruction");
1546        let made = func.create_inst(InstData::new(Opcode::Unreachable), &[], Span::DUMMY);
1547        func.insert_after(made, first);
1548        let opcodes: Vec<&str> = func.insts(entry).map(|inst| func[inst].opcode.name()).collect();
1549        assert_eq!(opcodes, ["iconst", "unreachable", "icmp", "br_if"]);
1550        assert_eq!(func[entry].first, Some(first));
1551    }
1552
1553    #[test]
1554    #[should_panic(expected = "nothing goes after a terminator")]
1555    fn inserting_after_the_terminator_is_refused() {
1556        // A block ends where its branch is, so an instruction after one would be in no block that
1557        // control ever reaches, and the layout would be claiming otherwise.
1558        let (mut func, entry, _, _) = sum();
1559        let last = func.insts(entry).last().expect("a terminator");
1560        let made = func.create_inst(InstData::new(Opcode::Unreachable), &[], Span::DUMMY);
1561        func.insert_after(made, last);
1562    }
1563
1564    #[test]
1565    fn a_list_grows_in_place_while_it_is_the_last_thing_in_the_pool() {
1566        let mut func = Func::new(Symbol::from_raw(0), Signature::new());
1567        let block = func.create_block();
1568        let a = func.append_param(block, Type::int(32));
1569        let b = func.append_param(block, Type::int(32));
1570        let list = func.push_values(&[a]);
1571        let grown = func.append_arg(list, b);
1572        assert_eq!(func[grown], [a, b]);
1573        assert_eq!(grown.as_usize_range().start, list.as_usize_range().start);
1574    }
1575
1576    #[test]
1577    fn a_list_is_copied_when_something_is_behind_it() {
1578        let mut func = Func::new(Symbol::from_raw(0), Signature::new());
1579        let block = func.create_block();
1580        let a = func.append_param(block, Type::int(32));
1581        let b = func.append_param(block, Type::int(32));
1582        let list = func.push_values(&[a, a]);
1583        let behind = func.push_values(&[b]);
1584        let grown = func.append_arg(list, b);
1585        assert_eq!(func[grown], [a, a, b]);
1586        assert_eq!(func[list], [a, a], "the old run is still readable");
1587        assert_eq!(func[behind], [b], "and so is what was behind it");
1588        assert_ne!(grown.as_usize_range().start, list.as_usize_range().start);
1589    }
1590
1591    #[test]
1592    fn a_parameter_added_late_is_the_next_one_along() {
1593        // This is the shape SSA construction leaves: the loop header gains a parameter after
1594        // the blocks that branch to it already exist, and each of their branches grows an
1595        // argument to match.
1596        let (mut func, entry, header, _) = sum();
1597        let extra = func.append_param(header, Type::int(32));
1598        assert_eq!(func[header].params.len(), 3);
1599        assert_eq!(func[extra].def, Def::Param { block: header, index: 2 });
1600
1601        let br = func.terminator(entry).expect("a terminator");
1602        let call = func.successors(br).nth(1).expect("the branch to the header");
1603        let grown = func.append_arg(call.args, extra);
1604        assert_eq!(func[grown].len(), 3);
1605    }
1606
1607    #[test]
1608    fn a_span_rides_along_with_the_instruction() {
1609        let mut func = Func::new(Symbol::from_raw(0), Signature::new());
1610        let block = func.create_block();
1611        let span = Span::new(10, 20);
1612        let mut b = Builder::new(&mut func, block).at(span);
1613        let value = b.iconst(Type::int(32), 7);
1614        let inst = match func[value].def {
1615            Def::Result { inst, .. } => inst,
1616            Def::Param { .. } => unreachable!("a constant is not a parameter"),
1617        };
1618        assert_eq!(func.span(inst), span);
1619    }
1620
1621    #[test]
1622    fn a_store_produces_nothing_and_a_load_produces_one_value() {
1623        let mut func = Func::new(Symbol::from_raw(0), Signature::new());
1624        let block = func.create_block();
1625        let addr = func.append_param(block, Type::PTR);
1626        let info = MemInfo {
1627            size: 4,
1628            align: 4,
1629            order: MemOrder::NotAtomic,
1630            tbaa: None,
1631            owns: 0,
1632            restrict: Restrict::NONE,
1633        };
1634        let mut b = Builder::new(&mut func, block);
1635        let value = b.load(Type::int(32), addr, info, Flags::NONE);
1636        let store = b.store(value, addr, info, Flags::VOLATILE);
1637        assert_eq!(func[store].results, 0);
1638        assert_eq!(func[store].flags, Flags::VOLATILE);
1639        assert_eq!(func[value].ty, Type::int(32));
1640    }
1641
1642    #[test]
1643    fn a_call_produces_what_its_signature_returns() {
1644        let mut names = Interner::new();
1645        let mut func = Func::new(names.intern("caller"), Signature::new());
1646        let sig = func.add_signature(
1647            Signature::new().with_params(&[Type::int(32)]).with_returns(&[Type::int(64)]),
1648        );
1649        let block = func.create_block();
1650        let arg = func.append_param(block, Type::int(32));
1651        let callee = names.intern("callee");
1652        let mut b = Builder::new(&mut func, block);
1653        let call = b.call(callee, sig, &[arg]);
1654        assert_eq!(func[call].results, 1);
1655        let value = func[call].first_result.expect("a result");
1656        assert_eq!(func[value].ty, Type::int(64));
1657        assert_eq!(func[call].extra, Extra::Call(Idx::new(0)));
1658    }
1659
1660    #[test]
1661    fn the_counts_are_what_was_made() {
1662        let (func, _, _, _) = sum();
1663        let counts = func.counts();
1664        assert_eq!(counts.blocks, 3);
1665        assert_eq!(counts.insts, 9);
1666        // Four block parameters and five instruction results, which is the two constants, the
1667        // two additions and the two comparisons less the branches, which produce nothing.
1668        assert_eq!(counts.values, 4 + 6);
1669    }
1670
1671    #[test]
1672    #[should_panic(expected = "the instruction is in a block")]
1673    fn appending_an_instruction_twice_is_refused() {
1674        let (mut func, entry, _, _) = sum();
1675        let first = func.insts(entry).next().expect("an instruction");
1676        func.append_inst(entry, first);
1677    }
1678
1679    #[test]
1680    #[should_panic(expected = "the instruction is not in a block")]
1681    fn removing_an_instruction_twice_is_refused() {
1682        let (mut func, entry, _, _) = sum();
1683        let first = func.insts(entry).next().expect("an instruction");
1684        func.remove_inst(first);
1685        func.remove_inst(first);
1686    }
1687
1688    /// A store and a load with memory threaded through them, as memory SSA construction does it.
1689    fn threaded() -> (Func, Inst, Inst) {
1690        let mut names = Interner::new();
1691        let i32_ = Type::int(32);
1692        let mut func = Func::new(
1693            names.intern("thread"),
1694            Signature::new().with_params(&[Type::PTR]).with_returns(&[i32_]),
1695        );
1696        let entry = func.create_block();
1697        let addr = func.append_param(entry, Type::PTR);
1698        let info = MemInfo {
1699            size: 4,
1700            align: 4,
1701            order: MemOrder::NotAtomic,
1702            tbaa: None,
1703            owns: 0,
1704            restrict: Restrict::NONE,
1705        };
1706
1707        let mut b = Builder::new(&mut func, entry);
1708        let start = b.mem_entry();
1709        let seven = b.iconst(i32_, 7);
1710        let store = b.store(seven, addr, info, Flags::NONE);
1711        let value = b.load(i32_, addr, info, Flags::NONE);
1712        let Def::Result { inst: load, .. } = func[value].def else {
1713            panic!("the load produced it");
1714        };
1715
1716        let store = func.with_mem(store, start);
1717        let after = func.mem_out(store).expect("a store makes a new version");
1718        let load = func.with_mem(load, after);
1719        (func, store, load)
1720    }
1721
1722    #[test]
1723    fn threading_memory_puts_it_last_and_leaves_everything_else_where_it_was() {
1724        let (func, store, load) = threaded();
1725        assert_eq!(func.mem_in(store), func.mem_out(store).map(|_| func[func[store].args][2]));
1726        assert_eq!(func[func[store].args].len(), 3);
1727        assert!(func.carries_mem(store));
1728        assert!(func.carries_mem(load));
1729
1730        // The address of the load is still its first operand, which is the point of putting
1731        // memory last: nothing that read the operands before has to learn about it.
1732        assert_eq!(func[func[load].args][0], func[func.entry().expect("an entry")].params[0]);
1733        assert_eq!(func.mem_in(load), func.mem_out(store));
1734        assert_eq!(func.mem_out(load), None);
1735    }
1736
1737    #[test]
1738    #[should_panic(expected = "this is already on the memory chain")]
1739    fn threading_memory_through_the_same_instruction_twice_is_refused() {
1740        let (mut func, store, _) = threaded();
1741        let start = func.mem_in(store).expect("it was threaded");
1742        func.with_mem(store, start);
1743    }
1744
1745    /// Two copies of the same pair of addresses, the first of a fixed length and the second of one
1746    /// the caller passed in, both on the memory chain so that the length is not the last operand.
1747    fn copies() -> (Func, Inst, Inst) {
1748        let mut names = Interner::new();
1749        let i64_ = Type::int(64);
1750        let mut func = Func::new(
1751            names.intern("copies"),
1752            Signature::new().with_params(&[Type::PTR, Type::PTR, i64_]),
1753        );
1754        let entry = func.create_block();
1755        let to = func.append_param(entry, Type::PTR);
1756        let from = func.append_param(entry, Type::PTR);
1757        let length = func.append_param(entry, i64_);
1758        let info = MemInfo {
1759            size: 16,
1760            align: 4,
1761            order: MemOrder::NotAtomic,
1762            tbaa: None,
1763            owns: 0,
1764            restrict: Restrict::NONE,
1765        };
1766
1767        let mut b = Builder::new(&mut func, entry);
1768        let start = b.mem_entry();
1769        let mem = b.func().add_mem(info);
1770        let args = b.func().push_values(&[to, from]);
1771        let fixed =
1772            b.inst(InstData { args, extra: Extra::Mem(mem), ..InstData::new(Opcode::Memcpy) }, &[]);
1773        let mem = b.func().add_mem(MemInfo { size: 0, ..info });
1774        let args = b.func().push_values(&[to, from, length]);
1775        let computed =
1776            b.inst(InstData { args, extra: Extra::Mem(mem), ..InstData::new(Opcode::Memcpy) }, &[]);
1777        b.ret(&[]);
1778
1779        let fixed = func.with_mem(fixed, start);
1780        let after = func.mem_out(fixed).expect("a copy makes a new version");
1781        let computed = func.with_mem(computed, after);
1782        (func, fixed, computed)
1783    }
1784
1785    #[test]
1786    fn a_bulk_copy_hands_back_its_length_where_it_has_one_and_nothing_where_the_payload_has_it() {
1787        let (func, fixed, computed) = copies();
1788        let params = &func[func.entry().expect("an entry")].params;
1789        let [to, from, length] = params[..] else { panic!("three of them were appended") };
1790
1791        let bulk = func.bulk(fixed).expect("a memcpy is one");
1792        assert_eq!((bulk.to, bulk.with, bulk.length), (to, from, None));
1793
1794        let bulk = func.bulk(computed).expect("a memcpy is one");
1795        assert_eq!((bulk.to, bulk.with, bulk.length), (to, from, Some(length)));
1796    }
1797
1798    #[test]
1799    fn an_instruction_that_is_not_a_bulk_operation_is_not_taken_apart_as_one() {
1800        let (func, store, load) = threaded();
1801        assert_eq!(func.bulk(store), None);
1802        assert_eq!(func.bulk(load), None);
1803    }
1804}