内联汇编

inline-assembly.md
commit 28548db57d0acbc00ee80b43816953dbe31d53ba

为了极端底层操作和性能要求,你可能希望直接控制 CPU。Rust 通过asm!宏来支持使用内联汇编。

  1. asm!(assembly template
  2. : output operands
  3. : input operands
  4. : clobbers
  5. : options
  6. );

任何asm的使用需要功能通道(需要在包装箱上加上#![feature(asm)]来允许使用)并且当然也需要写在unsafe块中

注意:这里的例子使用了 x86/x86-64 汇编,不过所有平台都受支持。

汇编模板

assembly template是唯一需要的参数并且必须是原始字符串(就是""

  1. #![feature(asm)]
  2. #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
  3. fn foo() {
  4. unsafe {
  5. asm!("NOP");
  6. }
  7. }
  8. // Other platforms:
  9. #[cfg(not(any(target_arch = "x86", target_arch = "x86_64")))]
  10. fn foo() { /* ... */ }
  11. fn main() {
  12. // ...
  13. foo();
  14. // ...
  15. }

feature(asm)#[cfg]从现在开始将被忽略。)

输出操作数,输入操作数,覆盖和选项都是可选的,然而如果你要省略它们的话,你必选加上正确数量的:

  1. # #![feature(asm)]
  2. # #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
  3. # fn main() { unsafe {
  4. asm!("xor %eax, %eax"
  5. :
  6. :
  7. : "eax"
  8. );
  9. # } }
  10. # #[cfg(not(any(target_arch = "x86", target_arch = "x86_64")))]
  11. # fn main() {}

有空格在中间也没关系:

  1. # #![feature(asm)]
  2. # #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
  3. # fn main() { unsafe {
  4. asm!("xor %eax, %eax" ::: "eax");
  5. # } }
  6. # #[cfg(not(any(target_arch = "x86", target_arch = "x86_64")))]
  7. # fn main() {}

操作数

输入和输出操作数都有相同的格式:: "constraints1"(expr1), "constraints2"(expr2), ..."。输出操作数表达式必须是可变的左值,或还未赋值的:

  1. # #![feature(asm)]
  2. # #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
  3. fn add(a: i32, b: i32) -> i32 {
  4. let c: i32;
  5. unsafe {
  6. asm!("add $2, $0"
  7. : "=r"(c)
  8. : "0"(a), "r"(b)
  9. );
  10. }
  11. c
  12. }
  13. # #[cfg(not(any(target_arch = "x86", target_arch = "x86_64")))]
  14. # fn add(a: i32, b: i32) -> i32 { a + b }
  15. fn main() {
  16. assert_eq!(add(3, 14159), 14162)
  17. }

如果你想在这里使用真正的操作数,然而,要求你在你想使用的寄存器上套上大括号{},并且要求你指明操作数的大小。这在非常底层的编程中是很有用的,这时你使用哪个寄存器是很重要的:

  1. # #![feature(asm)]
  2. # #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
  3. # unsafe fn read_byte_in(port: u16) -> u8 {
  4. let result: u8;
  5. asm!("in %dx, %al" : "={al}"(result) : "{dx}"(port));
  6. result
  7. # }

覆盖(Clobbers)

一些指令修改的寄存器可能保存有不同的值,所以我们使用覆盖列表来告诉编译器不要假设任何装载在这些寄存器的值是有效的。

  1. # #![feature(asm)]
  2. # #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
  3. # fn main() { unsafe {
  4. // Put the value 0x200 in eax:
  5. asm!("mov $$0x200, %eax" : /* no outputs */ : /* no inputs */ : "eax");
  6. # } }
  7. # #[cfg(not(any(target_arch = "x86", target_arch = "x86_64")))]
  8. # fn main() {}

输入和输出寄存器并不需要列出因为这些信息已经通过给出的限制沟通过了。因此,任何其它的被使用的寄存器应该隐式或显式的被列出。

如果汇编修改了代码状态寄存器cc则需要在覆盖中被列出,如果汇编修改了内存,memory也应被指定。

选项(Options)

最后一部分,options是 Rust 特有的。格式是逗号分隔的基本字符串(也就是说,:"foo", "bar", "baz")。它被用来指定关于内联汇编的额外信息:

目前有效的选项有:

  1. volatile - 相当于 gcc/clang 中的__asm__ __volatile__ (...)
  2. alignstack - 特定的指令需要栈按特定方式对齐(比如,SSE)并且指定这个告诉编译器插入通常的栈对齐代码
  3. intel - 使用 intel 语法而不是默认的 AT&T 语法
  1. # #![feature(asm)]
  2. # #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
  3. # fn main() {
  4. let result: i32;
  5. unsafe {
  6. asm!("mov eax, 2" : "={eax}"(result) : : : "intel")
  7. }
  8. println!("eax is currently {}", result);
  9. # }
  10. # #[cfg(not(any(target_arch = "x86", target_arch = "x86_64")))]
  11. # fn main() {}

更多信息

目前asm!的实现是一个LLVM内联汇编表达式的直接绑定,所以请确保充分的阅读他们的文档来获取关于覆盖,限制等概念的更多信息。