1 unstable release

0.1.0 Jan 6, 2023

#1726 in Rust patterns

46 downloads per month

MIT license

41KB
849 lines

opcode-macros

This crate provides a opcode_match macro, which generates a complex match statement for opcodes encoded in bit fields.

Format

The basic format is like this:

use opcode_macros::opcode_match;
let opcode = 0u8;
mod namespace {
    pub const A_1: u8 = 0;
    pub const A_2: u8 = 0;
    pub const B_1: u8 = 0;
    pub const B_2: u8 = 0;
}
let result =
opcode_match! {
    opcode as u8 in namespace,
    [[A_1: a1, A_2: a2], [B_1: b1, B_2: b2]] => {
        // Code
        1
    }
    _ => {
        // Code
        0
    }
}
;
assert_eq!(result, 1);

It generates something like this:

let opcode = 0u8;
mod namespace {
    pub const A_1: u8 = 0;
    pub const A_2: u8 = 0;
    pub const B_1: u8 = 0;
    pub const B_2: u8 = 0;
}
const A_1_B_1: u8 = namespace::A_1 | namespace::B_1;
const A_1_B_2: u8 = namespace::A_1 | namespace::B_2;
const A_2_B_1: u8 = namespace::A_2 | namespace::B_1;
const A_2_B_2: u8 = namespace::A_2 | namespace::B_2;
match opcode {
    A_1_B_1 => { /* Code */ }
    A_1_B_2 => { /* Code */ }
    A_2_B_1 => { /* Code */ }
    A_2_B_2 => { /* Code */ }
    _ => { /* Code */ }
}

Match Arm Headers

Match arm headers is something like [[A_1: a1, A_2: a2], [B_1: b1, B_2: b2]].

For example, in eBPF opcodes, BPF_ALU | BPF_K | BPF_ADD is an opcode for 32-bit addition with constants, while BPF_ALU | BPF_K | BPF_SUB is an opcode for 32-bit subtraction with constants. To match against these opcodes, we use the following code:

use opcode_macros::opcode_match;
use ebpf_consts::*;
let opcode = 0x04u8;
let mut result = 0u64;
let dst = 10u64;
let imm = 10u64;
opcode_match! {
    opcode as u8 in ebpf_consts,
    [[BPF_ALU: _], [BPF_K: _],
     [BPF_ADD: add, BPF_SUB: sub]] => {
        result = dst.#"wrapping_{}"2(imm);
    }
    _ => {}
}
assert_eq!(result, 20);

We will talk about the templating rules later.

In the example above, you can also use some other variants:

  • [BPF_ADD: "add", BPF_SUB: "sub"]
  • [BPF_ADD: [add], BPF_SUB: [sub]]
  • [BPF_ADD: ["add"], BPF_SUB: ["sub"]]
  • [BPF_ADD: ["add", "extra1"], BPF_SUB: ["sub", "extra2"]]

If you want to substitutes parts of the code with symbols like "+", you will need to quote the symbols like [BPF_ADD: "+"].

Code Template

Substitution

This is not a real life example.

use opcode_macros::opcode_match;
use ebpf_consts::*;
use core::ops::Add;
let opcode = 0u8;
opcode_match! {
    opcode as u8 in ebpf_consts,
    [
        // Group 0
        [BPF_K:   ["const", 1, 2, "Constant Operation"]],
        // Group 1
        [BPF_ADD: ["add", 3, 4, "Addition Operation"]],
    ] => {
        // Use the first token in group 0 as a string
        assert_eq!(#0, "const");
        // Use the fourth token in group 0 as a string
        assert_eq!(#:3:0, "Constant Operation");
        assert_eq!(#:0:1, "add");
        assert_eq!(#:3:1, "Addition Operation");

        // Use raw tokens
        assert_eq!(#:1:1, "3");
        assert_eq!(#:1:=1, 3);
        // 30.add(40) == 70, where #=1 is just #:0:=1
        let value = 30isize;
        assert_eq!(value.#=1(40), 70);

        // With in-token substitution: add -> wrapping_add
        assert_eq!(value.#"wrapping_{}"1(40), 70);
    }
    _ => panic!(),
}

Conditional Blocks

use opcode_macros::opcode_match;
use ebpf_consts::*;

let opcode = BPF_X | BPF_ALU;
opcode_match! {
    opcode as u8 in ebpf_consts,
    [[BPF_X: x, BPF_K: k], [BPF_ALU: "alu", BPF_ALU64: "alu64"]] => {
        #?((x))
            println!("In V1 branch");
            assert_eq!(#0, "x");                ##

        #?((k))
            println!("In V2 && {} branch", #1);
            assert_eq!(#0, "k");                ##

        #?((!"k"))
            println!("In V2 && {} branch", #1);
            assert_eq!(#0, "x");                ##
        println!("Common");
    }
    _ => panic!(),
};

The grammar is #?((cond1, cond2)|(cond3|cond4)|...) CODE ##, making the code only injected if the opcode matches (cond1 && cond2) || (cond3 && cond4) || .... A condition can be negated with an exclamation mark !cond1.

We don't allow nested conditions.

Dependencies

~1.5MB
~33K SLoC