Rust泛型结构体的关联类型实现：模拟x86寄存器寻址

我希望模拟x86架构中al/ah/ax的字节寻址行为，其中al/ah分别指向16位ax寄存器的低字节和高字节。我希望实现泛化，不仅支持u16/u8类型组合，还能支持u32/u16、u64/u32。

我已经写出了如下Rust代码：

#[derive(Clone, Copy)]
#[repr(C)] // 保证字段顺序
struct Halves<H> {
    lo: H,
    hi: H,
}

#[derive(Clone, Copy)]
union Addressed<T: Copy, H: Copy> {
    whole: T,
    halves: Halves<H>,
}

当前使用时必须写成Addressed<u16, u8>，但u8其实可以由u16推导出来，我希望简化为Addressed<u16>。H类型依赖于T，类似关联类型的概念，请问Rust中有没有机制可以为给定的T绑定特定的H类型？

解决方案 #

可以通过Trait关联类型结合默认泛型参数实现这个需求，具体步骤如下：

1. 定义关联类型Trait #

先定义一个Trait，让目标完整类型T关联对应的半长类型H：

trait HasHalf {
    type Half: Copy;
}

// 为需要的无符号整数类型实现该Trait
impl HasHalf for u16 {
    type Half = u8;
}

impl HasHalf for u32 {
    type Half = u16;
}

impl HasHalf for u64 {
    type Half = u32;
}

impl HasHalf for u128 {
    type Half = u64;
}

2. 修改联合类型的泛型约束 #

调整Addressed的泛型参数，使用默认值绑定Trait的关联类型：

#[derive(Clone, Copy)]
#[repr(C)] // 保证字段顺序与内存布局一致
struct Halves<H> {
    lo: H,
    hi: H,
}

#[derive(Clone, Copy)]
union Addressed<T: Copy + HasHalf, H: Copy = <T as HasHalf>::Half> {
    whole: T,
    halves: Halves<H>,
}

3. 使用示例 #

现在可以直接用Addressed<u16>替代Addressed<u16, u8>，Rust会自动推导对应的半长类型：

fn main() {
    let mut reg = Addressed { whole: 0x1234u16 };
    unsafe {
        println!("lo: {:#x}", reg.halves.lo); // 输出 0x34
        println!("hi: {:#x}", reg.halves.hi); // 输出 0x12
        reg.halves.lo = 0x56;
        println!("whole: {:#x}", reg.whole); // 输出 0x1256
    }
}

注意事项 #

• 访问联合体的不同字段必须放在unsafe块中，这是Rust内存安全规则的要求。
• #[repr(C)]必须保留，确保Halves的字段顺序为低字节在前，匹配x86的小端内存布局。
• 若需支持更多类型，只需为该类型实现HasHalf Trait即可。

内容的提问来源于stack exchange，提问作者twig-froth