asm! 把目标架构汇编嵌入 Rust 生成代码,并通过操作数和 options 告诉编译器它的行为。它不是一堵“禁止所有优化”的墙;优化器仍根据你声明的输入、输出和副作用安排周围代码。
使用前先检查标准库或架构 intrinsic 是否已表达所需操作。直接汇编适合确有目标指令或 ABI 要求的边界,普通算术不必为了性能直觉就改写。
从一个边界清楚的指令开始
以下针对 x86-64,采用 asm! 默认的 Intel 语法:
#[cfg(target_arch = "x86_64")]
fn add_wrapping(mut left: u64, right: u64) -> u64 {
unsafe {
core::arch::asm!(
"add {left}, {right}",
left = inout(reg) left,
right = in(reg) right,
options(pure, nomem, nostack),
);
}
left
}
fn main() {
#[cfg(target_arch = "x86_64")]
{
assert_eq!(add_wrapping(4, 5), 9);
assert_eq!(add_wrapping(u64::MAX, 1), 0);
}
}
函数名明确了模 2^64 回绕语义,不能把它当作开启溢出检查时的普通 Rust + 的完全等价替换。add 会改标志,因此不能附加 preserves_flags。pure 在此表示输出只依赖输入且无其他可观察副作用,允许删除未使用结果等优化。
inout 与 inlateout 差一个关键承诺
| 操作数 | 承诺 |
| in(reg) | 初始输入;汇编结束时须满足未列为输出的寄存器约定 |
| out(reg) | 输出可在输入仍被使用时提前写入,避免与输入不当重叠 |
| inout(reg) | 同一操作数既输入又输出,较保守地约束分配 |
| lateout(reg) | 所有输入已读完后才写,可与输入寄存器复用 |
| inlateout(reg) | 输入输出,并允许满足条件的输入寄存器复用 |
lateout 允许与输入寄存器复用,前提是写入时机满足约束。多指令模板如果先改了输出,再继续读另一个输入,错误使用 lateout/inlateout 可能在优化器复用寄存器后得到错误结果。Rust 内联汇编
reg 由编译器分配,显式 "eax" 等则指定物理寄存器。显式寄存器操作数不能再用模板占位符引用,而应在模板中直接写寄存器名;子寄存器宽度也要与指令和 Rust 类型对应。
多指令的输出不能无条件改成 lateout
#[cfg(target_arch = "x86_64")]
fn sum(left: u64, right: u64) -> u64 {
let output: u64;
unsafe {
core::arch::asm!(
"mov {output}, {left}",
"add {output}, {right}",
output = out(reg) output,
left = in(reg) left,
right = in(reg) right,
options(pure, nomem, nostack),
);
}
output
}
fn main() {
#[cfg(target_arch = "x86_64")]
for left in [0, 1, 7, u64::MAX, 1 << 63] {
for right in [0, 1, 7, u64::MAX, 1 << 63] {
assert_eq!(sum(std::hint::black_box(left), std::hint::black_box(right)),
left.wrapping_add(right));
}
}
}
如果 output 与 right 被分配到同一个寄存器,第一条 mov 就会破坏第二条尚未使用的 right。把输出标为 lateout 会向编译器允许这种复用,不能因为变量名不同或调试构建恰好分配了不同寄存器就认为安全。操作数约束示例
Rust 类型宽度与模板寄存器宽度一起检查
#[cfg(target_arch = "x86_64")]
fn add_u32(mut left: u32, right: u32) -> u32 {
unsafe {
core::arch::asm!(
"add {left:e}, {right:e}",
left = inout(reg) left,
right = in(reg) right,
options(pure, nomem, nostack),
);
}
left
}
fn main() {
#[cfg(target_arch = "x86_64")]
{
assert_eq!(add_u32(10, 20), 30);
assert_eq!(add_u32(u32::MAX, 1), 0);
}
}
:e 在 x86 寄存器模板中指定 32 位名字。不能假定将 u32 放进 reg 后,模板默认打印的更宽寄存器高位就含有某个可依赖值;应让指令宽度与实际输入一致。跨架构的修饰符和寄存器类别不同,需要按目标文档选择。
第二条指令读到的还是原输入吗?
寄存器约束最难的地方是变量名不同不代表物理位置不同。下图让 output 与 right 故意共用,第一条 mov 的写入就变成第二条 add 的错误输入。看到这个反例后,out 所保留的独立位置就有了具体意义。
正在呈现知识画面
第二条指令读到的还是原输入吗?过早写 output 会破坏尚未读取的 right;此模板需要 out,不能随意改为 lateout。
{"id":"rust-asm-register-reuse","title":"第二条指令读到的还是原输入吗?","summary":"过早写 output 会破坏尚未读取的 right;此模板需要 out,不能随意改为 lateout。","height":1000,"html":"<h2 data-i18n=\"heading\"></h2><p class=\"intro\" data-i18n=\"intro\"></p><label class=\"check\"><input type=\"checkbox\" id=\"reuse\"><span data-i18n=\"reuse\"></span></label><section class=\"panel\"><h3 data-i18n=\"trace\"></h3><div id=\"trace\"></div></section><div class=\"metrics\"><div><span data-i18n=\"actual\"></span><strong id=\"actual\"></strong></div><div><span data-i18n=\"expected\"></span><strong id=\"expected\"></strong></div></div><div class=\"copy\"><p id=\"explanation\" role=\"status\"></p><p class=\"reserve\" aria-hidden=\"true\" data-i18n=\"ok\"></p><p class=\"reserve\" aria-hidden=\"true\" data-i18n=\"bad\"></p></div><details class=\"scope\"><summary data-i18n=\"scopeLabel\"></summary><p data-i18n=\"scope\"></p></details>","css":".intro{margin:8px 0 18px;color:var(--muted);font-size:14px}.caption,.hint{font-size:12px;color:var(--muted);margin-top:12px}.control{display:grid;grid-template-columns:minmax(0,1fr) auto;gap:6px 12px;align-items:center;margin:14px 0;font-size:13px}.control label{font-size:13px}.control output{color:var(--accent);min-width:5em;text-align:right;font-variant-numeric:tabular-nums}.control input{grid-column:1/-1;width:100%;margin:0}.presets{display:flex;gap:4px;align-items:stretch}.presets button{flex:1 1 0;min-width:0;overflow-wrap:anywhere;font-size:13px}.actions{display:flex;gap:8px;flex-wrap:wrap;margin:14px 0}.actions button{flex:1;min-width:0;font-size:13px;min-height:42px}.copy{display:grid;font-size:14px;margin:16px 0}.copy>*{grid-area:1/1;margin:0}.reserve{visibility:hidden}.scope{margin-top:14px;color:var(--muted);font-size:12px}.scope summary{padding:8px 0;cursor:pointer}.scope p{margin-top:10px}.two{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:12px;margin:16px 0}.card{border:1px solid var(--rule);border-radius:9px;padding:14px;min-width:0}.card header{font-size:14px;font-weight:600;margin-bottom:10px}.card button{font-size:12px;width:100%;min-width:0}.card small{display:block;font-size:12px;color:var(--muted)}.card code{font-size:12px;overflow-wrap:anywhere}.metrics{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:12px;margin:16px 0}.metrics>div{border-left:2px solid var(--accent);padding-left:10px;min-width:0}.metrics span{display:block;min-height:3.2em;color:var(--muted);font-size:12px}.metrics strong{font-size:23px;display:block;font-weight:550;overflow-wrap:anywhere;font-variant-numeric:tabular-nums}.formula{border-top:1px solid var(--rule);padding-top:12px;margin-top:14px;font-size:13px;font-variant-numeric:tabular-nums;overflow-wrap:anywhere}.value{font:550 20px ui-monospace,monospace}.legend{display:flex;flex-wrap:wrap;gap:6px 14px;font-size:12px;color:var(--muted);margin:12px 0}.legend span:before{content:'';display:inline-block;width:8px;height:8px;border-radius:2px;background:var(--accent);margin-right:6px}.legend .second:before{background:var(--second)}.check{display:flex;align-items:center;gap:8px;font-size:13px;margin:12px 0}.check input{width:16px;height:16px;flex-shrink:0;accent-color:var(--accent)}select,input[type=text],input[type=number]{background:var(--paper);color:var(--ink);font:inherit;font-size:13px;padding:10px;border:1px solid var(--rule);border-radius:7px;max-width:100%;min-width:0}select{width:100%;min-height:42px}.band{height:20px;display:flex;background:var(--surface);border-radius:4px;overflow:hidden}.band i{height:100%;display:block;background:var(--accent);transition:width .25s}.band i.second{background:var(--second)}button:disabled{cursor:default}.status{font-size:13px;min-height:4.8em}.mono{font-family:ui-monospace,monospace;font-variant-numeric:tabular-nums}.muted{color:var(--muted)}\n@media(max-width:480px){.two{gap:8px}.card{padding:12px 9px}.actions button,.presets button{font-size:12px}.metrics strong{font-size:20px}.metrics{gap:8px}}\n@media(prefers-reduced-motion:reduce){*,*::before,*::after{transition:none!important;animation:none!important}}\n.track-row{margin:16px 0}.track-row>span{display:block;font-size:12px;color:var(--muted);margin-bottom:7px}.track{display:flex;gap:5px;min-width:0}.cell{flex:1;min-width:0;min-height:48px;border:1px solid var(--rule);border-radius:5px;background:var(--surface);display:flex;align-items:center;justify-content:center;text-align:center;padding:5px 3px;font-size:12px;overflow-wrap:anywhere;font-variant-numeric:tabular-nums}.cell.on{background:var(--accent-soft);border-color:var(--accent)}.cell.gold{background:var(--second-soft);border-color:var(--second)}.cell.empty{border-style:dashed;color:var(--muted)}.cell.error{border-color:var(--second);text-decoration:line-through}.choices{display:grid;grid-template-columns:repeat(4,minmax(0,1fr));gap:6px;margin:14px 0}.choices button{min-width:0;padding:10px 4px;font-size:12px;overflow-wrap:anywhere}.choices button[aria-current=true]{border-color:var(--accent);background:var(--accent-soft)}.relation{display:grid;grid-template-columns:minmax(0,1fr) 28px minmax(0,1fr);align-items:center;gap:6px;margin:12px 0}.relation .cell{min-height:72px}.relation>i{text-align:center;font-style:normal;color:var(--accent)}.readout{font-size:13px;white-space:pre-line;min-height:6.4em;overflow-wrap:anywhere}.plot{position:relative;border-left:1px solid var(--rule);border-bottom:1px solid var(--rule);padding:8px 0;margin:18px 0}.plot-row{height:28px;position:relative;margin:4px 0;background:var(--surface)}.plot-row i{position:absolute;inset-block:5px;background:var(--accent);min-width:2px;transition:width .2s,left .2s}.plot-row b{position:relative;padding:4px 6px;font:11px/28px ui-monospace,monospace;background:var(--paper);border-radius:3px}.axis{display:flex;justify-content:space-between;font-size:11px;color:var(--muted)}\n\n/* Action labels must remain readable across locales and narrow embeds. */\n.actions{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:8px}\n.actions button{overflow-wrap:anywhere;white-space:normal;min-height:58px;padding:8px}\nbutton{overflow-wrap:anywhere}\n@media(min-width:640px){.actions{grid-template-columns:repeat(3,minmax(0,1fr))}}\n\n.metrics span{overflow-wrap:anywhere}\n.panel{border:1px solid var(--rule);border-radius:9px;padding:12px;margin:14px 0;min-width:0}.panel h3{font-size:13px;margin:0 0 10px;color:var(--muted)}.source{font:13px/1.8 ui-monospace,monospace;white-space:pre-wrap;overflow-wrap:anywhere;margin:12px 0}.source mark{background:var(--accent-soft);color:var(--ink);border-bottom:2px solid var(--accent)}.source .fault{background:var(--second-soft);border-color:var(--second)}.diagram{display:block;width:100%;height:auto;max-height:320px;margin:14px 0;overflow:visible}.diagram text{fill:var(--ink);font:12px ui-monospace,monospace}.diagram .edge{stroke:var(--muted);stroke-width:1.5;fill:none}.diagram .edge.on{stroke:var(--accent);stroke-width:3}.diagram .node{fill:var(--surface);stroke:var(--rule);stroke-width:1.5}.diagram .node.on{fill:var(--accent-soft);stroke:var(--accent)}.diagram .node.gold{fill:var(--second-soft);stroke:var(--second)}.log{font:12px/1.8 ui-monospace,monospace;white-space:pre-line;overflow-wrap:anywhere;min-height:7.2em}.table{display:grid;gap:5px;margin:14px 0}.table .tr{display:grid;grid-template-columns:repeat(3,minmax(0,1fr));gap:5px}.table .tr>*{padding:8px 5px;background:var(--surface);border-radius:4px;min-width:0;overflow-wrap:anywhere;font-size:12px}.table .tr .on{background:var(--accent-soft)}.table .tr .gold{background:var(--second-soft)}.choices button{min-height:44px}.metrics strong{min-height:1.6em}.code-line{display:block}.code-line.on{background:var(--accent-soft);border-left:2px solid var(--accent);padding-left:6px}.status-badge{font-size:12px;color:var(--accent)}.tile-grid{display:grid;grid-template-columns:repeat(4,minmax(0,1fr));gap:6px}.tile-grid .cell{min-height:40px}.actions button{min-height:48px}.form-field{display:grid;gap:8px;margin:14px 0;font-size:13px}.form-field input{width:100%;font-family:ui-monospace,monospace}.metrics.three{grid-template-columns:repeat(3,minmax(0,1fr))}.formula{min-height:3.2em}.presets{flex-wrap:wrap}.presets button{flex-basis:25%;min-height:44px}\n\n.budget{display:flex;height:28px;margin:16px 0;background:var(--surface);border-radius:5px;overflow:hidden}.budget i{display:block;background:var(--accent);transition:width .2s}.budget i.gold{background:var(--second)}.intervals{margin:16px 0}.interval{position:relative;height:36px;background:var(--surface);margin:6px 0;border-radius:4px;overflow:hidden}.interval i{position:absolute;height:100%;background:var(--accent-soft);border-left:2px solid var(--accent)}.interval i.gold{background:var(--second-soft);border-color:var(--second)}.interval span{position:relative;z-index:1;font:12px/36px ui-monospace,monospace;padding-left:7px}.table .tr>*{min-height:4.8em}.formula{white-space:pre-line;min-height:7.2em}.actions:has(>button:only-child){grid-template-columns:1fr}.metrics strong{min-height:3.2em}.rangegrid{display:grid;grid-template-columns:repeat(12,minmax(0,1fr));gap:3px;margin:14px 0}.rangegrid .cell{min-height:30px;padding:3px 1px;font-size:10px}.message{min-height:7.2em;white-space:pre-line;font:13px/1.8 ui-monospace,monospace;overflow-wrap:anywhere}\n/* Allow CJK hanging punctuation without overflowing a metric's border box. */\n.metrics>div{padding-right:4px}.metrics span{line-break:anywhere}\n\n.chart-wrap{margin:18px 0}.chart-label{font-size:12px;color:var(--muted);min-height:3.2em}.curve{display:block;width:100%;height:auto;max-height:320px;overflow:visible}.curve text{font:11px ui-monospace,monospace;fill:var(--muted)}.gridline{stroke:var(--rule);stroke-width:1}.metrics strong{font-size:20px;min-height:3.2em}.formula{font:12px/1.8 ui-monospace,monospace;min-height:9em}.readout{min-height:8em}.presets button{flex-basis:30%}.scope p{overflow-wrap:anywhere} .unit-note{font-size:12px;color:var(--muted);margin:10px 0}\n\n.metrics span{word-break:break-all;line-break:strict}.metrics>div{padding-right:6px}\n\n.cell{min-height:72px}.metrics strong{min-height:3.6em}\n\n.table .tr>*{min-height:4em}","js":"const $=s=>document.querySelector(s);const set=(id,v)=>$('#'+id).textContent=v;const t=k=>viz.t(k);function cells(id,values){$('#'+id).replaceChildren(...values.map(v=>{const e=document.createElement('div');e.className='cell '+(v.cls||'');e.textContent=v.text;e.title=v.title||v.text;return e;}));}function pressed(mode){document.querySelectorAll('[data-mode]').forEach(e=>e.setAttribute('aria-pressed',String(e.dataset.mode===mode)));}const esc=s=>String(s).replace(/[&<>\"']/g,c=>({'&':'&','<':'<','>':'>','\"':'"',\"'\":'''}[c]));const fmt=s=>'{'+[...s].sort().join(', ')+'}';function graph(id,nodes,edges){const el=$('#'+id);el.setAttribute('viewBox','0 0 360 200');el.innerHTML='<defs><marker id=\"arrow-'+id+'\" viewBox=\"0 0 10 10\" refX=\"8\" refY=\"5\" markerWidth=\"5\" markerHeight=\"5\" orient=\"auto-start-reverse\"><path d=\"M 0 0 L 10 5 L 0 10 z\" fill=\"var(--muted)\"/></marker></defs>'+edges.map(e=>{const a=nodes[e[0]],b=nodes[e[1]],dx=b.x-a.x,dy=b.y-a.y,d=Math.hypot(dx,dy)||1;if(edges.some(r=>r[0]===e[1]&&r[1]===e[0])){const nx=-dy/d,ny=dx/d;return '<path class=\"edge '+(e[2]?'on':'')+'\" d=\"M '+(a.x+dx/d*22+nx*12)+' '+(a.y+dy/d*22+ny*12)+' Q '+((a.x+b.x)/2+nx*38)+' '+((a.y+b.y)/2+ny*38)+' '+(b.x-dx/d*25+nx*12)+' '+(b.y-dy/d*25+ny*12)+'\" marker-end=\"url(#arrow-'+id+')\"/>';}return '<line class=\"edge '+(e[2]?'on':'')+'\" x1=\"'+(a.x+dx/d*25)+'\" y1=\"'+(a.y+dy/d*25)+'\" x2=\"'+(b.x-dx/d*28)+'\" y2=\"'+(b.y-dy/d*28)+'\" marker-end=\"url(#arrow-'+id+')\"/>';}).join('')+nodes.map(n=>'<g><circle class=\"node '+(n.cls||'')+'\" cx=\"'+n.x+'\" cy=\"'+n.y+'\" r=\"25\"/><text x=\"'+n.x+'\" y=\"'+(n.y+4)+'\" text-anchor=\"middle\">'+esc(n.name)+'</text>'+(n.sub?'<text x=\"'+n.x+'\" y=\"'+(n.y+43)+'\" text-anchor=\"middle\">'+esc(n.sub)+'</text>':'')+'</g>').join('');}function table(id,rows){$('#'+id).classList.add('table');$('#'+id).replaceChildren(...rows.map(r=>{const row=document.createElement('div');row.className='tr';row.style.gridTemplateColumns='repeat('+r.length+',minmax(0,1fr))';r.forEach(x=>{const c=document.createElement('div');if(typeof x==='object'){c.textContent=x.text;c.className=x.cls||'';}else c.textContent=x;row.append(c)});return row;}));}function band(id,parts,total){$('#'+id).replaceChildren(...parts.map((p,i)=>{const e=document.createElement('i');e.style.width=(100*p.value/total)+'%';e.className=i%2?'gold':'on';e.title=p.name+': '+p.value;return e}));}function intervals(id,items,total){$('#'+id).replaceChildren(...items.map(p=>{const row=document.createElement('div');row.className='interval';const bar=document.createElement('i');bar.style.left=(100*p.start/total)+'%';bar.style.width=(100*(p.end-p.start)/total)+'%';bar.className=p.cls||'';const label=document.createElement('span');label.textContent=p.label;row.append(bar,label);return row;}));}function plot(id,series,xr,yr,opts={}){const e=$('#'+id),X=x=>42+(x-xr[0])/(xr[1]-xr[0])*298,Y=y=>175-(y-yr[0])/(yr[1]-yr[0])*148;let out='<defs><clipPath id=\"clip-'+id+'\"><rect x=\"42\" y=\"27\" width=\"298\" height=\"148\"/></clipPath></defs>';for(let i=0;i<3;i++){let x=xr[0]+i*(xr[1]-xr[0])/2,y=yr[0]+i*(yr[1]-yr[0])/2;out+='<path class=\"gridline\" d=\"M '+X(x)+' 27V175M42 '+Y(y)+'H340\"/><text x=\"'+X(x)+'\" y=\"194\" text-anchor=\"middle\">'+esc(opts.xfmt?opts.xfmt(x):Number(x.toFixed(2)))+'</text><text x=\"36\" y=\"'+(Y(y)+4)+'\" text-anchor=\"end\">'+esc(opts.yfmt?opts.yfmt(y):Number(y.toFixed(2)))+'</text>';}out+='<g clip-path=\"url(#clip-'+id+')\">';for(const s of series){out+='<path fill=\"none\" stroke=\"'+(s.color||'var(--accent)')+'\" stroke-width=\"2.5\" '+(s.dash?'stroke-dasharray=\"5 4\"':'')+' d=\"'+s.pts.map((p,i)=>(i?'L':'M')+X(p[0]).toFixed(2)+','+Y(p[1]).toFixed(2)).join(' ')+'\"/>';}for(const p of opts.points||[])out+='<circle cx=\"'+X(p[0])+'\" cy=\"'+Y(p[1])+'\" r=\"4\" fill=\"var(--second)\" stroke=\"var(--paper)\" stroke-width=\"1.5\"/>';if(opts.cursor!==undefined)out+='<path d=\"M'+X(opts.cursor)+' 27V175\" stroke=\"var(--muted)\" stroke-dasharray=\"3 3\"/>';e.innerHTML=out+'</g>';e.dataset.curves=JSON.stringify(series.map(s=>s.pts));}const curve=(fn,lo,hi,n=120)=>Array.from({length:n+1},(_,i)=>{const x=lo+(hi-lo)*i/n;return[x,fn(x)]});function inputs(draw){document.querySelectorAll('input,select').forEach(e=>{e.addEventListener('input',draw);e.addEventListener('change',draw)});draw()}const n=id=>+$('#'+id).value;const show=(id,v,d=3)=>set(id,Number(v.toFixed(d)));function draw(){let r=$('#reuse').checked;table('trace',[['','output','right'],['start',r?'R1 = 7':'R2 = ?', 'R1 = 7'],['mov output, left',r?'R1 = 4':'R2 = 4',r?'R1 = 4':'R1 = 7'],['add output, right',r?'R1 = 8':'R2 = 11',r?'R1 = 8':'R1 = 7']]);set('actual',r?8:11);set('expected',11);set('explanation',t(r?'bad':'ok'));}inputs(draw);","audio":false,"strings":{"actual":"计算结果","bad":"mov 已把 right 的寄存器改成 4,add 读到 4+4。错误来自声明允许了危险复用。","expected":"预期 left + right","heading":"第二条指令读到的还是原输入吗?","intro":"切换 output 是否与 right 共用寄存器,逐行对照相同模板。","ok":"独立输出寄存器保留 right=7,最终得到 11。","reset":"重置","reuse":"模拟 output 与 right 复用","scope":"抽象寄存器 R0/R1/R2,值 left=4、right=7;模拟一种错误声明允许的分配,不预测 rustc 实际分配。","scopeLabel":"模型范围与约定","trace":"指令与寄存器变化"}}
RDTSC:低位在 EAX,高位在 EDX
下面仅读取计数值,不是完整的微基准计时边界。运行环境须允许用户态读取 TSC:
#[cfg(target_arch = "x86_64")]
fn read_tsc() -> u64 {
let low: u32;
let high: u32;
unsafe {
core::arch::asm!(
"rdtsc",
out("eax") low,
out("edx") high,
options(nomem, nostack, preserves_flags),
);
}
((high as u64) << 32) | low as u64
}
fn main() {
#[cfg(target_arch = "x86_64")]
println!("{}", read_tsc());
}
不能给 RAX 同时声明多个冲突输出,也不能只读取 RAX 就声称已经自动拼好了 64 位结果。RDTSC 随外部计数器变化,不能标记 pure。
RDTSC 不提供完整的指令序列化;要衡量前后代码,需要按处理器手册设计排序、计时器读取和编译器可见性,并评估迁核、虚拟化与频率。TSC tick 也不等于当前核心每个实际执行周期。Intel SDM
AArch64 的 CNTVCT_EL0 是虚拟系统计数器,频率与访问权限有自己的约定,不应笼统称为“核心周期计数器”。跨架构移植需要重建测量定义,而不只是替换一条 mrs。Arm 系统计数器
options 不是性能提示,而是可依赖的声明
nomem 表示不读写汇编可见的外部内存等相应保证,允许编译器更积极地处理周围内存操作;它不是“刷新或不刷新 CPU 缓存”。readonly 与 pure 也各有要求,不能把会写缓冲区或访问设备的指令标成纯读或纯计算。
nostack 承诺不使用规定之外的栈空间,包含目标 ABI 的 red zone 规则;preserves_flags 承诺保留相应架构标志。noreturn 声明不会回到后继 Rust 代码。虚假声明可能造成未定义行为,即使调试构建看起来正常。
普通内存读写、volatile、编译器重排约束和 CPU 内存屏障是不同层次;asm! 默认的保守内存影响也不自动让非原子共享访问变成线程安全。
传入地址,不等于声明了内存读取
#[cfg(target_arch = "x86_64")]
fn load(value: &u64) -> u64 {
let output: u64;
unsafe {
core::arch::asm!(
"mov {output}, qword ptr [{address}]",
address = in(reg) value as *const u64,
output = lateout(reg) output,
options(readonly, nostack, preserves_flags),
);
}
output
}
fn main() {
#[cfg(target_arch = "x86_64")]
{
let values = [0u64, 42, u64::MAX];
for value in &values { assert_eq!(load(value), *value); }
}
}
address 是寄存器输入,[address] 才执行内存读取,所以不能标记 nomem。单条 mov 在写回输出前读取地址操作数,因此这里的 lateout 可以与地址输入复用;与前面两条指令的模板不同。
readonly 表示不写相关外部内存,但仍会读取。该示例的安全入口接收 &u64,用引用已有的保证约束读取范围和生命周期;若换成任意裸地址,封装就必须增加额外调用前提。MMIO、volatile 和线程同步不能由这段普通读取模板替代。
| 实际行为 | 声明时应考虑的内容 |
| 寄存器纯运算 | 输入、输出、标志;满足条件时可 pure + nomem |
| 通过指针读普通内存 | 不能 nomem,读取前提由接口建立 |
| 写缓冲区 | 不能 readonly,需正确独占或同步及有效范围 |
| 读取变化的硬件计数器 | 不能仅因不写内存就标 pure |
| 调用外部函数 | ABI、寄存器破坏、栈、内存和展开约定一起核对 |
在汇编中调用函数还有完整 ABI 义务
clobber_abi("C") 声明指定 ABI 的调用者易失寄存器集合,但不替你放置参数、维持栈对齐、预留 Windows shadow space 或实现展开信息。与它一起使用输出还有显式寄存器等规则,应按文档逐项检查。
因此一行 call 加一个 clobber_abi 并不是可随处复制的安全调用包装。普通函数调用通常应交给 Rust extern 声明;确需手写入口或完整函数时,再考虑 global_asm!/naked_asm! 及对应目标约束。
验证不能只看“能编译”
在目标平台测试边界输入,对纯运算与 Rust 明确语义的实现比较;至少检查优化构建,因为寄存器复用错误可能只在优化后出现。查看最终反汇编以确认指令、寄存器宽度与周围代码安排。
编译器能发现部分冲突或非法操作数,却通常不能证明模板与 clobber/options 一致。Miri 对内联汇编能力有限,不能用未覆盖的执行结果证明汇编安全。条件编译保护架构语法,CPU 指令扩展的运行条件另行处理,参见 cfg。
如何验证而不把微基准写成指令神话
先将计算语义写成安全 Rust 参照实现,测试零值、边界值、相同输入以及溢出,再比较不同优化等级的结果。相同输入尤其可能影响编译器的寄存器共用选择;测试不应只覆盖几组互不相等的小数。
可以用 rustc --edition=2024 -O --emit=asm example.rs 输出优化后的汇编,检查操作数分配与模板周围代码。查看某个汇编片段只是当前目标、版本和优化配置的证据,不能把寄存器名字固定写进语言契约。
测量性能时另需考虑防止无用计算被删除、测量开销、分支分布、缓存和频率。black_box 能帮助避免部分优化,但既不保证精确计时,也不是安全边界;几次 RDTSC 相减不会自动排除这些变量。若标准 Rust 已生成同样指令,手写 asm 只会增加维护与跨平台成本。